nue_draw_michelDataMC.C

Go to the documentation of this file.

#include "CAFAna/Core/Spectrum.h"
#include "CAFAna/Core/LoadFromFile.h"
#include "CAFAna/Analysis/Plots.h"

#include "Utilities/func/MathUtil.h"

#include "TCanvas.h"
#include "TH2.h"
#include "TH1.h"
#include "TFile.h"
#include "TLegend.h"

#include <iostream>
#include <cmath>

using namespace ana;


void MakePlot(std::string fname, std::string type)
{
  Spectrum *da = LoadFromFile<Spectrum>(fname,type+"/"+type+"_DA").release();
  Spectrum *mc = LoadFromFile<Spectrum>(fname,type+"/"+type+"_MC").release();
  Spectrum *me = LoadFromFile<Spectrum>(fname,
                                        type+"/"+type+"_MCTrueME").release();

  double pot = da->POT();
  TH1 *hda = da->ToTH1(pot);
  TH1 *hmc = mc->ToTH1(pot);
  TH1 *hme = me->ToTH1(pot);
  TH1 *hbg = mc->ToTH1(pot);
  hbg->Add(hme,-1); // Background is total - true MEs

  // Scale histograms down to make them prettier
  hmc->Scale(1e-3);
  hme->Scale(1e-3);
  hbg->Scale(1e-3);
  for (int i = 1; i <= hmc->GetNbinsX(); i++){
    hda->SetBinContent(i,1e-3*hda->GetBinContent(i));
    hda->SetBinError(i,1e-3*hda->GetBinError(i)); // Do this to fix errors too
  }

  hmc->SetLineColor(kRed);
  hbg->SetLineColor(kGreen+2);
  hme->SetLineColor(kBlue);

  hmc->SetMaximum(1.2*std::max(hmc->GetMaximum(),hda->GetMaximum()));
  hmc->SetMinimum(0);

  hmc->GetXaxis()->CenterTitle();
  hmc->GetYaxis()->SetTitle("10^{3} Events / 8.09#times10^{20}");
  hmc->GetYaxis()->CenterTitle();

  hmc->Draw("hist");
  hbg->Draw("hist,same");
  hme->Draw("hist,same");
  hda->Draw("E,same");

  TLegend *leg = AutoPlaceLegend(0.3,0.35);
  leg->AddEntry(hda,"Data","le");
  leg->AddEntry(hmc,"Total MC","l");
  leg->AddEntry(hme,"True Michel","l");
  leg->AddEntry(hbg,"Non-Michel Bkgd","l");
  leg->Draw();
}

void MakePlots(std::string fname)
{
  TCanvas *canMID = new TCanvas();
  MakePlot(fname,"MID");
  TCanvas *canInput = new TCanvas(UniqueName().c_str(),"",1000,750);
  canInput->Divide(2,2);
  canInput->cd(1);
  MakePlot(fname,"calE");
  canInput->cd(2);
  MakePlot(fname,"NHit");
  canInput->cd(3);
  MakePlot(fname,"DeltaT");
  canInput->cd(4);
  MakePlot(fname,"Dist");
}

void nue_draw_michelDataMC(std::string fname)
{
  MakePlots(fname);
}