gstreamer/subprojects/gst-plugins-bad/sys/d3d11/gstd3d11winrtcapture.cpp

/*
 * GStreamer
 * Copyright (C) 2022 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

#ifdef WINAPI_PARTITION_APP
#undef WINAPI_PARTITION_APP
#endif

#define WINAPI_PARTITION_APP 1

#include "gstd3d11winrtcapture.h"
#include "gstd3d11pluginutils.h"
#include <gmodule.h>
#include <winstring.h>
#include <roapi.h>
#include <windows.graphics.capture.h>
#include <windows.graphics.capture.interop.h>
#include <windows.graphics.directx.direct3d11.h>
#include <windows.graphics.directx.direct3d11.interop.h>
#include <string.h>

#include <wrl.h>

#ifdef HAVE_WINMM
#include <mmsystem.h>
#endif

GST_DEBUG_CATEGORY_EXTERN (gst_d3d11_screen_capture_debug);
#define GST_CAT_DEFAULT gst_d3d11_screen_capture_debug

/* *INDENT-OFF* */
using namespace Microsoft::WRL;
using namespace ABI::Windows::Foundation;
using namespace ABI::Windows::Graphics;
using namespace ABI::Windows::Graphics::Capture;
using namespace ABI::Windows::Graphics::DirectX;
using namespace ABI::Windows::Graphics::DirectX::Direct3D11;
using namespace Windows::Graphics::DirectX::Direct3D11;

static SRWLOCK capture_list_lock = SRWLOCK_INIT;
static GList *capture_list = nullptr;

#define D3D11_WINRT_CAPTURE_PROP_NAME "gst-d3d11-winrt-capture"
#define WM_GST_D3D11_WINRT_CAPTURE_CLOSED (WM_USER + 1)

typedef struct
{
  gboolean loaded;

  /* d3d11.dll */
  HRESULT (WINAPI * CreateDirect3D11DeviceFromDXGIDevice) (IDXGIDevice *
      dxgi_device, IInspectable ** graphics_device);

  /* combase.dll */
  HRESULT (WINAPI * RoInitialize) (RO_INIT_TYPE init_type);
  HRESULT (WINAPI * RoUninitialize) (void);
  HRESULT (WINAPI * WindowsCreateString) (PCNZWCH source_string,
      UINT32 length, HSTRING * string);
  HRESULT (WINAPI * WindowsDeleteString) (HSTRING string);
  HRESULT (WINAPI * RoGetActivationFactory) (HSTRING activatable_class_id,
      REFIID iid, void ** factory);
} GstD3D11WinRTVTable;

static GstD3D11WinRTVTable winrt_vtable = { FALSE, };

template < typename InterfaceType, PCNZWCH runtime_class_id >
static HRESULT
GstGetActivationFactory (InterfaceType ** factory)
{
  if (!gst_d3d11_winrt_capture_load_library ())
    return E_NOINTERFACE;

  HSTRING class_id_hstring;
  HRESULT hr = winrt_vtable.WindowsCreateString (runtime_class_id,
      wcslen (runtime_class_id), &class_id_hstring);

  if (FAILED (hr))
    return hr;

  hr = winrt_vtable.RoGetActivationFactory (class_id_hstring,
      IID_PPV_ARGS (factory));

  if (FAILED (hr)) {
    winrt_vtable.WindowsDeleteString (class_id_hstring);
    return hr;
  }

  return winrt_vtable.WindowsDeleteString (class_id_hstring);
}

#define CLOSE_COM(obj) G_STMT_START { \
  if (obj) { \
    ComPtr<IClosable> closable; \
    obj.As (&closable); \
    if (closable) \
      closable->Close (); \
    obj = nullptr; \
  } \
} G_STMT_END

struct GstD3D11WinRTCaptureInner
{
  ~GstD3D11WinRTCaptureInner()
  {
    CLOSE_COM (session);
    CLOSE_COM (pool);
    CLOSE_COM (item);
    CLOSE_COM (d3d_device);
  }

  STDMETHODIMP
  OnClosed (IGraphicsCaptureItem * item, IInspectable * args)
  {
    GST_WARNING ("Item %p got closed", this);
    this->closed = true;

    return S_OK;
  }

  ComPtr < IDirect3DDevice > d3d_device;
  ComPtr < IGraphicsCaptureItem > item;
  ComPtr < IDirect3D11CaptureFramePool > pool;
  ComPtr < IGraphicsCaptureSession > session;

  bool closed = false;
};
/* *INDENT-ON* */

