d3d12: Add helper object for fence operation

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/5870>
This commit is contained in:
Seungha Yang 2023-12-28 21:34:17 +09:00 committed by GStreamer Marge Bot
parent eeb57061d2
commit 4198bd6932
4 changed files with 292 additions and 0 deletions

View file

@ -27,6 +27,7 @@
#include "gstd3d12commandqueue.h"
#include "gstd3d12descriptorpool.h"
#include "gstd3d12device.h"
#include "gstd3d12fencedatapool.h"
#include "gstd3d12format.h"
#include "gstd3d12memory.h"
#include "gstd3d12utils.h"

View file

@ -0,0 +1,240 @@
/* GStreamer
* Copyright (C) 2023 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 "gstd3d12fencedatapool.h"
#include <gst/base/gstqueuearray.h>
#include <mutex>
#include <queue>
GST_DEBUG_CATEGORY_STATIC (gst_d3d12_fence_data_pool_debug);
#define GST_CAT_DEFAULT gst_d3d12_fence_data_pool_debug
struct NotifyData
{
gpointer user_data = nullptr;
GDestroyNotify notify = nullptr;
};
static void
notify_data_clear_func (NotifyData * data)
{
if (data->notify)
data->notify (data->user_data);
data->user_data = nullptr;
data->notify = nullptr;
}
/* *INDENT-OFF* */
struct _GstD3D12FenceData : public GstMiniObject
{
_GstD3D12FenceData ()
{
queue = gst_queue_array_new_for_struct (sizeof (NotifyData), 4);
gst_queue_array_set_clear_func (queue,
(GDestroyNotify) notify_data_clear_func);
}
~_GstD3D12FenceData ()
{
gst_queue_array_free (queue);
}
GstD3D12FenceDataPool *pool = nullptr;
GstQueueArray *queue;
};
GST_DEFINE_MINI_OBJECT_TYPE (GstD3D12FenceData, gst_d3d12_fence_data);
struct GstD3D12FenceDataPoolPrivate
{
~GstD3D12FenceDataPoolPrivate ()
{
while (!data_pool.empty ()) {
auto data = data_pool.front ();
data_pool.pop ();
gst_mini_object_unref (data);
}
}
std::mutex lock;
std::queue<GstD3D12FenceData *>data_pool;
};
/* *INDENT-ON* */
struct _GstD3D12FenceDataPool
{
GstObject parent;
GstD3D12FenceDataPoolPrivate *priv;
};
static void gst_d3d12_fence_data_pool_finalize (GObject * object);
#define gst_d3d12_fence_data_pool_parent_class parent_class
G_DEFINE_TYPE (GstD3D12FenceDataPool,
gst_d3d12_fence_data_pool, GST_TYPE_OBJECT);
static void
gst_d3d12_fence_data_pool_class_init (GstD3D12FenceDataPoolClass * klass)
{
auto object_class = G_OBJECT_CLASS (klass);
object_class->finalize = gst_d3d12_fence_data_pool_finalize;
GST_DEBUG_CATEGORY_INIT (gst_d3d12_fence_data_pool_debug,
"d3d12fencedatapool", 0, "d3d12fencedatapool");
}
static void
gst_d3d12_fence_data_pool_init (GstD3D12FenceDataPool * self)
{
self->priv = new GstD3D12FenceDataPoolPrivate ();
}
static void
gst_d3d12_fence_data_pool_finalize (GObject * object)
{
auto self = GST_D3D12_FENCE_DATA_POOL (object);
delete self->priv;
G_OBJECT_CLASS (parent_class)->finalize (object);
}
GstD3D12FenceDataPool *
gst_d3d12_fence_data_pool_new (void)
{
auto self = (GstD3D12FenceDataPool *)
g_object_new (GST_TYPE_D3D12_FENCE_DATA_POOL, nullptr);
gst_object_ref_sink (self);
return self;
}
static void
gst_d3d12_fence_data_pool_release (GstD3D12FenceDataPool * pool,
GstD3D12FenceData * data)
{
auto priv = pool->priv;
gst_queue_array_clear (data->queue);
{
std::lock_guard < std::mutex > lk (priv->lock);
data->dispose = nullptr;
data->pool = nullptr;
priv->data_pool.push (data);
}
gst_object_unref (pool);
}
static void
gst_d3d12_fence_data_free (GstD3D12FenceData * data)
{
delete data;
}
static gboolean
gst_d3d12_fence_data_dispose (GstD3D12FenceData * data)
{
if (!data->pool)
return TRUE;
gst_mini_object_ref (data);
gst_d3d12_fence_data_pool_release (data->pool, data);
return FALSE;
}
static GstD3D12FenceData *
gst_d3d12_fence_data_new (void)
{
auto data = new GstD3D12FenceData ();
gst_mini_object_init (data, 0, gst_d3d12_fence_data_get_type (),
nullptr, nullptr, (GstMiniObjectFreeFunction) gst_d3d12_fence_data_free);
return data;
}
gboolean
gst_d3d12_fence_data_pool_acquire (GstD3D12FenceDataPool * pool,
GstD3D12FenceData ** data)
{
g_return_val_if_fail (GST_IS_D3D12_FENCE_DATA_POOL (pool), FALSE);
g_return_val_if_fail (data, FALSE);
auto priv = pool->priv;
GstD3D12FenceData *new_data = nullptr;
{
std::lock_guard < std::mutex > lk (priv->lock);
if (!priv->data_pool.empty ()) {
new_data = priv->data_pool.front ();
priv->data_pool.pop ();
}
}
if (!new_data)
new_data = gst_d3d12_fence_data_new ();
new_data->pool = (GstD3D12FenceDataPool *) gst_object_ref (pool);
new_data->dispose =
(GstMiniObjectDisposeFunction) gst_d3d12_fence_data_dispose;
*data = new_data;
return TRUE;
}
void
gst_d3d12_fence_data_add_notify (GstD3D12FenceData * data, gpointer user_data,
GDestroyNotify notify)
{
g_return_if_fail (data);
NotifyData notify_data;
notify_data.user_data = user_data;
notify_data.notify = notify;
gst_queue_array_push_tail_struct (data->queue, &notify_data);
}
GstD3D12FenceData *
gst_d3d12_fence_data_ref (GstD3D12FenceData * data)
{
return (GstD3D12FenceData *) gst_mini_object_ref (data);
}
void
gst_d3d12_fence_data_unref (GstD3D12FenceData * data)
{
gst_mini_object_unref (data);
}
void
gst_clear_d3d12_fence_data (GstD3D12FenceData ** data)
{
gst_clear_mini_object (data);
}

