Commit a93b4769 authored by George Kiagiadakis's avatar George Kiagiadakis

modules: re-implement the simple policy module using WpPolicy

parent 25d70c54
......@@ -41,11 +41,11 @@ shared_library(
)
shared_library(
'wireplumber-module-pw-simple-policy',
'wireplumber-module-simple-policy',
[
'module-pw-simple-policy.c'
'module-simple-policy.c'
],
c_args : [common_c_args, '-DG_LOG_DOMAIN="m-pw-simple-policy"'],
c_args : [common_c_args, '-DG_LOG_DOMAIN="m-simple-policy"'],
install : true,
install_dir : wireplumber_module_dir,
dependencies : [wp_dep, pipewire_dep],
......
/* WirePlumber
*
* Copyright © 2019 Collabora Ltd.
* @author Julian Bouzas <julian.bouzas@collabora.com>
*
* SPDX-License-Identifier: MIT
*/
/**
* module-pw-simple-policy connects the first audio output client endpoint with
* the first audio sink remote endpoint
*/
#include <wp/wp.h>
#include <pipewire/pipewire.h>
typedef void (*WpDoneCallback)(gpointer);
struct impl {
WpCore *wp_core;
/* Remote */
struct pw_remote *remote;
struct spa_hook remote_listener;
/* Core */
struct pw_core_proxy *core_proxy;
struct spa_hook core_listener;
int core_seq;
WpDoneCallback done_cb;
gpointer done_cb_data;
/* Endpoints */
WpEndpoint *ep_client;
WpEndpoint *ep_remote;
};
static void
sync_core_with_callabck(struct impl* impl, WpDoneCallback callback,
gpointer data)
{
/* Set the callback and data */
impl->done_cb = callback;
impl->done_cb_data = data;
/* Sync the core */
impl->core_seq = pw_core_proxy_sync(impl->core_proxy, 0, impl->core_seq);
}
static WpEndpoint *
endpoint_get_selected (WpCore *core, const char *media_class)
{
g_autoptr (GPtrArray) ptr_array = NULL;
int i;
/* Get all the endpoints with the specific media lcass*/
ptr_array = wp_endpoint_find (core, media_class);
if (!ptr_array)
return NULL;
/* Find and return the "selected" endpoint */
/* FIXME: fix the endpoint API, this is terrible */
for (i = 0; i < ptr_array->len; i++) {
WpEndpoint *ep = g_ptr_array_index (ptr_array, i);
GVariantIter iter;
g_autoptr (GVariant) controls = NULL;
g_autoptr (GVariant) value = NULL;
const gchar *name;
guint id;
controls = wp_endpoint_list_controls (ep);
g_variant_iter_init (&iter, controls);
while ((value = g_variant_iter_next_value (&iter))) {
if (!g_variant_lookup (value, "name", "&s", &name)
|| !g_str_equal (name, "selected")) {
g_variant_unref (value);
continue;
}
g_variant_lookup (value, "id", "u", &id);
g_variant_unref (value);
}
value = wp_endpoint_get_control_value (ep, id);
if (value && g_variant_get_boolean (value))
return ep;
}
/* If not found, return the first endpoint */
return (ptr_array->len > 1) ? g_ptr_array_index (ptr_array, 0) : NULL;
}
static void
link_endpoints(gpointer data)
{
struct impl *impl = data;
WpEndpointLink *ep_link = NULL;
/* Make sure the endpoints are valid */
if (!impl->ep_client || !impl->ep_remote) {
g_warning ("Endpoints not valid to link. Skipping...\n");
return;
}
/* Link the client with the remote */
ep_link = wp_endpoint_link_new(impl->wp_core, impl->ep_client, 0,
impl->ep_remote, 0, NULL);
if (!ep_link) {
g_warning ("Could not link endpoints. Skipping...\n");
return;
}
g_info ("Endpoints linked successfully\n");
}
static void
endpoint_added (WpCore *core, GQuark key, WpEndpoint *ep, struct impl * impl)
{
const char *media_class = NULL;
/* Reset endpoints */
impl->ep_remote = NULL;
impl->ep_client = NULL;
/* Make sure an endpoint has been added */
g_return_if_fail (key == WP_GLOBAL_ENDPOINT);
/* Get the media class */
media_class = wp_endpoint_get_media_class(ep);
/* Only process client endpoints */
if (!g_str_has_prefix(media_class, "Stream"))
return;
/* TODO: For now we only accept audio output clients */
if (!g_str_has_prefix(media_class, "Stream/Output/Audio"))
return;
impl->ep_client = ep;
/* Get the first endpoint with media class Audio/Sink */
impl->ep_remote = endpoint_get_selected (core, "Audio/Sink");
if (!impl->ep_remote) {
g_warning ("Could not get an Audio/Sink remote endpoint\n");
return;
}
/* Do the linking when core is done */
sync_core_with_callabck (impl, link_endpoints, impl);
}
static void core_done(void *data, uint32_t id, int seq)
{
struct impl * impl = data;
/* Call the done callback if it exists */
if (impl->done_cb)
impl->done_cb(impl->done_cb_data);
impl->done_cb = NULL;
impl->done_cb_data = NULL;
}
static const struct pw_core_proxy_events core_events = {
PW_VERSION_CORE_EVENTS,
.done = core_done
};
static void on_state_changed(void *_data, enum pw_remote_state old,
enum pw_remote_state state, const char *error)
{
struct impl *impl = _data;
switch (state) {
case PW_REMOTE_STATE_ERROR:
break;
case PW_REMOTE_STATE_CONNECTED:
/* Register the core event callbacks */
impl->core_proxy = pw_remote_get_core_proxy(impl->remote);
pw_core_proxy_add_listener(impl->core_proxy, &impl->core_listener,
&core_events, impl);
break;
case PW_REMOTE_STATE_UNCONNECTED:
break;
default:
break;
}
}
static const struct pw_remote_events remote_events = {
PW_VERSION_REMOTE_EVENTS,
.state_changed = on_state_changed,
};
static void
module_destroy (gpointer data)
{
struct impl *impl = data;
g_slice_free (struct impl, impl);
}
void
wireplumber__module_init (WpModule * module, WpCore * core, GVariant * args)
{
struct impl *impl = g_new0(struct impl, 1);
/* Set destroy callback for impl */
wp_module_set_destroy_callback (module, module_destroy, impl);
/* Set the core */
impl->wp_core = core;
/* Set the core remote */
impl->remote = wp_core_get_global(core, WP_GLOBAL_PW_REMOTE);
/* Add a state changed listener */
pw_remote_add_listener(impl->remote, &impl->remote_listener, &remote_events,
impl);
/* Register the endpoint added and removed callbacks */
g_signal_connect (core, "global-added::endpoint",
(GCallback) endpoint_added, impl);
}
/* WirePlumber
*
* Copyright © 2019 Collabora Ltd.
* @author George Kiagiadakis <george.kiagiadakis@collabora.com>
*
* SPDX-License-Identifier: MIT
*/
#include <wp/wp.h>
struct _WpSimplePolicy
{
WpPolicy parent;
};
G_DECLARE_FINAL_TYPE (WpSimplePolicy, simple_policy, WP, SIMPLE_POLICY, WpPolicy)
G_DEFINE_TYPE (WpSimplePolicy, simple_policy, WP_TYPE_POLICY)
static void
simple_policy_init (WpSimplePolicy *self)
{
}
static gboolean
simple_policy_handle_endpoint (WpPolicy *policy, WpEndpoint *ep)
{
const char *media_class = NULL;
GVariantDict d;
g_autoptr (GVariant) props = NULL;
g_autoptr (WpCore) core = NULL;
g_autoptr (WpEndpoint) target = NULL;
g_autoptr (GError) error = NULL;
guint32 stream_id;
/* TODO: For now we only accept audio output clients */
media_class = wp_endpoint_get_media_class(ep);
if (!g_str_equal (media_class, "Stream/Output/Audio"))
return FALSE;
/* Locate the target endpoint */
g_variant_dict_init (&d, NULL);
g_variant_dict_insert (&d, "action", "s", "link");
g_variant_dict_insert (&d, "media.class", "s", "Audio/Sink");
/* TODO: more properties are needed here */
props = g_variant_dict_end (&d);
core = wp_policy_get_core (policy);
target = wp_policy_find_endpoint (core, props, &stream_id);
if (!target) {
g_warning ("Could not find an Audio/Sink target endpoint\n");
/* TODO: we should kill the client, otherwise it's going to hang waiting */
return FALSE;
}
/* Link the client with the target */
if (!wp_endpoint_link_new (core, ep, 0, target, stream_id, &error)) {
g_warning ("Could not link endpoints: %s\n", error->message);
} else {
g_info ("Sucessfully linked '%s' to '%s'\n", wp_endpoint_get_name (ep),
wp_endpoint_get_name (target));
}
return TRUE;
}
static WpEndpoint *
simple_policy_find_endpoint (WpPolicy *policy, GVariant *props,
guint32 *stream_id)
{
g_autoptr (WpCore) core = NULL;
g_autoptr (GPtrArray) ptr_array = NULL;
const char *media_class = NULL;
WpEndpoint *ep;
int i;
core = wp_policy_get_core (policy);
/* Get all the endpoints with the specific media class*/
g_variant_lookup (props, "media.class", "&s", &media_class);
ptr_array = wp_endpoint_find (core, media_class);
if (!ptr_array)
return NULL;
/* TODO: for now we statically return the first stream
* we should be looking into the media.role eventually */
*stream_id = 0;
/* Find and return the "selected" endpoint */
/* FIXME: fix the endpoint API, this is terrible */
for (i = 0; i < ptr_array->len; i++) {
ep = g_ptr_array_index (ptr_array, i);
GVariantIter iter;
g_autoptr (GVariant) controls = NULL;
g_autoptr (GVariant) value = NULL;
const gchar *name;
guint id;
controls = wp_endpoint_list_controls (ep);
g_variant_iter_init (&iter, controls);
while ((value = g_variant_iter_next_value (&iter))) {
if (!g_variant_lookup (value, "name", "&s", &name)
|| !g_str_equal (name, "selected")) {
g_variant_unref (value);
continue;
}
g_variant_lookup (value, "id", "u", &id);
g_variant_unref (value);
}
value = wp_endpoint_get_control_value (ep, id);
if (value && g_variant_get_boolean (value))
return g_object_ref (ep);
}
/* If not found, return the first endpoint */
ep = (ptr_array->len > 1) ? g_ptr_array_index (ptr_array, 0) : NULL;
return g_object_ref (ep);
}
static void
simple_policy_class_init (WpSimplePolicyClass *klass)
{
WpPolicyClass *policy_class = (WpPolicyClass *) klass;
policy_class->handle_endpoint = simple_policy_handle_endpoint;
policy_class->find_endpoint = simple_policy_find_endpoint;
}
void
wireplumber__module_init (WpModule * module, WpCore * core, GVariant * args)
{
WpPolicy *p = g_object_new (simple_policy_get_type (),
"rank", WP_POLICY_RANK_UPSTREAM,
NULL);
wp_policy_register (p, core);
}
load-module C libwireplumber-module-pipewire
load-module C libwireplumber-module-pw-audio-softdsp-endpoint
load-module C libwireplumber-module-pw-alsa-udev
load-module C libwireplumber-module-pw-simple-policy
load-module C libwireplumber-module-simple-policy
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment