SDK-eBUS/Ubuntu-LTS/eBUS_SDK_JAI_Ubuntu-20.04-x86_64-7.0.1-7536.zip

// *****************************************************************************

//

//     Copyright (c) 2025, Pleora Technologies Inc., All rights reserved.

//

// *****************************************************************************


#ifndef __PVCUDABUFFER_H__

#define __PVCUDABUFFER_H__


#include <PvBufferLib.h>

#include <PvPayloadType.h>

#include <PvPixelType.h>


namespace PvBufferLib

{

class CudaBuffer;

class Buffer;

} // namespace PvBufferLib


enum class PvCudaMemorySpaceType

{

    PvCudaHost,

    PvCudaDevice,

    PvCudaUVM

};


class PvCudaBuffer;

class PvBuffer;


class PV_BUFFER_API PvCudaBuffer

{

public:

    PvCudaBuffer( PvPayloadType aPayloadType = PvPayloadTypeImage );


    virtual ~PvCudaBuffer();


    PvPayloadType GetPayloadType() const;


    PvCudaMemorySpaceType GetMemorySpaceType() const;


    const uint8_t* GetDataPointer() const;


    uint8_t* GetDataPointer();


    PvPixelType GetPixelType() const;


    uint32_t GetSize() const;


    PvResult AllocCudaHost( uint32_t aSize, PvPixelType aPixelType );


    PvResult AllocCudaDevice( uint32_t aSize, PvPixelType aPixelType );


    PvResult AllocCudaUVM( uint32_t aSize, PvPixelType aPixelType );


    bool IsAllocated() const;


    void Free();


    PvResult Reallocate( uint32_t aSize );


    PvResult CopyFrom( const PvBuffer* aSrcPvBuffer, unsigned int aSize );


    PvResult CopyTo( PvBuffer* aDstPvBuffer, unsigned int aSize );


    PvResult CopyFrom( const PvCudaBuffer* aSrcPvCudaBuffer, unsigned int aSize );


    PvResult CopyTo( PvCudaBuffer* aDstPvCudaBuffer, unsigned int aSize );


private:

    // Not implemented

    PvCudaBuffer( const PvCudaBuffer& );

    const PvCudaBuffer& operator=( const PvCudaBuffer& );


    friend class PvCudaBufferConverter;


    PvBufferLib::CudaBuffer* mThis;

};


#endif