CutFlow_Cosmic.C

Go to the documentation of this file.

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

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

#include "3FlavorAna/Vars/Binnings.h"
#include "3FlavorAna/Vars/NumuEFxs.h"
#include "3FlavorAna/Vars/NueEnergy2020.h"
#include "CAFAna/Vars/GenieWeights.h"
#include "CAFAna/Vars/PPFXWeights.h"
#include "CAFAna/Vars/XsecTunes.h"

#include "3FlavorAna/Ana2020/CutFlowTables/CutFlow_header.h"

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

using namespace ana;

#include <fstream>

void CutFlow_Cosmic( bool isFHC ) {
  
  std::string sFHC = "fhc";
  if (!isFHC) {
    sFHC = "rhc";
  }

  // --- Set my loader
  SpectrumLoader loader( "prod_caf_R19-11-18-prod5reco.i_fd_cosmic_"+sFHC+"_full_v1_goodruns" );
  loader.SetSpillCut(kStandardSpillCuts);
  
  // --- Define my cuts and their names.
  std::vector<std::string> NuMu_CutNames; std::vector<Cut> NuMu_TieredCuts;
  std::vector<std::string> Nue_CutNames;  std::vector<Cut> Nue_TieredCuts;
  MakeCutFlowVecs( NuMu_CutNames, NuMu_TieredCuts, Nue_CutNames, Nue_TieredCuts );

  size_t NuMuCuts = NuMu_TieredCuts.size();
  size_t NueCuts  = Nue_TieredCuts .size();

  // --- Define my objects.
  Spectrum* NuMu_ReconEnergy[NuMuCuts];
  Spectrum* Nue_ReconEnergy [NueCuts ];
  
  const Var kMyWgt = kUnweighted;
  const Binning kSimpBins  = Binning::Simple(102, -0.01, 1.01 );

  // --- Set my objects.
  for(size_t cut = 0; cut < NuMuCuts; ++cut) {
    const Cut         kThisCut  = NuMu_TieredCuts[cut] && kInCosmicTimingWindow;
    const std::string kThisName = "NuMuCCEnergy_"+NuMu_CutNames[cut];
    NuMu_ReconEnergy[cut] = new Spectrum( kThisName, kNumuCCEOptimisedBinning, loader, kCCE          , kThisCut, kNoShift, kMyWgt );
  }
  for(size_t cut = 0; cut < NueCuts; ++cut) {
    const Cut         kThisCut  = Nue_TieredCuts[cut] && kInCosmicTimingWindow;
    const std::string kThisName = "NueCCEnergy_"+Nue_CutNames[cut];
    Nue_ReconEnergy [cut] = new Spectrum( kThisName, kSimpBins               , loader, kCVNe_looseptp, kThisCut, kNoShift, kMyWgt );
  }

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

  // --- Define an outfile.
  std::string OutName = "CosmicCutFlow_"+sFHC+".root";
  TFile *OutFile = new TFile(OutName.c_str(), "RECREATE");
  OutFile  -> cd();
  for(size_t cut = 0; cut < NuMuCuts; ++cut){
    NuMu_ReconEnergy[cut] -> SaveTo( OutFile,  TString("NuMuCCEnergy_")+TString(NuMu_CutNames[cut]) ) ;
  }
  for(size_t cut = 0; cut < NueCuts; ++cut) {
    Nue_ReconEnergy [cut] -> SaveTo( OutFile,  TString("NueCCEnergy_" )+TString(Nue_CutNames [cut]) ) ;
  }
} // End of function