gmyth: gst-plugins-nuvdemux/nuvdemux/gstnuvdemux.c@803672c406ec

     1 /* GStreamer

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

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

     4  *

     5  * This library is free software; you can redistribute it and/or

     6  * modify it under the terms of the GNU Library General Public

     7  * License as published by the Free Software Foundation; either

     8  * version 2 of the License, or (at your option) any later version.

     9  *

    10  * This library is distributed in the hope that it will be useful,

    11  * but WITHOUT ANY WARRANTY; without even the implied warranty of

    12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

    13  * Library General Public License for more details.

    14  *

    15  * You should have received a copy of the GNU Library General Public

    16  * License along with this library; if not, write to the

    17  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,

    18  * Boston, MA 02111-1307, USA.

    19  */

    20 /* Element-Checklist-Version: 5 */

    22 /**

    23  * SECTION:element-nuvdemux

    24  *

    25  * <refsect2>

    26  * <para>

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

    28  * </para>

    29  * <para>

    30  * This element currently only supports pull-based scheduling.

    31  * </para>

    32  * <title>Example launch line</title>

    33  * <para>

    34  * <programlisting>

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

    36  * </programlisting>

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

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

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

    40  * right decoder elements/plugins installed.

    41  * </para>

    42  * </refsect2>

    43  *

    44  */

    46 #ifdef HAVE_CONFIG_H

    47 #include "config.h"

    48 #endif

    50 #include <gst/gst.h>

    51 #include <gst/gsterror.h>

    52 #include <gst/gstplugin.h>

    53 #include <string.h>

    54 #include <math.h>

    56 #include "glib/gi18n.h"

    57 #include "gstnuvdemux.h"

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

    61 GST_DEBUG_CATEGORY_STATIC (nuvdemux_debug);

    62 #define GST_CAT_DEFAULT nuvdemux_debug

    63 #define GST_FLOW_ERROR_NO_DATA  -101

    65 enum

    66 {

    67    NUV_PUSH_MODE = 0,

    68    NUV_PULL_MODE

    69 };

    71 GST_DEBUG_CATEGORY_EXTERN (GST_CAT_EVENT);

    73 static const GstElementDetails gst_nuv_demux_details =

    74 GST_ELEMENT_DETAILS ("Nuv demuxer",

    75     "Codec/Demuxer",

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

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

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

    81 /* file header */

    82 typedef struct

    83 {

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

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

    87     gint  i_width;

    88     gint  i_height;

    89     gint  i_width_desired;

    90     gint  i_height_desired;

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

    94     gdouble  d_aspect;       /* 1.0 squared pixel */

    95     gdouble  d_fps;

    96     //fps num/denom

    97     gint     i_fpsn;

    98     gint     i_fpsd;

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

   101     gint     i_audio_blocks;

   102     gint     i_text_blocks;

   104     gint     i_keyframe_distance;

   106 } nuv_header;

   108 /* frame header */

   109 typedef struct

   110 {

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

   112                            R: Seekpoint (string:RTjjjjjjjj)

   113                            D: Extra data for codec */

   114     gchar i_compression; /* V: 0 uncompressed

   115                               1 RTJpeg

   116                               2 RTJpeg+lzo

   117                               N black frame

   118                               L copy last

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

   120                               1 lzo

   121                               2 layer 2

   122                               3 layer 3

   123                               F flac

   124                               S shorten

   125                               N null frame loudless

   126                               L copy last

   127                             S: B audio and vdeo sync point

   128                                A audio sync info (timecode == effective

   129                                     dsp frequency*100)

   130                                V next video sync (timecode == next video

   131                                     frame num)

   132                                S audio,video,text correlation */

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

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

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

   136                            0x04: cartoon filter */

   138     gint32 i_timecode;     /* ms */

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

   141                            S: length of packet correl */

   142 } nuv_frame_header;

   144 /* FIXME Not sure of this one */

   145 typedef struct

   146 {

   147     gint             i_version;

   148     guint32		     i_video_fcc;

   150     guint32		     i_audio_fcc;

   151     gint             i_audio_sample_rate;

   152     gint             i_audio_bits_per_sample;

   153     gint             i_audio_channels;

   154     gint             i_audio_compression_ratio;

   155     gint             i_audio_quality;

   156     gint             i_rtjpeg_quality;

   157     gint             i_rtjpeg_luma_filter;

   158     gint             i_rtjpeg_chroma_filter;

   159     gint             i_lavc_bitrate;

   160     gint             i_lavc_qmin;

   161     gint             i_lavc_qmax;

   162     gint             i_lavc_maxqdiff;

   163     gint64         	 i_seekable_offset;

   164     gint64           i_keyframe_adjust_offset;

   166 } nuv_extended_header;

   168 typedef enum {

   169   GST_NUV_DEMUX_START,

   170   GST_NUV_DEMUX_HEADER_DATA,

   171   GST_NUV_DEMUX_EXTRA_DATA,

   172   GST_NUV_DEMUX_MPEG_DATA,

   173   GST_NUV_DEMUX_EXTEND_HEADER,

   174   GST_NUV_DEMUX_EXTEND_HEADER_DATA,

   175   GST_NUV_DEMUX_FRAME_HEADER,

   176   GST_NUV_DEMUX_MOVI,

   177   GST_NUV_DEMUX_INVALID_DATA

   178 } GstNuvDemuxState;

   180 struct _GstNuvDemuxPrivate {

   181   /* used for indicate the mode */

   182   guint         mode;

   184   /* used on push mode */

   185   GstAdapter    *adapter;

   187   /* pads */

   188   GstPad        *sinkpad;

   189   GstPad        *src_video_pad;

   190   GstPad        *src_audio_pad;

   192   /* Flow control */

   193   GstFlowReturn last_video_return;

   194   GstFlowReturn last_audio_return;

   195   gboolean      more_data;

   197   /* NUV decoding state */

   198   GstNuvDemuxState  state;

   199   GstSegment        segment;

   200   guint64           last_update;

   201   guint64           offset;

   202   guint64           streamer_offset;

   204   /* duration information */

   205   gint64            duration_bytes;

   206   gint64            duration_time;

   207   gint64            duration_average;

   209   /* segment control info */

   210   gboolean          new_audio_segment;

   211   gboolean          new_video_segment;

   213   /* Mpeg ExtraData */

   214   guint64       mpeg_data_size;

   215   GstBuffer     *mpeg_buffer;

   217   /* Headers */

   218   nuv_header h;

   219   nuv_extended_header eh;

   220   nuv_frame_header fh;

   222   /* anothers info */

   223   gint64            time_start;

   224   gint64            time_diff;

   225   gint64            time_qos;

   226   guint64           last_frame_time;

   227 };

   230 static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink",

   231     GST_PAD_SINK,

   232     GST_PAD_ALWAYS,

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

   235 static GstStaticPadTemplate audio_src_template =

   236 GST_STATIC_PAD_TEMPLATE ("audio_src",

   237     GST_PAD_SRC,

   238     GST_PAD_SOMETIMES,

   239     GST_STATIC_CAPS_ANY);

   241 static GstStaticPadTemplate video_src_template =

   242 GST_STATIC_PAD_TEMPLATE ("video_src",

   243     GST_PAD_SRC,

   244     GST_PAD_SOMETIMES,

   245     GST_STATIC_CAPS_ANY);

   247 static void gst_nuv_demux_dispose (GObject * object);

   248 static void gst_nuv_demux_finalize (GObject * object);

   249 static GstStateChangeReturn gst_nuv_demux_change_state (GstElement * element,

   250     GstStateChange transition);

   251 static void gst_nuv_demux_loop (GstPad * pad);

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

   253 static GstFlowReturn gst_nuv_demux_play (GstPad * pad);

   254 static gboolean gst_nuv_demux_sink_activate_pull (GstPad * sinkpad,

   255     gboolean active);

   256 static gboolean gst_nuv_demux_sink_activate_push (GstPad * pad,

   257     gboolean active);

   258 static gboolean gst_nuv_demux_sink_activate (GstPad * sinkpad);

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

   260 static GstFlowReturn gst_nuv_demux_read_bytes (GstNuvDemux * nuv, guint64 size,

   261     gboolean move, GstBuffer ** buffer);

   262 static void gst_nuv_demux_reset (GstNuvDemux * nuv);

   263 static void gst_nuv_demux_destoy_src_pad (GstNuvDemux * nuv);

   264 static void gst_nuv_demux_send_eos (GstNuvDemux * nuv);

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

   266 static gint64 gst_nuv_demux_get_bytes_duration (GstNuvDemux *nuv);

   267 static gint64 gst_nuv_demux_get_time_duration (GstNuvDemux *nuv);

   269 #if (GST_VERSION_MINOR == 10) && (GST_VERSION_MICRO < 6)

   270 GstBuffer * gst_adapter_take_buffer (GstAdapter * adapter, guint nbytes);

   271 #endif

   274 GST_BOILERPLATE (GstNuvDemux, gst_nuv_demux, GstElement, GST_TYPE_ELEMENT);

   276 /******************************************************************************

   277  * Utils function

   278  ******************************************************************************/

   279 #if G_BYTE_ORDER == G_BIG_ENDIAN

   280 static inline gdouble

   281 _gdouble_swap_le_be (gdouble * d)

   282 {

   283   union

   284   {

   285     guint64 i;

   286     gdouble d;

   287   } u;

   289   u.d = *d;

   290   u.i = GUINT64_SWAP_LE_BE (u.i);

   291   return u.d;

   292 }

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

   295 #else /* G_BYTE_ORDER != G_BIG_ENDIAN */

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

   297 #endif /* G_BYTE_ORDER != G_BIG_ENDIAN */

   299 static void

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

   301 {

   302     if (in == 29.97) {

   303         *num = 30000;

   304         *denom = 1001;

   305     } else if (in == 23.976) {

   306         *num = 24000;

   307         *denom = 1001;

   308     } else {

   309         *denom = 1;

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

   311             *denom *= 10;

   312             in *= 10.0;

   313         }

   314         *num = (int)floor(in);

   315     }

   316 }

   318 /* GObject Functions */

   320 static void

   321 gst_nuv_demux_base_init (gpointer klass)

   322 {

   323   GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

   325   gst_element_class_add_pad_template (element_class,

   326       gst_static_pad_template_get (&audio_src_template));

   328   gst_element_class_add_pad_template (element_class,

   329       gst_static_pad_template_get (&video_src_template));

   331   gst_element_class_add_pad_template (element_class,

   332       gst_static_pad_template_get (&sink_template));

   333   gst_element_class_set_details (element_class, &gst_nuv_demux_details);

   334 }

   336 static void

   337 gst_nuv_demux_class_init (GstNuvDemuxClass * klass)

   338 {

   339   GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass);

   340   GObjectClass *gobject_class = (GObjectClass *) klass;

   342   GST_DEBUG_CATEGORY_INIT (nuvdemux_debug, "nuvdemux",

   343       0, "Demuxer for NUV streams");

   345   parent_class = g_type_class_peek_parent (klass);

   347   gobject_class->dispose = gst_nuv_demux_dispose;

   348   gobject_class->finalize = gst_nuv_demux_finalize;

   349   gstelement_class->change_state = gst_nuv_demux_change_state;

   351   g_type_class_add_private (gobject_class, sizeof (GstNuvDemuxPrivate));

   352 }

   354 static void

   355 gst_nuv_demux_init (GstNuvDemux * nuv, GstNuvDemuxClass * nuv_class)

   356 {

   357   nuv->priv = GST_NUV_DEMUX_GET_PRIVATE (nuv);

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

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

   360   gst_pad_set_activatepull_function (nuv->priv->sinkpad,

   361       gst_nuv_demux_sink_activate_pull);

   362   gst_pad_set_activatepush_function (nuv->priv->sinkpad,

   363       gst_nuv_demux_sink_activate_push);

   364   gst_pad_set_chain_function (nuv->priv->sinkpad,

   365       GST_DEBUG_FUNCPTR (gst_nuv_demux_chain));

   366   gst_pad_set_event_function (nuv->priv->sinkpad,

   367       gst_nuv_demux_sink_event);

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

   370   nuv->priv->new_audio_segment = TRUE;

   371   nuv->priv->new_video_segment = TRUE;

   373   /* creating adapter */

   374   nuv->priv->mode = NUV_PUSH_MODE;

   375   nuv->priv->adapter = gst_adapter_new ();

   377   gst_nuv_demux_reset (nuv);

   378 }

   380 static void

   381 gst_nuv_demux_dispose (GObject * object)

   382 {

   383   GstNuvDemux *nuv = GST_NUV_DEMUX (object);

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

   386     gst_buffer_unref (nuv->priv->mpeg_buffer);

   387   }

   389   gst_nuv_demux_reset (GST_NUV_DEMUX (object));

   390   gst_nuv_demux_destoy_src_pad (GST_NUV_DEMUX (object));

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

   393     gst_object_unref (nuv->priv->adapter);

   394   }

   395 }

   397 static void

   398 gst_nuv_demux_finalize (GObject * object)

   399 {

   400   G_OBJECT_CLASS (parent_class)->finalize (object);

   401 }

   404 /* HeaderLoad:

   405  */

   406 static GstFlowReturn

   407 gst_nuv_demux_header_load (GstNuvDemux * nuv, nuv_header *h)

   408 {

   409   GstBuffer *buffer = NULL;

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

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

   413       goto done;

   414   }

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

   432   GST_DEBUG_OBJECT (nuv,

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

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

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

   436       h->i_keyframe_distance);

   438 done:

   439   if (buffer != NULL) {

   440      gst_buffer_unref (buffer);

   441      buffer = NULL;

   442   }

   443   return res;

   444 }

   446 static GstFlowReturn

   447 gst_nuv_demux_stream_header_data (GstNuvDemux * nuv)

   448 {

   449   GstFlowReturn res;

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

   452   if (res == GST_FLOW_OK)

   453     nuv->priv->state = GST_NUV_DEMUX_EXTRA_DATA;

   454   return res;

   455 }

   457 /*

   458  * Read NUV file tag

   459  */

   460 static GstFlowReturn

   461 gst_nuv_demux_stream_file_header (GstNuvDemux * nuv)

   462 {

   463   GstFlowReturn res = GST_FLOW_OK;

   464   GstBuffer *file_header = NULL;

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

   467   if (res == GST_FLOW_OK) {

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

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

   470       nuv->priv->state = GST_NUV_DEMUX_HEADER_DATA;

   471     } else {

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

   473       nuv->priv->state = GST_NUV_DEMUX_INVALID_DATA;

   474       res = GST_FLOW_ERROR;

   475     }

   476   }

   478   if (file_header != NULL) {

   479     gst_buffer_unref (file_header);

   480     file_header = NULL;

   481   }

   482   return res;

   483 }

   485 /* FrameHeaderLoad:

   486  */

   487 static GstFlowReturn

   488 gst_nuv_demux_frame_header_load (GstNuvDemux * nuv, nuv_frame_header *h)

   489 {

   490   unsigned char *data;

   491   GstBuffer *buf = NULL;

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

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

   496       goto done;

   497   }

   499   data = buf->data;

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

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

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

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

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

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

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

   509       h->i_type,

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

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

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

   514 done:

   515   if (buf != NULL) {

   516       gst_buffer_unref (buf);

   517       buf = NULL;

   518   }

   519   return res;

   520 }

   522 static GstFlowReturn

   523 gst_nuv_demux_extended_header_load (GstNuvDemux * nuv,

   524     nuv_extended_header * h)

   525 {

   526   unsigned char *data;

   527   GstBuffer *buff = NULL;

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

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

   532       goto done;

   533   }

   535   data = buff->data;

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

   554   GST_DEBUG_OBJECT (nuv,

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

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

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

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

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

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

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

   562       h->i_keyframe_adjust_offset);

   564 done:

   565   if (buff != NULL) {

   566     gst_buffer_unref (buff);

   567     buff = NULL;

   568   }

   569   return res;

   570 }

   573 /* Query Functions */

   574 static const GstQueryType *

   575 gst_nuv_demux_get_src_query_types (GstPad * pad)

   576 {

   577   static const GstQueryType src_types[] = {

   578     GST_QUERY_POSITION,

   579     GST_QUERY_DURATION,

   580     0

   581   };

   583   return src_types;

   584 }

   586 static gboolean

   587 gst_nuv_demux_handle_src_query (GstPad * pad, GstQuery * query)

   588 {

   589   gboolean res = FALSE;

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

   592   switch (GST_QUERY_TYPE (query)) {

   593     case GST_QUERY_POSITION:

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

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

   597         res = TRUE;

   598       }

   599       break;

   600     case GST_QUERY_DURATION:

   601       {

   602         gint64 duration = 0;

   603         duration = gst_nuv_demux_get_time_duration (nuv);

   604         if (duration == GST_CLOCK_TIME_NONE) {

   605             duration = nuv->priv->duration_average;

   606         }

   607         if (duration != GST_CLOCK_TIME_NONE) {

   608             gst_query_set_duration (query, GST_FORMAT_TIME, duration);

   609             res = TRUE;

   610         }

   611       }

   612       break;

   613     default:

   614       res = FALSE;

   615       break;

   616   }

   618   gst_object_unref (nuv);

   620   return res;

   621 }

   623 static GstPad*

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

   625 {

   626     GstPad *pad = NULL;

   627     pad = gst_pad_new_from_static_template (template, name);

   628     gst_pad_set_caps (pad, caps);

   629     gst_pad_set_active (pad, TRUE);

   630     gst_pad_use_fixed_caps (pad);

   631     gst_element_add_pad (GST_ELEMENT (nuv), pad);

   633     return pad;

   634 }

   636 static void

   637 gst_nuv_demux_create_pads (GstNuvDemux * nuv)

   638 {

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

   640     GstCaps *video_caps = NULL;

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

   643         "divxversion", G_TYPE_INT, 4,

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

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

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

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

   648         "pixel-aspect-ratio", GST_TYPE_FRACTION,

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

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

   652     gst_caps_unref (video_caps);

   653   }

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

   656     GstCaps *audio_caps = NULL;

   658     audio_caps = gst_caps_new_simple ("audio/mpeg",

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

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

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

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

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

   665     gst_caps_unref (audio_caps);

   666   }

   668   gst_element_no_more_pads (GST_ELEMENT (nuv));

   669 }

   671 static GstFlowReturn

   672 gst_nuv_demux_read_head_frame (GstNuvDemux * nuv)

   673 {

   674   GstFlowReturn ret = GST_FLOW_OK;

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

   677   if (ret != GST_FLOW_OK)

   678     return ret;

   680   nuv->priv->state = GST_NUV_DEMUX_MOVI;

   681   return ret;

   682 }

   684 static gboolean

   685 gst_nuv_combine_flow (GstNuvDemux *nuv)

   686 {

   687     GstFlowReturn ret_video = nuv->priv->last_video_return;

   688     GstFlowReturn ret_audio = nuv->priv->last_audio_return;

   690     if ((ret_video != GST_FLOW_OK) &&

   691         (ret_audio != GST_FLOW_OK))

   692         return FALSE;

   694     if (GST_FLOW_IS_FATAL (ret_video))

   695         return FALSE;

   697     if (GST_FLOW_IS_FATAL (ret_audio))

   698         return FALSE;

   700     return TRUE;

   701 }

   703 static GstFlowReturn

   704 gst_nuv_demux_stream_data (GstNuvDemux * nuv)

   705 {

   706   GstFlowReturn ret = GST_FLOW_OK;

   707   GstPad *pad = NULL;

   708   guint64 timestamp;

   709   GstBuffer *buf = NULL;

   710   nuv_frame_header h;

   712   h = nuv->priv->fh;

   714   if (h.i_type == 'R')

   715     goto done;

   717   if (h.i_length > 0) {

   718 	  ret = gst_nuv_demux_read_bytes (nuv, h.i_length, TRUE, &buf);

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

   720           goto done;

   721       }

   723       if ((h.i_timecode < 0)) {

   724 	goto done;

   725       }

   727       timestamp = h.i_timecode * GST_MSECOND;

   728       GST_BUFFER_TIMESTAMP (buf) = timestamp;

   729   }

   730   else {

   731     goto done;

   732   }

   735   switch (h.i_type) {

   736     case 'V':

   737     {

   738       pad = nuv->priv->src_video_pad;

   740       if (nuv->priv->new_video_segment) {

   742         /* send new segment event*/

   743         gst_pad_push_event (nuv->priv->src_video_pad,

   744           gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, 0,

   745                                      GST_CLOCK_TIME_NONE, 0));

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

   748             nuv->priv->time_start = timestamp;

   749         }

   750         nuv->priv->new_video_segment = FALSE;

   751       }

   753       break;

   754     }

   755     case 'A':

   756     {

   757       pad = nuv->priv->src_audio_pad;

   759       if (nuv->priv->new_audio_segment) {

   760         /* send new segment event*/

   761         gst_pad_push_event (nuv->priv->src_audio_pad,

   762           gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, 0,

   763                                      GST_CLOCK_TIME_NONE, 0));

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

   766             nuv->priv->time_start = timestamp;

   767         }

   768         nuv->priv->new_audio_segment = FALSE;

   769       }

   771       break;

   772     }

   773     case 'S':

   774    {

   775       switch (h.i_compression) {

   776         case 'V':

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

   778           gst_pad_push_event (nuv->priv->src_video_pad,

   779               gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, h.i_timecode * GST_MSECOND,

   780               		GST_CLOCK_TIME_NONE, 0));

   781           break;

   782         case 'A':

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

   784           gst_pad_push_event (nuv->priv->src_audio_pad,

   785               gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, 0,

   786               		GST_CLOCK_TIME_NONE, 0));

   787           break;

   788         default:

   789           break;

   790       }

   791       goto done;

   792     }

   793     default:

   794       break;

   795   }

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

   798       /* update average time */

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

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

   801       gst_nuv_demux_update_duration (nuv, timestamp);

   803       /* pushing the buffer */

   804       gst_buffer_set_caps (buf, GST_PAD_CAPS (pad));

   805       ret = gst_pad_push (pad, buf);

   806       buf = NULL;

   808       if (ret != GST_FLOW_OK) {

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

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

   812             nuv->priv->last_video_return = ret;

   813         }

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

   815             nuv->priv->last_audio_return = ret;

   816         }

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

   819         if (gst_nuv_combine_flow (nuv) != FALSE) {

   820             ret = GST_FLOW_OK;

   821         }

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

   824       }

   825   }

   827 done:

   828   if (buf != NULL) {

   829       gst_buffer_unref (buf);

   830       buf = NULL;

   831   }

   832   if (ret == GST_FLOW_OK) {

   833      nuv->priv->state = GST_NUV_DEMUX_FRAME_HEADER;

   834      memset (&nuv->priv->fh, 0, sizeof (nuv->priv->fh));

   835   }

   836   return ret;

   837 }

   839 static GstFlowReturn

   840 gst_nuv_demux_stream_mpeg_data (GstNuvDemux * nuv)

   841 {

   842   GstFlowReturn ret = GST_FLOW_OK;

   844   /* ffmpeg extra data */

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

   846       &nuv->priv->mpeg_buffer);

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

   848     return ret;

   849   }

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

   852   nuv->priv->state = GST_NUV_DEMUX_EXTEND_HEADER;

   853   return ret;

   854 }

   856 static GstFlowReturn

   857 gst_nuv_demux_stream_extra_data (GstNuvDemux * nuv)

   858 {

   859   GstFlowReturn ret = GST_FLOW_OK;

   861   /* Load 'D' */

   862   nuv_frame_header h;

   864   ret = gst_nuv_demux_frame_header_load (nuv, &h);

   865   if (ret != GST_FLOW_OK)

   866     return ret;

   868   if (h.i_type != 'D') {

   869     GST_WARNING_OBJECT (nuv, "Unsuported rtjpeg");

   870     return GST_FLOW_NOT_SUPPORTED;

   871   }

   873   if (h.i_length > 0) {

   874     if (h.i_compression == 'F') {

   875       nuv->priv->state = GST_NUV_DEMUX_MPEG_DATA;

   876     } else {

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

   878       return GST_FLOW_NOT_SUPPORTED;

   879     }

   880   } else {

   881     nuv->priv->state = GST_NUV_DEMUX_EXTEND_HEADER;

   882   }

   884   return ret;

   885 }

   887 static GstFlowReturn

   888 gst_nuv_demux_stream_extend_header_data (GstNuvDemux * nuv)

   889 {

   890   GstFlowReturn ret = GST_FLOW_OK;

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

   893   if (ret != GST_FLOW_OK)

   894     return ret;

   896   gst_nuv_demux_create_pads (nuv);

   897   nuv->priv->state = GST_NUV_DEMUX_FRAME_HEADER;

   898   return ret;

   899 }

   901 static GstFlowReturn

   902 gst_nuv_demux_stream_extend_header (GstNuvDemux * nuv)

   903 {

   904   GstBuffer *buf = NULL;

   905   GstFlowReturn res = GST_FLOW_OK;

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

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

   909     if (buf != NULL) {

   910       gst_buffer_unref (buf);

   911     }

   912     return res;

   913   }

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

   916     gst_buffer_unref (buf);

   917     buf = NULL;

   918     nuv_frame_header h;

   920     res = gst_nuv_demux_frame_header_load (nuv, &h);

   921     if (res != GST_FLOW_OK)

   922       return res;

   924     if (h.i_length != 512) {

   925       return GST_FLOW_ERROR;

   926     }

   927     nuv->priv->state = GST_NUV_DEMUX_EXTEND_HEADER_DATA;

   928   } else {

   929     nuv->priv->state = GST_NUV_DEMUX_INVALID_DATA;

   930     g_object_unref (buf);

   931     GST_ELEMENT_WARNING (nuv, STREAM, FAILED,

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

   933     return GST_FLOW_ERROR;

   934   }

   935   return res;

   936 }

   938 static GstFlowReturn

   939 gst_nuv_demux_play (GstPad * pad)

   940 {

   941   GstFlowReturn res = GST_FLOW_OK;

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

   944   switch (nuv->priv->state) {

   945     case GST_NUV_DEMUX_START:

   946       res = gst_nuv_demux_stream_file_header (nuv);

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

   948         goto pause;

   949       }

   950       break;

   952     case GST_NUV_DEMUX_HEADER_DATA:

   953       res = gst_nuv_demux_stream_header_data (nuv);

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

   955         goto pause;

   956       }

   957       break;

   959     case GST_NUV_DEMUX_EXTRA_DATA:

   960       res = gst_nuv_demux_stream_extra_data (nuv);

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

   962         goto pause;

   963       }

   964       break;

   966     case GST_NUV_DEMUX_MPEG_DATA:

   967       res = gst_nuv_demux_stream_mpeg_data (nuv);

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

   969         goto pause;

   970       }

   971       break;

   973     case GST_NUV_DEMUX_EXTEND_HEADER:

   974       res = gst_nuv_demux_stream_extend_header (nuv);

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

   976         goto pause;

   977       }

   978       break;

   980     case GST_NUV_DEMUX_EXTEND_HEADER_DATA:

   981       res = gst_nuv_demux_stream_extend_header_data (nuv);

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

   983         goto pause;

   984       }

   985       break;

   987     case GST_NUV_DEMUX_FRAME_HEADER:

   988       res = gst_nuv_demux_read_head_frame (nuv);

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

   990         goto pause;

   991       }

   992       break;

   994     case GST_NUV_DEMUX_MOVI:

   995       res = gst_nuv_demux_stream_data (nuv);

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

   997         goto pause;

   998       }

   999       break;

  1001     case GST_NUV_DEMUX_INVALID_DATA:

  1002       goto pause;

  1003       break;

  1004     default:

  1005       g_assert_not_reached ();

  1006   }

  1008   return GST_FLOW_OK;

  1010 pause:

  1011   GST_LOG_OBJECT (nuv, "pausing task, reason %s", gst_flow_get_name (res));

  1012   gst_pad_pause_task (nuv->priv->sinkpad);

  1013   if (GST_FLOW_IS_FATAL (res)) {

  1014     GST_ELEMENT_ERROR (nuv, STREAM, FAILED,

  1015         (_("Internal data stream error.")),

  1016         ("streaming stopped, reason %s", gst_flow_get_name (res)));

  1018     gst_nuv_demux_send_eos (nuv);

  1019   }

  1020   return res;

  1021 }

  1023 static void

  1024 gst_nuv_demux_send_eos (GstNuvDemux * nuv)

  1025 {

  1026   gst_element_post_message (GST_ELEMENT (nuv),

  1027       gst_message_new_segment_done (GST_OBJECT (nuv), GST_FORMAT_TIME, -1));

  1029   if (nuv->priv->src_video_pad)

  1030     gst_pad_push_event (nuv->priv->src_video_pad, gst_event_new_eos ());

  1031   if (nuv->priv->src_audio_pad)

  1032     gst_pad_push_event (nuv->priv->src_audio_pad, gst_event_new_eos ());

  1033 }

  1035 static GstFlowReturn

  1036 gst_nuv_demux_read_bytes (GstNuvDemux * nuv, guint64 size, gboolean move,

  1037     GstBuffer ** buffer)

  1038 {

  1039   GstFlowReturn ret = GST_FLOW_OK;

  1041   if (size == 0) {

  1042     return ret;

  1043   }

  1045   if (nuv->priv->mode == NUV_PULL_MODE) {

  1046     ret = gst_pad_pull_range (nuv->priv->sinkpad, nuv->priv->offset, size, buffer);

  1047     if (ret == GST_FLOW_OK) {

  1048         if (move) {

  1049 		    nuv->priv->offset += size;

  1050 		}

  1051       /* got eos */

  1052     } else if (ret == GST_FLOW_UNEXPECTED) {

  1053       if (buffer != NULL)

  1054           gst_buffer_unref (buffer);

  1056       gst_nuv_demux_send_eos (nuv);

  1057       return GST_FLOW_WRONG_STATE;

  1058     }

  1059   } else {

  1060     if (gst_adapter_available (nuv->priv->adapter) < size) {

  1061       nuv->priv->more_data = TRUE;

  1062       return GST_FLOW_ERROR_NO_DATA;

  1063     }

  1064     if (move) {

  1065       *buffer = gst_adapter_take_buffer (nuv->priv->adapter, size);

  1066     } else {

  1067       guint8 *data = NULL;

  1068       data = (guint8 *) gst_adapter_peek (nuv->priv->adapter, size);

  1069       *buffer = gst_buffer_new ();

  1070       gst_buffer_set_data (*buffer, data, size);

  1071     }

  1072   }

  1073   return ret;

  1074 }

  1076 static gboolean

  1077 gst_nuv_demux_sink_activate (GstPad * sinkpad)

  1078 {

  1079   gboolean res = TRUE;

  1080   GstNuvDemux *nuv = GST_NUV_DEMUX (gst_pad_get_parent (sinkpad));

  1082   if (gst_pad_check_pull_range (sinkpad)) {

  1083     if (nuv->priv->adapter) {

  1084       gst_adapter_clear (nuv->priv->adapter);

  1085       g_object_unref (nuv->priv->adapter);

  1086       nuv->priv->adapter = NULL;

  1087     }

  1088     res = gst_pad_activate_pull (sinkpad, TRUE);

  1089   } else {

  1090     if (nuv->priv->adapter) {

  1091       gst_adapter_clear (nuv->priv->adapter);

  1092     } else {

  1093       nuv->priv->adapter = gst_adapter_new ();

  1094     }

  1095     res = gst_pad_activate_push (sinkpad, TRUE);

  1096   }

  1098   g_object_unref (nuv);

  1099   return res;

  1100 }

  1102 static gboolean

  1103 gst_nuv_demux_sink_activate_pull (GstPad * sinkpad, gboolean active)

  1104 {

  1105   GstNuvDemux *nuv = GST_NUV_DEMUX (gst_pad_get_parent (sinkpad));

  1107   if (active) {

  1108     GST_DEBUG_OBJECT (nuv, "activating pull function");

  1109     nuv->priv->mode = NUV_PULL_MODE;

  1110     if (nuv->priv->adapter) {

  1111       gst_adapter_clear (nuv->priv->adapter);

  1112       g_object_unref (nuv->priv->adapter);

  1113       nuv->priv->adapter = NULL;

  1114     }

  1115     gst_pad_start_task (sinkpad, (GstTaskFunction) gst_nuv_demux_loop, sinkpad);

  1116   } else {

  1117     GST_DEBUG_OBJECT (nuv, "deactivating pull function");

  1118     gst_pad_stop_task (sinkpad);

  1119   }

  1120   gst_object_unref (nuv);

  1122   return TRUE;

  1123 }

  1125 static gboolean

  1126 gst_nuv_demux_sink_activate_push (GstPad * pad, gboolean active)

  1127 {

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

  1131   if (active) {

  1132     GST_DEBUG_OBJECT (nuv, "activating push/chain function");

  1133   } else {

  1134     GST_DEBUG_OBJECT (nuv, "deactivating push/chain function");

  1135   }

  1137   gst_object_unref (nuv);

  1139   return TRUE;

  1140 }

  1143 static gboolean

  1144 gst_nuv_demux_sink_event (GstPad *pad, GstEvent *event)

  1145 {

  1146   gboolean res;

  1147   GstNuvDemux *nuv;

  1149   nuv = GST_NUV_DEMUX (gst_pad_get_parent (pad));

  1151   switch (GST_EVENT_TYPE (event)) {

  1152     case GST_EVENT_SEEK:

  1153     {

  1154       gdouble rate;

  1155       GstFormat format;

  1156       GstSeekFlags flags;

  1157       GstSeekType cur_type;

  1158       gint64 cur;

  1159       GstSeekType stop_type;

  1160       gint64 stop;

  1162       gst_event_parse_seek (event, &rate, &format, &flags, &cur_type, &cur, &stop_type, &stop);

  1163       GST_DEBUG_OBJECT (nuv, "got seek, start %" G_GINT64_FORMAT ", stop %" G_GINT64_FORMAT, cur, stop);

  1164       break;

  1165     }

  1166     case GST_EVENT_NEWSEGMENT:

  1167     {

  1168       gint64 start, stop, time;

  1169       gdouble rate, arate;

  1170       GstFormat format;

  1171       gboolean update;

  1173       GST_DEBUG_OBJECT (nuv, "got a new segment event");

  1174       gst_event_parse_new_segment (event, &update, &rate, &format, &start, &stop, &time);

  1176       GST_DEBUG_OBJECT (nuv, "got newsegment, start %" G_GINT64_FORMAT ", stop %" G_GINT64_FORMAT, start, stop);

  1177       nuv->priv->duration_bytes = stop - start;

  1178       gst_event_unref (event);

  1179       res = TRUE;

  1180       break;

  1181     }

  1182     case GST_EVENT_QOS:

  1183     {

  1184 /*

  1185       gdouble proportion;

  1186       GstClockTimeDiff diff;

  1187       GstClockTime timestamp;

  1189       gst_event_parse_qos (event, &proportion, &diff, &timestamp);

  1190       if (diff > 0)

  1191         nuv->time_qos = timecode + diff;

  1192       else

  1193         nuv->time_qos = -1;

  1194 */

  1195       break;

  1196     }

  1198     case GST_EVENT_EOS:

  1199     default:

  1200       res = gst_pad_event_default (pad, event);

  1201       break;

  1202   }

  1204   gst_object_unref (nuv);

  1205   return res;

  1206 }

  1208 static GstFlowReturn

  1209 gst_nuv_demux_chain (GstPad * pad, GstBuffer * buf)

  1210 {

  1211   GstFlowReturn ret = GST_FLOW_OK;

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

  1215   gst_adapter_push (nuv->priv->adapter, buf);

  1217   while ((ret == GST_FLOW_OK) && (nuv->priv->more_data == FALSE)) {

  1218       ret = gst_nuv_demux_play (pad);

  1219   }

  1221   nuv->priv->more_data = FALSE;

  1222   gst_object_unref (nuv);

  1224   return ret;

  1225 }

  1227 static void

  1228 gst_nuv_demux_loop (GstPad * pad)

  1229 {

  1230   gst_nuv_demux_play (pad);

  1231 }

  1233 static void

  1234 gst_nuv_demux_reset (GstNuvDemux * nuv)

  1235 {

  1236   nuv->priv->more_data = FALSE;

  1237   nuv->priv->state = GST_NUV_DEMUX_START;

  1238   nuv->priv->mode = 0;

  1239   nuv->priv->offset = 0;

  1240   nuv->priv->streamer_offset = 0;

  1241   nuv->priv->time_start = 0;

  1242   nuv->priv->time_qos = GST_CLOCK_TIME_NONE;

  1243   nuv->priv->duration_bytes = GST_CLOCK_TIME_NONE;

  1244   nuv->priv->duration_time = GST_CLOCK_TIME_NONE;

  1245   nuv->priv->duration_average = GST_CLOCK_TIME_NONE;

  1246   gst_segment_init (&nuv->priv->segment, GST_FORMAT_TIME);

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

  1249     gst_adapter_clear (nuv->priv->adapter);

  1250   }

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

  1253     gst_buffer_unref (nuv->priv->mpeg_buffer);

  1254     nuv->priv->mpeg_buffer = NULL;

  1255   }

  1256 }

  1258 static void

  1259 gst_nuv_demux_destoy_src_pad (GstNuvDemux * nuv)

  1260 {

  1261   if (nuv->priv->src_video_pad) {

  1262     gst_element_remove_pad (GST_ELEMENT (nuv), nuv->priv->src_video_pad);

  1263     nuv->priv->src_video_pad = NULL;

  1264   }

  1266   if (nuv->priv->src_audio_pad) {

  1267     gst_element_remove_pad (GST_ELEMENT (nuv), nuv->priv->src_audio_pad);

  1268     nuv->priv->src_audio_pad = NULL;

  1269   }

  1270 }

  1272 static GstStateChangeReturn

  1273 gst_nuv_demux_change_state (GstElement * element, GstStateChange transition)

  1274 {

  1275   GstStateChangeReturn ret = GST_STATE_CHANGE_SUCCESS;

  1277   switch (transition) {

  1278     case GST_STATE_CHANGE_NULL_TO_READY:

  1279       gst_nuv_demux_reset (GST_NUV_DEMUX (element));

  1280       gst_nuv_demux_destoy_src_pad (GST_NUV_DEMUX (element));

  1281       break;

  1282     default:

  1283       break;

  1284   }

  1286   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);

  1287   if (ret == GST_STATE_CHANGE_FAILURE) {

  1288     goto done;

  1289   }

  1291   switch (transition) {

  1292     case GST_STATE_CHANGE_READY_TO_NULL:

  1293       gst_nuv_demux_reset (GST_NUV_DEMUX (element));

  1294       gst_nuv_demux_destoy_src_pad (GST_NUV_DEMUX (element));

  1295       break;

  1296     default:

  1297       break;

  1298   }

  1300 done:

  1301   return ret;

  1302 }

  1304 static void

  1305 gst_nuv_demux_update_duration (GstNuvDemux *nuv, guint64 current_timestamp)

  1306 {

  1307     guint64 interval = 0;

  1309     if (gst_nuv_demux_get_time_duration (nuv) != GST_CLOCK_TIME_NONE)

  1310         return;

  1312     interval  = current_timestamp - nuv->priv->last_update;

  1314     if (interval > (10 * GST_SECOND)) {

  1315         GstMessage* msg = NULL;

  1316         gint64 average = 0;

  1317         gint64 duration_bytes = gst_nuv_demux_get_bytes_duration (nuv);

  1319         if (duration_bytes == GST_CLOCK_TIME_NONE) {

  1320             return;

  1321         }

  1323         interval = gst_util_uint64_scale (1, current_timestamp - nuv->priv->time_start, GST_SECOND);

  1324         if (interval > 0) {

  1325             average = gst_util_uint64_scale (1 , nuv->priv->streamer_offset, interval);

  1326         }

  1327         if (average > 0) {

  1328             nuv->priv->duration_average = gst_util_uint64_scale (GST_SECOND, duration_bytes, average);

  1329         }

  1331         nuv->priv->last_update = current_timestamp;

  1332         msg = gst_message_new_duration (GST_OBJECT (nuv), GST_FORMAT_TIME, nuv->priv->duration_average);

  1333         gst_element_post_message (GST_ELEMENT (nuv), msg);

  1334         GST_DEBUG_OBJECT (nuv, "New Duration Average %"G_GUINT64_FORMAT, nuv->priv->duration_average);

  1335         g_debug ("New Duration Average %"G_GUINT64_FORMAT, nuv->priv->duration_average);

  1336     }

  1337 }

  1339 static gint64

  1340 gst_nuv_demux_get_bytes_duration (GstNuvDemux *nuv)

  1341 {

  1342     if (nuv->priv->duration_bytes == GST_CLOCK_TIME_NONE) {

  1343         GstPad *peer = gst_pad_get_peer (nuv->priv->sinkpad);

  1344         GstQuery *query = gst_query_new_duration (GST_FORMAT_BYTES);

  1345         if (gst_pad_query (peer, query)) {

  1346             gint64 duration;

  1348             gst_query_parse_duration (query, NULL, &duration);

  1349             nuv->priv->duration_bytes = duration;

  1350         }

  1351         gst_object_unref (peer);

  1352         gst_query_unref (query);

  1353     }

  1354     return nuv->priv->duration_bytes;

  1355 }

  1357 static gint64

  1358 gst_nuv_demux_get_time_duration (GstNuvDemux *nuv)

  1359 {

  1360     if (nuv->priv->duration_time == GST_CLOCK_TIME_NONE) {

  1361         GstPad *peer = gst_pad_get_peer (nuv->priv->sinkpad);

  1362         GstQuery *query = gst_query_new_duration (GST_FORMAT_TIME);

  1363         if (gst_pad_query (peer, query)) {

  1364             gint64 duration;

  1365             gst_query_parse_duration (query, NULL, &duration);

  1366             nuv->priv->duration_time = duration;

  1367         }

  1368         gst_object_unref (peer);

  1369         gst_query_unref (query);

  1370     }

  1371     return nuv->priv->duration_time;

  1372 }

  1374 #if (GST_VERSION_MINOR == 10) && (GST_VERSION_MICRO < 6)

  1375 GstBuffer *

  1376 gst_adapter_take_buffer (GstAdapter * adapter, guint nbytes)

  1377 {

  1378   GstBuffer *buffer;

  1379   GstBuffer *cur;

  1380   guint8 *data;

  1382   g_return_val_if_fail (GST_IS_ADAPTER (adapter), NULL);

  1383   g_return_val_if_fail (nbytes > 0, NULL);

  1385   GST_LOG_OBJECT (adapter, "taking buffer of %u bytes", nbytes);

  1387   /* we don't have enough data, return NULL. This is unlikely

  1388    * as one usually does an _available() first instead of peeking a

  1389    * random size. */

  1390   if (G_UNLIKELY (nbytes > adapter->size))

  1391     return NULL;

  1393   /* our head buffer has enough data left, return it */

  1394   cur = adapter->buflist->data;

  1395   if (GST_BUFFER_SIZE (cur) >= nbytes + adapter->skip) {

  1396     GST_LOG_OBJECT (adapter, "providing buffer of %d bytes via sub-buffer",

  1397         nbytes);

  1398     buffer = gst_buffer_create_sub (cur, adapter->skip, nbytes);

  1400     gst_adapter_flush (adapter, nbytes);

  1402     return buffer;

  1403   }

  1405   data = gst_adapter_take (adapter, nbytes);

  1406   if (data == NULL)

  1407     return NULL;

  1409   buffer = gst_buffer_new ();

  1410   GST_BUFFER_DATA (buffer) = data;

  1411   GST_BUFFER_MALLOCDATA (buffer) = data;

  1412   GST_BUFFER_SIZE (buffer) = nbytes;

  1414   return buffer;

  1415 }

  1416 #endif

  1419 static gboolean

  1420 plugin_init (GstPlugin * plugin)

  1421 {

  1422 #ifdef ENABLE_NLS

  1423   setlocale (LC_ALL, "");

  1424   bindtextdomain (GETTEXT_PACKAGE, LOCALEDIR);

  1425 #endif /* ENABLE_NLS */

  1427   if (!gst_element_register (plugin, "nuvdemux", GST_RANK_SECONDARY,

  1428           GST_TYPE_NUV_DEMUX)) {

  1429     return FALSE;

  1430   }

  1431   return TRUE;

  1432 }

  1434 GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,

  1435     GST_VERSION_MINOR,

  1436     "nuvdemux",

  1437     "Demuxes and muxes audio and video",

  1438      plugin_init, VERSION, "LGPL", "NuvDemux", "")

author	renatofilho
	Mon Dec 18 18:18:26 2006 +0000 (2006-12-18)
branch	trunk
changeset 222	803672c406ec
parent 209	b8a07bf0375b
child 251	9f0346db9a46
permissions	-rw-r--r--