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@…>, 4 years ago

add DCA from TrackCovariance

  • Property mode set to 100644
File size: 4.6 KB
RevLine 
[ff9fb2d9]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 *
[1388e73]23 * \authors P. Demin - UCLouvain, Louvain-la-Neuve
24 * M. Selvaggi - CERN
[ff9fb2d9]25 *
26 */
27
[7d790c1]28//FIXME add reference to Bedeschi-code
[1388e73]29//FIXME make sure about units of P, X
30//FIXME fix pt > 200 GeV issue and angle > 6.41
[7d790c1]31
[ff9fb2d9]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
[1388e73]44#include <iostream>
45#include <sstream>
46
47using namespace std;
48
[ff9fb2d9]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;
[1388e73]97 Double_t mass, p, pt, q, ct;
98 Double_t dd0, ddz, dphi, dct, dp, dpt;
99
[ff9fb2d9]100
101 fItInputArray->Reset();
102 while((candidate = static_cast<Candidate *>(fItInputArray->Next())))
103 {
[1388e73]104 const TLorentzVector &candidatePosition = candidate->InitialPosition;
[ff9fb2d9]105 const TLorentzVector &candidateMomentum = candidate->Momentum;
106
[7b518f0]107 mass = candidateMomentum.M();
[ff9fb2d9]108
109 ObsTrk track(candidatePosition.Vect(), candidateMomentum.Vect(), candidate->Charge, fBz, fCovariance);
110
[46c8df8]111 mother = candidate;
[ff9fb2d9]112 candidate = static_cast<Candidate *>(candidate->Clone());
[1388e73]113
[7b518f0]114 candidate->Momentum.SetVectM(track.GetObsP(), mass);
[1388e73]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
[46c8df8]122 candidate->Xd = track.GetObsX().X();
123 candidate->Yd = track.GetObsX().Y();
124 candidate->Zd = track.GetObsX().Z();
125
[1388e73]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;
[ff9fb2d9]150
151 candidate->AddCandidate(mother);
152
153 fOutputArray->Add(candidate);
154 }
155}
156
157//------------------------------------------------------------------------------
Note: See TracBrowser for help on using the repository browser.