16 |
|
#include "MitAna/DataTree/interface/TrackFwd.h" |
17 |
|
#include "MitAna/DataTree/interface/Electron.h" |
18 |
|
#include "MitAna/DataTree/interface/ElectronCol.h" |
19 |
+ |
#include "MitAna/DataTree/interface/PFCandidateCol.h" |
20 |
+ |
#include "MitAna/DataTree/interface/PileupEnergyDensityCol.h" |
21 |
|
#include "MitCommon/MathTools/interface/MathUtils.h" |
22 |
|
|
23 |
|
class TRandom3; |
34 |
|
enum MVAType { |
35 |
|
kBaseline = 0, // SigmaIEtaIEta, DEtaIn, DPhiIn, FBrem, SigmaIPhiIPhi, NBrem, OneOverEMinusOneOverP |
36 |
|
kNoIPInfo, // kBaseline + EOverP, ESeedClusterOverPout, ESeedClusterOverPIn |
37 |
< |
kWithIPInfo // kV2 + d0 , IP3d, IP3dSig |
37 |
> |
kWithIPInfo, // kBaseline + d0 , IP3d, IP3dSig |
38 |
> |
kIDIsoCombined // ID variables , PFIso03 , PFIso04 |
39 |
|
}; |
40 |
|
|
41 |
|
|
50 |
|
|
51 |
|
Bool_t IsInitialized() const { return fIsInitialized; } |
52 |
|
Double_t MVAValue(const Electron *ele, const Vertex *vertex); |
53 |
< |
Double_t MVAValue(Double_t ElePt , Double_t EleSCEta, |
53 |
> |
Double_t MVAValue(const Electron *ele, const Vertex *vertex, |
54 |
> |
const PFCandidateCol *PFCands, const PileupEnergyDensityCol *PileupEnergyDensity); |
55 |
> |
Double_t MVAValue(Double_t ElePt , Double_t EleEta, |
56 |
|
Double_t EleSigmaIEtaIEta, |
57 |
|
Double_t EleDEtaIn, |
58 |
|
Double_t EleDPhiIn, |
67 |
|
Double_t EleOneOverEMinusOneOverP, |
68 |
|
Double_t EleESeedClusterOverPIn, |
69 |
|
Double_t EleIP3d, |
70 |
< |
Double_t EleIP3dSig ); |
70 |
> |
Double_t EleIP3dSig); |
71 |
> |
Double_t MVAValue(Double_t ElePt , Double_t EleEta, |
72 |
> |
Double_t PileupEnergyDensity, |
73 |
> |
Double_t EleSigmaIEtaIEta, |
74 |
> |
Double_t EleDEtaIn, |
75 |
> |
Double_t EleDPhiIn, |
76 |
> |
Double_t EleHoverE, |
77 |
> |
Double_t EleD0, |
78 |
> |
Double_t EleDZ, |
79 |
> |
Double_t EleFBrem, |
80 |
> |
Double_t EleEOverP, |
81 |
> |
Double_t EleESeedClusterOverPout, |
82 |
> |
Double_t EleSigmaIPhiIPhi, |
83 |
> |
Double_t EleNBrem, |
84 |
> |
Double_t EleOneOverEMinusOneOverP, |
85 |
> |
Double_t EleESeedClusterOverPIn, |
86 |
> |
Double_t EleIP3d, |
87 |
> |
Double_t EleIP3dSig, |
88 |
> |
Double_t EleGsfTrackChi2OverNdof, |
89 |
> |
Double_t EledEtaCalo, |
90 |
> |
Double_t EledPhiCalo, |
91 |
> |
Double_t EleR9, |
92 |
> |
Double_t EleSCEtaWidth, |
93 |
> |
Double_t EleSCPhiWidth, |
94 |
> |
Double_t EleCovIEtaIPhi, |
95 |
> |
Double_t ElePreShowerOverRaw, |
96 |
> |
Double_t EleChargedIso03, |
97 |
> |
Double_t EleNeutralHadronIso03, |
98 |
> |
Double_t EleGammaIso03, |
99 |
> |
Double_t EleChargedIso04, |
100 |
> |
Double_t EleNeutralHadronIso04, |
101 |
> |
Double_t EleGammaIso04 ); |
102 |
|
|
103 |
|
|
104 |
|
protected: |
122 |
|
Float_t fMVAVar_EleESeedClusterOverPIn; |
123 |
|
Float_t fMVAVar_EleIP3d; |
124 |
|
Float_t fMVAVar_EleIP3dSig; |
125 |
< |
|
126 |
< |
|
125 |
> |
Float_t fMVAVar_EleGsfTrackChi2OverNdof; |
126 |
> |
Float_t fMVAVar_EledEtaCalo; |
127 |
> |
Float_t fMVAVar_EledPhiCalo; |
128 |
> |
Float_t fMVAVar_EleR9; |
129 |
> |
Float_t fMVAVar_EleSCEtaWidth; |
130 |
> |
Float_t fMVAVar_EleSCPhiWidth; |
131 |
> |
Float_t fMVAVar_EleCovIEtaIPhi; |
132 |
> |
Float_t fMVAVar_ElePreShowerOverRaw; |
133 |
> |
Float_t fMVAVar_EleChargedIso03OverPt; |
134 |
> |
Float_t fMVAVar_EleNeutralHadronIso03OverPt; |
135 |
> |
Float_t fMVAVar_EleGammaIso03OverPt; |
136 |
> |
Float_t fMVAVar_EleChargedIso04OverPt; |
137 |
> |
Float_t fMVAVar_EleNeutralHadronIso04OverPt; |
138 |
> |
Float_t fMVAVar_EleGammaIso04OverPt; |
139 |
> |
|
140 |
|
ClassDef(ElectronIDMVA, 0) // Muon tools |
141 |
|
}; |
142 |
|
} |