Mods/interface/PhotonIDMod.h

//--------------------------------------------------------------------------------------------------
// $Id: PhotonIDMod.h,v 1.15 2011/04/06 18:03:48 fabstoec 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                SetEffAreas(Double_t ecal, Double_t hcal, Double_t track) { 
        fEffAreaEcal = ecal; fEffAreaHcal = hcal; fEffAreaTrack = track;}
      void                SetTriggerObjectsName(const char *n)   { fTrigObjectsName = n;       }
        
    
      enum EPhIdType {
        kIdUndef = 0,       //not defined
        kTight,             //"Tight"
        kLoose,             //"Loose"
        kLooseEM,           //"LooseEM"
        kCustomId           //"Custom"
      };
      enum EPhIsoType {
        kIsoUndef = 0,      //not defined        
        kNoIso,             //"NoIso"
        kCombinedIso,       //"CombinedIso"
        kCustomIso,         //"Custom"
        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            fEffAreaEcal;
      Double_t            fEffAreaHcal;
      Double_t            fEffAreaTrack;
      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.16
Committed:	Tue Apr 12 22:14:21 2011 UTC (14 years ago) by bendavid
Content type:	text/plain
Branch:	MAIN
Changes since 1.15:	+20 -3 lines
Log Message:	Add PhotonTools with conversion id, trigger matching and diphoton categories, add functionality to PhotonIdMod
#	User	Rev	Content
1	sixie	1.1	//--------------------------------------------------------------------------------------------------
2	bendavid	1.16	// $Id: PhotonIDMod.h,v 1.15 2011/04/06 18:03:48 fabstoec 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	fabstoec	1.15	void SetEffAreas(Double_t ecal, Double_t hcal, Double_t track) {
67			fEffAreaEcal = ecal; fEffAreaHcal = hcal; fEffAreaTrack = track;}
68	bendavid	1.16	void SetTriggerObjectsName(const char *n) { fTrigObjectsName = n; }
69
70	fabstoec	1.15
71
72	sixie	1.1	enum EPhIdType {
73			kIdUndef = 0, //not defined
74			kTight, //"Tight"
75			kLoose, //"Loose"
76	ceballos	1.2	kLooseEM, //"LooseEM"
77	sixie	1.1	kCustomId //"Custom"
78			};
79			enum EPhIsoType {
80			kIsoUndef = 0, //not defined
81			kNoIso, //"NoIso"
82	ceballos	1.2	kCombinedIso, //"CombinedIso"
83	fabstoec	1.15	kCustomIso, //"Custom"
84			kMITPUCorrected //PileUp Corrected Hgg Isolation
85	sixie	1.1	};
86
87			protected:
88	loizides	1.5	void Process();
89			void SlaveBegin();
90
91	loizides	1.6	TString fPhotonBranchName; //name of photon collection (input)
92			TString fGoodPhotonsName; //name of exported "good photon" collection
93	fabstoec	1.15	TString fTrackBranchName; // name of the track collection (only needed for PU corrected isolation)
94			TString fBeamspotBranchName; //name of the Beamspot collection (only needed for PU corrected isolation)
95	bendavid	1.16	TString fPileUpDenName; //name of the PU density collection
96			TString fConversionName; //name of conversion branch
97			TString fElectronName;
98			TString fTrigObjectsName; //name of trigger object collection
99	loizides	1.6	TString fPhotonIDType; //type of photon identification we impose
100			TString fPhotonIsoType; //type of photon isolation we impose
101	loizides	1.4	Double_t fPhotonPtMin; //min pt cut
102	loizides	1.6	Double_t fHadOverEmMax; //maximum of hadronic/em energy
103	sixie	1.13	Bool_t fApplySpikeRemoval; //whether apply spike removal
104	loizides	1.6	Bool_t fApplyPixelSeed; //=true then apply pixel seed constraint
105	bendavid	1.16	Bool_t fApplyElectronVeto; //=true then apply electron veto (with no conversion recovery)
106			Bool_t fApplyElectronVetoConvRecovery; //=true then apply electron veto with conversion recovery
107			Bool_t fApplyConversionId; //=true then apply conversion id cuts
108			Bool_t fApplyTriggerMatching; //match to hlt photon (default=0)
109	ceballos	1.8	Double_t fPhotonR9Min; //min R9 value
110	loizides	1.4	EPhIdType fPhIdType; //!identification scheme
111			EPhIsoType fPhIsoType; //!isolation scheme
112	ceballos	1.11	Bool_t fFiduciality; //=true then apply fiducual requirement
113			Double_t fEtaWidthEB; //max Eta Width in ECAL Barrel
114			Double_t fEtaWidthEE; //max Eta Width in ECAL End Cap
115	ceballos	1.12	Double_t fAbsEtaMax; //max Abs Eta
116	bendavid	1.14	Bool_t fApplyR9Min; //apply R9 min
117	fabstoec	1.15	Double_t fEffAreaEcal;
118			Double_t fEffAreaHcal;
119			Double_t fEffAreaTrack;
120	loizides	1.10	const PhotonCol *fPhotons; //!photon branch
121	fabstoec	1.15	const TrackCol *fTracks; //!track branch
122			const BeamSpotCol *fBeamspots; //!beamspot branch
123	bendavid	1.16	const PileupEnergyDensityCol *fPileUpDen; //!rho branch
124			const DecayParticleCol *fConversions; //!conversion branch
125			const ElectronCol *fElectrons; //!electron branch
126	fabstoec	1.15
127
128	loizides	1.6	ClassDef(PhotonIDMod, 1) // Photon identification module
129	sixie	1.1	};
130			}
131			#endif