Fork me on GitHub

source: git/modules/TrackCovariance.cc@ bd33fce

Last change on this file since bd33fce was c9e7466, checked in by michele <michele.selvaggi@…>, 4 years ago

fixed trailing printouts

  • Property mode set to 100644
File size: 5.4 KB
Line 
1/*
2 * Delphes: a framework for fast simulation of a generic collider experiment
3 * Copyright (C) 2020 Universite catholique de Louvain (UCLouvain), Belgium
4 *
5 * This program is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation, either version 3 of the License, or
8 * (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
17 */
18
19/** \class TrackCovariance
20 *
21 * Smears track parameters according to appropriate covariance matrix.
22 *
23 * \authors F. Bedeschi - INFN Pisa
24* P. Demin - UCLouvain, Louvain-la-Neuve
25 * M. Selvaggi - CERN
26 *
27 *
28 */
29
30//FIXME add reference to Bedeschi-code
31//FIXME make sure about units of P, X
32//FIXME fix pt > 200 GeV issue and angle > 6.41
33
34#include "modules/TrackCovariance.h"
35
36#include "classes/DelphesClasses.h"
37
38#include "TrackCovariance/SolGeom.h"
39#include "TrackCovariance/SolGridCov.h"
40#include "TrackCovariance/ObsTrk.h"
41
42#include "TLorentzVector.h"
43#include "TMath.h"
44#include "TObjArray.h"
45
46#include <iostream>
47#include <sstream>
48
49using namespace std;
50
51//------------------------------------------------------------------------------
52
53TrackCovariance::TrackCovariance() :
54 fGeometry(0), fCovariance(0), fAcx(0), fItInputArray(0)
55{
56 fGeometry = new SolGeom();
57 fCovariance = new SolGridCov();
58}
59
60//------------------------------------------------------------------------------
61
62TrackCovariance::~TrackCovariance()
63{
64 if(fGeometry) delete fGeometry;
65 if(fCovariance) delete fCovariance;
66}
67
68//------------------------------------------------------------------------------
69
70void TrackCovariance::Init()
71{
72 fBz = GetDouble("Bz", 0.0);
73 fGeometry->Read(GetString("DetectorGeometry", ""));
74 fNMinHits = GetInt("NMinHits", 6);
75
76 // load geometry
77 fCovariance->Calc(fGeometry);
78 fCovariance->SetMinHits(fNMinHits);
79 // load geometry
80 fAcx = fCovariance->AccPnt();
81
82 // import input array
83 fInputArray = ImportArray(GetString("InputArray", "TrackMerger/tracks"));
84 fItInputArray = fInputArray->MakeIterator();
85
86 // create output array
87
88 fOutputArray = ExportArray(GetString("OutputArray", "tracks"));
89}
90
91//------------------------------------------------------------------------------
92
93void TrackCovariance::Finish()
94{
95 if(fItInputArray) delete fItInputArray;
96}
97
98//------------------------------------------------------------------------------
99
100void TrackCovariance::Process()
101{
102 Candidate *candidate, *mother, *particle;
103 Double_t mass, p, pt, q, ct;
104 Double_t dd0, ddz, dphi, dct, dp, dpt, dC;
105
106
107 fItInputArray->Reset();
108 while((candidate = static_cast<Candidate *>(fItInputArray->Next())))
109 {
110
111 // converting to meters
112 particle = static_cast<Candidate *>(candidate->GetCandidates()->At(0));
113
114 // converting to meters
115 const TLorentzVector &candidatePosition = particle->Position*1e-03;
116 const TLorentzVector &candidateMomentum = particle->Momentum;
117
118 if ( !fCovariance->IsAccepted(candidateMomentum.Vect()) ) continue;
119
120 mass = candidateMomentum.M();
121
122 ObsTrk track(candidatePosition.Vect(), candidateMomentum.Vect(), candidate->Charge, fBz, fCovariance);
123
124 mother = candidate;
125 candidate = static_cast<Candidate*>(candidate->Clone());
126
127 candidate->Momentum.SetVectM(track.GetObsP(), mass);
128
129 // converting back to mm
130 candidate->InitialPosition.SetXYZT(track.GetObsX().X()*1e03,track.GetObsX().Y()*1e03,track.GetObsX().Z()*1e03,candidatePosition.T()*1e03);
131
132 // save full covariance 5x5 matrix internally (D0, phi, Curvature, dz, ctg(theta))
133 candidate->TrackCovariance = track.GetCov();
134
135 pt = candidate->Momentum.Pt();
136 p = candidate->Momentum.P();
137 q = track.GetObsQ();
138 ct = track.GetObsPar()[4];
139
140 candidate->Xd = track.GetObsX().X()*1e03;
141 candidate->Yd = track.GetObsX().Y()*1e03;
142 candidate->Zd = track.GetObsX().Z()*1e03;
143
144 candidate->D0 = track.GetObsPar()[0]*1e03;
145 candidate->Phi = track.GetObsPar()[1];
146
147 // inverse of curvature
148 candidate->C = track.GetObsPar()[2]*1e-03;
149 candidate->DZ = track.GetObsPar()[3]*1e03;
150 candidate->CtgTheta = track.GetObsPar()[4];
151 candidate->P = track.GetObsP().Mag();
152 candidate->PT = pt;
153 candidate->Charge = q;
154
155 dd0 = TMath::Sqrt(track.GetCov()(0, 0))*1e03;
156 ddz = TMath::Sqrt(track.GetCov()(3, 3))*1e03;
157 dphi = TMath::Sqrt(track.GetCov()(1, 1));
158 dct = TMath::Sqrt(track.GetCov()(4, 4));
159 dpt = 2 * TMath::Sqrt( track.GetCov()(2, 2))*pt*pt / (0.2998*fBz);
160 dp = TMath::Sqrt((1.+ct*ct)*dpt*dpt + 4*pt*pt*ct*ct*dct*dct/(1.+ct*ct)/(1.+ct*ct));
161 dC = TMath::Sqrt(track.GetCov()(2, 2))*1e-03;
162
163 candidate->ErrorD0 = dd0;
164 candidate->ErrorDZ = ddz;
165 candidate->ErrorP = dp;
166 candidate->ErrorC = dC;
167 candidate->ErrorCtgTheta = dct;
168 candidate->ErrorPhi = dphi;
169 candidate->ErrorPT = dpt;
170 //candidate->TrackResolution = dpt / pt;
171 candidate->TrackResolution = dp / p;
172
173 candidate->AddCandidate(mother);
174
175 fOutputArray->Add(candidate);
176 }
177}
178
179//------------------------------------------------------------------------------
Note: See TracBrowser for help on using the repository browser.