MaCh3Factory.h

Go to the documentation of this file.

#pragma once
 
// MaCh3 includes
#include "Algorithms/FitterBase.h"
#include "Algorithms/MR2T2.h"
#include "Algorithms/DelayedMR2T2.h"
#include "Algorithms/PSO.h"
#include "Algorithms/LikelihoodFit.h"
#ifdef MaCh3_MINUIT2
#include "Algorithms/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, __FILE__ , __LINE__);
  // do we pca whole matrix or only submatrix
  auto PCAParamRegion = GetFromManager<std::vector<int>>(Settings[std::string(PreFix) + "PCAParams"], {-999, -999}, __FILE__ , __LINE__);
 
  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"], {}, __FILE__ , __LINE__);
 
  // Fixed CovObject parameters loop
  if (FixParams.size() == 1 && FixParams.at(0) == "All") {
    CovObject->SetFixAllParameters();
  } else {
    for (unsigned int j = 0; j < FixParams.size(); j++) {
      CovObject->SetFixParameter(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);
    Handlers[i] = Sample;
  }
  return Handlers;
}