ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/cbrown/AnalysisFramework/Plotting/Modules/Systematics.C
Revision: 1.49
Committed: Wed Nov 9 14:25:05 2011 UTC (13 years, 6 months ago) by buchmann
Content type: text/plain
Branch: MAIN
Changes since 1.48: +1 -0 lines
Log Message:
Keeping return format intact for special uses (i.e. replace original reso entry by 0)

File Contents

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