ATLAS Offline Software
Loading...
Searching...
No Matches
TrkToLeptonPVTool Class Reference

#include <TrkToLeptonPVTool.h>

Inheritance diagram for TrkToLeptonPVTool:
Collaboration diagram for TrkToLeptonPVTool:

Public Member Functions

 TrkToLeptonPVTool (const std::string &type, const std::string &name, const IInterface *parent)
virtual ~TrkToLeptonPVTool ()
virtual StatusCode initialize () override
virtual StatusCode finalize () override
virtual std::unique_ptr< xAOD::VertexmatchTrkToPV (const xAOD::TrackParticle *trk, const xAOD::Vertex *PV, const xAOD::EventInfo *=0) const override final
 Method to match any tracks to a known Primary Vertex.
virtual std::unique_ptr< xAOD::VertexnpartVertex (const std::vector< const xAOD::TrackParticle * > &, const xAOD::EventInfo *=0) const override final
 Method to create a Primary Vertex using N(>=2) identified tracks (e.g.
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

SG::ReadCondHandleKey< InDet::BeamSpotDatam_beamSpotKey { this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot" }
ToolHandle< Trk::IVertexFitterm_fitterSvc {this,"VertexFitter","Trk::TrkVKalVrtFitter/VertexFitterTool","Vertex Fitter tool instance"}
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 30 of file TrkToLeptonPVTool.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

◆ TrkToLeptonPVTool()

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

Definition at line 18 of file TrkToLeptonPVTool.cxx.

20 :
21 AthAlgTool(type,name,parent)
22 {
23 declareInterface<ITrkToLeptonPV>(this);
24 }
AthAlgTool()
Default constructor:

◆ ~TrkToLeptonPVTool()

TrkToLeptonPVTool::~TrkToLeptonPVTool ( )
virtual

Definition at line 27 of file TrkToLeptonPVTool.cxx.

27 {
28 ATH_MSG_DEBUG("TrkToLeptonPVTool destructor called");
29 }
#define ATH_MSG_DEBUG(x)

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 }
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T, V, H > &t)

◆ 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

◆ finalize()

StatusCode TrkToLeptonPVTool::finalize ( )
overridevirtual

Definition at line 46 of file TrkToLeptonPVTool.cxx.

47 {
48 ATH_MSG_DEBUG(" finalize()");
49 return StatusCode::SUCCESS;
50 }

◆ initialize()

StatusCode TrkToLeptonPVTool::initialize ( )
overridevirtual

Definition at line 32 of file TrkToLeptonPVTool.cxx.

32 {
33 //-----
34 if (m_fitterSvc.retrieve().isFailure()) {
35 ATH_MSG_DEBUG("Can't find Trk::TrkVKalVrtFitter");
36 return StatusCode::FAILURE;
37 } else {
38 ATH_MSG_DEBUG("TrkToLeptonPVTool TrkVKalVrtFitter found");
39 }
40 //-----
41 ATH_CHECK(m_beamSpotKey.initialize());
42 //-----
43 return StatusCode::SUCCESS;
44 }
#define ATH_CHECK
Evaluate an expression and check for errors.
SG::ReadCondHandleKey< InDet::BeamSpotData > m_beamSpotKey
ToolHandle< Trk::IVertexFitter > m_fitterSvc

◆ 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 & ITrkToLeptonPV::interfaceID ( )
inlinestaticinherited

Definition at line 36 of file ITrkToLeptonPV.h.

36{ return IID_TrkToLeptonPV;}
static const InterfaceID IID_TrkToLeptonPV("ITrkToLeptonPV", 1, 0)
Interface for PV creation based on identified tracks.

◆ matchTrkToPV()

std::unique_ptr< xAOD::Vertex > TrkToLeptonPVTool::matchTrkToPV ( const xAOD::TrackParticle * ,
const xAOD::Vertex * ,
const xAOD::EventInfo * = 0 ) const
finaloverridevirtual

Method to match any tracks to a known Primary Vertex.

A combined PV+track vertex is returned. Track compatibility to PV can be estimated with Chi2 of the PV+track vertex (NDF=2)

Implements ITrkToLeptonPV.

Definition at line 53 of file TrkToLeptonPVTool.cxx.

