mirror of
https://github.com/kunkundi/crossdesk.git
synced 2026-10-28 16:54:11 +08:00
[feat] enable CUDA/OpenGL video rendering on Linux
This commit is contained in:
@@ -48,6 +48,7 @@
|
|||||||
#include <X11/Xlib.h>
|
#include <X11/Xlib.h>
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
|
|
||||||
|
#include "platform/linux/gui/cuda_graphics.h"
|
||||||
#include "platform/linux/gui/tray/linux_tray.h"
|
#include "platform/linux/gui/tray/linux_tray.h"
|
||||||
#endif
|
#endif
|
||||||
#include "rd_log.h"
|
#include "rd_log.h"
|
||||||
@@ -1030,6 +1031,7 @@ void GuiApplication::InitializeSettings() {
|
|||||||
|
|
||||||
bool GuiApplication::InitializeSDL() {
|
bool GuiApplication::InitializeSDL() {
|
||||||
#if defined(__linux__) && !defined(__APPLE__)
|
#if defined(__linux__) && !defined(__APPLE__)
|
||||||
|
ConfigureLinuxCudaGraphics(config_center_->IsHardwareVideoCodec());
|
||||||
// Standard Wayland top-level windows cannot choose an absolute screen
|
// Standard Wayland top-level windows cannot choose an absolute screen
|
||||||
// position. Use the session's XWayland compatibility server for the GUI so
|
// position. Use the session's XWayland compatibility server for the GUI so
|
||||||
// auxiliary windows such as the controlled-side status window can be
|
// auxiliary windows such as the controlled-side status window can be
|
||||||
|
|||||||
@@ -147,9 +147,9 @@ int GuiRuntime::CreateConnectionPeer() {
|
|||||||
sizeof(params_.log_path) - 1);
|
sizeof(params_.log_path) - 1);
|
||||||
params_.log_path[sizeof(params_.log_path) - 1] = '\0';
|
params_.log_path[sizeof(params_.log_path) - 1] = '\0';
|
||||||
params_.hardware_acceleration = config_center_->IsHardwareVideoCodec();
|
params_.hardware_acceleration = config_center_->IsHardwareVideoCodec();
|
||||||
#if defined(_WIN32) || defined(__APPLE__)
|
#if defined(_WIN32) || defined(__linux__) || defined(__APPLE__)
|
||||||
// Windows renderers retain pooled CPU NV12 frames or CUDA device frames;
|
// Windows and Linux retain pooled CPU NV12 frames or CUDA device frames;
|
||||||
// macOS retains VideoToolbox CVPixelBuffers for direct Metal sampling. Both
|
// macOS retains VideoToolbox CVPixelBuffers for direct Metal sampling. All
|
||||||
// platforms fall back to a packed CPU copy when native upload is unavailable.
|
// platforms fall back to a packed CPU copy when native upload is unavailable.
|
||||||
params_.native_video_output = true;
|
params_.native_video_output = true;
|
||||||
#else
|
#else
|
||||||
|
|||||||
@@ -0,0 +1,82 @@
|
|||||||
|
#include "platform/common/gui/cuda_driver.h"
|
||||||
|
|
||||||
|
#if CROSSDESK_HAS_CUDA_DRIVER
|
||||||
|
#if defined(_WIN32)
|
||||||
|
#include <windows.h>
|
||||||
|
#else
|
||||||
|
#include <dlfcn.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include <type_traits>
|
||||||
|
|
||||||
|
#include "rd_log.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
const CudaDriver* CudaDriver::Get() {
|
||||||
|
static const CudaDriver driver;
|
||||||
|
return driver.loaded_ ? &driver : nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
void* CudaDriver::Symbol(const char* name) const {
|
||||||
|
#if defined(_WIN32)
|
||||||
|
return reinterpret_cast<void*>(
|
||||||
|
GetProcAddress(static_cast<HMODULE>(module_), name));
|
||||||
|
#else
|
||||||
|
return dlsym(module_, name);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
CudaDriver::CudaDriver() {
|
||||||
|
#if defined(_WIN32)
|
||||||
|
module_ = LoadLibraryW(L"nvcuda.dll");
|
||||||
|
#else
|
||||||
|
module_ = dlopen("libcuda.so.1", RTLD_LAZY | RTLD_LOCAL);
|
||||||
|
#endif
|
||||||
|
if (!module_) return;
|
||||||
|
const auto load = [this](auto& function, const char* name) {
|
||||||
|
function = reinterpret_cast<std::decay_t<decltype(function)>>(Symbol(name));
|
||||||
|
return function != nullptr;
|
||||||
|
};
|
||||||
|
loaded_ = load(init, "cuInit") &&
|
||||||
|
load(device_get_count, "cuDeviceGetCount") &&
|
||||||
|
load(device_get, "cuDeviceGet") &&
|
||||||
|
load(context_create, "cuCtxCreate_v2") &&
|
||||||
|
load(context_destroy, "cuCtxDestroy_v2") &&
|
||||||
|
load(context_push, "cuCtxPushCurrent_v2") &&
|
||||||
|
load(context_pop, "cuCtxPopCurrent_v2") &&
|
||||||
|
load(error_name, "cuGetErrorName") &&
|
||||||
|
load(mem_alloc, "cuMemAlloc_v2") &&
|
||||||
|
load(mem_free, "cuMemFree_v2") &&
|
||||||
|
load(memcpy_2d, "cuMemcpy2D_v2") &&
|
||||||
|
load(memcpy_2d_async, "cuMemcpy2DAsync_v2") &&
|
||||||
|
load(stream_create, "cuStreamCreate") &&
|
||||||
|
load(stream_destroy, "cuStreamDestroy_v2") &&
|
||||||
|
load(stream_synchronize, "cuStreamSynchronize") &&
|
||||||
|
load(register_gl_buffer, "cuGraphicsGLRegisterBuffer") &&
|
||||||
|
load(unregister_resource, "cuGraphicsUnregisterResource") &&
|
||||||
|
load(map_resources, "cuGraphicsMapResources") &&
|
||||||
|
load(unmap_resources, "cuGraphicsUnmapResources") &&
|
||||||
|
load(mapped_pointer, "cuGraphicsResourceGetMappedPointer_v2");
|
||||||
|
}
|
||||||
|
|
||||||
|
CudaDriver::~CudaDriver() {
|
||||||
|
if (!module_) return;
|
||||||
|
#if defined(_WIN32)
|
||||||
|
FreeLibrary(static_cast<HMODULE>(module_));
|
||||||
|
#else
|
||||||
|
dlclose(module_);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
bool CudaDriver::Check(CUresult result, const char* operation) const {
|
||||||
|
if (result == CUDA_SUCCESS) return true;
|
||||||
|
const char* name = nullptr;
|
||||||
|
error_name(result, &name);
|
||||||
|
LOG_WARN("CUDA/OpenGL {} failed: {} ({})", operation,
|
||||||
|
name ? name : "unknown CUDA error", static_cast<int>(result));
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
|
#endif
|
||||||
@@ -0,0 +1,63 @@
|
|||||||
|
/*
|
||||||
|
* @Author: DI JUNKUN
|
||||||
|
* @Date: 2026-09-07
|
||||||
|
* Copyright (c) 2026 by DI JUNKUN, All Rights Reserved.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef _CUDA_DRIVER_H_
|
||||||
|
#define _CUDA_DRIVER_H_
|
||||||
|
|
||||||
|
#if defined(USE_CUDA) && USE_CUDA && \
|
||||||
|
(defined(_WIN32) || \
|
||||||
|
(defined(__linux__) && (defined(__x86_64__) || defined(__amd64__))))
|
||||||
|
#define CROSSDESK_HAS_CUDA_DRIVER 1
|
||||||
|
|
||||||
|
#include <cuda.h>
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
// Application-owned driver entry points. No codec library or MiniRTC internals
|
||||||
|
// are involved; the module remains loaded for this singleton's lifetime.
|
||||||
|
class CudaDriver {
|
||||||
|
public:
|
||||||
|
static const CudaDriver* Get();
|
||||||
|
bool Check(CUresult result, const char* operation) const;
|
||||||
|
|
||||||
|
decltype(&cuInit) init = nullptr;
|
||||||
|
decltype(&cuDeviceGetCount) device_get_count = nullptr;
|
||||||
|
decltype(&cuDeviceGet) device_get = nullptr;
|
||||||
|
CUresult(CUDAAPI* context_create)(CUcontext*, unsigned int, CUdevice) = nullptr;
|
||||||
|
decltype(&cuCtxDestroy) context_destroy = nullptr;
|
||||||
|
decltype(&cuCtxPushCurrent) context_push = nullptr;
|
||||||
|
decltype(&cuCtxPopCurrent) context_pop = nullptr;
|
||||||
|
decltype(&cuGetErrorName) error_name = nullptr;
|
||||||
|
decltype(&cuMemAlloc) mem_alloc = nullptr;
|
||||||
|
decltype(&cuMemFree) mem_free = nullptr;
|
||||||
|
decltype(&cuMemcpy2D) memcpy_2d = nullptr;
|
||||||
|
decltype(&cuMemcpy2DAsync) memcpy_2d_async = nullptr;
|
||||||
|
decltype(&cuStreamCreate) stream_create = nullptr;
|
||||||
|
decltype(&cuStreamDestroy) stream_destroy = nullptr;
|
||||||
|
decltype(&cuStreamSynchronize) stream_synchronize = nullptr;
|
||||||
|
CUresult(CUDAAPI* register_gl_buffer)(CUgraphicsResource*, unsigned int,
|
||||||
|
unsigned int) = nullptr;
|
||||||
|
decltype(&cuGraphicsUnregisterResource) unregister_resource = nullptr;
|
||||||
|
decltype(&cuGraphicsMapResources) map_resources = nullptr;
|
||||||
|
decltype(&cuGraphicsUnmapResources) unmap_resources = nullptr;
|
||||||
|
decltype(&cuGraphicsResourceGetMappedPointer) mapped_pointer = nullptr;
|
||||||
|
|
||||||
|
private:
|
||||||
|
CudaDriver();
|
||||||
|
~CudaDriver();
|
||||||
|
CudaDriver(const CudaDriver&) = delete;
|
||||||
|
CudaDriver& operator=(const CudaDriver&) = delete;
|
||||||
|
void* Symbol(const char* name) const;
|
||||||
|
void* module_ = nullptr;
|
||||||
|
bool loaded_ = false;
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
|
#else
|
||||||
|
#define CROSSDESK_HAS_CUDA_DRIVER 0
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif
|
||||||
@@ -0,0 +1,198 @@
|
|||||||
|
#include "platform/common/gui/cuda_opengl_interop.h"
|
||||||
|
#include "platform/common/gui/cuda_driver.h"
|
||||||
|
|
||||||
|
#include <memory>
|
||||||
|
#include <new>
|
||||||
|
|
||||||
|
#include "platform/common/gui/native_video_frame_ref.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
#if CROSSDESK_HAS_CUDA_DRIVER
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
class ScopedCudaContext {
|
||||||
|
public:
|
||||||
|
ScopedCudaContext(const CudaDriver& driver, CUcontext context)
|
||||||
|
: cuda(driver),
|
||||||
|
active_(cuda.Check(cuda.context_push(context), "cuCtxPushCurrent")) {}
|
||||||
|
~ScopedCudaContext() { Pop(); }
|
||||||
|
explicit operator bool() const { return active_; }
|
||||||
|
bool Pop() {
|
||||||
|
if (!active_) return true;
|
||||||
|
active_ = false;
|
||||||
|
CUcontext context = nullptr;
|
||||||
|
return cuda.Check(cuda.context_pop(&context), "cuCtxPopCurrent");
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
const CudaDriver& cuda;
|
||||||
|
bool active_;
|
||||||
|
};
|
||||||
|
|
||||||
|
bool IsCudaFrame(const MiniRtcNativeVideoFrame* frame) {
|
||||||
|
return frame && GetOpenGlNativeFrame(*frame) &&
|
||||||
|
frame->type == MiniRtcNativeVideoFrameCudaNv12;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
struct CudaOpenGlVideoBuffer {
|
||||||
|
const CudaDriver* driver = nullptr;
|
||||||
|
CUcontext context = nullptr;
|
||||||
|
CUstream stream = nullptr;
|
||||||
|
CUgraphicsResource resource = nullptr;
|
||||||
|
bool mapped = false;
|
||||||
|
bool failed = false;
|
||||||
|
NativeVideoFrameRef source_frame;
|
||||||
|
};
|
||||||
|
|
||||||
|
bool IsCudaDeviceAvailable() {
|
||||||
|
const auto* cuda = CudaDriver::Get();
|
||||||
|
int count = 0;
|
||||||
|
return cuda && cuda->init(0) == CUDA_SUCCESS &&
|
||||||
|
cuda->device_get_count(&count) == CUDA_SUCCESS && count > 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
CudaOpenGlVideoBuffer* CreateCudaOpenGlVideoBuffer(
|
||||||
|
const MiniRtcNativeVideoFrame* frame, uint32_t gl_buffer) {
|
||||||
|
const auto* driver = CudaDriver::Get();
|
||||||
|
if (!IsCudaFrame(frame) || gl_buffer == 0 || !driver) {
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
std::unique_ptr<CudaOpenGlVideoBuffer, decltype(&DestroyCudaOpenGlVideoBuffer)>
|
||||||
|
buffer(new (std::nothrow) CudaOpenGlVideoBuffer,
|
||||||
|
&DestroyCudaOpenGlVideoBuffer);
|
||||||
|
if (!buffer) return nullptr;
|
||||||
|
buffer->driver = driver;
|
||||||
|
const auto& cuda = *driver;
|
||||||
|
buffer->context = static_cast<CUcontext>(frame->payload.cuda_nv12.context);
|
||||||
|
buffer->source_frame = NativeVideoFrameRef(frame);
|
||||||
|
ScopedCudaContext context(cuda, buffer->context);
|
||||||
|
if (!context ||
|
||||||
|
!cuda.Check(cuda.stream_create(&buffer->stream,
|
||||||
|
CU_STREAM_NON_BLOCKING),
|
||||||
|
"cuStreamCreate") ||
|
||||||
|
!cuda.Check(cuda.register_gl_buffer(
|
||||||
|
&buffer->resource, gl_buffer,
|
||||||
|
CU_GRAPHICS_REGISTER_FLAGS_WRITE_DISCARD),
|
||||||
|
"cuGraphicsGLRegisterBuffer") ||
|
||||||
|
!context.Pop()) {
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
return buffer.release();
|
||||||
|
}
|
||||||
|
|
||||||
|
int UploadCudaFrameToOpenGlBuffer(
|
||||||
|
CudaOpenGlVideoBuffer* buffer, const MiniRtcNativeVideoFrame* frame) {
|
||||||
|
if (!buffer || buffer->failed || !IsCudaFrame(frame) ||
|
||||||
|
frame->payload.cuda_nv12.context != buffer->context) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
const auto& cuda = *buffer->driver;
|
||||||
|
ScopedCudaContext context(cuda, buffer->context);
|
||||||
|
if (!context) {
|
||||||
|
buffer->failed = true;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
// The caller has completed this buffer's previous GL use. Hold the new
|
||||||
|
// surface through asynchronous copies, including partially failed uploads.
|
||||||
|
buffer->source_frame = NativeVideoFrameRef(frame);
|
||||||
|
const char* operation = "cuGraphicsMapResources";
|
||||||
|
CUresult result =
|
||||||
|
cuda.map_resources(1, &buffer->resource, buffer->stream);
|
||||||
|
buffer->mapped = result == CUDA_SUCCESS;
|
||||||
|
CUdeviceptr destination = 0;
|
||||||
|
size_t mapped_size = 0;
|
||||||
|
if (result == CUDA_SUCCESS) {
|
||||||
|
operation = "cuGraphicsResourceGetMappedPointer";
|
||||||
|
result = cuda.mapped_pointer(
|
||||||
|
&destination, &mapped_size, buffer->resource);
|
||||||
|
}
|
||||||
|
const size_t y_size = static_cast<size_t>(frame->width) * frame->height;
|
||||||
|
if (result == CUDA_SUCCESS && mapped_size < y_size + y_size / 2) {
|
||||||
|
operation = "validate mapped buffer size";
|
||||||
|
result = CUDA_ERROR_INVALID_VALUE;
|
||||||
|
}
|
||||||
|
if (result == CUDA_SUCCESS) {
|
||||||
|
CUDA_MEMCPY2D copy{};
|
||||||
|
copy.srcMemoryType = CU_MEMORYTYPE_DEVICE;
|
||||||
|
copy.srcDevice = frame->payload.cuda_nv12.y_device_pointer;
|
||||||
|
copy.srcPitch = frame->payload.cuda_nv12.y_stride;
|
||||||
|
copy.dstMemoryType = CU_MEMORYTYPE_DEVICE;
|
||||||
|
copy.dstDevice = destination;
|
||||||
|
copy.dstPitch = frame->width;
|
||||||
|
copy.WidthInBytes = frame->width;
|
||||||
|
copy.Height = frame->height;
|
||||||
|
operation = "cuMemcpy2DAsync (Y plane)";
|
||||||
|
result = cuda.memcpy_2d_async(©, buffer->stream);
|
||||||
|
if (result == CUDA_SUCCESS) {
|
||||||
|
copy.srcDevice = frame->payload.cuda_nv12.uv_device_pointer;
|
||||||
|
copy.srcPitch = frame->payload.cuda_nv12.uv_stride;
|
||||||
|
copy.dstDevice = destination + y_size;
|
||||||
|
copy.Height = frame->height / 2;
|
||||||
|
operation = "cuMemcpy2DAsync (UV plane)";
|
||||||
|
result = cuda.memcpy_2d_async(©, buffer->stream);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (buffer->mapped) {
|
||||||
|
const CUresult unmap =
|
||||||
|
cuda.unmap_resources(1, &buffer->resource, buffer->stream);
|
||||||
|
buffer->mapped = unmap != CUDA_SUCCESS;
|
||||||
|
if (result == CUDA_SUCCESS) {
|
||||||
|
operation = "cuGraphicsUnmapResources";
|
||||||
|
result = unmap;
|
||||||
|
} else {
|
||||||
|
cuda.Check(unmap, "cuGraphicsUnmapResources (cleanup)");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
const bool copied = cuda.Check(result, operation);
|
||||||
|
const bool popped = context.Pop();
|
||||||
|
buffer->failed = !copied || !popped;
|
||||||
|
return buffer->failed ? -1 : 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void DestroyCudaOpenGlVideoBuffer(CudaOpenGlVideoBuffer* buffer) {
|
||||||
|
if (!buffer) return;
|
||||||
|
// Keep the frame's device context alive until after the context is popped.
|
||||||
|
std::unique_ptr<CudaOpenGlVideoBuffer> owner(buffer);
|
||||||
|
const auto& cuda = *buffer->driver;
|
||||||
|
ScopedCudaContext context(cuda, buffer->context);
|
||||||
|
if (!context) return;
|
||||||
|
if (buffer->stream) {
|
||||||
|
cuda.Check(cuda.stream_synchronize(buffer->stream),
|
||||||
|
"cuStreamSynchronize (destroy)");
|
||||||
|
}
|
||||||
|
if (buffer->mapped) {
|
||||||
|
cuda.Check(cuda.unmap_resources(
|
||||||
|
1, &buffer->resource, buffer->stream),
|
||||||
|
"cuGraphicsUnmapResources (destroy)");
|
||||||
|
cuda.Check(cuda.stream_synchronize(buffer->stream),
|
||||||
|
"cuStreamSynchronize (unmap)");
|
||||||
|
}
|
||||||
|
if (buffer->resource) {
|
||||||
|
cuda.Check(cuda.unregister_resource(buffer->resource),
|
||||||
|
"cuGraphicsUnregisterResource");
|
||||||
|
}
|
||||||
|
if (buffer->stream) {
|
||||||
|
cuda.Check(cuda.stream_destroy(buffer->stream), "cuStreamDestroy");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#else
|
||||||
|
|
||||||
|
bool IsCudaDeviceAvailable() { return false; }
|
||||||
|
CudaOpenGlVideoBuffer* CreateCudaOpenGlVideoBuffer(
|
||||||
|
const MiniRtcNativeVideoFrame*, uint32_t) {
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
int UploadCudaFrameToOpenGlBuffer(
|
||||||
|
CudaOpenGlVideoBuffer*, const MiniRtcNativeVideoFrame*) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
void DestroyCudaOpenGlVideoBuffer(CudaOpenGlVideoBuffer*) {}
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
@@ -0,0 +1,35 @@
|
|||||||
|
/*
|
||||||
|
* @Author: DI JUNKUN
|
||||||
|
* @Date: 2026-09-07
|
||||||
|
* Copyright (c) 2026 by DI JUNKUN, All Rights Reserved.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef _CUDA_OPENGL_INTEROP_H_
|
||||||
|
#define _CUDA_OPENGL_INTEROP_H_
|
||||||
|
|
||||||
|
#include "minirtc.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
bool IsCudaDeviceAvailable();
|
||||||
|
|
||||||
|
struct CudaOpenGlVideoBuffer;
|
||||||
|
|
||||||
|
// These operations require the matching GL context to be current. The caller
|
||||||
|
// owns the GL buffer and must allocate its storage before registering it.
|
||||||
|
CudaOpenGlVideoBuffer* CreateCudaOpenGlVideoBuffer(
|
||||||
|
const MiniRtcNativeVideoFrame* frame, uint32_t gl_buffer);
|
||||||
|
|
||||||
|
// Asynchronous, GPU-only NV12 copy. Retains the source through the next upload
|
||||||
|
// or destruction, including on failure. Before reusing the buffer, the caller
|
||||||
|
// must wait for the previous GL fence. Frames must use the same CUDA context.
|
||||||
|
int UploadCudaFrameToOpenGlBuffer(CudaOpenGlVideoBuffer* buffer,
|
||||||
|
const MiniRtcNativeVideoFrame* frame);
|
||||||
|
|
||||||
|
// Finish prior GL use before calling. Waits for CUDA work and releases retained
|
||||||
|
// frames; the caller may then resize/delete the GL buffer. Accepts null.
|
||||||
|
void DestroyCudaOpenGlVideoBuffer(CudaOpenGlVideoBuffer* buffer);
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
|
|
||||||
|
#endif
|
||||||
@@ -0,0 +1,99 @@
|
|||||||
|
/*
|
||||||
|
* @Author: DI JUNKUN
|
||||||
|
* @Date: 2026-09-07
|
||||||
|
* Copyright (c) 2026 by DI JUNKUN, All Rights Reserved.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef _NATIVE_VIDEO_FRAME_REF_H_
|
||||||
|
#define _NATIVE_VIDEO_FRAME_REF_H_
|
||||||
|
|
||||||
|
#include <limits>
|
||||||
|
#include <utility>
|
||||||
|
|
||||||
|
#include "minirtc.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
class NativeVideoFrameRef {
|
||||||
|
public:
|
||||||
|
NativeVideoFrameRef() = default;
|
||||||
|
|
||||||
|
explicit NativeVideoFrameRef(const MiniRtcNativeVideoFrame *frame) {
|
||||||
|
if (frame && frame->owner && frame->retain && frame->release) {
|
||||||
|
frame_ = *frame;
|
||||||
|
frame_.struct_size = sizeof(frame_);
|
||||||
|
frame_.retain(frame_.owner);
|
||||||
|
valid_ = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
NativeVideoFrameRef(const NativeVideoFrameRef &other)
|
||||||
|
: NativeVideoFrameRef(other.Get()) {}
|
||||||
|
|
||||||
|
NativeVideoFrameRef(NativeVideoFrameRef &&other) noexcept
|
||||||
|
: frame_(other.frame_), valid_(std::exchange(other.valid_, false)) {
|
||||||
|
other.frame_ = {};
|
||||||
|
}
|
||||||
|
|
||||||
|
NativeVideoFrameRef &operator=(NativeVideoFrameRef other) noexcept {
|
||||||
|
Swap(other);
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
~NativeVideoFrameRef() { Reset(); }
|
||||||
|
|
||||||
|
void Reset() {
|
||||||
|
if (valid_ && frame_.owner && frame_.release) {
|
||||||
|
frame_.release(frame_.owner);
|
||||||
|
}
|
||||||
|
frame_ = {};
|
||||||
|
valid_ = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
void Swap(NativeVideoFrameRef &other) noexcept {
|
||||||
|
std::swap(frame_, other.frame_);
|
||||||
|
std::swap(valid_, other.valid_);
|
||||||
|
}
|
||||||
|
|
||||||
|
const MiniRtcNativeVideoFrame *Get() const { return valid_ ? &frame_ : nullptr; }
|
||||||
|
explicit operator bool() const { return valid_; }
|
||||||
|
|
||||||
|
private:
|
||||||
|
MiniRtcNativeVideoFrame frame_{};
|
||||||
|
bool valid_ = false;
|
||||||
|
};
|
||||||
|
|
||||||
|
inline const MiniRtcNativeVideoFrame *GetOpenGlNativeFrame(
|
||||||
|
const MiniRtcNativeVideoFrame &frame) {
|
||||||
|
const MiniRtcNativeVideoFrame *native = &frame;
|
||||||
|
if (native->struct_size < static_cast<uint32_t>(sizeof(MiniRtcNativeVideoFrame)) ||
|
||||||
|
!native->owner || !native->retain || !native->release ||
|
||||||
|
!native->copy_to_nv12 || native->width == 0 || native->height == 0 ||
|
||||||
|
(native->width & 1U) != 0 || (native->height & 1U) != 0 ||
|
||||||
|
native->width > static_cast<uint32_t>(std::numeric_limits<int>::max()) ||
|
||||||
|
native->height > static_cast<uint32_t>(std::numeric_limits<int>::max())) {
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
if (native->type == MiniRtcNativeVideoFrameCpuNv12) {
|
||||||
|
return native->payload.cpu_nv12.y_plane &&
|
||||||
|
native->payload.cpu_nv12.uv_plane &&
|
||||||
|
native->payload.cpu_nv12.y_stride >= native->width &&
|
||||||
|
native->payload.cpu_nv12.uv_stride >= native->width
|
||||||
|
? native
|
||||||
|
: nullptr;
|
||||||
|
}
|
||||||
|
if (native->type == MiniRtcNativeVideoFrameCudaNv12) {
|
||||||
|
return native->payload.cuda_nv12.y_device_pointer != 0 &&
|
||||||
|
native->payload.cuda_nv12.uv_device_pointer != 0 &&
|
||||||
|
native->payload.cuda_nv12.y_stride >= native->width &&
|
||||||
|
native->payload.cuda_nv12.uv_stride >= native->width &&
|
||||||
|
native->payload.cuda_nv12.context
|
||||||
|
? native
|
||||||
|
: nullptr;
|
||||||
|
}
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
|
|
||||||
|
#endif
|
||||||
@@ -19,6 +19,7 @@
|
|||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <array>
|
#include <array>
|
||||||
#include <atomic>
|
#include <atomic>
|
||||||
|
#include <cstdio>
|
||||||
#include <cstring>
|
#include <cstring>
|
||||||
#include <limits>
|
#include <limits>
|
||||||
#include <mutex>
|
#include <mutex>
|
||||||
@@ -26,98 +27,25 @@
|
|||||||
#include <utility>
|
#include <utility>
|
||||||
|
|
||||||
#include "rd_log.h"
|
#include "rd_log.h"
|
||||||
#if defined(_WIN32) && USE_CUDA
|
#include "platform/common/gui/cuda_opengl_interop.h"
|
||||||
#include "nvcodec_api.h"
|
#include "platform/common/gui/native_video_frame_ref.h"
|
||||||
|
#if defined(USE_CUDA) && USE_CUDA && \
|
||||||
|
(defined(_WIN32) || \
|
||||||
|
(defined(__linux__) && (defined(__x86_64__) || defined(__amd64__))))
|
||||||
|
#define CROSSDESK_OPENGL_CUDA_INTEROP 1
|
||||||
|
#else
|
||||||
|
#define CROSSDESK_OPENGL_CUDA_INTEROP 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
namespace crossdesk {
|
namespace crossdesk {
|
||||||
namespace {
|
namespace {
|
||||||
|
|
||||||
constexpr size_t kFrameSlotCount = 3;
|
constexpr size_t kFrameSlotCount = 3;
|
||||||
#if defined(_WIN32) && USE_CUDA
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
constexpr size_t kCudaUploadSlotCount = 3;
|
constexpr size_t kCudaUploadSlotCount = 3;
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
class NativeVideoFrameRef {
|
|
||||||
public:
|
|
||||||
NativeVideoFrameRef() = default;
|
|
||||||
|
|
||||||
explicit NativeVideoFrameRef(const MiniRtcNativeVideoFrame *frame) {
|
|
||||||
if (frame && frame->owner && frame->retain && frame->release) {
|
|
||||||
frame_ = *frame;
|
|
||||||
frame_.struct_size = sizeof(frame_);
|
|
||||||
frame_.retain(frame_.owner);
|
|
||||||
valid_ = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
NativeVideoFrameRef(const NativeVideoFrameRef &other)
|
|
||||||
: NativeVideoFrameRef(other.Get()) {}
|
|
||||||
|
|
||||||
NativeVideoFrameRef(NativeVideoFrameRef &&other) noexcept
|
|
||||||
: frame_(other.frame_), valid_(std::exchange(other.valid_, false)) {
|
|
||||||
other.frame_ = {};
|
|
||||||
}
|
|
||||||
|
|
||||||
NativeVideoFrameRef &operator=(NativeVideoFrameRef other) noexcept {
|
|
||||||
Swap(other);
|
|
||||||
return *this;
|
|
||||||
}
|
|
||||||
|
|
||||||
~NativeVideoFrameRef() { Reset(); }
|
|
||||||
|
|
||||||
void Reset() {
|
|
||||||
if (valid_ && frame_.owner && frame_.release) {
|
|
||||||
frame_.release(frame_.owner);
|
|
||||||
}
|
|
||||||
frame_ = {};
|
|
||||||
valid_ = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
void Swap(NativeVideoFrameRef &other) noexcept {
|
|
||||||
std::swap(frame_, other.frame_);
|
|
||||||
std::swap(valid_, other.valid_);
|
|
||||||
}
|
|
||||||
|
|
||||||
const MiniRtcNativeVideoFrame *Get() const { return valid_ ? &frame_ : nullptr; }
|
|
||||||
explicit operator bool() const { return valid_; }
|
|
||||||
|
|
||||||
private:
|
|
||||||
MiniRtcNativeVideoFrame frame_{};
|
|
||||||
bool valid_ = false;
|
|
||||||
};
|
|
||||||
|
|
||||||
const MiniRtcNativeVideoFrame *GetOpenGlNativeFrame(
|
|
||||||
const MiniRtcNativeVideoFrame &frame) {
|
|
||||||
const MiniRtcNativeVideoFrame *native = &frame;
|
|
||||||
if (native->struct_size < static_cast<uint32_t>(sizeof(MiniRtcNativeVideoFrame)) ||
|
|
||||||
!native->owner || !native->retain || !native->release ||
|
|
||||||
!native->copy_to_nv12 || native->width == 0 || native->height == 0 ||
|
|
||||||
(native->width & 1U) != 0 || (native->height & 1U) != 0 ||
|
|
||||||
native->width > static_cast<uint32_t>(std::numeric_limits<int>::max()) ||
|
|
||||||
native->height > static_cast<uint32_t>(std::numeric_limits<int>::max())) {
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
if (native->type == MiniRtcNativeVideoFrameCpuNv12) {
|
|
||||||
return native->payload.cpu_nv12.y_plane &&
|
|
||||||
native->payload.cpu_nv12.uv_plane &&
|
|
||||||
native->payload.cpu_nv12.y_stride >= native->width &&
|
|
||||||
native->payload.cpu_nv12.uv_stride >= native->width
|
|
||||||
? native
|
|
||||||
: nullptr;
|
|
||||||
}
|
|
||||||
if (native->type == MiniRtcNativeVideoFrameCudaNv12) {
|
|
||||||
return native->payload.cuda_nv12.y_device_pointer != 0 &&
|
|
||||||
native->payload.cuda_nv12.uv_device_pointer != 0 &&
|
|
||||||
native->payload.cuda_nv12.y_stride >= native->width &&
|
|
||||||
native->payload.cuda_nv12.uv_stride >= native->width &&
|
|
||||||
native->payload.cuda_nv12.context
|
|
||||||
? native
|
|
||||||
: nullptr;
|
|
||||||
}
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
enum class SlotUse {
|
enum class SlotUse {
|
||||||
available,
|
available,
|
||||||
pending,
|
pending,
|
||||||
@@ -194,7 +122,7 @@ struct OpenGlFunctions {
|
|||||||
PFNGLUNIFORM4FPROC uniform_4f = nullptr;
|
PFNGLUNIFORM4FPROC uniform_4f = nullptr;
|
||||||
PFNGLUSEPROGRAMPROC use_program = nullptr;
|
PFNGLUSEPROGRAMPROC use_program = nullptr;
|
||||||
PFNGLVERTEXATTRIBPOINTERPROC vertex_attrib_pointer = nullptr;
|
PFNGLVERTEXATTRIBPOINTERPROC vertex_attrib_pointer = nullptr;
|
||||||
#if defined(_WIN32) && USE_CUDA
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
PFNGLFENCESYNCPROC fence_sync = nullptr;
|
PFNGLFENCESYNCPROC fence_sync = nullptr;
|
||||||
PFNGLCLIENTWAITSYNCPROC client_wait_sync = nullptr;
|
PFNGLCLIENTWAITSYNCPROC client_wait_sync = nullptr;
|
||||||
PFNGLDELETESYNCPROC delete_sync = nullptr;
|
PFNGLDELETESYNCPROC delete_sync = nullptr;
|
||||||
@@ -235,7 +163,7 @@ struct OpenGlFunctions {
|
|||||||
LoadOpenGlFunction(&vertex_attrib_pointer, "glVertexAttribPointer");
|
LoadOpenGlFunction(&vertex_attrib_pointer, "glVertexAttribPointer");
|
||||||
}
|
}
|
||||||
|
|
||||||
#if defined(_WIN32) && USE_CUDA
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
bool LoadSyncFunctions() {
|
bool LoadSyncFunctions() {
|
||||||
return LoadOpenGlFunction(&fence_sync, "glFenceSync") &&
|
return LoadOpenGlFunction(&fence_sync, "glFenceSync") &&
|
||||||
LoadOpenGlFunction(&client_wait_sync, "glClientWaitSync") &&
|
LoadOpenGlFunction(&client_wait_sync, "glClientWaitSync") &&
|
||||||
@@ -276,6 +204,21 @@ GLuint CompileShader(const OpenGlFunctions &gl, GLenum type,
|
|||||||
|
|
||||||
bool IsEnabled(GLenum capability) { return glIsEnabled(capability) == GL_TRUE; }
|
bool IsEnabled(GLenum capability) { return glIsEnabled(capability) == GL_TRUE; }
|
||||||
|
|
||||||
|
bool HasOpenGlExtension(const char *extensions, const char *extension) {
|
||||||
|
if (!extensions) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
const size_t length = std::strlen(extension);
|
||||||
|
for (const char *match = std::strstr(extensions, extension); match;
|
||||||
|
match = std::strstr(match + length, extension)) {
|
||||||
|
if ((match == extensions || match[-1] == ' ') &&
|
||||||
|
(match[length] == '\0' || match[length] == ' ')) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
void RestoreCapability(GLenum capability, bool enabled) {
|
void RestoreCapability(GLenum capability, bool enabled) {
|
||||||
if (enabled) {
|
if (enabled) {
|
||||||
glEnable(capability);
|
glEnable(capability);
|
||||||
@@ -346,10 +289,14 @@ struct OpenGlVideoRenderer::Impl {
|
|||||||
GLint video_enabled_location = -1;
|
GLint video_enabled_location = -1;
|
||||||
int texture_width = 0;
|
int texture_width = 0;
|
||||||
int texture_height = 0;
|
int texture_height = 0;
|
||||||
|
int cpu_texture_width = 0;
|
||||||
|
int cpu_texture_height = 0;
|
||||||
|
bool pixel_unpack_buffer_available = false;
|
||||||
|
bool unpack_subimage_available = false;
|
||||||
std::string uploaded_stream;
|
std::string uploaded_stream;
|
||||||
uint64_t uploaded_sequence = 0;
|
uint64_t uploaded_sequence = 0;
|
||||||
|
|
||||||
#if defined(_WIN32) && USE_CUDA
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
enum class CudaUploadResult {
|
enum class CudaUploadResult {
|
||||||
ready,
|
ready,
|
||||||
busy,
|
busy,
|
||||||
@@ -363,17 +310,14 @@ struct OpenGlVideoRenderer::Impl {
|
|||||||
size_t capacity = 0;
|
size_t capacity = 0;
|
||||||
int texture_width = 0;
|
int texture_width = 0;
|
||||||
int texture_height = 0;
|
int texture_height = 0;
|
||||||
CUgraphicsResource resource = nullptr;
|
CudaOpenGlVideoBuffer *resource = nullptr;
|
||||||
GLsync fence = nullptr;
|
GLsync fence = nullptr;
|
||||||
};
|
};
|
||||||
|
|
||||||
std::array<CudaUploadSlot, kCudaUploadSlotCount> cuda_upload_slots;
|
std::array<CudaUploadSlot, kCudaUploadSlotCount> cuda_upload_slots;
|
||||||
size_t next_cuda_upload_slot = 0;
|
size_t next_cuda_upload_slot = 0;
|
||||||
size_t displayed_cuda_upload_slot = kCudaUploadSlotCount;
|
size_t displayed_cuda_upload_slot = kCudaUploadSlotCount;
|
||||||
CUcontext cuda_context = nullptr;
|
void *cuda_context = nullptr;
|
||||||
CUstream cuda_upload_stream = nullptr;
|
|
||||||
NativeVideoFrameRef cuda_context_owner;
|
|
||||||
bool cuda_sync_functions_available = false;
|
|
||||||
bool cuda_interop_disabled = false;
|
bool cuda_interop_disabled = false;
|
||||||
std::atomic<bool> cuda_cpu_fallback{false};
|
std::atomic<bool> cuda_cpu_fallback{false};
|
||||||
bool cuda_native_logged = false;
|
bool cuda_native_logged = false;
|
||||||
@@ -392,6 +336,7 @@ struct OpenGlVideoRenderer::Impl {
|
|||||||
const GLenum wait_result =
|
const GLenum wait_result =
|
||||||
gl.client_wait_sync(slot.fence, 0, 0);
|
gl.client_wait_sync(slot.fence, 0, 0);
|
||||||
if (wait_result == GL_WAIT_FAILED) {
|
if (wait_result == GL_WAIT_FAILED) {
|
||||||
|
LOG_WARN("CUDA/OpenGL glClientWaitSync failed, error={}", glGetError());
|
||||||
DisableCudaInterop();
|
DisableCudaInterop();
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
@@ -419,28 +364,10 @@ struct OpenGlVideoRenderer::Impl {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (cuda_context && minirtc::cuCtxPushCurrent_ld &&
|
|
||||||
minirtc::cuCtxPushCurrent_ld(cuda_context) == CUDA_SUCCESS) {
|
|
||||||
if (cuda_upload_stream && minirtc::cuStreamSynchronize_ld) {
|
|
||||||
minirtc::cuStreamSynchronize_ld(cuda_upload_stream);
|
|
||||||
}
|
|
||||||
for (auto &slot : cuda_upload_slots) {
|
|
||||||
if (slot.resource && minirtc::cuGraphicsUnregisterResource_ld) {
|
|
||||||
minirtc::cuGraphicsUnregisterResource_ld(slot.resource);
|
|
||||||
slot.resource = nullptr;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (cuda_upload_stream && minirtc::cuStreamDestroy_ld) {
|
|
||||||
minirtc::cuStreamDestroy_ld(cuda_upload_stream);
|
|
||||||
}
|
|
||||||
CUcontext popped_context = nullptr;
|
|
||||||
minirtc::cuCtxPopCurrent_ld(&popped_context);
|
|
||||||
}
|
|
||||||
cuda_upload_stream = nullptr;
|
|
||||||
cuda_context = nullptr;
|
cuda_context = nullptr;
|
||||||
cuda_context_owner.Reset();
|
|
||||||
|
|
||||||
for (auto &slot : cuda_upload_slots) {
|
for (auto &slot : cuda_upload_slots) {
|
||||||
|
DestroyCudaOpenGlVideoBuffer(slot.resource);
|
||||||
if (slot.buffer != 0 && gl.delete_buffers) {
|
if (slot.buffer != 0 && gl.delete_buffers) {
|
||||||
gl.delete_buffers(1, &slot.buffer);
|
gl.delete_buffers(1, &slot.buffer);
|
||||||
}
|
}
|
||||||
@@ -458,57 +385,32 @@ struct OpenGlVideoRenderer::Impl {
|
|||||||
|
|
||||||
bool PrepareCudaContext(const NativeVideoFrameRef &native_frame) {
|
bool PrepareCudaContext(const NativeVideoFrameRef &native_frame) {
|
||||||
const MiniRtcNativeVideoFrame *frame = native_frame.Get();
|
const MiniRtcNativeVideoFrame *frame = native_frame.Get();
|
||||||
if (!frame || !frame->payload.cuda_nv12.context ||
|
if (!frame || !frame->payload.cuda_nv12.context || cuda_interop_disabled) {
|
||||||
cuda_interop_disabled ||
|
|
||||||
!cuda_sync_functions_available ||
|
|
||||||
minirtc::LoadCudaGraphicsInterop() != 0) {
|
|
||||||
DisableCudaInterop();
|
DisableCudaInterop();
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
auto frame_context =
|
void *frame_context = frame->payload.cuda_nv12.context;
|
||||||
static_cast<CUcontext>(frame->payload.cuda_nv12.context);
|
|
||||||
if (cuda_context && cuda_context != frame_context) {
|
if (cuda_context && cuda_context != frame_context) {
|
||||||
ReleaseCudaInterop();
|
ReleaseCudaInterop();
|
||||||
}
|
}
|
||||||
if (!cuda_context) {
|
cuda_context = frame_context;
|
||||||
cuda_context_owner = native_frame;
|
|
||||||
cuda_context = frame_context;
|
|
||||||
if (minirtc::cuCtxPushCurrent_ld(cuda_context) != CUDA_SUCCESS) {
|
|
||||||
ReleaseCudaInterop();
|
|
||||||
DisableCudaInterop();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
const CUresult create_result = minirtc::cuStreamCreate_ld(
|
|
||||||
&cuda_upload_stream, CU_STREAM_NON_BLOCKING);
|
|
||||||
CUcontext popped_context = nullptr;
|
|
||||||
const CUresult pop_result =
|
|
||||||
minirtc::cuCtxPopCurrent_ld(&popped_context);
|
|
||||||
if (create_result != CUDA_SUCCESS || pop_result != CUDA_SUCCESS) {
|
|
||||||
ReleaseCudaInterop();
|
|
||||||
DisableCudaInterop();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool PrepareCudaSlot(CudaUploadSlot &slot, size_t required_size) {
|
bool PrepareCudaSlot(CudaUploadSlot &slot, size_t required_size,
|
||||||
|
const MiniRtcNativeVideoFrame *frame) {
|
||||||
if (slot.buffer == 0) {
|
if (slot.buffer == 0) {
|
||||||
gl.gen_buffers(1, &slot.buffer);
|
gl.gen_buffers(1, &slot.buffer);
|
||||||
}
|
}
|
||||||
if (slot.buffer == 0) {
|
if (slot.buffer == 0) {
|
||||||
|
LOG_WARN("CUDA/OpenGL glGenBuffers failed, error={}", glGetError());
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (slot.capacity < required_size) {
|
if (slot.capacity < required_size) {
|
||||||
if (slot.resource) {
|
DestroyCudaOpenGlVideoBuffer(slot.resource);
|
||||||
if (minirtc::cuGraphicsUnregisterResource_ld(slot.resource) !=
|
slot.resource = nullptr;
|
||||||
CUDA_SUCCESS) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
slot.resource = nullptr;
|
|
||||||
}
|
|
||||||
GLint previous_unpack_buffer = 0;
|
GLint previous_unpack_buffer = 0;
|
||||||
glGetIntegerv(GL_PIXEL_UNPACK_BUFFER_BINDING, &previous_unpack_buffer);
|
glGetIntegerv(GL_PIXEL_UNPACK_BUFFER_BINDING, &previous_unpack_buffer);
|
||||||
gl.bind_buffer(GL_PIXEL_UNPACK_BUFFER, slot.buffer);
|
gl.bind_buffer(GL_PIXEL_UNPACK_BUFFER, slot.buffer);
|
||||||
@@ -517,21 +419,18 @@ struct OpenGlVideoRenderer::Impl {
|
|||||||
GL_STREAM_DRAW);
|
GL_STREAM_DRAW);
|
||||||
gl.bind_buffer(GL_PIXEL_UNPACK_BUFFER,
|
gl.bind_buffer(GL_PIXEL_UNPACK_BUFFER,
|
||||||
static_cast<GLuint>(previous_unpack_buffer));
|
static_cast<GLuint>(previous_unpack_buffer));
|
||||||
if (glGetError() != GL_NO_ERROR) {
|
const GLenum buffer_error = glGetError();
|
||||||
|
if (buffer_error != GL_NO_ERROR) {
|
||||||
|
LOG_WARN("CUDA/OpenGL glBufferData failed, error={}", buffer_error);
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
slot.capacity = required_size;
|
slot.capacity = required_size;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!slot.resource) {
|
if (!slot.resource) {
|
||||||
if (minirtc::cuGraphicsGLRegisterBuffer_ld(
|
slot.resource = CreateCudaOpenGlVideoBuffer(frame, slot.buffer);
|
||||||
&slot.resource, slot.buffer,
|
|
||||||
CU_GRAPHICS_REGISTER_FLAGS_WRITE_DISCARD) != CUDA_SUCCESS) {
|
|
||||||
slot.resource = nullptr;
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return true;
|
return slot.resource != nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool PrepareCudaTextures(CudaUploadSlot &slot) {
|
bool PrepareCudaTextures(CudaUploadSlot &slot) {
|
||||||
@@ -577,6 +476,9 @@ struct OpenGlVideoRenderer::Impl {
|
|||||||
selected_index = index;
|
selected_index = index;
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
if (cuda_interop_disabled) {
|
||||||
|
return CudaUploadResult::unavailable;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
if (selected_index == kCudaUploadSlotCount) {
|
if (selected_index == kCudaUploadSlotCount) {
|
||||||
++cuda_busy_drop_count;
|
++cuda_busy_drop_count;
|
||||||
@@ -587,68 +489,11 @@ struct OpenGlVideoRenderer::Impl {
|
|||||||
return CudaUploadResult::busy;
|
return CudaUploadResult::busy;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (minirtc::cuCtxPushCurrent_ld(cuda_context) != CUDA_SUCCESS) {
|
|
||||||
DisableCudaInterop();
|
|
||||||
return CudaUploadResult::unavailable;
|
|
||||||
}
|
|
||||||
|
|
||||||
CudaUploadSlot &slot = cuda_upload_slots[selected_index];
|
CudaUploadSlot &slot = cuda_upload_slots[selected_index];
|
||||||
const size_t packed_size =
|
const size_t packed_size =
|
||||||
static_cast<size_t>(frame->width) * frame->height * 3U / 2U;
|
static_cast<size_t>(frame->width) * frame->height * 3U / 2U;
|
||||||
CUresult result = PrepareCudaSlot(slot, packed_size)
|
if (!PrepareCudaSlot(slot, packed_size, frame) ||
|
||||||
? CUDA_SUCCESS
|
UploadCudaFrameToOpenGlBuffer(slot.resource, frame) != 0) {
|
||||||
: CUDA_ERROR_INVALID_VALUE;
|
|
||||||
CUgraphicsResource resource = slot.resource;
|
|
||||||
bool mapped = false;
|
|
||||||
CUdeviceptr destination = 0;
|
|
||||||
size_t mapped_size = 0;
|
|
||||||
if (result == CUDA_SUCCESS) {
|
|
||||||
result = minirtc::cuGraphicsMapResources_ld(
|
|
||||||
1, &resource, cuda_upload_stream);
|
|
||||||
mapped = result == CUDA_SUCCESS;
|
|
||||||
}
|
|
||||||
if (result == CUDA_SUCCESS) {
|
|
||||||
result = minirtc::cuGraphicsResourceGetMappedPointer_ld(
|
|
||||||
&destination, &mapped_size, slot.resource);
|
|
||||||
}
|
|
||||||
if (result == CUDA_SUCCESS && mapped_size < packed_size) {
|
|
||||||
result = CUDA_ERROR_INVALID_VALUE;
|
|
||||||
}
|
|
||||||
if (result == CUDA_SUCCESS) {
|
|
||||||
CUDA_MEMCPY2D copy{};
|
|
||||||
copy.srcMemoryType = CU_MEMORYTYPE_DEVICE;
|
|
||||||
copy.srcDevice =
|
|
||||||
static_cast<CUdeviceptr>(
|
|
||||||
frame->payload.cuda_nv12.y_device_pointer);
|
|
||||||
copy.srcPitch = frame->payload.cuda_nv12.y_stride;
|
|
||||||
copy.dstMemoryType = CU_MEMORYTYPE_DEVICE;
|
|
||||||
copy.dstDevice = destination;
|
|
||||||
copy.dstPitch = frame->width;
|
|
||||||
copy.WidthInBytes = frame->width;
|
|
||||||
copy.Height = frame->height;
|
|
||||||
result = minirtc::cuMemcpy2DAsync_ld(©, cuda_upload_stream);
|
|
||||||
if (result == CUDA_SUCCESS) {
|
|
||||||
copy.srcDevice =
|
|
||||||
static_cast<CUdeviceptr>(
|
|
||||||
frame->payload.cuda_nv12.uv_device_pointer);
|
|
||||||
copy.srcPitch = frame->payload.cuda_nv12.uv_stride;
|
|
||||||
copy.dstDevice = destination +
|
|
||||||
static_cast<CUdeviceptr>(frame->width) * frame->height;
|
|
||||||
copy.Height = frame->height / 2U;
|
|
||||||
result = minirtc::cuMemcpy2DAsync_ld(©, cuda_upload_stream);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (mapped) {
|
|
||||||
const CUresult unmap_result = minirtc::cuGraphicsUnmapResources_ld(
|
|
||||||
1, &resource, cuda_upload_stream);
|
|
||||||
if (result == CUDA_SUCCESS) {
|
|
||||||
result = unmap_result;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
CUcontext popped_context = nullptr;
|
|
||||||
const CUresult pop_result =
|
|
||||||
minirtc::cuCtxPopCurrent_ld(&popped_context);
|
|
||||||
if (result != CUDA_SUCCESS || pop_result != CUDA_SUCCESS) {
|
|
||||||
DisableCudaInterop();
|
DisableCudaInterop();
|
||||||
return CudaUploadResult::unavailable;
|
return CudaUploadResult::unavailable;
|
||||||
}
|
}
|
||||||
@@ -674,9 +519,11 @@ struct OpenGlVideoRenderer::Impl {
|
|||||||
video_enabled_location = -1;
|
video_enabled_location = -1;
|
||||||
texture_width = 0;
|
texture_width = 0;
|
||||||
texture_height = 0;
|
texture_height = 0;
|
||||||
|
cpu_texture_width = 0;
|
||||||
|
cpu_texture_height = 0;
|
||||||
uploaded_stream.clear();
|
uploaded_stream.clear();
|
||||||
uploaded_sequence = 0;
|
uploaded_sequence = 0;
|
||||||
#if defined(_WIN32) && USE_CUDA
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
displayed_cuda_upload_slot = kCudaUploadSlotCount;
|
displayed_cuda_upload_slot = kCudaUploadSlotCount;
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
@@ -697,14 +544,37 @@ bool OpenGlVideoRenderer::Setup() {
|
|||||||
LOG_WARN("OpenGL NV12 underlay unavailable: required functions missing");
|
LOG_WARN("OpenGL NV12 underlay unavailable: required functions missing");
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
#if defined(_WIN32) && USE_CUDA
|
const auto *version = reinterpret_cast<const char *>(glGetString(GL_VERSION));
|
||||||
impl_->cuda_sync_functions_available = impl_->gl.LoadSyncFunctions();
|
const bool is_gles = version && std::strstr(version, "OpenGL ES") != nullptr;
|
||||||
impl_->cuda_interop_disabled = !impl_->cuda_sync_functions_available;
|
int major = 0;
|
||||||
impl_->cuda_cpu_fallback.store(!impl_->cuda_sync_functions_available,
|
int minor = 0;
|
||||||
std::memory_order_release);
|
if (version) {
|
||||||
|
std::sscanf(is_gles ? version + std::strlen("OpenGL ES ") : version,
|
||||||
|
"%d.%d", &major, &minor);
|
||||||
|
}
|
||||||
|
const auto *extensions = major < 3
|
||||||
|
? reinterpret_cast<const char *>(glGetString(GL_EXTENSIONS)) : nullptr;
|
||||||
|
impl_->pixel_unpack_buffer_available = major >= 3 ||
|
||||||
|
(!is_gles && ((major == 2 && minor >= 1) ||
|
||||||
|
HasOpenGlExtension(extensions, "GL_ARB_pixel_buffer_object")));
|
||||||
|
impl_->unpack_subimage_available = !is_gles || major >= 3 ||
|
||||||
|
HasOpenGlExtension(extensions, "GL_EXT_unpack_subimage");
|
||||||
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
|
// GLX may return entry points even if the current GLES 2 context cannot
|
||||||
|
// use them. Gate CUDA interop on context capabilities as well as symbols.
|
||||||
|
const bool sync_available = is_gles ? major >= 3
|
||||||
|
: major > 3 || (major == 3 && minor >= 2) ||
|
||||||
|
HasOpenGlExtension(extensions, "GL_ARB_sync");
|
||||||
|
impl_->cuda_interop_disabled = !impl_->pixel_unpack_buffer_available ||
|
||||||
|
!sync_available || !impl_->gl.LoadSyncFunctions();
|
||||||
|
impl_->cuda_cpu_fallback.store(impl_->cuda_interop_disabled,
|
||||||
|
std::memory_order_release);
|
||||||
impl_->cuda_native_logged = false;
|
impl_->cuda_native_logged = false;
|
||||||
impl_->cuda_fallback_logged = false;
|
impl_->cuda_fallback_logged = false;
|
||||||
impl_->cuda_busy_drop_count = 0;
|
impl_->cuda_busy_drop_count = 0;
|
||||||
|
if (impl_->cuda_interop_disabled) {
|
||||||
|
LOG_WARN("CUDA/OpenGL sync functions unavailable; using CPU fallback");
|
||||||
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
static constexpr char kGlesVertexShader[] = R"glsl(
|
static constexpr char kGlesVertexShader[] = R"glsl(
|
||||||
@@ -815,8 +685,6 @@ void main() {
|
|||||||
}
|
}
|
||||||
)glsl";
|
)glsl";
|
||||||
|
|
||||||
const auto *version = reinterpret_cast<const char *>(glGetString(GL_VERSION));
|
|
||||||
const bool is_gles = version && std::strstr(version, "OpenGL ES") != nullptr;
|
|
||||||
const char *vertex_source =
|
const char *vertex_source =
|
||||||
is_gles ? kGlesVertexShader : kDesktopVertexShader;
|
is_gles ? kGlesVertexShader : kDesktopVertexShader;
|
||||||
const char *fragment_source =
|
const char *fragment_source =
|
||||||
@@ -939,14 +807,17 @@ void main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl_->ready.store(true, std::memory_order_release);
|
impl_->ready.store(true, std::memory_order_release);
|
||||||
LOG_INFO("OpenGL NV12 underlay initialized ({})",
|
const auto *vendor = reinterpret_cast<const char *>(glGetString(GL_VENDOR));
|
||||||
version ? version : "unknown OpenGL version");
|
const auto *renderer = reinterpret_cast<const char *>(glGetString(GL_RENDERER));
|
||||||
|
LOG_INFO("OpenGL NV12 underlay initialized ({}), vendor={}, renderer={}",
|
||||||
|
version ? version : "unknown OpenGL version",
|
||||||
|
vendor ? vendor : "unknown", renderer ? renderer : "unknown");
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
void OpenGlVideoRenderer::Teardown() {
|
void OpenGlVideoRenderer::Teardown() {
|
||||||
impl_->ready.store(false, std::memory_order_release);
|
impl_->ready.store(false, std::memory_order_release);
|
||||||
#if defined(_WIN32) && USE_CUDA
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
impl_->ReleaseCudaInterop();
|
impl_->ReleaseCudaInterop();
|
||||||
#endif
|
#endif
|
||||||
if (impl_->y_texture != 0)
|
if (impl_->y_texture != 0)
|
||||||
@@ -1143,7 +1014,7 @@ OpenGlVideoRenderer::SubmitNativeFrame(std::string_view remote_id,
|
|||||||
return SubmitResult::dropped;
|
return SubmitResult::dropped;
|
||||||
}
|
}
|
||||||
|
|
||||||
#if defined(_WIN32) && USE_CUDA
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
if (native->type == MiniRtcNativeVideoFrameCudaNv12 &&
|
if (native->type == MiniRtcNativeVideoFrameCudaNv12 &&
|
||||||
impl_->cuda_cpu_fallback.load(std::memory_order_acquire)) {
|
impl_->cuda_cpu_fallback.load(std::memory_order_acquire)) {
|
||||||
const size_t required_size =
|
const size_t required_size =
|
||||||
@@ -1152,27 +1023,18 @@ OpenGlVideoRenderer::SubmitNativeFrame(std::string_view remote_id,
|
|||||||
if (native->copy_to_nv12(native->owner, target->bytes.data(),
|
if (native->copy_to_nv12(native->owner, target->bytes.data(),
|
||||||
target->bytes.size()) != 0) {
|
target->bytes.size()) != 0) {
|
||||||
target->bytes.clear();
|
target->bytes.clear();
|
||||||
|
target->native_frame.Reset();
|
||||||
target->valid = false;
|
target->valid = false;
|
||||||
target->use = SlotUse::available;
|
target->use = SlotUse::available;
|
||||||
return SubmitResult::failed;
|
return SubmitResult::failed;
|
||||||
}
|
}
|
||||||
target->native_frame.Reset();
|
target->native_frame.Reset();
|
||||||
target->width = static_cast<int>(native->width);
|
} else
|
||||||
target->height = static_cast<int>(native->height);
|
|
||||||
target->remote_id.assign(remote_id);
|
|
||||||
target->sequence = frames.next_sequence++;
|
|
||||||
target->use = SlotUse::pending;
|
|
||||||
target->valid = true;
|
|
||||||
return SubmitResult::submitted;
|
|
||||||
}
|
|
||||||
#endif
|
#endif
|
||||||
|
{
|
||||||
NativeVideoFrameRef retained(native);
|
target->bytes.clear();
|
||||||
if (!retained) {
|
target->native_frame = NativeVideoFrameRef(native);
|
||||||
return SubmitResult::failed;
|
|
||||||
}
|
}
|
||||||
target->bytes.clear();
|
|
||||||
target->native_frame = std::move(retained);
|
|
||||||
target->width = static_cast<int>(native->width);
|
target->width = static_cast<int>(native->width);
|
||||||
target->height = static_cast<int>(native->height);
|
target->height = static_cast<int>(native->height);
|
||||||
target->remote_id.assign(remote_id);
|
target->remote_id.assign(remote_id);
|
||||||
@@ -1235,10 +1097,11 @@ OpenGlVideoRenderer::RenderLatest(std::string_view remote_id,
|
|||||||
GLint previous_program = 0;
|
GLint previous_program = 0;
|
||||||
GLint previous_active_texture = GL_TEXTURE0;
|
GLint previous_active_texture = GL_TEXTURE0;
|
||||||
GLint previous_array_buffer = 0;
|
GLint previous_array_buffer = 0;
|
||||||
#if defined(_WIN32) && USE_CUDA
|
|
||||||
GLint previous_unpack_buffer = 0;
|
GLint previous_unpack_buffer = 0;
|
||||||
#endif
|
|
||||||
GLint previous_unpack_alignment = 4;
|
GLint previous_unpack_alignment = 4;
|
||||||
|
GLint previous_unpack_row_length = 0;
|
||||||
|
GLint previous_unpack_skip_rows = 0;
|
||||||
|
GLint previous_unpack_skip_pixels = 0;
|
||||||
GLint previous_viewport[4] = {};
|
GLint previous_viewport[4] = {};
|
||||||
GLint previous_scissor_box[4] = {};
|
GLint previous_scissor_box[4] = {};
|
||||||
GLfloat previous_clear_color[4] = {};
|
GLfloat previous_clear_color[4] = {};
|
||||||
@@ -1246,9 +1109,18 @@ OpenGlVideoRenderer::RenderLatest(std::string_view remote_id,
|
|||||||
glGetIntegerv(GL_CURRENT_PROGRAM, &previous_program);
|
glGetIntegerv(GL_CURRENT_PROGRAM, &previous_program);
|
||||||
glGetIntegerv(GL_ACTIVE_TEXTURE, &previous_active_texture);
|
glGetIntegerv(GL_ACTIVE_TEXTURE, &previous_active_texture);
|
||||||
glGetIntegerv(GL_ARRAY_BUFFER_BINDING, &previous_array_buffer);
|
glGetIntegerv(GL_ARRAY_BUFFER_BINDING, &previous_array_buffer);
|
||||||
#if defined(_WIN32) && USE_CUDA
|
if (impl_->pixel_unpack_buffer_available) {
|
||||||
glGetIntegerv(GL_PIXEL_UNPACK_BUFFER_BINDING, &previous_unpack_buffer);
|
glGetIntegerv(GL_PIXEL_UNPACK_BUFFER_BINDING, &previous_unpack_buffer);
|
||||||
#endif
|
impl_->gl.bind_buffer(GL_PIXEL_UNPACK_BUFFER, 0);
|
||||||
|
}
|
||||||
|
if (impl_->unpack_subimage_available) {
|
||||||
|
glGetIntegerv(GL_UNPACK_ROW_LENGTH, &previous_unpack_row_length);
|
||||||
|
glGetIntegerv(GL_UNPACK_SKIP_ROWS, &previous_unpack_skip_rows);
|
||||||
|
glGetIntegerv(GL_UNPACK_SKIP_PIXELS, &previous_unpack_skip_pixels);
|
||||||
|
glPixelStorei(GL_UNPACK_ROW_LENGTH, 0);
|
||||||
|
glPixelStorei(GL_UNPACK_SKIP_ROWS, 0);
|
||||||
|
glPixelStorei(GL_UNPACK_SKIP_PIXELS, 0);
|
||||||
|
}
|
||||||
glGetIntegerv(GL_UNPACK_ALIGNMENT, &previous_unpack_alignment);
|
glGetIntegerv(GL_UNPACK_ALIGNMENT, &previous_unpack_alignment);
|
||||||
glGetIntegerv(GL_VIEWPORT, previous_viewport);
|
glGetIntegerv(GL_VIEWPORT, previous_viewport);
|
||||||
glGetIntegerv(GL_SCISSOR_BOX, previous_scissor_box);
|
glGetIntegerv(GL_SCISSOR_BOX, previous_scissor_box);
|
||||||
@@ -1302,7 +1174,7 @@ OpenGlVideoRenderer::RenderLatest(std::string_view remote_id,
|
|||||||
y_data = native->payload.cpu_nv12.y_plane;
|
y_data = native->payload.cpu_nv12.y_plane;
|
||||||
uv_data = native->payload.cpu_nv12.uv_plane;
|
uv_data = native->payload.cpu_nv12.uv_plane;
|
||||||
} else if (native->type == MiniRtcNativeVideoFrameCudaNv12) {
|
} else if (native->type == MiniRtcNativeVideoFrameCudaNv12) {
|
||||||
#if defined(_WIN32) && USE_CUDA
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
const auto cuda_result = impl_->CopyCudaFrameToUploadBuffer(
|
const auto cuda_result = impl_->CopyCudaFrameToUploadBuffer(
|
||||||
active_native_frame, &cuda_upload_slot);
|
active_native_frame, &cuda_upload_slot);
|
||||||
use_pixel_unpack_buffer =
|
use_pixel_unpack_buffer =
|
||||||
@@ -1310,7 +1182,7 @@ OpenGlVideoRenderer::RenderLatest(std::string_view remote_id,
|
|||||||
should_upload =
|
should_upload =
|
||||||
cuda_result != OpenGlVideoRenderer::Impl::CudaUploadResult::busy;
|
cuda_result != OpenGlVideoRenderer::Impl::CudaUploadResult::busy;
|
||||||
if (use_pixel_unpack_buffer && !impl_->cuda_native_logged) {
|
if (use_pixel_unpack_buffer && !impl_->cuda_native_logged) {
|
||||||
LOG_INFO("Windows native NV12 using asynchronous CUDA/OpenGL ring");
|
LOG_INFO("Native NV12 using asynchronous CUDA/OpenGL ring");
|
||||||
impl_->cuda_native_logged = true;
|
impl_->cuda_native_logged = true;
|
||||||
}
|
}
|
||||||
if (should_upload && !use_pixel_unpack_buffer &&
|
if (should_upload && !use_pixel_unpack_buffer &&
|
||||||
@@ -1338,7 +1210,7 @@ OpenGlVideoRenderer::RenderLatest(std::string_view remote_id,
|
|||||||
|
|
||||||
GLuint upload_y_texture = impl_->y_texture;
|
GLuint upload_y_texture = impl_->y_texture;
|
||||||
GLuint upload_uv_texture = impl_->uv_texture;
|
GLuint upload_uv_texture = impl_->uv_texture;
|
||||||
#if defined(_WIN32) && USE_CUDA
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
if (should_upload && upload_succeeded && use_pixel_unpack_buffer) {
|
if (should_upload && upload_succeeded && use_pixel_unpack_buffer) {
|
||||||
auto &cuda_slot = impl_->cuda_upload_slots[cuda_upload_slot];
|
auto &cuda_slot = impl_->cuda_upload_slots[cuda_upload_slot];
|
||||||
upload_succeeded = impl_->PrepareCudaTextures(cuda_slot);
|
upload_succeeded = impl_->PrepareCudaTextures(cuda_slot);
|
||||||
@@ -1356,7 +1228,7 @@ OpenGlVideoRenderer::RenderLatest(std::string_view remote_id,
|
|||||||
impl_->gl.active_texture(GL_TEXTURE1);
|
impl_->gl.active_texture(GL_TEXTURE1);
|
||||||
glBindTexture(GL_TEXTURE_2D, upload_uv_texture);
|
glBindTexture(GL_TEXTURE_2D, upload_uv_texture);
|
||||||
if (use_pixel_unpack_buffer) {
|
if (use_pixel_unpack_buffer) {
|
||||||
#if defined(_WIN32) && USE_CUDA
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
impl_->gl.bind_buffer(
|
impl_->gl.bind_buffer(
|
||||||
GL_PIXEL_UNPACK_BUFFER,
|
GL_PIXEL_UNPACK_BUFFER,
|
||||||
impl_->cuda_upload_slots[cuda_upload_slot].buffer);
|
impl_->cuda_upload_slots[cuda_upload_slot].buffer);
|
||||||
@@ -1369,9 +1241,9 @@ OpenGlVideoRenderer::RenderLatest(std::string_view remote_id,
|
|||||||
use_pixel_unpack_buffer
|
use_pixel_unpack_buffer
|
||||||
? reinterpret_cast<const void *>(static_cast<uintptr_t>(y_size))
|
? reinterpret_cast<const void *>(static_cast<uintptr_t>(y_size))
|
||||||
: static_cast<const void *>(uv_data);
|
: static_cast<const void *>(uv_data);
|
||||||
int uploaded_texture_width = impl_->texture_width;
|
int uploaded_texture_width = impl_->cpu_texture_width;
|
||||||
int uploaded_texture_height = impl_->texture_height;
|
int uploaded_texture_height = impl_->cpu_texture_height;
|
||||||
#if defined(_WIN32) && USE_CUDA
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
if (use_pixel_unpack_buffer) {
|
if (use_pixel_unpack_buffer) {
|
||||||
uploaded_texture_width =
|
uploaded_texture_width =
|
||||||
impl_->cuda_upload_slots[cuda_upload_slot].texture_width;
|
impl_->cuda_upload_slots[cuda_upload_slot].texture_width;
|
||||||
@@ -1398,12 +1270,6 @@ OpenGlVideoRenderer::RenderLatest(std::string_view remote_id,
|
|||||||
source_height / 2, GL_LUMINANCE_ALPHA,
|
source_height / 2, GL_LUMINANCE_ALPHA,
|
||||||
GL_UNSIGNED_BYTE, uv_pixels);
|
GL_UNSIGNED_BYTE, uv_pixels);
|
||||||
}
|
}
|
||||||
if (use_pixel_unpack_buffer) {
|
|
||||||
#if defined(_WIN32) && USE_CUDA
|
|
||||||
impl_->gl.bind_buffer(GL_PIXEL_UNPACK_BUFFER,
|
|
||||||
static_cast<GLuint>(previous_unpack_buffer));
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
const GLenum upload_error = glGetError();
|
const GLenum upload_error = glGetError();
|
||||||
upload_succeeded = upload_error == GL_NO_ERROR;
|
upload_succeeded = upload_error == GL_NO_ERROR;
|
||||||
if (upload_succeeded) {
|
if (upload_succeeded) {
|
||||||
@@ -1411,7 +1277,11 @@ OpenGlVideoRenderer::RenderLatest(std::string_view remote_id,
|
|||||||
impl_->texture_height = source_height;
|
impl_->texture_height = source_height;
|
||||||
impl_->uploaded_stream.assign(remote_id);
|
impl_->uploaded_stream.assign(remote_id);
|
||||||
impl_->uploaded_sequence = sequence;
|
impl_->uploaded_sequence = sequence;
|
||||||
#if defined(_WIN32) && USE_CUDA
|
if (!use_pixel_unpack_buffer) {
|
||||||
|
impl_->cpu_texture_width = source_width;
|
||||||
|
impl_->cpu_texture_height = source_height;
|
||||||
|
}
|
||||||
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
if (use_pixel_unpack_buffer) {
|
if (use_pixel_unpack_buffer) {
|
||||||
auto &cuda_slot = impl_->cuda_upload_slots[cuda_upload_slot];
|
auto &cuda_slot = impl_->cuda_upload_slots[cuda_upload_slot];
|
||||||
cuda_slot.texture_width = source_width;
|
cuda_slot.texture_width = source_width;
|
||||||
@@ -1423,6 +1293,13 @@ OpenGlVideoRenderer::RenderLatest(std::string_view remote_id,
|
|||||||
#endif
|
#endif
|
||||||
} else {
|
} else {
|
||||||
LOG_WARN("OpenGL NV12 texture upload failed, error={}", upload_error);
|
LOG_WARN("OpenGL NV12 texture upload failed, error={}", upload_error);
|
||||||
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
|
if (use_pixel_unpack_buffer) {
|
||||||
|
// No draw fence will cover this failed upload. The upload handle retains its source
|
||||||
|
// until the upload handle is destroyed.
|
||||||
|
impl_->DisableCudaInterop();
|
||||||
|
}
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1470,7 +1347,7 @@ OpenGlVideoRenderer::RenderLatest(std::string_view remote_id,
|
|||||||
}
|
}
|
||||||
GLuint displayed_y_texture = impl_->y_texture;
|
GLuint displayed_y_texture = impl_->y_texture;
|
||||||
GLuint displayed_uv_texture = impl_->uv_texture;
|
GLuint displayed_uv_texture = impl_->uv_texture;
|
||||||
#if defined(_WIN32) && USE_CUDA
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
if (impl_->displayed_cuda_upload_slot < impl_->cuda_upload_slots.size()) {
|
if (impl_->displayed_cuda_upload_slot < impl_->cuda_upload_slots.size()) {
|
||||||
const auto &cuda_slot =
|
const auto &cuda_slot =
|
||||||
impl_->cuda_upload_slots[impl_->displayed_cuda_upload_slot];
|
impl_->cuda_upload_slots[impl_->displayed_cuda_upload_slot];
|
||||||
@@ -1510,7 +1387,7 @@ OpenGlVideoRenderer::RenderLatest(std::string_view remote_id,
|
|||||||
impl_->gl.enable_vertex_attrib_array(
|
impl_->gl.enable_vertex_attrib_array(
|
||||||
static_cast<GLuint>(impl_->texcoord_location));
|
static_cast<GLuint>(impl_->texcoord_location));
|
||||||
glDrawArrays(GL_TRIANGLE_STRIP, 0, 4);
|
glDrawArrays(GL_TRIANGLE_STRIP, 0, 4);
|
||||||
#if defined(_WIN32) && USE_CUDA
|
#if CROSSDESK_OPENGL_CUDA_INTEROP
|
||||||
if (impl_->displayed_cuda_upload_slot < impl_->cuda_upload_slots.size()) {
|
if (impl_->displayed_cuda_upload_slot < impl_->cuda_upload_slots.size()) {
|
||||||
auto &cuda_slot =
|
auto &cuda_slot =
|
||||||
impl_->cuda_upload_slots[impl_->displayed_cuda_upload_slot];
|
impl_->cuda_upload_slots[impl_->displayed_cuda_upload_slot];
|
||||||
@@ -1539,10 +1416,15 @@ OpenGlVideoRenderer::RenderLatest(std::string_view remote_id,
|
|||||||
texcoord_state);
|
texcoord_state);
|
||||||
impl_->gl.bind_buffer(GL_ARRAY_BUFFER,
|
impl_->gl.bind_buffer(GL_ARRAY_BUFFER,
|
||||||
static_cast<GLuint>(previous_array_buffer));
|
static_cast<GLuint>(previous_array_buffer));
|
||||||
#if defined(_WIN32) && USE_CUDA
|
if (impl_->pixel_unpack_buffer_available) {
|
||||||
impl_->gl.bind_buffer(GL_PIXEL_UNPACK_BUFFER,
|
impl_->gl.bind_buffer(GL_PIXEL_UNPACK_BUFFER,
|
||||||
static_cast<GLuint>(previous_unpack_buffer));
|
static_cast<GLuint>(previous_unpack_buffer));
|
||||||
#endif
|
}
|
||||||
|
if (impl_->unpack_subimage_available) {
|
||||||
|
glPixelStorei(GL_UNPACK_ROW_LENGTH, previous_unpack_row_length);
|
||||||
|
glPixelStorei(GL_UNPACK_SKIP_ROWS, previous_unpack_skip_rows);
|
||||||
|
glPixelStorei(GL_UNPACK_SKIP_PIXELS, previous_unpack_skip_pixels);
|
||||||
|
}
|
||||||
impl_->gl.active_texture(GL_TEXTURE0);
|
impl_->gl.active_texture(GL_TEXTURE0);
|
||||||
glBindTexture(GL_TEXTURE_2D, static_cast<GLuint>(previous_texture_0));
|
glBindTexture(GL_TEXTURE_2D, static_cast<GLuint>(previous_texture_0));
|
||||||
impl_->gl.active_texture(GL_TEXTURE1);
|
impl_->gl.active_texture(GL_TEXTURE1);
|
||||||
|
|||||||
@@ -0,0 +1,53 @@
|
|||||||
|
#include "platform/linux/gui/cuda_graphics.h"
|
||||||
|
|
||||||
|
#include <cstdlib>
|
||||||
|
#include <initializer_list>
|
||||||
|
|
||||||
|
#include "rd_log.h"
|
||||||
|
|
||||||
|
#include "platform/common/gui/cuda_opengl_interop.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
void ConfigureLinuxCudaGraphics(bool hardware_acceleration) {
|
||||||
|
#if USE_CUDA && (defined(__x86_64__) || defined(__amd64__))
|
||||||
|
if (!hardware_acceleration) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Respect explicit GPU/driver selection, including a request to use Mesa
|
||||||
|
// or software rendering. Native frames can still use the CPU fallback.
|
||||||
|
for (const char* name : {"__NV_PRIME_RENDER_OFFLOAD",
|
||||||
|
"__NV_PRIME_RENDER_OFFLOAD_PROVIDER",
|
||||||
|
"__GLX_VENDOR_LIBRARY_NAME",
|
||||||
|
"__EGL_VENDOR_LIBRARY_FILENAMES",
|
||||||
|
"__EGL_VENDOR_LIBRARY_DIRS", "DRI_PRIME",
|
||||||
|
"LIBGL_ALWAYS_SOFTWARE",
|
||||||
|
"MESA_LOADER_DRIVER_OVERRIDE"}) {
|
||||||
|
const char* value = std::getenv(name);
|
||||||
|
if (value && value[0] != '\0') {
|
||||||
|
LOG_INFO("Keeping explicit Linux graphics selection: {}={}", name,
|
||||||
|
value);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!IsCudaDeviceAvailable()) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Both EGL and GLX need to render on NVIDIA to share CUDA frame buffers.
|
||||||
|
// These are process-local, not desktop settings.
|
||||||
|
// https://download.nvidia.com/XFree86/Linux-x86_64/595.45.04/README/primerenderoffload.html
|
||||||
|
if (setenv("__NV_PRIME_RENDER_OFFLOAD", "1", 1) != 0 ||
|
||||||
|
setenv("__GLX_VENDOR_LIBRARY_NAME", "nvidia", 1) != 0) {
|
||||||
|
LOG_WARN("Unable to select NVIDIA PRIME rendering for CUDA video");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
LOG_INFO("Selected NVIDIA PRIME rendering for CUDA/OpenGL video interop");
|
||||||
|
#else
|
||||||
|
(void)hardware_acceleration;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
@@ -0,0 +1,18 @@
|
|||||||
|
/*
|
||||||
|
* @Author: DI JUNKUN
|
||||||
|
* @Date: 2026-09-07
|
||||||
|
* Copyright (c) 2026 by DI JUNKUN, All Rights Reserved.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef _CUDA_GRAPHICS_H_
|
||||||
|
#define _CUDA_GRAPHICS_H_
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
// Call before SDL or Slint initializes EGL/GLX. Hybrid laptops otherwise
|
||||||
|
// create the GUI context on the integrated GPU while NVDEC uses NVIDIA.
|
||||||
|
void ConfigureLinuxCudaGraphics(bool hardware_acceleration);
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
|
|
||||||
|
#endif
|
||||||
@@ -333,6 +333,8 @@ function setup_targets()
|
|||||||
"apps/desktop/src/platform/windows/gui/slint_backend.cpp",
|
"apps/desktop/src/platform/windows/gui/slint_backend.cpp",
|
||||||
"apps/desktop/src/platform/windows/gui/slint_renderer_probe.cpp",
|
"apps/desktop/src/platform/windows/gui/slint_renderer_probe.cpp",
|
||||||
"apps/desktop/src/platform/common/gui/opengl_video_renderer.cpp",
|
"apps/desktop/src/platform/common/gui/opengl_video_renderer.cpp",
|
||||||
|
"apps/desktop/src/platform/common/gui/cuda_driver.cpp",
|
||||||
|
"apps/desktop/src/platform/common/gui/cuda_opengl_interop.cpp",
|
||||||
"apps/desktop/src/platform/common/gui/video_renderer_factory_opengl.cpp",
|
"apps/desktop/src/platform/common/gui/video_renderer_factory_opengl.cpp",
|
||||||
"apps/desktop/src/platform/windows/gui/runtime/windows_service_runtime.cpp",
|
"apps/desktop/src/platform/windows/gui/runtime/windows_service_runtime.cpp",
|
||||||
"apps/desktop/src/platform/windows/gui/application/portable_service_integration.cpp")
|
"apps/desktop/src/platform/windows/gui/application/portable_service_integration.cpp")
|
||||||
@@ -350,7 +352,10 @@ function setup_targets()
|
|||||||
elseif is_os("linux") then
|
elseif is_os("linux") then
|
||||||
add_links("GL")
|
add_links("GL")
|
||||||
add_files("apps/desktop/src/platform/linux/gui/tray/linux_tray.cpp",
|
add_files("apps/desktop/src/platform/linux/gui/tray/linux_tray.cpp",
|
||||||
|
"apps/desktop/src/platform/linux/gui/cuda_graphics.cpp",
|
||||||
"apps/desktop/src/platform/common/gui/opengl_video_renderer.cpp",
|
"apps/desktop/src/platform/common/gui/opengl_video_renderer.cpp",
|
||||||
|
"apps/desktop/src/platform/common/gui/cuda_driver.cpp",
|
||||||
|
"apps/desktop/src/platform/common/gui/cuda_opengl_interop.cpp",
|
||||||
"apps/desktop/src/platform/common/gui/video_renderer_factory_opengl.cpp")
|
"apps/desktop/src/platform/common/gui/video_renderer_factory_opengl.cpp")
|
||||||
add_includedirs("apps/desktop/src/platform/common/gui",
|
add_includedirs("apps/desktop/src/platform/common/gui",
|
||||||
"apps/desktop/src/platform/linux/gui/tray",
|
"apps/desktop/src/platform/linux/gui/tray",
|
||||||
|
|||||||
Vendored
+1
-1
Submodule deps/submodules/minirtc updated: 645477187a...6c39589408
Reference in New Issue
Block a user