MakeNus18ExtrapPred.C

Go to the documentation of this file.

//////////////////////////////////////////////////////////////
// MakeNus18ExtrapPred.C
// M Wallbank (wallbank@fnal.gov)
//
// Make extrapolated predictions.
//////////////////////////////////////////////////////////////

// framework
#include "CAFAna/Analysis/Calcs.h"
#include "CAFAna/Analysis/Prod4Loaders.h"
#include "CAFAna/Core/Loaders.h"
#include "CAFAna/Core/Binning.h"
#include "CAFAna/Core/SpectrumLoader.h"
#include "NuXAna/Cuts/NusCuts18.h"
#include "CAFAna/Cuts/SpillCuts.h"
#include "CAFAna/Cuts/TimingCuts.h"
#include "CAFAna/Vars/GenieWeights.h"
#include "CAFAna/Vars/HistAxes.h"
#include "NuXAna/Vars/HistAxes.h"
#include "CAFAna/Vars/PPFXWeights.h"
#include "NuXAna/macros/Nus18/Nus18PredictionHandlers.h"
#include "NuXAna/macros/Nus18/Nus18Utilities.h"
#include "OscLib/OscCalcSterile.h"

// stl
#include <utility>

using namespace ana;

void MakeNus18ExtrapPred(TString opt) {

  TH1::AddDirectory(0);

  // Get running options
  bool rhc = opt.Contains("rhc", TString::kIgnoreCase);
  bool fhc = opt.Contains("fhc", TString::kIgnoreCase);
  assert(rhc || fhc);

  bool concat = opt.Contains("concat", TString::kIgnoreCase);
  bool caf = opt.Contains("caf", TString::kIgnoreCase);
  assert(concat || caf);

  bool systematics = opt.Contains("syst", TString::kIgnoreCase);
  bool cosmics = opt.Contains("cos", TString::kIgnoreCase);

  // Weighting
  const Var kReweight = kXSecCVWgt2018*kPPFXFluxCVWgt;

  // Set up the loaders
  Loaders* loaders = new Prod4NomLoaders(
    concat? ECAFType::kNusConcat : ECAFType::kFullCAF,
    rhc? Loaders::FluxType::kRHC : Loaders::FluxType::kFHC);
  loaders->SetSpillCut(kStandardSpillCuts);

  // Cosmic loaders
  SpectrumLoader* loaders_cos;
  if (rhc)
    loaders_cos = new SpectrumLoader(kFDCosmicRHCProd4, kCosmic);
  else
    loaders_cos = new SpectrumLoader(kFDCosmicFHCProd4, kCosmic);
  loaders_cos->SetSpillCut(kStandardSpillCuts);

  // Get systematics
  std::vector<const ISyst*> systs;
  // if (systematics) {
  //   NusSystematics* nus_systs = LoadSystematics();
  //   systs = nus_systs->GetISysts();
  // }

  // Set up oscillation calculator
  osc::OscCalcSterile* calc = DefaultSterileCalc(3);

  // Set up the PredictionHandler object
  FDPredictionHandler prediction("Extrap");
  prediction.AddLoaders(loaders);
  if (cosmics)
    prediction.AddCosmics(loaders_cos);
  if (systematics)
    prediction.AddSystematics(systs, calc);
  prediction.AddVar("Nus18Extrap", kNus18AxisE, kNus18FD, kNoShift, kReweight);

  // Set off the loaders!
  loaders->Go();
  loaders_cos->Go();

  // Write the data
  std::stringstream outFileName; outFileName << "pred_nus18_";
  rhc ? outFileName << "rhc" : outFileName << "fhc";
  outFileName << "_extrap";
  if (systematics) outFileName << "_syst";
  outFileName << ".root";
  TFile* outFile = new TFile(outFileName.str().c_str(), "RECREATE");
  prediction.Save(outFile);
  outFile->Close();
  delete outFile;

  return;

}