ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitPhysics/Mods/interface/PhotonTreeWriter.h
Revision: 1.40
Committed: Thu Nov 7 17:15:45 2013 UTC (11 years, 6 months ago) by veverka
Content type: text/plain
Branch: MAIN
Changes since 1.39: +13 -1 lines
Log Message:
Fixed bug in dilepton selection for VH lep cat

File Contents

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