BCModelManager.h

#ifndef __BCMODELMANAGER__H
#define __BCMODELMANAGER__H

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

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

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

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

class BCModelManager
{
public:

   BCModelManager();

   BCModelManager(const BCModelManager & modelmanager);

   virtual ~BCModelManager();

   BCModelManager & operator = (const BCModelManager & modelmanager);

   unsigned int GetNModels()
   { return fModelContainer -> size(); };

   BCModel * GetModel(int index)
   { return fModelContainer -> at(index); };

   int GetNDataPoints()
   { return (fDataSet) ? fDataSet -> GetNDataPoints() : 0; };

   BCDataPoint * GetDataPoint(int index)
   { return fDataSet -> GetDataPoint(index); };

   BCDataSet * GetDataSet()
   { return fDataSet; };

   void SetDataSet(BCDataSet * dataset);

   void SetSingleDataPoint(BCDataPoint * datapoint);

   void SetSingleDataPoint(BCDataSet * dataset, unsigned int index);

   void SetNIterationsMax(int niterations);

   void SetNIterationsMin(int niterations);

   void SetNIterationsPrecisionCheck(int niterations);

   void SetNIterationsOutput(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 SetDataPointLowerBoundaries(BCDataPoint * datasetlowerboundaries);

   void SetDataPointUpperBoundaries(BCDataPoint* datasetupperboundaries);

   void SetDataPointLowerBoundary(int index, double lowerboundary);

   void SetDataPointUpperBoundary(int index, double upperboundary);

   void SetDataBoundaries(int index, double lowerboundary, double upperboundary);

   void SetNChains(unsigned int n);

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

   void AddDataPoint(BCDataPoint * datapoint)
   { fDataSet -> AddDataPoint(datapoint); };

   int ReadDataFromFile(const char * filename, const char * treename, const char * branchnames)
   { return this ->  ReadDataFromFileTree(filename, treename, branchnames); };

   int ReadDataFromFile(const char * filename, int nvariables)
   { return this -> ReadDataFromFileTxt(filename, nvariables); };

   int ReadDataFromFileTree(const char * filename, const char * treename, const char * branchnames);

   int ReadDataFromFileTxt(const char * filename, int nbranches);

   void Integrate();

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

   void FindMode();

   void MarginalizeAll();

   void WriteMarkovChain(bool flag);

   void ResetDataSet()
   { fDataSet -> Reset(); };

   void PrintModelComparisonSummary(const char * filename=0);

   void PrintSummary(const char * filename=0);

   void PrintResults();

   void CalculatePValue(bool flag_histogram=false);

private:

   void Copy(BCModelManager & modelmanager) const;

   BCModelContainer * fModelContainer;

   BCDataSet * fDataSet;

};

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

#endif