draw_plots_util.h File Reference

#include "CAFAna/Analysis/Exposures.h"
#include "CAFAna/Analysis/Plots.h"
#include "CAFAna/Analysis/Calcs.h"
#include "3FlavorAna/Plotting/NuePlotStyle.h"
#include "CAFAna/Analysis/Style.h"
#include "CAFAna/Core/Binning.h"
#include "CAFAna/Core/HistAxis.h"
#include "CAFAna/Core/LoadFromFile.h"
#include "CAFAna/Core/Utilities.h"
#include "CAFAna/Prediction/IPrediction.h"
#include "CAFAna/Prediction/PredictionInterp.h"
#include "CAFAna/Prediction/PredictionExtrap.h"
#include "CAFAna/Prediction/PredictionNoExtrap.h"
#include "3FlavorAna/Prediction/PredictionExtendToPeripheral.h"
#include "3FlavorAna/Prediction/PredictionSystJoint2018.h"
#include "CAFAna/Vars/FitVars.h"
#include "CAFAna/Decomp/IDecomp.h"
#include "CAFAna/Extrap/IExtrap.h"
#include "CAFAna/Extrap/ModularExtrap.h"
#include "CAFAna/Extrap/ModularExtrapComponent.h"
#include "OscLib/OscCalcDumb.h"
#include "OscLib/IOscCalc.h"
#include "TStyle.h"
#include "TFile.h"
#include "TH1.h"
#include "TCanvas.h"
#include "TGraph.h"
#include "TGraphAsymmErrors.h"
#include "TGaxis.h"
#include "TLatex.h"
#include "TPaveText.h"
#include "TLegend.h"
#include "TLine.h"
#include "TSystem.h"
#include <iostream>
#include <fstream>
#include <iomanip>
#include "tex_tables_util.h"

Go to the source code of this file.

Classes
struct	ana::DataMCComponents

Namespaces
	ana
	Cuts and Vars for the 2020 FD DiF Study.

Enumerations
enum	AxisType {   kNue1bin, kNue3bin, kNue4bin, kNumuOpt,   kNumuOther, kNue1bin, kNue3bin, kNumuOpt,   kNumuOther, kNue1bin, kNue3bin, kNumuOpt,   kNumuOther, kNue1bin, kNue3bin, kNueNDbin,   kNumuOpt, kNumuOther }
enum	ana::EModExtrapComps {   ana::kEEextrap, ana::kEEAntiextrap, ana::kMMextrap, ana::kMMAntiextrap,   ana::kMEextrap, ana::kMEAntiextrap, ana::kEMextrap, ana::kEMAntiextrap,   ana::kNCTotalextrap, ana::kNCextrap, ana::kNCAntiextrap, ana::kMTextrap,   ana::kMTAntiextrap, ana::kETextrap, ana::kETAntiextrap, ana::kEEextrap,   ana::kEEAntiextrap, ana::kMMextrap, ana::kMMAntiextrap, ana::kMEextrap,   ana::kMEAntiextrap, ana::kEMextrap, ana::kEMAntiextrap, ana::kNCTotalextrap,   ana::kNCextrap, ana::kNCAntiextrap, ana::kMTextrap, ana::kMTAntiextrap,   ana::kETextrap, ana::kETAntiextrap, ana::kEEextrap, ana::kEEAntiextrap,   ana::kMMextrap, ana::kMMAntiextrap, ana::kMEextrap, ana::kMEAntiextrap,   ana::kEMextrap, ana::kEMAntiextrap, ana::kNCTotalextrap, ana::kNCextrap,   ana::kNCAntiextrap, ana::kMTextrap, ana::kMTAntiextrap, ana::kETextrap,   ana::kETAntiextrap }

