mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-04 23:46:43 +00:00
e0a9a73adf
GstD3D11ScreenCapture object is pipeline-independent global object and the object can be shared by multiple src elements, in order to overcome a limitation of DXGI Desktop Duplication API. Note that the API allows only single capture session in a process for a monitor. Therefore GstD3D11ScreenCapture object must be able to handle a case where a src element holds different GstD3D11Device object. Which can happen when GstD3D11Device context is not shared by pipelines. What's changed: * Allocates capture texture with D3D11_RESOURCE_MISC_SHARED for the texture to be able to copied into other device's texture * Holds additional shader objects per src element and use it when drawing mouse Fixes: https://gitlab.freedesktop.org/gstreamer/gstreamer/-/issues/1197 Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/2366>
254 lines
7.6 KiB
C++
254 lines
7.6 KiB
C++
/* GStreamer
|
|
* Copyright (C) 2021 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/gst.h>
|
|
#include <windows.h>
|
|
|
|
static GstDevice *
|
|
enum_devices (gboolean only_show, gint monitor_idx, guint64 monitor_handle)
|
|
{
|
|
GstDeviceMonitor *monitor;
|
|
GstCaps *caps;
|
|
GList *devices, *iter;
|
|
gboolean ret;
|
|
guint i;
|
|
GstDevice *target = nullptr;
|
|
|
|
monitor = gst_device_monitor_new ();
|
|
|
|
/* Filtering by using d3d11 memory caps with "Source/Monitor" class */
|
|
caps = gst_caps_from_string ("video/x-raw(memory:D3D11Memory)");
|
|
ret = gst_device_monitor_add_filter (monitor, "Source/Monitor", caps);
|
|
gst_caps_unref (caps);
|
|
|
|
if (!ret) {
|
|
gst_object_unref (monitor);
|
|
g_warning ("Failed to setup device monitor");
|
|
return nullptr;
|
|
}
|
|
|
|
gst_device_monitor_start (monitor);
|
|
devices = gst_device_monitor_get_devices (monitor);
|
|
|
|
if (!devices) {
|
|
g_warning ("No detected d3d11 monitor device");
|
|
gst_device_monitor_stop (monitor);
|
|
gst_object_unref (monitor);
|
|
return nullptr;
|
|
}
|
|
|
|
gst_println ("Found %d monitor device(s)", g_list_length (devices));
|
|
|
|
for (iter = devices, i = 0; iter; iter = g_list_next (iter), i++) {
|
|
GstDevice *dev = GST_DEVICE (iter->data);
|
|
GstStructure *s;
|
|
guint disp_coord_left, disp_coord_top, disp_coord_right, disp_coord_bottom;
|
|
gchar *name = nullptr;
|
|
gchar *adapter_desc = nullptr;
|
|
guint64 hmonitor;
|
|
HMONITOR handle;
|
|
gboolean primary;
|
|
|
|
s = gst_device_get_properties (dev);
|
|
|
|
gst_structure_get (s,
|
|
"display.coordinates.left", G_TYPE_INT, &disp_coord_left,
|
|
"display.coordinates.top", G_TYPE_INT, &disp_coord_top,
|
|
"display.coordinates.right", G_TYPE_INT, &disp_coord_right,
|
|
"display.coordinates.bottom", G_TYPE_INT, &disp_coord_bottom,
|
|
"device.adapter.description", G_TYPE_STRING, &adapter_desc,
|
|
"device.hmonitor", G_TYPE_UINT64, &hmonitor,
|
|
"device.primary", G_TYPE_BOOLEAN, &primary, nullptr);
|
|
|
|
name = gst_device_get_display_name (dev);
|
|
|
|
handle = (HMONITOR) hmonitor;
|
|
|
|
gst_println ("Monitor %d (%s - %s):", i, name, adapter_desc);
|
|
gst_println (" HMONITOR: %p (%" G_GUINT64_FORMAT ")", handle, hmonitor);
|
|
gst_println (" Display Coordinates (left:top:right:bottom): %d:%d:%d:%d\n",
|
|
disp_coord_left, disp_coord_top, disp_coord_right, disp_coord_bottom);
|
|
|
|
g_free (adapter_desc);
|
|
g_free (name);
|
|
|
|
if (!only_show && !target) {
|
|
if (monitor_handle != 0) {
|
|
if (monitor_handle == hmonitor) {
|
|
target = (GstDevice *) gst_object_ref (dev);
|
|
}
|
|
} else if (monitor_idx < 0) {
|
|
if (primary) {
|
|
target = (GstDevice *) gst_object_ref (dev);
|
|
}
|
|
} else if (monitor_idx == i) {
|
|
target = (GstDevice *) gst_object_ref (dev);
|
|
}
|
|
|
|
if (target)
|
|
gst_println ("Found target monitor device");
|
|
}
|
|
}
|
|
g_list_free_full (devices, gst_object_unref);
|
|
|
|
return target;
|
|
}
|
|
|
|
static gboolean
|
|
bus_msg (GstBus * bus, GstMessage * msg, GMainLoop * loop)
|
|
{
|
|
switch (GST_MESSAGE_TYPE (msg)) {
|
|
case GST_MESSAGE_ERROR:{
|
|
GError *err;
|
|
gchar *dbg;
|
|
|
|
gst_message_parse_error (msg, &err, &dbg);
|
|
g_printerr ("ERROR %s \n", err->message);
|
|
if (dbg != NULL)
|
|
g_printerr ("ERROR debug information: %s\n", dbg);
|
|
g_clear_error (&err);
|
|
g_free (dbg);
|
|
|
|
g_main_loop_quit (loop);
|
|
break;
|
|
}
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
gint
|
|
main (gint argc, gchar ** argv)
|
|
{
|
|
GstElement *pipeline, *src, *queue, *sink;
|
|
GstElement *pipeline_1 = nullptr, *src_1, *queue_1, *sink_1;
|
|
GMainLoop *loop;
|
|
gboolean ret;
|
|
gboolean show_devices = FALSE;
|
|
gboolean multi_pipelines = FALSE;
|
|
gboolean show_cursor = FALSE;
|
|
gint64 hmonitor = 0;
|
|
gint monitor_index = -1;
|
|
GError *err = nullptr;
|
|
GstDevice *device;
|
|
GOptionContext *option_ctx;
|
|
GOptionEntry options[] = {
|
|
{"show-devices", 0, 0, G_OPTION_ARG_NONE, &show_devices,
|
|
"Display available monitor devices", nullptr},
|
|
{"hmonitor", 0, 0, G_OPTION_ARG_INT64, &hmonitor,
|
|
"Address of HMONITOR handle", nullptr},
|
|
{"index", 0, 0, G_OPTION_ARG_INT, &monitor_index,
|
|
"Monitor index to capture (-1 for primary monitor)", nullptr},
|
|
{"multi-pipelines", 0, 0, G_OPTION_ARG_NONE, &multi_pipelines,
|
|
"Run two separate pipelines for capturing a single monitor", nullptr},
|
|
{"show-cursor", 0, 0, G_OPTION_ARG_NONE, &show_cursor,
|
|
"Draw mouse cursor", nullptr},
|
|
{nullptr}
|
|
};
|
|
|
|
option_ctx = g_option_context_new ("D3D11 screen capture example");
|
|
g_option_context_add_main_entries (option_ctx, options, NULL);
|
|
g_option_context_add_group (option_ctx, gst_init_get_option_group ());
|
|
ret = g_option_context_parse (option_ctx, &argc, &argv, &err);
|
|
g_option_context_free (option_ctx);
|
|
|
|
if (!ret) {
|
|
g_printerr ("option parsing failed: %s\n", err->message);
|
|
g_clear_error (&err);
|
|
return 1;
|
|
}
|
|
|
|
device = enum_devices (show_devices, monitor_index, (guint64) hmonitor);
|
|
if (show_devices) {
|
|
gst_clear_object (&device);
|
|
return 0;
|
|
}
|
|
|
|
if (!device) {
|
|
gst_println ("Failed to find monitor device");
|
|
return 1;
|
|
}
|
|
|
|
src = gst_device_create_element (device, nullptr);
|
|
if (!src) {
|
|
g_warning ("Failed to create d3d11screencapture element");
|
|
return 1;
|
|
}
|
|
|
|
g_object_set (src, "show-cursor", show_cursor, nullptr);
|
|
|
|
if (multi_pipelines) {
|
|
src_1 = gst_device_create_element (device, nullptr);
|
|
if (!src_1) {
|
|
g_warning ("Failed to create second d3d11screencapture element");
|
|
return 1;
|
|
}
|
|
|
|
g_object_set (src_1, "show-cursor", show_cursor, nullptr);
|
|
}
|
|
|
|
gst_object_unref (device);
|
|
|
|
loop = g_main_loop_new (nullptr, FALSE);
|
|
pipeline = gst_pipeline_new (nullptr);
|
|
|
|
queue = gst_element_factory_make ("queue", nullptr);
|
|
sink = gst_element_factory_make ("d3d11videosink", nullptr);
|
|
|
|
gst_bin_add_many (GST_BIN (pipeline), src, queue, sink, nullptr);
|
|
gst_element_link_many (src, queue, sink, nullptr);
|
|
|
|
gst_bus_add_watch (GST_ELEMENT_BUS (pipeline), (GstBusFunc) bus_msg, loop);
|
|
gst_element_set_state (pipeline, GST_STATE_PLAYING);
|
|
|
|
if (multi_pipelines) {
|
|
pipeline_1 = gst_pipeline_new (nullptr);
|
|
|
|
queue_1 = gst_element_factory_make ("queue", nullptr);
|
|
sink_1 = gst_element_factory_make ("d3d11videosink", nullptr);
|
|
|
|
gst_bin_add_many (GST_BIN (pipeline_1), src_1, queue_1, sink_1, nullptr);
|
|
gst_element_link_many (src_1, queue_1, sink_1, nullptr);
|
|
|
|
gst_bus_add_watch (GST_ELEMENT_BUS (pipeline_1), (GstBusFunc) bus_msg, loop);
|
|
gst_element_set_state (pipeline_1, GST_STATE_PLAYING);
|
|
}
|
|
|
|
g_main_loop_run (loop);
|
|
|
|
gst_element_set_state (pipeline, GST_STATE_NULL);
|
|
gst_bus_remove_watch (GST_ELEMENT_BUS (pipeline));
|
|
gst_object_unref (pipeline);
|
|
|
|
if (multi_pipelines) {
|
|
gst_element_set_state (pipeline_1, GST_STATE_NULL);
|
|
gst_bus_remove_watch (GST_ELEMENT_BUS (pipeline));
|
|
gst_object_unref (pipeline_1);
|
|
}
|
|
|
|
g_main_loop_unref (loop);
|
|
|
|
return 0;
|
|
}
|