#define LOAD_SYMBOL(module,name,func) G_STMT_START { \
  if (!g_module_symbol (module, G_STRINGIFY (name), (gpointer *) &winrt_vtable.func)) { \
    GST_WARNING ("Failed to load '%s', %s", G_STRINGIFY (name), g_module_error()); \
    if (d3d11_module) \
      g_module_close (d3d11_module); \
    if (combase_module) \
      g_module_close (combase_module); \
    return; \
  } \
} G_STMT_END

gboolean
gst_d3d11_winrt_capture_load_library (void)
{
  static GModule *d3d11_module = nullptr;
  static GModule *combase_module = nullptr;

  GST_D3D11_CALL_ONCE_BEGIN {
    d3d11_module = g_module_open ("d3d11.dll", G_MODULE_BIND_LAZY);
    /* Shouldn't happen... */
    if (!d3d11_module)
      return;

    combase_module = g_module_open ("combase.dll", G_MODULE_BIND_LAZY);
    if (!combase_module) {
      g_module_close (d3d11_module);
      return;
    }

    LOAD_SYMBOL (d3d11_module, CreateDirect3D11DeviceFromDXGIDevice,
        CreateDirect3D11DeviceFromDXGIDevice);
    LOAD_SYMBOL (combase_module, RoInitialize, RoInitialize);
    LOAD_SYMBOL (combase_module, RoUninitialize, RoUninitialize);
    LOAD_SYMBOL (combase_module, WindowsCreateString, WindowsCreateString);
    LOAD_SYMBOL (combase_module, WindowsDeleteString, WindowsDeleteString);
    LOAD_SYMBOL (combase_module, RoGetActivationFactory,
        RoGetActivationFactory);

    winrt_vtable.loaded = TRUE;
  }
  GST_D3D11_CALL_ONCE_END;

  return winrt_vtable.loaded;
}

enum
{
  PROP_0,
  PROP_D3D11_DEVICE,
  PROP_MONITOR_HANDLE,
  PROP_WINDOW_HANDLE,
};

struct _GstD3D11WinRTCapture
{
  GstD3D11ScreenCapture parent;

  GstD3D11Device *device;
  GstD3D11WinRTCaptureInner *inner;
  /* Reported by WGC API */
  SizeInt32 pool_size;
  /* Actual texture resolution */
  UINT width;
  UINT height;

  gboolean flushing;
  boolean show_mouse;
  boolean show_border;

  GThread *thread;
  GMainContext *context;
  GMainLoop *loop;

  CRITICAL_SECTION lock;
  CONDITION_VARIABLE cond;
  LARGE_INTEGER frequency;

  HMONITOR monitor_handle;
  HWND window_handle;

  HWND hidden_window;
};

static void gst_d3d11_winrt_capture_constructed (GObject * object);
static void gst_d3d11_winrt_capture_dispose (GObject * object);
static void gst_d3d11_winrt_capture_finalize (GObject * object);
static void gst_d3d11_winrt_capture_set_property (GObject * object,
    guint prop_id, const GValue * value, GParamSpec * pspec);

static GstFlowReturn
gst_d3d11_winrt_capture_prepare (GstD3D11ScreenCapture * capture);
static gboolean
gst_d3d11_winrt_capture_get_size (GstD3D11ScreenCapture * capture,
    guint * width, guint * height);
static gboolean
gst_d3d11_winrt_capture_get_colorimetry (GstD3D11ScreenCapture * capture,
    GstVideoColorimetry * colorimetry);
static gboolean
gst_d3d11_winrt_capture_unlock (GstD3D11ScreenCapture * capture);
static gboolean
gst_d3d11_winrt_capture_unlock_stop (GstD3D11ScreenCapture * capture);
static void
gst_d3d11_winrt_capture_show_border (GstD3D11ScreenCapture * capture,
    gboolean show);
static GstFlowReturn
gst_d3d11_winrt_capture_do_capture (GstD3D11ScreenCapture * capture,
    GstD3D11Device * device, ID3D11Texture2D * texture,
    ID3D11RenderTargetView * rtv, ID3D11VertexShader * vs,
    ID3D11PixelShader * ps, ID3D11InputLayout * layout,
    ID3D11SamplerState * sampler, ID3D11BlendState * blend,
    D3D11_BOX * crop_box, gboolean draw_mouse);
static gpointer
gst_d3d11_winrt_capture_thread_func (GstD3D11WinRTCapture * self);

#define gst_d3d11_winrt_capture_parent_class parent_class
G_DEFINE_TYPE (GstD3D11WinRTCapture, gst_d3d11_winrt_capture,
    GST_TYPE_D3D11_SCREEN_CAPTURE);

