Utils/interface/JetIDMVA.h

//--------------------------------------------------------------------------------------------------
// $Id $
//
// JetIDMVA
//
// Helper Class for Jet Id MVA
//
// Authors: P. Harris
//--------------------------------------------------------------------------------------------------

#ifndef MITPHYSICS_UTILS_JetIDMVA_H
#define MITPHYSICS_UTILS_JetIDMVA_H

#include "CondFormats/JetMETObjects/interface/FactorizedJetCorrector.h"
#include "MitAna/DataTree/interface/PFJetFwd.h"
#include "MitAna/DataTree/interface/VertexFwd.h"
#include "MitAna/DataTree/interface/TrackFwd.h"
#include "MitAna/DataTree/interface/PFJet.h"
#include "MitAna/DataTree/interface/PFJetCol.h"
#include "MitAna/DataTree/interface/PFCandidateCol.h"
#include "MitAna/DataTree/interface/PileupEnergyDensityCol.h"
#include "MitCommon/MathTools/interface/MathUtils.h"

class TRandom3;
namespace TMVA {
  class Reader;
}

namespace mithep {
  class JetIDMVA {
    public:
      JetIDMVA();
      ~JetIDMVA(); 

      enum MVAType {
        kBaseline = 0
      };

      enum CutType {
        kTight     = 0,
        kMedium    = 1,
        kLoose     = 2,
        kMET       = 3
      };

      void     Initialize(JetIDMVA::CutType iCutType,
                          TString           iLowPtWeights ="$CMSSW_BASE/src/MitPhysics/data/mva_JetID_lowpt.weights.xml",
                          TString           iHighPtWeights="$CMSSW_BASE/src/MitPhysics/data/mva_JetID_highpt.weights.xml",
                          JetIDMVA::MVAType iType=kBaseline,
                          TString           iCutFileName  ="$CMSSW_BASE/src/MitPhysics/Utils/python/JetIdParams_cfi.py");
      
      Bool_t   IsInitialized() const { return fIsInitialized; }
      Double_t MVAValue(    
                        Float_t iNPV    ,
                        Float_t iJPt1   ,
                        Float_t iJEta1  ,
                        Float_t iJPhi1  ,
                        Float_t iJD01   ,
                        Float_t iJDZ1   ,
                        Float_t iBeta   ,
                        Float_t iBetaStar,
                        Float_t iNCharged,
                        Float_t iNNeutrals,
                        Float_t iDRMean  ,
                        Float_t iFrac01  ,
                        Float_t iFrac02  ,
                        Float_t iFrac03  ,
                        Float_t iFrac04  ,
                        Float_t iFrac05  
                        );

      //UNcorrected Jets
      Bool_t   pass(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices,
                    FactorizedJetCorrector *iJetCorrector,
                    const PileupEnergyDensityCol *iPileupEnergyDensity);
      
      //Corrected Jets
      Bool_t   pass(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices);
                                        
      //Uncorrected Jets
      Double_t MVAValue(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices,
                        FactorizedJetCorrector *iJetCorrector,
                        const PileupEnergyDensityCol *iPileupEnergyDensity,
                        Bool_t printDebug=false);

      //Corrected Jets
      Double_t MVAValue(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices,
                        Bool_t printDebug=false);


      double  correctedPt(const PFJet *iJet, FactorizedJetCorrector *iJetCorrector,
                          const PileupEnergyDensityCol *iPUEnergyDensity);

      Float_t                  fJetPtMin;
      Float_t                  fDZCut;

    protected:      
      TMVA::Reader            *fReader;
      TString                  fLowPtMethodName;
      TString                  fHighPtMethodName;
      MVAType                  fType;
      CutType                  fCutType;
      Bool_t                   fIsInitialized;
      Float_t                  fMVACut[4][4]; //Fix the cut array

      Float_t fNVtx     ;
      Float_t fJPt1     ;
      Float_t fJEta1    ;
      Float_t fJPhi1    ;
      Float_t fJD01     ;
      Float_t fJDZ1     ;
      Float_t fBeta     ;
      Float_t fBetaStar ;
      Float_t fNCharged ;
      Float_t fNNeutrals;
      Float_t fDRMean   ;
      Float_t fFrac01   ;
      Float_t fFrac02   ;
      Float_t fFrac03   ;
      Float_t fFrac04   ;
      Float_t fFrac05   ;

      ClassDef(JetIDMVA,0)
        };
}


