Mods/interface/ElectronIDMod.h

//--------------------------------------------------------------------------------------------------
// $Id: ElectronIDMod.h,v 1.29 2010/04/07 06:57:47 sixie Exp $
//
// ElectronIDMod
//
// This module applies electron identification criteria and exports a pointer to a collection
// of "good electrons" according to the specified identification scheme.
//
// See http://indico.cern.ch/contributionDisplay.py?contribId=1&confId=42251
//
// Authors: S.Xie, C.Loizides
//--------------------------------------------------------------------------------------------------

#ifndef MITPHYSICS_MODS_ELECTRONIDMOD_H
#define MITPHYSICS_MODS_ELECTRONIDMOD_H

#include "MitAna/TreeMod/interface/BaseMod.h" 
#include "MitAna/DataTree/interface/ElectronFwd.h"
#include "MitAna/DataTree/interface/VertexFwd.h"
#include "MitAna/DataTree/interface/DecayParticleFwd.h"

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

     enum EElIdType {
        kIdUndef = 0,       //not defined
        kTight,             //"Tight"
        kLoose,             //"Loose"
        kLikelihood,        //"Likelihood"
        kNoId,              //"NoId"
        kZeeId,             //"ZeeId"
        kCustomIdLoose,     //"CustomLoose"
        kCustomIdTight,     //"CustomTight"
        kVBTFWorkingPoint95Id,
        kVBTFWorkingPoint90Id,
        kVBTFWorkingPoint80Id,
        kVBTFWorkingPoint70Id
      };

      enum EElIsoType {
        kIsoUndef = 0,      //not defined
        kTrackCalo,         //"TrackCalo"
        kTrackJura,         //"TrackJura"
        kTrackJuraCombined, //"TrackJuraCombined"
        kTrackJuraSliding,  //"TrackJuraSliding"
        kNoIso,             //"NoIso"
        kZeeIso,            //"ZeeIso"
        kCustomIso,          //"Custom"
        kVBTFWorkingPoint95Iso,
        kVBTFWorkingPoint90Iso,
        kVBTFWorkingPoint80Iso,
        kVBTFWorkingPoint70Iso
      };

      Bool_t              GetApplyConversionFilter()  const { return fApplyConvFilter;        }
      Bool_t              GetApplyD0Cut()             const { return fApplyD0Cut;             }
      Double_t            GetCaloIsoCut()             const { return fCaloIsolationCut;       }
      Double_t            GetEcalJurIsoCut()          const { return fEcalJuraIsoCut;         }
      Double_t            GetCombIsoCut()             const { return fCombIsolationCut;       }
      const char         *GetGoodName()               const { return GetGoodElectronsName();  }   
      const char         *GetGoodElectronsName()      const { return fGoodElectronsName;      }   
      Double_t            GetHcalIsoCut()             const { return fHcalIsolationCut;       }
      Double_t            GetIDLikelihoodCut()        const { return fIDLikelihoodCut;        }
      const char         *GetIDType()                 const { return fElectronIDType;         }
      const char         *GetInputName()              const { return fElectronBranchName;     }   
      const char         *GetIsoType()                const { return fElectronIsoType;        }
      const char         *GetOutputName()             const { return GetGoodElectronsName();  }
      Double_t            GetPtMin()                  const { return fElectronPtMin;          }
      Bool_t              GetReverseD0Cut()           const { return fReverseD0Cut;           }
      Bool_t              GetReverseIsoCut()          const { return fReverseIsoCut;          }
      Double_t            GetTrackIsoCut()            const { return fTrackIsolationCut;      }
      Bool_t              GetChargeFilter()           const { return fChargeFilter;           }
      Bool_t              PassChargeFilter(const Electron *el) const;
      Bool_t              PassConversionFilter(const Electron *el, const DecayParticleCol *conversions) const;
      Bool_t              PassD0Cut(const Electron *el, const VertexCol *vertices) const;
      Bool_t              PassIDCut(const Electron *el, EElIdType idType) const;
      Bool_t              PassIsolationCut(const Electron *el, EElIsoType isoType) const;
      Bool_t              PassSpikeRemovalFilter(const Electron *ele) const;
      void                SetApplyConversionFilter(Bool_t b)    { fApplyConvFilter    = b;    }
      void                SetApplyD0Cut(Bool_t b)               { fApplyD0Cut         = b;    }
      void                SetCaloIsoCut(Double_t cut)           { fCaloIsolationCut   = cut;  }
      void                SetCombIsoCut(Double_t cut)           { fCombIsolationCut  = cut;   }
      void                SetD0Cut(Double_t cut)                { fD0Cut = cut;               }
      void                SetEcalJurIsoCut(Double_t cut)        { fEcalJuraIsoCut     = cut;  }
      void                SetGoodElectronsName(const char *n)   { fGoodElectronsName  = n;    }   
      void                SetGoodName(const char *n)            { SetGoodElectronsName(n);    }   
      void                SetHcalIsoCut(Double_t cut)           { fHcalIsolationCut   = cut;  }
      void                SetIDLikelihoodCut(Double_t cut)      { fIDLikelihoodCut    = cut;  }
      void                SetIDType(const char *type)           { fElectronIDType     = type; }
      void                SetInputName(const char *n)           { fElectronBranchName = n;    }   
      void                SetIsoType(const char *type)          { fElectronIsoType    = type; }
      void                SetOutputName(const char *n)          { SetGoodElectronsName(n);    }   
      void                SetPtMin(Double_t pt)                 { fElectronPtMin      = pt;   }
      void                SetReverseD0Cut(Bool_t b)             { fReverseD0Cut = b;          }
      void                SetReverseIsoCut(Bool_t b)            { fReverseIsoCut = b;         }
      void                SetTrackIsoCut(Double_t cut)          { fTrackIsolationCut  = cut;  }
      void                SetChargeFilter(Bool_t b)             { fChargeFilter = b;          }
      void                SetWrongHitsRequirement(Bool_t b)     { fWrongHitsRequirement = b;  }
      void                Setup();

    protected:
      Bool_t              PassCustomID(const Electron *el) const;
      Bool_t              PassCustomIso(const Electron *el) const;
      void                Process();
      void                SetCustomIDCuts(EElIdType idt);
      void                SetCustomIsoCuts(EElIsoType idt);
      void                SlaveBegin();


      TString                 fElectronBranchName;     //name of electron collection (input)
      TString                 fConversionBranchName;   //name of electron collection (input)
      TString                 fGoodElectronsName;      //name of exported "good electrons" col
      TString                 fVertexName;             //name of vertex collection
      TString                 fElectronIDType;         //type of electron ID we impose
      TString                 fElectronIsoType;        //type of electron Isolation we impose
      Double_t                fElectronPtMin;          //min pt cut
      Double_t                fIDLikelihoodCut;        //cut value for ID likelihood
      Double_t                fTrackIsolationCut;      //cut value for track isolation
      Double_t                fCaloIsolationCut;       //cut value for calo isolation
      Double_t                fEcalJuraIsoCut;         //cut value for ecal jurassic isolation
      Double_t                fHcalIsolationCut;       //cut value for hcal isolation
      Double_t                fCombIsolationCut;       //cut value for combined isolation
      Bool_t                  fApplyConvFilter;        //whether remove conversions
      Bool_t                  fWrongHitsRequirement;   //whether to use wrong hits req for conversion removal
      Bool_t                  fApplyD0Cut;             //whether apply d0 cut
      Bool_t                  fChargeFilter;           //whether apply GSF and CFT equal requirement
      Double_t                fD0Cut;                  //max d0
      Bool_t                  fReverseIsoCut;          //apply reversion iso cut (default=0)
      Bool_t                  fReverseD0Cut;           //apply reversion d0 cut (default=0)
      EElIdType               fElIdType;               //!identification scheme
      EElIsoType              fElIsoType;              //!isolation scheme
      const ElectronCol      *fElectrons;              //!electron collection
      const DecayParticleCol *fConversions;            //!conversion collection
      const VertexCol        *fVertices;               //!vertices branches
      Double_t                fCuts[6][8];             //!custom id cuts
      Double_t                fIsoCuts[4][2];          //!custom isolation cuts

    ClassDef(ElectronIDMod, 1) // Electron identification module
  };
}
#endif
Revision:	1.30
Committed:	Sat Apr 10 16:29:51 2010 UTC (15 years ago) by sixie
Content type:	text/plain
Branch:	MAIN
Changes since 1.29:	+3 -1 lines
Log Message:	Add 95% working point cut
#	User	Rev	Content
1	loizides	1.1	//--------------------------------------------------------------------------------------------------
2	sixie	1.30	// $Id: ElectronIDMod.h,v 1.29 2010/04/07 06:57:47 sixie Exp $
3	loizides	1.1	//
4			// ElectronIDMod
5			//
6	loizides	1.4	// This module applies electron identification criteria and exports a pointer to a collection
7	loizides	1.5	// of "good electrons" according to the specified identification scheme.
8	loizides	1.1	//
9	loizides	1.13	// See http://indico.cern.ch/contributionDisplay.py?contribId=1&confId=42251
10			//
11	loizides	1.4	// Authors: S.Xie, C.Loizides
12	loizides	1.1	//--------------------------------------------------------------------------------------------------
13
14	loizides	1.4	#ifndef MITPHYSICS_MODS_ELECTRONIDMOD_H
15			#define MITPHYSICS_MODS_ELECTRONIDMOD_H
16	loizides	1.1
17			#include "MitAna/TreeMod/interface/BaseMod.h"
18	loizides	1.20	#include "MitAna/DataTree/interface/ElectronFwd.h"
19			#include "MitAna/DataTree/interface/VertexFwd.h"
20			#include "MitAna/DataTree/interface/DecayParticleFwd.h"
21	loizides	1.1
22			namespace mithep
23			{
24			class ElectronIDMod : public BaseMod
25			{
26			public:
27			ElectronIDMod(const char *name="ElectronIDMod",
28	loizides	1.4	const char *title="Electron identification module");
29
30	sixie	1.26	enum EElIdType {
31			kIdUndef = 0, //not defined
32			kTight, //"Tight"
33			kLoose, //"Loose"
34			kLikelihood, //"Likelihood"
35			kNoId, //"NoId"
36			kZeeId, //"ZeeId"
37			kCustomIdLoose, //"CustomLoose"
38	sixie	1.29	kCustomIdTight, //"CustomTight"
39	sixie	1.30	kVBTFWorkingPoint95Id,
40	sixie	1.29	kVBTFWorkingPoint90Id,
41			kVBTFWorkingPoint80Id,
42			kVBTFWorkingPoint70Id
43	sixie	1.26	};
44
45			enum EElIsoType {
46			kIsoUndef = 0, //not defined
47			kTrackCalo, //"TrackCalo"
48			kTrackJura, //"TrackJura"
49	ceballos	1.28	kTrackJuraCombined, //"TrackJuraCombined"
50	sixie	1.26	kTrackJuraSliding, //"TrackJuraSliding"
51			kNoIso, //"NoIso"
52			kZeeIso, //"ZeeIso"
53	sixie	1.29	kCustomIso, //"Custom"
54	sixie	1.30	kVBTFWorkingPoint95Iso,
55	sixie	1.29	kVBTFWorkingPoint90Iso,
56			kVBTFWorkingPoint80Iso,
57			kVBTFWorkingPoint70Iso
58	sixie	1.26	};
59
60	loizides	1.13	Bool_t GetApplyConversionFilter() const { return fApplyConvFilter; }
61	ceballos	1.14	Bool_t GetApplyD0Cut() const { return fApplyD0Cut; }
62	loizides	1.8	Double_t GetCaloIsoCut() const { return fCaloIsolationCut; }
63			Double_t GetEcalJurIsoCut() const { return fEcalJuraIsoCut; }
64	ceballos	1.28	Double_t GetCombIsoCut() const { return fCombIsolationCut; }
65	loizides	1.8	const char *GetGoodName() const { return GetGoodElectronsName(); }
66			const char *GetGoodElectronsName() const { return fGoodElectronsName; }
67			Double_t GetHcalIsoCut() const { return fHcalIsolationCut; }
68			Double_t GetIDLikelihoodCut() const { return fIDLikelihoodCut; }
69			const char *GetIDType() const { return fElectronIDType; }
70			const char *GetInputName() const { return fElectronBranchName; }
71			const char *GetIsoType() const { return fElectronIsoType; }
72			const char *GetOutputName() const { return GetGoodElectronsName(); }
73			Double_t GetPtMin() const { return fElectronPtMin; }
74	loizides	1.22	Bool_t GetReverseD0Cut() const { return fReverseD0Cut; }
75			Bool_t GetReverseIsoCut() const { return fReverseIsoCut; }
76	loizides	1.8	Double_t GetTrackIsoCut() const { return fTrackIsolationCut; }
77	ceballos	1.23	Bool_t GetChargeFilter() const { return fChargeFilter; }
78	sixie	1.26	Bool_t PassChargeFilter(const Electron *el) const;
79			Bool_t PassConversionFilter(const Electron el, const DecayParticleCol conversions) const;
80			Bool_t PassD0Cut(const Electron el, const VertexCol vertices) const;
81			Bool_t PassIDCut(const Electron *el, EElIdType idType) const;
82			Bool_t PassIsolationCut(const Electron *el, EElIsoType isoType) const;
83	sixie	1.29	Bool_t PassSpikeRemovalFilter(const Electron *ele) const;
84	loizides	1.13	void SetApplyConversionFilter(Bool_t b) { fApplyConvFilter = b; }
85	ceballos	1.14	void SetApplyD0Cut(Bool_t b) { fApplyD0Cut = b; }
86	loizides	1.8	void SetCaloIsoCut(Double_t cut) { fCaloIsolationCut = cut; }
87	ceballos	1.28	void SetCombIsoCut(Double_t cut) { fCombIsolationCut = cut; }
88	loizides	1.22	void SetD0Cut(Double_t cut) { fD0Cut = cut; }
89	loizides	1.8	void SetEcalJurIsoCut(Double_t cut) { fEcalJuraIsoCut = cut; }
90	loizides	1.22	void SetGoodElectronsName(const char *n) { fGoodElectronsName = n; }
91	loizides	1.8	void SetGoodName(const char *n) { SetGoodElectronsName(n); }
92			void SetHcalIsoCut(Double_t cut) { fHcalIsolationCut = cut; }
93			void SetIDLikelihoodCut(Double_t cut) { fIDLikelihoodCut = cut; }
94			void SetIDType(const char *type) { fElectronIDType = type; }
95			void SetInputName(const char *n) { fElectronBranchName = n; }
96			void SetIsoType(const char *type) { fElectronIsoType = type; }
97			void SetOutputName(const char *n) { SetGoodElectronsName(n); }
98			void SetPtMin(Double_t pt) { fElectronPtMin = pt; }
99	loizides	1.22	void SetReverseD0Cut(Bool_t b) { fReverseD0Cut = b; }
100			void SetReverseIsoCut(Bool_t b) { fReverseIsoCut = b; }
101	loizides	1.8	void SetTrackIsoCut(Double_t cut) { fTrackIsolationCut = cut; }
102	ceballos	1.23	void SetChargeFilter(Bool_t b) { fChargeFilter = b; }
103	bendavid	1.24	void SetWrongHitsRequirement(Bool_t b) { fWrongHitsRequirement = b; }
104	ceballos	1.27	void Setup();
105	loizides	1.4
106	loizides	1.1	protected:
107	sixie	1.26	Bool_t PassCustomID(const Electron *el) const;
108	sixie	1.29	Bool_t PassCustomIso(const Electron *el) const;
109	sixie	1.26	void Process();
110			void SetCustomIDCuts(EElIdType idt);
111	sixie	1.29	void SetCustomIsoCuts(EElIsoType idt);
112	sixie	1.26	void SlaveBegin();
113	loizides	1.22
114	loizides	1.7
115	ceballos	1.12	TString fElectronBranchName; //name of electron collection (input)
116			TString fConversionBranchName; //name of electron collection (input)
117	loizides	1.16	TString fGoodElectronsName; //name of exported "good electrons" col
118	ceballos	1.12	TString fVertexName; //name of vertex collection
119			TString fElectronIDType; //type of electron ID we impose
120			TString fElectronIsoType; //type of electron Isolation we impose
121			Double_t fElectronPtMin; //min pt cut
122			Double_t fIDLikelihoodCut; //cut value for ID likelihood
123			Double_t fTrackIsolationCut; //cut value for track isolation
124			Double_t fCaloIsolationCut; //cut value for calo isolation
125			Double_t fEcalJuraIsoCut; //cut value for ecal jurassic isolation
126			Double_t fHcalIsolationCut; //cut value for hcal isolation
127	ceballos	1.28	Double_t fCombIsolationCut; //cut value for combined isolation
128	loizides	1.13	Bool_t fApplyConvFilter; //whether remove conversions
129	bendavid	1.24	Bool_t fWrongHitsRequirement; //whether to use wrong hits req for conversion removal
130	loizides	1.16	Bool_t fApplyD0Cut; //whether apply d0 cut
131	loizides	1.25	Bool_t fChargeFilter; //whether apply GSF and CFT equal requirement
132	loizides	1.13	Double_t fD0Cut; //max d0
133	loizides	1.19	Bool_t fReverseIsoCut; //apply reversion iso cut (default=0)
134			Bool_t fReverseD0Cut; //apply reversion d0 cut (default=0)
135	loizides	1.13	EElIdType fElIdType; //!identification scheme
136			EElIsoType fElIsoType; //!isolation scheme
137	ceballos	1.12	const ElectronCol *fElectrons; //!electron collection
138			const DecayParticleCol *fConversions; //!conversion collection
139	loizides	1.16	const VertexCol *fVertices; //!vertices branches
140	loizides	1.22	Double_t fCuts[6][8]; //!custom id cuts
141	sixie	1.29	Double_t fIsoCuts[4][2]; //!custom isolation cuts
142	ceballos	1.17
143	loizides	1.8	ClassDef(ElectronIDMod, 1) // Electron identification module
144	loizides	1.1	};
145			}
146			#endif