static void
gst_d3d11_winrt_capture_class_init (GstD3D11WinRTCaptureClass * klass)
{
  GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
  GstD3D11ScreenCaptureClass *capture_class =
      GST_D3D11_SCREEN_CAPTURE_CLASS (klass);

  gobject_class->constructed = gst_d3d11_winrt_capture_constructed;
  gobject_class->dispose = gst_d3d11_winrt_capture_dispose;
  gobject_class->finalize = gst_d3d11_winrt_capture_finalize;
  gobject_class->set_property = gst_d3d11_winrt_capture_set_property;

  g_object_class_install_property (gobject_class, PROP_D3D11_DEVICE,
      g_param_spec_object ("d3d11device", "D3D11 Device",
          "GstD3D11Device object for operating",
          GST_TYPE_D3D11_DEVICE, (GParamFlags)
          (G_PARAM_WRITABLE | G_PARAM_CONSTRUCT_ONLY |
              G_PARAM_STATIC_STRINGS)));
  g_object_class_install_property (gobject_class, PROP_MONITOR_HANDLE,
      g_param_spec_pointer ("monitor-handle", "Monitor Handle",
          "A HMONITOR handle of monitor to capture", (GParamFlags)
          (G_PARAM_WRITABLE | G_PARAM_CONSTRUCT_ONLY |
              G_PARAM_STATIC_STRINGS)));
  g_object_class_install_property (gobject_class, PROP_WINDOW_HANDLE,
      g_param_spec_pointer ("window-handle", "Window Handle",
          "A HWND handle of window to capture", (GParamFlags)
          (G_PARAM_WRITABLE | G_PARAM_CONSTRUCT_ONLY |
              G_PARAM_STATIC_STRINGS)));

  capture_class->prepare = GST_DEBUG_FUNCPTR (gst_d3d11_winrt_capture_prepare);
  capture_class->get_size =
      GST_DEBUG_FUNCPTR (gst_d3d11_winrt_capture_get_size);
  capture_class->get_colorimetry =
      GST_DEBUG_FUNCPTR (gst_d3d11_winrt_capture_get_colorimetry);
  capture_class->unlock = GST_DEBUG_FUNCPTR (gst_d3d11_winrt_capture_unlock);
  capture_class->unlock_stop =
      GST_DEBUG_FUNCPTR (gst_d3d11_winrt_capture_unlock_stop);
  capture_class->show_border =
      GST_DEBUG_FUNCPTR (gst_d3d11_winrt_capture_show_border);
  capture_class->do_capture =
      GST_DEBUG_FUNCPTR (gst_d3d11_winrt_capture_do_capture);
}

static void
gst_d3d11_winrt_capture_init (GstD3D11WinRTCapture * self)
{
  InitializeCriticalSection (&self->lock);
}

static void
gst_d3d11_winrt_capture_constructed (GObject * object)
{
  GstD3D11WinRTCapture *self = GST_D3D11_WINRT_CAPTURE (object);
  GstD3D11CSLockGuard lk (&self->lock);

  self->context = g_main_context_new ();
  self->loop = g_main_loop_new (self->context, FALSE);
  self->thread = g_thread_new ("GstD3D11WinRTCapture",
      (GThreadFunc) gst_d3d11_winrt_capture_thread_func, self);
  while (!g_main_loop_is_running (self->loop))
    SleepConditionVariableCS (&self->cond, &self->lock, INFINITE);

  G_OBJECT_CLASS (parent_class)->constructed (object);
}

static void
gst_d3d11_winrt_capture_dispose (GObject * object)
{
  GstD3D11WinRTCapture *self = GST_D3D11_WINRT_CAPTURE (object);

  if (self->loop)
    g_main_loop_quit (self->loop);

  g_clear_pointer (&self->thread, g_thread_join);
  g_clear_pointer (&self->loop, g_main_loop_unref);
  g_clear_pointer (&self->context, g_main_context_unref);

  gst_clear_object (&self->device);

  G_OBJECT_CLASS (parent_class)->dispose (object);
}

static void
gst_d3d11_winrt_capture_finalize (GObject * object)
{
  GstD3D11WinRTCapture *self = GST_D3D11_WINRT_CAPTURE (object);

  DeleteCriticalSection (&self->lock);

  G_OBJECT_CLASS (parent_class)->finalize (object);
}

