1 |
//--------------------------------------------------------------------------------------------------
|
2 |
// $Id: PhotonTreeWriter.h,v 1.27 2012/10/26 19:42:48 fabstoec Exp $
|
3 |
//
|
4 |
// PhotonTreeWriter
|
5 |
//
|
6 |
// Authors: J. Bendavid
|
7 |
//--------------------------------------------------------------------------------------------------
|
8 |
|
9 |
#ifndef MITPHYSICS_MODS_PHOTONTREEWRITER_H
|
10 |
#define MITPHYSICS_MODS_PHOTONTREEWRITER_H
|
11 |
|
12 |
#include "MitAna/TreeMod/interface/BaseMod.h"
|
13 |
#include "MitAna/DataTree/interface/PhotonFwd.h"
|
14 |
#include "MitAna/DataTree/interface/TrackCol.h"
|
15 |
#include "MitAna/DataTree/interface/VertexCol.h"
|
16 |
#include "MitAna/DataTree/interface/BeamSpotCol.h"
|
17 |
#include "MitAna/DataTree/interface/PFCandidateCol.h"
|
18 |
#include "MitAna/DataTree/interface/PileupEnergyDensityCol.h"
|
19 |
#include "MitAna/DataTree/interface/DecayParticleCol.h"
|
20 |
#include "MitAna/DataTree/interface/ElectronCol.h"
|
21 |
#include "MitAna/DataTree/interface/MuonCol.h"
|
22 |
#include "MitAna/DataTree/interface/DecayParticleCol.h"
|
23 |
#include "MitAna/DataTree/interface/PileupInfoCol.h"
|
24 |
#include "MitAna/DataTree/interface/MCParticleCol.h"
|
25 |
#include "MitAna/DataTree/interface/MCEventInfo.h"
|
26 |
#include "MitAna/DataTree/interface/SuperClusterCol.h"
|
27 |
#include "MitAna/DataTree/interface/PFMetCol.h"
|
28 |
#include "MitAna/DataTree/interface/JetCol.h"
|
29 |
#include "MitAna/DataTree/interface/PFJetCol.h"
|
30 |
#include "MitAna/DataTree/interface/GenJetCol.h"
|
31 |
#include "MitPhysics/Utils/interface/PhotonFix.h"
|
32 |
#include "MitPhysics/Utils/interface/PhotonTools.h"
|
33 |
#include "MitPhysics/Utils/interface/MVAMet.h"
|
34 |
#include "MitPhysics/Utils/interface/MVAVBF.h"
|
35 |
|
36 |
#include "MitPhysics/Utils/interface/VertexTools.h"
|
37 |
#include "MitPhysics/Utils/interface/ElectronIDMVA.h"
|
38 |
|
39 |
|
40 |
class TNtuple;
|
41 |
class TRandom3;
|
42 |
|
43 |
namespace mithep
|
44 |
{
|
45 |
|
46 |
class PhotonTreeWriterVtx
|
47 |
{
|
48 |
|
49 |
public:
|
50 |
void SetVars(const Vertex *v, const Photon *p1, const Photon *p2, const PFCandidateCol *pfcands, Int_t idx, Int_t numvtx, Float_t genvtxz);
|
51 |
|
52 |
private:
|
53 |
Int_t n;
|
54 |
Int_t nvtx;
|
55 |
Float_t ptgg;
|
56 |
Float_t phigg;
|
57 |
Float_t etagg;
|
58 |
Float_t pxgg;
|
59 |
Float_t pygg;
|
60 |
Float_t pzgg;
|
61 |
Float_t mgg;
|
62 |
Float_t zgen;
|
63 |
Float_t x;
|
64 |
Float_t y;
|
65 |
Float_t z;
|
66 |
Float_t pt;
|
67 |
Float_t phi;
|
68 |
Float_t eta;
|
69 |
Float_t px;
|
70 |
Float_t py;
|
71 |
Float_t pz;
|
72 |
Float_t sumpt;
|
73 |
Float_t sumptsq;
|
74 |
Int_t nchalltoward;
|
75 |
Int_t nchalltransverse;
|
76 |
Int_t nchallaway;
|
77 |
Int_t nchcuttoward;
|
78 |
Int_t nchcuttransverse;
|
79 |
Int_t nchcutaway;
|
80 |
};
|
81 |
|
82 |
template <int NClus=16>//ming: what's template
|
83 |
class PhotonTreeWriterPhoton
|
84 |
{
|
85 |
public:
|
86 |
void SetVars(const Photon *p, const DecayParticle *c, const Electron *ele, const SuperCluster *pfsc, const MCParticle *m, PhotonFix &phfixph, PhotonFix &phfixele, const TrackCol* trackCol,const VertexCol* vtxCol, const PFCandidateCol* candCol, Double_t _tRho, Bool_t fillclusterarrays, const ElectronCol* els, const DecayParticleCol *convs, const BaseVertex *bs, Bool_t applyElectronVeto=kTRUE, const Vertex* realVtx = NULL);
|
87 |
Float_t Ecor() const { return ecor; };
|
88 |
Float_t Ecorerr() const { return ecorerr; };
|
89 |
Float_t Ecorele() const { return ecorele; };
|
90 |
Float_t Ecoreleerr() const { return ecoreleerr; };
|
91 |
|
92 |
private:
|
93 |
UChar_t hasphoton;
|
94 |
Float_t e;
|
95 |
Float_t pt;
|
96 |
Float_t eta;
|
97 |
Float_t phi;
|
98 |
Float_t r9;
|
99 |
Float_t e3x3;
|
100 |
Float_t e5x5;
|
101 |
Float_t sce;
|
102 |
Float_t scrawe;
|
103 |
Float_t scpse;
|
104 |
Float_t scpssigmaxx;
|
105 |
Float_t scpssigmayy;
|
106 |
Float_t sceta;
|
107 |
Float_t scphi;
|
108 |
UInt_t scnclusters;
|
109 |
UInt_t scnhits;
|
110 |
Float_t scetawidth;
|
111 |
Float_t scphiwidth;
|
112 |
Float_t hovere;
|
113 |
Float_t hoveretower;
|
114 |
Float_t sigietaieta;
|
115 |
Bool_t isbarrel;
|
116 |
Bool_t isr9reco;
|
117 |
Bool_t isr9cat;
|
118 |
Char_t phcat;
|
119 |
Float_t eerr;
|
120 |
Float_t eerrsmeared;
|
121 |
Float_t esmearing;
|
122 |
Float_t idmva;
|
123 |
Float_t ecalisodr03;
|
124 |
Float_t hcalisodr03;
|
125 |
Float_t trkisohollowdr03;
|
126 |
Float_t ecalisodr04;
|
127 |
Float_t hcalisodr04;
|
128 |
Float_t trkisohollowdr04;
|
129 |
Float_t trackiso1;
|
130 |
Float_t trackiso2;
|
131 |
Float_t combiso1;
|
132 |
Float_t combiso2;
|
133 |
Bool_t passeleveto;
|
134 |
|
135 |
//quantities from seed basic cluster
|
136 |
Float_t eseed;
|
137 |
Float_t etaseed;
|
138 |
Float_t phiseed;
|
139 |
Int_t ietaseed;
|
140 |
Int_t iphiseed;
|
141 |
Int_t ixseed;
|
142 |
Int_t iyseed;
|
143 |
Float_t etacryseed;
|
144 |
Float_t phicryseed;
|
145 |
Float_t xcryseed;
|
146 |
Float_t ycryseed;
|
147 |
Float_t thetaaxisseed;
|
148 |
Float_t phiaxisseed;
|
149 |
Float_t sigietaietaseed;
|
150 |
Float_t sigiphiphiseed;
|
151 |
Float_t covietaiphiseed;
|
152 |
Float_t e3x3seed;
|
153 |
Float_t e5x5seed;
|
154 |
Float_t emaxseed;
|
155 |
Float_t e2ndseed;
|
156 |
Float_t etopseed;
|
157 |
Float_t ebottomseed;
|
158 |
Float_t eleftseed;
|
159 |
Float_t erightseed;
|
160 |
Float_t e1x3seed;
|
161 |
Float_t e3x1seed;
|
162 |
Float_t e1x5seed;
|
163 |
Float_t e2x2seed;
|
164 |
Float_t e4x4seed;
|
165 |
Float_t e2x5maxseed;
|
166 |
Float_t e2x5topseed;
|
167 |
Float_t e2x5bottomseed;
|
168 |
Float_t e2x5leftseed;
|
169 |
Float_t e2x5rightseed;
|
170 |
Float_t xseedseed;
|
171 |
Float_t yseedseed;
|
172 |
Float_t zseedseed;
|
173 |
UInt_t nhitsseed;
|
174 |
|
175 |
//quantities from second basic cluster, if present
|
176 |
Float_t ebc2;
|
177 |
Float_t etabc2;
|
178 |
Float_t phibc2;
|
179 |
Int_t ietabc2;
|
180 |
Int_t iphibc2;
|
181 |
Int_t ixbc2;
|
182 |
Int_t iybc2;
|
183 |
Float_t etacrybc2;
|
184 |
Float_t phicrybc2;
|
185 |
Float_t xcrybc2;
|
186 |
Float_t ycrybc2;
|
187 |
Float_t thetaaxisbc2;
|
188 |
Float_t phiaxisbc2;
|
189 |
Float_t sigietaietabc2;
|
190 |
Float_t sigiphiphibc2;
|
191 |
Float_t covietaiphibc2;
|
192 |
Float_t e3x3bc2;
|
193 |
Float_t e5x5bc2;
|
194 |
Float_t emaxbc2;
|
195 |
Float_t e2ndbc2;
|
196 |
Float_t etopbc2;
|
197 |
Float_t ebottombc2;
|
198 |
Float_t eleftbc2;
|
199 |
Float_t erightbc2;
|
200 |
Float_t e1x3bc2;
|
201 |
Float_t e3x1bc2;
|
202 |
Float_t e1x5bc2;
|
203 |
Float_t e2x2bc2;
|
204 |
Float_t e4x4bc2;
|
205 |
Float_t e2x5maxbc2;
|
206 |
Float_t e2x5topbc2;
|
207 |
Float_t e2x5bottombc2;
|
208 |
Float_t e2x5leftbc2;
|
209 |
Float_t e2x5rightbc2;
|
210 |
Float_t xbc2bc2;
|
211 |
Float_t ybc2bc2;
|
212 |
Float_t zbc2bc2;
|
213 |
UInt_t nhitsbc2;
|
214 |
|
215 |
//quantities from lowest energy basic cluster if present
|
216 |
Float_t ebclast;
|
217 |
Float_t etabclast;
|
218 |
Float_t phibclast;
|
219 |
Int_t ietabclast;
|
220 |
Int_t iphibclast;
|
221 |
Int_t ixbclast;
|
222 |
Int_t iybclast;
|
223 |
Float_t etacrybclast;
|
224 |
Float_t phicrybclast;
|
225 |
Float_t xcrybclast;
|
226 |
Float_t ycrybclast;
|
227 |
Float_t thetaaxisbclast;
|
228 |
Float_t phiaxisbclast;
|
229 |
Float_t sigietaietabclast;
|
230 |
Float_t sigiphiphibclast;
|
231 |
Float_t covietaiphibclast;
|
232 |
Float_t e3x3bclast;
|
233 |
Float_t e5x5bclast;
|
234 |
UInt_t nhitsbclast;
|
235 |
|
236 |
//quantities from second lowest energy basic cluster if present
|
237 |
Float_t ebclast2;
|
238 |
Float_t etabclast2;
|
239 |
Float_t phibclast2;
|
240 |
Int_t ietabclast2;
|
241 |
Int_t iphibclast2;
|
242 |
Int_t ixbclast2;
|
243 |
Int_t iybclast2;
|
244 |
Float_t etacrybclast2;
|
245 |
Float_t phicrybclast2;
|
246 |
Float_t xcrybclast2;
|
247 |
Float_t ycrybclast2;
|
248 |
Float_t thetaaxisbclast2;
|
249 |
Float_t phiaxisbclast2;
|
250 |
Float_t sigietaietabclast2;
|
251 |
Float_t sigiphiphibclast2;
|
252 |
Float_t covietaiphibclast2;
|
253 |
Float_t e3x3bclast2;
|
254 |
Float_t e5x5bclast2;
|
255 |
UInt_t nhitsbclast2;
|
256 |
|
257 |
//energy correction quantities from PhotonFix
|
258 |
Float_t ecor;
|
259 |
Float_t ecorerr;
|
260 |
Float_t ecorele;
|
261 |
Float_t ecoreleerr;
|
262 |
Float_t etac;
|
263 |
Float_t etas;
|
264 |
Float_t etam;
|
265 |
Float_t phic;
|
266 |
Float_t phis;
|
267 |
Float_t phim;
|
268 |
Float_t xz;
|
269 |
Float_t xc;
|
270 |
Float_t xs;
|
271 |
Float_t xm;
|
272 |
Float_t yz;
|
273 |
Float_t yc;
|
274 |
Float_t ys;
|
275 |
Float_t ym;
|
276 |
|
277 |
//conversion quantities
|
278 |
UChar_t hasconversion;
|
279 |
Float_t convp;
|
280 |
Float_t convpt;
|
281 |
Float_t conveta;
|
282 |
Float_t convphi;
|
283 |
Float_t convdeta;
|
284 |
Float_t convdphi;
|
285 |
Float_t convvtxrho;
|
286 |
Float_t convvtxz;
|
287 |
Float_t convvtxphi;
|
288 |
Float_t convleadpt;
|
289 |
Float_t convtrailpt;
|
290 |
Float_t convleadtrackpt;
|
291 |
Char_t convleadtrackalgo;
|
292 |
Char_t convleadtrackalgos;
|
293 |
Char_t convleadtrackcharge;
|
294 |
Float_t convtrailtrackpt;
|
295 |
Char_t convtrailtrackalgo;
|
296 |
Char_t convtrailtrackalgos;
|
297 |
Char_t trailtrackcharge;
|
298 |
|
299 |
//electron quantities
|
300 |
UChar_t haselectron;
|
301 |
UChar_t eleisecaldriven;
|
302 |
UChar_t eleistrackerdriven;
|
303 |
Float_t elee;
|
304 |
Float_t elept;
|
305 |
Float_t eleeta;
|
306 |
Float_t elephi;
|
307 |
Char_t elecharge;
|
308 |
Float_t elefbrem;
|
309 |
Float_t eledeta;
|
310 |
Float_t eledphi;
|
311 |
Float_t elep;
|
312 |
Float_t elepin;
|
313 |
Float_t elepout;
|
314 |
|
315 |
//pf supercluster quantities
|
316 |
UChar_t haspfsc;
|
317 |
Float_t pfsce;
|
318 |
Float_t pfscrawe;
|
319 |
Float_t pfsceta;
|
320 |
Float_t pfscphi;
|
321 |
UInt_t pfscnclusters;
|
322 |
UInt_t pfscnhits;
|
323 |
Float_t pfscetawidth;
|
324 |
Float_t pfscphiwidth;
|
325 |
UInt_t pfscnpsclusters;
|
326 |
|
327 |
//generator level quantities
|
328 |
UChar_t ispromptgen;
|
329 |
Float_t gene;
|
330 |
Float_t genpt;
|
331 |
Float_t geneta;
|
332 |
Float_t genphi;
|
333 |
Float_t genz;
|
334 |
Int_t pdgid;
|
335 |
Int_t motherpdgid;
|
336 |
|
337 |
// -----------------------------------------------------
|
338 |
// PF-CiC4 Debug Stuff
|
339 |
Float_t pfcic4_tIso1;
|
340 |
Float_t pfcic4_tIso2;
|
341 |
Float_t pfcic4_tIso3;
|
342 |
Float_t pfcic4_covIEtaIEta;
|
343 |
Float_t pfcic4_HoE;
|
344 |
Float_t pfcic4_R9;
|
345 |
Float_t pfcic4_wVtxInd;
|
346 |
Float_t pfcic4_ecalIso3;
|
347 |
Float_t pfcic4_ecalIso4;
|
348 |
Float_t pfcic4_trackIsoSel03;
|
349 |
Float_t pfcic4_trackIsoWorst04;
|
350 |
Float_t pfcic4_combIso1;
|
351 |
Float_t pfcic4_combIso2;
|
352 |
// -----------------------------------------------------
|
353 |
|
354 |
// -----------------------------------------------------
|
355 |
//id mva
|
356 |
//2011
|
357 |
Float_t idmva_tIso1abs;
|
358 |
Float_t idmva_tIso2abs;
|
359 |
Float_t idmva_tIso3abs;
|
360 |
Float_t idmva_absIsoEcal;
|
361 |
Float_t idmva_absIsoHcal;
|
362 |
//2012
|
363 |
Float_t idmva_CoviEtaiPhi;
|
364 |
Float_t idmva_s4ratio;
|
365 |
Float_t idmva_GammaIso;
|
366 |
Float_t idmva_ChargedIso_selvtx;
|
367 |
Float_t idmva_ChargedIso_0p2_selvtx;
|
368 |
Float_t idmva_ChargedIso_worstvtx;
|
369 |
Float_t idmva_PsEffWidthSigmaRR;
|
370 |
|
371 |
Float_t ebcs[NClus];
|
372 |
Float_t etabcs[NClus];
|
373 |
Float_t phibcs[NClus];
|
374 |
Int_t ietabcs[NClus];
|
375 |
Int_t iphibcs[NClus];
|
376 |
Int_t ixbcs[NClus];
|
377 |
Int_t iybcs[NClus];
|
378 |
Float_t etacrybcs[NClus];
|
379 |
Float_t phicrybcs[NClus];
|
380 |
Float_t xcrybcs[NClus];
|
381 |
Float_t ycrybcs[NClus];
|
382 |
Float_t sigietaietabcs[NClus];
|
383 |
Float_t sigiphiphibcs[NClus];
|
384 |
Float_t covietaiphibcs[NClus];
|
385 |
Float_t sigetaetabcs[NClus];
|
386 |
Float_t sigphiphibcs[NClus];
|
387 |
Float_t covetaphibcs[NClus];
|
388 |
Float_t e3x3bcs[NClus];
|
389 |
Float_t e5x5bcs[NClus];
|
390 |
Float_t emaxbcs[NClus];
|
391 |
Float_t e2ndbcs[NClus];
|
392 |
Float_t etopbcs[NClus];
|
393 |
Float_t ebottombcs[NClus];
|
394 |
Float_t eleftbcs[NClus];
|
395 |
Float_t erightbcs[NClus];
|
396 |
Float_t e1x3bcs[NClus];
|
397 |
Float_t e3x1bcs[NClus];
|
398 |
Float_t e1x5bcs[NClus];
|
399 |
Float_t e2x2bcs[NClus];
|
400 |
Float_t e4x4bcs[NClus];
|
401 |
Float_t e2x5maxbcs[NClus];
|
402 |
Float_t e2x5topbcs[NClus];
|
403 |
Float_t e2x5bottombcs[NClus];
|
404 |
Float_t e2x5leftbcs[NClus];
|
405 |
Float_t e2x5rightbcs[NClus];
|
406 |
UInt_t nhitsbcs[NClus];
|
407 |
|
408 |
Float_t epfbcs[NClus];
|
409 |
Float_t etapfbcs[NClus];
|
410 |
Float_t phipfbcs[NClus];
|
411 |
Int_t ietapfbcs[NClus];
|
412 |
Int_t iphipfbcs[NClus];
|
413 |
Int_t ixpfbcs[NClus];
|
414 |
Int_t iypfbcs[NClus];
|
415 |
Float_t etacrypfbcs[NClus];
|
416 |
Float_t phicrypfbcs[NClus];
|
417 |
Float_t xcrypfbcs[NClus];
|
418 |
Float_t ycrypfbcs[NClus];
|
419 |
Float_t sigietaietapfbcs[NClus];
|
420 |
Float_t sigiphiphipfbcs[NClus];
|
421 |
Float_t covietaiphipfbcs[NClus];
|
422 |
Float_t sigetaetapfbcs[NClus];
|
423 |
Float_t sigphiphipfbcs[NClus];
|
424 |
Float_t covetaphipfbcs[NClus];
|
425 |
Float_t e3x3pfbcs[NClus];
|
426 |
Float_t e5x5pfbcs[NClus];
|
427 |
Float_t emaxpfbcs[NClus];
|
428 |
Float_t e2ndpfbcs[NClus];
|
429 |
Float_t etoppfbcs[NClus];
|
430 |
Float_t ebottompfbcs[NClus];
|
431 |
Float_t eleftpfbcs[NClus];
|
432 |
Float_t erightpfbcs[NClus];
|
433 |
Float_t e1x3pfbcs[NClus];
|
434 |
Float_t e3x1pfbcs[NClus];
|
435 |
Float_t e1x5pfbcs[NClus];
|
436 |
Float_t e2x2pfbcs[NClus];
|
437 |
Float_t e4x4pfbcs[NClus];
|
438 |
Float_t e2x5maxpfbcs[NClus];
|
439 |
Float_t e2x5toppfbcs[NClus];
|
440 |
Float_t e2x5bottompfbcs[NClus];
|
441 |
Float_t e2x5leftpfbcs[NClus];
|
442 |
Float_t e2x5rightpfbcs[NClus];
|
443 |
UInt_t nhitspfbcs[NClus];
|
444 |
|
445 |
Float_t epsc[100];
|
446 |
Float_t etapsc[100];
|
447 |
Float_t phipsc[100];
|
448 |
UChar_t planepsc[100];
|
449 |
|
450 |
|
451 |
};
|
452 |
|
453 |
class PhotonTreeWriterDiphotonEvent
|
454 |
{
|
455 |
public:
|
456 |
// ------------ BTAG STUFF -------------------
|
457 |
Float_t btagJet1;
|
458 |
Float_t btagJet1Pt;
|
459 |
Float_t btagJet1Eta;
|
460 |
Float_t btagJet2;
|
461 |
Float_t btagJet2Pt;
|
462 |
Float_t btagJet2Eta;
|
463 |
// ----------- LEPTON TAG STUFF -------------
|
464 |
Int_t leptonTag;
|
465 |
// ----------- VEERTEX SYNCHING STUFF -------
|
466 |
|
467 |
Int_t vtxInd1;
|
468 |
Int_t vtxInd2;
|
469 |
Int_t vtxInd3;
|
470 |
|
471 |
Float_t vtxBestPtbal ;
|
472 |
Float_t vtxBestPtasym ;
|
473 |
Float_t vtxBestSumpt2 ;
|
474 |
Float_t vtxBestP2Conv ;
|
475 |
|
476 |
Float_t vtxMva1Z ;
|
477 |
Float_t vtxMva2Z ;
|
478 |
Float_t vtxMva3Z ;
|
479 |
|
480 |
Float_t vtxMva1 ;
|
481 |
Float_t vtxMva2 ;
|
482 |
Float_t vtxMva3 ;
|
483 |
|
484 |
Int_t vtxNleg1 ;
|
485 |
Int_t vtxNleg2 ;
|
486 |
Int_t vtxConvIdx1 ;
|
487 |
Int_t vtxConvIdx2 ;
|
488 |
Int_t vtxNconv ;
|
489 |
|
490 |
Float_t vtxConv1Z ;
|
491 |
Float_t vtxConv1DZ ;
|
492 |
Float_t vtxConv1Prob ;
|
493 |
|
494 |
Float_t vtxConv2Z ;
|
495 |
Float_t vtxConv2DZ ;
|
496 |
Float_t vtxConv2Prob ;
|
497 |
|
498 |
|
499 |
// ---------- MUON STUFF --------------------
|
500 |
Float_t muonPt;
|
501 |
Float_t muonEta;
|
502 |
Float_t muDR1;
|
503 |
Float_t muDR2;
|
504 |
Float_t muIso1;
|
505 |
Float_t muIso2;
|
506 |
Float_t muIso3;
|
507 |
Float_t muIso4;
|
508 |
Float_t muD0;
|
509 |
Float_t muDZ;
|
510 |
Int_t muNhits;
|
511 |
Float_t muChi2;
|
512 |
Int_t muNpixhits;
|
513 |
Int_t muNegs;
|
514 |
Int_t muNMatch;
|
515 |
// ----------- ELECTRON STUFF --------------
|
516 |
Float_t elePt;
|
517 |
Float_t eleEta;
|
518 |
Float_t eleSCEta;
|
519 |
Float_t eleIso1;
|
520 |
Float_t eleIso2;
|
521 |
Float_t eleIso3;
|
522 |
Float_t eleIso4;
|
523 |
Float_t eleDist;
|
524 |
Float_t eleDcot;
|
525 |
Float_t eleCoviee;
|
526 |
Float_t eleDphiin;
|
527 |
Float_t eleDetain;
|
528 |
Float_t eleDR1;
|
529 |
Float_t eleDR2;
|
530 |
Float_t eleMass1;
|
531 |
Float_t eleMass2;
|
532 |
Int_t eleNinnerHits;
|
533 |
Float_t eleIdMva;
|
534 |
// -----------------------------------------
|
535 |
Float_t rho;
|
536 |
Float_t rho25;
|
537 |
Float_t rhoold;
|
538 |
Float_t genHiggspt;
|
539 |
Float_t genHiggsZ;
|
540 |
Float_t genmass;
|
541 |
Float_t gencostheta;
|
542 |
Float_t genz;
|
543 |
Float_t bsX;
|
544 |
Float_t bsY;
|
545 |
Float_t bsZ;
|
546 |
Float_t bsSigmaZ;
|
547 |
Float_t vtxX;
|
548 |
Float_t vtxY;
|
549 |
Float_t vtxZ;
|
550 |
Int_t nVtx;
|
551 |
Int_t numPU;
|
552 |
Int_t numPUminus;
|
553 |
Int_t numPUplus;
|
554 |
Float_t mass;
|
555 |
Float_t masserr;
|
556 |
Float_t masserrsmeared;
|
557 |
Float_t masserrwrongvtx;
|
558 |
Float_t masserrsmearedwrongvtx;
|
559 |
Float_t vtxprob;
|
560 |
Float_t deltamvtx;
|
561 |
Float_t ptgg;
|
562 |
Float_t etagg;
|
563 |
Float_t phigg;
|
564 |
Float_t costheta;
|
565 |
Float_t massele;
|
566 |
Float_t ptee;
|
567 |
Float_t costhetaele;
|
568 |
Float_t mt;
|
569 |
Float_t cosphimet;
|
570 |
Float_t mtele;
|
571 |
Float_t cosphimetele;
|
572 |
UInt_t evt;
|
573 |
UInt_t run;
|
574 |
UInt_t lumi;
|
575 |
UChar_t evtcat;
|
576 |
UInt_t nobj;
|
577 |
Double_t fromZ; //added: Heng 2/14/2012
|
578 |
Double_t fromW; //added: Heng
|
579 |
Float_t zpt; //added: Heng
|
580 |
Float_t allZpt; //added: Heng
|
581 |
Float_t zEta; //added: Heng
|
582 |
Float_t allZEta; //added: Heng
|
583 |
|
584 |
//Met stuff
|
585 |
Float_t corrpfmet;
|
586 |
Float_t corrpfmetphi;
|
587 |
Float_t corrpfmetx;
|
588 |
Float_t corrpfmety;
|
589 |
Double_t dphiMetgg;
|
590 |
Double_t cosdphiMetgg;
|
591 |
Double_t dphiPhPh;
|
592 |
Float_t pfmet;
|
593 |
Float_t pfmetphi;
|
594 |
Float_t pfmetx;
|
595 |
Float_t pfmety;
|
596 |
|
597 |
|
598 |
Float_t mvametsel;
|
599 |
Float_t mvametselphi;
|
600 |
Float_t mvametselx;
|
601 |
Float_t mvametsely;
|
602 |
Float_t mvametselsig;
|
603 |
|
604 |
Float_t mvametfirst;
|
605 |
Float_t mvametfirstphi;
|
606 |
Float_t mvametfirstx;
|
607 |
Float_t mvametfirsty;
|
608 |
Float_t mvametfirstsig;
|
609 |
|
610 |
Double_t spfMet;
|
611 |
|
612 |
// ----------- VBF TAG STUFF -------------
|
613 |
Int_t vbfTag;
|
614 |
Float_t vbfbdt;
|
615 |
|
616 |
// ----------------------------------------
|
617 |
UChar_t ismc;
|
618 |
Int_t mcprocid;
|
619 |
|
620 |
//corrected quantities from PhotonFix corrections
|
621 |
Float_t masscor;
|
622 |
Float_t masscorerr;
|
623 |
Float_t masscorele;
|
624 |
Float_t masscoreleerr;
|
625 |
|
626 |
//jet quantities for met phi cut
|
627 |
Float_t jetleadNoIDpt;
|
628 |
Float_t jetleadNoIDeta;
|
629 |
Float_t jetleadNoIDphi;
|
630 |
Float_t jetleadNoIDmass;
|
631 |
|
632 |
//jet quantities
|
633 |
Float_t jet1pt;
|
634 |
Float_t jet1eta;
|
635 |
Float_t jet1phi;
|
636 |
Float_t jet1mass;
|
637 |
Float_t jet2pt;
|
638 |
Float_t jet2eta;
|
639 |
Float_t jet2phi;
|
640 |
Float_t jet2mass;
|
641 |
Float_t jetcentralpt;
|
642 |
Float_t jetcentraleta;
|
643 |
Float_t jetcentralphi;
|
644 |
Float_t jetcentralmass;
|
645 |
Float_t dijetpt;
|
646 |
Float_t dijeteta;
|
647 |
Float_t dijetphi;
|
648 |
Float_t dijetmass;
|
649 |
Float_t jetetaplus;
|
650 |
Float_t jetetaminus;
|
651 |
|
652 |
// Float_t uncorrjet1pt;
|
653 |
// Float_t uncorrjet1eta;
|
654 |
// Float_t uncorrjet1phi;
|
655 |
// Float_t uncorrjet1mass;
|
656 |
// Float_t uncorrjet2pt;
|
657 |
// Float_t uncorrjet2eta;
|
658 |
// Float_t uncorrjet2phi;
|
659 |
// Float_t uncorrjet2mass;
|
660 |
// Float_t uncorrjetcentralpt;
|
661 |
// Float_t uncorrjetcentraleta;
|
662 |
// Float_t uncorrjetcentralphi;
|
663 |
// Float_t uncorrjetcentralmass;
|
664 |
// Float_t diuncorrjetpt;
|
665 |
// Float_t diuncorrjeteta;
|
666 |
// Float_t diuncorrjetphi;
|
667 |
// Float_t diuncorrjetmass;
|
668 |
// Float_t uncorrjetetaplus;
|
669 |
// Float_t uncorrjetetaminus;
|
670 |
|
671 |
Float_t zeppenfeld;
|
672 |
Float_t dphidijetgg;
|
673 |
|
674 |
PhotonTreeWriterPhoton<16> photons[2];
|
675 |
};
|
676 |
|
677 |
class PhotonTreeWriter : public BaseMod
|
678 |
{
|
679 |
public:
|
680 |
PhotonTreeWriter(const char *name ="PhotonTreeWriter",
|
681 |
const char *title="Selecting PhotonPairs");
|
682 |
|
683 |
~PhotonTreeWriter();
|
684 |
|
685 |
// setting all the input Names
|
686 |
void SetInputPhotonsName(const char *n){ fPhotonBranchName= n; }
|
687 |
void SetPhotonsFromBranch(bool b) { fPhotonsFromBranch = b; }
|
688 |
void SetTrackName(const char *n) { fTrackBranchName = n; }
|
689 |
void SetElectronName(const char *n) { fElectronName = n; }
|
690 |
void SetConversionName(const char *n) { fConversionName = n; }
|
691 |
void SetPUDensityName(const char *n) { fPileUpDenName = n; }
|
692 |
void SetPVName(const char *n) { fPVName = n; }
|
693 |
void SetPVFromBranch(bool b) { fPVFromBranch = b; }
|
694 |
void SetMCParticle(const char *n) { fMCParticleName = n; }
|
695 |
void SetPUInfoName(const char *n) { fPileUpName = n; }
|
696 |
void SetBeamspotName(const char *n) { fBeamspotName = n; }
|
697 |
void SetPFCandName(const char *n) { fPFCandName = n; }
|
698 |
void SetSuperClusterName(const char *n) { fSuperClusterName = n; }
|
699 |
void SetPFJetName(const char *n) { fPFJetName = n; }
|
700 |
void SetGenJetName(const char *n) { fGenJetName = n; }
|
701 |
void SetuncorrPFJetName(const char *n) { funcorrPFJetName = n; }
|
702 |
void SetPFNoPileUpName(const char *n) { fPFNoPileUpName = n; }
|
703 |
void SetPFPileUpName(const char *n) { fPFPileUpName = n; }
|
704 |
|
705 |
|
706 |
void SetPFJetsFromBranch(Bool_t b) { fPFJetsFromBranch = b; }
|
707 |
void SetEnableJets(Bool_t b) { fEnableJets = b; }
|
708 |
void SetApplyJetId(Bool_t b) { fApplyJetId = b; }
|
709 |
void SetApplyLeptonTag(Bool_t b) { fApplyLeptonTag = b; }
|
710 |
void SetApplyVBFTag(Bool_t b) { fApplyVBFTag = b; }
|
711 |
void SetApplyBTag(Bool_t b) { fApplyBTag = b; }
|
712 |
void SetApplyPFMetCorr(Bool_t b) { fApplyPFMetCorrections = b; }
|
713 |
void SetPhFixDataFile(const char *n) { fPhFixDataFile = n; }
|
714 |
|
715 |
|
716 |
|
717 |
|
718 |
// set basic Cut variables (FOR PRE-SELECTION)
|
719 |
|
720 |
// is Data Or Not?
|
721 |
void SetIsData (Bool_t b) { fIsData = b; };
|
722 |
|
723 |
|
724 |
void SetApplyElectronVeto(Bool_t b) { fApplyElectronVeto = b; }
|
725 |
|
726 |
void SetTupleName(const char* c) { fTupleName = c; }
|
727 |
void SetGoodElectronsFromBranch(Bool_t b) { fGoodElectronsFromBranch = b; }
|
728 |
void SetGoodElectronName(TString name) { fGoodElectronName = name; }
|
729 |
void SetWriteDiphotonTree(Bool_t b) { fWriteDiphotonTree = b; }
|
730 |
void SetWriteSingleTree(Bool_t b) { fWriteSingleTree = b; }
|
731 |
void SetLoopOnGoodElectrons(Bool_t b) { fLoopOnGoodElectrons = b; }
|
732 |
void SetEnablePFPhotons(Bool_t b) { fEnablePFPhotons = b; }
|
733 |
void SetExcludeSinglePrompt(Bool_t b) { fExcludeSinglePrompt = b; }
|
734 |
void SetExcludeDoublePrompt(Bool_t b) { fExcludeDoublePrompt = b; }
|
735 |
|
736 |
void SetLeptonTagElectronsName(TString name) { fLeptonTagElectronsName = name; }
|
737 |
void SetLeptonTagMuonsName (TString name) { fLeptonTagMuonsName = name; }
|
738 |
void SetFillClusterArrays(Bool_t b) { fFillClusterArrays = b; }
|
739 |
|
740 |
void SetDo2012LepTag(Bool_t b) { fDo2012LepTag = b; }
|
741 |
|
742 |
void SetBeamspotWidth(Double_t x) { fBeamspotWidth = x; }
|
743 |
|
744 |
void SetElectronMVAWeightsSubdet0Pt10To20(TString s)
|
745 |
{ fElectronMVAWeights_Subdet0Pt10To20 = s; }
|
746 |
void SetElectronMVAWeightsSubdet1Pt10To20(TString s)
|
747 |
{ fElectronMVAWeights_Subdet1Pt10To20 = s; }
|
748 |
void SetElectronMVAWeightsSubdet2Pt10To20(TString s)
|
749 |
{ fElectronMVAWeights_Subdet2Pt10To20 = s; }
|
750 |
void SetElectronMVAWeightsSubdet0Pt20ToInf(TString s)
|
751 |
{ fElectronMVAWeights_Subdet0Pt20ToInf = s; }
|
752 |
void SetElectronMVAWeightsSubdet1Pt20ToInf(TString s)
|
753 |
{ fElectronMVAWeights_Subdet1Pt20ToInf = s; }
|
754 |
void SetElectronMVAWeightsSubdet2Pt20ToInf(TString s)
|
755 |
{ fElectronMVAWeights_Subdet2Pt20ToInf = s; }
|
756 |
|
757 |
void SetDoSynching(bool b) {fDoSynching = b;}
|
758 |
|
759 |
|
760 |
|
761 |
protected:
|
762 |
void Process();
|
763 |
void SlaveBegin();
|
764 |
// Private auxiliary methods...
|
765 |
void FindHiggsPtAndZ(Float_t& pt, Float_t& z, Float_t& mass);
|
766 |
Float_t GetEventCat (PhotonTools::CiCBaseLineCats cat1,
|
767 |
PhotonTools::CiCBaseLineCats cat2);
|
768 |
|
769 |
// Names for the input Collections
|
770 |
TString fPhotonBranchName;
|
771 |
TString fPFPhotonName;
|
772 |
TString fElectronName;
|
773 |
TString fGoodElectronName;
|
774 |
TString fConversionName;
|
775 |
TString fPFConversionName;
|
776 |
TString fTrackBranchName;
|
777 |
TString fPileUpDenName;
|
778 |
TString fPVName;
|
779 |
TString fBeamspotName;
|
780 |
TString fPFCandName;
|
781 |
TString fPFNoPileUpName; //name of pfnpu collection
|
782 |
TString fPFPileUpName; //name of pfpu collection
|
783 |
|
784 |
TString fMCParticleName;
|
785 |
TString fMCEventInfoName;
|
786 |
|
787 |
TString fPileUpName;
|
788 |
TString fSuperClusterName;
|
789 |
TString fPFMetName;
|
790 |
TString fPFJetName;
|
791 |
|
792 |
|
793 |
TString funcorrPFJetName;
|
794 |
TString fGenJetName; //added to do pfmet correction 05/01/2012
|
795 |
|
796 |
|
797 |
TString fLeptonTagElectronsName;
|
798 |
TString fLeptonTagMuonsName;
|
799 |
|
800 |
|
801 |
// is it Data or MC?
|
802 |
Bool_t fIsData;
|
803 |
|
804 |
// in case there's some PV pre-selection
|
805 |
Bool_t fPhotonsFromBranch;
|
806 |
Bool_t fPVFromBranch;
|
807 |
Bool_t fGoodElectronsFromBranch;
|
808 |
Bool_t fPFJetsFromBranch;
|
809 |
|
810 |
Bool_t fDoSynching;
|
811 |
|
812 |
const PhotonCol *fPhotons;
|
813 |
const PhotonCol *fPFPhotons;
|
814 |
const ElectronCol *fElectrons;
|
815 |
const ElectronCol *fGoodElectrons;
|
816 |
const DecayParticleCol *fConversions;
|
817 |
const DecayParticleCol *fPFConversions;
|
818 |
const TrackCol *fTracks;
|
819 |
const PileupEnergyDensityCol *fPileUpDen;
|
820 |
const VertexCol *fPV;
|
821 |
const BeamSpotCol *fBeamspot;
|
822 |
const PFCandidateCol *fPFCands;
|
823 |
const MCParticleCol *fMCParticles;
|
824 |
const MCEventInfo *fMCEventInfo;
|
825 |
const PileupInfoCol *fPileUp;
|
826 |
const SuperClusterCol *fSuperClusters;
|
827 |
const PFMetCol *fPFMet;
|
828 |
const JetCol *fPFJets;
|
829 |
const GenJetCol *fGenJets;
|
830 |
const PFJetCol *funcorrPFJets;
|
831 |
|
832 |
const ElectronCol *fLeptonTagElectrons;
|
833 |
const MuonCol *fLeptonTagMuons;
|
834 |
const PFCandidateCol *fPFNoPileUpCands; //!pfnpu collection
|
835 |
const PFCandidateCol *fPFPileUpCands; //!pfpu collection
|
836 |
|
837 |
// --------------------------------
|
838 |
Bool_t fLoopOnGoodElectrons; //loop over good elecs instead of photons
|
839 |
Bool_t fApplyElectronVeto; //invert elec veto (for cic sel. only atm)
|
840 |
Bool_t fWriteDiphotonTree;
|
841 |
Bool_t fWriteSingleTree;
|
842 |
|
843 |
Bool_t fEnablePFPhotons;
|
844 |
|
845 |
Bool_t fExcludeSinglePrompt;
|
846 |
Bool_t fExcludeDoublePrompt;
|
847 |
|
848 |
Bool_t fEnableJets;
|
849 |
Bool_t fApplyJetId;
|
850 |
|
851 |
Bool_t fApplyLeptonTag;
|
852 |
Bool_t fApplyVBFTag;
|
853 |
Bool_t fApplyBTag;
|
854 |
Bool_t fApplyPFMetCorrections;
|
855 |
|
856 |
Bool_t fFillClusterArrays;
|
857 |
Bool_t fFillVertexTree;
|
858 |
|
859 |
Bool_t fDo2012LepTag;
|
860 |
|
861 |
TString fPhFixDataFile;
|
862 |
PhotonFix fPhfixph;
|
863 |
PhotonFix fPhfixele;
|
864 |
|
865 |
Double_t fBeamspotWidth;
|
866 |
|
867 |
// --------------------------------
|
868 |
// variables for vbf
|
869 |
float jet1pt_vbf;
|
870 |
float jet2pt_vbf;
|
871 |
float deltajeteta_vbf;
|
872 |
float dijetmass_vbf;
|
873 |
float zeppenfeld_vbf;
|
874 |
float dphidijetgg_vbf;
|
875 |
float diphoptOverdiphomass_vbf;
|
876 |
float pho1ptOverdiphomass_vbf;
|
877 |
float pho2ptOverdiphomass_vbf;
|
878 |
|
879 |
// --------------------------------
|
880 |
// validation Tuple
|
881 |
TString fTupleName;
|
882 |
PhotonTreeWriterDiphotonEvent* fDiphotonEvent;
|
883 |
PhotonTreeWriterPhoton<16>* fSinglePhoton;
|
884 |
TTree* hCiCTuple;
|
885 |
TTree* hCiCTupleSingle;
|
886 |
|
887 |
PhotonTreeWriterVtx *fDiphotonVtx;
|
888 |
TTree *hVtxTree;
|
889 |
|
890 |
MVAMet fMVAMet;
|
891 |
JetIDMVA fJetId;
|
892 |
MVAVBF fMVAVBF;
|
893 |
|
894 |
VertexTools fVtxTools;
|
895 |
|
896 |
ElectronIDMVA *fElectronIDMVA;
|
897 |
TString fElectronMVAWeights_Subdet0Pt10To20;
|
898 |
TString fElectronMVAWeights_Subdet1Pt10To20;
|
899 |
TString fElectronMVAWeights_Subdet2Pt10To20;
|
900 |
TString fElectronMVAWeights_Subdet0Pt20ToInf;
|
901 |
TString fElectronMVAWeights_Subdet1Pt20ToInf;
|
902 |
TString fElectronMVAWeights_Subdet2Pt20ToInf;
|
903 |
|
904 |
RhoUtilities::RhoType fTheRhoType;
|
905 |
|
906 |
ClassDef(PhotonTreeWriter, 1) // Photon identification module
|
907 |
};
|
908 |
}
|
909 |
#endif
|