ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitPhysics/Utils/interface/MVAMet.h
(Generate patch)

Comparing UserCode/MitPhysics/Utils/interface/MVAMet.h (file contents):
Revision 1.3 by pharris, Wed Apr 4 14:43:11 2012 UTC vs.
Revision 1.8 by bendavid, Tue Apr 24 15:37:14 2012 UTC

# Line 8 | Line 8
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"
# Line 22 | Line 22
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 {
# Line 43 | Line 43 | namespace mithep {
43      //                  TString iPhiWeights="$CMSSW_BASE/src/MitPhysics/data/mva_JetID.weights.xml",
44      //                  MVAType iType=kBaseline);
45      void    Initialize(
46 <                       TString iJetMVAFile="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilPhiRegress_baseline.weights.xml",
47 <                       TString iU1Weights ="$CMSSW_BASE/src/MitPhysics/data/gbrmet.root",
48 <                       TString iPhiWeights="$CMSSW_BASE/src/MitPhysics/data/gbrmetphi.root",
49 <                       MVAType iType=kBaseline);
46 >                       TString iJetLowPtFile ="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilPhiRegress_baseline.weights.xml",
47 >                       TString iJetHighPtFile="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilPhiRegress_baseline.weights.xml",
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      ,
# Line 79 | Line 85 | namespace mithep {
85  
86      Met GetMet(         Bool_t iPhi,Float_t iPtVis,Float_t iPhiVis,Float_t iSumEtVis,
87                          const PFMet            *iMet  ,
88 <                        const PFCandidateCol   *iCands,const Vertex *iVertex,
88 >                        const PFCandidateCol   *iCands,
89 >                        const Vertex *iVertex,const VertexCol *iVertices,
90                          const PFJetCol         *iJets ,
91                          FactorizedJetCorrector *iJetCorrector,
92                          const PileupEnergyDensityCol *iPileupEnergyDensity,
# Line 88 | Line 95 | namespace mithep {
95  
96      Met GetMet(         Bool_t iPhi,Float_t iPtVis,Float_t iPhiVis,Float_t iSumEtVis,
97                          const PFMet            *iMet  ,
98 <                        const PFCandidateCol   *iCands,const Vertex *iVertex,
98 >                        const PFCandidateCol   *iCands,
99 >                        const Vertex *iVertex,const VertexCol *iVertices,
100                          const PFJetCol         *iJets ,
101                          int iNPV,
102                          Bool_t printDebug=false);
# Line 97 | Line 105 | namespace mithep {
105                          Float_t iPt1,Float_t iPhi1,Float_t iEta1,
106                          Float_t iPt2,Float_t iPhi2,Float_t iEta2,
107                          const PFMet            *iMet  ,
108 <                        const PFCandidateCol   *iCands,const Vertex *iVertex,
108 >                        const PFCandidateCol   *iCands,
109 >                        const Vertex *iVertex,const VertexCol *iVertices,
110                          const PFJetCol         *iJets ,
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,
118                          Float_t iPt2,Float_t iPhi2,Float_t iEta2,
119                          const PFMet            *iMet  ,
120 <                        const PFCandidateCol   *iCands,const Vertex *iVertex,
120 >                        const PFCandidateCol   *iCands,
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;
# Line 145 | Line 160 | namespace mithep {
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)

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines