FaSTTUBe/GST-Tensordecoder-ov_ep
37
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

vkimagebufferpool: set video meta to buffers if requested

Browse Source 
This patch adds the possibility to attach video metas to allocated buffers if
API users request that option.

The main reason to add this meta is for the future usage of
gst_buffer_find_memory(), where the requested offset determines the memory
index.

In the case of multi-memory Vulkan images the offset of every component is
larger than the default offset determined by the format and frame size, because
of the buffer minimum size and memory alignment determined by the used driver.
Then, the offset is set in the buffer video meta.

In the case of single memory Vulkan images, the default offset is set in the
video meta.

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/9219>
This commit is contained in:
Víctor Manuel Jáquez Leal 2025-06-13 11:28:35 +02:00 committed by GStreamer Marge Bot
parent bbd8fbb023
commit 44acc0bfb1
1 changed files with 27 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
subprojects/gst-plugins-bad/gst-libs/gst/vulkan/gstvkimagebufferpool.c
View File

@ -57,6 +57,7 @@ struct _GstVulkanImageBufferPoolPrivate
guint32 n_profiles;
GstVulkanVideoProfile profiles[2];
GstVulkanOperation *exec;
gboolean add_videometa;
};
static void gst_vulkan_image_buffer_pool_finalize (GObject * object);
@ -378,6 +379,10 @@ gst_vulkan_image_buffer_pool_set_config (GstBufferPool * pool,
priv->usage = requested_usage;
/* enable metadata based on config of the pool */
priv->add_videometa = gst_buffer_pool_config_has_option (config,
GST_BUFFER_POOL_OPTION_VIDEO_META);
return GST_BUFFER_POOL_CLASS (parent_class)->set_config (pool, config);
/* ERRORS */
@ -512,6 +517,7 @@ gst_vulkan_image_buffer_pool_alloc (GstBufferPool * pool, GstBuffer ** buffer,
VkImageCreateInfo image_info;
GstBuffer *buf;
guint i;
gsize offset[GST_VIDEO_MAX_PLANES] = { 0, };
/* *INDENT-OFF* */
image_info = (VkImageCreateInfo) {
@ -576,11 +582,24 @@ gst_vulkan_image_buffer_pool_alloc (GstBufferPool * pool, GstBuffer ** buffer,
goto mem_create_failed;
}
if (i < GST_VIDEO_MAX_PLANES - 1)
offset[i + 1] = mem->size;
gst_buffer_append_memory (buf, mem);
}
prepare_buffer (vk_pool, buf);
if (priv->add_videometa) {
gsize *off = (priv->n_imgs == 1) ? priv->v_info.offset : offset;
gst_buffer_add_video_meta_full (buf, GST_VIDEO_FRAME_FLAG_NONE,
GST_VIDEO_INFO_FORMAT (&priv->v_info),
GST_VIDEO_INFO_WIDTH (&priv->v_info),
GST_VIDEO_INFO_HEIGHT (&priv->v_info),
GST_VIDEO_INFO_N_PLANES (&priv->v_info), off, priv->v_info.stride);
}
*buffer = buf;
return GST_FLOW_OK;
@ -628,6 +647,13 @@ gst_vulkan_image_buffer_pool_reset_buffer (GstBufferPool * pool,
}
}
static const gchar **
gst_vulkan_image_buffer_pool_get_options (GstBufferPool * pool)
{
static const gchar *options[] = { GST_BUFFER_POOL_OPTION_VIDEO_META, NULL };
return options;
}
/**
* gst_vulkan_image_buffer_pool_new:
* @device: the #GstVulkanDevice to use
@ -663,6 +689,7 @@ gst_vulkan_image_buffer_pool_class_init (GstVulkanImageBufferPoolClass * klass)
gstbufferpool_class->alloc_buffer = gst_vulkan_image_buffer_pool_alloc;
gstbufferpool_class->stop = gst_vulkan_image_buffer_pool_stop;
gstbufferpool_class->reset_buffer = gst_vulkan_image_buffer_pool_reset_buffer;
gstbufferpool_class->get_options = gst_vulkan_image_buffer_pool_get_options;
}
static void