/* GStreamer

 * Copyright (C) <2005> Wim Taymans <wim@fluendo.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 

 */

/**

 * SECTION:gstbasertppayload

 * @short_description: Base class for RTP payloader

 *

 * <refsect2>

 * <para>

 * Provides a base class for RTP payloaders

 * </para>

 * </refsect2>

 */

#ifdef HAVE_CONFIG_H

#  include "config.h"

#endif

#include <string.h>

#include <gst/rtp/gstrtpbuffer.h>

#include "gstbasertppayload.h"

GST_DEBUG_CATEGORY_STATIC (basertppayload_debug);

#define GST_CAT_DEFAULT (basertppayload_debug)

#define GST_BASE_RTP_PAYLOAD_GET_PRIVATE(obj)  \

   (G_TYPE_INSTANCE_GET_PRIVATE ((obj), GST_TYPE_BASE_RTP_PAYLOAD, GstBaseRTPPayloadPrivate))

struct _GstBaseRTPPayloadPrivate

{

  gboolean ts_offset_random;

  gboolean seqnum_offset_random;

  gboolean ssrc_random;

  guint16 next_seqnum;

};

/* BaseRTPPayload signals and args */

enum

{

  /* FILL ME */

  LAST_SIGNAL

};

/* FIXME 0.11, a better default is the Ethernet MTU of 

 * 1500 - sizeof(headers) as pointed out by marcelm in IRC:

 * So an Ethernet MTU of 1500, minus 60 for the max IP, minus 8 for UDP, gives

 * 1432 bytes or so.  And that should be adjusted downward further for other

 * encapsulations like PPPoE, so 1400 at most.

 */

#define DEFAULT_MTU                     1400

#define DEFAULT_PT                      96

#define DEFAULT_SSRC                    -1

#define DEFAULT_TIMESTAMP_OFFSET        -1

#define DEFAULT_SEQNUM_OFFSET           -1

#define DEFAULT_MAX_PTIME               -1

#define DEFAULT_MIN_PTIME               0

enum

{

  PROP_0,

  PROP_MTU,

  PROP_PT,

  PROP_SSRC,

  PROP_TIMESTAMP_OFFSET,

  PROP_SEQNUM_OFFSET,

  PROP_MAX_PTIME,

  PROP_MIN_PTIME,

  PROP_TIMESTAMP,

  PROP_SEQNUM

};

static void gst_basertppayload_class_init (GstBaseRTPPayloadClass * klass);

static void gst_basertppayload_base_init (GstBaseRTPPayloadClass * klass);

static void gst_basertppayload_init (GstBaseRTPPayload * basertppayload,

    gpointer g_class);

static void gst_basertppayload_finalize (GObject * object);

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

static GstFlowReturn gst_basertppayload_chain (GstPad * pad,

    GstBuffer * buffer);

static void gst_basertppayload_set_property (GObject * object, guint prop_id,

    const GValue * value, GParamSpec * pspec);

static void gst_basertppayload_get_property (GObject * object, guint prop_id,

    GValue * value, GParamSpec * pspec);

static GstStateChangeReturn gst_basertppayload_change_state (GstElement *

    element, GstStateChange transition);

static GstElementClass *parent_class = NULL;

/* FIXME 0.11: API should be changed to gst_base_typ_payload_xyz */

#ifdef __SYMBIAN32__

EXPORT_C

#endif

GType

gst_basertppayload_get_type (void)

{

  static GType basertppayload_type = 0;

  if (!basertppayload_type) {

    static const GTypeInfo basertppayload_info = {

      sizeof (GstBaseRTPPayloadClass),

      (GBaseInitFunc) gst_basertppayload_base_init,

      NULL,

      (GClassInitFunc) gst_basertppayload_class_init,

      NULL,

      NULL,

      sizeof (GstBaseRTPPayload),

      0,

      (GInstanceInitFunc) gst_basertppayload_init,

    };

    basertppayload_type =

        g_type_register_static (GST_TYPE_ELEMENT, "GstBaseRTPPayload",

        &basertppayload_info, G_TYPE_FLAG_ABSTRACT);

  }

  return basertppayload_type;

}

static void

gst_basertppayload_base_init (GstBaseRTPPayloadClass * klass)

{

}

static void

gst_basertppayload_class_init (GstBaseRTPPayloadClass * klass)

