ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitPhysics/Utils/interface/MVATools.h
(Generate patch)

Comparing UserCode/MitPhysics/Utils/interface/MVATools.h (file contents):
Revision 1.1 by mingyang, Thu Oct 13 17:29:19 2011 UTC vs.
Revision 1.9 by mingyang, Sun May 27 19:41:21 2012 UTC

# Line 25 | Line 25
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
# Line 35 | Line 35 | namespace TMVA {//MVA
35   namespace mithep {
36    class MVATools {
37      public:
38 <      MVATools();
39 <  
40 <  
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);
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,double MVAPtMin);
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(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;
74 <    float RelE2x5Right;
75 <    float RelE5x5;
76 <    
77 <    //MVA variables 2
78 <    float EtaWidth;
79 <    float PhiWidth;
80 <    float CoviEtaiPhi;
81 <    float CoviPhiiPhi;
82 <    float RelPreshowerEnergy;
83 <
84 <    //variables used to compute mva variables
85 <
86 <    Bool_t PassElecVeto;  
87 <
88 <    double ecalIso3;
89 <    double ecalIso4;
90 <    double hcalIso4;
91 <    
92 <    unsigned int wVtxInd;
93 <    
94 <    double trackIso1;
95 <    
96 <    // track iso only
97 <    double trackIso3;
98 <    
99 <    // track iso worst vtx
100 <    double trackIso2;
101 <    
102 <    double combIso1;
103 <    double combIso2;
104 <    
105 <    double RawEnergy;
106 <    
107 <    double dRTrack;
108 <
109 <    //spectator variables
110 <    double Pt_MVA;
111 <    double ScEta_MVA;
112 <  
113 <    Bool_t isbarrel;
114 <    
115 <    // check which category it is ...
116 <    int _tCat;
117 <    
118 <    //MVA  
119 <    Bool_t PassMVA;
120 <    TMVA::Reader *reader;
121 <    Float_t bdt;
122 <    Int_t PassElecVetoInt;
123 <
124 <    ClassDef(MVATools, 0) // Muon tools
125 <      };
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
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 >      //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

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines