MaCh3Factory.h

Go to the documentation of this file.

#pragma once
 
// MaCh3 includes
#include "Fitters/FitterBase.h"
#include "Fitters/MR2T2.h"
#include "Fitters/DelayedMR2T2.h"
#include "Fitters/PSO.h"
#include "Fitters/LikelihoodFit.h"
#ifdef MaCh3_MINUIT2
#include "Fitters/MinuitFit.h"
#endif
 
 
#include "Parameters/ParameterHandlerGeneric.h"
 
 
std::unique_ptr<FitterBase> MaCh3FitterFactory(Manager *FitManager);
 
std::unique_ptr<Manager> MaCh3ManagerFactory(int argc, char **argv);
 
// ********************************************
template <typename CovType>
std::unique_ptr<CovType> MaCh3CovarianceFactory(Manager *FitManager, const std::string& PreFix) {
// ********************************************
  // config for our matrix
  YAML::Node Settings = FitManager->raw()["General"]["Systematics"];
  auto CovMatrixName = Get<std::string>(Settings[std::string(PreFix) + "CovName"], __FILE__, __LINE__);
  MACH3LOG_INFO("Initialising {} matrix", CovMatrixName);
 
  // yaml files initialising out matrix
  auto CovMatrixFile = Get<std::vector<std::string>>(Settings[std::string(PreFix) + "CovFile"], __FILE__, __LINE__);
 
  // PCA threshold, -1 means no pca
  auto PCAThreshold = GetFromManager<int>(Settings[std::string(PreFix) + "PCAThreshold"], -1);
  // do we pca whole matrix or only submatrix
  auto PCAParamRegion = GetFromManager<std::vector<int>>(Settings[std::string(PreFix) + "PCAParams"], {-999, -999});
 
  auto CovObject = std::make_unique<CovType>(CovMatrixFile, CovMatrixName, PCAThreshold, PCAParamRegion[0], PCAParamRegion[1]);
 
  // Fill the parameter values with their nominal values
  // should _ALWAYS_ be done before overriding with fix or flat
  CovObject->SetParameters();
 
  auto FixParams = GetFromManager<std::vector<std::string>>(Settings[std::string(PreFix) + "Fix"], {});
 
  // Fixed CovObject parameters loop
  if (FixParams.size() == 1 && FixParams.at(0) == "All") {
    for (int j = 0; j < CovObject->GetNumParams(); j++) {
      CovObject->ToggleFixParameter(j);
    }
  } else {
    for (unsigned int j = 0; j < FixParams.size(); j++) {
      CovObject->ToggleFixParameter(FixParams.at(j));
    }
  }
 
  if (CheckNodeExists(Settings, std::string(PreFix) + "Tune"))
  {
    CovObject->SetTune(Get<std::string>(Settings[std::string(PreFix) + "Tune"], __FILE__, __LINE__));
  }
 
  //Global step scale for matrix
  auto StepScale = Get<double>(Settings[std::string(PreFix) + "StepScale"], __FILE__, __LINE__);
 
  CovObject->SetStepScale(StepScale);
 
  // Adaptive MCMC stuff
  if (CheckNodeExists(FitManager->raw(), "AdaptionOptions")) {
    CovObject->InitialiseAdaption(FitManager->raw());
  }
  
  return CovObject;
}
 
// ********************************************
template <typename SampleType>
std::vector<SampleType*> MaCh3SampleHandlerFactory(const std::vector<std::string>& SampleConfig,
                                               ParameterHandlerGeneric* xsec) {
// ********************************************
  std::vector<SampleType*> Handlers(SampleConfig.size());
  for (size_t i = 0; i < SampleConfig.size(); ++i)
  {
    // Instantiate the sample using the specified class type
    SampleType* Sample = new SampleType(SampleConfig[i], xsec);
    Sample->Reweight();
 
    for(int iSample = 0; iSample < Sample->GetNsamples(); iSample++)
    {
      // Obtain sample name and create a TString version for histogram naming
      std::string name = Sample->GetSampleTitle(iSample);
      TString NameTString = TString(name.c_str());
 
      // Clone the 1D histogram with a modified name
      Sample->AddData(iSample, Sample->GetMCArray(iSample));
    }
    Handlers[i] = Sample;
  }
  return Handlers;
}