HiggsAnalysisTools/include/ElectronIDMVA.h at master · VecbosApp/HiggsAnalysisTools · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
//--------------------------------------------------------------------------------------------------
// $Id $
//
// ElectronIDMVA
//
// Helper Class for applying MVA electron ID selection
//
// Authors: S.Xie
//--------------------------------------------------------------------------------------------------

#ifndef HIGGSANALYSIS_ElectronIDMVA_H
#define HIGGSANALYSIS_ElectronIDMVA_H

#include "TMVA/Tools.h"
#include "TMVA/Reader.h"

class ElectronIDMVA {
  public:
    ElectronIDMVA();
    ~ElectronIDMVA();

    enum MVAType {
      kBaseline = 0,      // SigmaIEtaIEta, DEtaIn, DPhiIn, FBrem, SigmaIPhiIPhi, NBrem,
                          // OneOverEMinusOneOverP
      kNoIPInfo,          // kBaseline + EOverP, ESeedClusterOverPout, ESeedClusterOverPIn
      kWithIPInfo         // kV2 + d0 , IP3d, IP3dSig
    };

    void   Initialize(std::string methodName,
                      std::string Subdet0Pt10To20Weights ,
                      std::string Subdet1Pt10To20Weights ,
                      std::string Subdet2Pt10To20Weights,
                      std::string Subdet0Pt20ToInfWeights,
                      std::string Subdet1Pt20ToInfWeights,
                      std::string Subdet2Pt20ToInfWeights,
                      ElectronIDMVA::MVAType type );
    Bool_t IsInitialized() const { return fIsInitialized; }

    double MVAValue(double ElePt , double EleSCEta,
                    double EleSigmaIEtaIEta,
                    double EleDEtaIn,
                    double EleDPhiIn,
                    double EleHoverE,
                    double EleD0,
                    double EleFBrem,
                    double EleEOverP,
                    double EleESeedClusterOverPout,
                    double EleSigmaIPhiIPhi,
                    double EleNBrem,
                    double EleOneOverEMinusOneOverP,
                    double EleESeedClusterOverPIn,
                    double EleIP3d,
                    double EleIP3dSig );


  protected:
    TMVA::Reader             *fTMVAReader[6];
    std::string               fMethodname;
    MVAType                   fMVAType;

    Bool_t                    fIsInitialized;

    Float_t                   fMVAVar_EleSigmaIEtaIEta;
    Float_t                   fMVAVar_EleDEtaIn;
    Float_t                   fMVAVar_EleDPhiIn;
    Float_t                   fMVAVar_EleHoverE;
    Float_t                   fMVAVar_EleD0;
    Float_t                   fMVAVar_EleFBrem;
    Float_t                   fMVAVar_EleEOverP;
    Float_t                   fMVAVar_EleESeedClusterOverPout;
    Float_t                   fMVAVar_EleSigmaIPhiIPhi;
    Float_t                   fMVAVar_EleNBrem;
    Float_t                   fMVAVar_EleOneOverEMinusOneOverP;
    Float_t                   fMVAVar_EleESeedClusterOverPIn;
    Float_t                   fMVAVar_EleIP3d;
    Float_t                   fMVAVar_EleIP3dSig;

};

#endif