18 |
|
#include <TSystem.h> |
19 |
|
#include <TRandom3.h> |
20 |
|
|
21 |
– |
//#include "TTbar_stuff.C" |
21 |
|
using namespace std; |
22 |
|
|
23 |
|
using namespace PlottingSetup; |
43 |
|
} |
44 |
|
} |
45 |
|
|
46 |
+ |
vector<float> get_expected_points(float observed,int npoints, string RunDirectory, bool doPlot=false) { |
47 |
+ |
TF1 *gaus = new TF1("gaus","gaus(0)",0,10*observed); |
48 |
+ |
gaus->SetParameters(1,observed,2*observed); |
49 |
+ |
gaus->SetParameter(0,1/(gaus->Integral(0,10*observed))); |
50 |
+ |
float currentpoint=0.01; |
51 |
+ |
float stepsize=observed/100; |
52 |
+ |
vector<float> points; |
53 |
+ |
while(currentpoint<25*observed && points.size()<npoints) { |
54 |
+ |
|
55 |
+ |
if(gaus->Integral(0,currentpoint)>((points.size()+1.0)/(npoints+2.0))) { |
56 |
+ |
if(Verbosity>0) dout << "Added points for calculation at " << currentpoint << " (integral is " << gaus->Integral(0,currentpoint) << ")" << endl; |
57 |
+ |
points.push_back(currentpoint); |
58 |
+ |
if(points.size()==npoints) break; |
59 |
+ |
} |
60 |
+ |
currentpoint+=stepsize; |
61 |
+ |
} |
62 |
+ |
if(doPlot) { |
63 |
+ |
gaus->SetName("Expected limit computation points"); |
64 |
+ |
gaus->SetTitle("Expected limit computation points"); |
65 |
+ |
TCanvas *can = new TCanvas("can","can"); |
66 |
+ |
gaus->Draw(); |
67 |
+ |
TLine *lines[npoints]; |
68 |
+ |
for(int i=0;i<npoints;i++) { |
69 |
+ |
lines[i] = new TLine(points[i],0,points[i],gaus->GetMaximum()); |
70 |
+ |
lines[i]->SetLineColor(kBlue); |
71 |
+ |
lines[i]->Draw(); |
72 |
+ |
} |
73 |
+ |
can->SaveAs((RunDirectory+"/DistributionOfComputedPoints.png").c_str()); |
74 |
+ |
delete can; |
75 |
+ |
for(int i=0;i<npoints;i++) delete lines[i]; |
76 |
+ |
} |
77 |
+ |
delete gaus; |
78 |
+ |
|
79 |
+ |
return points; |
80 |
+ |
} |
81 |
+ |
|
82 |
|
void prepare_MC_datacard(string RunDirectory, TFile *f, float uJSU, float uPDF) { |
83 |
|
TH1F *dataob = (TH1F*)f->Get("data_obs"); |
84 |
|
TH1F *signal = (TH1F*)f->Get("signal"); |
296 |
|
return histo; |
297 |
|
} |
298 |
|
|
299 |
+ |
TH2F* empty2d(string hname) { |
300 |
+ |
TH2F *h = new TH2F(hname.c_str(),hname.c_str(),1000,0,1000,1000,0,1000); |
301 |
+ |
return h; |
302 |
+ |
} |
303 |
|
|
304 |
< |
void SQRT(TH1F *h) { |
305 |
< |
for (int i=1;i<=h->GetNbinsX();i++) { |
306 |
< |
h->SetBinContent(i,TMath::Sqrt(h->GetBinContent(i))); |
307 |
< |
} |
304 |
> |
TH2F* QuickDraw2d(TTree *events, string hname, string variable, string xlabel, string ylabel, TCut cut, string addcut, bool dataormc, float luminosity, map < pair<float, float>, map<string, float> > &xsec) { |
305 |
> |
TH2F *histo = empty2d(hname); |
306 |
> |
histo->Sumw2(); |
307 |
> |
stringstream drawcommand; |
308 |
> |
drawcommand << variable << ":mll>>" << hname; |
309 |
> |
if(SUSYScanSpace::SUSYscantype!=mSUGRA) { |
310 |
> |
events->Draw(drawcommand.str().c_str(),(essentialcut&&cut&&addcut.c_str())*cutWeight); |
311 |
> |
} else { |
312 |
> |
float totalxs=0; |
313 |
> |
for(int i=0;i<12;i++) { |
314 |
> |
stringstream drawcommand2; |
315 |
> |
drawcommand2 << variable << ">>+" << hname; |
316 |
> |
stringstream mSUGRAweight; |
317 |
> |
float xs = XSForProcessViaAddCutWrapper(addcut,xsec,i); |
318 |
> |
totalxs+=xs; |
319 |
> |
mSUGRAweight << "(0.5*(2*(process==" << i << "))*" << xs << ")"; |
320 |
> |
events->Draw(drawcommand2.str().c_str(),(essentialcut&&cut&&addcut.c_str())*cutWeight*mSUGRAweight.str().c_str()); |
321 |
> |
} |
322 |
> |
histo->Scale(1.0/totalxs); // to get back to 1 /pb normalization |
323 |
> |
} |
324 |
> |
events->Draw("eventNum",addcut.c_str(),"goff"); |
325 |
> |
float nevents = events->GetSelectedRows(); |
326 |
> |
QuickDrawNevents=nevents; |
327 |
> |
histo->Scale(luminosity/nevents); // this will give a histogram which is normalized to 1 pb so that any limit will be with respect to 1 pb |
328 |
> |
|
329 |
> |
return histo; |
330 |
|
} |
331 |
|
|
332 |
|
TH1F* Multiply(TH1F *h1, TH1F *h2) { |
443 |
|
TH1F *ZOSSFN = QuickDraw(signalevents,"ZOSSFN","-"+mcjzb,binning, "JZB", "events",cutmass&&cutOSSF&&limitnJetcut&&basiccut,addcut,mc,luminosity,xsec); |
444 |
|
TH1F *ZOSOFP; |
445 |
|
TH1F *ZOSOFN; |
446 |
< |
|
446 |
> |
|
447 |
> |
TH2F *ZOSSFP2d = QuickDraw2d(signalevents,"ZOSSFP2d",mcjzb, "JZB", "events",cutmass&&cutOSSF&&limitnJetcut&&basiccut,addcut,mc,luminosity,xsec); |
448 |
> |
TH2F *ZOSSFN2d = QuickDraw2d(signalevents,"ZOSSFN2d","-"+mcjzb,"JZB", "events",cutmass&&cutOSSF&&limitnJetcut&&basiccut,addcut,mc,luminosity,xsec); |
449 |
> |
TH2F *ZOSOFP2d; |
450 |
> |
TH2F *ZOSOFN2d; |
451 |
> |
|
452 |
|
if(!PlottingSetup::FullMCAnalysis) { |
453 |
|
ZOSOFP = QuickDraw(signalevents,"ZOSOFP",mcjzb,binning, "JZB", "events",cutmass&&cutOSOF&&limitnJetcut&&basiccut,addcut,mc,luminosity,xsec); |
454 |
|
ZOSOFN = QuickDraw(signalevents,"ZOSOFN","-"+mcjzb,binning, "JZB", "events",cutmass&&cutOSOF&&limitnJetcut&&basiccut,addcut,mc,luminosity,xsec); |
455 |
+ |
|
456 |
+ |
ZOSOFP2d = QuickDraw2d(signalevents,"ZOSOFP2d",mcjzb, "JZB", "events",cutmass&&cutOSOF&&limitnJetcut&&basiccut,addcut,mc,luminosity,xsec); |
457 |
+ |
ZOSOFN2d = QuickDraw2d(signalevents,"ZOSOFN2d","-"+mcjzb, "JZB", "events",cutmass&&cutOSOF&&limitnJetcut&&basiccut,addcut,mc,luminosity,xsec); |
458 |
|
} |
459 |
|
|
460 |
|
TH1F *SBOSSFP; |
462 |
|
TH1F *SBOSSFN; |
463 |
|
TH1F *SBOSOFN; |
464 |
|
|
465 |
+ |
TH2F *SBOSSFP2d; |
466 |
+ |
TH2F *SBOSOFP2d; |
467 |
+ |
TH2F *SBOSSFN2d; |
468 |
+ |
TH2F *SBOSOFN2d; |
469 |
+ |
|
470 |
|
if(PlottingSetup::RestrictToMassPeak&&!PlottingSetup::FullMCAnalysis) { |
471 |
|
SBOSSFP = QuickDraw(signalevents,"SBOSSFP",mcjzb,binning, "JZB", "events",cutOSSF&&limitnJetcut&&basiccut&&sidebandcut,addcut,mc,luminosity,xsec); |
472 |
|
SBOSOFP = QuickDraw(signalevents,"SBOSOFP",mcjzb,binning, "JZB", "events",cutOSOF&&limitnJetcut&&basiccut&&sidebandcut,addcut,mc,luminosity,xsec); |
473 |
|
SBOSSFN = QuickDraw(signalevents,"SBOSSFN","-"+mcjzb,binning, "JZB", "events",cutOSSF&&limitnJetcut&&basiccut&&sidebandcut,addcut,mc,luminosity,xsec); |
474 |
|
SBOSOFN = QuickDraw(signalevents,"SBOSOFN","-"+mcjzb,binning, "JZB", "events",cutOSOF&&limitnJetcut&&basiccut&&sidebandcut,addcut,mc,luminosity,xsec); |
475 |
+ |
|
476 |
+ |
SBOSSFP2d = QuickDraw2d(signalevents,"SBOSSFP2d",mcjzb,"JZB", "events",cutOSSF&&limitnJetcut&&basiccut&&sidebandcut,addcut,mc,luminosity,xsec); |
477 |
+ |
SBOSOFP2d = QuickDraw2d(signalevents,"SBOSOFP2d",mcjzb,"JZB", "events",cutOSOF&&limitnJetcut&&basiccut&&sidebandcut,addcut,mc,luminosity,xsec); |
478 |
+ |
SBOSSFN2d = QuickDraw2d(signalevents,"SBOSSFN2d","-"+mcjzb,"JZB", "events",cutOSSF&&limitnJetcut&&basiccut&&sidebandcut,addcut,mc,luminosity,xsec); |
479 |
+ |
SBOSOFN2d = QuickDraw2d(signalevents,"SBOSOFN2d","-"+mcjzb,"JZB", "events",cutOSOF&&limitnJetcut&&basiccut&&sidebandcut,addcut,mc,luminosity,xsec); |
480 |
|
} |
481 |
|
|
482 |
|
string signalname="signal"; |
483 |
< |
if(identifier!="") { |
405 |
< |
signalname="signal_"+identifier; |
406 |
< |
} |
483 |
> |
if(identifier!="") signalname="signal_"+identifier; |
484 |
|
|
485 |
|
TH1F *Lobs = new TH1F("Lobs","Lobs",binning.size()-1,&binning[0]); |
486 |
|
TH1F *Lpred = new TH1F("Lpred","Lpred",binning.size()-1,&binning[0]); |
487 |
|
|
488 |
+ |
TH2F *Lobs2d = empty2d("Lobs2d"); |
489 |
+ |
TH2F *Lpred2d = empty2d("Lobs2d"); |
490 |
+ |
|
491 |
|
TH1F *flippedLobs = new TH1F("flippedLobs","flippedLobs",binning.size()-1,&binning[0]); |
492 |
|
TH1F *flippedLpred = new TH1F("flippedLpred","flippedLpred",binning.size()-1,&binning[0]); |
493 |
|
|
494 |
+ |
TH2F *flippedLobs2d = empty2d("flippedLobs2d"); |
495 |
+ |
TH2F *flippedLpred2d = empty2d("flippedLpred2d"); |
496 |
+ |
|
497 |
|
Lobs->Add(ZOSSFP); |
498 |
< |
if(!PlottingSetup::FullMCAnalysis) Lpred->Add(ZOSSFN); |
498 |
> |
Lobs2d->Add(ZOSSFP2d); |
499 |
> |
if(!PlottingSetup::FullMCAnalysis) { |
500 |
> |
Lpred->Add(ZOSSFN); |
501 |
> |
Lpred2d->Add(ZOSSFN2d); |
502 |
> |
} |
503 |
|
|
504 |
|
dout << "SITUATION FOR SIGNAL: " << endl; |
418 |
– |
|
419 |
– |
|
505 |
|
if(PlottingSetup::FullMCAnalysis) { |
506 |
< |
dout << " OSSF JZB> 0 : " << ZOSSFP->Integral() << endl; |
506 |
> |
dout << " OSSF JZB> 0 : " << ZOSSFP->Integral() << endl; |
507 |
|
} else { |
508 |
< |
dout << " OSSF JZB> 0 : " << ZOSSFP->Integral() << " JZB < 0 :" << ZOSSFN->Integral() << endl; |
509 |
< |
dout << " OSOF JZB> 0 : " << ZOSOFP->Integral() << " JZB < 0 :" << ZOSOFN->Integral() << endl; |
510 |
< |
if(PlottingSetup::RestrictToMassPeak&&!PlottingSetup::FullMCAnalysis) { |
511 |
< |
dout << " OSSF SB JZB> 0 : " << SBOSSFP->Integral() << " JZB < 0 :" << SBOSSFN->Integral() << endl; |
512 |
< |
dout << " OSOF SB JZB> 0 : " << SBOSOFP->Integral() << " JZB < 0 :" << SBOSOFN->Integral() << endl; |
513 |
< |
} |
508 |
> |
dout << " OSSF JZB> 0 : " << ZOSSFP->Integral() << " JZB < 0 :" << ZOSSFN->Integral() << endl; |
509 |
> |
dout << " OSOF JZB> 0 : " << ZOSOFP->Integral() << " JZB < 0 :" << ZOSOFN->Integral() << endl; |
510 |
> |
|
511 |
> |
if(PlottingSetup::RestrictToMassPeak&&!PlottingSetup::FullMCAnalysis) { |
512 |
> |
dout << " OSSF SB JZB> 0 : " << SBOSSFP->Integral() << " JZB < 0 :" << SBOSSFN->Integral() << endl; |
513 |
> |
dout << " OSOF SB JZB> 0 : " << SBOSOFP->Integral() << " JZB < 0 :" << SBOSOFN->Integral() << endl; |
514 |
> |
} |
515 |
|
} |
430 |
– |
|
516 |
|
|
517 |
|
flippedLobs->Add(ZOSSFN); |
518 |
< |
if(!PlottingSetup::FullMCAnalysis) flippedLpred->Add(ZOSSFP); |
518 |
> |
flippedLobs2d->Add(ZOSSFN2d); |
519 |
> |
|
520 |
> |
if(!PlottingSetup::FullMCAnalysis) { |
521 |
> |
flippedLpred->Add(ZOSSFP); |
522 |
> |
flippedLpred2d->Add(ZOSSFP2d); |
523 |
> |
} |
524 |
|
|
525 |
|
if(!PlottingSetup::FullMCAnalysis) { |
526 |
|
if(PlottingSetup::RestrictToMassPeak) { |
527 |
< |
Lpred->Add(ZOSOFP,1.0/3); |
528 |
< |
Lpred->Add(ZOSOFN,-1.0/3); |
529 |
< |
Lpred->Add(SBOSSFP,1.0/3); |
530 |
< |
Lpred->Add(SBOSSFN,-1.0/3); |
531 |
< |
Lpred->Add(SBOSOFP,1.0/3); |
532 |
< |
Lpred->Add(SBOSOFN,-1.0/3); |
533 |
< |
|
534 |
< |
//flipped prediction |
535 |
< |
flippedLpred->Add(ZOSOFP,-1.0/3); |
536 |
< |
flippedLpred->Add(ZOSOFN,1.0/3); |
537 |
< |
flippedLpred->Add(SBOSSFP,-1.0/3); |
538 |
< |
flippedLpred->Add(SBOSSFN,1.0/3); |
539 |
< |
flippedLpred->Add(SBOSOFP,-1.0/3); |
540 |
< |
flippedLpred->Add(SBOSOFN,1.0/3); |
527 |
> |
Lpred->Add(ZOSOFP,1.0/3); |
528 |
> |
Lpred->Add(ZOSOFN,-1.0/3); |
529 |
> |
Lpred->Add(SBOSSFP,1.0/3); |
530 |
> |
Lpred->Add(SBOSSFN,-1.0/3); |
531 |
> |
Lpred->Add(SBOSOFP,1.0/3); |
532 |
> |
Lpred->Add(SBOSOFN,-1.0/3); |
533 |
> |
|
534 |
> |
//flipped prediction |
535 |
> |
flippedLpred->Add(ZOSOFP,-1.0/3); |
536 |
> |
flippedLpred->Add(ZOSOFN,1.0/3); |
537 |
> |
flippedLpred->Add(SBOSSFP,-1.0/3); |
538 |
> |
flippedLpred->Add(SBOSSFN,1.0/3); |
539 |
> |
flippedLpred->Add(SBOSOFP,-1.0/3); |
540 |
> |
flippedLpred->Add(SBOSOFN,1.0/3); |
541 |
> |
|
542 |
> |
//2d stuff: regular |
543 |
> |
Lpred2d->Add(ZOSOFP2d,1.0/3); |
544 |
> |
Lpred2d->Add(ZOSOFN2d,-1.0/3); |
545 |
> |
Lpred2d->Add(SBOSSFP2d,1.0/3); |
546 |
> |
Lpred2d->Add(SBOSSFN2d,-1.0/3); |
547 |
> |
Lpred2d->Add(SBOSOFP2d,1.0/3); |
548 |
> |
Lpred2d->Add(SBOSOFN2d,-1.0/3); |
549 |
> |
|
550 |
> |
//3d stuff: flipped prediction |
551 |
> |
flippedLpred2d->Add(ZOSOFP2d,-1.0/3); |
552 |
> |
flippedLpred2d->Add(ZOSOFN2d,1.0/3); |
553 |
> |
flippedLpred2d->Add(SBOSSFP2d,-1.0/3); |
554 |
> |
flippedLpred2d->Add(SBOSSFN2d,1.0/3); |
555 |
> |
flippedLpred2d->Add(SBOSOFP2d,-1.0/3); |
556 |
> |
flippedLpred2d->Add(SBOSOFN2d,1.0/3); |
557 |
|
} else { |
558 |
< |
Lpred->Add(ZOSOFP,1.0); |
559 |
< |
Lpred->Add(ZOSOFN,-1.0); |
560 |
< |
|
561 |
< |
//flipped prediction |
562 |
< |
flippedLpred->Add(ZOSOFP,-1.0); |
563 |
< |
flippedLpred->Add(ZOSOFN,1.0); |
558 |
> |
Lpred->Add(ZOSOFP,1.0); |
559 |
> |
Lpred->Add(ZOSOFN,-1.0); |
560 |
> |
|
561 |
> |
//flipped prediction |
562 |
> |
flippedLpred->Add(ZOSOFP,-1.0); |
563 |
> |
flippedLpred->Add(ZOSOFN,1.0); |
564 |
> |
|
565 |
> |
//2d stuff |
566 |
> |
Lpred2d->Add(ZOSOFP2d,1.0); |
567 |
> |
Lpred2d->Add(ZOSOFN2d,-1.0); |
568 |
> |
|
569 |
> |
//flipped prediction |
570 |
> |
flippedLpred2d->Add(ZOSOFP2d,-1.0); |
571 |
> |
flippedLpred2d->Add(ZOSOFN2d,1.0); |
572 |
|
} |
573 |
|
} |
574 |
|
|
575 |
|
TH1F *signal = (TH1F*)Lobs->Clone("signal"); |
576 |
+ |
TH1F *signal2d = (TH1F*)Lobs2d->Clone("signal2d"); |
577 |
+ |
|
578 |
|
if(!PlottingSetup::FullMCAnalysis) signal->Add(Lpred,-1); |
579 |
|
signal->SetName(signalname.c_str()); |
580 |
|
signal->SetTitle(signalname.c_str()); |
581 |
|
signal->Write(); |
582 |
|
|
583 |
+ |
if(!PlottingSetup::FullMCAnalysis) signal2d->Add(Lpred2d,-1); |
584 |
+ |
signal2d->SetName((signalname+"2d").c_str()); |
585 |
+ |
signal2d->SetTitle((signalname+"2d").c_str()); |
586 |
+ |
signal2d->Write(); |
587 |
+ |
|
588 |
|
TH1F *flippedsignal = (TH1F*)flippedLobs->Clone(); |
589 |
|
if(!PlottingSetup::FullMCAnalysis) flippedsignal->Add(flippedLpred,-1); |
590 |
|
flippedsignal->SetName(("flipped_"+signalname).c_str()); |
591 |
|
flippedsignal->Write(); |
592 |
|
|
593 |
+ |
TH2F *flippedsignal2d = (TH2F*)flippedLobs2d->Clone(); |
594 |
+ |
if(!PlottingSetup::FullMCAnalysis) flippedsignal2d->Add(flippedLpred,-1); |
595 |
+ |
flippedsignal2d->SetName(("flipped2d_"+signalname).c_str()); |
596 |
+ |
flippedsignal2d->Write(); |
597 |
+ |
|
598 |
|
if(identifier=="") { |
599 |
|
TH1F *signalStatUp = (TH1F*)signal->Clone(); |
600 |
|
signalStatUp->SetName(((string)signal->GetName()+"_StatUp").c_str()); |
604 |
|
signalStatDn->SetName(((string)signal->GetName()+"_StatDown").c_str()); |
605 |
|
signalStatDn->SetTitle(((string)signal->GetTitle()+"_StatDown").c_str()); |
606 |
|
|
607 |
+ |
TH2F *signalStatUp2d = (TH2F*)signal->Clone(); |
608 |
+ |
signalStatUp2d->SetName(((string)signal2d->GetName()+"_StatUp").c_str()); |
609 |
+ |
signalStatUp2d->SetTitle(((string)signal2d->GetTitle()+"_StatUp").c_str()); |
610 |
+ |
|
611 |
+ |
TH2F *signalStatDn2d = (TH2F*)signal->Clone(); |
612 |
+ |
signalStatDn2d->SetName(((string)signal2d->GetName()+"_StatDown").c_str()); |
613 |
+ |
signalStatDn2d->SetTitle(((string)signal2d->GetTitle()+"_StatDown").c_str()); |
614 |
+ |
|
615 |
+ |
|
616 |
|
for(int i=1;i<=signalStatDn->GetNbinsX();i++) { |
617 |
|
float staterr; |
618 |
|
if(!PlottingSetup::FullMCAnalysis) staterr = TMath::Sqrt(Lpred->GetBinContent(i) + Lobs->GetBinContent(i)); |
629 |
|
signal->SetBinError(i,staterr); |
630 |
|
} |
631 |
|
|
632 |
+ |
for(int i=1;i<=signalStatDn->GetNbinsX();i++) { |
633 |
+ |
for(int j=1;j<=signalStatDn->GetNbinsY();j++) { |
634 |
+ |
float staterr; |
635 |
+ |
if(!PlottingSetup::FullMCAnalysis) staterr = TMath::Sqrt(Lpred2d->GetBinContent(i,j) + Lobs2d->GetBinContent(i,j)); |
636 |
+ |
else staterr = TMath::Sqrt(Lobs2d->GetBinContent(i,j)); |
637 |
+ |
|
638 |
+ |
if(!PlottingSetup::FullMCAnalysis) { |
639 |
+ |
dout << "Stat err in bin " << i << ", " << j << " : " << staterr << endl; |
640 |
+ |
dout << " prediction: " << Lpred2d->GetBinContent(i,j) << " , observation: " << Lobs2d->GetBinContent(i,j) << " --> signal: " << signal2d->GetBinContent(i,j) << endl; |
641 |
+ |
dout << " we obtain : " << signal2d->GetBinContent(i,j)-staterr << " , " << signal2d->GetBinContent(i,j)+staterr << endl; |
642 |
+ |
} |
643 |
+ |
if(signal2d->GetBinContent(i,j)-staterr>0) signalStatDn2d->SetBinContent(i,j,signal2d->GetBinContent(i,j)-staterr); |
644 |
+ |
else signalStatDn2d->SetBinContent(i,j,0); |
645 |
+ |
signalStatUp2d->SetBinContent(i,signal2d->GetBinContent(i,j)+staterr); |
646 |
+ |
signal2d->SetBinError(i,j,staterr); |
647 |
+ |
} |
648 |
+ |
} |
649 |
+ |
|
650 |
+ |
//----------------------------------- ******** GOTTEN HERE *******--------------------------------------- |
651 |
+ |
|
652 |
|
signalStatDn->Write(); |
653 |
|
signalStatUp->Write(); |
654 |
|
|
1063 |
|
delete SBOSOFN; |
1064 |
|
} |
1065 |
|
} |
1066 |
< |
|
1066 |
> |
|
1067 |
> |
|
1068 |
> |
//----------------------------------- ******** GOTTEN HERE *******--------------------------------------- |
1069 |
> |
|
1070 |
> |
|
1071 |
> |
|
1072 |
> |
|
1073 |
> |
|
1074 |
|
} |
1075 |
|
|
1076 |
< |
ShapeDroplet LimitsFromShapes(bool asymptotic, float firstGuess, TTree *events, string addcut, string name, string mcjzb, string datajzb, vector<float> jzbbins, float jzbpeakerrormc, float jzbpeakerrordata) { |
1076 |
> |
void DoFullCls(string RunDirectory,float firstGuess) { |
1077 |
> |
vector<float> epoints = get_expected_points(firstGuess,25,RunDirectory,true);//get 20 points; |
1078 |
> |
ofstream RealScript; |
1079 |
> |
dout << "Going to write the full CLs script to " << RunDirectory << "/LimitScript.sh" << endl; |
1080 |
> |
RealScript.open ((RunDirectory+"/LimitScript.sh").c_str()); |
1081 |
> |
RealScript << "#!/bin/bash \n"; |
1082 |
> |
RealScript << "\n"; |
1083 |
> |
RealScript << "RunDirectory=" << RunDirectory << "\n"; |
1084 |
> |
RealScript << "CMSSWDirectory=" << PlottingSetup::CMSSW_dir << "\n"; |
1085 |
> |
RealScript << "ORIGTMP=$TMP\n"; |
1086 |
> |
RealScript << "ORIGTMPDIR=$TMPDIR\n"; |
1087 |
> |
RealScript << "export TMP=" << RunDirectory << "\n"; |
1088 |
> |
RealScript << "export TMPDIR=" << RunDirectory << "\n"; |
1089 |
> |
RealScript << "\n"; |
1090 |
> |
RealScript << "echo \"Going to compute limit in $RunDirectory\"\n"; |
1091 |
> |
RealScript << "ORIGDIR=`pwd`\n"; |
1092 |
> |
RealScript << "\n"; |
1093 |
> |
RealScript << "pushd $CMSSWDirectory\n"; |
1094 |
> |
RealScript << "cd ~/final_production_2011/CMSSW_4_2_8/src/HiggsAnalysis/\n"; |
1095 |
> |
RealScript << "origscramarch=$SCRAM_ARCH\n"; |
1096 |
> |
RealScript << "origbase=$CMSSW_BASE\n"; |
1097 |
> |
RealScript << "export SCRAM_ARCH=slc5_amd64_gcc434\n"; |
1098 |
> |
RealScript << "eval `scram ru -sh`\n"; |
1099 |
> |
RealScript << "\n"; |
1100 |
> |
RealScript << "mkdir -p $RunDirectory\n"; |
1101 |
> |
RealScript << "cd $RunDirectory\n"; |
1102 |
> |
RealScript << "echo `pwd`\n"; |
1103 |
> |
RealScript << "mkdir -p FullCLs\n"; |
1104 |
> |
RealScript << "cd FullCLs\n"; |
1105 |
> |
RealScript << "\n"; |
1106 |
> |
RealScript << "combineEXE=\"${CMSSW_BASE}/bin/${SCRAM_ARCH}/combine\"\n"; |
1107 |
> |
RealScript << "\n"; |
1108 |
> |
RealScript << "dobreak=0\n"; |
1109 |
> |
RealScript << "npoints=0\n"; |
1110 |
> |
RealScript << "time for i in {"; |
1111 |
> |
RealScript << epoints[0]/4 << ","; // adding a VERY VERY low point just to be totally safe |
1112 |
> |
RealScript << epoints[0]/2 << ","; // adding a VERY low point just to be safe |
1113 |
> |
for(int i=0;i<epoints.size();i++) RealScript << epoints[i] << ","; |
1114 |
> |
RealScript << 2*epoints[epoints.size()-1] << ","; // adding a VERY high point just to be safe |
1115 |
> |
RealScript << 4*epoints[epoints.size()-1]; // adding a VERY VERY high point just to be totally safe |
1116 |
> |
RealScript << "}; do \n"; |
1117 |
> |
RealScript << "let \"npoints = npoints +1\"\n"; |
1118 |
> |
RealScript << "if [[ $dobreak -gt 0 ]]; then \n"; |
1119 |
> |
RealScript << " continue \n"; |
1120 |
> |
RealScript << "fi \n"; |
1121 |
> |
RealScript << "echo -e \" Going to compute CLs for $i \\n\"\n"; |
1122 |
> |
RealScript << "echo $(date +%d%b%Y,%H:%M:%S)\n"; |
1123 |
> |
RealScript << "time " << PlottingSetup::cbafbasedir << "/DistributedModelCalculations/Utilities/TimeOut \"$combineEXE ../susydatacard.txt -M HybridNew --freq -T 500 --clsAcc 0 -v 0 -n TestPoint --saveHybridResult --saveToys -s -1 -i 8 --singlePoint $i >/dev/null 2>/dev/null\"\n"; |
1124 |
> |
// RealScript << "$combineEXE ../susydatacard.txt -M HybridNew --freq -T 500 --clsAcc 0 -v 0 -n TestPoint --saveHybridResult --saveToys -s -1 -i 8 --singlePoint $i >/dev/null 2>/dev/null \n"; |
1125 |
> |
RealScript << "errorsize=$?\n"; |
1126 |
> |
RealScript << "rm " << RunDirectory << "/rstat* 2>/dev/null \n"; |
1127 |
> |
RealScript << "if [[ $errorsize -gt 0 ]]; then \n"; |
1128 |
> |
RealScript << " echo \"Apparently something went wrong (had to be aborted)\"\n"; |
1129 |
> |
RealScript << " if [[ $npoints -gt 10 ]]; then \n"; |
1130 |
> |
RealScript << " dobreak=1 \n"; |
1131 |
> |
RealScript << " fi \n"; |
1132 |
> |
RealScript << "fi \n"; |
1133 |
> |
RealScript << "done\n"; |
1134 |
> |
|
1135 |
> |
RealScript << "\n"; |
1136 |
> |
RealScript << "hadd -f FullCLs.root higgsCombine*.root\n"; |
1137 |
> |
RealScript << "mkdir originalFiles\n"; |
1138 |
> |
RealScript << "mv higgsCombine* originalFiles\n"; |
1139 |
> |
RealScript << "$combineEXE ../susydatacard.txt -M HybridNew --freq --grid=FullCLs.root\n"; |
1140 |
> |
RealScript << "$combineEXE ../susydatacard.txt -M HybridNew --freq --grid=FullCLs.root --expectedFromGrid 0.5\n"; |
1141 |
> |
RealScript << "$combineEXE ../susydatacard.txt -M HybridNew --freq --grid=FullCLs.root --expectedFromGrid 0.84\n"; |
1142 |
> |
RealScript << "$combineEXE ../susydatacard.txt -M HybridNew --freq --grid=FullCLs.root --expectedFromGrid 0.16\n"; |
1143 |
> |
RealScript << "$combineEXE ../susydatacard.txt -M HybridNew --freq --grid=FullCLs.root --expectedFromGrid 0.975\n"; |
1144 |
> |
RealScript << "$combineEXE ../susydatacard.txt -M HybridNew --freq --grid=FullCLs.root --expectedFromGrid 0.025\n"; |
1145 |
> |
RealScript << "\n"; |
1146 |
> |
RealScript << "hadd FinalResult.root higgsCombineTest.HybridNew*\n"; |
1147 |
> |
RealScript << "\n"; |
1148 |
> |
RealScript << "g++ " << PlottingSetup::cbafbasedir << "/DistributedModelCalculations/ShapeLimits/ReadAndSave.C -o ReadAndSave.exec `root-config --glibs --cflags` \n"; |
1149 |
> |
RealScript << "./ReadAndSave.exec FinalResult.root " << RunDirectory << "/FullCLsShapeDropletResult.txt \n"; |
1150 |
> |
RealScript << "\n"; |
1151 |
> |
RealScript << "rm " << RunDirectory << "/rstat* \n"; |
1152 |
> |
RealScript << "\n"; |
1153 |
> |
RealScript << "#####\n"; |
1154 |
> |
RealScript << "echo \"Finalizing ...\"\n"; |
1155 |
> |
RealScript << "cd $ORIGDIR\n"; |
1156 |
> |
RealScript << "echo $ORIGDIR\n"; |
1157 |
> |
RealScript << "ls -ltrh ../SandBox/ | grep combine\n"; |
1158 |
> |
RealScript << "export SCRAM_ARCH=$origscramarch\n"; |
1159 |
> |
RealScript << "export TMP=$ORIGTMP\n"; |
1160 |
> |
RealScript << "export TMPDIR=$ORIGTMPDIR\n"; |
1161 |
> |
RealScript << "exit 0\n"; |
1162 |
> |
RealScript << "\n"; |
1163 |
> |
RealScript.close(); |
1164 |
> |
gSystem->Exec(("bash "+RunDirectory+"/LimitScript.sh").c_str()); |
1165 |
> |
} |
1166 |
> |
|
1167 |
> |
|
1168 |
> |
ShapeDroplet LimitsFromShapes(float low_fullCLs, float high_CLs, TTree *events, string addcut, string name, string mcjzb, string datajzb, vector<float> jzbbins, float jzbpeakerrormc, float jzbpeakerrordata) { |
1169 |
|
map < pair<float, float>, map<string, float> > xsec; |
1170 |
|
if(SUSYScanSpace::SUSYscantype==mSUGRA) xsec=getXsec(PlottingSetup::cbafbasedir+"/"+mSUGRAxsFile); |
1171 |
|
|
1179 |
|
|
1180 |
|
ensure_directory_exists(RunDirectory); |
1181 |
|
|
1182 |
< |
// write_warning(__FUNCTION__,"Modified the shape output file! PLEASE RESTORE!!!"); TFile *datafile = new TFile((PlottingSetup::cbafbasedir+"/DistributedModelCalculations/TEST___StoredShapes.root").c_str(),"READ"); |
929 |
< |
TFile *datafile = new TFile((PlottingSetup::cbafbasedir+"/DistributedModelCalculations/StoredShapes.root").c_str(),"READ"); |
1182 |
> |
TFile *datafile = new TFile((PlottingSetup::cbafbasedir+"/DistributedModelCalculations/StoredShapes.root").c_str()); |
1183 |
|
if(datafile->IsZombie()) { |
1184 |
|
write_error(__FUNCTION__,"Fatal error: The stored shapes are not available!"); |
1185 |
|
assert(!datafile->IsZombie()); |
1270 |
|
limfile->Close(); |
1271 |
|
dout << "Info: Shape root file and datacard have been generated in " << RunDirectory << endl; |
1272 |
|
stringstream command; |
1273 |
< |
if(asymptotic) { |
1274 |
< |
if(firstGuess>0) command << "bash " << PlottingSetup::cbafbasedir<< "/DistributedModelCalculations/ShapeLimits/CreateModel.sh " << RunDirectory << " susydatacard.txt" << " 1 0 0 " << firstGuess; // ASYMPTOTIC LIMITS WITH FIRST GUESS |
1275 |
< |
else command << "bash " << PlottingSetup::cbafbasedir<< "/DistributedModelCalculations/ShapeLimits/CreateModel.sh " << RunDirectory << " susydatacard.txt" << " 0 0 0 0"; // ASYMPTOTIC LIMITS |
1276 |
< |
} |
1277 |
< |
else command << "bash " << PlottingSetup::cbafbasedir<< "/DistributedModelCalculations/ShapeLimits/CreateModel.sh " << RunDirectory << " susydatacard.txt" << " 2 0 " << int(0.333 * firstGuess) << " " << int(firstGuess); // ASYMPTOTIC LIMITS |
1273 |
> |
string DropletLocation; |
1274 |
> |
int CreatedModelFileExitCode; |
1275 |
> |
//---------------------------------------------------------------- |
1276 |
> |
//do an asymptotic limit first |
1277 |
> |
command << "bash " << PlottingSetup::cbafbasedir<< "/DistributedModelCalculations/ShapeLimits/CreateModel.sh " << RunDirectory << " susydatacard.txt" << " 0 0 0 0"; // ASYMPTOTIC LIMITS |
1278 |
|
dout <<"Going to run : " << command.str() << endl; |
1279 |
< |
int CreatedModelFileExitCode = gSystem->Exec(command.str().c_str()); |
1279 |
> |
CreatedModelFileExitCode = gSystem->Exec(command.str().c_str()); |
1280 |
|
dout << "exit code of limit algorithm (CreateModel.sh) : " << CreatedModelFileExitCode << endl; |
1281 |
|
if(!(CreatedModelFileExitCode==0)) { |
1282 |
< |
write_warning(__FUNCTION__,"Something bad happened. It looks like a shape analysis is not the way to go. "); |
1283 |
< |
ShapeDroplet alpha; |
1284 |
< |
alpha.observed=-12345; // this is the flag to say watch out something went wrong with the signal ... |
1285 |
< |
alpha.SignalIntegral=1; |
1286 |
< |
return alpha; |
1282 |
> |
write_warning(__FUNCTION__,"Something bad happened. It looks like a shape analysis is not the way to go. "); |
1283 |
> |
ShapeDroplet beta; |
1284 |
> |
beta.observed=-12345; // this is the flag to say watch out something went wrong with the signal ... |
1285 |
> |
beta.SignalIntegral=1; |
1286 |
> |
return beta; |
1287 |
> |
} |
1288 |
> |
DropletLocation=RunDirectory+"/ShapeDropletResult.txt"; |
1289 |
> |
ShapeDroplet beta; |
1290 |
> |
beta.readDroplet(DropletLocation); |
1291 |
> |
float obsLimit = beta.observed/SignalIntegral; |
1292 |
> |
dout << "Checking if the obtained limit (" << beta.observed << " / " << SignalIntegral << " = " << beta.observed/SignalIntegral << " is in [" << low_fullCLs << " , " << high_CLs << "]" << endl; |
1293 |
> |
if(obsLimit<high_CLs && obsLimit>low_fullCLs) { |
1294 |
> |
//if(PlottingSetup::scantype!=mSUGRA||(obsLimit<high_CLs && obsLimit>low_fullCLs)) { |
1295 |
> |
dout << " It is! Full CLs ahead!" << endl; |
1296 |
> |
DoFullCls(RunDirectory,beta.observed); |
1297 |
> |
DropletLocation=RunDirectory+"/FullCLsShapeDropletResult.txt"; |
1298 |
> |
} else { |
1299 |
> |
dout << " It is not ... happy with asymptotic limits." << endl; |
1300 |
|
} |
1301 |
+ |
//---------------------------------------------------------------- |
1302 |
|
ShapeDroplet alpha; |
1303 |
< |
alpha.readDroplet(RunDirectory+"/ShapeDropletResult.txt"); |
1303 |
> |
alpha.readDroplet(DropletLocation); |
1304 |
|
alpha.PDF=PDFuncert; |
1305 |
|
alpha.JSU=JZBscale; |
1306 |
|
alpha.SignalIntegral=SignalIntegral; |
1307 |
|
dout << "Done reading limit information from droplet" << endl; |
1308 |
|
dout << alpha << endl; |
1042 |
– |
if(asymptotic) { |
1043 |
– |
dout << "Doing asymptotic limits, therefore adding an additional round!" << endl; |
1044 |
– |
command.str(""); |
1045 |
– |
command << "bash " << PlottingSetup::cbafbasedir<< "/DistributedModelCalculations/ShapeLimits/CreateModel.sh " << RunDirectory << " susydatacard.txt" << " 1 0 0 " << alpha.expected; // ASYMPTOTIC LIMITS WITH FIRST GUESS |
1046 |
– |
CreatedModelFileExitCode = gSystem->Exec(command.str().c_str()); |
1047 |
– |
dout << "exit code of limit algorithm (CreateModel.sh) : " << CreatedModelFileExitCode << endl; |
1048 |
– |
if(!(CreatedModelFileExitCode==0)) { |
1049 |
– |
write_warning(__FUNCTION__,"Something bad happened. It looks like a shape analysis is not the way to go. "); |
1050 |
– |
ShapeDroplet alpha; |
1051 |
– |
alpha.observed=-12345; // this is the flag to say watch out something went wrong with the signal ... |
1052 |
– |
alpha.SignalIntegral=1; |
1053 |
– |
return alpha; |
1054 |
– |
} |
1055 |
– |
alpha.readDroplet(RunDirectory+"/ShapeDropletResult.txt"); |
1056 |
– |
alpha.PDF=PDFuncert; |
1057 |
– |
alpha.JSU=JZBscale; |
1058 |
– |
alpha.SignalIntegral=SignalIntegral; |
1059 |
– |
dout << "Done reading updated limit information from droplet" << endl; |
1060 |
– |
dout << alpha << endl; |
1061 |
– |
} |
1309 |
|
|
1310 |
|
dout << "Everything is saved in " << RunDirectory << endl; |
1311 |
|
dout << "Cleaning up ... " << std::flush; |
1312 |
< |
/* dout << " 1) Make sure models directory exists ... " << std::flush; |
1313 |
< |
gSystem->Exec("mkdir -p models/"); |
1067 |
< |
dout << " ok!" << endl; |
1068 |
< |
dout << " 2) Deleting any previous model files with the same name ... " << std::flush; |
1069 |
< |
stringstream delcommand; |
1070 |
< |
delcommand << "rm models/model_" << name << ".root"; |
1071 |
< |
gSystem->Exec(delcommand.str().c_str()); |
1072 |
< |
stringstream delcommand2; |
1073 |
< |
delcommand2 << "rm models/model_" << name << ".txt"; |
1074 |
< |
gSystem->Exec(delcommand2.str().c_str()); |
1075 |
< |
dout << " ok!" << endl; |
1076 |
< |
dout << " 3) Transfering the new model files ... " << std::flush; |
1077 |
< |
stringstream copycommand; |
1078 |
< |
copycommand << "cp " << RunDirectory << "/model.root models/model_" << name << ".root"; |
1079 |
< |
gSystem->Exec(copycommand.str().c_str()); |
1080 |
< |
stringstream copycommand2; |
1081 |
< |
copycommand2 << "cp " << RunDirectory << "/susydatacard.txt models/model_" << name << ".txt"; |
1082 |
< |
gSystem->Exec(copycommand2.str().c_str()); |
1083 |
< |
stringstream copycommand3; |
1084 |
< |
copycommand3 << "cp " << RunDirectory << "/limitfile.root models/model_" << name << "_histo.root"; |
1085 |
< |
gSystem->Exec(copycommand3.str().c_str()); |
1312 |
> |
write_warning(__FUNCTION__,"NOT CLEANING UP"); |
1313 |
> |
// gSystem->Exec(("rm -rf "+RunDirectory).c_str()); |
1314 |
|
dout << " ok!" << endl; |
1087 |
– |
dout << " 4) Removing original working directory (" << RunDirectory << ") ... " << std::flush;*/ |
1088 |
– |
// gSystem->Exec(("rm -r "+RunDirectory).c_str()); |
1089 |
– |
dout << " ok!" << endl; |
1090 |
– |
write_warning(__FUNCTION__,"Not cleaning up right now!"); |
1315 |
|
delete limcan; |
1316 |
|
return alpha; |
1317 |
|
} |
1318 |
|
|
1319 |
|
void PrepareDataShapes(string mcjzb, string datajzb, vector<float> jzbbins, float jzbpeakerrordata) { |
1320 |
< |
|
1320 |
> |
dout << "Preparing data shapes ... " << endl; |
1321 |
|
map < pair<float, float>, map<string, float> > xsec; |
1322 |
|
TFile *datafile = new TFile("../DistributedModelCalculations/StoredShapes.root","RECREATE"); |
1323 |
|
TCanvas *limcan = new TCanvas("limcan","Canvas for calculating limits"); |
1334 |
|
generate_mc_shapes(false,faketree,mcjzb,datajzb,jzbbins,TCut("weightEffDown"),cutnJets,"","EffDown",xsec); |
1335 |
|
generate_mc_shapes(false,faketree,mcjzb,datajzb,jzbbins,cutWeight,cutnJets,"","StatUp",xsec); |
1336 |
|
generate_mc_shapes(false,faketree,mcjzb,datajzb,jzbbins,cutWeight,cutnJets,"","StatDown",xsec); |
1337 |
< |
/* |
1337 |
> |
|
1338 |
|
datafile->Close(); |
1339 |
< |
*/ |
1339 |
> |
|
1340 |
|
} |
1341 |
|
|