/* * GStreamer * Copyright (C) 2015 Matthew Waters * * 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 "gstvkdevice.h" #include "gstvkdebug.h" #include /** * SECTION:vkdevice * @title: GstVulkanDevice * @short_description: Vulkan device * @see_also: #GstVulkanInstance, #GstVulkanPhysicalDevice * * A #GstVulkanDevice encapsulates a VkDevice */ #define GST_CAT_DEFAULT gst_vulkan_device_debug GST_DEBUG_CATEGORY (GST_CAT_DEFAULT); GST_DEBUG_CATEGORY_STATIC (GST_CAT_CONTEXT); #define GET_PRIV(o) (gst_vulkan_device_get_instance_private (o)) enum { PROP_0, PROP_INSTANCE, PROP_PHYSICAL_DEVICE, }; static void gst_vulkan_device_dispose (GObject * object); static void gst_vulkan_device_finalize (GObject * object); struct _GstVulkanDevicePrivate { GPtrArray *enabled_layers; GPtrArray *enabled_extensions; gboolean opened; guint queue_family_id; guint n_queues; GstVulkanFenceCache *fence_cache; }; static void _init_debug (void) { static volatile gsize init; if (g_once_init_enter (&init)) { GST_DEBUG_CATEGORY_INIT (GST_CAT_DEFAULT, "vulkandevice", 0, "Vulkan Device"); GST_DEBUG_CATEGORY_GET (GST_CAT_CONTEXT, "GST_CONTEXT"); g_once_init_leave (&init, 1); } } #define gst_vulkan_device_parent_class parent_class G_DEFINE_TYPE_WITH_CODE (GstVulkanDevice, gst_vulkan_device, GST_TYPE_OBJECT, G_ADD_PRIVATE (GstVulkanDevice); _init_debug ()); /** * gst_vulkan_device_new: * @physical_device: the associated #GstVulkanPhysicalDevice * * Returns: (transfer full): a new #GstVulkanDevice * * Since: 1.18 */ GstVulkanDevice * gst_vulkan_device_new (GstVulkanPhysicalDevice * physical_device) { GstVulkanDevice *device; g_return_val_if_fail (GST_IS_VULKAN_PHYSICAL_DEVICE (physical_device), NULL); device = g_object_new (GST_TYPE_VULKAN_DEVICE, "physical-device", physical_device, NULL); gst_object_ref_sink (device); return device; } /** * gst_vulkan_device_new_with_index: * @instance: the associated #GstVulkanInstance * @device_index: the device index to create the new #GstVulkanDevice from * * Returns: (transfer full): a new #GstVulkanDevice * * Since: 1.18 */ GstVulkanDevice * gst_vulkan_device_new_with_index (GstVulkanInstance * instance, guint device_index) { GstVulkanPhysicalDevice *physical; GstVulkanDevice *device; g_return_val_if_fail (GST_IS_VULKAN_INSTANCE (instance), NULL); physical = gst_vulkan_physical_device_new (instance, device_index); device = gst_vulkan_device_new (physical); gst_object_unref (physical); return device; } static void gst_vulkan_device_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec) { GstVulkanDevice *device = GST_VULKAN_DEVICE (object); switch (prop_id) { case PROP_PHYSICAL_DEVICE: device->physical_device = g_value_dup_object (value); break; default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } } static void gst_vulkan_device_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec) { GstVulkanDevice *device = GST_VULKAN_DEVICE (object); switch (prop_id) { case PROP_INSTANCE: g_value_set_object (value, device->instance); break; case PROP_PHYSICAL_DEVICE: g_value_set_object (value, device->physical_device); break; default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } } static void gst_vulkan_device_init (GstVulkanDevice * device) { GstVulkanDevicePrivate *priv = GET_PRIV (device); priv->enabled_layers = g_ptr_array_new_with_free_func (g_free); priv->enabled_extensions = g_ptr_array_new_with_free_func (g_free); } static void gst_vulkan_device_constructed (GObject * object) { GstVulkanDevice *device = GST_VULKAN_DEVICE (object); g_object_get (device->physical_device, "instance", &device->instance, NULL); /* by default allow vkswapper to work for rendering to an output window. * Ignore the failure if the extension does not exist. */ gst_vulkan_device_enable_extension (device, VK_KHR_SWAPCHAIN_EXTENSION_NAME); G_OBJECT_CLASS (parent_class)->constructed (object); } static void gst_vulkan_device_class_init (GstVulkanDeviceClass * device_class) { GObjectClass *gobject_class = (GObjectClass *) device_class; gobject_class->set_property = gst_vulkan_device_set_property; gobject_class->get_property = gst_vulkan_device_get_property; gobject_class->finalize = gst_vulkan_device_finalize; gobject_class->dispose = gst_vulkan_device_dispose; gobject_class->constructed = gst_vulkan_device_constructed; g_object_class_install_property (gobject_class, PROP_INSTANCE, g_param_spec_object ("instance", "Instance", "Associated Vulkan Instance", GST_TYPE_VULKAN_INSTANCE, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS)); g_object_class_install_property (gobject_class, PROP_PHYSICAL_DEVICE, g_param_spec_object ("physical-device", "Physical Device", "Associated Vulkan Physical Device", GST_TYPE_VULKAN_PHYSICAL_DEVICE, G_PARAM_READWRITE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS)); } static void gst_vulkan_device_dispose (GObject * object) { GstVulkanDevice *device = GST_VULKAN_DEVICE (object); GstVulkanDevicePrivate *priv = GET_PRIV (device); if (priv->fence_cache) { /* clear any outstanding fences */ g_object_run_dispose (G_OBJECT (priv->fence_cache)); /* don't double free this device */ priv->fence_cache->parent.device = NULL; } gst_clear_object (&priv->fence_cache); G_OBJECT_CLASS (parent_class)->dispose (object); } static void gst_vulkan_device_finalize (GObject * object) { GstVulkanDevice *device = GST_VULKAN_DEVICE (object); GstVulkanDevicePrivate *priv = GET_PRIV (device); if (device->device) { vkDeviceWaitIdle (device->device); vkDestroyDevice (device->device, NULL); } device->device = VK_NULL_HANDLE; gst_clear_object (&device->physical_device); gst_clear_object (&device->instance); g_ptr_array_unref (priv->enabled_layers); priv->enabled_layers = NULL; g_ptr_array_unref (priv->enabled_extensions); priv->enabled_extensions = NULL; G_OBJECT_CLASS (parent_class)->finalize (object); } /** * gst_vulkan_device_open: * @device: a #GstVulkanDevice * @error: a #GError * * Attempts to create the internal #VkDevice object. * * Returns: whether a vulkan device could be created * * Since: 1.18 */ gboolean gst_vulkan_device_open (GstVulkanDevice * device, GError ** error) { GstVulkanDevicePrivate *priv = GET_PRIV (device); VkPhysicalDevice gpu; VkResult err; guint i; g_return_val_if_fail (GST_IS_VULKAN_DEVICE (device), FALSE); GST_OBJECT_LOCK (device); if (priv->opened) { GST_OBJECT_UNLOCK (device); return TRUE; } gpu = gst_vulkan_device_get_physical_device (device); /* FIXME: allow overriding/selecting */ for (i = 0; i < device->physical_device->n_queue_families; i++) { if (device->physical_device-> queue_family_props[i].queueFlags & VK_QUEUE_GRAPHICS_BIT) break; } if (i >= device->physical_device->n_queue_families) { g_set_error (error, GST_VULKAN_ERROR, VK_ERROR_INITIALIZATION_FAILED, "Failed to find a compatible queue family"); goto error; } priv->queue_family_id = i; priv->n_queues = 1; { VkDeviceQueueCreateInfo queue_info = { 0, }; VkDeviceCreateInfo device_info = { 0, }; gfloat queue_priority = 0.5; queue_info.sType = VK_STRUCTURE_TYPE_DEVICE_QUEUE_CREATE_INFO; queue_info.pNext = NULL; queue_info.queueFamilyIndex = priv->queue_family_id; queue_info.queueCount = priv->n_queues; queue_info.pQueuePriorities = &queue_priority; device_info.sType = VK_STRUCTURE_TYPE_DEVICE_CREATE_INFO; device_info.pNext = NULL; device_info.queueCreateInfoCount = 1; device_info.pQueueCreateInfos = &queue_info; device_info.enabledLayerCount = priv->enabled_layers->len; device_info.ppEnabledLayerNames = (const char *const *) priv->enabled_layers->pdata; device_info.enabledExtensionCount = priv->enabled_extensions->len; device_info.ppEnabledExtensionNames = (const char *const *) priv->enabled_extensions->pdata; device_info.pEnabledFeatures = NULL; err = vkCreateDevice (gpu, &device_info, NULL, &device->device); if (gst_vulkan_error_to_g_error (err, error, "vkCreateDevice") < 0) { goto error; } } priv->fence_cache = gst_vulkan_fence_cache_new (device); /* avoid reference loops between us and the fence cache */ gst_object_unref (device); priv->opened = TRUE; GST_OBJECT_UNLOCK (device); return TRUE; error: { GST_OBJECT_UNLOCK (device); return FALSE; } } /** * gst_vulkan_device_get_queue: * @device: a #GstVulkanDevice * @queue_family: a queue family to retrieve * @queue_i: index of the family to retrieve * * Returns: (transfer full): a new #GstVulkanQueue * * Since: 1.18 */ GstVulkanQueue * gst_vulkan_device_get_queue (GstVulkanDevice * device, guint32 queue_family, guint32 queue_i) { GstVulkanDevicePrivate *priv = GET_PRIV (device); GstVulkanQueue *ret; g_return_val_if_fail (GST_IS_VULKAN_DEVICE (device), NULL); g_return_val_if_fail (device->device != NULL, NULL); g_return_val_if_fail (priv->opened, NULL); g_return_val_if_fail (queue_family < priv->n_queues, NULL); g_return_val_if_fail (queue_i < device->physical_device->queue_family_props[queue_family].queueCount, NULL); ret = g_object_new (GST_TYPE_VULKAN_QUEUE, NULL); gst_object_ref_sink (ret); ret->device = gst_object_ref (device); ret->family = queue_family; ret->index = queue_i; vkGetDeviceQueue (device->device, queue_family, queue_i, &ret->queue); return ret; } /** * gst_vulkan_device_foreach_queue: * @device: a #GstVulkanDevice * @func: (scope call): a #GstVulkanDeviceForEachQueueFunc to run for each #GstVulkanQueue * @user_data: (closure func): user data to pass to each call of @func * * Iterate over each queue family available on #GstVulkanDevice * * Since: 1.18 */ void gst_vulkan_device_foreach_queue (GstVulkanDevice * device, GstVulkanDeviceForEachQueueFunc func, gpointer user_data) { GstVulkanDevicePrivate *priv = GET_PRIV (device); gboolean done = FALSE; guint i; for (i = 0; i < priv->n_queues; i++) { GstVulkanQueue *queue = gst_vulkan_device_get_queue (device, priv->queue_family_id, i); if (!func (device, queue, user_data)) done = TRUE; gst_object_unref (queue); if (done) break; } } /** * gst_vulkan_device_get_proc_address: * @device: a #GstVulkanDevice * @name: name of the function to retrieve * * Performs vkGetDeviceProcAddr() with @device and @name * * Returns: the function pointer for @name or %NULL * * Since: 1.18 */ gpointer gst_vulkan_device_get_proc_address (GstVulkanDevice * device, const gchar * name) { g_return_val_if_fail (GST_IS_VULKAN_DEVICE (device), NULL); g_return_val_if_fail (device->device != NULL, NULL); g_return_val_if_fail (name != NULL, NULL); GST_TRACE_OBJECT (device, "%s", name); return vkGetDeviceProcAddr (device->device, name); } /** * gst_vulkan_device_get_instance: * @device: a #GstVulkanDevice * * Returns: (transfer full): the #GstVulkanInstance used to create this @device * * Since: 1.18 */ GstVulkanInstance * gst_vulkan_device_get_instance (GstVulkanDevice * device) { g_return_val_if_fail (GST_IS_VULKAN_DEVICE (device), NULL); return device->instance ? gst_object_ref (device->instance) : NULL; } /** * gst_vulkan_device_get_physical_device: (skip) * @device: a #GstVulkanDevice * * Returns: The VkPhysicalDevice used to create @device * * Since: 1.18 */ VkPhysicalDevice gst_vulkan_device_get_physical_device (GstVulkanDevice * device) { g_return_val_if_fail (GST_IS_VULKAN_DEVICE (device), NULL); return gst_vulkan_physical_device_get_handle (device->physical_device); } /** * gst_context_set_vulkan_device: * @context: a #GstContext * @device: a #GstVulkanDevice * * Sets @device on @context * * Since: 1.18 */ void gst_context_set_vulkan_device (GstContext * context, GstVulkanDevice * device) { GstStructure *s; g_return_if_fail (context != NULL); g_return_if_fail (gst_context_is_writable (context)); if (device) GST_CAT_LOG (GST_CAT_CONTEXT, "setting GstVulkanDevice(%" GST_PTR_FORMAT ") on context(%" GST_PTR_FORMAT ")", device, context); s = gst_context_writable_structure (context); gst_structure_set (s, GST_VULKAN_DEVICE_CONTEXT_TYPE_STR, GST_TYPE_VULKAN_DEVICE, device, NULL); } /** * gst_context_get_vulkan_device: * @context: a #GstContext * @device: resulting #GstVulkanDevice * * Returns: Whether @device was in @context * * Since: 1.18 */ gboolean gst_context_get_vulkan_device (GstContext * context, GstVulkanDevice ** device) { const GstStructure *s; gboolean ret; g_return_val_if_fail (device != NULL, FALSE); g_return_val_if_fail (context != NULL, FALSE); s = gst_context_get_structure (context); ret = gst_structure_get (s, GST_VULKAN_DEVICE_CONTEXT_TYPE_STR, GST_TYPE_VULKAN_DEVICE, device, NULL); GST_CAT_LOG (GST_CAT_CONTEXT, "got GstVulkanDevice(%" GST_PTR_FORMAT ") from context(%" GST_PTR_FORMAT ")", *device, context); return ret; } /** * gst_vulkan_device_handle_context_query: * @element: a #GstElement * @query: a #GstQuery of type #GST_QUERY_CONTEXT * @device: the #GstVulkanDevice * * If a #GstVulkanDevice is requested in @query, sets @device as the reply. * * Intended for use with element query handlers to respond to #GST_QUERY_CONTEXT * for a #GstVulkanDevice. * * Returns: whether @query was responded to with @device * * Since: 1.18 */ gboolean gst_vulkan_device_handle_context_query (GstElement * element, GstQuery * query, GstVulkanDevice * device) { gboolean res = FALSE; const gchar *context_type; GstContext *context, *old_context; g_return_val_if_fail (element != NULL, FALSE); g_return_val_if_fail (query != NULL, FALSE); g_return_val_if_fail (GST_QUERY_TYPE (query) == GST_QUERY_CONTEXT, FALSE); if (!device) return FALSE; gst_query_parse_context_type (query, &context_type); if (g_strcmp0 (context_type, GST_VULKAN_DEVICE_CONTEXT_TYPE_STR) == 0) { gst_query_parse_context (query, &old_context); if (old_context) context = gst_context_copy (old_context); else context = gst_context_new (GST_VULKAN_DEVICE_CONTEXT_TYPE_STR, TRUE); gst_context_set_vulkan_device (context, device); gst_query_set_context (query, context); gst_context_unref (context); res = device != NULL; } return res; } /** * gst_vulkan_device_run_context_query: * @element: a #GstElement * @device: (inout): a #GstVulkanDevice * * Attempt to retrieve a #GstVulkanDevice using #GST_QUERY_CONTEXT from the * surrounding elements of @element. * * Returns: whether @device contains a valid #GstVulkanDevice * * Since: 1.18 */ gboolean gst_vulkan_device_run_context_query (GstElement * element, GstVulkanDevice ** device) { GstQuery *query; g_return_val_if_fail (GST_IS_ELEMENT (element), FALSE); g_return_val_if_fail (device != NULL, FALSE); _init_debug (); if (*device && GST_IS_VULKAN_DEVICE (*device)) return TRUE; if ((query = gst_vulkan_local_context_query (element, GST_VULKAN_DEVICE_CONTEXT_TYPE_STR))) { GstContext *context; gst_query_parse_context (query, &context); if (context) gst_context_get_vulkan_device (context, device); gst_query_unref (query); } GST_DEBUG_OBJECT (element, "found device %p", *device); if (*device) return TRUE; return FALSE; } GstVulkanFence * gst_vulkan_device_create_fence (GstVulkanDevice * device, GError ** error) { GstVulkanDevicePrivate *priv; g_return_val_if_fail (GST_IS_VULKAN_DEVICE (device), NULL); priv = GET_PRIV (device); return gst_vulkan_fence_cache_acquire (priv->fence_cache, error); } /* reimplement a specfic case of g_ptr_array_find_with_equal_func as that * requires Glib 2.54 */ static gboolean ptr_array_find_string (GPtrArray * array, const gchar * str, guint * index) { guint i; for (i = 0; i < array->len; i++) { gchar *val = (gchar *) g_ptr_array_index (array, i); if (g_strcmp0 (val, str) == 0) { if (index) *index = i; return TRUE; } } return FALSE; } static gboolean gst_vulkan_device_is_extension_enabled_unlocked (GstVulkanDevice * device, const gchar * name, guint * index) { GstVulkanDevicePrivate *priv = GET_PRIV (device); return ptr_array_find_string (priv->enabled_extensions, name, index); } /** * gst_vulkan_device_is_extension_enabled: * @device: a # GstVulkanDevice * @name: extension name * * Returns: whether extension @name is enabled */ gboolean gst_vulkan_device_is_extension_enabled (GstVulkanDevice * device, const gchar * name) { gboolean ret; g_return_val_if_fail (GST_IS_VULKAN_DEVICE (device), FALSE); g_return_val_if_fail (name != NULL, FALSE); GST_OBJECT_LOCK (device); ret = gst_vulkan_device_is_extension_enabled_unlocked (device, name, NULL); GST_OBJECT_UNLOCK (device); return ret; } static gboolean gst_vulkan_device_enable_extension_unlocked (GstVulkanDevice * device, const gchar * name) { GstVulkanDevicePrivate *priv = GET_PRIV (device); if (gst_vulkan_device_is_extension_enabled_unlocked (device, name, NULL)) /* extension is already enabled */ return TRUE; if (!gst_vulkan_physical_device_get_extension_info (device->physical_device, name, NULL)) return FALSE; g_ptr_array_add (priv->enabled_extensions, g_strdup (name)); return TRUE; } /** * gst_vulkan_device_enable_extension: * @device: a #GstVulkanDevice * @name: extension name to enable * * Enable an Vulkan extension by @name. Enabling an extension will * only have an effect before the call to gst_vulkan_device_open(). * * Returns: whether the Vulkan extension could be enabled. */ gboolean gst_vulkan_device_enable_extension (GstVulkanDevice * device, const gchar * name) { gboolean ret; g_return_val_if_fail (GST_IS_VULKAN_DEVICE (device), FALSE); g_return_val_if_fail (name != NULL, FALSE); GST_OBJECT_LOCK (device); ret = gst_vulkan_device_enable_extension_unlocked (device, name); GST_OBJECT_UNLOCK (device); return ret; } static gboolean gst_vulkan_device_disable_extension_unlocked (GstVulkanDevice * device, const gchar * name) { GstVulkanDevicePrivate *priv = GET_PRIV (device); guint i; if (!gst_vulkan_physical_device_get_extension_info (device->physical_device, name, NULL)) return FALSE; if (!gst_vulkan_device_is_extension_enabled_unlocked (device, name, &i)) /* extension is already disabled */ return TRUE; g_ptr_array_remove_index_fast (priv->enabled_extensions, i); return TRUE; } /** * gst_vulkan_device_disable_extension: * @device: a #GstVulkanDevice * @name: extension name to enable * * Disable an Vulkan extension by @name. Disabling an extension will only have * an effect before the call to gst_vulkan_device_open(). * * Returns: whether the Vulkan extension could be disabled. */ gboolean gst_vulkan_device_disable_extension (GstVulkanDevice * device, const gchar * name) { gboolean ret; g_return_val_if_fail (GST_IS_VULKAN_DEVICE (device), FALSE); g_return_val_if_fail (name != NULL, FALSE); GST_OBJECT_LOCK (device); ret = gst_vulkan_device_disable_extension_unlocked (device, name); GST_OBJECT_UNLOCK (device); return ret; } static gboolean gst_vulkan_device_is_layer_enabled_unlocked (GstVulkanDevice * device, const gchar * name) { GstVulkanDevicePrivate *priv = GET_PRIV (device); return ptr_array_find_string (priv->enabled_layers, name, NULL); } /** * gst_vulkan_device_is_layer_enabled: * @device: a # GstVulkanDevice * @name: layer name * * Returns: whether layer @name is enabled */ gboolean gst_vulkan_device_is_layer_enabled (GstVulkanDevice * device, const gchar * name) { gboolean ret; g_return_val_if_fail (GST_IS_VULKAN_DEVICE (device), FALSE); g_return_val_if_fail (name != NULL, FALSE); GST_OBJECT_LOCK (device); ret = gst_vulkan_device_is_layer_enabled_unlocked (device, name); GST_OBJECT_UNLOCK (device); return ret; } static gboolean gst_vulkan_device_enable_layer_unlocked (GstVulkanDevice * device, const gchar * name) { GstVulkanDevicePrivate *priv = GET_PRIV (device); if (gst_vulkan_device_is_layer_enabled_unlocked (device, name)) /* layer is already enabled */ return TRUE; if (!gst_vulkan_physical_device_get_layer_info (device->physical_device, name, NULL, NULL, NULL)) return FALSE; g_ptr_array_add (priv->enabled_layers, g_strdup (name)); return TRUE; } /** * gst_vulkan_device_enable_layer: * @device: a #GstVulkanDevice * @name: layer name to enable * * Enable an Vulkan layer by @name. Enabling a layer will * only have an effect before the call to gst_vulkan_device_open(). * * Returns: whether the Vulkan layer could be enabled. */ gboolean gst_vulkan_device_enable_layer (GstVulkanDevice * device, const gchar * name) { gboolean ret; g_return_val_if_fail (GST_IS_VULKAN_DEVICE (device), FALSE); g_return_val_if_fail (name != NULL, FALSE); GST_OBJECT_LOCK (device); ret = gst_vulkan_device_enable_layer_unlocked (device, name); GST_OBJECT_UNLOCK (device); return ret; }