ATLAS Offline Software
Loading...
Searching...
No Matches
Trk::TruthToTrack Class Reference

#include <TruthToTrack.h>

Inheritance diagram for Trk::TruthToTrack:
Collaboration diagram for Trk::TruthToTrack:

Public Member Functions

 TruthToTrack (const std::string &type, const std::string &name, const IInterface *parent)
virtual StatusCode initialize ()
virtual const Trk::TrackParametersmakeProdVertexParameters (HepMC::ConstGenParticlePtr part) const
 This function produces a Trk::TrackParameters object corresponding to the HepMC::GenParticle at the production vertex.
virtual const Trk::TrackParametersmakeProdVertexParameters (const xAOD::TruthParticle *part) const
virtual const Trk::TrackParametersmakePerigeeParameters (HepMC::ConstGenParticlePtr part) const
 This function extrapolates track to the perigee, and returns perigee parameters.
virtual const Trk::TrackParametersmakePerigeeParameters (const xAOD::TruthParticle *part) 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 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

Static Public Member Functions

static const InterfaceID & interfaceID ()

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

Gaudi::Details::PropertyBase & declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
 specialization for handling Gaudi::Property<SG::VarHandleKey>

Private Attributes

ToolHandle< Trk::IExtrapolatorm_extrapolator
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 27 of file TruthToTrack.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

◆ TruthToTrack()

Trk::TruthToTrack::TruthToTrack ( const std::string & type,
const std::string & name,
const IInterface * parent )

Definition at line 23 of file TruthToTrack.cxx.

24 : ::AthAlgTool(type,name,parent)
25 , m_extrapolator("Trk::Extrapolator/AtlasExtrapolator")
26{
27 declareInterface<ITruthToTrack>(this);
28 declareProperty("Extrapolator", m_extrapolator);
29}
AthAlgTool()
Default constructor:
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T, V, H > &t)
ToolHandle< Trk::IExtrapolator > m_extrapolator

Member Function Documentation

◆ 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

◆ initialize()

StatusCode Trk::TruthToTrack::initialize ( )
virtual

Definition at line 32 of file TruthToTrack.cxx.

32 {
33 ATH_CHECK( m_extrapolator.retrieve() );
34 return StatusCode::SUCCESS;
35}
#define ATH_CHECK
Evaluate an expression and check for errors.

◆ 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.

◆ interfaceID()

const InterfaceID & Trk::ITruthToTrack::interfaceID ( )
inlinestaticinherited

Definition at line 40 of file ITruthToTrack.h.

40 {
41 static const InterfaceID IID_ITruthToTrack("Trk::ITruthToTrack",1,0);
42 return IID_ITruthToTrack;
43 }

◆ makePerigeeParameters() [1/2]

const Trk::TrackParameters * Trk::TruthToTrack::makePerigeeParameters ( const xAOD::TruthParticle * part) const
virtual

Implements Trk::ITruthToTrack.

Definition at line 125 of file TruthToTrack.cxx.

125 {
126 const Trk::TrackParameters* generatedTrackPerigee = nullptr;
127
128 if(part && part->hasProdVtx() && m_extrapolator) {
129
130 std::unique_ptr<const Trk::TrackParameters> productionVertexTrackParams( makeProdVertexParameters(part) );
131 if(productionVertexTrackParams) {
132
133 // Extrapolate the TrackParameters object to the perigee. Direct extrapolation,
134 // no material effects.
135 generatedTrackPerigee = m_extrapolator->extrapolateDirectly(
136 Gaudi::Hive::currentContext(),
137 *productionVertexTrackParams,
138 Trk::PerigeeSurface(),
140 false,
141 Trk::nonInteracting ).release();
142 }
143 }
144
145 return generatedTrackPerigee;
146}
virtual const Trk::TrackParameters * makeProdVertexParameters(HepMC::ConstGenParticlePtr part) const
This function produces a Trk::TrackParameters object corresponding to the HepMC::GenParticle at the p...
@ anyDirection
ParametersBase< TrackParametersDim, Charged > TrackParameters

◆ makePerigeeParameters() [2/2]

const Trk::TrackParameters * Trk::TruthToTrack::makePerigeeParameters ( HepMC::ConstGenParticlePtr part) const
virtual

This function extrapolates track to the perigee, and returns perigee parameters.

The return value points to an object allocated with the operator new. The caller is responsible for deletion of the object.

Implements Trk::ITruthToTrack.

