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::ReadHandleKey< xAOD::MuonContainerm_recoMuKey
SG::WriteDecorHandleKey< xAOD::TruthParticleContainerm_muonTruthRecoLink
SG::WriteDecorHandleKey< xAOD::MuonContainerm_muonTruthParticleLink
SG::WriteDecorHandleKey< xAOD::MuonContainerm_muonTruthParticleType
SG::WriteDecorHandleKey< xAOD::MuonContainerm_muonTruthParticleOrigin
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 364 of file RecoToTruthAssociationAlg.cxx.

364 {
367 if (identifiers.empty()) { return; }
368 for (unsigned int i = 0; i < vec.size(); ++i) {
369 if (vec[i] != dummy_unsigned) continue;
370 for (unsigned j = 0; j < identifiers.size(); ++j) {
371 const Identifier id{identifiers[j]};
372 if (m_idHelperSvc->measuresPhi(id)) {
373 const auto phiIdx = static_cast<PhiIndex>(i);
374 if (m_idHelperSvc->phiIndex(id) == phiIdx) {
375 vec[i] = 0;
376 break;
377 }
378 } else {
379 const auto chIdx = static_cast<ChIndex>(i);
380 if (m_idHelperSvc->chamberIndex(id) == chIdx) {
381 vec[i] = 0;
382 break;
383 }
384 }
385 }
386 }
387}
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 264 of file RecoToTruthAssociationAlg.cxx.

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

