gstreamer/subprojects/gst-plugins-bad/sys/nvcodec/gstcudanvmm.c
Seungha Yang e5132a8508 cudaupload,cudadownload: Add support for dGPU NVMM
Implement NVMM <-> CUDA, GL, SYSTEM memory conversion. Jetson is
not supported yet. Note that NVMM <-> GL interop on Jetson platform
is supported by GstGL

Some example pipelines are:
- Convert NVMM to GstGL memory
  nvv4l2decoder ! "video/x-raw(memory:NVMM)" ! cudadownload ! "video/x-raw(memory:GLMemory)" ! glimagesink

- Upload system memory to NVMM and encode
  video/x-raw,format=NV12 ! cudaupload ! "video/x-raw(memory:NVMM)" ! nvv4l2h264enc

- Convert NVMM to GstCUDA memory and encode
  nvvideoconvert ! "video/x-raw(memory:NVMM)" ! cudaupload ! "video/x-raw(memory:CUDAMemory)" ! nvh264enc

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/1865>
2022-03-07 19:05:29 +00:00

101 lines
2.4 KiB
C

/* GStreamer
* Copyright (C) 2022 Seungha Yang <seungha@centricular.com>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "gstcudanvmm.h"
#include <gmodule.h>
#include <string.h>
GST_DEBUG_CATEGORY_EXTERN (gst_cuda_nvmm_debug);
#define GST_CAT_DEFAULT gst_cuda_nvmm_debug
#define LOAD_SYMBOL(name) G_STMT_START { \
if (!g_module_symbol (module, G_STRINGIFY (name), (gpointer *) &vtable->name)) { \
GST_ERROR ("Failed to load symbol '%s', %s", G_STRINGIFY (name), g_module_error()); \
goto error; \
} \
} G_STMT_END;
/* *INDENT-OFF* */
typedef struct _GstCudaNvmmVTable
{
gboolean loaded;
GstBufferPool * (*gst_nvds_buffer_pool_new) (void);
} GstCudaNvmmVTable;
/* *INDENT-ON* */
static GstCudaNvmmVTable gst_cuda_nvmm_vtable = { 0, };
static gboolean
gst_cuda_nvmm_load_library (void)
{
GModule *module;
GstCudaNvmmVTable *vtable;
if (gst_cuda_nvmm_vtable.loaded)
return TRUE;
module = g_module_open ("libnvdsbufferpool.so", G_MODULE_BIND_LAZY);
if (!module) {
GST_INFO ("libnvdsbufferpool library is unavailable");
return FALSE;
}
vtable = &gst_cuda_nvmm_vtable;
LOAD_SYMBOL (gst_nvds_buffer_pool_new);
vtable->loaded = TRUE;
return TRUE;
error:
g_module_close (module);
return FALSE;
}
gboolean
gst_cuda_nvmm_init_once (void)
{
static gboolean loaded = FALSE;
static gsize load_once = 0;
if (g_once_init_enter (&load_once)) {
loaded = gst_cuda_nvmm_load_library ();
g_once_init_leave (&load_once, 1);
}
return loaded;
}
GstBufferPool *
gst_cuda_nvmm_buffer_pool_new (void)
{
if (!gst_cuda_nvmm_init_once ())
return NULL;
g_assert (gst_cuda_nvmm_vtable.gst_nvds_buffer_pool_new != NULL);
return gst_cuda_nvmm_vtable.gst_nvds_buffer_pool_new ();
}