datamc_ND_numu_kinematics_RHC.C

Go to the documentation of this file.

#include "CAFAna/Analysis/Plots.h"
#include "CAFAna/Prediction/PredictionNoExtrap.h"
#include "StandardRecord/Proxy/SRProxy.h" 

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

// Cuts
#include "CAFAna/Cuts/Cuts.h"
#include "CAFAna/Cuts/SpillCuts.h"
#include "CAFAna/Cuts/NueCutsFirstAna.h"
#include "CAFAna/Cuts/NueCutsSecondAna.h"
#include "3FlavorAna/Cuts/NueCuts2017.h"
#include "3FlavorAna/Cuts/NueCuts2018.h"
#include "3FlavorAna/Cuts/NumuCuts.h"
#include "3FlavorAna/Cuts/NumuCuts2017.h"
#include "3FlavorAna/Cuts/NumuCuts2018.h"
#include "CAFAna/Cuts/TruthCuts.h"

// Vars
#include "CAFAna/Vars/Vars.h"
#include "CAFAna/Vars/GenieWeights.h"
#include "3FlavorAna/Vars/NueVarsExtra.h"
#include "3FlavorAna/Vars/NueVars.h"
#include "3FlavorAna/Vars/NumuVars.h"
#include "CAFAna/Vars/PPFXWeights.h"
#include "CAFAna/Vars/HistAxes.h"
#include "3FlavorAna/Vars/HistAxes.h"

#include "CAFAna/Analysis/Style.h"
#include "CAFAna/Decomp/IDecomp.h"
#include "CAFAna/Decomp/ProportionalDecomp.h"

#include "3FlavorAna/Decomp/NumuDecomp.h"
#include "CAFAna/Decomp/CheatDecomp.h"
#include "3FlavorAna/Decomp/MichelDecomp.h"

#include "CAFAna/Extrap/ModularExtrap.h"
#include "CAFAna/Prediction/PredictionExtrap.h"

#include "TFile.h"

#include "CAFAna/Core/Spectrum.h"
#include "CAFAna/Core/Utilities.h"



#include <fstream>
#include <iostream>
#include <cmath>
#include <string>

using namespace ana;

#include "Utilities/func/MathUtil.h"


void datamc_ND_numu_kinematics_RHC()
{
  std::string nameData = "prod_caf_R17-09-05-prod4recopreview.f_nd_numi_rhc_full_v1_addShortSimpleCVN_goodruns";
  std::string nameMC = "prod_caf_R17-11-14-prod4reco.e_nd_genie_nonswap_rhc_nova_v08_full_v1";
  
  //choose your samweb datasets or concat files: 
  SpectrumLoader loaderData(nameData);
  SpectrumLoader loaderMC(nameMC);

  loaderData.SetSpillCut(kStandardSpillCuts);
  loaderMC.SetSpillCut(kStandardDQCuts);

  //which variables do you want to look at?
  const int kNumVars = 7;
  
  const HistDef defs[kNumVars] = {
      {"numu", kNumuCCAxis},
      {"trueQ2", {"true Q^{2} (GeV^{2})", Binning::Simple(500, 0, 5), kTrueQ2}},
      {"recoQ2", {"reco Q^{2} (GeV^{2})", Binning::Simple(500, 0, 5), kRecoQ2}},
      {"trueW2", {"true W^{2} (GeV^{2})", Binning::Simple(100, 0, 2), kTrueW}},
      {"trueE", {"true E (GeV)", Binning::Simple(100, 0, 5), kTrueE}},
      {"PtP", {"p_{t}/p", Binning::Simple(500,0,1), kPtP}},
      {"CosNumi", {"CosNumi", Binning::Simple(500,0,1), kCosNumi}}
  };
   
  const int kNumSels = 1;
  const Cut sels[kNumSels] = {
      kNumuCutND2018 
      };
  const std::string selNames[kNumSels] = {
      "numuND"
      };

  Spectrum* spects[kNumSels][kNumVars];
  IPrediction* preds[kNumSels][kNumVars];

  //loop over the selectors and variables to create a full selection
  for(int selIdx = 0; selIdx < kNumSels; ++selIdx){
    for(int varIdx = 0; varIdx < kNumVars; ++varIdx){
      const HistAxis& axis = defs[varIdx].axis;
      spects[selIdx][varIdx] = new Spectrum(loaderData, axis, sels[selIdx], kNoShift);
      preds[selIdx][varIdx] = new PredictionNoExtrap(loaderMC, kNullLoader, kNullLoader,
                                                     axis.GetLabels()[0], axis.GetBinnings()[0], axis.GetVars()[0],
                                                     sels[selIdx], kNoShift, kXSecCVWgt2018*kPPFXFluxCVWgt);           
    }
  }

  loaderMC.Go();
  loaderData.Go();
  
  TString fname = "datamc_ND_numu_kinematics.root";
  TFile* fout = new TFile(fname, "RECREATE");
  for(int selIdx = 0; selIdx < kNumSels; ++selIdx){
    TDirectory* d = fout->mkdir(selNames[selIdx].c_str());
    for(int varIdx = 0; varIdx < kNumVars; ++varIdx){
      const char* name = defs[varIdx].name.c_str();
      spects[selIdx][varIdx]->SaveTo(d, TString::Format("spect_%s", name));
      preds[selIdx][varIdx]->SaveTo(d, TString::Format("pred_%s", name));
    }
  }
}