ATLAS Offline Software
Loading...
Searching...
No Matches
D3PD::TrackParticleVertexAssociationTool Class Referenceabstract

Associate from a TrackParticle to its reconstructed vertex. More...

#include <TrackParticleVertexAssociationTool.h>

Inheritance diagram for D3PD::TrackParticleVertexAssociationTool:
Collaboration diagram for D3PD::TrackParticleVertexAssociationTool:

Public Types

typedef SingleAssociationTool< Types< Rec::TrackParticle, xAOD::TrackParticle >, Types< Trk::VxCandidate, xAOD::Vertex > > Base

Public Member Functions

 TrackParticleVertexAssociationTool (const std::string &type, const std::string &name, const IInterface *parent)
 Standard Gaudi tool constructor.
virtual StatusCode initialize () override
 Standard Gaudi initialize method.
virtual StatusCode book () final
 book ntuple variables
virtual const Trk::VxCandidateget (const Rec::TrackParticle &p) override
 Return the target object.
virtual const xAOD::Vertexget (const xAOD::TrackParticle &p) override
 Return the target object.
virtual const void * getUntyped (const void *p)
 Return the target object.
virtual const std::type_info & fromTypeinfo () const
 Return the std::type_info for the source of the association.
virtual const Types< Trk::VxCandidate, xAOD::Vertex > * get (const Types< Rec::TrackParticle, xAOD::TrackParticle > &p)=0
 Return the target object.
virtual const std::type_info & typeinfo () const
 Return the type of object retrieved by this tool.
virtual void releaseObject (const typename SelectType< Types< Rec::TrackParticle, xAOD::TrackParticle >, 0 >::type *p)
 Release an object retrieved from the association.
virtual void releaseObject (const typename SelectType< TO_T, 0 >::type *p)
 Release an object retrieved from the association.
virtual void releaseObjectUntyped (const void *p)
 Release an object retrieved from the association.
virtual StatusCode configureD3PD (D3PD::IAddVariable *tree, const std::type_info &ti)
 Configure during initialization: type-check.
StatusCode configureD3PD (IAddVariable *tree)
 Configure the parent tree.
virtual StatusCode addVariable (const std::string &name, const std::type_info &ti, void *&ptr, const std::string &docstring="", const void *defval=0)
 Add a variable to the tuple.
StatusCode addVariable (const std::string &name, T *&ptr, const std::string &docstring="")
 Make the template implementation from IAddVariable visible.
StatusCode addVariable (const std::string &name, T *&ptr, const std::string &docstring, const U &defval)
 Make the template implementation from IAddVariable visible.
virtual StatusCode addDimensionedVariable (const std::string &name, const std::type_info &ti, void *&ptr, const std::string &dim, const std::string &docstring="", const void *defval=0)
 Add a variable to the tuple.
StatusCode addDimensionedVariable (const std::string &name, T *&ptr, const std::string &dim, const std::string &docstring="")
 Add a variable to the tuple.
StatusCode addDimensionedVariable (const std::string &name, T *&ptr, const std::string &dim, const std::string &docstring, const U &defval)
 Add a variable to the tuple.

Protected Member Functions

const void * doConversion (const void *p)
 Helper to convert pointers to source objects.
StatusCode configureMulti (D3PD::IAddVariable *tree, const std::type_info &ti, const std::vector< const std::type_info * > &tis, size_t &which)
 Configure during initialization: type-check.

Protected Attributes

ObjectMetadata m_metadata
 Metadata about the variables created by this object.

Private Attributes

std::string m_targetSGKey
 SG key for target collection.
bool m_fast
 flag to just return whatever is returned by TrackParticle::reconstructedVertex()
float * m_weight
 weight of the track at the associated vertex
SG::ReadHandleKey< xAOD::VertexContainerm_vertexKey { this, "VertexContainer", "PrimaryVertices", "primary vertex container" }
 xAOD vertex container handle key
std::string m_prefix
 Parameter: source object prefix.
std::string m_blockName
 Property: the name of this block.
TypeConverter m_inputConverter
 Helper to convert source object pointers.
const std::string & m_suffix
 Reference to the variable suffix.
IAddVariablem_tree
 Reference to the block name.

Static Private Attributes

static const std::string s_emptyString
 Used to provide a default value for blockName.

Detailed Description

Associate from a TrackParticle to its reconstructed vertex.

