Fork me on GitHub

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

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

add isolation
File size: 16.3 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_Secal")){curstring >> varname >> value; HAD_Secal = value;}
125 else if(strstr(temp_string.c_str(),"HAD_Necal")){curstring >> varname >> value; HAD_Necal = value;}
126 else if(strstr(temp_string.c_str(),"HAD_Cecal")){curstring >> varname >> value; HAD_Cecal = value;}
127 else if(strstr(temp_string.c_str(),"HAD_Shcal")){curstring >> varname >> value; HAD_Shcal = value;}
128 else if(strstr(temp_string.c_str(),"HAD_Nhcal")){curstring >> varname >> value; HAD_Nhcal = value;}
129 else if(strstr(temp_string.c_str(),"HAD_Chcal")){curstring >> varname >> value; HAD_Chcal = value;}
130 else if(strstr(temp_string.c_str(),"HAD_Shf")){curstring >> varname >> value; HAD_Shf = value;}
131 else if(strstr(temp_string.c_str(),"HAD_Nhf")){curstring >> varname >> value; HAD_Nhf = value;}
132 else if(strstr(temp_string.c_str(),"HAD_Chf")){curstring >> varname >> value; HAD_Chf = value;}
133 else if(strstr(temp_string.c_str(),"MU_SmearPt")){curstring >> varname >> value; MU_SmearPt = value;}
134 else if(strstr(temp_string.c_str(),"TAU_CONE_ENERGY")){curstring >> varname >> value; TAU_CONE_ENERGY = value;}
135 else if(strstr(temp_string.c_str(),"TAU_CONE_TRACKS")){curstring >> varname >> value; TAU_CONE_TRACKS = value;}
136 else if(strstr(temp_string.c_str(),"PT_TRACK_TAU")){curstring >> varname >> value; PT_TRACK_TAU = value;}
137 else if(strstr(temp_string.c_str(),"PT_TRACKS_MIN")){curstring >> varname >> value; PT_TRACKS_MIN = value;}
138 else if(strstr(temp_string.c_str(),"TAGGING_B")){curstring >> varname >> value; TAGGING_B = (int)value;}
139 else if(strstr(temp_string.c_str(),"MISTAGGING_C")){curstring >> varname >> value; MISTAGGING_C = (int)value;}
140 else if(strstr(temp_string.c_str(),"MISTAGGING_L")){curstring >> varname >> value; MISTAGGING_L = (int)value;}
141 else if(strstr(temp_string.c_str(),"CONERADIUS")){curstring >> varname >> value; CONERADIUS = value;}
142 else if(strstr(temp_string.c_str(),"JETALGO")){curstring >> varname >> value; JETALGO = (int)value;}
143 else if(strstr(temp_string.c_str(),"TRACKING_EFF")){curstring >> varname >> value; TRACKING_EFF = (int)value;}
144 }
145
146// Define Cone algorithm.
147 C_SEEDTHRESHOLD = 1.0;
148 C_ADJACENCYCUT = 2;
149 C_MAXITERATIONS = 100;
150 C_IRATCH = 1;
151 C_OVERLAPTHRESHOLD = 0.75;
152
153//Define MidPoint algorithm.
154 M_SEEDTHRESHOLD = 1.0;
155 M_CONEAREAFRACTION = 0.25;
156 M_MAXPAIRSIZE = 2;
157 M_MAXITERATIONS = 100;
158 M_OVERLAPTHRESHOLD = 0.75;
159
160}
161
162
163// **********Provides the smeared TLorentzVector for the electrons********
164// Smears the electron energy, and changes the 4-momentum accordingly
165// different smearing if the electron is central (eta < 2.5) or forward
166void RESOLution::SmearElectron(TLorentzVector &electron) {
167 // the 'electron' variable will be changed by the function
168 float energy = electron.E(); // before smearing
169 float energyS = 0.0; // after smearing // \sigma/E = C + N/E + S/\sqrt{E}
170
171 if(fabs(electron.Eta()) < MAX_TRACKER) { // if the electron is inside the tracker
172 energyS = gRandom->Gaus(energy, sqrt(
173 pow(ELG_Ncen,2) +
174 pow(ELG_Ccen*energy,2) +
175 pow(ELG_Scen*sqrt(energy),2) ));
176 } else { // outside the tracker
177 energyS = gRandom->Gaus(energy, sqrt(
178 pow(ELG_Nfwd,2) +
179 pow(ELG_Cfwd*energy,2) +
180 pow(ELG_Sfwd*sqrt(energy),2) ) );
181 }
182float theta=electron.Theta();
183float phi=electron.Phi();
184 float px = energyS * sin(theta) * cos(phi);
185 float py = energyS * sin(theta) * sin(phi);
186 float pz = energyS * cos(theta);
187 //cout<<" px "<<px<<" py "<<py<<" pz "<<pz<<endl;
188 electron.SetPtEtaPhiE(energyS/cosh(electron.Eta()), electron.Eta(), electron.Phi(), energyS);
189 //cout<<"nous "<<electron.Px()<<" "<<electron.Py()<<" "<<electron.Pz()<<endl;
190 if(electron.E() < 0)electron.SetPxPyPzE(0,0,0,0); // no negative values after smearing !
191}
192
193
194// **********Provides the smeared TLorentzVector for the muons********
195// Smears the muon pT and changes the 4-momentum accordingly
196void RESOLution::SmearMu(TLorentzVector &muon) {
197 // the 'muon' variable will be changed by the function
198 float pt = muon.Pt(); // before smearing
199 float ptS = gRandom->Gaus(pt, MU_SmearPt*pt ); // after smearing // \sigma/E = C + N/E + S/\sqrt{E}
200
201 muon.SetPtEtaPhiE(ptS, muon.Eta(), muon.Phi(), ptS*cosh(muon.Eta()));
202
203 if(muon.E() < 0)muon.SetPxPyPzE(0,0,0,0); // no negative values after smearing !
204}
205
206
207// **********Provides the smeared TLorentzVector for the hadrons********
208// Smears the hadron 4-momentum
209void RESOLution::SmearHadron(TLorentzVector &hadron, const float frac)
210 // the 'hadron' variable will be changed by the function
211 // the 'frac' variable describes the long-living particles. Should be 0.7 for K0S and Lambda, 1. otherwise
212{
213 float energy = hadron.E(); // before smearing
214 float energyS = 0.0; // after smearing // \sigma/E = C + N/E + S/\sqrt{E}
215 float energy_ecal = (1.0 - frac)*energy; // electromagnetic calorimeter
216 float energy_hcal = frac*energy; // hadronic calorimeter
217 // frac takes into account the decay of long-living particles, that decay in the calorimeters
218 // some of the particles decay mostly in the ecal, some mostly in the hcal
219
220 float energyS1,energyS2;
221 if(fabs(hadron.Eta()) < MAX_CALO_CEN) {
222 energyS1 = gRandom->Gaus(energy_hcal, sqrt(
223 pow(HAD_Nhcal,2) +
224 pow(HAD_Chcal*energy_hcal,2) +
225 pow(HAD_Shcal*sqrt(energy_hcal),2) )) ;
226
227
228 energyS2 = gRandom->Gaus(energy_ecal, sqrt(
229 pow(HAD_Necal,2) +
230 pow(HAD_Cecal*energy_ecal,2) +
231 pow(HAD_Secal*sqrt(energy_ecal),2) ) );
232
233 energyS = ((energyS1>0)?energyS1:0) + ((energyS2>0)?energyS2:0);
234 } else {
235 energyS = gRandom->Gaus(energy, sqrt(
236 pow(HAD_Nhf,2) +
237 pow(HAD_Chf*energy,2) +
238 pow(HAD_Shf*sqrt(energy),2) ));
239 }
240
241
242 hadron.SetPtEtaPhiE(energyS/cosh(hadron.Eta()),hadron.Eta(), hadron.Phi(), energyS);
243
244 if(hadron.E() < 0)hadron.SetPxPyPzE(0,0,0,0);
245}
246
247// **********Provides the energy in the cone of radius TAU_CONE_ENERGY for the tau identification********
248// to be taken into account, a calo tower should
249// 1) have a transverse energy \f$ E_T = \sqrt{E_X^2 + E_Y^2} \f$ above a given threshold
250// 2) be inside a cone with a radius R and the axis defined by (eta,phi)
251double RESOLution::EnergySmallCone(const vector<PhysicsTower> &towers, const float eta, const float phi) {
252 double Energie=0;
253 for(unsigned int i=0; i < towers.size(); i++) {
254 if(towers[i].fourVector.pt() < M_SEEDTHRESHOLD) continue;
255 if((DeltaR(phi,eta,towers[i].fourVector.phi(),towers[i].fourVector.eta()) < TAU_CONE_ENERGY)) {
256 Energie += towers[i].fourVector.E;
257 }
258 }
259 return Energie;
260}
261
262
263// **********Provides the number of tracks in the cone of radius TAU_CONE_TRACKS for the tau identification********
264// to be taken into account, a track should
265// 1) avec a transverse momentum \$f p_T \$ above a given threshold
266// 2) be inside a cone with a radius R and the axis defined by (eta,phi)
267// IMPORTANT REMARK !!!!!
268// previously, the argument 'phi' was before the argument 'eta'
269// this has been changed for consistency with the other functions
270// double check your running code that uses NumTracks !
271unsigned int RESOLution::NumTracks(const vector<TLorentzVector> &tracks, const float pt_track, const float eta, const float phi) {
272 unsigned int numtrack=0;
273 for(unsigned int i=0; i < tracks.size(); i++) {
274 if((tracks[i].Pt() < pt_track )||
275 (DeltaR(phi,eta,tracks[i].Phi(),tracks[i].Eta()) > TAU_CONE_TRACKS)
276 )continue;
277 numtrack++;
278 }
279 return numtrack;
280}
281
282
283//*** Returns the PID of the particle with the highest energy, in a cone with a radius CONERADIUS and an axis (eta,phi) *********
284//used by Btaggedjet
285///// Attention : bug removed => CONERADIUS/2 -> CONERADIUS !!
286int RESOLution::Bjets(const TSimpleArray<TRootGenParticle> &subarray, const float eta, const float phi) {
287 float emax=0;
288 int Ppid=0;
289 if(subarray.GetEntries()>0) {
290 for(int i=0; i < subarray.GetEntries();i++) { // should have pt>PT_JETMIN and a small cone radius (r<CONE_JET)
291 float genDeltaR = DeltaR(subarray[i]->Phi,subarray[i]->Eta,phi,eta);
292 if(genDeltaR < CONERADIUS && subarray[i]->E > emax) {
293 emax=subarray[i]->E;
294 Ppid=abs(subarray[i]->PID);
295 }
296 }
297 }
298 return Ppid;
299}
300
301
302//******************** Simulates the b-tagging efficiency for real bjet, or the misendentification for other jets****************
303bool RESOLution::Btaggedjet(const TLorentzVector &JET, const TSimpleArray<TRootGenParticle> &subarray) {
304 if( rand()%100 < (TAGGING_B+1) && Bjets(subarray,JET.Eta(),JET.Phi())==pB ) return true; // b-tag of b-jets is 40%
305 else if( rand()%100 < (MISTAGGING_C+1) && Bjets(subarray,JET.Eta(),JET.Phi())==pC ) return true; // b-tag of c-jets is 10%
306 else if( rand()%100 < (MISTAGGING_L+1) && Bjets(subarray,JET.Eta(),JET.Phi())!=0) return true; // b-tag of light jets is 1%
307 return false;
308}
309
310//***********************Isolation criteria***********************
311//****************************************************************
312bool RESOLution::Isolation(Float_t phi,Float_t eta,const vector<TLorentzVector> &tracks,float PT_TRACK2)
313{
314 bool isolated = false;
315 Float_t deltar=5000.; // Initial value; should be high; no further repercussion
316 // loop on all final charged particles, with p_t >2, close enough from the electron
317 for(unsigned int i=0; i < tracks.size(); i++)
318 {
319 if(tracks[i].Pt() < PT_TRACK2)continue;
320 Float_t genDeltaR = DeltaR(phi,eta,tracks[i].Phi(),tracks[i].Eta()); // slower to evaluate
321 if(
322 (genDeltaR > deltar) ||
323 (genDeltaR==0)
324 ) continue ;
325 deltar=genDeltaR;
326 }
327 if(deltar > 0.5)isolated = true; // returns the closest distance
328 return isolated;
329}
330
331
332//**************************** Returns the delta Phi ****************************
333float DeltaPhi(const float phi1, const float phi2) {
334 float deltaphi=phi1-phi2; // in here, -PI < phi < PI
335 if(fabs(deltaphi) > PI) deltaphi=2.*PI-fabs(deltaphi);// put deltaphi between 0 and PI
336 else deltaphi=fabs(deltaphi);
337
338 return deltaphi;
339}
340
341//**************************** Returns the delta R****************************
342float DeltaR(const float phi1, const float eta1, const float phi2, const float eta2) {
343 return sqrt(pow(DeltaPhi(phi1,phi2),2) + pow(eta1-eta2,2));
344}
345
346int sign(const int myint) {
347 if (myint >0) return 1;
348 else if (myint <0) return -1;
349 else return 0;
350}
351
352int sign(const float myfloat) {
353 if (myfloat >0) return 1;
354 else if (myfloat <0) return -1;
355 else return 0;
356}
357
358
359float Charge(const long int pid) {
360 // source: RPP chap 34 Monte Carlo Particle Numbering Scheme
361/* switch (abs(pid)) {
362 case 1: case 3: case 5: case 7: return (float) sign(pid)*(-1/3); break; // d, s, b, b'
363 case 2: case 4: case 6: case 8: return (float) sign(pid)*2/3; break; // u, c, t, t'
364
365 case 11: case 13: case 15: return (float) sign(pid)*(-1); break; // e, mu, tau
366 case 12: case 14: case 16: return (float) 0; break; // nu_e, nu_mu, nu_tau
367
368 case 9: case 21: case 22: case 23: case 25:
369 case 32: case 33: case 35: case 36: return (float) 0; break; // neutral gauge/higgs bosons
370 case 24: case 34: case 37: return (float) sign(pid); break; // charged gauge/higgs bosons
371 }
372*/
373 return 0;
374}
Note: See TracBrowser for help on using the repository browser.