Fork me on GitHub

source: git/modules/TrackCovariance.cc @ 46c8df8

Last change on this file since 46c8df8 was 46c8df8, checked in by Michele Selvaggi <michele.selvaggi@…>, 3 months ago

add DCA from TrackCovariance?

  • Property mode set to 100644
File size: 4.6 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 P. Demin - UCLouvain, Louvain-la-Neuve
24 *           M. Selvaggi - CERN
25 *
26 */
27
28//FIXME add reference to Bedeschi-code
29//FIXME make sure about units of P, X
30//FIXME fix pt > 200 GeV issue and angle > 6.41
31
32#include "modules/TrackCovariance.h"
33
34#include "classes/DelphesClasses.h"
35
36#include "TrackCovariance/SolGeom.h"
37#include "TrackCovariance/SolGridCov.h"
38#include "TrackCovariance/ObsTrk.h"
39
40#include "TLorentzVector.h"
41#include "TMath.h"
42#include "TObjArray.h"
43
44#include <iostream>
45#include <sstream>
46
47using namespace std;
48
49//------------------------------------------------------------------------------
50
51TrackCovariance::TrackCovariance() :
52  fGeometry(0), fCovariance(0), fItInputArray(0)
53{
54  fGeometry = new SolGeom();
55  fCovariance = new SolGridCov();
56}
57
58//------------------------------------------------------------------------------
59
60TrackCovariance::~TrackCovariance()
61{
62  if(fGeometry) delete fGeometry;
63  if(fCovariance) delete fCovariance;
64}
65
66//------------------------------------------------------------------------------
67
68void TrackCovariance::Init()
69{
70  fBz = GetDouble("Bz", 0.0);
71  fGeometry->Read(GetString("DetectorGeometry", ""));
72
73  fCovariance->Calc(fGeometry);
74
75  // import input array
76
77  fInputArray = ImportArray(GetString("InputArray", "TrackMerger/tracks"));
78  fItInputArray = fInputArray->MakeIterator();
79
80  // create output array
81
82  fOutputArray = ExportArray(GetString("OutputArray", "tracks"));
83}
84
85//------------------------------------------------------------------------------
86
87void TrackCovariance::Finish()
88{
89  if(fItInputArray) delete fItInputArray;
90}
91
92//------------------------------------------------------------------------------
93
94void TrackCovariance::Process()
95{
96  Candidate *candidate, *mother;
97  Double_t mass, p, pt, q, ct;
98  Double_t dd0, ddz, dphi, dct, dp, dpt;
99 
100
101  fItInputArray->Reset();
102  while((candidate = static_cast<Candidate *>(fItInputArray->Next())))
103  {
104    const TLorentzVector &candidatePosition = candidate->InitialPosition;
105    const TLorentzVector &candidateMomentum = candidate->Momentum;
106
107    mass = candidateMomentum.M();
108
109    ObsTrk track(candidatePosition.Vect(), candidateMomentum.Vect(), candidate->Charge, fBz, fCovariance);
110
111    mother    = candidate;
112    candidate = static_cast<Candidate *>(candidate->Clone());
113
114    candidate->Momentum.SetVectM(track.GetObsP(), mass);
115    candidate->InitialPosition.SetXYZT(track.GetObsX().X(),track.GetObsX().Y(),track.GetObsX().Z(),candidatePosition.T());
116
117    pt = candidate->Momentum.Pt();
118    p  = candidate->Momentum.P();
119    q  = track.GetObsQ();
120    ct = track.GetObsPar()[4];
121
122    candidate->Xd = track.GetObsX().X();
123    candidate->Yd = track.GetObsX().Y();
124    candidate->Zd = track.GetObsX().Z();
125   
126    candidate->D0 = track.GetObsPar()[0];
127    candidate->DZ = track.GetObsPar()[3];
128    candidate->P  = track.GetObsP().Mag();
129    candidate->CtgTheta = track.GetObsPar()[4];
130    candidate->Phi = track.GetObsPar()[1];
131
132    candidate->PT = pt;
133    candidate->Charge = q;
134
135    dd0       = TMath::Sqrt(track.GetCov()(0, 0));
136    ddz       = TMath::Sqrt(track.GetCov()(3, 3));
137    dphi      = TMath::Sqrt(track.GetCov()(1, 1));
138    dct       = TMath::Sqrt(track.GetCov()(4, 4));
139    dpt       = 2 * TMath::Sqrt( track.GetCov()(2, 2))*pt*pt / (0.2998*fBz);
140    dp        = TMath::Sqrt((1.+ct*ct)*dpt*dpt + 4*pt*pt*ct*ct*dct*dct/(1.+ct*ct)/(1.+ct*ct));
141
142    candidate->ErrorD0 = dd0;
143    candidate->ErrorDZ = ddz;
144    candidate->ErrorP = dp;
145    candidate->ErrorCtgTheta = dct;
146    candidate->ErrorPhi = dphi;
147    candidate->ErrorPT = dpt;
148    //candidate->TrackResolution = dpt / pt;
149    candidate->TrackResolution = dp / p;
150
151    candidate->AddCandidate(mother);
152
153    fOutputArray->Add(candidate);
154  }
155}
156
157//------------------------------------------------------------------------------
Note: See TracBrowser for help on using the repository browser.