ATLAS Offline Software
Loading...
Searching...
No Matches
DetailedIDNtupleTool.cxx
Go to the documentation of this file.
1/*
2 Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
3 */
4
5#include "DetailedIDNtupleTool.h"
6
7#include "TrkEventPrimitives/FitQuality.h"
8#include "TrkTrack/Track.h"
9#include "TrkRIO_OnTrack/RIO_OnTrack.h"
10#include "TrkEventPrimitives/TrackStateDefs.h"
11#include "TrkTrackSummary/TrackSummary.h"
12
13//for Amg::error helper function:
14#include "EventPrimitives/EventPrimitivesHelpers.h"
15
16#include "xAODEventInfo/EventInfo.h"
17#include "TrkVertexOnTrack/VertexOnTrack.h"
18
19#include "TrkToolInterfaces/ITrackSummaryTool.h"
20#include "TrkAlignEvent/AlignModule.h"
21#include "TrkAlignEvent/AlignTrack.h"
22
23#include "TrkTruthData/TrackTruth.h"
24#include "TrkTruthData/TrackTruthCollection.h"
25#include "AtlasHepMC/GenParticle.h"
26#include "AtlasHepMC/GenVertex.h"
27
28#include "TrkAlignInterfaces/IAlignModuleTool.h"
29
30#include "TFile.h"
31#include "TTree.h"
32#include "TMath.h"
33#include <cmath>
34
35
36namespace InDet {
37//________________________________________________________________________
38 DetailedIDNtupleTool::DetailedIDNtupleTool(const std::string& type, const std::string& name, const IInterface* parent)
39 : AthAlgTool(type, name, parent)
40 , m_trackSumTool("Trk::TrackSummaryTool/TrackSummaryTool", this)
41 , m_alignModuleTool{}
42 , m_storeNormalRefittedOnly{false}
43 , m_storeConstrainedOnly{true}
44 , m_storeTruth{true}
45 , m_matchProbability{0.8}
46 , m_file{}
47 , m_tree{}
48 , m_trackCollection{"Tracks"}
49 , m_tracksTruthName{"TrackTruthCollection"} {
50 declareInterface<Trk::IFillNtupleTool>(this);
51
52 declareProperty("FileName", m_filename = "IDAlign.root");
53 declareProperty("FilePath", m_filepath = "./");
54 declareProperty("TrackCollection", m_trackCollection);
55 declareProperty("TracksTruthName", m_tracksTruthName);
56 declareProperty("TrackSummaryTool", m_trackSumTool);
57 declareProperty("AlignModuleTool", m_alignModuleTool);
58 declareProperty("MatchProbability", m_matchProbability);
59 declareProperty("StoreTruth", m_storeTruth);
60 declareProperty("MatchProbability", m_matchProbability);
61 declareProperty("StoreNormalRefittedOnly", m_storeNormalRefittedOnly);
62 declareProperty("StoreConstrainedOnly", m_storeConstrainedOnly);
63 }
64
65//________________________________________________________________________
68
69//________________________________________________________________________
70 StatusCode DetailedIDNtupleTool::initialize() {
71 ATH_MSG_DEBUG("initialize() of DetailedIDNtupleTool");
72 ATH_MSG_DEBUG("creating file with name " << m_filepath << m_filename);
73
74 if (!m_tree) initializeNtuple();
75 ATH_CHECK(m_truthToTrack.retrieve());
76
77 /* Retrieve extrapolator tool */
78 ATH_CHECK(m_extrapolator.retrieve());
79
80 // get AlignModuleTool
81 if (m_alignModuleTool.empty()) {
82 ATH_MSG_FATAL("m_alignModuleTool is empty in DetailedIDNtupleTool::initialize()");
83 return StatusCode::FAILURE;
84 } else {
85 ATH_CHECK(m_alignModuleTool.retrieve());
86 }
87 return StatusCode::SUCCESS;
88 }
89
90//________________________________________________________________________
91 StatusCode DetailedIDNtupleTool::fillNtuple() {
92 if (m_file && m_file->IsOpen()) {
93 m_file->cd();
94 if (m_tree) m_tree->Write();
95 }
96 return StatusCode::SUCCESS;
97 }
98
99//________________________________________________________________________
100 StatusCode DetailedIDNtupleTool::finalize() {
101 ATH_MSG_DEBUG("finalize() of DetailedIDNtupleTool");
102 return StatusCode::SUCCESS;
103 }
104
105//________________________________________________________________________
106 void DetailedIDNtupleTool::dumpTrack(const EventContext& ctx, int /* itrk*/, const Trk::AlignTrack* alignTrack) {
107 ATH_MSG_DEBUG("In dumpTrack()");
108 Trk::AlignTrack::AlignTrackType type = alignTrack->type();
109 // if hope to dump all the tracks, should set:
110 // m_storeNormalRefittedOnly = false, m_storeConstrainedOnly=false
111 if ((m_storeNormalRefittedOnly && (type != Trk::AlignTrack::NormalRefitted)) ||
112 (m_storeConstrainedOnly && (type != Trk::AlignTrack::BeamspotConstrained))) {
113 return;
114 }
115
116 // get run and event numbers
117 ATH_MSG_DEBUG("Retrieving event info.");
118 const xAOD::EventInfo* eventInfo;
119 if (evtStore()->retrieve(eventInfo).isFailure()) {
120 ATH_MSG_ERROR("Could not retrieve event info.");
121 } else {
122 m_runNumber = eventInfo->runNumber();
123 m_evtNumber = eventInfo->eventNumber();
124 }
125
126 // initialize variables
127 constexpr double invalidParameterValue{-999.};
128 m_d0 = invalidParameterValue;
129 m_z0 = invalidParameterValue;
130 m_phi0 = invalidParameterValue;
131 m_theta = invalidParameterValue;
132 m_qoverp = invalidParameterValue;
133 m_pt = invalidParameterValue;
134 m_eta = invalidParameterValue;
135 constexpr double invalidChiSq{-1e12};
136 m_chi2 = invalidChiSq;
137 m_chi2prob = invalidChiSq;
138 constexpr int invalidDegreesOfFreedom{-999};
139
140 m_ndof = invalidDegreesOfFreedom;
141 m_xvtx = invalidParameterValue;
142 m_yvtx = invalidParameterValue;
143 m_zvtx = invalidParameterValue;
144
145 m_err_d0 = invalidParameterValue;
146 m_err_z0 = invalidParameterValue;
147 m_err_phi0 = invalidParameterValue;
148 m_err_theta = invalidParameterValue;
149 m_err_qoverp = invalidParameterValue;
150 m_toRef_d0 = invalidParameterValue;
151 m_toRef_z0 = invalidParameterValue;
152 m_toRef_phi0 = invalidParameterValue;
153 m_toRef_theta = invalidParameterValue;
154 m_toRef_qoverp = invalidParameterValue;
155
156
157 m_original_d0 = invalidParameterValue;
158 m_original_z0 = invalidParameterValue;
159 m_original_phi0 = invalidParameterValue;
160 m_original_theta = invalidParameterValue;
161 m_original_qoverp = invalidParameterValue;
162 m_original_pt =invalidParameterValue;
163 m_original_eta = -999.;
164 m_original_chi2 = invalidChiSq;
165 m_original_chi2prob = invalidChiSq;
166 m_original_ndof = invalidDegreesOfFreedom;
167 m_original_xvtx = invalidParameterValue;
168 m_original_yvtx = invalidParameterValue;
169 m_original_zvtx = invalidParameterValue;
170
171 m_original_err_d0 = invalidParameterValue;
172 m_original_err_z0 = invalidParameterValue;
173 m_original_err_phi0 = invalidParameterValue;
174 m_original_err_theta = invalidParameterValue;
175 m_original_err_qoverp = invalidParameterValue;;
176 m_original_toRef_d0 = invalidParameterValue;
177 m_original_toRef_z0 = invalidParameterValue;
178 m_original_toRef_phi0 = invalidParameterValue;
179 m_original_toRef_theta =invalidParameterValue;
180 m_original_toRef_qoverp = invalidParameterValue;;
181
182 m_truth_d0 = invalidParameterValue;
183 m_truth_z0 = invalidParameterValue;
184 m_truth_phi0 = invalidParameterValue;
185 m_truth_theta = invalidParameterValue;
186 m_truth_qoverp = invalidParameterValue;
187 m_truth_pt = invalidParameterValue;
188 m_truth_eta = invalidParameterValue;;
189 m_truth_prod_x = invalidParameterValue;
190 m_truth_prod_y = invalidParameterValue;
191 m_truth_prod_z = invalidParameterValue;;
192
193
194
195 const Trk::Perigee* aMeasPer = (alignTrack->perigeeParameters());
196
197 if (not aMeasPer) {
198 ATH_MSG_ERROR("Could not get Trk::Perigee of the alignTrack");
199 return;
200 } else {
201 m_d0 = aMeasPer->parameters()[Trk::d0];
202 m_z0 = aMeasPer->parameters()[Trk::z0];
203 m_phi0 = aMeasPer->parameters()[Trk::phi0];
204 m_theta = aMeasPer->parameters()[Trk::theta];
205 m_qoverp = aMeasPer->parameters()[Trk::qOverP];
206
207 const AmgSymMatrix(5) * locCov = aMeasPer->covariance();
208 m_err_d0 = Amg::error(*locCov, Trk::d0);
209 m_err_z0 = Amg::error(*locCov, Trk::z0);
210 m_err_phi0 = Amg::error(*locCov, Trk::phi0);
211 m_err_theta = Amg::error(*locCov, Trk::theta);
212 m_err_qoverp = Amg::error(*locCov, Trk::qOverP);
213
214 m_pt =
215 std::sqrt((aMeasPer->momentum().x()) * (aMeasPer->momentum().x()) + (aMeasPer->momentum().y()) *
216 (aMeasPer->momentum().y()));
217 m_eta = aMeasPer->eta();
218
219 m_xvtx = aMeasPer->position().x();
220 m_yvtx = aMeasPer->position().y();
221 m_zvtx = aMeasPer->position().z();
222
223
224
225 // get fit quality and chi2 probability of track
226 const Trk::FitQuality* fitQual = alignTrack->fitQuality();
227 if (not fitQual) ATH_MSG_ERROR("No fit quality assigned to the track");
228 else {
229 if (fitQual->chiSquared() > 0. && fitQual->numberDoF() > 0) {
230 m_chi2 = fitQual->chiSquared();
231 m_ndof = fitQual->numberDoF();
232 m_chi2prob = TMath::Prob(m_chi2, m_ndof);
233
234 ATH_MSG_DEBUG(" - chi2 : " << m_chi2);
235 ATH_MSG_DEBUG(" - ndof : " << m_ndof);
236 ATH_MSG_DEBUG(" - chi2/ndof : " << m_chi2 / (double) m_ndof);
237 ATH_MSG_DEBUG(" - chi2 propability : " << m_chi2prob);
238 }
239 }
240 }
241
242 const Trk::MeasurementBase* measbase = *(alignTrack->measurementsOnTrack()->begin());
243 Amg::Vector3D refPoint = measbase->associatedSurface().center();
244 const Trk::VertexOnTrack* vot = dynamic_cast<const Trk::VertexOnTrack*> (measbase);
245 if (!vot) {
246 ATH_MSG_ERROR(" Seems the pseudo-measuremnt in the alignTrack not exist!");
247 ATH_MSG_ERROR(" this pseudo-measurement has been rejected as outlier in the refitting!");
248 return;
249 } else {
250 ATH_MSG_DEBUG(" VoT get from the alignTrack:" << *vot);
251 }
252
253 ATH_MSG_DEBUG(" the pseudo-measurement position: " << refPoint);
254 const Trk::Track* originalTrack = alignTrack->originalTrack();
255
256 const Trk::Perigee* originalMeasPer = originalTrack->perigeeParameters();
257 if (!originalMeasPer) {
258 ATH_MSG_ERROR("No original track!");
259 } else {
260 m_original_d0 = originalMeasPer->parameters()[Trk::d0];
261 m_original_z0 = originalMeasPer->parameters()[Trk::z0];
262 m_original_phi0 = originalMeasPer->parameters()[Trk::phi0];
263 m_original_theta = originalMeasPer->parameters()[Trk::theta];
264 m_original_qoverp = originalMeasPer->parameters()[Trk::qOverP];
265
266 const AmgSymMatrix(5) * locError = originalMeasPer->covariance();
267 m_original_err_d0 = Amg::error(*locError, Trk::d0);
268 m_original_err_z0 = Amg::error(*locError, Trk::z0);
269 m_original_err_phi0 = Amg::error(*locError, Trk::phi0);
270 m_original_err_theta = Amg::error(*locError, Trk::theta);
271 m_original_err_qoverp = Amg::error(*locError, Trk::qOverP);
272
273 m_original_pt = (1.0 / m_original_qoverp) * std::sin(m_original_theta);
274 m_original_eta = -std::log(std::tan(0.5 * m_original_theta));
275
276 m_original_xvtx = originalMeasPer->position().x();
277 m_original_yvtx = originalMeasPer->position().y();
278 m_original_zvtx = originalMeasPer->position().z();
279
280 // get fit quality and chi2 probability of original Track
281 const Trk::FitQuality* originalFitQual = originalTrack->fitQuality();
282 if (not originalFitQual) ATH_MSG_ERROR("No fit quality assigned to the track");
283 else {
284 if (originalFitQual->chiSquared() > 0. && originalFitQual->numberDoF() > 0) {
285 m_original_chi2 = originalFitQual->chiSquared();
286 m_original_ndof = originalFitQual->numberDoF();
287 m_original_chi2prob = TMath::Prob(m_original_chi2, m_original_ndof);
288
289 ATH_MSG_DEBUG(" - original chi2 : " << m_original_chi2);
290 ATH_MSG_DEBUG(" - original ndof : " << m_original_ndof);
291 ATH_MSG_DEBUG(" - original chi2/ndof : " << m_original_chi2 / (double) m_original_ndof);
292 ATH_MSG_DEBUG(" - original chi2 propability : " << m_original_chi2prob);
293 }
294 }
295
296
297 const Trk::PerigeeSurface persf(refPoint);
298 const Trk::Perigee* originalPerigeeAtRef =
299 dynamic_cast<const Trk::Perigee*>(m_extrapolator->extrapolateTrack(ctx, *originalTrack, persf).release());
300 if (!originalPerigeeAtRef) {
301 const Trk::Perigee* originalTrackPerigee = originalTrack->perigeeParameters();
302 if (originalTrackPerigee && ((originalTrackPerigee->associatedSurface())) == persf) {
303 ATH_MSG_DEBUG("Perigee of Track is already expressed to given vertex, a copy is returned.");
304 originalPerigeeAtRef = originalTrackPerigee->clone();
305 } else ATH_MSG_DEBUG("Extrapolation to Perigee failed, NULL pointer is returned.");
306 }
307
308 if (originalPerigeeAtRef) {
309 std::unique_ptr<const Trk::Perigee > originalMeasPerAtRef(originalPerigeeAtRef);
310 m_original_toRef_d0 = originalMeasPerAtRef->parameters()[Trk::d0];
311 m_original_toRef_z0 = originalMeasPerAtRef->parameters()[Trk::z0];
312 m_original_toRef_phi0 = originalMeasPerAtRef->parameters()[Trk::phi0];
313 m_original_toRef_theta = originalMeasPerAtRef->parameters()[Trk::theta];
314 m_original_toRef_qoverp = originalMeasPerAtRef->parameters()[Trk::qOverP];
315 }
316
317 const Trk::Perigee* PerigeeAtRef =
318 dynamic_cast<const Trk::Perigee*>(m_extrapolator->extrapolateTrack(ctx, *alignTrack, persf).release());
319 if (!PerigeeAtRef) {
320 const Trk::Perigee* alignTrackPerigee = alignTrack->perigeeParameters();
321 if (alignTrackPerigee && ((alignTrackPerigee->associatedSurface())) == persf) {
322 ATH_MSG_DEBUG("Perigee of AlignTrack is already expressed to given vertex, a copy is returned.");
323 PerigeeAtRef = alignTrackPerigee->clone();
324 } else ATH_MSG_DEBUG("Extrapolation to Perigee failed, NULL pointer is returned.");
325 }
326
327 //post-eigen, can simply use the TrackParameters * returned by m_extrapolator->extrapolate?
328 if (PerigeeAtRef) {
329 std::unique_ptr<const Trk::Perigee > MeasPerAtRef((PerigeeAtRef));
330 m_toRef_d0 = MeasPerAtRef->parameters()[Trk::d0];
331 m_toRef_z0 = MeasPerAtRef->parameters()[Trk::z0];
332 m_toRef_phi0 = MeasPerAtRef->parameters()[Trk::phi0];
333 m_toRef_theta = MeasPerAtRef->parameters()[Trk::theta];
334 m_toRef_qoverp = MeasPerAtRef->parameters()[Trk::qOverP];
335 }
336 }
337
339 if (m_storeTruth) retrieveTruthInfo(alignTrack);
340
341
342 m_file->cd();
343 m_tree->Fill();
344 ATH_MSG_DEBUG("tree filled");
345
346 return;
347 }
348
349 bool DetailedIDNtupleTool::retrieveTruthInfo(const Trk::AlignTrack* alignTrack) {
350 // although we select tracks using the TrackSelectionTool, we still need to get a complete TrackCollection
351 // from StoreGate for use in the track-truth map, otherwise the track-truth matching is screwed up
352
353 const TrackCollection* RecCollection = nullptr;
354
355 if (evtStore()->retrieve(RecCollection, m_trackCollection).isFailure()) {
356 ATH_MSG_WARNING("Track collection \"" << m_trackCollection << "\" not found.");
357 return false;
358 }
359 if (RecCollection) {
360 ATH_MSG_DEBUG(
361 "Retrieved " << m_trackCollection << " with size " << RecCollection->size() <<
362 " reconstructed tracks from storegate");
363 } else {
364 ATH_MSG_WARNING("RecCollection is null pointer in DetailedIDNtupleTool::retrieveTruthInfo");
365 return false;
366 }
367
368 // get TrackTruthCollection
369 const TrackTruthCollection* TruthMap = nullptr;
370 if (StatusCode::SUCCESS != evtStore()->retrieve(TruthMap, m_tracksTruthName)) {
371 ATH_MSG_DEBUG("Cannot find " << m_tracksTruthName);
372 return false;
373 }
374
375 ATH_MSG_DEBUG(
376 "Track Truth Collection with name " << m_tracksTruthName << " with size " << TruthMap->size() <<
377 " found in StoreGate");
378
379 bool flag = false;
380 // get fit quality and chi2 probability of track
381 const Trk::Perigee* startPerigee = alignTrack->perigeeParameters();
382 const Trk::Perigee* measPer = startPerigee;
383
384 if (measPer == nullptr) {
385 ATH_MSG_DEBUG("No measured perigee parameters assigned to the track");
386 return false;
387 }
388
389 ElementLink<TrackCollection> tracklink;
390 tracklink.setElement(const_cast<Trk::Track*>(alignTrack->originalTrack()));
391 tracklink.setStorableObject(*RecCollection);
392 const ElementLink<TrackCollection> tracklink2 = tracklink;
393
394 TrackTruthCollection::const_iterator found = TruthMap->find(tracklink2);
395 if ((found != TruthMap->end()) && (found->second.probability() > m_matchProbability)) {
396 TrackTruth trtruth = found->second;
397 const HepMcParticleLink& HMPL = trtruth.particleLink();
398
399 if (HMPL.isValid()) {
400#ifdef HEPMC3
401 HepMC::ConstGenParticlePtr genparptr = HMPL.scptr();
402#else
403 const HepMC::GenParticle* genparptr = HMPL.cptr();
404#endif
405
406 if (genparptr) {
407 if (genparptr->production_vertex()) {
408 if (genparptr->pdg_id() == 0) {
409 ATH_MSG_INFO("PDG ID is zero in DetailedIDNtupleTool::retrieveTruthInfo");
410 } else {
411 const Trk::TrackParameters* generatedTrackPerigee = m_truthToTrack->makePerigeeParameters(genparptr);
412 if (!generatedTrackPerigee) ATH_MSG_WARNING("Unable to extrapolate genparticle to perigee!");
413 else {
414 flag = true;
415 m_truth_qoverpt = generatedTrackPerigee->parameters()[Trk::qOverP] / std::sin(
416 generatedTrackPerigee->parameters()[Trk::theta]);
417 m_truth_qoverp = generatedTrackPerigee->parameters()[Trk::qOverP];
418 m_truth_phi0 = generatedTrackPerigee->parameters()[Trk::phi0];
419 m_truth_d0 = generatedTrackPerigee->parameters()[Trk::d0];
420 m_truth_z0 = generatedTrackPerigee->parameters()[Trk::z0];
421 m_truth_theta = generatedTrackPerigee->parameters()[Trk::theta];
422 m_truth_eta = generatedTrackPerigee->eta();
423 m_truth_prod_x = genparptr->production_vertex()->position().x();
424 m_truth_prod_y = genparptr->production_vertex()->position().y();
425 m_truth_prod_z = genparptr->production_vertex()->position().z();
426
427 delete generatedTrackPerigee;
428 m_truth_pt = 1. / std::abs(m_truth_qoverpt);
429 m_truth_charge = 1;
430 if (m_truth_qoverpt < 0) m_truth_charge = -1;
431 if (m_truth_phi0 < 0) m_truth_phi0 += 2 * M_PI;
432 ATH_MSG_DEBUG("Found matched truth track with phi, PT = " << m_truth_phi0 << ", " << m_truth_pt);
433 }
434 }
435 }
436 }
437 }
438 }
439
440 return flag;
441 }
442
443//I think this is never used!
444 const Trk::TrackParameters* DetailedIDNtupleTool::perigeeParameter(const Trk::AlignTrack* alignTrack) const {
445 const Trk::TrackParameters* aMeasPer {};
446
447 for (const auto i:*(alignTrack->trackStateOnSurfaces())) {
448 if (i->type(Trk::TrackStateOnSurface::Perigee)) {
449 aMeasPer = (i->trackParameters());
450 break;
451 }
452 }
453 if (not aMeasPer) {
454 ATH_MSG_ERROR("Could not get Trk::MeasuredPerigee of the alignTrack");
455 }
456 return aMeasPer;
457 }
458
461
462//________________________________________________________________________
465
468
469//________________________________________________________________________
472
473//________________________________________________________________________
474 void DetailedIDNtupleTool::initializeNtuple() {
475 //m_file = new TFile((m_filepath+m_filename).c_str(), "RECREATE");
476 m_file->cd();
477 m_tree = new TTree("IDAlign", "Inner Detector Alignment Ntuple");
478
479 m_tree->Branch("run", &m_runNumber, "run/I");
480 m_tree->Branch("evt", &m_evtNumber, "evt/I");
481
482 m_tree->Branch("xvtx", &m_xvtx, "xvtx/D");
483 m_tree->Branch("yvtx", &m_yvtx, "yvtx/D");
484 m_tree->Branch("zvtx", &m_zvtx, "zvtx/D");
485 m_tree->Branch("d0", &m_d0, "d0/D");
486 m_tree->Branch("z0", &m_z0, "z0/D");
487 m_tree->Branch("phi0", &m_phi0, "phi0/D");
488 m_tree->Branch("theta", &m_theta, "theta/D");
489 m_tree->Branch("qoverp", &m_qoverp, "qoverp/D");
490 m_tree->Branch("pt", &m_pt, "pt/D");
491 m_tree->Branch("eta", &m_eta, "eta/D");
492 m_tree->Branch("chi2", &m_chi2, "chi2/D");
493 m_tree->Branch("ndof", &m_ndof, "ndof/I");
494 m_tree->Branch("chi2prob", &m_chi2prob, "chi2prob/D");
495 m_tree->Branch("err_d0", &m_err_d0, "err_d0/D");
496 m_tree->Branch("err_z0", &m_err_z0, "err_z0/D");
497 m_tree->Branch("err_phi0", &m_err_phi0, "err_phi0/D");
498 m_tree->Branch("err_theta", &m_err_theta, "err_theta/D");
499 m_tree->Branch("err_qoverp", &m_err_qoverp, "err_qoverp/D");
500
501 if (m_storeTruth) {
502 m_tree->Branch("m_truth_prod_x", &m_truth_prod_x, "truth_prod_x/D");
503 m_tree->Branch("m_truth_prod_y", &m_truth_prod_y, "truth_prod_y/D");
504 m_tree->Branch("m_truth_prod_z", &m_truth_prod_z, "truth_prod_z/D");
505 m_tree->Branch("truth_d0", &m_truth_d0, "truth_d0/D");
506 m_tree->Branch("truth_z0", &m_truth_z0, "truth_z0/D");
507 m_tree->Branch("truth_phi0", &m_truth_phi0, "truth_phi0/D");
508 m_tree->Branch("truth_theta", &m_truth_theta, "truth_theta/D");
509 m_tree->Branch("truth_qoverp", &m_truth_qoverp, "truth_qoverp/D");
510 m_tree->Branch("truth_pt", &m_truth_pt, "truth_pt/D");
511 m_tree->Branch("truth_eta", &m_truth_eta, "truth_eta/D");
512 }
513
514 m_tree->Branch("original_xvtx", &m_original_xvtx, "original_xvtx/D");
515 m_tree->Branch("original_yvtx", &m_original_yvtx, "original_yvtx/D");
516 m_tree->Branch("original_zvtx", &m_original_zvtx, "original_zvtx/D");
517 m_tree->Branch("original_d0", &m_original_d0, "original_d0/D");
518 m_tree->Branch("original_z0", &m_original_z0, "original_z0/D");
519 m_tree->Branch("original_phi0", &m_original_phi0, "original_phi0/D");
520 m_tree->Branch("original_theta", &m_original_theta, "original_theta/D");
521 m_tree->Branch("original_qoverp", &m_original_qoverp, "original_qoverp/D");
522 m_tree->Branch("original_pt", &m_original_pt, "original_pt/D");
523 m_tree->Branch("original_eta", &m_original_eta, "original_eta/D");
524 m_tree->Branch("original_chi2", &m_original_chi2, "original_chi2/D");
525 m_tree->Branch("original_ndof", &m_original_ndof, "original_ndof/I");
526 m_tree->Branch("original_chi2prob", &m_original_chi2prob, "original_chi2prob/D");
527
528 m_tree->Branch("original_err_d0", &m_original_err_d0, "original_err_d0/D");
529 m_tree->Branch("original_err_z0", &m_original_err_z0, "original_err_z0/D");
530 m_tree->Branch("original_err_phi0", &m_original_err_phi0, "original_err_phi0/D");
531 m_tree->Branch("original_err_theta", &m_original_err_theta, "original_err_theta/D");
532 m_tree->Branch("original_err_qoverp", &m_original_err_qoverp, "original_err_qoverp/D");
533
534 m_tree->Branch("toRef_d0", &m_toRef_d0, "toRef_d0/D");
535 m_tree->Branch("toRef_z0", &m_toRef_z0, "toRef_z0/D");
536 m_tree->Branch("toRef_phi0", &m_toRef_phi0, "toRef_phi0/D");
537 m_tree->Branch("toRef_theta", &m_toRef_theta, "toRef_theta/D");
538 m_tree->Branch("toRef_qoverp", &m_toRef_qoverp, "toRef_qoverp/D");
539 m_tree->Branch("original_toRef_d0", &m_original_toRef_d0, "original_toRef_d0/D");
540 m_tree->Branch("original_toRef_z0", &m_original_toRef_z0, "original_toRef_z0/D");
541 m_tree->Branch("original_toRef_phi0", &m_original_toRef_phi0, "original_toRef_phi0/D");
542 m_tree->Branch("original_toRef_theta", &m_original_toRef_theta, "original_toRef_theta/D");
543 m_tree->Branch("original_toRef_qoverp", &m_original_toRef_qoverp, "original_toRef_qoverp/D");
544
545 return;
546 }
547} // end namespace
M_PI
#define M_PI
Definition ActiveFraction.h:11
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_FATAL
#define ATH_MSG_FATAL(x)
Definition AthMsgStreamMacros.h:34
ATH_MSG_INFO
#define ATH_MSG_INFO(x)
Definition AthMsgStreamMacros.h:31
ATH_MSG_WARNING
#define ATH_MSG_WARNING(x)
Definition AthMsgStreamMacros.h:32
ATH_MSG_DEBUG
#define ATH_MSG_DEBUG(x)
Definition AthMsgStreamMacros.h:29
AmgSymMatrix
#define AmgSymMatrix(dim)
Definition EventPrimitives.h:50
RIO_OnTrack.h
TrackCollection
DataVector< Trk::Track > TrackCollection
This typedef represents a collection of Trk::Track objects.
Definition TrackCollection.h:19
AthAlgTool::AthAlgTool
AthAlgTool(const std::string &type, const std::string &name, const IInterface *parent)
Constructor with parameters:
Definition AthAlgTool.cxx:16
AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T, V, H > &t)
Definition AthCommonDataStore.h:145
AthCommonDataStore< AthCommonMsg< AlgTool > >::evtStore
ServiceHandle< StoreGateSvc > & evtStore()
Definition AthCommonDataStore.h:85
DataVector::size
size_type size() const noexcept
Returns the number of elements in the collection.
InDet::DetailedIDNtupleTool::m_storeTruth
bool m_storeTruth
retrieve the truth information
Definition DetailedIDNtupleTool.h:71
InDet::DetailedIDNtupleTool::showStatistics
virtual void showStatistics() override
write statistics out to log file
Definition DetailedIDNtupleTool.cxx:470
InDet::DetailedIDNtupleTool::m_filename
std::string m_filename
name of ntuple file
Definition DetailedIDNtupleTool.h:77
InDet::DetailedIDNtupleTool::m_storeConstrainedOnly
bool m_storeConstrainedOnly
onlystore the AlignTrack which is normally refitted
Definition DetailedIDNtupleTool.h:70
InDet::DetailedIDNtupleTool::dumpTrack
virtual void dumpTrack(const EventContext &ctx, int itrk, const Trk::AlignTrack *alignTrack) override
fills track information to ntuple
Definition DetailedIDNtupleTool.cxx:106
InDet::DetailedIDNtupleTool::DetailedIDNtupleTool
DetailedIDNtupleTool(const std::string &type, const std::string &name, const IInterface *parent)
Definition DetailedIDNtupleTool.cxx:38
InDet::DetailedIDNtupleTool::fillNtuple
virtual StatusCode fillNtuple() override
writes trees and histograms to ntuple
Definition DetailedIDNtupleTool.cxx:91
InDet::DetailedIDNtupleTool::m_extrapolator
PublicToolHandle< Trk::IExtrapolator > m_extrapolator
track extrapolator
Definition DetailedIDNtupleTool.h:67
InDet::DetailedIDNtupleTool::storeHitmap
virtual void storeHitmap() override
stores hitmap for writing to ntuple
Definition DetailedIDNtupleTool.cxx:459
InDet::DetailedIDNtupleTool::m_trackSumTool
ToolHandle< Trk::ITrackSummaryTool > m_trackSumTool
Pointer to track summary tool.
Definition DetailedIDNtupleTool.h:62
InDet::DetailedIDNtupleTool::m_storeNormalRefittedOnly
bool m_storeNormalRefittedOnly
only store the AlignTrack which is normally refitted
Definition DetailedIDNtupleTool.h:69
InDet::DetailedIDNtupleTool::finalize
virtual StatusCode finalize() override
Definition DetailedIDNtupleTool.cxx:100
InDet::DetailedIDNtupleTool::fillSummary
virtual void fillSummary() override
fills ntuple with event and track summary information
Definition DetailedIDNtupleTool.cxx:466
InDet::DetailedIDNtupleTool::m_filepath
std::string m_filepath
path to ntuple file
Definition DetailedIDNtupleTool.h:78
InDet::DetailedIDNtupleTool::m_truthToTrack
PublicToolHandle< Trk::ITruthToTrack > m_truthToTrack
the truth to track Tool
Definition DetailedIDNtupleTool.h:65
InDet::DetailedIDNtupleTool::fillHitmap
virtual void fillHitmap() override
fills ntuple with hit information
Definition DetailedIDNtupleTool.cxx:463
InDet::DetailedIDNtupleTool::m_alignModuleTool
ToolHandle< Trk::IAlignModuleTool > m_alignModuleTool
Pointer to AlignmModuleTool.
Definition DetailedIDNtupleTool.h:63
InDet::DetailedIDNtupleTool::perigeeParameter
const Trk::TrackParameters * perigeeParameter(const Trk::AlignTrack *track) const
Definition DetailedIDNtupleTool.cxx:444
InDet::DetailedIDNtupleTool::initialize
virtual StatusCode initialize() override
Definition DetailedIDNtupleTool.cxx:70
InDet::DetailedIDNtupleTool::retrieveTruthInfo
bool retrieveTruthInfo(const Trk::AlignTrack *trk)
Definition DetailedIDNtupleTool.cxx:349
InDet::DetailedIDNtupleTool::m_matchProbability
double m_matchProbability
the probabililty cut in the truth matching
Definition DetailedIDNtupleTool.h:72
TrackTruth
MC particle associated with a reco track + the quality of match.
Definition TrackTruth.h:14
TrackTruth::particleLink
const HepMcParticleLink & particleLink() const
Definition TrackTruth.h:26
Trk::AlignTrack::originalTrack
const Track * originalTrack() const
retrieve pointer to original track
Definition AlignTrack.h:90
Trk::AlignTrack::NormalRefitted
@ NormalRefitted
normally refitted, without adding any pseudo-measurement
Definition AlignTrack.h:48
Trk::AlignTrack::BeamspotConstrained
@ BeamspotConstrained
refitted with beamspot constraint
Definition AlignTrack.h:49
Trk::AlignTrack::type
AlignTrackType type() const
get and set the refit type
Definition AlignTrack.h:96
Trk::FitQuality
Class to represent and store fit qualities from track reconstruction in terms of and number of degre...
Definition FitQuality.h:97
Trk::FitQuality::numberDoF
int numberDoF() const
returns the number of degrees of freedom of the overall track or vertex fit as integer
Definition FitQuality.h:60
Trk::FitQuality::chiSquared
double chiSquared() const
returns the of the overall track fit
Definition FitQuality.h:56
Trk::MeasurementBase
This class is the pure abstract base class for all fittable tracking measurements.
Definition MeasurementBase.h:58
Trk::MeasurementBase::associatedSurface
virtual const Surface & associatedSurface() const =0
Interface method to get the associated Surface.
Trk::ParametersBase::eta
double eta() const
Access method for pseudorapidity - from momentum.
Trk::ParametersBase::momentum
const Amg::Vector3D & momentum() const
Access method for the momentum.
Trk::ParametersBase::position
const Amg::Vector3D & position() const
Access method for the position.
Trk::ParametersT::clone
virtual ParametersT< DIM, T, S > * clone() const override final
Virtual clone.
Trk::ParametersT::associatedSurface
virtual const S & associatedSurface() const override final
Access to the Surface method.
Trk::PerigeeSurface
Class describing the Line to which the Perigee refers to.
Definition PerigeeSurface.h:43
Trk::Surface::center
const Amg::Vector3D & center() const
Returns the center position of the Surface.
Trk::TrackStateOnSurface::Perigee
@ Perigee
This represents a perigee, and so will contain a Perigee object only.
Definition TrackStateOnSurface.h:117
Trk::Track::trackStateOnSurfaces
const Trk::TrackStates * trackStateOnSurfaces() const
return a pointer to a const DataVector of const TrackStateOnSurfaces.
Trk::Track::measurementsOnTrack
const DataVector< const MeasurementBase > * measurementsOnTrack() const
return a pointer to a vector of MeasurementBase (NOT including any that come from outliers).
Definition Tracking/TrkEvent/TrkTrack/src/Track.cxx:178
Trk::Track::perigeeParameters
const Perigee * perigeeParameters() const
return Perigee.
Definition Tracking/TrkEvent/TrkTrack/src/Track.cxx:163
Trk::Track::fitQuality
const FitQuality * fitQuality() const
return a pointer to the fit quality const-overload
Trk::VertexOnTrack
Class to handle Vertex On Tracks, it inherits from the common MeasurementBase.
Definition VertexOnTrack.h:45
xAOD::EventInfo_v1::runNumber
uint32_t runNumber() const
The current event's run number.
xAOD::EventInfo_v1::eventNumber
uint64_t eventNumber() const
The current event's event number.
release
static std::string release
Definition computils.h:50
Amg::error
double error(const Amg::MatrixX &mat, int index)
return diagonal error of the matrix caller should ensure the matrix is symmetric and the index is in ...
Definition EventPrimitivesHelpers.h:40
Amg::Vector3D
Eigen::Matrix< double, 3, 1 > Vector3D
Definition GeoPrimitives.h:47
HepMC::ConstGenParticlePtr
const GenParticle * ConstGenParticlePtr
Definition GenParticle.h:38
InDet
Primary Vertex Finder.
Definition VP1ErrorUtils.h:36
Trk::Perigee
ParametersT< TrackParametersDim, Charged, PerigeeSurface > Perigee
Definition Tracking/TrkEvent/TrkParameters/TrkParameters/TrackParameters.h:33
Trk::phi0
@ phi0
Definition ParamDefs.h:65
Trk::theta
@ theta
Definition ParamDefs.h:66
Trk::qOverP
@ qOverP
perigee
Definition ParamDefs.h:67
Trk::d0
@ d0
Definition ParamDefs.h:63
Trk::z0
@ z0
Definition ParamDefs.h:64
Trk::TrackParameters
ParametersBase< TrackParametersDim, Charged > TrackParameters
Definition Tracking/TrkEvent/TrkParameters/TrkParameters/TrackParameters.h:27
xAOD::EventInfo
EventInfo_v1 EventInfo
Definition of the latest event info version.
Definition IEventInfoCnvTool.h:16
type
Generated on for ATLAS Offline Software by doxygen 1.16.1