/* GStreamer

 * Copyright (C) <2006> Renato Araujo Oliveira Filho <renato.filho@indt.org.br>

 *                      Rosfran Borges <rosfran.borges@indt.org.br>

 *

 * 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.

 */

/* Element-Checklist-Version: 5 */

/**

 * SECTION:element-nuvdemux

 *

 * <refsect2>

 * <para>

 * Demuxes an .nuv file into raw or compressed audio and/or video streams.

 * </para>

 * <para>

 * This element currently only supports pull-based scheduling.

 * </para>

 * <title>Example launch line</title>

 * <para>

 * <programlisting>

 * gst-launch filesrc test.nuv ! nuvdemux name=demux  demux.audio_00 ! decodebin ! audioconvert ! audioresample ! autoaudiosink   demux.video_00 ! queue ! decodebin ! ffmpegcolorspace ! videoscale ! autovideosink

 * </programlisting>

 * Play (parse and decode) an .nuv file and try to output it to

 * an automatically detected soundcard and videosink. If the NUV file contains

 * compressed audio or video data, this will only work if you have the

 * right decoder elements/plugins installed.

 * </para>

 * </refsect2>

 *

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include <gst/gst.h>

#include <gst/gsterror.h>

#include <gst/gstplugin.h>

#include <string.h>

#include <math.h>

#include "glib/gi18n.h"

#include "gstnuvdemux.h"

#define GST_NUV_DEMUX_GET_PRIVATE(obj) (G_TYPE_INSTANCE_GET_PRIVATE ((obj), GST_TYPE_NUV_DEMUX, GstNuvDemuxPrivate))

GST_DEBUG_CATEGORY_STATIC (nuvdemux_debug);

#define GST_CAT_DEFAULT nuvdemux_debug

#define GST_FLOW_ERROR_NO_DATA  -101

enum

{

   NUV_PUSH_MODE = 0,

   NUV_PULL_MODE

};

GST_DEBUG_CATEGORY_EXTERN (GST_CAT_EVENT);

static const GstElementDetails gst_nuv_demux_details =

GST_ELEMENT_DETAILS ("Nuv demuxer",

    "Codec/Demuxer",

    "Demultiplex a .nuv file into audio and video",

    "Renato Araujo Oliveira Filho <renato.filho@indt.org.br>,"

    "Rosfran Borges <rosfran.borges@indt.org.br>");

/* file header */

typedef struct

{

    gchar id[12];       /* "NuppelVideo\0" or "MythTVVideo\0" */

    gchar version[5];    /* "x.xx\0" */

    gint  i_width;

    gint  i_height;

    gint  i_width_desired;

    gint  i_height_desired;

    gchar i_mode;            /* P progressive, I interlaced */

    gdouble  d_aspect;       /* 1.0 squared pixel */

    gdouble  d_fps;

    //fps num/denom

    gint     i_fpsn;

    gint     i_fpsd;

    gint     i_video_blocks; /* 0 no video, -1 unknown */

    gint     i_audio_blocks;

    gint     i_text_blocks;

    gint     i_keyframe_distance;

} nuv_header;

/* frame header */

typedef struct

{

    gchar i_type;        /* A: audio, V: video, S: sync; T: test

                           R: Seekpoint (string:RTjjjjjjjj)

                           D: Extra data for codec */

    gchar i_compression; /* V: 0 uncompressed

                              1 RTJpeg

                              2 RTJpeg+lzo

                              N black frame

                              L copy last

                           A: 0 uncompressed (44100 1-bits, 2ch)

                              1 lzo

                              2 layer 2

                              3 layer 3

                              F flac

                              S shorten

                              N null frame loudless

                              L copy last

                            S: B audio and vdeo sync point

                               A audio sync info (timecode == effective

                                    dsp frequency*100)

                               V next video sync (timecode == next video

                                    frame num)

                               S audio,video,text correlation */

    gchar i_keyframe;    /* 0 keyframe, else no no key frame */

    guint8 i_filters;  /* 0x01: gauss 5 pixel (8,2,2,2,2)/16

                           0x02: gauss 5 pixel (8,1,1,1,1)/12

                           0x04: cartoon filter */

    gint32 i_timecode;     /* ms */

    gint i_length;       /* V,A,T: length of following data

                           S: length of packet correl */

} nuv_frame_header;

/* FIXME Not sure of this one */

typedef struct

{

    gint             i_version;

    guint32		     i_video_fcc;

    guint32		     i_audio_fcc;

    gint             i_audio_sample_rate;

    gint             i_audio_bits_per_sample;

    gint             i_audio_channels;

    gint             i_audio_compression_ratio;

    gint             i_audio_quality;

    gint             i_rtjpeg_quality;

    gint             i_rtjpeg_luma_filter;

    gint             i_rtjpeg_chroma_filter;

    gint             i_lavc_bitrate;

    gint             i_lavc_qmin;

    gint             i_lavc_qmax;

    gint             i_lavc_maxqdiff;

    gint64         	 i_seekable_offset;

    gint64           i_keyframe_adjust_offset;

} nuv_extended_header;

typedef enum {

  GST_NUV_DEMUX_START,

  GST_NUV_DEMUX_HEADER_DATA,

  GST_NUV_DEMUX_EXTRA_DATA,

  GST_NUV_DEMUX_MPEG_DATA,

  GST_NUV_DEMUX_EXTEND_HEADER,

  GST_NUV_DEMUX_EXTEND_HEADER_DATA,

  GST_NUV_DEMUX_FRAME_HEADER,

  GST_NUV_DEMUX_MOVI,

  GST_NUV_DEMUX_INVALID_DATA

} GstNuvDemuxState;

struct _GstNuvDemuxPrivate {

  /* used for indicate the mode */

  guint         mode;

  /* used on push mode */

  GstAdapter    *adapter;

  /* pads */

  GstPad        *sinkpad;

  GstPad        *src_video_pad;

  GstPad        *src_audio_pad;

  /* Flow control */

  GstFlowReturn last_video_return;

  GstFlowReturn last_audio_return;

  /* NUV decoding state */

  GstNuvDemuxState  state;

  GstSegment        segment;

  guint64           last_update;

  guint64           offset;

  guint64           streamer_offset;

  /* duration information */

  gint64            duration_bytes;

  gint64            duration_time;

  gint64            duration_average;

  /* segment control info */

  gboolean          new_audio_segment;

  gboolean          new_video_segment;

  /* Mpeg ExtraData */

  guint64       mpeg_data_size;

  GstBuffer     *mpeg_buffer;

  /* Headers */

  nuv_header *h;

  nuv_extended_header *eh;

  nuv_frame_header *fh;

  /* anothers info */

  gint64            time_start;

  gint64            time_diff;

  gint64            time_qos;

  guint64           last_frame_time;

};

static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink",

    GST_PAD_SINK,

    GST_PAD_ALWAYS,

    GST_STATIC_CAPS ("video/x-nuv"));

static GstStaticPadTemplate audio_src_template =

GST_STATIC_PAD_TEMPLATE ("audio_src",

    GST_PAD_SRC,

    GST_PAD_SOMETIMES,

    GST_STATIC_CAPS_ANY);

static GstStaticPadTemplate video_src_template =

GST_STATIC_PAD_TEMPLATE ("video_src",

    GST_PAD_SRC,

    GST_PAD_SOMETIMES,

    GST_STATIC_CAPS_ANY);

static void gst_nuv_demux_dispose (GObject * object);

static void gst_nuv_demux_finalize (GObject * object);

static GstStateChangeReturn gst_nuv_demux_change_state (GstElement * element,

    GstStateChange transition);

static void gst_nuv_demux_loop (GstPad * pad);

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

static GstFlowReturn gst_nuv_demux_play (GstPad * pad);

static gboolean gst_nuv_demux_sink_activate_pull (GstPad * sinkpad,

    gboolean active);

static gboolean gst_nuv_demux_sink_activate_push (GstPad * pad,

    gboolean active);

static gboolean gst_nuv_demux_sink_activate (GstPad * sinkpad);

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

static GstFlowReturn gst_nuv_demux_read_bytes (GstNuvDemux * nuv, guint64 size,

    gboolean move, GstBuffer ** buffer);

static void gst_nuv_demux_reset (GstNuvDemux * nuv);

static void gst_nuv_demux_destoy_src_pad (GstNuvDemux * nuv);

static void gst_nuv_demux_send_eos (GstNuvDemux * nuv);

static void gst_nuv_demux_update_duration (GstNuvDemux *nuv, guint64 current_timestamp);

static gint64 gst_nuv_demux_get_bytes_duration (GstNuvDemux *nuv);

static gint64 gst_nuv_demux_get_time_duration (GstNuvDemux *nuv);

GST_BOILERPLATE (GstNuvDemux, gst_nuv_demux, GstElement, GST_TYPE_ELEMENT);

/******************************************************************************

 * Utils function

 ******************************************************************************/

#if G_BYTE_ORDER == G_BIG_ENDIAN

static inline gdouble

_gdouble_swap_le_be (gdouble * d)

{

  union

  {

    guint64 i;

    gdouble d;

  } u;

  u.d = *d;

  u.i = GUINT64_SWAP_LE_BE (u.i);

  return u.d;

}

#define READ_DOUBLE_FROM_LE(d) (_gdouble_swap_le_be((gdouble* ) d))

#else /* G_BYTE_ORDER != G_BIG_ENDIAN */

#define READ_DOUBLE_FROM_LE(d) *((gdouble* ) (d))

#endif /* G_BYTE_ORDER != G_BIG_ENDIAN */

static void

double2fraction (double in, int *num, int *denom)

{

    if (in == 29.97) {

        *num = 30000;

        *denom = 1001;

    } else if (in == 23.976) {

        *num = 24000;

        *denom = 1001;

    } else {

        *denom = 1;

        while (in - floor(in) >= 0.1) {

            *denom *= 10;

            in *= 10.0;

        }

        *num = (int)floor(in);

    }

}

/* GObject Functions */

static void

gst_nuv_demux_base_init (gpointer klass)

{

  GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

  gst_element_class_add_pad_template (element_class,

      gst_static_pad_template_get (&audio_src_template));

  gst_element_class_add_pad_template (element_class,

      gst_static_pad_template_get (&video_src_template));

  gst_element_class_add_pad_template (element_class,

      gst_static_pad_template_get (&sink_template));

  gst_element_class_set_details (element_class, &gst_nuv_demux_details);

}

static void

gst_nuv_demux_class_init (GstNuvDemuxClass * klass)

{

  GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass);

  GObjectClass *gobject_class = (GObjectClass *) klass;

  GST_DEBUG_CATEGORY_INIT (nuvdemux_debug, "nuvdemux",

      0, "Demuxer for NUV streams");

  parent_class = g_type_class_peek_parent (klass);

  gobject_class->dispose = gst_nuv_demux_dispose;

  gobject_class->finalize = gst_nuv_demux_finalize;

  gstelement_class->change_state = gst_nuv_demux_change_state;

  g_type_class_add_private (gobject_class, sizeof (GstNuvDemuxPrivate));

}

static void

gst_nuv_demux_init (GstNuvDemux * nuv, GstNuvDemuxClass * nuv_class)

{

  nuv->priv = GST_NUV_DEMUX_GET_PRIVATE (nuv);

  nuv->priv->sinkpad = gst_pad_new_from_static_template (&sink_template, "sink");

  gst_pad_set_activate_function (nuv->priv->sinkpad, gst_nuv_demux_sink_activate);

  gst_pad_set_activatepull_function (nuv->priv->sinkpad,

      gst_nuv_demux_sink_activate_pull);

  gst_pad_set_activatepush_function (nuv->priv->sinkpad,

      gst_nuv_demux_sink_activate_push);

  gst_pad_set_chain_function (nuv->priv->sinkpad,

      GST_DEBUG_FUNCPTR (gst_nuv_demux_chain));

  gst_pad_set_event_function (nuv->priv->sinkpad,

      gst_nuv_demux_sink_event);

  gst_element_add_pad (GST_ELEMENT (nuv), nuv->priv->sinkpad);

  nuv->priv->new_audio_segment = TRUE;

  nuv->priv->new_video_segment = TRUE;

  gst_nuv_demux_reset (nuv);

}

static void

gst_nuv_demux_dispose (GObject * object)

{

  GstNuvDemux *nuv = GST_NUV_DEMUX (object);

  if (nuv->priv->mpeg_buffer != NULL) {

    gst_buffer_unref (nuv->priv->mpeg_buffer);

  }

  gst_nuv_demux_reset (GST_NUV_DEMUX (object));

  gst_nuv_demux_destoy_src_pad (GST_NUV_DEMUX (object));

  if (nuv->priv->adapter != NULL) {

    gst_object_unref (nuv->priv->adapter);

  }

}

static void

gst_nuv_demux_finalize (GObject * object)

{

  G_OBJECT_CLASS (parent_class)->finalize (object);

}

/* HeaderLoad:

 */

static GstFlowReturn

gst_nuv_demux_header_load (GstNuvDemux * nuv, nuv_header ** h_ret)

