Utils/interface/ElectronTools.h

//--------------------------------------------------------------------------------------------------
// $Id $
//
// ElectronTools
//
// Helper Class for electron Identification decisions.
//
// Authors: S.Xie
//--------------------------------------------------------------------------------------------------

#ifndef MITPHYSICS_UTILS_ELECTRONTOOLS_H
#define MITPHYSICS_UTILS_ELECTRONTOOLS_H

#include "MitAna/DataTree/interface/Electron.h"
#include "MitAna/DataTree/interface/DecayParticleCol.h"
#include "MitAna/DataTree/interface/VertexCol.h"
#include "MitAna/DataTree/interface/BeamSpotCol.h"
#include "MitAna/DataTree/interface/TriggerObjectCol.h"
#include "MitCommon/MathTools/interface/MathUtils.h"
#include "MitPhysics/ElectronLikelihood/interface/ElectronLikelihood.h"

namespace mithep {
  class ElectronTools {
    public:
      ElectronTools();
  
     enum EElIdType {
        kIdUndef = 0,       //not defined
        kTight,             //"Tight"
        kLoose,             //"Loose"
        kLikelihood,        //"Likelihood"
        kNoId,              //"NoId"
        kZeeId,             //"ZeeId"
        kCustomIdLoose,     //"CustomLoose"
        kCustomIdTight,     //"CustomTight"
        kVBTFWorkingPointFakeableId,
        kVBTFWorkingPoint95Id,
        kVBTFWorkingPoint90Id,
        kVBTFWorkingPoint85Id,
        kVBTFWorkingPoint80Id,
        kVBTFWorkingPointLowPtId,
        kVBTFWorkingPoint70Id,
        kMVAID_BDTG_NoIPInfo,
        kMVAID_BDTG_WithIPInfo,
        kMVAID_BDTG_IDIsoCombined
      };

      enum EElIsoType {
        kIsoUndef = 0,                 //"not defined"
        kTrackCalo,                    //"TrackCalo"
        kTrackJura,                    //"TrackJura"
        kTrackJuraCombined,            //"TrackJuraCombined"
        kTrackJuraSliding,             //"TrackJuraSliding"
        kTrackJuraSlidingNoCorrection, //"TrackJuraSlidingNoCorrection"
        kCombinedRelativeConeAreaCorrected, //"CombinedRelativeConeAreaCorrected"
        kNoIso,                        //"NoIso"
        kPFIso,                        //"PFIso"
        kPFIsoNoL,                     //"PFIsoNoL"
        kZeeIso,                       //"ZeeIso"
        kCustomIso,                    //"Custom"
        kVBTFWorkingPoint95Iso,
        kVBTFWorkingPoint90Iso,
        kVBTFWorkingPoint85Iso,
        kVBTFWorkingPoint80Iso,
        kVBTFWorkingPoint70Iso,
        kMVAIso_BDTG_IDIsoCombined
      };

      enum EElectronEffectiveAreaType {
        kEleChargedIso03, 
        kEleNeutralHadronIso03, 
        kEleGammaIso03, 
        kEleGammaIsoVetoEtaStrip03, 
        kEleChargedIso04, 
        kEleNeutralHadronIso04, 
        kEleGammaIso04, 
        kEleGammaIsoVetoEtaStrip04, 
        kEleNeutralHadronIso007, 
        kEleHoverE, 
        kEleHcalDepth1OverEcal, 
        kEleHcalDepth2OverEcal,
        kEleGammaIsoDR0p0To0p1,
        kEleGammaIsoDR0p1To0p2,
        kEleGammaIsoDR0p2To0p3,
        kEleGammaIsoDR0p3To0p4,
        kEleGammaIsoDR0p4To0p5,
        kEleNeutralHadronIsoDR0p0To0p1,
        kEleNeutralHadronIsoDR0p1To0p2,
        kEleNeutralHadronIsoDR0p2To0p3,
        kEleNeutralHadronIsoDR0p3To0p4,
        kEleNeutralHadronIsoDR0p4To0p5
      };
      
      enum EElectronEffectiveAreaTarget {
        kEleEANoCorr,
        kEleEAData2011,
        kEleEASummer11MC,
        kEleEAFall11MC
      };

