ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitPhysics/Utils/interface/MVAMet.h
Revision: 1.13
Committed: Tue Sep 25 15:39:15 2012 UTC (12 years, 7 months ago) by pharris
Content type: text/plain
Branch: MAIN
Changes since 1.12: +2 -0 lines
Log Message: 
Updated MVA Met tool

File Contents

# User Rev Content
1 pharris 1.1 //--------------------------------------------------------------------------------------------------
2     // $Id $
3     //
4     // Met Regression
5     //
6     // Authors: P. Harris
7     //--------------------------------------------------------------------------------------------------
8    
9     #ifndef MITPHYSICS_UTILS_MVAMet_H
10     #define MITPHYSICS_UTILS_MVAMet_H
11 pharris 1.6 #include <TMatrixD.h>
12 pharris 1.1 #include "MitAna/DataTree/interface/PFJetFwd.h"
13     #include "MitAna/DataTree/interface/VertexFwd.h"
14     #include "MitAna/DataTree/interface/TrackFwd.h"
15 pharris 1.12 #include "MitAna/DataTree/interface/MuonCol.h"
16     #include "MitAna/DataTree/interface/PFTauCol.h"
17     #include "MitAna/DataTree/interface/ElectronCol.h"
18 pharris 1.1 #include "MitAna/DataTree/interface/Met.h"
19     #include "MitAna/DataTree/interface/PFMet.h"
20     #include "MitAna/DataTree/interface/PFMetCol.h"
21     #include "MitAna/DataTree/interface/PFJet.h"
22     #include "MitAna/DataTree/interface/PFJetCol.h"
23     #include "MitAna/DataTree/interface/PFCandidateCol.h"
24     #include "MitAna/DataTree/interface/PileupEnergyDensityCol.h"
25     #include "MitCommon/MathTools/interface/MathUtils.h"
26    
27     #include "MitPhysics/Utils/interface/RecoilTools.h"
28 pharris 1.13 #include "MitPhysics/Utils/interface/MetLeptonTools.h"
29 pharris 1.9 #include "CondFormats/EgammaObjects/interface/GBRForest.h"
30 pharris 1.1
31     class TRandom3;
32 bendavid 1.8 class GBRForest;
33 pharris 1.1
34     namespace mithep {
35     class MVAMet {
36     public:
37     MVAMet();
38     ~MVAMet();
39     enum MVAType {
40     kBaseline = 0
41     };
42    
43 pharris 1.3 //void setVariables(TMVA::Reader *iReader,bool iScale);
44     //void Initialize( TString iU1MethodName="U1MVA",
45     // TString iPhiMethodName="PhiMVA",
46     // TString iJetMVAFile="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilPhiRegress_baseline.weights.xml",
47     // TString iU1Weights="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilRegress_baseline.weights.xml",
48     // TString iPhiWeights="$CMSSW_BASE/src/MitPhysics/data/mva_JetID.weights.xml",
49     // MVAType iType=kBaseline);
50     void Initialize(
51 pharris 1.4 TString iJetLowPtFile ="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilPhiRegress_baseline.weights.xml",
52     TString iJetHighPtFile="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilPhiRegress_baseline.weights.xml",
53     TString iJetCutFile ="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilPhiRegress_baseline.weights.xml",
54     TString iU1Weights ="$CMSSW_BASE/src/MitPhysics/data/gbrmet.root",
55     TString iPhiWeights ="$CMSSW_BASE/src/MitPhysics/data/gbrmetphi.root",
56 pharris 1.6 TString iCovU1Weights ="$CMSSW_BASE/src/MitPhysics/data/gbrcovu1_52.root",
57     TString iCovU2Weights ="$CMSSW_BASE/src/MitPhysics/data/gbrcovu2_52.root",
58 pharris 1.4 MVAMet::MVAType iType=kBaseline);
59 pharris 1.3
60 pharris 1.1 Bool_t IsInitialized() const { return fIsInitialized; }
61 pharris 1.11 Float_t* getVals();
62 pharris 1.3 Double_t evaluatePhi();
63     Double_t evaluateU1();
64 pharris 1.6 Double_t evaluateCovU1();
65     Double_t evaluateCovU2();
66 pharris 1.1 Double_t MVAValue( bool iPhi,
67     Float_t iPFSumEt,
68     Float_t iU ,
69     Float_t iUPhi ,
70     Float_t iTKSumEt,
71     Float_t iTKU ,
72     Float_t iTKUPhi ,
73     Float_t iNPSumEt,
74     Float_t iNPU ,
75     Float_t iNPUPhi ,
76     Float_t iPUSumEt,
77     Float_t iPUMet ,
78     Float_t iPUMetPhi,
79     Float_t iPCSumEt,
80     Float_t iPCU ,
81     Float_t iPCUPhi ,
82     Float_t iJSPt1 ,
83     Float_t iJSEta1 ,
84     Float_t iJSPhi1 ,
85     Float_t iJSPt2 ,
86     Float_t iJSEta2 ,
87     Float_t iJSPhi2 ,
88     Float_t iNJet ,
89     Float_t iNAllJet,
90     Float_t iNPV );
91    
92 pharris 1.11 Met GetMet( Bool_t iPhi,
93     Float_t iPtVis,Float_t iPhiVis,Float_t iSumEtVis,
94     Float_t iPtQ ,Float_t iPhiQ ,Float_t iSumEtQ,
95 pharris 1.1 const PFMet *iMet ,
96 pharris 1.4 const PFCandidateCol *iCands,
97     const Vertex *iVertex,const VertexCol *iVertices,
98 pharris 1.1 const PFJetCol *iJets ,
99     FactorizedJetCorrector *iJetCorrector,
100     const PileupEnergyDensityCol *iPileupEnergyDensity,
101     int iNPV,
102     Bool_t printDebug=false);
103    
104 pharris 1.11 Met GetMet( Bool_t iPhi,
105     Float_t iPtVis,Float_t iPhiVis,Float_t iSumEtVis,
106     Float_t iPtQ ,Float_t iPhiQ ,Float_t iSumEtQ,
107 pharris 1.3 const PFMet *iMet ,
108 pharris 1.4 const PFCandidateCol *iCands,
109 pharris 1.9 const Vertex *iVertex,const VertexCol *iVertices,Double_t iRho,
110 pharris 1.3 const PFJetCol *iJets ,
111     int iNPV,
112     Bool_t printDebug=false);
113    
114 pharris 1.1 Met GetMet( Bool_t iPhi,
115 pharris 1.11 Float_t iPt1,Float_t iPhi1,Float_t iEta1,Float_t iChargedFrac1,
116     Float_t iPt2,Float_t iPhi2,Float_t iEta2,Float_t iChargedFrac2,
117 pharris 1.1 const PFMet *iMet ,
118 pharris 1.4 const PFCandidateCol *iCands,
119     const Vertex *iVertex,const VertexCol *iVertices,
120 pharris 1.1 const PFJetCol *iJets ,
121     FactorizedJetCorrector *iJetCorrector,
122     const PileupEnergyDensityCol *iPUEnergyDensity,
123     int iNPV,
124 pharris 1.5 Bool_t printDebug=false);
125 pharris 1.3
126     Met GetMet( Bool_t iPhi,
127 pharris 1.11 Float_t iPt1,Float_t iPhi1,Float_t iEta1,Float_t iChargedFrac1,
128     Float_t iPt2,Float_t iPhi2,Float_t iEta2,Float_t iChargedFrac2,
129 pharris 1.3 const PFMet *iMet ,
130 pharris 1.4 const PFCandidateCol *iCands,
131 pharris 1.9 const Vertex *iVertex,const VertexCol *iVertices,Double_t iRho,
132 pharris 1.3 const PFJetCol *iJets ,
133     int iNPV,
134 pharris 1.5 Bool_t printDebug=false);
135 pharris 1.6
136 pharris 1.12 Met GetMet( const MuonCol *iMuons,const ElectronCol *iElectrons,const PFTauCol *iTaus,
137     const PFCandidateCol *iCands,const PFJetCol *iJets,const Vertex *iPV,const VertexCol *iVertices,const PFMetCol *iPFMet,
138     FactorizedJetCorrector *iJetCorrector,const PileupEnergyDensityCol* iPUEnergyDensity);
139    
140 pharris 1.9 TMatrixD* GetMetCovariance() { return fCov; }
141     Float_t GetSignificance () { return fSignificance;}
142     Float_t GetUncertainty () { return fUncertainty;}
143 pharris 1.1 RecoilTools *fRecoilTools;
144 pharris 1.3
145     protected:
146 pharris 1.1 TString fPhiMethodName;
147     TString fU1MethodName;
148 pharris 1.6 TString fCovU1MethodName;
149     TString fCovU2MethodName;
150 pharris 1.1 Bool_t fIsInitialized;
151     MVAType fType;
152 pharris 1.10 Bool_t f42;
153 pharris 1.1
154 pharris 1.7 Float_t fSumEt ;
155 pharris 1.1 Float_t fU ;
156     Float_t fUPhi ;
157     Float_t fTKSumEt;
158     Float_t fTKU ;
159     Float_t fTKUPhi ;
160     Float_t fNPSumEt;
161     Float_t fNPU ;
162     Float_t fNPUPhi ;
163     Float_t fPUSumEt;
164     Float_t fPUMet ;
165     Float_t fPUMetPhi ;
166     Float_t fPCSumEt;
167     Float_t fPCU ;
168     Float_t fPCUPhi ;
169     Float_t fJSPt1 ;
170     Float_t fJSEta1 ;
171     Float_t fJSPhi1 ;
172     Float_t fJSPt2 ;
173     Float_t fJSEta2 ;
174     Float_t fJSPhi2 ;
175     Float_t fNJet ;
176     Float_t fNAllJet;
177     Float_t fNPV ;
178     Float_t fUPhiMVA;
179 pharris 1.6 Float_t fUMVA;
180 pharris 1.3
181     Float_t* fPhiVals;
182     Float_t* fU1Vals;
183 pharris 1.6 Float_t* fCovVals;
184 pharris 1.3
185     GBRForest *fPhiReader;
186     GBRForest *fU1Reader;
187 pharris 1.6 GBRForest *fCovU1Reader;
188     GBRForest *fCovU2Reader;
189    
190 pharris 1.7 Float_t fSignificance;
191 pharris 1.9 Float_t fUncertainty;
192 pharris 1.7 TMatrixD *fCov;
193 pharris 1.6
194 pharris 1.12 Int_t fNMuons;
195     Int_t fNElectrons;
196     Int_t fNTaus;
197    
198 pharris 1.13 MetLeptonTools *fMetLeptonTools;
199 pharris 1.3 //TMVA::Reader* fPhiReader;
200     //TMVA::Reader* fU1Reader;
201 pharris 1.1 ClassDef(MVAMet,0)
202     };
203     }
204     #endif