{

  GstBuffer *buffer = NULL;

  GstFlowReturn res = gst_nuv_demux_read_bytes (nuv, 72, TRUE, &buffer);

  if ((res != GST_FLOW_OK) || (buffer == NULL))

    return res;

  nuv_header *h = g_new0 (nuv_header, 1);

  memcpy (h->id, buffer->data, 12);

  memcpy (h->version, buffer->data + 12, 5);

  h->i_width = GST_READ_UINT32_LE (&buffer->data[20]);

  h->i_height = GST_READ_UINT32_LE (&buffer->data[24]);

  h->i_width_desired = GST_READ_UINT32_LE (&buffer->data[28]);

  h->i_height_desired = GST_READ_UINT32_LE (&buffer->data[32]);

  h->i_mode = GPOINTER_TO_INT (buffer->data[36]);

  h->d_aspect = READ_DOUBLE_FROM_LE (&buffer->data[40]);

  h->d_fps = READ_DOUBLE_FROM_LE (&buffer->data[48]);

  /* get the num and denom values from fps */

  double2fraction (h->d_fps, &h->i_fpsn, &h->i_fpsd);

  h->i_video_blocks = GST_READ_UINT32_LE (&buffer->data[56]);

  h->i_audio_blocks = GST_READ_UINT32_LE (&buffer->data[60]);

  h->i_text_blocks = GST_READ_UINT32_LE (&buffer->data[64]);

  h->i_keyframe_distance = GST_READ_UINT32_LE (&buffer->data[68]);

  GST_DEBUG_OBJECT (nuv,

      "nuv: h=%s v=%s %dx%d a=%f fps=%f v=%d a=%d t=%d kfd=%d", h->id,

      h->version, h->i_width, h->i_height, h->d_aspect, h->d_fps,

      h->i_video_blocks, h->i_audio_blocks, h->i_text_blocks,

      h->i_keyframe_distance);

  *h_ret = h;

  gst_buffer_unref (buffer);

  return res;

}

static GstFlowReturn

gst_nuv_demux_stream_header_data (GstNuvDemux * nuv)

{

  GstFlowReturn res;

  res = gst_nuv_demux_header_load (nuv, &nuv->priv->h);

  if (res == GST_FLOW_OK)

    nuv->priv->state = GST_NUV_DEMUX_EXTRA_DATA;

  return res;

}

/*

 * Read NUV file tag

 */

static GstFlowReturn

gst_nuv_demux_stream_file_header (GstNuvDemux * nuv)

{

  GstFlowReturn res = GST_FLOW_OK;

  GstBuffer *file_header = NULL;

  res = gst_nuv_demux_read_bytes (nuv, 12, FALSE, &file_header);

  if ((res != GST_FLOW_OK) || (file_header == NULL)) {

    return res;

  } else {

    if (strncmp ((gchar *) file_header->data, "MythTVVideo", 11) ||

        strncmp ((gchar *) file_header->data, "NuppelVideo", 11)) {

      nuv->priv->state = GST_NUV_DEMUX_HEADER_DATA;

    } else {

      GST_DEBUG_OBJECT (nuv, "error parsing file header");

      nuv->priv->state = GST_NUV_DEMUX_INVALID_DATA;

      res = GST_FLOW_ERROR;

    }

  }

  if (file_header != NULL) {

    gst_buffer_unref (file_header);

  }

  return res;

}

/* FrameHeaderLoad:

 */

static GstFlowReturn

gst_nuv_demux_frame_header_load (GstNuvDemux * nuv, nuv_frame_header ** h_ret)

{

  unsigned char *data;

  nuv_frame_header *h;

  GstBuffer *buf = NULL;

  GstFlowReturn res = gst_nuv_demux_read_bytes (nuv, 12, TRUE, &buf);

  if ((res != GST_FLOW_OK) || (buf == NULL)) {

    if (buf != NULL) {

      gst_buffer_unref (buf);

    }

    return res;

  }

  h = g_new0 (nuv_frame_header, 1);

  data = buf->data;

  h->i_type = GPOINTER_TO_INT (data[0]);

  h->i_compression = GPOINTER_TO_INT (data[1]);

  h->i_keyframe = GPOINTER_TO_INT (data[2]);

  h->i_filters = GPOINTER_TO_INT (data[3]);

  h->i_timecode = GST_READ_UINT32_LE (&data[4]);

  h->i_length = GST_READ_UINT32_LE (&data[8]);

  GST_DEBUG_OBJECT (nuv, "frame hdr: t=%c c=%c k=%d f=0x%x timecode=%d l=%d",

      h->i_type,

      h->i_compression ? h->i_compression : ' ',

      h->i_keyframe ? h->i_keyframe : ' ',

      h->i_filters, h->i_timecode, h->i_length);

  *h_ret = h;

  gst_buffer_unref (buf);

  return GST_FLOW_OK;

}

