/* -*- Mode: C; c-basic-offset: 4 -*- */
/*
    Copyright (C) 2002, 2003 Andy Wingo <wingo@pobox.com>

    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU 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
    General Public License for more details.

    You should have received a copy of the GNU General Public
    License along with this library; if not, write to the Free
    Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <stdlib.h>
#include <string.h>
#include "gstjack.h"
#include <gst/audio/audio.h>

/* TODO:

   - work out the src side (caps setting, etc)

   future core TODO:
   - make a jack clock provider
   - add GST_ELEMENT_FIXED_DATA_RATE, GST_ELEMENT_QOS,
     GST_ELEMENT_CHANGES_DATA_RATE element flags, and make the scheduler
     sensitive to them
*/

/* elementfactory information */
static GstElementDetails gst_jack_bin_details = {
  "Jack Bin",
  "Generic/Bin",
  "Jack processing bin",
  "Andy Wingo <wingo@pobox.com>",
};

static GstElementDetails gst_jack_sink_details = {
  "Jack Sink",
  "Sink/Audio",
  "Output to a Jack processing network",
  "Andy Wingo <wingo@pobox.com>",
};

static GstElementDetails gst_jack_src_details = {
  "Jack Src",
  "Source/Audio",
  "Input from a Jack processing network",
  "Andy Wingo <wingo@pobox.com>",
};


static GHashTable *port_name_counts = NULL;
static GstElementClass *parent_class = NULL;

static void gst_jack_base_init (gpointer g_class);
static void gst_jack_src_base_init (gpointer g_class);
static void gst_jack_sink_base_init (gpointer g_class);
static void gst_jack_init (GstJack * this);
static void gst_jack_class_init (GstJackClass * klass);
static void gst_jack_set_property (GObject * object, guint prop_id,
    const GValue * value, GParamSpec * pspec);
static void gst_jack_get_property (GObject * object, guint prop_id,
    GValue * value, GParamSpec * pspec);

static GstPadTemplate *gst_jack_src_request_pad_factory ();
static GstPadTemplate *gst_jack_sink_request_pad_factory ();
static GstPad *gst_jack_request_new_pad (GstElement * element,
    GstPadTemplate * templ, const gchar * name);
static GstStateChangeReturn gst_jack_change_state (GstElement * element,
    GstStateChange transition);
static GstPadLinkReturn gst_jack_link (GstPad * pad, const GstCaps * caps);

static void gst_jack_loop (GstElement * element);


enum
{
  ARG_0,
  ARG_PORT_NAME_PREFIX
};


GType
gst_jack_get_type (void)
{
  static GType jack_type = 0;

  if (!jack_type) {
    static const GTypeInfo jack_info = {
      sizeof (GstJackClass),
      gst_jack_base_init,
      NULL,
      NULL,
      NULL,
      NULL,
      sizeof (GstJack),
      0,
      NULL,
    };

    jack_type =
        g_type_register_static (GST_TYPE_ELEMENT, "GstJack", &jack_info, 0);
  }
  return jack_type;
}

GType
gst_jack_sink_get_type (void)
{
  static GType jack_type = 0;

  if (!jack_type) {
    static const GTypeInfo jack_info = {
      sizeof (GstJackClass),
      gst_jack_sink_base_init,
      NULL,
      (GClassInitFunc) gst_jack_class_init,
      NULL,
      NULL,
      sizeof (GstJack),
      0,
      (GInstanceInitFunc) gst_jack_init,
    };

    jack_type =
        g_type_register_static (GST_TYPE_JACK, "GstJackSink", &jack_info, 0);
  }
  return jack_type;
}

GType
gst_jack_src_get_type (void)
{
  static GType jack_type = 0;

  if (!jack_type) {
    static const GTypeInfo jack_info = {
      sizeof (GstJackClass),
      gst_jack_src_base_init,
      NULL,
      (GClassInitFunc) gst_jack_class_init,
      NULL,
      NULL,
      sizeof (GstJack),
      0,
      (GInstanceInitFunc) gst_jack_init,
    };

    jack_type =
        g_type_register_static (GST_TYPE_JACK, "GstJackSrc", &jack_info, 0);
  }
  return jack_type;
}

