ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/cbrown/AnalysisFramework/Plotting/Modules/Systematics.C
Revision: 1.29
Committed: Wed Sep 7 06:42:21 2011 UTC (13 years, 8 months ago) by buchmann
Content type: text/plain
Branch: MAIN
Changes since 1.28: +17 -1 lines
Log Message: 
Added advanced MC efficiency calculation for individual processes (to compute total XS)

File Contents

# User Rev Content
1 buchmann 1.1 #include <iostream>
2     #include <vector>
3     #include <sys/stat.h>
4    
5     #include <TMath.h>
6     #include <TColor.h>
7     #include <TPaveText.h>
8     #include <TRandom.h>
9     #include <TF1.h>
10    
11     #ifndef SampleClassLoaded
12     #include "ActiveSamples.C"
13     #endif
14    
15     #ifndef Verbosity
16     #define Verbosity 0
17     #endif
18    
19     #include <TFile.h>
20     #include <TTree.h>
21     #include <TH1.h>
22     #include <TCut.h>
23     #include <TMath.h>
24     #include <TLine.h>
25     #include <TCanvas.h>
26     #include <TProfile.h>
27     #include <TF1.h>
28    
29    
30    
31     Int_t nBins = 100;
32     Float_t jzbMin = -207;
33     Float_t jzbMax = 243;
34     Float_t jzbSel = 100;
35     int iplot=0;
36     int verbose=0;
37 buchmann 1.2 string geqleq;
38     string mcjzbexpression;
39 buchmann 1.4 bool automatized=false;//if we're running this fully automatized we don't want each function to flood the screen
40 buchmann 1.2
41     TString geq_or_leq() {
42     if(geqleq=="geq") return TString(">=");
43     if(geqleq=="leq") return TString("<=");
44     return TString("GEQ_OR_LEQ_ERROR");
45     }
46 buchmann 1.1
47 buchmann 1.6 TString ngeq_or_leq() {
48     if(geqleq=="geq") return TString("<=");
49     if(geqleq=="leq") return TString(">=");
50     return TString("NGEQ_OR_LEQ_ERROR");
51     }
52    
53 buchmann 1.1 //______________________________________________________________________________
54     Double_t Interpolate(Double_t x, TH1 *histo)
55     {
56     // Given a point x, approximates the value via linear interpolation
57     // based on the two nearest bin centers
58     // Andy Mastbaum 10/21/08
59     // in newer ROOT versions but not in the one I have so I had to work around that ...
60    
61     Int_t xbin = histo->FindBin(x);
62     Double_t x0,x1,y0,y1;
63    
64     if(x<=histo->GetBinCenter(1)) {
65     return histo->GetBinContent(1);
66     } else if(x>=histo->GetBinCenter(histo->GetNbinsX())) {
67     return histo->GetBinContent(histo->GetNbinsX());
68     } else {
69     if(x<=histo->GetBinCenter(xbin)) {
70     y0 = histo->GetBinContent(xbin-1);
71     x0 = histo->GetBinCenter(xbin-1);
72     y1 = histo->GetBinContent(xbin);
73     x1 = histo->GetBinCenter(xbin);
74     } else {
75     y0 = histo->GetBinContent(xbin);
76     x0 = histo->GetBinCenter(xbin);
77     y1 = histo->GetBinContent(xbin+1);
78     x1 = histo->GetBinCenter(xbin+1);
79     }
80     return y0 + (x-x0)*((y1-y0)/(x1-x0));
81     }
82     }
83    
84 buchmann 1.7 //____________________________________________________________________________________
85     // Plotting with all contributions, i.e. sidebands, peak, osof,ossf ... (for a systematic)
86     float allcontributionsplot(TTree* events, TCut kBaseCut, TCut kMassCut, TCut kSidebandCut, TCut JZBPosCut, TCut JZBNegCut) {
87     iplot++;
88     int count=iplot;
89     // Define new histogram
90     string hname=GetNumericHistoName();
91     TH1F* hossfp = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
92     events->Draw(TString(mcjzbexpression)+">>"+TString(hname),kBaseCut&&kMassCut&&JZBPosCut&&cutOSSF,"goff");
93     hname=GetNumericHistoName();
94     TH1F* hossfn = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
95     events->Draw(TString(mcjzbexpression)+">>"+TString(hname),kBaseCut&&kMassCut&&JZBNegCut&&cutOSSF,"goff");
96    
97     hname=GetNumericHistoName();
98     TH1F* hosofp = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
99     events->Draw(TString(mcjzbexpression)+">>"+TString(hname),kBaseCut&&kMassCut&&JZBPosCut&&cutOSOF,"goff");
100     hname=GetNumericHistoName();
101     TH1F* hosofn = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
102     events->Draw(TString(mcjzbexpression)+">>"+TString(hname),kBaseCut&&kMassCut&&JZBNegCut&&cutOSOF,"goff");
103    
104     hname=GetNumericHistoName();
105     TH1F* sbhossfp = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
106     events->Draw(TString(mcjzbexpression)+">>"+TString(hname),kBaseCut&&kSidebandCut&&JZBPosCut&&cutOSSF,"goff");
107     hname=GetNumericHistoName();
108     TH1F* sbhossfn = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
109     events->Draw(TString(mcjzbexpression)+">>"+TString(hname),kBaseCut&&kSidebandCut&&JZBNegCut&&cutOSSF,"goff");
110    
111     hname=GetNumericHistoName();
112     TH1F* sbhosofp = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
113     events->Draw(TString(mcjzbexpression)+">>"+TString(hname),kBaseCut&&kSidebandCut&&JZBPosCut&&cutOSOF,"goff");
114     hname=GetNumericHistoName();
115     TH1F* sbhosofn = new TH1F(hname.c_str(),hname.c_str(),1,-14000,14000);
116     events->Draw(TString(mcjzbexpression)+">>"+TString(hname),kBaseCut&&kSidebandCut&&JZBNegCut&&cutOSOF,"goff");
117    
118     float obs = hossfp->Integral();
119     float pred= hossfn->Integral() + (1.0/3)*( hosofp->Integral() - hosofn->Integral() + sbhossfp->Integral() - sbhossfn->Integral() + sbhosofp->Integral() - sbhosofn->Integral());
120    
121     delete hossfp,hossfn,hosofp,hosofn;
122     delete sbhossfp,sbhossfn,sbhosofp,sbhosofn;
123     return obs-pred;
124     }
125    
126 buchmann 1.1
127     //____________________________________________________________________________________
128     // Efficiency plot
129     TH1F* plotEff(TTree* events, TCut kbase, TString informalname) {
130     iplot++;
131     int count=iplot;
132     // Define new histogram
133     char hname[30]; sprintf(hname,"hJzbEff%d",count);
134     TH1F* hJzbEff = new TH1F(hname,"JZB selection efficiency ; JZB (GeV/c); Efficiency",
135     nBins,jzbMin,jzbMax);
136     Float_t step = (jzbMax-jzbMin)/static_cast<Float_t>(nBins);
137    
138 buchmann 1.6 events->Draw(mcjzbexpression.c_str(),"genJZB>-400"&&kbase,"goff");
139 buchmann 1.1 Float_t maxEff = events->GetSelectedRows();
140 buchmann 1.5 if(verbose>0) dout << hname << " (" << informalname <<") " << maxEff << std::endl;
141 buchmann 1.1
142 buchmann 1.5 if(verbose>0) dout << "JZB max = " << jzbMax << std::endl;
143 buchmann 1.1 // Loop over steps to get efficiency curve
144     char cut[256];
145     for ( Int_t iBin = 0; iBin<nBins; ++iBin ) {
146 buchmann 1.6 sprintf(cut,"genJZB>%3f",jzbMin+iBin*step);
147 buchmann 1.2 events->Draw(mcjzbexpression.c_str(),TCut(cut)&&kbase,"goff");
148 buchmann 1.1 Float_t eff = static_cast<Float_t>(events->GetSelectedRows())/maxEff;
149 buchmann 1.5 // dout << "COUCOU " << __LINE__ << std::endl;
150 buchmann 1.1 hJzbEff->SetBinContent(iBin+1,eff);
151     hJzbEff->SetBinError(iBin+1,TMath::Sqrt(eff*(1-eff)/maxEff));
152     }
153     return hJzbEff;
154    
155    
156     }
157    
158    
159     //________________________________________________________________________________________
160 pablom 1.23 // Master Formula
161     void master_formula(std::vector<float> eff, float &errHi, float &errLo) {
162    
163     float x0 = eff[0];
164     float deltaPos = 0, deltaNeg = 0;
165     for(int k = 0; k < (eff.size()-1)/2; k++) {
166     float xneg = eff[2*k+2];
167     float xpos = eff[2*k+1];
168     if(xpos-x0>0 || xneg-x0>0) {
169     if(xpos-x0 > xneg-x0) {
170     deltaPos += (xpos-x0)*(xpos-x0);
171     } else {
172     deltaPos += (xneg-x0)*(xneg-x0);
173     }
174     }
175     if(x0-xpos>0 || x0-xneg>0) {
176     if(x0-xpos > x0-xneg) {
177     deltaNeg += (xpos-x0)*(xpos-x0);
178     } else {
179     deltaNeg += (xneg-x0)*(xneg-x0);
180     }
181     }
182     }
183     errHi = sqrt(deltaPos);
184     errLo = sqrt(deltaNeg);
185    
186     }
187    
188    
189     //________________________________________________________________________________________
190     // Get normalization factor for the PDFs
191     float get_norm_pdf_factor(TTree *events, int k) {
192    
193 buchmann 1.25 TH1F *haux = new TH1F("haux", "", 10000, 0, 5);
194 pablom 1.23 char nameVar[20];
195     sprintf(nameVar, "pdfW[%d]", k);
196     events->Project("haux", nameVar);
197 buchmann 1.24 float thisW = haux->Integral();
198     events->Project("haux", "pdfW[0]");
199     float normW = haux->Integral();
200    
201     float factor=thisW/normW;
202 pablom 1.23
203     delete haux;
204    
205     return factor;
206    
207     }
208    
209    
210    
211     //________________________________________________________________________________________
212 buchmann 1.1 // Pile-up efficiency
213 buchmann 1.10 float pileup(TTree *events, bool requireZ, string informalname, string addcut="",Float_t myJzbMax = 140. ) {
214 buchmann 1.1 nBins = 16;
215     jzbMax = myJzbMax;
216    
217     // Acceptance cuts
218 buchmann 1.6 TCut kbase("abs(genMll-91.2)<20&&genNjets>2&&genZPt>0&&abs(mll-91.2)<20&&((id1+1)*(id2+1)*ch1*ch2)!=-2");
219 buchmann 1.10 if(addcut!="") kbase=kbase&&addcut.c_str();//this is mostly for SUSY scans (adding requirements on masses)
220    
221 buchmann 1.8 if(requireZ) kbase=kbase&&"TMath::Abs(genMID)==23";
222 buchmann 1.1 TH1F* hLM4 = plotEff(events,kbase,informalname);
223     hLM4->SetMinimum(0.);
224    
225     // Nominal function
226     TF1* func = new TF1("func","0.5*TMath::Erfc([0]*x-[1])",jzbMin,jzbMax);
227     func->SetParameter(0,0.03);
228     func->SetParameter(1,0.);
229     hLM4->Fit(func,"Q");
230    
231     // Pimped-up function
232     TF1* funcUp = (TF1*)func->Clone();
233     funcUp->SetParameter( 0., func->GetParameter(0)/1.1); // 10% systematic error (up in sigma => 0.1 in erfc)
234 buchmann 1.5 if(!automatized) dout << " PU: " << funcUp->Eval(jzbSel) << " " << func->Eval(jzbSel)
235 buchmann 1.1 << "(" << (funcUp->Eval(jzbSel)-func->Eval(jzbSel))/func->Eval(jzbSel)*100. << "%)" << std::endl;
236    
237 buchmann 1.18 return (funcUp->Eval(jzbSel)-func->Eval(jzbSel))/func->Eval(jzbSel);
238 buchmann 1.1
239     }
240    
241     //____________________________________________________________________________________
242 buchmann 1.6 // Effect of peak shifting
243 buchmann 1.10 void PeakError(TTree *events,float &result, string mcjzb, float peakerr,string addcut="") {
244 buchmann 1.6 TString peakup("("+TString(mcjzb)+"+"+TString(any2string(TMath::Abs(peakerr)))+")"+geq_or_leq()+TString(any2string(jzbSel)));
245     TString peakdown("("+TString(mcjzb)+"-"+TString(any2string(TMath::Abs(peakerr)))+")"+geq_or_leq()+TString(any2string(jzbSel)));
246     TString peakcentral("("+TString(mcjzb)+")"+geq_or_leq()+TString(any2string(jzbSel)));
247     TString npeakup("("+TString(mcjzb)+"+"+TString(any2string(TMath::Abs(peakerr)))+")"+ngeq_or_leq()+"-"+TString(any2string(jzbSel)));
248     TString npeakdown("("+TString(mcjzb)+"-"+TString(any2string(TMath::Abs(peakerr)))+")"+ngeq_or_leq()+"-"+TString(any2string(jzbSel)));
249     TString npeakcentral("("+TString(mcjzb)+")"+ngeq_or_leq()+"-"+TString(any2string(jzbSel)));
250    
251     nBins = 1;
252     string informalname="PeakErrorCalculation";
253     float resup,resdown,rescent;
254     for(int i=0;i<3;i++) {
255     string poscut,negcut;
256     if(i==0) {
257     poscut=peakcentral;
258     negcut=npeakcentral;
259     } else if(i==1) {
260     poscut=peakdown;
261     negcut=npeakdown;
262     } else if(i==2) {
263     poscut=peakup;
264     negcut=npeakup;
265     }
266 buchmann 1.10 float res;
267     if(addcut=="") res=allcontributionsplot(events,cutnJets,cutmass,sidebandcut,poscut.c_str(),negcut.c_str());
268     else res=allcontributionsplot(events,cutnJets&&addcut.c_str(),cutmass,sidebandcut,poscut.c_str(),negcut.c_str());
269 buchmann 1.7 if(i==0) rescent=res;
270     else if(i==1) resdown=res;
271     else if(i==2) resup=res;
272 buchmann 1.6 }
273 buchmann 1.18 if(TMath::Abs(rescent-resup)>TMath::Abs(rescent-resdown)) result=(TMath::Abs(rescent-resup)/rescent);
274     else result=(TMath::Abs(rescent-resdown)/rescent);
275 buchmann 1.6 }
276    
277     //____________________________________________________________________________________
278 buchmann 1.1 // Total selection efficiency (MC)
279 pablom 1.23 void MCefficiency(TTree *events,float &result, float &resulterr,string mcjzb,bool requireZ,int Neventsinfile, string addcut="", int k = 0) {
280 buchmann 1.1
281     char jzbSelStr[256]; sprintf(jzbSelStr,"%f",jzbSel);
282     // All acceptance cuts at gen. level
283 buchmann 1.14 //TCut kbase("abs(genMll-91.2)<20&&genNjets>2&&genZPt>0&&genJZB"+geq_or_leq()+TString(jzbSelStr)+"&&genId1==-genId2");
284     TCut kbase("");
285 buchmann 1.8 if(requireZ) kbase=kbase&&"TMath::Abs(genMID)==23";
286 buchmann 1.9 if(addcut!="") kbase=kbase&&addcut.c_str();//this is mostly for SUSY scans (adding requirements on masses)
287 buchmann 1.1 // Corresponding reco. cuts
288 buchmann 1.11 TCut ksel("pfJetGoodNum>2&&abs(mll-91.2)<20&&id1==id2&&"+TString(mcjzb)+geq_or_leq()+TString(jzbSelStr));
289 buchmann 1.15 TCut ksel2("pfJetGoodNum>2&&abs(mll-91.2)<20&&id1==id2&&"+TString(mcjzb)+ngeq_or_leq()+TString("-")+TString(jzbSelStr));
290 pablom 1.23 TCut posSide = kbase&&ksel;
291     TCut negSide = kbase&&ksel2;
292     string sposSide(posSide);
293     string snegSide(negSide);
294     char var[20];
295     sprintf(var, "pdfW[%d]", k);
296     string svar(var);
297     string newPosSide = "(" + sposSide + ")*" + svar;
298     string newNegSide = "(" + snegSide + ")*" + svar;
299    
300 buchmann 1.24 TH1F *effh= new TH1F("effh","effh",1,-14000,14000);
301 buchmann 1.26 if(k>=0)events->Draw((mcjzbexpression+">>effh").c_str(), newPosSide.c_str(),"goff");
302     else events->Draw((mcjzbexpression+">>effh").c_str(), sposSide.c_str(),"goff");
303 buchmann 1.24 Float_t sel = effh->Integral();
304 buchmann 1.21 Float_t nsel=0;
305     if(ConsiderSignalContaminationForLimits) {
306 buchmann 1.26 if(k>=0)events->Draw((mcjzbexpression+">>effh").c_str(), newNegSide.c_str(),"goff");
307     else events->Draw((mcjzbexpression+">>effh").c_str(), snegSide.c_str(),"goff");
308 buchmann 1.24 nsel = effh->Integral();
309 buchmann 1.21 }
310 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.
311 buchmann 1.26 float normFactor = 1;
312     if(k>=0) get_norm_pdf_factor(events, k);
313 pablom 1.23 sel = sel/normFactor;
314     nsel = nsel/normFactor;
315    
316 buchmann 1.13 // events->Draw(mcjzbexpression.c_str(),kbase,"goff");
317     // Float_t tot = events->GetSelectedRows();
318     Float_t tot = Neventsinfile;
319 buchmann 1.1
320 buchmann 1.21 if(ConsiderSignalContaminationForLimits) {
321     result=(sel-nsel)/tot;
322     resulterr=(1.0/tot)*TMath::Sqrt(sel+nsel+(sel-nsel)*(sel-nsel)/tot);
323     } else {//no signal contamination considered:
324     result=(sel)/tot;
325     resulterr=TMath::Sqrt(sel/tot*(1+sel/tot)/tot);
326     }
327 buchmann 1.24 if(!automatized) dout << " MC efficiency: " << result << "+-" << resulterr << " ( JZB>" << jzbSel << " : " << sel << " , JZB<-" << jzbSel << " : " << nsel << " and nevents=" << tot << ") with normFact=" << normFactor << std::endl;
328     delete effh;
329 buchmann 1.1 }
330    
331 buchmann 1.29
332     //____________________________________________________________________________________
333     // Total selection efficiency (MC)
334     vector<float> processMCefficiency(TTree *events,string mcjzb,bool requireZ,int Neventsinfile, string addcut) {
335     vector<float> process_efficiencies;
336     for(int iprocess=0;iprocess<=10;iprocess++) {
337     float this_process_efficiency,efferr;
338     stringstream addcutplus;
339     addcutplus<<addcut<<"&&(process=="<<iprocess<<")";
340     MCefficiency(events,this_process_efficiency, efferr,mcjzb,requireZ,Neventsinfile, addcutplus.str(),-1);
341     process_efficiencies.push_back(this_process_efficiency);
342     }
343     return process_efficiencies;
344     }
345    
346    
347 buchmann 1.10 void JZBefficiency(TTree *events, string informalname, float &jzbeff, float &jzbefferr, bool requireZ, string addcut="") {
348 buchmann 1.6 TCut kbase("abs(genMll-91.2)<20&&genNjets>2&&genZPt>0&&abs(mll-91.2)<20&&((id1+1)*(id2+1)*ch1*ch2)!=-2");
349 buchmann 1.10 if(addcut!="") kbase=kbase&&addcut.c_str();//this is mostly for SUSY scans (adding requirements on masses)
350 buchmann 1.8 if(requireZ) kbase=kbase&&"TMath::Abs(genMID)==23";
351 buchmann 1.1 TH1F* hLM4 = plotEff(events,kbase,informalname);
352     Int_t bin = hLM4->FindBin(jzbSel); // To get the error
353 buchmann 1.8 jzbeff=Interpolate(jzbSel,hLM4);
354     jzbefferr=hLM4->GetBinError(bin);
355 buchmann 1.5 if(!automatized) dout << " Efficiency at JZB==" << jzbSel << std::endl;
356 buchmann 1.8 if(!automatized) dout << " " << jzbeff << "+-" << jzbefferr << std::endl;
357 buchmann 1.1 }
358    
359     //________________________________________________________________________
360     // Effect of energy scale on efficiency
361 buchmann 1.10 void JZBjetScale(TTree *events, float &jesdown, float &jesup, string informalname,bool requireZ,string addcut="",float syst=0.1, Float_t jzbSelection=-1, TString plotName = "" ) {
362 buchmann 1.6 TCut kbase("abs(genMll-91.2)<20&&genZPt>0");
363 buchmann 1.10 if(addcut!="") kbase=kbase&&addcut.c_str();//this is mostly for SUSY scans (adding requirements on masses)
364 buchmann 1.8 if(requireZ) kbase=kbase&&"TMath::Abs(genMID)==23";
365    
366 buchmann 1.1 TCut ksel("abs(mll-91.2)<20&&((id1+1)*(id2+1)*ch1*ch2)!=-2");
367     TCut nJets("pfJetGoodNum>2");
368     stringstream down,up;
369     down << "pfJetGoodNum"<<30*(1-syst)<<">=3";
370     up << "pfJetGoodNum"<<30*(1+syst)<<">=3";
371    
372     TCut nJetsP(up.str().c_str());
373     TCut nJetsM(down.str().c_str());
374    
375     if ( !(plotName.Length()>1) ) plotName = informalname;
376    
377     nBins = 1; jzbMin = jzbSel*0.95; jzbMax = jzbSel*1.05;
378     TH1F* hist = plotEff(events,(kbase&&ksel&&nJets),informalname);
379    
380     TH1F* histp = plotEff(events,(kbase&&ksel&&nJetsP),informalname);
381    
382     TH1F* histm = plotEff(events,(kbase&&ksel&&nJetsM),informalname);
383    
384     // Dump some information
385     Float_t eff = Interpolate(jzbSel,hist);
386     Float_t effp = Interpolate(jzbSel,histp);
387     Float_t effm = Interpolate(jzbSel,histm);
388 buchmann 1.5 if(!automatized) dout << " Efficiency at JZB==" << jzbSel << std::endl;
389     if(!automatized) dout << " JESup: " << effp << " (" << (effp-eff)/eff*100. << "%)" << std::endl;
390     if(!automatized) dout << " central: " << eff << std::endl;
391     if(!automatized) dout << " JESdown: " << effm << " (" << (effm-eff)/eff*100. << "%)" << std::endl;
392 buchmann 1.18 jesup=(effp-eff)/eff;
393     jesdown=(effm-eff)/eff;
394 buchmann 1.1 }
395    
396     //________________________________________________________________________
397     // Effect of energy scale on JZB efficiency
398 buchmann 1.10 void doJZBscale(TTree *events, float &down, float &up, float &syst, float systematic, string informalname, bool requireZ, string addcut) {
399 buchmann 1.1
400 buchmann 1.6 TCut kbase("abs(genMll-91.2)<20&&genZPt>0&&genNjets>2");
401 buchmann 1.10 if(addcut!="") kbase=kbase&&addcut.c_str();//this is mostly for SUSY scans (adding requirements on masses)
402 buchmann 1.8 if(requireZ) kbase=kbase&&"TMath::Abs(genMID)==23";
403 buchmann 1.1 TCut ksel("abs(mll-91.2)<20&&((id1+1)*(id2+1)*ch1*ch2)!=-2");
404    
405     nBins = 50;
406     jzbMin = 0.5*jzbSel;
407     jzbMax = 2.0*jzbSel;
408    
409     TH1F* hist = plotEff(events,kbase&&ksel,informalname);
410    
411     // Dump some information
412     Float_t eff = Interpolate(jzbSel,hist);
413     Float_t effp = Interpolate(jzbSel*(1.+systematic),hist);
414     Float_t effm = Interpolate(jzbSel*(1.-systematic),hist);
415 buchmann 1.5 if(!automatized) dout << " efficiency at JZB==" << jzbSel*(1.+systematic) << "(-"<<syst*100<<"%) : " << effp << " (" << ((effp-eff)/eff)*100. << "%)" << std::endl;
416     if(!automatized) dout << " efficiency at JZB==" << jzbSel << ": " << eff << std::endl;
417     if(!automatized) dout << " efficiency at JZB==" << jzbSel*(1.-systematic) << "(-"<<syst*100<<"%) : " << effm << " (" << ((effm-eff)/eff)*100. << "%)" << std::endl;
418 buchmann 1.18 up=((effp-eff)/eff);
419     down=((effm-eff)/eff);
420 buchmann 1.1 }
421    
422     //________________________________________________________________________
423     // JZB response (true/reco. vs. true)
424 buchmann 1.10 void JZBresponse(TTree *events, bool requireZ, float &resp, float &resperr, string addcut="",bool isMET = kFALSE, Float_t myJzbMax = 200., Int_t nPeriods = 9 ) {
425 buchmann 1.1
426     jzbMin = 20;
427 buchmann 1.6 TCut kbase("abs(genMll-91.2)<20&&genZPt>0&&genNjets>2");
428 buchmann 1.10 if(addcut!="") kbase=kbase&&addcut.c_str();//this is mostly for SUSY scans (adding requirements on masses)
429 buchmann 1.8 if(requireZ) kbase=kbase&&"TMath::Abs(genMID)==23";
430 buchmann 1.1 TCut ksel("abs(mll-91.2)<20&&((id1+1)*(id2+1)*ch1*ch2)!=-2");
431    
432 buchmann 1.8 TProfile* hJzbResp = new TProfile("hJzbResp","JZB response ; JZB true (GeV/c); JZB reco. / JZB true", nPeriods, jzbMin, myJzbMax, "" );
433 buchmann 1.1
434 buchmann 1.6 if (!isMET) events->Project("hJzbResp","("+TString(mcjzbexpression)+")/genJZB:genJZB",kbase&&ksel);
435 buchmann 1.1 else events->Project("hJzbResp","met[4]/genMET:genMET",kbase&&ksel);
436    
437     hJzbResp->SetMaximum(1.2);
438     hJzbResp->SetMinimum(0.2);
439     hJzbResp->Fit("pol0","Q");
440     TF1 *fittedfunction = hJzbResp->GetFunction("pol0");
441 buchmann 1.8 resp=fittedfunction->GetParameter(0);
442     resperr=fittedfunction->GetParError(0);
443     if(!automatized) dout << " Response: " << resp << " +/- " << resperr << endl;
444 buchmann 1.5 delete hJzbResp;
445 buchmann 1.1 }
446    
447    
448 pablom 1.23 //________________________________________________________________________________________
449     // PDF uncertainty
450 buchmann 1.24 float get_pdf_uncertainty(TTree *events, string mcjzb, bool requireZ, int Neventsinfile, int NPdfs, string addcut="") {
451 pablom 1.23 std::vector<float> efficiency;
452 buchmann 1.24 for(int k = 1; k < NPdfs; k++) {
453 pablom 1.23 float result, resulterr;
454     MCefficiency(events, result, resulterr, mcjzb, requireZ, Neventsinfile, addcut, k);
455     efficiency.push_back(result);
456     }
457 buchmann 1.24 float errHi, errLow,err;
458 pablom 1.23 master_formula(efficiency, errHi, errLow);
459 buchmann 1.24 err=errLow;
460     if(errHi>errLow) err=errHi;
461     if(!automatized) dout << " Uncertainty from PDF: " << errLow << " (low) and " << errHi << "(high) ---> Picked " << err << endl;
462     return err;
463 pablom 1.23
464     }
465    
466 buchmann 1.24 int get_npdfs(TTree *events) {
467     int NPDFs;
468     events->SetBranchAddress("NPdfs",&NPDFs);
469     events->GetEntry(1);
470     return NPDFs;
471     }
472    
473 pablom 1.23
474 buchmann 1.22 void do_systematics_for_one_file(TTree *events,int Neventsinfile,string informalname, vector<vector<float> > &results,string mcjzb,string datajzb,float peakerror,bool requireZ=false, string addcut="", bool ismSUGRA=false) {
475 buchmann 1.1 float JetEnergyScaleUncert=0.1;
476     float JZBScaleUncert=0.1;
477 buchmann 1.2 mcjzbexpression=mcjzb;
478 buchmann 1.28 float triggereff=5.0/100;// in range [0,1]
479 buchmann 1.5 dout << "Trigger efficiency not implemented in this script yet, still using external one" << endl;
480 buchmann 1.18 float leptonseleff=2.0/100;// in range [0,1]
481 buchmann 1.5 dout << "Lepton selection efficiency not implemented in this script yet, still using external one" << endl;
482 buchmann 1.1
483 buchmann 1.27 int NPdfs=0;
484     if(ismSUGRA) NPdfs = get_npdfs(events);
485 buchmann 1.29
486 buchmann 1.8 float mceff,mcefferr,jzbeff,jzbefferr;
487 buchmann 1.5 if(!automatized) dout << "MC efficiencies:" << endl;
488 buchmann 1.26 MCefficiency(events,mceff,mcefferr,mcjzb,requireZ,Neventsinfile,addcut,-1);
489 buchmann 1.10 JZBefficiency(events,informalname,jzbeff,jzbefferr,requireZ,addcut);
490 buchmann 1.8 if(!automatized) dout << "JZB efficiency: " << jzbeff << "+/-" << jzbefferr << endl;
491 buchmann 1.1
492 buchmann 1.6 if(!automatized) dout << "Error from Peak position:" << endl;
493     float sysfrompeak=0;
494 buchmann 1.10 PeakError(events,sysfrompeak,mcjzb,peakerror,addcut);
495 buchmann 1.6
496 buchmann 1.5 if(!automatized) dout << "Jet energy scale: " << std::endl;
497 buchmann 1.1 float jesup,jesdown;
498 buchmann 1.10 JZBjetScale(events,jesdown,jesup,informalname,requireZ,addcut,JetEnergyScaleUncert);
499 buchmann 1.1
500 buchmann 1.5 if(!automatized) dout << "JZB scale: " << std::endl;
501 buchmann 1.1 float scaleup,scaledown,scalesyst;
502 buchmann 1.10 doJZBscale(events,scaledown,scaleup,scalesyst,JZBScaleUncert,informalname,requireZ,addcut);
503 buchmann 1.1
504 buchmann 1.5 if(!automatized) dout << "JZB response: " << std::endl;
505 buchmann 1.8 float resp,resperr;
506 buchmann 1.10 JZBresponse(events,requireZ,resp,resperr,addcut);
507 buchmann 1.1
508 buchmann 1.5 if(!automatized) dout << "Pileup: " << std::endl;
509 buchmann 1.10 float resolution=pileup(events,requireZ,informalname,addcut);
510 pablom 1.23
511 buchmann 1.22 float PDFuncert=0;
512 buchmann 1.24 if(ismSUGRA) PDFuncert = get_pdf_uncertainty(events, mcjzb, requireZ, Neventsinfile, NPdfs, addcut);
513 pablom 1.23
514 buchmann 1.5 dout << "_______________________________________________" << endl;
515 buchmann 1.10 dout << " SUMMARY FOR " << informalname << " with JZB>" << jzbSel << " (all in %) ";
516     if(addcut!="") dout << "With additional cut: " << addcut;
517     dout << endl;
518 buchmann 1.18 dout << "MC efficiency: " << mceff << "+/-" << mcefferr << endl; // in range [0,1]
519     dout << "Trigger efficiency: " << triggereff << endl; // in range [0,1]
520     dout << "Lepton Sel Eff: " << leptonseleff << endl; // in range [0,1]
521     dout << "Jet energy scale: " << jesup << " " << jesdown << endl; // in range [0,1]
522     dout << "JZB Scale Uncert: " << scaledown << " " << scaleup << endl; // in range [0,1]
523     dout << "Resolution : " << resolution << endl; // in range [0,1]
524     dout << "From peak : " << sysfrompeak << endl; // in range [0,1]
525 buchmann 1.24 if(ismSUGRA) dout << "PDF uncertainty : " << PDFuncert << endl; // in range [0,1]
526 buchmann 1.18 dout << "JZB efficiency: " << jzbeff << "+/-" << jzbefferr << " (not yet included below) " << endl; // in range [0,1]
527     dout << "JZB response : " << resp << " +/-" << resperr << " (not yet included below) " << endl; // in range [0,1]
528 buchmann 1.1
529 buchmann 1.4 float toterr=0;
530 buchmann 1.18 toterr+=(triggereff)*(triggereff);
531     toterr+=(leptonseleff)*(leptonseleff);
532     if(fabs(jesup)>fabs(jesdown)) toterr+=(jesup*jesup); else toterr+=(jesdown*jesdown);
533     if(fabs(scaleup)>fabs(scaledown)) toterr+=(scaleup*scaleup); else toterr+=(scaledown*scaledown);
534     toterr+=(resolution*resolution);
535     toterr+=(sysfrompeak*sysfrompeak);
536 buchmann 1.22 if(ismSUGRA) toterr+=(PDFuncert*PDFuncert);
537 buchmann 1.18 dout << "TOTAL SYSTEMATICS: " << TMath::Sqrt(toterr) << " --> " << TMath::Sqrt(toterr)*mceff << endl;
538 buchmann 1.19 float systerr=TMath::Sqrt(toterr)*mceff;
539     toterr=TMath::Sqrt(toterr*mceff*mceff+mcefferr*mcefferr);//also includes stat err!
540    
541     dout << "FINAL RESULT : " << 100*mceff << " +/- "<< 100*mcefferr << " (stat) +/- " << 100*systerr << " (syst) %" << endl;
542     dout << " we thus use the sqrt of the sum of the squares of the stat & syst err, which is : " << 100*toterr << endl;
543    
544     //Do not modify the lines below or mess with the order; this order is expected by all limit calculating functions!
545 buchmann 1.4 vector<float> res;
546     res.push_back(jzbSel);
547     res.push_back(mceff);
548     res.push_back(mcefferr);
549     res.push_back(toterr);
550     res.push_back(TMath::Sqrt((mcefferr)*(mcefferr)+(toterr*toterr)));
551 buchmann 1.18 if(fabs(jesup)>fabs(jesdown)) res.push_back(fabs(jesup)); else res.push_back(fabs(jesdown));
552     if(fabs(scaleup)>fabs(scaledown)) res.push_back(fabs(scaleup)); else res.push_back(fabs(scaledown));
553     res.push_back(fabs(resolution));
554 buchmann 1.22 if(ismSUGRA) res.push_back(PDFuncert);
555 buchmann 1.4 results.push_back(res);
556 buchmann 1.1 }
557    
558 buchmann 1.8 vector<vector<float> > compute_systematics(string mcjzb, float mcpeakerror, string datajzb, samplecollection &signalsamples, vector<float> bins, bool requireZ=false) {
559 buchmann 1.4 automatized=true;
560     vector< vector<float> > systematics;
561 buchmann 1.1 for (int isignal=0; isignal<signalsamples.collection.size();isignal++) {
562 buchmann 1.5 dout << "Looking at signal " << (signalsamples.collection)[isignal].filename << endl;
563 buchmann 1.1 for(int ibin=0;ibin<bins.size();ibin++) {
564     jzbSel=bins[ibin];
565 buchmann 1.2 geqleq="geq";
566 buchmann 1.13 do_systematics_for_one_file((signalsamples.collection)[isignal].events,(signalsamples.collection)[isignal].Nentries,(signalsamples.collection)[isignal].samplename,systematics,mcjzb,datajzb,mcpeakerror,requireZ);
567 buchmann 1.1 }//end of bin loop
568     }//end of signal loop
569 buchmann 1.4 return systematics;
570 buchmann 1.1 }