35 |
|
namespace mithep { |
36 |
|
class MVATools { |
37 |
|
public: |
38 |
< |
MVATools(); |
38 |
> |
MVATools(); |
39 |
|
|
40 |
< |
|
41 |
< |
//-------------------------- |
42 |
< |
//MVA |
43 |
< |
//-------------------------- |
44 |
< |
//void InitializeMVA(int VariableType, TString EndcapWeights,TString BarrelWeights); |
45 |
< |
//Bool_t PassMVASelection(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho,const ElectronCol *els); |
46 |
< |
|
47 |
< |
void InitializeMVA(int VariableType, TString EndcapWeights,TString BarrelWeights); |
48 |
< |
Bool_t PassMVASelection(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho,const ElectronCol* els,double MVAPtMin, Float_t bdtCutBarrel, Float_t bdtCutEndcap, Bool_t applyEleVeto = true); |
49 |
< |
Int_t PassElectronVetoInt(const Photon* p, const ElectronCol* els); |
40 |
> |
|
41 |
> |
//-------------------------- |
42 |
> |
//MVA |
43 |
> |
//-------------------------- |
44 |
> |
|
45 |
> |
void InitializeMVA(int VariableType, TString EndcapWeights,TString BarrelWeights); |
46 |
> |
Bool_t PassMVASelection(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho,const ElectronCol* els,Float_t bdtCutBarrel, Float_t bdtCutEndcap); |
47 |
> |
Int_t PassElectronVetoInt(const Photon* p, const ElectronCol* els); |
48 |
|
Float_t GetMVAbdtValue(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho,const ElectronCol* els); |
49 |
|
|
50 |
|
TMVA::Reader *fReaderEndcap; |
51 |
|
TMVA::Reader *fReaderBarrel; |
52 |
|
|
53 |
< |
//MVA variables 0 |
54 |
< |
Float_t HoE; |
55 |
< |
Float_t covIEtaIEta; |
56 |
< |
Float_t tIso1; |
57 |
< |
Float_t tIso3; |
58 |
< |
Float_t tIso2; |
59 |
< |
Float_t R9; |
60 |
< |
|
61 |
< |
//MVA variables 1 |
62 |
< |
float RelIsoEcal; |
63 |
< |
float RelIsoHcal; |
64 |
< |
|
65 |
< |
float RelEMax; |
66 |
< |
float RelETop; |
67 |
< |
float RelEBottom; |
68 |
< |
float RelELeft; |
69 |
< |
float RelERight; |
70 |
< |
float RelE2x5Max; |
71 |
< |
float RelE2x5Top; |
72 |
< |
float RelE2x5Bottom; |
73 |
< |
float RelE2x5Left; |
76 |
< |
float RelE2x5Right; |
77 |
< |
float RelE5x5; |
78 |
< |
|
79 |
< |
//MVA variables 2 |
80 |
< |
float EtaWidth; |
81 |
< |
float PhiWidth; |
82 |
< |
float CoviEtaiPhi; |
83 |
< |
float CoviPhiiPhi; |
84 |
< |
float RelPreshowerEnergy; |
53 |
> |
//MVA Variables |
54 |
> |
float HoE; |
55 |
> |
float covIEtaIEta; |
56 |
> |
float tIso1abs; |
57 |
> |
float tIso3abs; |
58 |
> |
float tIso2abs; |
59 |
> |
float R9; |
60 |
> |
|
61 |
> |
float absIsoEcal; |
62 |
> |
float absIsoHcal; |
63 |
> |
float RelEMax; |
64 |
> |
float RelETop; |
65 |
> |
float RelEBottom; |
66 |
> |
float RelELeft; |
67 |
> |
float RelERight; |
68 |
> |
float RelE2x5Max; |
69 |
> |
float RelE2x5Top; |
70 |
> |
float RelE2x5Bottom; |
71 |
> |
float RelE2x5Left; |
72 |
> |
float RelE2x5Right; |
73 |
> |
float RelE5x5; |
74 |
|
|
75 |
< |
//variables used to compute mva variables |
76 |
< |
|
77 |
< |
Bool_t PassElecVeto; |
75 |
> |
float EtaWidth; |
76 |
> |
float PhiWidth; |
77 |
> |
float CoviEtaiPhi; |
78 |
> |
float CoviPhiiPhi; |
79 |
|
|
80 |
< |
double ecalIso3; |
81 |
< |
double ecalIso4; |
92 |
< |
double hcalIso4; |
93 |
< |
|
94 |
< |
unsigned int wVtxInd; |
95 |
< |
|
96 |
< |
double trackIso1; |
97 |
< |
|
98 |
< |
// track iso only |
99 |
< |
double trackIso3; |
100 |
< |
|
101 |
< |
// track iso worst vtx |
102 |
< |
double trackIso2; |
103 |
< |
|
104 |
< |
double combIso1; |
105 |
< |
double combIso2; |
106 |
< |
|
107 |
< |
double RawEnergy; |
108 |
< |
|
109 |
< |
double dRTrack; |
110 |
< |
|
111 |
< |
//spectator variables |
112 |
< |
double Pt_MVA; |
113 |
< |
double ScEta_MVA; |
114 |
< |
|
115 |
< |
Bool_t isbarrel; |
116 |
< |
|
117 |
< |
// check which category it is ... |
118 |
< |
int _tCat; |
119 |
< |
|
120 |
< |
//MVA |
121 |
< |
Bool_t PassMVA; |
122 |
< |
TMVA::Reader *reader; |
123 |
< |
Float_t bdt; |
124 |
< |
Int_t PassElecVetoInt; |
80 |
> |
float NVertexes; |
81 |
> |
float RelPreshowerEnergy; |
82 |
|
|
83 |
< |
ClassDef(MVATools, 0) // Muon tools |
84 |
< |
}; |
83 |
> |
//variable for v2 and v1 |
84 |
> |
float RelIsoEcal; |
85 |
> |
float RelIsoHcal; |
86 |
> |
float tIso1; |
87 |
> |
float tIso3; |
88 |
> |
float tIso2; |
89 |
> |
|
90 |
> |
//variables used to compute mva variables |
91 |
> |
|
92 |
> |
Bool_t PassElecVeto; |
93 |
> |
|
94 |
> |
double ecalIso3; |
95 |
> |
double ecalIso4; |
96 |
> |
double hcalIso4; |
97 |
> |
|
98 |
> |
unsigned int wVtxInd; |
99 |
> |
|
100 |
> |
double trackIso1; |
101 |
> |
|
102 |
> |
// track iso only |
103 |
> |
double trackIso3; |
104 |
> |
|
105 |
> |
// track iso worst vtx |
106 |
> |
double trackIso2; |
107 |
> |
|
108 |
> |
double combIso1; |
109 |
> |
double combIso2; |
110 |
> |
|
111 |
> |
double RawEnergy; |
112 |
> |
|
113 |
> |
double dRTrack; |
114 |
> |
|
115 |
> |
//spectator variables |
116 |
> |
double Pt_MVA; |
117 |
> |
double ScEta_MVA; |
118 |
> |
|
119 |
> |
Bool_t isbarrel; |
120 |
> |
|
121 |
> |
// check which category it is ... |
122 |
> |
int _tCat; |
123 |
> |
|
124 |
> |
//MVA |
125 |
> |
Bool_t PassMVA; |
126 |
> |
TMVA::Reader *reader; |
127 |
> |
Float_t bdt; |
128 |
> |
Int_t PassElecVetoInt; |
129 |
> |
|
130 |
> |
ClassDef(MVATools, 0) |
131 |
> |
}; |
132 |
|
} |
133 |
|
|
134 |
|
#endif |