25 |
|
#include "MitAna/DataTree/interface/SuperCluster.h" |
26 |
|
#include "MitAna/DataTree/interface/SuperClusterCol.h" |
27 |
|
#include "MitCommon/MathTools/interface/MathUtils.h" |
28 |
< |
|
28 |
> |
#include "MitAna/DataTree/interface/PFCandidateCol.h" |
29 |
|
|
30 |
|
class TRandom3; |
31 |
|
namespace TMVA {//MVA |
35 |
|
namespace mithep { |
36 |
|
class MVATools { |
37 |
|
public: |
38 |
< |
MVATools(); |
39 |
< |
|
38 |
> |
MVATools(); |
39 |
|
|
40 |
< |
//-------------------------- |
41 |
< |
//MVA |
42 |
< |
//-------------------------- |
43 |
< |
//void InitializeMVA(int VariableType, TString EndcapWeights,TString BarrelWeights); |
44 |
< |
//Bool_t PassMVASelection(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho,const ElectronCol *els); |
45 |
< |
|
46 |
< |
void InitializeMVA(int VariableType, TString EndcapWeights,TString BarrelWeights); |
47 |
< |
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); |
48 |
< |
Int_t PassElectronVetoInt(const Photon* p, const ElectronCol* els); |
49 |
< |
Float_t GetMVAbdtValue(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho,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,Float_t bdtCutBarrel, Float_t bdtCutEndcap, const ElectronCol* els=0, Bool_t applyElectronVeto=kTRUE); |
47 |
> |
Int_t PassElectronVetoInt(const Photon* p, const ElectronCol* els); |
48 |
> |
Float_t GetMVAbdtValue_2011(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho,const ElectronCol* els=0, Bool_t applyElectronVeto=kTRUE); |
49 |
> |
Float_t GetMVAbdtValue_2012_globe(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho, const PFCandidateCol *fPFCands,const ElectronCol* els=0, Bool_t applyElectronVeto=kTRUE); |
50 |
|
|
51 |
|
TMVA::Reader *fReaderEndcap; |
52 |
|
TMVA::Reader *fReaderBarrel; |
53 |
|
|
54 |
< |
//MVA variables 0 |
55 |
< |
Float_t HoE; |
56 |
< |
Float_t covIEtaIEta; |
57 |
< |
Float_t tIso1; |
58 |
< |
Float_t tIso3; |
59 |
< |
Float_t tIso2; |
60 |
< |
Float_t R9; |
61 |
< |
|
62 |
< |
//MVA variables 1 |
63 |
< |
float RelIsoEcal; |
64 |
< |
float RelIsoHcal; |
65 |
< |
|
66 |
< |
float RelEMax; |
67 |
< |
float RelETop; |
68 |
< |
float RelEBottom; |
69 |
< |
float RelELeft; |
70 |
< |
float RelERight; |
71 |
< |
float RelE2x5Max; |
72 |
< |
float RelE2x5Top; |
73 |
< |
float RelE2x5Bottom; |
74 |
< |
float RelE2x5Left; |
75 |
< |
float RelE2x5Right; |
76 |
< |
float RelE5x5; |
77 |
< |
|
78 |
< |
//MVA variables 2 |
79 |
< |
float EtaWidth; |
80 |
< |
float PhiWidth; |
81 |
< |
float CoviEtaiPhi; |
82 |
< |
float CoviPhiiPhi; |
83 |
< |
float RelPreshowerEnergy; |
84 |
< |
|
85 |
< |
//variables used to compute mva variables |
86 |
< |
|
87 |
< |
Bool_t PassElecVeto; |
88 |
< |
|
89 |
< |
double ecalIso3; |
90 |
< |
double ecalIso4; |
91 |
< |
double hcalIso4; |
92 |
< |
|
93 |
< |
unsigned int wVtxInd; |
94 |
< |
|
95 |
< |
double trackIso1; |
96 |
< |
|
97 |
< |
// track iso only |
98 |
< |
double trackIso3; |
99 |
< |
|
100 |
< |
// track iso worst vtx |
101 |
< |
double trackIso2; |
102 |
< |
|
103 |
< |
double combIso1; |
104 |
< |
double combIso2; |
105 |
< |
|
106 |
< |
double RawEnergy; |
107 |
< |
|
108 |
< |
double dRTrack; |
109 |
< |
|
110 |
< |
//spectator variables |
111 |
< |
double Pt_MVA; |
112 |
< |
double ScEta_MVA; |
113 |
< |
|
114 |
< |
Bool_t isbarrel; |
115 |
< |
|
116 |
< |
// check which category it is ... |
117 |
< |
int _tCat; |
118 |
< |
|
119 |
< |
//MVA |
120 |
< |
Bool_t PassMVA; |
121 |
< |
TMVA::Reader *reader; |
122 |
< |
Float_t bdt; |
123 |
< |
Int_t PassElecVetoInt; |
54 |
> |
//MVA Variables |
55 |
> |
float HoE; |
56 |
> |
float covIEtaIEta; |
57 |
> |
float tIso1abs; |
58 |
> |
float tIso3abs; |
59 |
> |
float tIso2abs; |
60 |
> |
float R9; |
61 |
> |
|
62 |
> |
float absIsoEcal; |
63 |
> |
float absIsoHcal; |
64 |
> |
float RelEMax; |
65 |
> |
float RelETop; |
66 |
> |
float RelEBottom; |
67 |
> |
float RelELeft; |
68 |
> |
float RelERight; |
69 |
> |
float RelE2x5Max; |
70 |
> |
float RelE2x5Top; |
71 |
> |
float RelE2x5Bottom; |
72 |
> |
float RelE2x5Left; |
73 |
> |
float RelE2x5Right; |
74 |
> |
float RelE5x5; |
75 |
> |
|
76 |
> |
float EtaWidth; |
77 |
> |
float PhiWidth; |
78 |
> |
float CoviEtaiPhi; |
79 |
> |
float CoviPhiiPhi; |
80 |
> |
|
81 |
> |
float NVertexes; |
82 |
> |
float RelPreshowerEnergy; |
83 |
> |
|
84 |
> |
//variable for v2 and v1 |
85 |
> |
float RelIsoEcal; |
86 |
> |
float RelIsoHcal; |
87 |
> |
float tIso1; |
88 |
> |
float tIso3; |
89 |
> |
float tIso2; |
90 |
> |
|
91 |
> |
float ScEta; |
92 |
> |
|
93 |
> |
//variables used to compute mva variables |
94 |
> |
|
95 |
> |
Bool_t PassElecVeto; |
96 |
> |
|
97 |
> |
double ecalIso3; |
98 |
> |
double ecalIso4; |
99 |
> |
double hcalIso4; |
100 |
> |
|
101 |
> |
unsigned int wVtxInd; |
102 |
> |
|
103 |
> |
double trackIso1; |
104 |
> |
|
105 |
> |
// track iso only |
106 |
> |
double trackIso3; |
107 |
> |
|
108 |
> |
// track iso worst vtx |
109 |
> |
double trackIso2; |
110 |
> |
|
111 |
> |
double combIso1; |
112 |
> |
double combIso2; |
113 |
> |
|
114 |
> |
double RawEnergy; |
115 |
> |
|
116 |
> |
double dRTrack; |
117 |
> |
|
118 |
> |
//spectator variables |
119 |
> |
double Pt_MVA; |
120 |
> |
double ScEta_MVA; |
121 |
> |
|
122 |
> |
Bool_t isbarrel; |
123 |
> |
|
124 |
> |
// check which category it is ... |
125 |
> |
int _tCat; |
126 |
|
|
127 |
< |
ClassDef(MVATools, 0) // Muon tools |
128 |
< |
}; |
127 |
> |
//1201 variable |
128 |
> |
float myphoton_pfchargedisogood03; |
129 |
> |
float myphoton_pfchargedisobad03; |
130 |
> |
float myphoton_pfphotoniso03; |
131 |
> |
float myphoton_sieie; |
132 |
> |
float myphoton_sieip; |
133 |
> |
float myphoton_etawidth; |
134 |
> |
float myphoton_phiwidth; |
135 |
> |
float myphoton_r9; |
136 |
> |
float myphoton_s4ratio; |
137 |
> |
float myphoton_SCeta; |
138 |
> |
float event_rho; |
139 |
> |
float myphoton_ESEffSigmaRR; |
140 |
> |
|
141 |
> |
//MVA |
142 |
> |
Bool_t PassMVA; |
143 |
> |
TMVA::Reader *reader; |
144 |
> |
Float_t bdt; |
145 |
> |
Int_t PassElecVetoInt; |
146 |
> |
|
147 |
> |
ClassDef(MVATools, 0) |
148 |
> |
}; |
149 |
|
} |
150 |
|
|
151 |
|
#endif |