SampleSummary.h

Go to the documentation of this file.

#pragma once
 
// MaCh3 includes
#include "Samples/SampleHandlerBase.h"
#include "Fitters/StatisticalUtils.h"
#include "Fitters/MCMCProcessor.h"
 
namespace M3 {
  enum kInfCrit {
    kBIC,      
    kDIC,      
    kWAIC,     
    kInfCrits  
  };
}
// *******************
class SampleSummary {
// *******************
  public:
    SampleSummary(const int n_Samples, const std::string &Filename, SampleHandlerBase* const sample, const int nSteps);
    ~SampleSummary();
 
    void AddData(std::vector<TH2Poly*> &DataHist);
    void AddNominal(std::vector<TH2Poly*> &NominalHist, std::vector<TH2Poly*> &W2Nom);
    void AddThrow(std::vector<TH2Poly*> &MCHist, std::vector<TH2Poly*> &W2Hist, const double LLHPenalty = 0.0, const double Weight = 1.0, const int DrawNumber = 0);
    void AddThrowByMode(std::vector<std::vector<TH2Poly*>> &SampleVector_ByMode);
 
    void Write();
 
    inline void SetLikelihood(const TestStatistic TestStat){ likelihood = TestStat;};
    inline void SetNModelParams(const int nPars){ nModelParams = nPars;};
 
  private:
    inline void MakePredictive();
 
    inline void PrepareOutput();
 
    inline void CalcLLH(TH2Poly * const & Data, TH2Poly * const & MC, TH2Poly * const & W2);
    inline void CalcLLH(TH1D * const & Data, TH1D * const & MC, TH1D * const & W2);
 
    inline double GetLLH(TH2Poly * const & Data, TH2Poly * const & MC, TH2Poly * const & W2);
    inline double GetLLH(TH1D * const & Data, TH1D * const & MC, TH1D * const & W2);
 
    inline void PlotBetaParameters();
 
    inline void StudyKinematicCorrelations();
 
    inline void MakeCutLLH();
    // Make the 1D cut distribution and give the 1D p-value
    inline void MakeCutLLH1D(TH1D *Histogram, double llh_ref = -999);
    inline void MakeCutEventRate(TH1D *Histogram, const double DataRate);
    inline void MakeChi2Hists();
 
    inline bool CheckSamples(const int Length);
 
    inline TH1D* ProjectHist(TH2D* Histogram, const bool ProjectX);
    inline TH1D* ProjectPoly(TH2Poly* Histogram, const bool ProjectX, const int selection, const bool MakeErrorHist = false);
 
    inline void MakeFluctuatedHistogram(TH1D *FluctHist, TH1D* PolyHist);
 
    inline void MakeFluctuatedHistogram(TH2Poly *FluctHist, TH2Poly* PolyHist);
 
    inline void StudyInformationCriterion(M3::kInfCrit Criterion);
 
    inline void StudyBIC();
 
    inline void StudyDIC();
 
    inline void StudyWAIC();
 
    std::unique_ptr<TRandom3> rnd;
    bool first_pass;
 
    bool StandardFluctuation;
 
    std::vector<std::vector<TH2Poly*>> MCVector;
    std::vector<std::vector<TH2Poly*>> W2MCVector;
    std::vector<std::vector<std::vector<TH2Poly*>>> MCVectorByMode;
 
    std::vector<double> LLHPenaltyVector;
    std::vector<double> WeightVector;
 
    int nSamples;
 
    std::vector<std::string> SampleNames;
 
    std::vector<std::vector<std::unique_ptr<TH1D>>> PosteriorHist;
    std::vector<std::vector<std::unique_ptr<TH1D>>> w2Hist;
 
    std::vector<TH2D*> ViolinHists_ProjectX;
    std::vector<TH2D*> ViolinHists_ProjectY;
    
