genie_plot_utils.h

Go to the documentation of this file.

#pragma once

#include "TH1D.h"
#include "TCanvas.h"
#include "TFile.h"
#include "TStyle.h"
#include "TLine.h"
#include "TLegend.h"
#include "TLatex.h"

#include <iostream>

std::vector<int> axis_widths = {6,8,6,8};
std::vector<std::string> axis_labels_fv = {
  "N_{#mu}#nu_{#muQE}",
  "N_{#mu}#nu_{#mu}",
  "N_{#mu}NC",

  "N_{e}#nu_{eQE}",
  "N_{e}#nu_{e}",
  "N_{e}#nu_{#mu}",
  "N_{e}NC",

  "F_{#mu}#nu_{#muQE}",
  "F_{#mu}#nu_{#mu}",
  "F_{#mu}NC",

  "F_{e}#nu_{eQE}",
  "F_{e}#nu_{e}",
  "F_{e}#nu_{#mu}",
  "F_{e}NC"
};

std::vector<std::string> axis_labels = {
  "N_{#mu}#nu_{#muQE}",
  "N_{#mu}#nu_{#mu}",
  "N_{#mu}NC",

  "N_{e}#nu_{eQE}",
  "N_{e}#nu_{e}",
  "N_{e}#nu_{#mu}",
  "N_{e}NC",

  "F_{#mu}#nu_{#muQE}/N_{#mu}#nu_{#muQE}",
  "F_{#mu}#nu_{#mu}/N_{#mu}#nu_{#mu}",
  "F_{#mu}NC",

  "F_{e}#nu_{eQE}/N_{#mu}#nu_{#muQE}",
  "F_{e}#nu_{e}/N_{#mu}#nu_{#mu}",
  "F_{e}#nu_{#mu}/N_{e}#nu_{#mu}",
  "F_{e}NC/N_{e}NC"
};

TCanvas* RatioPlot (std::string name,
                    std::vector<TH1D*>    topHistos,
                    std::vector<TString> topOption,
                    std::vector<TH1D*>    bottomHistos,
                    std::vector<TString> bottomOption,
                    double min = 0.3,double max = 1.7,
                    std::vector<double> ls = {0.8,1,1.2})
{
  gStyle->SetTitleStyle(0);
  TCanvas *c = new TCanvas(name.c_str(), "canvas", 1000, 700);
  //define top and bottom pads
  TPad* p1 = new TPad("p1", "p1", 0, 0, 1, 1);
  TPad* p2 = new TPad("p2", "p2", 0, 0, 1, 1);
  p1->SetNumber(1);
  p2->SetNumber(2);
  p1->SetBottomMargin(.55);
  p2->SetTopMargin(.45);
  p2->SetBottomMargin(.35);

  for(auto p:{p1,p2}){
    p->SetFillStyle(0);
    p->Draw();
  }

  // format histograms
  auto h1 = (TH1D*) topHistos[0]->Clone();
  auto h3 = (TH1D*) bottomHistos[0]->Clone();

  for(auto & h:{h1,h3}){
    h->SetStats(0);

    h->GetYaxis()->SetTitleSize(26);
    h->GetYaxis()->SetTitleFont(43);
    h->GetYaxis()->SetTitleOffset(1.3);
    h->GetYaxis()->SetLabelFont(43);
    h->GetYaxis()->CenterTitle();

    h->GetXaxis()->SetTitleOffset(1.2);
    h->GetXaxis()->SetTitleSize(28);
    h->GetXaxis()->SetTitleFont(43);
    h->GetXaxis()->SetLabelFont(43);
    h->GetXaxis()->CenterTitle();
  }

  h1->GetYaxis()->SetLabelSize(18);

  h1->GetXaxis()->SetLabelSize(0);
  h1->GetXaxis()->SetTitleSize(0);

  h3->SetTitle("");
  h3->GetYaxis()->SetTitle("Ratio");
  h3->GetYaxis()->SetRangeUser(min,max);
  h3->GetYaxis()->SetLabelSize(15);
  h3->GetYaxis()->SetDecimals();
  h3->GetYaxis()->CenterTitle();

  h3->GetXaxis()->SetTitle(h1->GetXaxis()->GetTitle());
  h3->GetXaxis()->SetLabelSize(0);

  p1->cd();

  h1->Draw(topOption[0]);

  for(unsigned int ii = 1; ii < topHistos.size(); ++ii) {
    topHistos[ii]->Draw("same "+topOption[ii]);
  }
  h1->Draw("same "+topOption[0]);

  p1->RedrawAxis(); //white histograms cover axes

  p2->cd();
  h3->Draw(bottomOption[0]);
  for(unsigned int ii = 1; ii < bottomHistos.size(); ++ii){
    bottomHistos[ii]->Draw("same " + bottomOption[ii]);
  }
  //h3->Draw("same " + bottomOption[0]);

  TLine *lone = new TLine();
  lone->SetLineStyle(3);
  lone->SetLineColor(kGray+3);
  p2->Update();
  for(unsigned int i = 0;i < ls.size();++i) lone->DrawLine(p2->GetUxmin(),ls[i],p2->GetUxmax(),ls[i]);
  p2->RedrawAxis();

  return c;
}

void zeroRatio(TH1D* h){
  for(int i = 1;i <= h->GetNbinsX();++i){
    if(h->GetBinContent(i) == 0) h->SetBinContent(i,1);
  }
}

