datamc_ND_numu_kinematics_RHC_pTBins.C

Go to the documentation of this file.

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

// Core
#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 "3FlavorAna/Cuts/QuantileCuts2020.h"
#include "3FlavorAna/Cuts/NueCuts2020.h"
#include "3FlavorAna/Cuts/NumuCuts.h"
#include "3FlavorAna/Cuts/NumuCuts2020.h"
#include "CAFAna/Cuts/TruthCuts.h"

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

#include "CAFAna/Analysis/Style.h"
#include "CAFAna/Core/Utilities.h"

#include "TFile.h"
#include <fstream>
#include <iostream>
#include <cmath>
#include <string>

using namespace ana;

#include "Utilities/func/MathUtil.h"


void datamc_ND_numu_kinematics_RHC_pTBins()
{

  std::string nameData = "prod_caf_R19-11-18-prod5reco.g_nd_numi_rhc_full_v1_goodruns";
  std::string nameMC = "prod_caf_R19-11-18-prod5reco.d_nd_genie_N1810j0211a_nonswap_rhc_nova_v08_full_v1";

  SpectrumLoader loaderData(ndData);
  SpectrumLoader loaderMC(ndMC);

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

  //which variables do you want to look at?
  const int kNumVars = 5;
  
  struct HistDef{
    std::string name;
    const HistAxis axis;
  };

  const HistDef defs[kNumVars] = {
      {"numu", kNumuCCAxis},
      {"Muon pT", {"#mu p_{T}", Binning::Simple(50,0,5), kNumuMuonPt}},
      {"recoQ2", {"reco Q^{2} (GeV^{2})", Binning::Simple(50, 0, 5), kRecoQ2}},
      {"PtP", {"p_{t}/p", Binning::Simple(20,0,1), kPtP}},
      {"CosNumi", {"CosNumi", Binning::Simple(20,0,1), kCosNumi}}
  };
   
  const int kNumSels = 4;
  const unsigned int nQuants = 3;
  bool isRHC = true;
  std::vector<Cut> PtQuantCutsND = GetNueQuantCuts2020( isRHC, caf::kNEARDET, nQuants, ana::kExtrapPt );
  const Cut sels[kNumSels] = {
    kNumu2020ND,
    kNumu2020ND && PtQuantCutsND[0],
    kNumu2020ND && PtQuantCutsND[1],
    kNumu2020ND && PtQuantCutsND[2]
  };
  const std::string selNames[kNumSels] = {  
    "AllBins",
    "Quant1",
    "Quant2",
    "Quant3"
  };

  Spectrum* NDdata[kNumSels][kNumVars];
  Spectrum* NDmc[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;
      NDdata[selIdx][varIdx] = new Spectrum(loaderData, axis, sels[selIdx], kNoShift);
      NDmc[selIdx][varIdx] = new Spectrum(loaderMC, axis, sels[selIdx], kNoShift, kXSecCVWgt2020*kPPFXFluxCVWgt);

    }
  }

  loaderMC.Go();
  loaderData.Go();
 
  TString fname = "datamc_ND_numu_kinematics_RHC_pTBins.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();
      NDdata[selIdx][varIdx]->SaveTo(d, TString::Format("data_%s", name));
      NDmc[selIdx][varIdx]->SaveTo(d, TString::Format("mc_%s", name));
    }
  }
}