static GstFlowReturn

gst_nuv_demux_extended_header_load (GstNuvDemux * nuv,

    nuv_extended_header ** h_ret)

{

  unsigned char *data;

  GstBuffer *buff = NULL;

  nuv_extended_header *h;

  GstFlowReturn res = gst_nuv_demux_read_bytes (nuv, 512, TRUE, &buff);

  if ((res != GST_FLOW_OK) || (buff == NULL)) {

    if (buff != NULL) {

      gst_buffer_unref (buff);

    }

    return res;

  }

  h = g_new0 (nuv_extended_header, 1);

  data = buff->data;

  h->i_version = GST_READ_UINT32_LE (&data[0]);

  h->i_video_fcc = GST_MAKE_FOURCC (data[4], data[5], data[6], data[7]);

  h->i_audio_fcc = GST_MAKE_FOURCC (data[8], data[9], data[10], data[11]);

  h->i_audio_sample_rate = GST_READ_UINT32_LE (&data[12]);

  h->i_audio_bits_per_sample = GST_READ_UINT32_LE (&data[16]);

  h->i_audio_channels = GST_READ_UINT32_LE (&data[20]);

  h->i_audio_compression_ratio = GST_READ_UINT32_LE (&data[24]);

  h->i_audio_quality = GST_READ_UINT32_LE (&data[28]);

  h->i_rtjpeg_quality = GST_READ_UINT32_LE (&data[32]);

  h->i_rtjpeg_luma_filter = GST_READ_UINT32_LE (&data[36]);

  h->i_rtjpeg_chroma_filter = GST_READ_UINT32_LE (&data[40]);

  h->i_lavc_bitrate = GST_READ_UINT32_LE (&data[44]);

  h->i_lavc_qmin = GST_READ_UINT32_LE (&data[48]);

  h->i_lavc_qmin = GST_READ_UINT32_LE (&data[52]);

  h->i_lavc_maxqdiff = GST_READ_UINT32_LE (&data[56]);

  h->i_seekable_offset = GST_READ_UINT64_LE (&data[60]);

  h->i_keyframe_adjust_offset = GST_READ_UINT64_LE (&data[68]);

  GST_DEBUG_OBJECT (nuv,

      "ex hdr: v=%d vffc=%4.4s afcc=%4.4s %dHz %dbits ach=%d acr=%d aq=%d"

      "rtjpeg q=%d lf=%d lc=%d lavc br=%d qmin=%d qmax=%d maxqdiff=%d seekableoff=%lld keyfao=%lld",

      h->i_version, (gchar *) & h->i_video_fcc, (gchar *) & h->i_audio_fcc,

      h->i_audio_sample_rate, h->i_audio_bits_per_sample, h->i_audio_channels,

      h->i_audio_compression_ratio, h->i_audio_quality, h->i_rtjpeg_quality,

      h->i_rtjpeg_luma_filter, h->i_rtjpeg_chroma_filter, h->i_lavc_bitrate,

      h->i_lavc_qmin, h->i_lavc_qmax, h->i_lavc_maxqdiff, h->i_seekable_offset,

      h->i_keyframe_adjust_offset);

  *h_ret = h;

  gst_buffer_unref (buff);

  return res;

}

/* Query Functions */

static const GstQueryType *

gst_nuv_demux_get_src_query_types (GstPad * pad)

{

  static const GstQueryType src_types[] = {

    GST_QUERY_POSITION,

    GST_QUERY_DURATION,

    0

  };

  return src_types;

}

static gboolean

gst_nuv_demux_handle_src_query (GstPad * pad, GstQuery * query)

{

  gboolean res = FALSE;

  GstNuvDemux *nuv = GST_NUV_DEMUX (gst_pad_get_parent (pad));

  switch (GST_QUERY_TYPE (query)) {

    case GST_QUERY_POSITION:

      if (GST_CLOCK_TIME_IS_VALID (nuv->priv->last_frame_time)) {

        gst_query_set_position (query, GST_FORMAT_TIME, nuv->priv->last_frame_time);

        res = TRUE;

      }

      break;

    case GST_QUERY_DURATION:

      {

        gint64 duration = 0;

        duration = gst_nuv_demux_get_time_duration (nuv);

        if (duration == GST_CLOCK_TIME_NONE) {

            duration = nuv->priv->duration_average;

        }

        if (duration != GST_CLOCK_TIME_NONE) {

            gst_query_set_duration (query, GST_FORMAT_TIME, duration);

            res = TRUE;

        }

      }

      break;

    default:

      res = FALSE;

      break;

  }

  gst_object_unref (nuv);

  return res;

}

