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 std::string GetSampleTitle(const int Sample) const = 0;
  virtual std::string GetName() const = 0;
  virtual double GetSampleLikelihood(const int isample) const = 0;
  virtual void CleanMemoryBeforeFit() = 0;
  virtual void SaveAdditionalInfo(TDirectory* Dir) {(void) Dir;};
  MaCh3Modes* GetMaCh3Modes() const { return Modes.get(); }
      
  virtual void Reweight()=0;
  virtual double GetLikelihood() const = 0;
 
  virtual void PrintRates(const bool DataOnly = false) = 0;
 
  unsigned int GetNEvents() const {return nEvents;}
  virtual int GetNOscChannels(const int iSample) const = 0;
 
  virtual std::string GetKinVarName(const int iSample, const int Dimension) const = 0;
 
  virtual TH1* GetDataHist(const int Sample) = 0;
  virtual TH1* GetMCHist(const int Sample) = 0;
  virtual TH1* GetW2Hist(const int Sample) = 0;
 
  virtual int GetNDim(const int Sample) const = 0;
  virtual std::string GetFlavourName(const int iSample, const int iChannel) const = 0;
 
  virtual std::vector<double> ReturnKinematicParameterBinning(const int Sample, const std::string &KinematicParameter) const = 0;
 
  virtual TH1* Get1DVarHistByModeAndChannel(const int iSample, const std::string& ProjectionVar_Str,
                                            int kModeToFill = -1, int kChannelToFill = -1, int WeightStyle = 0, TAxis* Axis = nullptr) = 0;
 
  virtual TH2* Get2DVarHistByModeAndChannel(const int iSample, const std::string& ProjectionVar_StrX,
                                            const std::string& ProjectionVar_StrY, int kModeToFill = -1,
                                            int kChannelToFill = -1, int WeightStyle = 0,
                                            TAxis* AxisX = nullptr, TAxis* AxisY = nullptr) = 0;
  virtual TH1 *Get1DVarHist(const int iSample, const std::string &ProjectionVar,
                           const std::vector<KinematicCut> &EventSelectionVec = {}, int WeightStyle = 0,
                            TAxis *Axis = nullptr, const std::vector<KinematicCut> &SubEventSelectionVec = {}) = 0;
  virtual TH2* Get2DVarHist(const int iSample, const std::string& ProjectionVarX, const std::string& ProjectionVarY,
                            const std::vector< KinematicCut >& EventSelectionVec = {},
                            int WeightStyle = 0, TAxis* AxisX = nullptr, TAxis* AxisY = nullptr,
                            const std::vector< KinematicCut >& SubEventSelectionVec = {}) = 0;
 
  double GetPoissonLLH(const double data, const double mc) const;
 
  double GetTestStatLLH(const double data, const double mc, const double w2) const;
 
  void SetTestStatistic(TestStatistic testStat){ fTestStatistic = testStat; }
  TestStatistic GetTestStatistic() const { return fTestStatistic; }
 
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;
};