8 |
|
|
9 |
|
#ifndef MITPHYSICS_UTILS_MVAMet_H |
10 |
|
#define MITPHYSICS_UTILS_MVAMet_H |
11 |
< |
|
11 |
> |
#include <TMatrixD.h> |
12 |
|
#include "MitAna/DataTree/interface/PFJetFwd.h" |
13 |
|
#include "MitAna/DataTree/interface/VertexFwd.h" |
14 |
|
#include "MitAna/DataTree/interface/TrackFwd.h" |
22 |
|
#include "MitCommon/MathTools/interface/MathUtils.h" |
23 |
|
|
24 |
|
#include "MitPhysics/Utils/interface/RecoilTools.h" |
25 |
– |
#include "MitPhysics/Utils/interface/GBRForest.h" |
25 |
|
|
26 |
|
class TRandom3; |
27 |
+ |
class GBRForest; |
28 |
|
|
29 |
|
namespace mithep { |
30 |
|
class MVAMet { |
48 |
|
TString iJetCutFile ="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilPhiRegress_baseline.weights.xml", |
49 |
|
TString iU1Weights ="$CMSSW_BASE/src/MitPhysics/data/gbrmet.root", |
50 |
|
TString iPhiWeights ="$CMSSW_BASE/src/MitPhysics/data/gbrmetphi.root", |
51 |
+ |
TString iCovU1Weights ="$CMSSW_BASE/src/MitPhysics/data/gbrcovu1_52.root", |
52 |
+ |
TString iCovU2Weights ="$CMSSW_BASE/src/MitPhysics/data/gbrcovu2_52.root", |
53 |
|
MVAMet::MVAType iType=kBaseline); |
54 |
|
|
55 |
|
Bool_t IsInitialized() const { return fIsInitialized; } |
56 |
|
Double_t evaluatePhi(); |
57 |
|
Double_t evaluateU1(); |
58 |
+ |
Double_t evaluateCovU1(); |
59 |
+ |
Double_t evaluateCovU2(); |
60 |
|
Double_t MVAValue( bool iPhi, |
61 |
|
Float_t iPFSumEt, |
62 |
|
Float_t iU , |
111 |
|
FactorizedJetCorrector *iJetCorrector, |
112 |
|
const PileupEnergyDensityCol *iPUEnergyDensity, |
113 |
|
int iNPV, |
114 |
< |
Bool_t printDebug); |
114 |
> |
Bool_t printDebug=false); |
115 |
|
|
116 |
|
Met GetMet( Bool_t iPhi, |
117 |
|
Float_t iPt1,Float_t iPhi1,Float_t iEta1, |
121 |
|
const Vertex *iVertex,const VertexCol *iVertices, |
122 |
|
const PFJetCol *iJets , |
123 |
|
int iNPV, |
124 |
< |
Bool_t printDebug); |
125 |
< |
|
124 |
> |
Bool_t printDebug=false); |
125 |
> |
|
126 |
> |
TMatrixD* GetMetCovariance() { return fCov; } |
127 |
> |
double GetSignificance () { return fSignificance;} |
128 |
|
RecoilTools *fRecoilTools; |
129 |
|
|
130 |
|
protected: |
131 |
|
TString fPhiMethodName; |
132 |
|
TString fU1MethodName; |
133 |
+ |
TString fCovU1MethodName; |
134 |
+ |
TString fCovU2MethodName; |
135 |
|
Bool_t fIsInitialized; |
136 |
|
MVAType fType; |
137 |
|
|
138 |
+ |
Float_t fSumEt ; |
139 |
|
Float_t fU ; |
140 |
|
Float_t fUPhi ; |
141 |
|
Float_t fTKSumEt; |
160 |
|
Float_t fNAllJet; |
161 |
|
Float_t fNPV ; |
162 |
|
Float_t fUPhiMVA; |
163 |
+ |
Float_t fUMVA; |
164 |
|
|
165 |
|
Float_t* fPhiVals; |
166 |
|
Float_t* fU1Vals; |
167 |
< |
|
167 |
> |
Float_t* fCovVals; |
168 |
|
|
169 |
|
GBRForest *fPhiReader; |
170 |
|
GBRForest *fU1Reader; |
171 |
+ |
GBRForest *fCovU1Reader; |
172 |
+ |
GBRForest *fCovU2Reader; |
173 |
+ |
|
174 |
+ |
Float_t fSignificance; |
175 |
+ |
TMatrixD *fCov; |
176 |
+ |
|
177 |
|
//TMVA::Reader* fPhiReader; |
178 |
|
//TMVA::Reader* fU1Reader; |
179 |
|
ClassDef(MVAMet,0) |