ATLAS Offline Software
Loading...
Searching...
No Matches
Muon::RecoToTruthAssociationAlg Class Reference

#include <RecoToTruthAssociationAlg.h>

Inheritance diagram for Muon::RecoToTruthAssociationAlg:
Collaboration diagram for Muon::RecoToTruthAssociationAlg:

Public Member Functions

StatusCode initialize () override
StatusCode execute (const EventContext &ctx) const override
virtual StatusCode sysInitialize () override
 Override sysInitialize.
virtual bool isClonable () const override
 Specify if the algorithm is clonable.
virtual unsigned int cardinality () const override
 Cardinality (Maximum number of clones that can exist) special value 0 means that algorithm is reentrant.
virtual StatusCode sysExecute (const EventContext &ctx) override
 Execute an algorithm.
virtual const DataObjIDColl & extraOutputDeps () const override
 Return the list of extra output dependencies.
virtual bool filterPassed (const EventContext &ctx) const
virtual void setFilterPassed (bool state, const EventContext &ctx) const
ServiceHandle< StoreGateSvc > & evtStore ()
 The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.
const ServiceHandle< StoreGateSvc > & detStore () const
 The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.
virtual StatusCode sysStart () override
 Handle START transition.
virtual std::vector< Gaudi::DataHandle * > inputHandles () const override
 Return this algorithm's input handles.
virtual std::vector< Gaudi::DataHandle * > outputHandles () const override
 Return this algorithm's output handles.
Gaudi::Details::PropertyBase & declareProperty (Gaudi::Property< T, V, H > &t)
void updateVHKA (Gaudi::Details::PropertyBase &)
MsgStream & msg () const
bool msgLvl (const MSG::Level lvl) const

Protected Member Functions

void renounceArray (SG::VarHandleKeyArray &handlesArray)
 remove all handles from I/O resolution
std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > renounce (T &h)
void extraDeps_update_handler (Gaudi::Details::PropertyBase &ExtraDeps)
 Add StoreName to extra input/output deps as needed.

Private Types

typedef ServiceHandle< StoreGateSvcStoreGateSvc_t

Private Member Functions

void count_chamber_layers (const xAOD::IParticle *truth_particle, const Trk::Track *ptrk, std::vector< unsigned int > &nprecHitsPerChamberLayer, std::vector< unsigned int > &nphiHitsPerChamberLayer, std::vector< unsigned int > &ntrigEtaHitsPerChamberLayer) const
void clear_dummys (const std::vector< unsigned long long > &identifiers, std::vector< unsigned int > &vec) const
Gaudi::Details::PropertyBase & declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
 specialization for handling Gaudi::Property<SG::VarHandleKey>

Private Attributes

SG::ReadHandleKey< xAOD::TruthParticleContainerm_truthMuKey
 Key to the filtered muon truth particles.
SG::WriteDecorHandleKey< xAOD::TruthParticleContainerm_muonTruthRecoLink
 Decorations for the filtered muon truth particles.
SG::ReadHandleKey< xAOD::MuonContainerm_recoMuKey
SG::WriteDecorHandleKey< xAOD::MuonContainerm_muonTruthParticleLink
 Decorations for the reconstructed muon particles.