Functions
osc::IOscCalc *	GetCalculator (bool usingdumb)
TH1 *	PrettyRatio (TH1 *shi, TH1 *nom, int color, double alpha=1., TString titley="Shift / Nom")
double	FindLimitY (std::vector< TH1 * > histos, bool doMax=true)
void	FormatErrorBand (TH1 *hplus, TH1 *hminus, bool signal=false, bool fixbins=false)
TH1 *	HistNumuOpt (TH1 *orig)
void	PrettyTag (TString pid, int color, double xndc, double yndc)
void	PIDTag (TString pid)
void	HTag (TString hie)
TLegend *	CustomLegend (std::vector< TH1 * >h, std::vector< TString > titles, std::vector< TString > option, double x1=kLegendDefaultX1, double y1=kLegendDefaultY1, double x2=kLegendDefaultX2, double y2=kLegendDefaultY2)
void	FixLegend (TLegend *leg, TString opt="default")
TGraph *	DummyGraph (int color=kBlack, int style=1, int linewidth=2, int markerstyle=kFullCircle)
TLegend *	DefaultNueLegend (bool sig=true, bool tot=true, bool data=false, double x1=kLegendDefaultX1, double y1=kLegendDefaultY1, double x2=kLegendDefaultX2, double y2=kLegendDefaultY2)
TLegend *	DefaultNumuLegend (bool sig=true, bool tot=true, bool data=false, double x1=kLegendDefaultX1, double y1=kLegendDefaultY1, double x2=kLegendDefaultX2, double y2=kLegendDefaultY2)
void	PrintCanvasAll (TCanvas *c, const TString outname, const TString outdir="plots/", const std::vector< TString > exts={"pdf","png","root"})
TH1 *	RebinToShortAxis (TH1 *h)
void	NuePID3bin (TH1 *h, TString pidtag, bool binlabels=true)
TCanvas *	ExPIDPlot (std::vector< TH1 * > topHistos, std::vector< TString > topOption)
TCanvas *	RatioPlot (std::vector< TH1 * > topHistos, std::vector< TString > topOption, std::vector< TH1 * > bottomHistos, std::vector< TString > bottomOption, TString pidtag, AxisType pidaxis=kNue1bin)
void	PlotDataMC (std::vector< TH1 * >vnom, std::vector< bool >isdata, TLegend *leg, TString pidtag="", TString htag="", TString out_name="plot_FD", bool ratioplot=false, bool ratioerrors=false, AxisType pidaxis=kNue1bin)
void	PlotNDDataTotalMCComparison (TH1 *hdata, std::vector< TH1 * > htots, TLegend *leg, TString pidtag="", TString out_name="plot_nd", bool ratioplot=false, AxisType pidaxis=kNue1bin)
void	PlotMCComponentsComparison (std::vector< TH1 * >vnom, std::vector< TH1 * >vshi, TLegend *leg, TString pidtag="", TString out_name="plot_nd", bool ratioplot=false, bool ratioerrors=false, AxisType pidaxis=kNue1bin)
void	PlotMCComponentsErrorBand (std::vector< TH1 * >vnom, std::vector< TH1 * >vshi, TLegend *leg, TString pidtag="", TString out_name="plot_nd", bool ratioplot=false, bool ratioerrors=false, AxisType pidaxis=kNue1bin)
void	ana::DefaultFormatNue (DataMCComponents &comp, int linestyle=1)
const DataMCComponents	ana::GetNDComponents (const IDecomp *decomp)
struct DataMCComponets	ana::GetFDMCComponents (osc::IOscCalc *calc, IPrediction *pred_no, TString output_name="nue", int linestyle=1, bool bkgdetails=false)
void	ana::CompareNDDataOneMC (DataMCComponents h_no, DataMCComponents h_sh, TString plottitle, TString out_name, TString pidtag, AxisType pidaxis, bool printtable=true)
void	ana::CompareNDDataTwoMC (DataMCComponents hnom, DataMCComponents hplu, DataMCComponents hmin, TString plottitle, TString out_name, TString pidtag, AxisType pidaxis, bool printtable=true)
const IDecomp *	ana::GetDecomp (IPrediction *prediction, EModExtrapComps modExtrapComp)
void	ana::CompareNDDataMCFromVector (PredictionSystJoint2018 *predictionSyst, const ISyst *syst, EModExtrapComps modExtrapComp, TString plottitle, TString out_name, TString tag, AxisType pidaxis, bool printtable=true)
void	ana::CompareOneShiftPred (IPrediction *pred_no, IPrediction *pred_sh, TString plottitle, TString out_name, TString pidtag, bool printtable=true, AxisType pidaxis=kNue1bin, bool PrintParams=false)
void	ana::CompareTwoShiftPred (IPrediction *pred_no, IPrediction *pred_pl, IPrediction *pred_mi, TString plottitle, TString out_name, TString pidtag, bool printtable=true, AxisType pidaxis=kNue1bin, bool PrintParams=false)
void	ana::ComparePredictionsFromVector (PredictionSystJoint2018 *predictionSyst, const ISyst *syst, TString plottitle, TString out_name, bool printtable=true, AxisType pidaxis=kNue3bin)

Variables
static bool	isFHC =true
const bool	printratioplots =true
bool	usedumb = false
bool	mergePeripheral =false
const double	kLegendDefaultX1 = 0.63
const double	kLegendDefaultX2 = 0.85
const double	kLegendDefaultY1 = 0.68
const double	kLegendDefaultY2 = 0.88

Enumeration Type Documentation

enum AxisType

Enumerator
kNue1bin
kNue3bin
kNue4bin
kNumuOpt
kNumuOther
kNue1bin
kNue3bin
kNumuOpt
kNumuOther
kNue1bin
kNue3bin
kNumuOpt
kNumuOther
kNue1bin
kNue3bin
kNueNDbin
kNumuOpt
kNumuOther

Definition at line 50 of file draw_plots_util.h.

{kNue1bin, kNue3bin, kNumuOpt, kNumuOther};

Function Documentation

TLegend* CustomLegend	(	std::vector< TH1 * >	h,
		std::vector< TString >	titles,
		std::vector< TString >	option,
		double	x1 = kLegendDefaultX1,
		double	y1 = kLegendDefaultY1,
		double	x2 = kLegendDefaultX2,
		double	y2 = kLegendDefaultY2
	)

Definition at line 170 of file draw_plots_util.h.

References MECModelEnuComparisons::leg, registry_explorer::v, x1, submit_syst::x2, y1, and submit_syst::y2.

{
  TLegend* leg = new TLegend(x1, y1, x2, y2);
  leg->SetFillStyle(0);
  for (unsigned int v = 0; v < h.size(); ++v)
    leg->AddEntry(h[v], titles[v], option[v]);
  leg->SetBit(kCanDelete);
  return leg;
}

TLegend* DefaultNueLegend	(	bool	sig = true,
		bool	tot = true,
		bool	data = false,
		double	x1 = kLegendDefaultX1,
		double	y1 = kLegendDefaultY1,
		double	x2 = kLegendDefaultX2,
		double	y2 = kLegendDefaultY2
	)