static void
gst_d3d11_winrt_capture_set_property (GObject * object, guint prop_id,
    const GValue * value, GParamSpec * pspec)
{
  GstD3D11WinRTCapture *self = GST_D3D11_WINRT_CAPTURE (object);

  switch (prop_id) {
    case PROP_D3D11_DEVICE:
      self->device = (GstD3D11Device *) g_value_dup_object (value);
      break;
    case PROP_MONITOR_HANDLE:
      self->monitor_handle = (HMONITOR) g_value_get_pointer (value);
      break;
    case PROP_WINDOW_HANDLE:
      self->window_handle = (HWND) g_value_get_pointer (value);
      break;
    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
      break;
  }
}

static gboolean
gst_d3d11_winrt_capture_running_cb (GstD3D11WinRTCapture * self)
{
  GstD3D11CSLockGuard lk (&self->lock);
  WakeAllConditionVariable (&self->cond);

  return G_SOURCE_REMOVE;
}

static void
gst_d3d11_winrt_configure (GstD3D11WinRTCapture * self)
{
  HRESULT hr;
  GstD3D11Device *device = self->device;
  ComPtr < ID3D10Multithread > multi_thread;
  ComPtr < IGraphicsCaptureItemInterop > interop;
  ID3D11Device *device_handle;
  ComPtr < IDXGIDevice > dxgi_device;
  ComPtr < IInspectable > inspectable;
  ComPtr < IDirect3D11CaptureFramePoolStatics > pool_statics;
  ComPtr < IDirect3D11CaptureFramePoolStatics2 > pool_statics2;
  ComPtr < IGraphicsCaptureSession2 > session2;
  ComPtr < IGraphicsCaptureSession3 > session3;
  GstD3D11WinRTCaptureInner *inner = nullptr;

  device_handle = gst_d3d11_device_get_device_handle (device);
  hr = device_handle->QueryInterface (IID_PPV_ARGS (&multi_thread));
  if (!gst_d3d11_result (hr, device)) {
    GST_ERROR_OBJECT (self, "ID3D10Multithread interface is unavailable");
    return;
  }

  multi_thread->SetMultithreadProtected (TRUE);

  hr = GstGetActivationFactory < IGraphicsCaptureItemInterop,
      RuntimeClass_Windows_Graphics_Capture_GraphicsCaptureItem > (&interop);
  if (!gst_d3d11_result (hr, device)) {
    GST_WARNING_OBJECT (self, "IGraphicsCaptureItemInterop is not available");
    return;
  }

  inner = new GstD3D11WinRTCaptureInner ();

  if (self->monitor_handle) {
    hr = interop->CreateForMonitor (self->monitor_handle,
        IID_PPV_ARGS (&inner->item));
  } else if (self->window_handle) {
    hr = interop->CreateForWindow (self->window_handle,
        IID_PPV_ARGS (&inner->item));
  } else {
    g_assert_not_reached ();
    goto error;
  }

  if (!gst_d3d11_result (hr, device)) {
    GST_WARNING_OBJECT (self, "Could not create item");
    goto error;
  }

  hr = device_handle->QueryInterface (IID_PPV_ARGS (&dxgi_device));
  if (!gst_d3d11_result (hr, device)) {
    GST_WARNING_OBJECT (self, "IDXGIDevice is not available");
    goto error;
  }

  hr = winrt_vtable.CreateDirect3D11DeviceFromDXGIDevice (dxgi_device.Get (),
      &inspectable);
  if (!gst_d3d11_result (hr, device)) {
    GST_WARNING_OBJECT (self, "CreateDirect3D11DeviceFromDXGIDevice failed");
    goto error;
  }

  hr = inspectable.As (&inner->d3d_device);
  if (!gst_d3d11_result (hr, device)) {
    GST_WARNING_OBJECT (device, "IDirect3DDevice is not available");
    goto error;
  }

  hr = GstGetActivationFactory < IDirect3D11CaptureFramePoolStatics,
      RuntimeClass_Windows_Graphics_Capture_Direct3D11CaptureFramePool >
      (&pool_statics);
  if (!gst_d3d11_result (hr, self->device)) {
    GST_ERROR_OBJECT (self,
        "IDirect3D11CaptureFramePoolStatics is not available");
    goto error;
  }

  hr = pool_statics.As (&pool_statics2);
  if (!gst_d3d11_result (hr, self->device)) {
    GST_ERROR_OBJECT (self,
        "IDirect3D11CaptureFramePoolStatics2 is not available");
    goto error;
  }

  hr = inner->item->get_Size (&self->pool_size);
  if (!gst_d3d11_result (hr, self->device)) {
    GST_ERROR_OBJECT (self, "Could not get item size");
    goto error;
  }

  self->width = (UINT) self->pool_size.Width;
  self->height = (UINT) self->pool_size.Height;

  hr = pool_statics2->CreateFreeThreaded (inner->d3d_device.Get (),
      DirectXPixelFormat::DirectXPixelFormat_B8G8R8A8UIntNormalized,
      1, self->pool_size, &inner->pool);
  if (!gst_d3d11_result (hr, self->device)) {
    GST_ERROR_OBJECT (self, "Could not setup pool");
    goto error;
  }

  hr = inner->pool->CreateCaptureSession (inner->item.Get (), &inner->session);
  if (!gst_d3d11_result (hr, self->device)) {
    GST_ERROR_OBJECT (self, "Could not create session");
    goto error;
  }

  inner->session.As (&session2);
  if (session2)
    session2->put_IsCursorCaptureEnabled (FALSE);

  inner->session.As (&session3);
  if (session3)
    session3->put_IsBorderRequired (self->show_border);

  hr = inner->session->StartCapture ();
  if (!gst_d3d11_result (hr, self->device)) {
    GST_ERROR_OBJECT (self, "Could not start capture");
    goto error;
  }

  self->inner = inner;

  return;

error:
  if (inner)
    delete inner;
}

