diff options
Diffstat (limited to 'sys/wasapi/gstwasapidevice.c')
-rw-r--r-- | sys/wasapi/gstwasapidevice.c | 205 |
1 files changed, 201 insertions, 4 deletions
diff --git a/sys/wasapi/gstwasapidevice.c b/sys/wasapi/gstwasapidevice.c index 4e6379c3e..c13fc6a8a 100644 --- a/sys/wasapi/gstwasapidevice.c +++ b/sys/wasapi/gstwasapidevice.c @@ -1,5 +1,6 @@ /* GStreamer * Copyright (C) 2018 Nirbheek Chauhan <nirbheek@centricular.com> + * 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 @@ -23,11 +24,27 @@ #include "gstwasapidevice.h" +GST_DEBUG_CATEGORY_EXTERN (gst_wasapi_debug); +#define GST_CAT_DEFAULT gst_wasapi_debug + G_DEFINE_TYPE (GstWasapiDeviceProvider, gst_wasapi_device_provider, GST_TYPE_DEVICE_PROVIDER); static void gst_wasapi_device_provider_finalize (GObject * object); static GList *gst_wasapi_device_provider_probe (GstDeviceProvider * provider); +static gboolean gst_wasapi_device_provider_start (GstDeviceProvider * provider); +static void gst_wasapi_device_provider_stop (GstDeviceProvider * provider); + +static HRESULT +gst_wasapi_device_provider_device_added (GstMMDeviceEnumerator * enumerator, + LPCWSTR device_id, gpointer user_data); +static HRESULT +gst_wasapi_device_provider_device_removed (GstMMDeviceEnumerator * enumerator, + LPCWSTR device_id, gpointer user_data); +static HRESULT +gst_wasapi_device_provider_default_device_changed (GstMMDeviceEnumerator * + enumerator, EDataFlow flow, ERole role, LPCWSTR device_id, + gpointer user_data); static void gst_wasapi_device_provider_class_init (GstWasapiDeviceProviderClass * klass) @@ -38,6 +55,8 @@ gst_wasapi_device_provider_class_init (GstWasapiDeviceProviderClass * klass) gobject_class->finalize = gst_wasapi_device_provider_finalize; dm_class->probe = gst_wasapi_device_provider_probe; + dm_class->start = gst_wasapi_device_provider_start; + dm_class->stop = gst_wasapi_device_provider_stop; gst_device_provider_class_set_static_metadata (dm_class, "WASAPI (Windows Audio Session API) Device Provider", @@ -46,15 +65,68 @@ gst_wasapi_device_provider_class_init (GstWasapiDeviceProviderClass * klass) } static void -gst_wasapi_device_provider_init (GstWasapiDeviceProvider * provider) +gst_wasapi_device_provider_init (GstWasapiDeviceProvider * self) { - CoInitializeEx (NULL, COINIT_MULTITHREADED); + self->enumerator = gst_mm_device_enumerator_new (); +} + +static gboolean +gst_wasapi_device_provider_start (GstDeviceProvider * provider) +{ + GstWasapiDeviceProvider *self = GST_WASAPI_DEVICE_PROVIDER (provider); + GstMMNotificationClientCallbacks callbacks = { NULL, }; + GList *devices = NULL; + GList *iter; + + if (!self->enumerator) { + GST_WARNING_OBJECT (self, "Enumerator wasn't configured"); + return FALSE; + } + + callbacks.device_added = gst_wasapi_device_provider_device_added; + callbacks.device_removed = gst_wasapi_device_provider_device_removed; + callbacks.default_device_changed = + gst_wasapi_device_provider_default_device_changed; + + if (!gst_mm_device_enumerator_set_notification_callback (self->enumerator, + &callbacks, self)) { + GST_WARNING_OBJECT (self, "Failed to set callbacks"); + return FALSE; + } + + /* baseclass will not call probe() once it's started, but we can get + * notification only add/remove or change case. To this manually */ + devices = gst_wasapi_device_provider_probe (provider); + if (devices) { + for (iter = devices; iter; iter = g_list_next (iter)) { + gst_device_provider_device_add (provider, GST_DEVICE (iter->data)); + } + + g_list_free (devices); + } + + return TRUE; +} + +static void +gst_wasapi_device_provider_stop (GstDeviceProvider * provider) +{ + GstWasapiDeviceProvider *self = GST_WASAPI_DEVICE_PROVIDER (provider); + + if (self->enumerator) { + gst_mm_device_enumerator_set_notification_callback (self->enumerator, + NULL, NULL); + } } static void gst_wasapi_device_provider_finalize (GObject * object) { - CoUninitialize (); + GstWasapiDeviceProvider *self = GST_WASAPI_DEVICE_PROVIDER (object); + + gst_clear_object (&self->enumerator); + + G_OBJECT_CLASS (gst_wasapi_device_provider_parent_class)->finalize (object); } static GList * @@ -63,12 +135,137 @@ gst_wasapi_device_provider_probe (GstDeviceProvider * provider) GstWasapiDeviceProvider *self = GST_WASAPI_DEVICE_PROVIDER (provider); GList *devices = NULL; - if (!gst_wasapi_util_get_devices (GST_OBJECT (self), TRUE, &devices)) + if (!gst_wasapi_util_get_devices (self->enumerator, TRUE, &devices)) GST_ERROR_OBJECT (self, "Failed to enumerate devices"); return devices; } +static gboolean +gst_wasapi_device_is_in_list (GList * list, GstDevice * device) +{ + GList *iter; + GstStructure *s; + const gchar *device_id; + gboolean found = FALSE; + + s = gst_device_get_properties (device); + g_assert (s); + + device_id = gst_structure_get_string (s, "device.strid"); + g_assert (device_id); + + for (iter = list; iter; iter = g_list_next (iter)) { + GstStructure *other_s; + const gchar *other_id; + + other_s = gst_device_get_properties (GST_DEVICE (iter->data)); + g_assert (other_s); + + other_id = gst_structure_get_string (other_s, "device.strid"); + g_assert (other_id); + + if (g_ascii_strcasecmp (device_id, other_id) == 0) { + found = TRUE; + } + + gst_structure_free (other_s); + if (found) + break; + } + + gst_structure_free (s); + + return found; +} + +static void +gst_wasapi_device_provider_update_devices (GstWasapiDeviceProvider * self) +{ + GstDeviceProvider *provider = GST_DEVICE_PROVIDER_CAST (self); + GList *prev_devices = NULL; + GList *new_devices = NULL; + GList *to_add = NULL; + GList *to_remove = NULL; + GList *iter; + + GST_OBJECT_LOCK (self); + prev_devices = g_list_copy_deep (provider->devices, + (GCopyFunc) gst_object_ref, NULL); + GST_OBJECT_UNLOCK (self); + + new_devices = gst_wasapi_device_provider_probe (provider); + + /* Ownership of GstDevice for gst_device_provider_device_add() + * and gst_device_provider_device_remove() is a bit complicated. + * Remove floating reference here for things to be clear */ + for (iter = new_devices; iter; iter = g_list_next (iter)) + gst_object_ref_sink (iter->data); + + /* Check newly added devices */ + for (iter = new_devices; iter; iter = g_list_next (iter)) { + if (!gst_wasapi_device_is_in_list (prev_devices, GST_DEVICE (iter->data))) { + to_add = g_list_prepend (to_add, gst_object_ref (iter->data)); + } + } + + /* Check removed device */ + for (iter = prev_devices; iter; iter = g_list_next (iter)) { + if (!gst_wasapi_device_is_in_list (new_devices, GST_DEVICE (iter->data))) { + to_remove = g_list_prepend (to_remove, gst_object_ref (iter->data)); + } + } + + for (iter = to_remove; iter; iter = g_list_next (iter)) + gst_device_provider_device_remove (provider, GST_DEVICE (iter->data)); + + for (iter = to_add; iter; iter = g_list_next (iter)) + gst_device_provider_device_add (provider, GST_DEVICE (iter->data)); + + if (prev_devices) + g_list_free_full (prev_devices, (GDestroyNotify) gst_object_unref); + + if (to_add) + g_list_free_full (to_add, (GDestroyNotify) gst_object_unref); + + if (to_remove) + g_list_free_full (to_remove, (GDestroyNotify) gst_object_unref); +} + +static HRESULT +gst_wasapi_device_provider_device_added (GstMMDeviceEnumerator * enumerator, + LPCWSTR device_id, gpointer user_data) +{ + GstWasapiDeviceProvider *self = GST_WASAPI_DEVICE_PROVIDER (user_data); + + gst_wasapi_device_provider_update_devices (self); + + return S_OK; +} + +static HRESULT +gst_wasapi_device_provider_device_removed (GstMMDeviceEnumerator * enumerator, + LPCWSTR device_id, gpointer user_data) +{ + GstWasapiDeviceProvider *self = GST_WASAPI_DEVICE_PROVIDER (user_data); + + gst_wasapi_device_provider_update_devices (self); + + return S_OK; +} + +static HRESULT +gst_wasapi_device_provider_default_device_changed (GstMMDeviceEnumerator * + enumerator, EDataFlow flow, ERole role, LPCWSTR device_id, + gpointer user_data) +{ + GstWasapiDeviceProvider *self = GST_WASAPI_DEVICE_PROVIDER (user_data); + + gst_wasapi_device_provider_update_devices (self); + + return S_OK; +} + /* GstWasapiDevice begins */ enum |