make_nueFDprediction_kinematics_FHC.C

Go to the documentation of this file.

// Analysis
#include "CAFAna/Analysis/Exposures.h"
#include "CAFAna/Analysis/Plots.h"
#include "CAFAna/Analysis/Prod5Loaders.h"
#include "CAFAna/Analysis/Style.h"

// Core
#include "CAFAna/Core/Binning.h"
#include "CAFAna/Core/Cut.h"
#include "CAFAna/Core/HistAxis.h"
#include "CAFAna/Core/LoadFromFile.h"
#include "CAFAna/Core/Spectrum.h"
#include "CAFAna/Core/SpectrumLoader.h"
#include "CAFAna/Core/SpectrumLoaderBase.h"

// Cuts
#include "CAFAna/Cuts/Cuts.h"
#include "3FlavorAna/Cuts/NumuCuts.h"
#include "3FlavorAna/Cuts/NumuCuts2020.h"
#include "CAFAna/Cuts/SpillCuts.h"
#include "3FlavorAna/Cuts/NueCuts2020.h"

// Predict
#include "CAFAna/Prediction/IPrediction.h"
#include "3FlavorAna/Prediction/PredictionGenerator3Flavor.h"

// Vars
#include "CAFAna/Vars/XsecTunes.h"
#include "CAFAna/Vars/TruthVars.h"
#include "3FlavorAna/Vars/HistAxes.h"
#include "3FlavorAna/Vars/NueVars.h"
#include "CAFAna/Vars/Vars.h"
#include "CAFAna/Vars/PPFXWeights.h"

#include "StandardRecord/Proxy/SRProxy.h"

#include <iostream>
#include <iomanip>

using namespace ana;

void make_nueFDprediction_kinematics_FHC()
{  
  const std::string outfilename = "FDprediction_kinematics_FHC.root";
  const std::string sample = "NOReweight";

  Prod5NomLoaders loaders(ana::kNueConcat, ana::Loaders::kFHC);
  loaders.SetSpillCut(kStandardSpillCuts);

  struct GenDef{
    const IPredictionGenerator *gen;
    const TString cutname;
    const TString varname;
  };

  std::vector <GenDef> gens;
  std::vector <IPrediction * > predictions;

  gens.push_back({new NoExtrapGenerator(kNue2020Axis, kNue2020FDAllSamples, kXSecCVWgt2020*kPPFXFluxCVWgt),
        sample.c_str(), "nueAxis_NoExtrap"}); 

  gens.push_back({new NueSignalExtrapGenerator( kNue2020Axis,
                                                kNumuCCOptimisedAxis2020,
                                                kNue2020FDAllSamples,
                                                kNumu2020ND,
                                                kNoShift,
                                                kXSecCVWgt2020*kPPFXFluxCVWgt),
        sample.c_str(), "nueAxis_NueSignalExtrap"});

  for(auto & gen:gens){
    predictions.push_back(gen.gen->Generate(loaders).release());
  }

  loaders.Go();

  TFile* file = new TFile(outfilename.c_str(),"RECREATE");
  for (int i = 0 ; i < (int)gens.size(); ++i){
    auto dir = file->GetDirectory(gens[i].cutname);
    if(!dir) dir = file->mkdir(gens[i].cutname);
    predictions[i]->SaveTo(dir, gens[i].varname);
  }

  file->Close();
  
}