Mods/interface/ElectronIDMod.h

//--------------------------------------------------------------------------------------------------
// $Id: ElectronIDMod.h,v 1.9 2008/12/11 10:55:43 loizides Exp $
//
// ElectronIDMod
//
// This module applies electron identification criteria and exports a pointer to a collection
// of "good electrons" according to the specified identification scheme.
//
// 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/Collections.h"

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

      Bool_t              GetApplyConversionFilter()  const { return fApplyConversionFilter;  }
      Double_t            GetCaloIsoCut()             const { return fCaloIsolationCut;       }
      Double_t            GetEcalJurIsoCut()          const { return fEcalJuraIsoCut;         }
      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            GetTrackIsoCut()            const { return fTrackIsolationCut;      }
      void                SetApplyConversionFilter(Bool_t b)    { fApplyConversionFilter  = b;    }
      void                SetCaloIsoCut(Double_t cut)           { fCaloIsolationCut   = cut;  }
      void                SetEcalJurIsoCut(Double_t cut)        { fEcalJuraIsoCut     = cut;  }
      void                SetGoodName(const char *n)            { SetGoodElectronsName(n);    }   
      void                SetGoodElectronsName(const char *n)   { fGoodElectronsName  = 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                SetTrackIsoCut(Double_t cut)          { fTrackIsolationCut  = cut;  }
      void                SetD0Cut(Double_t cut)                { fD0Cut = cut;               }

      enum EElIdType {
        kIdUndef = 0,       //not defined
        kTight,             //"Tight"
        kLoose,             //"Loose"
        kLikelihood,        //"Likelihood"
        kNoId,              //"NoId"
        kCustomId           //"Custom"
      };
      enum EElIsoType {
        kIsoUndef = 0,      //not defined
        kTrackCalo,         //"TrackCalo"
        kTrackJura,         //"TrackJura"
        kTrackJuraSliding,  //"TrackJuraSliding"
        kNoIso,             //"NoIso"
        kCustomIso          //"Custom"
      };

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

      TString                 fElectronBranchName;     //name of electron collection (input)
      TString                 fConversionBranchName;   //name of electron collection (input)
      TString                 fGoodElectronsName;      //name of exported "good electrons" collection
      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
      const ElectronCol      *fElectrons;              //!electron collection
      const DecayParticleCol *fConversions;            //!conversion collection
      const VertexCol        *fVertices;               // Vertices branches
      EElIdType               fElIdType;               //!identification scheme
      EElIsoType              fElIsoType;              //!isolation scheme
      Bool_t                  fApplyConversionFilter;  //!whether remove conversions
      Double_t                fD0Cut;                  //max d0
    
    ClassDef(ElectronIDMod, 1) // Electron identification module
  };
}
#endif
Revision:	1.12
Committed:	Thu Apr 2 09:20:53 2009 UTC (16 years, 1 month ago) by ceballos
Content type:	text/plain
Branch:	MAIN
Changes since 1.11:	+26 -18 lines
Log Message:	new lepton ID
#	User	Rev	Content
1	loizides	1.1	//--------------------------------------------------------------------------------------------------
2	ceballos	1.12	// $Id: ElectronIDMod.h,v 1.9 2008/12/11 10:55:43 loizides 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.4	// Authors: S.Xie, C.Loizides
10	loizides	1.1	//--------------------------------------------------------------------------------------------------
11
12	loizides	1.4	#ifndef MITPHYSICS_MODS_ELECTRONIDMOD_H
13			#define MITPHYSICS_MODS_ELECTRONIDMOD_H
14	loizides	1.1
15			#include "MitAna/TreeMod/interface/BaseMod.h"
16			#include "MitAna/DataTree/interface/Collections.h"
17
18			namespace mithep
19			{
20			class ElectronIDMod : public BaseMod
21			{
22			public:
23			ElectronIDMod(const char *name="ElectronIDMod",
24	loizides	1.4	const char *title="Electron identification module");
25	ceballos	1.12	~ElectronIDMod() {}
26	loizides	1.4
27	ceballos	1.12	Bool_t GetApplyConversionFilter() const { return fApplyConversionFilter; }
28	loizides	1.8	Double_t GetCaloIsoCut() const { return fCaloIsolationCut; }
29			Double_t GetEcalJurIsoCut() const { return fEcalJuraIsoCut; }
30			const char *GetGoodName() const { return GetGoodElectronsName(); }
31			const char *GetGoodElectronsName() const { return fGoodElectronsName; }
32			Double_t GetHcalIsoCut() const { return fHcalIsolationCut; }
33			Double_t GetIDLikelihoodCut() const { return fIDLikelihoodCut; }
34			const char *GetIDType() const { return fElectronIDType; }
35			const char *GetInputName() const { return fElectronBranchName; }
36			const char *GetIsoType() const { return fElectronIsoType; }
37			const char *GetOutputName() const { return GetGoodElectronsName(); }
38			Double_t GetPtMin() const { return fElectronPtMin; }
39			Double_t GetTrackIsoCut() const { return fTrackIsolationCut; }
40	ceballos	1.12	void SetApplyConversionFilter(Bool_t b) { fApplyConversionFilter = b; }
41	loizides	1.8	void SetCaloIsoCut(Double_t cut) { fCaloIsolationCut = cut; }
42			void SetEcalJurIsoCut(Double_t cut) { fEcalJuraIsoCut = cut; }
43			void SetGoodName(const char *n) { SetGoodElectronsName(n); }
44			void SetGoodElectronsName(const char *n) { fGoodElectronsName = n; }
45			void SetHcalIsoCut(Double_t cut) { fHcalIsolationCut = cut; }
46			void SetIDLikelihoodCut(Double_t cut) { fIDLikelihoodCut = cut; }
47			void SetIDType(const char *type) { fElectronIDType = type; }
48			void SetInputName(const char *n) { fElectronBranchName = n; }
49			void SetIsoType(const char *type) { fElectronIsoType = type; }
50			void SetOutputName(const char *n) { SetGoodElectronsName(n); }
51			void SetPtMin(Double_t pt) { fElectronPtMin = pt; }
52			void SetTrackIsoCut(Double_t cut) { fTrackIsolationCut = cut; }
53	ceballos	1.12	void SetD0Cut(Double_t cut) { fD0Cut = cut; }
54	loizides	1.4
55			enum EElIdType {
56			kIdUndef = 0, //not defined
57			kTight, //"Tight"
58			kLoose, //"Loose"
59			kLikelihood, //"Likelihood"
60	loizides	1.9	kNoId, //"NoId"
61	loizides	1.4	kCustomId //"Custom"
62			};
63			enum EElIsoType {
64			kIsoUndef = 0, //not defined
65			kTrackCalo, //"TrackCalo"
66			kTrackJura, //"TrackJura"
67			kTrackJuraSliding, //"TrackJuraSliding"
68			kNoIso, //"NoIso"
69			kCustomIso //"Custom"
70			};
71
72	loizides	1.1	protected:
73	ceballos	1.12	void Process();
74			void SlaveBegin();
75	loizides	1.7
76	ceballos	1.12	TString fElectronBranchName; //name of electron collection (input)
77			TString fConversionBranchName; //name of electron collection (input)
78			TString fGoodElectronsName; //name of exported "good electrons" collection
79			TString fVertexName; //name of vertex collection
80			TString fElectronIDType; //type of electron ID we impose
81			TString fElectronIsoType; //type of electron Isolation we impose
82			Double_t fElectronPtMin; //min pt cut
83			Double_t fIDLikelihoodCut; //cut value for ID likelihood
84			Double_t fTrackIsolationCut; //cut value for track isolation
85			Double_t fCaloIsolationCut; //cut value for calo isolation
86			Double_t fEcalJuraIsoCut; //cut value for ecal jurassic isolation
87			Double_t fHcalIsolationCut; //cut value for hcal isolation
88			const ElectronCol *fElectrons; //!electron collection
89			const DecayParticleCol *fConversions; //!conversion collection
90			const VertexCol *fVertices; // Vertices branches
91			EElIdType fElIdType; //!identification scheme
92			EElIsoType fElIsoType; //!isolation scheme
93			Bool_t fApplyConversionFilter; //!whether remove conversions
94			Double_t fD0Cut; //max d0
95	loizides	1.1
96	loizides	1.8	ClassDef(ElectronIDMod, 1) // Electron identification module
97	loizides	1.1	};
98			}
99			#endif