Fork me on GitHub

source: git/modules/TimeSmearing.cc@ 6049672

Timing
Last change on this file since 6049672 was 6049672, checked in by Kaan Yüksel Oyulmaz <kaanyukseloyulmaz@…>, 5 years ago

New changes according to last meeting

  • 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 TimeSmearing
20 *
21 * Performs transverse momentum resolution smearing.
22 *
23 * \author P. Demin - UCL, Louvain-la-Neuve
24 *
25 */
26
27#include "modules/TimeSmearing.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
54TimeSmearing::TimeSmearing() :
55 fItInputArray(0), fFormula(0)
56{
57 fFormula = new DelphesFormula;
58}
59
60//------------------------------------------------------------------------------
61
62TimeSmearing::~TimeSmearing()
63{
64 if(fFormula) delete fFormula;
65}
66
67//------------------------------------------------------------------------------
68
69void TimeSmearing::Init()
70{
71 // read resolution formula
72 fFormula->Compile(GetString("TimeResolution", "0.001"));
73
74 // import input array
75 fEtaMax = GetDouble("EtaMax", 6.);
76 fInputArray = ImportArray(GetString("InputArray", "MuonMomentumSmearing/muons"));
77 fItInputArray = fInputArray->MakeIterator();
78 // create output array
79
80 fOutputArray = ExportArray(GetString("OutputArray", "muons"));
81}
82
83//------------------------------------------------------------------------------
84
85void TimeSmearing::Finish()
86{
87 if(fItInputArray) delete fItInputArray;
88}
89
90//------------------------------------------------------------------------------
91
92void TimeSmearing::Process()
93{
94 Candidate *candidate, *mother;
95 Double_t ti, tf_smeared, tf;
96 Double_t pt, eta, phi, e, p, l;
97 Double_t beta_particle;
98 Double_t timeResolution;
99
100
101 const Double_t c_light = 2.99792458E8;
102
103 fItInputArray->Reset();
104 while((candidate = static_cast<Candidate*>(fItInputArray->Next())))
105 {
106 ti = candidate->InitialPosition.T()*1.0E-3/c_light;
107 tf = candidate->Position.T()*1.0E-3/c_light;
108
109 // dummy, only need to properly call TFormula
110 const TLorentzVector &candidatePosition = candidate->Position;
111 const TLorentzVector &candidateMomentum = candidate->Momentum;
112 eta = candidatePosition.Eta();
113 phi = candidatePosition.Phi();
114 pt = candidateMomentum.Pt();
115 e = candidateMomentum.E();
116 p = candidateMomentum.P();
117 beta_particle = p/e;
118 l = candidate->L;
119 timeResolution = fFormula->Eval(e);
120
121 if(candidate->Charge != 0)
122 {
123 tf_smeared = tf + timeResolution*gRandom->Gaus(0, 1);
124 mother = candidate;
125 candidate = static_cast<Candidate*>(candidate->Clone()); // I am not sure that we need these lines !!!
126 candidate->AddCandidate(mother);
127 candidate->InitialPosition.SetT((100+ti)*1.0E3*c_light);
128 candidate->Position.SetT(tf_smeared*1.0E3*c_light);
129 candidate->ErrorT = timeResolution*1.0E3*c_light;
130 fOutputArray->Add(candidate);
131 }
132 else
133 {
134 ti = timeResolution - l*1.0E3/(c_light*beta_particle);
135 candidate->InitialPosition.SetT(ti);
136 candidate->ErrorT = timeResolution*1.0E3*c_light; // Do we need to sum with 100 like in upside ?
137 fOutputArray->Add(candidate);
138 }
139
140 }
141}
142
143//------------------------------------------------------------------------------
Note: See TracBrowser for help on using the repository browser.