BCModelManager.h

#ifndef __BCMODELMANAGER__H
#define __BCMODELMANAGER__H

/*
 * Copyright (C) 2007-2018, the BAT core developer team
 * All rights reserved.
 *
 * For the licensing terms see doc/COPYING.
 * For documentation see http://mpp.mpg.de/bat
 */

// ---------------------------------------------------------

#include "BCDataSet.h"
#include "BCModel.h"

#include <string>

// ---------------------------------------------------------

class BCModelManager
{
public:

    BCModelManager();

    BCModelManager(const BCModelManager& other);

    virtual ~BCModelManager();

    BCModelManager& operator=(BCModelManager modelmanager);

    friend void swap(BCModelManager& A, BCModelManager& B);

    unsigned int GetNModels()
    { return fModels.size(); }

    BCModel* GetModel(unsigned index)
    { return (index < fModels.size()) ? fModels[index] : 0; }

    BCDataSet* GetDataSet()
    { return fDataSet; };

    void SetDataSet(BCDataSet* dataset);

    void SetPrecision(BCEngineMCMC::Precision precision);

    void SetNIterationsMax(int niterations);

    void SetNIterationsMin(int niterations);

    void SetNIterationsPrecisionCheck(int niterations);

    void SetMarginalizationMethod(BCIntegrate::BCMarginalizationMethod method);

    void SetIntegrationMethod(BCIntegrate::BCIntegrationMethod method);

    void SetOptimizationMethod(BCIntegrate::BCOptimizationMethod method);

    void SetRelativePrecision(double relprecision);

    void SetAbsolutePrecision(double absprecision);

    void SetNbins(unsigned int n);

    void SetNChains(unsigned int n);

    void AddModel(BCModel* model, double prior_probability = 0.);

    void Integrate();

    double BayesFactor(const unsigned int imodel1, const unsigned int imodel2) const;

    void FindMode();

    void MarginalizeAll();

    void WriteMarkovChain(bool flag);

    void WriteMarkovChainRun(bool flag);

    void WriteMarkovChainPreRun(bool flag);

    void WriteMarkovChain(const std::string& prefix, const std::string& option, bool flag_run = true, bool flag_prerun = true);

    void PrintModelComparisonSummary() const;

    void PrintSummary() const;

private:

    std::vector<BCModel*> fModels;

    std::vector<double> fAPrioriProbability;

    std::vector<double> fAPosterioriProbability;

    BCDataSet* fDataSet;

};

// ---------------------------------------------------------

#endif