ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/cbrown/AnalysisFramework/Plotting/Modules/Systematics.C
Revision: 1.50
Committed: Wed Nov 9 16:50:36 2011 UTC (13 years, 6 months ago) by buchmann
Content type: text/plain
Branch: MAIN
Changes since 1.49: +20 -17 lines
Log Message:
Updated systematic treatment to have trigger eff considered

File Contents

# Content
1 #include <iostream>
2 #include <vector>
3 #include <sys/stat.h>
4 #include <algorithm>
5 #include <cmath>
6
7 #include <TMath.h>
8 #include <TColor.h>
9 #include <TPaveText.h>
10 #include <TRandom.h>
11 #include <TF1.h>
12
13 #ifndef SampleClassLoaded
14 #include "ActiveSamples.C"
15 #endif
16
17 #ifndef Verbosity
18 #define Verbosity 0
19 #endif
20
21 #include <TFile.h>
22 #include <TTree.h>
23 #include <TH1.h>
24 #include <TCut.h>
25 #include <TMath.h>
26 #include <TLine.h>
27 #include <TCanvas.h>
28 #include <TProfile.h>
29 #include <TF1.h>
30
31
32
33 Int_t nBins = 100;
34 Float_t jzbMin = -207;
35 Float_t jzbMax = 243;
36 Float_t jzbSel = 100;
37 int iplot=0;
38 int verbose=0;
39 string geqleq;
40 string mcjzbexpression;
41 bool automatized=false;//if we're running this fully automatized we don't want each function to flood the screen
42
43 TString geq_or_leq() {
44 if(geqleq=="geq") return TString(">=");
45 if(geqleq=="leq") return TString("<=");
46 return TString("GEQ_OR_LEQ_ERROR");
47 }
48
49 TString ngeq_or_leq() {
50 if(geqleq=="geq") return TString("<=");
51 if(geqleq=="leq") return TString(">=");
52 return TString("NGEQ_OR_LEQ_ERROR");
53 }
54
55 //______________________________________________________________________________
56 Double_t Interpolate(Double_t x, TH1 *histo)
57 {
58 // Given a point x, approximates the value via linear interpolation
59 // based on the two nearest bin centers
60 // Andy Mastbaum 10/21/08
61 // in newer ROOT versions but not in the one I have so I had to work around that ...
62
63 Int_t xbin = histo->FindBin(x);
64 Double_t x0,x1,y0,y1;
65
66 if(x<=histo->GetBinCenter(1)) {
67 return histo->GetBinContent(1);
68 } else if(x>=histo->GetBinCenter(histo->GetNbinsX())) {
69 return histo->GetBinContent(histo->GetNbinsX());
70 } else {
71 if(x<=histo->GetBinCenter(xbin)) {
72 y0 = histo->GetBinContent(xbin-1);
73 x0 = histo->GetBinCenter(xbin-1);
74 y1 = histo->GetBinContent(xbin);
75 x1 = histo->GetBinCenter(xbin);
76 } else {
77 y0 = histo->GetBinContent(xbin);
78 x0 = histo->GetBinCenter(xbin);
79 y1 = histo->GetBinContent(xbin+1);
80 x1 = histo->GetBinCenter(xbin+1);
81 }
82 return y0 + (x-x0)*((y1-y0)/(x1-x0));
83 }
84 }
85
86 //____________________________________________________________________________________
87 // Plotting with all contributions, i.e. sidebands, peak, osof,ossf ... (for a systematic)
88 float allcontributionsplot(TTree* events, TCut kBaseCut, TCut kMassCut, TCut kSidebandCut, TCut JZBPosCut, TCut JZBNegCut, int flipped) {
89 iplot++;
90 int count=iplot;
91 string locmcjzbexpression=mcjzbexpression;
92 // Define new histogram
93 string hname=GetNumericHistoName();
94 TH1F* hossfp = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
95 events->Draw("("+TString(locmcjzbexpression)+")>>"+TString(hname),kBaseCut&&kMassCut&&JZBPosCut&&cutOSSF,"goff");
96 hname=GetNumericHistoName();
97 TH1F* hossfn = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
98 events->Draw("("+TString(locmcjzbexpression)+")>>"+TString(hname),kBaseCut&&kMassCut&&JZBNegCut&&cutOSSF,"goff");
99
100 hname=GetNumericHistoName();
101 TH1F* hosofp = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
102 events->Draw("("+TString(locmcjzbexpression)+")>>"+TString(hname),kBaseCut&&kMassCut&&JZBPosCut&&cutOSOF,"goff");
103 hname=GetNumericHistoName();
104 TH1F* hosofn = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
105 events->Draw("("+TString(locmcjzbexpression)+")>>"+TString(hname),kBaseCut&&kMassCut&&JZBNegCut&&cutOSOF,"goff");
106
107 float obs=0;
108 float pred=0;
109 flag_this_change(__FUNCTION__,__LINE__,true);//PlottingSetup::RestrictToMassPeak
110 if(PlottingSetup::RestrictToMassPeak) {
111 hname=GetNumericHistoName();
112 TH1F* sbhossfp = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
113 events->Draw("("+TString(locmcjzbexpression)+")>>"+TString(hname),kBaseCut&&kSidebandCut&&JZBPosCut&&cutOSSF,"goff");
114 hname=GetNumericHistoName();
115 TH1F* sbhossfn = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
116 events->Draw("("+TString(locmcjzbexpression)+")>>"+TString(hname),kBaseCut&&kSidebandCut&&JZBNegCut&&cutOSSF,"goff");
117
118 hname=GetNumericHistoName();
119 TH1F* sbhosofp = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
120 events->Draw("("+TString(locmcjzbexpression)+")>>"+TString(hname),kBaseCut&&kSidebandCut&&JZBPosCut&&cutOSOF,"goff");
121 hname=GetNumericHistoName();
122 TH1F* sbhosofn = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
123 events->Draw("("+TString(locmcjzbexpression)+")>>"+TString(hname),kBaseCut&&kSidebandCut&&JZBNegCut&&cutOSOF,"goff");
124
125 obs = hossfp->Integral();
126 pred= hossfn->Integral() + (1.0/3)*( hosofp->Integral() - hosofn->Integral() + sbhossfp->Integral() - sbhossfn->Integral() + sbhosofp->Integral() - sbhosofn->Integral());
127
128 if(flipped>0) {
129 obs = hossfn->Integral();
130 pred= hossfp->Integral() - (1.0/3)*( hosofp->Integral() - hosofn->Integral() + sbhossfp->Integral() - sbhossfn->Integral() + sbhosofp->Integral() - sbhosofn->Integral());
131 }
132 delete sbhossfp,sbhossfn,sbhosofp,sbhosofn;
133 } else {
134 obs = hossfp->Integral();
135 pred= hossfn->Integral() + (hosofp->Integral() - hosofn->Integral());
136 if(flipped>0) {
137 obs = hossfn->Integral();
138 pred= hossfp->Integral() - (hosofp->Integral() - hosofn->Integral());;
139 }
140 }
141
142 delete hossfp,hossfn,hosofp,hosofn;
143 return obs-pred;
144 }
145
146
147 //____________________________________________________________________________________
148 // Efficiency plot
149 TH1F* plotEff(TTree* events, TCut kbase, TString informalname, int flipped) {
150 iplot++;
151 int count=iplot;
152 iplot++;
153 int count2=iplot;
154 // Define new histogram
155 char hname[30]; sprintf(hname,"hJzbEff%d",count);
156 char hname2[30]; sprintf(hname2,"hJzbEff%d",count2);
157 TH1F* hJzbEff = new TH1F(hname,"JZB selection efficiency ; JZB (GeV/c); Efficiency",nBins,jzbMin,jzbMax);
158 TH1F* hJzbEff2= new TH1F(hname2,"JZB selection efficiency ; JZB (GeV/c); Efficiency",1,-14000,14000);
159 Float_t step = (jzbMax-jzbMin)/static_cast<Float_t>(nBins);
160
161 if(flipped==0) events->Draw((mcjzbexpression+">>"+(string)hname2).c_str(),("genJZB>-400"&&kbase),"goff");
162 else events->Draw(("-"+mcjzbexpression).c_str(),("genJZB<400"&&kbase),"goff");
163 Float_t maxEff = hJzbEff2->Integral();
164 if(verbose>0) dout << hname << " (" << informalname <<") " << maxEff << std::endl;
165
166 if(verbose>0) dout << "JZB max = " << jzbMax << std::endl;
167 // Loop over steps to get efficiency curve
168 char cut[256];
169 for ( Int_t iBin = 0; iBin<nBins; ++iBin ) {
170 sprintf(cut,"genJZB>%3f",jzbMin+iBin*step);
171 if(flipped>0) sprintf(cut,"genJZB<-%3f",jzbMin+iBin*step);
172 if(flipped==0) events->Draw((mcjzbexpression+">>"+(string)hname2).c_str(),(TCut(cut)&&kbase),"goff");
173 if(flipped==1) events->Draw(("-"+mcjzbexpression+">>"+(string)hname2).c_str(),(TCut(cut)&&kbase),"goff");
174 Float_t eff = static_cast<Float_t>(hJzbEff2->Integral())/maxEff;
175 hJzbEff->SetBinContent(iBin+1,eff);
176 hJzbEff->SetBinError(iBin+1,TMath::Sqrt(eff*(1-eff)/maxEff));
177 }
178 delete hJzbEff2;
179 return hJzbEff;
180 }
181
182
183 //________________________________________________________________________________________
184 // Master Formula
185 void master_formula(std::vector<float> eff, float &errHi, float &errLo) {
186
187 float x0 = eff[0];
188 float deltaPos = 0, deltaNeg = 0;
189 for(int k = 0; k < (eff.size()-1)/2; k++) {
190 float xneg = eff[2*k+2];
191 float xpos = eff[2*k+1];
192 if(xpos-x0>0 || xneg-x0>0) {
193 if(xpos-x0 > xneg-x0) {
194 deltaPos += (xpos-x0)*(xpos-x0);
195 } else {
196 deltaPos += (xneg-x0)*(xneg-x0);
197 }
198 }
199 if(x0-xpos>0 || x0-xneg>0) {
200 if(x0-xpos > x0-xneg) {
201 deltaNeg += (xpos-x0)*(xpos-x0);
202 } else {
203 deltaNeg += (xneg-x0)*(xneg-x0);
204 }
205 }
206 }
207 errHi = sqrt(deltaPos);
208 errLo = sqrt(deltaNeg);
209
210 }
211
212
213 //________________________________________________________________________________________
214 // Get normalization factor for the PDFs
215 float get_norm_pdf_factor(TTree *events, int k, string addcut) {
216
217 TH1F *haux = new TH1F("haux", "", 10000, 0, 5);
218 char nameVar[20];
219 sprintf(nameVar, "pdfW[%d]", k);
220 events->Project("haux", nameVar, addcut.c_str());
221 float thisW = haux->Integral();
222 events->Project("haux", "pdfW[0]");
223 float normW = haux->Integral();
224
225 float factor=thisW/normW;
226
227 delete haux;
228
229 return factor;
230
231 }
232
233
234
235 //________________________________________________________________________________________
236 // Pile-up efficiency
237 float pileup(TTree *events, bool requireZ, string informalname, int flipped, string addcut="",Float_t myJzbMax = 140. ) {
238 nBins = 16;
239 jzbMax = myJzbMax;
240
241 // Acceptance cuts
242 flag_this_change(__FUNCTION__,__LINE__,true);//PlottingSetup::RestrictToMassPeak
243 TCut kbase(PlottingSetup::genMassCut&&"genNjets>2&&genZPt>0"&&cutmass&&cutOSSF);
244 if(addcut!="") kbase=kbase&&addcut.c_str();//this is mostly for SUSY scans (adding requirements on masses)
245
246 if(requireZ&&PlottingSetup::RestrictToMassPeak) kbase=kbase&&"TMath::Abs(genMID)==23";
247 TH1F* hLM4 = plotEff(events,kbase,informalname,flipped);
248 hLM4->SetMinimum(0.);
249
250 // Nominal function
251 TF1* func = new TF1("func","0.5*TMath::Erfc([0]*x-[1])",jzbMin,jzbMax);
252 func->SetParameter(0,0.03);
253 func->SetParameter(1,0.);
254 hLM4->Fit(func,"Q");
255
256 // Pimped-up function
257 TF1* funcUp = (TF1*)func->Clone();
258 funcUp->SetParameter( 0, func->GetParameter(0)/1.1); // 10% systematic error (up in sigma => 0.1 in erfc)
259 if(!automatized) dout << " PU: " << funcUp->Eval(jzbSel) << " " << func->Eval(jzbSel)
260 << "(" << (funcUp->Eval(jzbSel)-func->Eval(jzbSel))/func->Eval(jzbSel)*100. << "%)" << std::endl;
261
262 return (funcUp->Eval(jzbSel)-func->Eval(jzbSel))/func->Eval(jzbSel);
263
264 }
265
266 //____________________________________________________________________________________
267 // Effect of peak shifting
268 void PeakError(TTree *events,float &result, string mcjzb, float peakerr,int flipped,string addcut="") {
269 //Note: the cut used here is something like (JZBEXPRESSION+(peakerr)>50) without all the other cuts, to increase statistics (particularly for scans)
270 TString peakup("("+TString(mcjzb)+"+"+TString(any2string(TMath::Abs(peakerr)))+")"+geq_or_leq()+TString(any2string(jzbSel)));
271 TString peakdown("("+TString(mcjzb)+"-"+TString(any2string(TMath::Abs(peakerr)))+")"+geq_or_leq()+TString(any2string(jzbSel)));
272 TString peakcentral("("+TString(mcjzb)+")"+geq_or_leq()+TString(any2string(jzbSel)));
273 TString npeakup("("+TString(mcjzb)+"+"+TString(any2string(TMath::Abs(peakerr)))+")"+ngeq_or_leq()+"-"+TString(any2string(jzbSel)));
274 TString npeakdown("("+TString(mcjzb)+"-"+TString(any2string(TMath::Abs(peakerr)))+")"+ngeq_or_leq()+"-"+TString(any2string(jzbSel)));
275 TString npeakcentral("("+TString(mcjzb)+")"+ngeq_or_leq()+"-"+TString(any2string(jzbSel)));
276 nBins = 1;
277 string informalname="PeakErrorCalculation";
278 float resup,resdown,rescent;
279 for(int i=0;i<3;i++) {
280 string poscut,negcut;
281 if(i==0) {
282 poscut=peakcentral;
283 negcut=npeakcentral;
284 } else if(i==1) {
285 poscut=peakdown;
286 negcut=npeakdown;
287 } else if(i==2) {
288 poscut=peakup;
289 negcut=npeakup;
290 }
291 float res;
292 if(addcut=="") res=allcontributionsplot(events,cutnJets,cutmass,sidebandcut,poscut.c_str(),negcut.c_str(),flipped);
293 else res=allcontributionsplot(events,cutnJets&&addcut.c_str(),cutmass,sidebandcut,poscut.c_str(),negcut.c_str(),flipped);
294 if(i==0) rescent=res;
295 else if(i==1) resdown=res;
296 else if(i==2) resup=res;
297 }
298 if(TMath::Abs(rescent-resup)>TMath::Abs(rescent-resdown)) result=(TMath::Abs(rescent-resup)/(float)rescent);
299 else result=(TMath::Abs(rescent-resdown)/(float)rescent);
300 }
301
302
303 void MCPartialefficiency(TTree *events,float &result, float &resulterr,int flipped, string mcjzb,bool requireZ,int Neventsinfile, string addcut="", int k = 0, int type = 0) {
304 if(!events) {
305 write_error(__FUNCTION__,"Tree passed for efficiency calculation is invalid!");
306 result=0;resulterr=0;
307 return;
308 }
309
310 char jzbSelStr[256]; sprintf(jzbSelStr,"%f",jzbSel);
311 // All acceptance cuts at gen. level
312 //TCut kbase("abs(genMll-91.2)<20&&genNjets>2&&genZPt>0&&genJZB"+geq_or_leq()+TString(jzbSelStr)+"&&genId1==-genId2");
313 TCut kbase("");
314 if(requireZ&&PlottingSetup::RestrictToMassPeak) kbase=kbase&&"TMath::Abs(genMID)==23";
315 if(addcut!="") kbase=kbase&&addcut.c_str();//this is mostly for SUSY scans (adding requirements on masses)
316 // Corresponding reco. cuts
317
318 TCut acceptance("genPt2 != 0");
319 TCut massId(cutmass&&cutOSSF);
320 TCut njets(cutnJets);
321 TCut jzbp;
322 TCut jzbn;
323 if(flipped==0) {
324 jzbp=TCut((TString(mcjzb)+geq_or_leq()+TString(jzbSelStr)));
325 jzbn=TCut((TString(mcjzb)+ngeq_or_leq()+TString("-")+TString(jzbSelStr)));
326 } else {
327 jzbp=TCut(TString(mcjzb)+ngeq_or_leq()+TString("-")+TString(jzbSelStr));
328 jzbn=TCut(TString(mcjzb)+geq_or_leq()+TString(jzbSelStr));
329 }
330 float ntotal = events->Draw("pt1", kbase, "goff");
331 TCut theCut;
332 switch(type) {
333 case 1:
334 theCut = kbase+acceptance;
335 break;
336 case 2:
337 theCut = kbase+massId;
338 break;
339 case 3:
340 theCut = kbase+massId+njets;
341 break;
342 case 4:
343 theCut = kbase+massId+njets+jzbn;
344 break;
345 default:
346 theCut = kbase+massId+njets+jzbn;
347 break;
348 }
349
350 string stheCut(theCut);
351 char var[20];
352 sprintf(var, "pdfW[%d]", k);
353
354 string svar(var);
355 string newtheCut;
356 if(k>0) newtheCut = "(" + stheCut + ")*" + svar;
357 else newtheCut = "(" + stheCut + ")"; // for k==0 or even k==-1 we don't need to evaluate PDFs
358
359 TH1F *effh= new TH1F("effh","effh",1,-14000,14000);
360 if(k>=0) events->Draw((mcjzbexpression+">>effh").c_str(), newtheCut.c_str(),"goff");
361 else events->Draw((mcjzbexpression+">>effh").c_str(), theCut,"goff");
362 Float_t sel = effh->Integral();
363 Float_t nsel=0;
364 //Corrections due to normalization in the PDF. This has to be applied as well to the number of events in a file if the definition changes at some point.
365 float normFactor = 1;
366 if(k>=0) get_norm_pdf_factor(events, k, addcut);
367 sel = sel/normFactor;
368
369 result=(sel)/ntotal;
370 resulterr=TMath::Sqrt(sel/ntotal*(1+sel/ntotal)/ntotal);
371
372 }
373
374 //____________________________________________________________________________________
375 // Total selection efficiency (MC)
376 //returns the efficiency WITHOUT signal contamination, and the result and resulterr contain the result and the corresponding error
377 Value MCefficiency(TTree *events,float &result, float &resulterr, int flipped, string mcjzb,bool requireZ,int Neventsinfile, string addcut="", int k = 0) {
378 if(!events) {
379 write_error(__FUNCTION__,"Tree passed for efficiency calculation is invalid!");
380 result=0;
381 resulterr=0;
382 return Value(0,0);
383 }
384
385 char jzbSelStr[256]; sprintf(jzbSelStr,"%f",jzbSel);
386 // All acceptance cuts at gen. level
387 //TCut kbase("abs(genMll-91.2)<20&&genNjets>2&&genZPt>0&&genJZB"+geq_or_leq()+TString(jzbSelStr)+"&&genId1==-genId2");
388 TCut kbase("");
389
390 flag_this_change(__FUNCTION__,__LINE__,true);//PlottingSetup::RestrictToMassPeak
391 if(requireZ&&PlottingSetup::RestrictToMassPeak) kbase=kbase&&"TMath::Abs(genMID)==23";
392 if(addcut!="") kbase=kbase&&addcut.c_str();//this is mostly for SUSY scans (adding requirements on masses)
393 // Corresponding reco. cuts
394
395 flag_this_change(__FUNCTION__,__LINE__,true);//PlottingSetup::RestrictToMassPeak
396 TCut ksel;//("pfJetGoodNum>2"&&cutmass&&"id1==id2&&"+TString(mcjzb)+geq_or_leq()+TString(jzbSelStr));
397 TCut ksel2;//("pfJetGoodNum>2"&&cutmass&&"id1==id2&&"+TString(mcjzb)+ngeq_or_leq()+TString("-")+TString(jzbSelStr));
398 flag_this_change(__FUNCTION__,__LINE__,true);//PlottingSetup::RestrictToMassPeak
399 if(PlottingSetup::RestrictToMassPeak||!ConsiderSignalContaminationForLimits) {
400 ksel=TCut("pfJetGoodNum>2"&&cutmass&&"id1==id2&&"+TString(mcjzb)+geq_or_leq()+TString(jzbSelStr));
401 ksel2=TCut("pfJetGoodNum>2"&&cutmass&&"id1==id2&&"+TString(mcjzb)+ngeq_or_leq()+TString("-")+TString(jzbSelStr));
402 if(flipped>0) {
403 ksel=TCut("pfJetGoodNum>2"&&cutmass&&"id1==id2&&"+TString(mcjzb)+ngeq_or_leq()+TString("-")+TString(jzbSelStr));
404 ksel2=TCut("pfJetGoodNum>2"&&cutmass&&"id1==id2&&"+TString(mcjzb)+geq_or_leq()+TString(jzbSelStr));
405 }
406 } else {
407 //for off peak analysis we don't use the OSSF condition here yet so we can recycle these two cuts for the em condition!
408 ksel=TCut("pfJetGoodNum>2"&&cutmass&&(TString(mcjzb)+geq_or_leq()+TString(jzbSelStr)));
409 ksel2=TCut("pfJetGoodNum>2"&&cutmass&&(TString(mcjzb)+ngeq_or_leq()+TString("-")+TString(jzbSelStr)));
410 if(flipped>0) {
411 ksel=TCut("pfJetGoodNum>2"&&cutmass&&(TString(mcjzb)+ngeq_or_leq()+TString("-")+TString(jzbSelStr)));
412 ksel2=TCut("pfJetGoodNum>2"&&cutmass&&(TString(mcjzb)+geq_or_leq()+TString(jzbSelStr)));
413 }
414 }
415
416 TCut posSide = kbase&&ksel;
417 TCut negSide = kbase&&ksel2;
418 string sposSide(posSide);
419 string snegSide(negSide);
420 char var[20];
421 sprintf(var, "pdfW[%d]", k);
422 if(k==-1) sprintf(var,"1.0");//case in which we don't want to evaluate PDFs
423 string svar(var);
424 string newPosSide = "((id1==id2)&&(" + sposSide + "))*" + svar;
425 string newNegSide = "((id1==id2)&&(" + snegSide + "))*" + svar;
426 string emnewPosSide = "((id1!=id2)&&(" + sposSide + "))*" + svar; // only used for off peak analysis
427 string emnewNegSide = "((id1!=id2)&&(" + snegSide + "))*" + svar; // only used for off peak analysis
428
429 TH1F *effh= new TH1F("effh","effh",1,-14000,14000);
430 if(k>=0)events->Draw((mcjzbexpression+">>effh").c_str(), TCut(newPosSide.c_str())*PlottingSetup::Weight,"goff");
431 else events->Draw((mcjzbexpression+">>effh").c_str(), TCut((sposSide+"&&(id1==id2)").c_str())*PlottingSetup::Weight,"goff");//the OSSF condition is added for the offpeak analysis, in onpeak case it's there already but doesn't change anything.
432 Float_t sel = effh->Integral();
433 Float_t nsel=0;
434
435 ///----------------------------------------------- THIS PART REQUIRES STUDYING! -------------------------
436
437 if(ConsiderSignalContaminationForLimits) {
438 flag_this_change(__FUNCTION__,__LINE__,true);//PlottingSetup::RestrictToMassPeak
439 if(PlottingSetup::RestrictToMassPeak) {
440 events->Draw((mcjzbexpression+">>effh").c_str(), (TCut(newNegSide.c_str()))*PlottingSetup::Weight,"goff");
441 nsel += effh->Integral();
442 } else {
443 events->Draw((mcjzbexpression+">>effh").c_str(), (TCut(newNegSide.c_str()))*PlottingSetup::Weight,"goff");
444 nsel += effh->Integral();
445 events->Draw((mcjzbexpression+">>effh").c_str(), (TCut(emnewPosSide.c_str()))*PlottingSetup::Weight,"goff");
446 nsel += effh->Integral();
447 events->Draw((mcjzbexpression+">>effh").c_str(), (TCut(emnewNegSide.c_str()))*PlottingSetup::Weight,"goff");
448 nsel -= effh->Integral();
449 }
450 }
451
452 //Corrections due to normalization in the PDF. This has to be applied as well to the number of events in a file if the definition changes at some point.
453 float normFactor = 1;
454 if(k>=0) get_norm_pdf_factor(events, k, addcut);
455 sel = sel/normFactor;
456 nsel = nsel/normFactor;
457
458 // events->Draw(mcjzbexpression.c_str(),kbase,"goff");
459 // Float_t tot = events->GetSelectedRows();
460 Float_t tot = Neventsinfile;
461
462 Value result_wo_signalcont;
463
464 if(ConsiderSignalContaminationForLimits) {
465 result=(sel-nsel)/tot;
466 resulterr=(1.0/tot)*TMath::Sqrt(sel+nsel+(sel-nsel)*(sel-nsel)/tot);
467 result_wo_signalcont=Value(sel/tot,TMath::Sqrt(sel/tot*(1+sel/tot)/tot));
468 } else {//no signal contamination considered:
469 result=(sel)/tot;
470 resulterr=TMath::Sqrt(sel/tot*(1+sel/tot)/tot);
471 result_wo_signalcont=Value(result,resulterr);
472 }
473 if(!automatized && k>0 ) dout << "PDF assessment [" << k << "] : ";
474 if(!automatized) dout << " MC efficiency: " << result << "+-" << resulterr << " ( JZB>" << jzbSel << " : " << sel << " , signal contamination : " << nsel << " and nevents=" << tot << ") with normFact=" << normFactor << std::endl;
475 delete effh;
476 return result_wo_signalcont;
477 }
478
479
480
481 //____________________________________________________________________________________
482 // Selection efficiency for one process (MC)
483 // not in use anymore.
484 /*
485 vector<float> processMCefficiency(TTree *events,int flipped, string mcjzb,bool requireZ,int Neventsinfile, string addcut) {
486 vector<float> process_efficiencies;
487 for(int iprocess=0;iprocess<=10;iprocess++) {
488 float this_process_efficiency,efferr;
489 stringstream addcutplus;
490 addcutplus<<addcut<<"&&(process=="<<iprocess<<")";
491 MCefficiency(events,this_process_efficiency, efferr,flipped,mcjzb,requireZ,Neventsinfile, addcutplus.str(),-1);
492 process_efficiencies.push_back(this_process_efficiency);
493 }
494 return process_efficiencies;
495 }
496 */
497
498 void JZBefficiency(TTree *events, string informalname, float &jzbeff, float &jzbefferr, int flipped, bool requireZ, string addcut="") {
499 TCut kbase(genMassCut&&"genNjets>2&&genZPt>0"&&cutmass&&cutOSSF);
500 if(addcut!="") kbase=kbase&&addcut.c_str();//this is mostly for SUSY scans (adding requirements on masses)
501 if(requireZ&&PlottingSetup::RestrictToMassPeak) kbase=kbase&&"TMath::Abs(genMID)==23";
502 TH1F* hLM4 = plotEff(events,kbase,informalname,flipped);
503 Int_t bin = hLM4->FindBin(jzbSel); // To get the error
504 jzbeff=Interpolate(jzbSel,hLM4);
505 jzbefferr=hLM4->GetBinError(bin);
506 if(!automatized) dout << " Efficiency at JZB==" << jzbSel << std::endl;
507 if(!automatized) dout << " " << jzbeff << "+-" << jzbefferr << std::endl;
508 }
509
510 //________________________________________________________________________
511 // Effect of energy scale on efficiency
512 void JZBjetScale(TTree *events, float &jesdown, float &jesup, string informalname, int flipped, bool requireZ,string addcut="",Float_t jzbSelection=-1, TString plotName = "" ) {
513 TCut kbase(genMassCut&&"genZPt>0");
514 if(addcut!="") kbase=kbase&&addcut.c_str();//this is mostly for SUSY scans (adding requirements on masses)
515 flag_this_change(__FUNCTION__,__LINE__,true);//PlottingSetup::RestrictToMassPeak
516 if(requireZ&&PlottingSetup::RestrictToMassPeak) kbase=kbase&&"TMath::Abs(genMID)==23";
517
518 TCut ksel(cutmass&&cutOSSF);
519 TCut nJets("pfJetGoodNum>2");
520 stringstream down,up;
521 down << "pfJetGoodNumn1sigma>=3";
522 up << "pfJetGoodNump1sigma>=3";
523
524 TCut nJetsP(up.str().c_str());
525 TCut nJetsM(down.str().c_str());
526
527 if ( !(plotName.Length()>1) ) plotName = informalname;
528
529 nBins = 1; jzbMin = jzbSel*0.95; jzbMax = jzbSel*1.05;
530 TH1F* hist = plotEff(events,(kbase&&ksel&&nJets),informalname,flipped);
531
532 TH1F* histp = plotEff(events,(kbase&&ksel&&nJetsP),informalname,flipped);
533
534 TH1F* histm = plotEff(events,(kbase&&ksel&&nJetsM),informalname,flipped);
535
536 // Dump some information
537 Float_t eff = Interpolate(jzbSel,hist);
538 Float_t effp = Interpolate(jzbSel,histp);
539 Float_t effm = Interpolate(jzbSel,histm);
540 if(!automatized) dout << " Efficiency at JZB==" << jzbSel << std::endl;
541 if(!automatized) dout << " JESup: " << effp << " (" << (effp-eff)/eff*100. << "%)" << std::endl;
542 if(!automatized) dout << " central: " << eff << std::endl;
543 if(!automatized) dout << " JESdown: " << effm << " (" << (effm-eff)/eff*100. << "%)" << std::endl;
544 jesup=(effp-eff)/eff;
545 jesdown=(effm-eff)/eff;
546 }
547
548 //________________________________________________________________________
549 // Effect of energy scale on JZB efficiency
550 void doJZBscale(TTree *events, float &down, float &up, float &syst, float systematic, string informalname, int flipped, bool requireZ, string addcut) {
551
552 TCut kbase(genMassCut&&"genZPt>0&&genNjets>2");
553 if(addcut!="") kbase=kbase&&addcut.c_str();//this is mostly for SUSY scans (adding requirements on masses)
554 flag_this_change(__FUNCTION__,__LINE__,true);//PlottingSetup::RestrictToMassPeak
555 if(requireZ&&PlottingSetup::RestrictToMassPeak) kbase=kbase&&"TMath::Abs(genMID)==23";
556 TCut ksel(cutmass&&cutOSSF);
557
558 nBins = 50;
559 jzbMin = 0.5*jzbSel;
560 jzbMax = 2.0*jzbSel;
561
562 TH1F* hist = plotEff(events,kbase&&ksel,informalname,flipped);
563
564 // Dump some information
565 Float_t eff = Interpolate(jzbSel,hist);
566 Float_t effp = Interpolate(jzbSel*(1.+systematic),hist);
567 Float_t effm = Interpolate(jzbSel*(1.-systematic),hist);
568 if(!automatized) dout << " efficiency at JZB==" << jzbSel*(1.+systematic) << "(-"<<systematic*100<<"%) : " << effp << " (" << ((effp-eff)/eff)*100. << "%)" << std::endl;
569 if(!automatized) dout << " efficiency at JZB==" << jzbSel << ": " << eff << std::endl;
570 if(!automatized) dout << " efficiency at JZB==" << jzbSel*(1.-systematic) << "(-"<<systematic*100<<"%) : " << effm << " (" << ((effm-eff)/eff)*100. << "%)" << std::endl;
571 up=((effp-eff)/eff);
572 down=((effm-eff)/eff);
573 }
574
575 //________________________________________________________________________
576 // JZB response (true/reco. vs. true)
577 void JZBresponse(TTree *events, bool requireZ, float &resp, float &resperr, int flipped, string addcut="", bool isMET = kFALSE, Float_t myJzbMax = 200., Int_t nPeriods = 9 ) {
578
579 jzbMin = 20;
580 flag_this_change(__FUNCTION__,__LINE__,false);//PlottingSetup::RestrictToMassPeak
581 TCut kbase(genMassCut&&"genZPt>0&&genNjets>2");
582 if(addcut!="") kbase=kbase&&addcut.c_str();//this is mostly for SUSY scans (adding requirements on masses)
583 flag_this_change(__FUNCTION__,__LINE__,false);//PlottingSetup::RestrictToMassPeak
584 if(requireZ&&PlottingSetup::RestrictToMassPeak) kbase=kbase&&"TMath::Abs(genMID)==23";
585 flag_this_change(__FUNCTION__,__LINE__,false);//PlottingSetup::RestrictToMassPeak
586 TCut ksel(cutmass&&cutOSSF);
587
588 TProfile* hJzbResp = new TProfile("hJzbResp","JZB response ; JZB true (GeV/c); JZB reco. / JZB true", nPeriods, jzbMin, myJzbMax, "" );
589
590 string locmcjzbexpression=mcjzbexpression;
591 if(flipped>0) locmcjzbexpression="-"+locmcjzbexpression;
592 string possibleminus="";
593 if(flipped>0) possibleminus="-";
594 if (!isMET) events->Project("hJzbResp","("+TString(locmcjzbexpression)+")/("+possibleminus+"genJZB):("+possibleminus+"genJZB)",kbase&&ksel);
595 else events->Project("hJzbResp","met[4]/genMET:genMET",kbase&&ksel);
596
597 hJzbResp->SetMaximum(1.2);
598 hJzbResp->SetMinimum(0.2);
599 hJzbResp->Fit("pol0","Q");
600 TF1 *fittedfunction = hJzbResp->GetFunction("pol0");
601 if(!fittedfunction) {
602 // in case there are not enough points passing our selection
603 cout << "OOPS response function invalid, assuming 100% error !!!!" << endl;
604 resp=1;
605 resperr=1;
606 } else {
607 resp=fittedfunction->GetParameter(0);
608 resperr=fittedfunction->GetParError(0);
609 if(!automatized) dout << " Response: " << resp << " +/- " << resperr << endl;
610 }
611 delete hJzbResp;
612 }
613
614
615 //________________________________________________________________________________________
616 // PDF uncertainty
617 float get_pdf_uncertainty(TTree *events, int flipped, string mcjzb, bool requireZ, int Neventsinfile, int NPdfs, string addcut="") {
618 std::vector<float> efficiency;
619 for(int k = 1; k < NPdfs; k++) {
620 float result, resulterr;
621 Value flipval;
622 MCefficiency(events, result, resulterr, flipped, mcjzb, requireZ, Neventsinfile, addcut, k);
623 efficiency.push_back(result);
624 }
625 float errHi, errLow,err;
626 master_formula(efficiency, errHi, errLow);
627 err=errLow;
628 if(errHi>errLow) err=errHi;
629 if(!automatized) dout << " Uncertainty from PDF: " << errLow << " (low) and " << errHi << "(high) ---> Picked " << err << endl;
630 return err;
631
632 }
633
634 int get_npdfs(TTree *events) {
635 int NPDFs;
636 events->SetBranchAddress("NPdfs",&NPDFs);
637 events->GetEntry(1);
638 return NPDFs;
639 }
640
641
642 void do_systematics_for_one_file(TTree *events,int Neventsinfile,string informalname, vector<vector<float> > &results,int flipped, string mcjzb,string datajzb,float peakerror,bool requireZ=false, string addcut="", bool ismSUGRA=false) {
643 float JZBScaleUncert=0.05;
644 mcjzbexpression=mcjzb;
645 float triggereff=2.0/100;// in range [0,1]
646 dout << "Trigger efficiency not implemented in this script yet, still using external one" << endl;
647 float leptonseleff=2.0/100;// in range [0,1]
648 leptonseleff=TMath::Sqrt(leptonseleff*leptonseleff+leptonseleff*leptonseleff); // because the 2% is per lepton
649 dout << "Lepton selection efficiency not implemented in this script yet, still using external one" << endl;
650
651 int NPdfs=0;
652 if(ismSUGRA) NPdfs = get_npdfs(events);
653
654 float mceff,mcefferr,jzbeff,jzbefferr;
655 if(!automatized) dout << "MC efficiencies:" << endl;
656 Value flipefficiency;
657 Value mceff_nosigcont = MCefficiency(events,mceff,mcefferr,flipped,mcjzb,requireZ,Neventsinfile,addcut,-1);
658 if(!automatized) cout << " Without signal contamination, we find an efficiency of " << mceff_nosigcont << endl;
659
660 if(PlottingSetup::computeJZBefficiency) JZBefficiency(events,informalname,jzbeff,jzbefferr,flipped,requireZ,addcut);
661 if(!automatized) dout << "JZB efficiency: " << jzbeff << "+/-" << jzbefferr << endl;
662
663 if(!automatized) dout << "Error from Peak position:" << endl;
664 float sysfrompeak=0;
665 PeakError(events,sysfrompeak,mcjzb,peakerror,flipped,addcut);
666
667 if(!automatized) dout << "Jet energy scale: " << std::endl;
668 float jesup,jesdown;
669 JZBjetScale(events,jesdown,jesup,informalname,flipped,requireZ,addcut);
670
671 if(!automatized) dout << "JZB scale: " << std::endl;
672 float scaleup,scaledown,scalesyst;
673 doJZBscale(events,scaledown,scaleup,scalesyst,JZBScaleUncert,informalname,flipped,requireZ,addcut);
674
675 if(!automatized) dout << "JZB response: " << std::endl;
676 float resp,resperr;
677 if(PlottingSetup::computeJZBresponse) {
678 if(!automatized) dout << "JZB response: " << std::endl;
679 if(!ismSUGRA) JZBresponse(events,requireZ,resp,resperr,flipped,addcut);
680 }
681
682 if(!automatized) dout << "Pileup: " << std::endl;
683 // float resolution;
684 //resolution=pileup(events,requireZ,informalname,flipped,addcut);
685
686 float PDFuncert=0;
687 if(!automatized) dout << "Assessing PDF uncertainty: " << std::endl;
688 if(ismSUGRA) PDFuncert = get_pdf_uncertainty(events, flipped, mcjzb, requireZ, Neventsinfile, NPdfs, addcut);
689
690 dout << "_______________________________________________" << endl;
691 dout << " SUMMARY FOR " << informalname << " with JZB>" << jzbSel << " (all in %) ";
692 if(addcut!="") dout << "With additional cut: " << addcut;
693 dout << endl;
694 dout << "MC efficiency: " << mceff << "+/-" << mcefferr << endl; // in range [0,1]
695 dout << "Trigger efficiency: " << triggereff << endl; // in range [0,1]
696 dout << "Lepton Sel Eff: " << leptonseleff << endl; // in range [0,1]
697 dout << "Jet energy scale: " << jesup << " " << jesdown << endl; // in range [0,1]
698 dout << "JZB Scale Uncert: " << scaledown << " " << scaleup << endl; // in range [0,1]
699 // dout << "Resolution : " << resolution << endl; // in range [0,1]
700 dout << "From peak : " << sysfrompeak << endl; // in range [0,1]
701 if(ismSUGRA) dout << "PDF uncertainty : " << PDFuncert << endl; // in range [0,1]
702 if(PlottingSetup::computeJZBefficiency) dout << "JZB efficiency: " << jzbeff << "+/-" << jzbefferr << " (not yet included below) " << endl; // in range [0,1]
703 if(PlottingSetup::computeJZBresponse)dout << "JZB response : " << resp << " +/-" << resperr << " (not yet included below) " << endl; // in range [0,1]
704
705 float toterr=0;
706 toterr+=(triggereff)*(triggereff);
707 toterr+=(leptonseleff)*(leptonseleff);
708 if(fabs(jesup)>fabs(jesdown)) toterr+=(jesup*jesup); else toterr+=(jesdown*jesdown);
709 if(fabs(scaleup)>fabs(scaledown)) toterr+=(scaleup*scaleup); else toterr+=(scaledown*scaledown);
710 // toterr+=(resolution*resolution);
711 toterr+=(sysfrompeak*sysfrompeak);
712 if(ismSUGRA) toterr+=(PDFuncert*PDFuncert);
713 dout << "TOTAL SYSTEMATICS: " << TMath::Sqrt(toterr) << " --> " << TMath::Sqrt(toterr)*mceff << endl;
714 float systerr=TMath::Sqrt(toterr)*mceff;
715 toterr=TMath::Sqrt(toterr*mceff*mceff+mcefferr*mcefferr);//also includes stat err!
716
717 dout << "FINAL RESULT : " << 100*mceff << " +/- "<< 100*mcefferr << " (stat) +/- " << 100*systerr << " (syst) %" << endl;
718 dout << " we thus use the sqrt of the sum of the squares of the stat & syst err, which is : " << 100*toterr << endl;
719 dout << "_______________________________________________" << endl;
720
721 //Do not modify the lines below or mess with the order; this order is expected by all limit calculating functions!
722 vector<float> res;
723 res.push_back(jzbSel);
724 res.push_back(mceff);
725 res.push_back(mcefferr);
726 res.push_back(toterr);
727 res.push_back(TMath::Sqrt((mcefferr)*(mcefferr)+(toterr*toterr)));
728 if(fabs(jesup)>fabs(jesdown)) res.push_back(fabs(jesup)); else res.push_back(fabs(jesdown));
729 if(fabs(scaleup)>fabs(scaledown)) res.push_back(fabs(scaleup)); else res.push_back(fabs(scaledown));
730 // res.push_back(fabs(resolution));
731 res.push_back(0.0);
732 res.push_back(mceff_nosigcont.getValue());
733 res.push_back(mceff_nosigcont.getError());
734 if(ismSUGRA) res.push_back(PDFuncert);
735 results.push_back(res);
736 }
737
738 vector<vector<float> > compute_systematics(string mcjzb, float mcpeakerror, int flipped, string datajzb, samplecollection &signalsamples, vector<float> bins, bool requireZ=false) {
739 automatized=true;
740 vector< vector<float> > systematics;
741 for (int isignal=0; isignal<signalsamples.collection.size();isignal++) {
742 dout << "Looking at signal " << (signalsamples.collection)[isignal].filename << endl;
743 for(int ibin=0;ibin<bins.size();ibin++) {
744 jzbSel=bins[ibin];
745 geqleq="geq";
746 do_systematics_for_one_file((signalsamples.collection)[isignal].events,(signalsamples.collection)[isignal].Nentries,(signalsamples.collection)[isignal].samplename,systematics,flipped,mcjzb,datajzb,mcpeakerror,requireZ);
747 }//end of bin loop
748 }//end of signal loop
749 return systematics;
750 }