static void
gst_jack_base_init (gpointer g_class)
{
  GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);

  gst_element_class_set_details (element_class, &gst_jack_bin_details);
}

static void
gst_jack_src_base_init (gpointer g_class)
{
  GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);

  gst_element_class_add_pad_template (element_class,
      gst_jack_src_request_pad_factory ());
  gst_element_class_set_details (element_class, &gst_jack_src_details);
}

static void
gst_jack_sink_base_init (gpointer g_class)
{
  GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);

  gst_element_class_add_pad_template (element_class,
      gst_jack_sink_request_pad_factory ());
  gst_element_class_set_details (element_class, &gst_jack_sink_details);
}

static void
gst_jack_class_init (GstJackClass * klass)
{
  GObjectClass *object_class;
  GstElementClass *element_class;
  GParamSpec *pspec;
  gchar *prefix;

  object_class = (GObjectClass *) klass;
  element_class = (GstElementClass *) klass;

  if (parent_class == NULL)
    parent_class = g_type_class_peek_parent (klass);

  object_class->get_property = gst_jack_get_property;
  object_class->set_property = gst_jack_set_property;

  if (GST_IS_JACK_SINK_CLASS (klass))
    prefix = "gst-out-";
  else
    prefix = "gst-in-";

  pspec = g_param_spec_string ("port-name-prefix", "Port name prefix",
      "String to prepend to jack port names",
      prefix, G_PARAM_READWRITE | G_PARAM_CONSTRUCT);
  g_object_class_install_property (object_class, ARG_PORT_NAME_PREFIX, pspec);

  element_class->change_state = gst_jack_change_state;

  element_class->request_new_pad = gst_jack_request_new_pad;
}

static void
gst_jack_init (GstJack * this)
{
  if (G_OBJECT_TYPE (this) == GST_TYPE_JACK_SRC)
    this->direction = GST_PAD_SRC;
  else if (G_OBJECT_TYPE (this) == GST_TYPE_JACK_SINK)
    this->direction = GST_PAD_SINK;
  else
    g_assert_not_reached ();

  gst_element_set_loop_function (GST_ELEMENT (this), gst_jack_loop);
}

static void
gst_jack_set_property (GObject * object, guint prop_id, const GValue * value,
    GParamSpec * pspec)
{
  GstJack *this = (GstJack *) object;

  switch (prop_id) {
    case ARG_PORT_NAME_PREFIX:
      if (this->port_name_prefix)
        g_free (this->port_name_prefix);
      this->port_name_prefix = g_strdup (g_value_get_string (value));
      break;
    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
      return;
  }
}

static void
gst_jack_get_property (GObject * object, guint prop_id, GValue * value,
    GParamSpec * pspec)
{
  GstJack *this = (GstJack *) object;

  switch (prop_id) {
    case ARG_PORT_NAME_PREFIX:
      g_value_set_string (value, this->port_name_prefix);
      break;
    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
      break;
  }
}

static GstPadTemplate *
gst_jack_src_request_pad_factory (void)
{
  static GstPadTemplate *template = NULL;

  if (!template) {
    GstCaps *caps;

    caps = gst_caps_from_string (GST_AUDIO_FLOAT_STANDARD_PAD_TEMPLATE_CAPS);
    template = gst_pad_template_new ("%s", GST_PAD_SRC, GST_PAD_REQUEST, caps);
  }

  return template;
}

static GstPadTemplate *
gst_jack_sink_request_pad_factory (void)
{
  static GstPadTemplate *template = NULL;

  if (!template) {
    GstCaps *caps;

    caps = gst_caps_from_string (GST_AUDIO_FLOAT_STANDARD_PAD_TEMPLATE_CAPS);
    template = gst_pad_template_new ("%s", GST_PAD_SINK, GST_PAD_REQUEST, caps);
  }

  return template;
}

