make_uDST.h

Go to the documentation of this file.

#include "NCUtils/NCType.h"
#include "NCUtils/NCRunUtil.h"
#include "JobControl/JobC.h"
#include "JobControl/JobCModule.h"
#include "Registry/Registry.h"
#include "Conventions/Detector.h"
#include "Conventions/BeamType.h"
#include "Conventions/ReleaseType.h"
#include "MCReweight/MCReweight.h"
#include "MCReweight/NeugenWeightCalculator.h"

#include <iostream>
#include <string>

enum EDataType {
  kUnknown=-1, kMC, kData, kMockData, kBlind, kElectron, kTau
};

Registry reg;

EDataType datatype=kUnknown;
Detector::Detector_t det=Detector::kUnknown;
BeamType::BeamType_t beamtype=BeamType::kUnknown;
NC::RunUtil::ERunType runPeriod=NC::RunUtil::kRunAll;
string month="";

//===========================================================

void set_month(string m)
{
  month=m;
}

//===========================================================

void set_detector(Detector::Detector_t _det)
{
  det=_det;
}

//===========================================================

// Options are kMC, kData, kMockData, kBlind
void set_datatype(EDataType t)
{
  datatype=t;
}

//===========================================================

void set_runperiod(NC::RunUtil::ERunType r)
{
  runPeriod=r;
}

//===========================================================

string get_designator(Detector::Detector_t d, EDataType t)
{
  if(d==Detector::kFar){
    switch(datatype){
    case kMC:       return "f210";
    case kData:     // Fallthrough
    case kMockData: // Fallthrough
    case kBlind:
                    return "F";
    case kElectron: return "f214";
    case kTau:      return "f213";
    default: assert(0 && "Something went horribly wrong");
    }
  } else if(d==Detector::kNear){
    if(t==kMC) return "n";
    else return "N";
  }

  assert(0 && "Something Wrong: unknown detector");
}

//===========================================================

void set_dir(string path)
{
  assert(det!=Detector::kUnknown);
  assert(datatype!=kUnknown);
  string runPeriodStr;
  switch(runPeriod){
  case NC::RunUtil::kRunI:
    runPeriodStr="r1"; break;
  case NC::RunUtil::kRunII:
    runPeriodStr="r2"; break;
  case NC::RunUtil::kRunIII:
    runPeriodStr="r3"; break;
  case NC::RunUtil::kRunAll:
    // This will happen for data
    runPeriodStr=""; break;
  default:
    assert(0 && "Unknown runPeriod");
  }
  string glob=path+"/"+
    get_designator(det, datatype)+"*"+runPeriodStr+"*.antp*.root";
  reg.Set("FilePath", glob.c_str());
}

//===========================================================

void set_mc_version(ReleaseType::Release_t r)
{
  reg.Set("MCVersion", r);
}

//===========================================================

void set_beamtype(BeamType::BeamType_t b)
{
  beamtype=b;
  reg.Set("BeamType", BeamType::AsString(beamtype));
}

//===========================================================

void set_output(string path, string tag="")
{
  assert(det!=Detector::kUnknown);
  assert(datatype!=kUnknown);
  assert(beamtype!=BeamType::kUnknown);
  // Make sure the run period is set for MC
  assert(runPeriod!=NC::RunUtil::kRunAll || datatype==kData);

  string fileName(path+"/");
  if (det == Detector::kFar) fileName += "far";
  else if (det==Detector::kNear) fileName += "near";
  else assert(0 && "Unknown detector. Have you called set_detector?");

  fileName += "_";
  fileName += BeamType::AsString(beamtype);
  fileName += "_";
  // Only add "run<n>" to the filename if we've been given a specific runperiod
  // This means that data doesn't get one
  if(runPeriod!=NC::RunUtil::kRunAll){
    // This relies on NC::RunUtil::kRunI == 1 etc, which is indeed the case
    fileName += Form("run%d_", (int)runPeriod);
  }

  switch(datatype){
  case kMC:
    fileName += "mc";
    break;
  case kElectron:
    fileName += "electron";
    break;
  case kTau:
    fileName += "tau";
    break;
  case kData:
    fileName += "data_"+month;
    break;
  case kMockData: 
    fileName += "mock";
    break;
  case kBlind:
    fileName += "blind";
    break;
  case kUnknown:
    assert(0 && "datatype not set");
  }

  fileName += tag+".uDST.root";

  reg.Set("FileName", fileName.c_str());
}

//===========================================================

void set_cuts(NCType::ECuts c)
{
  reg.Set("CutSuite", int(c));
}

//===========================================================

void do_extractions(string codenames)
{
  reg.Set("ExtractionsList", codenames.c_str());
}

//===========================================================

void set_file_limit(int limit)
{
  if(limit>0) reg.Set("FileCountLimit", limit);
}

// Extraction-specific things

//===========================================================

void set_pdf_training_file(TString s)
{
  reg.Set("ExtractionPDFTrainingFilePath", s.Data());
}

//===========================================================

void set_mda_mc_path(TString s)
{
  reg.Set("MDAMCPath", s.Data());
}

//===========================================================
void set_ann_use_lowE(bool useLowE)
{
  reg.Set("RPAnnUseLowETrain", useLowE);
}

//===========================================================

void run()
{

  // Check we have a meaningful combination of options
  if (det==Detector::kNear && (datatype==kTau || datatype==kElectron)){
    assert(0 && "Tau/Electron files don't exist for ND");
  }

  reg.Set("UseAll185i", false); //use all -185kA data
  reg.Set("ReadPDFs", true);
  reg.Set("RunPeriod", (int)runPeriod);

  // According to some error message, this is done somewhere else...
  //MCReweight &mcReweight = MCReweight::Instance();
  //NeugenWeightCalculator *neugenWeightCal = new NeugenWeightCalculator;
  //mcReweight.AddWeightCalculator(neugenWeightCal);

  JobC jc;

  // set message service verbosity levels before creating path
  // so that messages from the constructor don't get lost
  jc.Msg.SetLevel("MicroDSTMaker","Info");
  jc.Msg.SetLevel("NCAnalysisUtils","Info");
  jc.Msg.SetLevel("NCAnalysisCuts","Info");
  jc.Msg.SetLevel("NCUtils","Info");

  //Create path
  jc.Path.Create("Analysis", "MicroDSTMaker::Ana");

  JobCModule& jm=jc.Path("Analysis").Mod("MicroDSTMaker");

  jm.GetConfig().UnLockValues();
  jm.GetConfig() = reg;
  jm.Config(jm.GetConfig());
  jm.GetConfig().LockKeys();

  jc.Path("Analysis").Run();
  //Job Report
  jc.Path("Analysis").Report();
  //Get Message Statistics
  jc.Msg.Stats();
}

---