Definition at line 99 of file TruthToTrack.cxx.

99 {
100 const Trk::TrackParameters* generatedTrackPerigee = nullptr;
101
102 if(part && part->production_vertex() && m_extrapolator) {
103
104 std::unique_ptr<const Trk::TrackParameters> productionVertexTrackParams( makeProdVertexParameters(part) );
105 if(productionVertexTrackParams) {
106
107 // Extrapolate the TrackParameters object to the perigee. Direct extrapolation,
108 // no material effects.
109 generatedTrackPerigee = m_extrapolator->extrapolateDirectly(
110 Gaudi::Hive::currentContext(),
111 *productionVertexTrackParams,
112 Trk::PerigeeSurface(),
114 false,
115 Trk::nonInteracting ).release();
116 }
117 }
118
119 return generatedTrackPerigee;
120}

◆ makeProdVertexParameters() [1/2]

const Trk::TrackParameters * Trk::TruthToTrack::makeProdVertexParameters ( const xAOD::TruthParticle * part) const
virtual

Implements Trk::ITruthToTrack.

Definition at line 70 of file TruthToTrack.cxx.

70 {
72
73 if(part && part->hasProdVtx()) {
74 Amg::Vector3D hv(part->prodVtx()->x(),part->prodVtx()->y(),part->prodVtx()->z());
75 const Amg::Vector3D& globalPos = hv;
76
77 Amg::Vector3D hv2(part->p4().Px(),part->p4().Py(),part->p4().Pz());
78 const Amg::Vector3D& globalMom = hv2;
79
80 const int id = part->pdg_id();
81 if (id) {
82 const double charge = MC::charge(id);
83 Amg::Translation3D tmpTransl(hv);
84 Amg::Transform3D tmpTransf = tmpTransl * Amg::RotationMatrix3D::Identity();
85 const Trk::PlaneSurface surface(tmpTransf);
86 result = new Trk::AtaPlane(globalPos, globalMom, charge, surface);
87 }
88 else {
89 ATH_MSG_WARNING("Could not get particle data for particle ID="<<id);
90 }
91 }
92
93 return result;
94}
#define ATH_MSG_WARNING(x)
double charge(const T &p)
Definition AtlasPID.h:997
Eigen::Affine3d Transform3D
Eigen::Matrix< double, 3, 1 > Vector3D
Eigen::Translation< double, 3 > Translation3D
double charge(const T &p)
ParametersT< TrackParametersDim, Charged, PlaneSurface > AtaPlane

◆ makeProdVertexParameters() [2/2]

const Trk::TrackParameters * Trk::TruthToTrack::makeProdVertexParameters ( HepMC::ConstGenParticlePtr part) const
virtual

This function produces a Trk::TrackParameters object corresponding to the HepMC::GenParticle at the production vertex.

The object is allocated using operator new and returned by pointer. The caller is responsible for deletion of the object. (Wrapping it into a smart pointer may be the most convenient way to make sure the memory is freed.)

Implements Trk::ITruthToTrack.

Definition at line 40 of file TruthToTrack.cxx.

40 {
42
43 if(part && part->production_vertex()) {
44 HepMC::FourVector tv = part->production_vertex()->position();
45 Amg::Vector3D hv(tv.x(),tv.y(),tv.z());
46 const Amg::Vector3D& globalPos = hv;
47
48 const HepMC::FourVector& fv = part->momentum();
49 Amg::Vector3D hv2(fv.px(),fv.py(),fv.pz());
50 const Amg::Vector3D& globalMom = hv2;
51
52 const int id = part->pdg_id();
53 if (id) {
54 const double charge = MC::charge(id);
55 Amg::Translation3D tmpTransl(hv);
56 Amg::Transform3D tmpTransf = tmpTransl * Amg::RotationMatrix3D::Identity();
57 const Trk::PlaneSurface surface(tmpTransf);
58 result = new Trk::AtaPlane(globalPos, globalMom, charge, surface);
59 }
60 else {
61 ATH_MSG_WARNING("Could not get particle data for particle ID="<<id);
62 }
63 }
64
65 return result;
66}

◆ msg()

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

Definition at line 24 of file AthCommonMsg.h.

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

◆ 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.

◆ 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_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_extrapolator

ToolHandle<Trk::IExtrapolator> Trk::TruthToTrack::m_extrapolator
private

Definition at line 52 of file TruthToTrack.h.

◆ 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: