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

Comparing UserCode/MitPhysics/Utils/interface/JetIDMVA.h (file contents):
Revision 1.3 by pharris, Sat Apr 7 09:36:32 2012 UTC vs.
Revision 1.9 by pharris, Tue Sep 25 15:39:15 2012 UTC

# Line 21 | Line 21
21   #include "MitAna/DataTree/interface/PileupEnergyDensityCol.h"
22   #include "MitCommon/MathTools/interface/MathUtils.h"
23  
24 + // for Rho definitons
25 + #include "MitPhysics/Utils/interface/RhoUtilities.h"
26 +
27   class TRandom3;
28   namespace TMVA {
29    class Reader;
# Line 33 | Line 36 | namespace mithep {
36        ~JetIDMVA();
37  
38        enum MVAType {
39 <        kBaseline = 0
39 >        kBaseline = 0,
40 >        k42       = 1,
41 >        k52       = 2,
42 >        kCut      = 3,
43 >        kQGP      = 4
44        };
45  
46        enum CutType {
47          kTight     = 0,
48          kMedium    = 1,
49 <        kLoose     = 2
49 >        kLoose     = 2,
50 >        kMET       = 3
51        };
52  
53        void     Initialize(JetIDMVA::CutType iCutType,
# Line 47 | Line 55 | namespace mithep {
55                            TString           iHighPtWeights="$CMSSW_BASE/src/MitPhysics/data/mva_JetID_highpt.weights.xml",
56                            JetIDMVA::MVAType iType=kBaseline,
57                            TString           iCutFileName  ="$CMSSW_BASE/src/MitPhysics/Utils/python/JetIdParams_cfi.py");
58 <      
58 >    
59        Bool_t   IsInitialized() const { return fIsInitialized; }
60        Double_t MVAValue(    
61                          Float_t iNPV    ,
# Line 65 | Line 73 | namespace mithep {
73                          Float_t iFrac02  ,
74                          Float_t iFrac03  ,
75                          Float_t iFrac04  ,
76 <                        Float_t iFrac05  
76 >                        Float_t iFrac05  ,
77 >                        Float_t iDR2Mean
78                          );
79 +      //Cut Based
80 +      Bool_t passCut(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices);
81  
82 +      Bool_t passPt(const PFJet *iJet, FactorizedJetCorrector *iJetCorrector=0,
83 +                    const PileupEnergyDensityCol *iPileupEnergyDensity=0,
84 +                    RhoUtilities::RhoType type=RhoUtilities::DEFAULT);
85        //UNcorrected Jets
86        Bool_t   pass(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices,
87                      FactorizedJetCorrector *iJetCorrector,
88 <                    const PileupEnergyDensityCol *iPileupEnergyDensity);
89 <      
88 >                    const PileupEnergyDensityCol *iPileupEnergyDensity,
89 >                    RhoUtilities::RhoType type = RhoUtilities::DEFAULT);
90 >
91        //Corrected Jets
92        Bool_t   pass(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices);
93                                          
# Line 82 | Line 97 | namespace mithep {
97                          const PileupEnergyDensityCol *iPileupEnergyDensity,
98                          Bool_t printDebug=false);
99  
100 +      Double_t* QGValue(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices, //Vertex here is the PV
101 +                        FactorizedJetCorrector *iJetCorrector,
102 +                        const PileupEnergyDensityCol *iPileupEnergyDensity,
103 +                        Bool_t printDebug);
104 +
105        //Corrected Jets
106        Double_t MVAValue(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices,
107                          Bool_t printDebug=false);
108  
109  
110        double  correctedPt(const PFJet *iJet, FactorizedJetCorrector *iJetCorrector,
111 <                          const PileupEnergyDensityCol *iPUEnergyDensity);
111 >                          const PileupEnergyDensityCol *iPUEnergyDensity,
112 >                          RhoUtilities::RhoType type = RhoUtilities::DEFAULT,int iId=-1);
113  
114        Float_t                  fJetPtMin;
115        Float_t                  fDZCut;
116  
117      protected:      
118        TMVA::Reader            *fReader;
119 +      TMVA::Reader            *fLowPtReader;
120        TString                  fLowPtMethodName;
121        TString                  fHighPtMethodName;
122        MVAType                  fType;
123        CutType                  fCutType;
124        Bool_t                   fIsInitialized;
125        Float_t                  fMVACut[4][4]; //Fix the cut array
126 <
126 >      Float_t                  fRMSCut[4][4];
127 >      Float_t                  fBetaStarCut[4][4];
128 >      
129        Float_t fNVtx     ;
130        Float_t fJPt1     ;
131        Float_t fJEta1    ;
# Line 112 | Line 136 | namespace mithep {
136        Float_t fBetaStar ;
137        Float_t fNCharged ;
138        Float_t fNNeutrals;
139 +      Float_t fNParticles;
140        Float_t fDRMean   ;
141 +      Float_t fPtD      ;
142        Float_t fFrac01   ;
143        Float_t fFrac02   ;
144        Float_t fFrac03   ;
145        Float_t fFrac04   ;
146        Float_t fFrac05   ;
147 +      Float_t fDR2Mean  ;
148  
149        ClassDef(JetIDMVA,0)
150          };

Diff Legend

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