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; |
36 |
|
~JetIDMVA(); |
37 |
|
|
38 |
|
enum MVAType { |
39 |
< |
kBaseline = 0 |
39 |
> |
kBaseline = 0, |
40 |
> |
k42 = 1, |
41 |
> |
k52 = 2, |
42 |
> |
kCut = 3 |
43 |
|
}; |
44 |
|
|
45 |
|
enum CutType { |
46 |
|
kTight = 0, |
47 |
|
kMedium = 1, |
48 |
< |
kLoose = 2 |
48 |
> |
kLoose = 2, |
49 |
> |
kMET = 3 |
50 |
|
}; |
51 |
|
|
52 |
|
void Initialize(JetIDMVA::CutType iCutType, |
72 |
|
Float_t iFrac02 , |
73 |
|
Float_t iFrac03 , |
74 |
|
Float_t iFrac04 , |
75 |
< |
Float_t iFrac05 |
75 |
> |
Float_t iFrac05 , |
76 |
> |
Float_t iDR2Mean |
77 |
|
); |
78 |
+ |
//Cut Based |
79 |
+ |
Bool_t passCut(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices); |
80 |
|
|
81 |
|
//UNcorrected Jets |
82 |
|
Bool_t pass(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices, |
83 |
|
FactorizedJetCorrector *iJetCorrector, |
84 |
< |
const PileupEnergyDensityCol *iPileupEnergyDensity); |
85 |
< |
|
84 |
> |
const PileupEnergyDensityCol *iPileupEnergyDensity, |
85 |
> |
RhoUtilities::RhoType type = RhoUtilities::DEFAULT); |
86 |
> |
|
87 |
|
//Corrected Jets |
88 |
|
Bool_t pass(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices); |
89 |
|
|
99 |
|
|
100 |
|
|
101 |
|
double correctedPt(const PFJet *iJet, FactorizedJetCorrector *iJetCorrector, |
102 |
< |
const PileupEnergyDensityCol *iPUEnergyDensity); |
102 |
> |
const PileupEnergyDensityCol *iPUEnergyDensity, |
103 |
> |
RhoUtilities::RhoType type = RhoUtilities::DEFAULT); |
104 |
|
|
105 |
|
Float_t fJetPtMin; |
106 |
|
Float_t fDZCut; |
107 |
|
|
108 |
|
protected: |
109 |
|
TMVA::Reader *fReader; |
110 |
+ |
TMVA::Reader *fLowPtReader; |
111 |
|
TString fLowPtMethodName; |
112 |
|
TString fHighPtMethodName; |
113 |
|
MVAType fType; |
114 |
|
CutType fCutType; |
115 |
|
Bool_t fIsInitialized; |
116 |
|
Float_t fMVACut[4][4]; //Fix the cut array |
117 |
< |
|
117 |
> |
Float_t fRMSCut[4][4]; |
118 |
> |
Float_t fBetaStarCut[4][4]; |
119 |
> |
|
120 |
|
Float_t fNVtx ; |
121 |
|
Float_t fJPt1 ; |
122 |
|
Float_t fJEta1 ; |
133 |
|
Float_t fFrac03 ; |
134 |
|
Float_t fFrac04 ; |
135 |
|
Float_t fFrac05 ; |
136 |
+ |
Float_t fDR2Mean ; |
137 |
|
|
138 |
|
ClassDef(JetIDMVA,0) |
139 |
|
}; |