55 {
56 static const SG::ConstAccessor<float> vyAcc("vy");
57 if(vyAcc.isAvailable(*trk)) {
58 std::vector<const xAOD::TrackParticle *> tpv(1,trk);
59 return std::unique_ptr<xAOD::Vertex>(m_fitterSvc->fit(tpv,(*PV)));
60 }
61
62 //---DAOD case
63 if( !eventINFO ) return {nullptr};
64 std::unique_ptr< SG::AuxStoreInternal > pAux;
66 std::vector<const xAOD::TrackParticle*> wrkTrkC(1);
67 pAux = std::make_unique< SG::AuxStoreInternal >();
68 TPC.setStore( pAux.get() );
69 TPC.reserve( 1 );
70 TPC.push_back(new (std::nothrow) xAOD::TrackParticle(*trk));
71 if(!TPC[0])return {nullptr};
72 const EventContext& ctx = Gaudi::Hive::currentContext();
73 const float mvx= (eventINFO) ? eventINFO->beamPosX() : 0.;
74 const float mvy= (eventINFO) ? eventINFO->beamPosY() : 0.;
75 static const SG::ConstAccessor<float> vzAcc("vz");
76 const float mvz= (vzAcc.isAvailable(*trk)) ? trk->vz() : 0.;
77 TPC[0]->setParametersOrigin( mvx, mvy, mvz);
78 wrkTrkC[0]=TPC[0];
79 return std::unique_ptr<xAOD::Vertex>(m_fitterSvc->fit(ctx,wrkTrkC,(*PV)));
80
81 }
void reserve(size_type n)
Attempt to preallocate enough memory for a specified number of elements.
value_type push_back(value_type pElem)
Add an element to the end of the collection.
TrackParticle_v1 TrackParticle
Reference the current persistent version:
TrackParticleContainer_v1 TrackParticleContainer
Definition of the current "TrackParticle container version".

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

◆ npartVertex()

std::unique_ptr< xAOD::Vertex > TrkToLeptonPVTool::npartVertex ( const std::vector< const xAOD::TrackParticle * > & ,
const xAOD::EventInfo * = 0 ) const
finaloverridevirtual

Method to create a Primary Vertex using N(>=2) identified tracks (e.g.

leptons) and beam constraint, The obtained N-track PV can be used as input for matchTrkToPV() to check another track. However, better way to check another track compatibility is to call again npartVertex() with N+1 tracks and check the obtained Chi2 (NDF=2*N_trk-1)

Implements ITrkToLeptonPV.

Definition at line 84 of file TrkToLeptonPVTool.cxx.

