ATLAS Offline Software
Loading...
Searching...
No Matches
TrackParametersPlots.cxx
Go to the documentation of this file.
1/*
2 Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
3*/
4
9
11#include "TrackParametersPlots.h"
12#include "../TrackParametersHelper.h"
13#include "../OfflineObjectDecorHelper.h"
14
15
19IDTPM::TrackParametersPlots::TrackParametersPlots(
20 PlotMgr* pParent, const std::string& dirName,
21 const std::string& anaTag, const std::string& trackType,
22 bool plotErrors, bool recomputeIP ) :
23 PlotMgr( dirName, anaTag, pParent ),
24 m_trackType( trackType ),
25 m_plotErrors( plotErrors ), m_recomputeIP( recomputeIP ) { }
26
27
31void IDTPM::TrackParametersPlots::initializePlots()
32{
33 StatusCode sc = bookPlots();
34 if( sc.isFailure() ) {
35 ATH_MSG_ERROR( "Failed to book track parameters plots" );
36 }
37}
38
39
40StatusCode IDTPM::TrackParametersPlots::bookPlots()
41{
42 ATH_MSG_DEBUG( "Booking track parameters plots in " << getDirectory() );
43
44 ATH_CHECK( retrieveAndBook( m_pt, m_trackType+"_pt" ) );
45 ATH_CHECK( retrieveAndBook( m_eta, m_trackType+"_eta" ) );
46 ATH_CHECK( retrieveAndBook( m_phi, m_trackType+"_phi" ) );
47 ATH_CHECK( retrieveAndBook( m_d0, m_trackType+"_d0" ) );
48 ATH_CHECK( retrieveAndBook( m_z0, m_trackType+"_z0" ) );
49 ATH_CHECK( retrieveAndBook( m_z0sin, m_trackType+"_z0sin" ) );
50 ATH_CHECK( retrieveAndBook( m_theta, m_trackType+"_theta" ) );
51 ATH_CHECK( retrieveAndBook( m_qoverp, m_trackType+"_qoverp" ) );
52 if( m_trackType == "truth" ) {
53 ATH_CHECK( retrieveAndBook( m_prodR, m_trackType+"_prodR" ) );
54 ATH_CHECK( retrieveAndBook( m_prodZ, m_trackType+"_prodZ" ) );
55 ATH_CHECK( retrieveAndBook( m_nSiHits, m_trackType+"_nSiHits" ) );
56 ATH_CHECK( retrieveAndBook( m_nSiHits_vs_eta, m_trackType+"_nSiHits_vs_eta" ) );
57 } else {
58 ATH_CHECK( retrieveAndBook( m_chi2, m_trackType+"_chi2" ) );
59 ATH_CHECK( retrieveAndBook( m_ndof, m_trackType+"_ndof" ) );
60 ATH_CHECK( retrieveAndBook( m_chi2OverNdof, m_trackType+"_chi2OverNdof" ) );
61 ATH_CHECK( retrieveAndBook( m_author, m_trackType+"_author" ) );
62 ATH_CHECK( retrieveAndBook( m_time, m_trackType+"_time" ) );
63 ATH_CHECK( retrieveAndBook( m_hasValidTime_eff_vs_eta, m_trackType+"_hasValidTime_eff_vs_eta" ) );
64 }
65 ATH_CHECK( retrieveAndBook( m_eta_vs_pt, m_trackType+"_eta_vs_pt" ) );
66 ATH_CHECK( retrieveAndBook( m_eta_vs_phi, m_trackType+"_eta_vs_phi" ) );
67 ATH_CHECK( retrieveAndBook( m_z0_vs_d0, m_trackType+"_z0_vs_d0" ) );
68 ATH_CHECK( retrieveAndBook( m_z0sin_vs_d0, m_trackType+"_z0sin_vs_d0" ) );
69
70 if( m_plotErrors ) {
72 ATH_CHECK( retrieveAndBook( m_sigma_pt, "sigma_"+m_trackType+"_pt" ) );
73 ATH_CHECK( retrieveAndBook( m_sigma_eta, "sigma_"+m_trackType+"_eta" ) );
74 ATH_CHECK( retrieveAndBook( m_sigma_phi, "sigma_"+m_trackType+"_phi" ) );
75 ATH_CHECK( retrieveAndBook( m_sigma_d0, "sigma_"+m_trackType+"_d0" ) );
76 ATH_CHECK( retrieveAndBook( m_sigma_z0, "sigma_"+m_trackType+"_z0" ) );
77 ATH_CHECK( retrieveAndBook( m_sigma_z0sin, "sigma_"+m_trackType+"_z0sin" ) );
78 ATH_CHECK( retrieveAndBook( m_sigma_theta, "sigma_"+m_trackType+"_theta" ) );
79
81 ATH_CHECK( retrieveAndBook( m_significance_pt, "significance_"+m_trackType+"_pt" ) );
82 ATH_CHECK( retrieveAndBook( m_significance_eta, "significance_"+m_trackType+"_eta" ) );
83 ATH_CHECK( retrieveAndBook( m_significance_phi, "significance_"+m_trackType+"_phi" ) );
84 ATH_CHECK( retrieveAndBook( m_significance_d0, "significance_"+m_trackType+"_d0" ) );
85 ATH_CHECK( retrieveAndBook( m_significance_z0, "significance_"+m_trackType+"_z0" ) );
86 ATH_CHECK( retrieveAndBook( m_significance_z0sin, "significance_"+m_trackType+"_z0sin" ) );
87 ATH_CHECK( retrieveAndBook( m_significance_theta, "significance_"+m_trackType+"_theta" ) );
88 }
89
90 return StatusCode::SUCCESS;
91}
92
93
97template< typename PARTICLE >
98StatusCode IDTPM::TrackParametersPlots::fillPlots(
99 const PARTICLE& particle, float weight )
100{
102 float ppt = pT( particle ) / Gaudi::Units::GeV;
103 float peta = eta( particle );
104 float pphi = phi( particle );
105 float pd0 = d0( particle );
106 if( m_recomputeIP ) pd0 = getD0TrackInJet( particle );
107 float pz0 = z0( particle );
108 float pz0sin = z0SinTheta( particle );
109 float ptheta = theta( particle );
110 float pqoverp = qOverP( particle ); // in MeV^-1
111 float pprodR = prodR( particle );
112 float pprodZ = prodZ( particle );
113 float pnSiHits = nSiHits( particle );
114 float pchi2 = chiSquared( particle );
115 float pndof = ndof( particle );
116 float pchi2OverNdof = ( pndof > 0 ) ? pchi2 / pndof : 0.;
117 std::vector< unsigned int > pauthor = author( particle );
118
120 ATH_CHECK( fill( m_pt, ppt, weight ) );
121 ATH_CHECK( fill( m_eta, peta, weight ) );
122 ATH_CHECK( fill( m_phi, pphi, weight ) );
123 ATH_CHECK( fill( m_d0, pd0, weight ) );
124 ATH_CHECK( fill( m_z0, pz0, weight ) );
125 ATH_CHECK( fill( m_z0sin, pz0sin, weight ) );
126 ATH_CHECK( fill( m_theta, ptheta, weight ) );
127 ATH_CHECK( fill( m_qoverp, pqoverp, weight ) );
128 if( m_trackType == "truth" ) {
129 ATH_CHECK( fill( m_prodR, pprodR, weight ) );
130 ATH_CHECK( fill( m_prodZ, pprodZ, weight ) );
131 ATH_CHECK( fill( m_nSiHits, pnSiHits, weight ) );
132 ATH_CHECK( fill( m_nSiHits_vs_eta, peta, pnSiHits, weight ) );
133 } else {
134 ATH_CHECK( fill( m_chi2, pchi2, weight ) );
135 ATH_CHECK( fill( m_ndof, pndof, weight ) );
136 ATH_CHECK( fill( m_chi2OverNdof, pchi2OverNdof, weight ) );
137 for( unsigned int i=0; i<pauthor.size(); i++ ) {
138 ATH_CHECK( fill( m_author, pauthor[i], weight ) );
139 }
140 static const SG::Accessor< uint8_t > accValidTime("hasValidTime");
141 static const SG::Accessor< float > accTime("time");
142 if( accValidTime.isAvailable( particle ) ) {
143 bool phasValidTime = hasValidTime( particle );
144 ATH_CHECK( fill( m_hasValidTime_eff_vs_eta, peta, phasValidTime, weight ) );
145 if( accTime.isAvailable( particle ) and phasValidTime ) {
146 ATH_CHECK( fill( m_time, time( particle ), weight ) );
147 }
148 }
149 }
150 ATH_CHECK( fill( m_eta_vs_pt, ppt, peta, weight ) );
151 ATH_CHECK( fill( m_eta_vs_phi, pphi, peta, weight ) );
152 ATH_CHECK( fill( m_z0_vs_d0, pd0, pz0, weight ) );
153 ATH_CHECK( fill( m_z0sin_vs_d0, pz0*std::sin(ptheta), pd0, weight ) );
154
155 if( m_plotErrors ) {
156 float ppt_err = pTError( particle ) / Gaudi::Units::GeV;
157 float peta_err = etaError( particle );
158 float pphi_err = error( particle, Trk::phi );
159 float pd0_err = error( particle, Trk::d0 );
160 float pz0_err = error( particle, Trk::z0 );
161 float pz0sin_err = z0SinThetaError( particle );
162 float ptheta_err = error( particle, Trk::theta );
163
164 ATH_CHECK( fill( m_sigma_pt, ppt_err, weight ) );
165 ATH_CHECK( fill( m_sigma_eta, peta_err, weight ) );
166 ATH_CHECK( fill( m_sigma_phi, pphi_err, weight ) );
167 ATH_CHECK( fill( m_sigma_d0, pd0_err, weight ) );
168 ATH_CHECK( fill( m_sigma_z0, pz0_err, weight ) );
169 ATH_CHECK( fill( m_sigma_z0sin, pz0sin_err, weight ) );
170 ATH_CHECK( fill( m_sigma_theta, ptheta_err, weight ) );
171
172 float ppt_sig = ppt_err > 0. ? ppt / ppt_err : -9999.;
173 float peta_sig = peta_err > 0. ? peta / peta_err : -9999.;
174 float pphi_sig = pphi_err > 0. ? pphi / pphi_err : -9999.;
175 float pd0_sig = pd0_err > 0. ? pd0 / pd0_err : -9999.;
176 float pz0_sig = pz0_err > 0. ? pz0 / pz0_err : -9999.;
177 float pz0sin_sig = pz0sin_err > 0. ? pz0sin / pz0sin_err : -9999.;
178 float ptheta_sig = ptheta_err > 0. ? ptheta / ptheta_err : -9999.;
179
180 ATH_CHECK( fill( m_significance_pt, ppt_sig, weight ) );
181 ATH_CHECK( fill( m_significance_eta, peta_sig, weight ) );
182 ATH_CHECK( fill( m_significance_phi, pphi_sig, weight ) );
183 ATH_CHECK( fill( m_significance_d0, pd0_sig, weight ) );
184 ATH_CHECK( fill( m_significance_z0, pz0_sig, weight ) );
185 ATH_CHECK( fill( m_significance_z0sin, pz0sin_sig, weight ) );
186 ATH_CHECK( fill( m_significance_theta, ptheta_sig, weight ) );
187 }
188
189 return StatusCode::SUCCESS;
190}
191
192template StatusCode IDTPM::TrackParametersPlots::fillPlots< xAOD::TrackParticle >(
193 const xAOD::TrackParticle&, float weight );
194
195template StatusCode IDTPM::TrackParametersPlots::fillPlots< xAOD::TruthParticle >(
196 const xAOD::TruthParticle&, float weight );
197
198
202void IDTPM::TrackParametersPlots::finalizePlots()
203{
204 ATH_MSG_DEBUG( "Finalising track parameters plots" );
206}
eta
Scalar eta() const
pseudorapidity method
Definition AmgMatrixBasePlugin.h:83
phi
Scalar phi() const
phi method
Definition AmgMatrixBasePlugin.h:67
theta
Scalar theta() const
theta method
Definition AmgMatrixBasePlugin.h:75
ATH_CHECK
#define ATH_CHECK
Evaluate an expression and check for errors.
Definition AthCheckMacros.h:40
ATH_MSG_ERROR
#define ATH_MSG_ERROR(x)
Definition AthMsgStreamMacros.h:33
ATH_MSG_DEBUG
#define ATH_MSG_DEBUG(x)
Definition AthMsgStreamMacros.h:29
sc
static Double_t sc
Definition LArPhysWaveHECTool.cxx:37
OfflineObjectDecorHelper.h
Utility methods to access offline object decorations.
TrackParametersHelper.h
Utility methods to access track/truth particles parmeters in a consitent way in this package.
IDTPM::PlotMgr::retrieveAndBook
StatusCode retrieveAndBook(P *&pHisto, const std::string &identifier, const std::string &folderOverride="", const std::string &nameOverride="")
Definition PlotMgr.h:64
IDTPM::PlotMgr::PlotMgr
PlotMgr(const std::string &dirName, const std::string &anaTag, PlotMgr *pParent=nullptr)
Constructor taking parent node and directory name for plots pParent = nullptr by default to book plot...
Definition PlotMgr.cxx:25
IDTPM::TrackParametersPlots::TrackParametersPlots
TrackParametersPlots(PlotMgr *pParent, const std::string &dirName, const std::string &anaTag, const std::string &trackType, bool plotErrors=false, bool recomputeIP=false)
Constructor.
Definition TrackParametersPlots.cxx:19
IDTPM::TrackParametersPlots::m_significance_pt
TH1 * m_significance_pt
significance plots
Definition TrackParametersPlots.h:85
IDTPM::TrackParametersPlots::fillPlots
StatusCode fillPlots(const PARTICLE &particle, float weight)
Dedicated fill method (for tracks and/or truth particles)
Definition TrackParametersPlots.cxx:98
IDTPM::TrackParametersPlots::initializePlots
void initializePlots()
Book the histograms.
Definition TrackParametersPlots.cxx:31
IDTPM::TrackParametersPlots::finalizePlots
void finalizePlots()
Print out final stats on histograms.
Definition TrackParametersPlots.cxx:202
PlotBase::getDirectory
const std::string & getDirectory()
Definition PlotBase.h:88
SG::Accessor
Helper class to provide type-safe access to aux data.
Definition Control/AthContainers/AthContainers/Accessor.h:68
SG::ConstAccessor::isAvailable
bool isAvailable(const ELT &e) const
Test to see if this variable exists in the store.
IDTPM::getD0TrackInJet
float getD0TrackInJet(const xAOD::TrackParticle &track, const std::string &quality)
recompute d0 projecting in jet direction w.r.t. origin
Definition OfflineObjectDecorHelper.cxx:53
IDTPM::pT
float pT(const U &p)
Accessor utility function for getting the value of pT.
Definition TrackParametersHelper.h:33
IDTPM::z0SinTheta
float z0SinTheta(const U &p)
Definition TrackParametersHelper.h:75
IDTPM::prodZ
float prodZ(const U &p)
Definition TrackParametersHelper.h:102
IDTPM::qOverP
float qOverP(const U &p)
Definition TrackParametersHelper.h:111
IDTPM::chiSquared
float chiSquared(const U &p)
Definition TrackParametersHelper.h:128
IDTPM::pTError
float pTError(const U &p)
Definition TrackParametersHelper.h:213
IDTPM::z0
float z0(const U &p)
Definition TrackParametersHelper.h:72
IDTPM::d0
float d0(const U &p)
Definition TrackParametersHelper.h:84
IDTPM::prodR
float prodR(const U &p)
Definition TrackParametersHelper.h:93
IDTPM::ndof
float ndof(const U &p)
Definition TrackParametersHelper.h:134
IDTPM::time
float time(const U &p)
Definition TrackParametersHelper.h:159
IDTPM::nSiHits
float nSiHits(const U &p)
Definition TrackParametersHelper.h:427
IDTPM::author
std::vector< unsigned int > author(const U &p)
Definition TrackParametersHelper.h:147
IDTPM::etaError
float etaError(const U &p)
Definition TrackParametersHelper.h:222
IDTPM::z0SinThetaError
float z0SinThetaError(const U &p)
Definition TrackParametersHelper.h:234
IDTPM::hasValidTime
uint8_t hasValidTime(const U &p)
Definition TrackParametersHelper.h:153
Trk::theta
@ theta
Definition ParamDefs.h:66
Trk::phi
@ phi
Definition ParamDefs.h:75
Trk::d0
@ d0
Definition ParamDefs.h:63
Trk::z0
@ z0
Definition ParamDefs.h:64
xAOD::TrackParticle
TrackParticle_v1 TrackParticle
Reference the current persistent version:
Definition Event/xAOD/xAODTracking/xAODTracking/TrackParticle.h:13
xAOD::TruthParticle
TruthParticle_v1 TruthParticle
Typedef to implementation.
Definition Event/xAOD/xAODTruth/xAODTruth/TruthParticle.h:15
fill
void fill(H5::Group &out_file, size_t iterations)
Definition test-half-precision.cxx:64
Generated on for ATLAS Offline Software by doxygen 1.14.0