make_extrap_figure.C File Reference

#include "TCanvas.h"
#include "TColor.h"
#include "TFile.h"
#include "TGraph.h"
#include "TH2.h"
#include "TLatex.h"
#include "TLegend.h"
#include "TLine.h"
#include "TPad.h"
#include "TStyle.h"
#include <iostream>
#include <vector>

Go to the source code of this file.

Functions
void	no_titles (std::vector< TH1 * > hs)
TPad *	pad (std::string name, double x0, double y0, double x1, double y1)
void	latex (double x, double y, std::string txt, double ang=0, int align=22)
TGraph *	rotate_curve (TH1 *h)
TGraph *	rotate_hist (TH1 *h, bool neg=false)
void	make_extrap_figure ()

Variables
const double	kVertSplit = .6
const double	kLRMargin = .05
const double	kTopMargin = .05
const double	kBottomMargin = .1
const double	kMatrixWidth = .15
const double	kXPadding = .0025
const double	kYPadding = .005

Function Documentation

void latex	(	double	x,
		double	y,
		std::string	txt,
		double	ang = 0,
		int	align = 22
	)

Definition at line 50 of file make_extrap_figure.C.

{
  TLatex* ltx = new TLatex(x, y, txt.c_str());
  ltx->SetNDC();
  ltx->SetTextAlign(align);
  ltx->SetTextAngle(ang);
  ltx->Draw();
}

void make_extrap_figure

(

)

Definition at line 85 of file make_extrap_figure.C.

References file_size_ana::axes, b, plot_validation_datamc::c, e, MakeMiniprodValidationCuts::f, MECModelEnuComparisons::g, MECModelEnuComparisons::i, kBlue, kBottomMargin, kLRMargin, kMatrixWidth, kRed, kTopMargin, kVertSplit, latex(), no_titles(), pad(), r(), rotate_curve(), and rotate_hist().