Definition at line 221 of file draw_plots_util.h.

References DummyGraph(), ana::kBeamNueBackgroundColor, ana::kNCBackgroundColor, ana::kNueSignalColor, ana::kNumuBackgroundColor, ana::kTotalMCColor, MECModelEnuComparisons::leg, PandAna.Demos.pi0_spectra::tot, x1, submit_syst::x2, y1, and submit_syst::y2.

{
  TLegend* leg = new TLegend(x1,y1,x2,y2);
  leg->SetFillStyle(0);
  if(sig)  leg->AddEntry( DummyGraph(kNueSignalColor), "#nu_{e} Signal","l");
  if(data) leg->AddEntry( DummyGraph(),                "ND Data","epl");
  if(tot)  leg->AddEntry( DummyGraph(kTotalMCColor),   "Total Bkg","l");
  leg->AddEntry( DummyGraph(kBeamNueBackgroundColor),  "Beam #nu_{e}","l");
  leg->AddEntry( DummyGraph(kNCBackgroundColor),       "NC","l");
  leg->AddEntry( DummyGraph(kNumuBackgroundColor),     "#nu_{#mu} CC","l");
  leg->SetBit(kCanDelete);
  return leg;

}

TLegend* DefaultNumuLegend	(	bool	sig = true,
		bool	tot = true,
		bool	data = false,
		double	x1 = kLegendDefaultX1,
		double	y1 = kLegendDefaultY1,
		double	x2 = kLegendDefaultX2,
		double	y2 = kLegendDefaultY2
	)

Definition at line 240 of file draw_plots_util.h.

References DummyGraph(), isNumuAna, ana::kBeamNueBackgroundColor, ana::kNCBackgroundColor, ana::kNueSignalColor, ana::kNumuBackgroundColor, ana::kNuTauBackgroundColor, ana::kTotalMCColor, MECModelEnuComparisons::leg, PandAna.Demos.pi0_spectra::tot, x1, submit_syst::x2, y1, and submit_syst::y2.

{
  TLegend* leg = new TLegend(x1,y1,x2,y2);
  leg->SetFillStyle(0);
  if(sig)  leg->AddEntry( DummyGraph(kNueSignalColor), "#nu_{#mu} Signal","l");
  if(data) leg->AddEntry( DummyGraph(),                "ND Data","epl");
  if(tot)  leg->AddEntry( DummyGraph(kTotalMCColor),   "Total Bkg","l");
  if(isNumuAna) leg->AddEntry( DummyGraph(kNumuBackgroundColor),     "#nu_{#mu} app","l");
  else          leg->AddEntry( DummyGraph(kNumuBackgroundColor),     "#nu_{#mu} CC" ,"l");
  leg->AddEntry( DummyGraph(kNCBackgroundColor),       "NC","l");
  leg->AddEntry( DummyGraph(kBeamNueBackgroundColor),  "#nu_{e} CC","l");
  leg->AddEntry( DummyGraph(kNuTauBackgroundColor),    "#nu_{#tau} CC","l");
  leg->SetBit(kCanDelete);
  return leg;

}

TGraph* DummyGraph	(	int	color = kBlack,
		int	style = 1,
		int	linewidth = 2,
		int	markerstyle = kFullCircle
	)

Definition at line 209 of file draw_plots_util.h.

References plot_validation_datamc_2018::color.

{
  auto gr = new TGraph();
  gr->SetLineColor(color);
  gr->SetLineStyle(style);
  gr->SetLineWidth(linewidth);
  gr->SetMarkerStyle(markerstyle);
  gr->SetMarkerColor(color);
  return gr;
}

TCanvas* ExPIDPlot	(	std::vector< TH1 * >	topHistos,
		std::vector< TString >	topOption
	)

Definition at line 311 of file draw_plots_util.h.

References plot_validation_datamc::Clone(), Ebins, FindLimitY(), make_syst_table_plots::h, he, calib::j, and pidBins.

{
  TCanvas *cpid = new TCanvas("cpid", "cpid", 410, 850);
  float Ebins = 10;
  int pidBins = 3;

  TPad* p[pidBins];
  p[0] = new TPad("", "", 0, 0, 1, 1);
  p[1] = new TPad("", "", 0, 0, 1, 1);
  p[2] = new TPad("", "", 0, 0, 1, 1);
  p[0]->SetTopMargin(0.1);
  p[0]->SetBottomMargin(.634);
  p[1]->SetTopMargin(.369);
  p[1]->SetBottomMargin(.369);
  p[2]->SetTopMargin(.634);
  p[2]->SetBottomMargin(0.1);
  p[0]->SetFillStyle(0);
  p[1]->SetFillStyle(0);
  p[2]->SetFillStyle(0);
  p[0]->Draw();
  p[1]->Draw();
  p[2]->Draw();

  TH1* he[pidBins];
  TH1* h[topHistos.size()][pidBins];
  for(int k=pidBins-1; k>=0; --k){
    he[k] = (TH1*) topHistos[0]->Clone();
    he[k]->GetYaxis()->SetRangeUser(FindLimitY(topHistos,0) ,1.2*FindLimitY(topHistos,1));
    he[k]->GetXaxis()->SetRangeUser(Ebins*k,Ebins*(k+1));
    if ( k>0 ) he[k]->GetXaxis()->SetLabelSize(0);
    p[pidBins-1-k]->cd();
    he[k]->Draw(topOption[0]);
    he[k]->GetYaxis()->SetTitleSize(24);
    he[k]->GetYaxis()->SetTitleFont(43);
    he[k]->GetYaxis()->SetTitleOffset(1.25);
    he[k]->GetYaxis()->SetLabelFont(43);
    he[k]->GetYaxis()->SetLabelSize(14);
    he[k]->GetXaxis()->SetLabelOffset(-0.008);
    he[k]->GetXaxis()->SetTitleOffset(0.5);
    p[pidBins-1-k]->SetGridy();
  }
  for(unsigned int ii=0;ii<topHistos.size();ii++){
    for(int j=pidBins-1; j>=0; --j){
      h[ii][j] = (TH1*) topHistos[ii]->Clone();
      h[ii][j]->GetXaxis()->SetRangeUser(Ebins*j,Ebins*(j+1));
      p[pidBins-1-j]->cd();
      h[ii][j]->Draw("same "+topOption[ii]);
    }
  }
  for(int h=pidBins-1; h>=0; --h){
    he[h]->SetStats(0);
    p[pidBins-1-h]->cd();
    he[h]->Draw("same "+topOption[0]);
  }
  p[0]->RedrawAxis(); //needed for white histograms e.g. beam
  p[1]->RedrawAxis();
  p[2]->RedrawAxis();
  return cpid;
}

double FindLimitY	(	std::vector< TH1 * >	histos,
		bool	doMax = true
	)

Definition at line 93 of file draw_plots_util.h.

References make_mec_shifts_plots::GetMaximum(), MECModelEnuComparisons::i, cet::sqlite::max(), and min().

{
  double max = histos[0]->GetMaximum();
  double min = histos[0]->GetMinimum();

  for(unsigned int i=0; i<histos.size(); i++){
    if ( histos[i]->GetMaximum() > max ) max = histos[i]->GetMaximum();
    if ( histos[i]->GetMinimum() < min ) min = histos[i]->GetMinimum();
  }

  if ( doMax ) return max;
  else         return min;
}

void FixLegend	(	TLegend *	leg,
		TString	opt = "default"
	)

Definition at line 185 of file draw_plots_util.h.

References kLegendDefaultX1, kLegendDefaultX2, kLegendDefaultY1, kLegendDefaultY2, and MECModelEnuComparisons::opt.

{
  leg->SetFillStyle(0);
  if(opt=="default")
  {
    leg->SetX1(kLegendDefaultX1); leg->SetX2(kLegendDefaultX2);
    leg->SetY1(kLegendDefaultY1); leg->SetY2(kLegendDefaultY2);
  }
  if(opt=="ExPID")
  {
    leg->SetX1(0.405); leg->SetX2(0.565);
    leg->SetY1(0.63); leg->SetY2(0.82);
    leg->SetBorderSize(0);
    leg->SetTextSize(0.033);
  }
  if(opt=="ExPIDPads")
  {
    leg->SetX1(0.55); leg->SetX2(0.89);
    leg->SetY1(0.75); leg->SetY2(0.89);
    leg->SetTextSize(0.045);
  }
}

void FormatErrorBand	(	TH1 *	hplus,
		TH1 *	hminus,
		bool	signal = false,
		bool	fixbins = false
	)

Definition at line 108 of file draw_plots_util.h.

References om::cout, allTimeWatchdog::endl, MECModelEnuComparisons::i, kBlue, kRed, ana::kTotalMCErrorBandColor, kViolet, cet::sqlite::max(), and min().

{
  int mccolor = signal ? kViolet -9 : kTotalMCErrorBandColor;
  std::cout << "hplus: " << hplus->Integral() << std::endl;
  std::cout << "hminus: " << hminus->Integral() << std::endl;
  hplus ->SetLineColor(mccolor);
  hminus->SetLineColor(mccolor);
  hplus ->SetFillColor(mccolor);
  hminus->SetFillColor(10);
  hplus ->SetMarkerColor(kRed-7);
  hminus->SetMarkerColor(kBlue-7);

  if (fixbins)
  {
    for(int i=1; i <= hplus->GetNbinsX(); i++)
    {
      double tplus = hplus->GetBinContent(i);
      double tminus = hminus->GetBinContent(i);
      hplus->SetBinContent(i,max(tplus, tminus));
      hminus->SetBinContent(i,min(tplus,tminus));
    }
  }
}

osc::IOscCalc* GetCalculator

(

bool

usingdumb

)

Definition at line 55 of file draw_plots_util.h.

References calc, osc::OscCalcDumb::Copy(), osc::_IOscCalc< T >::Copy(), ana::DefaultOscCalc(), dmsq32, ana::kFitDeltaInPiUnits, ana::kFitDmSq32, ana::kFitSinSqTheta23, ana::FitVarWithPrior::SetValue(), ana::ConstrainedFitVarWithPrior::SetValue(), ana::FitDmSq32::SetValue(), and ssth23.

                                            {
  double dCP_Pi = 1.48;
  double ssth23 = 0.558;
  double dmsq32 = 2.44e-3;

  osc::IOscCalc * calc;
  if(usingdumb){
    osc::OscCalcDumb dumb;
    calc = dumb.Copy();
  }
  else{
    auto calc2 = DefaultOscCalc();
    kFitDeltaInPiUnits.SetValue(calc2, dCP_Pi);
    kFitSinSqTheta23.SetValue(calc2, ssth23);
    kFitDmSq32.SetValue(calc2, dmsq32);
    calc = calc2->Copy();
  }
  return calc;
}

