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.4 by pharris, Sat Apr 7 09:36:32 2012 UTC vs.
Revision 1.11 by pharris, Fri Jul 20 18:21:49 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 > #include "CondFormats/EgammaObjects/interface/GBRForest.h"
26  
27   class TRandom3;
28 + class GBRForest;
29  
30   namespace mithep {
31    class MVAMet {
# Line 48 | Line 49 | namespace mithep {
49                         TString iJetCutFile   ="$CMSSW_BASE/src/MitPhysics/data/mva_RecoilPhiRegress_baseline.weights.xml",
50                         TString iU1Weights    ="$CMSSW_BASE/src/MitPhysics/data/gbrmet.root",
51                         TString iPhiWeights   ="$CMSSW_BASE/src/MitPhysics/data/gbrmetphi.root",
52 +                       TString iCovU1Weights   ="$CMSSW_BASE/src/MitPhysics/data/gbrcovu1_52.root",
53 +                       TString iCovU2Weights   ="$CMSSW_BASE/src/MitPhysics/data/gbrcovu2_52.root",
54                         MVAMet::MVAType  iType=kBaseline);
55          
56      Bool_t   IsInitialized() const { return fIsInitialized; }
57 +    Float_t* getVals();
58      Double_t evaluatePhi();
59      Double_t evaluateU1();
60 +    Double_t evaluateCovU1();
61 +    Double_t evaluateCovU2();
62      Double_t MVAValue(  bool iPhi,
63                          Float_t iPFSumEt,
64                          Float_t iU      ,
# Line 79 | Line 85 | namespace mithep {
85                          Float_t iNAllJet,
86                          Float_t iNPV    );
87  
88 <    Met GetMet(         Bool_t iPhi,Float_t iPtVis,Float_t iPhiVis,Float_t iSumEtVis,
88 >    Met GetMet(         Bool_t iPhi,
89 >                        Float_t iPtVis,Float_t iPhiVis,Float_t iSumEtVis,
90 >                        Float_t iPtQ  ,Float_t iPhiQ  ,Float_t iSumEtQ,
91                          const PFMet            *iMet  ,
92                          const PFCandidateCol   *iCands,
93                          const Vertex *iVertex,const VertexCol *iVertices,
# Line 89 | Line 97 | namespace mithep {
97                          int iNPV,
98                          Bool_t printDebug=false);
99  
100 <    Met GetMet(         Bool_t iPhi,Float_t iPtVis,Float_t iPhiVis,Float_t iSumEtVis,
100 >    Met GetMet(         Bool_t iPhi,
101 >                        Float_t iPtVis,Float_t iPhiVis,Float_t iSumEtVis,
102 >                        Float_t iPtQ  ,Float_t iPhiQ  ,Float_t iSumEtQ,
103                          const PFMet            *iMet  ,
104                          const PFCandidateCol   *iCands,
105 <                        const Vertex *iVertex,const VertexCol *iVertices,
105 >                        const Vertex *iVertex,const VertexCol *iVertices,Double_t iRho,
106                          const PFJetCol         *iJets ,
107                          int iNPV,
108                          Bool_t printDebug=false);
109  
110      Met GetMet(         Bool_t iPhi,
111 <                        Float_t iPt1,Float_t iPhi1,Float_t iEta1,
112 <                        Float_t iPt2,Float_t iPhi2,Float_t iEta2,
111 >                        Float_t iPt1,Float_t iPhi1,Float_t iEta1,Float_t iChargedFrac1,
112 >                        Float_t iPt2,Float_t iPhi2,Float_t iEta2,Float_t iChargedFrac2,
113                          const PFMet            *iMet  ,
114                          const PFCandidateCol   *iCands,
115                          const Vertex *iVertex,const VertexCol *iVertices,
# Line 107 | Line 117 | namespace mithep {
117                          FactorizedJetCorrector *iJetCorrector,
118                          const PileupEnergyDensityCol *iPUEnergyDensity,
119                          int iNPV,
120 <                        Bool_t printDebug);
120 >                        Bool_t printDebug=false);
121  
122      Met GetMet(         Bool_t iPhi,
123 <                        Float_t iPt1,Float_t iPhi1,Float_t iEta1,
124 <                        Float_t iPt2,Float_t iPhi2,Float_t iEta2,
123 >                        Float_t iPt1,Float_t iPhi1,Float_t iEta1,Float_t iChargedFrac1,
124 >                        Float_t iPt2,Float_t iPhi2,Float_t iEta2,Float_t iChargedFrac2,
125                          const PFMet            *iMet  ,
126                          const PFCandidateCol   *iCands,
127 <                        const Vertex *iVertex,const VertexCol *iVertices,
127 >                        const Vertex *iVertex,const VertexCol *iVertices,Double_t iRho,
128                          const PFJetCol         *iJets ,
129                          int iNPV,
130 <                        Bool_t printDebug);
131 <    
130 >                        Bool_t printDebug=false);
131 >
132 >    TMatrixD*   GetMetCovariance() { return fCov;         }
133 >    Float_t     GetSignificance () { return fSignificance;}
134 >    Float_t     GetUncertainty  () { return fUncertainty;}
135      RecoilTools *fRecoilTools;
136      
137    protected:
138      TString      fPhiMethodName;
139      TString      fU1MethodName;
140 +    TString      fCovU1MethodName;
141 +    TString      fCovU2MethodName;
142      Bool_t       fIsInitialized;
143      MVAType      fType;
144 +    Bool_t       f42;
145      
146 +    Float_t fSumEt  ;
147      Float_t fU      ;
148      Float_t fUPhi   ;
149      Float_t fTKSumEt;
# Line 151 | Line 168 | namespace mithep {
168      Float_t fNAllJet;
169      Float_t fNPV    ;
170      Float_t fUPhiMVA;
171 +    Float_t fUMVA;
172      
173      Float_t* fPhiVals;
174      Float_t* fU1Vals;
175 <    
175 >    Float_t* fCovVals;
176      
177      GBRForest *fPhiReader;
178      GBRForest *fU1Reader;
179 +    GBRForest *fCovU1Reader;
180 +    GBRForest *fCovU2Reader;
181 +
182 +    Float_t   fSignificance;
183 +    Float_t   fUncertainty;
184 +    TMatrixD *fCov;
185 +
186      //TMVA::Reader* fPhiReader;
187      //TMVA::Reader* fU1Reader;
188      ClassDef(MVAMet,0)

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines