Mods/interface/PhotonIDMod.h

//--------------------------------------------------------------------------------------------------
// $Id: PhotonIDMod.h,v 1.18 2011/05/18 14:01:18 bendavid Exp $
//
// PhotonIDMod
//
// This module applies photon identification criteria and exports a pointer to a collection
// of "good photons" according to the specified identification scheme.
//
// Authors: S.Xie, C.Loizides
//--------------------------------------------------------------------------------------------------

#ifndef MITPHYSICS_MODS_PHOTONIDMOD_H
#define MITPHYSICS_MODS_PHOTONIDMOD_H

#include "MitAna/TreeMod/interface/BaseMod.h" 
#include "MitAna/DataTree/interface/PhotonFwd.h"
#include "MitAna/DataTree/interface/TrackCol.h"
#include "MitAna/DataTree/interface/BeamSpotCol.h"
#include "MitAna/DataTree/interface/PileupEnergyDensityCol.h"
#include "MitAna/DataTree/interface/DecayParticleCol.h"
#include "MitAna/DataTree/interface/ElectronCol.h"

namespace mithep 
{
  class PhotonIDMod : public BaseMod
  {
    public:
      PhotonIDMod(const char *name="PhotonIDMod", 
                  const char *title="Photon identification module");

      Bool_t              GetApplySpikeRemoval()      const { return fApplySpikeRemoval;   }
      Bool_t              GetApplyPixelSeed()         const { return fApplyPixelSeed;      }
      const char         *GetGoodName()               const { return GetGoodPhotonsName(); }   
      const char         *GetGoodPhotonsName()        const { return fGoodPhotonsName;     }   
      Double_t            GetHadOverEmMax()           const { return fHadOverEmMax;        }
      const char         *GetIDType()                 const { return fPhotonIDType;        }
      const char         *GetInputName()              const { return fPhotonBranchName;    }   
      const char         *GetIsoType()                const { return fPhotonIsoType;       }
      const char         *GetOutputName()             const { return GetGoodPhotonsName(); }   
      Double_t            GetPtMin()                  const { return fPhotonPtMin;         }
      Bool_t              GetApplyFiduciality()       const { return fFiduciality;         }
      Double_t            GetEtaWidthEB()             const { return fEtaWidthEB;          }
      Double_t            GetEtaWidthEE()             const { return fEtaWidthEE;          }
      Double_t            GetAbsEtaMax()              const { return fAbsEtaMax;           }
      void                SetApplySpikeRemoval(Bool_t b)    { fApplySpikeRemoval  = b;     }
      void                SetApplyPixelSeed(Bool_t b)       { fApplyPixelSeed  = b;        }
      void                SetApplyElectronVeto(Bool_t b)    { fApplyElectronVeto = b;      }
      void                SetApplyElectronVetoConvRecovery(Bool_t b) { fApplyElectronVetoConvRecovery = b; }
      void                SetApplyConversionId(Bool_t b)    { fApplyConversionId = b;      }
      void                SetApplyTriggerMatching(Bool_t b)      { fApplyTriggerMatching = b;  }      
      void                SetGoodName(const char *n)        { SetGoodPhotonsName(n);       }   
      void                SetGoodPhotonsName(const char *n) { fGoodPhotonsName = n;        }   
      void                SetHadOverEmMax(Double_t hoe)     { fHadOverEmMax    = hoe;      }
      void                SetIDType(const char *type)       { fPhotonIDType    = type;     }
      void                SetInputName(const char *n)       { fPhotonBranchName= n;        }   
      void                SetTrackName(const char *n)       { fTrackBranchName = n;        }   
      void                SetBeamspotName(const char *n)    { fBeamspotBranchName = n;     }   
      void                SetIsoType(const char *type)      { fPhotonIsoType   = type;     }
      void                SetOutputName(const char *n)      { SetGoodPhotonsName(n);       }    
      void                SetPtMin(Double_t pt)             { fPhotonPtMin     = pt;       }
      void                SetR9Min(Double_t x)              { fPhotonR9Min     = x;        }
      void                SetEtaWidthEB(Double_t x)         { fEtaWidthEB      = x;        }
      void                SetEtaWidthEE(Double_t x)         { fEtaWidthEE      = x;        }
      void                SetAbsEtaMax(Double_t x)          { fAbsEtaMax       = x;        }
      void                SetApplyR9Min(Bool_t b)           { fApplyR9Min      = b;        }
      void                SetApplyFiduciality(Bool_t b)     { fFiduciality = b;            }      
      void                SetEffAreas(Double_t ecal, Double_t hcal, Double_t track) { 
        fEffAreaEcalEE = ecal; fEffAreaHcalEE = hcal; fEffAreaTrackEE = track;
        fEffAreaEcalEB = ecal; fEffAreaHcalEB = hcal; fEffAreaTrackEB = track;
      }
      void                SetEffAreasEEEB(Double_t ecalEE, Double_t hcalEE, Double_t trackEE,
                                          Double_t ecalEB, Double_t hcalEB, Double_t trackEB) { 
        fEffAreaEcalEE = ecalEE; fEffAreaHcalEE = hcalEE; fEffAreaTrackEE = trackEE;
        fEffAreaEcalEB = ecalEB; fEffAreaHcalEB = hcalEB; fEffAreaTrackEB = trackEB;
      }
      void                SetTriggerObjectsName(const char *n)   { fTrigObjectsName = n;       }
    
