void getEff5(TH1D* h1, TH1D* h2, TH1D* h3, TH1D* h4, TH1D* h5, TString varType, TString INPUTDIR_PREFIX, double from, double to) {
gStyle->SetOptTitle(0);
gStyle->SetEndErrorSize(2);
gStyle->SetErrorX(0.5);
TLegend* leg = MakeLegend();
TPaveText* pt = MakeTPave();
TCanvas* c1 = new TCanvas("c1","c1",800,600);
c1->cd();
if (varType != "rapidity") c1->SetLogx();
h1->SetMarkerSize(0);
h1->SetLineWidth(1.2);
//range
if (varType == "pt") h1->GetXaxis()->SetRangeUser(8,180);
else if (varType == "vtx") h1->GetXaxis()->SetRangeUser(0,20.4);
else if (varType == "mass") h1->GetXaxis()->SetRangeUser(15, 999);
h1->GetYaxis()->SetRangeUser(from,to);
h1->GetXaxis()->SetMoreLogLabels();
h1->GetYaxis()->SetTitle("#epsilon");
h1->SetLineColor(kRed);
h1->SetFillColor(kRed);
h2->SetMarkerStyle(20);
h2->SetMarkerSize(1.1);
h2->SetMarkerColor(kBlack);
h2->SetLineColor(kBlack);
h3->SetMarkerStyle(20);
h3->SetMarkerSize(1.1);
h3->SetMarkerColor(kGreen);
h3->SetLineColor(kGreen);
h4->SetMarkerStyle(20);
h4->SetMarkerSize(1.1);
h4->SetMarkerColor(kBlue);
h4->SetLineColor(kBlue);
h5->SetMarkerStyle(20);
h5->SetMarkerSize(1.1);
h5->SetMarkerColor(kViolet);
h5->SetLineColor(kViolet);
//if you compare data to data
h1->Draw("E2");
h1->GetXaxis()->SetTitle(getTitle(varType, true));
h2->Draw("PEsame");
h3->Draw("PEsame");
h4->Draw("PEsame");
h5->Draw("PEsame");
leg->AddEntry(h1,"MC","f");
leg->AddEntry(h2,"Data, Run A","pl");
leg->AddEntry(h3,"Data, Run B","pl");
leg->AddEntry(h4,"Data, Run A+B combined","pl");
leg->AddEntry(h5,"Data, Run A+B split","pl");
leg->Draw("same");
//if (dataOnly && !(varType == "rrr")) pt->Draw("same");
//if (varType == "pt") c1->SetLogx();
//Save
c1->SaveAs(INPUTDIR_PREFIX+"/"+h1->GetName()+varType+".png");
}
void getComparisonABCDratio(TH1D* hmc, TH1D* hAB, TH1D* hC, TH1D* hD, TString varType, bool dataOnly, TString INPUTDIR_PREFIX, TString run) {
gStyle->SetOptTitle(0);
gStyle->SetEndErrorSize(2);
gStyle->SetErrorX(0.5);
TLegend* leg = MakeLegend();
TPaveText* pt = MakeTPave();
TCanvas* c1 = new TCanvas("c1","c1",800,600);
c1->cd();
//c1->SetLogx();
//range
if (varType == "pt" || varType == "pt2") {
//h1->GetXaxis()->SetRangeUser(8,180);
c1->SetLogx();
}
//else if (varType == "vtx") h1->GetXaxis()->SetRangeUser(0,20.4);
//get ratios
hAB->Divide(hAB,hmc);
hC->Divide(hC,hmc);
hD->Divide(hD,hmc);
hAB->GetYaxis()->SetRangeUser(0.92,1.12);
hAB->GetXaxis()->SetMoreLogLabels();
hAB->GetYaxis()->SetTitle("#epsilon");
hAB->SetMarkerStyle(20);
hAB->SetMarkerSize(1.1);
hC->SetMarkerStyle(20);
hC->SetMarkerSize(1.1);
hC->SetMarkerColor(kBlue);
hC->SetLineColor(kBlue);
hD->SetMarkerStyle(20);
hD->SetMarkerSize(1.1);
hD->SetMarkerColor(kViolet);
hD->SetLineColor(kViolet);
//hmc->Draw("E2");
//hmc->GetXaxis()->SetTitle(getTitle(varType));
hAB->Draw("PE");
hAB->GetXaxis()->SetTitle(getTitle(varType));
hC->Draw("samePE");
hD->Draw("samePE");
//leg->AddEntry(hmc,"MC, 53X","f");
leg->AddEntry(hAB,"Data Run A+B/MC","pl");
leg->AddEntry(hC,"Data Run C/MC","pl");
leg->AddEntry(hD,"Data Run D/MC","pl");
leg->Draw("same");
//if (dataOnly && !(varType == "rrr")) pt->Draw("same");
//if (varType == "pt") c1->SetLogx();
c1->SaveAs(INPUTDIR_PREFIX+"/"+hmc->GetName()+varType+".png");
c1->SaveAs(INPUTDIR_PREFIX+"/"+hmc->GetName()+varType+".C");
}
void getEff2(TH1D* h1, TH1D* h2, bool isData, TString varType, TString INPUTDIR_PREFIX, double from, double to) {
gStyle->SetOptTitle(0);
gStyle->SetEndErrorSize(2);
gStyle->SetErrorX(0.5);
TLegend* leg = MakeLegend();
TPaveText* pt = MakeTPave();
TCanvas* c1 = new TCanvas("c1","c1",800,600);
c1->cd();
if (varType != "rapidity") c1->SetLogx();
h1->SetMarkerSize(0);
h1->SetLineWidth(1.2);
//range
if (varType == "pt") h1->GetXaxis()->SetRangeUser(8,180);
else if (varType == "vtx") h1->GetXaxis()->SetRangeUser(0,20.4);
else if (varType == "mass") h1->GetXaxis()->SetRangeUser(15, 999);
h1->GetYaxis()->SetRangeUser(from,to);
h1->GetXaxis()->SetMoreLogLabels();
h1->GetYaxis()->SetTitle("#epsilon");
h1->SetLineColor(kRed);
h1->SetFillColor(kRed);
if (!isData) {
h2->SetLineColor(kGreen+1);
h2->SetFillColor(kGreen+1);
h2->SetFillStyle(3144);
} else {
h2->SetMarkerStyle(20);
}
//if you compare data to data
h1->Draw("E2");
h1->GetXaxis()->SetTitle(getTitle(varType,true));
if (!isData) {
h2->Draw("E2same");
} else {
h2->Draw("PEsame");
}
leg->AddEntry(h1,"MC, combined PF iso","f");
if (!isData) {
leg->AddEntry(h2,"MC, PF iso no e-#gamma","f");
} else {
leg->AddEntry(h2,"T&P data","pl");
}
leg->Draw("same");
//if (dataOnly && !(varType == "rrr")) pt->Draw("same");
//if (varType == "pt") c1->SetLogx();
//Save
c1->SaveAs(INPUTDIR_PREFIX+"/"+h1->GetName()+getTitle(varType,true)+".png");
}
//FIXME special for Hwidong
void getEffRatSpecial(TH1D* h1, TH1D* h2, TH1D* h3, TH1D* h4, TString varType, TString INPUTDIR_PREFIX, double from, double to) {
gStyle->SetOptTitle(0);
gStyle->SetEndErrorSize(2);
gStyle->SetErrorX(0.5);
TLegend* leg = MakeLegend();
TPaveText* pt = MakeTPave();
TCanvas* c1 = new TCanvas("c1","c1",800,600);
c1->cd();
if (varType != "rapidity") c1->SetLogx();
h1->SetMarkerSize(0);
h1->SetLineWidth(1.2);
//range
if (varType == "pt") h1->GetXaxis()->SetRangeUser(8,180);
else if (varType == "vtx") h1->GetXaxis()->SetRangeUser(0,20.4);
else if (varType == "mass") h1->GetXaxis()->SetRangeUser(15, 999);
h1->GetYaxis()->SetRangeUser(from,to);
h1->GetXaxis()->SetMoreLogLabels();
h1->GetYaxis()->SetTitle("#epsilon");
h1->SetLineColor(kBlack);
h1->SetFillColor(kBlack);
h2->SetLineColor(kBlue);
h2->SetFillColor(kBlue);
h2->SetFillStyle(3144);
h3->SetLineColor(kRed);
h3->SetFillColor(kRed);
h3->SetFillStyle(3244);
h4->SetLineColor(kGreen);
h4->SetFillColor(kGreen);
h4->SetFillStyle(3344);
//if you compare data to data
h1->Draw("E2");
h1->GetXaxis()->SetTitle(getTitle(varType, true));
h2->Draw("E2same");
h3->Draw("E2same");
h4->Draw("E2same");
leg->AddEntry(h1,"Binning 1","f");
leg->AddEntry(h2,"Binning 2","f");
leg->AddEntry(h3,"Binning 3","f");
leg->AddEntry(h4,"Binning 4","f");
leg->Draw("same");
//if (dataOnly && !(varType == "rrr")) pt->Draw("same");
//if (varType == "pt") c1->SetLogx();
//Save
cout << "Trying to save" << endl;
c1->SaveAs(INPUTDIR_PREFIX+"/"+h1->GetName()+varType+".png");
}
void getEff1(TH1D* h1, bool isData, TString varType, TString INPUTDIR_PREFIX, double from, double to) {
gStyle->SetOptTitle(0);
gStyle->SetEndErrorSize(2);
gStyle->SetErrorX(0.5);
TLegend* leg = MakeLegend();
TPaveText* pt = MakeTPave();
TCanvas* c1 = new TCanvas("c1","c1",800,600);
c1->cd();
if (varType != "rapidity") c1->SetLogx();
h1->SetMarkerSize(0);
h1->SetLineWidth(1.2);
//range
if (varType == "pt") h1->GetXaxis()->SetRangeUser(8,180);
else if (varType == "vtx") h1->GetXaxis()->SetRangeUser(0,20.4);
else if (varType == "mass") h1->GetXaxis()->SetRangeUser(15, 999);
h1->GetYaxis()->SetRangeUser(from,to);
h1->GetXaxis()->SetMoreLogLabels();
h1->GetYaxis()->SetTitle("#epsilon");
if (!isData) {
h1->SetLineColor(kRed);
h1->SetFillColor(kRed);
} else {
h1->SetMarkerStyle(20);
}
//if you compare data to data
if (!isData) {
h1->Draw("E2");
} else {
h1->Draw("PE");
}
cout << "WARNING: I hardcode that this is used for T&P" << endl;
h1->GetXaxis()->SetTitle(getTitle(varType,true));
if (!isData) {
leg->AddEntry(h1,"MC","f");
} else {
leg->AddEntry(h1,"MC","pl");
}
//leg->Draw("COLZsame");
//if (dataOnly && !(varType == "rrr")) pt->Draw("same");
//if (varType == "pt") c1->SetLogx();
//Save
c1->SaveAs(INPUTDIR_PREFIX+"/"+h1->GetName()+getTitle(varType, true)+".png");
}
void comparison_2D(void) {
bool DRAW_CHI2_ONLY = false;
gROOT->ProcessLine(".L ../../tools/histotools.C");
gROOT->ProcessLine(".L ../../tools/plottools.C");
TString result_type = "_fsrUnfGood"; //_fsrBbB
TString beautiful[6] = {"20-30 GeV","30-45 GeV","45-60 GeV","60-120 GeV","120-200 GeV","200-1500 GeV"};
TString filenames[6] = {"20to30","30to45","45to60","60to120","120to200","200to1500"};
TFile* input_2D = new TFile(result_type+"/xSecDET_results_2D.root");
input_2D->cd();
TMatrixT<double> absex_input = (TMatrixT<double>)gDirectory->Get("XSec");
TMatrixT<double> absex_input_err = (TMatrixT<double>)gDirectory->Get("XSecErr");
TMatrixT<double> absex_input_syserr = (TMatrixT<double>)gDirectory->Get("XSecSystErr");
TMatrixT<double> rshape_input = (TMatrixT<double>)gDirectory->Get("normXSecByBin");
TMatrixT<double> rshape_input_err = (TMatrixT<double>)gDirectory->Get("normXSecErrByBin");
TMatrixT<double> rshape_input_staterr = (TMatrixT<double>)gDirectory->Get("normXSecErrByBinStat");
TMatrixT<double> rshape_input_syserr = (TMatrixT<double>)gDirectory->Get("normXSecErrByBinSyst");
TH1D* hrshape = new TH1D("rshape","rshape",132,0,132);
TH1D* habsex = new TH1D("absex","absex",132,0,132);
for (int imass = 0; imass < 7; imass++) {
double binw = 0.1;
if (imass == 6) binw = 0.2;
for (int irap = 0; irap < 24; irap++) {
//WARNING: there was a mess with shifts/dimentions. Please check it first!
if (imass != 6) {
if (imass !=0) {
hrshape->SetBinContent(irap+1+(imass-1)*24,rshape_input(imass,irap));
hrshape->SetBinError(irap+1+(imass-1)*24,rshape_input_err(imass,irap));
habsex->SetBinContent(irap+1+(imass-1)*24,absex_input(imass,irap)/binw);
habsex->SetBinError(irap+1+(imass-1)*24,absex_input_err(imass,irap)/binw);
}
} else {
if (irap >= 12) continue;
hrshape->SetBinContent(irap+1+120,rshape_input(imass,irap));
hrshape->SetBinError(irap+1+120,rshape_input_err(imass,irap));
habsex->SetBinContent(irap+1+120,absex_input(imass,irap)/binw);
habsex->SetBinError(irap+1+120,absex_input_err(imass,irap)/binw);
}
}
}
TH1D* h1_ee = get2DSlice(habsex, "absrap2030",false, true); //WHAT IS THIS FALSE?? adds +23 shift!
TH1D* h2_ee = get2DSlice(habsex, "absrap3045",false, true);
TH1D* h3_ee = get2DSlice(habsex, "absrap4560",false, true);
TH1D* h4_ee = get2DSlice(habsex, "absrap60120",false, true);
TH1D* h5_ee = get2DSlice(habsex, "absrap120200",false, true);
TH1D* h6_ee = get2DSlice(habsex, "absrap200",false, true);
TFile* fout2 = new TFile("../absex_DET2D_EE.root","recreate");
fout2->cd();
h1_ee->Write("20to30");
h2_ee->Write("30to45");
h3_ee->Write("45to60");
h4_ee->Write("60to120");
h5_ee->Write("120to200");
h6_ee->Write("200to1500");
habsex->Write("hxsec");
fout2->Close();
TH1D* h1_er = get2DSlice(hrshape, "absrap2030",false, true); //WHAT IS THIS FALSE?? adds +23 shift!
TH1D* h2_er = get2DSlice(hrshape, "absrap3045",false, true);
TH1D* h3_er = get2DSlice(hrshape, "absrap4560",false, true);
TH1D* h4_er = get2DSlice(hrshape, "absrap60120",false, true);
TH1D* h5_er = get2DSlice(hrshape, "absrap120200",false, true);
TH1D* h6_er = get2DSlice(hrshape, "absrap200",false, true);
//Save output in the format same as muons
TFile* fout1 = new TFile("rshape_DET2D_EE.root","recreate");
fout1->cd();
h1_er->Write("20to30");
h1_er->Write("30to45");
h1_er->Write("45to60");
h1_er->Write("60to120");
h1_er->Write("120to200");
h1_er->Write("200to1500");
fout1->Close();
////////////////
//NOW PLOT THE COMPARISON TO MUONS, FOR ABSOLUTES
////////////////
//NORMALIZATION CROSS SECTION
cout << "Normalization cross section in the dielectron channel is: " << h4_ee->Integral() << endl;
std::vector<TH1D*> h__ee;
h__ee.push_back(h1_ee);
h__ee.push_back(h2_ee);
h__ee.push_back(h3_ee);
h__ee.push_back(h4_ee);
h__ee.push_back(h5_ee);
h__ee.push_back(h6_ee);
//NOW make a comparison
TFile* fmumu = new TFile("../absex_DET2D.root");
fmumu->cd();
TH1D* h1 = (TH1D*)gDirectory->Get("20to30");
TH1D* h2 = (TH1D*)gDirectory->Get("30to45");
TH1D* h3 = (TH1D*)gDirectory->Get("45to60");
TH1D* h4 = (TH1D*)gDirectory->Get("60to120");
cout << "Normalization cross section in the dimuon channel is: " << h4->Integral() << endl;
TH1D* h5 = (TH1D*)gDirectory->Get("120to200");
TH1D* h6 = (TH1D*)gDirectory->Get("200to1500");
std::vector<TH1D*> h__mumu;
h__mumu.push_back(h1);
h__mumu.push_back(h2);
h__mumu.push_back(h3);
h__mumu.push_back(h4);
h__mumu.push_back(h5);
h__mumu.push_back(h6);
//TH1D* hmumu = sliceCombiner(h1,h2,h3,h4,h5,h6,true);
//PLOT superimposed
TString WHICHES[6] = {"2030","3045","4560","60120","120200","2001500"};
TH1D* hratio[6];
for (int i = 0; i < h__mumu.size(); i++) {
if (!DRAW_CHI2_ONLY) {
TCanvas* c = new TCanvas(WHICHES[i],WHICHES[i]);
c->Draw();
c->cd();
TPad *p1 = new TPad("p1", "",0.0,0.25,1.0,1.0,0,0,0);
p1->Draw();
p1->cd();
//p1->SetLogy();
p1->SetGridx(kFALSE);
p1->SetGridy(kFALSE);
p1->SetBottomMargin(0.01);
//h__mumu.at(i)->SetMaximum(640);
h__mumu.at(i)->SetMaximum(h__mumu.at(i)->GetMaximum()*1.9);
h__mumu.at(i)->SetMarkerStyle(22);
h__mumu.at(i)->GetYaxis()->SetTitle("d#sigma/dMdy");
h__mumu.at(i)->GetXaxis()->SetTitle("|y|");
h__mumu.at(i)->Draw("P");
//h__ee.at(i)->Scale(h__mumu.at(i)->Integral()/h__ee.at(i)->Integral());
h__ee.at(i)->SetMarkerStyle(21);
h__ee.at(i)->SetMarkerColor(kRed);
h__ee.at(i)->SetLineColor(kRed);
h__ee.at(i)->Draw("Psame");
TLegend* leg = MakeLegend();
leg->AddEntry(h__mumu.at(i),"Dimuons","lp");
leg->AddEntry(h__ee.at(i),"Dielectrons","lp");
leg->Draw("same");
TPaveText* pt = MakeTPave();
pt->AddText(beautiful[i]);
pt->Draw("");
}
//Set up the ratio
//cout << h__mumu.at(i)->GetNbinsX() << " XX " << h__ee.at(i)->GetNbinsX() << endl;
hratio[i] = (TH1D*)h__mumu.at(i)->Clone();
//FindRatio(hratio[i],h__mumu.at(i),h__ee.at(i));
hratio[i]->Divide(h__mumu.at(i),h__ee.at(i));
if (!DRAW_CHI2_ONLY ) {
//tmp fix of errors at high mass
hratio[i]->SetBinError(22,1.5*hratio[i]->GetBinError(22));
hratio[i]->SetBinError(23,1.7*hratio[i]->GetBinError(23));
hratio[i]->SetBinError(24,2.5*hratio[i]->GetBinError(24));
gStyle->SetOptTitle(0);
TPad *p2 = new TPad("p2", "",0.0,0.0,1.0,0.247,0,0,0);
c->cd();
p2->Draw();
p2->cd();
//p2->SetLogx();
p2->SetTopMargin(0.007);
p2->SetBottomMargin(0.35);
p2->SetTickx();
//p2->SetGridx(kFALSE);
p2->SetGridy(kFALSE);
hratio[i]->GetYaxis()->SetTitle("Mu/Ele");
hratio[i]->SetStats(kFALSE);
hratio[i]->GetYaxis()->SetTitleOffset(0.29);
hratio[i]->GetXaxis()->SetTitle("|Y_{ll}|");
hratio[i]->GetYaxis()->SetTitleSize(0.15);
hratio[i]->GetXaxis()->SetTitleSize(0.15);
hratio[i]->GetYaxis()->SetLabelSize(0.10);
hratio[i]->GetXaxis()->SetLabelSize(0.15);
hratio[i]->SetMarkerSize(1.3);
hratio[i]->SetMaximum(1.45);
hratio[i]->SetMinimum(0.55);
hratio[i]->SetMarkerStyle(20);
hratio[i]->Draw("ep");
//c->Update();
TLine *line=new TLine();
line->SetLineColor(kBlue);
line->SetLineWidth(1);
line->DrawLine(0,1,2.4,1);
std::ostringstream pprint;
pprint.str("");
pprint<<i+1;
string label = pprint.str();
c->SaveAs("EMushapeR_Mhslice"+TString(label)+"_inAcc.pdf");
//c->SaveAs("EMu"+TString(label)+"_inAcc.png");
}
//CHI2 TEST
TH1D* chi2 = new TH1D("chi2","chi2",h__mumu.at(i)->GetNbinsX(),0,2.4);
for (int ii =0; ii < h__mumu.at(i)->GetNbinsX(); ii++) {
chi2->SetBinContent(ii+1,fabs(h__mumu.at(i)->GetBinContent(ii+1)-h__ee.at(i)->GetBinContent(ii+1))/sqrt(h__mumu.at(i)->GetBinError(ii+1)*h__ee.at(i)->GetBinError(ii+1)));
chi2->SetBinError(ii+1,0.0);
}
cout << "Full chi2 per NDF " << chi2->Integral()/double(hratio[i]->GetNbinsX()) << " " << hratio[i]->GetNbinsX() << endl;
TCanvas* c0 = new TCanvas("EMu"+filenames[i]+"_chi2","EMu"+filenames[i]+"_chi2");
c0->cd();
chi2->GetYaxis()->SetRangeUser(0,4.);
chi2->GetYaxis()->SetTitle("#chi^{2}");
chi2->GetXaxis()->SetTitle("|y|");
TPaveText* pt0 = MakeTPave();
std::ostringstream pprint;
pprint.str("");
pprint<<chi2->Integral()/double(hratio[i]->GetNbinsX());
pt0->AddText("#chi^{2}/ndf (CTEQ12) = "+TString(pprint.str()));
pprint.str("");
chi2->SetMarkerStyle(20);
chi2->SetMarkerColor(kRed);
chi2->SetLineColor(kRed);
chi2->Draw("PL");
pt0->Draw("same");
}
}
//TString WHICHHIST1, TString WHICHHIST2, assume 2 is data
void getDataMC(TH1D* h1, TH1D* h2, bool isData2, bool drawRatio) {
gStyle->SetEndErrorSize(2);
gStyle->SetErrorX(0.5);
TCanvas* c = new TCanvas();
c->cd();
TPad *p1 = new TPad("p1", "",0.0,0.242,1.0,1.0,0,0,0);
if (drawRatio) {
p1->Draw();
p1->cd();
p1->SetBottomMargin(0.013);
//p1->SetLogx();
}
double max = h1->GetMaximum() > h2->GetMaximum() ? h1->GetMaximum() : h2->GetMaximum();
h1->SetMaximum(1.2*max);
h1->GetYaxis()->SetTitleSize(0.065);
h1->GetYaxis()->SetTitle("Events/bin");
h1->SetLineColor(kOrange+3);
h1->SetFillColor(kOrange-2);
h1->SetMarkerSize(0);
h1->SetLineWidth(1.2);
h1->Draw("E2");
h2->SetMarkerSize(1.3);
h2->SetMarkerStyle(20);
h2->Draw("EPsame");
TLegend* leg = MakeLegend();
leg->AddEntry(h1,"MC based estimation","f");
leg->AddEntry(h2,"Data-driven prediction","lp");
leg->Draw("same");
//Draw ratios
if (drawRatio) {
gStyle->SetOptTitle(0);
gStyle->SetPadGridY(false);
TH1D* ratio = (TH1D*)h1->Clone();
ratio->Divide(h2,h1);
TPad *p2 = new TPad("p2", "",0.0,0.0,1.0,0.24,0,0,0);
c->cd();
p2->Draw();
p2->cd();
//p2->SetLogx();
p2->SetTopMargin(0.005);
p2->SetBottomMargin(0.35);
//p2->SetTickx();
ratio->GetYaxis()->SetTitle("e#mu/MC");
ratio->SetStats(kFALSE);
double ranger = ratio->GetBinError(1) > ratio->GetBinError(ratio->GetXaxis()->GetNbins()-1) ? ratio->GetBinError(1) : ratio->GetBinError(ratio->GetXaxis()->GetNbins()-1);
ratio->GetYaxis()->SetRangeUser(0.01,1.2*ranger+fabs(ratio->GetMaximum()));
ratio->GetYaxis()->SetNdivisions(5);
ratio->GetYaxis()->SetTitleOffset(0.29);
ratio->GetXaxis()->SetTitle("|Y_{#mu#mu}|");
ratio->GetXaxis()->SetMoreLogLabels();
ratio->GetYaxis()->SetTitleSize(0.23);
ratio->GetXaxis()->SetTitleSize(0.18);
//control digit sizes
ratio->GetYaxis()->SetLabelSize(1.2*0.15);
ratio->GetXaxis()->SetLabelSize(0.15);
//ratio->SetTickLength(ratio->GetTickLength()*2.);
ratio->SetFillStyle(3001);
ratio->SetLineColor(kBlack);
ratio->SetFillColor(kBlack);
ratio->SetMarkerSize(0);
ratio->SetLineWidth(1.2);
ratio->Draw("E3");
}
c->SaveAs(TString(h1->GetName())+".png");
}
//Assume heff_mct, htp, htpd, rat
void getEff3Stats(TH1D* h1, TH1D* h2, TH1D* h3, TH1D* h4, bool isData, TString varType, TString INPUTDIR_PREFIX, double from, double to) {
gStyle->SetOptTitle(0);
gStyle->SetEndErrorSize(2);
gStyle->SetErrorX(0.5);
TLegend* leg = MakeLegend();
TPaveText* pt = MakeTPave();
TCanvas* c1 = new TCanvas("c1","c1",800,600);
c1->cd();
if (varType != "rapidity") c1->SetLogx();
h1->SetMarkerSize(0);
h1->SetLineWidth(1.2);
//range
if (varType == "pt") h1->GetXaxis()->SetRangeUser(8,180);
else if (varType == "vtx") h1->GetXaxis()->SetRangeUser(0,20.4);
else if (varType == "mass") h1->GetXaxis()->SetRangeUser(15, 999);
h1->GetYaxis()->SetRangeUser(from,to);
h1->GetXaxis()->SetMoreLogLabels();
h1->GetYaxis()->SetTitle("#epsilon");
h1->SetLineColor(kRed);
h1->SetFillColor(kRed);
h2->SetLineColor(kBlue);
h2->SetFillColor(kBlue);
h4->SetLineColor(kGreen);
h4->SetFillColor(kGreen);
h4->SetFillStyle(3144);
//if you compare data to data
h1->Draw("E2");
h1->GetXaxis()->SetTitle(getTitle(varType, true));
h2->Draw("E2same");
h4->Draw("E2same");
leg->AddEntry(h1,"MC Truth","f");
leg->AddEntry(h2,"MC T&P","f");
leg->AddEntry(h4,"Ratio","f");
leg->Draw("same");
//if (dataOnly && !(varType == "rrr")) pt->Draw("same");
//if (varType == "pt") c1->SetLogx();
gROOT->ProcessLine(".L ../tools/stattools.C");
//Make TPave
ostringstream tmpstrm1;
ostringstream tmpstrm2;
ostringstream tmpstrm3;
tmpstrm3 << INPUTDIR_PREFIX << " " << h1->GetName();
tmpstrm1 << estimatorPerMassHisto1(h2,h3);
tmpstrm2 << estimatorPerMassHisto2(h2,h3,h1);
TPaveText *pt = new TPaveText(0.25,0.77,0.40,0.92,"brNDC");
pt->SetFillColor(0);
// pt->SetTextFont(72);
pt->SetTextSize(0.040);
pt->SetBorderSize(0);
pt->SetTextAlign(12);
pt->AddText(TString(tmpstrm3.str().c_str()));
pt->AddText("Simple estimator: "+TString(tmpstrm1.str().c_str()));
pt->AddText("Combined estimator: "+TString(tmpstrm2.str().c_str()));
pt->Draw("same");
//Save
c1->SaveAs(INPUTDIR_PREFIX+"/stats"+h1->GetName()+varType+".png");
}