TCanvas* QuickUnivRatioPlot(std::string name,TH1D* hNom,std::vector<TH1D*> others){
  std::vector<TH1D*> h0;
  std::vector<TString> h0opt;
  std::vector<TH1D*> hRat;
  std::vector<TString> hRatopt;

  h0.push_back(hNom);
  h0opt.push_back("hist");
  for(unsigned int i = 0;i < others.size();++i){
    //others[i]->SetLineColor(kGray);
    h0.push_back(others[i]);
    h0opt.push_back("hist");
    TH1D* h = (TH1D*)others[i]->Clone("h");
    h->Divide(hNom);
    zeroRatio(h);
    hRat.push_back(h);
    hRatopt.push_back("hist");
  }

  return RatioPlot(name,h0,h0opt,hRat,hRatopt);
}

void drawCoverLines(TH1D* hist){
  TLine *lone = new TLine();

  std::vector<double> coverage = {0.9,0.95,0.99,0.999,0.9999};
  std::vector<int> coverageBins;
  for(unsigned int i = 0;i < coverage.size();++i) coverageBins.push_back(hist->FindFirstBinAbove(coverage[i]));

  std::cout<<"Coverage & PCs \\\\"<<std::endl;
  for(unsigned int i = 0;i < coverage.size();++i){
    double x = hist->GetBinCenter(coverageBins[i]);
    lone->DrawLine(x,gPad->GetUymin(),x,gPad->GetUymax());
    
    std::cout<<100*coverage[i]<<"\\% & "<<(int)(x+0.5)<<" \\\\"<<std::endl;
  }

}

void drawBinLines(std::vector<int> width){
  TLine *lone = new TLine();
  const int bins = 20;

  double loc = 0;
  for(unsigned int i = 0;i < width.size();++i){
    for(int j = 0;j < width[i];++j){
      loc+=bins;
      if(j == width[i]-1 && i != width.size()-1){
        lone->SetLineWidth(3);
        lone->DrawLine(loc,gPad->GetUymin(),loc,gPad->GetUymax());
        continue;
      }
      lone->SetLineWidth(1);
      lone->DrawLine(loc,gPad->GetUymin(),loc,gPad->GetUymax());
    }
  }
}

void drawLabels(std::vector<std::string> labels)
{
  double left = gPad->GetLeftMargin();
  double right = 1-gPad->GetRightMargin();
  double top = 1-gPad->GetTopMargin();
  double bottom = gPad->GetBottomMargin();

  TLatex *tLabel = new TLatex();
  tLabel->SetTextSize(0.025);
  tLabel->SetTextAlign(12);
  tLabel->SetTextSize(1./30);
  tLabel->SetNDC();
  tLabel->SetTextAngle(-90);

  double center=0;
  for(unsigned int i = 0;i < labels.size();++i){
    center = left+(right-left)*(1+2*i)/(labels.size()*2);
    tLabel->DrawLatex(center, 0.95*bottom, labels[i].c_str());
  }
}

TCanvas* SplitPlot (std::string name,
                    std::vector<TH1D*> topHistos,
                    std::vector<TH1D*> bottomHistos)
{
  gStyle->SetTitleStyle(0);
  TCanvas *c = new TCanvas(name.c_str(), "canvas", 1000, 700);
  //define top and bottom pads
  TPad* p1 = new TPad("p1", "p1", 0, 0, 1, 1);
  TPad* p2 = new TPad("p2", "p2", 0, 0, 1, 1);
  p1->SetNumber(1);
  p2->SetNumber(2);
  p1->SetBottomMargin(.5);
  p2->SetTopMargin(.5);

  for(auto p:{p1,p2}){
    p->SetFillStyle(0);
    p->Draw();
  }

  // format histograms
  auto h1 = (TH1D*) topHistos[0]->Clone();
  auto h3 = (TH1D*) bottomHistos[0]->Clone();

  for(auto & h:{h1,h3}){
    h->SetStats(0);

    h->GetYaxis()->SetTitleSize(26);
    h->GetYaxis()->SetTitleFont(43);
    h->GetYaxis()->SetTitleOffset(1.3);
    h->GetYaxis()->SetLabelFont(43);
    h->GetYaxis()->CenterTitle();
    h->GetYaxis()->SetRangeUser(0.75,1.25);

    h->GetXaxis()->SetTitleOffset(1.2);
    h->GetXaxis()->SetTitleSize(28);
    h->GetXaxis()->SetTitleFont(43);
    h->GetXaxis()->SetLabelFont(43);
    h->GetXaxis()->CenterTitle();
  }

  h1->GetYaxis()->SetTitle("Shift");
  h1->GetYaxis()->SetLabelSize(18);

  h1->GetXaxis()->SetLabelSize(0);
  h1->GetXaxis()->SetTitleSize(0);

  h3->SetTitle("");
  h3->GetYaxis()->SetTitle("Shift");
  h3->GetYaxis()->SetLabelSize(15);
  h3->GetYaxis()->SetDecimals();
  h3->GetYaxis()->CenterTitle();

  h3->GetXaxis()->SetTitle(h1->GetXaxis()->GetTitle());
  h3->GetXaxis()->SetLabelSize(0);

  p1->cd();

  h1->Draw("hist");

  for(unsigned int ii = 1; ii < topHistos.size(); ++ii) {
    topHistos[ii]->Draw("hist same");
  }
  h1->Draw("hist same");

  p1->RedrawAxis(); //white histograms cover axes

  p2->cd();
  h3->Draw("hist");
  for(unsigned int ii = 1; ii < bottomHistos.size(); ++ii){
    bottomHistos[ii]->Draw("hist same");
  }
  h3->Draw("hist same");

  TLine *lone = new TLine();
  lone->SetLineStyle(3);
  lone->SetLineColor(kGray+3);
  p2->Update();
  //for(unsigned int i = 0;i < ls.size();++i) lone->DrawLine(p2->GetUxmin(),ls[i],p2->GetUxmax(),ls[i]);
  p2->RedrawAxis();

  return c;
}