get_spectra_2dplots.C

Go to the documentation of this file.

#include "CAFAna/Core/SpectrumLoader.h"
#include "CAFAna/Core/Loaders.h"
#include "CAFAna/Core/Spectrum.h"
#include "CAFAna/Cuts/TimingCuts.h"
#include "CAFAna/Cuts/SpillCuts.h"
#include "3FlavorAna/Cuts/NueCuts2020.h"
#include "CAFAna/Prediction/PredictionNoExtrap.h"
#include "CAFAna/Vars/XsecTunes.h"
#include "CAFAna/Vars/PPFXWeights.h"
#include "CAFAna/Vars/Vars.h"

#include "TFile.h"


using namespace ana;

void get_spectra_2dplots(TString beam = "fhc", TString suffix = "cosmics")
{

 ////////////////////////////////////////////////////////////////////////////////
 // Variables and cuts
 ////////////////////////////////////////////////////////////////////////////////

  const Cut core_no_pid = kNue2020CorePresel && kNue2020FDNearestSlice;
  const Cut peri_no_pid = kNue2020PeripheralBasicEventCut && !kNue2020FD && kNue2020FDNearestSlice;
  
  const HistAxis axis_core ("CVN Score", Binning::Simple(50, 0.75, 1.0), kCVNe_looseptp,
                            "Cosmic BDT", Binning::Simple(50, 0.3, 0.8), kCosPIDCoreBDT);
  const HistAxis axis_peri ("CVN Score", Binning::Simple(50, 0.9, 1.0), kCVNe_looseptp,
                            "Cosmic BDT", Binning::Simple(50, 0.3, 0.8), kCosPIDPeriBDT);
  
  struct SpectDef{
    TString name;
    const HistAxis axis;
    const Cut cut;
  };

  std::vector <SpectDef> defs;
  defs.push_back({"core_no_pid", axis_core, core_no_pid});
  defs.push_back({"peri_no_pid", axis_peri, peri_no_pid});

  ////////////////////////////////////////////////////////////////////////////////
  // Loaders and spectra for cosmic trigger
  //////////////////////////////////////////////////////////////////////////////// 
  if(suffix.Contains("cosmics")){
    std::string ldr;
    if(beam=="fhc") ldr = "prod_sumdecaf_R19-11-18-prod5reco.i_fd_cosmic_fhc_full_v1_goodruns_nue2020";
    if(beam=="rhc") ldr = "prod_sumdecaf_R19-11-18-prod5reco.i_fd_cosmic_rhc_full_v1_goodruns_nue2020";
    SpectrumLoader loader(ldr);
    loader.SetSpillCut(kStandardSpillCuts);
    
    std::vector <Spectrum*> spects;
    for(auto const & def:defs){
      spects.push_back(new Spectrum(loader,def.axis,def.cut && kInCosmicTimingWindow));
    }
    loader.Go();
    
    TString filename = "spectra_"+ beam + "_" + suffix + ".root";
    auto file = new TFile(filename,"recreate"); 
    for(uint i =0; i < spects.size(); ++i){
      spects[i]->SaveTo(file, "cosm_" + defs[i].name);
    }
    file->Close();
    std::cout << "Saved to " << filename << std::endl;
    return;
  }//end cosmics
  
  ////////////////////////////////////////////////////////////////////////////////
  // Loaders and predictions for MC
  ////////////////////////////////////////////////////////////////////////////////
  if(suffix.Contains("prediction")){
    Loaders loaders;
    std::string ldrFDNonSwap, ldrFDFlxSwap, ldrFDTauSwap;
    if(beam=="fhc"){
      ldrFDNonSwap = "prod_sumdecaf_R19-11-18-prod5reco.f_fd_genie_N1810j0211a_nonswap_fhc_nova_v08_full_v1_nue2020";
      ldrFDFlxSwap = "prod_sumdecaf_R19-11-18-prod5reco.f_fd_genie_N1810j0211a_fluxswap_fhc_nova_v08_full_v1_nue2020";
      ldrFDTauSwap = "prod_sumdecaf_R19-11-18-prod5reco.f_fd_genie_N1810j0211a_tau_fhc_nova_v08_full_v1_nue2020";
    }
    if(beam=="rhc"){
      ldrFDNonSwap = "prod_sumdecaf_R19-11-18-prod5reco.f_fd_genie_N1810j0211a_nonswap_rhc_nova_v08_full_v1_nue2020";
      ldrFDFlxSwap = "prod_sumdecaf_R19-11-18-prod5reco.f_fd_genie_N1810j0211a_fluxswap_rhc_nova_v08_full_v1_nue2020";
      ldrFDTauSwap = "prod_sumdecaf_R19-11-18-prod5reco.f_fd_genie_N1810j0211a_tau_rhc_nova_v08_full_v1_nue2020";
    }

    loaders.SetLoaderPath(ldrFDNonSwap,caf::kFARDET, Loaders::kMC, ana::kBeam, Loaders::kNonSwap);
    loaders.SetLoaderPath(ldrFDFlxSwap,caf::kFARDET, Loaders::kMC, ana::kBeam, Loaders::kFluxSwap);
    loaders.SetLoaderPath(ldrFDTauSwap,caf::kFARDET, Loaders::kMC, ana::kBeam, Loaders::kTauSwap);
    loaders.SetSpillCut(kStandardSpillCuts);
    
    std::vector <IPrediction *> preds;
    for(auto const & def:defs){
      preds.push_back(new PredictionNoExtrap(loaders,def.axis,def.cut,kNoShift,
                                             kXSecCVWgt2020*kPPFXFluxCVWgt));
    }
    loaders.Go();

    TString filename = "preds_"+ beam + "_" + suffix + ".root";
    auto file = new TFile(filename,"recreate"); 
    for(uint i =0; i < preds.size(); ++i){
      preds[i]->SaveTo(file, "pred_" + defs[i].name);
    }
    file->Close();
    std::cout << "Saved to " << filename << std::endl;
    return;
    }//end predictions
  else{
    return;
  }
}