ATLAS Offline Software
Loading...
Searching...
No Matches
PMGTools::PMGDecayProductsSelectionTool Class Referencefinal

an IAsgSelectionTool that select particles based on the allowed decay chain More...

#include <PMGDecayProductsSelectionTool.h>

Inheritance diagram for PMGTools::PMGDecayProductsSelectionTool:
Collaboration diagram for PMGTools::PMGDecayProductsSelectionTool:

Public Member Functions

 PMGDecayProductsSelectionTool (const std::string &name)
 standard constructor
virtual StatusCode initialize () override
 Dummy implementation of the initialisation function.
virtual const asg::AcceptInfogetAcceptInfo () const override
 Declare the interface ID for this pure-virtual interface class to the Athena framework.
virtual asg::AcceptData accept (const xAOD::IParticle *) const override
 The main accept method: the actual cuts are applied here.
virtual void print () const
 Print the state of the tool.
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 sysInitialize () override
 Perform system initialization for an algorithm.
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
Additional helper functions, not directly mimicking Athena
template<class T>
const T * getProperty (const std::string &name) const
 Get one of the tool's properties.
const std::string & msg_level_name () const __attribute__((deprecated))
 A deprecated function for getting the message level's name.
const std::string & getName (const void *ptr) const
 Get the name of an object that is / should be in the event store.
SG::sgkey_t getKey (const void *ptr) const
 Get the (hashed) key of an object that is in the event store.

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

asg::AcceptData hasRequiredInitialParent (const xAOD::TruthParticle *truthParticle, asg::AcceptData &acceptData) const
 Helper function to check for required parent particles.
size_t getNParents (const xAOD::TruthParticle *truthParticle) const
 Helper function to get the number of parent particles.
const xAOD::TruthParticlegetParent (const xAOD::TruthParticle *truthParticle, size_t index) const
 Helper function get a parent by index.
Gaudi::Details::PropertyBase & declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
 specialization for handling Gaudi::Property<SG::VarHandleKey>

Private Attributes

int m_truthParticleIndex { -1 }
 Index for the truth particle link.
int m_requiredParentIndex { -1 }
 Index for the required parent particles.
asg::AcceptInfo m_accept
 the AcceptInfo we are using
std::unique_ptr< const SG::AuxElement::Accessor< std::vector< ElementLink< xAOD::TruthParticleContainer > > > > m_parentsAccessor {}
 common parents accessor
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
std::vector< int > m_requiredParentPDGIDs
 tool properties
std::vector< int > m_allowedIntermediatePDGIDs

Detailed Description

an IAsgSelectionTool that select particles based on the allowed decay chain

Definition at line 23 of file PMGDecayProductsSelectionTool.h.

Member Typedef Documentation

◆ StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< AlgTool > >::StoreGateSvc_t
privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

◆ PMGDecayProductsSelectionTool()

PMGTools::PMGDecayProductsSelectionTool::PMGDecayProductsSelectionTool ( const std::string & name)

standard constructor

Definition at line 26 of file PMGDecayProductsSelectionTool.cxx.

28 : AsgTool (name)
29 {
30 declareProperty ("requiredParentPDGIDs", m_requiredParentPDGIDs, "required parent particle PDG IDs (positive only)");
31 declareProperty ("allowedIntermediatePDGIDs", m_allowedIntermediatePDGIDs, "allowed intermediate particles PDG IDs (positive only)");
32 }
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T, V, H > &t)
std::vector< int > m_requiredParentPDGIDs
tool properties
AsgTool(const std::string &name)
Constructor specifying the tool instance's name.
Definition AsgTool.cxx:58

Member Function Documentation

◆ accept()

asg::AcceptData PMGTools::PMGDecayProductsSelectionTool::accept ( const xAOD::IParticle * ) const
overridevirtual

The main accept method: the actual cuts are applied here.

Implements IAsgSelectionTool.

Definition at line 80 of file PMGDecayProductsSelectionTool.cxx.

