BCModelManager.h

Go to the documentation of this file.

#ifndef __BCMODELMANAGER__H
#define __BCMODELMANAGER__H

/*
 * Copyright (C) 2008, Daniel Kollar and Kevin Kroeninger.
 * All rights reserved.
 *
 * For the licensing terms see doc/COPYING.
 */

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

#include "BAT/BCModel.h"
#include "BAT/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);

      /*
       * Fixes an axis */
      void FixDataAxis(int index, bool fixed);

      /*
       * Sets the number of Markov chains */
      void SetNChains(unsigned int n);

      /*
       * Sets the flag for PCA */
      void SetFlagPCA(bool flag);

      /* @} */

      /* @{ */

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

      void FindMode();

      void MarginalizeAll();

      /*
       * Flag for writing Markov chain to file */
      void WriteMarkovChain(bool flag);

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

      void PrintModelComparisonSummary(const char * filename=0);

      void PrintSummary(const char * filename=0);

      /*
       * Prints summaries of all files */
      void PrintResults();

      /*
       * Calculates the p-value for all models. */
      void CalculatePValue(bool flag_histogram=false);

      /* @} */

   private:

      /*
       * Copies this BCModelManager into another one */
      void Copy(BCModelManager & modelmanager) const;

      BCModelContainer * fModelContainer;

      BCDataSet * fDataSet;

};

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

#endif

---