Fork me on GitHub

source: svn/trunk/src/SmearUtil.cc@ 32

Last change on this file since 32 was 32, checked in by severine ovyn, 16 years ago

remove HCALecal
File size: 16.0 KB
Line 
1/*
2 ---- Delphes ----
3 A Fast Simulator for general purpose LHC detector
4 S. Ovyn ~~~~ severine.ovyn@uclouvain.be
5
6 Center for Particle Physics and Phenomenology (CP3)
7 Universite Catholique de Louvain (UCL)
8 Louvain-la-Neuve, Belgium
9*/
10
11/// \file SmearUtil.cc
12/// \brief RESOLution class, and some generic definitions
13
14
15#include "interface/SmearUtil.h"
16#include "TRandom.h"
17
18#include <iostream>
19#include <sstream>
20#include <fstream>
21using namespace std;
22
23//------------------------------------------------------------------------------
24
25RESOLution::RESOLution() {
26
27MAX_TRACKER = 2.5; // tracker coverage
28MAX_CALO_CEN = 3.0; // central calorimeter coverage
29MAX_CALO_FWD = 5.0; // forward calorimeter pseudorapidity coverage
30MAX_MU = 2.4; // muon chambers pseudorapidity coverage
31MIN_CALO_VFWD= 5.2; // very forward calorimeter (if any), like CASTOR
32MAX_CALO_VFWD= 6.6; // very forward calorimeter (if any), like CASTOR
33MIN_ZDC = 8.3; // zero-degree calorimeter, coverage
34
35ZDC_S = 140.; // ZDC distance to IP
36RP220_S = 220; // distance of the RP to the IP, in meters
37RP220_X = 0.002;// distance of the RP to the beam, in meters
38FP420_S = 420; // distance of the RP to the IP, in meters
39FP420_X = 0.004;// distance of the RP to the beam, in meters
40
41
42ELG_Scen = 0.028; // S term for central ECAL
43ELG_Ncen = 0.124 ; // N term for central ECAL
44ELG_Ccen = 0.0026 ; // C term for central ECAL
45ELG_Cfwd = 0.107 ; // S term for forward ECAL
46ELG_Sfwd = 2.084 ; // C term for forward ECAL
47ELG_Nfwd = 0.0 ; // N term for central ECAL
48
49HAD_Secal = 0.05 ; // S term for central ECAL // electromagnetic calorimeter
50HAD_Necal = 0.25 ; // N term for central ECAL
51HAD_Cecal = 0.0055 ; // C term for central ECAL
52HAD_Shcal = 0.91 ; // S term for central HCAL // hadronic calorimeter
53HAD_Nhcal = 0. ; // N term for central HCAL
54HAD_Chcal = 0.038 ; // C term for central HCAL
55HAD_Shf = 2.7 ; // S term for central HF // forward calorimeter
56HAD_Nhf = 0. ; // N term for central HF
57HAD_Chf = 0.13 ; // C term for central HF
58
59MU_SmearPt = 0.01 ;
60
61TAU_CONE_ENERGY = 0.15 ; // Delta R = radius of the cone // for "electromagnetic collimation"
62TAU_EM_COLLIMATION = 0.95;
63TAU_CONE_TRACKS= 0.4 ; //Delta R for tracker isolation for tau's
64PT_TRACK_TAU = 2.0 ; // GeV // 6 GeV ????
65
66
67PT_TRACKS_MIN = 0.9 ; // minimal pt needed to reach the calorimeter, in GeV
68PT_QUARKS_MIN = 2.0 ; // minimal pt needed by quarks to reach the tracker, in GeV (??????)
69TRACKING_EFF = 90;
70
71
72TAGGING_B = 40;
73MISTAGGING_C = 10;
74MISTAGGING_L = 1;
75
76
77CONERADIUS = 0.7; // generic jet radius ; not for tau's !!!
78JETALGO = 1; // 1 for Cone algorithm, 2 for MidPoint algorithm, 3 for SIScone algorithm, 4 for kt algorithm
79// Define Cone algorithm.
80C_SEEDTHRESHOLD = 1.0;
81C_ADJACENCYCUT = 2;
82C_MAXITERATIONS = 100;
83C_IRATCH = 1;
84C_OVERLAPTHRESHOLD = 0.75;
85
86//Define MidPoint algorithm.
87M_SEEDTHRESHOLD = 1.0;
88M_CONEAREAFRACTION = 0.25;
89M_MAXPAIRSIZE = 2;
90M_MAXITERATIONS = 100;
91M_OVERLAPTHRESHOLD = 0.75;
92
93}
94
95//------------------------------------------------------------------------------
96void RESOLution::ReadDataCard(const string datacard) {
97
98 string temp_string;
99 istringstream curstring;
100
101 ifstream fichier_a_lire(datacard.c_str());
102 if(!fichier_a_lire.good()) {
103 cout << datacard << "Datadard " << datacard << " not found, use default values" << endl;
104 return;
105 }
106
107 while (getline(fichier_a_lire,temp_string)) {
108 curstring.clear(); // needed when using several times istringstream::str(string)
109 curstring.str(temp_string);
110 string varname;
111 float value;
112
113 if(strstr(temp_string.c_str(),"#")) { }
114 else if(strstr(temp_string.c_str(),"MAX_TRACKER")){curstring >> varname >> value; MAX_TRACKER = value;}
115 else if(strstr(temp_string.c_str(),"MAX_CALO_CEN")){curstring >> varname >> value; MAX_CALO_CEN = value;}
116 else if(strstr(temp_string.c_str(),"MAX_CALO_FWD")){curstring >> varname >> value; MAX_CALO_FWD = value;}
117 else if(strstr(temp_string.c_str(),"MAX_MU")){curstring >> varname >> value; MAX_MU = value;}
118 else if(strstr(temp_string.c_str(),"ELG_Scen")){curstring >> varname >> value; ELG_Scen = value;}
119 else if(strstr(temp_string.c_str(),"ELG_Ncen")){curstring >> varname >> value; ELG_Ncen = value;}
120 else if(strstr(temp_string.c_str(),"ELG_Ccen")){curstring >> varname >> value; ELG_Ccen = value;}
121 else if(strstr(temp_string.c_str(),"ELG_Sfwd")){curstring >> varname >> value; ELG_Sfwd = value;}
122 else if(strstr(temp_string.c_str(),"ELG_Cfwd")){curstring >> varname >> value; ELG_Cfwd = value;}
123 else if(strstr(temp_string.c_str(),"ELG_Nfwd")){curstring >> varname >> value; ELG_Nfwd = value;}
124 else if(strstr(temp_string.c_str(),"HAD_Shcal")){curstring >> varname >> value; HAD_Shcal = value;}
125 else if(strstr(temp_string.c_str(),"HAD_Nhcal")){curstring >> varname >> value; HAD_Nhcal = value;}
126 else if(strstr(temp_string.c_str(),"HAD_Chcal")){curstring >> varname >> value; HAD_Chcal = value;}
127 else if(strstr(temp_string.c_str(),"HAD_Shf")){curstring >> varname >> value; HAD_Shf = value;}
128 else if(strstr(temp_string.c_str(),"HAD_Nhf")){curstring >> varname >> value; HAD_Nhf = value;}
129 else if(strstr(temp_string.c_str(),"HAD_Chf")){curstring >> varname >> value; HAD_Chf = value;}
130 else if(strstr(temp_string.c_str(),"MU_SmearPt")){curstring >> varname >> value; MU_SmearPt = value;}
131 else if(strstr(temp_string.c_str(),"TAU_CONE_ENERGY")){curstring >> varname >> value; TAU_CONE_ENERGY = value;}
132 else if(strstr(temp_string.c_str(),"TAU_CONE_TRACKS")){curstring >> varname >> value; TAU_CONE_TRACKS = value;}
133 else if(strstr(temp_string.c_str(),"PT_TRACK_TAU")){curstring >> varname >> value; PT_TRACK_TAU = value;}
134 else if(strstr(temp_string.c_str(),"PT_TRACKS_MIN")){curstring >> varname >> value; PT_TRACKS_MIN = value;}
135 else if(strstr(temp_string.c_str(),"TAGGING_B")){curstring >> varname >> value; TAGGING_B = (int)value;}
136 else if(strstr(temp_string.c_str(),"MISTAGGING_C")){curstring >> varname >> value; MISTAGGING_C = (int)value;}
137 else if(strstr(temp_string.c_str(),"MISTAGGING_L")){curstring >> varname >> value; MISTAGGING_L = (int)value;}
138 else if(strstr(temp_string.c_str(),"CONERADIUS")){curstring >> varname >> value; CONERADIUS = value;}
139 else if(strstr(temp_string.c_str(),"JETALGO")){curstring >> varname >> value; JETALGO = (int)value;}
140 else if(strstr(temp_string.c_str(),"TRACKING_EFF")){curstring >> varname >> value; TRACKING_EFF = (int)value;}
141 }
142
143// Define Cone algorithm.
144 C_SEEDTHRESHOLD = 1.0;
145 C_ADJACENCYCUT = 2;
146 C_MAXITERATIONS = 100;
147 C_IRATCH = 1;
148 C_OVERLAPTHRESHOLD = 0.75;
149
150//Define MidPoint algorithm.
151 M_SEEDTHRESHOLD = 1.0;
152 M_CONEAREAFRACTION = 0.25;
153 M_MAXPAIRSIZE = 2;
154 M_MAXITERATIONS = 100;
155 M_OVERLAPTHRESHOLD = 0.75;
156
157}
158
159
160// **********Provides the smeared TLorentzVector for the electrons********
161// Smears the electron energy, and changes the 4-momentum accordingly
162// different smearing if the electron is central (eta < 2.5) or forward
163void RESOLution::SmearElectron(TLorentzVector &electron) {
164 // the 'electron' variable will be changed by the function
165 float energy = electron.E(); // before smearing
166 float energyS = 0.0; // after smearing // \sigma/E = C + N/E + S/\sqrt{E}
167
168 if(fabs(electron.Eta()) < MAX_TRACKER) { // if the electron is inside the tracker
169 energyS = gRandom->Gaus(energy, sqrt(
170 pow(ELG_Ncen,2) +
171 pow(ELG_Ccen*energy,2) +
172 pow(ELG_Scen*sqrt(energy),2) ));
173 } else { // outside the tracker
174 energyS = gRandom->Gaus(energy, sqrt(
175 pow(ELG_Nfwd,2) +
176 pow(ELG_Cfwd*energy,2) +
177 pow(ELG_Sfwd*sqrt(energy),2) ) );
178 }
179float theta=electron.Theta();
180float phi=electron.Phi();
181 float px = energyS * sin(theta) * cos(phi);
182 float py = energyS * sin(theta) * sin(phi);
183 float pz = energyS * cos(theta);
184 //cout<<" px "<<px<<" py "<<py<<" pz "<<pz<<endl;
185 electron.SetPtEtaPhiE(energyS/cosh(electron.Eta()), electron.Eta(), electron.Phi(), energyS);
186 //cout<<"nous "<<electron.Px()<<" "<<electron.Py()<<" "<<electron.Pz()<<endl;
187 if(electron.E() < 0)electron.SetPxPyPzE(0,0,0,0); // no negative values after smearing !
188}
189
190
191// **********Provides the smeared TLorentzVector for the muons********
192// Smears the muon pT and changes the 4-momentum accordingly
193void RESOLution::SmearMu(TLorentzVector &muon) {
194 // the 'muon' variable will be changed by the function
195 float pt = muon.Pt(); // before smearing
196 float ptS = gRandom->Gaus(pt, MU_SmearPt*pt ); // after smearing // \sigma/E = C + N/E + S/\sqrt{E}
197
198 muon.SetPtEtaPhiE(ptS, muon.Eta(), muon.Phi(), ptS*cosh(muon.Eta()));
199
200 if(muon.E() < 0)muon.SetPxPyPzE(0,0,0,0); // no negative values after smearing !
201}
202
203
204// **********Provides the smeared TLorentzVector for the hadrons********
205// Smears the hadron 4-momentum
206void RESOLution::SmearHadron(TLorentzVector &hadron, const float frac)
207 // the 'hadron' variable will be changed by the function
208 // the 'frac' variable describes the long-living particles. Should be 0.7 for K0S and Lambda, 1. otherwise
209{
210 float energy = hadron.E(); // before smearing
211 float energyS = 0.0; // after smearing // \sigma/E = C + N/E + S/\sqrt{E}
212 float energy_ecal = (1.0 - frac)*energy; // electromagnetic calorimeter
213 float energy_hcal = frac*energy; // hadronic calorimeter
214 // frac takes into account the decay of long-living particles, that decay in the calorimeters
215 // some of the particles decay mostly in the ecal, some mostly in the hcal
216
217 float energyS1,energyS2;
218 if(fabs(hadron.Eta()) < MAX_CALO_CEN) {
219 energyS1 = gRandom->Gaus(energy_hcal, sqrt(
220 pow(HAD_Nhcal,2) +
221 pow(HAD_Chcal*energy_hcal,2) +
222 pow(HAD_Shcal*sqrt(energy_hcal),2) )) ;
223
224
225 energyS2 = gRandom->Gaus(energy_ecal, sqrt(
226 pow(ELG_Ncen,2) +
227 pow(ELG_Ccen*energy_ecal,2) +
228 pow(ELG_Scen*sqrt(energy_ecal),2) ) );
229
230 energyS = ((energyS1>0)?energyS1:0) + ((energyS2>0)?energyS2:0);
231 } else {
232 energyS = gRandom->Gaus(energy, sqrt(
233 pow(HAD_Nhf,2) +
234 pow(HAD_Chf*energy,2) +
235 pow(HAD_Shf*sqrt(energy),2) ));
236 }
237
238
239 hadron.SetPtEtaPhiE(energyS/cosh(hadron.Eta()),hadron.Eta(), hadron.Phi(), energyS);
240
241 if(hadron.E() < 0)hadron.SetPxPyPzE(0,0,0,0);
242}
243
244// **********Provides the energy in the cone of radius TAU_CONE_ENERGY for the tau identification********
245// to be taken into account, a calo tower should
246// 1) have a transverse energy \f$ E_T = \sqrt{E_X^2 + E_Y^2} \f$ above a given threshold
247// 2) be inside a cone with a radius R and the axis defined by (eta,phi)
248double RESOLution::EnergySmallCone(const vector<PhysicsTower> &towers, const float eta, const float phi) {
249 double Energie=0;
250 for(unsigned int i=0; i < towers.size(); i++) {
251 if(towers[i].fourVector.pt() < M_SEEDTHRESHOLD) continue;
252 if((DeltaR(phi,eta,towers[i].fourVector.phi(),towers[i].fourVector.eta()) < TAU_CONE_ENERGY)) {
253 Energie += towers[i].fourVector.E;
254 }
255 }
256 return Energie;
257}
258
259
260// **********Provides the number of tracks in the cone of radius TAU_CONE_TRACKS for the tau identification********
261// to be taken into account, a track should
262// 1) avec a transverse momentum \$f p_T \$ above a given threshold
263// 2) be inside a cone with a radius R and the axis defined by (eta,phi)
264// IMPORTANT REMARK !!!!!
265// previously, the argument 'phi' was before the argument 'eta'
266// this has been changed for consistency with the other functions
267// double check your running code that uses NumTracks !
268unsigned int RESOLution::NumTracks(const vector<TLorentzVector> &tracks, const float pt_track, const float eta, const float phi) {
269 unsigned int numtrack=0;
270 for(unsigned int i=0; i < tracks.size(); i++) {
271 if((tracks[i].Pt() < pt_track )||
272 (DeltaR(phi,eta,tracks[i].Phi(),tracks[i].Eta()) > TAU_CONE_TRACKS)
273 )continue;
274 numtrack++;
275 }
276 return numtrack;
277}
278
279
280//*** Returns the PID of the particle with the highest energy, in a cone with a radius CONERADIUS and an axis (eta,phi) *********
281//used by Btaggedjet
282///// Attention : bug removed => CONERADIUS/2 -> CONERADIUS !!
283int RESOLution::Bjets(const TSimpleArray<TRootGenParticle> &subarray, const float eta, const float phi) {
284 float emax=0;
285 int Ppid=0;
286 if(subarray.GetEntries()>0) {
287 for(int i=0; i < subarray.GetEntries();i++) { // should have pt>PT_JETMIN and a small cone radius (r<CONE_JET)
288 float genDeltaR = DeltaR(subarray[i]->Phi,subarray[i]->Eta,phi,eta);
289 if(genDeltaR < CONERADIUS && subarray[i]->E > emax) {
290 emax=subarray[i]->E;
291 Ppid=abs(subarray[i]->PID);
292 }
293 }
294 }
295 return Ppid;
296}
297
298
299//******************** Simulates the b-tagging efficiency for real bjet, or the misendentification for other jets****************
300bool RESOLution::Btaggedjet(const TLorentzVector &JET, const TSimpleArray<TRootGenParticle> &subarray) {
301 if( rand()%100 < (TAGGING_B+1) && Bjets(subarray,JET.Eta(),JET.Phi())==pB ) return true; // b-tag of b-jets is 40%
302 else if( rand()%100 < (MISTAGGING_C+1) && Bjets(subarray,JET.Eta(),JET.Phi())==pC ) return true; // b-tag of c-jets is 10%
303 else if( rand()%100 < (MISTAGGING_L+1) && Bjets(subarray,JET.Eta(),JET.Phi())!=0) return true; // b-tag of light jets is 1%
304 return false;
305}
306
307//***********************Isolation criteria***********************
308//****************************************************************
309bool RESOLution::Isolation(Float_t phi,Float_t eta,const vector<TLorentzVector> &tracks,float PT_TRACK2)
310{
311 bool isolated = false;
312 Float_t deltar=5000.; // Initial value; should be high; no further repercussion
313 // loop on all final charged particles, with p_t >2, close enough from the electron
314 for(unsigned int i=0; i < tracks.size(); i++)
315 {
316 if(tracks[i].Pt() < PT_TRACK2)continue;
317 Float_t genDeltaR = DeltaR(phi,eta,tracks[i].Phi(),tracks[i].Eta()); // slower to evaluate
318 if(
319 (genDeltaR > deltar) ||
320 (genDeltaR==0)
321 ) continue ;
322 deltar=genDeltaR;
323 }
324 if(deltar > 0.5)isolated = true; // returns the closest distance
325 return isolated;
326}
327
328
329//**************************** Returns the delta Phi ****************************
330float DeltaPhi(const float phi1, const float phi2) {
331 float deltaphi=phi1-phi2; // in here, -PI < phi < PI
332 if(fabs(deltaphi) > PI) deltaphi=2.*PI-fabs(deltaphi);// put deltaphi between 0 and PI
333 else deltaphi=fabs(deltaphi);
334
335 return deltaphi;
336}
337
338//**************************** Returns the delta R****************************
339float DeltaR(const float phi1, const float eta1, const float phi2, const float eta2) {
340 return sqrt(pow(DeltaPhi(phi1,phi2),2) + pow(eta1-eta2,2));
341}
342
343int sign(const int myint) {
344 if (myint >0) return 1;
345 else if (myint <0) return -1;
346 else return 0;
347}
348
349int sign(const float myfloat) {
350 if (myfloat >0) return 1;
351 else if (myfloat <0) return -1;
352 else return 0;
353}
354
355
356float Charge(const long int pid) {
357 // source: RPP chap 34 Monte Carlo Particle Numbering Scheme
358/* switch (abs(pid)) {
359 case 1: case 3: case 5: case 7: return (float) sign(pid)*(-1/3); break; // d, s, b, b'
360 case 2: case 4: case 6: case 8: return (float) sign(pid)*2/3; break; // u, c, t, t'
361
362 case 11: case 13: case 15: return (float) sign(pid)*(-1); break; // e, mu, tau
363 case 12: case 14: case 16: return (float) 0; break; // nu_e, nu_mu, nu_tau
364
365 case 9: case 21: case 22: case 23: case 25:
366 case 32: case 33: case 35: case 36: return (float) 0; break; // neutral gauge/higgs bosons
367 case 24: case 34: case 37: return (float) sign(pid); break; // charged gauge/higgs bosons
368 }
369*/
370 return 0;
371}
Note: See TracBrowser for help on using the repository browser.