TH1* HistNumuOpt

(

TH1 *

orig

)

Definition at line 134 of file draw_plots_util.h.

References runNovaSAM::ret, and ana::UniqueName().

                            {
  auto ret = (TH1*) orig->Clone(UniqueName().c_str());
  ret->Scale(0.1,"width");
  ret->GetYaxis()->SetTitle("Events / 0.1 GeV");
  ret->SetBit(kCanDelete);
  ret->SetDirectory(gROOT);
  return ret;
}

void HTag

(

TString

hie

)

Definition at line 159 of file draw_plots_util.h.

References kOrange, and PrettyTag().

{ PrettyTag (hie, (hie=="NH")?kAzure+3:kOrange+2, .18, .825);}

void NuePID3bin	(	TH1 *	h,
		TString	pidtag,
		bool	binlabels = true
	)

Definition at line 304 of file draw_plots_util.h.

References ana::Nue2018ThreeBinDivisions(), and ana::Nue2018ThreeBinLabels().

{
  Nue2018ThreeBinDivisions();
  if (binlabels) Nue2018ThreeBinLabels();
  //if (h) PID2DAxis(h);
}

void PIDTag

(

TString

pid

)

Definition at line 158 of file draw_plots_util.h.

References PrettyTag().

{ PrettyTag (pid, kAzure+3, .83, .92); }

void PlotDataMC	(	std::vector< TH1 * >	vnom,
		std::vector< bool >	isdata,
		TLegend *	leg,
		TString	pidtag = "",
		TString	htag = "",
		TString	out_name = "plot_FD",
		bool	ratioplot = false,
		bool	ratioerrors = false,
		AxisType	pidaxis = kNue1bin
	)

Definition at line 466 of file draw_plots_util.h.

References bin, demo5::c2, FixLegend(), genie::utils::style::Format(), ana::GraphWithPoissonErrors(), HTag(), kNue3bin, ana::Nue2018ThreeBinDivisions(), ana::Nue2018ThreeBinLabels(), submit_concat_project::out_name, PrettyRatio(), PrettyTag(), PrintCanvasAll(), ana::PrintLatexFigure(), RatioPlot(), and registry_explorer::v.

{
  gStyle->SetPadLeftMargin(0.15);
  gStyle->SetPadTopMargin(0.08);

  // Fill empty bins with 0 if in range and <0 if out of range
  for ( int bin = 1; bin<=vnom[0]->GetNbinsX(); ++bin){
    bool filledbin=false;
    for ( unsigned int v = 0; v<vnom.size(); ++v){
      if ( !isdata[v] )
  if (vnom[v]->GetBinContent(bin) > 0 ) filledbin=true;
    }
    for ( uint v = 0; v<vnom.size(); ++v){
      if (vnom[v]->GetBinContent(bin) == 0 ){
  if ( !filledbin ) vnom[v]->SetBinContent(bin,-0.01);
  else vnom[v]->SetBinContent(bin,0.001);
      }
      vnom[v]->GetYaxis()->SetTitleOffset(1.05);
    }
  }

  // Draw marker w/error or line when appropriate
  int datahist=-1;
  for ( unsigned int v = 0; v<vnom.size(); ++v)
    if (vnom[v]->GetLineColor() == kBlack ) datahist=v;

  // Option to make a ratio plot given
  if(!ratioplot){
    auto c2 = new TCanvas ("c2","c2",715,500);
    for ( unsigned int v = 0; v<vnom.size(); ++v){
      TGraphAsymmErrors* gr = GraphWithPoissonErrors(vnom[v], 0, 1);
      if ( isdata[v] ){
  if (v == datahist ) gr->Draw("ep same");
  else vnom[v]->Draw("p same");
      }
      else             vnom[v]->Draw("hist same");
    }
    gPad->RedrawAxis();

    // Adjust for ExPID axis
    if(pidaxis==kNue3bin)
    {
      FixLegend(leg);
      Nue2018ThreeBinDivisions();
      Nue2018ThreeBinLabels();
    }

    // pot label
    PrettyTag(TString::Format("%g#times10^{20} POT equiv.", 8.85).Data(),
        kBlack, .175, .74);
    HTag(htag);
    leg->Draw("same");

    gPad->SetFillStyle(0);
    c2->SetFillStyle(0);
    PrintCanvasAll (c2,out_name+"_datamc","plots/",{"pdf","eps","png","C","root"});
    PrintLatexFigure(out_name+"_datamc");
    //removed: prelim tag
  }
  else{
    std::vector<TH1*> vratio;
    for(unsigned int ii=1;ii<vnom.size();ii++){
      auto htemp = PrettyRatio (vnom[datahist], vnom[ii], vnom[ii]->GetLineColor(), 1.,
        "Data / MC");
      htemp->GetYaxis()->SetRangeUser(0.5,1.5);
      vratio.push_back(htemp);
    }
    std::vector<TString> topOption  = {" ","hist"};
    std::vector<TString> bottomOption (vratio.size(),ratioerrors ?"p":"hist p");
    auto cratio2= RatioPlot(vnom, topOption,vratio,bottomOption,pidtag,pidaxis);
    cratio2->cd();
    leg->Draw("same");
    HTag(pidtag);
    PrintCanvasAll(cratio2,out_name+"_datamc_ratio");
    PrintLatexFigure("/png_"+out_name+"_datamc_ratio");
  }
}

