Fork me on GitHub

source: git/modules/ImpactParameterSmearing.cc@ bd33fce

Last change on this file since bd33fce was 341014c, checked in by Pavel Demin <pavel-demin@…>, 6 years ago

apply .clang-format to all .h, .cc and .cpp files

  • Property mode set to 100644
File size: 4.2 KB
Line 
1/*
2 * Delphes: a framework for fast simulation of a generic collider experiment
3 * Copyright (C) 2012-2014 Universite catholique de Louvain (UCL), 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 ImpactParameterSmearing
20 *
21 * Performs transverse impact parameter smearing.
22 *
23 * \author M. Selvaggi - UCL, Louvain-la-Neuve
24 *
25 */
26
27#include "modules/ImpactParameterSmearing.h"
28
29#include "classes/DelphesClasses.h"
30#include "classes/DelphesFactory.h"
31#include "classes/DelphesFormula.h"
32
33#include "ExRootAnalysis/ExRootClassifier.h"
34#include "ExRootAnalysis/ExRootFilter.h"
35#include "ExRootAnalysis/ExRootResult.h"
36
37#include "TDatabasePDG.h"
38#include "TFormula.h"
39#include "TLorentzVector.h"
40#include "TMath.h"
41#include "TObjArray.h"
42#include "TRandom3.h"
43#include "TString.h"
44
45#include <algorithm>
46#include <iostream>
47#include <sstream>
48#include <stdexcept>
49
50using namespace std;
51
52//------------------------------------------------------------------------------
53
54ImpactParameterSmearing::ImpactParameterSmearing() :
55 fFormula(0), fItInputArray(0)
56{
57 fFormula = new DelphesFormula;
58}
59
60//------------------------------------------------------------------------------
61
62ImpactParameterSmearing::~ImpactParameterSmearing()
63{
64 if(fFormula) delete fFormula;
65}
66
67//------------------------------------------------------------------------------
68
69void ImpactParameterSmearing::Init()
70{
71 // read resolution formula
72
73 fFormula->Compile(GetString("ResolutionFormula", "0.0"));
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 ImpactParameterSmearing::Finish()
88{
89 if(fItInputArray) delete fItInputArray;
90}
91
92//------------------------------------------------------------------------------
93
94void ImpactParameterSmearing::Process()
95{
96 Candidate *candidate, *particle, *mother;
97 Double_t xd, yd, zd, d0, sx, sy, sz, dd0;
98 Double_t pt, eta, px, py, phi, e;
99
100 fItInputArray->Reset();
101 while((candidate = static_cast<Candidate *>(fItInputArray->Next())))
102 {
103
104 // take momentum before smearing (otherwise apply double smearing on d0)
105 particle = static_cast<Candidate *>(candidate->GetCandidates()->At(0));
106
107 const TLorentzVector &candidateMomentum = particle->Momentum;
108
109 eta = candidateMomentum.Eta();
110 pt = candidateMomentum.Pt();
111 phi = candidateMomentum.Phi();
112 e = candidateMomentum.E();
113
114 px = candidateMomentum.Px();
115 py = candidateMomentum.Py();
116
117 // calculate coordinates of closest approach to track circle in transverse plane xd, yd, zd
118 xd = candidate->Xd;
119 yd = candidate->Yd;
120 zd = candidate->Zd;
121
122 // calculate smeared values
123 sx = gRandom->Gaus(0.0, fFormula->Eval(pt, eta, phi, e));
124 sy = gRandom->Gaus(0.0, fFormula->Eval(pt, eta, phi, e));
125 sz = gRandom->Gaus(0.0, fFormula->Eval(pt, eta, phi, e));
126
127 xd += sx;
128 yd += sy;
129 zd += sz;
130
131 // calculate impact parameter (after-smearing)
132 d0 = (xd * py - yd * px) / pt;
133
134 dd0 = gRandom->Gaus(0.0, fFormula->Eval(pt, eta, phi, e));
135
136 // fill smeared values in candidate
137 mother = candidate;
138
139 candidate = static_cast<Candidate *>(candidate->Clone());
140 candidate->Xd = xd;
141 candidate->Yd = yd;
142 candidate->Zd = zd;
143
144 candidate->D0 = d0;
145 candidate->ErrorD0 = dd0;
146
147 candidate->AddCandidate(mother);
148 fOutputArray->Add(candidate);
149 }
150}
151
152//------------------------------------------------------------------------------
Note: See TracBrowser for help on using the repository browser.