ATLAS Offline Software
MuonTruthClassifierFallback.cxx
Go to the documentation of this file.
1 /*
2  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
3 */
4 // Runs on muons without a truth particle link.
5 // Finds the nearest stable truth particle and adds its info to the muon.
7 
10 #include "xAODMuon/MuonContainer.h"
11 
12 namespace {
13  static const SG::AuxElement::Decorator<float> decorator_dR("MCTFallback_dR");
14  static const SG::AuxElement::Decorator<int> decorator_type("MCTFallback_truthType");
15  static const SG::AuxElement::Decorator<int> decorator_origin("MCTFallback_truthOrigin");
16 
17  static const SG::AuxElement::Decorator<float> decorator_pu_dR("MCTFallbackPU_dR");
18  static const SG::AuxElement::Decorator<int> decorator_pu_type("MCTFallbackPU_truthType");
19  static const SG::AuxElement::Decorator<int> decorator_pu_origin("MCTFallbackPU_truthOrigin");
20 
21  static const SG::AuxElement::Accessor<int> acc_tT("truthType");
22  static const SG::AuxElement::Accessor<int> acc_tO("truthOrigin");
23 
24 } // namespace
25 // Constructor
27  const IInterface* p) :
28  base_class(t, n, p) {
29 }
30 
31 // Athena initialize and finalize
33  ATH_MSG_VERBOSE("initialize() ...");
34  ATH_CHECK(m_mcTruthClassifier.retrieve());
35 
36  ATH_CHECK(m_containerKey.initialize());
37  ATH_CHECK(m_truthSGKey.initialize());
38  ATH_CHECK(m_truthPileupSGKey.initialize());
39  ATH_CHECK(m_truthMuonSGKey.initialize());
40 
42  m_Truth_dR_Key = m_containerKey.key() + ".MCTFallback_dR";
43  m_Truth_type_Key = m_containerKey.key() + ".MCTFallback_truthType";
44  m_Truth_origin_Key = m_containerKey.key() + ".MCTFallback_truthOrigin";
45  ATH_CHECK(m_Truth_dR_Key.initialize());
46  ATH_CHECK(m_Truth_type_Key.initialize());
47  ATH_CHECK(m_Truth_origin_Key.initialize());
48 
49  m_Truth_PU_dR_Key = m_containerKey.key() + ".MCTFallbackPU_dR";
50  m_Truth_PU_type_Key = m_containerKey.key() + ".MCTFallbackPU_truthType";
51  m_Truth_PU_origin_Key = m_containerKey.key() + ".MCTFallbackPU_truthOrigin";
52  ATH_CHECK(m_Truth_PU_dR_Key.initialize());
53  ATH_CHECK(m_Truth_PU_type_Key.initialize());
54  ATH_CHECK(m_Truth_PU_origin_Key.initialize());
55 
56  return StatusCode::SUCCESS;
57 }
58 
60  // Retrieve main particle collection
61  const EventContext& ctx = Gaudi::Hive::currentContext();
62 
64  if (!parts.isValid()) {
65  ATH_MSG_ERROR("No Muon collection with name " << m_containerKey.fullKey() << " found in StoreGate!");
66  return StatusCode::FAILURE;
67  }
68  SG::ReadHandle<xAOD::TruthEventContainer> tec{m_truthSGKey, ctx};
69  if (!tec.isValid()) {
70  ATH_MSG_ERROR("No truth collection with name " << m_truthSGKey.fullKey() << " found in StoreGate!");
71  return StatusCode::FAILURE;
72  }
73  SG::ReadHandle<xAOD::TruthPileupEventContainer> tpec{m_truthPileupSGKey, ctx};
74  if (!tpec.isValid()) {
75  ATH_MSG_DEBUG("No truth pileup collection with name " << m_truthPileupSGKey.fullKey() << " found in StoreGate. Pile-up information will not be filled");
76  }
77 
78  SG::ReadHandle<xAOD::TruthParticleContainer> truthMuons{m_truthMuonSGKey, ctx};
79  if (!truthMuons.isValid()) {
80  ATH_MSG_ERROR("No truth muon collection with name " << m_truthMuonSGKey.fullKey() << " found in StoreGate!");
81  return StatusCode::FAILURE;
82  }
83 
84  // Set up the decorator
85 
86  for (const xAOD::IParticle* part : *parts) {
87  const xAOD::TruthParticle* closest = nullptr;
88  float minDR{FLT_MAX};
89  for (const xAOD::TruthParticle* muTruth : *truthMuons) {
90  const float dR = xAOD::P4Helpers::deltaR2(muTruth, part);
91  if (dR > minDR) continue;
92  closest = muTruth;
93  minDR = dR;
94  }
95 
96  for (const xAOD::TruthEvent* event : *tec) {
97  for (size_t parti = 0; parti < event->nTruthParticles(); parti++) {
98  const xAOD::TruthParticle* tpart = event->truthParticle(parti);
99  if (!tpart || !MC::isStable(tpart) || HepMC::is_simulation_particle(tpart) || !tpart->charge() || tpart->isMuon() ||
100  tpart->pt() < m_minPt)
101  continue;
102  const float dR = xAOD::P4Helpers::deltaR2(tpart, part);
103  if (dR > minDR) continue;
104  closest = tpart;
105  minDR = dR;
106  }
107  }
108 
109  decorator_dR(*part) = closest ? std::sqrt(minDR) : -1;
110 
111  int newType{-1}, newOrigin{-1};
112  if (closest && closest->isMuon()) {
113  newType = acc_tT(*closest);
114  newOrigin = acc_tO(*closest);
115  } else if (closest) {
116  auto res = m_mcTruthClassifier->particleTruthClassifier(closest);
117  newType = res.first;
118  newOrigin = res.second;
119  }
120  decorator_type(*part) = newType;
121  decorator_origin(*part) = newOrigin;
122 
123  decorator_pu_dR(*part) = -1;
124  decorator_pu_type(*part) = -1;
125  decorator_pu_origin(*part) = -1;
126 
127  minDR = FLT_MAX;
128 
129  if (tpec.isValid()) {
130  const xAOD::TruthParticle* closestPileup = nullptr;
131  for (auto event : *tpec) {
132  for (size_t parti = 0; parti < event->nTruthParticles(); parti++) {
133  const xAOD::TruthParticle* tpart = event->truthParticle(parti);
134  if (!tpart || !MC::isStable(tpart) || HepMC::is_simulation_particle(tpart) || !tpart->charge() || tpart->isMuon() ||
135  tpart->pt() < m_minPt)
136  continue;
137  const float dR = xAOD::P4Helpers::deltaR2(tpart, part);
138  if (dR > minDR) continue;
139  closestPileup = tpart;
140  minDR = dR;
141  }
142  }
143  decorator_pu_dR(*part) = (closestPileup ? std::sqrt(minDR) : -1);
144  int newPileupType{-1}, newPileupOrigin{-1};
145  if (closestPileup) {
146  auto res = m_mcTruthClassifier->particleTruthClassifier(closestPileup);
147  newPileupType = res.first;
148  newPileupOrigin = res.second;
149  }
150  decorator_pu_type(*part) = newPileupType;
151  decorator_pu_origin(*part) = newPileupOrigin;
152  }
153  }
154 
155  return StatusCode::SUCCESS;
156 }
LArG4FSStartPointFilter.part
part
Definition: LArG4FSStartPointFilter.py:21
SG::Accessor
Helper class to provide type-safe access to aux data.
Definition: Control/AthContainers/AthContainers/Accessor.h:68
SG::ReadHandle
Definition: StoreGate/StoreGate/ReadHandle.h:67
xAODP4Helpers.h
read_hist_ntuple.t
t
Definition: read_hist_ntuple.py:5
ATH_MSG_VERBOSE
#define ATH_MSG_VERBOSE(x)
Definition: AthMsgStreamMacros.h:28
xAOD::P4Helpers::deltaR2
double deltaR2(double rapidity1, double phi1, double rapidity2, double phi2)
from bare rapidity,phi
Definition: xAODP4Helpers.h:111
xAOD::IParticle
Class providing the definition of the 4-vector interface.
Definition: Event/xAOD/xAODBase/xAODBase/IParticle.h:41
python.utils.AtlRunQueryDQUtils.p
p
Definition: AtlRunQueryDQUtils.py:209
ATH_MSG_ERROR
#define ATH_MSG_ERROR(x)
Definition: AthMsgStreamMacros.h:33
DerivationFramework::MuonTruthClassifierFallback::addBranches
virtual StatusCode addBranches() const override
Definition: MuonTruthClassifierFallback.cxx:59
event
POOL::TEvent event(POOL::TEvent::kClassAccess)
SG::Decorator
Helper class to provide type-safe access to aux data.
Definition: Decorator.h:59
HepMC::is_simulation_particle
bool is_simulation_particle(const T &p)
Method to establish if a particle (or barcode) was created during the simulation (TODO update to be s...
Definition: MagicNumbers.h:354
beamspotman.n
n
Definition: beamspotman.py:729
EL::StatusCode
::StatusCode StatusCode
StatusCode definition for legacy code.
Definition: PhysicsAnalysis/D3PDTools/EventLoop/EventLoop/StatusCode.h:22
ATH_MSG_DEBUG
#define ATH_MSG_DEBUG(x)
Definition: AthMsgStreamMacros.h:29
DerivationFramework::MuonTruthClassifierFallback::initialize
StatusCode initialize() override
Definition: MuonTruthClassifierFallback.cxx:32
xAOD::TruthParticle_v1
Class describing a truth particle in the MC record.
Definition: TruthParticle_v1.h:37
res
std::pair< std::vector< unsigned int >, bool > res
Definition: JetGroupProductTest.cxx:11
DerivationFramework::MuonTruthClassifierFallback::MuonTruthClassifierFallback
MuonTruthClassifierFallback(const std::string &t, const std::string &n, const IInterface *p)
Constructor with parameters.
Definition: MuonTruthClassifierFallback.cxx:26
xAOD::TruthEvent_v1
Class describing a signal truth event in the MC record.
Definition: TruthEvent_v1.h:35
ATH_CHECK
#define ATH_CHECK
Definition: AthCheckMacros.h:40
MuonTruthClassifierFallback.h
MuonContainer.h
MC::isStable
bool isStable(const T &p)
Identify if the particle is stable, i.e. has not decayed.
Definition: HepMCHelpers.h:45
xAOD::TruthParticle_v1::isMuon
bool isMuon() const
Whether the particle is a muon (or antimuon)
xAOD::TruthParticle_v1::pt
virtual double pt() const override final
The transverse momentum ( ) of the particle.
Definition: TruthParticle_v1.cxx:161
doL1CaloHVCorrections.parts
parts
Definition: doL1CaloHVCorrections.py:334
xAOD::TruthParticle_v1::charge
double charge() const
Physical charge.
HepMCHelpers.h