GetSpectra.h

Go to the documentation of this file.

const int kNumPlots = 11;

struct Plot
{
  std::string label;
  std::string fname;
  ana::Binning bins;
  Var var;
};


void getPredictions(unsigned int timestamp, unsigned int nDays, std::string naming){
  
  std::cout << "\n=================" << std::endl;
  std::cout << "getPredictions.C"   << std::endl;
  std::cout << "=================\n" <<std::endl;
  Plot plots[kNumPlots] =
    {
      {"Slice time [#mus]", "tus", ana::Binning::Simple(45,1,541), ana::kSliceTime},
      {"Run", "run", ana::Binning::Simple(15000, 15000, 30000), SIMPLEVAR(hdr.run)},
      {"Time", "time", ana::Binning::Simple(nDays<3 ? 24*nDays : nDays,timestamp - 86400*nDays,timestamp), SIMPLEVAR(hdr.unixtime)},
      {"Max y [m]", "maxy", kXYBins, kBoxMaxY},
      {"Slice time [#mus]", "tighttus", ana::Binning::Simple(360,205,241), ana::kSliceTime},
      {"No. Kalman tracks", "nkal", ana::Binning::Simple(10, 0, 10), SIMPLEVAR(trk.kalman.ntracks)},
      {"No. Hits", "nhit", ana::Binning::Simple(10, 0, 500), ana::kNHit},
      {"Kalman angle", "angkal", ana::Binning::Simple(5, 0, 1), SIMPLEVAR(sel.cosrej.anglekal)},
      {"Z direction", "dirz", ana::Binning::Simple(5, 0.5, 1), ana::kDirZ},
      {"BDT", "bdt", ana::Binning::Simple(20, 0, 1), ana::kNumuContPID},
      {"p_{T}/p", "ptp", ana::Binning::Simple(20, 0, 1), ana::kPtP}
    };


  std::cout << "Setting loaders ..." << std::endl;
  //SpectrumLoader fd_nonswap("prod_caf_R19-11-18-prod5reco.f_fd_genie_N1810j0211a_nonswap_rhc_nova_v08_full_v1");
  //SpectrumLoader fd_fluxswap("prod_caf_R19-11-18-prod5reco.f_fd_genie_N1810j0211a_fluxswap_rhc_nova_v08_full_v1");
  SpectrumLoader fd_nonswap("prod_caf_R19-11-18-prod5reco.f_fd_genie_N1810j0211a_nonswap_fhc_nova_v08_full_v1");
  SpectrumLoader fd_fluxswap("prod_caf_R19-11-18-prod5reco.f_fd_genie_N1810j0211a_fluxswap_fhc_nova_v08_full_v1");
  
  PredictionNoExtrap* hPred[kNumPlots];

  for(int i = 0; i < kNumPlots; ++i){
    Plot p = plots[i];
    std::cout << "Defining PredictionNoExtrap " << i << std::endl;
    hPred[i]=new PredictionNoExtrap(fd_nonswap,fd_fluxswap,kNullLoader,p.label, p.bins, p.var, kCut, kNoShift , kPPFXFluxCVWgt*kXSecCVWgt2020);
  }
  
  Spectrum nonSwapDenom("NHits", ana::Binning::Simple(100, 0, 1000), fd_nonswap, ana::kNHit, kNoCut, kNoShift, kWOscDumbHere);
  Spectrum swapDenom("NHits", ana::Binning::Simple(100, 0, 1000), fd_fluxswap, ana::kNHit, kNoCut, kNoShift, kWOscDumbHere);

  std::cout << "Loaders Go ..." << std::endl;
  fd_nonswap.Go();
  fd_fluxswap.Go();


  std::string fileName = "predictions_" + naming + ".root";
  std::cerr << "Saving predictions to file: " << fileName << std::endl;
  TFile *file = TFile::Open(fileName.c_str(), "RECREATE");
  for(int i = 0; i < kNumPlots; ++i){
    Plot p = plots[i];
    hPred[i]->SaveTo(file , (p.fname).c_str() );
  }

  nonSwapDenom.SaveTo(file, "nonSwapDenom");
  swapDenom.SaveTo(file,"swapDenom");
  file->Close();

  std::cout << "Predictions made! \n" << std::endl; 


} // getPredictions



