4 |
|
#include "MitPhysics/Utils/interface/MVATools.h" |
5 |
|
#include "MitPhysics/Utils/interface/ElectronTools.h" |
6 |
|
#include "MitPhysics/Utils/interface/IsolationTools.h" |
7 |
+ |
#include "MitAna/DataTree/interface/PFCandidateCol.h" |
8 |
|
#include "MitAna/DataTree/interface/StableData.h" |
9 |
+ |
#include <TMath.h> |
10 |
|
#include <TFile.h> |
11 |
|
#include <TRandom3.h> |
12 |
|
#include "TMVA/Tools.h"//MVA |
13 |
|
#include "TMVA/Reader.h"//MVA |
14 |
|
|
15 |
+ |
|
16 |
|
ClassImp(mithep::MVATools) |
17 |
|
|
18 |
|
using namespace mithep; |
23 |
|
fReaderEndcap(0), |
24 |
|
fReaderBarrel(0), |
25 |
|
|
26 |
< |
//MVA variables 0 |
26 |
> |
//MVA Variables v4 |
27 |
|
HoE(0), |
28 |
|
covIEtaIEta(0), |
29 |
< |
tIso1(0), |
30 |
< |
tIso3(0), |
31 |
< |
tIso2(0), |
29 |
> |
tIso1abs(0), |
30 |
> |
tIso3abs(0), |
31 |
> |
tIso2abs(0), |
32 |
|
R9(0), |
33 |
|
|
34 |
< |
//MVA variables 1 |
35 |
< |
RelIsoEcal(0), |
33 |
< |
RelIsoHcal(0), |
34 |
< |
|
34 |
> |
absIsoEcal(0), |
35 |
> |
absIsoHcal(0), |
36 |
|
RelEMax(0), |
37 |
|
RelETop(0), |
38 |
|
RelEBottom(0), |
45 |
|
RelE2x5Right(0), |
46 |
|
RelE5x5(0), |
47 |
|
|
47 |
– |
//MVA variables 2 |
48 |
|
EtaWidth(0), |
49 |
|
PhiWidth(0), |
50 |
< |
CoviEtaiPhi(0), |
50 |
> |
CoviEtaiPhi(0), |
51 |
|
CoviPhiiPhi(0), |
52 |
– |
RelPreshowerEnergy(0) |
52 |
|
|
53 |
+ |
NVertexes(0), |
54 |
+ |
RelPreshowerEnergy(0), |
55 |
+ |
|
56 |
+ |
//MVA v2 and v1 |
57 |
+ |
RelIsoEcal(0), |
58 |
+ |
RelIsoHcal(0), |
59 |
+ |
tIso1(0), |
60 |
+ |
tIso3(0), |
61 |
+ |
tIso2(0), |
62 |
+ |
ScEta(0.) |
63 |
|
{ |
64 |
|
// Constructor. |
65 |
|
} |
78 |
|
readers[1] = fReaderBarrel; |
79 |
|
|
80 |
|
for (UInt_t i=0; i<2; ++i) { |
81 |
< |
|
81 |
> |
|
82 |
|
if(VariableType==0||VariableType==1||VariableType==2){ |
83 |
|
readers[i]->AddVariable( "HoE", &HoE ); |
84 |
|
readers[i]->AddVariable( "covIEtaIEta", &covIEtaIEta ); |
88 |
|
readers[i]->AddVariable( "R9", &R9 ); |
89 |
|
} |
90 |
|
|
91 |
+ |
if(VariableType==3||VariableType==4){ |
92 |
+ |
readers[i]->AddVariable( "HoE", &HoE ); |
93 |
+ |
readers[i]->AddVariable( "covIEtaIEta", &covIEtaIEta ); |
94 |
+ |
readers[i]->AddVariable( "tIso1abs", &tIso1abs ); |
95 |
+ |
readers[i]->AddVariable( "tIso3abs", &tIso3abs ); |
96 |
+ |
readers[i]->AddVariable( "tIso2abs", &tIso2abs ); |
97 |
+ |
readers[i]->AddVariable( "R9", &R9 ); |
98 |
+ |
} |
99 |
+ |
|
100 |
|
if(VariableType==1||VariableType==2){ |
101 |
< |
readers[i]->AddVariable( "RelIsoEcal", &RelIsoEcal); |
102 |
< |
readers[i]->AddVariable( "RelIsoHcal", &RelIsoHcal); |
103 |
< |
readers[i]->AddVariable( "RelEMax", &RelEMax); |
104 |
< |
readers[i]->AddVariable( "RelETop",&RelETop); |
105 |
< |
readers[i]->AddVariable( "RelEBottom", &RelEBottom); |
101 |
> |
readers[i]->AddVariable( "RelIsoEcal", &RelIsoEcal ); |
102 |
> |
readers[i]->AddVariable( "RelIsoHcal", &RelIsoHcal ); |
103 |
> |
readers[i]->AddVariable( "RelEMax", &RelEMax ); |
104 |
> |
readers[i]->AddVariable( "RelETop", &RelETop ); |
105 |
> |
readers[i]->AddVariable( "RelEBottom", &RelEBottom ); |
106 |
> |
readers[i]->AddVariable( "RelELeft", &RelELeft ); |
107 |
> |
readers[i]->AddVariable( "RelERight", &RelERight ); |
108 |
> |
readers[i]->AddVariable( "RelE2x5Max", &RelE2x5Max ); |
109 |
> |
readers[i]->AddVariable( "RelE2x5Top", &RelE2x5Top ); |
110 |
> |
readers[i]->AddVariable( "RelE2x5Bottom", &RelE2x5Bottom ); |
111 |
> |
readers[i]->AddVariable( "RelE2x5Left", &RelE2x5Left ); |
112 |
> |
readers[i]->AddVariable( "RelE2x5Right", &RelE2x5Right ); |
113 |
> |
readers[i]->AddVariable( "RelE5x5", &RelE5x5 ); |
114 |
> |
} |
115 |
> |
|
116 |
> |
if(VariableType==3||VariableType==4){ |
117 |
> |
readers[i]->AddVariable( "absIsoEcal", &absIsoEcal ); |
118 |
> |
readers[i]->AddVariable( "absIsoHcal", &absIsoHcal ); |
119 |
> |
readers[i]->AddVariable( "RelEMax", &RelEMax ); |
120 |
> |
readers[i]->AddVariable( "RelETop", &RelEBottom); |
121 |
> |
readers[i]->AddVariable( "RelEBottom", &RelEBottom ); |
122 |
|
readers[i]->AddVariable( "RelELeft", &RelELeft ); |
123 |
< |
readers[i]->AddVariable( "RelERight", &RelERight); |
123 |
> |
readers[i]->AddVariable( "RelERight", &RelERight ); |
124 |
|
readers[i]->AddVariable( "RelE2x5Max", &RelE2x5Max ); |
125 |
|
readers[i]->AddVariable( "RelE2x5Top", &RelE2x5Top ); |
126 |
|
readers[i]->AddVariable( "RelE2x5Bottom", &RelE2x5Bottom ); |
129 |
|
readers[i]->AddVariable( "RelE5x5", &RelE5x5 ); |
130 |
|
} |
131 |
|
|
132 |
< |
if(VariableType==2){ |
132 |
> |
if(VariableType==2||VariableType==3||VariableType==4){ |
133 |
|
readers[i]->AddVariable( "EtaWidth", &EtaWidth ); |
134 |
|
readers[i]->AddVariable( "PhiWidth", &PhiWidth ); |
135 |
|
readers[i]->AddVariable( "CoviEtaiPhi", &CoviEtaiPhi ); |
136 |
|
readers[i]->AddVariable( "CoviPhiiPhi", &CoviPhiiPhi ); |
137 |
+ |
if(VariableType==4){ |
138 |
+ |
readers[i]->AddVariable( "NVertexes", &NVertexes ); |
139 |
+ |
} |
140 |
|
if(i==0){ |
141 |
|
readers[i]->AddVariable( "RelPreshowerEnergy", &RelPreshowerEnergy ); |
142 |
|
} |
143 |
|
} |
144 |
+ |
|
145 |
+ |
if(VariableType==6){ |
146 |
+ |
readers[i]->AddVariable( "HoE", &HoE ); |
147 |
+ |
readers[i]->AddVariable( "covIEtaIEta", &covIEtaIEta ); |
148 |
+ |
readers[i]->AddVariable( "tIso1abs", &tIso1abs ); |
149 |
+ |
readers[i]->AddVariable( "tIso3abs", &tIso3abs ); |
150 |
+ |
readers[i]->AddVariable( "tIso2abs", &tIso2abs ); |
151 |
+ |
readers[i]->AddVariable( "R9", &R9 ); |
152 |
+ |
readers[i]->AddVariable( "absIsoEcal", &absIsoEcal ); |
153 |
+ |
readers[i]->AddVariable( "absIsoHcal", &absIsoHcal ); |
154 |
+ |
readers[i]->AddVariable( "RelE5x5", &RelE5x5 ); |
155 |
+ |
readers[i]->AddVariable( "EtaWidth", &EtaWidth ); |
156 |
+ |
readers[i]->AddVariable( "PhiWidth", &PhiWidth ); |
157 |
+ |
readers[i]->AddVariable( "CoviEtaiPhi", &CoviEtaiPhi ); |
158 |
+ |
readers[i]->AddVariable( "CoviPhiiPhi", &CoviPhiiPhi ); |
159 |
+ |
readers[i]->AddVariable( "NVertexes", &NVertexes ); |
160 |
+ |
if(i==0){ |
161 |
+ |
readers[i]->AddVariable( "RelPreshowerEnergy", &RelPreshowerEnergy ); |
162 |
+ |
} |
163 |
+ |
} |
164 |
+ |
|
165 |
+ |
if(VariableType==7){ |
166 |
+ |
readers[i]->AddVariable( "HoE", &HoE ); |
167 |
+ |
readers[i]->AddVariable( "covIEtaIEta", &covIEtaIEta ); |
168 |
+ |
readers[i]->AddVariable( "tIso1abs", &tIso1abs ); |
169 |
+ |
readers[i]->AddVariable( "tIso3abs", &tIso3abs ); |
170 |
+ |
readers[i]->AddVariable( "tIso2abs", &tIso2abs ); |
171 |
+ |
readers[i]->AddVariable( "R9", &R9 ); |
172 |
+ |
readers[i]->AddVariable( "absIsoEcal", &absIsoEcal ); |
173 |
+ |
readers[i]->AddVariable( "absIsoHcal", &absIsoHcal ); |
174 |
+ |
readers[i]->AddVariable( "NVertexes", &NVertexes ); |
175 |
+ |
readers[i]->AddVariable( "ScEta", &ScEta ); |
176 |
+ |
} |
177 |
+ |
|
178 |
+ |
if(VariableType==10){ |
179 |
+ |
readers[i]->AddVariable( "HoE", &HoE ); |
180 |
+ |
readers[i]->AddVariable( "covIEtaIEta", &covIEtaIEta ); |
181 |
+ |
readers[i]->AddVariable( "tIso1abs", &tIso1abs ); |
182 |
+ |
readers[i]->AddVariable( "tIso3abs", &tIso3abs ); |
183 |
+ |
readers[i]->AddVariable( "tIso2abs", &tIso2abs ); |
184 |
+ |
readers[i]->AddVariable( "R9", &R9 ); |
185 |
+ |
readers[i]->AddVariable( "absIsoEcal", &absIsoEcal ); |
186 |
+ |
readers[i]->AddVariable( "absIsoHcal", &absIsoHcal ); |
187 |
+ |
readers[i]->AddVariable( "NVertexes", &NVertexes ); |
188 |
+ |
readers[i]->AddVariable( "ScEta", &ScEta ); |
189 |
+ |
readers[i]->AddVariable( "EtaWidth", &EtaWidth ); |
190 |
+ |
readers[i]->AddVariable( "PhiWidth", &PhiWidth ); |
191 |
+ |
} |
192 |
+ |
|
193 |
+ |
if(VariableType==1201){ |
194 |
+ |
readers[i]->AddVariable( "myphoton_pfchargedisogood03", &myphoton_pfchargedisogood03); |
195 |
+ |
readers[i]->AddVariable( "myphoton_pfchargedisobad03", &myphoton_pfchargedisobad03); |
196 |
+ |
readers[i]->AddVariable( "myphoton_pfphotoniso03", &myphoton_pfphotoniso03 ); |
197 |
+ |
readers[i]->AddVariable( "myphoton_sieie", &myphoton_sieie ); |
198 |
+ |
readers[i]->AddVariable( "myphoton_sieip", &myphoton_sieip ); |
199 |
+ |
readers[i]->AddVariable( "myphoton_etawidth", &myphoton_etawidth ); |
200 |
+ |
readers[i]->AddVariable( "myphoton_phiwidth", &myphoton_phiwidth ); |
201 |
+ |
readers[i]->AddVariable( "myphoton_r9", &myphoton_r9 ); |
202 |
+ |
readers[i]->AddVariable( "myphoton_s4ratio", &myphoton_s4ratio ); |
203 |
+ |
readers[i]->AddVariable( "myphoton_SCeta", &myphoton_SCeta ); |
204 |
+ |
readers[i]->AddVariable( "event_rho", &event_rho ); |
205 |
+ |
if(i==0){ |
206 |
+ |
readers[i]->AddVariable( "myphoton_ESEffSigmaRR", &myphoton_ESEffSigmaRR); |
207 |
+ |
} |
208 |
+ |
} |
209 |
+ |
|
210 |
|
} |
211 |
|
|
212 |
|
fReaderEndcap->BookMVA("BDT method",EndcapWeights); |
213 |
|
fReaderBarrel->BookMVA("BDT method",BarrelWeights); |
214 |
< |
|
214 |
> |
|
215 |
|
assert(fReaderEndcap); |
216 |
|
assert(fReaderBarrel); |
217 |
|
|
218 |
|
} |
219 |
|
|
220 |
< |
//-------------------------------------------------------------------------------------------------- |
220 |
> |
Bool_t MVATools::PassMVASelection(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho,Float_t bdtCutBarrel, Float_t bdtCutEndcap, const ElectronCol* els, Bool_t applyElectronVeto) { |
221 |
> |
|
222 |
> |
//initilize the bool value |
223 |
> |
PassMVA=kFALSE; |
224 |
> |
|
225 |
> |
Float_t photon_bdt = MVATools::GetMVAbdtValue_2011(p,vtx,trackCol,vtxCol, _tRho, els, applyElectronVeto); |
226 |
> |
|
227 |
> |
if (isbarrel) { |
228 |
> |
if(bdt>bdtCutBarrel){ |
229 |
> |
PassMVA=kTRUE; |
230 |
> |
} |
231 |
> |
} |
232 |
> |
else { |
233 |
> |
if(bdt>bdtCutEndcap){ |
234 |
> |
PassMVA=kTRUE; |
235 |
> |
} |
236 |
> |
} |
237 |
> |
return PassMVA; |
238 |
> |
} |
239 |
|
|
240 |
< |
Bool_t MVATools::PassMVASelection(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho,const ElectronCol* els,double MVAPtMin, Float_t bdtCutBarrel, Float_t bdtCutEndcap, Bool_t applyEleVeto) { |
240 |
> |
//--------------------------------------------------------------------------------- |
241 |
> |
Int_t MVATools::PassElectronVetoInt(const Photon* p, const ElectronCol* els) { |
242 |
|
|
243 |
|
// these values are taken from the H2GGlobe code... (actually from Marco/s mail) |
244 |
|
float cic4_allcuts_temp_sublead[] = { |
251 |
|
1., 0.062, 0.97, 0.97, |
252 |
|
1.5, 1.5, 1.5, 1.5 }; // the last line is PixelmatchVeto and un-used |
253 |
|
|
132 |
– |
|
254 |
|
//initilize the bool value |
255 |
< |
PassMVA=kFALSE; |
135 |
< |
PassElecVeto=kFALSE; |
136 |
< |
//Bool_t PassElecVeto=PhotonTools::PassElectronVeto(p,els); |
255 |
> |
PassElecVetoInt=0; |
256 |
|
|
257 |
+ |
dRTrack = PhotonTools::ElectronVetoCiC(p, els); |
258 |
+ |
|
259 |
+ |
ScEta_MVA=p->SCluster()->Eta(); |
260 |
+ |
|
261 |
+ |
isbarrel = (fabs(ScEta_MVA)<1.4442); |
262 |
+ |
|
263 |
+ |
R9 = p->R9(); |
264 |
+ |
//R9 = p->E33()/p->SCluster()->RawEnergy(); |
265 |
+ |
|
266 |
+ |
// check which category it is ... |
267 |
+ |
_tCat = 1; |
268 |
+ |
if ( !isbarrel ) _tCat = 3; |
269 |
+ |
if ( R9 < 0.94 ) _tCat++; |
270 |
+ |
|
271 |
+ |
//Electron Veto |
272 |
+ |
if(dRTrack > cic4_allcuts_temp_sublead[_tCat-1+6*4]){ |
273 |
+ |
PassElecVetoInt=1; |
274 |
+ |
} |
275 |
+ |
|
276 |
+ |
return PassElecVetoInt; |
277 |
+ |
|
278 |
+ |
} |
279 |
+ |
|
280 |
+ |
//-------------------------------------------------------------------------------------------------- |
281 |
+ |
|
282 |
+ |
Float_t MVATools::GetMVAbdtValue_2012_globe(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho, const PFCandidateCol *fPFCands,const ElectronCol* els,Bool_t applyElectronVeto) { |
283 |
|
|
284 |
|
//get the variables used to compute MVA variables |
285 |
|
ecalIso3 = p->EcalRecHitIsoDr03(); |
288 |
|
|
289 |
|
wVtxInd = 0; |
290 |
|
|
291 |
< |
trackIso1 = IsolationTools::CiCTrackIsolation(p,vtx, 0.3, 0.02, 0.0, 0.0, 0.1, 1.0, trackCol, NULL, NULL, (!applyEleVeto ? els : NULL) );//Question Ming:whyfPV->At(0) instead of selected vertex using ranking method? |
292 |
< |
|
148 |
< |
// track iso only |
149 |
< |
trackIso3 = IsolationTools::CiCTrackIsolation(p,vtx, 0.3, 0.02, 0.0, 0.0, 0.1, 1.0, trackCol, NULL, NULL, (!applyEleVeto ? els : NULL)); |
150 |
< |
|
291 |
> |
trackIso1 = IsolationTools::CiCTrackIsolation(p,vtx, 0.3, 0.02, 0.0, 0.0, 0.1, 1.0, trackCol, NULL, NULL, (!applyElectronVeto ? els : NULL) );//Question Ming:whyfPV->At(0) instead of selected vertex using ranking method? |
292 |
> |
|
293 |
|
// track iso worst vtx |
294 |
< |
trackIso2 = IsolationTools::CiCTrackIsolation(p,vtx, 0.4, 0.02, 0.0, 0.0, 0.1, 1.0, trackCol, &wVtxInd,vtxCol, (!applyEleVeto ? els : NULL)); |
294 |
> |
trackIso2 = IsolationTools::CiCTrackIsolation(p,vtx, 0.4, 0.02, 0.0, 0.0, 0.1, 1.0, trackCol, &wVtxInd,vtxCol, (!applyElectronVeto ? els : NULL) ); |
295 |
|
|
296 |
|
combIso1 = ecalIso3+hcalIso4+trackIso1 - 0.17*_tRho; |
297 |
|
combIso2 = ecalIso4+hcalIso4+trackIso2 - 0.52*_tRho; |
298 |
|
|
299 |
|
RawEnergy = p->SCluster()->RawEnergy(); |
300 |
|
|
301 |
< |
dRTrack = PhotonTools::ElectronVetoCiC(p, els); |
301 |
> |
ScEta = p->SCluster()->Eta(); |
302 |
|
|
303 |
< |
//compute MVA variables 0 |
162 |
< |
HoE = p->HadOverEm(); |
163 |
< |
covIEtaIEta = p->CoviEtaiEta(); |
303 |
> |
//mva varialbes v1 and v2 |
304 |
|
tIso1 = (combIso1) *50./p->Et(); |
305 |
< |
tIso3 = (trackIso3)*50./p->Et(); |
305 |
> |
tIso3 = (trackIso1)*50./p->Et(); |
306 |
|
tIso2 = (combIso2) *50./(p->MomVtx(vtxCol->At(wVtxInd)->Position()).Pt()); |
167 |
– |
R9 = p->R9(); |
168 |
– |
|
169 |
– |
//newly added MVA variables 1 |
307 |
|
RelIsoEcal=(ecalIso3-0.17*_tRho)/p->Et(); |
308 |
|
RelIsoHcal=(hcalIso4-0.17*_tRho)/p->Et(); |
309 |
< |
|
309 |
> |
|
310 |
> |
//compute mva variables for v3 |
311 |
> |
HoE = p->HadOverEm(); |
312 |
> |
covIEtaIEta = p->CoviEtaiEta(); |
313 |
> |
tIso1abs = combIso1; |
314 |
> |
tIso3abs = trackIso1; |
315 |
> |
tIso2abs = combIso2; |
316 |
> |
R9 = p->R9(); |
317 |
> |
|
318 |
> |
absIsoEcal=(ecalIso3-0.17*_tRho); |
319 |
> |
absIsoHcal=(hcalIso4-0.17*_tRho); |
320 |
|
RelEMax=p->SCluster()->Seed()->EMax()/RawEnergy; |
321 |
|
RelETop=p->SCluster()->Seed()->ETop()/RawEnergy; |
322 |
|
RelEBottom=p->SCluster()->Seed()->EBottom()/RawEnergy; |
329 |
|
RelE2x5Right=p->SCluster()->Seed()->E2x5Right()/RawEnergy; |
330 |
|
RelE5x5=p->SCluster()->Seed()->E5x5()/RawEnergy; |
331 |
|
|
332 |
< |
//newly added MVA variables 2 |
333 |
< |
EtaWidth=p->SCluster()->EtaWidth(); |
187 |
< |
PhiWidth=p->SCluster()->PhiWidth(); |
332 |
> |
EtaWidth=p->EtaWidth(); |
333 |
> |
PhiWidth=p->PhiWidth(); |
334 |
|
CoviEtaiPhi=p->SCluster()->Seed()->CoviEtaiPhi(); |
335 |
|
CoviPhiiPhi=p->SCluster()->Seed()->CoviPhiiPhi(); |
336 |
+ |
|
337 |
|
RelPreshowerEnergy=p->SCluster()->PreshowerEnergy()/RawEnergy; |
338 |
+ |
NVertexes=vtxCol->GetEntries(); |
339 |
|
|
340 |
|
//spectator variables |
341 |
|
Pt_MVA=p->Pt(); |
342 |
|
ScEta_MVA=p->SCluster()->Eta(); |
343 |
< |
|
343 |
> |
|
344 |
> |
// |
345 |
> |
|
346 |
|
isbarrel = (fabs(ScEta_MVA)<1.4442); |
347 |
|
|
348 |
< |
// check which category it is ... |
349 |
< |
_tCat = 1; |
350 |
< |
if ( !isbarrel ) _tCat = 3; |
351 |
< |
if ( R9 < 0.94 ) _tCat++; |
352 |
< |
|
353 |
< |
//Electron Veto (made electron Veto optinal (Fabian) ) |
354 |
< |
if(dRTrack > cic4_allcuts_temp_sublead[_tCat-1+6*4] || !applyEleVeto ){ |
355 |
< |
PassElecVeto=kTRUE; |
356 |
< |
} |
357 |
< |
|
358 |
< |
if(PassElecVeto && Pt_MVA>MVAPtMin && ((fabs(ScEta_MVA)<1.4442)||(fabs(ScEta_MVA)>1.566 && fabs(ScEta_MVA)<2.5)) && tIso1<250 && tIso2<250 && tIso3<250){ |
359 |
< |
|
210 |
< |
if (isbarrel) { |
211 |
< |
reader = fReaderBarrel; |
212 |
< |
} |
213 |
< |
else { |
214 |
< |
reader = fReaderEndcap; |
215 |
< |
} |
216 |
< |
|
217 |
< |
bdt = reader->EvaluateMVA( "BDT method" ); |
218 |
< |
|
219 |
< |
if (isbarrel) { |
220 |
< |
if(bdt>bdtCutBarrel){ |
221 |
< |
PassMVA=kTRUE; |
222 |
< |
} |
223 |
< |
} |
224 |
< |
else { |
225 |
< |
if(bdt>bdtCutEndcap){ |
226 |
< |
PassMVA=kTRUE; |
227 |
< |
} |
228 |
< |
} |
348 |
> |
//variable 1201 |
349 |
> |
myphoton_pfchargedisogood03=IsolationTools::PFChargedIsolation(p,vtx,0.3,0,fPFCands); |
350 |
> |
myphoton_pfchargedisobad03=IsolationTools::PFChargedIsolation(p,vtx,0.3,0,fPFCands,&wVtxInd,vtxCol); |
351 |
> |
myphoton_pfphotoniso03=IsolationTools::PFGammaIsolation(p,0.3,0,fPFCands); |
352 |
> |
myphoton_sieie=covIEtaIEta; |
353 |
> |
myphoton_sieip=CoviEtaiPhi; |
354 |
> |
myphoton_etawidth=EtaWidth; |
355 |
> |
myphoton_phiwidth=PhiWidth; |
356 |
> |
myphoton_r9=R9; |
357 |
> |
myphoton_s4ratio=p->SCluster()->Seed()->E2x2()/p->SCluster()->Seed()->E5x5(); |
358 |
> |
myphoton_SCeta=ScEta_MVA; |
359 |
> |
event_rho= _tRho; |
360 |
|
|
361 |
< |
} |
231 |
< |
|
232 |
< |
return PassMVA; |
233 |
< |
} |
361 |
> |
myphoton_ESEffSigmaRR=-99; |
362 |
|
|
363 |
< |
//--------------------------------------------------------------------------------- |
364 |
< |
Int_t MVATools::PassElectronVetoInt(const Photon* p, const ElectronCol* els) { |
365 |
< |
|
366 |
< |
// these values are taken from the H2GGlobe code... (actually from Marco/s mail) |
367 |
< |
float cic4_allcuts_temp_sublead[] = { |
368 |
< |
3.8, 2.2, 1.77, 1.29, |
369 |
< |
11.7, 3.4, 3.9, 1.84, |
370 |
< |
3.5, 2.2, 2.3, 1.45, |
371 |
< |
0.0106, 0.0097, 0.028, 0.027, |
372 |
< |
0.082, 0.062, 0.065, 0.048, |
245 |
< |
0.94, 0.36, 0.94, 0.32, |
246 |
< |
1., 0.062, 0.97, 0.97, |
247 |
< |
1.5, 1.5, 1.5, 1.5 }; // the last line is PixelmatchVeto and un-used |
248 |
< |
|
249 |
< |
//initilize the bool value |
250 |
< |
PassElecVetoInt=0; |
251 |
< |
|
252 |
< |
dRTrack = PhotonTools::ElectronVetoCiC(p, els); |
253 |
< |
|
254 |
< |
ScEta_MVA=p->SCluster()->Eta(); |
363 |
> |
if(!isbarrel){ |
364 |
> |
myphoton_ESEffSigmaRR=float(sqrt((p->SCluster()->PsEffWidthSigmaXX())*(p->SCluster()->PsEffWidthSigmaXX())+(p->SCluster()->PsEffWidthSigmaYY())*(p->SCluster()->PsEffWidthSigmaYY()))); |
365 |
> |
} |
366 |
> |
|
367 |
> |
if (isbarrel) { |
368 |
> |
reader = fReaderBarrel; |
369 |
> |
} |
370 |
> |
else { |
371 |
> |
reader = fReaderEndcap; |
372 |
> |
} |
373 |
|
|
374 |
< |
isbarrel = (fabs(ScEta_MVA)<1.4442); |
374 |
> |
assert(reader); |
375 |
|
|
376 |
< |
R9 = p->R9(); |
376 |
> |
bdt = reader->EvaluateMVA("BDT method"); |
377 |
> |
|
378 |
> |
/* printf("HoE: %f\n",HoE); |
379 |
> |
printf("covIEtaIEta: %f\n",covIEtaIEta); |
380 |
> |
printf("tIso1abs: %f\n",tIso1abs); |
381 |
> |
printf("tIso3abs: %f\n",tIso3abs); |
382 |
> |
printf("tIso2abs: %f\n",tIso2abs); |
383 |
|
|
384 |
< |
// check which category it is ... |
385 |
< |
_tCat = 1; |
386 |
< |
if ( !isbarrel ) _tCat = 3; |
387 |
< |
if ( R9 < 0.94 ) _tCat++; |
384 |
> |
printf("absIsoEcal: %f\n",absIsoEcal); |
385 |
> |
printf("absIsoHcal: %f\n",absIsoHcal); |
386 |
> |
printf("RelEMax: %f\n",RelEMax); |
387 |
> |
printf("RelETop: %f\n",RelETop); |
388 |
> |
printf("RelEBottom: %f\n",RelEBottom); |
389 |
> |
printf("RelELeft: %f\n",RelELeft); |
390 |
> |
printf("RelERight: %f\n",RelERight); |
391 |
> |
printf("RelE2x5Max: %f\n",RelE2x5Max); |
392 |
> |
printf("RelE2x5Top: %f\n",RelE2x5Top); |
393 |
> |
printf("RelE2x5Bottom: %f\n",RelE2x5Bottom); |
394 |
> |
printf("RelE2x5Left: %f\n",RelE2x5Left); |
395 |
> |
printf("RelE2x5Right;: %f\n",RelE2x5Right); |
396 |
> |
printf("RelE5x5: %f\n",RelE5x5); |
397 |
|
|
398 |
< |
//Electron Veto |
399 |
< |
if(dRTrack > cic4_allcuts_temp_sublead[_tCat-1+6*4]){ |
400 |
< |
PassElecVetoInt=1; |
401 |
< |
} |
398 |
> |
printf("EtaWidth: %f\n",EtaWidth); |
399 |
> |
printf("PhiWidth: %f\n",PhiWidth); |
400 |
> |
printf("CoviEtaiPhi: %f\n",CoviEtaiPhi); |
401 |
> |
printf("CoviPhiiPhi: %f\n",CoviPhiiPhi); |
402 |
|
|
403 |
< |
return PassElecVetoInt; |
403 |
> |
if (!isbarrel) { |
404 |
> |
printf("RelPreshowerEnergy: %f\n",RelPreshowerEnergy); |
405 |
> |
}*/ |
406 |
|
|
407 |
+ |
return bdt; |
408 |
|
} |
409 |
|
|
410 |
< |
//-------------------------------------------------------------------------------------------------- |
275 |
< |
|
276 |
< |
Float_t MVATools::GetMVAbdtValue(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho,const ElectronCol* els) { |
410 |
> |
Float_t MVATools::GetMVAbdtValue_2011(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho,const ElectronCol* els,Bool_t applyElectronVeto) { |
411 |
|
|
412 |
|
//get the variables used to compute MVA variables |
413 |
|
ecalIso3 = p->EcalRecHitIsoDr03(); |
416 |
|
|
417 |
|
wVtxInd = 0; |
418 |
|
|
419 |
< |
trackIso1 = IsolationTools::CiCTrackIsolation(p,vtx, 0.3, 0.02, 0.0, 0.0, 0.1, 1.0, trackCol);//Question Ming:whyfPV->At(0) instead of selected vertex using ranking method? |
420 |
< |
|
287 |
< |
// track iso only |
288 |
< |
trackIso3 = IsolationTools::CiCTrackIsolation(p,vtx, 0.3, 0.02, 0.0, 0.0, 0.1, 1.0, trackCol); |
289 |
< |
|
419 |
> |
trackIso1 = IsolationTools::CiCTrackIsolation(p,vtx, 0.3, 0.02, 0.0, 0.0, 0.1, 1.0, trackCol, NULL, NULL, (!applyElectronVeto ? els : NULL) );//Question Ming:whyfPV->At(0) instead of selected vertex using ranking method? |
420 |
> |
|
421 |
|
// track iso worst vtx |
422 |
< |
trackIso2 = IsolationTools::CiCTrackIsolation(p,vtx, 0.4, 0.02, 0.0, 0.0, 0.1, 1.0, trackCol, &wVtxInd,vtxCol); |
422 |
> |
trackIso2 = IsolationTools::CiCTrackIsolation(p,vtx, 0.4, 0.02, 0.0, 0.0, 0.1, 1.0, trackCol, &wVtxInd,vtxCol, (!applyElectronVeto ? els : NULL) ); |
423 |
|
|
424 |
|
combIso1 = ecalIso3+hcalIso4+trackIso1 - 0.17*_tRho; |
425 |
|
combIso2 = ecalIso4+hcalIso4+trackIso2 - 0.52*_tRho; |
426 |
|
|
427 |
|
RawEnergy = p->SCluster()->RawEnergy(); |
428 |
|
|
429 |
< |
dRTrack = PhotonTools::ElectronVetoCiC(p, els); |
429 |
> |
ScEta = p->SCluster()->Eta(); |
430 |
|
|
431 |
< |
//compute MVA variables 0 |
301 |
< |
HoE = p->HadOverEm(); |
302 |
< |
covIEtaIEta = p->CoviEtaiEta(); |
431 |
> |
//mva varialbes v1 and v2 |
432 |
|
tIso1 = (combIso1) *50./p->Et(); |
433 |
< |
tIso3 = (trackIso3)*50./p->Et(); |
433 |
> |
tIso3 = (trackIso1)*50./p->Et(); |
434 |
|
tIso2 = (combIso2) *50./(p->MomVtx(vtxCol->At(wVtxInd)->Position()).Pt()); |
306 |
– |
R9 = p->R9(); |
307 |
– |
|
308 |
– |
//newly added MVA variables 1 |
435 |
|
RelIsoEcal=(ecalIso3-0.17*_tRho)/p->Et(); |
436 |
|
RelIsoHcal=(hcalIso4-0.17*_tRho)/p->Et(); |
437 |
< |
|
437 |
> |
|
438 |
> |
//compute mva variables for v3 |
439 |
> |
HoE = p->HadOverEm(); |
440 |
> |
covIEtaIEta = p->CoviEtaiEta(); |
441 |
> |
tIso1abs = combIso1; |
442 |
> |
tIso3abs = trackIso1; |
443 |
> |
tIso2abs = combIso2; |
444 |
> |
R9 = p->R9(); |
445 |
> |
|
446 |
> |
absIsoEcal=(ecalIso3-0.17*_tRho); |
447 |
> |
absIsoHcal=(hcalIso4-0.17*_tRho); |
448 |
|
RelEMax=p->SCluster()->Seed()->EMax()/RawEnergy; |
449 |
|
RelETop=p->SCluster()->Seed()->ETop()/RawEnergy; |
450 |
|
RelEBottom=p->SCluster()->Seed()->EBottom()/RawEnergy; |
457 |
|
RelE2x5Right=p->SCluster()->Seed()->E2x5Right()/RawEnergy; |
458 |
|
RelE5x5=p->SCluster()->Seed()->E5x5()/RawEnergy; |
459 |
|
|
460 |
< |
//newly added MVA variables 2 |
461 |
< |
EtaWidth=p->SCluster()->EtaWidth(); |
326 |
< |
PhiWidth=p->SCluster()->PhiWidth(); |
460 |
> |
EtaWidth=p->EtaWidth(); |
461 |
> |
PhiWidth=p->PhiWidth(); |
462 |
|
CoviEtaiPhi=p->SCluster()->Seed()->CoviEtaiPhi(); |
463 |
|
CoviPhiiPhi=p->SCluster()->Seed()->CoviPhiiPhi(); |
464 |
+ |
|
465 |
|
RelPreshowerEnergy=p->SCluster()->PreshowerEnergy()/RawEnergy; |
466 |
+ |
NVertexes=vtxCol->GetEntries(); |
467 |
|
|
468 |
|
//spectator variables |
469 |
|
Pt_MVA=p->Pt(); |
470 |
|
ScEta_MVA=p->SCluster()->Eta(); |
471 |
< |
|
471 |
> |
|
472 |
> |
// |
473 |
> |
|
474 |
|
isbarrel = (fabs(ScEta_MVA)<1.4442); |
475 |
|
|
476 |
|
if (isbarrel) { |
484 |
|
|
485 |
|
bdt = reader->EvaluateMVA("BDT method"); |
486 |
|
|
487 |
< |
/* printf("HoE: %f\n",HoE); |
487 |
> |
/* printf("HoE: %f\n",HoE); |
488 |
|
printf("covIEtaIEta: %f\n",covIEtaIEta); |
489 |
< |
printf("tIso1: %f\n",tIso1); |
490 |
< |
printf("tIso3: %f\n",tIso3); |
491 |
< |
printf("tIso2: %f\n",tIso2); |
353 |
< |
printf("tIso1: %f\n",tIso1); |
354 |
< |
printf("R9: %f\n",R9); |
355 |
< |
printf("RelIsoEcal: %f\n",RelIsoEcal); |
356 |
< |
printf("RelIsoHcal: %f\n",RelIsoHcal); |
489 |
> |
printf("tIso1abs: %f\n",tIso1abs); |
490 |
> |
printf("tIso3abs: %f\n",tIso3abs); |
491 |
> |
printf("tIso2abs: %f\n",tIso2abs); |
492 |
|
|
493 |
+ |
printf("absIsoEcal: %f\n",absIsoEcal); |
494 |
+ |
printf("absIsoHcal: %f\n",absIsoHcal); |
495 |
|
printf("RelEMax: %f\n",RelEMax); |
496 |
|
printf("RelETop: %f\n",RelETop); |
497 |
|
printf("RelEBottom: %f\n",RelEBottom); |
509 |
|
printf("CoviEtaiPhi: %f\n",CoviEtaiPhi); |
510 |
|
printf("CoviPhiiPhi: %f\n",CoviPhiiPhi); |
511 |
|
|
512 |
< |
if (isbarrel) { |
512 |
> |
if (!isbarrel) { |
513 |
|
printf("RelPreshowerEnergy: %f\n",RelPreshowerEnergy); |
514 |
|
}*/ |
515 |
|
|