static GstPad *
gst_jack_request_new_pad (GstElement * element, GstPadTemplate * templ,
    const gchar * name)
{
  GstJack *this;
  gchar *newname;
  GList *l, **pad_list;
  GstJackPad *pad;
  gint count;

  g_return_val_if_fail (GST_IS_JACK (element), NULL);
  this = GST_JACK (element);

  if (!this->bin)
    pad_list = &this->pads;
  else if (this->direction == GST_PAD_SRC)
    pad_list = &this->bin->src_pads;
  else
    pad_list = &this->bin->sink_pads;

  if (name) {
    l = *pad_list;
    while (l) {
      if (strcmp (GST_JACK_PAD (l)->name, name) == 0) {
        g_warning ("requested port name %s already in use.", name);
        return NULL;
      }
      l = l->next;
    }
    newname = g_strdup (name);
  } else {
    if (this->direction == GST_PAD_SINK)
      newname = g_strdup ("alsa_pcm:playback_1");
    else
      newname = g_strdup ("alsa_pcm:capture_1");
  }

  pad = g_new0 (GstJackPad, 1);

  if (!port_name_counts)
    port_name_counts = g_hash_table_new (g_str_hash, g_str_equal);

  count =
      GPOINTER_TO_INT (g_hash_table_lookup (port_name_counts,
          this->port_name_prefix));
  g_hash_table_insert (port_name_counts, g_strdup (this->port_name_prefix),
      GINT_TO_POINTER (count + 1));

  pad->name = g_strdup_printf ("%s%d", this->port_name_prefix, count);

  pad->peer_name = newname;
  pad->pad = gst_pad_new_from_template (templ, newname);
  gst_element_add_pad (GST_ELEMENT (this), pad->pad);
  gst_pad_set_link_function (pad->pad, gst_jack_link);

  this->pads = g_list_append (this->pads, pad);

  g_print ("returning from request_new_pad, pad %s created, to connect to %s\n",
      pad->name, pad->peer_name);
  return pad->pad;
}

static GstStateChangeReturn
gst_jack_change_state (GstElement * element, GstStateChange transition)
{
  GstJack *this;
  GList *l = NULL, **pads;
  GstJackPad *pad;
  GstCaps *caps;

  g_return_val_if_fail (element != NULL, FALSE);
  this = GST_JACK (element);

  switch (GST_STATE_PENDING (element)) {
    case GST_STATE_NULL:
      JACK_DEBUG ("%s: NULL", GST_OBJECT_NAME (GST_OBJECT (this)));

      break;

    case GST_STATE_READY:
      JACK_DEBUG ("%s: READY", GST_OBJECT_NAME (GST_OBJECT (this)));

      if (!this->bin) {
        if (!(this->bin = (GstJackBin *) gst_element_get_managing_bin (element))
            || !GST_IS_JACK_BIN (this->bin)) {
          this->bin = NULL;
          g_warning ("jack element %s needs to be contained in a jack bin.",
              GST_OBJECT_NAME (element));
          return GST_STATE_CHANGE_FAILURE;
        }

        /* fixme: verify that all names are unique */
        l = this->pads;
        pads =
            (this->direction ==
            GST_PAD_SRC) ? &this->bin->src_pads : &this->bin->sink_pads;
        while (l) {
          pad = GST_JACK_PAD (l);
          JACK_DEBUG ("%s: appending pad %s:%s to list", GST_OBJECT_NAME (this),
              pad->name, pad->peer_name);
          *pads = g_list_append (*pads, pad);
          l = g_list_next (l);
        }
      }
      break;

    case GST_STATE_PAUSED:
      JACK_DEBUG ("%s: PAUSED", GST_OBJECT_NAME (GST_OBJECT (this)));

      if (GST_STATE (element) == GST_STATE_READY) {
        /* we're in READY->PAUSED */
        l = this->pads;
        while (l) {
          pad = GST_JACK_PAD (l);
          caps = gst_caps_copy (gst_pad_get_negotiated_caps (pad->pad));
          gst_caps_set_simple (caps,
              "rate", G_TYPE_INT, (int) this->bin->rate,
              "buffer-frames", G_TYPE_INT, (gint) this->bin->nframes, NULL);
          if (gst_pad_try_set_caps (pad->pad, caps) <= 0)
            return GST_STATE_CHANGE_FAILURE;
          l = g_list_next (l);
        }
      }
      break;
    case GST_STATE_PLAYING:
      JACK_DEBUG ("%s: PLAYING", GST_OBJECT_NAME (GST_OBJECT (this)));
      break;
  }

  JACK_DEBUG ("%s: state change finished", GST_OBJECT_NAME (this));

  if (GST_ELEMENT_CLASS (parent_class)->change_state)
    return GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);

  return GST_STATE_CHANGE_SUCCESS;
}

