DrawCovMx.C

Go to the documentation of this file.

#include "CAFAna/Analysis/CovMxManager.h"
#include "NuXAna/macros/Nus20/Utilities.h"

#include "TCanvas.h"

using namespace ana;

using std::ostringstream;
using std::string;
using std::vector;

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

void DrawCovMx() {

  try {

    DontAddDirectory guard;

    string opt = "numusel_nuesel_rhc_fhc_neardet_fardet";

    TFile* plotFile = TFile::Open(("plots/covmx/plots_"+opt+".root").c_str(), "recreate");
    TDirectory* systDir = plotFile->mkdir("systs");

    vector<Sample> samples = GetSamplesFromOptString(opt.c_str());
    for (Sample& s : samples) SetEverything(s);
    vector<const ISyst*> systs = GetAllSysts();
    auto calc = DefaultSterileCalc(4);
    SetNus20Params(calc);
    ostringstream tag;
    for (size_t i = 0; i < samples.size(); ++i) {
      if (i > 0) tag << "_";
      tag << samples[i].GetID();
    }
    TCanvas c("c", "c", 1600, 900);
    TFile* inFile = TFile::Open((InputPath()+"/covmx/mx3flav.root").c_str(), "read");
    TDirectory* dir = inFile->GetDirectory(tag.str().c_str());
    CovarianceMatrix* mx = LoadFrom<CovarianceMatrix>(dir).release();
    TH2* hMx = mx->GetCovMxRelativeTH2(samples, calc);
    plotFile->WriteTObject(hMx, "MxScaled");
    hMx->Draw("colz");
    c.SaveAs("plots/covmx/MxScaled.png");
    delete hMx;
    hMx = mx->GetFullCovMxTH2();
    plotFile->WriteTObject(hMx, "MxFull");
    hMx->Draw("colz");
    c.SaveAs("plots/covmx/MxFull.png");
    delete hMx;
    hMx = mx->GetFullCorrMxTH2();
    plotFile->WriteTObject(hMx, "MxCorrFull");
    hMx->Draw("colz");
    c.SaveAs("plots/covmx/MxFullCorr.png");
    delete hMx;

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

}