renatofilho@608: /* GStreamer renatofilho@608: * Copyright (C) <2006> Renato Araujo Oliveira Filho renatofilho@608: * Rosfran Borges renatofilho@608: * renatofilho@608: * This library is free software; you can redistribute it and/or renatofilho@608: * modify it under the terms of the GNU Library General Public renatofilho@608: * License as published by the Free Software Foundation; either renatofilho@608: * version 2 of the License, or (at your option) any later version. renatofilho@608: * renatofilho@608: * This library is distributed in the hope that it will be useful, renatofilho@608: * but WITHOUT ANY WARRANTY; without even the implied warranty of renatofilho@608: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU renatofilho@608: * Library General Public License for more details. renatofilho@608: * renatofilho@608: * You should have received a copy of the GNU Library General Public renatofilho@608: * License along with this library; if not, write to the renatofilho@608: * Free Software Foundation, Inc., 59 Temple Place - Suite 330, renatofilho@608: * Boston, MA 02111-1307, USA. renatofilho@608: */ renatofilho@608: /* Element-Checklist-Version: 5 */ renatofilho@608: renatofilho@608: /** renatofilho@608: * SECTION:element-nuvdemux renatofilho@608: * renatofilho@608: * renatofilho@608: * renatofilho@608: * Demuxes an .nuv file into raw or compressed audio and/or video streams. renatofilho@608: * renatofilho@608: * renatofilho@608: * This element currently only supports pull-based scheduling. renatofilho@608: * renatofilho@608: * Example launch line renatofilho@608: * renatofilho@608: * renatofilho@608: * gst-launch filesrc test.nuv ! nuvdemux name=demux demux.audio_00 ! decodebin ! audioconvert ! audioresample ! autoaudiosink demux.video_00 ! queue ! decodebin ! ffmpegcolorspace ! videoscale ! autovideosink renatofilho@608: * renatofilho@608: * Play (parse and decode) an .nuv file and try to output it to renatofilho@608: * an automatically detected soundcard and videosink. If the NUV file contains renatofilho@608: * compressed audio or video data, this will only work if you have the renatofilho@608: * right decoder elements/plugins installed. renatofilho@608: * renatofilho@608: * renatofilho@608: * renatofilho@608: */ renatofilho@608: renatofilho@608: #ifdef HAVE_CONFIG_H renatofilho@608: #include "config.h" renatofilho@608: #endif renatofilho@608: renatofilho@608: #include renatofilho@608: #include renatofilho@608: #include renatofilho@608: #include renatofilho@608: #include renatofilho@608: renatofilho@608: #include "glib/gi18n.h" renatofilho@608: #include "gstnuvdemux.h" renatofilho@608: renatofilho@608: #define GST_NUV_DEMUX_GET_PRIVATE(obj) (G_TYPE_INSTANCE_GET_PRIVATE ((obj), GST_TYPE_NUV_DEMUX, GstNuvDemuxPrivate)) renatofilho@608: renatofilho@751: GST_DEBUG_CATEGORY_STATIC(nuvdemux_debug); renatofilho@608: #define GST_CAT_DEFAULT nuvdemux_debug renatofilho@608: #define GST_FLOW_ERROR_NO_DATA -101 renatofilho@608: #define GST_FLOW_ERROR_EOS -102 renatofilho@608: renatofilho@608: enum renatofilho@608: { renatofilho@751: NUV_PUSH_MODE = 0, renatofilho@751: NUV_PULL_MODE renatofilho@608: }; renatofilho@608: renatofilho@751: GST_DEBUG_CATEGORY_EXTERN(GST_CAT_EVENT); renatofilho@608: renatofilho@608: static const GstElementDetails gst_nuv_demux_details = renatofilho@751: GST_ELEMENT_DETAILS("Nuv demuxer", renatofilho@751: "Codec/Demuxer", renatofilho@751: "Demultiplex a .nuv file into audio and video", renatofilho@751: "Renato Araujo Oliveira Filho ," renatofilho@751: "Rosfran Borges "); renatofilho@608: renatofilho@608: renatofilho@608: /* file header */ renatofilho@608: typedef struct renatofilho@608: { renatofilho@751: gchar id[12]; /* "NuppelVideo\0" or "MythTVVideo\0" */ renatofilho@751: gchar version[5]; /* "x.xx\0" */ renatofilho@608: renatofilho@751: gint i_width; renatofilho@751: gint i_height; renatofilho@751: gint i_width_desired; renatofilho@751: gint i_height_desired; renatofilho@608: renatofilho@751: gchar i_mode; /* P progressive, I interlaced */ renatofilho@608: renatofilho@751: gdouble d_aspect; /* 1.0 squared pixel */ renatofilho@751: gdouble d_fps; renatofilho@751: //fps num/denom renatofilho@751: gint i_fpsn; renatofilho@751: gint i_fpsd; renatofilho@608: renatofilho@751: gint i_video_blocks; /* 0 no video, -1 unknown */ renatofilho@751: gint i_audio_blocks; renatofilho@751: gint i_text_blocks; renatofilho@608: renatofilho@751: gint i_keyframe_distance; renatofilho@608: renatofilho@608: } nuv_header; renatofilho@608: renatofilho@608: /* frame header */ renatofilho@608: typedef struct renatofilho@608: { renatofilho@751: gchar i_type; /* A: audio, V: video, S: sync; T: test renatofilho@751: R: Seekpoint (string:RTjjjjjjjj) renatofilho@751: D: Extra data for codec */ renatofilho@751: gchar i_compression; /* V: 0 uncompressed renatofilho@751: 1 RTJpeg renatofilho@751: 2 RTJpeg+lzo renatofilho@751: N black frame renatofilho@751: L copy last renatofilho@751: A: 0 uncompressed (44100 1-bits, 2ch) renatofilho@751: 1 lzo renatofilho@751: 2 layer 2 renatofilho@751: 3 layer 3 renatofilho@751: F flac renatofilho@751: S shorten renatofilho@751: N null frame loudless renatofilho@751: L copy last renatofilho@751: S: B audio and vdeo sync point renatofilho@751: A audio sync info (timecode == effective renatofilho@751: dsp frequency*100) renatofilho@751: V next video sync (timecode == next video renatofilho@751: frame num) renatofilho@751: S audio,video,text correlation */ renatofilho@751: gchar i_keyframe; /* 0 keyframe, else no no key frame */ renatofilho@751: guint8 i_filters; /* 0x01: gauss 5 pixel (8,2,2,2,2)/16 renatofilho@751: 0x02: gauss 5 pixel (8,1,1,1,1)/12 renatofilho@751: 0x04: cartoon filter */ renatofilho@608: renatofilho@751: gint32 i_timecode; /* ms */ renatofilho@608: renatofilho@751: gint i_length; /* V,A,T: length of following data renatofilho@751: S: length of packet correl */ renatofilho@608: } nuv_frame_header; renatofilho@608: renatofilho@608: renatofilho@608: /* FIXME Not sure of this one */ renatofilho@608: typedef struct renatofilho@608: { renatofilho@751: gint i_version; renatofilho@751: guint32 i_video_fcc; renatofilho@608: renatofilho@751: guint32 i_audio_fcc; renatofilho@751: gint i_audio_sample_rate; renatofilho@751: gint i_audio_bits_per_sample; renatofilho@751: gint i_audio_channels; renatofilho@751: gint i_audio_compression_ratio; renatofilho@751: gint i_audio_quality; renatofilho@751: gint i_rtjpeg_quality; renatofilho@751: gint i_rtjpeg_luma_filter; renatofilho@751: gint i_rtjpeg_chroma_filter; renatofilho@751: gint i_lavc_bitrate; renatofilho@751: gint i_lavc_qmin; renatofilho@751: gint i_lavc_qmax; renatofilho@751: gint i_lavc_maxqdiff; renatofilho@751: gint64 i_seekable_offset; renatofilho@751: gint64 i_keyframe_adjust_offset; renatofilho@608: renatofilho@608: } nuv_extended_header; renatofilho@608: renatofilho@608: typedef struct renatofilho@608: { renatofilho@608: gint64 timecode; renatofilho@608: gint64 offset; renatofilho@608: renatofilho@608: } frame_index_data; renatofilho@608: renatofilho@751: typedef enum renatofilho@751: { renatofilho@608: GST_NUV_DEMUX_START, renatofilho@608: GST_NUV_DEMUX_HEADER_DATA, renatofilho@608: GST_NUV_DEMUX_EXTRA_DATA, renatofilho@608: GST_NUV_DEMUX_MPEG_DATA, renatofilho@608: GST_NUV_DEMUX_EXTEND_HEADER, renatofilho@608: GST_NUV_DEMUX_EXTEND_HEADER_DATA, renatofilho@608: GST_NUV_DEMUX_INDEX_CREATE, renatofilho@608: GST_NUV_DEMUX_FRAME_HEADER, renatofilho@608: GST_NUV_DEMUX_MOVI, renatofilho@608: GST_NUV_DEMUX_INVALID_DATA renatofilho@608: } GstNuvDemuxState; renatofilho@608: renatofilho@751: struct _GstNuvDemuxPrivate renatofilho@751: { renatofilho@751: /* used for indicate the mode */ renatofilho@751: guint mode; renatofilho@608: renatofilho@751: /* used on push mode */ renatofilho@751: GstAdapter *adapter; renatofilho@608: renatofilho@751: /* pads */ renatofilho@751: GstPad *sinkpad; renatofilho@751: GstPad *src_video_pad; renatofilho@751: GstPad *src_audio_pad; renatofilho@608: renatofilho@751: /* Flow control */ renatofilho@751: GstFlowReturn last_video_return; renatofilho@751: GstFlowReturn last_audio_return; renatofilho@751: gboolean more_data; renatofilho@751: gboolean eos; renatofilho@751: gboolean new_file; renatofilho@751: guint segment; renatofilho@608: renatofilho@751: /* NUV decoding state */ renatofilho@751: GstNuvDemuxState state; renatofilho@751: guint64 offset; renatofilho@608: renatofilho@751: /* duration information */ renatofilho@751: guint64 duration_bytes; renatofilho@751: guint64 duration_time; renatofilho@751: guint64 segment_stop; renatofilho@751: guint64 segment_start; renatofilho@608: renatofilho@751: /* segment control info */ renatofilho@751: gboolean new_audio_segment; renatofilho@751: gboolean new_video_segment; renatofilho@608: renatofilho@751: /* Mpeg ExtraData */ renatofilho@751: guint64 mpeg_data_size; renatofilho@751: GstBuffer *mpeg_buffer; renatofilho@608: renatofilho@751: /* Headers */ renatofilho@608: nuv_header h; renatofilho@608: nuv_extended_header eh; renatofilho@608: nuv_frame_header fh; renatofilho@608: renatofilho@751: /* anothers info */ renatofilho@751: guint64 header_lengh; renatofilho@751: gint64 time_start; renatofilho@751: gint64 time_diff; renatofilho@751: gint64 time_qos; renatofilho@751: guint64 last_frame_time; renatofilho@751: GSList *index; renatofilho@608: }; renatofilho@608: renatofilho@608: renatofilho@751: static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE("sink", renatofilho@751: GST_PAD_SINK, renatofilho@751: GST_PAD_ALWAYS, renatofilho@751: GST_STATIC_CAPS renatofilho@751: ("video/x-nuv")); renatofilho@608: renatofilho@608: static GstStaticPadTemplate audio_src_template = renatofilho@751: GST_STATIC_PAD_TEMPLATE("audio_src", renatofilho@751: GST_PAD_SRC, renatofilho@751: GST_PAD_SOMETIMES, renatofilho@751: GST_STATIC_CAPS_ANY); renatofilho@608: renatofilho@608: static GstStaticPadTemplate video_src_template = renatofilho@751: GST_STATIC_PAD_TEMPLATE("video_src", renatofilho@751: GST_PAD_SRC, renatofilho@751: GST_PAD_SOMETIMES, renatofilho@751: GST_STATIC_CAPS_ANY); renatofilho@608: renatofilho@751: static void gst_nuv_demux_dispose(GObject * object); renatofilho@751: static void gst_nuv_demux_finalize(GObject * object); renatofilho@751: static GstStateChangeReturn gst_nuv_demux_change_state(GstElement * element, renatofilho@751: GstStateChange renatofilho@751: transition); renatofilho@751: static void gst_nuv_demux_loop(GstPad * pad); renatofilho@751: static GstFlowReturn gst_nuv_demux_chain(GstPad * pad, GstBuffer * buf); renatofilho@751: static GstFlowReturn gst_nuv_demux_play(GstPad * pad); renatofilho@751: static gboolean gst_nuv_demux_sink_activate_pull(GstPad * sinkpad, renatofilho@751: gboolean active); renatofilho@751: static gboolean gst_nuv_demux_sink_activate_push(GstPad * pad, renatofilho@751: gboolean active); renatofilho@751: static gboolean gst_nuv_demux_sink_activate(GstPad * sinkpad); renatofilho@751: static gboolean gst_nuv_demux_sink_event(GstPad * pad, GstEvent * event); renatofilho@751: static gboolean gst_nuv_demux_srcpad_event(GstPad * pad, GstEvent * event); renatofilho@751: static frame_index_data *gst_nuv_demux_do_seek_index(GstNuvDemux * nuv, renatofilho@751: gint64 seek_pos, renatofilho@751: gint64 segment_stop, renatofilho@751: GstFormat format); renatofilho@608: renatofilho@608: renatofilho@751: static GstFlowReturn gst_nuv_demux_move_bytes(GstNuvDemux * nuv, renatofilho@751: guint64 size); renatofilho@751: static GstFlowReturn gst_nuv_demux_read_bytes(GstNuvDemux * nuv, guint64 size, renatofilho@751: gboolean move, renatofilho@751: GstBuffer ** buffer); renatofilho@751: static void gst_nuv_demux_reset(GstNuvDemux * nuv); renatofilho@751: static void gst_nuv_demux_destoy_src_pad(GstNuvDemux * nuv); renatofilho@751: static void gst_nuv_demux_send_eos(GstNuvDemux * nuv); renatofilho@751: static void gst_nuv_demux_create_seek_index(GstNuvDemux * nuv); renatofilho@608: renatofilho@608: renatofilho@751: #if (GST_VERSION_MINOR == 10) && (GST_VERSION_MICRO < 6) renatofilho@751: GstBuffer *gst_adapter_take_buffer(GstAdapter * adapter, guint nbytes); renatofilho@608: #endif renatofilho@608: renatofilho@608: renatofilho@751: GST_BOILERPLATE(GstNuvDemux, gst_nuv_demux, GstElement, GST_TYPE_ELEMENT); renatofilho@608: renatofilho@608: /****************************************************************************** renatofilho@608: * Utils function renatofilho@608: ******************************************************************************/ renatofilho@608: #if G_BYTE_ORDER == G_BIG_ENDIAN renatofilho@608: static inline gdouble renatofilho@751: _gdouble_swap_le_be(gdouble * d) renatofilho@608: { renatofilho@608: union renatofilho@608: { renatofilho@751: guint64 i; renatofilho@751: gdouble d; renatofilho@608: } u; renatofilho@608: renatofilho@608: u.d = *d; renatofilho@751: u.i = GUINT64_SWAP_LE_BE(u.i); renatofilho@608: return u.d; renatofilho@608: } renatofilho@608: renatofilho@608: #define READ_DOUBLE_FROM_LE(d) (_gdouble_swap_le_be((gdouble* ) d)) renatofilho@608: #else /* G_BYTE_ORDER != G_BIG_ENDIAN */ renatofilho@608: #define READ_DOUBLE_FROM_LE(d) *((gdouble* ) (d)) renatofilho@608: #endif /* G_BYTE_ORDER != G_BIG_ENDIAN */ renatofilho@608: renatofilho@608: static void renatofilho@751: double2fraction(double in, int *num, int *denom) renatofilho@608: { renatofilho@751: if (in == 29.97) renatofilho@751: { renatofilho@751: *num = 30000; renatofilho@751: *denom = 1001; renatofilho@751: } renatofilho@751: else if (in == 23.976) renatofilho@751: { renatofilho@751: *num = 24000; renatofilho@751: *denom = 1001; renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: *denom = 1; renatofilho@751: while (in - floor(in) >= 0.1) renatofilho@751: { renatofilho@751: *denom *= 10; renatofilho@751: in *= 10.0; renatofilho@751: } renatofilho@751: *num = (int) floor(in); renatofilho@751: } renatofilho@608: } renatofilho@608: renatofilho@608: /* GObject Functions */ renatofilho@608: renatofilho@608: static void renatofilho@751: gst_nuv_demux_base_init(gpointer klass) renatofilho@608: { renatofilho@751: GstElementClass *element_class = GST_ELEMENT_CLASS(klass); renatofilho@608: renatofilho@751: gst_element_class_add_pad_template(element_class, renatofilho@751: gst_static_pad_template_get renatofilho@751: (&audio_src_template)); renatofilho@608: renatofilho@751: gst_element_class_add_pad_template(element_class, renatofilho@751: gst_static_pad_template_get renatofilho@751: (&video_src_template)); renatofilho@608: renatofilho@751: gst_element_class_add_pad_template(element_class, renatofilho@751: gst_static_pad_template_get renatofilho@751: (&sink_template)); renatofilho@751: gst_element_class_set_details(element_class, &gst_nuv_demux_details); renatofilho@608: } renatofilho@608: renatofilho@608: static void renatofilho@751: gst_nuv_demux_class_init(GstNuvDemuxClass * klass) renatofilho@608: { renatofilho@751: GstElementClass *gstelement_class = GST_ELEMENT_CLASS(klass); renatofilho@608: GObjectClass *gobject_class = (GObjectClass *) klass; renatofilho@608: renatofilho@751: GST_DEBUG_CATEGORY_INIT(nuvdemux_debug, "nuvdemux", renatofilho@751: 0, "Demuxer for NUV streams"); renatofilho@608: renatofilho@751: parent_class = g_type_class_peek_parent(klass); renatofilho@608: renatofilho@608: gobject_class->dispose = gst_nuv_demux_dispose; renatofilho@608: gobject_class->finalize = gst_nuv_demux_finalize; renatofilho@608: gstelement_class->change_state = gst_nuv_demux_change_state; renatofilho@608: renatofilho@751: g_type_class_add_private(gobject_class, sizeof(GstNuvDemuxPrivate)); renatofilho@608: } renatofilho@608: renatofilho@608: static void renatofilho@751: gst_nuv_demux_init(GstNuvDemux * nuv, GstNuvDemuxClass * nuv_class) renatofilho@608: { renatofilho@751: nuv->priv = GST_NUV_DEMUX_GET_PRIVATE(nuv); renatofilho@751: nuv->priv->sinkpad = renatofilho@751: gst_pad_new_from_static_template(&sink_template, "sink"); renatofilho@608: renatofilho@751: /* creating adapter */ renatofilho@608: nuv->priv->mode = NUV_PUSH_MODE; renatofilho@751: nuv->priv->adapter = gst_adapter_new(); renatofilho@608: renatofilho@608: nuv->priv->new_audio_segment = TRUE; renatofilho@608: nuv->priv->new_video_segment = TRUE; renatofilho@608: renatofilho@751: gst_pad_set_activate_function(nuv->priv->sinkpad, renatofilho@751: gst_nuv_demux_sink_activate); renatofilho@751: gst_pad_set_activatepull_function(nuv->priv->sinkpad, renatofilho@751: gst_nuv_demux_sink_activate_pull); renatofilho@751: gst_pad_set_activatepush_function(nuv->priv->sinkpad, renatofilho@751: gst_nuv_demux_sink_activate_push); renatofilho@751: gst_pad_set_chain_function(nuv->priv->sinkpad, renatofilho@751: GST_DEBUG_FUNCPTR(gst_nuv_demux_chain)); renatofilho@751: gst_pad_set_event_function(nuv->priv->sinkpad, renatofilho@751: GST_DEBUG_FUNCPTR(gst_nuv_demux_sink_event)); renatofilho@714: renatofilho@608: renatofilho@751: gst_element_add_pad(GST_ELEMENT(nuv), nuv->priv->sinkpad); renatofilho@608: renatofilho@608: } renatofilho@608: renatofilho@608: static void renatofilho@751: gst_nuv_demux_dispose(GObject * object) renatofilho@608: { renatofilho@751: GstNuvDemux *nuv = GST_NUV_DEMUX(object); renatofilho@608: renatofilho@608: renatofilho@751: if (nuv->priv->mpeg_buffer != NULL) renatofilho@751: { renatofilho@751: gst_buffer_unref(nuv->priv->mpeg_buffer); renatofilho@751: } renatofilho@608: renatofilho@751: gst_nuv_demux_reset(GST_NUV_DEMUX(object)); renatofilho@751: gst_nuv_demux_destoy_src_pad(GST_NUV_DEMUX(object)); renatofilho@608: renatofilho@751: if (nuv->priv->adapter != NULL) renatofilho@751: { renatofilho@751: gst_object_unref(nuv->priv->adapter); renatofilho@751: } renatofilho@608: } renatofilho@608: renatofilho@608: static void renatofilho@751: gst_nuv_demux_finalize(GObject * object) renatofilho@608: { renatofilho@751: G_OBJECT_CLASS(parent_class)->finalize(object); renatofilho@608: } renatofilho@608: renatofilho@608: renatofilho@608: /* HeaderLoad: renatofilho@608: */ renatofilho@608: static GstFlowReturn renatofilho@751: gst_nuv_demux_header_load(GstNuvDemux * nuv, nuv_header * h) renatofilho@608: { renatofilho@608: GstBuffer *buffer = NULL; renatofilho@751: GstFlowReturn res = gst_nuv_demux_read_bytes(nuv, 72, TRUE, &buffer); renatofilho@608: renatofilho@751: if ((res != GST_FLOW_OK) || (buffer == NULL)) renatofilho@751: { renatofilho@751: goto done; renatofilho@751: } renatofilho@608: renatofilho@751: if (h != NULL) renatofilho@751: { renatofilho@751: memcpy(h->id, buffer->data, 12); renatofilho@751: memcpy(h->version, buffer->data + 12, 5); renatofilho@751: h->i_width = GST_READ_UINT32_LE(&buffer->data[20]); renatofilho@751: h->i_height = GST_READ_UINT32_LE(&buffer->data[24]); renatofilho@751: h->i_width_desired = GST_READ_UINT32_LE(&buffer->data[28]); renatofilho@751: h->i_height_desired = GST_READ_UINT32_LE(&buffer->data[32]); renatofilho@751: h->i_mode = GPOINTER_TO_INT(buffer->data[36]); renatofilho@751: h->d_aspect = READ_DOUBLE_FROM_LE(&buffer->data[40]); renatofilho@751: h->d_fps = READ_DOUBLE_FROM_LE(&buffer->data[48]); renatofilho@751: /* get the num and denom values from fps */ renatofilho@751: double2fraction(h->d_fps, &h->i_fpsn, &h->i_fpsd); renatofilho@751: h->i_video_blocks = GST_READ_UINT32_LE(&buffer->data[56]); renatofilho@751: h->i_audio_blocks = GST_READ_UINT32_LE(&buffer->data[60]); renatofilho@751: h->i_text_blocks = GST_READ_UINT32_LE(&buffer->data[64]); renatofilho@751: h->i_keyframe_distance = GST_READ_UINT32_LE(&buffer->data[68]); renatofilho@608: renatofilho@751: GST_DEBUG_OBJECT(nuv, renatofilho@751: "nuv: h=%s v=%s %dx%d a=%f fps=%f v=%d a=%d t=%d kfd=%d", renatofilho@751: h->id, h->version, h->i_width, h->i_height, renatofilho@751: h->d_aspect, h->d_fps, h->i_video_blocks, renatofilho@751: h->i_audio_blocks, h->i_text_blocks, renatofilho@751: h->i_keyframe_distance); renatofilho@751: } renatofilho@608: renatofilho@608: done: renatofilho@751: if (buffer != NULL) renatofilho@751: { renatofilho@751: gst_buffer_unref(buffer); renatofilho@751: buffer = NULL; renatofilho@751: } renatofilho@608: return res; renatofilho@608: } renatofilho@608: renatofilho@608: static GstFlowReturn renatofilho@751: gst_nuv_demux_stream_header_data(GstNuvDemux * nuv) renatofilho@608: { renatofilho@608: GstFlowReturn res; renatofilho@608: renatofilho@751: if (nuv->priv->new_file) renatofilho@751: res = gst_nuv_demux_header_load(nuv, NULL); renatofilho@714: else renatofilho@751: res = gst_nuv_demux_header_load(nuv, &nuv->priv->h); renatofilho@714: renatofilho@608: if (res == GST_FLOW_OK) renatofilho@751: nuv->priv->state = GST_NUV_DEMUX_EXTRA_DATA; renatofilho@608: return res; renatofilho@608: } renatofilho@608: renatofilho@608: /* renatofilho@608: * Read NUV file tag renatofilho@608: */ renatofilho@608: static GstFlowReturn renatofilho@751: gst_nuv_demux_stream_file_header(GstNuvDemux * nuv) renatofilho@608: { renatofilho@608: GstFlowReturn res = GST_FLOW_OK; renatofilho@608: GstBuffer *file_header = NULL; renatofilho@608: renatofilho@751: res = gst_nuv_demux_read_bytes(nuv, 12, FALSE, &file_header); renatofilho@751: if (res == GST_FLOW_OK) renatofilho@751: { renatofilho@751: if (strncmp((gchar *) file_header->data, "MythTVVideo", 11) || renatofilho@751: strncmp((gchar *) file_header->data, "NuppelVideo", 11)) renatofilho@751: { renatofilho@751: nuv->priv->state = GST_NUV_DEMUX_HEADER_DATA; renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: GST_DEBUG_OBJECT(nuv, "error parsing file header"); renatofilho@751: nuv->priv->state = GST_NUV_DEMUX_INVALID_DATA; renatofilho@751: res = GST_FLOW_ERROR; renatofilho@751: } renatofilho@751: } renatofilho@608: renatofilho@751: if (file_header != NULL) renatofilho@751: { renatofilho@751: gst_buffer_unref(file_header); renatofilho@751: file_header = NULL; renatofilho@751: } renatofilho@608: return res; renatofilho@608: } renatofilho@608: renatofilho@608: /* FrameHeaderLoad: renatofilho@608: */ renatofilho@608: static GstFlowReturn renatofilho@751: gst_nuv_demux_frame_header_load(GstNuvDemux * nuv, nuv_frame_header * h) renatofilho@608: { renatofilho@608: unsigned char *data; renatofilho@608: GstBuffer *buf = NULL; renatofilho@608: renatofilho@751: GstFlowReturn res = gst_nuv_demux_read_bytes(nuv, 12, TRUE, &buf); renatofilho@608: renatofilho@751: if ((res != GST_FLOW_OK) || (buf == NULL)) renatofilho@751: { renatofilho@751: goto done; renatofilho@751: } renatofilho@608: renatofilho@714: if (h == NULL) renatofilho@714: goto done; renatofilho@751: renatofilho@608: data = buf->data; renatofilho@608: renatofilho@751: h->i_type = GPOINTER_TO_INT(data[0]); renatofilho@751: h->i_compression = GPOINTER_TO_INT(data[1]); renatofilho@751: h->i_keyframe = GPOINTER_TO_INT(data[2]); renatofilho@751: h->i_filters = GPOINTER_TO_INT(data[3]); renatofilho@751: h->i_timecode = GST_READ_UINT32_LE(&data[4]); renatofilho@751: h->i_length = GST_READ_UINT32_LE(&data[8]); renatofilho@608: renatofilho@751: GST_DEBUG_OBJECT(nuv, "frame hdr: t=%c c=%c k=%d f=0x%x timecode=%d l=%d", renatofilho@751: h->i_type, renatofilho@751: h->i_compression ? h->i_compression : ' ', renatofilho@751: h->i_keyframe ? h->i_keyframe : ' ', renatofilho@751: h->i_filters, h->i_timecode, h->i_length); renatofilho@608: renatofilho@751: done: renatofilho@751: if (buf != NULL) renatofilho@751: { renatofilho@751: gst_buffer_unref(buf); renatofilho@751: buf = NULL; renatofilho@751: } renatofilho@608: renatofilho@608: return res; renatofilho@608: } renatofilho@608: renatofilho@608: static GstFlowReturn renatofilho@751: gst_nuv_demux_extended_header_load(GstNuvDemux * nuv, nuv_extended_header * h) renatofilho@608: { renatofilho@608: unsigned char *data; renatofilho@608: GstBuffer *buff = NULL; renatofilho@608: renatofilho@751: GstFlowReturn res = gst_nuv_demux_read_bytes(nuv, 512, TRUE, &buff); renatofilho@608: renatofilho@751: if ((res != GST_FLOW_OK) || (buff == NULL)) renatofilho@751: { renatofilho@751: goto done; renatofilho@751: } renatofilho@608: renatofilho@714: if (h == NULL) renatofilho@751: goto done; renatofilho@714: renatofilho@608: data = buff->data; renatofilho@751: h->i_version = GST_READ_UINT32_LE(&data[0]); renatofilho@751: h->i_video_fcc = GST_MAKE_FOURCC(data[4], data[5], data[6], data[7]); renatofilho@751: h->i_audio_fcc = GST_MAKE_FOURCC(data[8], data[9], data[10], data[11]); renatofilho@751: h->i_audio_sample_rate = GST_READ_UINT32_LE(&data[12]); renatofilho@751: h->i_audio_bits_per_sample = GST_READ_UINT32_LE(&data[16]); renatofilho@751: h->i_audio_channels = GST_READ_UINT32_LE(&data[20]); renatofilho@751: h->i_audio_compression_ratio = GST_READ_UINT32_LE(&data[24]); renatofilho@751: h->i_audio_quality = GST_READ_UINT32_LE(&data[28]); renatofilho@751: h->i_rtjpeg_quality = GST_READ_UINT32_LE(&data[32]); renatofilho@751: h->i_rtjpeg_luma_filter = GST_READ_UINT32_LE(&data[36]); renatofilho@751: h->i_rtjpeg_chroma_filter = GST_READ_UINT32_LE(&data[40]); renatofilho@751: h->i_lavc_bitrate = GST_READ_UINT32_LE(&data[44]); renatofilho@751: h->i_lavc_qmin = GST_READ_UINT32_LE(&data[48]); renatofilho@751: h->i_lavc_qmin = GST_READ_UINT32_LE(&data[52]); renatofilho@751: h->i_lavc_maxqdiff = GST_READ_UINT32_LE(&data[56]); renatofilho@751: h->i_seekable_offset = GST_READ_UINT64_LE(&data[60]); renatofilho@751: h->i_keyframe_adjust_offset = GST_READ_UINT64_LE(&data[68]); renatofilho@608: renatofilho@751: GST_DEBUG_OBJECT(nuv, renatofilho@751: "ex hdr: v=%d vffc=%4.4s afcc=%4.4s %dHz %dbits ach=%d acr=%d aq=%d" renatofilho@751: "rtjpeg q=%d lf=%d lc=%d lavc br=%d qmin=%d qmax=%d maxqdiff=%d seekableoff=%lld keyfao=%lld", renatofilho@751: h->i_version, (gchar *) & h->i_video_fcc, renatofilho@751: (gchar *) & h->i_audio_fcc, h->i_audio_sample_rate, renatofilho@751: h->i_audio_bits_per_sample, h->i_audio_channels, renatofilho@751: h->i_audio_compression_ratio, h->i_audio_quality, renatofilho@751: h->i_rtjpeg_quality, h->i_rtjpeg_luma_filter, renatofilho@751: h->i_rtjpeg_chroma_filter, h->i_lavc_bitrate, renatofilho@751: h->i_lavc_qmin, h->i_lavc_qmax, h->i_lavc_maxqdiff, renatofilho@751: h->i_seekable_offset, h->i_keyframe_adjust_offset); renatofilho@608: renatofilho@608: done: renatofilho@751: if (buff != NULL) renatofilho@751: { renatofilho@751: gst_buffer_unref(buff); renatofilho@751: buff = NULL; renatofilho@751: } renatofilho@608: return res; renatofilho@608: } renatofilho@608: renatofilho@608: renatofilho@608: /* Query Functions */ renatofilho@608: static const GstQueryType * renatofilho@751: gst_nuv_demux_get_src_query_types(GstPad * pad) renatofilho@608: { renatofilho@608: static const GstQueryType src_types[] = { renatofilho@751: GST_QUERY_POSITION, renatofilho@751: GST_QUERY_DURATION, renatofilho@751: 0 renatofilho@608: }; renatofilho@608: renatofilho@608: return src_types; renatofilho@608: } renatofilho@608: renatofilho@608: static gboolean renatofilho@751: gst_nuv_demux_handle_src_query(GstPad * pad, GstQuery * query) renatofilho@608: { renatofilho@608: gboolean res = FALSE; renatofilho@751: GstNuvDemux *nuv = GST_NUV_DEMUX(gst_pad_get_parent(pad)); renatofilho@608: renatofilho@674: renatofilho@751: switch (GST_QUERY_TYPE(query)) renatofilho@674: { renatofilho@751: case GST_QUERY_POSITION: renatofilho@751: { renatofilho@674: GstFormat format; renatofilho@751: gst_query_parse_position(query, &format, NULL); renatofilho@751: switch (format) renatofilho@751: { renatofilho@751: case GST_FORMAT_TIME: renatofilho@751: if (GST_CLOCK_TIME_IS_VALID(nuv->priv->last_frame_time)) renatofilho@751: { renatofilho@751: gst_query_set_position(query, GST_FORMAT_TIME, renatofilho@751: nuv->priv->last_frame_time); renatofilho@751: res = TRUE; renatofilho@751: } renatofilho@751: break; renatofilho@751: default: renatofilho@751: break; renatofilho@751: } renatofilho@674: break; renatofilho@751: } renatofilho@751: case GST_QUERY_DURATION: renatofilho@674: { renatofilho@751: GstFormat format; renatofilho@751: gst_query_parse_duration(query, &format, NULL); renatofilho@751: switch (format) renatofilho@751: { renatofilho@751: case GST_FORMAT_TIME: renatofilho@751: if (nuv->priv->duration_time != GST_CLOCK_TIME_NONE) renatofilho@751: { renatofilho@751: gst_query_set_duration(query, GST_FORMAT_TIME, renatofilho@751: nuv->priv->duration_time); renatofilho@751: res = TRUE; renatofilho@751: } renatofilho@751: break; renatofilho@751: default: renatofilho@751: break; renatofilho@751: } renatofilho@751: break; renatofilho@674: } renatofilho@674: default: renatofilho@674: break; renatofilho@751: } renatofilho@608: renatofilho@751: if (res == FALSE) renatofilho@751: { renatofilho@751: res = gst_pad_query_default(pad, query); renatofilho@751: } renatofilho@674: renatofilho@751: gst_object_unref(nuv); renatofilho@608: return res; renatofilho@608: } renatofilho@608: renatofilho@751: static GstPad * renatofilho@751: gst_nuv_demux_create_pad(GstNuvDemux * nuv, GstCaps * caps, renatofilho@751: GstStaticPadTemplate * template, const gchar * name) renatofilho@608: { renatofilho@751: GstPad *pad = NULL; renatofilho@751: pad = gst_pad_new_from_static_template(template, name); renatofilho@751: gst_pad_set_caps(pad, caps); renatofilho@751: gst_pad_set_active(pad, TRUE); renatofilho@751: gst_pad_use_fixed_caps(pad); renatofilho@751: gst_element_add_pad(GST_ELEMENT(nuv), pad); renatofilho@608: renatofilho@751: gst_pad_set_event_function(pad, renatofilho@751: GST_DEBUG_FUNCPTR(gst_nuv_demux_srcpad_event)); renatofilho@608: renatofilho@751: gst_pad_set_query_type_function(pad, renatofilho@751: GST_DEBUG_FUNCPTR renatofilho@751: (gst_nuv_demux_get_src_query_types)); renatofilho@608: renatofilho@751: gst_pad_set_query_function(pad, renatofilho@751: GST_DEBUG_FUNCPTR renatofilho@751: (gst_nuv_demux_handle_src_query)); renatofilho@608: renatofilho@751: renatofilho@751: return pad; renatofilho@608: } renatofilho@608: renatofilho@608: static void renatofilho@751: gst_nuv_demux_create_pads(GstNuvDemux * nuv) renatofilho@608: { renatofilho@751: if (nuv->priv->h.i_video_blocks != 0) renatofilho@751: { renatofilho@751: GstCaps *video_caps = NULL; renatofilho@608: renatofilho@751: video_caps = gst_caps_new_simple("video/x-divx", renatofilho@751: "divxversion", G_TYPE_INT, 4, renatofilho@751: "width", G_TYPE_INT, renatofilho@751: nuv->priv->h.i_width, "height", renatofilho@751: G_TYPE_INT, nuv->priv->h.i_height, renatofilho@751: "framerate", GST_TYPE_FRACTION, renatofilho@751: nuv->priv->h.i_fpsn, renatofilho@751: nuv->priv->h.i_fpsd, "format", renatofilho@751: GST_TYPE_FOURCC, renatofilho@751: nuv->priv->eh.i_video_fcc, renatofilho@751: "pixel-aspect-ratio", renatofilho@751: GST_TYPE_FRACTION, renatofilho@751: (gint) (nuv->priv->h.d_aspect * renatofilho@751: 1000.0f), 1000, NULL); renatofilho@608: renatofilho@751: nuv->priv->src_video_pad = renatofilho@751: gst_nuv_demux_create_pad(nuv, video_caps, &video_src_template, renatofilho@751: "video_src"); renatofilho@751: gst_caps_unref(video_caps); renatofilho@751: } renatofilho@608: renatofilho@751: if (nuv->priv->h.i_audio_blocks != 0) renatofilho@751: { renatofilho@751: GstCaps *audio_caps = NULL; renatofilho@608: renatofilho@751: 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, "layer", G_TYPE_INT, 3, // fixme: magic number renatofilho@751: "mpegversion", G_TYPE_INT, renatofilho@751: nuv->priv->eh.i_version, NULL); renatofilho@608: renatofilho@751: nuv->priv->src_audio_pad = renatofilho@751: gst_nuv_demux_create_pad(nuv, audio_caps, &audio_src_template, renatofilho@751: "audio_src"); renatofilho@751: gst_caps_unref(audio_caps); renatofilho@751: } renatofilho@751: renatofilho@751: gst_element_no_more_pads(GST_ELEMENT(nuv)); renatofilho@608: } renatofilho@608: renatofilho@608: static gboolean renatofilho@751: gst_nuv_demux_validate_header(nuv_frame_header * h) renatofilho@608: { renatofilho@608: gboolean valid = FALSE; renatofilho@751: //g_usleep (1 * G_USEC_PER_SEC ); renatofilho@751: switch (h->i_type) renatofilho@751: { renatofilho@608: /* renatofilho@608: case 'V': renatofilho@608: if (h->i_compression == 0 || renatofilho@608: h->i_compression == 1 || renatofilho@608: h->i_compression == 2 || renatofilho@608: h->i_compression == 'N' || renatofilho@608: h->i_compression == 'L') { renatofilho@608: valid = TRUE; renatofilho@608: } renatofilho@608: break; renatofilho@608: case 'A': renatofilho@608: if (h->i_compression == 0 || renatofilho@608: h->i_compression == 1 || renatofilho@608: h->i_compression == 2 || renatofilho@608: h->i_compression == 3 || renatofilho@608: h->i_compression == 'F' || renatofilho@608: h->i_compression == 'S' || renatofilho@608: h->i_compression == 'N' || renatofilho@608: h->i_compression == 'L') { renatofilho@608: valid = TRUE; renatofilho@608: } renatofilho@608: break; renatofilho@608: case 'S': renatofilho@608: if (h->i_compression == 'B' || renatofilho@608: h->i_compression == 'A' || renatofilho@608: h->i_compression == 'V' || renatofilho@608: h->i_compression == 'S') { renatofilho@608: valid = TRUE; renatofilho@608: } renatofilho@608: break; renatofilho@608: */ renatofilho@751: case 'A': renatofilho@751: case 'V': renatofilho@751: case 'S': renatofilho@751: case 'R': renatofilho@751: case 'D': renatofilho@751: case 'Q': renatofilho@751: valid = TRUE; renatofilho@751: break; renatofilho@751: default: renatofilho@751: valid = FALSE; renatofilho@751: } renatofilho@608: renatofilho@608: return valid; renatofilho@608: } renatofilho@608: renatofilho@608: static GstFlowReturn renatofilho@751: gst_nuv_demux_read_head_frame(GstNuvDemux * nuv) renatofilho@608: { renatofilho@608: GstFlowReturn ret = GST_FLOW_OK; renatofilho@608: gboolean valid = FALSE; renatofilho@608: renatofilho@751: do renatofilho@751: { renatofilho@751: ret = gst_nuv_demux_frame_header_load(nuv, &nuv->priv->fh); renatofilho@751: if (ret != GST_FLOW_OK) renatofilho@751: { renatofilho@751: return ret; renatofilho@751: } renatofilho@608: renatofilho@751: if (gst_nuv_demux_validate_header(&nuv->priv->fh) == TRUE) renatofilho@751: valid = TRUE; renatofilho@608: renatofilho@751: } renatofilho@751: while (valid == FALSE); renatofilho@608: renatofilho@608: nuv->priv->state = GST_NUV_DEMUX_MOVI; renatofilho@608: return ret; renatofilho@608: } renatofilho@608: renatofilho@608: static gboolean renatofilho@751: gst_nuv_combine_flow(GstNuvDemux * nuv) renatofilho@608: { renatofilho@751: GstFlowReturn ret_video = nuv->priv->last_video_return; renatofilho@751: GstFlowReturn ret_audio = nuv->priv->last_audio_return; renatofilho@608: renatofilho@751: if ((ret_video != GST_FLOW_OK) && (ret_audio != GST_FLOW_OK)) renatofilho@751: return FALSE; renatofilho@608: renatofilho@751: if (GST_FLOW_IS_FATAL(ret_video)) renatofilho@751: return FALSE; renatofilho@608: renatofilho@751: if (GST_FLOW_IS_FATAL(ret_audio)) renatofilho@751: return FALSE; renatofilho@608: renatofilho@751: return TRUE; renatofilho@608: } renatofilho@608: renatofilho@608: static GstFlowReturn renatofilho@751: gst_nuv_demux_stream_data(GstNuvDemux * nuv) renatofilho@608: { renatofilho@608: GstFlowReturn ret = GST_FLOW_OK; renatofilho@608: GstPad *pad = NULL; renatofilho@608: guint64 timestamp; renatofilho@608: GstBuffer *buf = NULL; renatofilho@608: nuv_frame_header h; renatofilho@608: renatofilho@608: h = nuv->priv->fh; renatofilho@608: renatofilho@751: if (h.i_type == 'R') renatofilho@751: { renatofilho@751: goto done; renatofilho@608: } renatofilho@608: renatofilho@751: if (h.i_length > 0) renatofilho@751: { renatofilho@751: ret = gst_nuv_demux_read_bytes(nuv, h.i_length, TRUE, &buf); renatofilho@751: if ((ret != GST_FLOW_OK) || (buf == NULL)) renatofilho@751: { renatofilho@751: goto done; renatofilho@751: } renatofilho@751: renatofilho@751: if ((h.i_timecode < 0)) renatofilho@751: { renatofilho@751: h.i_timecode = 0; renatofilho@751: //goto done; renatofilho@751: } renatofilho@751: renatofilho@751: timestamp = h.i_timecode * GST_MSECOND; renatofilho@751: GST_BUFFER_TIMESTAMP(buf) = timestamp; renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: goto done; renatofilho@751: } renatofilho@751: renatofilho@751: renatofilho@751: switch (h.i_type) renatofilho@751: { renatofilho@751: case 'V': renatofilho@751: { renatofilho@751: pad = nuv->priv->src_video_pad; renatofilho@751: renatofilho@751: if (nuv->priv->new_video_segment) renatofilho@751: { renatofilho@751: renatofilho@751: /* send new segment event */ renatofilho@751: gst_pad_push_event(nuv->priv->src_video_pad, renatofilho@751: gst_event_new_new_segment(TRUE, 1.0, renatofilho@751: GST_FORMAT_TIME, 0, renatofilho@751: GST_CLOCK_TIME_NONE, renatofilho@751: 0)); renatofilho@751: renatofilho@751: if (nuv->priv->time_start == GST_CLOCK_TIME_NONE) renatofilho@751: { renatofilho@751: nuv->priv->time_start = timestamp; renatofilho@751: } renatofilho@751: nuv->priv->new_video_segment = FALSE; renatofilho@751: } renatofilho@751: renatofilho@751: break; renatofilho@751: } renatofilho@751: case 'A': renatofilho@751: { renatofilho@751: pad = nuv->priv->src_audio_pad; renatofilho@751: renatofilho@751: if (nuv->priv->new_audio_segment) renatofilho@751: { renatofilho@751: /* send new segment event */ renatofilho@751: gst_pad_push_event(nuv->priv->src_audio_pad, renatofilho@751: gst_event_new_new_segment(TRUE, 1.0, renatofilho@751: GST_FORMAT_TIME, 0, renatofilho@751: GST_CLOCK_TIME_NONE, renatofilho@751: 0)); renatofilho@751: renatofilho@751: if (nuv->priv->time_start == GST_CLOCK_TIME_NONE) renatofilho@751: { renatofilho@751: nuv->priv->time_start = timestamp; renatofilho@751: } renatofilho@751: nuv->priv->new_audio_segment = FALSE; renatofilho@751: } renatofilho@751: renatofilho@751: break; renatofilho@751: } renatofilho@751: case 'S': renatofilho@751: { renatofilho@751: switch (h.i_compression) renatofilho@751: { renatofilho@751: case 'V': renatofilho@751: GST_DEBUG_OBJECT(nuv, "sending new video segment: %d", renatofilho@751: h.i_timecode); renatofilho@751: gst_pad_push_event(nuv->priv->src_video_pad, renatofilho@751: gst_event_new_new_segment(TRUE, 1.0, renatofilho@751: GST_FORMAT_TIME, renatofilho@751: h.i_timecode * renatofilho@751: GST_MSECOND, renatofilho@751: GST_CLOCK_TIME_NONE, renatofilho@751: 0)); renatofilho@751: break; renatofilho@751: case 'A': renatofilho@751: GST_DEBUG_OBJECT(nuv, "sending new audio segment: %d", renatofilho@751: h.i_timecode); renatofilho@751: gst_pad_push_event(nuv->priv->src_audio_pad, renatofilho@751: gst_event_new_new_segment(TRUE, 1.0, renatofilho@751: GST_FORMAT_TIME, 0, renatofilho@751: GST_CLOCK_TIME_NONE, renatofilho@751: 0)); renatofilho@751: break; renatofilho@751: default: renatofilho@751: break; renatofilho@751: } renatofilho@751: goto done; renatofilho@751: } renatofilho@751: default: renatofilho@751: break; renatofilho@751: } renatofilho@751: renatofilho@751: if ((buf != NULL) && (pad != NULL)) renatofilho@751: { renatofilho@751: /* pushing the buffer */ renatofilho@751: gst_buffer_set_caps(buf, GST_PAD_CAPS(pad)); renatofilho@751: ret = gst_pad_push(pad, buf); renatofilho@751: buf = NULL; renatofilho@751: renatofilho@751: if (ret != GST_FLOW_OK) renatofilho@751: { renatofilho@751: GST_WARNING_OBJECT(nuv, "error: %d pushing on srcpad %s", ret, renatofilho@751: gst_pad_get_name(pad)); renatofilho@751: renatofilho@751: if (pad == nuv->priv->src_video_pad) renatofilho@751: { renatofilho@751: nuv->priv->last_video_return = ret; renatofilho@751: } renatofilho@751: else if (pad == nuv->priv->src_audio_pad) renatofilho@751: { renatofilho@751: nuv->priv->last_audio_return = ret; renatofilho@751: } renatofilho@751: renatofilho@751: /* verify anothers flow if is necessary stop task */ renatofilho@751: if (gst_nuv_combine_flow(nuv) != FALSE) renatofilho@751: { renatofilho@751: ret = GST_FLOW_OK; renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: GST_WARNING_OBJECT(nuv, "error: on push"); renatofilho@751: } renatofilho@751: renatofilho@751: } renatofilho@751: } renatofilho@608: renatofilho@608: done: renatofilho@751: if (buf != NULL) renatofilho@751: { renatofilho@751: gst_buffer_unref(buf); renatofilho@751: buf = NULL; renatofilho@751: } renatofilho@751: if (ret == GST_FLOW_OK) renatofilho@751: { renatofilho@751: nuv->priv->state = GST_NUV_DEMUX_FRAME_HEADER; renatofilho@751: memset(&nuv->priv->fh, 0, sizeof(nuv->priv->fh)); renatofilho@751: } renatofilho@608: return ret; renatofilho@608: } renatofilho@608: renatofilho@608: static GstFlowReturn renatofilho@751: gst_nuv_demux_stream_mpeg_data(GstNuvDemux * nuv) renatofilho@608: { renatofilho@608: GstFlowReturn ret = GST_FLOW_OK; renatofilho@608: renatofilho@751: /* ffmpeg extra data */ renatofilho@751: if (nuv->priv->new_file) renatofilho@751: { renatofilho@751: GstBuffer *buf; renatofilho@751: ret = renatofilho@751: gst_nuv_demux_read_bytes(nuv, nuv->priv->mpeg_data_size, TRUE, &buf); renatofilho@751: gst_buffer_unref(buf); renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: ret = gst_nuv_demux_read_bytes(nuv, nuv->priv->mpeg_data_size, TRUE, renatofilho@751: &nuv->priv->mpeg_buffer); renatofilho@751: } renatofilho@714: renatofilho@751: if ((ret != GST_FLOW_OK) || (nuv->priv->mpeg_buffer == NULL)) renatofilho@751: { renatofilho@751: return ret; renatofilho@751: } renatofilho@608: renatofilho@751: GST_BUFFER_SIZE(nuv->priv->mpeg_buffer) = nuv->priv->mpeg_data_size; renatofilho@608: nuv->priv->state = GST_NUV_DEMUX_EXTEND_HEADER; renatofilho@608: return ret; renatofilho@608: } renatofilho@608: renatofilho@608: static GstFlowReturn renatofilho@751: gst_nuv_demux_stream_extra_data(GstNuvDemux * nuv) renatofilho@608: { renatofilho@608: GstFlowReturn ret = GST_FLOW_OK; renatofilho@608: renatofilho@751: /* Load 'D' */ renatofilho@608: nuv_frame_header h; renatofilho@608: renatofilho@751: if (nuv->priv->new_file) renatofilho@751: ret = gst_nuv_demux_frame_header_load(nuv, NULL); renatofilho@751: else renatofilho@751: ret = gst_nuv_demux_frame_header_load(nuv, &h); renatofilho@714: renatofilho@608: if (ret != GST_FLOW_OK) renatofilho@751: return ret; renatofilho@608: renatofilho@751: if (h.i_type != 'D') renatofilho@751: { renatofilho@751: GST_WARNING_OBJECT(nuv, "Unsuported rtjpeg"); renatofilho@751: return GST_FLOW_NOT_SUPPORTED; renatofilho@751: } renatofilho@608: renatofilho@751: if (h.i_length > 0) renatofilho@751: { renatofilho@751: if (h.i_compression == 'F') renatofilho@751: { renatofilho@751: nuv->priv->state = GST_NUV_DEMUX_MPEG_DATA; renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: GST_WARNING_OBJECT(nuv, renatofilho@751: "only file with extended chunk are supported"); renatofilho@751: return GST_FLOW_NOT_SUPPORTED; renatofilho@751: } renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: nuv->priv->state = GST_NUV_DEMUX_EXTEND_HEADER; renatofilho@751: } renatofilho@608: renatofilho@608: return ret; renatofilho@608: } renatofilho@608: renatofilho@608: static GstFlowReturn renatofilho@751: gst_nuv_demux_stream_extend_header_data(GstNuvDemux * nuv) renatofilho@608: { renatofilho@608: GstFlowReturn ret = GST_FLOW_OK; renatofilho@608: renatofilho@714: if (nuv->priv->new_file) renatofilho@751: ret = gst_nuv_demux_extended_header_load(nuv, NULL); renatofilho@751: else renatofilho@751: { renatofilho@751: ret = gst_nuv_demux_extended_header_load(nuv, &nuv->priv->eh); renatofilho@714: if (ret != GST_FLOW_OK) renatofilho@751: return ret; renatofilho@751: gst_nuv_demux_create_pads(nuv); renatofilho@751: } renatofilho@608: renatofilho@608: nuv->priv->state = GST_NUV_DEMUX_INDEX_CREATE; renatofilho@608: return ret; renatofilho@608: } renatofilho@608: renatofilho@608: static GstFlowReturn renatofilho@751: gst_nuv_demux_stream_extend_header(GstNuvDemux * nuv) renatofilho@608: { renatofilho@608: GstBuffer *buf = NULL; renatofilho@608: GstFlowReturn res = GST_FLOW_OK; renatofilho@608: renatofilho@751: res = gst_nuv_demux_read_bytes(nuv, 1, FALSE, &buf); renatofilho@751: if ((res != GST_FLOW_OK) || (buf == NULL)) renatofilho@751: { renatofilho@751: if (buf != NULL) renatofilho@751: { renatofilho@751: gst_buffer_unref(buf); renatofilho@751: } renatofilho@751: return res; renatofilho@751: } renatofilho@608: renatofilho@751: if (buf->data[0] == 'X') renatofilho@751: { renatofilho@751: gst_buffer_unref(buf); renatofilho@751: buf = NULL; renatofilho@751: nuv_frame_header h; renatofilho@608: renatofilho@751: res = gst_nuv_demux_frame_header_load(nuv, &h); renatofilho@751: if (res != GST_FLOW_OK) renatofilho@751: return res; renatofilho@608: renatofilho@751: if (h.i_length != 512) renatofilho@751: { renatofilho@751: return GST_FLOW_ERROR; renatofilho@751: } renatofilho@751: nuv->priv->state = GST_NUV_DEMUX_EXTEND_HEADER_DATA; renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: nuv->priv->state = GST_NUV_DEMUX_INVALID_DATA; renatofilho@751: g_object_unref(buf); renatofilho@751: GST_ELEMENT_WARNING(nuv, STREAM, FAILED, renatofilho@751: (_("incomplete NUV support")), renatofilho@751: ("incomplete NUV support")); renatofilho@751: return GST_FLOW_ERROR; renatofilho@751: } renatofilho@608: return res; renatofilho@608: } renatofilho@608: renatofilho@608: static void renatofilho@751: gst_nuv_demux_create_seek_index(GstNuvDemux * nuv) renatofilho@608: { renatofilho@608: GstMessage *msg; renatofilho@608: nuv_frame_header h; renatofilho@608: renatofilho@751: while (gst_nuv_demux_frame_header_load(nuv, &h) == GST_FLOW_OK) renatofilho@751: { renatofilho@751: if ((h.i_type == 'V') && (h.i_keyframe == 0)) renatofilho@751: { renatofilho@751: frame_index_data *f = g_new0(frame_index_data, 1); renatofilho@608: renatofilho@751: f->offset = nuv->priv->offset - 12; renatofilho@751: f->timecode = h.i_timecode * GST_MSECOND; renatofilho@608: renatofilho@751: nuv->priv->index = g_slist_append(nuv->priv->index, f); renatofilho@751: } renatofilho@751: if (h.i_type != 'R') renatofilho@751: { renatofilho@751: nuv->priv->offset += h.i_length; renatofilho@608: if (h.i_type == 'A' || h.i_type == 'V') renatofilho@751: nuv->priv->duration_time = h.i_timecode * GST_MSECOND; renatofilho@751: } renatofilho@751: } renatofilho@751: GST_DEBUG_OBJECT(nuv, renatofilho@751: "CREATING INDEX: DONE : DURATION Bytes/Sec: %" renatofilho@751: G_GUINT64_FORMAT "/%" G_GUINT64_FORMAT, nuv->priv->offset, renatofilho@751: nuv->priv->duration_time); renatofilho@751: renatofilho@608: nuv->priv->duration_bytes = nuv->priv->offset; renatofilho@608: nuv->priv->offset = nuv->priv->header_lengh; renatofilho@608: renatofilho@751: msg = renatofilho@751: gst_message_new_duration(GST_OBJECT(nuv), GST_FORMAT_TIME, renatofilho@751: nuv->priv->duration_time); renatofilho@751: gst_element_post_message(GST_ELEMENT(nuv), msg); renatofilho@608: } renatofilho@608: renatofilho@608: static GstFlowReturn renatofilho@751: gst_nuv_demux_play(GstPad * pad) renatofilho@608: { renatofilho@608: GstFlowReturn res = GST_FLOW_OK; renatofilho@751: GstNuvDemux *nuv = GST_NUV_DEMUX(GST_PAD_PARENT(pad)); renatofilho@608: renatofilho@751: switch (nuv->priv->state) renatofilho@751: { renatofilho@751: case GST_NUV_DEMUX_START: renatofilho@751: res = gst_nuv_demux_stream_file_header(nuv); renatofilho@751: if ((res != GST_FLOW_OK) && (res != GST_FLOW_ERROR_NO_DATA)) renatofilho@751: { renatofilho@751: goto pause; renatofilho@751: } renatofilho@751: break; renatofilho@608: renatofilho@751: case GST_NUV_DEMUX_HEADER_DATA: renatofilho@751: res = gst_nuv_demux_stream_header_data(nuv); renatofilho@751: if ((res != GST_FLOW_OK) && (res != GST_FLOW_ERROR_NO_DATA)) renatofilho@751: { renatofilho@751: goto pause; renatofilho@751: } renatofilho@751: break; renatofilho@608: renatofilho@751: case GST_NUV_DEMUX_EXTRA_DATA: renatofilho@751: res = gst_nuv_demux_stream_extra_data(nuv); renatofilho@751: if ((res != GST_FLOW_OK) && (res != GST_FLOW_ERROR_NO_DATA)) renatofilho@751: { renatofilho@751: goto pause; renatofilho@751: } renatofilho@751: break; renatofilho@608: renatofilho@751: case GST_NUV_DEMUX_MPEG_DATA: renatofilho@751: res = gst_nuv_demux_stream_mpeg_data(nuv); renatofilho@751: if ((res != GST_FLOW_OK) && (res != GST_FLOW_ERROR_NO_DATA)) renatofilho@751: { renatofilho@751: goto pause; renatofilho@751: } renatofilho@751: break; renatofilho@608: renatofilho@751: case GST_NUV_DEMUX_EXTEND_HEADER: renatofilho@751: res = gst_nuv_demux_stream_extend_header(nuv); renatofilho@751: if ((res != GST_FLOW_OK) && (res != GST_FLOW_ERROR_NO_DATA)) renatofilho@751: { renatofilho@751: goto pause; renatofilho@751: } renatofilho@751: break; renatofilho@608: renatofilho@751: case GST_NUV_DEMUX_EXTEND_HEADER_DATA: renatofilho@751: res = gst_nuv_demux_stream_extend_header_data(nuv); renatofilho@751: if ((res != GST_FLOW_OK) && (res != GST_FLOW_ERROR_NO_DATA)) renatofilho@751: { renatofilho@751: goto pause; renatofilho@751: } renatofilho@751: //store file header size renatofilho@751: nuv->priv->header_lengh = nuv->priv->offset; renatofilho@751: break; renatofilho@608: renatofilho@751: case GST_NUV_DEMUX_INDEX_CREATE: renatofilho@751: if ((nuv->priv->mode == NUV_PULL_MODE) && (!nuv->priv->new_file)) renatofilho@751: { renatofilho@751: gst_nuv_demux_create_seek_index(nuv); renatofilho@751: } renatofilho@608: renatofilho@751: case GST_NUV_DEMUX_FRAME_HEADER: renatofilho@751: res = gst_nuv_demux_read_head_frame(nuv); renatofilho@751: if ((res != GST_FLOW_OK) && (res != GST_FLOW_ERROR_NO_DATA)) renatofilho@751: { renatofilho@751: goto pause; renatofilho@751: } renatofilho@751: break; renatofilho@608: renatofilho@751: case GST_NUV_DEMUX_MOVI: renatofilho@751: res = gst_nuv_demux_stream_data(nuv); renatofilho@751: if ((res != GST_FLOW_OK) && (res != GST_FLOW_ERROR_NO_DATA)) renatofilho@751: { renatofilho@751: goto pause; renatofilho@751: } renatofilho@751: break; renatofilho@608: renatofilho@751: case GST_NUV_DEMUX_INVALID_DATA: renatofilho@751: goto pause; renatofilho@751: break; renatofilho@751: default: renatofilho@751: g_assert_not_reached(); renatofilho@751: } renatofilho@608: renatofilho@608: return GST_FLOW_OK; renatofilho@608: renatofilho@608: pause: renatofilho@751: GST_LOG_OBJECT(nuv, "pausing task, reason %s", gst_flow_get_name(res)); renatofilho@751: gst_pad_pause_task(nuv->priv->sinkpad); renatofilho@608: renatofilho@751: if (res == GST_FLOW_ERROR_EOS) renatofilho@751: { renatofilho@751: gst_nuv_demux_send_eos(nuv); renatofilho@751: nuv->priv->eos = TRUE; renatofilho@751: res = GST_FLOW_OK; renatofilho@751: } renatofilho@608: renatofilho@751: if (GST_FLOW_IS_FATAL(res)) renatofilho@751: { renatofilho@751: GST_ELEMENT_ERROR(nuv, STREAM, FAILED, renatofilho@751: (_("Internal data stream error.")), renatofilho@751: ("streaming stopped, reason %s", renatofilho@751: gst_flow_get_name(res))); renatofilho@608: renatofilho@751: gst_nuv_demux_send_eos(nuv); renatofilho@751: } renatofilho@608: return res; renatofilho@608: } renatofilho@608: renatofilho@608: static void renatofilho@751: gst_nuv_demux_send_eos(GstNuvDemux * nuv) renatofilho@608: { renatofilho@751: gst_element_post_message(GST_ELEMENT(nuv), renatofilho@751: gst_message_new_segment_done(GST_OBJECT(nuv), renatofilho@751: GST_FORMAT_TIME, -1)); renatofilho@608: renatofilho@608: if (nuv->priv->src_video_pad) renatofilho@751: gst_pad_push_event(nuv->priv->src_video_pad, gst_event_new_eos()); renatofilho@608: if (nuv->priv->src_audio_pad) renatofilho@751: gst_pad_push_event(nuv->priv->src_audio_pad, gst_event_new_eos()); renatofilho@608: } renatofilho@608: renatofilho@608: static GstFlowReturn renatofilho@751: gst_nuv_demux_read_bytes(GstNuvDemux * nuv, guint64 size, gboolean move, renatofilho@751: GstBuffer ** buffer) renatofilho@608: { renatofilho@608: GstFlowReturn ret = GST_FLOW_OK; renatofilho@608: renatofilho@751: if (size == 0) renatofilho@751: { renatofilho@751: return ret; renatofilho@751: } renatofilho@608: renatofilho@751: if (nuv->priv->mode == NUV_PULL_MODE) renatofilho@751: { renatofilho@751: ret = renatofilho@751: gst_pad_pull_range(nuv->priv->sinkpad, nuv->priv->offset, size, renatofilho@751: buffer); renatofilho@751: if (ret == GST_FLOW_OK) renatofilho@751: { renatofilho@751: if (move) renatofilho@751: { renatofilho@751: nuv->priv->offset += size; renatofilho@751: } renatofilho@751: /* got eos */ renatofilho@751: } renatofilho@751: else if (ret == GST_FLOW_UNEXPECTED) renatofilho@751: { renatofilho@751: return GST_FLOW_ERROR_EOS; renatofilho@751: } renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: if (gst_adapter_available(nuv->priv->adapter) < size) renatofilho@751: { renatofilho@751: nuv->priv->more_data = TRUE; renatofilho@751: return GST_FLOW_ERROR_NO_DATA; renatofilho@751: } renatofilho@751: if (move) renatofilho@751: { renatofilho@751: *buffer = gst_adapter_take_buffer(nuv->priv->adapter, size); renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: guint8 *data = NULL; renatofilho@751: data = (guint8 *) gst_adapter_peek(nuv->priv->adapter, size); renatofilho@751: *buffer = gst_buffer_new(); renatofilho@751: gst_buffer_set_data(*buffer, data, size); renatofilho@751: } renatofilho@751: } renatofilho@608: return ret; renatofilho@608: } renatofilho@608: rosfran@713: static GstFlowReturn renatofilho@751: gst_nuv_demux_move_bytes(GstNuvDemux * nuv, guint64 size) rosfran@713: { rosfran@713: GstFlowReturn ret = GST_FLOW_OK; rosfran@713: renatofilho@751: if (size == 0) renatofilho@751: { renatofilho@751: return ret; renatofilho@751: } rosfran@713: renatofilho@751: if (nuv->priv->mode == NUV_PULL_MODE) renatofilho@751: { renatofilho@751: nuv->priv->offset += size; renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: if (gst_adapter_available(nuv->priv->adapter) < size) renatofilho@751: { renatofilho@751: nuv->priv->more_data = TRUE; renatofilho@751: return GST_FLOW_ERROR_NO_DATA; renatofilho@751: } renatofilho@751: gst_adapter_flush(nuv->priv->adapter, size); renatofilho@751: } rosfran@713: return ret; rosfran@713: } rosfran@713: renatofilho@608: static gboolean renatofilho@751: gst_nuv_demux_sink_activate(GstPad * sinkpad) renatofilho@608: { renatofilho@608: gboolean res = TRUE; renatofilho@751: GstNuvDemux *nuv = GST_NUV_DEMUX(gst_pad_get_parent(sinkpad)); renatofilho@751: renatofilho@751: if (gst_pad_check_pull_range(sinkpad)) renatofilho@751: { renatofilho@751: gst_adapter_clear(nuv->priv->adapter); renatofilho@751: res = gst_pad_activate_pull(sinkpad, TRUE); renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: gst_adapter_clear(nuv->priv->adapter); renatofilho@751: res = gst_pad_activate_push(sinkpad, TRUE); renatofilho@751: } renatofilho@751: renatofilho@751: g_object_unref(nuv); renatofilho@608: return res; renatofilho@608: } renatofilho@608: renatofilho@608: static gboolean renatofilho@751: gst_nuv_demux_sink_activate_pull(GstPad * sinkpad, gboolean active) renatofilho@608: { renatofilho@751: GstNuvDemux *nuv = GST_NUV_DEMUX(gst_pad_get_parent(sinkpad)); renatofilho@608: renatofilho@751: if (active) renatofilho@751: { renatofilho@751: GST_DEBUG_OBJECT(nuv, "activating pull function"); renatofilho@751: nuv->priv->mode = NUV_PULL_MODE; renatofilho@751: gst_adapter_clear(nuv->priv->adapter); renatofilho@608: renatofilho@751: gst_pad_start_task(sinkpad, (GstTaskFunction) gst_nuv_demux_loop, renatofilho@751: sinkpad); renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: GST_DEBUG_OBJECT(nuv, "deactivating pull function"); renatofilho@751: gst_pad_stop_task(sinkpad); renatofilho@751: } renatofilho@751: gst_object_unref(nuv); renatofilho@608: renatofilho@608: return TRUE; renatofilho@608: } renatofilho@608: renatofilho@608: static gboolean renatofilho@751: gst_nuv_demux_sink_activate_push(GstPad * pad, gboolean active) renatofilho@608: { renatofilho@751: GstNuvDemux *nuv = GST_NUV_DEMUX(gst_pad_get_parent(pad)); renatofilho@608: renatofilho@751: if (active) renatofilho@751: { renatofilho@751: nuv->priv->mode = NUV_PUSH_MODE; renatofilho@751: gst_adapter_clear(nuv->priv->adapter); renatofilho@608: renatofilho@751: GST_DEBUG_OBJECT(nuv, "activating push/chain function"); renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: GST_DEBUG_OBJECT(nuv, "deactivating push/chain function"); renatofilho@751: } renatofilho@608: renatofilho@751: gst_object_unref(nuv); renatofilho@608: renatofilho@608: return TRUE; renatofilho@608: } renatofilho@608: renatofilho@608: static frame_index_data * renatofilho@751: gst_nuv_demux_do_seek_index(GstNuvDemux * nuv, gint64 seek_pos, renatofilho@751: gint64 segment_stop, GstFormat format) renatofilho@608: { renatofilho@608: GSList *l; renatofilho@608: frame_index_data *ret = NULL; renatofilho@608: renatofilho@751: if (nuv->priv->index == NULL) renatofilho@751: { renatofilho@751: return NULL; renatofilho@751: } renatofilho@608: renatofilho@751: /* find keyframe closest to the requested position */ renatofilho@751: for (l = nuv->priv->index; l != NULL; l = l->next) renatofilho@751: { renatofilho@751: frame_index_data *f = (frame_index_data *) l->data; renatofilho@751: gint64 pos = 0; renatofilho@608: renatofilho@751: if (format == GST_FORMAT_BYTES) renatofilho@751: { renatofilho@751: pos = f->offset; renatofilho@751: } renatofilho@751: else if (format == GST_FORMAT_TIME) renatofilho@751: { renatofilho@751: pos = f->timecode; renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: return NULL; renatofilho@751: } renatofilho@608: renatofilho@751: if (pos >= seek_pos) renatofilho@751: { renatofilho@751: ret = f; renatofilho@751: break; renatofilho@751: } renatofilho@751: renatofilho@751: if ((segment_stop != -1) && (segment_stop != GST_CLOCK_TIME_NONE) renatofilho@751: && (pos > segment_stop)) renatofilho@751: { renatofilho@751: break; renatofilho@751: } renatofilho@751: } renatofilho@608: renatofilho@608: return ret; renatofilho@608: } renatofilho@608: renatofilho@608: static gboolean renatofilho@751: gst_nuv_demux_do_seek(GstNuvDemux * nuv, GstEvent * event) renatofilho@608: { renatofilho@608: gdouble rate; renatofilho@608: GstFormat format; renatofilho@608: GstSeekFlags flags; renatofilho@608: GstSeekType cur_type; renatofilho@608: gint64 cur; renatofilho@608: GstSeekType stop_type; renatofilho@608: gint64 stop; renatofilho@608: gboolean flush; renatofilho@608: frame_index_data *entry; renatofilho@608: gint64 segment_start; renatofilho@608: gint64 segment_stop; renatofilho@608: GstEvent *newsegment_event; renatofilho@608: renatofilho@751: if (nuv->priv->eos) renatofilho@751: { renatofilho@751: return FALSE; renatofilho@751: } renatofilho@608: renatofilho@751: if (nuv->priv->mode == NUV_PUSH_MODE) renatofilho@751: { renatofilho@751: return FALSE; renatofilho@751: } renatofilho@608: renatofilho@608: renatofilho@751: gst_event_parse_seek(event, &rate, &format, &flags, renatofilho@751: &cur_type, &cur, &stop_type, &stop); renatofilho@608: renatofilho@608: renatofilho@608: renatofilho@608: /* renatofilho@608: if (format == GST_FORMAT_TIME) { renatofilho@608: GST_DEBUG_OBJECT (nuv, "Can only seek on BYTES"); renatofilho@608: return FALSE; renatofilho@608: } renatofilho@608: */ renatofilho@608: renatofilho@751: if (rate <= 0.0) renatofilho@751: { renatofilho@751: GST_DEBUG_OBJECT(nuv, "Can only seek with positive rate"); renatofilho@751: return FALSE; renatofilho@751: } renatofilho@751: renatofilho@751: if (cur_type == GST_SEEK_TYPE_SET) renatofilho@751: { renatofilho@751: GST_OBJECT_LOCK(nuv); renatofilho@751: if (gst_nuv_demux_do_seek_index(nuv, cur, -1, format) == NULL) renatofilho@751: { renatofilho@751: GST_DEBUG_OBJECT(nuv, "No matching seek entry in index"); renatofilho@751: GST_OBJECT_UNLOCK(nuv); renatofilho@751: return FALSE; renatofilho@751: } renatofilho@751: GST_OBJECT_UNLOCK(nuv); renatofilho@751: } renatofilho@608: renatofilho@608: flush = !!(flags & GST_SEEK_FLAG_FLUSH); renatofilho@608: renatofilho@751: if (flush) renatofilho@751: { renatofilho@751: gst_pad_push_event(nuv->priv->sinkpad, gst_event_new_flush_start()); renatofilho@751: if (nuv->priv->src_video_pad != NULL) renatofilho@751: { renatofilho@751: gst_pad_push_event(nuv->priv->src_video_pad, renatofilho@751: gst_event_new_flush_start()); renatofilho@751: } renatofilho@608: renatofilho@751: if (nuv->priv->src_audio_pad != NULL) renatofilho@751: { renatofilho@751: gst_pad_push_event(nuv->priv->src_audio_pad, renatofilho@751: gst_event_new_flush_start()); renatofilho@751: } renatofilho@751: } renatofilho@751: else renatofilho@751: { renatofilho@751: gst_pad_pause_task(nuv->priv->sinkpad); renatofilho@751: } renatofilho@751: renatofilho@751: GST_PAD_STREAM_LOCK(nuv->priv->sinkpad); renatofilho@751: GST_OBJECT_LOCK(nuv); renatofilho@608: renatofilho@608: renatofilho@608: if (cur == GST_CLOCK_TIME_NONE) renatofilho@751: cur = 0; renatofilho@608: if (stop == GST_CLOCK_TIME_NONE) renatofilho@751: stop = nuv->priv->duration_time; renatofilho@608: renatofilho@608: if (cur_type == GST_SEEK_TYPE_SET) renatofilho@751: segment_start = cur; renatofilho@608: else if (cur_type == GST_SEEK_TYPE_CUR) renatofilho@751: segment_start = nuv->priv->segment_start + cur; renatofilho@608: else renatofilho@751: segment_start = nuv->priv->segment_start; renatofilho@608: renatofilho@608: if (stop_type == GST_SEEK_TYPE_SET) renatofilho@751: segment_stop = stop; renatofilho@608: else if (stop_type == GST_SEEK_TYPE_CUR) renatofilho@751: segment_stop = nuv->priv->segment_stop + stop; renatofilho@608: else renatofilho@751: segment_stop = nuv->priv->segment_stop; renatofilho@608: renatofilho@751: segment_start = CLAMP(segment_start, 0, nuv->priv->duration_time); renatofilho@751: segment_stop = CLAMP(segment_stop, 0, nuv->priv->duration_time); renatofilho@608: renatofilho@751: entry = gst_nuv_demux_do_seek_index(nuv, segment_start, renatofilho@751: segment_stop, format); renatofilho@608: renatofilho@751: if (entry == NULL) renatofilho@751: { renatofilho@751: GST_DEBUG_OBJECT(nuv, "No matching seek entry in index"); renatofilho@751: goto seek_error; renatofilho@751: } renatofilho@608: renatofilho@608: segment_start = entry->timecode; renatofilho@608: renatofilho@608: nuv->priv->segment_start = segment_start; renatofilho@608: nuv->priv->segment_stop = segment_stop; renatofilho@608: renatofilho@751: GST_OBJECT_UNLOCK(nuv); renatofilho@608: renatofilho@751: if (!nuv->priv->eos) renatofilho@751: { renatofilho@751: GstMessage *msg; renatofilho@751: msg = gst_message_new_segment_start(GST_OBJECT(nuv), GST_FORMAT_TIME, renatofilho@751: nuv->priv->segment_start); renatofilho@608: renatofilho@751: gst_element_post_message(GST_ELEMENT(nuv), msg); renatofilho@751: } renatofilho@608: renatofilho@751: GST_DEBUG_OBJECT(nuv, renatofilho@751: "NEW SEGMENT START %" G_GUINT64_FORMAT ", STOP %" renatofilho@751: G_GUINT64_FORMAT, segment_start, segment_stop); renatofilho@751: newsegment_event = renatofilho@751: gst_event_new_new_segment(FALSE, rate, GST_FORMAT_TIME, segment_start, renatofilho@751: segment_stop, segment_start); renatofilho@608: renatofilho@608: renatofilho@751: if (flush) renatofilho@751: { renatofilho@751: if (nuv->priv->src_video_pad != NULL) renatofilho@751: { renatofilho@751: gst_pad_push_event(nuv->priv->src_video_pad, renatofilho@751: gst_event_new_flush_stop()); renatofilho@751: } renatofilho@608: renatofilho@751: if (nuv->priv->src_audio_pad != NULL) renatofilho@751: { renatofilho@751: gst_pad_push_event(nuv->priv->src_audio_pad, renatofilho@751: gst_event_new_flush_stop()); renatofilho@751: } renatofilho@608: renatofilho@751: gst_pad_push_event(nuv->priv->sinkpad, gst_event_new_flush_stop()); renatofilho@751: } renatofilho@608: renatofilho@751: renatofilho@751: if (nuv->priv->src_video_pad != NULL) renatofilho@751: { renatofilho@751: gst_pad_push_event(nuv->priv->src_video_pad, renatofilho@751: gst_event_ref(newsegment_event)); renatofilho@751: } renatofilho@751: if (nuv->priv->src_audio_pad != NULL) renatofilho@751: { renatofilho@751: gst_pad_push_event(nuv->priv->src_audio_pad, renatofilho@751: gst_event_ref(newsegment_event)); renatofilho@751: } renatofilho@751: renatofilho@751: gst_event_unref(newsegment_event); renatofilho@608: renatofilho@608: nuv->priv->state = GST_NUV_DEMUX_FRAME_HEADER; renatofilho@608: nuv->priv->offset = entry->offset; renatofilho@608: renatofilho@751: gst_pad_start_task(nuv->priv->sinkpad, (GstTaskFunction) gst_nuv_demux_loop, renatofilho@751: nuv->priv->sinkpad); renatofilho@608: renatofilho@751: GST_PAD_STREAM_UNLOCK(nuv->priv->sinkpad); renatofilho@608: return TRUE; renatofilho@608: renatofilho@608: seek_error: renatofilho@751: GST_DEBUG_OBJECT(nuv, "Got a seek error"); renatofilho@751: GST_OBJECT_UNLOCK(nuv); renatofilho@751: GST_PAD_STREAM_UNLOCK(nuv->priv->sinkpad); renatofilho@608: return FALSE; renatofilho@608: renatofilho@608: } renatofilho@608: renatofilho@608: static gboolean renatofilho@751: gst_nuv_demux_srcpad_event(GstPad * pad, GstEvent * event) renatofilho@608: { renatofilho@608: gboolean res = FALSE; renatofilho@608: GstNuvDemux *nuv; renatofilho@608: renatofilho@751: nuv = GST_NUV_DEMUX(gst_pad_get_parent(pad)); renatofilho@751: renatofilho@751: switch (GST_EVENT_TYPE(event)) renatofilho@751: { renatofilho@751: case GST_EVENT_SEEK: renatofilho@751: res = gst_nuv_demux_do_seek(nuv, event); renatofilho@751: break; renatofilho@751: default: renatofilho@751: res = FALSE; renatofilho@751: break; renatofilho@751: } renatofilho@751: renatofilho@751: gst_object_unref(nuv); renatofilho@608: return res; renatofilho@608: } renatofilho@608: renatofilho@714: static gboolean renatofilho@751: gst_nuv_demux_sink_event(GstPad * pad, GstEvent * event) renatofilho@714: { renatofilho@751: gboolean res = FALSE; renatofilho@751: GstNuvDemux *nuv; renatofilho@714: renatofilho@751: nuv = GST_NUV_DEMUX(gst_pad_get_parent(pad)); renatofilho@714: renatofilho@751: switch (GST_EVENT_TYPE(event)) renatofilho@751: { renatofilho@751: case GST_EVENT_NEWSEGMENT: renatofilho@751: { renatofilho@751: gboolean update; renatofilho@751: gdouble rate; renatofilho@751: GstFormat format; renatofilho@751: gint64 start; renatofilho@751: gint64 stop; renatofilho@751: gint64 position; renatofilho@714: renatofilho@751: gst_event_parse_new_segment(event, &update, &rate, &format, &start, renatofilho@751: &stop, &position); renatofilho@751: if ((format == GST_FORMAT_BYTES) && (start == 0)) renatofilho@751: { renatofilho@751: g_debug("NEW SEGMENT 0"); renatofilho@751: if (nuv->priv->segment > 0) renatofilho@751: { renatofilho@751: nuv->priv->new_file = TRUE; renatofilho@751: nuv->priv->state = GST_NUV_DEMUX_START; renatofilho@751: nuv->priv->offset = 0; renatofilho@751: } renatofilho@751: nuv->priv->segment++; renatofilho@714: renatofilho@751: /* renatofilho@751: newsegment_event = gst_event_new_new_segment (FALSE, rate, renatofilho@751: GST_FORMAT_TIME, 0, GST_CLOCK_TIME_NONE, GST_CLOCK_TIME_NONE); rosfran@725: renatofilho@751: gst_pad_push_event (nuv->priv->src_audio_pad, gst_event_ref (newsegment_event)); renatofilho@751: gst_pad_push_event (nuv->priv->src_video_pad, gst_event_ref (newsegment_event)); renatofilho@751: gst_event_unref (newsegment_event); renatofilho@751: */ renatofilho@751: res = gst_pad_event_default(pad, event); renatofilho@751: } renatofilho@751: break; renatofilho@751: } renatofilho@751: case GST_EVENT_CUSTOM_DOWNSTREAM: renatofilho@751: { renatofilho@751: /* renatofilho@751: nuv->priv->new_file = TRUE; renatofilho@751: nuv->priv->state = GST_NUV_DEMUX_START; renatofilho@751: nuv->priv->offset = 0; renatofilho@751: */ renatofilho@751: GST_PAD_STREAM_LOCK(pad); renatofilho@751: gst_nuv_demux_reset(nuv); renatofilho@751: GST_PAD_STREAM_UNLOCK(pad); renatofilho@714: renatofilho@751: res = gst_pad_event_default(pad, event); renatofilho@751: break; renatofilho@751: } renatofilho@751: default: renatofilho@751: res = gst_pad_event_default(pad, event); renatofilho@751: break; renatofilho@751: } renatofilho@751: renatofilho@751: return res; renatofilho@714: } renatofilho@714: renatofilho@608: static GstFlowReturn renatofilho@751: gst_nuv_demux_chain(GstPad * pad, GstBuffer * buf) renatofilho@608: { renatofilho@608: GstFlowReturn ret = GST_FLOW_OK; renatofilho@751: GstNuvDemux *nuv = GST_NUV_DEMUX(gst_pad_get_parent(pad)); renatofilho@751: renatofilho@608: if (nuv->priv->mode != NUV_PUSH_MODE) renatofilho@751: return ret; renatofilho@608: renatofilho@751: gst_adapter_push(nuv->priv->adapter, buf); renatofilho@751: renatofilho@751: while ((ret == GST_FLOW_OK) && (nuv->priv->more_data == FALSE)) renatofilho@751: { renatofilho@751: ret = gst_nuv_demux_play(pad); renatofilho@751: } renatofilho@608: renatofilho@608: nuv->priv->more_data = FALSE; renatofilho@751: gst_object_unref(nuv); renatofilho@608: renatofilho@608: return ret; renatofilho@608: } renatofilho@608: renatofilho@608: static void renatofilho@751: gst_nuv_demux_loop(GstPad * pad) renatofilho@608: { renatofilho@751: gst_nuv_demux_play(pad); renatofilho@608: } renatofilho@608: renatofilho@608: static void renatofilho@751: gst_nuv_demux_index_free(gpointer data, gpointer user_data) renatofilho@608: { renatofilho@751: g_free(data); renatofilho@608: } renatofilho@608: renatofilho@608: static void renatofilho@751: gst_nuv_demux_reset(GstNuvDemux * nuv) renatofilho@608: { renatofilho@608: nuv->priv->eos = FALSE; renatofilho@608: nuv->priv->more_data = FALSE; renatofilho@608: nuv->priv->state = GST_NUV_DEMUX_START; renatofilho@608: nuv->priv->mode = NUV_PUSH_MODE; renatofilho@608: nuv->priv->offset = 0; renatofilho@608: nuv->priv->time_start = 0; renatofilho@608: nuv->priv->time_qos = GST_CLOCK_TIME_NONE; renatofilho@608: nuv->priv->duration_bytes = GST_CLOCK_TIME_NONE; renatofilho@608: nuv->priv->duration_time = GST_CLOCK_TIME_NONE; renatofilho@608: nuv->priv->last_video_return = GST_FLOW_OK; renatofilho@608: nuv->priv->last_audio_return = GST_FLOW_OK; renatofilho@608: nuv->priv->header_lengh = 0; renatofilho@608: nuv->priv->segment_stop = GST_CLOCK_TIME_NONE; renatofilho@608: nuv->priv->segment_start = GST_CLOCK_TIME_NONE; renatofilho@714: nuv->priv->new_file = FALSE; renatofilho@608: renatofilho@751: //clear index list renatofilho@751: g_slist_foreach(nuv->priv->index, gst_nuv_demux_index_free, NULL); renatofilho@751: g_slist_free(nuv->priv->index); renatofilho@608: nuv->priv->index = NULL; renatofilho@608: renatofilho@751: gst_adapter_clear(nuv->priv->adapter); renatofilho@608: renatofilho@751: if (nuv->priv->mpeg_buffer != NULL) renatofilho@751: { renatofilho@751: gst_buffer_unref(nuv->priv->mpeg_buffer); renatofilho@751: nuv->priv->mpeg_buffer = NULL; renatofilho@751: } renatofilho@608: } renatofilho@608: renatofilho@608: static void renatofilho@751: gst_nuv_demux_destoy_src_pad(GstNuvDemux * nuv) renatofilho@608: { renatofilho@751: if (nuv->priv->src_video_pad) renatofilho@751: { renatofilho@751: gst_element_remove_pad(GST_ELEMENT(nuv), nuv->priv->src_video_pad); renatofilho@751: nuv->priv->src_video_pad = NULL; renatofilho@751: } renatofilho@608: renatofilho@751: if (nuv->priv->src_audio_pad) renatofilho@751: { renatofilho@751: gst_element_remove_pad(GST_ELEMENT(nuv), nuv->priv->src_audio_pad); renatofilho@751: nuv->priv->src_audio_pad = NULL; renatofilho@751: } renatofilho@608: } renatofilho@608: renatofilho@608: static GstStateChangeReturn renatofilho@751: gst_nuv_demux_change_state(GstElement * element, GstStateChange transition) renatofilho@608: { renatofilho@608: GstStateChangeReturn ret = GST_STATE_CHANGE_SUCCESS; renatofilho@608: renatofilho@751: g_debug("Nuvdemux state_change"); renatofilho@751: switch (transition) renatofilho@751: { renatofilho@751: case GST_STATE_CHANGE_NULL_TO_READY: renatofilho@751: gst_nuv_demux_reset(GST_NUV_DEMUX(element)); renatofilho@751: gst_nuv_demux_destoy_src_pad(GST_NUV_DEMUX(element)); renatofilho@751: break; renatofilho@751: default: renatofilho@751: break; renatofilho@751: } renatofilho@608: renatofilho@751: g_debug("Nuvdemux state_change: 1"); renatofilho@692: renatofilho@751: ret = GST_ELEMENT_CLASS(parent_class)->change_state(element, transition); renatofilho@751: if (ret == GST_STATE_CHANGE_FAILURE) renatofilho@751: { renatofilho@751: goto done; renatofilho@751: } renatofilho@608: renatofilho@692: renatofilho@751: g_debug("Nuvdemux state_change: 2"); renatofilho@692: renatofilho@751: switch (transition) renatofilho@751: { renatofilho@751: case GST_STATE_CHANGE_READY_TO_NULL: renatofilho@751: gst_nuv_demux_reset(GST_NUV_DEMUX(element)); renatofilho@751: gst_nuv_demux_destoy_src_pad(GST_NUV_DEMUX(element)); renatofilho@751: break; renatofilho@751: default: renatofilho@751: break; renatofilho@751: } renatofilho@608: renatofilho@692: renatofilho@751: g_debug("Nuvdemux state_change: DONE"); renatofilho@692: renatofilho@608: done: renatofilho@608: return ret; renatofilho@608: } renatofilho@608: renatofilho@751: #if (GST_VERSION_MINOR == 10) && (GST_VERSION_MICRO < 6) renatofilho@608: GstBuffer * renatofilho@751: gst_adapter_take_buffer(GstAdapter * adapter, guint nbytes) renatofilho@608: { renatofilho@608: GstBuffer *buffer; renatofilho@608: GstBuffer *cur; renatofilho@608: guint8 *data; renatofilho@608: renatofilho@751: g_return_val_if_fail(GST_IS_ADAPTER(adapter), NULL); renatofilho@751: g_return_val_if_fail(nbytes > 0, NULL); renatofilho@608: renatofilho@751: GST_LOG_OBJECT(adapter, "taking buffer of %u bytes", nbytes); renatofilho@608: renatofilho@751: /* we don't have enough data, return NULL. This is unlikely renatofilho@751: * as one usually does an _available() first instead of peeking a renatofilho@751: * random size. */ renatofilho@751: if (G_UNLIKELY(nbytes > adapter->size)) renatofilho@751: return NULL; renatofilho@608: renatofilho@751: /* our head buffer has enough data left, return it */ renatofilho@608: cur = adapter->buflist->data; renatofilho@751: if (GST_BUFFER_SIZE(cur) >= nbytes + adapter->skip) renatofilho@751: { renatofilho@751: GST_LOG_OBJECT(adapter, "providing buffer of %d bytes via sub-buffer", renatofilho@751: nbytes); renatofilho@751: buffer = gst_buffer_create_sub(cur, adapter->skip, nbytes); renatofilho@608: renatofilho@751: gst_adapter_flush(adapter, nbytes); renatofilho@608: renatofilho@751: return buffer; renatofilho@751: } renatofilho@608: renatofilho@751: data = gst_adapter_take(adapter, nbytes); renatofilho@608: if (data == NULL) renatofilho@751: return NULL; renatofilho@608: renatofilho@751: buffer = gst_buffer_new(); renatofilho@751: GST_BUFFER_DATA(buffer) = data; renatofilho@751: GST_BUFFER_MALLOCDATA(buffer) = data; renatofilho@751: GST_BUFFER_SIZE(buffer) = nbytes; renatofilho@608: renatofilho@608: return buffer; renatofilho@608: } renatofilho@608: #endif renatofilho@608: renatofilho@608: static void renatofilho@751: gst_nuv_typefind(GstTypeFind * tf, gpointer unused) renatofilho@608: { renatofilho@751: guint8 *data = gst_type_find_peek(tf, 0, 11); renatofilho@608: renatofilho@751: if (data) renatofilho@751: { renatofilho@751: if (memcmp(data, "MythTVVideo", 11) == 0 renatofilho@751: || memcmp(data, "NuppelVideo", 11) == 0) renatofilho@751: { renatofilho@751: gst_type_find_suggest(tf, GST_TYPE_FIND_MAXIMUM, renatofilho@751: gst_caps_new_simple("video/x-nuv", NULL)); renatofilho@751: } renatofilho@751: } renatofilho@608: } renatofilho@608: renatofilho@608: static gboolean renatofilho@751: plugin_init(GstPlugin * plugin) renatofilho@608: { renatofilho@751: static gchar *exts[] = { "nuv", NULL }; renatofilho@608: #ifdef ENABLE_NLS renatofilho@751: setlocale(LC_ALL, ""); renatofilho@751: bindtextdomain(GETTEXT_PACKAGE, LOCALEDIR); renatofilho@608: #endif /* ENABLE_NLS */ renatofilho@608: renatofilho@751: if (!gst_element_register(plugin, "nuvdemux", GST_RANK_SECONDARY, renatofilho@751: GST_TYPE_NUV_DEMUX)) renatofilho@751: { renatofilho@751: return FALSE; renatofilho@751: } renatofilho@608: renatofilho@751: if (!gst_type_find_register(plugin, "video/x-nuv", GST_RANK_SECONDARY, renatofilho@751: gst_nuv_typefind, renatofilho@751: exts, renatofilho@751: gst_caps_new_simple("video/x-nuv", NULL), NULL, renatofilho@751: NULL)) renatofilho@751: { renatofilho@751: GST_WARNING("can't register typefind"); renatofilho@751: return FALSE; renatofilho@751: } renatofilho@608: renatofilho@608: return TRUE; renatofilho@608: } renatofilho@608: renatofilho@751: GST_PLUGIN_DEFINE(GST_VERSION_MAJOR, renatofilho@751: GST_VERSION_MINOR, renatofilho@751: "nuvdemux", renatofilho@751: "Demuxes and muxes audio and video", renatofilho@751: plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, renatofilho@751: GST_PACKAGE_ORIGIN)