HostPinnedRing.hpp

Go to the documentation of this file.

#pragma once
 
#include <Gfx/Graph/interop/GpuCapabilities.hpp>
#include <Gfx/Graph/interop/VideoPixelFormat.hpp>
 
#include <score_plugin_gfx_export.h>
 
#include <cstddef>
#include <cstdint>
#include <memory>
 
class QRhi;
class QRhiTexture;
class QRhiResourceUpdateBatch;
 
namespace score::gfx::interop
{
 
enum class HostPinnedRingBackend : uint8_t
{
  None = 0,
  Dvp,             
  AmdPinned,       
  CudaHostReg,     
  CpuStaging,      
};
 
SCORE_PLUGIN_GFX_EXPORT const char*
hostPinnedBackendName(HostPinnedRingBackend) noexcept;
 
enum class HostPinnedDirection : uint8_t
{
  CaptureToTexture = 0, 
  TextureToBuffer,      
};
 
struct SCORE_PLUGIN_GFX_EXPORT HostPinnedSlot
{
  void* host{};         
  std::size_t size{};   
  std::size_t stride{}; 
  void* backendOpaque{}; 
  bool cudaRegistered{};
};
 
struct HostPinnedRingConfig
{
  QRhi* rhi{};
  const GpuCapabilities* caps{}; 
  HostPinnedDirection direction{HostPinnedDirection::CaptureToTexture};
  VideoPixelFormat format{VideoPixelFormat::BGRA8};
  uint32_t width{};
  uint32_t height{};
  uint32_t stride{0}; 
  uint32_t slotCount{2};
  const char* debugName{"score-host-pinned-ring"};
};
 
class SCORE_PLUGIN_GFX_EXPORT HostPinnedRing
{
public:
  HostPinnedRing() noexcept;
  ~HostPinnedRing();
 
  HostPinnedRing(const HostPinnedRing&) = delete;
  HostPinnedRing& operator=(const HostPinnedRing&) = delete;
  HostPinnedRing(HostPinnedRing&&) noexcept;
  HostPinnedRing& operator=(HostPinnedRing&&) noexcept;
 
  bool create(const HostPinnedRingConfig& cfg);
 
  void destroy() noexcept;
 
  bool valid() const noexcept;
  HostPinnedRingBackend backend() const noexcept;
  std::size_t slotCount() const noexcept;
 
  HostPinnedSlot& slot(std::size_t i) noexcept;
  const HostPinnedSlot& slot(std::size_t i) const noexcept;
 
  void bindCudaDestination(
      QRhiTexture* dst, void* cuda_ctx, void* cuda_array) noexcept;
 
  bool uploadSlotToTexture(std::size_t i, QRhiTexture* dst,
                           QRhiResourceUpdateBatch* batch = nullptr);
 
  bool downloadTextureToSlot(QRhiTexture* src, std::size_t i,
                             QRhiResourceUpdateBatch* batch = nullptr);
 
  bool slotReady(std::size_t i) const noexcept;
 
  void resetSlotReady(std::size_t i) noexcept;
 
  std::size_t writeIndex() const noexcept;
  std::size_t advance() noexcept;
 
private:
  struct Impl;
  std::unique_ptr<Impl> m_impl;
};
 
} // namespace score::gfx::interop