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 |
|
}; |
43 |
|
|
44 |
|
enum CutType { |
45 |
|
kTight = 0, |
46 |
|
kMedium = 1, |
47 |
< |
kLoose = 2 |
47 |
> |
kLoose = 2, |
48 |
> |
kMET = 3 |
49 |
|
}; |
50 |
|
|
51 |
|
void Initialize(JetIDMVA::CutType iCutType, |
71 |
|
Float_t iFrac02 , |
72 |
|
Float_t iFrac03 , |
73 |
|
Float_t iFrac04 , |
74 |
< |
Float_t iFrac05 |
74 |
> |
Float_t iFrac05 , |
75 |
> |
Float_t iDR2Mean |
76 |
|
); |
77 |
|
|
78 |
|
//UNcorrected Jets |
79 |
|
Bool_t pass(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices, |
80 |
|
FactorizedJetCorrector *iJetCorrector, |
81 |
< |
const PileupEnergyDensityCol *iPileupEnergyDensity); |
82 |
< |
|
81 |
> |
const PileupEnergyDensityCol *iPileupEnergyDensity, |
82 |
> |
RhoUtilities::RhoType type = RhoUtilities::DEFAULT); |
83 |
> |
|
84 |
|
//Corrected Jets |
85 |
|
Bool_t pass(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices); |
86 |
|
|
96 |
|
|
97 |
|
|
98 |
|
double correctedPt(const PFJet *iJet, FactorizedJetCorrector *iJetCorrector, |
99 |
< |
const PileupEnergyDensityCol *iPUEnergyDensity); |
99 |
> |
const PileupEnergyDensityCol *iPUEnergyDensity, |
100 |
> |
RhoUtilities::RhoType type = RhoUtilities::DEFAULT); |
101 |
|
|
102 |
|
Float_t fJetPtMin; |
103 |
|
Float_t fDZCut; |
104 |
|
|
105 |
|
protected: |
106 |
|
TMVA::Reader *fReader; |
107 |
+ |
TMVA::Reader *fLowPtReader; |
108 |
|
TString fLowPtMethodName; |
109 |
|
TString fHighPtMethodName; |
110 |
|
MVAType fType; |
128 |
|
Float_t fFrac03 ; |
129 |
|
Float_t fFrac04 ; |
130 |
|
Float_t fFrac05 ; |
131 |
+ |
Float_t fDR2Mean ; |
132 |
|
|
133 |
|
ClassDef(JetIDMVA,0) |
134 |
|
}; |