void PlotMCComponentsComparison	(	std::vector< TH1 * >	vnom,
		std::vector< TH1 * >	vshi,
		TLegend *	leg,
		TString	pidtag = "",
		TString	out_name = "plot_nd",
		bool	ratioplot = false,
		bool	ratioerrors = false,
		AxisType	pidaxis = kNue1bin
	)

Definition at line 594 of file draw_plots_util.h.

References demo5::c2, FixLegend(), kNue3bin, NuePID3bin(), submit_concat_project::out_name, PIDTag(), PrettyRatio(), PrintCanvasAll(), ana::PrintLatexFigure(), RatioPlot(), ana::UniqueName(), and registry_explorer::v.

{
  if(pidaxis==kNue3bin) FixLegend(leg);

  if(!ratioplot)
  {
    auto c2 = new TCanvas (UniqueName().c_str(), "c2");
    for(auto const & v:vnom) v->Draw("hist same");
    for(auto const & v:vshi) v->Draw("hist same");
    if(pidaxis==kNue3bin) NuePID3bin (vnom[0],pidtag,true);
    PIDTag(pidtag);
    leg->Draw("same");
    PrintCanvasAll(c2, out_name+"_mccomp");
    PrintLatexFigure("/png_"+out_name+"_mccomp");
    c2->Close();
  }
  else
  {
    std::vector<TH1*> vratio;
    for(unsigned int ii=0;ii<vshi.size();ii++){
      auto htemp = PrettyRatio(vshi[ii], vnom[ii],
                               vnom[ii]->GetLineColor(), ii>0 ? 0.4:1.);
      htemp->GetYaxis()->SetRangeUser(0.5, 1.5);
      vratio.push_back (htemp);
    }
    vnom.insert(vnom.end(),vshi.begin(),vshi.end());
    std::vector<TString> topOption(vnom.size(), "hist");
    std::vector<TString> bottomOption(vratio.size(),
                                      ratioerrors ? "p":"hist p");
    auto cratio2= RatioPlot(vnom, topOption, vratio,
                            bottomOption, pidtag, pidaxis);
    cratio2->cd();
    leg->Draw("same");
    PIDTag(pidtag);
    PrintCanvasAll(cratio2, out_name+"_mccomp_ratio");
    PrintLatexFigure("/png_"+out_name+"_mccomp_ratio");
  }
  //if(pidaxis==kNue3bin) {
  //  vnom.insert(vnom.end(),vshi.begin(),vshi.end());
  //  std::vector<TString> topOption    (vnom.size(),"hist");
  //  auto cpads =ExPIDPlot(vnom,topOption);
  //  cpads->cd();
  //  FixLegend(leg,"default");
  //  leg->Draw("same");
  //  PIDTag(pidtag);
  //  PrintCanvasAll(cpads, out_name+"_mccomp_pads");
  //      PrintLatexFigure("/png_"+out_name+"_mccomp_pads");
  //}
}

void PlotMCComponentsErrorBand	(	std::vector< TH1 * >	vnom,
		std::vector< TH1 * >	vshi,
		TLegend *	leg,
		TString	pidtag = "",
		TString	out_name = "plot_nd",
		bool	ratioplot = false,
		bool	ratioerrors = false,
		AxisType	pidaxis = kNue1bin
	)

Definition at line 650 of file draw_plots_util.h.

References ana::DataMCComponents::beam, ana::DataMCComponents::bkg, demo5::c2, ana::CenterTitles(), plot_validation_datamc::Clone(), ana::DataMCComponents::data, ana::DefaultFormatNue(), FixLegend(), FormatErrorBand(), make_mec_shifts_plots::GetMaximum(), ana::kBeamNueBackgroundColor, ana::kNCBackgroundColor, kNue3bin, ana::kNueSignalColor, ana::kNueSignalWSColor, ana::kNumuBackgroundColor, ana::kTotalMCColor, ana::DataMCComponents::nc, NuePID3bin(), ana::DataMCComponents::nuetonumu, ana::DataMCComponents::nuetonutau, ana::DataMCComponents::numu, ana::DataMCComponents::numutonumu, ana::DataMCComponents::numutonutau, submit_concat_project::out_name, PIDTag(), PrettyRatio(), PrintCanvasAll(), ana::PrintLatexFigure(), RatioPlot(), ana::DataMCComponents::sig, ana::DataMCComponents::tau, registry_explorer::v, and ana::DataMCComponents::wrongSign.