static GstPad*

gst_nuv_demux_create_pad (GstNuvDemux *nuv, GstCaps *caps, GstStaticPadTemplate *template, const gchar* name)

{

    GstPad *pad = NULL;

    pad = gst_pad_new_from_static_template (template, name);

    gst_pad_use_fixed_caps (pad);

    gst_pad_set_caps (pad, caps);

    gst_pad_set_active (pad, TRUE);

    gst_element_add_pad (GST_ELEMENT (nuv), pad);

    return pad;

}

static void

gst_nuv_demux_create_pads (GstNuvDemux * nuv)

{

  if (nuv->priv->h->i_video_blocks != 0) {

    GstCaps *video_caps = NULL;

    video_caps = gst_caps_new_simple ("video/x-divx",

        "divxversion", G_TYPE_INT, 4,

        "width", G_TYPE_INT, nuv->priv->h->i_width,

        "height", G_TYPE_INT, nuv->priv->h->i_height,

        "framerate", GST_TYPE_FRACTION, nuv->priv->h->i_fpsn, nuv->priv->h->i_fpsd,

        "format", GST_TYPE_FOURCC, nuv->priv->eh->i_video_fcc,

        "pixel-aspect-ratio", GST_TYPE_FRACTION,

        (gint) (nuv->priv->h->d_aspect * 1000.0f), 1000, NULL);

    nuv->priv->src_video_pad = gst_nuv_demux_create_pad (nuv, video_caps, &video_src_template, "video_src");

    gst_caps_unref (video_caps);

  }

  if (nuv->priv->h->i_audio_blocks != 0) {

    GstCaps *audio_caps = NULL;

    audio_caps = gst_caps_new_simple ("audio/mpeg",

        "rate", G_TYPE_INT, nuv->priv->eh->i_audio_sample_rate,

        "format", GST_TYPE_FOURCC, nuv->priv->eh->i_audio_fcc,

        "channels", G_TYPE_INT, nuv->priv->eh->i_audio_channels,

        "mpegversion", G_TYPE_INT, nuv->priv->eh->i_version, NULL);

    nuv->priv->src_audio_pad = gst_nuv_demux_create_pad (nuv, audio_caps, &audio_src_template, "audio_src");

    gst_caps_unref (audio_caps);

  }

  gst_element_no_more_pads (GST_ELEMENT (nuv));

}

static GstFlowReturn

gst_nuv_demux_read_head_frame (GstNuvDemux * nuv)

{

  GstFlowReturn ret = GST_FLOW_OK;

  if (nuv->priv->fh != NULL)

  {

    g_free (nuv->priv->fh);

    nuv->priv->fh = NULL;

  }

  ret = gst_nuv_demux_frame_header_load (nuv, &nuv->priv->fh);

  if (ret != GST_FLOW_OK)

    return ret;

  nuv->priv->state = GST_NUV_DEMUX_MOVI;

  return ret;

}

static gboolean

gst_nuv_combine_flow (GstNuvDemux *nuv)

{

    GstFlowReturn ret_video = nuv->priv->last_video_return;

    GstFlowReturn ret_audio = nuv->priv->last_audio_return;

    if ((ret_video != GST_FLOW_OK) &&

        (ret_audio != GST_FLOW_OK))

        return FALSE;

    if (GST_FLOW_IS_FATAL (ret_video))

        return FALSE;

    if (GST_FLOW_IS_FATAL (ret_audio))

        return FALSE;

    return TRUE;

}

static GstFlowReturn

gst_nuv_demux_stream_data (GstNuvDemux * nuv)