89 {
90 const xAOD::TruthParticleContainer* muonTruthContainer{nullptr};
91 ATH_CHECK(SG::get(muonTruthContainer, m_truthMuKey , ctx));
92 std::unique_ptr<SG::WriteDecorHandle<xAOD::TruthParticleContainer, ElementLink<xAOD::MuonContainer>>> muonTruthParticleRecoLink{};
93 if (!m_muonTruthRecoLink.empty()) {
94 muonTruthParticleRecoLink = std::make_unique<SG::WriteDecorHandle<xAOD::TruthParticleContainer, ElementLink<xAOD::MuonContainer>>>(m_muonTruthRecoLink, ctx);
95 }
96
97 SG::WriteDecorHandle<xAOD::MuonContainer, ElementLink<xAOD::TruthParticleContainer> > muonTruthParticleLink(m_muonTruthParticleLink,
98 ctx);
99 if (!muonTruthParticleLink.isValid()) {
100 ATH_MSG_WARNING("muon particle container not valid");
101 return StatusCode::FAILURE;
102 }
103 SG::WriteDecorHandle<xAOD::MuonContainer, int> muonTruthParticleType(m_muonTruthParticleType, ctx);
104 SG::WriteDecorHandle<xAOD::MuonContainer, int> muonTruthParticleOrigin(m_muonTruthParticleOrigin, ctx);
105 SG::WriteDecorHandle<xAOD::MuonContainer, std::vector<unsigned int> > muonTruthParticleNPrecMatched(m_muonTruthParticleNPrecMatched,
106 ctx);
107 SG::WriteDecorHandle<xAOD::MuonContainer, std::vector<unsigned int> > muonTruthParticleNPhiMatched(m_muonTruthParticleNPhiMatched, ctx);
108 SG::WriteDecorHandle<xAOD::MuonContainer, std::vector<unsigned int> > muonTruthParticleNTrigEtaMatched(m_muonTruthParticleNTrigEtaMatched, ctx);
109
110 // add link to reco muons and viceversa
111 bool saw_staco = false;
112 bool decor_staco = false;
113
114 // loop over muons
115 for (const xAOD::Muon* muon : *muonTruthParticleLink) {
116 // 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
117 // combined is not truth-matched)
118 ATH_MSG_DEBUG("muon with pT " << muon->pt() << " MeV, eta: " << muon->eta() << ", phi " << muon->phi() << " and author "
119 << muon->author());
120 const xAOD::TrackParticle* tp = nullptr;
121 if (m_associateWithInDetTP || muon->author() == xAOD::Muon::STACO || muon->author() == xAOD::Muon::MuGirl) {
122 tp = muon->trackParticle(xAOD::Muon::InnerDetectorTrackParticle);
123 } else {
124 tp = muon->primaryTrackParticle();
125 }
126
127 bool foundTruth{false}, setOrigin{false};
128 if (tp) {
129 // Associate reco with truth muon. Loop over reconstructed muons, get track particle for each one.
130 // Each track particle should carry a link to the corresponding truth particle. Then compare this truth particle link with the
131 // given truth muon particle
132 if (acc_origin.isAvailable(*tp) && acc_origin(*tp) != 0) {
133 muonTruthParticleOrigin(*muon) = acc_origin(*tp);
134 muonTruthParticleType(*muon) = acc_type(*tp);
135 setOrigin = true;
136 }
137
138 ElementLink<xAOD::TruthParticleContainer> truthLink;
139 if (acc_link.isAvailable(*tp)) {
140 truthLink = acc_link(*tp);
141 } else {
142 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());
143 }
144
145 if (truthLink.isValid()) {
146 ATH_MSG_VERBOSE(" Got valid truth link for muon author " << muon->author() << " uniqueID " << HepMC::uniqueID(*truthLink));
147 // loop over truth particles
148 for (const xAOD::TruthParticle* truthParticle : *muonTruthContainer) {
149 if (!MC::isStable(truthParticle)) continue;
150 ATH_MSG_DEBUG("Got truth muon with uniqueID " << HepMC::uniqueID(truthParticle) << " pt " << truthParticle->pt());
151 if ( !HepMC::is_sim_descendant(*truthLink, truthParticle)) {
152 ATH_MSG_VERBOSE("UniqueID truth link: " << HepMC::uniqueID(*truthLink)
153 << " is not decendant of " << HepMC::uniqueID(truthParticle));
154 continue;
155 }
156 ATH_MSG_VERBOSE("Truth muon uniqueID matches -> creating link with truth particle " << HepMC::uniqueID(*truthLink));
157 foundTruth = true;
159 ElementLink<xAOD::TruthParticleContainer> muonTruthLink{*muonTruthContainer,
160 truthParticle->index(),
161 ctx};
162 muonTruthLink.toPersistent();
163 muonTruthParticleLink(*muon) = muonTruthLink;
164 if (!setOrigin) {
165 muonTruthParticleOrigin(*muon) = acc_origin(*tp);
166 muonTruthParticleType(*muon) = acc_type(*tp);
167 setOrigin = true;
168 }
170 if (muonTruthParticleRecoLink && muonTruthParticleRecoLink->operator()(*truthParticle).isValid()){
171 const xAOD::Muon* decor_muon = *muonTruthParticleRecoLink->operator()(*truthParticle);
172 ATH_MSG_VERBOSE("Truth particle is already decorated with reco muon "<<decor_muon->pt()*1.e-3
173 <<" eta: "<<decor_muon->eta()<<" phi: "<<decor_muon->phi()<<" charge: "<<
174 decor_muon->charge()<<" author: "<<decor_muon->author()<<" all authors: "<<
175 decor_muon->allAuthors());
176
177 // Check first if the exiting muon has a better author
178 if (MuonCombined::authorRank(decor_muon->author()) < MuonCombined::authorRank(muon->author())){
179 ATH_MSG_DEBUG("Author of the decorated muon is better than the one of the new candidate");
180 continue;
181 }
183 const int com_score = (muon->allAuthors() & com_bit) - (decor_muon->allAuthors() &com_bit);
184 if (com_score > 0){
185 ATH_MSG_DEBUG("Found two muons reconstructed by an equivalent author. But this one is from the commissioning chain");
186 continue;
187 }
189 if (deltaR2(muon,truthParticle) >= deltaR2(muon, decor_muon)) continue;
190 }
191
192
193 ElementLink<xAOD::MuonContainer> muonLink{muon, *muonTruthParticleLink, ctx};
194
196 std::vector<unsigned int> nprecHitsPerChamberLayer(toInt(ChIndex::ChIndexMax), dummy_unsigned);
197 std::vector<unsigned int> nphiHitsPerChamberLayer(toInt(PhiIndex::PhiIndexMax), dummy_unsigned);
198 std::vector<unsigned int> ntrigEtaHitsPerChamberLayer(toInt(PhiIndex::PhiIndexMax), dummy_unsigned);
199
200 constexpr int author_sel = (1<<xAOD::Muon::MuidCo) | (1<<xAOD::Muon::MuidSA) | (1<<xAOD::Muon::MuGirl);
201 count_chamber_layers(muon->allAuthors() & author_sel
202 ? truthParticle
203 : nullptr,
204 tp->track(), nprecHitsPerChamberLayer, nphiHitsPerChamberLayer, ntrigEtaHitsPerChamberLayer);
206 muonTruthParticleNPrecMatched(*muon) = nprecHitsPerChamberLayer;
207 muonTruthParticleNPhiMatched(*muon) = nphiHitsPerChamberLayer;
208 muonTruthParticleNTrigEtaMatched(*muon) = ntrigEtaHitsPerChamberLayer;
209
210 if (muonTruthParticleRecoLink) (*muonTruthParticleRecoLink)(*truthParticle) = muonLink;
211 break;
212 }
213 } else {
214 ATH_MSG_DEBUG("Invalid truth link");
215 }
216 } else {
217 ATH_MSG_WARNING("Could not find the appropiate track particle for muon with pT: " << muon->pt() * 1.e-3 << " GeV, eta: "
218 << muon->eta() << ", phi: " << muon->phi()
219 << " author: " << muon->author());
220 }
221
222 if (!setOrigin) {
223 muonTruthParticleOrigin(*muon) = 0;
224 muonTruthParticleType(*muon) = 0;
225 }
226 if (!foundTruth) {
227 muonTruthParticleLink(*muon) = ElementLink<xAOD::TruthParticleContainer>();
228 // add these empty vectors
229 muonTruthParticleNPrecMatched(*muon) = std::vector<unsigned int>{};
230 muonTruthParticleNPhiMatched(*muon) = std::vector<unsigned int>{};
231 muonTruthParticleNTrigEtaMatched(*muon) = std::vector<unsigned int>{};
232 }
235 if (muon->author() == xAOD::Muon::STACO) {
236 const xAOD::TrackParticle* cmb_trk = muon->trackParticle(xAOD::Muon::CombinedTrackParticle);
237 if (!cmb_trk){
238 ATH_MSG_WARNING("Even a STACO muon should have a combined track");
239 continue;
240 } else {
241 if (!saw_staco) {
242 saw_staco = true;
243 decor_staco = !dec_origin.isAvailable (*cmb_trk);
244 }
245 if (decor_staco) {
246 dec_origin(*cmb_trk) = acc_origin(*muon);
247 dec_type(*cmb_trk) = acc_type(*muon);
248 dec_link(*cmb_trk) = acc_link(*muon);
249 }
250 }
251 }
252 }
254 if (muonTruthParticleRecoLink && !muonTruthParticleRecoLink->isAvailable()) {
255 for (const xAOD::TruthParticle* truthParticle : **muonTruthParticleRecoLink) {
256 ATH_MSG_DEBUG("no reco muon link set, add an empty one");
257 (*muonTruthParticleRecoLink)(*truthParticle) = ElementLink<xAOD::MuonContainer>();
258 }
259 }
260
261 return StatusCode::SUCCESS;
262}
#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
SG::WriteDecorHandleKey< xAOD::MuonContainer > m_muonTruthParticleType
SG::WriteDecorHandleKey< xAOD::MuonContainer > m_muonTruthParticleLink
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 47 of file RecoToTruthAssociationAlg.cxx.

