Mods/interface/ElectronIDMod.h

//--------------------------------------------------------------------------------------------------
// $Id: ElectronIDMod.h,v 1.56 2012/04/24 11:45:54 fabstoec 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/TrackFwd.h"
#include "MitAna/DataTree/interface/DecayParticleFwd.h"
#include "MitPhysics/Utils/interface/ElectronTools.h"
#include "MitPhysics/Utils/interface/IsolationTools.h"
#include "MitPhysics/ElectronLikelihood/interface/ElectronLikelihood.h"
#include "MitPhysics/ElectronLikelihood/interface/LikelihoodSwitches.h"
#include "MitPhysics/ElectronLikelihood/interface/LikelihoodMeasurements.h"
#include "MitPhysics/Utils/interface/ElectronIDMVA.h"
#include "MitAna/DataTree/interface/PileupEnergyDensityCol.h"
#include <TFile.h>
#include <TDirectory.h>
#include "TMVA/Reader.h"

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

      Bool_t              GetApplyConversionFilterType1() const { return fApplyConvFilterType1;   }
      Bool_t              GetApplyConversionFilterType2() const { return fApplyConvFilterType2;   }
      Bool_t              GetApplySpikeRemoval()          const { return fApplySpikeRemoval;      }
      Bool_t              GetApplyD0Cut()                 const { return fApplyD0Cut;             }
      Bool_t              GetApplyDZCut()                 const { return fApplyDZCut;             }
      Double_t            GetCaloIsoCut()                 const { return fCaloIsolationCut;       }
      Double_t            GetEcalJurIsoCut()              const { return fEcalJuraIsoCut;         }
      Double_t            GetCombIsoCut()                 const { return fCombIsolationCut;       }
      Double_t            GetPFIsoCut()                   const { return fPFIsolationCut;         }
      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;          }
      Double_t            GetEtaMax()                     const { return fElectronEtaMax;         }
      Bool_t              GetApplyTriggerMatching()       const { return fApplyTriggerMatching;   }
      Double_t            GetTrackIsoCut()                const { return fTrackIsolationCut;      }
      Bool_t              GetChargeFilter()               const { return fChargeFilter;           }
      Bool_t              PassLikelihoodID(const Electron *ele) const;
      Bool_t              PassMVAID(const Electron *el, ElectronTools::EElIdType idType, 
                                    const Vertex *vertex, const PFCandidateCol *PFCands,
                                    const PileupEnergyDensityCol *PileupEnergyDensity) const;
      Bool_t              PassIDCut(const Electron *el, ElectronTools::EElIdType idType, 
                                    const Vertex *vertex) const;
      Bool_t              PassIsolationCut(const Electron *el, ElectronTools::EElIsoType isoType,
                                           const TrackCol *tracks, const Vertex *vertex, 
                                           const Double_t rho) const;
      Bool_t              GetCombinedIdCut()               const { return fCombinedIdCut;      }
      void                SetPrintMVADebugInfo(Bool_t b)         { fPrintMVADebugInfo = b;     }
      void                SetApplyConversionFilterType1(Bool_t b){ fApplyConvFilterType1 = b;  }
      void                SetApplyConversionFilterType2(Bool_t b){ fApplyConvFilterType2 = b;  }
      void                SetNExpectedHitsInnerCut(Int_t cut)    { fNExpectedHitsInnerCut = cut;}
      void                SetInvertNExpectedHitsInnerCut(Bool_t b){ fInvertNExpectedHitsInnerCut = b;}
      void                SetApplySpikeRemoval(Bool_t b)         { fApplySpikeRemoval  = b;    }
      void                SetApplyD0Cut(Bool_t b)                { fApplyD0Cut         = b;    }
      void                SetApplyDZCut(Bool_t b)                { fApplyDZCut         = b;    }
      void                SetCaloIsoCut(Double_t cut)            { fCaloIsolationCut   = cut;  }
      void                SetCombIsoCut(Double_t cut)            { fCombIsolationCut   = cut;  }
      void                SetCombRelativeIsoCut(Double_t cut, Double_t cut_EE = -1.)    { 
        fCombRelativeIsolationCut = cut; 
        if ( cut_EE < 0. )
          fCombRelativeIsolationCut_EE = cut;
        else
          fCombRelativeIsolationCut_EE = cut_EE;
      }
      void                SetPFIsoCut(Double_t cut)              { fPFIsolationCut     = cut;  }
      void                SetD0Cut(Double_t cut)                 { fD0Cut = cut;               }
      void                SetDZCut(Double_t cut)                 { fDZCut = cut;               }
      void                SetWhichVertex(Int_t d)                { fWhichVertex = d;           }
      void                SetEcalJurIsoCut(Double_t cut)         { fEcalJuraIsoCut     = cut;  }
      void                SetGoodElectronsName(const char *n)    { fGoodElectronsName  = n;    }  
      void                SetOldMuonsName(const char *n)         { fNonIsolatedMuonsName  = n;    }  
      void                SetOldElectronsName(const char *n)     { fNonIsolatedElectronsName  = 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                SetTriggerObjectsName(const char *n)   { fTrigObjectsName = n;       }
      void                SetOutputName(const char *n)           { SetGoodElectronsName(n);    }  
      void                SetPtMin(Double_t pt)                  { fElectronPtMin      = pt;   }
      void                SetEtMin(Double_t et)                  { fElectronEtMin      = et;   }      
      void                SetEtaMax(Double_t eta)                { fElectronEtaMax     = eta;  }
      void                SetApplyTriggerMatching(Bool_t b)      { fApplyTriggerMatching = b;  }
      void                SetTrackIsoCut(Double_t cut)           { fTrackIsolationCut  = cut;  }
      void                SetChargeFilter(Bool_t b)              { fChargeFilter = b;          }
      void                SetNWrongHitsMax(UInt_t n)             { fNWrongHitsMax = n;         }
      void                SetCombinedIdCut(Bool_t b)             { fCombinedIdCut  = b;        }
      void                SetApplyEcalFiducial(Bool_t b)         { fApplyEcalFiducial = b;     }
      void                SetApplyEcalSeeded(Bool_t b)           { fApplyEcalSeeded = b;       }      
      void                SetApplyCombinedIso(Bool_t b)          { fApplyCombinedIso = b;      }      
      void                SetElectronsFromBranch(Bool_t b)       { fElectronsFromBranch = b;   }      
      void                SetLH(ElectronLikelihood *l)           { fLH = l;                    }
      void                SetVertexName(TString name)            { fVertexName = name;         }
      void                SetIntRadius(Double_t dr)              { fIntRadius = dr;            }
      void                Setup();
 
      void                SetElectronMVAWeightsSubdet0Pt10To20(TString s)  
                          { fElectronMVAWeights_Subdet0Pt10To20  = s; }
      void                SetElectronMVAWeightsSubdet1Pt10To20(TString s)  
                          { fElectronMVAWeights_Subdet1Pt10To20  = s; }
      void                SetElectronMVAWeightsSubdet2Pt10To20(TString s)  
                          { fElectronMVAWeights_Subdet2Pt10To20  = s; }
      void                SetElectronMVAWeightsSubdet0Pt20ToInf(TString s) 
                          { fElectronMVAWeights_Subdet0Pt20ToInf = s; }
      void                SetElectronMVAWeightsSubdet1Pt20ToInf(TString s) 
                          { fElectronMVAWeights_Subdet1Pt20ToInf = s; }
      void                SetElectronMVAWeightsSubdet2Pt20ToInf(TString s) 
                          { fElectronMVAWeights_Subdet2Pt20ToInf = s; }

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

      Bool_t                    fPrintMVADebugInfo;      //print MVA debug information
      TString                   fElectronBranchName;     //name of electron collection (input)
      TString                   fConversionBranchName;   //name of electron collection (input)
      TString                   fGoodElectronsName;      //name of exported "good electrons" col
      TString                   fNonIsolatedMuonsName;    //name of imported "old muon" collection
      TString                   fNonIsolatedElectronsName;//name of imported "old electron" collection
      TString                   fVertexName;             //name of vertex collection
      TString                   fBeamSpotName;           //name of beamspot collection
      TString                   fTrackName;              //name of track collection
      TString                   fPFCandidatesName;       //name of pfcandidates collection
      TString                   fElectronIDType;         //type of electron ID we impose
      TString                   fElectronIsoType;        //type of electron Isolation we impose
      TString                   fTrigObjectsName;        //name of trigger object collection
      Double_t                  fElectronPtMin;          //min pt cut
      Double_t                  fElectronEtMin;          //min pt cut            
      Double_t                  fElectronEtaMax;         //max eta 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
      Double_t                  fCombRelativeIsolationCut; //cut value for combined relative isolation
      Double_t                  fCombRelativeIsolationCut_EE; //cut value for combined relative isolation (EE)
      Double_t                  fPFIsolationCut;         //cut value for PF isolation, default -1.0
      Bool_t                    fApplyConvFilterType1;   //whether remove conversions using fit method
      Bool_t                    fApplyConvFilterType2;   //whether remove conversions using DCotTheta method
      UInt_t                    fNWrongHitsMax;          //whether to use wrong hits req 
                                                         //for conversion removal
      Int_t                     fNExpectedHitsInnerCut;  //cut value for NExpectedHitsInner maximum
      Bool_t                    fInvertNExpectedHitsInnerCut; //whether to invert NExpectedHitsInner cut
      Bool_t                    fCombinedIdCut;          //whether to use full combined id
      Bool_t                    fApplySpikeRemoval;      //whether spike removal
      Bool_t                    fApplyD0Cut;             //whether apply d0 cut
      Bool_t                    fApplyDZCut;             //whether apply dz cut
      Bool_t                    fChargeFilter;           //whether apply GSF and CFT equal requirement
      Double_t                  fD0Cut;                  //max d0
      Double_t                  fDZCut;                  //max dz
      Int_t                     fWhichVertex;            //vertex to use (-2: beamspot, -1: closest in Z)
      Bool_t                    fApplyTriggerMatching;   //match to hlt electron (default=0)
      Bool_t                    fApplyEcalSeeded;        //require ecal seeded flag
      Bool_t                    fApplyCombinedIso;       //apply combined isolation
      Bool_t                    fApplyEcalFiducial;      //apply ecal fiducial cuts on supercluster eta
      Bool_t                    fElectronsFromBranch;    //where to get input electrons
      ElectronTools::EElIdType  fElIdType;              //!identification scheme
      ElectronTools::EElIsoType fElIsoType;              //!isolation scheme
      const ElectronCol        *fElectrons;              //!electron collection
      const DecayParticleCol   *fConversions;            //!conversion collection
      const VertexCol          *fVertices;               //!vertices branches
      const BeamSpotCol        *fBeamSpot;               //!beamspot branch
      const TrackCol           *fTracks;                 //!Track branch     
      const PFCandidateCol     *fPFCandidates;           //!pfcandidate branch
      PFCandidateCol           *fPFNoPileUpCands;        //!pfcandidate branch with PFNoPU     
      Double_t                  fIntRadius;              //!min IntRadius cut in pf isolation
      MuonCol                  *fNonIsolatedMuons;       //!pointer to old muon collection 
      ElectronCol              *fNonIsolatedElectrons;   //!pointer to old electron collection
      ElectronLikelihood       *fLH;                     //LH
      TString                   fPileupEnergyDensityName;
      const PileupEnergyDensityCol *fPileupEnergyDensity;
      ElectronIDMVA            *fElectronIDMVA;
      TString                   fElectronMVAWeights_Subdet0Pt10To20;
      TString                   fElectronMVAWeights_Subdet1Pt10To20;
      TString                   fElectronMVAWeights_Subdet2Pt10To20;
      TString                   fElectronMVAWeights_Subdet0Pt20ToInf;
      TString                   fElectronMVAWeights_Subdet1Pt20ToInf;
      TString                   fElectronMVAWeights_Subdet2Pt20ToInf;

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