86 {
87 if(particles.empty()) return {nullptr};
88
89 std::vector<const xAOD::TrackParticle*> tmpp(particles);
90 std::sort(tmpp.begin(),tmpp.end());
91 auto tst=std::unique(tmpp.begin(),tmpp.end());
92 if( tst != tmpp.end()) {
93 ATH_MSG_DEBUG(" Duplicated particles on input!");
94 return {nullptr};
95 }
96
97 static const SG::ConstAccessor<float> vyAcc("vy");
98 bool fullxAOD=false; if(vyAcc.isAvailable(*particles[0])) fullxAOD=true;
99
100 xAOD::Vertex BEAM;
101 BEAM.makePrivateStore();
102 BEAM.setX(0.); BEAM.setY(0.); BEAM.setZ(0.);
103 std::vector<float> defaultCovar {0.015f*0.015f,0.f,0.015f*0.015f,0.f,0.f,1.e6};
104 BEAM.setCovariance(defaultCovar);
105 float beamtiltX=0.;
106 float beamtiltY=0.;
107 AmgSymMatrix(3) beamcov;
108 beamcov.setIdentity();
109 //-------------------------------
110 if(eventINFO){
111 BEAM.setX(eventINFO->beamPosX());
112 BEAM.setY(eventINFO->beamPosY());
113 BEAM.setZ(eventINFO->beamPosZ());
114 beamcov(0,0) = eventINFO->beamPosSigmaX() * eventINFO->beamPosSigmaX();
115 beamcov(1,0) = beamcov(0,1) = eventINFO->beamPosSigmaXY();
116 beamcov(1,1) = eventINFO->beamPosSigmaY() * eventINFO->beamPosSigmaY();
117 beamcov(2,2) = eventINFO->beamPosSigmaZ() * eventINFO->beamPosSigmaZ()*1.e6; //Remove any constraint in Z direction
118 BEAM.setCovariancePosition(beamcov);
119 beamtiltX= eventINFO->beamTiltXZ();
120 beamtiltY= eventINFO->beamTiltYZ();
121 }
122 SG::ReadCondHandle<InDet::BeamSpotData> beamSpotHandle { m_beamSpotKey };
123
124 if(beamSpotHandle.isValid() && fullxAOD ){
125 ATH_MSG_DEBUG("Beam service is present");
126 BEAM.setPosition(beamSpotHandle->beamVtx().position());
127 beamcov = beamSpotHandle->beamVtx().covariancePosition();
128 beamcov(2,2) *= 1.e6; //Remove any constraint in Z direction
129 BEAM.setCovariancePosition(beamcov);
130 beamtiltX= beamSpotHandle->beamTilt(0);
131 beamtiltY= beamSpotHandle->beamTilt(1);
132 }
133 if(fullxAOD){ ATH_MSG_DEBUG("xAOD data"); }
134 else { ATH_MSG_DEBUG("DxAOD data");}
135 ATH_MSG_DEBUG("BEAM x,y,z="<<BEAM.x()<<","<<BEAM.y()<<","<<BEAM.z());
136 ATH_MSG_DEBUG("BEAM covariance="<<BEAM.covariance()[0]<<","<<BEAM.covariance()[1]<<","<<BEAM.covariance()[2]
137 <<","<<BEAM.covariance()[3]<<","<<BEAM.covariance()[4]<<","<<BEAM.covariance()[5]);
138 ATH_MSG_DEBUG("BEAM tiltX,Y="<<beamtiltX<<","<<beamtiltY);
139
140 const EventContext& ctx = Gaudi::Hive::currentContext();
141
142 if(fullxAOD){
143 //---If beam is tilted -> make pre-fit and translate beam constraint to pre-fitted position
144 if( beamtiltX!=0. || beamtiltY!=0.) {
145 std::unique_ptr<xAOD::Vertex> iniVertex = m_fitterSvc->fit(ctx,particles,BEAM);
146 if(!iniVertex) return {nullptr};
147 BEAM.setX(BEAM.x()+beamtiltX*iniVertex->z());
148 BEAM.setY(BEAM.y()+beamtiltY*iniVertex->z());
149 }
150 return std::unique_ptr<xAOD::Vertex>(m_fitterSvc->fit(ctx,particles,BEAM));
151 }
152
153 //
154 //---DxAOD case
155 //
156 int NPRT=particles.size();
157 std::unique_ptr< SG::AuxStoreInternal > pAux;
159 std::vector<const xAOD::TrackParticle*> wrkTrkC(NPRT);
160 pAux = std::make_unique< SG::AuxStoreInternal >();
161 TPC.setStore( pAux.get() );
162 TPC.reserve( NPRT );
163 static const SG::ConstAccessor<float> vzAcc("vz");
164 for(int i=0; i<NPRT; i++){
165 TPC.push_back(new (std::nothrow) xAOD::TrackParticle(*particles[i]));
166 if(!TPC[i])return {nullptr};
167 const float mvx= (eventINFO) ? eventINFO->beamPosX() : 0.;
168 const float mvy= (eventINFO) ? eventINFO->beamPosY() : 0.;
169 const float mvz= (vzAcc.isAvailable(*particles[i])) ? particles[i]->vz() : 0.;
170 TPC[i]->setParametersOrigin( mvx, mvy, mvz);
171 wrkTrkC[i]=TPC[i];
172 }
173
174 //---If beam is tilted -> make pre-fit and translate beam constraint to pre-fitted position
175 if( beamtiltX!=0. || beamtiltY!=0.) {
176 std::unique_ptr<xAOD::Vertex> iniVertex = m_fitterSvc->fit(ctx,wrkTrkC,BEAM);
177 if(!iniVertex) return {nullptr};
178 BEAM.setX(BEAM.x()+beamtiltX*iniVertex->z());
179 BEAM.setY(BEAM.y()+beamtiltY*iniVertex->z());
180 }
181 return std::unique_ptr<xAOD::Vertex>(m_fitterSvc->fit(ctx,wrkTrkC,BEAM));
182 }
#define AmgSymMatrix(dim)
if(febId1==febId2)
void makePrivateStore()
Create a new (empty) private store for this object.
float z() const
Returns the z position.
void setCovariancePosition(const AmgSymMatrix(3)&covariancePosition)
Sets the vertex covariance matrix.
void setZ(float value)
Sets the z position.
void setX(float value)
Sets the x position.
void setY(float value)
Sets the y position.
float y() const
Returns the y position.
void setCovariance(const std::vector< float > &value)
Sets the covariance matrix as a simple vector of values.
void setPosition(const Amg::Vector3D &position)
Sets the 3-position.
float x() const
Returns the x position.
const std::vector< float > & covariance() const
Returns the covariance matrix as a simple vector of values.
DataModel_detail::iterator< DVL > unique(typename DataModel_detail::iterator< DVL > beg, typename DataModel_detail::iterator< DVL > end)
Specialization of unique for DataVector/List.
void sort(typename DataModel_detail::iterator< DVL > beg, typename DataModel_detail::iterator< DVL > end)
Specialization of sort for DataVector/List.
Vertex_v1 Vertex
Define the latest version of the vertex class.

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

SG::ReadCondHandleKey<InDet::BeamSpotData> TrkToLeptonPVTool::m_beamSpotKey { this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot" }
private

Definition at line 50 of file TrkToLeptonPVTool.h.

50{ this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot" };

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

ToolHandle< Trk::IVertexFitter > TrkToLeptonPVTool::m_fitterSvc {this,"VertexFitter","Trk::TrkVKalVrtFitter/VertexFitterTool","Vertex Fitter tool instance"}
private

Definition at line 51 of file TrkToLeptonPVTool.h.

51{this,"VertexFitter","Trk::TrkVKalVrtFitter/VertexFitterTool","Vertex Fitter tool instance"};

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