PredictiveThrower.h

Go to the documentation of this file.

#pragma once
 
#include "Fitters/FitterBase.h"
 
class ParameterHandlerGeneric;
class BinningHandler;
class SampleHandlerFD;
 
// ***************************
struct PredictiveSample {
// ***************************
  // Name of sample
  std::string Name;
  const SampleHandlerFD* SamHandler;
  const BinningHandler* Binning;
  int LocalId;
  int Dimenstion;
};
 
 
class PredictiveThrower : public FitterBase {
 public:
  PredictiveThrower(Manager * const fitMan);
  virtual ~PredictiveThrower();
 
  void ProduceToys();
 
  void RunPredictiveAnalysis();
 
  void RunMCMC() override {
    MACH3LOG_ERROR("{} is not supported in {}", __func__, GetName());
    throw MaCh3Exception(__FILE__ , __LINE__ );
  };
 
 private:
  void SetParamters();
 
  void SetupToyGeneration();
 
  bool LoadToys();
 
  void SetupSampleInformation();
 
  std::vector<std::string> GetStoredFancyName(ParameterHandlerBase* Systematics) const;
 
  std::vector<std::unique_ptr<TH1>> MakePredictive(const std::vector<std::vector<std::unique_ptr<TH1>>>& Toys,
                                                   const std::vector<TDirectory*>& Director,
                                                   const std::string& suffix,
                                                   const bool DebugHistograms);
 
  std::vector<std::vector<std::unique_ptr<TH2D>>> ProduceSpectra(
                                                      const std::vector<std::vector<std::unique_ptr<TH1>>>& Toys,
                                                      const std::vector<TDirectory*>& Director,
                                                      const std::string suffix);
 
  void PosteriorPredictivepValue(const std::vector<std::unique_ptr<TH1>>& PostPred_mc,
                                 const std::vector<TDirectory*>& SampleDir);
 
 
  double GetLLH(const std::unique_ptr<TH1>& DatHist,
                                   const std::unique_ptr<TH1>& MCHist,
                                   const std::unique_ptr<TH1>& W2Hist,
                                   const SampleHandlerBase* SampleHandler);
 
  void MakeChi2Plots(const std::vector<std::vector<double>>& Chi2_x,
                     const std::string& Chi2_x_title,
                     const std::vector<std::vector<double>>& Chi2_y,
                     const std::string& Chi2_y_title,
                     const std::vector<TDirectory*>& SampleDir,
                     const std::string Title);
 
 
  void StudyBetaParameters(TDirectory* PredictiveDir);
 
  bool FullLLH;
  int NModelParams;
  bool Is_PriorPredictive;
 
  int TotalNumberOfSamples;
 
  std::vector<PredictiveSample> SampleInfo;
 
  int Ntoys;
  std::vector<std::string> ParameterGroupsNotVaried;
  std::unordered_set<int> ParameterOnlyToVary;
 
  ParameterHandlerGeneric* ModelSystematic;
 
  std::vector<std::unique_ptr<TH1>> Data_Hist;
  std::vector<std::unique_ptr<TH1>> MC_Nom_Hist;
  std::vector<std::unique_ptr<TH1>> W2_Nom_Hist;
 
  std::vector<std::vector<std::unique_ptr<TH1>>> MC_Hist_Toy;
  std::vector<std::vector<std::unique_ptr<TH1>>> W2_Hist_Toy;
 
  std::vector<double> ReweightWeight;
  std::vector<double> PenaltyTerm;
};