CudaImage.h

Go to the documentation of this file.

// CudaImage.h:
// ------------
//

#ifndef __IPSDKIMAGE_CUDAIMAGE_H__
#define __IPSDKIMAGE_CUDAIMAGE_H__

//#include <IPSDKImage/IPSDKImageExports.h>
#include <IPSDKImage/Image/BaseCompleteImage.h>
#include <IPSDKImage/Storage/CudaPlanStorage.h>

namespace ipsdk {
namespace image {


class IPSDKIMAGE_API CudaImage : public BaseCompleteImage
{
    // declare serial class
    IPSDK_DECLARE_SERIAL_WITHOUT_COPY(IPSDKImage, CudaImage);

    // friend class for protected members access
    friend boost::shared_ptr<CudaImage> boost::make_shared<CudaImage>();

// predefined public types
public:
    static const eImageType::domain g_imageType = eImageType::eIT_Gpu;

protected:
    CudaImage();

public:
    ~CudaImage();
    
// methods
public:
    eImageType              getImageType() const;

    ipBool                  isGpuImage() const;

    ipBool                  isDiskImage() const;

    virtual BoolResult      init(const BaseImageGeometry& geometry);

    eImageAccessMode        getAccessMode() const;

    BoolResult              resize(const BaseImageGeometry& geometry);

    const void*             getDataPtr(const PlanIndex& planIndex) const;
    void*                   getDataPtr(const PlanIndex& planIndex);

    const void*             getDataPtr(const ipUInt64 zPlanIdx,
                                       const ipUInt64 cPlanIdx,
                                       const ipUInt64 tPlanIdx) const;
    void*                   getDataPtr(const ipUInt64 zPlanIdx,
                                       const ipUInt64 cPlanIdx,
                                       const ipUInt64 tPlanIdx);
    
    void                    fromCPU(const BaseImage& inImage);
    
    virtual void            toCPU(ipsdk::image::ImagePtr pOutImage) const;
    virtual void            toCPU(ipsdk::image::ImagePtr pOutImage);

    virtual ImageConstPtr   toCPU() const;
    virtual ImagePtr        toCPU();

    const ipsdk::image::CudaBuffer&
    getBuffer(const ipUInt64 zPlanIdx,
              const ipUInt64 cPlanIdx,
              const ipUInt64 tPlanIdx) const;

    ipsdk::image::CudaBuffer&
    getBuffer(const ipUInt64 zPlanIdx,
              const ipUInt64 cPlanIdx,
              const ipUInt64 tPlanIdx);

    const char*
    getBufferData(const ipUInt64 zPlanIdx,
                  const ipUInt64 cPlanIdx,
                  const ipUInt64 tPlanIdx) const;

    char*
    getBufferData(const ipUInt64 zPlanIdx,
                  const ipUInt64 cPlanIdx,
                  const ipUInt64 tPlanIdx);

    ipUInt64 getNbDevices() const;

protected:
    BoolResult              createImagePlans(const BaseImageGeometry& geometry,
                                             PlanStorageColl3d& imagePlans);

// attributes
protected:
    
};


inline eImageType
CudaImage::getImageType() const
{
    return g_imageType;
}

inline ipBool
CudaImage::isGpuImage() const
{
    return true;
}

inline ipBool
CudaImage::isDiskImage() const
{
    return false;
}



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

#endif // __IPSDKIMAGE_CUDAIMAGE_H__