622 lines
20 KiB
C++
622 lines
20 KiB
C++
//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;
|
|
|
|
}
|