{
  TFile* f = new TFile("extrap_hists.root");

  //  f->GetDirectory("extrap/MMextrap")->ls();

  TH1* h_reco_dat_nd = (TH1*)f->Get("MMextrap/NDDatReco");
  TH1* h_reco_mc_nd = (TH1*)f->Get("MMextrap/NDMCReco");

  TH1* h_true_mc_nd = (TH1*)f->Get("MMextrap/NDMCTrue");
  TH1* h_true_rwt_nd = (TH1*)f->Get("MMextrap/NDRwtTrue");

  TH1* h_true_mc_fd = (TH1*)f->Get("MMextrap/FDMCTrue");
  TH1* h_true_ext_fd = (TH1*)f->Get("MMextrap/FDExtTrue");

  TH1* h_reco_ext_fd = (TH1*)f->Get("MMextrap/FDExtReco");
  TH1* h_reco_mc_fd = (TH1*)f->Get("MMextrap/FDMCReco");

  // This is very fragile. Would be better if SavePlotsNumu() put out a
  // histogram rather than a canvas here.
  TH2* h_nd_matrix = (TH2*)((TCanvas*)f->Get("MMextrap/NDRecoToTrue"))->FindObject("cafanauniq20");
  TH2* h_fd_matrix = (TH2*)((TCanvas*)f->Get("MMextrap/FDTrueToReco"))->FindObject("cafanauniq21");

  TH1* h_osc = (TH1*)f->Get("osc");

  TH1* h_fn = (TH1*)h_true_mc_fd->Clone();
  h_fn->Divide(h_true_mc_nd);

  // Now that we have the F/N ratio it's safe to oscillate the FD plots
  h_true_mc_fd->Multiply(h_osc);
  h_true_ext_fd->Multiply(h_osc);

  h_reco_dat_nd->Scale(1e-3);
  h_reco_mc_nd->Scale(1e-3);

  h_true_mc_nd->Scale(1e-3);
  h_true_rwt_nd->Scale(1e-3);

  no_titles({h_reco_dat_nd, h_reco_mc_nd, h_true_mc_nd, h_true_rwt_nd,
        h_true_mc_fd, h_true_ext_fd, h_reco_ext_fd, h_reco_mc_fd,
        h_nd_matrix, h_fd_matrix, h_osc, h_fn});


  TCanvas* c = new TCanvas("a", "b", 1800, 600);

  TPad* pNDReco = pad("NDReco", kLRMargin, kVertSplit, kLRMargin+kMatrixWidth, 1-kTopMargin);
  TPad* pFDReco = pad("FDReco", 1-kLRMargin-kMatrixWidth, kVertSplit, 1-kLRMargin, 1-kTopMargin);

  TPad* pNDMatrix = pad("NDMatrix", kLRMargin, kBottomMargin, kLRMargin+kMatrixWidth, kVertSplit);
  TPad* pFDMatrix = pad("FDMatrix", 1-kLRMargin-kMatrixWidth, kBottomMargin, 1-kLRMargin, kVertSplit);

  // Split the middle 4 panels equally
  const double kNDTrueEdge = (.5+(kLRMargin+kMatrixWidth))/2.;
  const double kCenterPlotWidth = .5-kNDTrueEdge;

  TPad* pNDTrue = pad("NDTrue", kLRMargin+kMatrixWidth, kBottomMargin, kNDTrueEdge, kVertSplit);
  TPad* pFDTrue = pad("FDTrue", 1-kNDTrueEdge, kBottomMargin, 1-kLRMargin-kMatrixWidth, kVertSplit);

  TPad* pFN = pad("FN", kNDTrueEdge, kBottomMargin, .5, kVertSplit);
  TPad* pOsc = pad("Osc", .5, kBottomMargin, 1-kNDTrueEdge, kVertSplit);


  Int_t myPalette[1000];
  // Double_t r[]    = {.95, 1, .5};
  // Double_t g[]    = {.95, 0, 0.};
  // Double_t b[]    = {.95, 0, 0.};
  Double_t r[]    = {1, 1, .5};
  Double_t g[]    = {1, 0, 0.};
  Double_t b[]    = {1, 0, 0.};
  Double_t stop[] = {0., .25, 1.0};
  Int_t FI = TColor::CreateGradientColorTable(3, stop, r, g, b, 1000);
  for (int i=0;i<1000;i++) myPalette[i] = FI+i;
  gStyle->SetPalette(1000, myPalette);


  pNDReco->cd();
  h_reco_mc_nd->SetLineColor(kRed);
  h_reco_mc_nd->GetXaxis()->SetLabelSize(0);
  h_reco_mc_nd->Draw("hist");
  h_reco_dat_nd->Draw("ep same");

  pFDReco->cd();
  h_reco_mc_fd->SetLineColor(kRed);
  h_reco_mc_fd->GetXaxis()->SetLabelSize(0);
  h_reco_mc_fd->Draw("hist Y+");
  h_reco_ext_fd->SetLineColor(kBlue);
  h_reco_ext_fd->Draw("hist same");

  pNDTrue->cd();
  h_true_mc_nd->SetLineColor(kRed);
  h_true_rwt_nd->SetLineColor(kBlue);

  TH2* axes = new TH2F("", "", 100, 0, 3, 100, 0, 5);
  axes->GetYaxis()->SetLabelSize(0);
  axes->Draw();
  rotate_hist(h_true_mc_nd)->Draw("l same");
  rotate_hist(h_true_rwt_nd)->Draw("l same");

  pFDTrue->cd();
  h_true_mc_fd->SetLineColor(kRed);
  h_true_ext_fd->SetLineColor(kBlue);

  axes = new TH2F("", "", 100, -6, 0, 100, 0, 5);
  axes->GetYaxis()->SetLabelSize(0);
  axes->Draw();
  rotate_hist(h_true_mc_fd, true)->Draw("l same");
  rotate_hist(h_true_ext_fd, true)->Draw("l same");

  pNDMatrix->cd();
  axes = new TH2F("", "", 100, 0, 5, 100, 0, 5);
  axes->Draw();
  h_nd_matrix->Draw("col same");

  pFDMatrix->cd();
  axes = new TH2F("", "", 100, 0, 5, 100, 0, 5);
  axes->Draw("Y+");
  h_fd_matrix->Draw("col same");

  pFN->cd();
  h_fn->SetLineColor(kRed);
  axes = new TH2F("", "", 100, 0, .03, 100, 0, 5);
  axes->GetYaxis()->SetLabelSize(0);
  axes->Draw();
  rotate_curve(h_fn)->Draw("l same");

  pOsc->cd();
  h_osc->SetLineColor(kRed);
  axes = new TH2F("", "", 100, 0, 1, 100, 0, 5);
  axes->GetYaxis()->SetLabelSize(0);
  axes->Draw();
  rotate_curve(h_osc)->Draw("l same");

  c->cd();

  latex(.5*kLRMargin, .5*kVertSplit + .5*(1-kTopMargin), "10^{3} ND Events", 90);
  latex(1-.5*kLRMargin, .5*kVertSplit + .5*(1-kTopMargin), "FD Events", 90);

  latex(.5*kLRMargin, .5*kVertSplit + .5*(kBottomMargin), "True Energy (GeV)", 90);
  latex(1-.5*kLRMargin, .5*kVertSplit + .5*(kBottomMargin), "True Energy (GeV)", 90);

  latex(kLRMargin+kMatrixWidth/2, .5*kBottomMargin, "ND Reco Energy (GeV)");
  latex(1-kLRMargin-kMatrixWidth/2, .5*kBottomMargin, "FD Reco Energy (GeV)");

  latex(kNDTrueEdge-.5*kCenterPlotWidth, .5*kBottomMargin, "10^{3} ND Events");
  latex(1-kNDTrueEdge+.5*kCenterPlotWidth, .5*kBottomMargin, "FD Events");

  latex(.5-kCenterPlotWidth/2, .5*kBottomMargin, "F/N Ratio");
  latex(.5+kCenterPlotWidth/2, .5*kBottomMargin, "P(#nu_{#mu}#rightarrow#nu_{#mu})");

  // Draw legend by hand
  TLine* lin = new TLine(.37, .85, .395, .85);
  lin->SetLineWidth(2);
  lin->Draw();
  lin = new TLine(.37, .8, .395, .8);
  lin->SetLineWidth(2);
  lin->SetLineColor(kRed);
  lin->Draw();
  lin = new TLine(.37, .75, .395, .75);
  lin->SetLineWidth(2);
  lin->SetLineColor(kBlue);
  lin->Draw();

  latex(.4, .85, "ND data", 0, 12);
  latex(.4, .8, "Base Simulation", 0, 12);
  latex(.4, .75, "Data-Driven Prediction", 0, 12);

  gPad->Print("extrap.pdf");
  gPad->Print("extrap.png");
}

