FitterBase.h

Go to the documentation of this file.

#pragma once
 
// C++ includes
#include <iostream>
#include <fstream>
#include <sstream>
 
// MaCh3 Includes
#include "Samples/SampleHandlerBase.h"
#include "Parameters/ParameterHandlerBase.h"
#include "Manager/Manager.h"
#include "Fitters/MCMCProcessor.h"
 
//KS: Joy of forward declaration https://gieseanw.wordpress.com/2018/02/25/the-joys-of-forward-declarations-results-from-the-real-world/
class TRandom3;
class TStopwatch;
class TTree;
class TGraphAsymmErrors;
class TDirectory;
 
class FitterBase {
 public:
  FitterBase(manager * const fitMan);
  virtual ~FitterBase();
 
  void AddSampleHandler(SampleHandlerBase* sample);
 
  void AddSystObj(ParameterHandlerBase* cov);
 
  virtual void RunMCMC() = 0;
 
  void DragRace(const int NLaps = 100);
 
  void RunLLHScan();
 
  void GetStepScaleBasedOnLLHScan();
 
  void Run2DLLHScan();
 
  void RunSigmaVar();
 
  void RunSigmaVarFD();
 
  virtual void StartFromPreviousFit(const std::string& FitName);
 
  inline std::string GetName() const {return AlgorithmName;};
 protected:
  void ProcessMCMC();
 
  void PrepareOutput();
 
  void SaveOutput();
 
  void SanitiseInputs();
 
  void SaveSettings();
 
  bool GetScaneRange(std::map<std::string, std::vector<double>>& scanRanges);
 
  bool CheckSkipParameter(const std::vector<std::string>& SkipVector, const std::string& ParamName) const;
 
 
  void CustomRange(const std::string& ParName, const double sigma, double& ParamShiftValue);
 
  manager *fitMan;
 
  unsigned int step;
  double logLCurr;
  double logLProp;
  double accProb;
  int accCount;
  unsigned int stepStart;
 
  std::vector<double> sample_llh;
  std::vector<double> syst_llh;
 
  std::vector<SampleHandlerBase*> samples;
  unsigned int TotalNSamples;
 
  std::vector<ParameterHandlerBase*> systematics;
 
  std::unique_ptr<TStopwatch> clock;
  std::unique_ptr<TStopwatch> stepClock;
  double stepTime;
 
  std::unique_ptr<TRandom3> random;
 
  TFile *outputFile;
  TDirectory *CovFolder;
  TDirectory *SampleFolder;
  TTree *outTree;
  int auto_save;
 
  bool fTestLikelihood;
 
  bool FileSaved;
  bool SettingsSaved;
  bool OutputPrepared;
 
  std::string AlgorithmName;
 
  #ifdef DEBUG
  bool debug;
  std::ofstream debugFile;
  #endif
};