MemoryManager.hpp

Go to the documentation of this file.

#pragma once
 
#include <vector>
#include <map>
#include <string>
#include <pthread.h>
 
#include "DataTypeInator.hpp"
#include "MemoryManagement/AllocInfo.hpp"
#include "Type/DataType.hpp"
 
#include "CheckpointAgent/NewCheckpointAgentBase.hpp"
 
class MemoryManager {
 
    public:
    MemoryManager (DataTypeInator * dataTypeInator);
 
    ~MemoryManager ();
 
    // no copying
    MemoryManager(const MemoryManager& other) = delete;
    
    void* declare_var( const std::string& declaration, void* suppliedAllocation = NULL);
 
 
 
    bool var_exists( const std::string& var_name);
 
    void clear_var( void* address);
 
    void clear_var( const std::string& var_name);
 
    void clear_all_vars();
 
    void delete_var(void* address);
 
    void delete_var(std::string var_name);
 
    void write_checkpoint( std::ostream& out_s);
 
    void write_checkpoint( const std::string& filename);
 
    void restore_checkpoint(const std::string& filename);
 
    void restore_checkpoint( std::istream& in_s);
 
    AllocInfo* getAllocInfoOf( void* address );
 
    AllocInfo* getAllocInfoAt( void* address );
 
    AllocInfo* getAllocInfoNamed( const std::string& name );
 
    std::shared_ptr<const DataType> getDataTypeOf(const std::string& name);
 
    void setCheckPointAgent( CheckpointAgentBase* agent);
 
    private:
 
    void do_write_checkpoint(std::ostream& outStream, std::vector<AllocInfo*>& dependencies);
    
    void* do_declare_var(const std::string& abstract_declarator, 
                                const std::string& variable_name,
                                void * supplied_allocation);
 
    void delete_allocation(AllocInfo * allocInfo);
 
    unsigned int debugLevel;
 
    CheckpointAgentBase * checkpointAgent;
    DataTypeInator* dataTypeInator;
 
    // TODO: Let's think about this.
    // I would like to switch to c++ threading instead using pthreads directly
    pthread_mutex_t allocInfoMapMutex;
 
    std::map<void*, AllocInfo*> allocInfoByAddressMap;
    std::map<std::string, AllocInfo*> allocInfoByNameMap;
 
};