gstreamer/sys/mediafoundation/gstmfcapturewinrt.cpp
2020-06-17 17:37:05 +09:00

600 lines
16 KiB
C++

/* GStreamer
* Copyright (C) 2020 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 <gst/video/video.h>
#include "gstmfcapturewinrt.h"
#include "gstmfutils.h"
#include "mediacapturewrapper.h"
#include <memorybuffer.h>
#include <memory>
using namespace Microsoft::WRL;
using namespace Microsoft::WRL::Wrappers;
using namespace ABI::Windows::Media::MediaProperties;
using namespace ABI::Windows::Graphics::Imaging;
using namespace ABI::Windows::Foundation;
G_BEGIN_DECLS
GST_DEBUG_CATEGORY_EXTERN (gst_mf_source_object_debug);
#define GST_CAT_DEFAULT gst_mf_source_object_debug
G_END_DECLS
struct _GstMFCaptureWinRT
{
GstMFSourceObject parent;
MediaCaptureWrapper *capture;
GThread *thread;
GMutex lock;
GCond cond;
GMainContext *context;
GMainLoop *loop;
/* protected by lock */
GQueue *queue;
GstCaps *supported_caps;
GstVideoInfo info;
gboolean flushing;
gboolean got_error;
};
static void gst_mf_capture_winrt_constructed (GObject * object);
static void gst_mf_capture_winrt_finalize (GObject * object);
static gboolean gst_mf_capture_winrt_start (GstMFSourceObject * object);
static gboolean gst_mf_capture_winrt_stop (GstMFSourceObject * object);
static GstFlowReturn gst_mf_capture_winrt_fill (GstMFSourceObject * object,
GstBuffer * buffer);
static gboolean gst_mf_capture_winrt_unlock (GstMFSourceObject * object);
static gboolean gst_mf_capture_winrt_unlock_stop (GstMFSourceObject * object);
static GstCaps * gst_mf_capture_winrt_get_caps (GstMFSourceObject * object);
static gboolean gst_mf_capture_winrt_set_caps (GstMFSourceObject * object,
GstCaps * caps);
static HRESULT gst_mf_capture_winrt_on_frame (ISoftwareBitmap * bitmap,
void * user_data);
static HRESULT gst_mf_capture_winrt_on_failed (const std::string &error,
UINT32 error_code, void * user_data);
static gpointer gst_mf_capture_winrt_thread_func (GstMFCaptureWinRT * self);
#define gst_mf_capture_winrt_parent_class parent_class
G_DEFINE_TYPE (GstMFCaptureWinRT, gst_mf_capture_winrt,
GST_TYPE_MF_SOURCE_OBJECT);
static void
gst_mf_capture_winrt_class_init (GstMFCaptureWinRTClass * klass)
{
GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
GstMFSourceObjectClass *source_class = GST_MF_SOURCE_OBJECT_CLASS (klass);
gobject_class->constructed = gst_mf_capture_winrt_constructed;
gobject_class->finalize = gst_mf_capture_winrt_finalize;
source_class->start = GST_DEBUG_FUNCPTR (gst_mf_capture_winrt_start);
source_class->stop = GST_DEBUG_FUNCPTR (gst_mf_capture_winrt_stop);
source_class->fill = GST_DEBUG_FUNCPTR (gst_mf_capture_winrt_fill);
source_class->unlock = GST_DEBUG_FUNCPTR (gst_mf_capture_winrt_unlock);
source_class->unlock_stop =
GST_DEBUG_FUNCPTR (gst_mf_capture_winrt_unlock_stop);
source_class->get_caps = GST_DEBUG_FUNCPTR (gst_mf_capture_winrt_get_caps);
source_class->set_caps = GST_DEBUG_FUNCPTR (gst_mf_capture_winrt_set_caps);
}
static void
gst_mf_capture_winrt_init (GstMFCaptureWinRT * self)
{
self->queue = g_queue_new ();
g_mutex_init (&self->lock);
g_cond_init (&self->cond);
}
static void
gst_mf_capture_winrt_constructed (GObject * object)
{
GstMFCaptureWinRT *self = GST_MF_CAPTURE_WINRT (object);
self->context = g_main_context_new ();
self->loop = g_main_loop_new (self->context, FALSE);
/* Create a new thread to ensure that COM thread can be MTA thread */
g_mutex_lock (&self->lock);
self->thread = g_thread_new ("GstMFCaptureWinRT",
(GThreadFunc) gst_mf_capture_winrt_thread_func, self);
while (!g_main_loop_is_running (self->loop))
g_cond_wait (&self->cond, &self->lock);
g_mutex_unlock (&self->lock);
done:
G_OBJECT_CLASS (parent_class)->constructed (object);
}
static void
gst_mf_capture_winrt_finalize (GObject * object)
{
GstMFCaptureWinRT *self = GST_MF_CAPTURE_WINRT (object);
g_main_loop_quit (self->loop);
g_thread_join (self->thread);
g_main_loop_unref (self->loop);
g_main_context_unref (self->context);
g_queue_free (self->queue);
gst_clear_caps (&self->supported_caps);
g_mutex_clear (&self->lock);
g_cond_clear (&self->cond);
G_OBJECT_CLASS (parent_class)->finalize (object);
}
static gboolean
gst_mf_capture_winrt_main_loop_running_cb (GstMFCaptureWinRT * self)
{
GST_DEBUG_OBJECT (self, "Main loop running now");
g_mutex_lock (&self->lock);
g_cond_signal (&self->cond);
g_mutex_unlock (&self->lock);
return G_SOURCE_REMOVE;
}
static gpointer
gst_mf_capture_winrt_thread_func (GstMFCaptureWinRT * self)
{
GstMFSourceObject *source = GST_MF_SOURCE_OBJECT (self);
HRESULT hr;
guint index;
GSource *idle_source;
std::shared_ptr<GstWinRTMediaFrameSourceGroup> target_group;
std::vector<GstWinRTMediaFrameSourceGroup> group_list;
MediaCaptureWrapperCallbacks callbacks;
RoInitializeWrapper init_wrapper (RO_INIT_MULTITHREADED);
self->capture = new MediaCaptureWrapper;
callbacks.frame_arrived = gst_mf_capture_winrt_on_frame;
callbacks.failed = gst_mf_capture_winrt_on_failed;
self->capture->RegisterCb (callbacks, self);
g_main_context_push_thread_default (self->context);
idle_source = g_idle_source_new ();
g_source_set_callback (idle_source,
(GSourceFunc) gst_mf_capture_winrt_main_loop_running_cb, self, NULL);
g_source_attach (idle_source, self->context);
g_source_unref (idle_source);
hr = self->capture->EnumrateFrameSourceGroup(group_list);
#ifndef GST_DISABLE_GST_DEBUG
index = 0;
for (const auto& iter: group_list) {
GST_DEBUG_OBJECT (self, "device %d, name: \"%s\", path: \"%s\"",
index, iter.display_name_.c_str(), iter.id_.c_str());
index++;
}
#endif
GST_DEBUG_OBJECT (self,
"Requested device index: %d, name: \"%s\", path \"%s\"",
source->device_index, GST_STR_NULL (source->device_name),
GST_STR_NULL (source->device_path));
index = 0;
for (const auto& iter: group_list) {
gboolean match;
if (source->device_path) {
match = g_ascii_strcasecmp (iter.id_.c_str(), source->device_path) == 0;
} else if (source->device_name) {
match = g_ascii_strcasecmp (iter.display_name_.c_str(),
source->device_name) == 0;
} else if (source->device_index >= 0) {
match = index == source->device_index;
} else {
/* pick the first entry */
match = TRUE;
}
if (match) {
target_group = std::make_shared<GstWinRTMediaFrameSourceGroup>(iter);
break;
}
index++;
}
if (!target_group) {
GST_WARNING_OBJECT (self, "No matching device");
goto run_loop;
}
self->capture->SetSourceGroup(*target_group);
for (auto iter: target_group->source_list_) {
if (!self->supported_caps)
self->supported_caps = gst_caps_ref (iter.caps_);
else
self->supported_caps =
gst_caps_merge (self->supported_caps, gst_caps_ref (iter.caps_));
}
GST_DEBUG_OBJECT (self, "Available output caps %" GST_PTR_FORMAT,
self->supported_caps);
source->opened = !!self->supported_caps;
if (source->opened) {
g_free (source->device_path);
source->device_path = g_strdup (target_group->id_.c_str());
g_free (source->device_name);
source->device_name = g_strdup (target_group->display_name_.c_str());
source->device_index = index;
}
run_loop:
GST_DEBUG_OBJECT (self, "Starting main loop");
g_main_loop_run (self->loop);
GST_DEBUG_OBJECT (self, "Stopped main loop");
g_main_context_pop_thread_default (self->context);
gst_mf_capture_winrt_stop (source);
delete self->capture;
self->capture = NULL;
return NULL;
}
static gboolean
gst_mf_capture_winrt_start (GstMFSourceObject * object)
{
GstMFCaptureWinRT *self = GST_MF_CAPTURE_WINRT (object);
HRESULT hr;
if (!self->capture) {
GST_ERROR_OBJECT (self, "No capture object was configured");
return FALSE;
}
hr = self->capture->StartCapture();
if (!gst_mf_result (hr)) {
GST_ERROR_OBJECT (self, "Capture object doesn't want to start capture");
return FALSE;
}
return TRUE;
}
static gboolean
gst_mf_capture_winrt_stop (GstMFSourceObject * object)
{
GstMFCaptureWinRT *self = GST_MF_CAPTURE_WINRT (object);
HRESULT hr;
if (!self->capture) {
GST_ERROR_OBJECT (self, "No capture object was configured");
return FALSE;
}
hr = self->capture->StopCapture();
while (!g_queue_is_empty (self->queue)) {
ISoftwareBitmap *buffer =
(ISoftwareBitmap *) g_queue_pop_head (self->queue);
buffer->Release ();
}
if (!gst_mf_result (hr)) {
GST_ERROR_OBJECT (self, "Capture object doesn't want to stop capture");
return FALSE;
}
return TRUE;
}
static HRESULT
gst_mf_capture_winrt_on_frame (ISoftwareBitmap * bitmap,
void * user_data)
{
GstMFCaptureWinRT *self = GST_MF_CAPTURE_WINRT (user_data);
g_mutex_lock (&self->lock);
if (self->flushing) {
g_mutex_unlock (&self->lock);
return S_OK;
}
g_queue_push_tail (self->queue, bitmap);
bitmap->AddRef ();
g_cond_broadcast (&self->cond);
g_mutex_unlock (&self->lock);
return S_OK;
}
static HRESULT
gst_mf_capture_winrt_on_failed (const std::string &error,
UINT32 error_code, void * user_data)
{
GstMFCaptureWinRT *self = GST_MF_CAPTURE_WINRT (user_data);
GST_DEBUG_OBJECT (self, "Have error %s (%d)", error.c_str(), error_code);
g_mutex_lock (&self->lock);
self->got_error = TRUE;
g_cond_broadcast (&self->cond);
g_mutex_unlock (&self->lock);
return S_OK;
}
static GstFlowReturn
gst_mf_capture_winrt_fill (GstMFSourceObject * object, GstBuffer * buffer)
{
GstMFCaptureWinRT *self = GST_MF_CAPTURE_WINRT (object);
GstFlowReturn ret = GST_FLOW_OK;
HRESULT hr;
GstVideoFrame frame;
BYTE *data;
UINT32 size;
gint i, j;
ComPtr<ISoftwareBitmap> bitmap;
ComPtr<IBitmapBuffer> bitmap_buffer;
ComPtr<IMemoryBuffer> mem_buf;
ComPtr<IMemoryBufferReference> mem_ref;
ComPtr<Windows::Foundation::IMemoryBufferByteAccess> byte_access;
INT32 plane_count;
BitmapPlaneDescription desc[GST_VIDEO_MAX_PLANES];
g_mutex_lock (&self->lock);
if (self->got_error) {
g_mutex_unlock (&self->lock);
return GST_FLOW_ERROR;
}
if (self->flushing) {
g_mutex_unlock (&self->lock);
return GST_FLOW_FLUSHING;
}
while (!self->flushing && !self->got_error && g_queue_is_empty (self->queue))
g_cond_wait (&self->cond, &self->lock);
if (self->got_error) {
g_mutex_unlock (&self->lock);
return GST_FLOW_ERROR;
}
if (self->flushing) {
g_mutex_unlock (&self->lock);
return GST_FLOW_FLUSHING;
}
bitmap.Attach ((ISoftwareBitmap *) g_queue_pop_head (self->queue));
g_mutex_unlock (&self->lock);
hr = bitmap->LockBuffer (BitmapBufferAccessMode::BitmapBufferAccessMode_Read,
&bitmap_buffer);
if (!gst_mf_result (hr)) {
GST_ERROR_OBJECT (self, "Cannot lock ISoftwareBitmap");
return GST_FLOW_ERROR;
}
hr = bitmap_buffer->GetPlaneCount (&plane_count);
if (!gst_mf_result (hr)) {
GST_ERROR_OBJECT (self, "Cannot get plane count");
return GST_FLOW_ERROR;
}
if (plane_count > GST_VIDEO_MAX_PLANES) {
GST_ERROR_OBJECT (self, "Invalid plane count %d", plane_count);
return GST_FLOW_ERROR;
}
if (plane_count != GST_VIDEO_INFO_N_PLANES (&self->info)) {
GST_ERROR_OBJECT (self, "Ambiguous plane count %d", plane_count);
return GST_FLOW_ERROR;
}
for (i = 0; i < plane_count; i++) {
hr = bitmap_buffer->GetPlaneDescription (i, &desc[i]);
if (!gst_mf_result (hr)) {
GST_ERROR_OBJECT (self, "Cannot get description for plane %d", i);
return GST_FLOW_ERROR;
}
}
hr = bitmap_buffer.As (&mem_buf);
if (!gst_mf_result (hr)) {
GST_ERROR_OBJECT (self, "Cannot get IMemoryBuffer");
return GST_FLOW_ERROR;
}
hr = mem_buf->CreateReference (&mem_ref);
if (!gst_mf_result (hr)) {
GST_ERROR_OBJECT (self, "Cannot get IMemoryBufferReference");
return GST_FLOW_ERROR;
}
hr = mem_ref.As(&byte_access);
if (!gst_mf_result (hr)) {
GST_ERROR_OBJECT (self, "Cannot get IMemoryBufferByteAccess");
return GST_FLOW_ERROR;
}
hr = byte_access->GetBuffer (&data, &size);
if (!gst_mf_result (hr)) {
GST_ERROR_OBJECT (self, "Cannot get raw buffer data");
return GST_FLOW_ERROR;
}
if (size < GST_VIDEO_INFO_SIZE (&self->info)) {
GST_ERROR_OBJECT (self, "Too small buffer size %d", size);
return GST_FLOW_ERROR;
}
if (!gst_video_frame_map (&frame, &self->info, buffer, GST_MAP_WRITE)) {
GST_ERROR_OBJECT (self, "Failed to map buffer");
return GST_FLOW_ERROR;
}
for (i = 0; i < GST_VIDEO_INFO_N_PLANES (&self->info); i++) {
guint8 *src, *dst;
gint src_stride, dst_stride;
gint width;
src = data + desc[i].StartIndex;
dst = (guint8 *) GST_VIDEO_FRAME_PLANE_DATA (&frame, i);
src_stride = desc[i].Stride;
dst_stride = GST_VIDEO_FRAME_PLANE_STRIDE (&frame, i);
width = GST_VIDEO_INFO_COMP_WIDTH (&self->info, i)
* GST_VIDEO_INFO_COMP_PSTRIDE (&self->info, i);
for (j = 0; j < GST_VIDEO_INFO_COMP_HEIGHT (&self->info, i); j++) {
memcpy (dst, src, width);
src += src_stride;
dst += dst_stride;
}
}
gst_video_frame_unmap (&frame);
return ret;
}
static gboolean
gst_mf_capture_winrt_unlock (GstMFSourceObject * object)
{
GstMFCaptureWinRT *self = GST_MF_CAPTURE_WINRT (object);
g_mutex_lock (&self->lock);
if (self->flushing) {
g_mutex_unlock (&self->lock);
return TRUE;
}
self->flushing = TRUE;
g_cond_broadcast (&self->cond);
g_mutex_unlock (&self->lock);
return TRUE;
}
static gboolean
gst_mf_capture_winrt_unlock_stop (GstMFSourceObject * object)
{
GstMFCaptureWinRT *self = GST_MF_CAPTURE_WINRT (object);
g_mutex_lock (&self->lock);
if (!self->flushing) {
g_mutex_unlock (&self->lock);
return TRUE;
}
self->flushing = FALSE;
g_cond_broadcast (&self->cond);
g_mutex_unlock (&self->lock);
return TRUE;
}
static GstCaps *
gst_mf_capture_winrt_get_caps (GstMFSourceObject * object)
{
GstMFCaptureWinRT *self = GST_MF_CAPTURE_WINRT (object);
if (self->supported_caps)
return gst_caps_ref (self->supported_caps);
return NULL;
}
static gboolean
gst_mf_capture_winrt_set_caps (GstMFSourceObject * object, GstCaps * caps)
{
GstMFCaptureWinRT *self = GST_MF_CAPTURE_WINRT (object);
std::vector<GstWinRTMediaDescription> desc_list;
HRESULT hr;
GstCaps *target_caps = NULL;
hr = self->capture->GetAvailableDescriptions(desc_list);
if (!gst_mf_result (hr) || desc_list.empty()) {
GST_ERROR_OBJECT (self, "No available media description");
return FALSE;
}
for (const auto& iter: desc_list) {
if (gst_caps_is_subset (iter.caps_, caps)) {
target_caps = gst_caps_ref (iter.caps_);
self->capture->SetMediaDescription(iter);
break;
}
}
if (!target_caps) {
GST_ERROR_OBJECT (self,
"Could not determine target media type with given caps %"
GST_PTR_FORMAT, caps);
return FALSE;
}
gst_video_info_from_caps (&self->info, target_caps);
gst_caps_unref (target_caps);
return TRUE;
}
GstMFSourceObject *
gst_mf_capture_winrt_new (GstMFSourceType type, gint device_index,
const gchar * device_name, const gchar * device_path)
{
GstMFSourceObject *self;
/* TODO: Add audio capture support */
g_return_val_if_fail (type == GST_MF_SOURCE_TYPE_VIDEO, NULL);
self = (GstMFSourceObject *) g_object_new (GST_TYPE_MF_CAPTURE_WINRT,
"source-type", type, "device-index", device_index, "device-name",
device_name, "device-path", device_path, NULL);
gst_object_ref_sink (self);
if (!self->opened) {
GST_WARNING_OBJECT (self, "Couldn't open device");
gst_object_unref (self);
return NULL;
}
return self;
}