47 {
48 ATH_CHECK(m_idHelperSvc.retrieve());
49 ATH_CHECK(m_truthMuKey.initialize());
50 ATH_CHECK(m_recoMuKey.initialize());
51
52 if (m_recoLink.empty()){
54 } else {
56 }
57
58 ATH_CHECK(m_muonTruthRecoLink.initialize(!m_recoLink.empty()));
65 for (const std::string& trk_coll : m_assocTrkContainers.value()){
66 m_inputDecorKey.emplace_back(trk_coll + ".truthParticleLink");
67 }
68
69 m_inputDecorKey.emplace_back(m_truthMuKey, SG::AuxTypeRegistry::instance().getName(acc_origin.auxid()));
70 m_inputDecorKey.emplace_back(m_truthMuKey, SG::AuxTypeRegistry::instance().getName(acc_type.auxid()));
71 if (m_idHelperSvc->hasMDT()) {
72 m_inputDecorKey.emplace_back(m_truthMuKey, SG::AuxTypeRegistry::instance().getName(truthMdtHitsAcc.auxid()));
73 }
74 if (m_idHelperSvc->hasRPC()) {
75 m_inputDecorKey.emplace_back(m_truthMuKey, SG::AuxTypeRegistry::instance().getName(truthRpcHitsAcc.auxid()));
76 }
77 if (m_idHelperSvc->hasTGC()) {
78 m_inputDecorKey.emplace_back(m_truthMuKey, SG::AuxTypeRegistry::instance().getName(truthTgcHitsAcc.auxid()));
79 }
80 if (m_idHelperSvc->hasCSC()) {
81 m_inputDecorKey.emplace_back(m_truthMuKey, SG::AuxTypeRegistry::instance().getName(truthCscHitsAcc.auxid()));
82 }
83
84 ATH_CHECK(m_inputDecorKey.initialize());
85 return StatusCode::SUCCESS;
86}
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 68 of file RecoToTruthAssociationAlg.h.

