GetSpectra.h

Go to the documentation of this file.

const int kNumPlots = 11;

struct Plot
{
  std::string label;
  std::string fname;
  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", Binning::Simple(25,73,373), kSliceTime},
      {"Run", "run", Binning::Simple(15000, 15000, 30000), SIMPLEVAR(hdr.run)},
      {"Time", "time", Binning::Simple(nDays<3 ? 24*nDays : nDays,timestamp - 86400*nDays,timestamp), SIMPLEVAR(hdr.unixtime)},
      {"Max y [m]", "maxy", kXYBins, kBoxMaxY},
      {"Slice time [#mus]", "tighttus", Binning::Simple(360,205,241), kSliceTime},
      {"No. Kalman tracks", "nkal", Binning::Simple(10, 0, 10), SIMPLEVAR(trk.kalman.ntracks)},
      {"No. Hits", "nhit", Binning::Simple(10, 0, 500), kNHit},
      {"Kalman angle", "angkal", Binning::Simple(5, 0, 1), SIMPLEVAR(sel.cosrej.anglekal)},
      {"Z direction", "dirz", Binning::Simple(5, 0.5, 1), kDirZ},
      {"BDT", "bdt", Binning::Simple(20, 0, 1), kNumuContPID},
      {"p_{T}/p", "ptp", Binning::Simple(20, 0, 1), kPtP}
    };


  std::cout << "Setting loaders ..." << std::endl;
  //SpectrumLoader fd_nonswap("prod_caf_R17-11-14-prod4reco.e_fd_genie_nonswap_rhc_nova_v08_full_v1");
  //SpectrumLoader fd_fluxswap("prod_caf_R17-11-14-prod4reco.e_fd_genie_fluxswap_rhc_nova_v08_full_v1");
  SpectrumLoader fd_nonswap("prod_caf_R17-11-14-prod4reco.d_fd_genie_nonswap_fhc_nova_v08_full_v1");
  SpectrumLoader fd_fluxswap("prod_caf_R17-11-14-prod4reco.d_fd_genie_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*kXSecCVWgt2018);
  }
  
  Spectrum nonSwapDenom("NHits", Binning::Simple(100, 0, 1000), fd_nonswap, kNHit, kNoCut, kNoShift, kWOscDumbHere);
  Spectrum swapDenom("NHits", Binning::Simple(100, 0, 1000), fd_fluxswap, 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->mkdir( (p.fname).c_str() ));
  }

  nonSwapDenom.SaveTo(file->mkdir("nonSwapDenom"));
  swapDenom.SaveTo(file->mkdir("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;
  //fFiles = "defname: " + samdef;
  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);

  SpectrumLoader loader(fFiles);

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

  Plot plots[kNumPlots] =
    {
      {"Slice time [#mus]", "tus", Binning::Simple(25,73,373), kSliceTime},
      {"Run", "run", Binning::Simple(15000, 15000, 30000), SIMPLEVAR(hdr.run)},
      {"Time", "time", Binning::Simple(nDays<3 ? 24*nDays : nDays,timestamp - 86400*nDays,timestamp), SIMPLEVAR(hdr.unixtime)},
      {"Max y [m]", "maxy", kXYBins, kBoxMaxY},
      {"Slice time [#mus]", "tighttus", Binning::Simple(360,205,241), kSliceTime},
      {"No. Kalman tracks", "nkal", Binning::Simple(10, 0, 10), SIMPLEVAR(trk.kalman.ntracks)},
      {"No. Hits", "nhit", Binning::Simple(10, 0, 500), kNHit},
      {"Kalman angle", "angkal", Binning::Simple(5, 0, 1), SIMPLEVAR(sel.cosrej.anglekal)},
      {"Z direction", "dirz", Binning::Simple(5, 0.5, 1), kDirZ},
      {"BDT", "bdt", Binning::Simple(20, 0, 1), kNumuContPID},
      {"p_{T}/p", "ptp", Binning::Simple(20, 0, 1), 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, Binning::Simple(nDays<3 ? 24*nDays : nDays,timestamp - 86400*nDays,timestamp), SIMPLEVAR(hdr.unixtime), Binning::Simple(500,0,50e12),kMassFracPOT, kNoCut, kNoShift, kUnweighted);
  Spectrum massDenom("", Binning::Simple(1, 0, 1), loader, kUnweighted, kNoCut, kNoShift, kSpillPOT);
  Spectrum massNum("", 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->mkdir( ("data_"+p.fname).c_str() ));
    hBkg[i]->SaveTo(file->mkdir( ("bkg_"+p.fname).c_str() ));    
  }

  potTime.SaveTo(file->mkdir("potTime"));
  massDenom.SaveTo(file->mkdir("massDenom"));
  massNum.SaveTo(file->mkdir("massNum"));

  file->Close();

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

}// getData