ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/cbrown/AnalysisFramework/Plotting/Modules/LimitCalculation.C
Revision: 1.9
Committed: Mon Aug 8 14:20:46 2011 UTC (13 years, 9 months ago) by fronga
Content type: text/plain
Branch: MAIN
Changes since 1.8: +3 -3 lines
Log Message:
Adapt to new interface (includes error on # observed events)

File Contents

# Content
1 #include <iostream>
2 #include <vector>
3 #include <sys/stat.h>
4 #include <fstream>
5
6 #include <TCut.h>
7 #include <TROOT.h>
8 #include <TCanvas.h>
9 #include <TMath.h>
10 #include <TColor.h>
11 #include <TPaveText.h>
12 #include <TRandom.h>
13 #include <TH1.h>
14 #include <TH2.h>
15 #include <TF1.h>
16 #include <TSQLResult.h>
17 #include <TProfile.h>
18
19 //#include "TTbar_stuff.C"
20 using namespace std;
21
22 using namespace PlottingSetup;
23
24
25 void rediscover_the_top(string mcjzb, string datajzb) {
26 dout << "Hi! today we are going to (try to) rediscover the top!" << endl;
27 TCanvas *c3 = new TCanvas("c3","c3");
28 c3->SetLogy(1);
29 vector<float> binning;
30 //binning=allsamples.get_optimal_binsize(mcjzb,cutmass&&cutOSSF&&cutnJets,20,50,800);
31 /*
32 binning.push_back(50);
33 binning.push_back(100);
34 binning.push_back(150);
35 binning.push_back(200);
36 binning.push_back(500);
37
38
39 TH1F *dataprediction = allsamples.Draw("dataprediction", "-"+datajzb, binning, "JZB [GeV]", "events", cutmass&&cutOSSF&&cutnJets,data, luminosity);
40 TH1F *puresignal = allsamples.Draw("puresignal", datajzb, binning, "JZB [GeV]", "events", cutmass&&cutOSOF&&cutnJets,data, luminosity);
41 // TH1F *puresignal = allsamples.Draw("puresignal", mcjzb, binning, "JZB [GeV]", "events", cutmass&&cutOSSF&&cutnJets,mc, luminosity,allsamples.FindSample("TTJets"));
42 TH1F *observed = allsamples.Draw("observed", datajzb,binning, "JZB [GeV]", "events", cutmass&&cutOSSF&&cutnJets,data,luminosity);
43 /*
44 ofstream myfile;
45 TH1F *ratio = (TH1F*)observed->Clone();
46 ratio->Divide(dataprediction);
47 ratio->GetYaxis()->SetTitle("Ratio obs/pred");
48 ratio->Draw();
49 c3->SaveAs("testratio.png");
50 myfile.open ("ShapeFit_log.txt");
51 establish_upper_limits(observed,dataprediction,puresignal,"LM4",myfile);
52 myfile.close();
53 */
54
55
56 int nbins=100;
57 float low=0;
58 float hi=500;
59 TCanvas *c4 = new TCanvas("c4","c4",900,900);
60 c4->Divide(2,2);
61 c4->cd(1);
62 c4->cd(1)->SetLogy(1);
63 TH1F *datapredictiont = allsamples.Draw("datapredictiont", "-"+datajzb, nbins,low,hi, "JZB [GeV]", "events", cutmass&&cutOSSF&&cutnJets,data, luminosity);
64 TH1F *datapredictiono = allsamples.Draw("datapredictiono", "-"+datajzb, nbins,low,hi, "JZB [GeV]", "events", cutmass&&cutOSOF&&cutnJets,data, luminosity);
65 datapredictiont->Add(datapredictiono,-1);
66 dout << "Second way of doing this !!!! Analytical shape to the left :-D" << endl;
67 vector<TF1*> functions = do_cb_fit_to_plot(datapredictiont,10);
68 datapredictiont->SetMarkerColor(kRed);
69 datapredictiont->SetLineColor(kRed);
70 datapredictiont->Draw();
71 functions[1]->Draw("same");
72 TText *title1 = write_title("Top Background Prediction (JZB<0, with osof subtr)");
73 title1->Draw();
74
75 c4->cd(2);
76 c4->cd(2)->SetLogy(1);
77 TH1F *observedt = allsamples.Draw("observedt", datajzb, nbins,low,hi, "JZB [GeV]", "events", cutmass&&cutOSSF&&cutnJets,data, luminosity);
78 observedt->Draw();
79 datapredictiont->Draw("histo,same");
80 functions[1]->Draw("same");
81 TText *title2 = write_title("Observed and predicted background");
82 title2->Draw();
83
84 c4->cd(3);
85 c4->cd(3)->SetLogy(1);
86 // TH1F *ratio = (TH1F*)observedt->Clone();
87
88 TH1F *analytical_background_prediction= new TH1F("analytical_background_prediction","",nbins,low,hi);
89 for(int i=0;i<=nbins;i++) {
90 analytical_background_prediction->SetBinContent(i+1,functions[1]->Eval(((hi-low)/((float)nbins))*(i+0.5)));
91 analytical_background_prediction->SetBinError(i+1,TMath::Sqrt(functions[1]->Eval(((hi-low)/((float)nbins))*(i+0.5))));
92 }
93 analytical_background_prediction->GetYaxis()->SetTitle("JZB [GeV]");
94 analytical_background_prediction->GetYaxis()->CenterTitle();
95 TH1F *analyticaldrawonly = (TH1F*)analytical_background_prediction->Clone();
96 analytical_background_prediction->SetFillColor(TColor::GetColor("#3399FF"));
97 analytical_background_prediction->SetMarkerSize(0);
98 analytical_background_prediction->Draw("e5");
99 analyticaldrawonly->Draw("histo,same");
100 functions[1]->Draw("same");
101 TText *title3 = write_title("Analytical bg pred histo");
102 title3->Draw();
103
104 c4->cd(4);
105 // c4->cd(4)->SetLogy(1);
106 vector<float> ratio_binning;
107 ratio_binning.push_back(0);
108 ratio_binning.push_back(5);
109 ratio_binning.push_back(10);
110 ratio_binning.push_back(20);
111 ratio_binning.push_back(50);
112 // ratio_binning.push_back(60);
113 /*
114 ratio_binning.push_back(51);
115 ratio_binning.push_back(52);
116 ratio_binning.push_back(53);
117 ratio_binning.push_back(54);
118 ratio_binning.push_back(55);
119 ratio_binning.push_back(56);
120 ratio_binning.push_back(57);
121 ratio_binning.push_back(58);
122 ratio_binning.push_back(59);
123 ratio_binning.push_back(60);
124 // ratio_binning.push_back(70);*/
125 // ratio_binning.push_back(80);
126 // ratio_binning.push_back(90);
127 ratio_binning.push_back(80);
128 // ratio_binning.push_back(110);
129 ratio_binning.push_back(500);
130
131 TH1F *observedtb = allsamples.Draw("observedtb", datajzb, ratio_binning, "JZB [GeV]", "events", cutmass&&cutOSSF&&cutnJets,data, luminosity);
132 TH1F *datapredictiontb = allsamples.Draw("datapredictiontb", "-"+datajzb, ratio_binning, "JZB [GeV]", "events", cutmass&&cutOSSF&&cutnJets,data, luminosity);
133 TH1F *datapredictiontbo = allsamples.Draw("datapredictiontbo", "-"+datajzb, ratio_binning, "JZB [GeV]", "events", cutmass&&cutOSOF&&cutnJets,data, luminosity);
134 datapredictiontb->Add(datapredictiontbo,-1);
135 TH1F *analytical_background_predictionb = allsamples.Draw("analytical_background_predictionb",datajzb, ratio_binning, "JZB [GeV]", "events", cutmass&&cutOSSF&&cutnJets&&"mll<2",data, luminosity);
136 for(int i=0;i<=ratio_binning.size();i++) {
137 analytical_background_predictionb->SetBinContent(i+1,functions[1]->Eval(analytical_background_predictionb->GetBinCenter(i)));
138 analytical_background_predictionb->SetBinError(i+1,TMath::Sqrt(functions[1]->Eval(analytical_background_predictionb->GetBinCenter(i))));
139 }
140
141 TH1F *ratio = (TH1F*) observedtb->Clone();
142 ratio->Divide(datapredictiontb);
143
144 for (int i=0;i<=ratio_binning.size();i++) {
145 dout << observedtb->GetBinLowEdge(i+1) << ";"<<observedtb->GetBinContent(i+1) << ";" << datapredictiontb->GetBinContent(i+1) << " --> " << ratio->GetBinContent(i+1) << "+/-" << ratio->GetBinError(i+1) << endl;
146 }
147
148 // ratio->Divide(datapredictiontb);
149 // ratio->Divide(analytical_background_predictionb);
150 // TGraphAsymmErrors *JZBratio= histRatio(observedtb,analytical_background_predictionb,data,ratio_binning);
151 // ratio->Divide(analytical_background_prediction);
152 // ratio->Divide(datapredictiont);
153 // ratio->GetYaxis()->SetTitle("obs/pred");
154 // JZBratio->Draw("AP");
155 ratio->GetYaxis()->SetRangeUser(0,10);
156 ratio->Draw();
157 //analytical_background_predictionb->Draw();
158 // JZBratio->SetTitle("");
159 TText *title4 = write_title("Ratio of observed to predicted");
160 title4->Draw();
161
162 // CompleteSave(c4,"test/ttbar_discovery_dataprediction___analytical_function");
163 CompleteSave(c4,"test/ttbar_discovery_dataprediction__analytical__new_binning_one_huge_bin_from_80");
164
165
166
167
168
169 }
170
171 vector<float> compute_one_upper_limit(float mceff,float mcefferr, int ibin, string mcjzb, bool doobserved=false) {
172 float sigma95=0.0,sigma95A=0.0;
173 int nuisancemodel=1;
174 dout << "Now calling : CL95(" << luminosity << "," << lumiuncert*luminosity << "," << mceff << "," << mcefferr << "," << Npred[ibin] << "," << Nprederr[ibin] << "," << Nobs[ibin] << "," << false << "," << nuisancemodel<< ") " << endl;
175 sigma95 = CL95(luminosity, lumiuncert*luminosity, mceff, mcefferr, Npred[ibin], Nprederr[ibin], Nobs[ibin], false, nuisancemodel);
176 if(doobserved) {
177 dout << "Now calling : CLA(" << luminosity << "," << lumiuncert*luminosity << "," << mceff << "," << mcefferr << "," << Npred[ibin] << "," << Nprederr[ibin] << "," << nuisancemodel<< ") " << endl;
178 sigma95A = CLA(luminosity, lumiuncert*luminosity, mceff, mcefferr, Npred[ibin], Nprederr[ibin], nuisancemodel);
179 }
180 vector<float> sigmas;
181 sigmas.push_back(sigma95);
182 sigmas.push_back(sigma95A);
183 return sigmas;
184 }
185
186 void compute_upper_limits_from_counting_experiment(vector<vector<float> > uncertainties,vector<float> jzbcuts, string mcjzb, bool doobserved) {
187 dout << "Doing counting experiment ... " << endl;
188 vector<vector<string> > limits;
189 vector<vector<float> > vlimits;
190
191
192 for(int isample=0;isample<signalsamples.collection.size();isample++) {
193 vector<string> rows;
194 vector<float> vrows;
195 dout << "Considering sample " << signalsamples.collection[isample].samplename << endl;
196 rows.push_back(signalsamples.collection[isample].samplename);
197 for(int ibin=0;ibin<jzbcuts.size();ibin++) {
198 dout << "_________________________________________________________________________________" << endl;
199 float JZBcutat=uncertainties[isample*jzbcuts.size()+ibin][0];
200 float mceff=uncertainties[isample*jzbcuts.size()+ibin][1];
201 float staterr=uncertainties[isample*jzbcuts.size()+ibin][2];
202 float systerr=uncertainties[isample*jzbcuts.size()+ibin][3];
203 float toterr =uncertainties[isample*jzbcuts.size()+ibin][4];
204 float observed,observederr,null,result;
205 fill_result_histos(observed,observederr, null,null,null,null,null,null,null,mcjzb,JZBcutat,(int)5,result,(signalsamples.FindSample(signalsamples.collection[isample].filename)),signalsamples);
206 observed-=result;//this is the actual excess we see!
207 float expected=observed/luminosity;
208
209 dout << "Sample: " << signalsamples.collection[isample].samplename << ", JZB>"<<JZBcutat<< " : " << mceff << " +/- " << staterr << " (stat) +/- " << systerr << " (syst) --> toterr = " << toterr << endl;
210 vector<float> sigmas = compute_one_upper_limit(mceff,toterr,ibin,mcjzb,doobserved);
211
212 if(doobserved) {
213 rows.push_back(any2string(sigmas[0])+";"+any2string(sigmas[1])+";"+"("+any2string(expected)+")");
214 vrows.push_back(sigmas[0]);
215 vrows.push_back(sigmas[1]);
216 vrows.push_back(expected);
217 }
218 else {
219 rows.push_back(any2string(sigmas[0])+"("+any2string(expected)+")");
220 vrows.push_back(sigmas[0]);
221 vrows.push_back(expected);
222 }
223 }//end of bin loop
224 limits.push_back(rows);
225 vlimits.push_back(vrows);
226 }//end of sample loop
227 dout << endl << endl << "PAS table 3: " << endl << endl;
228 dout << "\t";
229 for (int irow=0;irow<jzbcuts.size();irow++) {
230 dout << jzbcuts[irow] << "\t";
231 }
232 dout << endl;
233 for(int irow=0;irow<limits.size();irow++) {
234 for(int ientry=0;ientry<limits[irow].size();ientry++) {
235 dout << limits[irow][ientry] << "\t";
236 }
237 dout << endl;
238 }
239
240 if(!doobserved) {
241 dout << endl << endl << "LIMITS: " << endl;
242 dout << "\t";
243 for (int irow=0;irow<jzbcuts.size();irow++) {
244 dout << jzbcuts[irow] << "\t";
245 }
246 dout << endl;
247 for(int irow=0;irow<limits.size();irow++) {
248 dout << limits[irow][0] << "\t";
249 for(int ientry=0;ientry<jzbcuts.size();ientry++) {
250 dout << Round(vlimits[irow][2*ientry] / vlimits[irow][2*ientry+1],3)<< "\t";
251 }
252 dout << endl;
253 }
254 }//do observed
255
256 dout << endl << endl << "Final selection efficiencies with total statistical and systematic errors, and corresponding observed and expected upper limits (UL) on ($\\sigma\\times$ BR $\\times$ acceptance) for the LM4 and LM8 scenarios, in the different regions. The last column contains the predicted ($\\sigma \\times $BR$\\times$ acceptance) at NLO obtained from Monte Carlo simulation." << endl;
257 dout << "Scenario \t Efficiency [%] \t Upper limits [pb] \t Prediction [pb]" << endl;
258 for(int icut=0;icut<jzbcuts.size();icut++) {
259 dout << "Region with JZB>" << jzbcuts[icut] << endl;
260 for(int isample=0;isample<signalsamples.collection.size();isample++) {
261 dout << limits[isample][0] << "\t" << Round(100*uncertainties[isample*jzbcuts.size()+icut][1],1) << "+/-" << Round(100*uncertainties[isample*jzbcuts.size()+icut][2],1) << " (stat) +/- " << Round(100*uncertainties[isample*jzbcuts.size()+icut][3],1) << " (syst) \t" << Round((vlimits[isample][2*icut]),3) << "\t" << Round(vlimits[isample][2*icut+1],3) << endl;
262 }
263 dout << endl;
264 }
265
266 write_warning("compute_upper_limits_from_counting_experiment","Still need to update the script");
267 }
268
269
270
271 /********************************************************************** new : Limits using SHAPES ***********************************
272
273
274 SSSSSSSSSSSSSSS hhhhhhh
275 SS:::::::::::::::Sh:::::h
276 S:::::SSSSSS::::::Sh:::::h
277 S:::::S SSSSSSSh:::::h
278 S:::::S h::::h hhhhh aaaaaaaaaaaaa ppppp ppppppppp eeeeeeeeeeee ssssssssss
279 S:::::S h::::hh:::::hhh a::::::::::::a p::::ppp:::::::::p ee::::::::::::ee ss::::::::::s
280 S::::SSSS h::::::::::::::hh aaaaaaaaa:::::ap:::::::::::::::::p e::::::eeeee:::::eess:::::::::::::s
281 SS::::::SSSSS h:::::::hhh::::::h a::::app::::::ppppp::::::pe::::::e e:::::es::::::ssss:::::s
282 SSS::::::::SS h::::::h h::::::h aaaaaaa:::::a p:::::p p:::::pe:::::::eeeee::::::e s:::::s ssssss
283 SSSSSS::::S h:::::h h:::::h aa::::::::::::a p:::::p p:::::pe:::::::::::::::::e s::::::s
284 S:::::S h:::::h h:::::h a::::aaaa::::::a p:::::p p:::::pe::::::eeeeeeeeeee s::::::s
285 S:::::S h:::::h h:::::ha::::a a:::::a p:::::p p::::::pe:::::::e ssssss s:::::s
286 SSSSSSS S:::::S h:::::h h:::::ha::::a a:::::a p:::::ppppp:::::::pe::::::::e s:::::ssss::::::s
287 S::::::SSSSSS:::::S h:::::h h:::::ha:::::aaaa::::::a p::::::::::::::::p e::::::::eeeeeeee s::::::::::::::s
288 S:::::::::::::::SS h:::::h h:::::h a::::::::::aa:::ap::::::::::::::pp ee:::::::::::::e s:::::::::::ss
289 SSSSSSSSSSSSSSS hhhhhhh hhhhhhh aaaaaaaaaa aaaap::::::pppppppp eeeeeeeeeeeeee sssssssssss
290 p:::::p
291 p:::::p
292 p:::::::p
293 p:::::::p
294 p:::::::p
295 ppppppppp
296
297
298 *********************************************************************** new : Limits using SHAPES ***********************************/
299
300
301 void limit_shapes_for_systematic_effect(TFile *limfile, string identifier, string mcjzb, string datajzb, int JES,vector<float> binning, TCanvas *limcan) {
302 dout << "Creatig shape templates ... ";
303 if(identifier!="") dout << "for systematic called "<<identifier;
304 dout << endl;
305 int dataormc=mcwithsignal;//this is only for tests - for real life you want dataormc=data !!!
306 if(dataormc!=data) write_warning(__FUNCTION__,"WATCH OUT! Not using data for limits!!!! this is ok for tests, but not ok for anything official!");
307
308 TCut limitnJetcut;
309 if(JES==noJES) limitnJetcut=cutnJets;
310 else {
311 if(JES==JESdown) limitnJetcut=cutnJetsJESdown;
312 if(JES==JESup) limitnJetcut=cutnJetsJESup;
313 }
314 TH1F *ZOSSFP = allsamples.Draw("ZOSSFP",datajzb,binning, "JZB4limits", "events",cutmass&&cutOSSF&&limitnJetcut&&basiccut,dataormc,luminosity);
315 TH1F *ZOSOFP = allsamples.Draw("ZOSOFP",datajzb,binning, "JZB4limits", "events",cutmass&&cutOSOF&&limitnJetcut&&basiccut,dataormc,luminosity);
316 TH1F *ZOSSFN = allsamples.Draw("ZOSSFN","-"+datajzb,binning, "JZB4limits", "events",cutmass&&cutOSSF&&limitnJetcut&&basiccut,dataormc,luminosity);
317 TH1F *ZOSOFN = allsamples.Draw("ZOSOFN","-"+datajzb,binning, "JZB4limits", "events",cutmass&&cutOSOF&&limitnJetcut&&basiccut,dataormc,luminosity);
318
319 TH1F *SBOSSFP = allsamples.Draw("SBOSSFP",datajzb,binning, "JZB4limits", "events",cutOSSF&&limitnJetcut&&basiccut&&sidebandcut,dataormc,luminosity);
320 TH1F *SBOSOFP = allsamples.Draw("SBOSOFP",datajzb,binning, "JZB4limits", "events",cutOSOF&&limitnJetcut&&basiccut&&sidebandcut,dataormc,luminosity);
321 TH1F *SBOSSFN = allsamples.Draw("SBOSSFN","-"+datajzb,binning, "JZB4limits", "events",cutOSSF&&limitnJetcut&&basiccut&&sidebandcut,dataormc,luminosity);
322 TH1F *SBOSOFN = allsamples.Draw("SBOSOFN","-"+datajzb,binning, "JZB4limits", "events",cutOSOF&&limitnJetcut&&basiccut&&sidebandcut,dataormc,luminosity);
323
324 TH1F *LZOSSFP = allsamples.Draw("LZOSSFP",mcjzb,binning, "JZB4limits", "events",cutmass&&cutOSSF&&limitnJetcut&&basiccut,mc,luminosity,allsamples.FindSample("LM4"));
325 TH1F *LZOSOFP = allsamples.Draw("LZOSOFP",mcjzb,binning, "JZB4limits", "events",cutmass&&cutOSOF&&limitnJetcut&&basiccut,mc,luminosity,allsamples.FindSample("LM4"));
326 TH1F *LZOSSFN = allsamples.Draw("LZOSSFN","-"+mcjzb,binning, "JZB4limits", "events",cutmass&&cutOSSF&&limitnJetcut&&basiccut,mc,luminosity,allsamples.FindSample("LM4"));
327 TH1F *LZOSOFN = allsamples.Draw("LZOSOFN","-"+mcjzb,binning, "JZB4limits", "events",cutmass&&cutOSOF&&limitnJetcut&&basiccut,mc,luminosity,allsamples.FindSample("LM4"));
328
329 TH1F *LSBOSSFP = allsamples.Draw("LSBOSSFP",mcjzb,binning, "JZB4limits", "events",cutOSSF&&limitnJetcut&&basiccut&&sidebandcut,mc,luminosity,allsamples.FindSample("LM4"));
330 TH1F *LSBOSOFP = allsamples.Draw("LSBOSOFP",mcjzb,binning, "JZB4limits", "events",cutOSOF&&limitnJetcut&&basiccut&&sidebandcut,mc,luminosity,allsamples.FindSample("LM4"));
331 TH1F *LSBOSSFN = allsamples.Draw("LSBOSSFN","-"+mcjzb,binning, "JZB4limits", "events",cutOSSF&&limitnJetcut&&basiccut&&sidebandcut,mc,luminosity,allsamples.FindSample("LM4"));
332 TH1F *LSBOSOFN = allsamples.Draw("LSBOSOFN","-"+mcjzb,binning, "JZB4limits", "events",cutOSOF&&limitnJetcut&&basiccut&&sidebandcut,mc,luminosity,allsamples.FindSample("LM4"));
333
334 string obsname="data_obs";
335 string predname="background";
336 string signalname="signal";
337 if(identifier!="") {
338 obsname=("data_"+identifier);
339 predname=("background_"+identifier);
340 signalname="signal_"+identifier;
341 }
342
343 TH1F *obs = (TH1F*)ZOSSFP->Clone();
344 obs->SetName(obsname.c_str());
345 obs->Write();
346 TH1F *pred = (TH1F*)ZOSSFN->Clone();
347 pred->Add(ZOSOFP,1.0/3);
348 pred->Add(ZOSOFN,-1.0/3);
349 pred->Add(SBOSSFP,1.0/3);
350 pred->Add(SBOSSFN,-1.0/3);
351 pred->Add(SBOSOFP,1.0/3);
352 pred->Add(SBOSOFN,-1.0/3);
353 pred->SetName(predname.c_str());
354 pred->Write();
355
356 // TH1F *Lobs = (TH1F*)LZOSSFP->Clone();
357 // TH1F *Lpred = (TH1F*)LZOSSFN->Clone();
358
359 TH1F *Lobs = new TH1F("Lobs","Lobs",binning.size()-1,&binning[0]);
360 TH1F *Lpred = new TH1F("Lpred","Lpred",binning.size()-1,&binning[0]);
361 Lobs->Add(LZOSSFP);
362 Lpred->Add(LZOSSFN);
363 Lpred->Add(LZOSOFP,1.0/3);
364 Lpred->Add(LZOSOFN,-1.0/3);
365 Lpred->Add(LSBOSSFP,1.0/3);
366 Lpred->Add(LSBOSSFN,-1.0/3);
367 Lpred->Add(LSBOSOFP,1.0/3);
368 Lpred->Add(LSBOSOFN,-1.0/3);
369 TH1F *signal = (TH1F*)Lobs->Clone();
370 signal->Add(Lpred,-1);
371 signal->SetName(signalname.c_str());
372 signal->Write();
373
374 delete Lobs;
375 delete Lpred;
376
377 delete ZOSSFP;
378 delete ZOSOFP;
379 delete ZOSSFN;
380 delete ZOSOFN;
381
382 delete SBOSSFP;
383 delete SBOSOFP;
384 delete SBOSSFN;
385 delete SBOSOFN;
386
387 delete LZOSSFP;
388 delete LZOSOFP;
389 delete LZOSSFN;
390 delete LZOSOFN;
391
392 delete LSBOSSFP;
393 delete LSBOSOFP;
394 delete LSBOSSFN;
395 delete LSBOSOFN;
396
397 }
398
399 void prepare_datacard(TFile *f) {
400 TH1F *dataob = (TH1F*)f->Get("data_obs");
401 TH1F *signal = (TH1F*)f->Get("signal");
402 TH1F *background = (TH1F*)f->Get("background");
403
404 ofstream datacard;
405 ensure_directory_exists(get_directory()+"/limits");
406 datacard.open ((get_directory()+"/limits/susydatacard.txt").c_str());
407 datacard << "Writing this to a file.\n";
408 datacard << "imax 1\n";
409 datacard << "jmax 1\n";
410 datacard << "kmax *\n";
411 datacard << "---------------\n";
412 datacard << "shapes * * limitfile.root $PROCESS $PROCESS_$SYSTEMATIC\n";
413 datacard << "---------------\n";
414 datacard << "bin 1\n";
415 datacard << "observation "<<dataob->Integral()<<"\n";
416 datacard << "------------------------------\n";
417 datacard << "bin 1 1\n";
418 datacard << "process signal background\n";
419 datacard << "process 0 1\n";
420 datacard << "rate "<<signal->Integral()<<" "<<background->Integral()<<"\n";
421 datacard << "--------------------------------\n";
422 datacard << "lumi lnN 1.10 1.0\n";
423 datacard << "bgnorm lnN 1.00 1.4 uncertainty on our prediction (40%)\n";
424 datacard << "JES shape 1 1 uncertainty on background shape and normalization\n";
425 datacard << "peak shape 1 1 uncertainty on signal resolution. Assume the histogram is a 2 sigma shift, \n";
426 datacard << "# so divide the unit gaussian by 2 before doing the interpolation\n";
427 datacard.close();
428 }
429
430
431 void prepare_limits(string mcjzb, string datajzb, float jzbpeakerrordata, float jzbpeakerrormc, vector<float> jzbbins) {
432 ensure_directory_exists(get_directory()+"/limits");
433 TFile *limfile = new TFile((get_directory()+"/limits/limitfile.root").c_str(),"RECREATE");
434 TCanvas *limcan = new TCanvas("limcan","Canvas for calculating limits");
435 limit_shapes_for_systematic_effect(limfile,"",mcjzb,datajzb,noJES,jzbbins,limcan);
436 limit_shapes_for_systematic_effect(limfile,"peakUp",newjzbexpression(mcjzb,jzbpeakerrormc),newjzbexpression(datajzb,jzbpeakerrordata),noJES,jzbbins,limcan);
437 limit_shapes_for_systematic_effect(limfile,"peakDown",newjzbexpression(mcjzb,-jzbpeakerrormc),newjzbexpression(datajzb,-jzbpeakerrordata),noJES,jzbbins,limcan);
438 limit_shapes_for_systematic_effect(limfile,"JESUp",mcjzb,datajzb,JESup,jzbbins,limcan);
439 limit_shapes_for_systematic_effect(limfile,"JESDown",mcjzb,datajzb,JESdown,jzbbins,limcan);
440
441 prepare_datacard(limfile);
442 limfile->Close();
443 write_info("prepare_limits","limitfile.root and datacard.txt have been generated. You can now use them to calculate limits!");
444
445 }