BCModelManager.h

Go to the documentation of this file.

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

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

#ifndef __BCMODELMANAGER__H
#define __BCMODELMANAGER__H

#include "BCModel.h"

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

class BCModelManager
{

 public:

   /* @{ */

   BCModelManager(); 

   BCModelManager(const BCModelManager & modelmanager); 

   virtual ~BCModelManager(); 

   /* @} */ 

   /* @{ */ 

   BCModelManager & operator = (const BCModelManager & modelmanager); 

   /* @} */ 

   /* @{ */ 

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

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

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

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

   BCDataSet * GetDataSet()
   { return fDataSet; }; 

   /* @} */ 

   /* @{ */ 
   
   void SetDataSet(BCDataSet * dataset); 

   void SetSingleDataPoint(BCDataPoint * datapoint); 

   void SetSingleDataPoint(BCDataSet * dataset, 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(int niterations); 

   void SetNSamplesPer2DBin(int n); 

   void SetRelativePrecision(double relprecision); 

   void SetNbins(int n);

   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(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 ReadDataFromFileTree(const char * filename, const char * treename, const char * branchnames); 

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

   int ReadDataFromFileUser(const char *  filename, std::vector<int> options_int, std::vector<double> options_double);

   void Normalize(); 

   void FindMode(); 

   void MarginalizeAll(); 

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

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

   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

---