CutFlow_Data.C

Go to the documentation of this file.

#include "CAFAna/Cuts/Cuts.h"
#include "3FlavorAna/Cuts/NumuCuts.h"
#include "3FlavorAna/Cuts/NumuCuts2017.h"
#include "CAFAna/Cuts/SpillCuts.h"
#include "CAFAna/Cuts/TimingCuts.h"

#include "CAFAna/Core/SpectrumLoader.h"
#include "CAFAna/Core/Spectrum.h"
#include "CAFAna/Core/ReweightableSpectrum.h"

#include "CAFAna/Vars/Vars.h"
#include "3FlavorAna/Vars/Binnings.h"
#include "3FlavorAna/Vars/NumuVars.h"

#include "CAFAna/Analysis/Exposures.h"
#include "CAFAna/Analysis/Prod3Loaders.h"

using namespace ana;

#include "TStyle.h"
#include "TCanvas.h"
#include "TH1.h"
#include "TLegend.h"
#include "TLatex.h"

#include <fstream>

struct Plot {
  std::string label;
  std::string fname;
  Binning bins;
  Var var;
};

void CutFlow_Data()
{
  // --- Set my loader.
  SpectrumLoader loader( "prod_restrictedcaf_R17-03-01-prod3reco.k_fd_numi_fhc_full_v1_goodruns" );
  //SpectrumLoader loader( "/pnfs/nova/production/R17-03-01-prod3reco.k/numi/fd/numu_contain_sa_2017/prod_sumrestricteddecaf*.root" );
  loader.SetSpillCut(kStandardSpillCuts);

  // --- Define my cuts and their names.
  std::vector<Cut> TieredCuts;
  std::vector<std::string> CutNames;
  CutNames.emplace_back("NoCut");       TieredCuts.emplace_back(kNumu2017Veto && kNoCut);
  CutNames.emplace_back("Quality");     TieredCuts.emplace_back(kNumu2017Veto && kNumuQuality);
  CutNames.emplace_back("Containment"); TieredCuts.emplace_back(kNumu2017Veto && kNumuQuality && kNumuContainFD2017);
  CutNames.emplace_back("Cosmic_Rej");  TieredCuts.emplace_back(kNumu2017Veto && kNumuQuality && kNumuContainFD2017 && kNumuCosmicRej2017);
  CutNames.emplace_back("Particle_ID"); TieredCuts.emplace_back(kNumu2017Veto && kNumuQuality && kNumuContainFD2017 && kNumuCosmicRej2017 && kNumuPID2017 );

  size_t NCuts = TieredCuts.size();

  // --- Define my objects.
  Plot plot = {"CC Energy [GeV]", "NuMuCCEnergy", kNumuCCEOptimisedBinning, kCCE};
  Spectrum* Cosmics[NCuts];
  Spectrum* BeamNus[NCuts];

  // --- Set my objects.
  for(size_t cut = 0; cut < NCuts; ++cut) {
    const Cut Cut_Cosm = TieredCuts[cut] && kInTimingSideband;
    const Cut Cut_Beam = TieredCuts[cut] && kInBeamSpill;
    Cosmics[cut] = new Spectrum( plot.label, plot.bins, loader, plot.var, Cut_Cosm, kNoShift, kTimingSidebandWeight );
    BeamNus[cut] = new Spectrum( plot.label, plot.bins, loader, plot.var, Cut_Beam, kNoShift, kUnweighted           );
  }

  // --- Set the loader off!
  loader.Go();

  // --- Define an outfile.
  std::string OutName = "DataCutFlow.root";
  TFile *OutFile = new TFile(OutName.c_str(), "RECREATE");
  OutFile  -> cd();
  for(size_t cut = 0; cut < NCuts; ++cut){
    Cosmics[cut] -> SaveTo( OutFile,  TString(CutNames[cut])+TString("_Cosmics") ) ;
    BeamNus[cut] -> SaveTo( OutFile,  TString(CutNames[cut])+TString("_BeamNus") ) ;
  }
} // End of function