Definition at line 34 of file TrackParticleVertexAssociationTool.h.

Member Typedef Documentation

◆ Base

Constructor & Destructor Documentation

◆ TrackParticleVertexAssociationTool()

D3PD::TrackParticleVertexAssociationTool::TrackParticleVertexAssociationTool ( const std::string & type,
const std::string & name,
const IInterface * parent )

Standard Gaudi tool constructor.

Parameters
typeThe name of the tool type.
nameThe tool name.
parentThe tool's Gaudi parent.

Definition at line 31 of file TrackParticleVertexAssociationTool.cxx.

35 : Base (type, name, parent)
36{
37 declareProperty ("VxSGKey", m_targetSGKey="VxPrimaryCandidate");
38 declareProperty ("Fast", m_fast = false);
39
40 book().ignore(); // Avoid coverity warnings.
41}
std::string m_targetSGKey
SG key for target collection.
bool m_fast
flag to just return whatever is returned by TrackParticle::reconstructedVertex()
SingleAssociationTool< Types< Rec::TrackParticle, xAOD::TrackParticle >, Types< Trk::VxCandidate, xAOD::Vertex > > Base
virtual StatusCode book() final
book ntuple variables

Member Function Documentation

◆ addDimensionedVariable() [1/3]

StatusCode D3PD::AddVariable::addDimensionedVariable ( const std::string & name,
const std::type_info & ti,
void *& ptr,
const std::string & dim,
const std::string & docstring = "",
const void * defval = 0 )
virtualinherited

Add a variable to the tuple.

Parameters
nameThe name of the variable.
typeThe type of the variable.
ptrPointer to the type of the variable. The pointer need not be initialized; the D3PD software will set the pointer prior to calling fill().
dimDimension for the variable. (Presently unimplemented!)
docstringDocumentation string for this variable.
defvalPointer to the default value to use for this variable. Null for no default (generally means to fill with zeros). Of the type given by ti. Only works for basic types.

If called from the constructor, the only effect is to clear ptr.

Definition at line 119 of file AddVariable.cxx.

129{
130 if (!m_tree) {
131 // Called from constructor --- just initialize pointer.
132 ptr = 0;
133 return StatusCode::SUCCESS;
134 }
135
137 ti, ptr, dim,
138 docstring, defval ) );
140 ti, ptr, dim,
142}
#define CHECK(...)
Evaluate an expression and check for errors.
virtual StatusCode addDimensionedVariable(const std::string &name, const std::type_info &ti, void *&ptr, const std::string &dim, const std::string &docstring="", const void *defval=0)
Add a variable to the tuple.
Type-safe wrapper for single associator tools.

◆ addDimensionedVariable() [2/3]

StatusCode D3PD::IAddVariable::addDimensionedVariable ( const std::string & name,
T *& ptr,
const std::string & dim,
const std::string & docstring,
const U & defval )
inherited

Add a variable to the tuple.

Parameters
nameThe name of the variable.
ptrPointer to the type of the variable. The pointer need not be initialized; the D3PD software will set the pointer prior to calling fill().
dimDimension for the variable. (Presently unimplemented!)
docstringDocumentation string for this variable.
defvalPointer to the default value to use for this variable. Only works for basic types.

If called from the constructor, the only effect is to clear ptr.

◆ addDimensionedVariable() [3/3]

StatusCode D3PD::IAddVariable::addDimensionedVariable ( const std::string & name,
T *& ptr,
const std::string & dim,
const std::string & docstring = "" )
inherited

Add a variable to the tuple.

Parameters
nameThe name of the variable.
ptrPointer to the type of the variable. The pointer need not be initialized; the D3PD software will set the pointer prior to calling fill().
dimDimension for the variable. (Presently unimplemented!)
docstringDocumentation string for this variable.

If called from the constructor, the only effect is to clear ptr.

◆ addVariable() [1/3]

StatusCode D3PD::AddVariable::addVariable ( const std::string & name,
const std::type_info & ti,
void *& ptr,
const std::string & docstring = "",
const void * defval = 0 )
virtualinherited

Add a variable to the tuple.

Parameters
nameThe name of the variable.
typeThe type of the variable.
ptrPointer to the type of the variable. The pointer need not be initialized; the D3PD software will set the pointer prior to calling fill().
docstringDocumentation string for this variable.
defvalPointer to the default value to use for this variable. Null for no default (generally means to fill with zeros). Of the type given by ti. Only works for basic types.