SG::WriteDecorHandleKey< xAOD::MuonContainerm_muonTruthParticleType
SG::WriteDecorHandleKey< xAOD::MuonContainerm_muonTruthParticleOrigin
SG::WriteDecorHandleKey< xAOD::MuonContainerm_muonTruthParticleClassification
SG::WriteDecorHandleKey< xAOD::MuonContainerm_muonTruthParticleNPrecMatched
SG::WriteDecorHandleKey< xAOD::MuonContainerm_muonTruthParticleNPhiMatched
SG::WriteDecorHandleKey< xAOD::MuonContainerm_muonTruthParticleNTrigEtaMatched
SG::ReadDecorHandleKeyArray< xAOD::IParticleContainerm_inputDecorKey
Gaudi::Property< std::vector< std::string > > m_assocTrkContainers
Gaudi::Property< std::string > m_recoLink {this, "RecoLinkName", "recoMuonLink" , "Decoration to the truth particle pointing to the muon"}
Gaudi::Property< bool > m_associateWithInDetTP {this, "AssociateWithInDetTP", false, "force use of ID track particles for association"}
ServiceHandle< Muon::IMuonIdHelperSvcm_idHelperSvc {this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"}
DataObjIDColl m_extendedExtraObjects
 Extra output dependency collection, extended by AthAlgorithmDHUpdate to add symlinks.
StoreGateSvc_t m_evtStore
 Pointer to StoreGate (event store by default)
StoreGateSvc_t m_detStore
 Pointer to StoreGate (detector store by default)
std::vector< SG::VarHandleKeyArray * > m_vhka
bool m_varHandleArraysDeclared

Detailed Description

Definition at line 18 of file RecoToTruthAssociationAlg.h.

Member Typedef Documentation

◆ StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::StoreGateSvc_t
privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Member Function Documentation

◆ cardinality()

unsigned int AthCommonReentrantAlgorithm< Gaudi::Algorithm >::cardinality ( ) const
overridevirtualinherited

Cardinality (Maximum number of clones that can exist) special value 0 means that algorithm is reentrant.

Override this to return 0 for reentrant algorithms.

Definition at line 75 of file AthCommonReentrantAlgorithm.cxx.

64{
65 return 0;
66}

◆ clear_dummys()

void Muon::RecoToTruthAssociationAlg::clear_dummys ( const std::vector< unsigned long long > & identifiers,
std::vector< unsigned int > & vec ) const
private

If the identifiers are empty then there is no change that a dummy value could be cleared from this list

Definition at line 372 of file RecoToTruthAssociationAlg.cxx.

372 {
375 if (identifiers.empty()) { return; }
376 for (unsigned int i = 0; i < vec.size(); ++i) {
377 if (vec[i] != dummy_unsigned) continue;
378 for (unsigned j = 0; j < identifiers.size(); ++j) {
379 const Identifier id{identifiers[j]};
380 if (m_idHelperSvc->measuresPhi(id)) {
381 const auto phiIdx = static_cast<PhiIndex>(i);
382 if (m_idHelperSvc->phiIndex(id) == phiIdx) {
383 vec[i] = 0;
384 break;
385 }
386 } else {
387 const auto chIdx = static_cast<ChIndex>(i);
388 if (m_idHelperSvc->chamberIndex(id) == chIdx) {
389 vec[i] = 0;
390 break;
391 }
392 }
393 }
394 }
395}
std::vector< size_t > vec
ServiceHandle< Muon::IMuonIdHelperSvc > m_idHelperSvc
std::vector< Identifier > identifiers(const InDetSimDataCollection &coll)
PhiIndex
enum to classify the different phi layers in the muon spectrometer
ChIndex
enum to classify the different chamber layers in the muon spectrometer

◆ count_chamber_layers()

void Muon::RecoToTruthAssociationAlg::count_chamber_layers ( const xAOD::IParticle * truth_particle,
const Trk::Track * ptrk,
std::vector< unsigned int > & nprecHitsPerChamberLayer,
std::vector< unsigned int > & nphiHitsPerChamberLayer,
std::vector< unsigned int > & ntrigEtaHitsPerChamberLayer ) const
private

Definition at line 272 of file RecoToTruthAssociationAlg.cxx.

275 {
276
277 if (!truthParticle || !truthMdtHitsAcc.isAvailable(*truthParticle)) {
278 ATH_MSG_DEBUG("muon has no truth hits vector in the truth association alg");
279 nprecHitsPerChamberLayer.clear();
280 nphiHitsPerChamberLayer.clear();
281 ntrigEtaHitsPerChamberLayer.clear();
282 return;
283 }
284 const std::vector<unsigned long long>& mdtTruth = truthMdtHitsAcc(*truthParticle);
285 std::vector<unsigned long long> cscTruth;
286
287 if (m_idHelperSvc->hasCSC()) cscTruth = truthCscHitsAcc(*truthParticle);
288 const std::vector<unsigned long long>& rpcTruth = truthRpcHitsAcc(*truthParticle);
289 const std::vector<unsigned long long>& tgcTruth = truthTgcHitsAcc(*truthParticle);
290
291 for (const Trk::TrackStateOnSurface* tsit : *ptrk->trackStateOnSurfaces()) {
292 if (!tsit || !tsit->trackParameters() || !tsit->measurementOnTrack()) continue;
293 const Trk::MeasurementBase* meas = tsit->measurementOnTrack();
294 Identifier id;
295 const Trk::RIO_OnTrack* rot = dynamic_cast<const Trk::RIO_OnTrack*>(meas);
296 if (rot)
297 id = rot->identify();
298 else {
299 const Muon::CompetingMuonClustersOnTrack* crot = dynamic_cast<const Muon::CompetingMuonClustersOnTrack*>(meas);
300 if (crot && !crot->containedROTs().empty() && crot->containedROTs().front()) id = crot->containedROTs().front()->identify();
301 }
302 if (!m_idHelperSvc->isMuon(id)) continue;
303
304 bool measPhi = m_idHelperSvc->measuresPhi(id);
305 bool isTgc = m_idHelperSvc->isTgc(id);
306 ChIndex chIndex = !isTgc ? m_idHelperSvc->chamberIndex(id) : ChIndex::ChUnknown;
307 if (m_idHelperSvc->isMdt(id)) {
308 for (unsigned int i = 0; i < mdtTruth.size(); ++i) {
309 if (id == mdtTruth[i]) {
310 if (chIndex != ChIndex::ChUnknown) {
311 increment_unsigned(nprecHitsPerChamberLayer.at(toInt(chIndex)));
312 }
313 break;
314 }
315 }
316 } else if (m_idHelperSvc->hasCSC() && m_idHelperSvc->isCsc(id)) {
317 for (unsigned int i = 0; i < cscTruth.size(); ++i) {
318 if (id != cscTruth[i]) continue;
319 if (measPhi) {
320 PhiIndex index = m_idHelperSvc->phiIndex(id);
321 if (index != PhiIndex::PhiUnknown) {
322 increment_unsigned(nphiHitsPerChamberLayer.at(toInt(index)));
323 }
324 } else {
325 if (chIndex != ChIndex::ChUnknown) {
326 increment_unsigned(nprecHitsPerChamberLayer.at(toInt(chIndex)));
327 }
328 }
329 break;
330 }
331 } else if (m_idHelperSvc->isRpc(id)) {
332 for (unsigned int i = 0; i < rpcTruth.size(); ++i) {
333 if (id != rpcTruth[i]) { continue; }
334 PhiIndex index = m_idHelperSvc->phiIndex(id);
335 if (index != PhiIndex::PhiUnknown) {
336 if (measPhi) {
337 increment_unsigned(nphiHitsPerChamberLayer.at(toInt(index)));
338 } else {
339 increment_unsigned(ntrigEtaHitsPerChamberLayer.at(toInt(index)));
340 }
341 }
342 break;
343 }
344 } else if (m_idHelperSvc->isTgc(id)) {
345 for (unsigned int i = 0; i < tgcTruth.size(); ++i) {
346 if (id != tgcTruth[i]) { continue; }
347 PhiIndex index = m_idHelperSvc->phiIndex(id);
348 if (index != PhiIndex::PhiUnknown) {
349 if (measPhi) {
350 increment_unsigned(nphiHitsPerChamberLayer.at(toInt(index)));
351 } else {
352 increment_unsigned(ntrigEtaHitsPerChamberLayer.at(toInt(index)));
353 }
354 }
355 break;
356 }
357 }
358 } // end loop over TSOS
359 ATH_MSG_DEBUG("finished loop over TSOS");
360
361 // now, have to check if there are non-zero truth hits in indices without reco hits
362 clear_dummys(mdtTruth, nprecHitsPerChamberLayer);
363 clear_dummys(cscTruth, nprecHitsPerChamberLayer);
364
365 clear_dummys(cscTruth, nphiHitsPerChamberLayer);
366 clear_dummys(rpcTruth, nphiHitsPerChamberLayer);
367 clear_dummys(tgcTruth, nphiHitsPerChamberLayer);
368
369 clear_dummys(rpcTruth, ntrigEtaHitsPerChamberLayer);
370 clear_dummys(tgcTruth, ntrigEtaHitsPerChamberLayer);
371}
#define ATH_MSG_DEBUG(x)
const std::vector< const MuonClusterOnTrack * > & containedROTs() const
returns the vector of SCT_ClusterOnTrack objects .
void clear_dummys(const std::vector< unsigned long long > &identifiers, std::vector< unsigned int > &vec) const
Identifier identify() const
return the identifier -extends MeasurementBase
const Trk::TrackStates * trackStateOnSurfaces() const
return a pointer to a const DataVector of const TrackStateOnSurfaces.
str index
Definition DeMoScan.py:362
ChIndex chIndex(const std::string &index)
convert ChIndex name string to enum
constexpr int toInt(const EnumType enumVal)

◆ declareGaudiProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareGaudiProperty ( Gaudi::Property< T, V, H > & hndl,
const SG::VarHandleKeyType &  )
inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleKey>

Definition at line 156 of file AthCommonDataStore.h.

158 {
160 hndl.value(),
161 hndl.documentation());
162
163 }
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T, V, H > &t)

◆ declareProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareProperty ( Gaudi::Property< T, V, H > & t)
inlineinherited

Definition at line 145 of file AthCommonDataStore.h.

145 {
146 typedef typename SG::HandleClassifier<T>::type htype;
148 }
Gaudi::Details::PropertyBase & declareGaudiProperty(Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
specialization for handling Gaudi::Property<SG::VarHandleKey>

◆ detStore()

const ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::detStore ( ) const
inlineinherited

The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 95 of file AthCommonDataStore.h.

◆ evtStore()

ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::evtStore ( )
inlineinherited

The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 85 of file AthCommonDataStore.h.

◆ execute()

StatusCode Muon::RecoToTruthAssociationAlg::execute ( const EventContext & ctx) const
override

Link the truth particle to the muon

Check first if the truth link already exists

May be both muons are reconstructed by the same author but one is commissioning

The last judge is a simple dR cut but this will hopefully never trigger

Zero supression do not want to store meaningless zeros

Decorate the results

Patch for STACO muons: Copy the truth information from the muon back to the combined track to avoid file corruptions reported in ATLASRECTS-6454

one more thing: need to have muonlink set for all truth particles to avoid ELReset errors

Definition at line 92 of file RecoToTruthAssociationAlg.cxx.

92 {
93 const xAOD::TruthParticleContainer* muonTruthContainer{nullptr};
94 ATH_CHECK(SG::get(muonTruthContainer, m_truthMuKey , ctx));
95 std::unique_ptr<SG::WriteDecorHandle<xAOD::TruthParticleContainer, ElementLink<xAOD::MuonContainer>>> muonTruthParticleRecoLink{};
96 if (!m_muonTruthRecoLink.empty()) {
97 muonTruthParticleRecoLink = std::make_unique<SG::WriteDecorHandle<xAOD::TruthParticleContainer, ElementLink<xAOD::MuonContainer>>>(m_muonTruthRecoLink, ctx);
98 }
99
100 SG::WriteDecorHandle<xAOD::MuonContainer, ElementLink<xAOD::TruthParticleContainer> > muonTruthParticleLink(m_muonTruthParticleLink,
101 ctx);
102 if (!muonTruthParticleLink.isValid()) {
103 ATH_MSG_WARNING("muon particle container not valid");
104 return StatusCode::FAILURE;
105 }
106 SG::WriteDecorHandle<xAOD::MuonContainer, int> muonTruthParticleType(m_muonTruthParticleType, ctx);
107 SG::WriteDecorHandle<xAOD::MuonContainer, int> muonTruthParticleOrigin(m_muonTruthParticleOrigin, ctx);
108 SG::WriteDecorHandle<xAOD::MuonContainer, unsigned int> muonTruthParticleClassification(m_muonTruthParticleClassification, ctx);
109 SG::WriteDecorHandle<xAOD::MuonContainer, std::vector<unsigned int> > muonTruthParticleNPrecMatched(m_muonTruthParticleNPrecMatched,
110 ctx);
111 SG::WriteDecorHandle<xAOD::MuonContainer, std::vector<unsigned int> > muonTruthParticleNPhiMatched(m_muonTruthParticleNPhiMatched, ctx);
112 SG::WriteDecorHandle<xAOD::MuonContainer, std::vector<unsigned int> > muonTruthParticleNTrigEtaMatched(m_muonTruthParticleNTrigEtaMatched, ctx);
113
114 // add link to reco muons and viceversa
115 bool saw_staco = false;
116 bool decor_staco = false;
117
118 // loop over muons
119 for (const xAOD::Muon* muon : *muonTruthParticleLink) {
120 // use primary track particle to get the truth link (except for the case of STACO, where we must use the ID track particle, as the
121 // combined is not truth-matched)
122 ATH_MSG_DEBUG("muon with pT " << muon->pt() << " MeV, eta: " << muon->eta() << ", phi " << muon->phi() << " and author "
123 << muon->author());
124 const xAOD::TrackParticle* tp = nullptr;
125 if (m_associateWithInDetTP || muon->author() == xAOD::Muon::STACO || muon->author() == xAOD::Muon::MuGirl) {
126 tp = muon->trackParticle(xAOD::Muon::InnerDetectorTrackParticle);
127 } else {
128 tp = muon->primaryTrackParticle();
129 }
130
131 bool foundTruth{false}, setOrigin{false};
132 if (tp) {
133 // Associate reco with truth muon. Loop over reconstructed muons, get track particle for each one.
134 // Each track particle should carry a link to the corresponding truth particle. Then compare this truth particle link with the
135 // given truth muon particle
136 if (acc_origin.isAvailable(*tp) && acc_origin(*tp) != 0) {
137 muonTruthParticleOrigin(*muon) = acc_origin(*tp);
138 muonTruthParticleType(*muon) = acc_type(*tp);
139 muonTruthParticleClassification(*muon) = acc_classification(*tp);
140 setOrigin = true;
141 }
142
143 ElementLink<xAOD::TruthParticleContainer> truthLink;
144 if (acc_link.isAvailable(*tp)) {
145 truthLink = acc_link(*tp);
146 } else {
147 ATH_MSG_DEBUG("Could not find any truth link associated with track having pt:"<<tp->pt()<<" MeV, eta: "<<tp->eta()<<", phi: "<<tp->phi()<<", charge: "<<tp->charge()<<". d0:"<<tp->d0()<<", z0: "<<tp->z0());
148 }
149
150 if (truthLink.isValid()) {
151 ATH_MSG_VERBOSE(" Got valid truth link for muon author " << muon->author() << " uniqueID " << HepMC::uniqueID(*truthLink));
152 // loop over truth particles
153 for (const xAOD::TruthParticle* truthParticle : *muonTruthContainer) {
154 if (!MC::isStable(truthParticle)) continue;
155 ATH_MSG_DEBUG("Got truth muon with uniqueID " << HepMC::uniqueID(truthParticle) << " pt " << truthParticle->pt());
156 if ( !HepMC::is_sim_descendant(*truthLink, truthParticle)) {
157 ATH_MSG_VERBOSE("UniqueID truth link: " << HepMC::uniqueID(*truthLink)
158 << " is not decendant of " << HepMC::uniqueID(truthParticle));
159 continue;
160 }
161 ATH_MSG_VERBOSE("Truth muon uniqueID matches -> creating link with truth particle " << HepMC::uniqueID(*truthLink));
162 foundTruth = true;
164 ElementLink<xAOD::TruthParticleContainer> muonTruthLink{*muonTruthContainer,
165 truthParticle->index(),
166 ctx};
167 muonTruthLink.toPersistent();
168 muonTruthParticleLink(*muon) = muonTruthLink;
169 if (!setOrigin) {
170 muonTruthParticleOrigin(*muon) = acc_origin(*tp);
171 muonTruthParticleType(*muon) = acc_type(*tp);
172 muonTruthParticleClassification(*muon) = acc_classification(*tp);
173 setOrigin = true;
174 }
176 if (muonTruthParticleRecoLink && muonTruthParticleRecoLink->operator()(*truthParticle).isValid()){
177 const xAOD::Muon* decor_muon = *muonTruthParticleRecoLink->operator()(*truthParticle);
178 ATH_MSG_VERBOSE("Truth particle is already decorated with reco muon "<<decor_muon->pt()*1.e-3
179 <<" eta: "<<decor_muon->eta()<<" phi: "<<decor_muon->phi()<<" charge: "<<
180 decor_muon->charge()<<" author: "<<decor_muon->author()<<" all authors: "<<
181 decor_muon->allAuthors());
182
183 // Check first if the exiting muon has a better author
184 if (MuonCombined::authorRank(decor_muon->author()) < MuonCombined::authorRank(muon->author())){
185 ATH_MSG_DEBUG("Author of the decorated muon is better than the one of the new candidate");
186 continue;
187 }
189 const int com_score = (muon->allAuthors() & com_bit) - (decor_muon->allAuthors() &com_bit);
190 if (com_score > 0){
191 ATH_MSG_DEBUG("Found two muons reconstructed by an equivalent author. But this one is from the commissioning chain");
192 continue;
193 }
195 if (deltaR2(muon,truthParticle) >= deltaR2(muon, decor_muon)) continue;
196 }
197
198
199 ElementLink<xAOD::MuonContainer> muonLink{muon, *muonTruthParticleLink, ctx};
200
202 std::vector<unsigned int> nprecHitsPerChamberLayer(toInt(ChIndex::ChIndexMax), dummy_unsigned);
203 std::vector<unsigned int> nphiHitsPerChamberLayer(toInt(PhiIndex::PhiIndexMax), dummy_unsigned);
204 std::vector<unsigned int> ntrigEtaHitsPerChamberLayer(toInt(PhiIndex::PhiIndexMax), dummy_unsigned);
205
206 constexpr int author_sel = (1<<xAOD::Muon::MuidCo) | (1<<xAOD::Muon::MuidSA) | (1<<xAOD::Muon::MuGirl);
207 count_chamber_layers(muon->allAuthors() & author_sel
208 ? truthParticle
209 : nullptr,
210 tp->track(), nprecHitsPerChamberLayer, nphiHitsPerChamberLayer, ntrigEtaHitsPerChamberLayer);
212 muonTruthParticleNPrecMatched(*muon) = nprecHitsPerChamberLayer;
213 muonTruthParticleNPhiMatched(*muon) = nphiHitsPerChamberLayer;
214 muonTruthParticleNTrigEtaMatched(*muon) = ntrigEtaHitsPerChamberLayer;
215
216 if (muonTruthParticleRecoLink) (*muonTruthParticleRecoLink)(*truthParticle) = muonLink;
217 break;
218 }
219 } else {
220 ATH_MSG_DEBUG("Invalid truth link");
221 }
222 } else {
223 ATH_MSG_WARNING("Could not find the appropiate track particle for muon with pT: " << muon->pt() * 1.e-3 << " GeV, eta: "
224 << muon->eta() << ", phi: " << muon->phi()
225 << " author: " << muon->author());
226 }
227
228 if (!setOrigin) {
229 muonTruthParticleOrigin(*muon) = 0;
230 muonTruthParticleType(*muon) = 0;
231 muonTruthParticleClassification(*muon) = 0;
232 }
233 if (!foundTruth) {
234 muonTruthParticleLink(*muon) = ElementLink<xAOD::TruthParticleContainer>();
235 // add these empty vectors
236 muonTruthParticleNPrecMatched(*muon) = std::vector<unsigned int>{};
237 muonTruthParticleNPhiMatched(*muon) = std::vector<unsigned int>{};
238 muonTruthParticleNTrigEtaMatched(*muon) = std::vector<unsigned int>{};
239 }
242 if (muon->author() == xAOD::Muon::STACO) {
243 const xAOD::TrackParticle* cmb_trk = muon->trackParticle(xAOD::Muon::CombinedTrackParticle);
244 if (!cmb_trk){
245 ATH_MSG_WARNING("Even a STACO muon should have a combined track");
246 continue;
247 } else {
248 if (!saw_staco) {
249 saw_staco = true;
250 decor_staco = !dec_origin.isAvailable (*cmb_trk);
251 }
252 if (decor_staco) {
253 dec_origin(*cmb_trk) = acc_origin(*muon);
254 dec_classification(*cmb_trk) = acc_classification(*muon);
255 dec_type(*cmb_trk) = acc_type(*muon);
256 dec_link(*cmb_trk) = acc_link(*muon);
257 }
258 }
259 }
260 }
262 if (muonTruthParticleRecoLink && !muonTruthParticleRecoLink->isAvailable()) {
263 for (const xAOD::TruthParticle* truthParticle : **muonTruthParticleRecoLink) {
264 ATH_MSG_DEBUG("no reco muon link set, add an empty one");
265 (*muonTruthParticleRecoLink)(*truthParticle) = ElementLink<xAOD::MuonContainer>();
266 }
267 }
268
269 return StatusCode::SUCCESS;
270}
#define ATH_CHECK
Evaluate an expression and check for errors.
#define ATH_MSG_VERBOSE(x)
#define ATH_MSG_WARNING(x)
SG::WriteDecorHandleKey< xAOD::MuonContainer > m_muonTruthParticleOrigin
SG::WriteDecorHandleKey< xAOD::MuonContainer > m_muonTruthParticleNPhiMatched
Gaudi::Property< bool > m_associateWithInDetTP
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer > m_muonTruthRecoLink
Decorations for the filtered muon truth particles.
SG::WriteDecorHandleKey< xAOD::MuonContainer > m_muonTruthParticleClassification
SG::WriteDecorHandleKey< xAOD::MuonContainer > m_muonTruthParticleType
SG::WriteDecorHandleKey< xAOD::MuonContainer > m_muonTruthParticleLink
Decorations for the reconstructed muon particles.
SG::WriteDecorHandleKey< xAOD::MuonContainer > m_muonTruthParticleNTrigEtaMatched
void count_chamber_layers(const xAOD::IParticle *truth_particle, const Trk::Track *ptrk, std::vector< unsigned int > &nprecHitsPerChamberLayer, std::vector< unsigned int > &nphiHitsPerChamberLayer, std::vector< unsigned int > &ntrigEtaHitsPerChamberLayer) const
SG::ReadHandleKey< xAOD::TruthParticleContainer > m_truthMuKey
Key to the filtered muon truth particles.
SG::WriteDecorHandleKey< xAOD::MuonContainer > m_muonTruthParticleNPrecMatched
virtual double eta() const
The pseudorapidity ( ) of the particle.
virtual double phi() const
The azimuthal angle ( ) of the particle.
virtual double pt() const
The transverse momentum ( ) of the particle.
float charge() const
uint16_t allAuthors() const
Get all the authors of this Muon.
Author author() const
int uniqueID(const T &p)
bool is_sim_descendant(const T1 &p1, const T2 &p2)
Method to check if the first particle is a descendant of the second in the simulation,...
bool isStable(const T &p)
Identify if the particle is stable, i.e. has not decayed.
int authorRank(const xAOD::Muon::Author &a)
Definition TagBase.h:23
const T * get(const ReadCondHandleKey< T > &key, const EventContext &ctx)
Convenience function to retrieve an object given a ReadCondHandleKey.
double deltaR2(double rapidity1, double phi1, double rapidity2, double phi2)
from bare rapidity,phi
TrackParticle_v1 TrackParticle
Reference the current persistent version:
TruthParticle_v1 TruthParticle
Typedef to implementation.
Muon_v1 Muon
Reference the current persistent version:
TruthParticleContainer_v1 TruthParticleContainer
Declare the latest version of the truth particle container.

◆ extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::extraDeps_update_handler ( Gaudi::Details::PropertyBase & ExtraDeps)
protectedinherited

Add StoreName to extra input/output deps as needed.

use the logic of the VarHandleKey to parse the DataObjID keys supplied via the ExtraInputs and ExtraOuputs Properties to add the StoreName if it's not explicitly given

◆ extraOutputDeps()

const DataObjIDColl & AthCommonReentrantAlgorithm< Gaudi::Algorithm >::extraOutputDeps ( ) const
overridevirtualinherited

Return the list of extra output dependencies.

This list is extended to include symlinks implied by inheritance relations.

Definition at line 94 of file AthCommonReentrantAlgorithm.cxx.

90{
91 // If we didn't find any symlinks to add, just return the collection
92 // from the base class. Otherwise, return the extended collection.
93 if (!m_extendedExtraObjects.empty()) {
95 }
97}
An algorithm that can be simultaneously executed in multiple threads.

◆ filterPassed()

virtual bool AthCommonReentrantAlgorithm< Gaudi::Algorithm >::filterPassed ( const EventContext & ctx) const
inlinevirtualinherited

Definition at line 96 of file AthCommonReentrantAlgorithm.h.

96 {
97 return execState( ctx ).filterPassed();
98 }
virtual bool filterPassed(const EventContext &ctx) const

◆ initialize()

StatusCode Muon::RecoToTruthAssociationAlg::initialize ( )
override

Definition at line 49 of file RecoToTruthAssociationAlg.cxx.

49 {
50 ATH_CHECK(m_idHelperSvc.retrieve());
51 ATH_CHECK(m_truthMuKey.initialize());
52 ATH_CHECK(m_recoMuKey.initialize());
53
54 if (m_recoLink.empty()){
56 } else {
58 }
59
60 ATH_CHECK(m_muonTruthRecoLink.initialize(!m_recoLink.empty()));
68 for (const std::string& trk_coll : m_assocTrkContainers.value()){
69 m_inputDecorKey.emplace_back(trk_coll + ".truthParticleLink");
70 }
71
72 m_inputDecorKey.emplace_back(m_truthMuKey, SG::AuxTypeRegistry::instance().getName(acc_origin.auxid()));
73 m_inputDecorKey.emplace_back(m_truthMuKey, SG::AuxTypeRegistry::instance().getName(acc_type.auxid()));
74 if (m_idHelperSvc->hasMDT()) {
75 m_inputDecorKey.emplace_back(m_truthMuKey, SG::AuxTypeRegistry::instance().getName(truthMdtHitsAcc.auxid()));
76 }
77 if (m_idHelperSvc->hasRPC()) {
78 m_inputDecorKey.emplace_back(m_truthMuKey, SG::AuxTypeRegistry::instance().getName(truthRpcHitsAcc.auxid()));
79 }
80 if (m_idHelperSvc->hasTGC()) {
81 m_inputDecorKey.emplace_back(m_truthMuKey, SG::AuxTypeRegistry::instance().getName(truthTgcHitsAcc.auxid()));
82 }
83 if (m_idHelperSvc->hasCSC()) {
84 m_inputDecorKey.emplace_back(m_truthMuKey, SG::AuxTypeRegistry::instance().getName(truthCscHitsAcc.auxid()));
85 }
86
87 ATH_CHECK(m_inputDecorKey.initialize());
88 return StatusCode::SUCCESS;
89}
SG::ReadHandleKey< xAOD::MuonContainer > m_recoMuKey
SG::ReadDecorHandleKeyArray< xAOD::IParticleContainer > m_inputDecorKey
Gaudi::Property< std::vector< std::string > > m_assocTrkContainers
Gaudi::Property< std::string > m_recoLink
static AuxTypeRegistry & instance()
Return the singleton registry instance.

◆ inputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::inputHandles ( ) const
overridevirtualinherited

Return this algorithm's input handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ isClonable()

◆ msg()

MsgStream & AthCommonMsg< Gaudi::Algorithm >::msg ( ) const
inlineinherited

Definition at line 24 of file AthCommonMsg.h.

24 {
25 return this->msgStream();
26 }

◆ msgLvl()

bool AthCommonMsg< Gaudi::Algorithm >::msgLvl ( const MSG::Level lvl) const
inlineinherited

Definition at line 30 of file AthCommonMsg.h.

30 {
31 return this->msgLevel(lvl);
32 }

◆ outputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::outputHandles ( ) const
overridevirtualinherited

Return this algorithm's output handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ renounce()

std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::renounce ( T & h)
inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

381 {
382 h.renounce();
384 }
std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > renounce(T &h)

◆ renounceArray()

void AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::renounceArray ( SG::VarHandleKeyArray & handlesArray)
inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

364 {
366 }

◆ setFilterPassed()

virtual void AthCommonReentrantAlgorithm< Gaudi::Algorithm >::setFilterPassed ( bool state,
const EventContext & ctx ) const
inlinevirtualinherited

Definition at line 100 of file AthCommonReentrantAlgorithm.h.

100 {
102 }
virtual void setFilterPassed(bool state, const EventContext &ctx) const

◆ sysExecute()

StatusCode AthCommonReentrantAlgorithm< Gaudi::Algorithm >::sysExecute ( const EventContext & ctx)
overridevirtualinherited

Execute an algorithm.

We override this in order to work around an issue with the Algorithm base class storing the event context in a member variable that can cause crashes in MT jobs.

Definition at line 85 of file AthCommonReentrantAlgorithm.cxx.

77{
78 return BaseAlg::sysExecute (ctx);
79}

◆ sysInitialize()

StatusCode AthCommonReentrantAlgorithm< Gaudi::Algorithm >::sysInitialize ( )
overridevirtualinherited

Override sysInitialize.

Override sysInitialize from the base class.

Loop through all output handles, and if they're WriteCondHandles, automatically register them and this Algorithm with the CondSvc

Scan through all outputHandles, and if they're WriteCondHandles, register them with the CondSvc

Reimplemented from AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >.

Reimplemented in HypoBase, and InputMakerBase.

Definition at line 61 of file AthCommonReentrantAlgorithm.cxx.

107 {
109
110 if (sc.isFailure()) {
111 return sc;
112 }
113
114 ServiceHandle<ICondSvc> cs("CondSvc",name());
115 for (auto h : outputHandles()) {
116 if (h->isCondition() && h->mode() == Gaudi::DataHandle::Writer) {
117 // do this inside the loop so we don't create the CondSvc until needed
118 if ( cs.retrieve().isFailure() ) {
119 ATH_MSG_WARNING("no CondSvc found: won't autoreg WriteCondHandles");
120 return StatusCode::SUCCESS;
121 }
122 if (cs->regHandle(this,*h).isFailure()) {
124 ATH_MSG_ERROR("unable to register WriteCondHandle " << h->fullKey()
125 << " with CondSvc");
126 }
127 }
128 }
129 return sc;
130}
#define ATH_MSG_ERROR(x)
virtual std::vector< Gaudi::DataHandle * > outputHandles() const override

◆ sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::sysStart ( )
overridevirtualinherited

Handle START transition.

We override this in order to make sure that conditions handle keys can cache a pointer to the conditions container.

◆ updateVHKA()

void AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::updateVHKA ( Gaudi::Details::PropertyBase & )
inlineinherited

Definition at line 308 of file AthCommonDataStore.h.

308 {
309 // debug() << "updateVHKA for property " << p.name() << " " << p.toString()
310 // << " size: " << m_vhka.size() << endmsg;
311 for (auto &a : m_vhka) {
313 for (auto k : keys) {
314 k->setOwner(this);
315 }
316 }
317 }

Member Data Documentation

◆ m_associateWithInDetTP

Gaudi::Property<bool> Muon::RecoToTruthAssociationAlg::m_associateWithInDetTP {this, "AssociateWithInDetTP", false, "force use of ID track particles for association"}
private

Definition at line 74 of file RecoToTruthAssociationAlg.h.

74{this, "AssociateWithInDetTP", false, "force use of ID track particles for association"};

◆ m_assocTrkContainers

Gaudi::Property<std::vector<std::string> > Muon::RecoToTruthAssociationAlg::m_assocTrkContainers
private
Initial value:
{this, "TrackContainers", {"CombinedMuonTrackParticles",
"ExtrapolatedMuonTrackParticles",
"MSOnlyExtrapolatedMuonTrackParticles"},
"Collection of track containers to be decorated before this alg can be scheduled. truthLink decoration exploited "}

Definition at line 69 of file RecoToTruthAssociationAlg.h.

69 {this, "TrackContainers", {"CombinedMuonTrackParticles",
70 "ExtrapolatedMuonTrackParticles",
71 "MSOnlyExtrapolatedMuonTrackParticles"},
72 "Collection of track containers to be decorated before this alg can be scheduled. truthLink decoration exploited "};

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_detStore
privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_evtStore
privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

◆ m_extendedExtraObjects

DataObjIDColl AthCommonReentrantAlgorithm< Gaudi::Algorithm >::m_extendedExtraObjects
privateinherited

Extra output dependency collection, extended by AthAlgorithmDHUpdate to add symlinks.

Empty if no symlinks were found.

Definition at line 114 of file AthCommonReentrantAlgorithm.h.

◆ m_idHelperSvc

ServiceHandle<Muon::IMuonIdHelperSvc> Muon::RecoToTruthAssociationAlg::m_idHelperSvc {this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"}
private

Definition at line 76 of file RecoToTruthAssociationAlg.h.

76{this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};

◆ m_inputDecorKey

SG::ReadDecorHandleKeyArray<xAOD::IParticleContainer> Muon::RecoToTruthAssociationAlg::m_inputDecorKey
private
Initial value:
{
this, "TrkTruthLinkKeys", {},
"Declare the decoration dependencies of this algorithm. Steer via TrackContainers property"}

Definition at line 66 of file RecoToTruthAssociationAlg.h.

66 {
67 this, "TrkTruthLinkKeys", {},
68 "Declare the decoration dependencies of this algorithm. Steer via TrackContainers property"};

◆ m_muonTruthParticleClassification

SG::WriteDecorHandleKey<xAOD::MuonContainer> Muon::RecoToTruthAssociationAlg::m_muonTruthParticleClassification
private
Initial value:
{
this, "MuonTruthParticleClassification", m_recoMuKey, "truthClassification",
"muon truth classification auxdata name; name will be reset in initialize() based on m_muonName"}

Definition at line 51 of file RecoToTruthAssociationAlg.h.

51 {
52 this, "MuonTruthParticleClassification", m_recoMuKey, "truthClassification",
53 "muon truth classification auxdata name; name will be reset in initialize() based on m_muonName"};

◆ m_muonTruthParticleLink

SG::WriteDecorHandleKey<xAOD::MuonContainer> Muon::RecoToTruthAssociationAlg::m_muonTruthParticleLink
private
Initial value:
{
this, "MuonTruthParticleLink", m_recoMuKey, "truthParticleLink",
"muon truth particle link auxdata name; name will be reset in initialize() based on m_muonName"}

Decorations for the reconstructed muon particles.

Definition at line 42 of file RecoToTruthAssociationAlg.h.

42 {
43 this, "MuonTruthParticleLink", m_recoMuKey, "truthParticleLink",
44 "muon truth particle link auxdata name; name will be reset in initialize() based on m_muonName"};

◆ m_muonTruthParticleNPhiMatched

SG::WriteDecorHandleKey<xAOD::MuonContainer> Muon::RecoToTruthAssociationAlg::m_muonTruthParticleNPhiMatched
private
Initial value:
{
this, "MuonTruthParticleNPhiMatched", m_recoMuKey, "nphiMatchedHitsPerChamberLayer",
"muon vector of number of phi matched hits per chamber layer auxdata name; name will be reset in initialize() based on m_muonName"}

Definition at line 58 of file RecoToTruthAssociationAlg.h.

58 {
59 this, "MuonTruthParticleNPhiMatched", m_recoMuKey, "nphiMatchedHitsPerChamberLayer",
60 "muon vector of number of phi matched hits per chamber layer auxdata name; name will be reset in initialize() based on m_muonName"};

◆ m_muonTruthParticleNPrecMatched

SG::WriteDecorHandleKey<xAOD::MuonContainer> Muon::RecoToTruthAssociationAlg::m_muonTruthParticleNPrecMatched
private
Initial value:
{
this, "MuonTruthParticleNPrecMatched", m_recoMuKey, "nprecMatchedHitsPerChamberLayer",
"muon vector of number of precision matched hits per chamber layer auxdata name; name will be reset in initialize() based on "
"m_muonName"}

Definition at line 54 of file RecoToTruthAssociationAlg.h.

54 {
55 this, "MuonTruthParticleNPrecMatched", m_recoMuKey, "nprecMatchedHitsPerChamberLayer",
56 "muon vector of number of precision matched hits per chamber layer auxdata name; name will be reset in initialize() based on "
57 "m_muonName"};

◆ m_muonTruthParticleNTrigEtaMatched

SG::WriteDecorHandleKey<xAOD::MuonContainer> Muon::RecoToTruthAssociationAlg::m_muonTruthParticleNTrigEtaMatched
private
Initial value:
{
this, "MuonTruthParticleNTrigEtaMatched", m_recoMuKey, "ntrigEtaMatchedHitsPerChamberLayer",
"muon vector of number of phi matched hits per chamber layer auxdata name; name will be reset in initialize() based on m_muonName"}

Definition at line 61 of file RecoToTruthAssociationAlg.h.

61 {
62 this, "MuonTruthParticleNTrigEtaMatched", m_recoMuKey, "ntrigEtaMatchedHitsPerChamberLayer",
63 "muon vector of number of phi matched hits per chamber layer auxdata name; name will be reset in initialize() based on m_muonName"};

◆ m_muonTruthParticleOrigin

SG::WriteDecorHandleKey<xAOD::MuonContainer> Muon::RecoToTruthAssociationAlg::m_muonTruthParticleOrigin
private
Initial value:
{
this, "MuonTruthParticleOrigin", m_recoMuKey, "truthOrigin",
"muon truth origin auxdata name; name will be reset in initialize() based on m_muonName"}

Definition at line 48 of file RecoToTruthAssociationAlg.h.

48 {
49 this, "MuonTruthParticleOrigin", m_recoMuKey, "truthOrigin",
50 "muon truth origin auxdata name; name will be reset in initialize() based on m_muonName"};

◆ m_muonTruthParticleType

SG::WriteDecorHandleKey<xAOD::MuonContainer> Muon::RecoToTruthAssociationAlg::m_muonTruthParticleType
private
Initial value:
{
this, "MuonTruthParticleType", m_recoMuKey, "truthType",
"muon truth type auxdata name; name will be reset in initialize() based on m_muonName"}

Definition at line 45 of file RecoToTruthAssociationAlg.h.

45 {
46 this, "MuonTruthParticleType", m_recoMuKey, "truthType",
47 "muon truth type auxdata name; name will be reset in initialize() based on m_muonName"};

◆ m_muonTruthRecoLink

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> Muon::RecoToTruthAssociationAlg::m_muonTruthRecoLink
private
Initial value:
{
this, "MuonTruthParticleRecoLink", m_truthMuKey, "",
"container name for muon truth particles; the full handle name, including the reco muon link auxdata, is set in initialize()"}

Decorations for the filtered muon truth particles.

Definition at line 33 of file RecoToTruthAssociationAlg.h.

33 {
34 this, "MuonTruthParticleRecoLink", m_truthMuKey, "",
35 "container name for muon truth particles; the full handle name, including the reco muon link auxdata, is set in initialize()"};

◆ m_recoLink

Gaudi::Property<std::string> Muon::RecoToTruthAssociationAlg::m_recoLink {this, "RecoLinkName", "recoMuonLink" , "Decoration to the truth particle pointing to the muon"}
private

Definition at line 73 of file RecoToTruthAssociationAlg.h.

73{this, "RecoLinkName", "recoMuonLink" , "Decoration to the truth particle pointing to the muon"};

◆ m_recoMuKey

SG::ReadHandleKey<xAOD::MuonContainer> Muon::RecoToTruthAssociationAlg::m_recoMuKey
private
Initial value:
{
this, "MuonContainerName", "Muons",
"container name for reco muon particles; the full handle name, including the reco muon link auxdata, is set in initialize()"}

Definition at line 37 of file RecoToTruthAssociationAlg.h.

37 {
38 this, "MuonContainerName", "Muons",
39 "container name for reco muon particles; the full handle name, including the reco muon link auxdata, is set in initialize()"};

◆ m_truthMuKey

SG::ReadHandleKey<xAOD::TruthParticleContainer> Muon::RecoToTruthAssociationAlg::m_truthMuKey
private
Initial value:
{
this, "MuonTruthParticleContainerName", "MuonTruthParticles"}

Key to the filtered muon truth particles.

Definition at line 29 of file RecoToTruthAssociationAlg.h.

29 {
30 this, "MuonTruthParticleContainerName", "MuonTruthParticles"};

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_varHandleArraysDeclared
privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vhka

std::vector<SG::VarHandleKeyArray*> AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_vhka
privateinherited

Definition at line 398 of file AthCommonDataStore.h.


The documentation for this class was generated from the following files: