mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-04 23:46:43 +00:00
ae3ed20f41
Since DXGI desktop duplication API does not work with Direct3D12 device, this element will use Direct3D11 device to acquire frame. Then other rendering operations (e.g., texture copy, render pipeline) will happen using Direct3D12 API Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/5883>
276 lines
6.9 KiB
C++
276 lines
6.9 KiB
C++
/* GStreamer
|
|
* Copyright (C) 2024 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 "gstd3d12screencapture.h"
|
|
#include "gstd3d12pluginutils.h"
|
|
#include <string.h>
|
|
|
|
#include <wrl.h>
|
|
|
|
GST_DEBUG_CATEGORY_EXTERN (gst_d3d12_screen_capture_debug);
|
|
#define GST_CAT_DEFAULT gst_d3d12_screen_capture_debug
|
|
|
|
/* *INDENT-OFF* */
|
|
using namespace Microsoft::WRL;
|
|
/* *INDENT-ON* */
|
|
|
|
#define gst_d3d12_screen_capture_parent_class parent_class
|
|
G_DEFINE_ABSTRACT_TYPE (GstD3D12ScreenCapture, gst_d3d12_screen_capture,
|
|
GST_TYPE_OBJECT);
|
|
|
|
static void
|
|
gst_d3d12_screen_capture_class_init (GstD3D12ScreenCaptureClass * klass)
|
|
{
|
|
}
|
|
|
|
static void
|
|
gst_d3d12_screen_capture_init (GstD3D12ScreenCapture * self)
|
|
{
|
|
}
|
|
|
|
GstFlowReturn
|
|
gst_d3d12_screen_capture_prepare (GstD3D12ScreenCapture * capture)
|
|
{
|
|
g_return_val_if_fail (GST_IS_D3D12_SCREEN_CAPTURE (capture), GST_FLOW_ERROR);
|
|
|
|
auto klass = GST_D3D12_SCREEN_CAPTURE_GET_CLASS (capture);
|
|
g_assert (klass->prepare);
|
|
|
|
return klass->prepare (capture);
|
|
}
|
|
|
|
gboolean
|
|
gst_d3d12_screen_capture_get_size (GstD3D12ScreenCapture * capture,
|
|
guint * width, guint * height)
|
|
{
|
|
g_return_val_if_fail (GST_IS_D3D12_SCREEN_CAPTURE (capture), FALSE);
|
|
g_return_val_if_fail (width != nullptr, FALSE);
|
|
g_return_val_if_fail (height != nullptr, FALSE);
|
|
|
|
auto klass = GST_D3D12_SCREEN_CAPTURE_GET_CLASS (capture);
|
|
g_assert (klass->get_size);
|
|
|
|
return klass->get_size (capture, width, height);
|
|
}
|
|
|
|
gboolean
|
|
gst_d3d12_screen_capture_unlock (GstD3D12ScreenCapture * capture)
|
|
{
|
|
g_return_val_if_fail (GST_IS_D3D12_SCREEN_CAPTURE (capture), FALSE);
|
|
|
|
auto klass = GST_D3D12_SCREEN_CAPTURE_GET_CLASS (capture);
|
|
|
|
if (klass->unlock)
|
|
return klass->unlock (capture);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
gboolean
|
|
gst_d3d12_screen_capture_unlock_stop (GstD3D12ScreenCapture * capture)
|
|
{
|
|
g_return_val_if_fail (GST_IS_D3D12_SCREEN_CAPTURE (capture), FALSE);
|
|
|
|
auto klass = GST_D3D12_SCREEN_CAPTURE_GET_CLASS (capture);
|
|
|
|
if (klass->unlock_stop)
|
|
return klass->unlock_stop (capture);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
void
|
|
gst_d3d12_screen_capture_show_border (GstD3D12ScreenCapture * capture,
|
|
gboolean show)
|
|
{
|
|
g_return_if_fail (GST_IS_D3D12_SCREEN_CAPTURE (capture));
|
|
|
|
auto klass = GST_D3D12_SCREEN_CAPTURE_GET_CLASS (capture);
|
|
|
|
if (klass->show_border)
|
|
klass->show_border (capture, show);
|
|
}
|
|
|
|
GstFlowReturn
|
|
gst_d3d12_screen_capture_do_capture (GstD3D12ScreenCapture * capture,
|
|
GstBuffer * buffer, const D3D12_BOX * crop_box, gboolean draw_mouse)
|
|
{
|
|
g_return_val_if_fail (GST_IS_D3D12_SCREEN_CAPTURE (capture), GST_FLOW_ERROR);
|
|
g_return_val_if_fail (GST_IS_BUFFER (buffer), GST_FLOW_ERROR);
|
|
|
|
auto klass = GST_D3D12_SCREEN_CAPTURE_GET_CLASS (capture);
|
|
g_assert (klass->do_capture);
|
|
|
|
return klass->do_capture (capture, buffer, crop_box, draw_mouse);
|
|
}
|
|
|
|
HRESULT
|
|
gst_d3d12_screen_capture_find_output_for_monitor (HMONITOR monitor,
|
|
IDXGIAdapter1 ** adapter, IDXGIOutput ** output)
|
|
{
|
|
ComPtr < IDXGIFactory1 > factory;
|
|
HRESULT hr = S_OK;
|
|
|
|
g_return_val_if_fail (monitor != nullptr, E_INVALIDARG);
|
|
|
|
hr = CreateDXGIFactory1 (IID_PPV_ARGS (&factory));
|
|
if (FAILED (hr))
|
|
return hr;
|
|
|
|
for (UINT adapter_idx = 0;; adapter_idx++) {
|
|
ComPtr < IDXGIAdapter1 > adapter_tmp;
|
|
|
|
hr = factory->EnumAdapters1 (adapter_idx, &adapter_tmp);
|
|
if (FAILED (hr))
|
|
break;
|
|
|
|
for (UINT output_idx = 0;; output_idx++) {
|
|
ComPtr < IDXGIOutput > output_tmp;
|
|
DXGI_OUTPUT_DESC desc;
|
|
|
|
hr = adapter_tmp->EnumOutputs (output_idx, &output_tmp);
|
|
if (FAILED (hr))
|
|
break;
|
|
|
|
hr = output_tmp->GetDesc (&desc);
|
|
if (FAILED (hr))
|
|
continue;
|
|
|
|
if (desc.Monitor == monitor) {
|
|
if (adapter)
|
|
*adapter = adapter_tmp.Detach ();
|
|
if (output)
|
|
*output = output_tmp.Detach ();
|
|
|
|
return S_OK;
|
|
}
|
|
}
|
|
}
|
|
|
|
return E_FAIL;
|
|
}
|
|
|
|
HRESULT
|
|
gst_d3d12_screen_capture_find_primary_monitor (HMONITOR * monitor,
|
|
IDXGIAdapter1 ** adapter, IDXGIOutput ** output)
|
|
{
|
|
ComPtr < IDXGIFactory1 > factory;
|
|
HRESULT hr = S_OK;
|
|
|
|
hr = CreateDXGIFactory1 (IID_PPV_ARGS (&factory));
|
|
if (FAILED (hr))
|
|
return hr;
|
|
|
|
for (UINT adapter_idx = 0;; adapter_idx++) {
|
|
ComPtr < IDXGIAdapter1 > adapter_tmp;
|
|
|
|
hr = factory->EnumAdapters1 (adapter_idx, &adapter_tmp);
|
|
if (FAILED (hr))
|
|
break;
|
|
|
|
for (UINT output_idx = 0;; output_idx++) {
|
|
ComPtr < IDXGIOutput > output_tmp;
|
|
DXGI_OUTPUT_DESC desc;
|
|
MONITORINFOEXW minfo;
|
|
|
|
hr = adapter_tmp->EnumOutputs (output_idx, &output_tmp);
|
|
if (FAILED (hr))
|
|
break;
|
|
|
|
hr = output_tmp->GetDesc (&desc);
|
|
if (FAILED (hr))
|
|
continue;
|
|
|
|
minfo.cbSize = sizeof (MONITORINFOEXW);
|
|
if (!GetMonitorInfoW (desc.Monitor, &minfo))
|
|
continue;
|
|
|
|
if ((minfo.dwFlags & MONITORINFOF_PRIMARY) != 0) {
|
|
if (monitor)
|
|
*monitor = desc.Monitor;
|
|
if (adapter)
|
|
*adapter = adapter_tmp.Detach ();
|
|
if (output)
|
|
*output = output_tmp.Detach ();
|
|
|
|
return S_OK;
|
|
}
|
|
}
|
|
}
|
|
|
|
return E_FAIL;
|
|
}
|
|
|
|
HRESULT
|
|
gst_d3d12_screen_capture_find_nth_monitor (guint index, HMONITOR * monitor,
|
|
IDXGIAdapter1 ** adapter, IDXGIOutput ** output)
|
|
{
|
|
ComPtr < IDXGIFactory1 > factory;
|
|
HRESULT hr = S_OK;
|
|
guint num_found = 0;
|
|
|
|
hr = CreateDXGIFactory1 (IID_PPV_ARGS (&factory));
|
|
if (FAILED (hr))
|
|
return hr;
|
|
|
|
for (UINT adapter_idx = 0;; adapter_idx++) {
|
|
ComPtr < IDXGIAdapter1 > adapter_tmp;
|
|
|
|
hr = factory->EnumAdapters1 (adapter_idx, &adapter_tmp);
|
|
if (FAILED (hr))
|
|
break;
|
|
|
|
for (UINT output_idx = 0;; output_idx++) {
|
|
ComPtr < IDXGIOutput > output_tmp;
|
|
DXGI_OUTPUT_DESC desc;
|
|
MONITORINFOEXW minfo;
|
|
|
|
hr = adapter_tmp->EnumOutputs (output_idx, &output_tmp);
|
|
if (FAILED (hr))
|
|
break;
|
|
|
|
hr = output_tmp->GetDesc (&desc);
|
|
if (FAILED (hr))
|
|
continue;
|
|
|
|
minfo.cbSize = sizeof (MONITORINFOEXW);
|
|
if (!GetMonitorInfoW (desc.Monitor, &minfo))
|
|
continue;
|
|
|
|
if (num_found == index) {
|
|
if (monitor)
|
|
*monitor = desc.Monitor;
|
|
if (adapter)
|
|
*adapter = adapter_tmp.Detach ();
|
|
if (output)
|
|
*output = output_tmp.Detach ();
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
num_found++;
|
|
}
|
|
}
|
|
|
|
return E_FAIL;
|
|
}
|