View file

@ -0,0 +1,50 @@
/* GStreamer
* Copyright (C) 2023 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.
*/
#pragma once
#include <gst/gst.h>
#include <gst/video/video.h>
#include "gstd3d12_fwd.h"
G_BEGIN_DECLS
#define GST_TYPE_D3D12_FENCE_DATA_POOL (gst_d3d12_fence_data_pool_get_type())
G_DECLARE_FINAL_TYPE (GstD3D12FenceDataPool,
gst_d3d12_fence_data_pool, GST, D3D12_FENCE_DATA_POOL, GstObject);
typedef struct _GstD3D12FenceData GstD3D12FenceData;
GstD3D12FenceDataPool * gst_d3d12_fence_data_pool_new (void);
gboolean gst_d3d12_fence_data_pool_acquire (GstD3D12FenceDataPool * pool,
GstD3D12FenceData ** data);
void gst_d3d12_fence_data_add_notify (GstD3D12FenceData * data,
gpointer user_data,
GDestroyNotify notify);
GstD3D12FenceData * gst_d3d12_fence_data_ref (GstD3D12FenceData * data);
void gst_d3d12_fence_data_unref (GstD3D12FenceData * data);
void gst_clear_d3d12_fence_data (GstD3D12FenceData ** data);
G_END_DECLS

View file

@ -9,6 +9,7 @@ d3d12_sources = [
'gstd3d12descriptorpool.cpp',
'gstd3d12device.cpp',
'gstd3d12download.cpp',
'gstd3d12fencedatapool.cpp',
'gstd3d12format.cpp',
'gstd3d12h264dec.cpp',
'gstd3d12h265dec.cpp',