void no_titles

(

std::vector< TH1 * >

hs

)

Definition at line 29 of file make_extrap_figure.C.

References make_syst_table_plots::h.

{
  for(TH1* h: hs){
    h->SetTitle("");
    h->GetXaxis()->SetTitleSize(0);
    h->GetYaxis()->SetTitleSize(0);
  }
}

TPad* pad	(	std::string	name,
		double	x0,
		double	y0,
		double	x1,
		double	y1
	)

Definition at line 38 of file make_extrap_figure.C.

References kXPadding, and kYPadding.

{
  TPad* p = new TPad(name.c_str(), "", 0, 0, 1, 1);
  p->SetLeftMargin(x0+kXPadding);
  p->SetRightMargin(1-x1+kXPadding);
  p->SetTopMargin(1-y1+kYPadding);
  p->SetBottomMargin(y0+kYPadding);
  p->SetFillStyle(0);
  p->Draw();
  return p;
}

TGraph* rotate_curve

(

TH1 *

h

)

Definition at line 59 of file make_extrap_figure.C.

References MECModelEnuComparisons::g, and MECModelEnuComparisons::i.

{
  TGraph* g = new TGraph;
  for(int i = 0; i < h->GetNbinsX()+2; ++i){
    g->SetPoint(i, h->GetBinContent(i), h->GetXaxis()->GetBinCenter(i));
  }
  g->SetLineWidth(2);
  g->SetLineColor(h->GetLineColor());
  return g;
}

TGraph* rotate_hist	(	TH1 *	h,
		bool	neg = false
	)

Definition at line 70 of file make_extrap_figure.C.

References MECModelEnuComparisons::g, MECModelEnuComparisons::i, and canMan::sign().

{
  int sign = neg ? -1 : +1;

  TGraph* g = new TGraph;
  for(int i = 0; i < h->GetNbinsX()+2; ++i){
    g->SetPoint(2*i, sign*h->GetBinContent(i), h->GetXaxis()->GetBinLowEdge(i));
    g->SetPoint(2*i+1, sign*h->GetBinContent(i), h->GetXaxis()->GetBinUpEdge(i));
  }
  g->SetLineWidth(2);
  g->SetLineColor(h->GetLineColor());

  return g;
}

Variable Documentation

const double kBottomMargin = .1

Definition at line 22 of file make_extrap_figure.C.

const double kLRMargin = .05

Definition at line 20 of file make_extrap_figure.C.

const double kMatrixWidth = .15

Definition at line 24 of file make_extrap_figure.C.

const double kTopMargin = .05

Definition at line 21 of file make_extrap_figure.C.

const double kVertSplit = .6

Definition at line 18 of file make_extrap_figure.C.

const double kXPadding = .0025

Definition at line 26 of file make_extrap_figure.C.

const double kYPadding = .005

Definition at line 27 of file make_extrap_figure.C.