      static Bool_t       PassChargeFilter(const Electron *el);
      static Bool_t       PassConversionFilter(const Electron *el, const DecayParticleCol *conversions, 
                                               const BaseVertex *vtx, UInt_t nWrongHitsMax=0, Double_t probMin=1e-6,
                                               Double_t lxyMin = 2.0, Bool_t matchCkf = kTRUE, Bool_t requireArbitratedMerged = kFALSE, Double_t trkptMin = -99.);
      static Bool_t       PassCustomID(const Electron *el, EElIdType idType);
      static Bool_t       PassCustomIso(const Electron *el, EElIsoType isoType,
                                        Bool_t useCombineIso = kTRUE);
      static Bool_t       PassD0Cut(const Electron *el, const VertexCol *vertices, Double_t fD0Cut, Int_t nVertex = 0);
      static Bool_t       PassD0Cut(const Electron *el, const BeamSpotCol *beamspots, Double_t fD0Cut);
      static Bool_t       PassDZCut(const Electron *el, const VertexCol *vertices, Double_t fDZCut, Int_t nVertex = 0);
      static Bool_t       PassSpikeRemovalFilter(const Electron *ele);
      static Bool_t       PassTriggerMatching(const Electron *ele, const TriggerObjectCol *trigobjs);
      static Int_t        Classify(const Electron *ele);
      static Int_t        PassTightId(const Electron *ele, const VertexCol *vertices, 
                                      const DecayParticleCol *conversions, const Int_t typeCuts,
                                      Double_t beta = 1.0);
      static bool         compute_cut(double x, double et, double cut_min, double cut_max, bool gtn=false);
      static Double_t     Likelihood(ElectronLikelihood *LH, const Electron *ele);
      static Double_t     ElectronEffectiveArea(EElectronEffectiveAreaType type, Double_t Eta, 
                                                EElectronEffectiveAreaTarget EffectiveAreaTarget = kEleEAData2011);

    ClassDef(ElectronTools, 0) // Muon tools
  };
}

#endif
Revision:	1.24
Committed:	Sun Apr 15 12:06:26 2012 UTC (13 years ago) by sixie
Content type:	text/plain
Branch:	MAIN
Changes since 1.23:	+20 -2 lines
Log Message:	add effective areas
#	User	Rev	Content
1	sixie	1.1	//--------------------------------------------------------------------------------------------------
2			// $Id $
3			//
4			// ElectronTools
5			//
6			// Helper Class for electron Identification decisions.
7			//
8			// Authors: S.Xie
9			//--------------------------------------------------------------------------------------------------
10
11			#ifndef MITPHYSICS_UTILS_ELECTRONTOOLS_H
12			#define MITPHYSICS_UTILS_ELECTRONTOOLS_H
13
14			#include "MitAna/DataTree/interface/Electron.h"
15			#include "MitAna/DataTree/interface/DecayParticleCol.h"
16			#include "MitAna/DataTree/interface/VertexCol.h"
17	sixie	1.2	#include "MitAna/DataTree/interface/BeamSpotCol.h"
18	bendavid	1.5	#include "MitAna/DataTree/interface/TriggerObjectCol.h"
19	sixie	1.1	#include "MitCommon/MathTools/interface/MathUtils.h"
20	sixie	1.20	#include "MitPhysics/ElectronLikelihood/interface/ElectronLikelihood.h"
21	sixie	1.1
22			namespace mithep {
23			class ElectronTools {
24			public:
25			ElectronTools();
26
27			enum EElIdType {
28			kIdUndef = 0, //not defined
29			kTight, //"Tight"
30			kLoose, //"Loose"
31			kLikelihood, //"Likelihood"
32			kNoId, //"NoId"
33			kZeeId, //"ZeeId"
34			kCustomIdLoose, //"CustomLoose"
35			kCustomIdTight, //"CustomTight"
36	ceballos	1.18	kVBTFWorkingPointFakeableId,
37	sixie	1.1	kVBTFWorkingPoint95Id,
38			kVBTFWorkingPoint90Id,
39	ceballos	1.4	kVBTFWorkingPoint85Id,
40	sixie	1.1	kVBTFWorkingPoint80Id,
41	ceballos	1.16	kVBTFWorkingPointLowPtId,
42	sixie	1.20	kVBTFWorkingPoint70Id,
43	sixie	1.23	kMVAID_BDTG_NoIPInfo,
44			kMVAID_BDTG_WithIPInfo,
45			kMVAID_BDTG_IDIsoCombined
46	sixie	1.1	};
47
48			enum EElIsoType {
49	ceballos	1.12	kIsoUndef = 0, //"not defined"
50			kTrackCalo, //"TrackCalo"
51			kTrackJura, //"TrackJura"
52			kTrackJuraCombined, //"TrackJuraCombined"
53			kTrackJuraSliding, //"TrackJuraSliding"
54			kTrackJuraSlidingNoCorrection, //"TrackJuraSlidingNoCorrection"
55	sixie	1.19	kCombinedRelativeConeAreaCorrected, //"CombinedRelativeConeAreaCorrected"
56	ceballos	1.12	kNoIso, //"NoIso"
57			kPFIso, //"PFIso"
58			kPFIsoNoL, //"PFIsoNoL"
59			kZeeIso, //"ZeeIso"
60			kCustomIso, //"Custom"
61	sixie	1.1	kVBTFWorkingPoint95Iso,
62			kVBTFWorkingPoint90Iso,
63	ceballos	1.4	kVBTFWorkingPoint85Iso,
64	sixie	1.1	kVBTFWorkingPoint80Iso,
65	sixie	1.23	kVBTFWorkingPoint70Iso,
66			kMVAIso_BDTG_IDIsoCombined
67	sixie	1.1	};
68
69	sixie	1.23	enum EElectronEffectiveAreaType {
70			kEleChargedIso03,
71			kEleNeutralHadronIso03,
72			kEleGammaIso03,
73			kEleGammaIsoVetoEtaStrip03,
74			kEleChargedIso04,
75			kEleNeutralHadronIso04,
76			kEleGammaIso04,
77			kEleGammaIsoVetoEtaStrip04,
78			kEleNeutralHadronIso007,
79			kEleHoverE,
80			kEleHcalDepth1OverEcal,
81	sixie	1.24	kEleHcalDepth2OverEcal,
82			kEleGammaIsoDR0p0To0p1,
83			kEleGammaIsoDR0p1To0p2,
84			kEleGammaIsoDR0p2To0p3,
85			kEleGammaIsoDR0p3To0p4,
86			kEleGammaIsoDR0p4To0p5,
87			kEleNeutralHadronIsoDR0p0To0p1,
88			kEleNeutralHadronIsoDR0p1To0p2,
89			kEleNeutralHadronIsoDR0p2To0p3,
90			kEleNeutralHadronIsoDR0p3To0p4,
91			kEleNeutralHadronIsoDR0p4To0p5
92	sixie	1.23	};
93
94	sixie	1.24	enum EElectronEffectiveAreaTarget {
95			kEleEANoCorr,
96			kEleEAData2011,
97			kEleEASummer11MC,
98			kEleEAFall11MC
99			};
100
101	sixie	1.1	static Bool_t PassChargeFilter(const Electron *el);
102			static Bool_t PassConversionFilter(const Electron el, const DecayParticleCol conversions,
103	bendavid	1.15	const BaseVertex *vtx, UInt_t nWrongHitsMax=0, Double_t probMin=1e-6,
104	bendavid	1.17	Double_t lxyMin = 2.0, Bool_t matchCkf = kTRUE, Bool_t requireArbitratedMerged = kFALSE, Double_t trkptMin = -99.);
105	sixie	1.1	static Bool_t PassCustomID(const Electron *el, EElIdType idType);
106	ceballos	1.4	static Bool_t PassCustomIso(const Electron *el, EElIsoType isoType,
107			Bool_t useCombineIso = kTRUE);
108	ceballos	1.14	static Bool_t PassD0Cut(const Electron el, const VertexCol vertices, Double_t fD0Cut, Int_t nVertex = 0);
109	ceballos	1.7	static Bool_t PassD0Cut(const Electron el, const BeamSpotCol beamspots, Double_t fD0Cut);
110	ceballos	1.16	static Bool_t PassDZCut(const Electron el, const VertexCol vertices, Double_t fDZCut, Int_t nVertex = 0);
111	sixie	1.1	static Bool_t PassSpikeRemovalFilter(const Electron *ele);
112	bendavid	1.5	static Bool_t PassTriggerMatching(const Electron ele, const TriggerObjectCol trigobjs);
113	ceballos	1.3	static Int_t Classify(const Electron *ele);
114			static Int_t PassTightId(const Electron ele, const VertexCol vertices,
115	ceballos	1.10	const DecayParticleCol *conversions, const Int_t typeCuts,
116			Double_t beta = 1.0);
117	ceballos	1.6	static bool compute_cut(double x, double et, double cut_min, double cut_max, bool gtn=false);
118	sixie	1.20	static Double_t Likelihood(ElectronLikelihood LH, const Electron ele);
119	sixie	1.24	static Double_t ElectronEffectiveArea(EElectronEffectiveAreaType type, Double_t Eta,
120			EElectronEffectiveAreaTarget EffectiveAreaTarget = kEleEAData2011);
121	sixie	1.1
122			ClassDef(ElectronTools, 0) // Muon tools
123			};
124			}
125
126			#endif