make_nueFDprediction_kinematics_FHC_pTExtrap.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"
#include "3FlavorAna/Cuts/QuantileCuts2020.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_pTExtrap()
{  
  const std::string outfilename = "FDprediction_kinematics_FHC_pTExtrap.root";
  const std::string sample = "NOReweight";

  const unsigned int nQuants = 3;

  bool isRHC = false;
  std::vector<Cut> PtQuantCutsND = GetNueQuantCuts2020( isRHC, caf::kNEARDET, nQuants, ana::kExtrapPt);
  std::vector<Cut> PtQuantCutsFD = GetNueQuantCuts2020( isRHC, caf::kFARDET, nQuants, ana::kExtrapPt);

  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"}); 

  //Loop over samples
  for ( unsigned int quant = 0; quant < nQuants; ++quant ){        
    gens.push_back({new NueSignalExtrapGenerator( kNue2020Axis,
                                                  kNumuCCOptimisedAxis2020,
                                                  kNue2020FDAllSamples && PtQuantCutsFD[quant],
                                                  kNumu2020ND && PtQuantCutsND[quant],
                                                  kNoShift,
                                                  kXSecCVWgt2020*kPPFXFluxCVWgt),
          sample.c_str(), Form("nueAxis_NueSignalExtrap_Quant%i", quant+1)});
  }
  
  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();
  
}