PlotSpectra.h

Go to the documentation of this file.

#pragma once

#include "StandardRecord/Proxy/SRProxy.h"
#include "CAFAna/Core/Binning.h"
#include "CAFAna/Core/Spectrum.h"
#include "CAFAna/Core/SpectrumLoader.h"
#include "CAFAna/Core/EventList.h"
#include "CAFAna/Core/Utilities.h"
#include "CAFAna/Cuts/Cuts.h"
#include "CAFAna/Vars/Vars.h"

#include "3FlavorAna/Vars/NueVarsExtra.h"
#include "3FlavorAna/Vars/NumuVarsExtra.h"
#include "3FlavorAna/Cuts/NueCuts2020.h"
#include "3FlavorAna/Cuts/NumuCuts2020.h"

#include "NuXAna/Vars/NusVars.h"
#include "NuXAna/Cuts/NusCuts20.h"
#include "NuXAna/Cuts/NusCuts.h"
#include "NuXAna/macros/Nus20/Selection/CutFlow.h"
#include "NuXAna/macros/Nus20/Selection/CutVariables.h"
#include "NuXAna/macros/Nus20/Selection/FOMUtilities.h"
#include "NuXAna/macros/Nus20/Selection/DecorrelateMatrix.h"

#include "TFile.h"
#include "TString.h"
#include "TH1.h"
#include "TCanvas.h"
#include "TPad.h"
#include "THStack.h"
#include "TLegend.h"
#include "TBox.h"
#include "TLatex.h"
#include "TStyle.h"

/// helper plotting scripts
#include "/nashome/a/alister1/NOvAPlotting.h"

enum Detector {
  kNear,
  kFar
};

std::vector<float> ndFractionalUncertainties = {};
std::vector<float> fdFractionalUncertainties = {};

std::string dirName;
std::vector<VarContainer> vars;
std::vector<CutContainer> cuts;
std::vector<CutContainer> subcuts;

// base histograms we use
TH1D* histTotal         = new TH1D(); /// total sig + bg
TH1D* histMCTotal       = new TH1D(); /// total mc only
TH1D* histMCSignal      = new TH1D(); /// total mc signal
TH1D* histMCSignalNoCut = new TH1D(); /// total mc signal (no cut)
TH1D* histCosmic        = new TH1D(); /// total cosmic
TH1D* histMCBg          = new TH1D(); /// mc-only bg (no cosmic)
TH1D* histData          = new TH1D(); /// total data


// histograms for FOMs
TH1D* EffTimesPurHist   = new TH1D();
TH1D* SOverSBHist       = new TH1D();
TH1D* SOverSSBHist      = new TH1D();
TH1D* BinnedSOverSSBSystHist = new TH1D();

TH2D* histMCTotalEnergy = nullptr;
TH2D* histMCSignalEnergy = nullptr;

bool isAreaNorm = true;

TCanvas* cRatio = new TCanvas("cRatio", "cRatio", 600, 600);
TPad*    topPad = new TPad("topPad", "", 0.005, 0.3, 0.995, 0.995);
TPad*    botPad = new TPad("botPad", "", 0.005, 0.005, 0.995, 0.3);

TCanvas* cNoRatio = new TCanvas("cNoRatio", "cNoRatio", 600, 500);
TPad* thisPad = new TPad("thisPad", "", 0.005, 0.005, 0.995, 0.995);

// loops histograms from file
void LoopHistograms(TFile* spectraFile, Detector thisDet);

// style histograms from file
void StyleHistograms(TH1D* h, Detector det, 
                     bool isBeamData, bool isCosmic, bool isTotal);

// set margins of pads 
void SetupCanvas(TPad* topPad, TPad* botPad, TPad* thisPad);

// determine whether to plot data for a given spectra
bool PlotData(std::string name);

// stle legends
void StyleLegend(TLegend* leg);

// draw boxes to show cuts
void DrawBoxes(TH1D* histMCTotal, std::vector<double> cutValues, CutSide thisCutSide);

// draw ratio plot
void DrawRatio(TPad* thisPad, TH1D* mc, TH1D* dat);

// get fractional uncertainties from covariance matrix
void GetFractionalUncertainties(TFile* covmx,
                                                                                                                                std::vector<float>* nd,
                                                                                                                                std::vector<float>* fd);

// initialise histograms we're going to use
void InitialiseHistograms(TFile* spectraFile, Detector thisDet, int iCut, int iVar,
                          std::string dirName, std::string histPathPre, std::string histPath);

// set up FOMs to draw
void InitialiseFOMHists(Detector thisDet, int iCut, int iVar);

// ...and then draw them
void DrawFOM(std::set<FomType>, int iVar);

// print selected events
void PrintSelectedEvents(int iCut, int iVar);

// draw functions
void DrawDataMC(Detector thisDet, int iCut, int iVar,
                bool isLogPlot, bool isCosmic, bool isAreaNormalised);
void DrawMCOnly(Detector thisDet, int iCut, int iVar,
                bool isLogPlot, bool isCosmic);
void DrawIndependentSpectra(Detector thisDet, int iCut, int iVar);

// get standard legend setup
TLegend* GetStandardLegend();