ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

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

Comparing UserCode/MitPhysics/Utils/src/JetIDMVA.cc (file contents):
Revision 1.13 by bendavid, Tue May 29 16:55:43 2012 UTC vs.
Revision 1.14 by bendavid, Tue May 29 17:09:02 2012 UTC

#	Line 58 | Line 58 | void JetIDMVA::Initialize( JetIDMVA::Cut
58		fIsInitialized = kTRUE;
59		fType          = iType;
60		fCutType       = iCutType;
61	+
62	+	std::string lCutId = "JetIdParams";
63	+	if(fType == k42)  lCutId = "PuJetIdOptMVA_wp";
64	+	if(fType == k52)  lCutId = "full_5x_wp";
65	+	if(fType == kCut) lCutId = "PuJetIdCutBased_wp";
66	+	//Load Cut Matrix
67	+	edm::ParameterSet lDefConfig = edm::readPSetsFrom(iCutFileName.Data())->getParameter<edm::ParameterSet>("JetIdParams");
68	+	edm::ParameterSet lConfig    = edm::readPSetsFrom(iCutFileName.Data())->getParameter<edm::ParameterSet>(lCutId);
69	+	std::string lCutType = "Tight";
70	+	if(fCutType == kMedium) lCutType = "Medium";
71	+	if(fCutType == kLoose ) lCutType = "Loose";
72	+	if(fCutType == kMET   ) lCutType = "MET";
73	+	if(fType != kCut) {
74	+	std::string lLowPtCut = "MET";
75	+	std::vector<double> lPt010  = lDefConfig.getParameter<std::vector<double> >(("Pt010_" +lLowPtCut).c_str());
76	+	std::vector<double> lPt1020 = lConfig.getParameter<std::vector<double> >(("Pt1020_"+lCutType).c_str());
77	+	std::vector<double> lPt2030 = lConfig.getParameter<std::vector<double> >(("Pt2030_"+lCutType).c_str());
78	+	std::vector<double> lPt3050 = lConfig.getParameter<std::vector<double> >(("Pt3050_"+lCutType).c_str());
79	+	for(int i0 = 0; i0 < 4; i0++) fMVACut[0][i0] = lPt010 [i0];
80	+	for(int i0 = 0; i0 < 4; i0++) fMVACut[1][i0] = lPt1020[i0];
81	+	for(int i0 = 0; i0 < 4; i0++) fMVACut[2][i0] = lPt2030[i0];
82	+	for(int i0 = 0; i0 < 4; i0++) fMVACut[3][i0] = lPt3050[i0];
83	+	}
84	+	if(fType == kCut) {
85	+	for(int i0 = 0; i0 < 2; i0++) {
86	+	std::string lFullCutType = lCutType;
87	+	if(i0 == 0) lFullCutType = "BetaStar"+ lCutType;
88	+	if(i0 == 1) lFullCutType = "RMS"     + lCutType;
89	+	std::vector<double> pt010  = lConfig.getParameter<std::vector<double> >(("Pt010_" +lFullCutType).c_str());
90	+	std::vector<double> pt1020 = lConfig.getParameter<std::vector<double> >(("Pt1020_"+lFullCutType).c_str());
91	+	std::vector<double> pt2030 = lConfig.getParameter<std::vector<double> >(("Pt2030_"+lFullCutType).c_str());
92	+	std::vector<double> pt3050 = lConfig.getParameter<std::vector<double> >(("Pt3050_"+lFullCutType).c_str());
93	+	if(i0 == 0) {
94	+	for(int i2 = 0; i2 < 4; i2++) fBetaStarCut[0][i2] = pt010 [i2];
95	+	for(int i2 = 0; i2 < 4; i2++) fBetaStarCut[1][i2] = pt1020[i2];
96	+	for(int i2 = 0; i2 < 4; i2++) fBetaStarCut[2][i2] = pt2030[i2];
97	+	for(int i2 = 0; i2 < 4; i2++) fBetaStarCut[3][i2] = pt3050[i2];
98	+	}
99	+	if(i0 == 1) {
100	+	for(int i2 = 0; i2 < 4; i2++) fRMSCut[0][i2] = pt010 [i2];
101	+	for(int i2 = 0; i2 < 4; i2++) fRMSCut[1][i2] = pt1020[i2];
102	+	for(int i2 = 0; i2 < 4; i2++) fRMSCut[2][i2] = pt2030[i2];
103	+	for(int i2 = 0; i2 < 4; i2++) fRMSCut[3][i2] = pt3050[i2];
104	+	}
105	+	}
106	+	return;
107	+	}
108	+
109	+
110		fLowPtReader   = 0;
111		fLowPtReader   = new TMVA::Reader( "!Color:!Silent:Error" );
112		fLowPtReader->AddVariable( "nvtx"     , &fNVtx      );
#	Line 134 | Line 183 | void JetIDMVA::Initialize( JetIDMVA::Cut
183		std::cout << "Jet ID MVA Initialization\n";
184		std::cout << "MethodName : " << fLowPtMethodName << " , type == " << fType << std::endl;
185
137	–	std::string lCutId = "JetIdParams";
138	–	if(fType == k42)  lCutId = "PuJetIdOptMVA_wp";
139	–	if(fType == k52)  lCutId = "full_5x_wp";
140	–	if(fType == kCut) lCutId = "PuJetIdCutBased_wp";
141	–	//Load Cut Matrix
142	–	edm::ParameterSet lDefConfig = edm::readPSetsFrom(iCutFileName.Data())->getParameter<edm::ParameterSet>("JetIdParams");
143	–	edm::ParameterSet lConfig    = edm::readPSetsFrom(iCutFileName.Data())->getParameter<edm::ParameterSet>(lCutId);
144	–	std::string lCutType = "Tight";
145	–	if(fCutType == kMedium) lCutType = "Medium";
146	–	if(fCutType == kLoose ) lCutType = "Loose";
147	–	if(fCutType == kMET   ) lCutType = "MET";
148	–	if(fType != kCut) {
149	–	std::string lLowPtCut = "MET";
150	–	std::vector<double> lPt010  = lDefConfig.getParameter<std::vector<double> >(("Pt010_" +lLowPtCut).c_str());
151	–	std::vector<double> lPt1020 = lConfig.getParameter<std::vector<double> >(("Pt1020_"+lCutType).c_str());
152	–	std::vector<double> lPt2030 = lConfig.getParameter<std::vector<double> >(("Pt2030_"+lCutType).c_str());
153	–	std::vector<double> lPt3050 = lConfig.getParameter<std::vector<double> >(("Pt3050_"+lCutType).c_str());
154	–	for(int i0 = 0; i0 < 4; i0++) fMVACut[0][i0] = lPt010 [i0];
155	–	for(int i0 = 0; i0 < 4; i0++) fMVACut[1][i0] = lPt1020[i0];
156	–	for(int i0 = 0; i0 < 4; i0++) fMVACut[2][i0] = lPt2030[i0];
157	–	for(int i0 = 0; i0 < 4; i0++) fMVACut[3][i0] = lPt3050[i0];
158	–	}
159	–	if(fType == kCut) {
160	–	for(int i0 = 0; i0 < 2; i0++) {
161	–	std::string lFullCutType = lCutType;
162	–	if(i0 == 0) lFullCutType = "BetaStar"+ lCutType;
163	–	if(i0 == 1) lFullCutType = "RMS"     + lCutType;
164	–	std::vector<double> pt010  = lConfig.getParameter<std::vector<double> >(("Pt010_" +lFullCutType).c_str());
165	–	std::vector<double> pt1020 = lConfig.getParameter<std::vector<double> >(("Pt1020_"+lFullCutType).c_str());
166	–	std::vector<double> pt2030 = lConfig.getParameter<std::vector<double> >(("Pt2030_"+lFullCutType).c_str());
167	–	std::vector<double> pt3050 = lConfig.getParameter<std::vector<double> >(("Pt3050_"+lFullCutType).c_str());
168	–	if(i0 == 0) {
169	–	for(int i2 = 0; i2 < 4; i2++) fBetaStarCut[0][i2] = pt010 [i2];
170	–	for(int i2 = 0; i2 < 4; i2++) fBetaStarCut[1][i2] = pt1020[i2];
171	–	for(int i2 = 0; i2 < 4; i2++) fBetaStarCut[2][i2] = pt2030[i2];
172	–	for(int i2 = 0; i2 < 4; i2++) fBetaStarCut[3][i2] = pt3050[i2];
173	–	}
174	–	if(i0 == 1) {
175	–	for(int i2 = 0; i2 < 4; i2++) fRMSCut[0][i2] = pt010 [i2];
176	–	for(int i2 = 0; i2 < 4; i2++) fRMSCut[1][i2] = pt1020[i2];
177	–	for(int i2 = 0; i2 < 4; i2++) fRMSCut[2][i2] = pt2030[i2];
178	–	for(int i2 = 0; i2 < 4; i2++) fRMSCut[3][i2] = pt3050[i2];
179	–	}
180	–	}
181	–	}
186		}
187
188		//--------------------------------------------------------------------------------------------------

Diff Legend

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