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 24 of file TruthToTrack.cxx.

25 : ::AthAlgTool(type,name,parent)
26 , m_extrapolator("Trk::Extrapolator/AtlasExtrapolator")
27{
28 declareInterface<ITruthToTrack>(this);
29 declareProperty("Extrapolator", m_extrapolator);
30}
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 33 of file TruthToTrack.cxx.

33 {
34 ATH_CHECK( m_extrapolator.retrieve() );
35 return StatusCode::SUCCESS;
36}
#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 126 of file TruthToTrack.cxx.

126 {
127 const Trk::TrackParameters* generatedTrackPerigee = nullptr;
128
129 if(part && part->hasProdVtx() && m_extrapolator) {
130
131 std::unique_ptr<const Trk::TrackParameters> productionVertexTrackParams( makeProdVertexParameters(part) );
132 if(productionVertexTrackParams) {
133
134 // Extrapolate the TrackParameters object to the perigee. Direct extrapolation,
135 // no material effects.
136 generatedTrackPerigee = m_extrapolator->extrapolateDirectly(
137 Gaudi::Hive::currentContext(),
138 *productionVertexTrackParams,
139 Trk::PerigeeSurface(),
141 false,
142 Trk::nonInteracting ).release();
143 }
144 }
145
146 return generatedTrackPerigee;
147}
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 100 of file TruthToTrack.cxx.

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

◆ makeProdVertexParameters() [1/2]

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

Implements Trk::ITruthToTrack.

Definition at line 71 of file TruthToTrack.cxx.

71 {
73
74 if(part && part->hasProdVtx()) {
75 Amg::Vector3D hv(part->prodVtx()->x(),part->prodVtx()->y(),part->prodVtx()->z());
76 const Amg::Vector3D& globalPos = hv;
77
78 Amg::Vector3D hv2(part->p4().Px(),part->p4().Py(),part->p4().Pz());
79 const Amg::Vector3D& globalMom = hv2;
80
81 const int id = part->pdg_id();
82 if (id) {
83 const double charge = MC::charge(id);
84 Amg::Translation3D tmpTransl(hv);
85 Amg::Transform3D tmpTransf = tmpTransl * Amg::RotationMatrix3D::Identity();
86 const Trk::PlaneSurface surface(tmpTransf);
87 result = new Trk::AtaPlane(globalPos, globalMom, charge, surface);
88 }
89 else {
90 ATH_MSG_WARNING("Could not get particle data for particle ID="<<id);
91 }
92 }
93
94 return result;
95}
#define ATH_MSG_WARNING(x)
double charge(const T &p)
Definition AtlasPID.h:1003
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 41 of file TruthToTrack.cxx.

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

◆ 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 >, and AthCheckedComponent<::AthAlgTool >.

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