ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitPhysics/Mods/interface/ElectronIDMod.h
Revision: 1.62
Committed: Mon Jan 7 22:15:21 2013 UTC (12 years, 4 months ago) by mingyang
Content type: text/plain
Branch: MAIN
CVS Tags: Mit_029c, Mit_029b, Mit_029a, HEAD
Changes since 1.61: +3 -1 lines
Log Message: 
beamspot constraint fVertices option added

File Contents

# User Rev Content
1 loizides 1.1 //--------------------------------------------------------------------------------------------------
2 mingyang 1.62 // $Id: ElectronIDMod.h,v 1.61 2012/10/08 17:18:09 mingyang 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 ceballos 1.41 #include "MitAna/DataTree/interface/TrackFwd.h"
21 loizides 1.20 #include "MitAna/DataTree/interface/DecayParticleFwd.h"
22 sixie 1.31 #include "MitPhysics/Utils/interface/ElectronTools.h"
23 ceballos 1.41 #include "MitPhysics/Utils/interface/IsolationTools.h"
24 ceballos 1.42 #include "MitPhysics/ElectronLikelihood/interface/ElectronLikelihood.h"
25     #include "MitPhysics/ElectronLikelihood/interface/LikelihoodSwitches.h"
26     #include "MitPhysics/ElectronLikelihood/interface/LikelihoodMeasurements.h"
27 sixie 1.53 #include "MitPhysics/Utils/interface/ElectronIDMVA.h"
28 ceballos 1.44 #include "MitAna/DataTree/interface/PileupEnergyDensityCol.h"
29 ceballos 1.42 #include <TFile.h>
30     #include <TDirectory.h>
31 sixie 1.53 #include "TMVA/Reader.h"
32 loizides 1.1
33 fabstoec 1.58 // for Rho definitons
34     #include "MitPhysics/Utils/interface/RhoUtilities.h"
35    
36 loizides 1.1 namespace mithep
37     {
38     class ElectronIDMod : public BaseMod
39     {
40     public:
41     ElectronIDMod(const char *name="ElectronIDMod",
42 loizides 1.4 const char *title="Electron identification module");
43    
44 ceballos 1.34 Bool_t GetApplyConversionFilterType1() const { return fApplyConvFilterType1; }
45     Bool_t GetApplyConversionFilterType2() const { return fApplyConvFilterType2; }
46     Bool_t GetApplySpikeRemoval() const { return fApplySpikeRemoval; }
47     Bool_t GetApplyD0Cut() const { return fApplyD0Cut; }
48 ceballos 1.45 Bool_t GetApplyDZCut() const { return fApplyDZCut; }
49 ceballos 1.34 Double_t GetCaloIsoCut() const { return fCaloIsolationCut; }
50     Double_t GetEcalJurIsoCut() const { return fEcalJuraIsoCut; }
51     Double_t GetCombIsoCut() const { return fCombIsolationCut; }
52 fabstoec 1.49 Double_t GetPFIsoCut() const { return fPFIsolationCut; }
53 ceballos 1.34 const char *GetGoodName() const { return GetGoodElectronsName(); }
54     const char *GetGoodElectronsName() const { return fGoodElectronsName; }
55     Double_t GetHcalIsoCut() const { return fHcalIsolationCut; }
56     Double_t GetIDLikelihoodCut() const { return fIDLikelihoodCut; }
57     const char *GetIDType() const { return fElectronIDType; }
58     const char *GetInputName() const { return fElectronBranchName; }
59     const char *GetIsoType() const { return fElectronIsoType; }
60     const char *GetOutputName() const { return GetGoodElectronsName(); }
61     Double_t GetPtMin() const { return fElectronPtMin; }
62 ceballos 1.37 Double_t GetEtaMax() const { return fElectronEtaMax; }
63 bendavid 1.35 Bool_t GetApplyTriggerMatching() const { return fApplyTriggerMatching; }
64 ceballos 1.34 Double_t GetTrackIsoCut() const { return fTrackIsolationCut; }
65     Bool_t GetChargeFilter() const { return fChargeFilter; }
66 sixie 1.53 Bool_t PassLikelihoodID(const Electron *ele) const;
67     Bool_t PassMVAID(const Electron *el, ElectronTools::EElIdType idType,
68 sixie 1.54 const Vertex *vertex, const PFCandidateCol *PFCands,
69     const PileupEnergyDensityCol *PileupEnergyDensity) const;
70 mingyang 1.61 Double_t EvaluateMVAID(const Electron *el, ElectronTools::EElIdType idType,
71     const Vertex *vertex, const PFCandidateCol *PFCands,
72     const PileupEnergyDensityCol *PileupEnergyDensity) const;
73 sixie 1.53 Bool_t PassIDCut(const Electron *el, ElectronTools::EElIdType idType,
74     const Vertex *vertex) const;
75 ceballos 1.41 Bool_t PassIsolationCut(const Electron *el, ElectronTools::EElIsoType isoType,
76 ceballos 1.44 const TrackCol *tracks, const Vertex *vertex,
77 ceballos 1.59 const Double_t rho,
78     ElectronTools::EElIdType idType = ElectronTools::kNoId) const;
79 ceballos 1.34 Bool_t GetCombinedIdCut() const { return fCombinedIdCut; }
80 sixie 1.55 void SetPrintMVADebugInfo(Bool_t b) { fPrintMVADebugInfo = b; }
81 ceballos 1.34 void SetApplyConversionFilterType1(Bool_t b){ fApplyConvFilterType1 = b; }
82     void SetApplyConversionFilterType2(Bool_t b){ fApplyConvFilterType2 = b; }
83 ceballos 1.50 void SetNExpectedHitsInnerCut(Int_t cut) { fNExpectedHitsInnerCut = cut;}
84     void SetInvertNExpectedHitsInnerCut(Bool_t b){ fInvertNExpectedHitsInnerCut = b;}
85 ceballos 1.34 void SetApplySpikeRemoval(Bool_t b) { fApplySpikeRemoval = b; }
86     void SetApplyD0Cut(Bool_t b) { fApplyD0Cut = b; }
87 ceballos 1.45 void SetApplyDZCut(Bool_t b) { fApplyDZCut = b; }
88 ceballos 1.34 void SetCaloIsoCut(Double_t cut) { fCaloIsolationCut = cut; }
89 fabstoec 1.49 void SetCombIsoCut(Double_t cut) { fCombIsolationCut = cut; }
90 fabstoec 1.56 void SetCombRelativeIsoCut(Double_t cut, Double_t cut_EE = -1.) {
91     fCombRelativeIsolationCut = cut;
92     if ( cut_EE < 0. )
93     fCombRelativeIsolationCut_EE = cut;
94     else
95     fCombRelativeIsolationCut_EE = cut_EE;
96     }
97 fabstoec 1.49 void SetPFIsoCut(Double_t cut) { fPFIsolationCut = cut; }
98 ceballos 1.34 void SetD0Cut(Double_t cut) { fD0Cut = cut; }
99 ceballos 1.45 void SetDZCut(Double_t cut) { fDZCut = cut; }
100     void SetWhichVertex(Int_t d) { fWhichVertex = d; }
101 ceballos 1.34 void SetEcalJurIsoCut(Double_t cut) { fEcalJuraIsoCut = cut; }
102     void SetGoodElectronsName(const char *n) { fGoodElectronsName = n; }
103 ceballos 1.43 void SetOldMuonsName(const char *n) { fNonIsolatedMuonsName = n; }
104     void SetOldElectronsName(const char *n) { fNonIsolatedElectronsName = n;}
105 ceballos 1.34 void SetGoodName(const char *n) { SetGoodElectronsName(n); }
106     void SetHcalIsoCut(Double_t cut) { fHcalIsolationCut = cut; }
107     void SetIDLikelihoodCut(Double_t cut) { fIDLikelihoodCut = cut; }
108     void SetIDType(const char *type) { fElectronIDType = type; }
109     void SetInputName(const char *n) { fElectronBranchName = n; }
110     void SetIsoType(const char *type) { fElectronIsoType = type; }
111 bendavid 1.36 void SetTriggerObjectsName(const char *n) { fTrigObjectsName = n; }
112 ceballos 1.34 void SetOutputName(const char *n) { SetGoodElectronsName(n); }
113     void SetPtMin(Double_t pt) { fElectronPtMin = pt; }
114 bendavid 1.38 void SetEtMin(Double_t et) { fElectronEtMin = et; }
115     void SetEtaMax(Double_t eta) { fElectronEtaMax = eta; }
116 ceballos 1.37 void SetApplyTriggerMatching(Bool_t b) { fApplyTriggerMatching = b; }
117 ceballos 1.34 void SetTrackIsoCut(Double_t cut) { fTrackIsolationCut = cut; }
118     void SetChargeFilter(Bool_t b) { fChargeFilter = b; }
119 bendavid 1.40 void SetNWrongHitsMax(UInt_t n) { fNWrongHitsMax = n; }
120 ceballos 1.34 void SetCombinedIdCut(Bool_t b) { fCombinedIdCut = b; }
121 bendavid 1.38 void SetApplyEcalFiducial(Bool_t b) { fApplyEcalFiducial = b; }
122     void SetApplyEcalSeeded(Bool_t b) { fApplyEcalSeeded = b; }
123     void SetApplyCombinedIso(Bool_t b) { fApplyCombinedIso = b; }
124 dkralph 1.39 void SetElectronsFromBranch(Bool_t b) { fElectronsFromBranch = b; }
125 ceballos 1.42 void SetLH(ElectronLikelihood *l) { fLH = l; }
126 bendavid 1.47 void SetVertexName(TString name) { fVertexName = name; }
127 ceballos 1.52 void SetIntRadius(Double_t dr) { fIntRadius = dr; }
128 fabstoec 1.60 void SetPFNoPileUpName(const char *n) { fPFNoPileUpName = n; }
129 mingyang 1.62 void SetPVName(const char *n) { fPVName = n; }
130 ceballos 1.27 void Setup();
131 sixie 1.53
132     void SetElectronMVAWeightsSubdet0Pt10To20(TString s)
133     { fElectronMVAWeights_Subdet0Pt10To20 = s; }
134     void SetElectronMVAWeightsSubdet1Pt10To20(TString s)
135     { fElectronMVAWeights_Subdet1Pt10To20 = s; }
136     void SetElectronMVAWeightsSubdet2Pt10To20(TString s)
137     { fElectronMVAWeights_Subdet2Pt10To20 = s; }
138     void SetElectronMVAWeightsSubdet0Pt20ToInf(TString s)
139     { fElectronMVAWeights_Subdet0Pt20ToInf = s; }
140     void SetElectronMVAWeightsSubdet1Pt20ToInf(TString s)
141     { fElectronMVAWeights_Subdet1Pt20ToInf = s; }
142     void SetElectronMVAWeightsSubdet2Pt20ToInf(TString s)
143     { fElectronMVAWeights_Subdet2Pt20ToInf = s; }
144 loizides 1.4
145 fabstoec 1.58 void SetRhoType(RhoUtilities::RhoType type) { fTheRhoType = type; };
146    
147    
148 loizides 1.1 protected:
149 sixie 1.26 void Process();
150     void SlaveBegin();
151 ceballos 1.42 void Terminate();
152 loizides 1.7
153 sixie 1.55 Bool_t fPrintMVADebugInfo; //print MVA debug information
154 sixie 1.31 TString fElectronBranchName; //name of electron collection (input)
155     TString fConversionBranchName; //name of electron collection (input)
156     TString fGoodElectronsName; //name of exported "good electrons" col
157 ceballos 1.43 TString fNonIsolatedMuonsName; //name of imported "old muon" collection
158     TString fNonIsolatedElectronsName;//name of imported "old electron" collection
159 sixie 1.31 TString fVertexName; //name of vertex collection
160 bendavid 1.40 TString fBeamSpotName; //name of beamspot collection
161 ceballos 1.41 TString fTrackName; //name of track collection
162     TString fPFCandidatesName; //name of pfcandidates collection
163 fabstoec 1.60 TString fPFNoPileUpName; //name of pfcandidates collection
164 sixie 1.31 TString fElectronIDType; //type of electron ID we impose
165     TString fElectronIsoType; //type of electron Isolation we impose
166 bendavid 1.35 TString fTrigObjectsName; //name of trigger object collection
167 sixie 1.31 Double_t fElectronPtMin; //min pt cut
168 bendavid 1.38 Double_t fElectronEtMin; //min pt cut
169 ceballos 1.37 Double_t fElectronEtaMax; //max eta cut
170 sixie 1.31 Double_t fIDLikelihoodCut; //cut value for ID likelihood
171     Double_t fTrackIsolationCut; //cut value for track isolation
172     Double_t fCaloIsolationCut; //cut value for calo isolation
173     Double_t fEcalJuraIsoCut; //cut value for ecal jurassic isolation
174     Double_t fHcalIsolationCut; //cut value for hcal isolation
175     Double_t fCombIsolationCut; //cut value for combined isolation
176 sixie 1.51 Double_t fCombRelativeIsolationCut; //cut value for combined relative isolation
177 fabstoec 1.56 Double_t fCombRelativeIsolationCut_EE; //cut value for combined relative isolation (EE)
178 fabstoec 1.49 Double_t fPFIsolationCut; //cut value for PF isolation, default -1.0
179 ceballos 1.34 Bool_t fApplyConvFilterType1; //whether remove conversions using fit method
180     Bool_t fApplyConvFilterType2; //whether remove conversions using DCotTheta method
181 bendavid 1.40 UInt_t fNWrongHitsMax; //whether to use wrong hits req
182 sixie 1.31 //for conversion removal
183 ceballos 1.50 Int_t fNExpectedHitsInnerCut; //cut value for NExpectedHitsInner maximum
184     Bool_t fInvertNExpectedHitsInnerCut; //whether to invert NExpectedHitsInner cut
185 ceballos 1.33 Bool_t fCombinedIdCut; //whether to use full combined id
186 ceballos 1.45 Bool_t fApplySpikeRemoval; //whether spike removal
187 sixie 1.31 Bool_t fApplyD0Cut; //whether apply d0 cut
188 ceballos 1.45 Bool_t fApplyDZCut; //whether apply dz cut
189 sixie 1.31 Bool_t fChargeFilter; //whether apply GSF and CFT equal requirement
190     Double_t fD0Cut; //max d0
191 ceballos 1.45 Double_t fDZCut; //max dz
192     Int_t fWhichVertex; //vertex to use (-2: beamspot, -1: closest in Z)
193 bendavid 1.35 Bool_t fApplyTriggerMatching; //match to hlt electron (default=0)
194 bendavid 1.38 Bool_t fApplyEcalSeeded; //require ecal seeded flag
195     Bool_t fApplyCombinedIso; //apply combined isolation
196     Bool_t fApplyEcalFiducial; //apply ecal fiducial cuts on supercluster eta
197 dkralph 1.39 Bool_t fElectronsFromBranch; //where to get input electrons
198 sixie 1.31 ElectronTools::EElIdType fElIdType; //!identification scheme
199     ElectronTools::EElIsoType fElIsoType; //!isolation scheme
200     const ElectronCol *fElectrons; //!electron collection
201     const DecayParticleCol *fConversions; //!conversion collection
202     const VertexCol *fVertices; //!vertices branches
203 bendavid 1.40 const BeamSpotCol *fBeamSpot; //!beamspot branch
204 ceballos 1.41 const TrackCol *fTracks; //!Track branch
205     const PFCandidateCol *fPFCandidates; //!pfcandidate branch
206 ceballos 1.57 PFCandidateCol *fPFNoPileUpCands; //!pfcandidate branch with PFNoPU
207 ceballos 1.52 Double_t fIntRadius; //!min IntRadius cut in pf isolation
208 ceballos 1.43 MuonCol *fNonIsolatedMuons; //!pointer to old muon collection
209     ElectronCol *fNonIsolatedElectrons; //!pointer to old electron collection
210 ceballos 1.48 ElectronLikelihood *fLH; //LH
211 ceballos 1.44 TString fPileupEnergyDensityName;
212     const PileupEnergyDensityCol *fPileupEnergyDensity;
213 sixie 1.53 ElectronIDMVA *fElectronIDMVA;
214     TString fElectronMVAWeights_Subdet0Pt10To20;
215     TString fElectronMVAWeights_Subdet1Pt10To20;
216     TString fElectronMVAWeights_Subdet2Pt10To20;
217     TString fElectronMVAWeights_Subdet0Pt20ToInf;
218     TString fElectronMVAWeights_Subdet1Pt20ToInf;
219     TString fElectronMVAWeights_Subdet2Pt20ToInf;
220 mingyang 1.62 TString fPVName;
221 ceballos 1.44
222 fabstoec 1.58 RhoUtilities::RhoType fTheRhoType;
223    
224    
225 loizides 1.8 ClassDef(ElectronIDMod, 1) // Electron identification module
226 loizides 1.1 };
227     }
228     #endif