      enum EPhIdType {
        kIdUndef = 0,       //not defined
        kTight,             //"Tight"
        kLoose,             //"Loose"
        kLooseEM,           //"LooseEM"
        kBaseLineId,        //"2011" Hgg BaseLine CiC
        kCustomId           //"Custom"
      };

      enum EPhIsoType {
        kIsoUndef = 0,      //not defined        
        kNoIso,             //"NoIso"
        kCombinedIso,       //"CombinedIso"
        kCustomIso,         //"Custom"
        kBaseLineIso,       //"2011" Hgg BaseLine CiC
        kMITPUCorrected     //PileUp Corrected Hgg Isolation
      };

    protected:
      void                Process();
      void                SlaveBegin();

      TString             fPhotonBranchName;     //name of photon collection (input)
      TString             fGoodPhotonsName;      //name of exported "good photon" collection
      TString             fTrackBranchName;      // name of the track collection (only needed for PU corrected isolation)
      TString             fBeamspotBranchName;   //name of the Beamspot collection (only needed for PU corrected isolation)
      TString             fPileUpDenName;        //name of the PU density collection      
      TString             fConversionName;       //name of conversion branch
      TString             fElectronName;
      TString             fTrigObjectsName;        //name of trigger object collection      
      TString             fPhotonIDType;         //type of photon identification we impose
      TString             fPhotonIsoType;        //type of photon isolation we impose
      Double_t            fPhotonPtMin;          //min pt cut
      Double_t            fHadOverEmMax;         //maximum of hadronic/em energy
      Bool_t              fApplySpikeRemoval;    //whether apply spike removal      
      Bool_t              fApplyPixelSeed;       //=true then apply pixel seed constraint
      Bool_t              fApplyElectronVeto;    //=true then apply electron veto (with no conversion recovery)
      Bool_t              fApplyElectronVetoConvRecovery; //=true then apply electron veto with conversion recovery
      Bool_t              fApplyConversionId;    //=true then apply conversion id cuts
      Bool_t              fApplyTriggerMatching;   //match to hlt photon (default=0)      
      Double_t            fPhotonR9Min;          //min R9 value
      EPhIdType           fPhIdType;             //!identification scheme
      EPhIsoType          fPhIsoType;            //!isolation scheme
      Bool_t              fFiduciality;          //=true then apply fiducual requirement
      Double_t            fEtaWidthEB;           //max Eta Width in ECAL Barrel
      Double_t            fEtaWidthEE;           //max Eta Width in ECAL End Cap
      Double_t            fAbsEtaMax;            //max Abs Eta
      Bool_t              fApplyR9Min;           //apply R9 min
      Double_t            fEffAreaEcalEE;
      Double_t            fEffAreaHcalEE;
      Double_t            fEffAreaTrackEE;
      Double_t            fEffAreaEcalEB;
      Double_t            fEffAreaHcalEB;
      Double_t            fEffAreaTrackEB;
      const PhotonCol    *fPhotons;              //!photon branch
      const TrackCol     *fTracks;               //!track branch
      const BeamSpotCol  *fBeamspots;            //!beamspot branch    
      const PileupEnergyDensityCol *fPileUpDen;  //!rho branch
      const DecayParticleCol *fConversions;      //!conversion branch
      const ElectronCol  *fElectrons;            //!electron branch


