Fork me on GitHub

source: git/modules/TrackCovariance.cc@ 4e8e72b

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

added first hit to track

  • Property mode set to 100644
File size: 6.1 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 // Get cylindrical box for fast track simulation
107 //
108 Double_t Rin = fGeometry->GetRmin();
109 Double_t ZinPos = fGeometry->GetZminPos();
110 Double_t ZinNeg = fGeometry->GetZminNeg();
111
112 fItInputArray->Reset();
113 while((candidate = static_cast<Candidate *>(fItInputArray->Next())))
114 {
115
116 // converting to meters
117 particle = static_cast<Candidate *>(candidate->GetCandidates()->At(0));
118
119 // converting to meters
120 const TLorentzVector &candidatePosition = particle->Position*1e-03;
121 const TLorentzVector &candidateMomentum = particle->Momentum;
122
123
124 Bool_t inside = TrkUtil::IsInside(candidatePosition.Vect(), Rin, ZinNeg, ZinPos); // Check if in inner box
125 Bool_t Accept = kTRUE;
126 if(inside) Accept = fCovariance->IsAccepted(candidateMomentum.Vect());
127 else Accept = fCovariance->IsAccepted(candidatePosition.Vect(),candidateMomentum.Vect(), fGeometry);
128 if(!Accept) continue;
129
130 mass = candidateMomentum.M();
131
132 ObsTrk track(candidatePosition.Vect(), candidateMomentum.Vect(), candidate->Charge, fCovariance, fGeometry);
133
134
135 mother = candidate;
136 candidate = static_cast<Candidate*>(candidate->Clone());
137
138 candidate->Momentum.SetVectM(track.GetObsP(), mass);
139
140 // converting back to mm
141 candidate->InitialPosition.SetXYZT(track.GetObsX().X()*1e03,track.GetObsX().Y()*1e03,track.GetObsX().Z()*1e03,candidatePosition.T()*1e03);
142
143 // save full covariance 5x5 matrix internally (D0, phi, Curvature, dz, ctg(theta))
144 candidate->TrackCovariance = track.GetCov();
145
146 pt = candidate->Momentum.Pt();
147 p = candidate->Momentum.P();
148 q = track.GetObsQ();
149 ct = track.GetObsPar()[4];
150
151 candidate->Xd = track.GetObsX().X()*1e03;
152 candidate->Yd = track.GetObsX().Y()*1e03;
153 candidate->Zd = track.GetObsX().Z()*1e03;
154
155 candidate->XFirstHit = track.GetFirstHit().X()*1e03;
156 candidate->YFirstHit = track.GetFirstHit().Y()*1e03;
157 candidate->ZFirstHit = track.GetFirstHit().Z()*1e03;
158
159 candidate->D0 = track.GetObsPar()[0]*1e03;
160 candidate->Phi = track.GetObsPar()[1];
161
162 // inverse of curvature
163 candidate->C = track.GetObsPar()[2]*1e-03;
164 candidate->DZ = track.GetObsPar()[3]*1e03;
165 candidate->CtgTheta = track.GetObsPar()[4];
166 candidate->P = track.GetObsP().Mag();
167 candidate->PT = pt;
168 candidate->Charge = q;
169
170 dd0 = TMath::Sqrt(track.GetCov()(0, 0))*1e03;
171 ddz = TMath::Sqrt(track.GetCov()(3, 3))*1e03;
172 dphi = TMath::Sqrt(track.GetCov()(1, 1));
173 dct = TMath::Sqrt(track.GetCov()(4, 4));
174 dpt = 2 * TMath::Sqrt( track.GetCov()(2, 2))*pt*pt / (0.2998*fBz);
175 dp = TMath::Sqrt((1.+ct*ct)*dpt*dpt + 4*pt*pt*ct*ct*dct*dct/(1.+ct*ct)/(1.+ct*ct));
176 dC = TMath::Sqrt(track.GetCov()(2, 2))*1e-03;
177
178 candidate->ErrorD0 = dd0;
179 candidate->ErrorDZ = ddz;
180 candidate->ErrorP = dp;
181 candidate->ErrorC = dC;
182 candidate->ErrorCtgTheta = dct;
183 candidate->ErrorPhi = dphi;
184 candidate->ErrorPT = dpt;
185 //candidate->TrackResolution = dpt / pt;
186 candidate->TrackResolution = dp / p;
187
188 candidate->AddCandidate(mother);
189
190 fOutputArray->Add(candidate);
191 }
192}
193
194//------------------------------------------------------------------------------
Note: See TracBrowser for help on using the repository browser.