| 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 , |
| 122 |
|
const PFJetCol *iJets , |
| 123 |
|
int iNPV, |
| 124 |
|
Bool_t printDebug=false); |
| 125 |
< |
|
| 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) |