82 {
83 asg::AcceptData acceptData (&m_accept);
84
85 // Check if xAOD::TruthParticle or if not if it has the TruthParticleLink
86 const xAOD::TruthParticle *truthParticle
87 = dynamic_cast<const xAOD::TruthParticle *> (particle);
88 if (truthParticle == nullptr)
89 {
90 // need to find the truth particle
91 truthParticle = xAOD::TruthHelpers::getTruthParticle(*particle);
92 if (truthParticle == nullptr)
93 {
94 acceptData.setCutResult (m_truthParticleIndex, false);
95 return acceptData;
96 }
97 }
98 acceptData.setCutResult (m_truthParticleIndex, true);
99
100 return hasRequiredInitialParent(truthParticle, acceptData);
101 }
int m_truthParticleIndex
Index for the truth particle link.
asg::AcceptInfo m_accept
the AcceptInfo we are using
asg::AcceptData hasRequiredInitialParent(const xAOD::TruthParticle *truthParticle, asg::AcceptData &acceptData) const
Helper function to check for required parent particles.
constexpr ParticleHypothesis particle[PARTICLEHYPOTHESES]
the array of masses
const xAOD::TruthParticle * getTruthParticle(const xAOD::IParticle &p)
Return the truthParticle associated to the given IParticle (if any)
TruthParticle_v1 TruthParticle
Typedef to implementation.

◆ declareGaudiProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< AlgTool > >::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 }

◆ declareProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< AlgTool > >::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< AlgTool > >::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< AlgTool > >::evtStore ( )
inlineinherited

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

Definition at line 85 of file AthCommonDataStore.h.

◆ extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::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

◆ getAcceptInfo()

const asg::AcceptInfo & PMGTools::PMGDecayProductsSelectionTool::getAcceptInfo ( ) const
overridevirtual

Declare the interface ID for this pure-virtual interface class to the Athena framework.

Method to get the AcceptInfo to query what cuts are defined.

Implements IAsgSelectionTool.

Definition at line 72 of file PMGDecayProductsSelectionTool.cxx.

74 {
75 return m_accept;
76 }

◆ getKey()

SG::sgkey_t asg::AsgTool::getKey ( const void * ptr) const
inherited

Get the (hashed) key of an object that is in the event store.

This is a bit of a special one. StoreGateSvc and xAOD::TEvent both provide ways for getting the SG::sgkey_t key for an object that is in the store, based on a bare pointer. But they provide different interfaces for doing so.

In order to allow tools to efficiently perform this operation, they can use this helper function.

See also
asg::AsgTool::getName
Parameters
ptrThe bare pointer to the object that the event store should know about
Returns
The hashed key of the object in the store. If not found, an invalid (zero) key.

Definition at line 119 of file AsgTool.cxx.

119 {
120
121#ifdef XAOD_STANDALONE
122 // In case we use @c xAOD::TEvent, we have a direct function call
123 // for this.
124 return evtStore()->event()->getKey( ptr );
125#else
126 const SG::DataProxy* proxy = evtStore()->proxy( ptr );
127 return ( proxy == nullptr ? 0 : proxy->sgkey() );
128#endif // XAOD_STANDALONE
129 }
ServiceHandle< StoreGateSvc > & evtStore()

◆ getName()

const std::string & asg::AsgTool::getName ( const void * ptr) const
inherited

Get the name of an object that is / should be in the event store.

This is a bit of a special one. StoreGateSvc and xAOD::TEvent both provide ways for getting the std::string name for an object that is in the store, based on a bare pointer. But they provide different interfaces for doing so.

In order to allow tools to efficiently perform this operation, they can use this helper function.

See also
asg::AsgTool::getKey
Parameters
ptrThe bare pointer to the object that the event store should know about
Returns
The string name of the object in the store. If not found, an empty string.

Definition at line 106 of file AsgTool.cxx.

106 {
107
108#ifdef XAOD_STANDALONE
109 // In case we use @c xAOD::TEvent, we have a direct function call
110 // for this.
111 return evtStore()->event()->getName( ptr );
112#else
113 const SG::DataProxy* proxy = evtStore()->proxy( ptr );
114 static const std::string dummy = "";
115 return ( proxy == nullptr ? dummy : proxy->name() );
116#endif // XAOD_STANDALONE
117 }

◆ getNParents()

size_t PMGTools::PMGDecayProductsSelectionTool::getNParents ( const xAOD::TruthParticle * truthParticle) const
private

Helper function to get the number of parent particles.

Definition at line 147 of file PMGDecayProductsSelectionTool.cxx.

149 {
150 if (m_parentsAccessor->isAvailable (*truthParticle))
151 {
152 return (*m_parentsAccessor)(*truthParticle).size();
153 }
154 else
155 {
156 return truthParticle->nParents();
157 }
158 }
std::unique_ptr< const SG::AuxElement::Accessor< std::vector< ElementLink< xAOD::TruthParticleContainer > > > > m_parentsAccessor
common parents accessor
size_t nParents() const
Number of parents of this particle.