If called from the constructor, the only effect is to clear ptr.

Definition at line 93 of file AddVariable.cxx.

90{
91 if (!m_tree) {
92 // Called from constructor --- just initialize pointer.
93 ptr = 0;
95 }
96
98 ti, ptr, docstring, defval ) );
101}
virtual StatusCode addVariable(const std::string &name, const std::type_info &ti, void *&ptr, const std::string &docstring="", const void *defval=0)
Add a variable to the tuple.

◆ addVariable() [2/3]

StatusCode D3PD::IAddVariable::addVariable ( const std::string & name,
T *& ptr,
const std::string & docstring,
const U & defval )
inherited

Make the template implementation from IAddVariable visible.

◆ addVariable() [3/3]

StatusCode D3PD::IAddVariable::addVariable ( const std::string & name,
T *& ptr,
const std::string & docstring = "" )
inherited

Make the template implementation from IAddVariable visible.

◆ book()

StatusCode D3PD::TrackParticleVertexAssociationTool::book ( )
finalvirtual

book ntuple variables

Reimplemented from D3PD::SingleAssociationToolImpl.

Definition at line 55 of file TrackParticleVertexAssociationTool.cxx.

◆ configureD3PD() [1/2]

StatusCode D3PD::SingleAssociationToolImpl::configureD3PD ( D3PD::IAddVariable * tree,
const std::type_info & ti )
virtualinherited

Configure during initialization: type-check.

Parameters
treeOur parent for tuple making.
tiGives the type of the object being passed to getUntyped.

configureD3PD should check that the type of the object coming as input (to getUntyped) is compatible with what it expects, and raise an error otherwise.

Parameters
treeOur parent for tuple making.
tiGives the type of the object being passed to resetUntyped.

configureD3PD should check that the type of the object coming as input (to getUntyped) is compatible with what it expects, and raise an error otherwise.

Definition at line 69 of file SingleAssociationToolImpl.cxx.

◆ configureD3PD() [2/2]

StatusCode D3PD::AddVariable::configureD3PD ( IAddVariable * tree)
inherited

Configure the parent tree.

Parameters
treeThe parent IAddVariable instance.

Definition at line 69 of file AddVariable.cxx.

62{
63 m_tree = tree;
64 m_metadata.clear();
66}

◆ configureMulti()

StatusCode D3PD::SingleAssociationToolImpl::configureMulti ( D3PD::IAddVariable * tree,
const std::type_info & ti,
const std::vector< const std::type_info * > & tis,
size_t & which )
protectedinherited

Configure during initialization: type-check.

Alternate implementation for the case where we have to choose among multiple possible input types.

Parameters
treeOur parent for tuple making.
tiGives the type of the object being passed to getUntyped.
tisList of possible input types that we can accept.
[out]whichIndex of the accepted type.

configureMulti should check that the type of the object coming as input (to getUntyped) is compatible with what it expects, and raise an error otherwise.

Definition at line 110 of file SingleAssociationToolImpl.cxx.

◆ doConversion()

const void * D3PD::SingleAssociationToolImpl::doConversion ( const void * p)
protectedinherited

Helper to convert pointers to source objects.

Parameters
pSource object, as a pointer to the type passed to configureD3PD. return Source object, as a pointer to the type given by fromTypeinfo.
pSource object, as a pointer to the type passed to configureD3PD. return Source object, as a pointer to the type given by typeinfoFrom.

Definition at line 94 of file SingleAssociationToolImpl.cxx.

104{
105 if (!p) return 0;
106 const void* pp = m_inputConverter.convertUntyped (p);
107 if (!pp) {
109 << "Pointer conversion from " << m_inputConverter.srcName() << " to "
110 << m_inputConverter.dstName() << " failed.";
111 return 0;
112 }
113 return pp;
114}
#define REPORT_MESSAGE(LVL)
Report a message.

◆ fromTypeinfo()

virtual const std::type_info & D3PD::SingleAssociationTool< Types< Rec::TrackParticle, xAOD::TrackParticle >, Types< Trk::VxCandidate, xAOD::Vertex > >::fromTypeinfo ( ) const
virtualinherited

Return the std::type_info for the source of the association.

Implements D3PD::SingleAssociationToolImpl.

◆ get() [1/3]

Return the target object.

Parameters
pThe source object for the association.

