Fork me on GitHub

source: git/modules/AngularSmearing.cc@ e5ea42e

Last change on this file since e5ea42e was 95aa610, checked in by Michele Selvaggi <michele.selvaggi@…>, 10 years ago

fixed phi, energy dependence in efficiency, smearing, identification, btagging, angular smearing modules

  • Property mode set to 100644
File size: 3.7 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
20/** \class AngularSmearing
21 *
22 * Performs transverse angular resolution smearing.
23 *
24 * \author M. Selvaggi - UCL, Louvain-la-Neuve
25 *
26 */
27
28#include "modules/AngularSmearing.h"
29
30#include "classes/DelphesClasses.h"
31#include "classes/DelphesFactory.h"
32#include "classes/DelphesFormula.h"
33
34#include "ExRootAnalysis/ExRootResult.h"
35#include "ExRootAnalysis/ExRootFilter.h"
36#include "ExRootAnalysis/ExRootClassifier.h"
37
38#include "TMath.h"
39#include "TString.h"
40#include "TFormula.h"
41#include "TRandom3.h"
42#include "TObjArray.h"
43#include "TDatabasePDG.h"
44#include "TLorentzVector.h"
45
46#include <algorithm>
47#include <stdexcept>
48#include <iostream>
49#include <sstream>
50
51using namespace std;
52
53//------------------------------------------------------------------------------
54
55AngularSmearing::AngularSmearing() :
56 fFormulaEta(0), fFormulaPhi(0), fItInputArray(0)
57{
58 fFormulaEta = new DelphesFormula;
59 fFormulaPhi = new DelphesFormula;
60}
61
62//------------------------------------------------------------------------------
63
64AngularSmearing::~AngularSmearing()
65{
66 if(fFormulaEta) delete fFormulaEta;
67 if(fFormulaPhi) delete fFormulaPhi;
68}
69
70//------------------------------------------------------------------------------
71
72void AngularSmearing::Init()
73{
74 // read resolution formula
75
76 fFormulaEta->Compile(GetString("EtaResolutionFormula", "0.0"));
77 fFormulaPhi->Compile(GetString("PhiResolutionFormula", "0.0"));
78
79
80 // import input array
81
82 fInputArray = ImportArray(GetString("InputArray", "ParticlePropagator/stableParticles"));
83 fItInputArray = fInputArray->MakeIterator();
84
85 // create output array
86
87 fOutputArray = ExportArray(GetString("OutputArray", "stableParticles"));
88}
89
90//------------------------------------------------------------------------------
91
92void AngularSmearing::Finish()
93{
94 if(fItInputArray) delete fItInputArray;
95}
96
97//------------------------------------------------------------------------------
98
99void AngularSmearing::Process()
100{
101 Candidate *candidate, *mother;
102 Double_t pt, eta, phi, e;
103
104 fItInputArray->Reset();
105 while((candidate = static_cast<Candidate*>(fItInputArray->Next())))
106 {
107 const TLorentzVector &candidatePosition = candidate->Position;
108 const TLorentzVector &candidateMomentum = candidate->Momentum;
109 eta = candidatePosition.Eta();
110 phi = candidatePosition.Phi();
111 pt = candidateMomentum.Pt();
112 e = candidateMomentum.E();
113
114 // apply smearing formula for eta,phi
115
116 eta = gRandom->Gaus(eta, fFormulaEta->Eval(pt, eta, phi, e));
117 phi = gRandom->Gaus(phi, fFormulaPhi->Eval(pt, eta, phi, e));
118
119 if(pt <= 0.0) continue;
120
121 mother = candidate;
122 candidate = static_cast<Candidate*>(candidate->Clone());
123 eta = candidateMomentum.Eta();
124 phi = candidateMomentum.Phi();
125 candidate->Momentum.SetPtEtaPhiE(pt, eta, phi, pt*TMath::CosH(eta));
126 candidate->AddCandidate(mother);
127
128 fOutputArray->Add(candidate);
129 }
130}
131
132//------------------------------------------------------------------------------
Note: See TracBrowser for help on using the repository browser.