/* GStreamer

 * Copyright (C) 2003 Julien Moutte <julien@moutte.net>

 * Copyright (C) 2005 Ronald S. Bultje <rbultje@ronald.bitfreak.net>

 * Copyright (C) 2005 Jan Schmidt <thaytan@mad.scientist.com>

 * Copyright (C) 2007 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., 59 Temple Place - Suite 330,

 * Boston, MA 02111-1307, USA.

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include <string.h>

#include "gststreamselector.h"

GST_DEBUG_CATEGORY_STATIC (stream_selector_debug);

#define GST_CAT_DEFAULT stream_selector_debug

static const GstElementDetails gst_stream_selector_details =

GST_ELEMENT_DETAILS ("StreamSelector",

    "Generic",

    "N-to-1 input stream_selectoring",

    "Julien Moutte <julien@moutte.net>\n"

    "Ronald S. Bultje <rbultje@ronald.bitfreak.net>\n"

    "Jan Schmidt <thaytan@mad.scientist.com>\n"

    "Wim Taymans <wim.taymans@gmail.com>");

static GstStaticPadTemplate gst_stream_selector_sink_factory =

GST_STATIC_PAD_TEMPLATE ("sink%d",

    GST_PAD_SINK,

    GST_PAD_REQUEST,

    GST_STATIC_CAPS_ANY);

static GstStaticPadTemplate gst_stream_selector_src_factory =

GST_STATIC_PAD_TEMPLATE ("src",

    GST_PAD_SRC,

    GST_PAD_ALWAYS,

    GST_STATIC_CAPS_ANY);

enum

{

  PROP_0,

  PROP_N_PADS,

  PROP_ACTIVE_PAD,

  PROP_LAST

};

static gboolean gst_stream_selector_is_active_sinkpad (GstStreamSelector * sel,

    GstPad * pad);

static GstPad *gst_stream_selector_activate_sinkpad (GstStreamSelector * sel,

    GstPad * pad);

static GstPad *gst_stream_selector_get_linked_pad (GstPad * pad,

    gboolean strict);

#define GST_TYPE_SELECTOR_PAD \

  (gst_selector_pad_get_type())

#define GST_SELECTOR_PAD(obj) \

  (G_TYPE_CHECK_INSTANCE_CAST ((obj), GST_TYPE_SELECTOR_PAD, GstSelectorPad))

#define GST_SELECTOR_PAD_CLASS(klass) \

  (G_TYPE_CHECK_CLASS_CAST ((klass), GST_TYPE_SELECTOR_PAD, GstSelectorPadClass))

#define GST_IS_SELECTOR_PAD(obj) \

  (G_TYPE_CHECK_INSTANCE_TYPE ((obj), GST_TYPE_SELECTOR_PAD))

#define GST_IS_SELECTOR_PAD_CLASS(klass) \

  (G_TYPE_CHECK_CLASS_TYPE ((klass), GST_TYPE_SELECTOR_PAD))

#define GST_SELECTOR_PAD_CAST(obj) \

  ((GstSelectorPad *)(obj))

typedef struct _GstSelectorPad GstSelectorPad;

typedef struct _GstSelectorPadClass GstSelectorPadClass;

struct _GstSelectorPad

{

  GstPad parent;

  gboolean active;

  gboolean eos;

  gboolean segment_pending;

  GstSegment segment;

  GstTagList *tags;

};

struct _GstSelectorPadClass

{

  GstPadClass parent;

};

static void gst_selector_pad_class_init (GstSelectorPadClass * klass);

static void gst_selector_pad_init (GstSelectorPad * pad);

static void gst_selector_pad_finalize (GObject * object);

static void gst_selector_pad_get_property (GObject * object,

    guint prop_id, GValue * value, GParamSpec * pspec);

static GstPadClass *selector_pad_parent_class = NULL;

static void gst_selector_pad_reset (GstSelectorPad * pad);

static gboolean gst_selector_pad_event (GstPad * pad, GstEvent * event);

static GstCaps *gst_selector_pad_getcaps (GstPad * pad);

static GList *gst_selector_pad_get_linked_pads (GstPad * pad);

static GstFlowReturn gst_selector_pad_chain (GstPad * pad, GstBuffer * buf);

static GstFlowReturn gst_selector_pad_bufferalloc (GstPad * pad,

    guint64 offset, guint size, GstCaps * caps, GstBuffer ** buf);

enum

{

  PROP_PAD_0,

  PROP_PAD_TAGS,

  PROP_PAD_ACTIVE,

  PROP_PAD_LAST

};

static GType

gst_selector_pad_get_type (void)

{

  static GType selector_pad_type = 0;

  if (!selector_pad_type) {

    static const GTypeInfo selector_pad_info = {

      sizeof (GstSelectorPadClass),

      NULL,

      NULL,

      (GClassInitFunc) gst_selector_pad_class_init,

      NULL,

      NULL,

      sizeof (GstSelectorPad),

      0,

      (GInstanceInitFunc) gst_selector_pad_init,

    };

    selector_pad_type =

        &selector_pad_info, 0);

  }

  return selector_pad_type;

}

static void

gst_selector_pad_class_init (GstSelectorPadClass * klass)

{

  GObjectClass *gobject_class;

  gobject_class = (GObjectClass *) klass;

  selector_pad_parent_class = g_type_class_peek_parent (klass);

  gobject_class->finalize = gst_selector_pad_finalize;

  gobject_class->get_property =

      GST_DEBUG_FUNCPTR (gst_selector_pad_get_property);

  g_object_class_install_property (gobject_class, PROP_PAD_TAGS,

      g_param_spec_boxed ("tags", "Tags",

          "The currently active tags on the pad", GST_TYPE_TAG_LIST,

  g_object_class_install_property (gobject_class, PROP_PAD_ACTIVE,

      g_param_spec_boolean ("active", "Active",

}

static void

gst_selector_pad_init (GstSelectorPad * pad)

{

  gst_selector_pad_reset (pad);

}

static void

gst_selector_pad_finalize (GObject * object)

{

  GstSelectorPad *pad;

  pad = GST_SELECTOR_PAD_CAST (object);

  if (pad->tags)

    gst_tag_list_free (pad->tags);

  G_OBJECT_CLASS (selector_pad_parent_class)->finalize (object);

}

static void

gst_selector_pad_get_property (GObject * object,

    guint prop_id, GValue * value, GParamSpec * pspec)

{

  GstSelectorPad *pad;

  pad = GST_SELECTOR_PAD (object);

  switch (prop_id) {

    case PROP_PAD_TAGS:

      GST_OBJECT_LOCK (object);

      g_value_set_boxed (value, pad->tags);

      GST_OBJECT_UNLOCK (object);

      break;

    case PROP_PAD_ACTIVE:

    {

      GstStreamSelector *sel;

      sel = GST_STREAM_SELECTOR (gst_pad_get_parent (pad));

      g_value_set_boolean (value, gst_stream_selector_is_active_sinkpad (sel,

              GST_PAD_CAST (pad)));

      gst_object_unref (sel);

      break;

    }

    default:

      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);

      break;

  }

}

static void

gst_selector_pad_reset (GstSelectorPad * pad)

{

  pad->active = FALSE;

  pad->eos = FALSE;

  gst_segment_init (&pad->segment, GST_FORMAT_UNDEFINED);

}

/* strictly get the linked pad from the sinkpad. If the pad is active we return

 * the srcpad else we return NULL */