Return the target of the association, or 0.

◆ get() [2/3]

const Trk::VxCandidate * D3PD::TrackParticleVertexAssociationTool::get ( const Rec::TrackParticle & p)
overridevirtual

Return the target object.

Parameters
pThe source object for the association.

Return the target of the association, or 0.

Definition at line 68 of file TrackParticleVertexAssociationTool.cxx.

68 {
69
70 if(m_fast){
71 return p.reconstructedVertex();
72 }
73
74 *m_weight = 0;
75
76 const VxContainer* vxContainer = 0;
77 StatusCode sc = evtStore()->retrieve(vxContainer, m_targetSGKey);
78 if (sc.isFailure() || !vxContainer) {
79 REPORT_MESSAGE (MSG::WARNING) << "Could not retrieve primary vertex container: " << m_targetSGKey;
80 return 0;
81 }
82
83 int nVx = vxContainer->size();
84 if(nVx < 1){
85 return 0;
86 }
87
88 for(int v=0; v<nVx-1; v++){
89 const Trk::VxCandidate* vx = vxContainer->at(v);
90 const std::vector<Trk::VxTrackAtVertex*>* tracks = vx->vxTrackAtVertex();
91 int nTracks = 0;
92 if(tracks) nTracks = tracks->size();
93 for(int t=0; t<nTracks; t++) {
94 Trk::VxTrackAtVertex* track = tracks->at(t);
95 if(track){
96 Trk::ITrackLink* link = track->trackOrParticleLink();
97 if (link) {
98 Trk::LinkToTrackParticleBase* linkToTrackParticle = dynamic_cast<Trk::LinkToTrackParticleBase*>(link);
99 if(linkToTrackParticle){
100 const Trk::TrackParticleBase *particle_base = **linkToTrackParticle;
101 if(particle_base == &p){
102 *m_weight = track->weight();
103 return vx;
104 }
105 }
106 }
107 }
108 }
109 }
110
111 return vxContainer->at(nVx-1);
112}
static Double_t sc
Athena::TPCnvVers::Old VxContainer
const T * at(size_type n) const
Access an element, as an rvalue.
size_type size() const noexcept
Returns the number of elements in the collection.
std::vector< Trk::VxTrackAtVertex * > * vxTrackAtVertex(void)
Unconst pointer to the vector of tracks Required by some of the vertex fitters.
::StatusCode StatusCode
StatusCode definition for legacy code.

◆ get() [3/3]

const xAOD::Vertex * D3PD::TrackParticleVertexAssociationTool::get ( const xAOD::TrackParticle & p)
overridevirtual

Return the target object.

Parameters
pThe source object for the association.

Return the target of the association, or 0.

Definition at line 122 of file TrackParticleVertexAssociationTool.cxx.

123{
124 const xAOD::Vertex* foundVertex { nullptr };
125 SG::ReadHandle<xAOD::VertexContainer> vertices { m_vertexKey };
126 if (!vertices.isValid())
127 {
128 ATH_MSG_WARNING("No VertexContainer with key = " << m_vertexKey.key() );
129 return foundVertex;
130 }
131 for ( const auto vx : *vertices )
132 {
133 for ( const auto& tpLink : vx->trackParticleLinks() )
134 {
135 if ( *tpLink == &p )
136 {
137 foundVertex = vx;
138 break;
139 }
140 }
141 if (foundVertex) break;
142 }
143 return foundVertex;
144}
#define ATH_MSG_WARNING(x)
SG::ReadHandleKey< xAOD::VertexContainer > m_vertexKey
xAOD vertex container handle key
virtual bool isValid() override final
Can the handle be successfully dereferenced?
Vertex_v1 Vertex
Define the latest version of the vertex class.

◆ getUntyped()

virtual const void * D3PD::SingleAssociationTool< Types< Rec::TrackParticle, xAOD::TrackParticle >, Types< Trk::VxCandidate, xAOD::Vertex > >::getUntyped ( const void * p)
virtualinherited

Return the target object.

Parameters
pThe source object for the association.

Return the target of the association, or 0. Should be of the type given by typeinfo.

◆ initialize()

StatusCode D3PD::TrackParticleVertexAssociationTool::initialize ( )
overridevirtual

Standard Gaudi initialize method.

Definition at line 46 of file TrackParticleVertexAssociationTool.cxx.

