Fork me on GitHub

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

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

fixed momentum cut and generalised to neutrals

  • Property mode set to 100644
File size: 3.5 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 time smearing.
22 *
23 * \author M. Selvaggi - CERN
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
53TimeSmearing::TimeSmearing() :
54 fItInputArray(0), fResolutionFormula(0)
55{
56 fResolutionFormula = new DelphesFormula;
57}
58
59//------------------------------------------------------------------------------
60
61TimeSmearing::~TimeSmearing()
62{
63 if(fResolutionFormula) delete fResolutionFormula;
64}
65
66//------------------------------------------------------------------------------
67
68void TimeSmearing::Init()
69{
70 // read resolution formula
71
72 // read time resolution formula in seconds
73 fResolutionFormula->Compile(GetString("TimeResolution", "30e-12"));
74
75 // import track input array
76 fInputArray = ImportArray(GetString("InputArray", "MuonMomentumSmearing/muons"));
77 fItInputArray = fInputArray->MakeIterator();
78
79
80 // create output array
81 fOutputArray = ExportArray(GetString("OutputArray", "tracks"));
82}
83
84//------------------------------------------------------------------------------
85
86void TimeSmearing::Finish()
87{
88 if(fItInputArray) delete fItInputArray;
89}
90
91//------------------------------------------------------------------------------
92
93void TimeSmearing::Process()
94{
95 Candidate *candidate, *mother;
96 Double_t tf_smeared, tf;
97 Double_t eta, energy;
98 Double_t timeResolution;
99
100 const Double_t c_light = 2.99792458E8;
101
102 fItInputArray->Reset();
103 while((candidate = static_cast<Candidate *>(fItInputArray->Next())))
104 {
105
106 const TLorentzVector &candidateFinalPosition = candidate->Position;
107 const TLorentzVector &candidateMomentum = candidate->Momentum;
108
109 tf = candidateFinalPosition.T() * 1.0E-3 / c_light;
110
111 eta = candidateMomentum.Eta();
112 energy = candidateMomentum.E();
113
114 // apply smearing formula
115 timeResolution = fResolutionFormula->Eval(0.0, eta, 0.0, energy);
116 tf_smeared = gRandom->Gaus(tf, timeResolution);
117
118 mother = candidate;
119 candidate = static_cast<Candidate *>(candidate->Clone());
120
121 candidate->Position.SetT(tf_smeared * 1.0E3 * c_light);
122 candidate->ErrorT = timeResolution * 1.0E3 * c_light;
123
124 candidate->AddCandidate(mother);
125 fOutputArray->Add(candidate);
126 }
127}
Note: See TracBrowser for help on using the repository browser.