drawShortTerm_2019.C

Go to the documentation of this file.

#include "CAFAna/Cuts/Cuts.h"
#include "CAFAna/Cuts/SpillCuts.h"
#include "CAFAna/Cuts/TimingCuts.h"
#include "CAFAna/Cuts/NumuCuts2018.h"
#include "CAFAna/Core/Loaders.h"
#include "CAFAna/Core/SpectrumLoader.h"
#include "CAFAna/Core/Spectrum.h"
#include "CAFAna/Core/ReweightableSpectrum.h"
#include "CAFAna/Vars/NumuVars.h"
#include "CAFAna/Vars/Vars.h"
#include "CAFAna/Analysis/Calcs.h"
#include "CAFAna/Analysis/Plots.h"

#include "StandardRecord/Proxy/SRProxy.h"

#include "TStyle.h"
#include "TCanvas.h"
#include "TGraph.h"
#include "TGraphAsymmErrors.h"
#include "TFile.h"
#include "TH1.h"
#include "TH2.h"
#include "TLegend.h"
#include "TLatex.h"
#include "TProfile.h"

#include "VarsAndCuts_2019.h"
#include "Periods_2019.h"
#include "CustomFunctions_2019.h"


using namespace ana;

void drawShortTerm_2019(std::string samDef, unsigned int endTime, unsigned int nDays, std::string s_PeriodLength, std::string outDir)
{
  //SET SIG FIGS.
  gStyle->SetStatFormat("6.3g"); 
  gStyle->SetPaintTextFormat("6.3g");
  gStyle->SetFitFormat("6.3g");

  //DEFINE A TFILE TO WRITE ALL CANVASES TO.
  TFile *f_Out = new TFile((TString)(outDir+"BeamMonitoring_"+s_PeriodLength+"_ShortTerm.root"),"RECREATE");

  //APPLY TIME CUTS TO DATASET, DEFINE THE LOADER AND SET SPILL CUTS.
  unsigned int startTime = endTime - 86400*nDays;
  const std::string fFiles = "dataset_def_name_newest_snapshot " + samDef + Form(" and online.subrunendtime >= '%u' and online.subrunendtime <= '%u'", startTime, endTime);
  SpectrumLoader loader(fFiles);
  loader.SetSpillCut(kStandardSpillCuts);

  const Binning kTimeBinning = Binning::Simple(nDays < 3 ? 24*nDays : nDays+1, startTime, startTime + 86400*(nBins+1));
  const Cut     kNumuNDCut   = kNumuCutND2018;

  //DEFINE PLOTS OF THE VARIABLES WE WANT TO SHOW.
  std::vector<Plot> vec_Plots = {
    {"Central Time", "Spill POT (#times10^{12})",            "spillPOT", Binning::Simple(100,0,100), kSpillPOT,  kNoCut,      1.0,   1. },
    {"Central Time", "Reconstructed Neutrino Energy, (GeV)", "ccE",      Binning::Simple(20,0,5),    kCCE,       kNumuNDCut,  1.5,  2.2 },
    {"Central Time", "Reconstructed Hadronic Energy, (GeV)", "hadE",     Binning::Simple(30,0,3),    kHadE,      kNumuNDCut,  0.15, 0.50},
    {"Central Time", "Reconstructed Muon Energy, (GeV)",     "muonE",    Binning::Simple(45,0,4.5),  kMuE,       kNumuNDCut,  1.0,  2.0 },
    {"Central Time", "Hadronic Energy Fraction",             "hadEfrac", Binning::Simple(20,0,1.0),  kHadEFrac,  kNumuNDCut,  0.05, 0.3 },
    {"Central Time", "No. of Hits",                          "nhit",     Binning::Simple(25,0,250),  kNHit,      kNumuNDCut,  90,   110 },
    {"Central Time", "Vertex X, (m)",                        "startX",   Binning::Simple(40,-2,2),   kTrkStartX, kNumuNDCut, -0.40, 0.0 },
    {"Central Time", "Vertex Y, (m)",                        "startY",   Binning::Simple(40,-2,2),   kTrkStartY, kNumuNDCut, -0.25, 0.15},
    {"Central Time", "Vertex Z, (m)",                        "startZ",   Binning::Simple(24,0,12),   kTrkStartZ, kNumuNDCut,  4.8,  6.25}
  };

  std::vector<Spectrum*> vec_Spectrum;
  std::vector<TCanvas*>  vec_Canvas;

  for(unsigned int i = 0; i < vec_Plots.size(); i++)
  {
    Spectrum *h_Spectrum = new Spectrum(vec_Plots.at(i).fLabelX, vec_Plots.at(i).fLabelY, loader, kTimeBinning, kTime, vec_Plots.at(i).fBins, vec_Plots.at(i).fVar, vec_Plots.at(i).fCut);
    vec_Spectrum.push_back(h_Spectrum);
    TCanvas  *c_Spectrum = new TCanvas((TString)("c_"+vec_Plots.at(i).fName+"_"+s_PeriodLength),(TString)("c_"+vec_Plots.at(i).fName+"_"+s_PeriodLength));
    c_Spectrum->SetBottomMargin(0.15);
    c_Spectrum->SetLeftMargin  (0.15);
    c_Spectrum->SetRightMargin (0.15);
    vec_Canvas.push_back(c_Spectrum);
  }

  loader.Go();

  for(unsigned int i = 0; i < vec_Plots.size(); i++)
  {
    vec_Canvas.at(i)->cd(); 
    TH2 *h_2D = vec_Spectrum.at(i)->ToTH2(vec_Spectrum.at(i)->POT());
    h_2D->SetName((TString)("h_2D_"+vec_Plots.at(i).fName+s_PeriodLength));
    h_2D->SetTitle("");
    h_2D->GetXaxis()->CenterTitle();
    h_2D->GetYaxis()->CenterTitle();
    h_2D->GetXaxis()->SetTimeDisplay(1);
    if (nDays >= 15)                {h_2D->GetXaxis()->SetTimeFormat("%y-%m-%d%F1970-01-01 00:00:00");      }
    else if(nDays > 3 && nDays < 15){h_2D->GetXaxis()->SetTimeFormat("%y-%m-%d-%H:%M%F1970-01-01 00:00:00");}
    else                            {h_2D->GetXaxis()->SetTimeFormat("%H:%M%F1970-01-01 00:00:00");         }

    unsigned int minBin = h_2D->GetXaxis()->FindBin(startTime);
    unsigned int maxBin = h_2D->GetXaxis()->FindBin(endTime);

    double minZoom = h_2D->GetXaxis()->GetBinLowEdge(minBin-1);
    double maxZoom = h_2D->GetXaxis()->GetBinLowEdge(maxBin+2);

    h_2D->GetXaxis()->SetRangeUser(minZoom, maxZoom);

    TProfile *h_Prof1 = h_2D->ProfileX((TString)("h_Prof1_"+vec_Plots.at(i).fName+"_"+s_PeriodLength));
    TProfile *h_Prof2 = h_2D->ProfileX((TString)("h_Prof2_"+vec_Plots.at(i).fName+"_"+s_PeriodLength));
    h_Prof1->SetMarkerStyle(24);
    h_Prof1->SetMarkerColor(kBlack);
    h_Prof1->SetLineColor  (kBlack);
    h_Prof1->SetMarkerSize(1.4);
    h_Prof2->SetMarkerStyle(20);
    h_Prof2->SetMarkerColor(kYellow);
    h_Prof2->SetLineColor  (kYellow);
    h_Prof2->SetMarkerSize(1.4);

    h_2D   ->Draw("COLZ");
    h_Prof2->Draw("SAME");
    h_Prof1->Draw("SAME");

    TLatex *txt_Exposure = new TLatex(0.2, 0.8, Form("%6.3g POT", vec_Spectrum.at(i)->POT()));
    txt_Exposure->SetNDC();
    txt_Exposure->SetTextSize(0.04);
    txt_Exposure->Draw();
    lastUpdated(endTime);

    vec_Canvas.at(i)->SaveAs((TString)(outDir+vec_Plots.at(i).fName+"_"+s_PeriodLength+"_ShortTerm.png"));
    f_Out->cd();
    vec_Canvas.at(i)->Write();
    vec_Canvas.at(i)->Clear();
    vec_Canvas.at(i)->SetName((TString)("c_1D_"+vec_Plots.at(i).fName+"_"+s_PeriodLength));

    TH1D *h_1D = h_2D->ProjectionY((TString)("h_1D_"+vec_Plots.at(i).fName+s_PeriodLength));
    h_1D->Draw("HIST");
    vec_Canvas.at(i)->Write();
  }

  std::cout <<  "ALL PLOTS MADE" << std::endl;
}