{

  GstFlowReturn ret = GST_FLOW_OK;

  GstPad *pad = NULL;

  guint64 timestamp;

  GstBuffer *buf = NULL;

  nuv_frame_header *h = NULL;

  h = nuv->priv->fh;

  if (h->i_type == 'R')

    goto done;

  if (h->i_length > 0) {

	  ret = gst_nuv_demux_read_bytes (nuv, h->i_length, TRUE, &buf);

	  if ((ret != GST_FLOW_OK) || (buf == NULL))

		  return ret;

      if (h->i_timecode < 0) {

          h->i_timecode = h->i_timecode * -1;

          nuv->priv->time_diff = h->i_timecode;

      }

      else

          h->i_timecode += nuv->priv->time_diff;

      timestamp = h->i_timecode * GST_MSECOND;

      GST_BUFFER_SIZE (buf) = h->i_length;

      GST_BUFFER_TIMESTAMP (buf) = timestamp;

  }

  else {

    goto done;

  }

  switch (h->i_type) {

    case 'V':

    {

      pad = nuv->priv->src_video_pad;

      if (nuv->priv->new_video_segment) {

        /* send new segment event*/

        gst_pad_push_event (nuv->priv->src_video_pad,

          gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, 0,

                                     GST_CLOCK_TIME_NONE, 0));

        if (nuv->priv->time_start == GST_CLOCK_TIME_NONE) {

            nuv->priv->time_start = timestamp;

        }

        nuv->priv->new_video_segment = FALSE;

      }

      break;

    }

    case 'A':

    {

      pad = nuv->priv->src_audio_pad;

      if (nuv->priv->new_audio_segment) {

        /* send new segment event*/

        gst_pad_push_event (nuv->priv->src_audio_pad,

          gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, 0,

                                     GST_CLOCK_TIME_NONE, 0));

        if (nuv->priv->time_start == GST_CLOCK_TIME_NONE) {

            nuv->priv->time_start = timestamp;

        }

        nuv->priv->new_audio_segment = FALSE;

      }

      break;

    }

    case 'S':

    {

      switch (h->i_compression) {

        case 'V':

          GST_DEBUG_OBJECT (nuv, "sending new video segment: %d", h->i_timecode);

          gst_pad_push_event (nuv->priv->src_video_pad,

              gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, h->i_timecode * GST_MSECOND,

              		GST_CLOCK_TIME_NONE, 0));

          break;

        case 'A':

          GST_DEBUG_OBJECT (nuv, "sending new audio segment: %d", h->i_timecode);

          gst_pad_push_event (nuv->priv->src_audio_pad,

              gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, 0,

              		GST_CLOCK_TIME_NONE, 0));

          break;

        default:

          break;

      }

      if (buf != NULL)

        gst_buffer_unref (buf);

      goto done;

    }

    default:

      if (buf != NULL)

        gst_buffer_unref (buf);

      break;

  }

  if ((buf != NULL) && (pad != NULL)) {

      /* update average time */

      nuv->priv->streamer_offset += h->i_length;

      gst_segment_set_last_stop (&nuv->priv->segment, GST_FORMAT_TIME, timestamp);

      gst_nuv_demux_update_duration (nuv, timestamp);

      /* pushing the buffer */

      gst_buffer_set_caps (buf, GST_PAD_CAPS (pad));

      ret = gst_pad_push (pad, buf);

      if (ret != GST_FLOW_OK) {

        if (pad == nuv->priv->src_video_pad) {

            nuv->priv->last_video_return = ret;

        }

        else if (pad == nuv->priv->src_audio_pad) {

            nuv->priv->last_audio_return = ret;

        }

        /* verify anothers flow if is necessary stop task */

        if (gst_nuv_combine_flow (nuv) != FALSE) {

            ret = GST_FLOW_OK;

        }

        GST_WARNING_OBJECT (nuv, "error: %d pushing on srcpad %s", ret, gst_pad_get_name (pad));

      }

  }

done:

  nuv->priv->state = GST_NUV_DEMUX_FRAME_HEADER;

  g_free (nuv->priv->fh);

  nuv->priv->fh = NULL;

  return ret;

}

static GstFlowReturn

gst_nuv_demux_stream_mpeg_data (GstNuvDemux * nuv)

{

  GstFlowReturn ret = GST_FLOW_OK;

  /* ffmpeg extra data */

  ret = gst_nuv_demux_read_bytes (nuv, nuv->priv->mpeg_data_size, TRUE,

      &nuv->priv->mpeg_buffer);

  if ((ret != GST_FLOW_OK) || (nuv->priv->mpeg_buffer == NULL)) {

    return ret;

  }

  GST_BUFFER_SIZE (nuv->priv->mpeg_buffer) = nuv->priv->mpeg_data_size;

  nuv->priv->state = GST_NUV_DEMUX_EXTEND_HEADER;

  return ret;

}

static GstFlowReturn

gst_nuv_demux_stream_extra_data (GstNuvDemux * nuv)