47{
48 ATH_CHECK( m_vertexKey.initialize() );
49 return Base::initialize();
50}
#define ATH_CHECK
Evaluate an expression and check for errors.

◆ releaseObject() [1/2]

virtual void D3PD::SingleAssociationToolTo< Types< Rec::TrackParticle, xAOD::TrackParticle > >::releaseObject ( const typename SelectType< Types< Rec::TrackParticle, xAOD::TrackParticle >, 0 >::type * p)
virtualinherited

Release an object retrieved from the association.

Parameters
pThe object to release.

Call this when you are done with the object returned by get(). The default implementation is a no-op, but if the association dynamically allocated the object which it returned, this gives it a chance to free it.

◆ releaseObject() [2/2]

template<typename TO_T>
virtual void D3PD::SingleAssociationToolTo< TO_T >::releaseObject ( const typename SelectType< TO_T, 0 >::type * p)
virtualinherited

Release an object retrieved from the association.

Parameters
pThe object to release.

Call this when you are done with the object returned by get(). The default implementation is a no-op, but if the association dynamically allocated the object which it returned, this gives it a chance to free it.

Reimplemented in D3PD::SingleAssociationTool< Types< T0 >, TO_T >.

◆ releaseObjectUntyped()

virtual void D3PD::SingleAssociationToolTo< Types< Rec::TrackParticle, xAOD::TrackParticle > >::releaseObjectUntyped ( const void * p)
virtualinherited

Release an object retrieved from the association.

Parameters
pThe object to release.

Call this when you are done with the object returned by getUntyped(). The default implementation is a no-op, but if the association dynamically allocated the object which it returned, this gives it a chance to free it.

◆ typeinfo()

virtual const std::type_info & D3PD::SingleAssociationToolTo< Types< Rec::TrackParticle, xAOD::TrackParticle > >::typeinfo ( ) const
virtualinherited

Return the type of object retrieved by this tool.

Member Data Documentation

◆ m_blockName

std::string D3PD::SingleAssociationToolImpl::m_blockName
privateinherited

Property: the name of this block.

Definition at line 121 of file SingleAssociationToolImpl.h.

◆ m_fast

bool D3PD::TrackParticleVertexAssociationTool::m_fast
private

flag to just return whatever is returned by TrackParticle::reconstructedVertex()

Definition at line 81 of file TrackParticleVertexAssociationTool.h.

◆ m_inputConverter

TypeConverter D3PD::SingleAssociationToolImpl::m_inputConverter
privateinherited

Helper to convert source object pointers.

Definition at line 124 of file SingleAssociationToolImpl.h.

◆ m_metadata

ObjectMetadata D3PD::AddVariable::m_metadata
protectedinherited

Metadata about the variables created by this object.

Definition at line 129 of file AddVariable.h.

◆ m_prefix

std::string D3PD::SingleAssociationToolImpl::m_prefix
privateinherited

Parameter: source object prefix.

Definition at line 118 of file SingleAssociationToolImpl.h.

◆ m_suffix

const std::string& D3PD::AddVariable::m_suffix
privateinherited

Reference to the variable suffix.

Definition at line 137 of file AddVariable.h.

◆ m_targetSGKey

std::string D3PD::TrackParticleVertexAssociationTool::m_targetSGKey
private

SG key for target collection.

Definition at line 78 of file TrackParticleVertexAssociationTool.h.

◆ m_tree

IAddVariable* D3PD::AddVariable::m_tree
privateinherited

Reference to the block name.

The parent IAddVariable instance.

Definition at line 143 of file AddVariable.h.

◆ m_vertexKey

SG::ReadHandleKey<xAOD::VertexContainer> D3PD::TrackParticleVertexAssociationTool::m_vertexKey { this, "VertexContainer", "PrimaryVertices", "primary vertex container" }
private

xAOD vertex container handle key

Definition at line 87 of file TrackParticleVertexAssociationTool.h.

87{ this, "VertexContainer", "PrimaryVertices", "primary vertex container" };

◆ m_weight

float* D3PD::TrackParticleVertexAssociationTool::m_weight
private

weight of the track at the associated vertex

Definition at line 84 of file TrackParticleVertexAssociationTool.h.

◆ s_emptyString

const std::string D3PD::AddVariable::s_emptyString
staticprivateinherited

Used to provide a default value for blockName.

Definition at line 41 of file AddVariable.h.


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