◆ getParent()

const xAOD::TruthParticle * PMGTools::PMGDecayProductsSelectionTool::getParent ( const xAOD::TruthParticle * truthParticle,
size_t index ) const
private

Helper function get a parent by index.

Definition at line 162 of file PMGDecayProductsSelectionTool.cxx.

165 {
166 if (m_parentsAccessor->isAvailable (*truthParticle))
167 {
168 ElementLink<xAOD::TruthParticleContainer> parentElementLink
169 = (*m_parentsAccessor)(*truthParticle).at(index);
170 return parentElementLink.isValid() ? *parentElementLink : nullptr;
171 }
172 else
173 {
174 return truthParticle->parent(index);
175 }
176 }
const TruthParticle_v1 * parent(size_t i) const
Retrieve the i-th mother (TruthParticle) of this TruthParticle.

◆ getProperty()

template<class T>
const T * asg::AsgTool::getProperty ( const std::string & name) const
inherited

Get one of the tool's properties.

◆ hasRequiredInitialParent()

asg::AcceptData PMGTools::PMGDecayProductsSelectionTool::hasRequiredInitialParent ( const xAOD::TruthParticle * truthParticle,
asg::AcceptData & acceptData ) const
private

Helper function to check for required parent particles.

Definition at line 105 of file PMGDecayProductsSelectionTool.cxx.

107 {
108 size_t nParents = getNParents (truthParticle);
109 for (size_t i = 0; i < nParents; i++)
110 {
111 const xAOD::TruthParticle *parent = getParent(truthParticle, i);
112 if (parent)
113 {
114 if (std::find(m_requiredParentPDGIDs.begin(), m_requiredParentPDGIDs.end(), std::abs(parent->pdgId())) != m_requiredParentPDGIDs.end())
115 {
116 acceptData.setCutResult (m_requiredParentIndex, true);
117 return acceptData;
118 }
119 else if (m_allowedIntermediatePDGIDs.empty() || std::find(m_allowedIntermediatePDGIDs.begin(), m_allowedIntermediatePDGIDs.end(), std::abs(parent->pdgId())) != m_allowedIntermediatePDGIDs.end())
120 {
121 return hasRequiredInitialParent(parent, acceptData);
122 }
123 else
124 {
125 ATH_MSG_VERBOSE("Removing particle as parent is not allowed: " << parent->pdgId());
126 return acceptData;
127 }
128 }
129 else
130 {
131 ATH_MSG_WARNING("Particle parent is not valid");
132 return acceptData;
133 }
134 }
135
136 if (std::find(m_requiredParentPDGIDs.begin(), m_requiredParentPDGIDs.end(), std::abs(truthParticle->pdgId())) != m_requiredParentPDGIDs.end())
137 {
138 acceptData.setCutResult (m_requiredParentIndex, true);
139 return acceptData;
140 }
141
142 return acceptData;
143 }
#define ATH_MSG_VERBOSE(x)
#define ATH_MSG_WARNING(x)
const xAOD::TruthParticle * getParent(const xAOD::TruthParticle *truthParticle, size_t index) const
Helper function get a parent by index.
int m_requiredParentIndex
Index for the required parent particles.
size_t getNParents(const xAOD::TruthParticle *truthParticle) const
Helper function to get the number of parent particles.
void setCutResult(const std::string &cutName, bool cutResult)
Set the result of a cut, based on the cut name (safer)
Definition AcceptData.h:134
int pdgId() const
PDG ID code.

◆ initialize()

StatusCode PMGTools::PMGDecayProductsSelectionTool::initialize ( void )
overridevirtual

Dummy implementation of the initialisation function.

It's here to allow the dual-use tools to skip defining an initialisation function. Since many are doing so...

Reimplemented from asg::AsgTool.

Definition at line 36 of file PMGDecayProductsSelectionTool.cxx.

