mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-04 23:46:43 +00:00
301 lines
7.9 KiB
C++
301 lines
7.9 KiB
C++
|
/* 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 "gstdwritebitmapmemory.h"
|
||
|
#include <mutex>
|
||
|
#include <condition_variable>
|
||
|
#include <thread>
|
||
|
#include <wrl.h>
|
||
|
|
||
|
/* *INDENT-OFF* */
|
||
|
using namespace Microsoft::WRL;
|
||
|
/* *INDENT-ON* */
|
||
|
|
||
|
GST_DEBUG_CATEGORY_STATIC (gst_dwrite_bitmap_allocator_debug);
|
||
|
#define GST_CAT_DEFAULT gst_dwrite_bitmap_allocator_debug
|
||
|
|
||
|
#define GST_DWRITE_BITMAP_MEMORY_NAME "DWriteBitmapMemory"
|
||
|
|
||
|
struct GstDWriteBitmapAllocatorPrivate
|
||
|
{
|
||
|
GstDWriteBitmapAllocatorPrivate ()
|
||
|
{
|
||
|
com_thread = std::thread (&GstDWriteBitmapAllocatorPrivate::ComThreadFunc,
|
||
|
this);
|
||
|
std::unique_lock < std::mutex > lk (init_lock);
|
||
|
while (!running)
|
||
|
init_cond.wait (lk);
|
||
|
}
|
||
|
|
||
|
~GstDWriteBitmapAllocatorPrivate ()
|
||
|
{
|
||
|
thread_lock.lock ();
|
||
|
terminate = true;
|
||
|
thread_cond.notify_one ();
|
||
|
thread_lock.unlock ();
|
||
|
|
||
|
com_thread.join ();
|
||
|
}
|
||
|
|
||
|
void ComThreadFunc ()
|
||
|
{
|
||
|
HRESULT hr;
|
||
|
CoInitializeEx (nullptr, COINIT_MULTITHREADED);
|
||
|
|
||
|
hr = CoCreateInstance (CLSID_WICImagingFactory, nullptr,
|
||
|
CLSCTX_INPROC_SERVER, IID_PPV_ARGS (&factory));
|
||
|
if (FAILED (hr))
|
||
|
GST_ERROR ("Couldn't create image factory");
|
||
|
|
||
|
init_lock.lock ();
|
||
|
running = true;
|
||
|
init_cond.notify_one ();
|
||
|
init_lock.unlock ();
|
||
|
|
||
|
std::unique_lock < std::mutex > lk (thread_lock);
|
||
|
while (!terminate)
|
||
|
thread_cond.wait (lk);
|
||
|
|
||
|
if (factory)
|
||
|
factory->Release ();
|
||
|
|
||
|
CoUninitialize ();
|
||
|
}
|
||
|
|
||
|
IWICImagingFactory *factory = nullptr;
|
||
|
|
||
|
std::mutex init_lock;
|
||
|
std::condition_variable init_cond;
|
||
|
|
||
|
std::mutex thread_lock;
|
||
|
std::condition_variable thread_cond;
|
||
|
|
||
|
bool running = false;
|
||
|
bool terminate = false;
|
||
|
|
||
|
std::thread com_thread;
|
||
|
};
|
||
|
|
||
|
struct _GstDWriteBitmapAllocator
|
||
|
{
|
||
|
GstAllocator parent;
|
||
|
|
||
|
GstDWriteBitmapAllocatorPrivate *priv;
|
||
|
};
|
||
|
|
||
|
static void gst_dwrite_bitmap_allocator_finalize (GObject * object);
|
||
|
|
||
|
static GstMemory *gst_dwrite_bitmap_allocator_dummy_alloc (GstAllocator * alloc,
|
||
|
gsize size, GstAllocationParams * params);
|
||
|
static void gst_dwrite_bitmap_allocator_free (GstAllocator * alloc,
|
||
|
GstMemory * mem);
|
||
|
|
||
|
static gpointer gst_dwrite_bitmap_allocator_map (GstMemory * mem,
|
||
|
GstMapInfo * info, gsize maxsize);
|
||
|
static void gst_dwrite_bitmap_allocator_unmap (GstMemory * mem,
|
||
|
GstMapInfo * info);
|
||
|
static GstMemory *gst_dwrite_bitmap_allocator_share (GstMemory * mem,
|
||
|
gssize offset, gssize size);
|
||
|
|
||
|
#define gst_dwrite_bitmap_allocator_parent_class parent_class
|
||
|
G_DEFINE_TYPE (GstDWriteBitmapAllocator, gst_dwrite_bitmap_allocator,
|
||
|
GST_TYPE_ALLOCATOR);
|
||
|
|
||
|
static void
|
||
|
gst_dwrite_bitmap_allocator_class_init (GstDWriteBitmapAllocatorClass * klass)
|
||
|
{
|
||
|
GObjectClass *object_class = G_OBJECT_CLASS (klass);
|
||
|
GstAllocatorClass *alloc_class = GST_ALLOCATOR_CLASS (klass);
|
||
|
|
||
|
object_class->finalize = gst_dwrite_bitmap_allocator_finalize;
|
||
|
|
||
|
alloc_class->alloc = gst_dwrite_bitmap_allocator_dummy_alloc;
|
||
|
alloc_class->free = gst_dwrite_bitmap_allocator_free;
|
||
|
|
||
|
GST_DEBUG_CATEGORY_INIT (gst_dwrite_bitmap_allocator_debug,
|
||
|
"dwritebitmapallocator", 0, "dwritebitmapallocator");
|
||
|
}
|
||
|
|
||
|
static void
|
||
|
gst_dwrite_bitmap_allocator_init (GstDWriteBitmapAllocator * self)
|
||
|
{
|
||
|
GstAllocator *alloc = GST_ALLOCATOR_CAST (self);
|
||
|
|
||
|
alloc->mem_type = GST_DWRITE_BITMAP_MEMORY_NAME;
|
||
|
alloc->mem_map_full = gst_dwrite_bitmap_allocator_map;
|
||
|
alloc->mem_unmap_full = gst_dwrite_bitmap_allocator_unmap;
|
||
|
alloc->mem_share = gst_dwrite_bitmap_allocator_share;
|
||
|
|
||
|
GST_OBJECT_FLAG_SET (alloc, GST_ALLOCATOR_FLAG_CUSTOM_ALLOC);
|
||
|
|
||
|
self->priv = new GstDWriteBitmapAllocatorPrivate ();
|
||
|
}
|
||
|
|
||
|
static void
|
||
|
gst_dwrite_bitmap_allocator_finalize (GObject * object)
|
||
|
{
|
||
|
GstDWriteBitmapAllocator *self = GST_DWRITE_BITMAP_ALLOCATOR (object);
|
||
|
|
||
|
delete self->priv;
|
||
|
|
||
|
G_OBJECT_CLASS (parent_class)->finalize (object);
|
||
|
}
|
||
|
|
||
|
static GstMemory *
|
||
|
gst_dwrite_bitmap_allocator_dummy_alloc (GstAllocator * alloc, gsize size,
|
||
|
GstAllocationParams * params)
|
||
|
{
|
||
|
g_assert_not_reached ();
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
static void
|
||
|
gst_dwrite_bitmap_allocator_free (GstAllocator * alloc, GstMemory * mem)
|
||
|
{
|
||
|
GstDWriteBitmapMemory *dmem = (GstDWriteBitmapMemory *) mem;
|
||
|
|
||
|
if (dmem->bitmap)
|
||
|
dmem->bitmap->Release ();
|
||
|
|
||
|
g_free (dmem);
|
||
|
}
|
||
|
|
||
|
static GstMemory *
|
||
|
gst_dwrite_bitmap_allocator_share (GstMemory * mem, gssize offset, gssize size)
|
||
|
{
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
static gpointer
|
||
|
gst_dwrite_bitmap_allocator_map (GstMemory * mem, GstMapInfo * info,
|
||
|
gsize maxsize)
|
||
|
{
|
||
|
GstDWriteBitmapMemory *dmem = (GstDWriteBitmapMemory *) mem;
|
||
|
ComPtr < IWICBitmapLock > bitmap_lock;
|
||
|
HRESULT hr;
|
||
|
WICRect rect = { 0, 0, dmem->info.width, dmem->info.height };
|
||
|
DWORD map_flags = 0;
|
||
|
BYTE *ptr = nullptr;
|
||
|
UINT size;
|
||
|
|
||
|
info->user_data[0] = nullptr;
|
||
|
|
||
|
if ((info->flags & GST_MAP_READ) == GST_MAP_READ)
|
||
|
map_flags |= (gint) WICBitmapLockRead;
|
||
|
|
||
|
if ((info->flags & GST_MAP_WRITE) == GST_MAP_WRITE)
|
||
|
map_flags |= (gint) WICBitmapLockWrite;
|
||
|
|
||
|
|
||
|
hr = dmem->bitmap->Lock (&rect, map_flags, &bitmap_lock);
|
||
|
if (FAILED (hr)) {
|
||
|
GST_ERROR_OBJECT (mem->allocator,
|
||
|
"Couldn't map bitmap, hr: 0x%x", (guint) hr);
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
hr = bitmap_lock->GetDataPointer (&size, &ptr);
|
||
|
if (FAILED (hr)) {
|
||
|
GST_ERROR_OBJECT (mem->allocator, "Couldn't get data pointer, hr: 0x%x",
|
||
|
(guint) hr);
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
info->user_data[0] = (gpointer) bitmap_lock.Detach ();
|
||
|
|
||
|
return (gpointer) ptr;
|
||
|
}
|
||
|
|
||
|
static void
|
||
|
gst_dwrite_bitmap_allocator_unmap (GstMemory * mem, GstMapInfo * info)
|
||
|
{
|
||
|
IWICBitmapLock *bitmap_lock;
|
||
|
|
||
|
bitmap_lock = (IWICBitmapLock *) info->user_data[0];
|
||
|
if (!bitmap_lock) {
|
||
|
GST_WARNING_OBJECT (mem->allocator, "No attached bitmap lock");
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
bitmap_lock->Release ();
|
||
|
info->user_data[0] = nullptr;
|
||
|
}
|
||
|
|
||
|
GstDWriteBitmapAllocator *
|
||
|
gst_dwrite_bitmap_allocator_new (void)
|
||
|
{
|
||
|
GstDWriteBitmapAllocator *self = (GstDWriteBitmapAllocator *)
|
||
|
g_object_new (GST_TYPE_DWRITE_BITMAP_ALLOCATOR, nullptr);
|
||
|
|
||
|
gst_object_ref_sink (self);
|
||
|
|
||
|
if (!self->priv->factory) {
|
||
|
gst_object_unref (self);
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
return self;
|
||
|
}
|
||
|
|
||
|
GstMemory *
|
||
|
gst_dwrite_bitmap_allocator_alloc (GstDWriteBitmapAllocator * alloc,
|
||
|
guint width, guint height)
|
||
|
{
|
||
|
HRESULT hr;
|
||
|
IWICImagingFactory *factory = alloc->priv->factory;
|
||
|
ComPtr < IWICBitmap > bitmap;
|
||
|
ComPtr < IWICBitmapLock > bitmap_lock;
|
||
|
WICRect rect = { 0, 0, (INT) width, (INT) height };
|
||
|
guint stride = width * 4;
|
||
|
GstDWriteBitmapMemory *mem;
|
||
|
gsize size;
|
||
|
|
||
|
hr = factory->CreateBitmap (width, height, GUID_WICPixelFormat32bppPBGRA,
|
||
|
WICBitmapCacheOnDemand, &bitmap);
|
||
|
if (FAILED (hr)) {
|
||
|
GST_ERROR_OBJECT (alloc, "Couldn't create bitmap, hr: 0x%x", (guint) hr);
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
hr = bitmap->Lock (&rect, WICBitmapLockRead, &bitmap_lock);
|
||
|
if (FAILED (hr)) {
|
||
|
GST_ERROR_OBJECT (alloc, "Couldn't lock bitmap, hr: 0x%x", (guint) hr);
|
||
|
return nullptr;
|
||
|
}
|
||
|
|
||
|
bitmap_lock = nullptr;
|
||
|
size = stride * height;
|
||
|
|
||
|
mem = g_new0 (GstDWriteBitmapMemory, 1);
|
||
|
gst_memory_init (GST_MEMORY_CAST (mem), (GstMemoryFlags) 0,
|
||
|
GST_ALLOCATOR_CAST (alloc), nullptr, size, 0, 0, size);
|
||
|
|
||
|
gst_video_info_set_format (&mem->info, GST_VIDEO_FORMAT_BGRA, width, height);
|
||
|
mem->info.size = size;
|
||
|
mem->info.stride[0] = stride;
|
||
|
|
||
|
mem->bitmap = bitmap.Detach ();
|
||
|
|
||
|
return GST_MEMORY_CAST (mem);
|
||
|
}
|