{
  auto hplu_plot = (TH1*) vshi[0]->Clone();
  auto hmin_plot = (TH1*) vshi[1]->Clone();
  FormatErrorBand(hplu_plot,hmin_plot,out_name.Contains("_sig"),true);
  std::vector<TH1*>vshi_plot={hplu_plot,hmin_plot};

  if(pidaxis==kNue3bin)
    FixLegend(leg);

  if(!ratioplot){
    auto c2 = new TCanvas ("c2","c2");
    vnom[0]->SetMaximum(1.2*vnom[0]->GetMaximum());
    vnom[0]->Draw("hist");
    for(auto v:vshi_plot) v->Draw("hist same");
    for(auto v:vnom) v->Draw("hist same");
    if(pidaxis==kNue3bin) NuePID3bin (vnom[0],pidtag,true);
    PIDTag(pidtag);
    leg->Draw("same");
    PrintCanvasAll(c2, out_name+"_mccomp");
    PrintLatexFigure("/png_"+out_name+"_mccomp");
    c2->Close();
  }
  else{
    std::vector<TH1*> vratio;
    for(unsigned int ii=0;ii<vshi.size();ii++){
      auto htemp = PrettyRatio(vshi[ii],vnom[0],vshi[ii]->GetMarkerColor());
      vratio.push_back(htemp);
    }
    vratio[0]->GetYaxis()->SetRangeUser(0.5,1.5);
    vnom.insert(vnom.begin()+1,vshi_plot.begin(),vshi_plot.end());
    std::vector<TString> topOption    (vnom.size(),"hist");
    std::vector<TString> bottomOption (vratio.size(),ratioerrors ?"p":"hist p");
    auto cratio2= RatioPlot(vnom, topOption,vratio,bottomOption,pidtag,pidaxis);
    cratio2->cd();
    leg->Draw("same");
    PIDTag(pidtag);
    PrintCanvasAll(cratio2, out_name+"_mccomp_ratio");
    PrintLatexFigure("/png_"+out_name+"_mccomp_ratio");
  }
}

void PlotNDDataTotalMCComparison	(	TH1 *	hdata,
		std::vector< TH1 * >	htots,
		TLegend *	leg,
		TString	pidtag = "",
		TString	out_name = "plot_nd",
		bool	ratioplot = false,
		AxisType	pidaxis = kNue1bin
	)

Definition at line 549 of file draw_plots_util.h.

References demo5::c1, FixLegend(), make_mec_shifts_plots::GetMaximum(), make_syst_table_plots::h, kNue3bin, NuePID3bin(), submit_concat_project::out_name, PIDTag(), PrettyRatio(), PrintCanvasAll(), ana::PrintLatexFigure(), PandAna.reco_validation.prod5_pid_validation::ratio(), and RatioPlot().

{
  if(pidaxis==kNue3bin)
    FixLegend(leg);
  std::vector<TString> h_opt;
  for(auto const & h:htots) h_opt.push_back("hist");
  h_opt.push_back("p");

  if(!ratioplot)
  {
    auto c1 = new TCanvas ("c1","c1");
    htots[0]->SetMaximum(1.3*htots[0]->GetMaximum());
    for(auto const & h:htots) h->Draw("hist same");
    htots[0]->Draw("hist same"); //need to redraw for error bands
    hdata->Draw("p same");
    PIDTag(pidtag);
    if(pidaxis==kNue3bin) NuePID3bin (htots[0],pidtag,true);
    leg->Draw("same");
    PrintCanvasAll(c1, out_name+"_datamc");
    PrintLatexFigure("/png_"+out_name+"_datamc");
  }
  else
  {
    std::vector<TH1*> ratios;
    std::vector<TString> r_opt;
    for(auto const & h:htots){
      auto ratio = PrettyRatio(hdata,h, h->GetLineColor(), 1., "Data / MC");
      ratio->GetYaxis()->SetRangeUser(0.5,1.5);
      ratios.push_back(ratio);
      r_opt.push_back("e1");
    }
    htots.push_back(hdata);
    auto cratio1 = RatioPlot(htots,h_opt,ratios,r_opt,pidtag,pidaxis);
    cratio1->cd();
    leg->Draw("same");
    PIDTag(pidtag);

    PrintCanvasAll(cratio1,out_name+"_datamc_ratio");
    PrintLatexFigure("/png_"+out_name+"_datamc_ratio");
  }
}

TH1* PrettyRatio	(	TH1 *	shi,
		TH1 *	nom,
		int	color,
		double	alpha = 1.,
		TString	titley = "Shift / Nom"
	)

Definition at line 80 of file draw_plots_util.h.

References ana::UniqueName().

                                                                                                  {
  auto htemp = (TH1*) nom->Clone(ana::UniqueName().c_str());
  htemp->Divide(shi,nom);
  htemp->SetMarkerColorAlpha(color,alpha);
  htemp->SetLineColorAlpha(color,alpha);
  htemp->SetMarkerStyle(34);
  htemp->SetMarkerSize(1.4);
  htemp->GetYaxis()->SetTitle(titley);
  return htemp;
}

void PrettyTag	(	TString	pid,
		int	color,
		double	xndc,
		double	yndc
	)

Definition at line 147 of file draw_plots_util.h.

References getGoodRuns4SAM::tag.

                                                                {
  TLatex* tag = new TLatex(xndc, yndc, pid);
  tag->SetTextColor(color);
  tag->SetNDC();
  tag->SetTextSize(0.04);
  tag->SetTextFont(42);
  tag->SetTextAlign(22);
  tag->Draw();
  tag->SetBit(kCanDelete);
}

void PrintCanvasAll	(	TCanvas *	c,
		const TString	outname,
		const TString	outdir = "plots/",
		const std::vector< TString >	exts = {"pdf","png","root"}
	)

Definition at line 260 of file draw_plots_util.h.

