pipewire/pinos/modules/spa/spa-v4l2-monitor.c

/* Pinos
 * Copyright (C) 2015 Wim Taymans <wim.taymans@gmail.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.
 */

#include <dlfcn.h>
#include <errno.h>
#include <poll.h>
#include <string.h>
#include <fcntl.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/mman.h>

#include <gio/gio.h>

#include <spa/include/spa/node.h>
#include <spa/include/spa/monitor.h>

#include "spa-v4l2-monitor.h"

#define PINOS_SPA_V4L2_MONITOR_GET_PRIVATE(obj)  \
     (G_TYPE_INSTANCE_GET_PRIVATE ((obj), PINOS_TYPE_SPA_V4L2_MONITOR, PinosSpaV4l2MonitorPrivate))

typedef struct {
  uint32_t node;
  uint32_t clock;
} URI;

struct _PinosSpaV4l2MonitorPrivate
{
  PinosDaemon *daemon;

  SpaHandle *handle;
  SpaMonitor *monitor;

  URI uri;

  GHashTable *nodes;
};

enum
{
  PROP_0,
  PROP_DAEMON,
  PROP_MONITOR,
};

G_DEFINE_TYPE (PinosSpaV4l2Monitor, pinos_spa_v4l2_monitor, G_TYPE_OBJECT);

static SpaResult
make_handle (PinosDaemon *daemon, SpaHandle **handle, const char *lib, const char *name, const SpaDict *info)
{
  SpaResult res;
  void *hnd, *state = NULL;
  SpaEnumHandleFactoryFunc enum_func;

  if ((hnd = dlopen (lib, RTLD_NOW)) == NULL) {
    g_error ("can't load %s: %s", lib, dlerror());
    return SPA_RESULT_ERROR;
  }
  if ((enum_func = dlsym (hnd, "spa_enum_handle_factory")) == NULL) {
    g_error ("can't find enum function");
    return SPA_RESULT_ERROR;
  }

  while (true) {
    const SpaHandleFactory *factory;

    if ((res = enum_func (&factory, &state)) < 0) {
      if (res != SPA_RESULT_ENUM_END)
        g_error ("can't enumerate factories: %d", res);
      break;
    }
    if (strcmp (factory->name, name))
      continue;

    *handle = g_malloc0 (factory->size);
    if ((res = spa_handle_factory_init (factory, *handle, info, daemon->support, daemon->n_support)) < 0) {
      g_error ("can't make factory instance: %d", res);
      return res;
    }
    return SPA_RESULT_OK;
  }
  return SPA_RESULT_ERROR;
}

static void
add_item (PinosSpaV4l2Monitor *this, SpaMonitorItem *item)
{
  PinosSpaV4l2MonitorPrivate *priv = this->priv;
  SpaResult res;
  SpaHandle *handle;
  PinosNode *node;
  void *node_iface;
  void *clock_iface;
  PinosProperties *props = NULL;

  g_debug ("v4l2-monitor %p: add: \"%s\" (%s)", this, item->name, item->id);

  handle = calloc (1, item->factory->size);
  if ((res = spa_handle_factory_init (item->factory,
                                      handle,
                                      item->info,
                                      priv->daemon->support,
                                      priv->daemon->n_support)) < 0) {
    g_error ("can't make factory instance: %d", res);
    return;
  }
  if ((res = spa_handle_get_interface (handle, priv->uri.node, &node_iface)) < 0) {
    g_error ("can't get NODE interface: %d", res);
    return;
  }
  if ((res = spa_handle_get_interface (handle, priv->uri.clock, &clock_iface)) < 0) {
    g_error ("can't get CLOCK interface: %d", res);
    return;
  }

  if (item->info) {
    unsigned int i;

    props = pinos_properties_new (NULL, NULL);

    for (i = 0; i < item->info->n_items; i++)
      pinos_properties_set (props,
                            item->info->items[i].key,
                            item->info->items[i].value);
  }

  node = g_object_new (PINOS_TYPE_NODE,
                       "daemon", priv->daemon,
                       "name", item->factory->name,
                       "node", node_iface,
                       "clock", clock_iface,
                       "properties", props,
                       NULL);

  g_hash_table_insert (priv->nodes, g_strdup (item->id), node);
}

static void
remove_item (PinosSpaV4l2Monitor *this, SpaMonitorItem *item)
{
  PinosSpaV4l2MonitorPrivate *priv = this->priv;
  PinosNode *node;

  g_debug ("v4l2-monitor %p: remove: \"%s\" (%s)", this, item->name, item->id);

  node = g_hash_table_lookup (priv->nodes, item->id);
  if (node) {
    pinos_node_remove (node);
    g_hash_table_remove (priv->nodes, item->id);
  }
}

static void
on_monitor_event  (SpaMonitor      *monitor,
                   SpaMonitorEvent *event,
                   void            *user_data)
{
  PinosSpaV4l2Monitor *this = user_data;

  switch (event->type) {
    case SPA_MONITOR_EVENT_TYPE_ADDED:
    {
      SpaMonitorItem *item = event->data;
      add_item (this, item);
      break;
    }
    case SPA_MONITOR_EVENT_TYPE_REMOVED:
    {
      SpaMonitorItem *item = event->data;
      remove_item (this, item);
    }
    case SPA_MONITOR_EVENT_TYPE_CHANGED:
    {
      SpaMonitorItem *item = event->data;
      g_debug ("v4l2-monitor %p: changed: \"%s\"", this, item->name);
      break;
    }
    default:
      break;
  }
}

