PtrMemory.hpp

/*
 * PtrMemory.hpp
 *
 *  Created on: Apr 15, 2015
 *      Author: Pietro Incardona
 */

#ifndef PTRMEMORY_HPP_
#define PTRMEMORY_HPP_


#include "config.h"
#include "memory.hpp"
#include <cstddef>
#include <cstdint>
#include <iostream>

#ifdef SE_CLASS2
#include "Memleak_check.hpp"
#endif

class PtrMemory : public memory
{
    size_t spm;

    void * dm;

    long int ref_cnt;

    bool copyDeviceToDevice(const PtrMemory & m);

    bool copyFromPointer(const void * ptr, size_t sz);

    void setAlignment(size_t align);

public:

    virtual bool flush() {return true;};
    virtual bool allocate(size_t sz);
    virtual void destroy();
    virtual bool copy(const memory & m);
    virtual size_t size() const;
    virtual bool resize(size_t sz);
    virtual void * getPointer();

    virtual const void * getPointer() const;

    virtual void * getDevicePointer();

    virtual void deviceToHost(){};

    virtual void incRef()
    {ref_cnt++;}

    virtual void decRef()
    {ref_cnt--;}

    virtual long int ref()
    {
        return ref_cnt;
    }

    bool isInitialized()
    {
        return true;
    }

    // Default constructor
    PtrMemory():spm(0),dm(NULL),ref_cnt(0)
    {
    };

    PtrMemory(void * ptr, size_t sz):spm(sz),dm(ptr),ref_cnt(0)
    {
    };

    ~PtrMemory()
    {
        if(ref_cnt == 0)
            destroy();
        else
            std::cerr << "Error: " << __FILE__ << " " << __LINE__ << " destroying a live object" << "\n";
    };
};


#endif /* PTRMEMORY_HPP_ */