References plot_validation_datamc::c, galleryMaker::ext, and outdir.

                                          {"pdf","png","root"} )
{
  for(auto const & ext:exts){
    gSystem -> MakeDirectory( outdir );
    gSystem -> MakeDirectory( outdir+"rootfiles/" );
    if(ext=="png")       c->SaveAs(outdir+"png_"+outname + "." + ext);
    else if(ext=="root") c->SaveAs(outdir+"rootfiles/"+outname + "." + ext);
    else                 c->SaveAs(outdir+outname + "." + ext);
  }
  delete c;
}

TCanvas* RatioPlot	(	std::vector< TH1 * >	topHistos,
		std::vector< TString >	topOption,
		std::vector< TH1 * >	bottomHistos,
		std::vector< TString >	bottomOption,
		TString	pidtag,
		AxisType	pidaxis = kNue1bin
	)

Definition at line 375 of file draw_plots_util.h.

References plot_validation_datamc::c, plot_validation_datamc::Clone(), FindLimitY(), make_syst_table_plots::h, h1, h3, kNue3bin, cet::sqlite::max(), NuePID3bin(), plot_validation_datamc::p1, and plot_validation_datamc::p2.

{
  gStyle->SetTitleStyle(0);
  TCanvas *c = new TCanvas("c", "canvas", 600, 700);
  if(pidaxis == kNue3bin) c->SetCanvasSize(750,650);
  //define top and bottom pads
  TPad* p1 = new TPad("", "", 0, 0, 1, 1);
  TPad* p2 = new TPad("", "", 0, 0, 1, 1);
  p1->SetBottomMargin(.3);
  p2->SetTopMargin(.7);

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

  // format histograms
  auto h1 = (TH1*) topHistos[0]->Clone();
  auto h3 = (TH1*) 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->GetXaxis()->SetTitleOffset(1.2);
    h->GetXaxis()->SetTitleSize(28);
    h->GetXaxis()->SetTitleFont(43);
    h->GetXaxis()->SetLabelFont(43);
  }

  h1->GetYaxis()->SetRangeUser(FindLimitY(topHistos,0), 1.2*FindLimitY(topHistos,1));
  h1->GetYaxis()->SetLabelSize(18);

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

  h3->SetTitle("");
  h3->GetYaxis()->SetRangeUser(max(0.,2-FindLimitY(bottomHistos,1)),
             1.01*FindLimitY(bottomHistos,1));
  h3->GetYaxis()->SetLabelSize(15);
  h3->GetYaxis()->SetDecimals();

  h3->GetXaxis()->SetTitle(h1->GetXaxis()->GetTitle());
  h3->GetXaxis()->SetLabelSize(18);
  h3->GetXaxis()->SetLabelOffset(0.005);

  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

  if(pidaxis==kNue3bin) NuePID3bin(0,pidtag);

  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]);

  auto lone = new TLine();
  lone->SetLineStyle(3);
  lone->SetLineColor(kGray+3);
  p2->Update();
  lone->DrawLine(p2->GetUxmin(),1.0,p2->GetUxmax(),1.0);
  lone->DrawLine(p2->GetUxmin(),1.2,p2->GetUxmax(),1.2);
  lone->DrawLine(p2->GetUxmin(),0.8,p2->GetUxmax(),0.8);
  float p2Y = p2->GetUymax()-p2->GetUymin();
  h3->GetYaxis()->SetRangeUser(p2->GetUymin()-(0.01*p2Y),p2->GetUymax()+(0.01*p2Y));
  p2->RedrawAxis();
  if(pidaxis==kNue3bin) NuePID3bin(h3,pidtag, false);

  return c;
}

TH1* RebinToShortAxis

(

TH1 *

h

)

Definition at line 278 of file draw_plots_util.h.

References it, runNovaSAM::ret, and ana::UniqueName().

{
  TH1* ret = new TH1D(UniqueName().c_str(), h->GetTitle(),
                      18, 0, 18);
  ret->GetXaxis()->SetTitle(h->GetXaxis()->GetTitle());
  ret->GetYaxis()->SetTitle(h->GetYaxis()->GetTitle());

  std::vector<std::pair<int, int>> binMap = {{3, 2},
                                             {4, 3},
                                             {5, 4},
                                             {6, 5}, // end of 1st spectrum
                                             {13, 8},
                                             {14, 9},
                                             {15, 10},
                                             {16, 11}, // end of 2nd
                                             {23, 14},
                                             {24, 15},
                                             {25, 16},
                                             {26, 17}};

  for(auto it: binMap) ret->SetBinContent(it.second, h->GetBinContent(it.first));

  return ret;
}

Variable Documentation

bool isFHC =true

static

Definition at line 45 of file draw_plots_util.h.

Referenced by ana::GetNDComponents().

const double kLegendDefaultX1 = 0.63

Definition at line 165 of file draw_plots_util.h.

const double kLegendDefaultX2 = 0.85

Definition at line 166 of file draw_plots_util.h.

const double kLegendDefaultY1 = 0.68

Definition at line 167 of file draw_plots_util.h.

const double kLegendDefaultY2 = 0.88

Definition at line 168 of file draw_plots_util.h.

bool mergePeripheral =false

Definition at line 53 of file draw_plots_util.h.

const bool printratioplots =true

Definition at line 48 of file draw_plots_util.h.

bool usedumb = false

Definition at line 52 of file draw_plots_util.h.