|
|
//Renata Kopecna
#include <design.hh>
#include <constants.hh>
#include <helpers.hh>
#include <TROOT.h>
#include <TBox.h>
#include <TPaveText.h>
#include <TF1.h>
#include <TLegend.h>
#include <TMultiGraph.h>
#include <TStyle.h>
#include <TFile.h>
#include <TGaxis.h>
#include <sstream>
#include <iostream>
//=========================================
//
// Latex in plots
//
//=========================================
std::string latex_decay(bool DTF){ //return "K_{S}^{0}#pi^{+}#mu^{+}#mu^{-}";
if (DTF) return "K^{+}#pi^{0}#mu^{+}#mu^{-}"; else return "#K^{+}#gamma#gamma#mu^{+}#mu^{-}"; }
std::string latex_Kst_decay(bool DTF){ if (DTF) return "K^{+}#pi^{0}"; else return "#K^{+}#gamma#gamma"; }
std::string MeVc(){ return "MeV"; } std::string GeVc(){ return "GeV"; } std::string GeVc_2(){ return "GeV^{2}"; }
std::string latex_pull(){ return "(x-x_{0})/#sigma"; }
std::string latex_angle(std::string angle, bool &is_ctl, bool &is_ctk, bool &is_phi){
if(strcmp(angle.c_str(), "ctk") == 0){ is_ctk = true; return "cos(#Theta_{K})"; } else if(strcmp(angle.c_str(), "ctl") == 0){ is_ctl = true; return "cos(#Theta_{L})"; } else if(strcmp(angle.c_str(), "phi") == 0){ is_phi = true; return "#phi [rad]"; } return ""; }
//Define the angles as chars, because ROOT
//Very not cool, but also very not cool
//of ROOT to not accept strings and I cannot be bothered
const char *LATEX_Q2_CH() { return LATEX_Q2.c_str(); } const char *LATEX_TL_CH() { return LATEX_TL.c_str(); } const char *LATEX_CTL_CH(){ return LATEX_CTL.c_str();} const char *LATEX_TK_CH() { return LATEX_TK.c_str(); } const char *LATEX_CTK_CH(){ return LATEX_CTK.c_str();} const char *LATEX_PHI_CH(){ return LATEX_PHI.c_str();}
std::string q2_label(double low_q2, double high_q2){ std::ostringstream qqbin; //TODO
qqbin << std::fixed << std::setprecision(2) <<low_q2<< " < q^{2} < " << high_q2; return qqbin.str();
} //=========================================
//
// General
//
//=========================================
void set_gStyle(){ gStyle->SetOptStat(0); gStyle->SetOptFit(0); gStyle->SetLabelFont(132, "xyz"); gStyle->SetTitleFont(132, "xyz"); gStyle->SetLegendFont(132); gStyle->SetStatFont(132); gStyle->SetPaintTextFormat(".1f"); }
//Save canvas with a given name and automatically append the type
//Type should be specified without the dot, so eg "eps" instead of ".eps
void saveTCanvas(TCanvas *c, std::string name, std::string type){ c->Print((name+"."+type).c_str(), type.c_str()); return; }
//Save several file types in one go
void saveTCanvas(TCanvas *c, std::string name, std::vector<std::string> types){ for (auto type: types) saveTCanvas(c,name,type); return; }
TLine *threeSigmaLine(double min, double max, bool plus){ int sigma = plus ? +3.0 : -3.0; TLine * line = new TLine(min, sigma, max, sigma); line->SetLineStyle(5); line->SetLineWidth(gStyle->GetLineWidth()); return line; } TLine *oneSigmaLine(double min, double max, bool plus){ int sigma = plus ? +1.0 : -1.0; TLine * line = new TLine(min, sigma, max, sigma); line->SetLineStyle(7); line->SetLineWidth(gStyle->GetLineWidth()); return line; }
void plotAndSave(TH1D *hist, std::string canvasName, std::string targetPath, std::string plotType){
TCanvas* cnvs = new TCanvas(canvasName.c_str(), canvasName.c_str(), 1600, 1200); cnvs->cd(); hist->Draw(); saveTCanvas(cnvs,targetPath, plotType);
delete cnvs; return;
} void plotAndSave(TH2D *hist, std::string canvasName, std::string targetPath, std::string plotType){
TCanvas* cnvs = new TCanvas(canvasName.c_str(), canvasName.c_str(), 1600, 1200); cnvs->cd(); hist->Draw("COLZ"); saveTCanvas(cnvs,targetPath, plotType);
delete cnvs; return;
}
void plotAndSaveWithLine(TH1D* hist, TF1 *line, std::string canvasName, std::string targetPath, std::string plotType){
//Create canvas
TCanvas* cnvs = new TCanvas(canvasName.c_str(), canvasName.c_str(), 1600, 1200); cnvs->cd();
//Add legend
//TLatex cannot do break lines and I cannot. Even.
TPaveText *leg = new TPaveText(0.55,0.65,0.87,0.87); leg->Paint("NDC"); leg->SetFillColor(kWhite); leg->SetTextFont(132); //Loop over params and print them
for (int n = 0; n < line->GetNpar(); n++){ std::ostringstream sParams; sParams << line->GetParName(n) << ": " << std::fixed << std::setprecision(4) << line->GetParameter(n) << "#pm" << line->GetParError(n); leg->AddText(sParams.str().c_str()); }
//Draw
hist->Draw("E1"); line->Draw("same"); leg->Draw("same");
//Save
cnvs->Print((targetPath+"."+plotType).c_str(), plotType.c_str());
//Delete and return
delete leg; delete cnvs; return;
}
//Sets TLatex
TLatex *getPrettyTex(double TextFontSize, int TextAlign){ TLatex *tex = new TLatex(); tex->SetNDC(true); tex->SetTextFont(132); tex->SetTextSize(TextFontSize); tex->SetTextAlign(TextAlign); return tex; }
//Add myThesis tag
void addThesistag(double x = 0.6, double y = 0.85, int color = 1, double scaling = 1.0){ if (!PLOT_THISTHESIS_TAG) return; TLatex* lhcbtext = getPrettyTex(0.07*scaling, 13); lhcbtext->SetTextColor(color); lhcbtext->DrawLatex(x,y,THISTHESIS_TAG.c_str()); return; } //Add lhcb tag
void addLHCbtag(double x = 0.6, double y = 0.85, std::string suffix = "", int color = 1, double scaling = 1.0){ TLatex* lhcbtext = getPrettyTex(0.07*scaling, 13); lhcbtext->SetTextColor(color); lhcbtext->DrawLatex(x,y,("LHCb "+suffix).c_str()); return; }
//=========================================
//
// Angular correction plots
//
//=========================================
//plot eff histograms for angular corrections
void plotAngular(TH1D *hist, TH1D *moments, bool write,std::string name, std::string appendix, std::string q2region, TLatex *tex, std::string folderName){ TCanvas* c0 = new TCanvas("c0", "c0", 1600, 1200); c0->cd(); hist->Scale(hist->GetNbinsX()/hist->Integral()); hist->SetMinimum(0.0); hist->Draw(); hist->GetYaxis()->SetTitle("Normalized entries (a.u.)"); moments->Scale(moments->GetNbinsX()/moments->Integral()); addLHCbtag(0.15,0.27,"simulation",1,1.0); addThesistag(0.15,0.20,1,0.9); moments->Draw("samel");
if (q2region != "") tex->DrawLatex(0.85, 0.15, q2region.c_str()); if (write) saveTCanvas(c0,folderName+name+appendix,"eps"); delete c0; }
//Overload the plot without plotting q2 region
void plotAngular(TH1D *hist, TH1D *moments, bool write,std::string name, std::string appendix, std::string folderName){ TLatex *tex = new TLatex(); plotAngular(hist, moments, write, name, appendix, "", tex, folderName); }
//plot 2D histograms for angular corrections
void plotAngular(TH2D *hist, TH2D *moments, bool write, std::string name, std::string appendix, std::string q2region, TLatex *tex, std::string folderName){ TCanvas* c0 = new TCanvas("c0", "c0", 1600, 800); c0->Divide(2,1); c0->cd(1); hist->Scale(hist->GetNbinsX()*hist->GetNbinsY()/hist->Integral()); hist->SetMinimum(0.0); hist->SetMaximum(1.4); hist->Draw("colz"); c0->cd(2); moments->Scale(moments->GetNbinsX()*moments->GetNbinsY()/moments->Integral()); moments->SetMinimum(0.0); moments->SetMaximum(1.4); moments->Draw("colz"); if (q2region != "") tex->DrawLatex(0.85, 0.15, q2region.c_str()); if (write) saveTCanvas(c0,folderName+name+appendix,"eps"); delete c0; }
//Overload the plot without plotting q2 region
void plotAngular(TH2D *hist, TH2D *moments, bool write,std::string name, std::string appendix, std::string folderName){ TLatex *tex = new TLatex(); plotAngular(hist, moments, write, name, appendix, "", tex, folderName); }
//Plot all the angular corrections in one canvas
void plotAngularInOne(TCanvas *c, int whichC, TH1D *moments, TH1D *ub, TH1D *hist, TH1D* ml, bool run_minuit, double q2min, double q2max, double y, bool cross){ c->cd(whichC)->SetMargin(0.15,0.05,0.1,0.05); moments->Draw("l");
TLine* line= new TLine(); line->SetLineStyle(kDashed); if (cross){ line->SetLineColor(kGray+1); line->DrawLine(q2min, -0.5, q2max, +0.5); line->DrawLine(q2min, +0.5, q2max, -0.5); } else{ ub->Draw("same"); hist->Draw("same"); if (run_minuit) ml->Draw("lsame"); line->SetLineColor(kRed); line->DrawLine(q2min, y, q2max, y); } }
//Label axes for angular correction polynomial scan
void labelAngularScan(std::vector<int>scan_low, std::vector<int>scan_range,TH2D *hist){
//label the x-axis
//loop over the range in thetak and thetal and print all possible combination
for(int tk = 0; tk < scan_range.at(1); tk++){ for(int tl = 0; tl < scan_range.at(2); tl++){ hist->GetXaxis()->SetBinLabel(tk * scan_range.at(2) + tl + 1, //+1 because bins start at 1
("#bf{"+std::to_string(tk+scan_low.at(1))+"} - "+std::to_string(tl+scan_low.at(2))).c_str()); } }
//label the y-axis
for(int qq = 0; qq < scan_range.at(0); qq++){ for(int fi =0; fi < scan_range.at(3); fi++){ hist->GetYaxis()->SetBinLabel(qq * scan_range.at(3) + fi + 1, ("#bf{"+std::to_string(qq+scan_low.at(0))+"} - "+std::to_string(fi+scan_low.at(3))).c_str()); } }
//draw labels on x-axis vertically
hist->GetXaxis()->LabelsOption("v");
return; }
//=========================================
//
// Fit results plots
//
//=========================================
int design_pull_basic(TH1D *pull, Float_t textsize, double eff_pullHeight){ pull->GetXaxis()->SetNoExponent(); //<-- spoils MaxDigits settings, so don't use it on other axis
pull->GetXaxis()->SetNdivisions(gStyle->GetNdivisions("X"));
pull->GetYaxis()->SetTitleSize(textsize/eff_pullHeight); pull->GetYaxis()->SetLabelSize(textsize/eff_pullHeight); pull->GetYaxis()->SetTitleOffset(gStyle->GetTitleOffset()*eff_pullHeight);
return 0; } int design_pull(TH1D *pull, Color_t lineColor, Color_t fillColor, double eff_pullHeight, double pullFrameRange){ Float_t textsize = gStyle->GetTextSize(); pull->GetXaxis()->SetTitleOffset(1.05); design_pull_basic(pull, textsize, eff_pullHeight);
pull->GetXaxis()->SetTitleSize (textsize/eff_pullHeight); pull->GetXaxis()->SetLabelSize (textsize/eff_pullHeight); pull->GetXaxis()->SetTickLength (pull->GetXaxis()->GetTickLength()/(eff_pullHeight/(1-eff_pullHeight)));
pull->GetYaxis()->CenterTitle(); pull->GetYaxis()->SetNdivisions (3, 9, 0, kTRUE);//gStyle->GetNdivisions("Y"));
pull->GetYaxis()->SetRangeUser (-pullFrameRange, pullFrameRange); pull->SetFillColor(fillColor); pull->SetLineColor(lineColor); pull->SetLineWidth(1);
return 0; } int design_pull(TGraphErrors *pull, Color_t lineColor, Color_t fillColor, double eff_pullHeight, double pullFrameRange){ Float_t textsize = gStyle->GetTextSize(); pull->GetXaxis()->SetTitleOffset(1.05); pull->GetXaxis()->SetNoExponent(); //<-- spoils MaxDigits settings, so don't use it on other axis
pull->GetYaxis()->SetTitleSize(textsize/eff_pullHeight); pull->GetYaxis()->SetLabelSize(textsize/eff_pullHeight); pull->GetYaxis()->SetTitleOffset(gStyle->GetTitleOffset()*eff_pullHeight);
pull->GetXaxis()->SetTitleSize (textsize/eff_pullHeight); pull->GetXaxis()->SetLabelSize (textsize/eff_pullHeight); pull->GetXaxis()->SetTickLength (pull->GetXaxis()->GetTickLength()/(eff_pullHeight/(1-eff_pullHeight)));
pull->GetYaxis()->CenterTitle(); pull->GetYaxis()->SetRangeUser (-pullFrameRange, pullFrameRange); pull->SetFillColor(fillColor); pull->SetLineColor(lineColor); pull->SetLineWidth(1);
return 0; }
void drawResonances(TCanvas *c, double min, double max){ c->cd(); TBox* box = new TBox(); box->SetFillColor(21); box->SetLineColor(21); box->SetFillStyle(1001); box->DrawBox(0.98, min, 1.10, max); box->DrawBox(8.0, min, 11.0, max); box->DrawBox(12.5, min, 15.0, max); c->Update(); //Probably not needed, but just ot be sure
return; } void drawResonances(TPad *c, double min, double max){ c->cd(); TBox* box = new TBox(); box->SetFillColor(21); box->SetLineColor(21); box->SetFillStyle(1001); box->DrawBox(0.98, min, 1.10, max); box->DrawBox(8.0, min, 11.0, max); box->DrawBox(12.5, min, 15.0, max); c->Update(); //Probably not needed, but just ot be sure
return; }
void designTGraph(TGraphErrors *graph, std::string title, std::string xAxisName, std::string yAxisName, Color_t color, Int_t markerStyle){ graph->SetName(title.c_str()); graph->SetLineColor(color); //graph->SetLineWidth(0);
graph->SetMarkerStyle(markerStyle); graph->SetMarkerColor(color); graph->GetXaxis()->SetTitle(xAxisName.c_str()); graph->GetYaxis()->SetTitle(yAxisName.c_str());
graph->GetXaxis()->SetTitleOffset(1.0); graph->GetYaxis()->SetTitleOffset(1.7); graph->SetTitle(""); graph->GetXaxis()->SetLabelSize(0.09); graph->GetYaxis()->SetLabelSize(0.09); graph->GetXaxis()->SetTitleSize(0.07); graph->GetYaxis()->SetTitleSize(0.05); return; }
void designMultiGraph(TMultiGraph *multGraph, std::string title, std::string xAxisName, std::string yAxisName){
//One needs to draw the multGraph first!
multGraph->SetName(title.c_str()); multGraph->GetXaxis()->SetTitle(xAxisName.c_str()); multGraph->GetYaxis()->SetTitle(yAxisName.c_str()); gStyle->SetOptStat(0);
multGraph->GetXaxis()->SetTitleOffset(1.0); multGraph->GetYaxis()->SetTitleOffset(1.4); multGraph->SetTitle(""); multGraph->GetXaxis()->SetLabelSize(0.05); multGraph->GetYaxis()->SetLabelSize(0.05); multGraph->GetXaxis()->SetTitleSize(0.05); multGraph->GetYaxis()->SetTitleSize(0.055); return; }
int design_YieldInQ2(int Run, TGraphErrors *graphSig, TGraphErrors *graphBkg, TGraphErrors *graphSignificance, TGraphErrors *CMS, bool fixRange, std::string savePath, std::vector<std::string> extensions){
std::string mainName = "Q2_Run" + std::to_string(Run); double y_scale = 1.5; double y_scale_1 = (Run==1) ? 2.5 : 1.75;
//Create a TCanvas
TCanvas *c = new TCanvas(mainName.c_str(),mainName.c_str(),750, 575); c->SetRightMargin(0.13); c->SetLeftMargin(0.23); c->SetTopMargin(0.075); c->SetBottomMargin(0.12); c->cd();
designTGraph(graphSig,"Signal","Q2","Yield",kBlack,20); designTGraph(graphBkg,"Background","Q2","Yield",kRed,20); graphSig->SetLineStyle(1); graphSig->SetLineWidth(1); graphSig->SetMarkerSize(0.5); graphBkg->SetLineStyle(1); graphBkg->SetLineWidth(1); graphBkg->SetMarkerSize(0.65);
designTGraph(graphSignificance,"Significance","Q2","S/(S+B)",kBlue,20); designTGraph(CMS,"Significance_CMS","Q2","S/(S+B)",kGreen+2,20);
graphSignificance->GetYaxis()->SetRangeUser(0.0,graphSignificance->GetY()[0]*y_scale); graphSignificance->GetXaxis()->SetRangeUser(Q2_MIN_RANGE,Q2_MAX_RANGE); CMS->GetXaxis()->SetRangeUser(Q2_MIN_RANGE,Q2_MAX_RANGE); graphSignificance->SetLineWidth(3); CMS->SetLineWidth(3);
//Create multigraphs
TMultiGraph *mg = new TMultiGraph(); mg->Add(graphSig,"AP"); mg->Add(graphBkg,"AP"); mg->Draw("AP"); mg->GetYaxis()->SetRangeUser(0.0,graphBkg->GetY()[1]*y_scale); mg->GetXaxis()->SetRangeUser(Q2_MIN_RANGE,Q2_MAX_RANGE); designMultiGraph(mg,"","q^{2} [GeV^{2}]","Yield");
TMultiGraph *mg_FoM = new TMultiGraph(); mg_FoM->Add(graphSignificance,"AP"); mg_FoM->Add(CMS,"AP"); mg_FoM->Draw("AP"); mg_FoM->GetXaxis()->SetRangeUser(Q2_MIN_RANGE,Q2_MAX_RANGE); designMultiGraph(mg_FoM,"","q^{2} [GeV^{2}]","S/(S+B)");
//Create two pads (to get significance y-axis on the right)
TPad *pad1 = new TPad("pad1","",0,0,1,1); TPad *pad2 = new TPad("pad2","",0,0,1,1);
pad2->SetFillStyle(4000); //will be transparent
pad2->SetFrameFillStyle(0);
pad1->SetRightMargin(0.13); pad1->SetLeftMargin(0.15); pad1->SetTopMargin(0.07); pad1->SetBottomMargin(0.12);
pad2->SetRightMargin(0.13); pad2->SetLeftMargin(0.15); pad2->SetTopMargin(0.07); pad2->SetBottomMargin(0.12);
pad1->Draw(); pad1->cd();
//Add legends
TLegend *leg = new TLegend(0.15,0.93,0.52,0.82); leg->AddEntry(graphSig, "Signal yield","l"); leg->AddEntry(graphBkg, "Background yield","l");
TLegend *legSignificance = new TLegend(0.52,0.93,0.8,0.82); legSignificance->AddEntry(graphSignificance, "Significance","l"); legSignificance->AddEntry(CMS, "CMS","l");
//Plot it
pad1->cd(); gStyle->SetGridColor(kGray); pad1->SetGridy(); pad1->SetGridx(); mg->Draw("SAME"); mg->GetYaxis()->SetTitle("Yield"); mg->GetYaxis()->SetRangeUser(0.0,graphBkg->GetY()[1]*y_scale); mg->GetXaxis()->SetRangeUser(Q2_MIN_RANGE,Q2_MAX_RANGE); mg->GetXaxis()->SetLabelSize(0); mg->GetYaxis()->SetTickLength(0); leg->Draw("SAME"); pad1->Update(); c->cd();
pad2->Draw(); pad2->cd(); //Get axis on the left side
TGaxis *axis = new TGaxis(Q2_MAX_RANGE,0.0,Q2_MAX_RANGE,graphSignificance->GetY()[0]*y_scale_1,0,graphSignificance->GetY()[0]*y_scale_1,510,"+L"); axis->SetTitle("S/sqrt(S+B)"); axis->SetTextFont(21); mg_FoM->Draw("AP"); mg_FoM->GetXaxis()->SetRangeUser(Q2_MIN_RANGE,Q2_MAX_RANGE); mg_FoM->GetYaxis()->SetRangeUser(0.0,graphSignificance->GetY()[0]*y_scale_1); mg_FoM->GetYaxis()->SetLabelSize(0); mg_FoM->GetYaxis()->SetTickLength(0); mg_FoM->GetYaxis()->SetTitle(""); axis->Draw("SAME"); legSignificance->Draw("SAME"); pad2->Update(); c->cd();
//Save it
if (!fixRange) savePath = savePath + "_sigma"; TFile *file = new TFile ((savePath+".root").c_str(), "RECREATE"); file->cd(); graphSig->Write(); graphBkg->Write(); graphSignificance->Write(); CMS->Write(); c->Write(); file->Close(); saveTCanvas(c,savePath,extensions);
return 0; }
int design_SignificanceInQ2(int Run, TGraphErrors *graphSignificance, TGraphErrors *CMS, bool fixRange, std::string savePath, std::vector<std::string>extensions){
std::string mainName = "Q2_Run" + std::to_string(Run); double y_scale = 1.4;
//Create a TCanvas
TCanvas *c = new TCanvas(mainName.c_str(),mainName.c_str(),750, 600); c->SetRightMargin(0.13); c->SetLeftMargin(0.15); c->SetTopMargin(0.07); c->SetBottomMargin(0.12); gStyle->SetGridColor(kGray); c->SetGridy(); c->SetGridx(); c->SetBottomMargin(0.16);
designTGraph(graphSignificance,"Significance","Q2","S/(S+B)",kBlue,20); designTGraph(CMS,"Significance_CMS","Q2","S/(S+B)",kGreen+2,20);
graphSignificance->GetYaxis()->SetRangeUser(0.0,graphSignificance->GetY()[0]*y_scale); graphSignificance->GetXaxis()->SetRangeUser(Q2_MIN_RANGE,Q2_MAX_RANGE); CMS->GetXaxis()->SetRangeUser(Q2_MIN_RANGE,Q2_MAX_RANGE); graphSignificance->SetLineWidth(3); CMS->SetLineWidth(3);
//Create multigraph
TMultiGraph *mg_FoM = new TMultiGraph(); mg_FoM->Add(graphSignificance,"AP"); mg_FoM->Add(CMS,"AP"); mg_FoM->Draw("AP"); designMultiGraph(mg_FoM,"","q^{2} [GeV^{2}]","S/(S+B)");
//Add legend
TLegend *legSignificance = new TLegend(0.35,0.9,0.65,0.75); legSignificance->AddEntry(graphSignificance, "Significance","l"); legSignificance->AddEntry(CMS, "CMS","l");
//Plot it
c->cd(); mg_FoM->Draw("AP"); mg_FoM->GetYaxis()->SetTitle("S/(S+B)"); mg_FoM->GetXaxis()->SetRangeUser(Q2_MIN_RANGE,Q2_MAX_RANGE); mg_FoM->GetYaxis()->SetRangeUser(0.0,graphSignificance->GetY()[0]*y_scale); mg_FoM->GetXaxis()->SetTitle("q^{2} [GeV^{2}]"); legSignificance->Draw("SAME"); c->Update();
//Save it
if (!fixRange) savePath = savePath + "_sigma"; TFile *file = new TFile ((savePath+".root").c_str(), "RECREATE"); file->cd(); graphSignificance->Write(); CMS->Write(); c->Write(); file->Close(); saveTCanvas(c,savePath,extensions);
return 0;
}
|