get_fdmc_spectra.C

Go to the documentation of this file.

#include "CAFAna/Analysis/Prod3Loaders.h"
#include "CAFAna/Core/Binning.h"
#include "CAFAna/Cuts/SpillCuts.h"
#include "CAFAna/Vars/GenieWeights.h"
#include "CAFAna/Vars/PPFXWeights.h"
#include "CAFAna/Core/Binning.h"
#include "3FlavorAna/Cuts/NueCuts2017.h"
#include "3FlavorAna/Cuts/NumuCuts2017.h"
#include "CAFAna/Prediction/IPrediction.h"
#include "3FlavorAna/Prediction/PredictionGenerator3Flavor.h"
#include "3FlavorAna/Vars/HistAxes.h"
#include "3FlavorAna/Ana2017/nue/sidebands/vars.h"

//#include "/nova/app/users/ecatanom/nova_code/releases/rel_oct11/CAFAna/nue/Ana2017/sidebands/vars.h"
using namespace ana;
void get_fdmc_spectra(std::string suffix = "full", bool extrapolate = false)
{
   Prod3NomLoaders loaders(ana::kNueConcat);
  // Prod3NomLoaders loaders(ana::kDecaf); //test
  loaders.SetSpillCut(kStandardSpillCuts);
  
  auto axis_numu2017 =  kNumuCCOptimisedAxis;
  auto cut_numund2017 = kNumuCutND2017;
  auto axis_nue2017 = kNue2017AxisMergedPeripheral;
  auto cut_nuend2017 = kNue2017NDCVNSsb;
  auto weight = kPPFXFluxCVWgt*kXSecCVWgt2017;
  
  struct GenDef{
    const IPredictionGenerator *gen;
    const TString cutname;
    const TString varname;
  };
  
  std::vector <GenDef> gens;
  std::vector <IPrediction * > predictions;
  
  for(int selIdx = 0; selIdx < kNumSels; ++selIdx){
    for(int varIdx=0; varIdx < kNumVars; ++varIdx){
      auto axis = defs[varIdx].axis;
      gens.push_back({new NoExtrapGenerator(axis, sels[selIdx], weight),
            selNames[selIdx], defs[varIdx].name});
  
       if(extrapolate)    {
      gens.push_back({new NueComboExtrapGenerator( axis,
                                                   axis_numu2017,
                                                   sels[selIdx],
                                                   cut_nuend2017,
                                                   cut_numund2017,
                                                   kNoShift,
                                                   weight),
            selNames[selIdx], defs[varIdx].name+"_extrap_combo"});
    gens.push_back({new NuePropExtrapGenerator( axis,
                                                axis_numu2017,
                                                sels[selIdx],
                                                cut_nuend2017,
                                                cut_numund2017,
                                                kNoShift,
                                                weight),
          selNames[selIdx], defs[varIdx].name+"_extrap_prop"});
       }//end if extrapolation
     } //end var
  }//end selection
  
  for(auto & gen:gens){
    predictions.push_back(gen.gen->Generate(loaders).release());
  }
  
  loaders.Go();
  
  TFile* file = new TFile(("fhc_preds_all_epochs_fdnue_cuts_" + suffix + ".root").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);
  }
  
}