    std::vector<TH2Poly*> DataHist;
    std::vector<TH1D*> DataHist_ProjectX;
    std::vector<TH1D*> DataHist_ProjectY;
    std::vector<TH2Poly*> NominalHist;
    std::vector<TH2Poly*> W2NomHist;
    std::vector<TH2Poly*> W2MeanHist;
    std::vector<TH2Poly*> W2ModeHist;
 
    std::unique_ptr<TH1D> lnLHist;
    std::unique_ptr<TH1D> lnLHist_drawfluc;
    std::unique_ptr<TH1D> lnLHist_drawflucdraw;
    std::unique_ptr<TH1D> lnLHist_drawdata;
    std::unique_ptr<TH2D> lnLDrawHist;
    std::unique_ptr<TH2D> lnLFlucHist;
 
    std::unique_ptr<TH2D> lnLDrawHistRate;
    std::unique_ptr<TH2D> lnLFlucHist_ProjectX;
 
    std::vector<TH1D*> lnLHist_Sample_DrawData;
    std::vector<TH1D*> lnLHist_Sample_DrawflucDraw;
    std::vector<TH1D*> lnLHist_Sample_PredflucDraw;
 
    std::vector<TH2Poly*> lnLHist_Mean;
    std::vector<TH2Poly*> lnLHist_Mode;
 
    std::vector<TH1D*> lnLHist_Mean_ProjectX;
 
    std::vector<TH2Poly*> MeanHist;
    std::vector<TH2Poly*> MeanHistCorrected;
    std::vector<TH2Poly*> ModeHist;
 
    std::vector<TH1D*> lnLHist_Mean1D;
    std::vector<TH1D*> lnLHist_Mode1D;
 
    std::unique_ptr<TH1D> RandomHist;
 
    std::vector<std::vector<std::unique_ptr<TH1D>>> BetaHist;
    bool DoBetaParam;
 
    unsigned int nChainSteps;
 
    bool isPriorPredictive;
 
    bool doShapeOnly;
 
    unsigned int nThrows;
 
    std::vector<int> maxBins;
 
    double llh_total;
 
    std::string OutputName;
    TFile *Outputfile;
    std::vector<TDirectory*> Dir;
 
    TTree *OutputTree;
    std::vector<double> llh_data_draw;
    std::vector<double> llh_drawfluc_draw;
    std::vector<double> llh_predfluc_draw;
 
    std::vector<double> llh_rate_data_draw;
    std::vector<double> llh_rate_predfluc_draw;
 
    std::vector<double> llh_data_drawfluc;
    std::vector<double> llh_data_predfluc;
    std::vector<double> llh_draw_pred;
    std::vector<double> llh_drawfluc_pred;
 
    std::vector<double> llh_predfluc_pred;
    std::vector<double> llh_drawfluc_predfluc;
    std::vector<double> llh_datafluc_draw;
 
    std::vector<double> llh_data_draw_ProjectX;
    std::vector<double> llh_drawfluc_draw_ProjectX;
 
    double llh_penalty;
 
    double total_llh_data_draw;
    double total_llh_drawfluc_draw;
    double total_llh_predfluc_draw;
 
    double total_llh_rate_data_draw;
    double total_llh_rate_predfluc_draw;
 
    double total_llh_data_predfluc;
    double total_llh_data_drawfluc;
    double total_llh_draw_pred;
    double total_llh_drawfluc_pred;
    double total_llh_drawfluc_predfluc;
    double total_llh_datafluc_draw;
    double total_llh_predfluc_pred;
 
    double total_llh_data_draw_ProjectX;
    double total_llh_drawfluc_draw_ProjectX;
 
    bool DoByModePlots;
    std::vector<std::vector<TH2Poly*>> MeanHist_ByMode;
    TH1D ****PosteriorHist_ByMode;
    
    SampleHandlerBase* SampleHandler;
 
    MaCh3Modes* Modes;
 
    TestStatistic likelihood;
 
    int nModelParams;
 
    int Debug;
};