mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2025-01-10 17:35:59 +00:00
Code Issues 1 Projects Releases Wiki Activity

cudabufferpool: Add support for CUDA stream use in memory

Browse source 
* Use GstCudaPoolAllocator
* Pass configured GstCudaStream object to allocator

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/3629>
This commit is contained in:
Seungha Yang 2022-12-23 00:37:36 +09:00 committed by GStreamer Marge Bot
parent 30d06e03c2
commit d409c35367
2 changed files with 107 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

106
subprojects/gst-plugins-bad/gst-libs/gst/cuda/gstcudabufferpool.c
View file

@ -31,6 +31,8 @@ GST_DEBUG_CATEGORY_STATIC (gst_cuda_buffer_pool_debug);
struct _GstCudaBufferPoolPrivate
{
GstVideoInfo info;
GstCudaStream *stream;
GstCudaPoolAllocator *alloc;
};
#define gst_cuda_buffer_pool_parent_class parent_class
@ -73,6 +75,20 @@ gst_cuda_buffer_pool_set_config (GstBufferPool * pool, GstStructure * config)
return FALSE;
}
if (priv->alloc) {
gst_cuda_allocator_set_active (GST_CUDA_ALLOCATOR (priv->alloc), FALSE);
gst_object_unref (priv->alloc);
}
gst_clear_cuda_stream (&priv->stream);
priv->stream = gst_buffer_pool_config_get_cuda_stream (config);
priv->alloc = gst_cuda_pool_allocator_new (self->context, priv->stream,
&info);
if (!priv->alloc) {
GST_ERROR_OBJECT (self, "Couldn't create allocator");
return FALSE;
}
mem = gst_cuda_allocator_alloc (NULL, self->context, NULL, &info);
if (!mem) {
GST_WARNING_OBJECT (self, "Failed to allocate memory");
@ -101,19 +117,19 @@ gst_cuda_buffer_pool_alloc (GstBufferPool * pool, GstBuffer ** buffer,
GstBuffer *buf;
GstMemory *mem;
GstCudaMemory *cmem;
GstFlowReturn ret;
mem = gst_cuda_allocator_alloc (NULL, self->context, NULL, &priv->info);
if (!mem) {
GST_WARNING_OBJECT (pool, "Cannot create CUDA memory");
return GST_FLOW_ERROR;
ret = gst_cuda_pool_allocator_acquire_memory (priv->alloc, &mem);
if (ret != GST_FLOW_OK) {
GST_WARNING_OBJECT (self, "Couldn't acquire memory");
return ret;
}
cmem = GST_CUDA_MEMORY_CAST (mem);
buf = gst_buffer_new ();
gst_buffer_append_memory (buf, mem);
cmem = GST_CUDA_MEMORY_CAST (mem);
GST_DEBUG_OBJECT (pool, "adding GstVideoMeta");
gst_buffer_add_video_meta_full (buf, GST_VIDEO_FRAME_FLAG_NONE,
GST_VIDEO_INFO_FORMAT (info), GST_VIDEO_INFO_WIDTH (info),
@ -125,6 +141,32 @@ gst_cuda_buffer_pool_alloc (GstBufferPool * pool, GstBuffer ** buffer,
return GST_FLOW_OK;
}
static gboolean
gst_cuda_buffer_pool_start (GstBufferPool * pool)
{
GstCudaBufferPool *self = GST_CUDA_BUFFER_POOL_CAST (pool);
GstCudaBufferPoolPrivate *priv = self->priv;
if (!gst_cuda_allocator_set_active (GST_CUDA_ALLOCATOR (priv->alloc), TRUE)) {
GST_ERROR_OBJECT (self, "Couldn't activate allocator");
return FALSE;
}
return GST_BUFFER_POOL_CLASS (parent_class)->start (pool);
}
static gboolean
gst_cuda_buffer_pool_stop (GstBufferPool * pool)
{
GstCudaBufferPool *self = GST_CUDA_BUFFER_POOL_CAST (pool);
GstCudaBufferPoolPrivate *priv = self->priv;
if (priv->alloc)
gst_cuda_allocator_set_active (GST_CUDA_ALLOCATOR (priv->alloc), FALSE);
return GST_BUFFER_POOL_CLASS (parent_class)->stop (pool);
}
/**
* gst_cuda_buffer_pool_new:
* @context: The #GstCudaContext to use for the new buffer pool
@ -148,11 +190,59 @@ gst_cuda_buffer_pool_new (GstCudaContext * context)
return GST_BUFFER_POOL_CAST (self);
}
/**
* gst_buffer_pool_config_get_cuda_stream:
* @config: a buffer pool config
*
* Returns: (transfer full) (nullable): the currently configured #GstCudaStream
* on @config or %NULL if @config doesn't hold #GstCudaStream
*
* Since: 1.24
*/
GstCudaStream *
gst_buffer_pool_config_get_cuda_stream (GstStructure * config)
{
GstCudaStream *stream = NULL;
g_return_val_if_fail (config, NULL);
gst_structure_get (config, "cuda-stream", GST_TYPE_CUDA_STREAM, &stream,
NULL);
return stream;
}
/**
* gst_buffer_pool_config_set_cuda_stream:
* @config: a buffer pool config
* @stream: a #GstCudaStream
*
* Sets @stream on @config
*
* Since: 1.24
*/
void
gst_buffer_pool_config_set_cuda_stream (GstStructure * config,
GstCudaStream * stream)
{
g_return_if_fail (config);
g_return_if_fail (GST_IS_CUDA_STREAM (stream));
gst_structure_set (config, "cuda-stream", GST_TYPE_CUDA_STREAM, stream, NULL);
}
static void
gst_cuda_buffer_pool_dispose (GObject * object)
{
GstCudaBufferPool *self = GST_CUDA_BUFFER_POOL_CAST (object);
GstCudaBufferPoolPrivate *priv = self->priv;
if (priv->alloc) {
gst_cuda_allocator_set_active (GST_CUDA_ALLOCATOR (priv->alloc), FALSE);
gst_clear_object (&priv->alloc);
}
gst_clear_cuda_stream (&priv->stream);
gst_clear_object (&self->context);
G_OBJECT_CLASS (parent_class)->dispose (object);
@ -168,6 +258,8 @@ gst_cuda_buffer_pool_class_init (GstCudaBufferPoolClass * klass)
bufferpool_class->get_options = gst_cuda_buffer_pool_get_options;
bufferpool_class->set_config = gst_cuda_buffer_pool_set_config;
bufferpool_class->start = gst_cuda_buffer_pool_start;
bufferpool_class->stop = gst_cuda_buffer_pool_stop;
bufferpool_class->alloc_buffer = gst_cuda_buffer_pool_alloc;
GST_DEBUG_CATEGORY_INIT (gst_cuda_buffer_pool_debug, "cudabufferpool", 0,

8
subprojects/gst-plugins-bad/gst-libs/gst/cuda/gstcudabufferpool.h
View file

@ -23,6 +23,7 @@
#include <gst/cuda/cuda-prelude.h>
#include <gst/cuda/gstcudamemory.h>
#include <gst/cuda/gstcudastream.h>
G_BEGIN_DECLS
@ -71,5 +72,12 @@ GType gst_cuda_buffer_pool_get_type (void);
GST_CUDA_API
GstBufferPool * gst_cuda_buffer_pool_new (GstCudaContext * context);
GST_CUDA_API
GstCudaStream * gst_buffer_pool_config_get_cuda_stream (GstStructure * config);
GST_CUDA_API
void gst_buffer_pool_config_set_cuda_stream (GstStructure * config,
GstCudaStream * stream);
G_END_DECLS