ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitPhysics/Mods/interface/PhotonTreeWriter.h
Revision: 1.23
Committed: Tue Jul 24 15:46:36 2012 UTC (12 years, 9 months ago) by bendavid
Content type: text/plain
Branch: MAIN
Changes since 1.22: +10 -3 lines
Log Message:
fix for shower scaling and some new options for beamspot width

File Contents

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