Fork me on GitHub

source: svn/trunk/modules/MomentumSmearing.cc@ 1383

Last change on this file since 1383 was 911, checked in by Pavel Demin, 12 years ago

set smeared particle momentum to its value at vertex

  • Property svn:keywords set to Id Revision Date
File size: 2.7 KB
Line 
1
2/** \class MomentumSmearing
3 *
4 * Performs transverse momentum resolution smearing.
5 *
6 * $Date: 2013-02-13 15:58:53 +0000 (Wed, 13 Feb 2013) $
7 * $Revision: 911 $
8 *
9 *
10 * \author P. Demin - UCL, Louvain-la-Neuve
11 *
12 */
13
14#include "modules/MomentumSmearing.h"
15
16#include "classes/DelphesClasses.h"
17#include "classes/DelphesFactory.h"
18#include "classes/DelphesFormula.h"
19
20#include "ExRootAnalysis/ExRootResult.h"
21#include "ExRootAnalysis/ExRootFilter.h"
22#include "ExRootAnalysis/ExRootClassifier.h"
23
24#include "TMath.h"
25#include "TString.h"
26#include "TFormula.h"
27#include "TRandom3.h"
28#include "TObjArray.h"
29#include "TDatabasePDG.h"
30#include "TLorentzVector.h"
31
32#include <algorithm>
33#include <stdexcept>
34#include <iostream>
35#include <sstream>
36
37using namespace std;
38
39//------------------------------------------------------------------------------
40
41MomentumSmearing::MomentumSmearing() :
42 fFormula(0), fItInputArray(0)
43{
44 fFormula = new DelphesFormula;
45}
46
47//------------------------------------------------------------------------------
48
49MomentumSmearing::~MomentumSmearing()
50{
51 if(fFormula) delete fFormula;
52}
53
54//------------------------------------------------------------------------------
55
56void MomentumSmearing::Init()
57{
58 // read resolution formula
59
60 fFormula->Compile(GetString("ResolutionFormula", "0.0"));
61
62 // import input array
63
64 fInputArray = ImportArray(GetString("InputArray", "ParticlePropagator/stableParticles"));
65 fItInputArray = fInputArray->MakeIterator();
66
67 // create output array
68
69 fOutputArray = ExportArray(GetString("OutputArray", "stableParticles"));
70}
71
72//------------------------------------------------------------------------------
73
74void MomentumSmearing::Finish()
75{
76 if(fItInputArray) delete fItInputArray;
77}
78
79//------------------------------------------------------------------------------
80
81void MomentumSmearing::Process()
82{
83 Candidate *candidate, *mother;
84 Double_t pt, eta, phi;
85
86 fItInputArray->Reset();
87 while((candidate = static_cast<Candidate*>(fItInputArray->Next())))
88 {
89 const TLorentzVector &candidatePosition = candidate->Position;
90 const TLorentzVector &candidateMomentum = candidate->Momentum;
91 eta = candidatePosition.Eta();
92 phi = candidatePosition.Phi();
93 pt = candidateMomentum.Pt();
94
95 // apply smearing formula
96 pt = gRandom->Gaus(pt, fFormula->Eval(pt, eta) * pt);
97
98 if(pt <= 0.0) continue;
99
100 mother = candidate;
101 candidate = static_cast<Candidate*>(candidate->Clone());
102 eta = candidateMomentum.Eta();
103 phi = candidateMomentum.Phi();
104 candidate->Momentum.SetPtEtaPhiE(pt, eta, phi, pt*TMath::CosH(eta));
105 candidate->AddCandidate(mother);
106
107 fOutputArray->Add(candidate);
108 }
109}
110
111//------------------------------------------------------------------------------
Note: See TracBrowser for help on using the repository browser.