datamc_ND_numu_kinematics_REW.C
Go to the documentation of this file.
4 
5 // Core
8 #include "CAFAna/Core/Spectrum.h"
10 #include "CAFAna/Core/HistAxis.h"
11 #include "CAFAna/Core/Binning.h"
12 #include "CAFAna/Core/Cut.h"
13 
14 // Cuts
15 #include "CAFAna/Cuts/Cuts.h"
16 #include "CAFAna/Cuts/SpillCuts.h"
17 #include "CAFAna/Cuts/NueCutsFirstAna.h"
18 #include "CAFAna/Cuts/NueCutsSecondAna.h"
21 #include "CAFAna/Cuts/TruthCuts.h"
22 
23 // Vars
24 #include "CAFAna/Vars/Vars.h"
30 
31 // Systs
32 #include "CAFAna/Systs/Systs.h"
33 #include "CAFAna/Analysis/Style.h"
34 #include "CAFAna/Decomp/IDecomp.h"
36 
37 #include "OscLib/OscCalcPMNSOpt.h"
38 
39 
40 #include "CAFAna/Core/SystShifts.h"
47 #include "TFile.h"
48 #include "OscLib/OscCalcPMNSOpt.h"
49 #include "OscLib/OscCalcDumb.h"
50 #include "CAFAna/Core/Spectrum.h"
51 #include "TH1.h"
53 #include "TCanvas.h"
55 #include "CAFAna/Analysis/Plots.h"
56 #include "CAFAna/Vars/FitVars.h"
57 #include "CAFAna/Cuts/Cuts.h"
59 
60 #include "TStopwatch.h"
61 #include "CAFAna/Core/Utilities.h"
62 
63 
64 
65 #include <fstream>
66 #include <iostream>
67 #include <cmath>
68 #include <string>
69 
70 using namespace ana;
71 
72 #include "OscLib/IOscCalc.h"
73 
74 // ROOT
75 #include "TCanvas.h"
76 #include "TFile.h"
77 #include "TH1.h"
78 #include "TLegend.h"
79 #include "TPad.h"
80 #include "TMath.h"
81 #include "TH2.h"
82 #include "TLine.h"
83 #include "TText.h"
84 #include "TString.h"
85 
86 #include "Utilities/func/MathUtil.h"
88 
89 
90 //define some useful variables:
91 
93 {
94 
95  std::string nameData = "defname: prod_decaf_R17-03-01-prod3reco.d_nd_numi_fhc_full_nue_or_numu_or_nus_contain_v1_goodruns";
96  std::string nameMC = "defname: prod_decaf_R17-03-01-prod3reco.d_nd_genie_nonswap_fhc_nova_v08_full_nue_or_numu_or_nus_contain_v1";
97 
98  //choose your samweb datasets or concat files:
99  SpectrumLoader loaderData(nameData);
100  SpectrumLoader loaderMC(nameMC);
101 
102  loaderData.SetSpillCut(kStandardSpillCuts);
103  loaderMC.SetSpillCut(kStandardDQCuts);
104 
105  //which variables do you want to look at?
106  const int kNumVars = 7;
107 
108  const HistDef defs[kNumVars] = {
109  {"recoE", {"E_{reco} (GeV)", Binning::Simple(100, 0, 5), kCCE}},
110  {"trueQ2", {"true Q^{2} (GeV^{2})", Binning::Simple(500, 0, 5), kTrueQ2}},
111  {"recoQ2", {"reco Q^{2} (GeV^{2})", Binning::Simple(500, 0, 5), kRecoQ2}},
112  {"trueW2", {"true W^{2} (GeV^{2})", Binning::Simple(100, 0, 2), kTrueW}},
113  {"trueE", {"true E (GeV)", Binning::Simple(100, 0, 5), kTrueE}},
114  {"PtP", {"p_{t}/p", Binning::Simple(500,0,1), kPtP}},
115  {"CosNumi", {"CosNumi", Binning::Simple(500,0,1), kCosNumi}}
116  };
117 
118  TFile *weight_file = new TFile("/nova/app/users/nostom66/nue/extrapsysts/combined/Weights.root","READ");
119 
120  //trueQ2 weight
121  TH1 *h = new TH1();
122  gDirectory->GetObject("trueQ2_weight_AllSamples",h);
123 
124  std::cout << h->GetNbinsX() << endl;
125  int nbins = h->GetNbinsX();
126  double bins[nbins], binslow[nbins], binshigh[nbins];
127 
128  for(int i = 1; i <= nbins; ++i){
129  bins[i-1] = h->GetBinContent(i);
130  binslow[i-1] = h->GetBinLowEdge(i);
131  binshigh[i-1] = h->GetBinLowEdge(i) + h->GetBinWidth(i);
132  }
133 
134  const Var kTrueQ2Weight([nbins, &binslow, &binshigh, &bins](const caf::SRProxy *sr)
135  {
136  for(int i = 0; i < nbins; ++i){
137  if((kRecoQ2(sr) > binslow[i]) && (kRecoQ2(sr) < binshigh[i])){return bins[i];}
138  }
139  return 1.0;
140  }
141  );
142 
143  //PtP weight
144  TH1 *hptp = new TH1();
145  gDirectory->GetObject("PtP_weight_AllSamples",hptp);
146 
147  std::cout << hptp->GetNbinsX() << endl;
148  int nbins_ptp = hptp->GetNbinsX();
149  double bins_ptp[nbins_ptp], binslow_ptp[nbins_ptp], binshigh_ptp[nbins_ptp];
150 
151  for(int i = 1; i <= nbins_ptp; ++i){
152  bins_ptp[i-1] = hptp->GetBinContent(i);
153  binslow_ptp[i-1] = hptp->GetBinLowEdge(i);
154  binshigh_ptp[i-1] = hptp->GetBinLowEdge(i) + hptp->GetBinWidth(i);
155  }
156 
157  const Var kPtPWeight([nbins_ptp, &binslow_ptp, &binshigh_ptp, &bins_ptp](const caf::SRProxy *sr)
158  {
159  for(int i = 0; i < nbins_ptp; ++i){
160  if((kPtP(sr) > binslow_ptp[i]) && (kPtP(sr) < binshigh_ptp[i])){return bins_ptp[i];}
161  }
162  return 1.0;
163  }
164  );
165 
166  //Cos weight
167  TH1 *hcos = new TH1();
168  gDirectory->GetObject("CosNumi_weight_AllSamples",hcos);
169 
170  std::cout << hcos->GetNbinsX() << endl;
171  int nbins_cos = hptp->GetNbinsX();
172  double bins_cos[nbins_cos], binslow_cos[nbins_cos], binshigh_cos[nbins_cos];
173 
174  for(int i = 1; i <= nbins_cos; ++i){
175  bins_cos[i-1] = hcos->GetBinContent(i);
176  binslow_cos[i-1] = hcos->GetBinLowEdge(i);
177  binshigh_cos[i-1] = hcos->GetBinLowEdge(i) + hcos->GetBinWidth(i);
178  }
179 
180  const Var kCosWeight([nbins_cos, &binslow_cos, &binshigh_cos, &bins_cos](const caf::SRProxy *sr)
181  {
182  for(int i = 0; i < nbins_cos; ++i){
183  if((kCosNumi(sr) > binslow_cos[i]) && (kCosNumi(sr) < binshigh_cos[i])){return bins_cos[i];}
184  }
185  return 1.0;
186  }
187  );
188 
189  weight_file->Close();
190 
191  const Cut kIsRock(
192  [](const caf::SRProxy* sr)
193  {
194  if (sr->mc.nnu == 0)
195  return false;
196  if (fabs(sr->mc.nu[0].vtx.X())>180 ||
197  fabs(sr->mc.nu[0].vtx.Y())>180 ||
198  sr->mc.nu[0].vtx.Z()<0 ||
199  sr->mc.nu[0].vtx.Z()>1250)
200  return true;
201  return false;
202  });
203 
204  const int kNumSels = 1;
205  const Cut sels[kNumSels] = {
206  kNumuNDCvn
207  };
208  const std::string selNames[kNumSels] = {
209  "numuND"
210  };
211 
212  Spectrum* spects_trueQ2[kNumSels][kNumVars];
213  Spectrum* spects_PtP[kNumSels][kNumVars];
214  Spectrum* spects_Cos[kNumSels][kNumVars];
215  IPrediction* preds_trueQ2[kNumSels][kNumVars];
216  IPrediction* preds_PtP[kNumSels][kNumVars];
217  IPrediction* preds_Cos[kNumSels][kNumVars];
218 
219  //loop over the selectors and variables to create a full selection
220  for(int selIdx = 0; selIdx < kNumSels; ++selIdx){
221  for(int varIdx = 0; varIdx < kNumVars; ++varIdx){
222  const HistAxis& axis = defs[varIdx].axis;
223 
224  // trueQ2 reweighted
225  spects_trueQ2[selIdx][varIdx] = new Spectrum(loaderData, axis, sels[selIdx], kNoShift, kTrueQ2Weight);
226  preds_trueQ2[selIdx][varIdx] = new PredictionNoExtrap(loaderMC, kNullLoader, kNullLoader,
227  axis.GetLabels()[0], axis.GetBinnings()[0], axis.GetVars()[0],
228  sels[selIdx], kNoShift, kXSecCVWgt2017*kPPFXFluxCVWgt*kTrueQ2Weight);
229  // PtP reweighted
230  spects_PtP[selIdx][varIdx] = new Spectrum(loaderData, axis, sels[selIdx], kNoShift, kPtPWeight);
231  preds_PtP[selIdx][varIdx] = new PredictionNoExtrap(loaderMC, kNullLoader, kNullLoader,
232  axis.GetLabels()[0], axis.GetBinnings()[0], axis.GetVars()[0],
233  sels[selIdx], kNoShift, kXSecCVWgt2017*kPPFXFluxCVWgt*kPtPWeight);
234  // CosTheta reweighted
235  spects_Cos[selIdx][varIdx] = new Spectrum(loaderData, axis, sels[selIdx], kNoShift, kCosWeight);
236  preds_Cos[selIdx][varIdx] = new PredictionNoExtrap(loaderMC, kNullLoader, kNullLoader,
237  axis.GetLabels()[0], axis.GetBinnings()[0], axis.GetVars()[0],
238  sels[selIdx], kNoShift, kXSecCVWgt2017*kPPFXFluxCVWgt*kCosWeight);
239 
240  }
241  }
242 
243  loaderMC.Go();
244  loaderData.Go();
245 
246  TString fname = "datamc_ND_numu_kinematics_REW.root";
247  TFile* fout = new TFile(fname, "RECREATE");
248  for(int selIdx = 0; selIdx < kNumSels; ++selIdx){
249  TDirectory* d = fout->mkdir(selNames[selIdx].c_str());
250  for(int varIdx = 0; varIdx < kNumVars; ++varIdx){
251  const char* name = defs[varIdx].name.c_str();
252  spects_trueQ2[selIdx][varIdx]->SaveTo(d, TString::Format("spect_trueQ2_%s", name));
253  preds_trueQ2[selIdx][varIdx]->SaveTo(d, TString::Format("pred_trueQ2_%s", name));
254 
255  spects_PtP[selIdx][varIdx]->SaveTo(d, TString::Format("spect_PtP_%s", name));
256  preds_PtP[selIdx][varIdx]->SaveTo(d, TString::Format("pred_PtP_%s", name));
257 
258  spects_Cos[selIdx][varIdx]->SaveTo(d, TString::Format("spect_Cos_%s", name));
259  preds_Cos[selIdx][varIdx]->SaveTo(d, TString::Format("pred_Cos_%s", name));
260  }
261  }
262 }
const XML_Char * name
Definition: expat.h:151
const ana::Var kRecoQ2([](const caf::SRProxy *sr){const double M_mu_sqrd=util::sqr(0.1056);double E_mu=kMuE(sr);double p_mu=sqrt(util::sqr(E_mu)-M_mu_sqrd);return 2 *kCCE(sr)*(E_mu-p_mu *kCosNumi(sr))-M_mu_sqrd;})
Reconstructed four-momentum transfer invariant (Q^2)
Definition: NumuVars.h:146
const int kNumVars
Definition: vars.h:14
Cuts and Vars for the 2020 FD DiF Study.
Definition: vars.h:6
fvar< T > fabs(const fvar< T > &x)
Definition: fabs.hpp:15
const std::vector< T > & GetVars() const
Definition: HistAxis.h:92
const Var kPPFXFluxCVWgt
Definition: PPFXWeights.h:16
Proxy for caf::StandardRecord.
Definition: SRProxy.h:2085
caf::Proxy< std::vector< caf::SRNeutrino > > nu
Definition: SRProxy.h:574
std::string name
Definition: NuePlotLists.h:12
caf::Proxy< short int > nnu
Definition: SRProxy.h:573
virtual void SaveTo(TDirectory *dir, const std::string &name) const
void SetSpillCut(const SpillCut &cut)
void datamc_ND_numu_kinematics_REW()
const Var kTrueQ2
Definition: TruthVars.h:27
Representation of a spectrum in any variable, with associated POT.
Definition: Spectrum.h:40
const int nbins
Definition: cellShifts.C:15
const Var kPtP
Transverse momentum fraction in slice.
Definition: NueVars.cxx:90
const Var kTrueE([](const caf::SRProxy *sr){return(sr->mc.nnu==0)?0.:float(sr->mc.nu[0].E);})
Definition: Vars.cxx:85
if(dump)
const Cut sels[kNumSels]
Definition: vars.h:44
const Cut kNumuNDCvn
Definition: NumuCuts.h:62
const SpillCut kStandardDQCuts([](const caf::SRSpillProxy *spill){if(spill->dcmedgematchfrac==0 &&spill->fracdcm3hits==0 &&spill->nmissingdcmslg==0) return bool(spill->isgoodspill); if(spill->det==caf::kNEARDET && (spill->fracdcm3hits > 0.45|| spill->nmissingdcms > 0)) return false; if(spill->eventincomplete) return false; if(spill->det==caf::kFARDET && spill->nmissingdcmslg > 0) return false; if(spill->det==caf::kFARDET && !spill->ismc && spill->dcmedgematchfrac<=0.2) return false;return true;})
Cut out events with a noisy detector or with parts missing.
Definition: SpillCuts.h:16
const Var kCCE
Definition: NumuVars.h:21
Float_t d
Definition: plot.C:236
caf::StandardRecord * sr
virtual void Go() override
Load all the registered spectra.
void SaveTo(TDirectory *dir, const std::string &name) const
Definition: Spectrum.cxx:517
const HistDef defs[kNumVars]
Definition: vars.h:15
const int kNumSels
Definition: vars.h:43
std::vector< float > Spectrum
Definition: Constants.h:570
HistAxis axis
Definition: NuePlotLists.h:13
const SystShifts kNoShift
Definition: SystShifts.cxx:21
OStream cout
Definition: OStream.cxx:6
caf::Proxy< caf::SRTruthBranch > mc
Definition: SRProxy.h:2097
const std::vector< Binning > & GetBinnings() const
Definition: LabelsAndBins.h:69
const Binning bins
Definition: NumuCC_CPiBin.h:8
::xsd::cxx::tree::string< char, simple_type > string
Definition: Database.h:154
Collaborates with Spectrum and OscillatableSpectrum to fill spectra from CAF files.
Standard interface to all prediction techniques.
Definition: IPrediction.h:57
const SpillCut kStandardSpillCuts
Apply this unless you&#39;re doing something special.
Definition: SpillCuts.h:49
void Format(TGraph *gr, int lcol, int lsty, int lwid, int mcol, int msty, double msiz)
Definition: Style.cxx:154
Prediction that just uses FD MC, with no extrapolation.
const std::string selNames[kNumSels]
Definition: vars.h:46
const Var kXSecCVWgt2017
Definition: XsecTunes.h:37
static Binning Simple(int n, double lo, double hi, const std::vector< std::string > &labels={})
Definition: Binning.cxx:114
const Var kTrueW
Definition: TruthVars.h:22
const Var kCosNumi([](const caf::SRProxy *sr){if(sr->trk.kalman.ntracks > 0 &&sr->trk.kalman.idxremid!=999){if(sr->hdr.det==1){return sr->trk.kalman.tracks[0].dir.Dot(beamDirND);}if(sr->hdr.det==2){return sr->trk.kalman.tracks[0].dir.Dot(beamDirFD);}}return-5.f;})
Definition: NumuVars.h:43
static NullLoader kNullLoader
Dummy loader that doesn&#39;t load any files.
const std::vector< std::string > & GetLabels() const
Definition: LabelsAndBins.h:68