RHatCalculator.h

Go to the documentation of this file.

#pragma once
 
// MaCh3 includes
#include "Fitters/StatisticalUtils.h"
#include "Samples/HistogramUtils.h"
#include "Parameters/ParameterHandlerUtils.h"
 
class RHatCalculator {
  public:
    RHatCalculator(bool HighMemory, std::vector<std::string>& Inputs, int entries);
    virtual ~RHatCalculator();
 
    void RunDiagnostic();
  private:
    void PrepareChains_HighMem();
    void PrepareChains();
 
    void InitialiseArrays();
 
    void CalcRhat_HighMem();
    void CalcRhat();
 
    void SaveResults();
 
    bool HighMemoryMode;
    int nDraw;
    int Nchains;
 
    std::vector<TString> BranchNames;
    std::vector<std::string> MCMCFile;
    std::vector<bool> ValidPar;
 
    std::vector<std::vector<double>> Mean;
    std::vector<std::vector<double>> StandardDeviation;
 
    std::vector<double> MeanGlobal;
    std::vector<double> StandardDeviationGlobal;
 
    std::vector<double> BetweenChainVariance;
    std::vector<double> MarginalPosteriorVariance;
    std::vector<double> RHat;
    std::vector<double> EffectiveSampleSize;
 
    int Ntoys;
    std::vector<std::vector<std::vector<double>>> Draws;
    std::vector<std::vector<std::vector<double>>> DrawsFolded;
 
    std::vector<double> MedianArr;
 
    std::vector<std::vector<double>> MeanFolded;
    std::vector<std::vector<double>> StandardDeviationFolded;
 
    std::vector<double> MeanGlobalFolded;
    std::vector<double> StandardDeviationGlobalFolded;
 
    std::vector<double> BetweenChainVarianceFolded;
    std::vector<double> MarginalPosteriorVarianceFolded;
    std::vector<double> RHatFolded;
    std::vector<double> EffectiveSampleSizeFolded;
 
    std::vector<int> Ntoys_requested;
    std::vector<int> Ntoys_filled;
    int TotToys;
    unsigned int NThin;
 
    std::vector<double> S1_global;
    std::vector<double> S2_global;
    std::vector<std::vector<double>> S1_chain;
    std::vector<std::vector<double>> S2_chain;
};