fileservices.h

Go to the documentation of this file.

 
#ifndef FILESERVICES_H
#define FILESERVICES_H
 
#include <list>
#include <string>
#include <map>
#include "service.h"
#include "param.h"
 
class Metapop;
class Patch;
 
class FileHandler;
class FHLogWriter;
class FileServices : public Service {
 
private:
  Metapop*    _popPtr;
  FHLogWriter* _logWriter;
  
  list< FileHandler* > _writers;
  
  list< FileHandler* > _readers;  
  
  string _rep_filename;
  
  string _basename;
  
  string _root_dir;
  
  list< ParamSet* > _params;
  
  unsigned int _mode;
  
  TMatrix* _patch_sample_size;  // num_patch x num_age
 
  TMatrix* _patch_sample_age;   // num_patch x num_age
 
  TMatrix* _patch_sampled;      // 1 x num_patch
 
  int _sampled_at_replicate, _sampled_at_generation;
 
  Metapop *_sample_pop;
 
public:
        
  typedef std::list< FileHandler* >::const_iterator file_it;
    
  FileServices ( );
  
  virtual ~FileServices ( );
  
  virtual bool init ( ) {return false;}
 
  virtual void notify ();
  bool init (list< ParamSet* >&  params);
  
  virtual Metapop*   get_pop_ptr ( )      {return _popPtr;}
  
  virtual void set_pop_ptr (Metapop* pop) {_popPtr = pop;}
  
  void setMode(unsigned int m) {_mode = m;}
  
  unsigned int getMode() {return _mode;}
  
  void setBasename (string name);
  
  void setRootDir (string name);
  
  void log_simparams();
  
  void log (string message);
  
  list< ParamSet* >& get_params() {return _params;};
  
  file_it getFirstWriter() {return _writers.begin();}
  
  file_it getLastWriter () {return _writers.end();}  
  
  file_it getFirstReader() {return _readers.begin();}
  
  file_it getLastReader () {return _readers.end();}
  
  FileHandler* getReader (string& type);
  
  string&  getBaseFileName ();
  
  string&  getRootDir ();
  
  string  getReplicateCounter ();
  
  string&  getFirstReplicateFileName ();
  
  string&  getReplicateFileName ();
  
  string  getGenerationCounter ();
  
  string  getGenerationReplicateFileName ();
  
 
  void set_sampling_params(TMatrix* size, TMatrix* age, TMatrix* patches) {_patch_sample_size = size; _patch_sample_age = age; _patch_sampled = patches;}
 
  void empty_sampled_pop ();
 
  void reset_sampled_pop ();
 
  Metapop* getSampledPop ();
 
  void subSamplePatch(sex_t SEX, age_idx AGE, unsigned int num, Patch* source_patch, Patch* patch);
 
  virtual void load ( SimComponent* sc );
  
  virtual void attach ( Handler* FH );
  
  virtual void attach_reader ( FileHandler* FH );
  
  virtual void reset ( );
 
};
#endif //FILESERVICES_H