38 {
39 if (m_requiredParentPDGIDs.empty())
40 {
41 ATH_MSG_ERROR ("required parent particles need to be set");
42 return StatusCode::FAILURE;
43 }
44
45 m_truthParticleIndex = m_accept.addCut ("truthParticle", "has truth particle cut");
46
47 std::ostringstream particles;
48 for (int pdgId : m_requiredParentPDGIDs)
49 {
50 particles << " " << pdgId;
51 }
52 ATH_MSG_DEBUG ("Performing required parent particle selection with parent PDG IDs:" << particles.str());
53 m_requiredParentIndex = m_accept.addCut ("requiredParents", "required parent particles cut");
54
55 if (!m_allowedIntermediatePDGIDs.empty())
56 {
57 std::ostringstream particles;
58 for (int pdgId : m_allowedIntermediatePDGIDs)
59 {
60 particles << " " << pdgId;
61 }
62 ATH_MSG_DEBUG ("Performing allowed intermediate particle selection with allowed PDG IDs:" << particles.str());
63 }
64
65 m_parentsAccessor = std::make_unique<const SG::AuxElement::Accessor<std::vector<ElementLink<xAOD::TruthParticleContainer>>>>("parentLinks");
66
67 return StatusCode::SUCCESS;
68 }
#define ATH_MSG_ERROR(x)
#define ATH_MSG_DEBUG(x)

◆ inputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< AlgTool > >::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.

◆ msg()

MsgStream & AthCommonMsg< AlgTool >::msg ( ) const
inlineinherited

Definition at line 24 of file AthCommonMsg.h.

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

◆ msg_level_name()

const std::string & asg::AsgTool::msg_level_name ( ) const
inherited

A deprecated function for getting the message level's name.

Instead of using this, weirdly named function, user code should get the string name of the current minimum message level (in case they really need it...), with:

MSG::name( msg().level() )

This function's name doesn't follow the ATLAS coding rules, and as such will be removed in the not too distant future.

Returns
The string name of the current minimum message level that's printed

Definition at line 101 of file AsgTool.cxx.

101 {
102
103 return MSG::name( msg().level() );
104 }
MsgStream & msg() const
const std::string & name(Level lvl)
Convenience function for translating message levels to strings.
Definition MsgLevel.cxx:19

◆ msgLvl()

bool AthCommonMsg< AlgTool >::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< AlgTool > >::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.

◆ print()

◆ 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< AlgTool > >::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< AlgTool > >::renounceArray ( SG::VarHandleKeyArray & handlesArray)
inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

364 {
366 }

◆ sysInitialize()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysInitialize ( )
overridevirtualinherited

Perform system initialization for an algorithm.

We override this to declare all the elements of handle key arrays at the end of initialization. See comments on updateVHKA.

Reimplemented in asg::AsgMetadataTool, AthCheckedComponent< AthAlgTool >, AthCheckedComponent<::AthAlgTool >, and DerivationFramework::CfAthAlgTool.

◆ sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::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< AlgTool > >::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 }
std::vector< SG::VarHandleKeyArray * > m_vhka

Member Data Documentation

◆ m_accept

asg::AcceptInfo PMGTools::PMGDecayProductsSelectionTool::m_accept
private

the AcceptInfo we are using

Definition at line 83 of file PMGDecayProductsSelectionTool.h.

◆ m_allowedIntermediatePDGIDs

std::vector<int> PMGTools::PMGDecayProductsSelectionTool::m_allowedIntermediatePDGIDs
private

Definition at line 71 of file PMGDecayProductsSelectionTool.h.

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_detStore
privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_evtStore
privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

◆ m_parentsAccessor

std::unique_ptr<const SG::AuxElement::Accessor<std::vector<ElementLink<xAOD::TruthParticleContainer> > > > PMGTools::PMGDecayProductsSelectionTool::m_parentsAccessor {}
private

common parents accessor

Definition at line 87 of file PMGDecayProductsSelectionTool.h.

87{};

◆ m_requiredParentIndex

int PMGTools::PMGDecayProductsSelectionTool::m_requiredParentIndex { -1 }
private

Index for the required parent particles.

Definition at line 79 of file PMGDecayProductsSelectionTool.h.

79{ -1 };

◆ m_requiredParentPDGIDs

std::vector<int> PMGTools::PMGDecayProductsSelectionTool::m_requiredParentPDGIDs
private

tool properties

Definition at line 70 of file PMGDecayProductsSelectionTool.h.

◆ m_truthParticleIndex

int PMGTools::PMGDecayProductsSelectionTool::m_truthParticleIndex { -1 }
private

Index for the truth particle link.

Definition at line 77 of file PMGDecayProductsSelectionTool.h.

77{ -1 };

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< AlgTool > >::m_varHandleArraysDeclared
privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vhka

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

Definition at line 398 of file AthCommonDataStore.h.


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