{

  GObjectClass *gobject_class;

  GstElementClass *gstelement_class;

  gobject_class = (GObjectClass *) klass;

  gstelement_class = (GstElementClass *) klass;

  g_type_class_add_private (klass, sizeof (GstBaseRTPPayloadPrivate));

  parent_class = g_type_class_peek_parent (klass);

  gobject_class->finalize = gst_basertppayload_finalize;

  gobject_class->set_property = gst_basertppayload_set_property;

  gobject_class->get_property = gst_basertppayload_get_property;

  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_MTU,

      g_param_spec_uint ("mtu", "MTU",

          "Maximum size of one packet",

  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_PT,

      g_param_spec_uint ("pt", "payload type",

  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_SSRC,

      g_param_spec_uint ("ssrc", "SSRC",

  g_object_class_install_property (G_OBJECT_CLASS (klass),

      PROP_TIMESTAMP_OFFSET, g_param_spec_uint ("timestamp-offset",

          "Timestamp Offset",

          "Offset to add to all outgoing timestamps (default = random)", 0,

  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_SEQNUM_OFFSET,

      g_param_spec_int ("seqnum-offset", "Sequence number Offset",

          "Offset to add to all outgoing seqnum (-1 = random)", -1, G_MAXUINT16,

  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_MAX_PTIME,

      g_param_spec_int64 ("max-ptime", "Max packet time",

          "Maximum duration of the packet data in ns (-1 = unlimited up to MTU)",

  /**

   * GstBaseRTPAudioPayload:min-ptime:

   *

   * Minimum duration of the packet data in ns (can't go above MTU)

   *

   * Since: 0.10.13

   **/

  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_MIN_PTIME,

      g_param_spec_int64 ("min-ptime", "Min packet time",

          "Minimum duration of the packet data in ns (can't go above MTU)",

  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_TIMESTAMP,

      g_param_spec_uint ("timestamp", "Timestamp",

          "The RTP timestamp of the last processed packet",

  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_SEQNUM,

      g_param_spec_uint ("seqnum", "Sequence number",

          "The RTP sequence number of the last processed packet",

  gstelement_class->change_state = gst_basertppayload_change_state;

  GST_DEBUG_CATEGORY_INIT (basertppayload_debug, "basertppayload", 0,

      "Base class for RTP Payloaders");

}

static void

gst_basertppayload_init (GstBaseRTPPayload * basertppayload, gpointer g_class)

{

  GstPadTemplate *templ;

  GstBaseRTPPayloadPrivate *priv;

  basertppayload->priv = priv =

      GST_BASE_RTP_PAYLOAD_GET_PRIVATE (basertppayload);

  templ =

      gst_element_class_get_pad_template (GST_ELEMENT_CLASS (g_class), "src");

  g_return_if_fail (templ != NULL);

  basertppayload->srcpad = gst_pad_new_from_template (templ, "src");

  gst_element_add_pad (GST_ELEMENT (basertppayload), basertppayload->srcpad);

  templ =

      gst_element_class_get_pad_template (GST_ELEMENT_CLASS (g_class), "sink");

  g_return_if_fail (templ != NULL);

  basertppayload->sinkpad = gst_pad_new_from_template (templ, "sink");

  gst_pad_set_setcaps_function (basertppayload->sinkpad,

  gst_pad_set_getcaps_function (basertppayload->sinkpad,

  gst_pad_set_event_function (basertppayload->sinkpad,

      gst_basertppayload_event);

  gst_pad_set_chain_function (basertppayload->sinkpad,

      gst_basertppayload_chain);

  gst_element_add_pad (GST_ELEMENT (basertppayload), basertppayload->sinkpad);

  basertppayload->seq_rand = g_rand_new ();

  basertppayload->ssrc_rand = g_rand_new ();

  basertppayload->ts_rand = g_rand_new ();

  basertppayload->mtu = DEFAULT_MTU;

  basertppayload->pt = DEFAULT_PT;

  basertppayload->seqnum_offset = DEFAULT_SEQNUM_OFFSET;

  basertppayload->ssrc = DEFAULT_SSRC;

  basertppayload->ts_offset = DEFAULT_TIMESTAMP_OFFSET;

  priv->seqnum_offset_random = (basertppayload->seqnum_offset == -1);

  priv->ts_offset_random = (basertppayload->ts_offset == -1);

  priv->ssrc_random = (basertppayload->ssrc == -1);

  basertppayload->max_ptime = DEFAULT_MAX_PTIME;

  basertppayload->min_ptime = DEFAULT_MIN_PTIME;

  basertppayload->media = NULL;

  basertppayload->encoding_name = NULL;

  basertppayload->clock_rate = 0;

}

static void

gst_basertppayload_finalize (GObject * object)

{

  GstBaseRTPPayload *basertppayload;

  basertppayload = GST_BASE_RTP_PAYLOAD (object);

  g_rand_free (basertppayload->seq_rand);

  basertppayload->seq_rand = NULL;

  g_rand_free (basertppayload->ssrc_rand);

  basertppayload->ssrc_rand = NULL;

  g_rand_free (basertppayload->ts_rand);

  basertppayload->ts_rand = NULL;

  g_free (basertppayload->media);

  basertppayload->media = NULL;

  g_free (basertppayload->encoding_name);

  basertppayload->encoding_name = NULL;

  G_OBJECT_CLASS (parent_class)->finalize (object);

}

static gboolean

{

  GstBaseRTPPayload *basertppayload;

  GstBaseRTPPayloadClass *basertppayload_class;

  gboolean ret = TRUE;

  GST_DEBUG_OBJECT (pad, "setting caps %" GST_PTR_FORMAT, caps);

  basertppayload = GST_BASE_RTP_PAYLOAD (gst_pad_get_parent (pad));

  basertppayload_class = GST_BASE_RTP_PAYLOAD_GET_CLASS (basertppayload);

  if (basertppayload_class->set_caps)

    ret = basertppayload_class->set_caps (basertppayload, caps);

  gst_object_unref (basertppayload);

  return ret;

}

static GstCaps *

{

  GstBaseRTPPayload *basertppayload;

  GstBaseRTPPayloadClass *basertppayload_class;

  GstCaps *caps = NULL;

  GST_DEBUG_OBJECT (pad, "getting caps");

  basertppayload = GST_BASE_RTP_PAYLOAD (gst_pad_get_parent (pad));

  basertppayload_class = GST_BASE_RTP_PAYLOAD_GET_CLASS (basertppayload);

  if (basertppayload_class->get_caps)

    caps = basertppayload_class->get_caps (basertppayload, pad);

  if (!caps) {

    caps = GST_PAD_TEMPLATE_CAPS (GST_PAD_PAD_TEMPLATE (pad));

    GST_DEBUG_OBJECT (pad,

        "using pad template %p with caps %p %" GST_PTR_FORMAT,

        GST_PAD_PAD_TEMPLATE (pad), caps, caps);

    caps = gst_caps_ref (caps);

  }

  gst_object_unref (basertppayload);

  return caps;

}

static gboolean

gst_basertppayload_event (GstPad * pad, GstEvent * event)

{

  GstBaseRTPPayload *basertppayload;

  GstBaseRTPPayloadClass *basertppayload_class;

  gboolean res;

  basertppayload = GST_BASE_RTP_PAYLOAD (gst_pad_get_parent (pad));

  basertppayload_class = GST_BASE_RTP_PAYLOAD_GET_CLASS (basertppayload);

  if (basertppayload_class->handle_event) {

    res = basertppayload_class->handle_event (pad, event);

    if (res)

      goto done;

  }

  switch (GST_EVENT_TYPE (event)) {

    case GST_EVENT_FLUSH_START:

      res = gst_pad_event_default (pad, event);

      break;

    case GST_EVENT_FLUSH_STOP:

      res = gst_pad_event_default (pad, event);

      gst_segment_init (&basertppayload->segment, GST_FORMAT_UNDEFINED);

      break;

    case GST_EVENT_NEWSEGMENT:

    {

      gboolean update;

      gdouble rate;

      GstFormat fmt;

      gint64 start, stop, position;

      gst_event_parse_new_segment (event, &update, &rate, &fmt, &start, &stop,

          &position);

      gst_segment_set_newsegment (&basertppayload->segment, update, rate, fmt,

          start, stop, position);

      /* fallthrough */

    }

    default:

      res = gst_pad_event_default (pad, event);

      break;

  }

done:

  gst_object_unref (basertppayload);

  return res;

}

static GstFlowReturn

gst_basertppayload_chain (GstPad * pad, GstBuffer * buffer)

{

  GstBaseRTPPayload *basertppayload;

  GstBaseRTPPayloadClass *basertppayload_class;

  GstFlowReturn ret;

  basertppayload = GST_BASE_RTP_PAYLOAD (gst_pad_get_parent (pad));

  basertppayload_class = GST_BASE_RTP_PAYLOAD_GET_CLASS (basertppayload);

  if (!basertppayload_class->handle_buffer)

    goto no_function;

  ret = basertppayload_class->handle_buffer (basertppayload, buffer);

  gst_object_unref (basertppayload);

  return ret;

  /* ERRORS */

no_function:

  {

    GST_ELEMENT_ERROR (basertppayload, STREAM, NOT_IMPLEMENTED, (NULL),

        ("subclass did not implement handle_buffer function"));

    gst_object_unref (basertppayload);

    gst_buffer_unref (buffer);

    return GST_FLOW_ERROR;

  }

}

/**

 * gst_basertppayload_set_options:

 * @payload: a #GstBaseRTPPayload

 * @media: the media type (typically "audio" or "video")

 * @dynamic: if the payload type is dynamic

 * @encoding_name: the encoding name 

 * @clock_rate: the clock rate of the media

 *

 * Set the rtp options of the payloader. These options will be set in the caps

 * of the payloader. Subclasses must call this method before calling

 * gst_basertppayload_push() or gst_basertppayload_set_outcaps().

 */

#ifdef __SYMBIAN32__

EXPORT_C

#endif

void

gst_basertppayload_set_options (GstBaseRTPPayload * payload,

    gchar * media, gboolean dynamic, gchar * encoding_name, guint32 clock_rate)

{

  g_return_if_fail (payload != NULL);

  g_return_if_fail (clock_rate != 0);

  g_free (payload->media);

  payload->media = g_strdup (media);

  payload->dynamic = dynamic;

  g_free (payload->encoding_name);

  payload->encoding_name = g_strdup (encoding_name);

  payload->clock_rate = clock_rate;

}

static gboolean

copy_fixed (GQuark field_id, const GValue * value, GstStructure * dest)

{

  if (gst_value_is_fixed (value)) {

    gst_structure_id_set_value (dest, field_id, value);

  }

  return TRUE;

}

/**

 * gst_basertppayload_set_outcaps:

 * @payload: a #GstBaseRTPPayload

 * @fieldname: the first field name or %NULL

 * @...: field values

 *

 * Configure the output caps with the optional parameters.

 *

 * Variable arguments should be in the form field name, field type

 * (as a GType), value(s).  The last variable argument should be NULL.

 *

 * Returns: %TRUE if the caps could be set.

 */

#ifdef __SYMBIAN32__

EXPORT_C

#endif

gboolean

gst_basertppayload_set_outcaps (GstBaseRTPPayload * payload, gchar * fieldname,

    ...)

{

  GstCaps *srccaps, *peercaps;

  srccaps = gst_caps_new_simple ("application/x-rtp",

      "media", G_TYPE_STRING, payload->media,

      "clock-rate", G_TYPE_INT, payload->clock_rate,

      "encoding-name", G_TYPE_STRING, payload->encoding_name, NULL);

  GST_DEBUG_OBJECT (payload, "defaults: %" GST_PTR_FORMAT, srccaps);

  if (fieldname) {

    va_list varargs;

    /* override with custom properties */

    va_start (varargs, fieldname);

    gst_caps_set_simple_valist (srccaps, fieldname, varargs);

    va_end (varargs);

    GST_DEBUG_OBJECT (payload, "custom added: %" GST_PTR_FORMAT, srccaps);

  }

  /* the peer caps can override some of the defaults */

  peercaps = gst_pad_peer_get_caps (payload->srcpad);

  if (peercaps == NULL) {

    /* no peer caps, just add the other properties */

    gst_caps_set_simple (srccaps,

        "payload", G_TYPE_INT, GST_BASE_RTP_PAYLOAD_PT (payload),

        "ssrc", G_TYPE_UINT, payload->current_ssrc,

        "clock-base", G_TYPE_UINT, payload->ts_base,

        "seqnum-base", G_TYPE_UINT, payload->seqnum_base, NULL);

    GST_DEBUG_OBJECT (payload, "no peer caps: %" GST_PTR_FORMAT, srccaps);

  } else {

    GstCaps *temp;

    GstStructure *s, *d;

    const GValue *value;

    gint pt;

    /* peer provides caps we can use to fixate, intersect. This always returns a

     * writable caps. */

    temp = gst_caps_intersect (srccaps, peercaps);

    gst_caps_unref (srccaps);

    gst_caps_unref (peercaps);