static GList *

gst_selector_pad_get_linked_pads (GstPad * pad)

{

  GstPad *otherpad;

  otherpad = gst_stream_selector_get_linked_pad (pad, TRUE);

  if (!otherpad)

    return NULL;

  /* need to drop the ref, internal linked pads is not MT safe */

  gst_object_unref (otherpad);

  return g_list_append (NULL, otherpad);

}

static gboolean

gst_selector_pad_event (GstPad * pad, GstEvent * event)

{

  gboolean res = TRUE;

  gboolean forward = TRUE;

  GstStreamSelector *sel;

  GstSelectorPad *selpad;

  sel = GST_STREAM_SELECTOR (gst_pad_get_parent (pad));

  selpad = GST_SELECTOR_PAD_CAST (pad);

  /* only forward if we are dealing with the active sinkpad */

  switch (GST_EVENT_TYPE (event)) {

    case GST_EVENT_FLUSH_STOP:

      gst_selector_pad_reset (selpad);

      break;

    case GST_EVENT_NEWSEGMENT:

    {

      gboolean update;

      GstFormat format;

      gdouble rate, arate;

      gint64 start, stop, time;

      gst_event_parse_new_segment_full (event, &update, &rate, &arate, &format,

          &start, &stop, &time);

      GST_DEBUG_OBJECT (selpad,

          "configured NEWSEGMENT update %d, rate %lf, applied rate %lf, "

          "format %d, "

          "%" G_GINT64_FORMAT " -- %" G_GINT64_FORMAT ", time %"

          G_GINT64_FORMAT, update, rate, arate, format, start, stop, time);

      gst_segment_set_newsegment_full (&selpad->segment, update,

          rate, arate, format, start, stop, time);

      /* if we are not going to forward the segment, mark the segment as

       * pending */

      if (!forward)

        selpad->segment_pending = TRUE;

      break;

    }

    case GST_EVENT_TAG:

    {

      GstTagList *tags;

      GST_OBJECT_LOCK (selpad);

      if (selpad->tags)

        gst_tag_list_free (selpad->tags);

      gst_event_parse_tag (event, &tags);

      if (tags)

        tags = gst_tag_list_copy (tags);

      selpad->tags = tags;

      GST_DEBUG_OBJECT (sel, "received tags %" GST_PTR_FORMAT, selpad->tags);

      GST_OBJECT_UNLOCK (selpad);

      break;

    }

    case GST_EVENT_EOS:

      selpad->eos = TRUE;

      break;

    default:

      break;

  }

  if (forward)

    res = gst_pad_push_event (sel->srcpad, event);

  else

    gst_event_unref (event);

  gst_object_unref (sel);

  return res;

}

static GstCaps *

gst_selector_pad_getcaps (GstPad * pad)

{

  GstStreamSelector *sel;

  GstCaps *caps;

  sel = GST_STREAM_SELECTOR (gst_pad_get_parent (pad));

  GST_DEBUG_OBJECT (sel, "Getting caps of srcpad peer");

  caps = gst_pad_peer_get_caps (sel->srcpad);

  if (caps == NULL)

    caps = gst_caps_new_any ();

  gst_object_unref (sel);

  return caps;

}

static GstFlowReturn

gst_selector_pad_bufferalloc (GstPad * pad, guint64 offset,

    guint size, GstCaps * caps, GstBuffer ** buf)

{

  GstStreamSelector *sel;

  GstFlowReturn result;

  GstPad *active_sinkpad;

  sel = GST_STREAM_SELECTOR (gst_pad_get_parent (pad));

  active_sinkpad = gst_stream_selector_activate_sinkpad (sel, pad);

  /* Fallback allocation for buffers from pads except the selected one */

  if (pad != active_sinkpad) {

    GST_DEBUG_OBJECT (sel,

        "Pad %s:%s is not selected. Performing fallback allocation",

        GST_DEBUG_PAD_NAME (pad));

    *buf = NULL;

    result = GST_FLOW_OK;

  } else {

    result = gst_pad_alloc_buffer (sel->srcpad, offset, size, caps, buf);

    /* FIXME: HACK. If buffer alloc returns not-linked, perform a fallback

     * allocation.  This should NOT be necessary, because playbin should

     * properly block the source pad from running until it's finished hooking 

     * everything up, but playbin needs refactoring first. */

    if (result == GST_FLOW_NOT_LINKED) {

      GST_DEBUG_OBJECT (sel,

          "No peer pad yet - performing fallback allocation for pad %s:%s",

          GST_DEBUG_PAD_NAME (pad));

      *buf = NULL;

      result = GST_FLOW_OK;

    }

  }

  gst_object_unref (sel);

  return result;

}

static GstFlowReturn

gst_selector_pad_chain (GstPad * pad, GstBuffer * buf)

{

  GstStreamSelector *sel;

  GstFlowReturn res;

  GstPad *active_sinkpad;

  GstSelectorPad *selpad;

  GstClockTime timestamp;

  GstSegment *seg;

  sel = GST_STREAM_SELECTOR (gst_pad_get_parent (pad));

  selpad = GST_SELECTOR_PAD_CAST (pad);

  seg = &selpad->segment;

  active_sinkpad = gst_stream_selector_activate_sinkpad (sel, pad);

  timestamp = GST_BUFFER_TIMESTAMP (buf);

  if (GST_CLOCK_TIME_IS_VALID (timestamp)) {

    GST_DEBUG_OBJECT (sel, "received timestamp %" GST_TIME_FORMAT,

        GST_TIME_ARGS (timestamp));

    gst_segment_set_last_stop (seg, seg->format, timestamp);

  }

  /* Ignore buffers from pads except the selected one */

  if (pad != active_sinkpad)

    goto ignore;

  /* if we have a pending segment, push it out now */

  if (selpad->segment_pending) {

    gst_pad_push_event (sel->srcpad, gst_event_new_new_segment_full (FALSE,

            seg->rate, seg->applied_rate, seg->format, seg->start, seg->stop,

            seg->time));

    selpad->segment_pending = FALSE;

  }

  /* forward */

  GST_DEBUG_OBJECT (sel, "Forwarding buffer %p from pad %s:%s", buf,

      GST_DEBUG_PAD_NAME (pad));

  res = gst_pad_push (sel->srcpad, buf);

done:

  gst_object_unref (sel);

  return res;

  /* dropped buffers */

ignore:

  {

    GST_DEBUG_OBJECT (sel, "Ignoring buffer %p from pad %s:%s",

        buf, GST_DEBUG_PAD_NAME (pad));

    gst_buffer_unref (buf);

    res = GST_FLOW_NOT_LINKED;

    goto done;

  }

}

static void gst_stream_selector_dispose (GObject * object);

static void gst_stream_selector_finalize (GObject * object);

static void gst_stream_selector_init (GstStreamSelector * sel);

static void gst_stream_selector_base_init (GstStreamSelectorClass * klass);

static void gst_stream_selector_class_init (GstStreamSelectorClass * klass);

static void gst_stream_selector_set_property (GObject * object,

    guint prop_id, const GValue * value, GParamSpec * pspec);

static void gst_stream_selector_get_property (GObject * object,

    guint prop_id, GValue * value, GParamSpec * pspec);

static GstPad *gst_stream_selector_request_new_pad (GstElement * element,

    GstPadTemplate * templ, const gchar * unused);

static void gst_stream_selector_release_pad (GstElement * element,

    GstPad * pad);

static GList *gst_stream_selector_get_linked_pads (GstPad * pad);

static GstCaps *gst_stream_selector_getcaps (GstPad * pad);

static GstElementClass *parent_class = NULL;

#ifdef __SYMBIAN32__

EXPORT_C

#endif

GType

gst_stream_selector_get_type (void)

{

  static GType stream_selector_type = 0;

  if (!stream_selector_type) {

    static const GTypeInfo stream_selector_info = {

      sizeof (GstStreamSelectorClass),

      (GBaseInitFunc) gst_stream_selector_base_init,

      NULL,

      (GClassInitFunc) gst_stream_selector_class_init,

      NULL,

      NULL,

      sizeof (GstStreamSelector),

      0,

      (GInstanceInitFunc) gst_stream_selector_init,

    };

    stream_selector_type =

        g_type_register_static (GST_TYPE_ELEMENT,

        "GstStreamSelector", &stream_selector_info, 0);

    GST_DEBUG_CATEGORY_INIT (stream_selector_debug,

        "streamselector", 0, "A stream-selector element");

  }

  return stream_selector_type;

}

static void

gst_stream_selector_base_init (GstStreamSelectorClass * klass)

{

  GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

  gst_element_class_set_details (element_class, &gst_stream_selector_details);

  gst_element_class_add_pad_template (element_class,

      gst_static_pad_template_get (&gst_stream_selector_sink_factory));

  gst_element_class_add_pad_template (element_class,

      gst_static_pad_template_get (&gst_stream_selector_src_factory));

}

static void

gst_stream_selector_class_init (GstStreamSelectorClass * klass)

{

  GObjectClass *gobject_class = G_OBJECT_CLASS (klass);

  GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass);

  parent_class = g_type_class_peek_parent (klass);

  gobject_class->dispose = gst_stream_selector_dispose;

  gobject_class->finalize = gst_stream_selector_finalize;

  gobject_class->set_property =

      GST_DEBUG_FUNCPTR (gst_stream_selector_set_property);

  gobject_class->get_property =

      GST_DEBUG_FUNCPTR (gst_stream_selector_get_property);

  g_object_class_install_property (gobject_class, PROP_N_PADS,

      g_param_spec_uint ("n-pads", "Number of Pads",

  g_object_class_install_property (gobject_class, PROP_ACTIVE_PAD,

      g_param_spec_object ("active-pad", "Active Pad",