static GstPadLinkReturn
gst_jack_link (GstPad * pad, const GstCaps * caps)
{
  GstJack *this;
  gint rate, buffer_frames;
  GstStructure *structure;

  this = GST_JACK (GST_OBJECT_PARENT (pad));

  structure = gst_caps_get_structure (caps, 0);
  gst_structure_get_int (structure, "rate", &rate);
  gst_structure_get_int (structure, "buffer-frames", &buffer_frames);
  if (this->bin && (rate != this->bin->rate ||
          buffer_frames != this->bin->nframes))
    return GST_PAD_LINK_REFUSED;

  return GST_PAD_LINK_OK;
}

static void
gst_jack_loop (GstElement * element)
{
  GstJack *this;
  GList *pads;
  gint len;
  GstJackPad *pad;
  GstBuffer *buffer;

  this = GST_JACK (element);

  len = this->bin->nframes * sizeof (sample_t);

  pads = this->pads;
  while (pads) {
    pad = GST_JACK_PAD (pads);

    if (this->direction == GST_PAD_SINK) {
      buffer = GST_BUFFER (gst_pad_pull (pad->pad));

      if (GST_IS_EVENT (buffer)) {
        GstEvent *event = GST_EVENT (buffer);

        switch (GST_EVENT_TYPE (buffer)) {
          case GST_EVENT_EOS:
            gst_element_set_eos (element);
            gst_event_unref (event);
            return;
          default:
            gst_pad_event_default (pad->pad, event);
            return;
        }
      }

      /* if the other plugins only give out buffer-frames or less (as
         they should), if the length of the GstBuffer is different
         from nframes then the buffer is short and we will get EOS
         next */
      memcpy (pad->data, GST_BUFFER_DATA (buffer), GST_BUFFER_SIZE (buffer));
      if (len != GST_BUFFER_SIZE (buffer))
        memset (pad->data + GST_BUFFER_SIZE (buffer), 0,
            len - GST_BUFFER_SIZE (buffer));

      gst_buffer_unref (buffer);
    } else {
      buffer = gst_buffer_new ();
      gst_buffer_set_data (buffer, pad->data, len);
      GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_DONTFREE);

      gst_pad_push (pad->pad, GST_DATA (buffer));
    }
    pads = g_list_next (pads);
  }
}

static gboolean
plugin_init (GstPlugin * plugin)
{
  if (!gst_element_register (plugin, "jackbin", GST_RANK_NONE,
          GST_TYPE_JACK_BIN))
    return FALSE;

  if (!gst_element_register (plugin, "jacksrc", GST_RANK_NONE,
          GST_TYPE_JACK_SRC))
    return FALSE;

  if (!gst_element_register (plugin, "jacksink", GST_RANK_NONE,
          GST_TYPE_JACK_SINK))
    return FALSE;

  return TRUE;
}

GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
    GST_VERSION_MINOR,
    "jack",
    "Jack Plugin Library", plugin_init, VERSION, "GPL", GST_PACKAGE_NAME,
    GST_PACKAGE_ORIGIN)