ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitPhysics/Mods/interface/PhotonTreeWriter.h
Revision: 1.34
Committed: Fri Aug 30 23:25:24 2013 UTC (11 years, 8 months ago) by veverka
Content type: text/plain
Branch: MAIN
Changes since 1.33: +17 -8 lines
Log Message:
Cleaned up ttH tag, added a prototype of the V(lep)H tag

File Contents

# Content
1 //--------------------------------------------------------------------------------------------------
2 // $Id: PhotonTreeWriter.h,v 1.33 2013/08/30 12:02:37 veverka Exp $
3 //
4 // PhotonTreeWriter
5 //
6 // Authors: J. Bendavid, J. Veverka
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 class TNtuple;
40 class TRandom3;
41
42 namespace mithep
43 {
44
45 class PhotonTreeWriterVtx
46 {
47
48 public:
49 void SetVars(const Vertex *v, const Photon *p1, const Photon *p2, const PFCandidateCol *pfcands, Int_t idx, Int_t numvtx, Float_t genvtxz);
50
51 private:
52 Int_t n;
53 Int_t nvtx;
54 Float_t ptgg;
55 Float_t phigg;
56 Float_t etagg;
57 Float_t pxgg;
58 Float_t pygg;
59 Float_t pzgg;
60 Float_t mgg;
61 Float_t zgen;
62 Float_t x;
63 Float_t y;
64 Float_t z;
65 Float_t pt;
66 Float_t phi;
67 Float_t eta;
68 Float_t px;
69 Float_t py;
70 Float_t pz;
71 Float_t sumpt;
72 Float_t sumptsq;
73 Int_t nchalltoward;
74 Int_t nchalltransverse;
75 Int_t nchallaway;
76 Int_t nchcuttoward;
77 Int_t nchcuttransverse;
78 Int_t nchcutaway;
79 };
80
81 template <int NClus=16>//ming: what's template
82 class PhotonTreeWriterPhoton
83 {
84 public:
85 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);
86 Float_t Ecor() const { return ecor; };
87 Float_t Ecorerr() const { return ecorerr; };
88 Float_t Ecorele() const { return ecorele; };
89 Float_t Ecoreleerr() const { return ecoreleerr; };
90 Float_t Pt() const {return pt;};
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; // flavor-based, Moriond 2013
465 Int_t leptonTag2; // MET-based, since legacy paper 2013
466 // ----------- VEERTEX SYNCHING STUFF -------
467
468 Int_t vtxInd1;
469 Int_t vtxInd2;
470 Int_t vtxInd3;
471
472 Float_t vtxBestPtbal ;
473 Float_t vtxBestPtasym ;
474 Float_t vtxBestSumpt2 ;
475 Float_t vtxBestP2Conv ;
476
477 Float_t vtxMva1Z ;
478 Float_t vtxMva2Z ;
479 Float_t vtxMva3Z ;
480
481 Float_t vtxMva1 ;
482 Float_t vtxMva2 ;
483 Float_t vtxMva3 ;
484
485 Int_t vtxNleg1 ;
486 Int_t vtxNleg2 ;
487 Int_t vtxConvIdx1 ;
488 Int_t vtxConvIdx2 ;
489 Int_t vtxNconv ;
490
491 Float_t vtxConv1Z ;
492 Float_t vtxConv1DZ ;
493 Float_t vtxConv1Prob ;
494
495 Float_t vtxConv2Z ;
496 Float_t vtxConv2DZ ;
497 Float_t vtxConv2Prob ;
498
499
500 // ---------- MUON STUFF --------------------
501 Float_t muonPt;
502 Float_t muonEta;
503 Float_t muDR1;
504 Float_t muDR2;
505 Float_t muIso1;
506 Float_t muIso2;
507 Float_t muIso3;
508 Float_t muIso4;
509 Float_t muD0;
510 Float_t muDZ;
511 Int_t muNhits;
512 Float_t muChi2;
513 Int_t muNpixhits;
514 Int_t muNegs;
515 Int_t muNMatch;
516 // ----------- ELECTRON STUFF --------------
517 Float_t elePt;
518 Float_t eleEta;
519 Float_t eleSCEta;
520 Float_t eleIso1;
521 Float_t eleIso2;
522 Float_t eleIso3;
523 Float_t eleIso4;
524 Float_t eleDist;
525 Float_t eleDcot;
526 Float_t eleCoviee;
527 Float_t eleDphiin;
528 Float_t eleDetain;
529 Float_t eleDR1;
530 Float_t eleDR2;
531 Float_t eleMass1;
532 Float_t eleMass2;
533 Int_t eleNinnerHits;
534 Float_t eleIdMva;
535 // -----------------------------------------
536 Float_t rho;
537 Float_t rho25;
538 Float_t rhoold;
539 Float_t genHiggspt;
540 Float_t genHiggsZ;
541 Float_t genmass;
542 Float_t gencostheta;
543 Float_t genz;
544 Float_t bsX;
545 Float_t bsY;
546 Float_t bsZ;
547 Float_t bsSigmaZ;
548 Float_t vtxX;
549 Float_t vtxY;
550 Float_t vtxZ;
551 Int_t nVtx;
552 Int_t numPU;
553 Int_t numPUminus;
554 Int_t numPUplus;
555 Float_t mass;
556 Float_t masserr;
557 Float_t masserrsmeared;
558 Float_t masserrwrongvtx;
559 Float_t masserrsmearedwrongvtx;
560 Float_t vtxprob;
561 Float_t deltamvtx;
562 Float_t ptgg;
563 Float_t etagg;
564 Float_t phigg;
565 Float_t costheta;
566 Float_t massele;
567 Float_t ptee;
568 Float_t costhetaele;
569 Float_t mt;
570 Float_t cosphimet;
571 Float_t mtele;
572 Float_t cosphimetele;
573 UInt_t evt;
574 UInt_t run;
575 UInt_t lumi;
576 UChar_t evtcat;
577 UInt_t nobj;
578 Double_t fromZ; //added: Heng 2/14/2012
579 Double_t fromW; //added: Heng
580 Float_t zpt; //added: Heng
581 Float_t allZpt; //added: Heng
582 Float_t zEta; //added: Heng
583 Float_t allZEta; //added: Heng
584
585 //Met stuff
586 Float_t corrpfmet;
587 Float_t corrpfmetphi;
588 Float_t corrpfmetx;
589 Float_t corrpfmety;
590 Double_t dphiMetgg;
591 Double_t cosdphiMetgg;
592 Double_t dphiPhPh;
593 Float_t pfmet;
594 Float_t pfmetphi;
595 Float_t pfmetx;
596 Float_t pfmety;
597
598
599 Float_t mvametsel;
600 Float_t mvametselphi;
601 Float_t mvametselx;
602 Float_t mvametsely;
603 Float_t mvametselsig;
604
605 Float_t mvametfirst;
606 Float_t mvametfirstphi;
607 Float_t mvametfirstx;
608 Float_t mvametfirsty;
609 Float_t mvametfirstsig;
610
611 Double_t spfMet;
612
613 // ----------- VBF TAG STUFF -------------
614 Int_t vbfTag;
615 Float_t vbfbdt;
616
617 // ----------- TTH TAG STUFF -------------
618 Int_t tthTag;
619
620 // ----------------------------------------
621 UChar_t ismc;
622 Int_t mcprocid;
623 Double_t mcweight;
624
625 //corrected quantities from PhotonFix corrections
626 Float_t masscor;
627 Float_t masscorerr;
628 Float_t masscorele;
629 Float_t masscoreleerr;
630
631 //jet quantities for met phi cut
632 Float_t jetleadNoIDpt;
633 Float_t jetleadNoIDeta;
634 Float_t jetleadNoIDphi;
635 Float_t jetleadNoIDmass;
636
637 //jet quantities
638 Float_t jet1pt;
639 Float_t jet1eta;
640 Float_t jet1phi;
641 Float_t jet1mass;
642 Float_t jet2pt;
643 Float_t jet2eta;
644 Float_t jet2phi;
645 Float_t jet2mass;
646 Float_t jetcentralpt;
647 Float_t jetcentraleta;
648 Float_t jetcentralphi;
649 Float_t jetcentralmass;
650 Float_t dijetpt;
651 Float_t dijeteta;
652 Float_t dijetphi;
653 Float_t dijetmass;
654 Float_t jetetaplus;
655 Float_t jetetaminus;
656
657 // Float_t uncorrjet1pt;
658 // Float_t uncorrjet1eta;
659 // Float_t uncorrjet1phi;
660 // Float_t uncorrjet1mass;
661 // Float_t uncorrjet2pt;
662 // Float_t uncorrjet2eta;
663 // Float_t uncorrjet2phi;
664 // Float_t uncorrjet2mass;
665 // Float_t uncorrjetcentralpt;
666 // Float_t uncorrjetcentraleta;
667 // Float_t uncorrjetcentralphi;
668 // Float_t uncorrjetcentralmass;
669 // Float_t diuncorrjetpt;
670 // Float_t diuncorrjeteta;
671 // Float_t diuncorrjetphi;
672 // Float_t diuncorrjetmass;
673 // Float_t uncorrjetetaplus;
674 // Float_t uncorrjetetaminus;
675
676 Float_t zeppenfeld;
677 Float_t dphidijetgg;
678
679 PhotonTreeWriterPhoton<16> photons[2];
680 }; // PhotonTreeWriterDiphotonEvent
681
682 class PhotonTreeWriter : public BaseMod
683 {
684 public:
685 PhotonTreeWriter(const char *name ="PhotonTreeWriter",
686 const char *title="Selecting PhotonPairs");
687
688 ~PhotonTreeWriter();
689
690 // setting all the input Names
691 void SetInputPhotonsName(const char *n){ fPhotonBranchName= n; }
692 void SetPhotonsFromBranch(bool b) { fPhotonsFromBranch = b; }
693 void SetTrackName(const char *n) { fTrackBranchName = n; }
694 void SetElectronName(const char *n) { fElectronName = n; }
695 void SetConversionName(const char *n) { fConversionName = n; }
696 void SetPUDensityName(const char *n) { fPileUpDenName = n; }
697 void SetPVName(const char *n) { fPVName = n; }
698 void SetPVFromBranch(bool b) { fPVFromBranch = b; }
699 void SetMCParticle(const char *n) { fMCParticleName = n; }
700 void SetPUInfoName(const char *n) { fPileUpName = n; }
701 void SetBeamspotName(const char *n) { fBeamspotName = n; }
702 void SetPFCandName(const char *n) { fPFCandName = n; }
703 void SetSuperClusterName(const char *n) { fSuperClusterName = n; }
704 void SetPFJetName(const char *n) { fPFJetName = n; }
705 void SetGenJetName(const char *n) { fGenJetName = n; }
706 void SetuncorrPFJetName(const char *n) { funcorrPFJetName = n; }
707 void SetPFNoPileUpName(const char *n) { fPFNoPileUpName = n; }
708 void SetPFPileUpName(const char *n) { fPFPileUpName = n; }
709
710
711 void SetPFJetsFromBranch(Bool_t b) { fPFJetsFromBranch = b; }
712 void SetEnableJets(Bool_t b) { fEnableJets = b; }
713 void SetEnableGenJets(Bool_t b) { fEnableGenJets = b; }
714 void SetApplyJetId(Bool_t b) { fApplyJetId = b; }
715 void SetApplyLeptonTag(Bool_t b) { fApplyLeptonTag = b; }
716 void SetApplyLeptonTag2(Bool_t b) { fApplyLeptonTag2 = b; }
717 void SetApplyVBFTag(Bool_t b) { fApplyVBFTag = b; }
718 void SetApplyTTHTag(Bool_t b) { fApplyTTHTag = b; }
719 void SetApplyBTag(Bool_t b) { fApplyBTag = b; }
720 void SetApplyPFMetCorr(Bool_t b) { fApplyPFMetCorrections = b; }
721 void SetPhFixDataFile(const char *n) { fPhFixDataFile = n; }
722 void SetVerbosityLevel(Bool_t b) { fVerbosityLevel = b; }
723
724
725
726
727 // set basic Cut variables (FOR PRE-SELECTION)
728
729 // is Data Or Not?
730 void SetIsData (Bool_t b) { fIsData = b; };
731
732
733 void SetApplyElectronVeto(Bool_t b) { fApplyElectronVeto = b; }
734
735 void SetTupleName(const char* c) { fTupleName = c; }
736 void SetGoodElectronsFromBranch(Bool_t b) { fGoodElectronsFromBranch = b; }
737 void SetGoodElectronName(TString name) { fGoodElectronName = name; }
738 void SetWriteDiphotonTree(Bool_t b) { fWriteDiphotonTree = b; }
739 void SetWriteSingleTree(Bool_t b) { fWriteSingleTree = b; }
740 void SetLoopOnGoodElectrons(Bool_t b) { fLoopOnGoodElectrons = b; }
741 void SetEnablePFPhotons(Bool_t b) { fEnablePFPhotons = b; }
742 void SetExcludeSinglePrompt(Bool_t b) { fExcludeSinglePrompt = b; }
743 void SetExcludeDoublePrompt(Bool_t b) { fExcludeDoublePrompt = b; }
744
745 void SetLeptonTagElectronsName(TString name) { fLeptonTagElectronsName = name; }
746 void SetLeptonTagMuonsName (TString name) { fLeptonTagMuonsName = name; }
747 void SetFillClusterArrays(Bool_t b) { fFillClusterArrays = b; }
748
749 void SetDo2012LepTag(Bool_t b) { fDo2012LepTag = b; }
750
751 void SetBeamspotWidth(Double_t x) { fBeamspotWidth = x; }
752
753 void SetElectronMVAWeightsSubdet0Pt10To20(TString s)
754 { fElectronMVAWeights_Subdet0Pt10To20 = s; }
755 void SetElectronMVAWeightsSubdet1Pt10To20(TString s)
756 { fElectronMVAWeights_Subdet1Pt10To20 = s; }
757 void SetElectronMVAWeightsSubdet2Pt10To20(TString s)
758 { fElectronMVAWeights_Subdet2Pt10To20 = s; }
759 void SetElectronMVAWeightsSubdet0Pt20ToInf(TString s)
760 { fElectronMVAWeights_Subdet0Pt20ToInf = s; }
761 void SetElectronMVAWeightsSubdet1Pt20ToInf(TString s)
762 { fElectronMVAWeights_Subdet1Pt20ToInf = s; }
763 void SetElectronMVAWeightsSubdet2Pt20ToInf(TString s)
764 { fElectronMVAWeights_Subdet2Pt20ToInf = s; }
765
766 void SetDoSynching(bool b) {fDoSynching = b;}
767
768
769
770 protected:
771 void Process();
772 void SlaveBegin();
773 void Terminate();
774 // Private auxiliary methods...
775 void FindHiggsPtAndZ(Float_t& pt, Float_t& z, Float_t& mass);
776 Float_t GetEventCat (PhotonTools::CiCBaseLineCats cat1,
777 PhotonTools::CiCBaseLineCats cat2);
778 void ApplyLeptonTag(const Photon *phHard,
779 const Photon *phSoft,
780 const Vertex *selvtx);
781 void ApplyLeptonTag2(const Photon *phHard,
782 const Photon *phSoft,
783 const Vertex *selvtx);
784 void ApplyTTHTag(const Photon *phHard, const Photon *phSoft,const Vertex *selvtx);
785 // void PrintTTHDebugInfo();
786 // void PrintTTHDecay();
787 // void PrintGenElectrons();
788 // void PrintElectrons(const char *, const ElectronCol*);
789
790 // Names for the input Collections
791 TString fPhotonBranchName;
792 TString fPFPhotonName;
793 TString fElectronName;
794 TString fGoodElectronName;
795 TString fConversionName;
796 TString fPFConversionName;
797 TString fTrackBranchName;
798 TString fPileUpDenName;
799 TString fPVName;
800 TString fBeamspotName;
801 TString fPFCandName;
802 TString fPFNoPileUpName; //name of pfnpu collection
803 TString fPFPileUpName; //name of pfpu collection
804
805 TString fMCParticleName;
806 TString fMCEventInfoName;
807
808 TString fPileUpName;
809 TString fSuperClusterName;
810 TString fPFMetName;
811 TString fPFJetName;
812
813
814 TString funcorrPFJetName;
815 TString fGenJetName; //added to do pfmet correction 05/01/2012
816
817
818 TString fLeptonTagElectronsName;
819 TString fLeptonTagMuonsName;
820
821
822 // is it Data or MC?
823 Bool_t fIsData;
824
825 // in case there's some PV pre-selection
826 Bool_t fPhotonsFromBranch;
827 Bool_t fPVFromBranch;
828 Bool_t fGoodElectronsFromBranch;
829 Bool_t fPFJetsFromBranch;
830
831 Bool_t fDoSynching;
832
833 const PhotonCol *fPhotons;
834 const PhotonCol *fPFPhotons;
835 const ElectronCol *fElectrons;
836 const ElectronCol *fGoodElectrons;
837 const DecayParticleCol *fConversions;
838 const DecayParticleCol *fPFConversions;
839 const TrackCol *fTracks;
840 const PileupEnergyDensityCol *fPileUpDen;
841 const VertexCol *fPV;
842 const BeamSpotCol *fBeamspot;
843 const PFCandidateCol *fPFCands;
844 const MCParticleCol *fMCParticles;
845 const MCEventInfo *fMCEventInfo;
846 const PileupInfoCol *fPileUp;
847 const SuperClusterCol *fSuperClusters;
848 const PFMetCol *fPFMet;
849 const JetCol *fPFJets;
850 const GenJetCol *fGenJets;
851 const PFJetCol *funcorrPFJets;
852
853 const ElectronCol *fLeptonTagElectrons;
854 const MuonCol *fLeptonTagMuons;
855 const PFCandidateCol *fPFNoPileUpCands; //!pfnpu collection
856 const PFCandidateCol *fPFPileUpCands; //!pfpu collection
857
858 // --------------------------------
859 Bool_t fLoopOnGoodElectrons; //loop over good elecs instead of photons
860 Bool_t fApplyElectronVeto; //invert elec veto (for cic sel. only atm)
861 Bool_t fWriteDiphotonTree;
862 Bool_t fWriteSingleTree;
863
864 Bool_t fEnablePFPhotons;
865
866 Bool_t fExcludeSinglePrompt;
867 Bool_t fExcludeDoublePrompt;
868
869 Bool_t fEnableJets;
870 Bool_t fEnableGenJets;
871 Bool_t fApplyJetId;
872
873 Bool_t fApplyLeptonTag;
874 Bool_t fApplyLeptonTag2;
875 Bool_t fApplyVBFTag;
876 Bool_t fApplyTTHTag;
877 Bool_t fApplyBTag;
878 Bool_t fApplyPFMetCorrections;
879
880 Bool_t fFillClusterArrays;
881 Bool_t fFillVertexTree;
882
883 Bool_t fDo2012LepTag;
884
885 Int_t fVerbosityLevel;
886
887 TString fPhFixDataFile;
888 PhotonFix fPhfixph;
889 PhotonFix fPhfixele;
890
891 Double_t fBeamspotWidth;
892
893 TFile *fTmpFile;
894
895 // --------------------------------
896 // variables for vbf
897 float jet1pt_vbf;
898 float jet2pt_vbf;
899 float deltajeteta_vbf;
900 float dijetmass_vbf;
901 float zeppenfeld_vbf;
902 float dphidijetgg_vbf;
903 float diphoptOverdiphomass_vbf;
904 float pho1ptOverdiphomass_vbf;
905 float pho2ptOverdiphomass_vbf;
906
907 // --------------------------------
908 // validation Tuple
909 TString fTupleName;
910 PhotonTreeWriterDiphotonEvent* fDiphotonEvent;
911 PhotonTreeWriterPhoton<16>* fSinglePhoton;
912 TTree* hCiCTuple;
913 TTree* hCiCTupleSingle;
914
915 PhotonTreeWriterVtx *fDiphotonVtx;
916 TTree *hVtxTree;
917
918 MVAMet fMVAMet;
919 JetIDMVA fJetId;
920 MVAVBF fMVAVBF;
921
922 VertexTools fVtxTools;
923
924 ElectronIDMVA *fElectronIDMVA;
925 TString fElectronMVAWeights_Subdet0Pt10To20;
926 TString fElectronMVAWeights_Subdet1Pt10To20;
927 TString fElectronMVAWeights_Subdet2Pt10To20;
928 TString fElectronMVAWeights_Subdet0Pt20ToInf;
929 TString fElectronMVAWeights_Subdet1Pt20ToInf;
930 TString fElectronMVAWeights_Subdet2Pt20ToInf;
931
932 RhoUtilities::RhoType fTheRhoType;
933
934 ClassDef(PhotonTreeWriter, 1) // Photon identification module
935 };
936 }
937 #endif