ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/cbrown/Development/Plotting/Modules/GeneralToolBox.C
(Generate patch)

Comparing UserCode/cbrown/Development/Plotting/Modules/GeneralToolBox.C (file contents):
Revision 1.22 by fronga, Tue Nov 13 13:27:37 2012 UTC vs.
Revision 1.33 by buchmann, Fri Feb 1 11:52:38 2013 UTC

# Line 40 | Line 40
40   #include <TRandom.h>
41   #include <TGraphErrors.h>
42   #include <TROOT.h>
43 + #include <TPolyLine.h>
44 +
45   #ifndef Verbosity
46   #define Verbosity 0
47   #endif
# Line 64 | Line 66 | namespace PlottingSetup {
66      bool is2012=true;
67      bool is53reco=true;
68      bool openBox = true;
69 +    vector<TH1F*> FakeHistoHeap;
70 +
71   }
72  
73   bool dopng=false;
# Line 257 | Line 261 | void ensure_directory_exists(string this
261    }
262   }
263  
264 + void DeleteStack(THStack mcstack) {
265 +  TH1F* h;
266 +  TIter nextHist(mcstack.GetHists());
267 +  while ( h = (TH1F*)nextHist() ) {
268 +    h->Delete();
269 +  }
270 +  delete h;
271 + }
272 +
273 +
274   void DeadEnd(string message) {
275    dout << endl;
276    dout << "\033[1;31m     _                _                  _  " << endl;
# Line 425 | Line 439 | void CompleteSave(TVirtualPad *can, stri
439  
440  
441   void setlumi(float l) {
442 < generaltoolboxlumi=l;
442 >  generaltoolboxlumi=l;
443   }
444  
445   int write_first_line(vector<vector<string> > &entries) {
# Line 1293 | Line 1307 | TGraphAsymmErrors* produce_ratio_graph(T
1307   }
1308  
1309  
1310 + TText* WriteSelection(int njets) {
1311 +  string sel="Loose selection";
1312 +  if(njets==3) sel="Tight selection";
1313 +  assert(njets==2||njets==3);
1314 +  TText *sele = new TText(0.97,0.135,sel.c_str());
1315 +  sele->SetNDC(true);
1316 +  sele->SetTextColor(TColor::GetColor("#848484"));
1317 +  sele->SetTextFont(42);
1318 +  sele->SetTextAlign(32);
1319 +  sele->SetTextSize(0.03);
1320 +  sele->SetTextAngle(270);
1321 +  return sele;
1322 + }
1323 +
1324   Double_t MarcosChi2TestX(const TH1* h1, const TH1* h2, Double_t &chi2, Int_t &ndf, Int_t &igood, Option_t *option)
1325   {
1326  
# Line 1315 | Line 1343 | Double_t MarcosChi2TestX(const TH1* h1,
1343     return prob;
1344   }
1345  
1346 < /*void save_with_difference(TH1F *obs, vector<TH1F*> predictions, vector<float> prediction_weights, vector<float> prediction_uncerts, TVirtualPad *canvas, string savemeas) {
1347 <  if(predictions.size()<1) {
1320 <    write_warning(__FUNCTION__,"Cannot make a pred-obs prediction plot because the prediction vector is empty. aborting!");
1321 <    return;
1322 <  }
1323 <  TH1F *comppred = predictions[0]->Clone("comppred");
1324 <  comppred->Scale(prediction_weights[0]);
1325 <  for(int i=1;i<predictions.size();i++) comppred->Add(predictions[i],prediction_weights[i]);
1326 <  
1327 <  TH1F *statpred = (TH1F*)comppred->Clone("statpred");
1328 <  for(int i=1;i<=statpred->GetNbinsX();i++) statpred->SetBinContent(0);
1329 <  
1330 <  TH1F *syspred = predictions[0];
1331 <  for(int i=1;i<=syspred->GetNbinsX();i++) syspred->SetBinContent(i,prediction_weights[0]*prediction_weights[0]*prediction_uncerts[0]*prediction_uncerts[0]*syspred->GetBinContent(i)*syspred->GetBinContent(i));
1332 <  for(int i=1;i<predictions.size();i++) {
1333 <    for(int ibin=1;ibin<=predictions[i]->GetNbinsX();ibin++) {
1334 <      float newentry=syspred->GetBinContent(ibin);
1335 <      newentry+=(predictions[i]->GetBinContent(ibin))*(predictions[i]->GetBinContent(ibin))*prediction_weights[i]*prediction_weights[i]*prediction_uncerts[i]*prediction_uncerts[i];
1336 <      syspred->SetBinContent(ibin,newentry);
1337 <    }
1338 <  }
1339 <  for(int ibin=1;ibin<=syspred->GetNbinsX();ibin++) {
1340 <    syspred->SetBinError(ibin,syspred->GetBinContent(i)+statpred->GetBinError());
1341 <    syspred->SetBinContent(ibin,0);
1342 <  }
1346 > void Save_With_Ratio(TH1F *nominator, TH1F *denominator, TVirtualPad *orig_canvas, string savemeas, bool do_bpred_ratio=false, bool extendrange=false, string yaxistitle="ratio",TH1F *syshisto=NULL) {
1347 >  //this function saves the pad being passed as well as a new one including the ratio.
1348    
1349 +  orig_canvas->cd();
1350 +  orig_canvas->Update();
1351 +  CompleteSave(orig_canvas,savemeas);
1352 +  TVirtualPad *canvas = (TVirtualPad*) orig_canvas->Clone("TempCanvas");//otherwise the Ratio_main_canvas will own our pad and destroy it upon deletion
1353    
1345 }
1346 */
1347 void save_with_ratio(TH1F *nominator, TH1F *denominator, TVirtualPad *canvas, string savemeas, bool do_bpred_ratio=false, bool extendrange=false, string yaxistitle="ratio",TH1F *syshisto=NULL) {
1348  //this function saves the pad being passed as well as a new one including the ratio.
1349  CompleteSave(canvas,savemeas);
1350
1354    float bottommargin=gStyle->GetPadBottomMargin();
1355    float canvas_height=gStyle->GetCanvasDefH();
1356    float canvas_width=gStyle->GetCanvasDefW();
# Line 1358 | Line 1361 | void save_with_ratio(TH1F *nominator, TH
1361    
1362    float xstretchfactor=((1-ratiospace)*(1-gStyle->GetPadTopMargin()))/((1)*ratiospace);
1363    
1364 <  TCanvas *main_canvas = new TCanvas("main_canvas","main_canvas",(Int_t)canvas_width,(Int_t)(canvas_height*(1+ratiospace)));
1364 >  TCanvas *Ratio_main_canvas = new TCanvas("Ratio_main_canvas","Ratio_main_canvas",(Int_t)canvas_width,(Int_t)(canvas_height*(1+ratiospace)));
1365    TPad *mainpad = new TPad("mainpad","mainpad",0,1-(1.0/(1+ratiospace)),1,1);//top (main) pad
1366    TPad *coverpad = new TPad("coverpad","coverpad",gStyle->GetPadLeftMargin()-0.008,1-(1.0/(1+ratiospace))-0.04,1,1-(1.0/(1+ratiospace))+0.103);//pad covering up the x scale
1367    TPad *bottompad = new TPad("bottompad", "Ratio Pad",0,0,1,(1-(1-bottommargin)/(1+ratiospace))-0.015); //bottom pad
1368    
1369 <  main_canvas->Range(0,0,1,1);
1370 <  main_canvas->SetBorderSize(0);
1371 <  main_canvas->SetFrameFillColor(0);
1369 >  Ratio_main_canvas->Range(0,0,1,1);
1370 >  Ratio_main_canvas->SetBorderSize(0);
1371 >  Ratio_main_canvas->SetFrameFillColor(0);
1372    
1373    mainpad->Draw();
1374    mainpad->cd();
# Line 1376 | Line 1379 | void save_with_ratio(TH1F *nominator, TH
1379    canvas->Range(0,0,1,1);
1380    canvas->Draw("same");
1381    mainpad->Modified();
1382 <  main_canvas->cd();
1382 >  Ratio_main_canvas->cd();
1383    coverpad->Draw();
1384    coverpad->cd();
1385    coverpad->Range(0,0,1,1);
# Line 1384 | Line 1387 | void save_with_ratio(TH1F *nominator, TH
1387    coverpad->SetBorderSize(0);
1388    coverpad->SetFrameFillColor(0);
1389    coverpad->Modified();
1390 <  main_canvas->cd();
1390 >  Ratio_main_canvas->cd();
1391    bottompad->SetTopMargin(ratiotopmargin);
1392    bottompad->SetBottomMargin(ratiobottommargin);
1393    bottompad->Draw();
# Line 1394 | Line 1397 | void save_with_ratio(TH1F *nominator, TH
1397    TH1F *ratio = (TH1F*)nominator->Clone(GetNumericHistoName().c_str());
1398    ratio->Divide(denominator);
1399  
1397
1400    TGraphAsymmErrors *eratio;
1401    TH1F *SystDown;
1402    TH1F *SystUp;
# Line 1420 | Line 1422 | void save_with_ratio(TH1F *nominator, TH
1422       }
1423      
1424    }
1423  eratio->SetFillColor(TColor::GetColor("#00ADE1"));
1425    
1426 +  if(syshisto && !do_bpred_ratio) {
1427 +       SystDown = (TH1F*)syshisto->Clone("SystDown");
1428 +       SystUp = (TH1F*)syshisto->Clone("SystUp");
1429 +       for(int i=1;i<=syshisto->GetNbinsX();i++) {
1430 +         SystDown->SetBinContent(i,1-syshisto->GetBinContent(i));
1431 +         SystUp->SetBinContent(i,1+syshisto->GetBinContent(i));
1432 +       }
1433 +       SystDown->SetLineColor(TColor::GetColor("#006DE1"));
1434 +       SystUp->SetLineColor(TColor::GetColor("#006DE1"));
1435 +  }
1436 +  eratio->SetFillColor(TColor::GetColor("#00ADE1"));
1437  
1438    ratio->SetTitle("");
1439    ratio->GetYaxis()->SetRangeUser(0.0,2.0);
# Line 1441 | Line 1453 | void save_with_ratio(TH1F *nominator, TH
1453    ratio->SetFillColor(TColor::GetColor("#58D3F7"));
1454    ratio->SetMarkerSize(0);
1455    ratio->Draw("e2");
1456 +  
1457 +  TGraphAsymmErrors *ratio_center = (TGraphAsymmErrors*)eratio->Clone("ratio_center");
1458 +  for(int i=0;i<ratio_center->GetN();i++) {
1459 +    ratio_center->SetPointError(i,ratio_center->GetErrorXlow(i),ratio_center->GetErrorXhigh(i),0.005,0.005);
1460 +  }
1461 +  
1462 +  ratio_center->SetFillColor(TColor::GetColor("#006381"));
1463 +  
1464    if(syshisto!=0) {
1465   //    sysratio->Draw("2");
1466   //    eratio->Draw("2,same");
# Line 1450 | Line 1470 | void save_with_ratio(TH1F *nominator, TH
1470    } else {
1471      eratio->Draw("2");
1472    }
1473 +  ratio_center->Draw("2");
1474    ratio->Draw("same,axis");
1475    TLine *oneline = new TLine(ratio->GetXaxis()->GetBinLowEdge(1),1,ratio->GetXaxis()->GetBinLowEdge(ratio->GetNbinsX())+ratio->GetXaxis()->GetBinWidth(ratio->GetNbinsX()),1);
1476    oneline->SetLineStyle(2);
1477    oneline->SetLineColor(kBlue);
1478    oneline->Draw("same");
1479  
1480 <  main_canvas->cd();
1481 <  main_canvas->Modified();
1482 <  main_canvas->cd();
1483 <  main_canvas->SetSelected(main_canvas);
1480 >  Ratio_main_canvas->cd();
1481 >  Ratio_main_canvas->Modified();
1482 >  Ratio_main_canvas->cd();
1483 >  Ratio_main_canvas->SetSelected(Ratio_main_canvas);
1484    
1485 <  CompleteSave(main_canvas,savemeas+"_withratio");
1485 >  CompleteSave(Ratio_main_canvas,savemeas+"_withratio");
1486    bottompad->cd();
1487    
1488    Double_t chi2;
# Line 1489 | Line 1510 | void save_with_ratio(TH1F *nominator, TH
1510    CompleteRatio->SetTextAlign(31);
1511    CompleteRatio->Draw();
1512    
1513 <  CompleteSave(main_canvas,savemeas+"_withratio_and_Chi2");
1513 >  CompleteSave(Ratio_main_canvas,savemeas+"_withratio_and_Chi2");
1514  
1515   //  float KS   = nominator->KolmogorovTest(denominator);
1516   //  stringstream KStext;
1517   //  Chi2text << "KS = " << KS << endl;
1518   //cout << "Found : " << KStext.str() << endl;
1519  
1520 <
1521 <  delete main_canvas;
1520 >  delete eratio;
1521 >  delete ratio_center;
1522 >  delete ratio;
1523 >  delete Ratio_main_canvas;
1524   }
1525  
1526 < void save_with_ratio_and_sys_band(TH1F *nominator, TH1F *denominator, TVirtualPad *canvas, string savemeas, float systematic, string yaxistitle="SysRatio") {
1526 > void save_with_ratio_and_sys_band(TH1F *nominator, TH1F *denominator, TVirtualPad *orig_canvas, string savemeas, bool do_bpred_ratio=false, bool extendrange=false, string yaxistitle="ratio",TH1F *syshisto=NULL) {
1527    //this function saves the pad being passed as well as a new one including the SysRatio.
1528 <  CompleteSave(canvas,savemeas);
1528 >  orig_canvas->cd();
1529 >  orig_canvas->Update();
1530 >  CompleteSave(orig_canvas,savemeas);
1531 >  
1532 >  TVirtualPad *canvas = (TVirtualPad*) orig_canvas->Clone("TempCanvas");//otherwise the Ratio_main_canvas will own our pad and destroy it upon deletion
1533  
1534    float bottommargin=gStyle->GetPadBottomMargin();
1535    float canvas_height=gStyle->GetCanvasDefH();
1536    float canvas_width=gStyle->GetCanvasDefW();
1537 <  float SysRatiospace=0.25;// space the SysRatio should take up (relative to original pad)
1537 >  float ratiospace=0.25;// space the ratio should take up (relative to original pad)
1538 >  
1539 >  float ratiobottommargin=0.3;
1540 >  float ratiotopmargin=0.1;
1541    
1542 <  float SysRatiobottommargin=0.3;
1543 <  float SysRatiotopmargin=0.1;
1542 >  float xstretchfactor=((1-ratiospace)*(1-gStyle->GetPadTopMargin()))/((1)*ratiospace);
1543 >  
1544 >  TCanvas *Ratio_main_canvas = new TCanvas("Ratio_main_canvas","Ratio_main_canvas",(Int_t)canvas_width,(Int_t)(canvas_height*(1+ratiospace)));
1545 >  TPad *mainpad = new TPad("mainpad","mainpad",0,1-(1.0/(1+ratiospace)),1,1);//top (main) pad
1546 >  TPad *coverpad = new TPad("coverpad","coverpad",gStyle->GetPadLeftMargin()-0.008,1-(1.0/(1+ratiospace))-0.04,1,1-(1.0/(1+ratiospace))+0.103);//pad covering up the x scale
1547 >  TPad *bottompad = new TPad("bottompad", "Ratio Pad",0,0,1,(1-(1-bottommargin)/(1+ratiospace))-0.015); //bottom pad
1548    
1549 <  float xstretchfactor=((1-SysRatiospace)*(1-gStyle->GetPadTopMargin()))/((1)*SysRatiospace);
1549 >  Ratio_main_canvas->Range(0,0,1,1);
1550 >  Ratio_main_canvas->SetBorderSize(0);
1551 >  Ratio_main_canvas->SetFrameFillColor(0);
1552    
1553 <  TCanvas *MainCanvas = new TCanvas("MainCanvas","MainCanvas",(Int_t)canvas_width,(Int_t)(canvas_height*(1+SysRatiospace)));
1554 <  TPad *MainPad = new TPad("MainPad","MainPad",0,1-(1.0/(1+SysRatiospace)),1,1);//top (main) pad
1555 <  TPad *CoverPad = new TPad("CoverPad","CoverPad",gStyle->GetPadLeftMargin()-0.008,1-(1.0/(1+SysRatiospace))-0.04,1,1-(1.0/(1+SysRatiospace))+0.103);//pad covering up the x scale
1556 <  TPad *BottomPad = new TPad("BottomPad", "Ratio Pad",0,0,1,(1-(1-bottommargin)/(1+SysRatiospace))-0.015); //bottom pad
1557 <  
1558 <  MainCanvas->Range(0,0,1,1);
1523 <  MainCanvas->SetBorderSize(0);
1524 <  MainCanvas->SetFrameFillColor(0);
1525 <  
1526 <  MainPad->Draw();
1527 <  MainPad->cd();
1528 <  MainPad->Range(0,0,1,1);
1529 <  MainPad->SetFillColor(kWhite);
1530 <  MainPad->SetBorderSize(0);
1531 <  MainPad->SetFrameFillColor(0);
1553 >  mainpad->Draw();
1554 >  mainpad->cd();
1555 >  mainpad->Range(0,0,1,1);
1556 >  mainpad->SetFillColor(kWhite);
1557 >  mainpad->SetBorderSize(0);
1558 >  mainpad->SetFrameFillColor(0);
1559    canvas->Range(0,0,1,1);
1560    canvas->Draw("same");
1561 <  MainPad->Modified();
1562 <  MainCanvas->cd();
1563 <  CoverPad->Draw();
1564 <  CoverPad->cd();
1565 <  CoverPad->Range(0,0,1,1);
1566 <  CoverPad->SetFillColor(kWhite);
1567 <  CoverPad->SetBorderSize(0);
1568 <  CoverPad->SetFrameFillColor(0);
1569 <  CoverPad->Modified();
1570 <  MainCanvas->cd();
1571 <  BottomPad->SetTopMargin(SysRatiotopmargin);
1572 <  BottomPad->SetBottomMargin(SysRatiobottommargin);
1573 <  BottomPad->Draw();
1574 <  BottomPad->cd();
1575 <  BottomPad->Range(0,0,1,1);
1576 <  BottomPad->SetFillColor(kWhite);
1577 <  TH1F *SysRatio = (TH1F*)nominator->Clone(GetNumericHistoName().c_str());
1578 <  SysRatio->Divide(denominator);
1561 >  mainpad->Modified();
1562 >  Ratio_main_canvas->cd();
1563 >  coverpad->Draw();
1564 >  coverpad->cd();
1565 >  coverpad->Range(0,0,1,1);
1566 >  coverpad->SetFillColor(kWhite);
1567 >  coverpad->SetBorderSize(0);
1568 >  coverpad->SetFrameFillColor(0);
1569 >  coverpad->Modified();
1570 >  Ratio_main_canvas->cd();
1571 >  bottompad->SetTopMargin(ratiotopmargin);
1572 >  bottompad->SetBottomMargin(ratiobottommargin);
1573 >  bottompad->Draw();
1574 >  bottompad->cd();
1575 >  bottompad->Range(0,0,1,1);
1576 >  bottompad->SetFillColor(kWhite);
1577 >  TH1F *ratio = (TH1F*)nominator->Clone(GetNumericHistoName().c_str());
1578 >  ratio->Divide(denominator);
1579  
1580  
1581 +  TGraphAsymmErrors *eratio;
1582 +  TGraphAsymmErrors *SysEnvelope = new TGraphAsymmErrors();
1583 +  
1584 +  if(syshisto && !do_bpred_ratio) {
1585 +       for(int i=1;i<=syshisto->GetNbinsX();i++) {
1586 +         float dx=syshisto->GetBinWidth(i)/2.0;
1587 +         float dy=syshisto->GetBinContent(i);
1588 +         if(dy<0.01) dy=0.07;
1589 +         SysEnvelope->SetPoint(i-1,syshisto->GetBinCenter(i),1.0);
1590 +         SysEnvelope->SetPointError(i-1,dx,dx,dy,dy);
1591 +       }
1592 +       SysEnvelope->SetFillColor(TColor::GetColor("#006DE1"));
1593 +  }
1594  
1595 <  SysRatio->SetTitle("");
1596 <  SysRatio->GetYaxis()->SetRangeUser(0.0,2.0);
1597 <  SysRatio->GetXaxis()->SetTitle(nominator->GetXaxis()->GetTitle());
1598 <  SysRatio->GetXaxis()->CenterTitle();
1599 <  SysRatio->GetYaxis()->SetTitle(yaxistitle.c_str());
1600 <  SysRatio->GetYaxis()->SetTitleOffset(0.4);
1601 <  SysRatio->GetYaxis()->CenterTitle();
1602 <  SysRatio->SetStats(0);
1603 <  SysRatio->GetXaxis()->SetLabelSize(xstretchfactor*SysRatio->GetXaxis()->GetLabelSize());
1604 <  SysRatio->GetYaxis()->SetLabelSize(xstretchfactor*SysRatio->GetYaxis()->GetLabelSize());
1605 <  SysRatio->GetXaxis()->SetTitleSize(xstretchfactor*gStyle->GetTitleSize());
1606 <  SysRatio->GetYaxis()->SetTitleSize(xstretchfactor*gStyle->GetTitleSize());
1607 <  SysRatio->GetYaxis()->SetNdivisions(502,false);
1608 <  SysRatio->SetFillColor(TColor::GetColor("#58D3F7"));
1609 <  TBox *sysbox = new TBox(SysRatio->GetXaxis()->GetBinLowEdge(1),1-systematic,SysRatio->GetXaxis()->GetBinLowEdge(SysRatio->GetNbinsX())+SysRatio->GetXaxis()->GetBinWidth(SysRatio->GetNbinsX()),1+systematic);
1610 <  sysbox->SetFillColor(TColor::GetColor("#82FA58")); // light green
1611 <  
1572 <  SysRatio->Draw("e1");
1573 <  sysbox->Draw();
1574 <  SysRatio->Draw("e1,same");
1575 <  SysRatio->Draw("e1,axis,same");
1595 >  ratio->SetTitle("");
1596 >  ratio->GetYaxis()->SetRangeUser(0.0,2.0);
1597 >  if(do_bpred_ratio) ratio->GetYaxis()->SetRangeUser(0.0,2.0);
1598 >  if(extendrange) ratio->GetYaxis()->SetRangeUser(0.0,4.0);
1599 >  ratio->GetXaxis()->SetTitle(nominator->GetXaxis()->GetTitle());
1600 >  ratio->GetXaxis()->CenterTitle();
1601 >  ratio->GetYaxis()->SetTitle(yaxistitle.c_str());
1602 >  ratio->GetYaxis()->SetTitleOffset(0.4);
1603 >  ratio->GetYaxis()->CenterTitle();
1604 >  ratio->SetStats(0);
1605 >  ratio->GetXaxis()->SetLabelSize(xstretchfactor*ratio->GetXaxis()->GetLabelSize());
1606 >  ratio->GetYaxis()->SetLabelSize(xstretchfactor*ratio->GetYaxis()->GetLabelSize());
1607 >  ratio->GetXaxis()->SetTitleSize(xstretchfactor*gStyle->GetTitleSize());
1608 >  ratio->GetYaxis()->SetTitleSize(xstretchfactor*gStyle->GetTitleSize());
1609 >  ratio->GetYaxis()->SetNdivisions(502,false);
1610 >  ratio->SetFillColor(TColor::GetColor("#58D3F7"));
1611 >  ratio->Draw("e1");
1612    
1613 <  TLine *oneline = new TLine(SysRatio->GetXaxis()->GetBinLowEdge(1),1,SysRatio->GetXaxis()->GetBinLowEdge(SysRatio->GetNbinsX())+SysRatio->GetXaxis()->GetBinWidth(SysRatio->GetNbinsX()),1);
1613 >  if(syshisto!=0) {
1614 >    SysEnvelope->SetFillColor(TColor::GetColor("#FE9A2E"));
1615 >    SysEnvelope->Draw("2,same");
1616 >    ratio->Draw("e1,same");
1617 >  } else {
1618 >    eratio->Draw("1");
1619 >  }
1620 >  ratio->Draw("same,axis");
1621 >  TLine *oneline = new TLine(ratio->GetXaxis()->GetBinLowEdge(1),1,ratio->GetXaxis()->GetBinLowEdge(ratio->GetNbinsX())+ratio->GetXaxis()->GetBinWidth(ratio->GetNbinsX()),1);
1622    oneline->SetLineStyle(2);
1623    oneline->SetLineColor(kBlue);
1624    oneline->Draw("same");
1625  
1626 <  MainCanvas->cd();
1627 <  MainCanvas->Modified();
1628 <  MainCanvas->cd();
1629 <  MainCanvas->SetSelected(MainCanvas);
1626 >  Ratio_main_canvas->cd();
1627 >  Ratio_main_canvas->Modified();
1628 >  Ratio_main_canvas->cd();
1629 >  Ratio_main_canvas->SetSelected(Ratio_main_canvas);
1630 >  
1631 >  CompleteSave(Ratio_main_canvas,savemeas+"_withSysRatio");
1632 >  bottompad->cd();
1633    
1634 <  CompleteSave(MainCanvas,savemeas+"_withSYSSysRatio");
1635 <  delete SysRatio;
1636 <  delete MainCanvas;
1634 >  Double_t chi2;
1635 >  Int_t ndf,igood;
1636 >  Double_t chi2prob = MarcosChi2TestX(nominator,denominator,chi2,ndf,igood,"UW");
1637 >
1638 >  stringstream Chi2text;
1639 >  Chi2text << "#chi^{2} / ndf: " << chi2 << " / " << ndf;
1640 >  stringstream Chi2probtext;
1641 >  Chi2probtext << "#chi^{2} prob: " << chi2prob;
1642 >  TText* chi2box = write_text(0.02,0.11,Chi2text.str());
1643 >  chi2box->SetTextSize(0.08);
1644 >  chi2box->SetTextAlign(11);
1645 >  chi2box->Draw();
1646 >  TText* chi2probbox = write_text(0.02,0.04,Chi2probtext.str());
1647 >  chi2probbox->SetTextSize(0.08);
1648 >  chi2probbox->SetTextAlign(11);
1649 >  chi2probbox->Draw();
1650 >  
1651 >  stringstream CompRatio;
1652 >  CompRatio << "Ratio: " << nominator->Integral() << " / " << denominator->Integral() << " \n " << nominator->Integral()/denominator->Integral();
1653 >  TText *CompleteRatio = write_text(0.98,0.04,CompRatio.str());
1654 >  CompleteRatio->SetTextSize(0.08);
1655 >  CompleteRatio->SetTextAlign(31);
1656 >  CompleteRatio->Draw();
1657 >  
1658 >  CompleteSave(Ratio_main_canvas,savemeas+"_withSysRatio_and_Chi2");
1659 >  delete Ratio_main_canvas;
1660 >  delete ratio;
1661   }
1662  
1663 < TH1F* CollapseStack(THStack stack,TString hname="base") {
1663 > TH1F* CollapseStack(THStack stack,TString hname="CollapsedStack") {
1664     TH1F *bhist = ((TH1F*)((stack.GetHists())->At(0)));
1665     TH1F *basehisto = (TH1F*)bhist->Clone(hname);
1666     TIter next(stack.GetHists());
# Line 1603 | Line 1674 | TH1F* CollapseStack(THStack stack,TStrin
1674     return basehisto;
1675   }
1676  
1677 < void save_with_ratio(TH1F *nominator, THStack denominator, TVirtualPad *canvas, string savemeas, bool do_bpred_ratio=false) {
1678 <  save_with_ratio(nominator, CollapseStack(denominator), canvas, savemeas, do_bpred_ratio);
1677 > void Save_With_Ratio(TH1F *nominator, THStack denominator, TVirtualPad *canvas, string savemeas, bool do_bpred_ratio=false) {
1678 >  TH1F *denominator_histo = (TH1F*) CollapseStack(denominator);
1679 >  Save_With_Ratio(nominator, denominator_histo, canvas, savemeas, do_bpred_ratio);
1680 >  delete denominator_histo;
1681   }
1682  
1683   void flag_this_change(string function, int line, int checked=0) {
# Line 1789 | Line 1862 | string removefunnystring(string name) {
1862    name=ReplaceCharacter(name,"]","_");
1863    name=ReplaceCharacter(name,"{","_");
1864    name=ReplaceCharacter(name,"}","_");
1865 +  name=ReplaceCharacter(name,"(","_");
1866 +  name=ReplaceCharacter(name,")","_");
1867 +  name=ReplaceCharacter(name," ","_");
1868    name=ReplaceCharacter(name,".","_");
1869    name=ReplaceCharacter(name,",","_");
1870    name=ReplaceCharacter(name,";","_");
# Line 1803 | Line 1879 | string removefunnystring(string name) {
1879    return name;
1880   }
1881  
1882 + TPolyLine* GetFitUncertaintyShape(TF1 *fit, float low, float high,float x0,float x1) {
1883 +  int nPoints=1000;
1884 +  double x[nPoints+1];
1885 +  double y[nPoints];
1886 +  
1887 +  float par1=fit->GetParameter(0);
1888 +  float dpar1=fit->GetParError(0);
1889 +  float par2=fit->GetParameter(1);
1890 +  float dpar2=fit->GetParError(1);
1891 +    
1892 +  float step=(x1-x0)/(nPoints/2.0 -1);
1893 +  for(int i=0;i<nPoints/2.0;i++) {
1894 +    x[i]=x0+i*step;
1895 +    y[i]=(par1+x[i]*par2)    -    sqrt(dpar1*dpar1+dpar2*dpar2*x[i]*x[i]);
1896 +    x[nPoints-1-i]=x[i];
1897 +    y[nPoints-1-i]=(par1+x[i]*par2)    +    sqrt(dpar1*dpar1+dpar2*dpar2*x[i]*x[i]);
1898 +    if(y[i]<low) y[i]=low;
1899 +    if(y[i]>high) y[i]=high;
1900 +    if(y[nPoints-1-i]<low) y[nPoints-1-i]=low;
1901 +    if(y[nPoints-1-i]>high) y[nPoints-1-i]=high;
1902 +  }
1903 +  
1904 +  x[nPoints]=x[0];
1905 +  y[nPoints]=y[0];
1906 +  
1907 +  TPolyLine *l = new TPolyLine(nPoints+1,x,y);
1908 +  l->SetFillColor(TColor::GetColor("#5858FA"));
1909 +  l->SetLineColor(TColor::GetColor("#5858FA"));
1910 +  l->SetLineWidth(1);
1911 +  return l;
1912 + }  
1913 +
1914 + //code for ReplaceAll copied from
1915 + //http://stackoverflow.com/questions/5343190/c-how-do-i-replace-all-instances-of-of-a-string-with-another-string
1916 + string ReplaceAll(string str, string from, string to) {
1917 +    size_t start_pos = 0;
1918 +    while((start_pos = str.find(from, start_pos)) != std::string::npos) {
1919 +         str.replace(start_pos, from.length(), to);
1920 +         start_pos += to.length(); // ...
1921 +    }
1922 +    return str;
1923 + }
1924 +
1925 + TPolyLine* GetFitUncertaintyShape(TH1 *histo, string fitname, float low, float high) {
1926 +  TF1 *fit = (TF1*)histo->GetFunction(fitname.c_str());
1927 +  if(!fit) {
1928 +    cout << "CANNOT PRODUCE FIT UNCERTAINTY AS THERE IS NO FIT CALLED " << fitname << " IN " << histo->GetName() << endl;
1929 +    return NULL;
1930 +  }
1931 +  float x0=histo->GetBinLowEdge(1);
1932 +  float x1=histo->GetBinLowEdge(histo->GetNbinsX())+histo->GetBinWidth(histo->GetNbinsX());
1933 +  return GetFitUncertaintyShape(fit, low, high,x0,x1);
1934 + }
1935 +
1936 + TPolyLine* GetFitUncertaintyShape(TGraphErrors *gr, string fitname, float low, float high, float minx=-999, float maxx=-999) {
1937 +  TF1 *fit = (TF1*)gr->GetFunction(fitname.c_str());
1938 +  if(!fit) {
1939 +    cout << "CANNOT PRODUCE FIT UNCERTAINTY AS THERE IS NO FIT CALLED " << fitname << " IN " << gr->GetName() << endl;
1940 +    return NULL;
1941 +  }
1942 +  double x,y;
1943 +  gr->GetPoint(0,x,y);
1944 +  float min=x;
1945 +  gr->GetPoint(gr->GetN()-1,x,y);
1946 +  float max=x;
1947 +  if(minx>-999 && maxx>-999 && minx<maxx) {
1948 +    max=maxx;
1949 +    min=minx;
1950 +  }
1951 +  return GetFitUncertaintyShape(fit, low, high,min,max);
1952 + }
1953 +
1954   stringstream all_bugs;
1955  
1956   void bug_tracker(string function, int line, string description) {
# Line 1853 | Line 2001 | float generalizedpSTAR(float massmother,
2001          res=TMath::Sqrt(res)/(2*massmother);
2002          return res;
2003   }
2004 +
2005 + void FindMinMax(TGraphErrors *gr, float &min, float &max) {
2006 +  double x,y;
2007 +  gr->GetPoint(0,x,y);
2008 +  min=500*y;
2009 +  max=0.0002*y;
2010 +  for(int i=0;i<gr->GetN();i++) {
2011 +    gr->GetPoint(i,x,y);
2012 +    cout << "Point at " << x << "/" << y << " with error " << gr->GetErrorY(i) << endl;
2013 +    float potmin=y-1.5*gr->GetErrorY(i); // taking 1.5 times the error to allow for enough space to present the plot
2014 +    float potmax=y+1.5*gr->GetErrorY(i); // taking 1.5 times the error to allow for enough space to present the plot
2015 +    if(potmin<min) min=potmin;
2016 +    if(potmax>max) max=potmax;
2017 +  }
2018 + }
2019 +
2020 +
2021 + void CleanLegends() {
2022 +  for(int ihist=(int)PlottingSetup::FakeHistoHeap.size()-1;ihist>=0;ihist--) {
2023 +    PlottingSetup::FakeHistoHeap[ihist]->Delete();
2024 +    PlottingSetup::FakeHistoHeap.pop_back();
2025 +  }
2026 + }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines