statservices.h

Go to the documentation of this file.

 
#ifndef STATSERVICES_H
#define STATSERVICES_H
 
#include <list>
#include <string>
#include <fstream>
#include <map>
#include <vector>
#include <assert.h>
#include <cstring> //for memcpy
#include "service.h"
#include "statrecorder.h"
 
using namespace std;
 
class Metapop;
 
class StatHandlerBase;
 
class StatServices : public Service {
  
private:
        
  Metapop* _popPtr;
  
  list< StatHandlerBase* > _statHandlers;
  
  list< StatRecBase *> _statRecorders;
  
  unsigned int _numRecorders;
  
  vector< vector< double*> > _statValues;
  
  double* _currentStatValues;
  
  string _statArg;
  
  unsigned int _occurrence;
  
  list< unsigned int > _occurrences;
  
  list< unsigned int >::const_iterator _current_occurrence;
  
  //output format:
  unsigned int _width, _precision;
  
  unsigned char _separator;
  
  bool _printAverages;
  
public:
    
  typedef list< StatHandlerBase* >::const_iterator stat_it;
        
  typedef list< StatRecBase* >::const_iterator rec_it;
  
  StatServices ( ) : _popPtr(0), _occurrence(0), _currentStatValues(0), 
  _numRecorders(0), _width(12), _precision(6), _separator('\t'), _printAverages(1)
  { }
  
  virtual ~StatServices ( ) { reset_stat_table(); }
  
  virtual bool init ( );
  
  Metapop* get_pop_ptr          ( )      {return _popPtr;}
  
  void set_pop_ptr              (Metapop* pop) {_popPtr=pop;}
  
  void set                      (string& str, unsigned int occ) 
  {_statArg = str; _occurrence = occ;}
  
  void setStatOptions           (string& str) {_statArg = str;}
  
  string& getStatArg       ( )  {return _statArg;}
  
  //OCURRENCE
  unsigned int getOccurrence    ( ) {return _occurrence;}
  
  void setOcccurrence           (unsigned int value) {_occurrence = value;}
  void setOccurrences           (map<unsigned int, unsigned int> timeTable);
  unsigned int getNumOccurrences( ) {return _occurrences.size();}
  
  unsigned int getNumOccurrences(unsigned int replicate) 
  {
    return _statValues[replicate-1].size();
  }
  unsigned int getCurrentOccurrence () {return *_current_occurrence;}
  void resetCurrentOccurrence () {_current_occurrence = _occurrences.begin();}
  
  //PRINT
  void printStatHeaders         (ofstream& FH);
  void doPrintAverages     ( )  {_printAverages = 1;}
  void cancelPrintAverages ( )  {_printAverages = 0;}
  bool getPrintAveragesOpt ( )  {return _printAverages;}
  
  void printStatValue           (ofstream& FH, unsigned int repl_idx);
  
  void printStatAverage         (ofstream& FH);
  
  void setCompactOutputFormat   () {_width = 0; _separator = ' ';}
  
  void setFieldWidth            (unsigned int val) {_width = val;}
  
  void setFieldPrecision        (unsigned int val) {_precision = val;}
  
  void setFieldSeparator        (unsigned char c)  {_separator = c;}
  
  void setDefaultOutputFormat   () {_width = 12; _precision = 6; _separator = '\t';}
  
  //STAT RECORDS
  list<StatRecBase*> getAllStats ( );
  
  unsigned int getNumStats ( ) {return _statRecorders.size();}
  
  stat_it getFirst () {return _statHandlers.begin();}
  
  stat_it getLast () {return _statHandlers.end();}
  
  vector<double*>* getReplicateStatRecords (unsigned int replicate) 
  {
    return &_statValues[ replicate-1 ];
  }
    
  double* getGenerationStatValues (unsigned int replicate, unsigned int occurence) const
  {
    assert(occurence < _occurrences.size());
    return _statValues[ replicate-1 ][ occurence ];
  }
  
  void copyGenerationStatValues (unsigned int replicate, unsigned int occurence, double* values, unsigned int size)
  {
    assert( _statValues[ replicate-1 ].size() < _occurrences.size() );
    
    assert( size == _numRecorders+2);
    
    double* new_record = new double[size];
    
    memcpy(new_record, values, size*sizeof(double));
    
    _statValues[ replicate-1 ].push_back(new_record);
  }
  void recordStats              (unsigned int gen);
  
  virtual void notify ();
  virtual void load ( SimComponent* sc );
  virtual void attach ( Handler* H);
  virtual void reset ( );
  
  void reset_stat_table();
};
#endif //STATSERVICES_H