{

  GstFlowReturn ret = GST_FLOW_OK;

  /* Load 'D' */

  nuv_frame_header *h;

  ret = gst_nuv_demux_frame_header_load (nuv, &h);

  if (ret != GST_FLOW_OK)

    return ret;

  if (h->i_type != 'D') {

    g_free (h);

    GST_WARNING_OBJECT (nuv, "Unsuported rtjpeg");

    return GST_FLOW_NOT_SUPPORTED;

  }

  if (h->i_length > 0) {

    if (h->i_compression == 'F') {

      nuv->priv->state = GST_NUV_DEMUX_MPEG_DATA;

    } else {

      g_free (h);

      GST_WARNING_OBJECT (nuv, "only file with extended chunk are supported");

      return GST_FLOW_NOT_SUPPORTED;

    }

  } else {

    nuv->priv->state = GST_NUV_DEMUX_EXTEND_HEADER;

  }

  g_free (h);

  h = NULL;

  return ret;

}

static GstFlowReturn

gst_nuv_demux_stream_extend_header_data (GstNuvDemux * nuv)

{

  GstFlowReturn ret = GST_FLOW_OK;

  ret = gst_nuv_demux_extended_header_load (nuv, &nuv->priv->eh);

  if (ret != GST_FLOW_OK)

    return ret;

  gst_nuv_demux_create_pads (nuv);

  nuv->priv->state = GST_NUV_DEMUX_FRAME_HEADER;

  return ret;

}

static GstFlowReturn

gst_nuv_demux_stream_extend_header (GstNuvDemux * nuv)

{

  GstBuffer *buf = NULL;

  GstFlowReturn res = GST_FLOW_OK;

  res = gst_nuv_demux_read_bytes (nuv, 1, FALSE, &buf);

  if ((res != GST_FLOW_OK) || (buf == NULL)) {

    if (buf != NULL) {

      gst_buffer_unref (buf);

    }

    return res;

  }

  if (buf->data[0] == 'X') {

    gst_buffer_unref (buf);

    buf = NULL;

    nuv_frame_header *h = NULL;

    res = gst_nuv_demux_frame_header_load (nuv, &h);

    if (res != GST_FLOW_OK)

      return res;

    if (h->i_length != 512) {

      g_free (h);

      return GST_FLOW_ERROR;

    }

    g_free (h);

    h = NULL;

    nuv->priv->state = GST_NUV_DEMUX_EXTEND_HEADER_DATA;

  } else {

    nuv->priv->state = GST_NUV_DEMUX_INVALID_DATA;

    g_object_unref (buf);

    GST_ELEMENT_WARNING (nuv, STREAM, FAILED,

        (_("incomplete NUV support")), ("incomplete NUV support"));

    return GST_FLOW_ERROR;

  }

  return res;

}

static GstFlowReturn

gst_nuv_demux_play (GstPad * pad)

{

  GstFlowReturn res = GST_FLOW_OK;

  GstNuvDemux *nuv = GST_NUV_DEMUX (GST_PAD_PARENT (pad));

  switch (nuv->priv->state) {

    case GST_NUV_DEMUX_START:

      res = gst_nuv_demux_stream_file_header (nuv);

      if ((res != GST_FLOW_OK) && (res != GST_FLOW_ERROR_NO_DATA)) {

        goto pause;

      }

      if (nuv->priv->state != GST_NUV_DEMUX_HEADER_DATA)

        break;

    case GST_NUV_DEMUX_HEADER_DATA:

      res = gst_nuv_demux_stream_header_data (nuv);

      if ((res != GST_FLOW_OK) && (res != GST_FLOW_ERROR_NO_DATA)) {

        goto pause;

      }

      if (nuv->priv->state != GST_NUV_DEMUX_EXTRA_DATA)

        break;

    case GST_NUV_DEMUX_EXTRA_DATA:

      res = gst_nuv_demux_stream_extra_data (nuv);

      if ((res != GST_FLOW_OK) && (res != GST_FLOW_ERROR_NO_DATA)) {

        goto pause;

      }

      if (nuv->priv->state != GST_NUV_DEMUX_MPEG_DATA)

        break;

    case GST_NUV_DEMUX_MPEG_DATA:

      res = gst_nuv_demux_stream_mpeg_data (nuv);

      if ((res != GST_FLOW_OK) && (res != GST_FLOW_ERROR_NO_DATA)) {

        goto pause;

      }

      if (nuv->priv->state != GST_NUV_DEMUX_EXTEND_HEADER)

        break;

    case GST_NUV_DEMUX_EXTEND_HEADER:

      res = gst_nuv_demux_stream_extend_header (nuv);

      if ((res != GST_FLOW_OK) && (res != GST_FLOW_ERROR_NO_DATA)) {

        goto pause;

      }

      if (nuv->priv->state != GST_NUV_DEMUX_EXTEND_HEADER_DATA)

        break;