    ClassDef(PhotonIDMod, 1) // Photon identification module
  };
}
#endif
Revision:	1.19
Committed:	Wed Jun 1 18:11:52 2011 UTC (13 years, 11 months ago) by fabstoec
Content type:	text/plain
Branch:	MAIN
CVS Tags:	Mit_023, Mit_022a, Mit_022
Changes since 1.18:	+4 -1 lines
Log Message:	CiC BaseLine Selection
#	User	Rev	Content
1	sixie	1.1	//--------------------------------------------------------------------------------------------------
2	fabstoec	1.19	// $Id: PhotonIDMod.h,v 1.18 2011/05/18 14:01:18 bendavid Exp $
3	sixie	1.1	//
4			// PhotonIDMod
5			//
6			// This module applies photon identification criteria and exports a pointer to a collection
7			// of "good photons" according to the specified identification scheme.
8			//
9			// Authors: S.Xie, C.Loizides
10			//--------------------------------------------------------------------------------------------------
11
12			#ifndef MITPHYSICS_MODS_PHOTONIDMOD_H
13			#define MITPHYSICS_MODS_PHOTONIDMOD_H
14
15			#include "MitAna/TreeMod/interface/BaseMod.h"
16	loizides	1.10	#include "MitAna/DataTree/interface/PhotonFwd.h"
17	fabstoec	1.15	#include "MitAna/DataTree/interface/TrackCol.h"
18			#include "MitAna/DataTree/interface/BeamSpotCol.h"
19			#include "MitAna/DataTree/interface/PileupEnergyDensityCol.h"
20	bendavid	1.16	#include "MitAna/DataTree/interface/DecayParticleCol.h"
21			#include "MitAna/DataTree/interface/ElectronCol.h"
22	sixie	1.1
23			namespace mithep
24			{
25			class PhotonIDMod : public BaseMod
26			{
27			public:
28			PhotonIDMod(const char *name="PhotonIDMod",
29	loizides	1.3	const char *title="Photon identification module");
30	sixie	1.1
31	sixie	1.13	Bool_t GetApplySpikeRemoval() const { return fApplySpikeRemoval; }
32	loizides	1.6	Bool_t GetApplyPixelSeed() const { return fApplyPixelSeed; }
33			const char *GetGoodName() const { return GetGoodPhotonsName(); }
34			const char *GetGoodPhotonsName() const { return fGoodPhotonsName; }
35			Double_t GetHadOverEmMax() const { return fHadOverEmMax; }
36			const char *GetIDType() const { return fPhotonIDType; }
37			const char *GetInputName() const { return fPhotonBranchName; }
38			const char *GetIsoType() const { return fPhotonIsoType; }
39			const char *GetOutputName() const { return GetGoodPhotonsName(); }
40			Double_t GetPtMin() const { return fPhotonPtMin; }
41	ceballos	1.11	Bool_t GetApplyFiduciality() const { return fFiduciality; }
42			Double_t GetEtaWidthEB() const { return fEtaWidthEB; }
43			Double_t GetEtaWidthEE() const { return fEtaWidthEE; }
44	ceballos	1.12	Double_t GetAbsEtaMax() const { return fAbsEtaMax; }
45	sixie	1.13	void SetApplySpikeRemoval(Bool_t b) { fApplySpikeRemoval = b; }
46	loizides	1.6	void SetApplyPixelSeed(Bool_t b) { fApplyPixelSeed = b; }
47	bendavid	1.16	void SetApplyElectronVeto(Bool_t b) { fApplyElectronVeto = b; }
48			void SetApplyElectronVetoConvRecovery(Bool_t b) { fApplyElectronVetoConvRecovery = b; }
49			void SetApplyConversionId(Bool_t b) { fApplyConversionId = b; }
50			void SetApplyTriggerMatching(Bool_t b) { fApplyTriggerMatching = b; }
51	loizides	1.6	void SetGoodName(const char *n) { SetGoodPhotonsName(n); }
52			void SetGoodPhotonsName(const char *n) { fGoodPhotonsName = n; }
53			void SetHadOverEmMax(Double_t hoe) { fHadOverEmMax = hoe; }
54			void SetIDType(const char *type) { fPhotonIDType = type; }
55			void SetInputName(const char *n) { fPhotonBranchName= n; }
56	fabstoec	1.15	void SetTrackName(const char *n) { fTrackBranchName = n; }
57			void SetBeamspotName(const char *n) { fBeamspotBranchName = n; }
58	loizides	1.6	void SetIsoType(const char *type) { fPhotonIsoType = type; }
59			void SetOutputName(const char *n) { SetGoodPhotonsName(n); }
60			void SetPtMin(Double_t pt) { fPhotonPtMin = pt; }
61	ceballos	1.11	void SetR9Min(Double_t x) { fPhotonR9Min = x; }
62			void SetEtaWidthEB(Double_t x) { fEtaWidthEB = x; }
63			void SetEtaWidthEE(Double_t x) { fEtaWidthEE = x; }
64	ceballos	1.12	void SetAbsEtaMax(Double_t x) { fAbsEtaMax = x; }
65	bendavid	1.14	void SetApplyR9Min(Bool_t b) { fApplyR9Min = b; }
66	bendavid	1.18	void SetApplyFiduciality(Bool_t b) { fFiduciality = b; }
67	fabstoec	1.15	void SetEffAreas(Double_t ecal, Double_t hcal, Double_t track) {
68	fabstoec	1.17	fEffAreaEcalEE = ecal; fEffAreaHcalEE = hcal; fEffAreaTrackEE = track;
69			fEffAreaEcalEB = ecal; fEffAreaHcalEB = hcal; fEffAreaTrackEB = track;
70			}
71			void SetEffAreasEEEB(Double_t ecalEE, Double_t hcalEE, Double_t trackEE,
72			Double_t ecalEB, Double_t hcalEB, Double_t trackEB) {
73			fEffAreaEcalEE = ecalEE; fEffAreaHcalEE = hcalEE; fEffAreaTrackEE = trackEE;
74			fEffAreaEcalEB = ecalEB; fEffAreaHcalEB = hcalEB; fEffAreaTrackEB = trackEB;
75			}
76	bendavid	1.16	void SetTriggerObjectsName(const char *n) { fTrigObjectsName = n; }
77	fabstoec	1.15
78	sixie	1.1	enum EPhIdType {
79			kIdUndef = 0, //not defined
80			kTight, //"Tight"
81			kLoose, //"Loose"
82	ceballos	1.2	kLooseEM, //"LooseEM"
83	fabstoec	1.19	kBaseLineId, //"2011" Hgg BaseLine CiC
84	sixie	1.1	kCustomId //"Custom"
85			};
86	fabstoec	1.19
87	sixie	1.1	enum EPhIsoType {
88			kIsoUndef = 0, //not defined
89			kNoIso, //"NoIso"
90	ceballos	1.2	kCombinedIso, //"CombinedIso"
91	fabstoec	1.15	kCustomIso, //"Custom"
92	fabstoec	1.19	kBaseLineIso, //"2011" Hgg BaseLine CiC
93	fabstoec	1.15	kMITPUCorrected //PileUp Corrected Hgg Isolation
94	sixie	1.1	};
95
96			protected:
97	loizides	1.5	void Process();
98			void SlaveBegin();
99
100	loizides	1.6	TString fPhotonBranchName; //name of photon collection (input)
101			TString fGoodPhotonsName; //name of exported "good photon" collection
102	fabstoec	1.15	TString fTrackBranchName; // name of the track collection (only needed for PU corrected isolation)
103			TString fBeamspotBranchName; //name of the Beamspot collection (only needed for PU corrected isolation)
104	bendavid	1.16	TString fPileUpDenName; //name of the PU density collection
105			TString fConversionName; //name of conversion branch
106			TString fElectronName;
107			TString fTrigObjectsName; //name of trigger object collection
108	loizides	1.6	TString fPhotonIDType; //type of photon identification we impose
109			TString fPhotonIsoType; //type of photon isolation we impose
110	loizides	1.4	Double_t fPhotonPtMin; //min pt cut
111	loizides	1.6	Double_t fHadOverEmMax; //maximum of hadronic/em energy
112	sixie	1.13	Bool_t fApplySpikeRemoval; //whether apply spike removal
113	loizides	1.6	Bool_t fApplyPixelSeed; //=true then apply pixel seed constraint
114	bendavid	1.16	Bool_t fApplyElectronVeto; //=true then apply electron veto (with no conversion recovery)
115			Bool_t fApplyElectronVetoConvRecovery; //=true then apply electron veto with conversion recovery
116			Bool_t fApplyConversionId; //=true then apply conversion id cuts
117			Bool_t fApplyTriggerMatching; //match to hlt photon (default=0)
118	ceballos	1.8	Double_t fPhotonR9Min; //min R9 value
119	loizides	1.4	EPhIdType fPhIdType; //!identification scheme
120			EPhIsoType fPhIsoType; //!isolation scheme
121	ceballos	1.11	Bool_t fFiduciality; //=true then apply fiducual requirement
122			Double_t fEtaWidthEB; //max Eta Width in ECAL Barrel
123			Double_t fEtaWidthEE; //max Eta Width in ECAL End Cap
124	ceballos	1.12	Double_t fAbsEtaMax; //max Abs Eta
125	bendavid	1.14	Bool_t fApplyR9Min; //apply R9 min
126	fabstoec	1.17	Double_t fEffAreaEcalEE;
127			Double_t fEffAreaHcalEE;
128			Double_t fEffAreaTrackEE;
129			Double_t fEffAreaEcalEB;
130			Double_t fEffAreaHcalEB;
131			Double_t fEffAreaTrackEB;
132	loizides	1.10	const PhotonCol *fPhotons; //!photon branch
133	fabstoec	1.15	const TrackCol *fTracks; //!track branch
134			const BeamSpotCol *fBeamspots; //!beamspot branch
135	bendavid	1.16	const PileupEnergyDensityCol *fPileUpDen; //!rho branch
136			const DecayParticleCol *fConversions; //!conversion branch
137			const ElectronCol *fElectrons; //!electron branch
138	fabstoec	1.15
139
140	loizides	1.6	ClassDef(PhotonIDMod, 1) // Photon identification module
141	sixie	1.1	};
142			}
143			#endif