#endif
Revision:	1.4
Committed:	Mon Apr 23 15:37:59 2012 UTC (13 years ago) by pharris
Content type:	text/plain
Branch:	MAIN
Changes since 1.3:	+2 -1 lines
Log Message:	52 MET update w/Covariance matrix
#	User	Rev	Content
1	pharris	1.1	//--------------------------------------------------------------------------------------------------
2			// $Id $
3			//
4			// JetIDMVA
5			//
6			// Helper Class for Jet Id MVA
7			//
8			// Authors: P. Harris
9			//--------------------------------------------------------------------------------------------------
10
11			#ifndef MITPHYSICS_UTILS_JetIDMVA_H
12			#define MITPHYSICS_UTILS_JetIDMVA_H
13
14			#include "CondFormats/JetMETObjects/interface/FactorizedJetCorrector.h"
15			#include "MitAna/DataTree/interface/PFJetFwd.h"
16			#include "MitAna/DataTree/interface/VertexFwd.h"
17			#include "MitAna/DataTree/interface/TrackFwd.h"
18			#include "MitAna/DataTree/interface/PFJet.h"
19			#include "MitAna/DataTree/interface/PFJetCol.h"
20			#include "MitAna/DataTree/interface/PFCandidateCol.h"
21			#include "MitAna/DataTree/interface/PileupEnergyDensityCol.h"
22			#include "MitCommon/MathTools/interface/MathUtils.h"
23
24			class TRandom3;
25			namespace TMVA {
26			class Reader;
27			}
28
29			namespace mithep {
30			class JetIDMVA {
31			public:
32			JetIDMVA();
33			~JetIDMVA();
34
35			enum MVAType {
36			kBaseline = 0
37			};
38
39	pharris	1.3	enum CutType {
40			kTight = 0,
41			kMedium = 1,
42	pharris	1.4	kLoose = 2,
43			kMET = 3
44	pharris	1.3	};
45
46			void Initialize(JetIDMVA::CutType iCutType,
47			TString iLowPtWeights ="$CMSSW_BASE/src/MitPhysics/data/mva_JetID_lowpt.weights.xml",
48			TString iHighPtWeights="$CMSSW_BASE/src/MitPhysics/data/mva_JetID_highpt.weights.xml",
49			JetIDMVA::MVAType iType=kBaseline,
50			TString iCutFileName ="$CMSSW_BASE/src/MitPhysics/Utils/python/JetIdParams_cfi.py");
51	pharris	1.1
52			Bool_t IsInitialized() const { return fIsInitialized; }
53			Double_t MVAValue(
54			Float_t iNPV ,
55			Float_t iJPt1 ,
56			Float_t iJEta1 ,
57			Float_t iJPhi1 ,
58			Float_t iJD01 ,
59			Float_t iJDZ1 ,
60	pharris	1.3	Float_t iBeta ,
61			Float_t iBetaStar,
62			Float_t iNCharged,
63			Float_t iNNeutrals,
64			Float_t iDRMean ,
65			Float_t iFrac01 ,
66			Float_t iFrac02 ,
67			Float_t iFrac03 ,
68			Float_t iFrac04 ,
69			Float_t iFrac05
70	pharris	1.1	);
71
72	pharris	1.2	//UNcorrected Jets
73	pharris	1.3	Bool_t pass(const PFJet iJet,const Vertex iVertex,const VertexCol *iVertices,
74	pharris	1.1	FactorizedJetCorrector *iJetCorrector,
75			const PileupEnergyDensityCol *iPileupEnergyDensity);
76	pharris	1.2
77			//Corrected Jets
78	pharris	1.3	Bool_t pass(const PFJet iJet,const Vertex iVertex,const VertexCol *iVertices);
79	pharris	1.2
80			//Uncorrected Jets
81	pharris	1.3	Double_t MVAValue(const PFJet iJet,const Vertex iVertex,const VertexCol *iVertices,
82	pharris	1.1	FactorizedJetCorrector *iJetCorrector,
83			const PileupEnergyDensityCol *iPileupEnergyDensity,
84			Bool_t printDebug=false);
85
86	pharris	1.2	//Corrected Jets
87	pharris	1.3	Double_t MVAValue(const PFJet iJet,const Vertex iVertex,const VertexCol *iVertices,
88	pharris	1.2	Bool_t printDebug=false);
89
90	pharris	1.1
91			double correctedPt(const PFJet iJet, FactorizedJetCorrector iJetCorrector,
92			const PileupEnergyDensityCol *iPUEnergyDensity);
93
94			Float_t fJetPtMin;
95	pharris	1.3	Float_t fDZCut;
96	pharris	1.1
97			protected:
98			TMVA::Reader *fReader;
99	pharris	1.3	TString fLowPtMethodName;
100			TString fHighPtMethodName;
101	pharris	1.1	MVAType fType;
102	pharris	1.3	CutType fCutType;
103	pharris	1.1	Bool_t fIsInitialized;
104	pharris	1.3	Float_t fMVACut[4][4]; //Fix the cut array
105
106			Float_t fNVtx ;
107			Float_t fJPt1 ;
108			Float_t fJEta1 ;
109			Float_t fJPhi1 ;
110			Float_t fJD01 ;
111			Float_t fJDZ1 ;
112			Float_t fBeta ;
113			Float_t fBetaStar ;
114			Float_t fNCharged ;
115			Float_t fNNeutrals;
116			Float_t fDRMean ;
117			Float_t fFrac01 ;
118			Float_t fFrac02 ;
119			Float_t fFrac03 ;
120			Float_t fFrac04 ;
121			Float_t fFrac05 ;
122
123	pharris	1.1	ClassDef(JetIDMVA,0)
124			};
125			}
126
127
128			#endif