MergePredictions.C

Go to the documentation of this file.

#include "CAFAna/Core/LoadFromFile.h"
#include "CAFAna/Core/Sample.h"
#include "CAFAna/Prediction/PredictionInterp.h"

#include "NuXAna/Prediction/NDPredictionSterile.h"
#include "NuXAna/Prediction/FDPredictionSterile.h"
#include "NuXAna/macros/Nus20/Utilities.h"

using namespace ana;

using std::cout;
using std::endl;
using std::string;

using covmx::Sample;

void CopyPred(TDirectory* in, TDirectory* out, Sample s) {

  cout << "Moving prediction " << s.GetTag() << "...";

  time_t before, after;
  time(&before);

  TDirectory* predDir = (s.selection == covmx::kCCNue) ?
    in->GetDirectory(("pred_"+s.GetTag()).c_str()) :
    in->GetDirectory(s.GetTag().c_str())->GetDirectory("pred");
  IPrediction* pred = (s.detector == covmx::kNearDet) ?
      (IPrediction*)LoadFrom<NDPredictionSterile>(predDir).release()
    : (IPrediction*)LoadFrom<FDPredictionSterile>(predDir).release();
  pred->SaveTo(out->mkdir("pred"));
  delete pred;
  predDir = (s.selection == covmx::kCCNue) ?
    in->GetDirectory(("pred_interp_"+s.GetTag()).c_str()) :
    in->GetDirectory(s.GetTag().c_str())->GetDirectory("pred_interp");
  LoadSysts(s);
  GetXsecSysts();
  pred = LoadFrom<PredictionInterp>(predDir).release();
  pred->SaveTo(out->mkdir("pred_interp"));
  delete pred;

  time(&after);
  double seconds = difftime(after, before);
  cout << " done in " << seconds << " seconds." << endl;

}

void MergePredictions(TString opt) {

  string path = "/nova/data/users/jhewes15/";

  TFile* outFile = TFile::Open((path+"predictions.root").c_str(), "update");

  Sample s = GetSampleFromOptString(opt);
  string inFileName = (s.selection == covmx::kCCNue) ? "pred.root" : "pred_existing.root";
  TFile* inFile = TFile::Open((path+inFileName).c_str(), "read");
  CopyPred(inFile, outFile->mkdir(s.GetTag().c_str()), s);

  delete inFile;
  delete outFile;
}