sidebandfittest.C

Go to the documentation of this file.

#ifdef __CINT__
void sidebandfittest()
{
  std::cout << "Sorry, you must run in compiled mode" << std::endl;
}
#else

#include "CAFAna/Core/SpectrumLoader.h"
#include "CAFAna/Core/Spectrum.h"
#include "CAFAna/Core/SystShifts.h"

#include "CAFAna/Cuts/SpillCuts.h"
#include "CAFAna/Cuts/TruthCuts.h"
#include "3FlavorAna/Cuts/NueCutsSecondAna.h"

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

#include "CAFAna/Prediction/PredictionScaleComp.h"

#include "NDAna/nuecc_inc/NueCCIncVars.h"
#include "NDAna/nuecc_inc/NueCCIncCuts.h"
#include "NDAna/nuecc_inc/NueCCIncBins.h"

#include "TH1.h"
#include "TH2.h"
#include "THStack.h"
#include "TCanvas.h"
#include "TLegend.h"

#include <vector>
#include <string>

using namespace ana;

void sidebandfittest()
{
  //Setup the loaders and apply spill cuts
  const std::string MCfiles = "defname: prod_caf_R17-03-01-prod3reco.d_nd_genie_nonswap_fhc_nova_v08_full_v1 with stride 4 offset 1";
  SpectrumLoader lNDMC(MCfiles);
  lNDMC.SetSpillCut(kStandardSpillCuts);

  const std::string Datafiles = "defname: prod_caf_R17-03-01-prod3reco.d_nd_genie_nonswap_fhc_nova_v08_full_v1 with stride 4 offset 0";
  SpectrumLoader lNDData(Datafiles);
  lNDData.SetSpillCut(kStandardSpillCuts);


  //Rewighting for Mock Data Challenge
  const Var kReweight = kNumuSBWeight * kNCSBWeight * kTuftsWeightCC;


  //Now make the data spectra
  Spectrum data_reweighted (lNDData, sideband_x_axis, sideband_y_axis,
                            kCVNCutSideband, kNoShift, kReweight);
  Spectrum data_tufts (lNDData, sideband_x_axis, sideband_y_axis,
                       kCVNCutSideband, kNoShift, kTuftsWeightCC);  



  //Break down of MC and Data by truth cut in sideband
  Spectrum data_sig_sb(lNDData, sideband_x_axis, sideband_y_axis, 
                       kCVNCutSideband && kIsNueCC, 
                       kNoShift, kTuftsWeightCC);
  Spectrum data_bkg_numu_sb(lNDData, sideband_x_axis, sideband_y_axis, 
                            kCVNCutSideband && kIsNumuBkg, 
                            kNoShift, kTuftsWeightCC);
  Spectrum data_bkg_nc_sb(lNDData, sideband_x_axis, sideband_y_axis,
                          kCVNCutSideband && kIsNC, 
                          kNoShift, kTuftsWeightCC);
  Spectrum data_sig_anti_sb(lNDData, sideband_x_axis, sideband_y_axis, 
                            kCVNCutSideband && kIsAntiNueCC, 
                            kNoShift, kTuftsWeightCC);
  Spectrum data_bkg_other_sb(lNDData, sideband_x_axis, sideband_y_axis, 
                             kCVNCutSideband && kIsOther_nueccinc, 
                             kNoShift, kTuftsWeightCC);

  Spectrum data_sig_sb_weight(lNDData, sideband_x_axis, sideband_y_axis, 
                              kCVNCutSideband && kIsNueCC, 
                              kNoShift, kReweight);

  Spectrum data_bkg_numu_sb_weight(lNDData, sideband_x_axis, sideband_y_axis, 
                                   kCVNCutSideband && kIsNumuBkg, 
                                   kNoShift, kReweight);
  Spectrum data_bkg_nc_sb_weight(lNDData, sideband_x_axis, sideband_y_axis, 
                                 kCVNCutSideband && kIsNC, 
                                 kNoShift,  kReweight);
  Spectrum data_sig_anti_sb_weight(lNDData, sideband_x_axis, sideband_y_axis, 
                                   kCVNCutSideband && kIsAntiNueCC, 
                                   kNoShift, kReweight);
  Spectrum data_bkg_other_sb_weight(lNDData, sideband_x_axis, sideband_y_axis, 
                                    kCVNCutSideband && kIsOther_nueccinc, 
                                    kNoShift, kReweight);

  Spectrum mc_sig_sb(lNDMC, sideband_x_axis, sideband_y_axis,
                     kCVNCutSideband && kIsNueCC, 
                     kNoShift, kTuftsWeightCC);
  Spectrum mc_bkg_numu_sb(lNDMC, sideband_x_axis, sideband_y_axis, 
                          kCVNCutSideband && kIsNumuBkg, 
                          kNoShift, kTuftsWeightCC);
  Spectrum mc_bkg_nc_sb(lNDMC, sideband_x_axis, sideband_y_axis, 
                        kCVNCutSideband && kIsNC, 
                        kNoShift, kTuftsWeightCC);
  Spectrum mc_sig_anti_sb(lNDMC, sideband_x_axis, sideband_y_axis, 
                          kCVNCutSideband && kIsAntiNueCC, 
                          kNoShift, kTuftsWeightCC);
  Spectrum mc_bkg_other_sb(lNDMC, sideband_x_axis, sideband_y_axis, 
                           kCVNCutSideband && kIsOther_nueccinc, 
                           kNoShift, kTuftsWeightCC);


  //Do it.
  lNDMC.Go();
  lNDData.Go();

  TFile* out = new TFile("SidebandFittest.root", "recreate");
  out->cd();
  
  data_reweighted.SaveTo(out->mkdir("data_reweighted"));
  data_tufts.SaveTo(out->mkdir("data_tufts"));
  data_sig_sb.SaveTo(out->mkdir("data_sig_sb"));
  data_bkg_numu_sb.SaveTo(out->mkdir("data_bkg_numu_sb"));
  data_bkg_nc_sb.SaveTo(out->mkdir("data_bkg_nc_sb"));
  data_sig_anti_sb.SaveTo(out->mkdir("data_sig_anti_sb"));
  data_bkg_other_sb.SaveTo(out->mkdir("data_bkg_other_sb"));
  mc_sig_sb.SaveTo(out->mkdir("mc_sig_sb"));
  mc_bkg_numu_sb.SaveTo(out->mkdir("mc_bkg_numu_sb"));
  mc_bkg_nc_sb.SaveTo(out->mkdir("mc_bkg_nc_sb"));
  mc_sig_anti_sb.SaveTo(out->mkdir("mc_sig_anti_sb"));
  mc_bkg_other_sb.SaveTo(out->mkdir("mc_bkg_other_sb"));
  data_sig_sb_weight.SaveTo(out->mkdir("data_sig_sb_weight"));
  data_bkg_numu_sb_weight.SaveTo(out->mkdir("data_bkg_numu_sb_weight"));
  data_bkg_nc_sb_weight.SaveTo(out->mkdir("data_bkg_nc_sb_weight"));
  data_sig_anti_sb_weight.SaveTo(out->mkdir("data_sig_anti_sb_weight"));
  data_bkg_other_sb_weight.SaveTo(out->mkdir("data_bkg_other_sb_weight"));
  out->Close();
  
}


#endif