void getData(unsigned int timestamp, unsigned int nDays, std::string naming, std::string samdef){

  std::cout << "\n=================" << std::endl;
  std::cout << "getData.C"   << std::endl;
  std::cout << "=================\n" <<std::endl;

  std::cout << "Setting loaders ..." << std::endl;

  unsigned int fhcStart = 1551204000; //12pm Central time, Feb 26th 2019.
  std::string fFiles;
  
  if(naming == "All")
    fFiles = "defname: " + samdef + Form(" and online.subrunendtime <= '%u' and online.subrunendtime >= '%u'", timestamp, fhcStart);
  else
    fFiles = "defname: " + samdef + Form(" and online.subrunendtime >= '%u' and online.subrunendtime <= '%u' and online.subrunendtime >= '%u'", timestamp - 86400*nDays, timestamp, fhcStart);


  fFiles = "defname: " + samdef;
  SpectrumLoader loader(fFiles);

  std::cout << "Defining spectrums ..." << std::endl;
  Spectrum* hData[kNumPlots];               
  Spectrum* hBkg[kNumPlots];

  Plot plots[kNumPlots] =
    {
      {"Slice time [#mus]", "tus", ana::Binning::Simple(45,1,541), ana::kSliceTime},
      {"Run", "run", ana::Binning::Simple(15000, 15000, 30000), SIMPLEVAR(hdr.run)},
      {"Time", "time", ana::Binning::Simple(nDays<3 ? 24*nDays : nDays,timestamp - 86400*nDays,timestamp), SIMPLEVAR(hdr.unixtime)},
      {"Max y [m]", "maxy", kXYBins, kBoxMaxY},
      {"Slice time [#mus]", "tighttus", ana::Binning::Simple(360,205,241), ana::kSliceTime},
      {"No. Kalman tracks", "nkal", ana::Binning::Simple(10, 0, 10), SIMPLEVAR(trk.kalman.ntracks)},
      {"No. Hits", "nhit", ana::Binning::Simple(10, 0, 500), ana::kNHit},
      {"Kalman angle", "angkal", ana::Binning::Simple(5, 0, 1), SIMPLEVAR(sel.cosrej.anglekal)},
      {"Z direction", "dirz", ana::Binning::Simple(5, 0.5, 1), ana::kDirZ},
      {"BDT", "bdt", ana::Binning::Simple(20, 0, 1), ana::kNumuContPID},
      {"p_{T}/p", "ptp", ana::Binning::Simple(20, 0, 1), ana::kPtP}
    };
  

  for(int i = 0; i < kNumPlots; ++i){
    Plot p = plots[i];
    hData[i] = new Spectrum(p.label, p.bins, loader, p.var, i <= 1 ? kCut : kCut && kInTimeTight, kNoShift, kUnweighted);
    hBkg[i]  = new Spectrum(p.label, p.bins, loader, p.var, kCut && !kInTimeTight, kNoShift, kUnweighted);
  }

  Spectrum potTime("Date","POT #times Mass", loader, ana::Binning::Simple(nDays<3 ? 24*nDays : nDays,timestamp - 86400*nDays,timestamp), SIMPLEVAR(hdr.unixtime), ana::Binning::Simple(500,0,50e12),kMassFracPOT, kNoCut, kNoShift, kUnweighted);
  Spectrum massDenom("", ana::Binning::Simple(1, 0, 1), loader, kUnweighted, kNoCut, kNoShift, kSpillPOT);
  Spectrum massNum("", ana::Binning::Simple(1, 0, 1), loader, kUnweighted, kNoCut, kNoShift, kMassFracPOT);
  
  std::cout << "Loaders Go ..." << std::endl;
  loader.Go();


  std::string fileName = "dataspectra_" + naming + ".root";
  std::cerr << "Saving predictions to file: " << fileName << std::endl;
  TFile *file = TFile::Open(fileName.c_str(), "RECREATE");

  for(int i = 0; i < kNumPlots; ++i){
    Plot p = plots[i];
    hData[i]->SaveTo(file , ("data_"+p.fname).c_str() );
    hBkg[i]->SaveTo(file,  ("bkg_"+p.fname).c_str() );    
  }

  potTime.SaveTo(file , "potTime" );
  massDenom.SaveTo(file, "massDenom");
  massNum.SaveTo(file, "massNum");

  file->Close();

  std::cout << "Spectra made! \n" << std::endl;

}// getData