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.2 by pharris, Wed Apr 4 07:57:00 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,
50 +        kMET       = 3
51 +      };
52  
53 <      void     Initialize(TString           iMethodName="JetIDMVA",
54 <                          TString           iWeights="$CMSSW_BASE/src/MitPhysics/data/mva_JetID.weights.xml",
55 <                          JetIDMVA::MVAType iType=kBaseline );
56 <      
53 >      void     Initialize(JetIDMVA::CutType iCutType,
54 >                          TString           iLowPtWeights ="$CMSSW_BASE/src/MitPhysics/data/mva_JetID_lowpt.weights.xml",
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 >    
59        Bool_t   IsInitialized() const { return fIsInitialized; }
60        Double_t MVAValue(    
61                          Float_t iNPV    ,
# Line 49 | Line 64 | namespace mithep {
64                          Float_t iJPhi1  ,
65                          Float_t iJD01   ,
66                          Float_t iJDZ1   ,
67 <                        Float_t iJM1    ,
68 <                        Float_t iNPart1 ,
69 <                        Float_t iLPt1   ,
70 <                        Float_t iLEta1  ,
71 <                        Float_t iLPhi1  ,
72 <                        Float_t iSPt1   ,
73 <                        Float_t iSEta1  ,
74 <                        Float_t iSPhi1  ,
75 <                        Float_t iNEPt1  ,
76 <                        Float_t iNEEta1 ,
77 <                        Float_t iNEPhi1 ,
63 <                        Float_t iEMPt1  ,
64 <                        Float_t iEMEta1 ,
65 <                        Float_t iEMPhi1 ,
66 <                        Float_t iChPt1  ,
67 <                        Float_t iChPhi1 ,
68 <                        Float_t iLFr1   ,
69 <                        Float_t iDRlC1  ,
70 <                        Float_t iDRLS1  ,
71 <                        Float_t iDRM1   ,
72 <                        Float_t iDRMNE1 ,
73 <                        Float_t iDREM1  ,
74 <                        Float_t iDRCH1  
67 >                        Float_t iBeta   ,
68 >                        Float_t iBetaStar,
69 >                        Float_t iNCharged,
70 >                        Float_t iNNeutrals,
71 >                        Float_t iDRMean  ,
72 >                        Float_t iFrac01  ,
73 >                        Float_t iFrac02  ,
74 >                        Float_t iFrac03  ,
75 >                        Float_t iFrac04  ,
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,
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);
92 >      Bool_t   pass(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices);
93                                          
94        //Uncorrected Jets
95 <      Double_t MVAValue(const PFJet *iJet,const Vertex *iVertex,
95 >      Double_t MVAValue(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices,
96                          FactorizedJetCorrector *iJetCorrector,
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,
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 <      TString                  fMethodName;
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 +      Float_t                  fRMSCut[4][4];
127 +      Float_t                  fBetaStarCut[4][4];
128        
129 <      Float_t fNPV;
130 <      Float_t fJPt1;
131 <      Float_t fJEta1;
132 <      Float_t fJPhi1;
133 <      Float_t fJD01 ;
134 <      Float_t fJDZ1 ;
135 <      Float_t fJM1  ;
136 <      Float_t fNPart1;
137 <      Float_t fLPt1 ;
138 <      Float_t fLEta1;
139 <      Float_t fLPhi1;
140 <      Float_t fSPt1 ;
141 <      Float_t fSEta1;
142 <      Float_t fSPhi1;
143 <      Float_t fNEPt1;
144 <      Float_t fNEEta1;
145 <      Float_t fNEPhi1;
146 <      Float_t fEMPt1;
147 <      Float_t fEMEta1;
148 <      Float_t fEMPhi1;
127 <      Float_t fChPt1;
128 <      Float_t fChPhi1;
129 <      Float_t fLFr1 ;
130 <      Float_t fDRLC1;
131 <      Float_t fDRLS1;
132 <      Float_t fDRM1 ;
133 <      Float_t fDRNE1;
134 <      Float_t fDREM1;
135 <      Float_t fDRCH1;
136 <        
129 >      Float_t fNVtx     ;
130 >      Float_t fJPt1     ;
131 >      Float_t fJEta1    ;
132 >      Float_t fJPhi1    ;
133 >      Float_t fJD01     ;
134 >      Float_t fJDZ1     ;
135 >      Float_t fBeta     ;
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          };
151   }

Diff Legend

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