Belle II Software development
Public Member Functions | Protected Member Functions | Protected Attributes | List of all members
RecoTrackExtractor Class Reference
Packages » tracking

class to extract results from qualityEstimation More...

#include <RecoTrackExtractor.h>

Inheritance diagram for RecoTrackExtractor:
VariableExtractor

Public Member Functions

 RecoTrackExtractor (std::vector< Named< float * > > &variableSet, const std::string &prefix="")
 Define names of variables that get extracted.
 
void extractVariables (const RecoTrack &recoTrack)
 extract the actual variables and write into a variable set
 

Protected Member Functions

void addVariable (const std::string &identifier, std::vector< Named< float * > > &variables)
 add a variable to the variable set
 

Protected Attributes

std::string m_prefix
 prefix for RecoTrack extracted variables
 
std::unordered_map< std::string, float > m_variables
 unordered_map to associate float value with a string name
 

Detailed Description

class to extract results from qualityEstimation

Definition at line 30 of file RecoTrackExtractor.h.

Constructor & Destructor Documentation

◆ RecoTrackExtractor()

RecoTrackExtractor ( std::vector< Named< float * > > & variableSet,
const std::string & prefix = "" )
inlineexplicit

Define names of variables that get extracted.

Definition at line 34 of file RecoTrackExtractor.h.

34 :
35 VariableExtractor(), m_prefix(prefix)
36 {
37 addVariable(prefix + "pdg_ID", variableSet);
38 addVariable(prefix + "pdg_ID_Mother", variableSet);
39 addVariable(prefix + "is_Vzero_Daughter", variableSet);
40 addVariable(prefix + "is_Primary", variableSet);
41
42 addVariable(prefix + "z0", variableSet);
43 addVariable(prefix + "d0", variableSet);
44
45 addVariable(prefix + "seed_Charge", variableSet);
46
47 addVariable(prefix + "seed_Pos_Pt", variableSet);
48 addVariable(prefix + "seed_Pos_Z", variableSet);
49 addVariable(prefix + "seed_Pos_Mag", variableSet);
50 addVariable(prefix + "seed_Pos_Theta", variableSet);
51 addVariable(prefix + "seed_Pos_Phi", variableSet);
52
53 addVariable(prefix + "seed_Mom_Pt", variableSet);
54 addVariable(prefix + "seed_Mom_Z", variableSet);
55 addVariable(prefix + "seed_Mom_Mag", variableSet);
56 addVariable(prefix + "seed_Mom_Theta", variableSet);
57 addVariable(prefix + "seed_Mom_Phi", variableSet);
58
59 addVariable(prefix + "seed_Time", variableSet);
60
61 addVariable(prefix + "N_tracking_hits", variableSet);
62 addVariable(prefix + "N_CDC_hits", variableSet);
63 addVariable(prefix + "N_SVD_hits", variableSet);
64 addVariable(prefix + "N_PXD_hits", variableSet);
65
66 addVariable(prefix + "Fit_Charge", variableSet);
67 addVariable(prefix + "Fit_Chi2", variableSet);
68 addVariable(prefix + "Fit_Ndf", variableSet);
69 addVariable(prefix + "Fit_NFailedPoints", variableSet);
70 addVariable(prefix + "Fit_PVal", variableSet);
71 addVariable(prefix + "Fit_Successful", variableSet);
72
73 addVariable(prefix + "POCA_Pos_Pt", variableSet);
74 addVariable(prefix + "POCA_Pos_Z", variableSet);
75 addVariable(prefix + "POCA_Pos_Mag", variableSet);
76 addVariable(prefix + "POCA_Pos_Theta", variableSet);
77 addVariable(prefix + "POCA_Pos_Phi", variableSet);
78
79 addVariable(prefix + "POCA_Mom_Pt", variableSet);
80 addVariable(prefix + "POCA_Mom_Z", variableSet);
81 addVariable(prefix + "POCA_Mom_Mag", variableSet);
82 addVariable(prefix + "POCA_Mom_Theta", variableSet);
83 addVariable(prefix + "POCA_Mom_Phi", variableSet);
84 }

Member Function Documentation

◆ addVariable()

void addVariable ( const std::string & identifier,
std::vector< Named< float * > > & variables )
inlineprotectedinherited

add a variable to the variable set

Definition at line 27 of file VariableExtractor.h.

28 {
29 //todo: verify if it is faster to check explicitly or not
30 auto value = m_variables.emplace(identifier, NAN).first;
31 variables.emplace_back(identifier, &(value->second));
32 }

