ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

root/cvsroot/UserCode/MitPhysics/Utils/interface/MVATools.h

Comparing UserCode/MitPhysics/Utils/interface/MVATools.h (file contents):
Revision 1.11 by mingyang, Sun Jun 10 16:29:09 2012 UTC vs.
Revision 1.14 by mingyang, Tue Jul 30 21:09:10 2013 UTC

#	Line 34 | Line 34 | namespace TMVA {//MVA
34
35		namespace mithep {
36		class MVATools {
37	<	public:
37	>	public:
38		MVATools();
39	<
39	>
40	>	// MVA Typles
41	>	enum IdMVAType {
42	>	kNone      = 0,
43	>	k2011IdMVA,
44	>	k2012IdMVA_globe,
45	>	k2012IdMVA,        // same as above, but more logical name...
46	>	k2013FinalIdMVA,
47	>	k2011IdMVA_HZg
48	>	};
49
50		//--------------------------
51		//MVA
52		//--------------------------
53
54	<	void InitializeMVA(int VariableType, TString EndcapWeights,TString BarrelWeights);
55	<	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
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;
54	>	//void InitializeMVA(int VariableType, TString EndcapWeights,TString BarrelWeights);
55	>	void InitializeMVA(IdMVAType type);
56
57	<	float NVertexes;
58	<	float RelPreshowerEnergy;
59	<
60	<	//variable for v2 and v1
61	<	float RelIsoEcal;
62	<	float RelIsoHcal;
63	<	float tIso1;
64	<	float tIso3;
65	<	float tIso2;
66	<
67	<	float ScEta;
68	<
69	<	//variables used to compute mva variables
70	<
71	<	Bool_t PassElecVeto;
72	<
73	<	double ecalIso3;
74	<	double ecalIso4;
75	<	double hcalIso4;
76	<
77	<	unsigned int wVtxInd;
78	<
79	<	double trackIso1;
80	<
81	<	// track iso only
82	<	double trackIso3;
83	<
84	<	// track iso worst vtx
85	<	double trackIso2;
86	<
87	<	double combIso1;
88	<	double combIso2;
89	<
90	<	double RawEnergy;
91	<
92	<	double dRTrack;
93	<
94	<	//spectator variables
95	<	double Pt_MVA;
96	<	double ScEta_MVA;
97	<
98	<	Bool_t isbarrel;
99	<
100	<	// check which category it is ...
101	<	int _tCat;
57	>	// removed this (fab): if a mod want to cuty on BDT values, compute them in the mod using GetMVAbdtValue(..) and make the cut in the mod
58	>	//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);
59	>	// removed this (fab): no needed naywhere...
60	>	//Int_t  PassElectronVetoInt(const Photon* p, const ElectronCol* els);
61	>
62	>	Double_t GetMVAbdtValue   (const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho, const PFCandidateCol *fPFCands=NULL ,const ElectronCol* els=0, Bool_t applyElectronVeto=kTRUE);
63	>
64	>	// these we can remove at some point
65	>	/*     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); */
66	>	/*     Float_t GetMVAbdtValue_2012_globe(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho, const PFCandidateCol *fPFCands=NULL ,const ElectronCol* els=0, Bool_t applyElectronVeto=kTRUE); */
67	>
68	>
69	>	private:
70	>
71	>	TMVA::Reader *fReaderEndcap;
72	>	TMVA::Reader *fReaderBarrel;
73	>
74	>	IdMVAType fMVAType;
75	>	std::vector<float> mvaVars;
76	>	std::vector<std::string> varNames;
77	>	std::map<std::string,unsigned int> mvaVarMapEB;
78	>	std::map<std::string,unsigned int> mvaVarMapEE;
79	>
80	>	// -------------------------------------------------
81	>	// fab: these guys should all go away...
82	>	//MVA Variables
83	>	/*     float HoE; */
84	>	/*     float covIEtaIEta; */
85	>	/*     float tIso1abs; */
86	>	/*     float tIso3abs; */
87	>	/*     float tIso2abs; */
88	>	/*     float R9; */
89	>
90	>	/*     float absIsoEcal; */
91	>	/*     float absIsoHcal; */
92	>	/*     float RelEMax; */
93	>	/*     float RelETop; */
94	>	/*     float RelEBottom; */
95	>	/*     float RelELeft; */
96	>	/*     float RelERight; */
97	>	/*     float RelE2x5Max; */
98	>	/*     float RelE2x5Top; */
99	>	/*     float RelE2x5Bottom; */
100	>	/*     float RelE2x5Left; */
101	>	/*     float RelE2x5Right; */
102	>	/*     float RelE5x5; */
103	>
104	>	/*     float EtaWidth; */
105	>	/*     float PhiWidth; */
106	>	/*     float CoviEtaiPhi; */
107	>	/*     float CoviPhiiPhi; */
108	>
109	>	/*     float NVertexes; */
110	>	/*     float RelPreshowerEnergy; */
111	>
112	>	/*     //variable for v2 and v1 */
113	>	/*     float RelIsoEcal; */
114	>	/*     float RelIsoHcal; */
115	>	/*     float tIso1; */
116	>	/*     float tIso3; */
117	>	/*     float tIso2; */
118	>
119	>	/*     float ScEta; */
120	>
121	>	/*     //variables used to compute mva variables */
122	>
123	>	/*     Bool_t PassElecVeto;   */
124	>
125	>	/*     double ecalIso3; */
126	>	/*     double ecalIso4; */
127	>	/*     double hcalIso4;  */
128	>
129	>	/*     unsigned int wVtxInd; */
130	>
131	>	/*     double trackIso1; */
132	>
133	>	/*     // track iso only */
134	>	/*     double trackIso3; */
135	>
136	>	/*     // track iso worst vtx */
137	>	/*     double trackIso2;  */
138	>
139	>	/*     double combIso1; */
140	>	/*     double combIso2; */
141	>
142	>	/*     double RawEnergy; */
143	>
144	>	/*     double dRTrack; */
145	>
146	>	/*     //spectator variables */
147	>	/*     double Pt_MVA; */
148	>	/*     double ScEta_MVA; */
149	>
150	>	/*     Bool_t isbarrel; */
151	>
152	>	/*     // check which category it is ... */
153	>	/*     int _tCat; */
154	>
155	>	/*     //1201 variable */
156	>	/*     float myphoton_pfchargedisogood03; */
157	>	/*     float myphoton_pfchargedisobad03;  */
158	>	/*     float myphoton_pfphotoniso03; */
159	>	/*     float myphoton_sieie;  */
160	>	/*     float myphoton_sieip; */
161	>	/*     float myphoton_etawidth;  */
162	>	/*     float myphoton_phiwidth; */
163	>	/*     float myphoton_r9;  */
164	>	/*     float myphoton_s4ratio; */
165	>	/*     float myphoton_SCeta;  */
166	>	/*     float event_rho; */
167	>	/*     float myphoton_ESEffSigmaRR; */
168	>
169	>	//MVA
170	>	/*     Bool_t PassMVA; */
171	>	/*     //Float_t bdt;   -> removed this, we should not have this a memeber variable !!! (fab) */
172	>	/*     Int_t PassElecVetoInt; */
173
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;
174
175	<	//MVA
176	<	Bool_t PassMVA;
177	<	TMVA::Reader *reader;
144	<	Float_t bdt;
145	<	Int_t PassElecVetoInt;
146	<
147	<	ClassDef(MVATools, 0)
148	<	};
175	>	TMVA::Reader *reader;
176	>	ClassDef(MVATools, 0)
177	>	};
178		}
179
180		#endif

Diff Legend

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