SampleHandlerBase.h

Go to the documentation of this file.

#pragma once
 
//C++ includes
#include <assert.h>
 
//MaCh3 includes
#include "Samples/SampleStructs.h"
#include "Samples/HistogramUtils.h"
#include "Manager/Manager.h"
#include "Manager/MaCh3Modes.h"
 
_MaCh3_Safe_Include_Start_ //{
//ROOT includes
#include "TTree.h"
#include "TH1D.h"
#include "TH2D.h"
#include "TMath.h"
#include "TFile.h"
#include "TROOT.h"
#include "TRandom3.h"
#include "TString.h"
_MaCh3_Safe_Include_End_ //}
 
class SampleHandlerBase
{
 public:
   SampleHandlerBase();
  virtual ~SampleHandlerBase();
 
 
 
  virtual inline M3::int_t GetNsamples(){ return nSamples; };
  virtual inline std::string GetTitle()const {return "SampleHandler";};
  virtual std::string GetSampleName(int Sample) const = 0;
  virtual inline double GetSampleLikelihood(const int isample){(void) isample; return GetLikelihood();};
  virtual void CleanMemoryBeforeFit() = 0;
  virtual void SaveAdditionalInfo(TDirectory* Dir) {(void) Dir;};
  MaCh3Modes* GetMaCh3Modes() const { return Modes.get(); }
      
  virtual void Reweight()=0;
  virtual double GetLikelihood() = 0;
 
  unsigned int GetNEvents() const {return nEvents;}
  virtual int GetNMCSamples() { return nSamples; }
  virtual int GetNOscChannels(){ return 1; }
 
  // WARNING KS: Needed for sigma var
  virtual void SetupBinning(const M3::int_t Selection, std::vector<double> &BinningX, std::vector<double> &BinningY){
    (void) Selection; (void) BinningX; (void) BinningY; throw MaCh3Exception(__FILE__ , __LINE__ , "Not implemented");}
  virtual TH1* GetData(const int Selection) { (void) Selection; throw MaCh3Exception(__FILE__ , __LINE__ , "Not implemented"); }
  virtual TH2Poly* GetW2(const int Selection){ (void) Selection; throw MaCh3Exception(__FILE__ , __LINE__ , "Not implemented");}
  virtual TH1* GetPDF(const int Selection){ (void) Selection; throw MaCh3Exception(__FILE__ , __LINE__ , "Not implemented");}
  virtual inline TH1* GetPDFMode(const int Selection, const int Mode) {
    (void) Selection; (void) Mode; throw MaCh3Exception(__FILE__ , __LINE__ , "Not implemented"); }
  virtual inline std::string GetKinVarLabel(const int sample, const int Dimension) {
    (void) sample; (void) Dimension; throw MaCh3Exception(__FILE__ , __LINE__ , "Not implemented");  };
 
  double GetTestStatLLH(double data, double mc) const;
  double GetTestStatLLH(const double data, const double mc, const double w2) const;
  inline void SetTestStatistic(TestStatistic testStat){ fTestStatistic = testStat; }
 
protected:
  void QuietPlease();
  void NowTalk();
 
  template <typename T>
  bool MatchCondition(const std::vector<T>& allowedValues, const T& value) {
    if (allowedValues.empty()) {
      return true;  // Apply to all if no specific values are specified
    }
    return std::find(allowedValues.begin(), allowedValues.end(), value) != allowedValues.end();
  }
 
  TestStatistic fTestStatistic;
 
  std::streambuf *buf;
  std::streambuf *errbuf;
 
  M3::int_t nSamples;
 
  unsigned int nEvents;
 
  std::unique_ptr<MaCh3Modes> Modes;
};