BCModelManager.h

Go to the documentation of this file.

#ifndef __BCMODELMANAGER__H
#define __BCMODELMANAGER__H

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

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

// BAT classes
class BCDataPoint;

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

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);

      // DEBUG DELETE?
      //   void SetNIterationsMax(int niterations);

      void SetMarginalizationMethod(BCIntegrate::BCMarginalizationMethod method);

      void SetIntegrationMethod(BCIntegrate::BCIntegrationMethod method);

      void SetOptimizationMethod(BCIntegrate::BCOptimizationMethod method);

      void SetNiterationsPerDimension(unsigned int niterations);

      void SetNSamplesPer2DBin(unsigned int n);

      void SetRelativePrecision(double relprecision);

      void SetNbins(unsigned int n);

      void SetFillErrorBand(bool flag = true);

      void UnetFillErrorBand()
         { SetFillErrorBand(false); };

      void SetFitFunctionIndexX(int index);

      void SetFitFunctionIndexY(int index);

      void SetFitFunctionIndices(int indexx, int indexy);

      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 FixDataAxis(int index, bool fixed);

      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 Normalize();

      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