static void
monitor_constructed (GObject * object)
{
  PinosSpaV4l2Monitor *this = PINOS_SPA_V4L2_MONITOR (object);
  PinosSpaV4l2MonitorPrivate *priv = this->priv;
  SpaResult res;
  void *state = NULL;

  g_debug ("spa-monitor %p: constructed", this);

  G_OBJECT_CLASS (pinos_spa_v4l2_monitor_parent_class)->constructed (object);

  priv->uri.node = spa_id_map_get_id (priv->daemon->map, SPA_NODE_URI);
  priv->uri.clock = spa_id_map_get_id (priv->daemon->map, SPA_CLOCK_URI);

  while (TRUE) {
    SpaMonitorItem *item;

    if ((res = spa_monitor_enum_items (priv->monitor, &item, &state)) < 0) {
      if (res != SPA_RESULT_ENUM_END)
        g_debug ("spa_monitor_enum_items: got error %d\n", res);
      break;
    }
    add_item (this, item);
  }
  spa_monitor_set_event_callback (priv->monitor, on_monitor_event, this);
}

static void
monitor_get_property (GObject    *_object,
                      guint       prop_id,
                      GValue     *value,
                      GParamSpec *pspec)
{
  PinosSpaV4l2Monitor *this = PINOS_SPA_V4L2_MONITOR (_object);
  PinosSpaV4l2MonitorPrivate *priv = this->priv;

  switch (prop_id) {
    case PROP_DAEMON:
      g_value_set_object (value, priv->daemon);
      break;

    case PROP_MONITOR:
      g_value_set_pointer (value, priv->monitor);
      break;

    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (this, prop_id, pspec);
      break;
  }
}

static void
monitor_set_property (GObject      *_object,
                      guint         prop_id,
                      const GValue *value,
                      GParamSpec   *pspec)
{
  PinosSpaV4l2Monitor *this = PINOS_SPA_V4L2_MONITOR (_object);
  PinosSpaV4l2MonitorPrivate *priv = this->priv;

  switch (prop_id) {
    case PROP_DAEMON:
      priv->daemon = g_value_dup_object (value);
      break;

    case PROP_MONITOR:
      priv->monitor = g_value_get_pointer (value);
      break;

    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (this, prop_id, pspec);
      break;
  }
}

static void
monitor_finalize (GObject * object)
{
  PinosSpaV4l2Monitor *this = PINOS_SPA_V4L2_MONITOR (object);
  PinosSpaV4l2MonitorPrivate *priv = this->priv;

  g_debug ("spa-monitor %p: dispose", this);
  spa_handle_clear (priv->handle);
  g_free (priv->handle);
  g_hash_table_unref (priv->nodes);

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

static void
pinos_spa_v4l2_monitor_class_init (PinosSpaV4l2MonitorClass * klass)
{
  GObjectClass *gobject_class = G_OBJECT_CLASS (klass);

  g_type_class_add_private (klass, sizeof (PinosSpaV4l2MonitorPrivate));

  gobject_class->constructed = monitor_constructed;
  gobject_class->finalize = monitor_finalize;
  gobject_class->set_property = monitor_set_property;
  gobject_class->get_property = monitor_get_property;

  g_object_class_install_property (gobject_class,
                                   PROP_DAEMON,
                                   g_param_spec_object ("daemon",
                                                        "Daemon",
                                                        "The Daemon",
                                                        PINOS_TYPE_DAEMON,
                                                        G_PARAM_READWRITE |
                                                        G_PARAM_CONSTRUCT_ONLY |
                                                        G_PARAM_STATIC_STRINGS));
  g_object_class_install_property (gobject_class,
                                   PROP_MONITOR,
                                   g_param_spec_pointer ("monitor",
                                                         "Monitor",
                                                         "The SPA monitor",
                                                         G_PARAM_READWRITE |
                                                         G_PARAM_CONSTRUCT_ONLY |
                                                         G_PARAM_STATIC_STRINGS));
}

static void
pinos_spa_v4l2_monitor_init (PinosSpaV4l2Monitor * this)
{
  PinosSpaV4l2MonitorPrivate *priv = this->priv = PINOS_SPA_V4L2_MONITOR_GET_PRIVATE (this);

  priv->nodes = g_hash_table_new_full (g_str_hash,
                                       g_str_equal,
                                       g_free,
                                       g_object_unref);
}

GObject *
pinos_spa_v4l2_monitor_new (PinosDaemon *daemon)
{
  GObject *monitor;
  SpaHandle *handle;
  SpaResult res;
  void *iface;

  if ((res = make_handle (daemon,
                          &handle,
                          "build/spa/plugins/v4l2/libspa-v4l2.so",
                          "v4l2-monitor",
                          NULL)) < 0) {
    g_error ("can't create v4l2-monitor: %d", res);
    return NULL;
  }

  if ((res = spa_handle_get_interface (handle,
                                       spa_id_map_get_id (daemon->map, SPA_MONITOR_URI),
                                       &iface)) < 0) {
    g_free (handle);
    g_error ("can't get MONITOR interface: %d", res);
    return NULL;
  }

  monitor = g_object_new (PINOS_TYPE_SPA_V4L2_MONITOR,
                          "daemon", daemon,
                          "monitor", iface,
                          NULL);
  return monitor;
}