MakeCovMx.C

Go to the documentation of this file.

/// Macro to load the manager class and use it to produce and save a covariance matrix
#include "CAFAna/Core/LoadFromFile.h"
#include "CAFAna/Analysis/CovMxManager.h"
#include "NuXAna/macros/Nus20/Utilities.h"

#include "TFile.h"

using namespace ana;

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

using covmx::CovarianceMatrix;
using covmx::Sample;

void MakeCovMx(TString opt) {

  try {

    string filePath = InputPath() + "/covmx/mgr.root";
    TFile* inFile = TFile::Open(filePath.c_str(), "read");

    vector<Sample> samples = GetSamplesFromOptString(opt);
    vector<const ISyst*> systs = GetAllSysts();
    auto mgr = LoadFrom<CovMxManager>(inFile);
    CovarianceMatrix* mx = mgr->GetCovarianceMatrix(samples, systs);

    TFile* outFile = TFile::Open("covmx.root", "recreate");
    ostringstream dirName;
    for (size_t i = 0; i < samples.size(); ++i) {
      if (i > 0) dirName << "_";
      dirName << samples[i].GetID();
    }
    mx->SaveTo(outFile->mkdir(dirName.str().c_str()));
    delete outFile; 

  } catch(const exception& e) {
    cout << endl << "Exception occurred! " << e.what() << endl;
  }

}