◆ extractVariables()

void extractVariables ( const RecoTrack & recoTrack)
inline

extract the actual variables and write into a variable set

Definition at line 87 of file RecoTrackExtractor.h.

88 {
89 float pdgID = 0;
90 float pdgIDMother = 0;
91 float isVzeroDaughter = -1;
92 float isPrimary = -1;
93 auto mcparticle = recoTrack.getRelated<MCParticle>();
94 if (mcparticle) {
95 pdgID = static_cast<float>(mcparticle->getPDG());
96 pdgIDMother = static_cast<float>(mcparticle->getMother()->getPDG());
97 isPrimary = static_cast<float>(mcparticle->isPrimaryParticle());
98 if (std::abs(pdgIDMother) == 310 || std::abs(pdgIDMother) == 3122) {
99 isVzeroDaughter = 1;
100 } else {
101 isVzeroDaughter = 0;
102 }
103 }
104 float z0 = -999;
105 float d0 = -999;
106 auto genfitTrack = recoTrack.getRelated<Track>("MDSTTracks");
107 if (genfitTrack) {
108 auto trackFitResult = genfitTrack->getTrackFitResultWithClosestMass(Const::pion);
109 if (trackFitResult) {
110 z0 = trackFitResult->getZ0();
111 d0 = trackFitResult->getD0();
112 }
113 }
114 m_variables.at(m_prefix + "pdg_ID") = pdgID;
115 m_variables.at(m_prefix + "pdg_ID_Mother") = pdgIDMother;
116 m_variables.at(m_prefix + "is_Vzero_Daughter") = isVzeroDaughter;
117 m_variables.at(m_prefix + "is_Primary") = isPrimary;
118
119 m_variables.at(m_prefix + "z0") = z0;
120 m_variables.at(m_prefix + "d0") = d0;
121
122 m_variables.at(m_prefix + "seed_Charge") = recoTrack.getChargeSeed();
123
124 m_variables.at(m_prefix + "seed_Pos_Pt") = recoTrack.getPositionSeed().Rho();
125 m_variables.at(m_prefix + "seed_Pos_Z") = recoTrack.getPositionSeed().Z();
126 m_variables.at(m_prefix + "seed_Pos_Mag") = recoTrack.getPositionSeed().R();
127 m_variables.at(m_prefix + "seed_Pos_Theta") = recoTrack.getPositionSeed().Theta();
128 m_variables.at(m_prefix + "seed_Pos_Phi") = recoTrack.getPositionSeed().Phi();
129
130 m_variables.at(m_prefix + "seed_Mom_Pt") = recoTrack.getMomentumSeed().Rho();
131 m_variables.at(m_prefix + "seed_Mom_Z") = recoTrack.getMomentumSeed().Z();
132 m_variables.at(m_prefix + "seed_Mom_Mag") = recoTrack.getMomentumSeed().R();
133 m_variables.at(m_prefix + "seed_Mom_Theta") = recoTrack.getMomentumSeed().Theta();
134 m_variables.at(m_prefix + "seed_Mom_Phi") = recoTrack.getMomentumSeed().Phi();
135
136 m_variables.at(m_prefix + "seed_Time") = recoTrack.getTimeSeed();
137
138 m_variables.at(m_prefix + "N_tracking_hits") = recoTrack.getNumberOfTrackingHits();
139 m_variables.at(m_prefix + "N_CDC_hits") = recoTrack.getNumberOfCDCHits();
140 m_variables.at(m_prefix + "N_SVD_hits") = recoTrack.getNumberOfSVDHits();
141 m_variables.at(m_prefix + "N_PXD_hits") = recoTrack.getNumberOfPXDHits();
142
143 const genfit::FitStatus* rt_TrackFitStatus = recoTrack.getTrackFitStatus();
144 if (rt_TrackFitStatus) {
145 m_variables.at(m_prefix + "Fit_Charge") = rt_TrackFitStatus->getCharge();
146 m_variables.at(m_prefix + "Fit_Chi2") = rt_TrackFitStatus->getChi2();
147 m_variables.at(m_prefix + "Fit_Ndf") = rt_TrackFitStatus->getNdf();
148 m_variables.at(m_prefix + "Fit_NFailedPoints") = rt_TrackFitStatus->getNFailedPoints();
149 m_variables.at(m_prefix + "Fit_PVal") = rt_TrackFitStatus->getPVal();
150 } else {
151 m_variables.at(m_prefix + "Fit_Charge") = -std::numeric_limits<float>::max();
152 m_variables.at(m_prefix + "Fit_Chi2") = -1.;
153 m_variables.at(m_prefix + "Fit_Ndf") = -1.;
154 m_variables.at(m_prefix + "Fit_NFailedPoints") = -1.;
155 m_variables.at(m_prefix + "Fit_PVal") = -1.;
156 }
157
158 m_variables.at(m_prefix + "Fit_Successful") = (float)recoTrack.wasFitSuccessful();
159
160 if (recoTrack.wasFitSuccessful()) {
161 B2Vector3D linePoint(0., 0., 0.);
162 B2Vector3D lineDirection(0., 0., 1.);
163
164 genfit::MeasuredStateOnPlane reco_sop;
165 try {
166 reco_sop = recoTrack.getMeasuredStateOnPlaneFromFirstHit();
167 reco_sop.extrapolateToLine(linePoint, lineDirection);
168 m_variables.at(m_prefix + "POCA_Pos_Pt") = reco_sop.getPos().Pt();
169 m_variables.at(m_prefix + "POCA_Pos_Z") = reco_sop.getPos().Z();
170 m_variables.at(m_prefix + "POCA_Pos_Mag") = reco_sop.getPos().Mag();
171 m_variables.at(m_prefix + "POCA_Pos_Theta") = reco_sop.getPos().Theta();
172 m_variables.at(m_prefix + "POCA_Pos_Phi") = reco_sop.getPos().Phi();
173
174 m_variables.at(m_prefix + "POCA_Mom_Pt") = reco_sop.getMom().Pt();
175 m_variables.at(m_prefix + "POCA_Mom_Z") = reco_sop.getMom().Z();
176 m_variables.at(m_prefix + "POCA_Mom_Mag") = reco_sop.getMom().Mag();
177 m_variables.at(m_prefix + "POCA_Mom_Theta") = reco_sop.getMom().Theta();
178 m_variables.at(m_prefix + "POCA_Mom_Phi") = reco_sop.getMom().Phi();
179 } catch (genfit::Exception const& e) {
180 // extrapolation not possible, skip this track
181 B2WARNING("RecoTrackExtractor: recoTrack BeamPipe POCA extrapolation failed!\n"
182 << "-->" << e.what());
183 m_variables.at(m_prefix + "POCA_Pos_Pt") = -1;
184 m_variables.at(m_prefix + "POCA_Pos_Z") = -std::numeric_limits<float>::max();
185 m_variables.at(m_prefix + "POCA_Pos_Mag") = -1;
186 m_variables.at(m_prefix + "POCA_Pos_Theta") = -10;
187 m_variables.at(m_prefix + "POCA_Pos_Phi") = -10;
188
189 m_variables.at(m_prefix + "POCA_Mom_Pt") = -1;
190 m_variables.at(m_prefix + "POCA_Mom_Z") = -std::numeric_limits<float>::max();
191 m_variables.at(m_prefix + "POCA_Mom_Mag") = -1;
192 m_variables.at(m_prefix + "POCA_Mom_Theta") = -10;
193 m_variables.at(m_prefix + "POCA_Mom_Phi") = -10;
194 }
195 } else {
196 m_variables.at(m_prefix + "POCA_Pos_Pt") = -1;
197 m_variables.at(m_prefix + "POCA_Pos_Z") = -std::numeric_limits<float>::max();
198 m_variables.at(m_prefix + "POCA_Pos_Mag") = -1;
199 m_variables.at(m_prefix + "POCA_Pos_Theta") = -10;
200 m_variables.at(m_prefix + "POCA_Pos_Phi") = -10;
201
202 m_variables.at(m_prefix + "POCA_Mom_Pt") = -1;
203 m_variables.at(m_prefix + "POCA_Mom_Z") = -std::numeric_limits<float>::max();
204 m_variables.at(m_prefix + "POCA_Mom_Mag") = -1;
205 m_variables.at(m_prefix + "POCA_Mom_Theta") = -10;
206 m_variables.at(m_prefix + "POCA_Mom_Phi") = -10;
207 }
208
209 }
Belle2::B2Vector3D
B2Vector3< double > B2Vector3D
typedef for common usage with double
Definition B2Vector3.h:516

Member Data Documentation

◆ m_prefix

std::string m_prefix
protected

prefix for RecoTrack extracted variables

Definition at line 213 of file RecoTrackExtractor.h.

◆ m_variables

std::unordered_map<std::string, float> m_variables
protectedinherited

unordered_map to associate float value with a string name

Definition at line 35 of file VariableExtractor.h.

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