CudaPlanStorage.h

Go to the documentation of this file.

// CudaPlanStorage.h:
// ------------------
//

#ifndef __IPSDKIMAGE_CUDAPLANSTORAGE_H__
#define __IPSDKIMAGE_CUDAPLANSTORAGE_H__

// suppression warning
// warning C4251: 'ipsdk::image::CudaPlanStorage::_pBuffer' : class 'boost::shared_ptr<ipsdk::image::CudaBuffer>' needs to have dll-interface to be used by clients of class 'ipsdk::processor::BaseOutputOptionalValueAttribute<T>'
// warning C4251: 'ipsdk::image::CudaPlanStorage::_dataType' : class 'boost::shared_ptr<ipsdk::image::CudaBuffer>' needs to have dll-interface to be used by clients of class 'ipsdk::processor::BaseOutputOptionalValueAttribute<T>'
#pragma warning (push)
#pragma warning (disable : 4251)

//#include <IPSDKImage/IPSDKImageExports.h>
#include <IPSDKImage/Storage/BasePlanStorage.h>
#include <IPSDKImage/Geometry/BaseImageGeometry.h>
#include <IPSDKImage/Storage/Buffers/CudaBuffer.cuh>

namespace ipsdk {
namespace image {


class IPSDKIMAGE_API CudaPlanStorage : public BasePlanStorage
{
    // declare dynamic serializable class
    IPSDK_DECLARE_SERIAL_WITHOUT_COPY(IPSDKImage, CudaPlanStorage);

// predefined public types
public:
    static const ePlanStorageType::domain g_planStorageType = ePlanStorageType::ePST_Gpu;

    typedef boost::shared_ptr<CudaPlanStorage> CudaPlanStoragePtr;
    typedef boost::shared_ptr<const CudaPlanStorage> CudaPlanStorageConstPtr;

    typedef boost::shared_ptr<CudaBuffer> CudaBufferPtr;
    typedef boost::shared_ptr<const CudaBuffer> CudaBufferConstPtr;

public:
    CudaPlanStorage();
    ~CudaPlanStorage();

// methods
public:
    ePlanStorageType        getPlanStorageType() const;

    BoolResult              init(const ImagePlanGeometry& geometry, const ipUInt64 deviceIdx = 0);

    eImageAccessMode        getAccessMode() const;

    const void*             getDataPtr() const;
    void*                   getDataPtr();

    bool                    canBeRemotlyAccessed() const;

    bool                    hasBinary8BitsStorage() const;

    ipUInt64                getFullBufferSize() const;

    const CudaBuffer&       getBuffer() const;
    CudaBuffer&             getBuffer();

    ipUInt64                getDeviceIdx() const;

    BoolResult              resizePlan(const ImagePlanGeometry& planGeometry);

    BoolResult              resize(const ImagePlanGeometry& geometry, const ipUInt64 deviceIdx = 0);

protected:
    BoolResult              resizePlan(const ImagePlanGeometry& geometry, const ipUInt64 deviceIdx = 0);

    BoolResult              readRawData(const ipUInt64 startOffset,
                                        const ipUInt64 size,
                                        void* pData) const;

    BoolResult              writeRawData(const ipUInt64 startOffset,
                                         const ipUInt64 size,
                                         const void* pData);

    void                    clearPlanData();

// attributes
protected:
    CudaBufferPtr           _pBuffer;

    ipUInt64                _bufferBaseSize;

    ipUInt64                _deviceIdx;

    eDataType               _dataType;
};


inline ePlanStorageType
CudaPlanStorage::getPlanStorageType() const
{
    return g_planStorageType;
}


} // end of namespace image
} // end of namespace ipsdk

#pragma warning (pop)

#endif // __IPSDKIMAGE_CUDAPLANSTORAGE_H__