static LRESULT CALLBACK
gst_d3d11_winrt_capture_proc (HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
{
  GstD3D11WinRTCapture *self;

  if (msg == WM_CREATE) {
    self = GST_D3D11_WINRT_CAPTURE (((LPCREATESTRUCTA) lparam)->lpCreateParams);

    SetPropA (hwnd, D3D11_WINRT_CAPTURE_PROP_NAME, self);
  } else if (GetPropA (hwnd, D3D11_WINRT_CAPTURE_PROP_NAME) &&
      msg == WM_GST_D3D11_WINRT_CAPTURE_CLOSED) {
    HANDLE handle = GetPropA (hwnd, D3D11_WINRT_CAPTURE_PROP_NAME);

    if (!GST_IS_D3D11_WINRT_CAPTURE (handle)) {
      GST_WARNING ("%p is not d3d11window object", handle);
      return DefWindowProcA (hwnd, msg, wparam, lparam);
    }

    self = GST_D3D11_WINRT_CAPTURE (handle);
    GST_INFO_OBJECT (self, "Target window got closed");
    GstD3D11CSLockGuard lk (&self->lock);
    if (self->inner)
      self->inner->closed = true;
    WakeAllConditionVariable (&self->cond);

    return 0;
  }

  return DefWindowProcA (hwnd, msg, wparam, lparam);
}

static HWND
gst_d3d11_winrt_create_hidden_window (GstD3D11WinRTCapture * self)
{
  static SRWLOCK lock = SRWLOCK_INIT;
  WNDCLASSEXA wc;
  ATOM atom;
  HINSTANCE inst = GetModuleHandle (nullptr);

  AcquireSRWLockExclusive (&lock);
  atom = GetClassInfoExA (inst, "GstD3D11WinRTCapture", &wc);
  if (atom == 0) {
    ZeroMemory (&wc, sizeof (WNDCLASSEXA));

    wc.cbSize = sizeof (WNDCLASSEXA);
    wc.lpfnWndProc = gst_d3d11_winrt_capture_proc;
    wc.hInstance = inst;
    wc.style = CS_OWNDC;
    wc.lpszClassName = "GstD3D11WinRTCapture";

    atom = RegisterClassExA (&wc);
    ReleaseSRWLockExclusive (&lock);

    if (atom == 0) {
      GST_ERROR_OBJECT (self, "Failed to register window class 0x%x",
          (guint) GetLastError ());
      return nullptr;
    }
  } else {
    ReleaseSRWLockExclusive (&lock);
  }

  return CreateWindowExA (0, "GstD3D11WinRTCapture", "GstD3D11WinRTCapture",
      WS_POPUP, 0, 0, 1, 1, nullptr, nullptr, inst, self);
}

static void CALLBACK
event_hook_func (HWINEVENTHOOK hook, DWORD event, HWND hwnd, LONG id_obj,
    LONG id_child, DWORD id_event_thread, DWORD event_time)
{
  if (event != EVENT_OBJECT_DESTROY || id_obj != OBJID_WINDOW ||
      id_child != INDEXID_CONTAINER || !hwnd) {
    return;
  }

  GstD3D11SRWLockGuard lk (&capture_list_lock);
  GList *iter;

  for (iter = capture_list; iter; iter = g_list_next (iter)) {
    GstD3D11WinRTCapture *capture = GST_D3D11_WINRT_CAPTURE (iter->data);
    GstD3D11CSLockGuard capture_lk (&capture->lock);

    if (capture->hidden_window && capture->window_handle == hwnd) {
      PostMessageA (capture->hidden_window, WM_GST_D3D11_WINRT_CAPTURE_CLOSED,
          0, 0);
      return;
    }
  }
}

static gboolean
gst_d3d11_winrt_capture_msg_cb (GIOChannel * source, GIOCondition condition,
    gpointer data)
{
  MSG msg;

  if (!PeekMessage (&msg, NULL, 0, 0, PM_REMOVE))
    return G_SOURCE_CONTINUE;

  TranslateMessage (&msg);
  DispatchMessage (&msg);

  return G_SOURCE_CONTINUE;
}

static void
gst_d3d11_winrt_capture_weak_ref_notify (gpointer data,
    GstD3D11WinRTCapture * self)
{
  GstD3D11SRWLockGuard lk (&capture_list_lock);
  capture_list = g_list_remove (capture_list, self);
}

static gpointer
gst_d3d11_winrt_capture_thread_func (GstD3D11WinRTCapture * self)
{
  GSource *source;
  GSource *msg_source = nullptr;
  GIOChannel *msg_io_channel = nullptr;
  HWINEVENTHOOK hook = nullptr;
#if HAVE_WINMM
  TIMECAPS time_caps;
  guint timer_res = 0;

  if (timeGetDevCaps (&time_caps, sizeof (TIMECAPS)) == TIMERR_NOERROR) {
    guint resolution;

    resolution = MIN (MAX (time_caps.wPeriodMin, 1), time_caps.wPeriodMax);

    if (timeBeginPeriod (resolution) != TIMERR_NOERROR)
      timer_res = resolution;
  }
#endif

  QueryPerformanceFrequency (&self->frequency);

  winrt_vtable.RoInitialize (RO_INIT_MULTITHREADED);
  g_main_context_push_thread_default (self->context);

  source = g_idle_source_new ();
  g_source_set_callback (source,
      (GSourceFunc) gst_d3d11_winrt_capture_running_cb, self, nullptr);
  g_source_attach (source, self->context);
  g_source_unref (source);

  gst_d3d11_winrt_configure (self);
  if (self->inner && self->window_handle) {
    /* hold list of capture objects to send target window closed event */
    AcquireSRWLockExclusive (&capture_list_lock);
    g_object_weak_ref (G_OBJECT (self),
        (GWeakNotify) gst_d3d11_winrt_capture_weak_ref_notify, nullptr);
    capture_list = g_list_append (capture_list, self);
    ReleaseSRWLockExclusive (&capture_list_lock);

    self->hidden_window = gst_d3d11_winrt_create_hidden_window (self);
    if (self->hidden_window) {
      DWORD process_id, thread_id;

      thread_id = GetWindowThreadProcessId (self->window_handle, &process_id);
      if (thread_id) {
        hook = SetWinEventHook (EVENT_OBJECT_DESTROY, EVENT_OBJECT_DESTROY,
            nullptr, event_hook_func, process_id, thread_id,
            WINEVENT_OUTOFCONTEXT);
      }

      msg_io_channel =
          g_io_channel_win32_new_messages ((guintptr) self->hidden_window);
      msg_source = g_io_create_watch (msg_io_channel, G_IO_IN);
      g_source_set_callback (msg_source,
          (GSourceFunc) gst_d3d11_winrt_capture_msg_cb, self, nullptr);
      g_source_attach (msg_source, self->context);
    }
  }

  g_main_loop_run (self->loop);

  if (hook)
    UnhookWinEvent (hook);

  EnterCriticalSection (&self->lock);
  if (self->hidden_window) {
    RemovePropA (self->hidden_window, D3D11_WINRT_CAPTURE_PROP_NAME);
    DestroyWindow (self->hidden_window);
    self->hidden_window = nullptr;
  }
  LeaveCriticalSection (&self->lock);

  if (msg_source) {
    g_source_destroy (msg_source);
    g_source_unref (msg_source);
  }

  if (msg_io_channel)
    g_io_channel_unref (msg_io_channel);

  if (self->inner)
    delete self->inner;
  self->inner = nullptr;

  g_main_context_pop_thread_default (self->context);
  winrt_vtable.RoUninitialize ();

#if HAVE_WINMM
  if (timer_res != 0)
    timeEndPeriod (timer_res);
#endif

  return nullptr;
}

static GstFlowReturn
gst_d3d11_winrt_capture_prepare (GstD3D11ScreenCapture * capture)
{
  GstD3D11WinRTCapture *self = GST_D3D11_WINRT_CAPTURE (capture);

  g_assert (self->inner != nullptr);

  return GST_FLOW_OK;
}

static gboolean
gst_d3d11_winrt_capture_get_size (GstD3D11ScreenCapture * capture,
    guint * width, guint * height)
{
  GstD3D11WinRTCapture *self = GST_D3D11_WINRT_CAPTURE (capture);

  *width = self->width;
  *height = self->height;

  return TRUE;
}

static gboolean
gst_d3d11_winrt_capture_get_colorimetry (GstD3D11ScreenCapture * capture,
    GstVideoColorimetry * colorimetry)
{
  return FALSE;
}

static gboolean
gst_d3d11_winrt_capture_unlock (GstD3D11ScreenCapture * capture)
{
  GstD3D11WinRTCapture *self = GST_D3D11_WINRT_CAPTURE (capture);
  GstD3D11CSLockGuard lk (&self->lock);

  self->flushing = TRUE;
  WakeAllConditionVariable (&self->cond);

  return TRUE;
}

static gboolean
gst_d3d11_winrt_capture_unlock_stop (GstD3D11ScreenCapture * capture)
{
  GstD3D11WinRTCapture *self = GST_D3D11_WINRT_CAPTURE (capture);
  GstD3D11CSLockGuard lk (&self->lock);

  self->flushing = FALSE;
  WakeAllConditionVariable (&self->cond);

  return TRUE;
}

static void
gst_d3d11_winrt_capture_show_border (GstD3D11ScreenCapture * capture,
    gboolean show)
{
  GstD3D11WinRTCapture *self = GST_D3D11_WINRT_CAPTURE (capture);
  GstD3D11CSLockGuard lk (&self->lock);

  self->show_border = show;
  if (self->inner->session) {
    ComPtr < IGraphicsCaptureSession3 > session3;
    self->inner->session.As (&session3);

    if (session3)
      session3->put_IsBorderRequired (self->show_border);
  }
}

static GstFlowReturn
gst_d3d11_winrt_capture_do_capture (GstD3D11ScreenCapture * capture,
    GstD3D11Device * device, ID3D11Texture2D * texture,
    ID3D11RenderTargetView * rtv, ID3D11VertexShader * vs,
    ID3D11PixelShader * ps, ID3D11InputLayout * layout,
    ID3D11SamplerState * sampler, ID3D11BlendState * blend,
    D3D11_BOX * crop_box, gboolean draw_mouse)
{
  GstD3D11WinRTCapture *self = GST_D3D11_WINRT_CAPTURE (capture);
  GstD3D11WinRTCaptureInner *inner = self->inner;
  ComPtr < IDirect3D11CaptureFrame > frame;
  ComPtr < IDirect3DSurface > surface;
  ComPtr < IDirect3DDxgiInterfaceAccess > access;
  ComPtr < ID3D11Texture2D > captured_texture;
  ID3D11DeviceContext *context_handle;
  SizeInt32 size;
  HRESULT hr;
  LARGE_INTEGER now;
  LONGLONG timeout;
  D3D11_TEXTURE2D_DESC desc;
  gboolean size_changed = FALSE;

  GstD3D11CSLockGuard lk (&self->lock);
again:
  frame = nullptr;
  surface = nullptr;
  access = nullptr;
  captured_texture = nullptr;

  if (inner->closed) {
    GST_ERROR_OBJECT (self, "Item was closed");
    return GST_FLOW_ERROR;
  }

  if (self->flushing) {
    GST_INFO_OBJECT (self, "We are flushing");
    return GST_FLOW_FLUSHING;
  }

  if ((draw_mouse && !self->show_mouse) || (!draw_mouse && self->show_mouse)) {
    ComPtr < IGraphicsCaptureSession2 > session2;
    self->show_mouse = draw_mouse;

    inner->session.As (&session2);
    if (session2) {
      hr = session2->put_IsCursorCaptureEnabled (draw_mouse);
      if (!gst_d3d11_result (hr, self->device))
        GST_DEBUG_OBJECT (self, "Could not set IsCursorCaptureEnabled");
    } else {
      GST_LOG_OBJECT (self, "IGraphicsCaptureSession2 is not available");
    }
  }

  /* Magic number 5 sec timeout */
  QueryPerformanceCounter (&now);
  timeout = now.QuadPart + 5 * self->frequency.QuadPart;

  do {
    hr = inner->pool->TryGetNextFrame (&frame);
    if (frame)
      break;

    if (!gst_d3d11_result (hr, self->device)) {
      GST_ERROR_OBJECT (self, "Could not capture frame");
      return GST_FLOW_ERROR;
    }

    SleepConditionVariableCS (&self->cond, &self->lock, 1);
    QueryPerformanceCounter (&now);
  } while (!inner->closed && !self->flushing && now.QuadPart < timeout);

  if (self->flushing) {
    GST_INFO_OBJECT (self, "We are flushing");
    return GST_FLOW_FLUSHING;
  }

  if (inner->closed) {
    GST_WARNING_OBJECT (self, "Capture item was closed");
    return GST_FLOW_ERROR;
  }

  if (!frame) {
    GST_WARNING_OBJECT (self, "No frame available");
    return GST_D3D11_SCREEN_CAPTURE_FLOW_EXPECTED_ERROR;
  }

  hr = frame->get_ContentSize (&size);
  if (!gst_d3d11_result (hr, self->device)) {
    GST_ERROR_OBJECT (self, "Could not get content size");
    return GST_FLOW_ERROR;
  }

  if (size.Width != self->pool_size.Width ||
      size.Height != self->pool_size.Height) {
    GST_DEBUG_OBJECT (self, "Size changed %dx%d -> %dx%d",
        self->pool_size.Width, self->pool_size.Height, size.Width, size.Height);
    self->pool_size = size;
    frame = nullptr;
    hr = inner->pool->Recreate (inner->d3d_device.Get (),
        DirectXPixelFormat::DirectXPixelFormat_B8G8R8A8UIntNormalized,
        1, self->pool_size);
    if (!gst_d3d11_result (hr, self->device)) {
      GST_ERROR_OBJECT (self, "Could not recreate");
      return GST_FLOW_ERROR;
    }

    size_changed = TRUE;
    goto again;
  }

  hr = frame->get_Surface (&surface);
  if (!gst_d3d11_result (hr, self->device)) {
    GST_ERROR_OBJECT (self, "Could not get IDirect3DSurface");
    return GST_FLOW_ERROR;
  }

  hr = surface.As (&access);
  if (!gst_d3d11_result (hr, self->device)) {
    GST_ERROR_OBJECT (self, "Could not get IDirect3DDxgiInterfaceAccess");
    return GST_FLOW_ERROR;
  }

  hr = access->GetInterface (IID_PPV_ARGS (&captured_texture));
  if (!gst_d3d11_result (hr, self->device)) {
    GST_ERROR_OBJECT (self, "Could not get texture from frame");
    return GST_FLOW_ERROR;
  }

  /* XXX: actual texture size can be different from reported pool size */
  captured_texture->GetDesc (&desc);
  if (desc.Width != self->width || desc.Height != self->height) {
    GST_DEBUG_OBJECT (self, "Texture size changed %dx%d -> %dx%d",
        self->width, self->height, desc.Width, desc.Height);
    self->width = desc.Width;
    self->height = desc.Height;
    size_changed = TRUE;
  }

  if (size_changed)
    return GST_D3D11_SCREEN_CAPTURE_FLOW_SIZE_CHANGED;

  context_handle = gst_d3d11_device_get_device_context_handle (self->device);
  GstD3D11DeviceLockGuard device_lk (self->device);
  context_handle->CopySubresourceRegion (texture, 0, 0, 0, 0,
      captured_texture.Get (), 0, crop_box);

  return GST_FLOW_OK;
}

GstD3D11ScreenCapture *
gst_d3d11_winrt_capture_new (GstD3D11Device * device, HMONITOR monitor_handle,
    HWND window_handle)
{
  GstD3D11WinRTCapture *self;

  if (window_handle && !IsWindow (window_handle)) {
    GST_WARNING_OBJECT (device, "Not a valid window handle");
    return nullptr;
  }

  if (!gst_d3d11_winrt_capture_load_library ())
    return nullptr;

  self = (GstD3D11WinRTCapture *) g_object_new (GST_TYPE_D3D11_WINRT_CAPTURE,
      "d3d11device", device, "monitor-handle", (gpointer) monitor_handle,
      "window-handle", (gpointer) window_handle, nullptr);
  if (!self->inner) {
    gst_clear_object (&self);
    return nullptr;
  }

  gst_object_ref_sink (self);

  return GST_D3D11_SCREEN_CAPTURE_CAST (self);
}