RdmaVideoOutput.hpp

Go to the documentation of this file.

#pragma once
 
#include <Gfx/Graph/encoders/ComputeEncoder.hpp>
#include <Gfx/Graph/interop/ComputeRingDispatcher.hpp>
#include <Gfx/Graph/interop/CudaInterop.h>
#include <Gfx/Graph/interop/ImportedGpuBufferRing.hpp>
#include <Gfx/Graph/interop/InteropFence.hpp>
#include <Gfx/Graph/interop/VendorDmaRegistrar.hpp>
#include <score_plugin_gfx_export.h>
 
#include <cstdint>
#include <functional>
#include <memory>
 
class QRhi;
class QRhiCommandBuffer;
class QRhiTexture;
 
namespace score::gfx
{
struct RenderState;
}
 
namespace score::gfx::interop
{
 
// VendorDmaRegistrar (the per-slot pin/unpin hooks supplied by the vendor
// adapter) lives in VendorDmaRegistrar.hpp, shared with CpuStagedVideoOutput.
 
struct RdmaVideoOutputConfig
{
  QRhi* rhi{};
  const score::gfx::RenderState* state{};
  QRhiTexture* sourceTexture{};
  int width{};
  int height{};
  std::uint32_t frameByteSize{};
  int slotCount{2};
  const char* debugName{"score-gpu-direct-output"};
 
  std::function<std::unique_ptr<score::gfx::ComputeEncoder>()> encoderFactory;
 
  QString colorConversion;
 
  VendorDmaRegistrar registrar;
};
 
class SCORE_PLUGIN_GFX_EXPORT RdmaVideoOutput
{
public:
  RdmaVideoOutput() = default;
  ~RdmaVideoOutput();
 
  RdmaVideoOutput(const RdmaVideoOutput&) = delete;
  RdmaVideoOutput& operator=(const RdmaVideoOutput&) = delete;
 
  bool init(const RdmaVideoOutputConfig& cfg);
 
  void release();
 
  bool valid() const noexcept { return m_ring.valid(); }
  std::size_t slotCount() const noexcept { return m_ring.slotCount(); }
 
  void encodeFrame(QRhiCommandBuffer& cb);
 
  void* prepareNextFrame();
 
  void advance();
 
  std::uint64_t lostFrames() const noexcept { return m_lostFrames; }
 
private:
  void reportLostFrame(const char* stage);
 
  RdmaVideoOutputConfig m_cfg{};
  CudaInteropContextHandle m_cudaCtx{};
  ImportedGpuBufferRing m_ring;
  std::unique_ptr<InteropFence> m_fence;
  ComputeRingDispatcher m_dispatcher;
  std::vector<void*> m_bounce;
  std::vector<bool> m_pinned;  
  std::uint64_t m_fenceValue{0};
  std::uint64_t m_lostFrames{0};
};
 
} // namespace score::gfx::interop