ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

root/cvsroot/UserCode/MitPhysics/Utils/src/ElectronIDMVA.cc

Comparing UserCode/MitPhysics/Utils/src/ElectronIDMVA.cc (file contents):
Revision 1.2 by sixie, Sun Sep 25 15:03:41 2011 UTC vs.
Revision 1.17 by ceballos, Sat Apr 28 07:10:42 2012 UTC

#	Line 15 | Line 15 | using namespace mithep;
15		//--------------------------------------------------------------------------------------------------
16		ElectronIDMVA::ElectronIDMVA() :
17		fMethodname("BDTG method"),
18	<	fLH(0),
19	<	fIsInitialized(kFALSE)
18	>	fIsInitialized(kFALSE),
19	>	fMVAType(ElectronIDMVA::kUninitialized),
20	>	fUseBinnedVersion(kTRUE),
21	>	fNMVABins(0)
22		{
23		// Constructor.
22	–	for(UInt_t i=0; i<6; ++i) {
23	–	fTMVAReader[i] = 0;
24	–	}
24		}
25
26
27		//--------------------------------------------------------------------------------------------------
28		ElectronIDMVA::~ElectronIDMVA()
29		{
30	<	for(UInt_t i=0; i<6; ++i) {
30	>	for(UInt_t i=0; i<fTMVAReader.size(); ++i) {
31		if (fTMVAReader[i]) delete fTMVAReader[i];
32		}
33		}
34
35		//--------------------------------------------------------------------------------------------------
36	+	void ElectronIDMVA::Initialize( std::string methodName,
37	+	std::string weightsfile,
38	+	ElectronIDMVA::MVAType type)
39	+	{
40	+
41	+	std::vector<std::string> tempWeightFileVector;
42	+	tempWeightFileVector.push_back(weightsfile);
43	+	Initialize(methodName,type,kFALSE,tempWeightFileVector);
44	+	}
45	+
46	+	//--------------------------------------------------------------------------------------------------
47		void ElectronIDMVA::Initialize( TString methodName,
48		TString Subdet0Pt10To20Weights ,
49		TString Subdet1Pt10To20Weights ,
#	Line 41 | Line 51 | void ElectronIDMVA::Initialize( TString
51		TString Subdet0Pt20ToInfWeights,
52		TString Subdet1Pt20ToInfWeights,
53		TString Subdet2Pt20ToInfWeights,
54	<	ElectronLikelihood *LH) {
54	>	ElectronIDMVA::MVAType type) {
55	>
56	>	std::vector<std::string> tempWeightFileVector;
57	>	tempWeightFileVector.push_back(std::string(Subdet0Pt10To20Weights.Data()));
58	>	tempWeightFileVector.push_back(std::string(Subdet1Pt10To20Weights.Data()));
59	>	tempWeightFileVector.push_back(std::string(Subdet2Pt10To20Weights.Data()));
60	>	tempWeightFileVector.push_back(std::string(Subdet0Pt20ToInfWeights.Data()));
61	>	tempWeightFileVector.push_back(std::string(Subdet1Pt20ToInfWeights.Data()));
62	>	tempWeightFileVector.push_back(std::string(Subdet2Pt20ToInfWeights.Data()));
63	>	Initialize(std::string(methodName.Data()),type,kTRUE,tempWeightFileVector);
64	>
65	>	}
66	>
67	>
68	>	//--------------------------------------------------------------------------------------------------
69	>	void ElectronIDMVA::Initialize(  std::string methodName,
70	>	ElectronIDMVA::MVAType type,
71	>	Bool_t useBinnedVersion,
72	>	std::vector<std::string> weightsfiles
73	>
74	>	) {
75	>
76	>	//clean up first
77	>	for (uint i=0;i<fTMVAReader.size(); ++i) {
78	>	if (fTMVAReader[i]) delete fTMVAReader[i];
79	>	}
80	>	fTMVAReader.clear();
81
82	+	//initialize
83		fIsInitialized = kTRUE;
47	–
84		fMethodname = methodName;
85	<	fLH = LH;
86	<	if (!fLH) { std::cout << "Error: Likelihood is not properly initialized.\n"; assert(fLH); }
85	>	fMVAType = type;
86	>	fUseBinnedVersion = useBinnedVersion;
87	>
88	>	//Define expected number of bins
89	>	UInt_t ExpectedNBins = 0;
90	>	if (!fUseBinnedVersion) {
91	>	ExpectedNBins = 1;
92	>	} else if (type == kBaseline
93	>	||type == kNoIPInfo
94	>	||type == kWithIPInfo
95	>	||type == kIDIsoCombined) {
96	>	ExpectedNBins = 6;
97	>	} else if (type == kIDEGamma2012TrigV0 ||
98	>	type == kIDEGamma2012NonTrigV0 ||
99	>	type == kIDHWW2012TrigV0) {
100	>	ExpectedNBins = 6;
101	>	} else if (type == kIsoRingsV0) {
102	>	ExpectedNBins = 4;
103	>	}
104	>	fNMVABins = ExpectedNBins;
105	>
106	>	//Check number of weight files given
107	>	if (fNMVABins != weightsfiles.size() ) {
108	>	std::cout << "Error: Expected Number of bins = " << fNMVABins << " does not equal to weightsfiles.size() = "
109	>	<< weightsfiles.size() << std::endl;
110	>	assert(fNMVABins == weightsfiles.size());
111	>	}
112	>
113	>
114	>	for(UInt_t i=0; i<fNMVABins; ++i) {
115	>	TMVA::Reader *tmpTMVAReader = new TMVA::Reader( "!Color:!Silent:Error" );
116	>	tmpTMVAReader->SetVerbose(kTRUE);
117	>
118	>	if (type == kBaseline) {
119	>	tmpTMVAReader->AddVariable( "SigmaIEtaIEta",         &fMVAVar_EleSigmaIEtaIEta            );
120	>	tmpTMVAReader->AddVariable( "DEtaIn",                &fMVAVar_EleDEtaIn                   );
121	>	tmpTMVAReader->AddVariable( "DPhiIn",                &fMVAVar_EleDPhiIn                   );
122	>	tmpTMVAReader->AddVariable( "FBrem",                 &fMVAVar_EleFBrem                    );
123	>	tmpTMVAReader->AddVariable( "SigmaIPhiIPhi",         &fMVAVar_EleSigmaIPhiIPhi            );
124	>	tmpTMVAReader->AddVariable( "NBrem",                 &fMVAVar_EleNBrem                    );
125	>	tmpTMVAReader->AddVariable( "OneOverEMinusOneOverP", &fMVAVar_EleOneOverEMinusOneOverP    );
126	>	}
127	>	if (type == kNoIPInfo) {
128	>	tmpTMVAReader->AddVariable( "SigmaIEtaIEta",         &fMVAVar_EleSigmaIEtaIEta            );
129	>	tmpTMVAReader->AddVariable( "DEtaIn",                &fMVAVar_EleDEtaIn                   );
130	>	tmpTMVAReader->AddVariable( "DPhiIn",                &fMVAVar_EleDPhiIn                   );
131	>	tmpTMVAReader->AddVariable( "FBrem",                 &fMVAVar_EleFBrem                    );
132	>	tmpTMVAReader->AddVariable( "EOverP",                &fMVAVar_EleEOverP                   );
133	>	tmpTMVAReader->AddVariable( "ESeedClusterOverPout",  &fMVAVar_EleESeedClusterOverPout     );
134	>	tmpTMVAReader->AddVariable( "SigmaIPhiIPhi",         &fMVAVar_EleSigmaIPhiIPhi            );
135	>	tmpTMVAReader->AddVariable( "NBrem",                 &fMVAVar_EleNBrem                    );
136	>	tmpTMVAReader->AddVariable( "OneOverEMinusOneOverP", &fMVAVar_EleOneOverEMinusOneOverP    );
137	>	tmpTMVAReader->AddVariable( "ESeedClusterOverPIn",   &fMVAVar_EleESeedClusterOverPIn      );
138	>	}
139	>	if (type == kWithIPInfo) {
140	>	tmpTMVAReader->AddVariable( "SigmaIEtaIEta",         &fMVAVar_EleSigmaIEtaIEta            );
141	>	tmpTMVAReader->AddVariable( "DEtaIn",                &fMVAVar_EleDEtaIn                   );
142	>	tmpTMVAReader->AddVariable( "DPhiIn",                &fMVAVar_EleDPhiIn                   );
143	>	tmpTMVAReader->AddVariable( "D0",                    &fMVAVar_EleD0                       );
144	>	tmpTMVAReader->AddVariable( "FBrem",                 &fMVAVar_EleFBrem                    );
145	>	tmpTMVAReader->AddVariable( "EOverP",                &fMVAVar_EleEOverP                   );
146	>	tmpTMVAReader->AddVariable( "ESeedClusterOverPout",  &fMVAVar_EleESeedClusterOverPout     );
147	>	tmpTMVAReader->AddVariable( "SigmaIPhiIPhi",         &fMVAVar_EleSigmaIPhiIPhi            );
148	>	tmpTMVAReader->AddVariable( "NBrem",                 &fMVAVar_EleNBrem                    );
149	>	tmpTMVAReader->AddVariable( "OneOverEMinusOneOverP", &fMVAVar_EleOneOverEMinusOneOverP    );
150	>	tmpTMVAReader->AddVariable( "ESeedClusterOverPIn",   &fMVAVar_EleESeedClusterOverPIn      );
151	>	tmpTMVAReader->AddVariable( "IP3d",                  &fMVAVar_EleIP3d                     );
152	>	tmpTMVAReader->AddVariable( "IP3dSig",               &fMVAVar_EleIP3dSig                  );
153	>	}
154	>	if (type == kIDIsoCombined) {
155	>	tmpTMVAReader->AddVariable( "SigmaIEtaIEta",         &fMVAVar_EleSigmaIEtaIEta            );
156	>	tmpTMVAReader->AddVariable( "DEtaIn",                &fMVAVar_EleDEtaIn                   );
157	>	tmpTMVAReader->AddVariable( "DPhiIn",                &fMVAVar_EleDPhiIn                   );
158	>	tmpTMVAReader->AddVariable( "D0",                    &fMVAVar_EleD0                       );
159	>	tmpTMVAReader->AddVariable( "FBrem",                 &fMVAVar_EleFBrem                    );
160	>	tmpTMVAReader->AddVariable( "EOverP",                &fMVAVar_EleEOverP                   );
161	>	tmpTMVAReader->AddVariable( "ESeedClusterOverPout",  &fMVAVar_EleESeedClusterOverPout     );
162	>	tmpTMVAReader->AddVariable( "SigmaIPhiIPhi",         &fMVAVar_EleSigmaIPhiIPhi            );
163	>	tmpTMVAReader->AddVariable( "OneOverEMinusOneOverP", &fMVAVar_EleOneOverEMinusOneOverP    );
164	>	tmpTMVAReader->AddVariable( "ESeedClusterOverPIn",   &fMVAVar_EleESeedClusterOverPIn      );
165	>	tmpTMVAReader->AddVariable( "IP3d",                  &fMVAVar_EleIP3d                     );
166	>	tmpTMVAReader->AddVariable( "IP3dSig",               &fMVAVar_EleIP3dSig                  );
167	>
168	>	tmpTMVAReader->AddVariable( "GsfTrackChi2OverNdof",  &fMVAVar_EleGsfTrackChi2OverNdof     );
169	>	tmpTMVAReader->AddVariable( "dEtaCalo",              &fMVAVar_EledEtaCalo                 );
170	>	tmpTMVAReader->AddVariable( "dPhiCalo",              &fMVAVar_EledPhiCalo                 );
171	>	tmpTMVAReader->AddVariable( "R9",                    &fMVAVar_EleR9                       );
172	>	tmpTMVAReader->AddVariable( "SCEtaWidth",            &fMVAVar_EleSCEtaWidth               );
173	>	tmpTMVAReader->AddVariable( "SCPhiWidth",            &fMVAVar_EleSCPhiWidth               );
174	>	tmpTMVAReader->AddVariable( "CovIEtaIPhi",           &fMVAVar_EleCovIEtaIPhi              );
175	>	if (i == 2 || i == 5) {
176	>	tmpTMVAReader->AddVariable( "PreShowerOverRaw",      &fMVAVar_ElePreShowerOverRaw       );
177	>	}
178	>	tmpTMVAReader->AddVariable( "ChargedIso03",          &fMVAVar_EleChargedIso03OverPt       );
179	>	tmpTMVAReader->AddVariable( "NeutralHadronIso03",    &fMVAVar_EleNeutralHadronIso03OverPt );
180	>	tmpTMVAReader->AddVariable( "GammaIso03",            &fMVAVar_EleGammaIso03OverPt         );
181	>	tmpTMVAReader->AddVariable( "ChargedIso04",          &fMVAVar_EleChargedIso04OverPt       );
182	>	tmpTMVAReader->AddVariable( "NeutralHadronIso04",    &fMVAVar_EleNeutralHadronIso04OverPt );
183	>	tmpTMVAReader->AddVariable( "GammaIso04",            &fMVAVar_EleGammaIso04OverPt         );
184	>
185	>	}
186	>
187	>	if (type == kIDEGamma2012TrigV0 || type == kIDHWW2012TrigV0) {
188	>	// Pure tracking variables
189	>	tmpTMVAReader->AddVariable("fbrem",           &fMVAVar_EleFBrem);
190	>	tmpTMVAReader->AddVariable("kfchi2",          &fMVAVar_EleKFTrkChiSqr);
191	>	tmpTMVAReader->AddVariable("kfhits",          &fMVAVar_EleKFTrkNLayers);  //Don't have this in (BAMBU <= 025)
192	>	if(type == kIDEGamma2012TrigV0)
193	>	tmpTMVAReader->AddVariable("kfhitsall",       &fMVAVar_EleKFTrkNHits);
194	>	tmpTMVAReader->AddVariable("gsfchi2",         &fMVAVar_EleGsfTrackChi2OverNdof);
195	>	tmpTMVAReader->AddVariable("deta",            &fMVAVar_EleDEtaIn);
196	>	tmpTMVAReader->AddVariable("dphi",            &fMVAVar_EleDPhiIn);
197	>	tmpTMVAReader->AddVariable("detacalo",        &fMVAVar_EledEtaCalo);
198	>	tmpTMVAReader->AddVariable("see",             &fMVAVar_EleSigmaIEtaIEta);
199	>	tmpTMVAReader->AddVariable("spp",             &fMVAVar_EleSigmaIPhiIPhi);
200	>	tmpTMVAReader->AddVariable("etawidth",        &fMVAVar_EleSCEtaWidth);
201	>	tmpTMVAReader->AddVariable("phiwidth",        &fMVAVar_EleSCPhiWidth);
202	>	tmpTMVAReader->AddVariable("e1x5e5x5",        &fMVAVar_EleE1x5OverE5x5);
203	>	tmpTMVAReader->AddVariable("R9",              &fMVAVar_EleR9);
204	>	tmpTMVAReader->AddVariable("HoE",             &fMVAVar_EleHoverE);
205	>	tmpTMVAReader->AddVariable("EoP",             &fMVAVar_EleEOverP);
206	>	tmpTMVAReader->AddVariable("IoEmIoP",         &fMVAVar_EleOneOverEMinusOneOverP);
207	>	tmpTMVAReader->AddVariable("eleEoPout",       &fMVAVar_EleEEleClusterOverPout); //Don't have this in (BAMBU <= 025)
208	>	if(type == kIDEGamma2012TrigV0)
209	>	tmpTMVAReader->AddVariable("EoPout",          &fMVAVar_EleESeedClusterOverPout);
210	>	if (i == 2 || i == 5) {
211	>	tmpTMVAReader->AddVariable( "PreShowerOverRaw",      &fMVAVar_ElePreShowerOverRaw       );
212	>	}
213	>	tmpTMVAReader->AddVariable( "d0",             &fMVAVar_EleD0);
214	>	tmpTMVAReader->AddVariable( "ip3d",           &fMVAVar_EleIP3d);
215
216	<	for(UInt_t i=0; i<6; ++i) {
217	<	if (fTMVAReader[i]) delete fTMVAReader[i];
216	>	tmpTMVAReader->AddSpectator("eta",            &fMVAVar_EleEta);
217	>	tmpTMVAReader->AddSpectator("pt",             &fMVAVar_ElePt);
218	>	}
219
220	<	fTMVAReader[i] = new TMVA::Reader( "!Color:!Silent:Error" );
221	<	fTMVAReader[i]->SetVerbose(kTRUE);
222	<	fTMVAReader[i]->AddVariable( "SigmaIEtaIEta",         &fMVAVar_EleSigmaIEtaIEta         );
223	<	fTMVAReader[i]->AddVariable( "DEtaIn",                &fMVAVar_EleDEtaIn                );
224	<	fTMVAReader[i]->AddVariable( "DPhiIn",                &fMVAVar_EleDPhiIn                );
225	<	fTMVAReader[i]->AddVariable( "HoverE",                &fMVAVar_EleHoverE                );
226	<	fTMVAReader[i]->AddVariable( "D0",                    &fMVAVar_EleD0                    );
227	<	fTMVAReader[i]->AddVariable( "FBrem",                 &fMVAVar_EleFBrem                 );
228	<	fTMVAReader[i]->AddVariable( "EOverP",                &fMVAVar_EleEOverP                );
229	<	fTMVAReader[i]->AddVariable( "ESeedClusterOverPout",  &fMVAVar_EleESeedClusterOverPout  );
230	<	fTMVAReader[i]->AddVariable( "SigmaIPhiIPhi",         &fMVAVar_EleSigmaIPhiIPhi         );
231	<	fTMVAReader[i]->AddVariable( "NBrem",                 &fMVAVar_EleNBrem                 );
232	<	fTMVAReader[i]->AddVariable( "OneOverEMinusOneOverP", &fMVAVar_EleOneOverEMinusOneOverP );
233	<	fTMVAReader[i]->AddVariable( "ESeedClusterOverPIn",   &fMVAVar_EleESeedClusterOverPIn   );
234	<	fTMVAReader[i]->AddVariable( "IP3d",                  &fMVAVar_EleIP3d                  );
235	<	fTMVAReader[i]->AddVariable( "IP3dSig",               &fMVAVar_EleIP3dSig               );
236	<	fTMVAReader[i]->AddVariable( "StandardLikelihood",    &fMVAVar_EleStandardLikelihood    );
237	<
238	<	if (i==0) fTMVAReader[i]->BookMVA(fMethodname , Subdet0Pt10To20Weights );
239	<	if (i==1) fTMVAReader[i]->BookMVA(fMethodname , Subdet1Pt10To20Weights );
240	<	if (i==2) fTMVAReader[i]->BookMVA(fMethodname , Subdet2Pt10To20Weights );
241	<	if (i==3) fTMVAReader[i]->BookMVA(fMethodname , Subdet0Pt20ToInfWeights );
242	<	if (i==4) fTMVAReader[i]->BookMVA(fMethodname , Subdet1Pt20ToInfWeights );
243	<	if (i==5) fTMVAReader[i]->BookMVA(fMethodname , Subdet2Pt20ToInfWeights );
220	>	if (type == kIDEGamma2012NonTrigV0 ) {
221	>	// Pure tracking variables
222	>	tmpTMVAReader->AddVariable("fbrem",           &fMVAVar_EleFBrem);
223	>	tmpTMVAReader->AddVariable("kfchi2",          &fMVAVar_EleKFTrkChiSqr);
224	>	tmpTMVAReader->AddVariable("kfhitsall",       &fMVAVar_EleKFTrkNHits);
225	>	tmpTMVAReader->AddVariable("gsfchi2",         &fMVAVar_EleGsfTrackChi2OverNdof);
226	>	tmpTMVAReader->AddVariable("deta",            &fMVAVar_EleDEtaIn);
227	>	tmpTMVAReader->AddVariable("dphi",            &fMVAVar_EleDPhiIn);
228	>	tmpTMVAReader->AddVariable("detacalo",        &fMVAVar_EledEtaCalo);
229	>	tmpTMVAReader->AddVariable("see",             &fMVAVar_EleSigmaIEtaIEta);
230	>	tmpTMVAReader->AddVariable("spp",             &fMVAVar_EleSigmaIPhiIPhi);
231	>	tmpTMVAReader->AddVariable("etawidth",        &fMVAVar_EleSCEtaWidth);
232	>	tmpTMVAReader->AddVariable("phiwidth",        &fMVAVar_EleSCPhiWidth);
233	>	tmpTMVAReader->AddVariable("e1x5e5x5",        &fMVAVar_EleE1x5OverE5x5);
234	>	tmpTMVAReader->AddVariable("R9",              &fMVAVar_EleR9);
235	>	tmpTMVAReader->AddVariable("HoE",             &fMVAVar_EleHoverE);
236	>	tmpTMVAReader->AddVariable("EoP",             &fMVAVar_EleEOverP);
237	>	tmpTMVAReader->AddVariable("IoEmIoP",         &fMVAVar_EleOneOverEMinusOneOverP);
238	>	tmpTMVAReader->AddVariable("EoPout",          &fMVAVar_EleESeedClusterOverPout);
239	>	if (i==2 || i==5) {
240	>	tmpTMVAReader->AddVariable("PreShowerOverRaw",&fMVAVar_ElePreShowerOverRaw);
241	>	}
242	>	tmpTMVAReader->AddSpectator("eta",            &fMVAVar_EleEta);
243	>	tmpTMVAReader->AddSpectator("pt",             &fMVAVar_ElePt);
244	>	}
245	>
246	>	if (type == kIsoRingsV0) {
247	>	tmpTMVAReader->AddVariable( "ChargedIso_DR0p0To0p1",         &fMVAVar_ChargedIso_DR0p0To0p1        );
248	>	tmpTMVAReader->AddVariable( "ChargedIso_DR0p1To0p2",         &fMVAVar_ChargedIso_DR0p1To0p2        );
249	>	tmpTMVAReader->AddVariable( "ChargedIso_DR0p2To0p3",         &fMVAVar_ChargedIso_DR0p2To0p3        );
250	>	tmpTMVAReader->AddVariable( "ChargedIso_DR0p3To0p4",         &fMVAVar_ChargedIso_DR0p3To0p4        );
251	>	tmpTMVAReader->AddVariable( "ChargedIso_DR0p4To0p5",         &fMVAVar_ChargedIso_DR0p4To0p5        );
252	>	tmpTMVAReader->AddVariable( "GammaIso_DR0p0To0p1",           &fMVAVar_GammaIso_DR0p0To0p1          );
253	>	tmpTMVAReader->AddVariable( "GammaIso_DR0p1To0p2",           &fMVAVar_GammaIso_DR0p1To0p2          );
254	>	tmpTMVAReader->AddVariable( "GammaIso_DR0p2To0p3",           &fMVAVar_GammaIso_DR0p2To0p3          );
255	>	tmpTMVAReader->AddVariable( "GammaIso_DR0p3To0p4",           &fMVAVar_GammaIso_DR0p3To0p4          );
256	>	tmpTMVAReader->AddVariable( "GammaIso_DR0p4To0p5",           &fMVAVar_GammaIso_DR0p4To0p5          );
257	>	tmpTMVAReader->AddVariable( "NeutralHadronIso_DR0p0To0p1",   &fMVAVar_NeutralHadronIso_DR0p0To0p1  );
258	>	tmpTMVAReader->AddVariable( "NeutralHadronIso_DR0p1To0p2",   &fMVAVar_NeutralHadronIso_DR0p1To0p2  );
259	>	tmpTMVAReader->AddVariable( "NeutralHadronIso_DR0p2To0p3",   &fMVAVar_NeutralHadronIso_DR0p2To0p3  );
260	>	tmpTMVAReader->AddVariable( "NeutralHadronIso_DR0p3To0p4",   &fMVAVar_NeutralHadronIso_DR0p3To0p4  );
261	>	tmpTMVAReader->AddVariable( "NeutralHadronIso_DR0p4To0p5",   &fMVAVar_NeutralHadronIso_DR0p4To0p5  );
262	>	tmpTMVAReader->AddSpectator( "eta",   &fMVAVar_EleEta );
263	>	tmpTMVAReader->AddSpectator( "pt" ,   &fMVAVar_ElePt  );
264	>	}
265	>
266	>	tmpTMVAReader->BookMVA(fMethodname , weightsfiles[i] );
267	>	std::cout << "MVABin " << i << " : MethodName = " << fMethodname
268	>	<< " , type == " << type << " , "
269	>	<< "Load weights file : " << weightsfiles[i]
270	>	<< std::endl;
271	>	fTMVAReader.push_back(tmpTMVAReader);
272
273		}
274	+	std::cout << "Electron ID MVA Completed\n";
275	+	}
276
82	–	std::cout << "Electron ID MVA Initialization\n";
83	–	std::cout << "MethodName : " << fMethodname << std::endl;
84	–	std::cout << "Load weights file : " << Subdet0Pt10To20Weights << std::endl;
85	–	std::cout << "Load weights file : " << Subdet1Pt10To20Weights << std::endl;
86	–	std::cout << "Load weights file : " << Subdet2Pt10To20Weights << std::endl;
87	–	std::cout << "Load weights file : " << Subdet0Pt20ToInfWeights << std::endl;
88	–	std::cout << "Load weights file : " << Subdet1Pt20ToInfWeights << std::endl;
89	–	std::cout << "Load weights file : " << Subdet2Pt20ToInfWeights << std::endl;
277
278	+	//--------------------------------------------------------------------------------------------------
279	+	UInt_t ElectronIDMVA::GetMVABin( double eta, double pt) const {
280	+
281	+	//Default is to return the first bin
282	+	uint bin = 0;
283	+
284	+	//return the first bin if not using binned version
285	+	if (!fUseBinnedVersion) return 0;
286	+
287	+	if (fMVAType == ElectronIDMVA::kBaseline
288	+	||fMVAType == ElectronIDMVA::kNoIPInfo
289	+	||fMVAType == ElectronIDMVA::kWithIPInfo
290	+	||fMVAType == ElectronIDMVA::kIDIsoCombined) {
291	+	if (pt < 20 && fabs(eta) < 1.0) bin = 0;
292	+	if (pt < 20 && fabs(eta) >= 1.0 && fabs(eta) < 1.479) bin = 1;
293	+	if (pt < 20 && fabs(eta) >= 1.479) bin = 2;
294	+	if (pt >= 20 && fabs(eta) < 1.0) bin = 3;
295	+	if (pt >= 20 && fabs(eta) >= 1.0 && fabs(eta) < 1.479) bin = 4;
296	+	if (pt >= 20 && fabs(eta) >= 1.479) bin = 5;
297	+	}
298	+
299	+	if (fMVAType == ElectronIDMVA::kIsoRingsV0) {
300	+	if (pt < 10 && fabs(eta) < 1.479) bin = 0;
301	+	if (pt < 10 && fabs(eta) >= 1.479) bin = 1;
302	+	if (pt >= 10 && fabs(eta) < 1.479) bin = 2;
303	+	if (pt >= 10 && fabs(eta) >= 1.479) bin = 3;
304	+	}
305	+
306	+	if (fMVAType == ElectronIDMVA::kIDEGamma2012TrigV0 ||
307	+	fMVAType == ElectronIDMVA::kIDEGamma2012NonTrigV0 ||
308	+	fMVAType == ElectronIDMVA::kIDHWW2012TrigV0) {
309	+	bin = 0;
310	+	if (pt < 10 && fabs(eta) < 0.8) bin = 0;
311	+	if (pt < 10 && fabs(eta) >= 0.8 && fabs(eta) < 1.479 ) bin = 1;
312	+	if (pt < 10 && fabs(eta) >= 1.479) bin = 2;
313	+	if (pt >= 10 && fabs(eta) < 0.8) bin = 3;
314	+	if (pt >= 10 && fabs(eta) >= 0.8 && fabs(eta) < 1.479 ) bin = 4;
315	+	if (pt >= 10 && fabs(eta) >= 1.479) bin = 5;
316	+	}
317	+
318	+	return bin;
319		}
320
321
322	+
323		//--------------------------------------------------------------------------------------------------
324	<	Double_t ElectronIDMVA::MVAValue(const Electron *ele, const Vertex *vertex) {
324	>	Double_t ElectronIDMVA::MVAValue(Double_t ElePt , Double_t EleEta,
325	>	Double_t EleSigmaIEtaIEta,
326	>	Double_t EleDEtaIn,
327	>	Double_t EleDPhiIn,
328	>	Double_t EleHoverE,
329	>	Double_t EleD0,
330	>	Double_t EleDZ,
331	>	Double_t EleFBrem,
332	>	Double_t EleEOverP,
333	>	Double_t EleESeedClusterOverPout,
334	>	Double_t EleSigmaIPhiIPhi,
335	>	Double_t EleNBrem,
336	>	Double_t EleOneOverEMinusOneOverP,
337	>	Double_t EleESeedClusterOverPIn,
338	>	Double_t EleIP3d,
339	>	Double_t EleIP3dSig
340	>	) {
341
342		if (!fIsInitialized) {
343		std::cout << "Error: ElectronIDMVA not properly initialized.\n";
344		return -9999;
345		}
346
347	<	Int_t subdet = 0;
348	<	if (ele->SCluster()->AbsEta() < 1.0) subdet = 0;
349	<	else if (ele->SCluster()->AbsEta() < 1.479) subdet = 1;
350	<	else subdet = 2;
351	<	Int_t ptBin = 0;
352	<	if (ele->Pt() > 20.0) ptBin = 1;
347	>	//set all input variables
348	>	fMVAVar_EleSigmaIEtaIEta = EleSigmaIEtaIEta;
349	>	fMVAVar_EleDEtaIn = EleDEtaIn;
350	>	fMVAVar_EleDPhiIn = EleDPhiIn;
351	>	fMVAVar_EleHoverE = EleHoverE;
352	>	fMVAVar_EleD0 = EleD0;
353	>	fMVAVar_EleDZ = EleDZ;
354	>	fMVAVar_EleFBrem = EleFBrem;
355	>	fMVAVar_EleEOverP = EleEOverP;
356	>	fMVAVar_EleESeedClusterOverPout = EleESeedClusterOverPout;
357	>	fMVAVar_EleSigmaIPhiIPhi = EleSigmaIPhiIPhi;
358	>	fMVAVar_EleNBrem = EleNBrem;
359	>	fMVAVar_EleOneOverEMinusOneOverP = EleOneOverEMinusOneOverP;
360	>	fMVAVar_EleESeedClusterOverPIn = EleESeedClusterOverPIn;
361	>	fMVAVar_EleIP3d = EleIP3d;
362	>	fMVAVar_EleIP3dSig = EleIP3dSig;
363	>
364	>	Double_t mva = -9999;
365	>	TMVA::Reader *reader = 0;
366	>	reader = fTMVAReader[GetMVABin( EleEta, ElePt)];
367	>
368	>	mva = reader->EvaluateMVA( fMethodname );
369	>
370	>	return mva;
371	>	}
372	>
373	>	//--------------------------------------------------------------------------------------------------
374	>	Double_t ElectronIDMVA::MVAValue(Double_t ElePt , Double_t EleEta, Double_t PileupEnergyDensity,
375	>	Double_t EleSigmaIEtaIEta,
376	>	Double_t EleDEtaIn,
377	>	Double_t EleDPhiIn,
378	>	Double_t EleHoverE,
379	>	Double_t EleD0,
380	>	Double_t EleDZ,
381	>	Double_t EleFBrem,
382	>	Double_t EleEOverP,
383	>	Double_t EleESeedClusterOverPout,
384	>	Double_t EleSigmaIPhiIPhi,
385	>	Double_t EleNBrem,
386	>	Double_t EleOneOverEMinusOneOverP,
387	>	Double_t EleESeedClusterOverPIn,
388	>	Double_t EleIP3d,
389	>	Double_t EleIP3dSig,
390	>	Double_t EleGsfTrackChi2OverNdof,
391	>	Double_t EledEtaCalo,
392	>	Double_t EledPhiCalo,
393	>	Double_t EleR9,
394	>	Double_t EleSCEtaWidth,
395	>	Double_t EleSCPhiWidth,
396	>	Double_t EleCovIEtaIPhi,
397	>	Double_t ElePreShowerOverRaw,
398	>	Double_t EleChargedIso03,
399	>	Double_t EleNeutralHadronIso03,
400	>	Double_t EleGammaIso03,
401	>	Double_t EleChargedIso04,
402	>	Double_t EleNeutralHadronIso04,
403	>	Double_t EleGammaIso04,
404	>	Bool_t printDebug
405	>	) {
406
407	+	if (!fIsInitialized) {
408	+	std::cout << "Error: ElectronIDMVA not properly initialized.\n";
409	+	return -9999;
410	+	}
411	+
412	+	Double_t Rho = 0;
413	+	if (!(TMath::IsNaN(PileupEnergyDensity) || isinf(PileupEnergyDensity))) Rho = PileupEnergyDensity;
414	+
415	+	//set all input variables
416	+	fMVAVar_EleSigmaIEtaIEta = EleSigmaIEtaIEta;
417	+	fMVAVar_EleDEtaIn = EleDEtaIn;
418	+	fMVAVar_EleDPhiIn = EleDPhiIn;
419	+	fMVAVar_EleHoverE = EleHoverE;
420	+	fMVAVar_EleD0 = EleD0;
421	+	fMVAVar_EleDZ = EleDZ;
422	+	fMVAVar_EleFBrem = EleFBrem;
423	+	fMVAVar_EleEOverP = EleEOverP;
424	+	fMVAVar_EleESeedClusterOverPout = EleESeedClusterOverPout;
425	+	fMVAVar_EleSigmaIPhiIPhi = EleSigmaIPhiIPhi;
426	+	fMVAVar_EleNBrem = EleNBrem;
427	+	fMVAVar_EleOneOverEMinusOneOverP = EleOneOverEMinusOneOverP;
428	+	fMVAVar_EleESeedClusterOverPIn = EleESeedClusterOverPIn;
429	+	fMVAVar_EleIP3d = EleIP3d;
430	+	fMVAVar_EleIP3dSig = EleIP3dSig;
431	+	fMVAVar_EleGsfTrackChi2OverNdof = EleGsfTrackChi2OverNdof;
432	+	fMVAVar_EledEtaCalo = EledEtaCalo;
433	+	fMVAVar_EledPhiCalo = EledPhiCalo;
434	+	fMVAVar_EleR9 = EleR9;
435	+	fMVAVar_EleSCEtaWidth = EleSCEtaWidth;
436	+	fMVAVar_EleSCPhiWidth = EleSCPhiWidth;
437	+	fMVAVar_EleCovIEtaIPhi = EleCovIEtaIPhi;
438	+	fMVAVar_ElePreShowerOverRaw = ElePreShowerOverRaw;
439	+	fMVAVar_EleChargedIso03OverPt
440	+	= (EleChargedIso03
441	+	- Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleChargedIso03, EleEta)) / ElePt;
442	+	fMVAVar_EleNeutralHadronIso03OverPt
443	+	= (EleNeutralHadronIso03
444	+	- Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleNeutralHadronIso03, EleEta)
445	+	+ Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleNeutralHadronIso007,EleEta)) / ElePt;
446	+	fMVAVar_EleGammaIso03OverPt
447	+	= (EleGammaIso03
448	+	- Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleGammaIso03, EleEta)
449	+	+ Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleGammaIsoVetoEtaStrip03,EleEta))/ElePt;
450	+	fMVAVar_EleChargedIso04OverPt
451	+	= (EleChargedIso04
452	+	- Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleChargedIso04, EleEta))/ElePt;
453	+	fMVAVar_EleNeutralHadronIso04OverPt
454	+	= (EleNeutralHadronIso04
455	+	- Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleNeutralHadronIso04, EleEta)
456	+	+ Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleNeutralHadronIso007,EleEta))/ElePt;
457	+	fMVAVar_EleGammaIso04OverPt
458	+	= (EleGammaIso04
459	+	- Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleGammaIso04, EleEta)
460	+	+ Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleGammaIsoVetoEtaStrip04,EleEta))/ElePt;
461	+
462	+
463	+
464	+
465	+	Double_t mva = -9999;
466	+	TMVA::Reader *reader = 0;
467	+	reader = fTMVAReader[GetMVABin( EleEta, ElePt)];
468	+	mva = reader->EvaluateMVA( fMethodname );
469	+
470	+	if (printDebug == kTRUE) {
471	+	std::cout << "Debug Electron MVA: "
472	+	<< ElePt << " " << EleEta << " " << " --> MVABin " << GetMVABin( EleEta, ElePt) << " : "
473	+	<< fMVAVar_EleSigmaIEtaIEta << " "
474	+	<< fMVAVar_EleDEtaIn << " "
475	+	<< fMVAVar_EleDPhiIn << " "
476	+	<< fMVAVar_EleHoverE << " "
477	+	<< fMVAVar_EleD0 << " "
478	+	<< fMVAVar_EleDZ << " "
479	+	<< fMVAVar_EleFBrem << " "
480	+	<< fMVAVar_EleEOverP << " "
481	+	<< fMVAVar_EleESeedClusterOverPout << " "
482	+	<< fMVAVar_EleSigmaIPhiIPhi << " "
483	+	<< fMVAVar_EleNBrem << " "
484	+	<< fMVAVar_EleOneOverEMinusOneOverP << " "
485	+	<< fMVAVar_EleESeedClusterOverPIn << " "
486	+	<< fMVAVar_EleIP3d << " "
487	+	<< fMVAVar_EleIP3dSig << " "
488	+	<< fMVAVar_EleGsfTrackChi2OverNdof << " "
489	+	<< fMVAVar_EledEtaCalo << " "
490	+	<< fMVAVar_EledPhiCalo << " "
491	+	<< fMVAVar_EleR9 << " "
492	+	<< fMVAVar_EleSCEtaWidth << " "
493	+	<< fMVAVar_EleSCPhiWidth << " "
494	+	<< fMVAVar_EleCovIEtaIPhi << " "
495	+	<< fMVAVar_ElePreShowerOverRaw << " "
496	+	<< fMVAVar_EleChargedIso03OverPt  << " "
497	+	<< fMVAVar_EleNeutralHadronIso03OverPt  << " "
498	+	<< fMVAVar_EleGammaIso03OverPt  << " "
499	+	<< fMVAVar_EleChargedIso04OverPt  << " "
500	+	<< fMVAVar_EleNeutralHadronIso04OverPt  << " "
501	+	<< fMVAVar_EleGammaIso04OverPt  << " "
502	+	<< " === : === "
503	+	<< mva
504	+	<< std::endl;
505	+	}
506	+
507	+	return mva;
508	+	}
509	+
510	+	Double_t ElectronIDMVA::MVAValue_IsoRings( Double_t ElePt,
511	+	Double_t EleSCEta,
512	+	Double_t ChargedIso_DR0p0To0p1,
513	+	Double_t ChargedIso_DR0p1To0p2,
514	+	Double_t ChargedIso_DR0p2To0p3,
515	+	Double_t ChargedIso_DR0p3To0p4,
516	+	Double_t ChargedIso_DR0p4To0p5,
517	+	Double_t GammaIso_DR0p0To0p1,
518	+	Double_t GammaIso_DR0p1To0p2,
519	+	Double_t GammaIso_DR0p2To0p3,
520	+	Double_t GammaIso_DR0p3To0p4,
521	+	Double_t GammaIso_DR0p4To0p5,
522	+	Double_t NeutralHadronIso_DR0p0To0p1,
523	+	Double_t NeutralHadronIso_DR0p1To0p2,
524	+	Double_t NeutralHadronIso_DR0p2To0p3,
525	+	Double_t NeutralHadronIso_DR0p3To0p4,
526	+	Double_t NeutralHadronIso_DR0p4To0p5,
527	+	Bool_t printDebug) {
528	+
529	+	if (fMVAType != ElectronIDMVA::kIsoRingsV0) {
530	+	std::cout << "Error: This function is only supported for MVAType == kIsoRingsV0.\n" << std::endl;
531	+	assert(kFALSE);
532	+	}
533	+
534	+	fMVAVar_ElePt = ElePt;
535	+	fMVAVar_EleEta = EleSCEta;
536	+	fMVAVar_ChargedIso_DR0p0To0p1 = ChargedIso_DR0p0To0p1;
537	+	fMVAVar_ChargedIso_DR0p1To0p2 = ChargedIso_DR0p1To0p2;
538	+	fMVAVar_ChargedIso_DR0p2To0p3 = ChargedIso_DR0p2To0p3;
539	+	fMVAVar_ChargedIso_DR0p3To0p4 = ChargedIso_DR0p3To0p4;
540	+	fMVAVar_ChargedIso_DR0p4To0p5 = ChargedIso_DR0p4To0p5;
541	+	fMVAVar_GammaIso_DR0p0To0p1 = GammaIso_DR0p0To0p1;
542	+	fMVAVar_GammaIso_DR0p1To0p2 = GammaIso_DR0p1To0p2;
543	+	fMVAVar_GammaIso_DR0p2To0p3 = GammaIso_DR0p2To0p3;
544	+	fMVAVar_GammaIso_DR0p3To0p4 = GammaIso_DR0p3To0p4;
545	+	fMVAVar_GammaIso_DR0p4To0p5 = GammaIso_DR0p4To0p5;
546	+	fMVAVar_NeutralHadronIso_DR0p0To0p1 = NeutralHadronIso_DR0p0To0p1;
547	+	fMVAVar_NeutralHadronIso_DR0p1To0p2 = NeutralHadronIso_DR0p1To0p2;
548	+	fMVAVar_NeutralHadronIso_DR0p2To0p3 = NeutralHadronIso_DR0p2To0p3;
549	+	fMVAVar_NeutralHadronIso_DR0p3To0p4 = NeutralHadronIso_DR0p3To0p4;
550	+	fMVAVar_NeutralHadronIso_DR0p4To0p5 = NeutralHadronIso_DR0p4To0p5;
551	+
552	+	Double_t mva = -9999;
553	+	TMVA::Reader *reader = 0;
554	+
555	+	if (printDebug == kTRUE) {
556	+	std::cout <<" -> BIN: " << fMVAVar_EleEta << " " << fMVAVar_ElePt << " : " << GetMVABin( fMVAVar_EleEta , fMVAVar_ElePt) << std::endl;
557	+	}
558	+	reader = fTMVAReader[GetMVABin( fMVAVar_EleEta , fMVAVar_ElePt)];
559	+	mva = reader->EvaluateMVA( fMethodname );
560	+
561	+	if (printDebug == kTRUE) {
562	+
563	+	std::cout << "Debug Electron MVA: \n";
564	+	std::cout << fMVAVar_ChargedIso_DR0p0To0p1 << " "
565	+	<< fMVAVar_ChargedIso_DR0p1To0p2 << " "
566	+	<< fMVAVar_ChargedIso_DR0p2To0p3 << " "
567	+	<< fMVAVar_ChargedIso_DR0p3To0p4 << " "
568	+	<< fMVAVar_ChargedIso_DR0p4To0p5 << " "
569	+	<< fMVAVar_GammaIso_DR0p0To0p1 << " "
570	+	<< fMVAVar_GammaIso_DR0p1To0p2 << " "
571	+	<< fMVAVar_GammaIso_DR0p2To0p3 << " "
572	+	<< fMVAVar_GammaIso_DR0p3To0p4 << " "
573	+	<< fMVAVar_GammaIso_DR0p4To0p5 << " "
574	+	<< fMVAVar_NeutralHadronIso_DR0p0To0p1 << " "
575	+	<< fMVAVar_NeutralHadronIso_DR0p1To0p2 << " "
576	+	<< fMVAVar_NeutralHadronIso_DR0p2To0p3 << " "
577	+	<< fMVAVar_NeutralHadronIso_DR0p3To0p4 << " "
578	+	<< fMVAVar_NeutralHadronIso_DR0p4To0p5 << " "
579	+	<< std::endl;
580	+	std::cout << "MVA: " << mva << " "
581	+	<< std::endl;
582	+	}
583	+	return mva;
584	+	}
585	+
586	+	Double_t ElectronIDMVA::MVAValue_IDNonTrig( Double_t ElePt,
587	+	Double_t EleSCEta,
588	+	Double_t EleFBrem,
589	+	Double_t EleKFTrkChiSqr,
590	+	Double_t EleKFTrkNHits,
591	+	Double_t EleGsfTrackChi2OverNdof,
592	+	Double_t EleDEtaIn,
593	+	Double_t EleDPhiIn,
594	+	Double_t EledEtaCalo,
595	+	Double_t EleSigmaIEtaIEta,
596	+	Double_t EleSigmaIPhiIPhi,
597	+	Double_t EleSCEtaWidth,
598	+	Double_t EleSCPhiWidth,
599	+	Double_t EleE1x5OverE5x5,
600	+	Double_t EleR9,
601	+	Double_t EleHoverE,
602	+	Double_t EleEOverP,
603	+	Double_t EleOneOverEMinusOneOverP,
604	+	Double_t EleESeedClusterOverPout,
605	+	Double_t ElePreShowerOverRaw,
606	+	Bool_t printDebug) {
607	+
608	+	if (fMVAType != ElectronIDMVA::kIDEGamma2012NonTrigV0) {
609	+	std::cout << "Error: This function is only supported for MVAType == kIDEGamma2012NonTrigV0.\n" << std::endl;
610	+	assert(kFALSE);
611	+	}
612	+
613	+	fMVAVar_ElePt = ElePt;
614	+	fMVAVar_EleEta = EleSCEta;
615	+	fMVAVar_EleFBrem = EleFBrem;
616	+	fMVAVar_EleKFTrkChiSqr = EleKFTrkChiSqr;
617	+	fMVAVar_EleKFTrkNHits = EleKFTrkNHits;
618	+	fMVAVar_EleGsfTrackChi2OverNdof = EleGsfTrackChi2OverNdof;
619	+	fMVAVar_EleDEtaIn = EleDEtaIn;
620	+	fMVAVar_EleDPhiIn = EleDPhiIn;
621	+	fMVAVar_EledEtaCalo = EledEtaCalo;
622	+	fMVAVar_EleSigmaIEtaIEta = EleSigmaIEtaIEta;
623	+	fMVAVar_EleSigmaIPhiIPhi = EleSigmaIPhiIPhi;
624	+	fMVAVar_EleSCEtaWidth = EleSCEtaWidth;
625	+	fMVAVar_EleSCPhiWidth = EleSCPhiWidth;
626	+	fMVAVar_EleE1x5OverE5x5 = EleE1x5OverE5x5;
627	+	fMVAVar_EleR9 = EleR9;
628	+	fMVAVar_EleHoverE = EleHoverE;
629	+	fMVAVar_EleEOverP = EleEOverP;
630	+	fMVAVar_EleOneOverEMinusOneOverP = EleOneOverEMinusOneOverP;
631	+	fMVAVar_EleESeedClusterOverPout = EleESeedClusterOverPout;
632	+	fMVAVar_ElePreShowerOverRaw = ElePreShowerOverRaw;
633	+
634	+	Double_t mva = -9999;
635	+	TMVA::Reader *reader = 0;
636	+
637	+	if (printDebug == kTRUE) {
638	+	std::cout <<" -> BIN: " << fMVAVar_EleEta << " " << fMVAVar_ElePt << " : " << GetMVABin( fMVAVar_EleEta , fMVAVar_ElePt) << std::endl;
639	+	}
640	+	reader = fTMVAReader[GetMVABin( fMVAVar_EleEta , fMVAVar_ElePt)];
641	+	mva = reader->EvaluateMVA( fMethodname );
642	+
643	+	if (printDebug == kTRUE) {
644	+	std::cout << "Debug Electron MVA: \n";
645	+	std::cout << " fbrem " <<  fMVAVar_EleFBrem
646	+	<< " kfchi2 " << fMVAVar_EleKFTrkChiSqr
647	+	<< " kfhits " << fMVAVar_EleKFTrkNHits
648	+	<< " kfhitsall " << fMVAVar_EleKFTrkNHits
649	+	<< " gsfchi2 " << fMVAVar_EleGsfTrackChi2OverNdof
650	+	<< " deta " <<  fMVAVar_EleDEtaIn
651	+	<< " dphi " << fMVAVar_EleDPhiIn
652	+	<< " detacalo " << fMVAVar_EledEtaCalo
653	+	<< " see " << fMVAVar_EleSigmaIEtaIEta
654	+	<< " spp " << fMVAVar_EleSigmaIPhiIPhi
655	+	<< " etawidth " << fMVAVar_EleSCEtaWidth
656	+	<< " phiwidth " << fMVAVar_EleSCPhiWidth
657	+	<< " e1x5e5x5 " << fMVAVar_EleE1x5OverE5x5
658	+	<< " R9 " << fMVAVar_EleR9
659	+	<< " HoE " << fMVAVar_EleHoverE
660	+	<< " EoP " << fMVAVar_EleEOverP
661	+	<< " IoEmIoP " << fMVAVar_EleOneOverEMinusOneOverP
662	+	<< " eleEoPout " << fMVAVar_EleESeedClusterOverPout
663	+	<< " EoPout " << fMVAVar_EleESeedClusterOverPout
664	+	<< " d0 " << fMVAVar_EleD0
665	+	<< " ip3d " << fMVAVar_EleIP3d
666	+	<< " eta " << fMVAVar_EleEta
667	+	<< " pt " << fMVAVar_ElePt << std::endl;
668	+	std::cout << "MVA: " << mva << " "
669	+	<< std::endl;
670	+	}
671	+	return mva;
672	+	}
673	+
674	+
675	+	//--------------------------------------------------------------------------------------------------
676	+	Double_t ElectronIDMVA::MVAValue(const Electron *ele, const Vertex *vertex,
677	+	const PFCandidateCol *PFCands,
678	+	const PileupEnergyDensityCol *PileupEnergyDensity,
679	+	Double_t intRadius,
680	+	Bool_t printDebug) {
681	+
682	+	if (!fIsInitialized) {
683	+	std::cout << "Error: ElectronIDMVA not properly initialized.\n";
684	+	return -9999;
685	+	}
686	+
687	+	Double_t Rho = 0;
688	+	if (!(TMath::IsNaN(PileupEnergyDensity->At(0)->Rho()) || isinf(PileupEnergyDensity->At(0)->Rho()))) Rho = PileupEnergyDensity->At(0)->Rho();
689	+
690		//set all input variables
691		fMVAVar_EleSigmaIEtaIEta = ele->CoviEtaiEta() ;
692		fMVAVar_EleDEtaIn = ele->DeltaEtaSuperClusterTrackAtVtx();
#	Line 119 | Line 700 | Double_t ElectronIDMVA::MVAValue(const E
700		if (!TMath::IsNaN(ele->SCluster()->Seed()->CoviPhiiPhi())) fMVAVar_EleSigmaIPhiIPhi = TMath::Sqrt(ele->SCluster()->Seed()->CoviPhiiPhi());
701		else fMVAVar_EleSigmaIPhiIPhi = ele->CoviEtaiEta();
702		fMVAVar_EleNBrem = ele->NumberOfClusters() - 1;
703	<	fMVAVar_EleOneOverEMinusOneOverP = (1.0/(ele->ESuperClusterOverP()*ele->BestTrk()->P())) - 1.0 / ele->BestTrk()->P();
703	>	fMVAVar_EleOneOverEMinusOneOverP = (1.0/(ele->SCluster()->Energy())) - 1.0 / ele->BestTrk()->P();
704		fMVAVar_EleESeedClusterOverPIn = ele->ESeedClusterOverPIn();
705		fMVAVar_EleIP3d = ele->Ip3dPV();
706		fMVAVar_EleIP3dSig = ele->Ip3dPVSignificance();
707	<	fMVAVar_EleStandardLikelihood = ElectronTools::Likelihood(fLH, ele);
707	>	fMVAVar_EleGsfTrackChi2OverNdof = ele->BestTrk()->Chi2() / ele->BestTrk()->Ndof();
708	>	fMVAVar_EledEtaCalo =  ele->DeltaEtaSeedClusterTrackAtCalo();
709	>	fMVAVar_EledPhiCalo = ele->DeltaPhiSeedClusterTrackAtCalo();
710	>	fMVAVar_EleR9 = ele->SCluster()->R9();
711	>	fMVAVar_EleSCEtaWidth = ele->SCluster()->EtaWidth();
712	>	fMVAVar_EleSCPhiWidth = ele->SCluster()->PhiWidth();
713	>	fMVAVar_EleCovIEtaIPhi = ele->SCluster()->Seed()->CoviEtaiPhi();
714	>	fMVAVar_ElePreShowerOverRaw = ele->SCluster()->PreshowerEnergy() / ele->SCluster()->RawEnergy();
715	>	fMVAVar_EleChargedIso03OverPt
716	>	= (IsolationTools::PFElectronIsolation(ele, PFCands, vertex, 0.1, 99999, 0.3, intRadius)
717	>	- Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleChargedIso03, ele->SCluster()->Eta())) / ele->Pt();
718	>	fMVAVar_EleNeutralHadronIso03OverPt
719	>	= (IsolationTools::PFElectronIsolation(ele, PFCands, vertex, 0.1, 0.5, 0.3, intRadius, PFCandidate::eNeutralHadron)
720	>	- Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleNeutralHadronIso03, ele->SCluster()->Eta())
721	>	+ Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleNeutralHadronIso007,ele->SCluster()->Eta())) / ele->Pt();
722	>	fMVAVar_EleGammaIso03OverPt
723	>	= (IsolationTools::PFElectronIsolation(ele, PFCands, vertex, 0.1, 0.5, 0.3, intRadius, PFCandidate::eGamma)
724	>	- Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleGammaIso03, ele->SCluster()->Eta())
725	>	+ Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleGammaIsoVetoEtaStrip03,ele->SCluster()->Eta())) / ele->Pt();
726	>	fMVAVar_EleChargedIso04OverPt
727	>	= (IsolationTools::PFElectronIsolation(ele, PFCands, vertex, 0.1, 99999, 0.4, intRadius)
728	>	- Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleChargedIso04, ele->SCluster()->Eta())) / ele->Pt();
729	>	fMVAVar_EleNeutralHadronIso04OverPt
730	>	= (IsolationTools::PFElectronIsolation(ele, PFCands, vertex, 0.1, 0.5, 0.4, intRadius, PFCandidate::eNeutralHadron)
731	>	- Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleNeutralHadronIso04, ele->SCluster()->Eta())
732	>	+ Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleNeutralHadronIso007,ele->SCluster()->Eta())) / ele->Pt() ;
733	>	fMVAVar_EleGammaIso04OverPt
734	>	= (IsolationTools::PFElectronIsolation(ele, PFCands, vertex, 0.1, 0.5, 0.4, intRadius, PFCandidate::eGamma)
735	>	- Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleGammaIso04, ele->SCluster()->Eta())
736	>	+ Rho * ElectronTools::ElectronEffectiveArea(ElectronTools::kEleGammaIsoVetoEtaStrip04,ele->SCluster()->Eta())) / ele->Pt();
737	>
738	>	//Additional vars
739	>	fMVAVar_EleEEleClusterOverPout = ele->EEleClusterOverPout();
740	>	if (ele->TrackerTrk()) {
741	>	fMVAVar_EleKFTrkChiSqr = ele->TrackerTrk()->RChi2();
742	>	fMVAVar_EleKFTrkNHits = ele->TrackerTrk()->NHits();
743	>	} else {
744	>	fMVAVar_EleKFTrkChiSqr = -1;
745	>	fMVAVar_EleKFTrkNHits = 0;
746	>	}
747	>	fMVAVar_EleE1x5OverE5x5 = ele->SCluster()->Seed()->E1x5() / ele->SCluster()->Seed()->E5x5();
748	>
749
750		Double_t mva = -9999;
751		TMVA::Reader *reader = 0;
752	<	Int_t MVABin = -1;
131	<	if (subdet == 0 && ptBin == 0) MVABin = 0;
132	<	if (subdet == 1 && ptBin == 0) MVABin = 1;
133	<	if (subdet == 2 && ptBin == 0) MVABin = 2;
134	<	if (subdet == 0 && ptBin == 1) MVABin = 3;
135	<	if (subdet == 1 && ptBin == 1) MVABin = 4;
136	<	if (subdet == 2 && ptBin == 1) MVABin = 5;
137	<	assert(MVABin >= 0 && MVABin <= 5);
138	<	reader = fTMVAReader[MVABin];
139	<
752	>	reader = fTMVAReader[GetMVABin( ele->SCluster()->Eta(), ele->Pt())];
753		mva = reader->EvaluateMVA( fMethodname );
754
755	+	if (printDebug == kTRUE) {
756	+	std::cout << "Debug Electron MVA: "
757	+	<< ele->Pt() << " " << ele->Eta() << " " << ele->Phi() << " : "
758	+	<< ele->Pt() << " " << ele->SCluster()->AbsEta() << " --> MVABin " << GetMVABin( ele->SCluster()->Eta(), ele->Pt()) << " : "
759	+	<< fMVAVar_EleSigmaIEtaIEta << " "
760	+	<< fMVAVar_EleDEtaIn << " "
761	+	<< fMVAVar_EleDPhiIn << " "
762	+	<< fMVAVar_EleHoverE << " "
763	+	<< fMVAVar_EleD0 << " "
764	+	<< fMVAVar_EleDZ << " "
765	+	<< fMVAVar_EleFBrem << " "
766	+	<< fMVAVar_EleEOverP << " "
767	+	<< fMVAVar_EleESeedClusterOverPout << " "
768	+	<< fMVAVar_EleSigmaIPhiIPhi << " "
769	+	<< fMVAVar_EleNBrem << " "
770	+	<< fMVAVar_EleOneOverEMinusOneOverP << " "
771	+	<< fMVAVar_EleESeedClusterOverPIn << " "
772	+	<< fMVAVar_EleIP3d << " "
773	+	<< fMVAVar_EleIP3dSig << " "
774	+	<< fMVAVar_EleGsfTrackChi2OverNdof << " "
775	+	<< fMVAVar_EledEtaCalo << " "
776	+	<< fMVAVar_EledPhiCalo << " "
777	+	<< fMVAVar_EleR9 << " "
778	+	<< fMVAVar_EleSCEtaWidth << " "
779	+	<< fMVAVar_EleSCPhiWidth << " "
780	+	<< fMVAVar_EleCovIEtaIPhi << " "
781	+	<< fMVAVar_ElePreShowerOverRaw << " "
782	+	<< fMVAVar_EleKFTrkChiSqr  << " "
783	+	<< fMVAVar_EleKFTrkNHits  << " "
784	+	<< fMVAVar_EleE1x5OverE5x5  << " "
785	+	<< " ::: "
786	+
787	+	<< " === : === "
788	+	<< mva << " "
789	+	<< std::endl;
790	+
791	+	}
792	+
793		return mva;
794		}
795	+
796	+	//--------------------------------------------------------------------------------------------------
797	+	Double_t ElectronIDMVA::MVAValue(const Electron *ele, const Vertex *vertex,
798	+	Bool_t printDebug) {
799	+
800	+	if (!fIsInitialized) {
801	+	std::cout << "Error: ElectronIDMVA not properly initialized.\n";
802	+	return -9999;
803	+	}
804	+
805	+	fMVAVar_ElePt = ele->Pt();
806	+	fMVAVar_EleEta = ele->Eta();
807	+
808	+	//set all input variables
809	+	fMVAVar_EleSigmaIEtaIEta = ele->CoviEtaiEta() ;
810	+	fMVAVar_EleDEtaIn = ele->DeltaEtaSuperClusterTrackAtVtx();
811	+	fMVAVar_EleDPhiIn = ele->DeltaPhiSuperClusterTrackAtVtx();
812	+	fMVAVar_EleHoverE = ele->HadronicOverEm();
813	+	fMVAVar_EleD0 = ele->BestTrk()->D0Corrected(*vertex);
814	+	fMVAVar_EleDZ = ele->BestTrk()->DzCorrected(*vertex);
815	+	fMVAVar_EleFBrem = ele->FBrem();
816	+	fMVAVar_EleEOverP = ele->ESuperClusterOverP();
817	+	fMVAVar_EleESeedClusterOverPout = ele->ESeedClusterOverPout();
818	+	if (!TMath::IsNaN(ele->SCluster()->Seed()->CoviPhiiPhi())) fMVAVar_EleSigmaIPhiIPhi = TMath::Sqrt(ele->SCluster()->Seed()->CoviPhiiPhi());
819	+	else fMVAVar_EleSigmaIPhiIPhi = ele->CoviEtaiEta();
820	+	fMVAVar_EleNBrem = ele->NumberOfClusters() - 1;
821	+	fMVAVar_EleOneOverEMinusOneOverP = (1.0/(ele->SCluster()->Energy())) - 1.0 / ele->BestTrk()->P();
822	+	fMVAVar_EleESeedClusterOverPIn = ele->ESeedClusterOverPIn();
823	+	fMVAVar_EleIP3d = ele->Ip3dPV();
824	+	fMVAVar_EleIP3dSig = ele->Ip3dPVSignificance();
825	+
826	+
827	+	fMVAVar_EleEEleClusterOverPout = 0;
828	+	if (ele->TrackerTrk()) {
829	+	fMVAVar_EleKFTrkChiSqr = ele->TrackerTrk()->RChi2();
830	+	fMVAVar_EleKFTrkNHits = ele->TrackerTrk()->NHits();
831	+	} else {
832	+	fMVAVar_EleKFTrkChiSqr = -1;
833	+	fMVAVar_EleKFTrkNHits = 0;
834	+	}
835	+	fMVAVar_EleGsfTrackChi2OverNdof = ele->BestTrk()->Chi2() / ele->BestTrk()->Ndof();
836	+	fMVAVar_EledEtaCalo =  ele->DeltaEtaSeedClusterTrackAtCalo();
837	+	fMVAVar_EleSCEtaWidth = ele->SCluster()->EtaWidth();
838	+	fMVAVar_EleSCPhiWidth = ele->SCluster()->PhiWidth();
839	+	fMVAVar_EleE1x5OverE5x5 = ele->SCluster()->Seed()->E1x5() / ele->SCluster()->Seed()->E5x5();
840	+	fMVAVar_EleR9 = ele->SCluster()->R9();
841	+	fMVAVar_EleHoverE = ele->HadronicOverEm();
842	+	fMVAVar_EleEOverP = ele->ESuperClusterOverP();
843	+	fMVAVar_EleOneOverEMinusOneOverP = (1.0/(ele->SCluster()->Energy())) - 1.0 / ele->BestTrk()->P();
844	+	fMVAVar_EleR9 = ele->SCluster()->R9();
845	+	fMVAVar_ElePreShowerOverRaw = ele->SCluster()->PreshowerEnergy() / ele->SCluster()->RawEnergy();
846	+
847	+
848	+	Double_t mva = -9999;
849	+	TMVA::Reader *reader = 0;
850	+	reader = fTMVAReader[GetMVABin( ele->SCluster()->Eta(), ele->Pt())];
851	+	mva = reader->EvaluateMVA( fMethodname );
852	+
853	+	if (printDebug == kTRUE) {
854	+	std::cout << "Debug Electron MVA: "
855	+	<< ele->Pt() << " " << ele->Eta() << " " << ele->Phi() << " : "
856	+	<< ele->Pt() << " " << ele->SCluster()->AbsEta() << " --> MVABin " << GetMVABin( ele->SCluster()->Eta(), ele->Pt()) << " : "
857	+	<< fMVAVar_EleSigmaIEtaIEta << " "
858	+	<< fMVAVar_EleDEtaIn << " "
859	+	<< fMVAVar_EleDPhiIn << " "
860	+	<< fMVAVar_EleHoverE << " "
861	+	<< fMVAVar_EleD0 << " "
862	+	<< fMVAVar_EleDZ << " "
863	+	<< fMVAVar_EleFBrem << " "
864	+	<< fMVAVar_EleEOverP << " "
865	+	<< fMVAVar_EleESeedClusterOverPout << " "
866	+	<< fMVAVar_EleSigmaIPhiIPhi << " "
867	+	<< fMVAVar_EleNBrem << " "
868	+	<< fMVAVar_EleOneOverEMinusOneOverP << " "
869	+	<< fMVAVar_EleESeedClusterOverPIn << " "
870	+	<< fMVAVar_EleIP3d << " "
871	+	<< fMVAVar_EleIP3dSig << " "
872	+	<< fMVAVar_EleGsfTrackChi2OverNdof << " "
873	+	<< fMVAVar_EledEtaCalo << " "
874	+	<< fMVAVar_EledPhiCalo << " "
875	+	<< fMVAVar_EleR9 << " "
876	+	<< fMVAVar_EleSCEtaWidth << " "
877	+	<< fMVAVar_EleSCPhiWidth << " "
878	+	<< fMVAVar_EleCovIEtaIPhi << " "
879	+	<< fMVAVar_ElePreShowerOverRaw << " "
880	+	<< fMVAVar_EleKFTrkChiSqr  << " "
881	+	<< fMVAVar_EleKFTrkNHits  << " "
882	+	<< fMVAVar_EleE1x5OverE5x5  << " "
883	+	<< " === : === "
884	+	<< mva << " "
885	+	<< std::endl;
886	+
887	+	}
888	+
889	+
890	+
891	+	return mva;
892	+	}
893	+
894	+
895	+
896	+
897	+	//--------------------------------------------------------------------------------------------------
898	+	//MVA Includes Isolation with removal of other leptons
899	+	//
900	+	Double_t ElectronIDMVA::MVAValue(const Electron *ele, const Vertex *vertex,
901	+	const PFCandidateCol *PFCands,
902	+	const PileupEnergyDensityCol *PileupEnergyDensity,
903	+	ElectronTools::EElectronEffectiveAreaTarget EffectiveAreaTarget,
904	+	const ElectronCol *goodElectrons,
905	+	const MuonCol *goodMuons,
906	+	Bool_t printDebug) {
907	+
908	+	if (!fIsInitialized) {
909	+	std::cout << "Error: ElectronIDMVA not properly initialized.\n";
910	+	return -9999;
911	+	}
912	+
913	+	Double_t Rho = 0;
914	+	if (!(TMath::IsNaN(PileupEnergyDensity->At(0)->Rho()) || isinf(PileupEnergyDensity->At(0)->Rho()))) Rho = PileupEnergyDensity->At(0)->Rho();
915	+
916	+	//set all input variables
917	+	fMVAVar_ElePt = ele->Pt();
918	+	fMVAVar_EleEta = ele->SCluster()->Eta();
919	+	fMVAVar_EleSigmaIEtaIEta = ele->CoviEtaiEta() ;
920	+
921	+	if (fMVAType == ElectronIDMVA::kIDEGamma2012TrigV0 ||
922	+	fMVAType == ElectronIDMVA::kIDEGamma2012NonTrigV0 ||
923	+	fMVAType == ElectronIDMVA::kIDHWW2012TrigV0) {
924	+	fMVAVar_EleDEtaIn = TMath::Min(fabs(double(ele->DeltaEtaSuperClusterTrackAtVtx())),0.06); ;
925	+	fMVAVar_EleDPhiIn = TMath::Min(fabs(double(ele->DeltaPhiSuperClusterTrackAtVtx())),0.6);
926	+	fMVAVar_EleFBrem = TMath::Max(double(ele->FBrem()),-1.0);
927	+	fMVAVar_EleEOverP = TMath::Min(double(ele->ESuperClusterOverP()), 20.0);
928	+	fMVAVar_EleESeedClusterOverPout = TMath::Min(double(ele->ESeedClusterOverPout()),20.0);
929	+	fMVAVar_EleOneOverEMinusOneOverP = (1.0/(ele->SCluster()->Energy())) - 1.0 / ele->P();
930	+	fMVAVar_EleGsfTrackChi2OverNdof = TMath::Min(double( ele->BestTrk()->Chi2() / ele->BestTrk()->Ndof()),200.0);
931	+	fMVAVar_EledEtaCalo =  TMath::Min(fabs(double(ele->DeltaEtaSeedClusterTrackAtCalo())),0.2);
932	+	fMVAVar_EleR9 = TMath::Min(double(ele->SCluster()->R9()), 5.0);
933	+	} else {
934	+	fMVAVar_EleDEtaIn = ele->DeltaEtaSuperClusterTrackAtVtx();
935	+	fMVAVar_EleDPhiIn = ele->DeltaPhiSuperClusterTrackAtVtx();
936	+	fMVAVar_EleFBrem = ele->FBrem();
937	+	fMVAVar_EleEOverP = ele->ESuperClusterOverP();
938	+	fMVAVar_EleESeedClusterOverPout = ele->ESeedClusterOverPout();
939	+	fMVAVar_EleOneOverEMinusOneOverP = (1.0/(ele->SCluster()->Energy())) - 1.0 / ele->BestTrk()->P();
940	+	fMVAVar_EleGsfTrackChi2OverNdof = ele->BestTrk()->Chi2() / ele->BestTrk()->Ndof();
941	+	fMVAVar_EledEtaCalo =  ele->DeltaEtaSeedClusterTrackAtCalo();
942	+	fMVAVar_EleR9 = ele->SCluster()->R9();
943	+	}
944	+
945	+	fMVAVar_EleHoverE = ele->HadronicOverEm();
946	+	fMVAVar_EleD0 = ele->BestTrk()->D0Corrected(*vertex);
947	+	fMVAVar_EleDZ = ele->BestTrk()->DzCorrected(*vertex);
948	+	if (!TMath::IsNaN(ele->SCluster()->Seed()->CoviPhiiPhi())) fMVAVar_EleSigmaIPhiIPhi = TMath::Sqrt(ele->SCluster()->Seed()->CoviPhiiPhi());
949	+	else fMVAVar_EleSigmaIPhiIPhi = ele->CoviEtaiEta();
950	+	fMVAVar_EleNBrem = ele->NumberOfClusters() - 1;
951	+	fMVAVar_EleESeedClusterOverPIn = ele->ESeedClusterOverPIn();
952	+	fMVAVar_EleIP3d = ele->Ip3dPV();
953	+	fMVAVar_EleIP3dSig = ele->Ip3dPVSignificance();
954	+	fMVAVar_EledPhiCalo = ele->DeltaPhiSeedClusterTrackAtCalo();
955	+	fMVAVar_EleSCEtaWidth = ele->SCluster()->EtaWidth();
956	+	fMVAVar_EleSCPhiWidth = ele->SCluster()->PhiWidth();
957	+	fMVAVar_EleCovIEtaIPhi = ele->SCluster()->Seed()->CoviEtaiPhi();
958	+	fMVAVar_ElePreShowerOverRaw = ele->SCluster()->PreshowerEnergy() / ele->SCluster()->RawEnergy();
959	+
960	+	//Additional vars
961	+	fMVAVar_EleEEleClusterOverPout = 0;
962	+	if (ele->TrackerTrk()) {
963	+	if (fMVAType == ElectronIDMVA::kIDEGamma2012TrigV0 ||
964	+	fMVAType == ElectronIDMVA::kIDEGamma2012NonTrigV0 ||
965	+	fMVAType == ElectronIDMVA::kIDHWW2012TrigV0 ) {
966	+	fMVAVar_EleKFTrkChiSqr = TMath::Min(double(ele->TrackerTrk()->RChi2()),10.0);
967	+	} else {
968	+	fMVAVar_EleKFTrkChiSqr = ele->TrackerTrk()->RChi2();
969	+	}
970	+	fMVAVar_EleKFTrkNHits = ele->TrackerTrk()->NHits();
971	+	fMVAVar_EleKFTrkNLayers = ele->CTFTrkNLayersWithMeasurement();
972	+	} else {
973	+	fMVAVar_EleKFTrkChiSqr = -1;
974	+	fMVAVar_EleKFTrkNHits = 0;
975	+	fMVAVar_EleKFTrkNLayers = 0;
976	+	}
977	+
978	+	if( ele->SCluster()->Seed()->E5x5() > 0.0 ) {
979	+	if (fMVAType == ElectronIDMVA::kIDEGamma2012TrigV0 ||
980	+	fMVAType == ElectronIDMVA::kIDEGamma2012NonTrigV0 ||
981	+	fMVAType == ElectronIDMVA::kIDHWW2012TrigV0 ) {
982	+	fMVAVar_EleE1x5OverE5x5 = TMath::Min(TMath::Max(1 - double(ele->SCluster()->Seed()->E1x5()/ele->SCluster()->Seed()->E5x5()) , -1.0),2.0);
983	+	} else {
984	+	fMVAVar_EleE1x5OverE5x5 = ele->SCluster()->Seed()->E1x5()/ele->SCluster()->Seed()->E5x5();
985	+	}
986	+	} else {
987	+	fMVAVar_EleE1x5OverE5x5 = -1.0;
988	+	}
989	+
990	+
991	+	Double_t tmpChargedIso_DR0p0To0p1  = 0;
992	+	Double_t tmpChargedIso_DR0p1To0p2  = 0;
993	+	Double_t tmpChargedIso_DR0p2To0p3  = 0;
994	+	Double_t tmpChargedIso_DR0p3To0p4  = 0;
995	+	Double_t tmpChargedIso_DR0p4To0p5  = 0;
996	+	Double_t tmpGammaIso_DR0p0To0p1  = 0;
997	+	Double_t tmpGammaIso_DR0p1To0p2  = 0;
998	+	Double_t tmpGammaIso_DR0p2To0p3  = 0;
999	+	Double_t tmpGammaIso_DR0p3To0p4  = 0;
1000	+	Double_t tmpGammaIso_DR0p4To0p5  = 0;
1001	+	Double_t tmpNeutralHadronIso_DR0p0To0p1  = 0;
1002	+	Double_t tmpNeutralHadronIso_DR0p1To0p2  = 0;
1003	+	Double_t tmpNeutralHadronIso_DR0p2To0p3  = 0;
1004	+	Double_t tmpNeutralHadronIso_DR0p3To0p4  = 0;
1005	+	Double_t tmpNeutralHadronIso_DR0p4To0p5  = 0;
1006	+
1007	+	for (UInt_t p=0; p<PFCands->GetEntries();p++) {
1008	+	const PFCandidate *pf = PFCands->At(p);
1009	+
1010	+	//exclude the electron itself
1011	+	if(pf->GsfTrk() && ele->GsfTrk() &&
1012	+	pf->GsfTrk() == ele->GsfTrk()) continue;
1013	+	if(pf->TrackerTrk() && ele->TrackerTrk() &&
1014	+	pf->TrackerTrk() == ele->TrackerTrk()) continue;
1015	+
1016	+	//************************************************************
1017	+	// New Isolation Calculations
1018	+	//************************************************************
1019	+	Double_t dr = MathUtils::DeltaR(ele->Mom(), pf->Mom());
1020	+
1021	+	if (dr < 1.0) {
1022	+	Bool_t IsLeptonFootprint = kFALSE;
1023	+	//************************************************************
1024	+	// Lepton Footprint Removal
1025	+	//************************************************************
1026	+	for (UInt_t q=0; q < goodElectrons->GetEntries() ; ++q) {
1027	+	//if pf candidate matches an electron passing ID cuts, then veto it
1028	+	if(pf->GsfTrk() && goodElectrons->At(q)->GsfTrk() &&
1029	+	pf->GsfTrk() == goodElectrons->At(q)->GsfTrk()) IsLeptonFootprint = kTRUE;
1030	+	if(pf->TrackerTrk() && goodElectrons->At(q)->TrackerTrk() &&
1031	+	pf->TrackerTrk() == goodElectrons->At(q)->TrackerTrk()) IsLeptonFootprint = kTRUE;
1032	+	//if pf candidate lies in veto regions of electron passing ID cuts, then veto it
1033	+	if(pf->BestTrk() && fabs(goodElectrons->At(q)->SCluster()->Eta()) >= 1.479
1034	+	&& MathUtils::DeltaR(goodElectrons->At(q)->Mom(), pf->Mom()) < 0.015) IsLeptonFootprint = kTRUE;
1035	+	if(pf->PFType() == PFCandidate::eGamma && fabs(goodElectrons->At(q)->SCluster()->Eta()) >= 1.479 &&
1036	+	MathUtils::DeltaR(goodElectrons->At(q)->Mom(), pf->Mom()) < 0.08) IsLeptonFootprint = kTRUE;
1037	+	}
1038	+	for (UInt_t q=0; q < goodMuons->GetEntries() ; ++q) {
1039	+	//if pf candidate matches an muon passing ID cuts, then veto it
1040	+	if(pf->TrackerTrk() && goodMuons->At(q)->TrackerTrk() &&
1041	+	pf->TrackerTrk() == goodMuons->At(q)->TrackerTrk()) IsLeptonFootprint = kTRUE;
1042	+	//if pf candidate lies in veto regions of muon passing ID cuts, then veto it
1043	+	if(pf->BestTrk() && MathUtils::DeltaR(goodMuons->At(q)->Mom(), pf->Mom()) < 0.01) IsLeptonFootprint = kTRUE;
1044	+	}
1045	+
1046	+	if (!IsLeptonFootprint) {
1047	+	Bool_t passVeto = kTRUE;
1048	+	//Charged
1049	+	if(pf->BestTrk()) {
1050	+	if (!(fabs(pf->BestTrk()->DzCorrected(*vertex) - ele->BestTrk()->DzCorrected(*vertex)) < 0.2)) passVeto = kFALSE;
1051	+	//************************************************************
1052	+	// Veto any PFmuon, or PFEle
1053	+	if (pf->PFType() == PFCandidate::eElectron || pf->PFType() == PFCandidate::eMuon) passVeto = kFALSE;
1054	+	//************************************************************
1055	+	//************************************************************
1056	+	// Footprint Veto
1057	+	if (fabs(ele->SCluster()->Eta()) >= 1.479 && dr < 0.015) passVeto = kFALSE;
1058	+	//************************************************************
1059	+	if (passVeto) {
1060	+	if (dr < 0.1) tmpChargedIso_DR0p0To0p1 += pf->Pt();
1061	+	if (dr >= 0.1 && dr < 0.2) tmpChargedIso_DR0p1To0p2 += pf->Pt();
1062	+	if (dr >= 0.2 && dr < 0.3) tmpChargedIso_DR0p2To0p3 += pf->Pt();
1063	+	if (dr >= 0.3 && dr < 0.4) tmpChargedIso_DR0p3To0p4 += pf->Pt();
1064	+	if (dr >= 0.4 && dr < 0.5) tmpChargedIso_DR0p4To0p5 += pf->Pt();
1065	+	} //pass veto
1066	+
1067	+	}
1068	+	//Gamma
1069	+	else if (pf->PFType() == PFCandidate::eGamma) {
1070	+	//************************************************************
1071	+	// Footprint Veto
1072	+	if (fabs(ele->SCluster()->Eta()) >= 1.479) {
1073	+	if (dr < 0.08) passVeto = kFALSE;
1074	+	}
1075	+	//************************************************************
1076	+
1077	+	if (passVeto) {
1078	+	if (dr < 0.1) tmpGammaIso_DR0p0To0p1 += pf->Pt();
1079	+	if (dr >= 0.1 && dr < 0.2) tmpGammaIso_DR0p1To0p2 += pf->Pt();
1080	+	if (dr >= 0.2 && dr < 0.3) tmpGammaIso_DR0p2To0p3 += pf->Pt();
1081	+	if (dr >= 0.3 && dr < 0.4) tmpGammaIso_DR0p3To0p4 += pf->Pt();
1082	+	if (dr >= 0.4 && dr < 0.5) tmpGammaIso_DR0p4To0p5 += pf->Pt();
1083	+	}
1084	+	}
1085	+	//NeutralHadron
1086	+	else {
1087	+	if (dr < 0.1) tmpNeutralHadronIso_DR0p0To0p1 += pf->Pt();
1088	+	if (dr >= 0.1 && dr < 0.2) tmpNeutralHadronIso_DR0p1To0p2 += pf->Pt();
1089	+	if (dr >= 0.2 && dr < 0.3) tmpNeutralHadronIso_DR0p2To0p3 += pf->Pt();
1090	+	if (dr >= 0.3 && dr < 0.4) tmpNeutralHadronIso_DR0p3To0p4 += pf->Pt();
1091	+	if (dr >= 0.4 && dr < 0.5) tmpNeutralHadronIso_DR0p4To0p5 += pf->Pt();
1092	+	}
1093	+	} //not lepton footprint
1094	+	} //in 1.0 dr cone
1095	+	} //loop over PF candidates
1096	+
1097	+	Double_t fMVAVar_ChargedIso_DR0p0To0p1  = 0;
1098	+	Double_t fMVAVar_ChargedIso_DR0p1To0p2  = 0;
1099	+	Double_t fMVAVar_ChargedIso_DR0p2To0p3  = 0;
1100	+	Double_t fMVAVar_ChargedIso_DR0p3To0p4  = 0;
1101	+	Double_t fMVAVar_ChargedIso_DR0p4To0p5  = 0;
1102	+	Double_t fMVAVar_GammaIso_DR0p0To0p1  = 0;
1103	+	Double_t fMVAVar_GammaIso_DR0p1To0p2  = 0;
1104	+	Double_t fMVAVar_GammaIso_DR0p2To0p3  = 0;
1105	+	Double_t fMVAVar_GammaIso_DR0p3To0p4  = 0;
1106	+	Double_t fMVAVar_GammaIso_DR0p4To0p5  = 0;
1107	+	Double_t fMVAVar_NeutralHadronIso_DR0p0To0p1  = 0;
1108	+	Double_t fMVAVar_NeutralHadronIso_DR0p1To0p2  = 0;
1109	+	Double_t fMVAVar_NeutralHadronIso_DR0p2To0p3  = 0;
1110	+	Double_t fMVAVar_NeutralHadronIso_DR0p3To0p4  = 0;
1111	+	Double_t fMVAVar_NeutralHadronIso_DR0p4To0p5  = 0;
1112	+
1113	+	fMVAVar_ChargedIso_DR0p0To0p1   = TMath::Min((tmpChargedIso_DR0p0To0p1)/ele->Pt(), 2.5);
1114	+	fMVAVar_ChargedIso_DR0p1To0p2   = TMath::Min((tmpChargedIso_DR0p1To0p2)/ele->Pt(), 2.5);
1115	+	fMVAVar_ChargedIso_DR0p2To0p3 = TMath::Min((tmpChargedIso_DR0p2To0p3)/ele->Pt(), 2.5);
1116	+	fMVAVar_ChargedIso_DR0p3To0p4 = TMath::Min((tmpChargedIso_DR0p3To0p4)/ele->Pt(), 2.5);
1117	+	fMVAVar_ChargedIso_DR0p4To0p5 = TMath::Min((tmpChargedIso_DR0p4To0p5)/ele->Pt(), 2.5);
1118	+	fMVAVar_GammaIso_DR0p0To0p1 = TMath::Max(TMath::Min((tmpGammaIso_DR0p0To0p1 - Rho*ElectronTools::ElectronEffectiveArea(ElectronTools::kEleGammaIsoDR0p0To0p1, ele->SCluster()->Eta(), EffectiveAreaTarget))/ele->Pt(), 2.5), 0.0);
1119	+	fMVAVar_GammaIso_DR0p1To0p2 = TMath::Max(TMath::Min((tmpGammaIso_DR0p1To0p2 - Rho*ElectronTools::ElectronEffectiveArea(ElectronTools::kEleGammaIsoDR0p1To0p2, ele->SCluster()->Eta(), EffectiveAreaTarget))/ele->Pt(), 2.5), 0.0);
1120	+	fMVAVar_GammaIso_DR0p2To0p3 = TMath::Max(TMath::Min((tmpGammaIso_DR0p2To0p3 - Rho*ElectronTools::ElectronEffectiveArea(ElectronTools::kEleGammaIsoDR0p2To0p3, ele->SCluster()->Eta(), EffectiveAreaTarget))/ele->Pt(), 2.5), 0.0);
1121	+	fMVAVar_GammaIso_DR0p3To0p4 = TMath::Max(TMath::Min((tmpGammaIso_DR0p3To0p4 - Rho*ElectronTools::ElectronEffectiveArea(ElectronTools::kEleGammaIsoDR0p3To0p4, ele->SCluster()->Eta(), EffectiveAreaTarget))/ele->Pt(), 2.5), 0.0);
1122	+	fMVAVar_GammaIso_DR0p4To0p5 = TMath::Max(TMath::Min((tmpGammaIso_DR0p4To0p5 - Rho*ElectronTools::ElectronEffectiveArea(ElectronTools::kEleGammaIsoDR0p4To0p5, ele->SCluster()->Eta(), EffectiveAreaTarget))/ele->Pt(), 2.5), 0.0);
1123	+	fMVAVar_NeutralHadronIso_DR0p0To0p1 = TMath::Max(TMath::Min((tmpNeutralHadronIso_DR0p0To0p1 - Rho*ElectronTools::ElectronEffectiveArea(ElectronTools::kEleNeutralHadronIsoDR0p0To0p1, ele->SCluster()->Eta(), EffectiveAreaTarget))/ele->Pt(), 2.5), 0.0);
1124	+	fMVAVar_NeutralHadronIso_DR0p1To0p2 = TMath::Max(TMath::Min((tmpNeutralHadronIso_DR0p1To0p2 - Rho*ElectronTools::ElectronEffectiveArea(ElectronTools::kEleNeutralHadronIsoDR0p1To0p2, ele->SCluster()->Eta(), EffectiveAreaTarget))/ele->Pt(), 2.5), 0.0);
1125	+	fMVAVar_NeutralHadronIso_DR0p2To0p3 = TMath::Max(TMath::Min((tmpNeutralHadronIso_DR0p2To0p3 - Rho*ElectronTools::ElectronEffectiveArea(ElectronTools::kEleNeutralHadronIsoDR0p2To0p3, ele->SCluster()->Eta(), EffectiveAreaTarget))/ele->Pt(), 2.5), 0.0);
1126	+	fMVAVar_NeutralHadronIso_DR0p3To0p4 = TMath::Max(TMath::Min((tmpNeutralHadronIso_DR0p3To0p4 - Rho*ElectronTools::ElectronEffectiveArea(ElectronTools::kEleNeutralHadronIsoDR0p3To0p4, ele->SCluster()->Eta(), EffectiveAreaTarget))/ele->Pt(), 2.5), 0.0);
1127	+	fMVAVar_NeutralHadronIso_DR0p4To0p5 = TMath::Max(TMath::Min((tmpNeutralHadronIso_DR0p4To0p5 - Rho*ElectronTools::ElectronEffectiveArea(ElectronTools::kEleNeutralHadronIsoDR0p4To0p5, ele->SCluster()->Eta(), EffectiveAreaTarget))/ele->Pt(), 2.5), 0.0);
1128	+
1129	+	//Do Binding of MVA input variables
1130	+	if (   fMVAType == ElectronIDMVA::kIDEGamma2012TrigV0
1131	+	|| fMVAType == ElectronIDMVA::kIDEGamma2012NonTrigV0
1132	+	|| fMVAType == ElectronIDMVA::kIsoRingsV0
1133	+	|| fMVAType == ElectronIDMVA::kIDHWW2012TrigV0) {
1134	+	bindVariables();
1135	+	}
1136	+
1137	+	Double_t mva = -9999;
1138	+	TMVA::Reader *reader = 0;
1139	+
1140	+	if (printDebug == kTRUE) {
1141	+	std::cout <<" -> BIN: " << fMVAVar_EleEta << " " << fMVAVar_ElePt << " : " << GetMVABin( fMVAVar_EleEta , fMVAVar_ElePt) << std::endl;
1142	+	}
1143	+	reader = fTMVAReader[GetMVABin( fMVAVar_EleEta , fMVAVar_ElePt)];
1144	+	mva = reader->EvaluateMVA( fMethodname );
1145	+
1146	+	if (printDebug == kTRUE) {
1147	+
1148	+	std::cout << "Debug Electron MVA: \n";
1149	+	std::cout << " fbrem " <<  fMVAVar_EleFBrem
1150	+	<< " kfchi2 " << fMVAVar_EleKFTrkChiSqr
1151	+	<< " kfhits " << fMVAVar_EleKFTrkNHits
1152	+	<< " kfhitsall " << fMVAVar_EleKFTrkNHits
1153	+	<< " gsfchi2 " << fMVAVar_EleGsfTrackChi2OverNdof
1154	+	<< " deta " <<  fMVAVar_EleDEtaIn
1155	+	<< " dphi " << fMVAVar_EleDPhiIn
1156	+	<< " detacalo " << fMVAVar_EledEtaCalo
1157	+	<< " see " << fMVAVar_EleSigmaIEtaIEta
1158	+	<< " spp " << fMVAVar_EleSigmaIPhiIPhi
1159	+	<< " etawidth " << fMVAVar_EleSCEtaWidth
1160	+	<< " phiwidth " << fMVAVar_EleSCPhiWidth
1161	+	<< " e1x5e5x5 " << fMVAVar_EleE1x5OverE5x5
1162	+	<< " R9 " << fMVAVar_EleR9
1163	+	<< " HoE " << fMVAVar_EleHoverE
1164	+	<< " EoP " << fMVAVar_EleEOverP
1165	+	<< " IoEmIoP " << fMVAVar_EleOneOverEMinusOneOverP
1166	+	<< " eleEoPout " << fMVAVar_EleESeedClusterOverPout
1167	+	<< " EoPout " << fMVAVar_EleESeedClusterOverPout
1168	+	<< " d0 " << fMVAVar_EleD0
1169	+	<< " ip3d " << fMVAVar_EleIP3d
1170	+	<< " eta " << fMVAVar_EleEta
1171	+	<< " pt " << fMVAVar_ElePt << std::endl;
1172	+
1173	+	std::cout << fMVAVar_ChargedIso_DR0p0To0p1 << " "
1174	+	<< fMVAVar_ChargedIso_DR0p1To0p2 << " "
1175	+	<< fMVAVar_ChargedIso_DR0p2To0p3 << " "
1176	+	<< fMVAVar_ChargedIso_DR0p3To0p4 << " "
1177	+	<< fMVAVar_ChargedIso_DR0p4To0p5 << " "
1178	+	<< fMVAVar_GammaIso_DR0p0To0p1 << " "
1179	+	<< fMVAVar_GammaIso_DR0p1To0p2 << " "
1180	+	<< fMVAVar_GammaIso_DR0p2To0p3 << " "
1181	+	<< fMVAVar_GammaIso_DR0p3To0p4 << " "
1182	+	<< fMVAVar_GammaIso_DR0p4To0p5 << " "
1183	+	<< fMVAVar_NeutralHadronIso_DR0p0To0p1 << " "
1184	+	<< fMVAVar_NeutralHadronIso_DR0p1To0p2 << " "
1185	+	<< fMVAVar_NeutralHadronIso_DR0p2To0p3 << " "
1186	+	<< fMVAVar_NeutralHadronIso_DR0p3To0p4 << " "
1187	+	<< fMVAVar_NeutralHadronIso_DR0p4To0p5 << " "
1188	+	<< std::endl;
1189	+	std::cout << "MVA: " << mva << " "
1190	+	<< std::endl;
1191	+	}
1192	+
1193	+	return mva;
1194	+	}
1195	+
1196	+
1197	+	void ElectronIDMVA::bindVariables() {
1198	+
1199	+	// this binding is needed for variables that sometime diverge.
1200	+
1201	+	if(fMVAVar_EleFBrem < -1.)
1202	+	fMVAVar_EleFBrem = -1.;
1203	+
1204	+	fMVAVar_EleDEtaIn = fabs(fMVAVar_EleDEtaIn);
1205	+	if(fMVAVar_EleDEtaIn > 0.06)
1206	+	fMVAVar_EleDEtaIn = 0.06;
1207	+
1208	+
1209	+	fMVAVar_EleDPhiIn = fabs(fMVAVar_EleDPhiIn);
1210	+	if(fMVAVar_EleDPhiIn > 0.6)
1211	+	fMVAVar_EleDPhiIn = 0.6;
1212	+
1213	+
1214	+	if(fMVAVar_EleESeedClusterOverPout > 20.)
1215	+	fMVAVar_EleESeedClusterOverPout = 20.;
1216	+
1217	+	if(fMVAVar_EleEOverP > 20.)
1218	+	fMVAVar_EleEOverP = 20.;
1219	+
1220	+	if(fMVAVar_EleEEleClusterOverPout > 20.)
1221	+	fMVAVar_EleEEleClusterOverPout = 20.;
1222	+
1223	+
1224	+	fMVAVar_EledEtaCalo = fabs(fMVAVar_EledEtaCalo);
1225	+	if(fMVAVar_EledEtaCalo > 0.2)
1226	+	fMVAVar_EledEtaCalo = 0.2;
1227	+
1228	+
1229	+	if(fMVAVar_EleE1x5OverE5x5 < -1.)
1230	+	fMVAVar_EleE1x5OverE5x5 = -1;
1231	+
1232	+	if(fMVAVar_EleE1x5OverE5x5 > 2.)
1233	+	fMVAVar_EleE1x5OverE5x5 = 2.;
1234	+
1235	+
1236	+
1237	+	if(fMVAVar_EleR9 > 5)
1238	+	fMVAVar_EleR9 = 5;
1239	+
1240	+	if(fMVAVar_EleGsfTrackChi2OverNdof > 200.)
1241	+	fMVAVar_EleGsfTrackChi2OverNdof = 200;
1242	+
1243	+
1244	+	if(fMVAVar_EleKFTrkChiSqr > 10.)
1245	+	fMVAVar_EleKFTrkChiSqr = 10.;
1246	+
1247	+	// Needed for a bug in CMSSW_420, fixed in more recent CMSSW versions
1248	+	if(std::isnan(fMVAVar_EleSigmaIPhiIPhi))
1249	+	fMVAVar_EleSigmaIPhiIPhi = 0.;
1250	+
1251	+
1252	+	return;
1253	+	}

Diff Legend

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