68{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 63 of file RecoToTruthAssociationAlg.h.

63 {this, "TrackContainers", {"CombinedMuonTrackParticles",
64 "ExtrapolatedMuonTrackParticles",
65 "MSOnlyExtrapolatedMuonTrackParticles"},
66 "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 70 of file RecoToTruthAssociationAlg.h.

70{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 60 of file RecoToTruthAssociationAlg.h.

60 {
61 this, "TrkTruthLinkKeys", {},
62 "Declare the decoration dependencies of this algorithm. Steer via TrackContainers property"};

◆ 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"}

Definition at line 39 of file RecoToTruthAssociationAlg.h.

39 {
40 this, "MuonTruthParticleLink", m_recoMuKey, "truthParticleLink",
41 "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 52 of file RecoToTruthAssociationAlg.h.

52 {
53 this, "MuonTruthParticleNPhiMatched", m_recoMuKey, "nphiMatchedHitsPerChamberLayer",
54 "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 48 of file RecoToTruthAssociationAlg.h.

48 {
49 this, "MuonTruthParticleNPrecMatched", m_recoMuKey, "nprecMatchedHitsPerChamberLayer",
50 "muon vector of number of precision matched hits per chamber layer auxdata name; name will be reset in initialize() based on "
51 "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 55 of file RecoToTruthAssociationAlg.h.

55 {
56 this, "MuonTruthParticleNTrigEtaMatched", m_recoMuKey, "ntrigEtaMatchedHitsPerChamberLayer",
57 "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 45 of file RecoToTruthAssociationAlg.h.

45 {
46 this, "MuonTruthParticleOrigin", m_recoMuKey, "truthOrigin",
47 "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 42 of file RecoToTruthAssociationAlg.h.

42 {
43 this, "MuonTruthParticleType", m_recoMuKey, "truthType",
44 "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()"}

Definition at line 36 of file RecoToTruthAssociationAlg.h.

36 {
37 this, "MuonTruthParticleRecoLink", m_truthMuKey, "",
38 "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 67 of file RecoToTruthAssociationAlg.h.

67{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 muon truth particles; the full handle name, including the reco muon link auxdata, is set in initialize()"}

Definition at line 32 of file RecoToTruthAssociationAlg.h.

32 {
33 this, "MuonContainerName", "Muons",
34 "container name for muon truth 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: