MakeISysts.C

Go to the documentation of this file.

//////////////////////////////////////////////////////////////
// MakeISysts.C
// Jeremy Hewes (jhewes15@fnal.gov)
//
// Make ISysts from SystMakers for 2020 FHC NC analysis
//////////////////////////////////////////////////////////////

#include "TFile.h"

#include "CAFAna/Core/LoadFromFile.h"
#include "CAFAna/Analysis/Exposures.h"
#include "3FlavorAna/Systs/CovMxSysts.h"
#include "NuXAna/Systs/CovMxSysts.h"
#include "NuXAna/Analysis/SystMaker.h"

#include "NuXAna/macros/Nus20/Utilities.h"

using namespace ana;

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

using covmx::Sample;

// -------------------------------------------------------------------------------------
void MakeISysts(string inFileName = "") {

  try {

    if (inFileName.empty()) inFileName = kNus20Path+"/systs/systs.root";
    TFile* inFile  = TFile::Open(inFileName.c_str(), "read");
    std::string outFileName = "isysts.root";
    TFile* outFile = TFile::Open(outFileName.c_str(), "recreate");

    auto calc = DefaultSterileCalc(4);
    SetNus20Params(calc);

    TIter next(inFile->GetListOfKeys());
    TKey* key;
    while ((key = (TKey*)next())) {

      Sample sample = GetSampleFromOptString(key->GetName());
      cout << "Generating ISysts for sample " << sample.GetTag() << endl;
      SetInfo(sample);

      SystematicsMaker* systsMaker = SystematicsMaker::LoadFrom((TDirectory*)key->ReadObj()).release();
 
      // Run the syst makers
      systsMaker->Go(sample, calc);

      // Get and save the isysts
      vector<NuISyst*> systs = systsMaker->MakeISysts();
      TDirectory* dir = outFile->mkdir(("isysts_"+sample.GetTag()).c_str());
      for (NuISyst* syst : systs) {
        if (sample.selection == covmx::kNC)
          syst->SaveTo(dir->mkdir((sample.GetTag()+"_"+syst->BaseName()).c_str()), "NCSyst");
        else if (sample.selection == covmx::kCCNumu)
          syst->SaveTo(dir->mkdir((sample.GetTag()+"_"+syst->BaseName()).c_str()), "NumuSyst");
        else if (sample.selection == covmx::kCCNue)
          syst->SaveTo(dir->mkdir((sample.GetTag()+"_"+syst->BaseName()).c_str()), "NueSyst");
      }
    }

    delete inFile;
    delete outFile; 

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

}