17 |
|
#include "MitAna/DataTree/interface/Electron.h" |
18 |
|
#include "MitAna/DataTree/interface/ElectronCol.h" |
19 |
|
#include "MitCommon/MathTools/interface/MathUtils.h" |
20 |
– |
#include "MitPhysics/ElectronLikelihood/interface/ElectronLikelihood.h" |
21 |
– |
#include "MitPhysics/ElectronLikelihood/interface/LikelihoodSwitches.h" |
22 |
– |
#include "MitPhysics/ElectronLikelihood/interface/LikelihoodMeasurements.h" |
23 |
– |
|
20 |
|
|
21 |
|
class TRandom3; |
22 |
|
namespace TMVA { |
29 |
|
ElectronIDMVA(); |
30 |
|
~ElectronIDMVA(); |
31 |
|
|
32 |
+ |
enum MVAType { |
33 |
+ |
kBaseline = 0, // SigmaIEtaIEta, DEtaIn, DPhiIn, FBrem, SigmaIPhiIPhi, NBrem, OneOverEMinusOneOverP |
34 |
+ |
kV1, // kBaseline + EOverP, HoverE, ESeedClusterOverPout, ESeedClusterOverPIn |
35 |
+ |
kV2 // kV2 + d0 , IP3d, IP3dSig |
36 |
+ |
}; |
37 |
+ |
|
38 |
+ |
|
39 |
|
void Initialize(TString methodName, |
40 |
|
TString Subdet0Pt10To20Weights , |
41 |
|
TString Subdet1Pt10To20Weights , |
43 |
|
TString Subdet0Pt20ToInfWeights, |
44 |
|
TString Subdet1Pt20ToInfWeights, |
45 |
|
TString Subdet2Pt20ToInfWeights, |
46 |
< |
ElectronLikelihood *LH ); |
46 |
> |
ElectronIDMVA::MVAType type ); |
47 |
|
|
48 |
|
Bool_t IsInitialized() const { return fIsInitialized; } |
49 |
|
Double_t MVAValue(const Electron *ele, const Vertex *vertex); |
50 |
< |
|
50 |
> |
Double_t MVAValue(Double_t ElePt , Double_t EleSCEta, |
51 |
> |
Double_t EleSigmaIEtaIEta, |
52 |
> |
Double_t EleDEtaIn, |
53 |
> |
Double_t EleDPhiIn, |
54 |
> |
Double_t EleHoverE, |
55 |
> |
Double_t EleD0, |
56 |
> |
Double_t EleDZ, |
57 |
> |
Double_t EleFBrem, |
58 |
> |
Double_t EleEOverP, |
59 |
> |
Double_t EleESeedClusterOverPout, |
60 |
> |
Double_t EleSigmaIPhiIPhi, |
61 |
> |
Double_t EleNBrem, |
62 |
> |
Double_t EleOneOverEMinusOneOverP, |
63 |
> |
Double_t EleESeedClusterOverPIn, |
64 |
> |
Double_t EleIP3d, |
65 |
> |
Double_t EleIP3dSig ); |
66 |
> |
|
67 |
> |
|
68 |
|
protected: |
69 |
|
TMVA::Reader *fTMVAReader[6]; |
70 |
|
TString fMethodname; |
71 |
|
|
52 |
– |
ElectronLikelihood *fLH; //Likelihood |
72 |
|
Bool_t fIsInitialized; |
73 |
|
|
74 |
|
Float_t fMVAVar_EleSigmaIEtaIEta; |
86 |
|
Float_t fMVAVar_EleESeedClusterOverPIn; |
87 |
|
Float_t fMVAVar_EleIP3d; |
88 |
|
Float_t fMVAVar_EleIP3dSig; |
70 |
– |
Float_t fMVAVar_EleStandardLikelihood; |
89 |
|
|
90 |
|
|
91 |
|
ClassDef(ElectronIDMVA, 0) // Muon tools |