gstreamer/gst/mpegtsdemux/tsdemux.c
2011-05-18 10:31:38 +03:00

1507 lines
46 KiB
C

/*
* tsdemux.c
* Copyright (C) 2009 Zaheer Abbas Merali
* 2010 Edward Hervey
*
* Authors:
* Zaheer Abbas Merali <zaheerabbas at merali dot org>
* Edward Hervey <edward.hervey@collabora.co.uk>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <stdlib.h>
#include <string.h>
#include "mpegtsbase.h"
#include "tsdemux.h"
#include "gstmpegdesc.h"
#include "gstmpegdefs.h"
#include "mpegtspacketizer.h"
/* latency in mseconds */
#define TS_LATENCY 700
#define TABLE_ID_UNSET 0xFF
/* Size of the pendingbuffers array. */
#define TS_MAX_PENDING_BUFFERS 256
GST_DEBUG_CATEGORY_STATIC (ts_demux_debug);
#define GST_CAT_DEFAULT ts_demux_debug
static GQuark QUARK_TSDEMUX;
static GQuark QUARK_PID;
static GQuark QUARK_PCR;
static GQuark QUARK_OPCR;
static GQuark QUARK_PTS;
static GQuark QUARK_DTS;
static GQuark QUARK_OFFSET;
typedef enum
{
PENDING_PACKET_EMPTY = 0, /* No pending packet/buffer
* Push incoming buffers to the array */
PENDING_PACKET_HEADER, /* PES header needs to be parsed
* Push incoming buffers to the array */
PENDING_PACKET_BUFFER, /* Currently filling up output buffer
* Push incoming buffers to the bufferlist */
PENDING_PACKET_DISCONT /* Discontinuity in incoming packets
* Drop all incoming buffers */
} PendingPacketState;
typedef struct _TSDemuxStream TSDemuxStream;
struct _TSDemuxStream
{
MpegTSBaseStream stream;
GstPad *pad;
/* set to FALSE before a push and TRUE after */
gboolean pushed;
/* the return of the latest push */
GstFlowReturn flow_return;
/* Output data */
PendingPacketState state;
/* Pending buffers array. */
/* These buffers are stored in this array until the PES header (if needed)
* is succesfully parsed. */
GstBuffer *pendingbuffers[TS_MAX_PENDING_BUFFERS];
guint8 nbpending;
/* Current data to be pushed out */
GstBufferList *current;
GstBufferListIterator *currentit;
GList *currentlist;
GstClockTime pts;
};
#define VIDEO_CAPS \
GST_STATIC_CAPS (\
"video/mpeg, " \
"mpegversion = (int) { 1, 2, 4 }, " \
"systemstream = (boolean) FALSE; " \
"video/x-h264,stream-format=(string)byte-stream," \
"alignment=(string)nal;" \
"video/x-dirac;" \
"video/x-wmv," \
"wmvversion = (int) 3, " \
"format = (fourcc) WVC1" \
)
#define AUDIO_CAPS \
GST_STATIC_CAPS ( \
"audio/mpeg, " \
"mpegversion = (int) { 1, 4 };" \
"audio/x-lpcm, " \
"width = (int) { 16, 20, 24 }, " \
"rate = (int) { 48000, 96000 }, " \
"channels = (int) [ 1, 8 ], " \
"dynamic_range = (int) [ 0, 255 ], " \
"emphasis = (boolean) { FALSE, TRUE }, " \
"mute = (boolean) { FALSE, TRUE }; " \
"audio/x-ac3; audio/x-eac3;" \
"audio/x-dts;" \
"audio/x-private-ts-lpcm" \
)
/* Can also use the subpicture pads for text subtitles? */
#define SUBPICTURE_CAPS \
GST_STATIC_CAPS ("subpicture/x-pgs; video/x-dvd-subpicture")
static GstStaticPadTemplate video_template =
GST_STATIC_PAD_TEMPLATE ("video_%04x", GST_PAD_SRC,
GST_PAD_SOMETIMES,
VIDEO_CAPS);
static GstStaticPadTemplate audio_template =
GST_STATIC_PAD_TEMPLATE ("audio_%04x",
GST_PAD_SRC,
GST_PAD_SOMETIMES,
AUDIO_CAPS);
static GstStaticPadTemplate subpicture_template =
GST_STATIC_PAD_TEMPLATE ("subpicture_%04x",
GST_PAD_SRC,
GST_PAD_SOMETIMES,
SUBPICTURE_CAPS);
static GstStaticPadTemplate private_template =
GST_STATIC_PAD_TEMPLATE ("private_%04x",
GST_PAD_SRC,
GST_PAD_SOMETIMES,
GST_STATIC_CAPS_ANY);
enum
{
ARG_0,
PROP_PROGRAM_NUMBER,
PROP_EMIT_STATS,
/* FILL ME */
};
/* Pad functions */
static const GstQueryType *gst_ts_demux_srcpad_query_types (GstPad * pad);
static gboolean gst_ts_demux_srcpad_query (GstPad * pad, GstQuery * query);
/* mpegtsbase methods */
static void
gst_ts_demux_program_started (MpegTSBase * base, MpegTSBaseProgram * program);
static void
gst_ts_demux_program_stopped (MpegTSBase * base, MpegTSBaseProgram * program);
static GstFlowReturn
gst_ts_demux_push (MpegTSBase * base, MpegTSPacketizerPacket * packet,
MpegTSPacketizerSection * section);
static void
gst_ts_demux_stream_added (MpegTSBase * base, MpegTSBaseStream * stream,
MpegTSBaseProgram * program);
static void
gst_ts_demux_stream_removed (MpegTSBase * base, MpegTSBaseStream * stream);
static GstFlowReturn
find_timestamps (MpegTSBase * base, guint64 initoff, guint64 * offset);
static void gst_ts_demux_set_property (GObject * object, guint prop_id,
const GValue * value, GParamSpec * pspec);
static void gst_ts_demux_get_property (GObject * object, guint prop_id,
GValue * value, GParamSpec * pspec);
static void gst_ts_demux_finalize (GObject * object);
static GstFlowReturn
process_pcr (MpegTSBase * base, guint64 initoff, GstClockTime * pcr,
guint numpcr, gboolean isinitial);
static gboolean push_event (MpegTSBase * base, GstEvent * event);
static void _extra_init (GType type);
GST_BOILERPLATE_FULL (GstTSDemux, gst_ts_demux, MpegTSBase,
GST_TYPE_MPEGTS_BASE, _extra_init);
static void
_extra_init (GType type)
{
QUARK_TSDEMUX = g_quark_from_string ("tsdemux");
QUARK_PID = g_quark_from_string ("pid");
QUARK_PCR = g_quark_from_string ("pcr");
QUARK_OPCR = g_quark_from_string ("opcr");
QUARK_PTS = g_quark_from_string ("pts");
QUARK_DTS = g_quark_from_string ("dts");
QUARK_OFFSET = g_quark_from_string ("offset");
}
static void
gst_ts_demux_base_init (gpointer klass)
{
GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
gst_element_class_add_pad_template (element_class,
gst_static_pad_template_get (&video_template));
gst_element_class_add_pad_template (element_class,
gst_static_pad_template_get (&audio_template));
gst_element_class_add_pad_template (element_class,
gst_static_pad_template_get (&subpicture_template));
gst_element_class_add_pad_template (element_class,
gst_static_pad_template_get (&private_template));
gst_element_class_set_details_simple (element_class,
"MPEG transport stream demuxer",
"Codec/Demuxer",
"Demuxes MPEG2 transport streams",
"Zaheer Abbas Merali <zaheerabbas at merali dot org>\n"
"Edward Hervey <edward.hervey@collabora.co.uk>");
}
static void
gst_ts_demux_class_init (GstTSDemuxClass * klass)
{
GObjectClass *gobject_class;
MpegTSBaseClass *ts_class;
gobject_class = G_OBJECT_CLASS (klass);
gobject_class->set_property = gst_ts_demux_set_property;
gobject_class->get_property = gst_ts_demux_get_property;
gobject_class->finalize = gst_ts_demux_finalize;
g_object_class_install_property (gobject_class, PROP_PROGRAM_NUMBER,
g_param_spec_int ("program-number", "Program number",
"Program Number to demux for (-1 to ignore)", -1, G_MAXINT,
-1, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
g_object_class_install_property (gobject_class, PROP_EMIT_STATS,
g_param_spec_boolean ("emit-stats", "Emit statistics",
"Emit messages for every pcr/opcr/pts/dts", FALSE,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
ts_class = GST_MPEGTS_BASE_CLASS (klass);
ts_class->push = GST_DEBUG_FUNCPTR (gst_ts_demux_push);
ts_class->push_event = GST_DEBUG_FUNCPTR (push_event);
ts_class->program_started = GST_DEBUG_FUNCPTR (gst_ts_demux_program_started);
ts_class->program_stopped = GST_DEBUG_FUNCPTR (gst_ts_demux_program_stopped);
ts_class->stream_added = gst_ts_demux_stream_added;
ts_class->stream_removed = gst_ts_demux_stream_removed;
ts_class->find_timestamps = GST_DEBUG_FUNCPTR (find_timestamps);
}
static void
gst_ts_demux_init (GstTSDemux * demux, GstTSDemuxClass * klass)
{
demux->need_newsegment = TRUE;
demux->program_number = -1;
demux->duration = GST_CLOCK_TIME_NONE;
GST_MPEGTS_BASE (demux)->stream_size = sizeof (TSDemuxStream);
}
static void
gst_ts_demux_finalize (GObject * object)
{
if (G_OBJECT_CLASS (parent_class)->finalize)
G_OBJECT_CLASS (parent_class)->finalize (object);
}
static void
gst_ts_demux_set_property (GObject * object, guint prop_id,
const GValue * value, GParamSpec * pspec)
{
GstTSDemux *demux = GST_TS_DEMUX (object);
switch (prop_id) {
case PROP_PROGRAM_NUMBER:
/* FIXME: do something if program is switched as opposed to set at
* beginning */
demux->program_number = g_value_get_int (value);
break;
case PROP_EMIT_STATS:
demux->emit_statistics = g_value_get_boolean (value);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
}
}
static void
gst_ts_demux_get_property (GObject * object, guint prop_id,
GValue * value, GParamSpec * pspec)
{
GstTSDemux *demux = GST_TS_DEMUX (object);
switch (prop_id) {
case PROP_PROGRAM_NUMBER:
g_value_set_int (value, demux->program_number);
break;
case PROP_EMIT_STATS:
g_value_set_boolean (value, demux->emit_statistics);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
}
}
static const GstQueryType *
gst_ts_demux_srcpad_query_types (GstPad * pad)
{
static const GstQueryType query_types[] = {
GST_QUERY_DURATION,
0
};
return query_types;
}
static gboolean
gst_ts_demux_srcpad_query (GstPad * pad, GstQuery * query)
{
gboolean res = TRUE;
GstTSDemux *demux;
demux = GST_TS_DEMUX (gst_pad_get_parent (pad));
switch (GST_QUERY_TYPE (query)) {
case GST_QUERY_DURATION:
{
GstFormat format;
gst_query_parse_duration (query, &format, NULL);
/* can only get position in time */
if (format != GST_FORMAT_TIME)
goto wrong_format;
gst_query_set_duration (query, GST_FORMAT_TIME, demux->duration);
break;
}
default:
res = gst_pad_query_default (pad, query);
break;
}
done:
gst_object_unref (demux);
return res;
wrong_format:
{
GST_DEBUG_OBJECT (demux, "only query duration on TIME is supported");
res = FALSE;
goto done;
}
}
static gboolean
push_event (MpegTSBase * base, GstEvent * event)
{
GstTSDemux *demux = (GstTSDemux *) base;
guint i;
if (G_UNLIKELY (demux->program == NULL))
return FALSE;
for (i = 0; i < 0x2000; i++) {
if (demux->program->streams[i]) {
if (((TSDemuxStream *) demux->program->streams[i])->pad) {
gst_event_ref (event);
gst_pad_push_event (((TSDemuxStream *) demux->program->streams[i])->pad,
event);
}
}
}
return TRUE;
}
static GstFlowReturn
tsdemux_combine_flows (GstTSDemux * demux, TSDemuxStream * stream,
GstFlowReturn ret)
{
guint i;
/* Store the value */
stream->flow_return = ret;
/* any other error that is not-linked can be returned right away */
if (ret != GST_FLOW_NOT_LINKED)
goto done;
/* Only return NOT_LINKED if all other pads returned NOT_LINKED */
for (i = 0; i < 0x2000; i++) {
if (demux->program->streams[i]) {
stream = (TSDemuxStream *) demux->program->streams[i];
if (stream->pad) {
ret = stream->flow_return;
/* some other return value (must be SUCCESS but we can return
* other values as well) */
if (ret != GST_FLOW_NOT_LINKED)
goto done;
}
}
/* if we get here, all other pads were unlinked and we return
* NOT_LINKED then */
}
done:
return ret;
}
static GstPad *
create_pad_for_stream (MpegTSBase * base, MpegTSBaseStream * bstream,
MpegTSBaseProgram * program)
{
TSDemuxStream *stream = (TSDemuxStream *) bstream;
gchar *name = NULL;
GstCaps *caps = NULL;
GstPadTemplate *template = NULL;
guint8 *desc = NULL;
GstPad *pad = NULL;
GST_LOG ("Attempting to create pad for stream 0x%04x with stream_type %d",
bstream->pid, bstream->stream_type);
switch (bstream->stream_type) {
case ST_VIDEO_MPEG1:
case ST_VIDEO_MPEG2:
GST_LOG ("mpeg video");
template = gst_static_pad_template_get (&video_template);
name = g_strdup_printf ("video_%04x", bstream->pid);
caps = gst_caps_new_simple ("video/mpeg",
"mpegversion", G_TYPE_INT,
bstream->stream_type == ST_VIDEO_MPEG1 ? 1 : 2, "systemstream",
G_TYPE_BOOLEAN, FALSE, NULL);
break;
case ST_AUDIO_MPEG1:
case ST_AUDIO_MPEG2:
GST_LOG ("mpeg audio");
template = gst_static_pad_template_get (&audio_template);
name = g_strdup_printf ("audio_%04x", bstream->pid);
caps =
gst_caps_new_simple ("audio/mpeg", "mpegversion", G_TYPE_INT, 1,
NULL);
break;
case ST_PRIVATE_DATA:
GST_LOG ("private data");
desc =
mpegts_get_descriptor_from_stream ((MpegTSBaseStream *) stream,
DESC_DVB_AC3);
if (desc) {
GST_LOG ("ac3 audio");
template = gst_static_pad_template_get (&audio_template);
name = g_strdup_printf ("audio_%04x", bstream->pid);
caps = gst_caps_new_simple ("audio/x-ac3", NULL);
g_free (desc);
break;
}
desc =
mpegts_get_descriptor_from_stream ((MpegTSBaseStream *) stream,
DESC_DVB_ENHANCED_AC3);
if (desc) {
GST_LOG ("ac3 audio");
template = gst_static_pad_template_get (&audio_template);
name = g_strdup_printf ("audio_%04x", bstream->pid);
caps = gst_caps_new_simple ("audio/x-eac3", NULL);
g_free (desc);
break;
}
desc =
mpegts_get_descriptor_from_stream ((MpegTSBaseStream *) stream,
DESC_DVB_TELETEXT);
if (desc) {
GST_LOG ("teletext");
template = gst_static_pad_template_get (&private_template);
name = g_strdup_printf ("private_%04x", bstream->pid);
caps = gst_caps_new_simple ("private/teletext", NULL);
g_free (desc);
break;
}
desc =
mpegts_get_descriptor_from_stream ((MpegTSBaseStream *) stream,
DESC_DVB_SUBTITLING);
if (desc) {
GST_LOG ("subtitling");
template = gst_static_pad_template_get (&private_template);
name = g_strdup_printf ("private_%04x", bstream->pid);
caps = gst_caps_new_simple ("subpicture/x-dvb", NULL);
g_free (desc);
}
/* hack for itv hd (sid 10510, video pid 3401 */
if (program->program_number == 10510 && bstream->pid == 3401) {
template = gst_static_pad_template_get (&video_template);
name = g_strdup_printf ("video_%04x", bstream->pid);
caps = gst_caps_new_simple ("video/x-h264",
"stream-format", G_TYPE_STRING, "byte-stream",
"alignment", G_TYPE_STRING, "nal", NULL);
}
break;
case ST_HDV_AUX_V:
/* We don't expose those streams since they're only helper streams */
/* template = gst_static_pad_template_get (&private_template); */
/* name = g_strdup_printf ("private_%04x", bstream->pid); */
/* caps = gst_caps_new_simple ("hdv/aux-v", NULL); */
break;
case ST_HDV_AUX_A:
/* We don't expose those streams since they're only helper streams */
/* template = gst_static_pad_template_get (&private_template); */
/* name = g_strdup_printf ("private_%04x", bstream->pid); */
/* caps = gst_caps_new_simple ("hdv/aux-a", NULL); */
break;
case ST_PRIVATE_SECTIONS:
case ST_MHEG:
case ST_DSMCC:
case ST_DSMCC_A:
case ST_DSMCC_B:
case ST_DSMCC_C:
case ST_DSMCC_D:
base->is_pes[bstream->pid] = FALSE;
break;
case ST_AUDIO_AAC:
template = gst_static_pad_template_get (&audio_template);
name = g_strdup_printf ("audio_%04x", bstream->pid);
caps = gst_caps_new_simple ("audio/mpeg",
"mpegversion", G_TYPE_INT, 4, NULL);
break;
case ST_VIDEO_MPEG4:
template = gst_static_pad_template_get (&video_template);
name = g_strdup_printf ("video_%04x", bstream->pid);
caps = gst_caps_new_simple ("video/mpeg",
"mpegversion", G_TYPE_INT, 4,
"systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
break;
case ST_VIDEO_H264:
template = gst_static_pad_template_get (&video_template);
name = g_strdup_printf ("video_%04x", bstream->pid);
caps = gst_caps_new_simple ("video/x-h264",
"stream-format", G_TYPE_STRING, "byte-stream",
"alignment", G_TYPE_STRING, "nal", NULL);
break;
case ST_VIDEO_DIRAC:
desc =
mpegts_get_descriptor_from_stream ((MpegTSBaseStream *) stream,
DESC_REGISTRATION);
if (desc) {
if (DESC_LENGTH (desc) >= 4) {
if (DESC_REGISTRATION_format_identifier (desc) == 0x64726163) {
GST_LOG ("dirac");
/* dirac in hex */
template = gst_static_pad_template_get (&video_template);
name = g_strdup_printf ("video_%04x", bstream->pid);
caps = gst_caps_new_simple ("video/x-dirac", NULL);
}
}
g_free (desc);
}
break;
case ST_PRIVATE_EA: /* Try to detect a VC1 stream */
{
desc =
mpegts_get_descriptor_from_stream ((MpegTSBaseStream *) stream,
DESC_REGISTRATION);
if (desc) {
if (DESC_LENGTH (desc) >= 4) {
if (DESC_REGISTRATION_format_identifier (desc) == DRF_ID_VC1) {
GST_WARNING ("0xea private stream type found but no descriptor "
"for VC1. Assuming plain VC1.");
template = gst_static_pad_template_get (&video_template);
name = g_strdup_printf ("video_%04x", bstream->pid);
caps = gst_caps_new_simple ("video/x-wmv",
"wmvversion", G_TYPE_INT, 3,
"format", GST_TYPE_FOURCC, GST_MAKE_FOURCC ('W', 'V', 'C', '1'),
NULL);
}
}
g_free (desc);
}
break;
}
case ST_BD_AUDIO_AC3:
{
/* REGISTRATION DRF_ID_HDMV */
desc = mpegts_get_descriptor_from_program (program, DESC_REGISTRATION);
if (desc) {
if (DESC_REGISTRATION_format_identifier (desc) == DRF_ID_HDMV) {
template = gst_static_pad_template_get (&audio_template);
name = g_strdup_printf ("audio_%04x", bstream->pid);
caps = gst_caps_new_simple ("audio/x-eac3", NULL);
}
g_free (desc);
}
if (template)
break;
/* DVB_ENHANCED_AC3 */
desc =
mpegts_get_descriptor_from_stream ((MpegTSBaseStream *) stream,
DESC_DVB_ENHANCED_AC3);
if (desc) {
template = gst_static_pad_template_get (&audio_template);
name = g_strdup_printf ("audio_%04x", bstream->pid);
caps = gst_caps_new_simple ("audio/x-eac3", NULL);
g_free (desc);
break;
}
/* DVB_AC3 */
desc =
mpegts_get_descriptor_from_stream ((MpegTSBaseStream *) stream,
DESC_DVB_AC3);
if (!desc)
GST_WARNING ("AC3 stream type found but no corresponding "
"descriptor to differentiate between AC3 and EAC3. "
"Assuming plain AC3.");
else
g_free (desc);
template = gst_static_pad_template_get (&audio_template);
name = g_strdup_printf ("audio_%04x", bstream->pid);
caps = gst_caps_new_simple ("audio/x-ac3", NULL);
break;
}
case ST_BD_AUDIO_EAC3:
template = gst_static_pad_template_get (&audio_template);
name = g_strdup_printf ("audio_%04x", bstream->pid);
caps = gst_caps_new_simple ("audio/x-eac3", NULL);
break;
case ST_PS_AUDIO_DTS:
template = gst_static_pad_template_get (&audio_template);
name = g_strdup_printf ("audio_%04x", bstream->pid);
caps = gst_caps_new_simple ("audio/x-dts", NULL);
break;
case ST_PS_AUDIO_LPCM:
template = gst_static_pad_template_get (&audio_template);
name = g_strdup_printf ("audio_%04x", bstream->pid);
caps = gst_caps_new_simple ("audio/x-lpcm", NULL);
break;
case ST_BD_AUDIO_LPCM:
template = gst_static_pad_template_get (&audio_template);
name = g_strdup_printf ("audio_%04x", bstream->pid);
caps = gst_caps_new_simple ("audio/x-private-ts-lpcm", NULL);
break;
case ST_PS_DVD_SUBPICTURE:
template = gst_static_pad_template_get (&subpicture_template);
name = g_strdup_printf ("subpicture_%04x", bstream->pid);
caps = gst_caps_new_simple ("video/x-dvd-subpicture", NULL);
break;
case ST_BD_PGS_SUBPICTURE:
template = gst_static_pad_template_get (&subpicture_template);
name = g_strdup_printf ("subpicture_%04x", bstream->pid);
caps = gst_caps_new_simple ("subpicture/x-pgs", NULL);
break;
}
if (template && name && caps) {
GST_LOG ("stream:%p creating pad with name %s and caps %s", stream, name,
gst_caps_to_string (caps));
pad = gst_pad_new_from_template (template, name);
gst_pad_use_fixed_caps (pad);
gst_pad_set_caps (pad, caps);
gst_pad_set_query_type_function (pad, gst_ts_demux_srcpad_query_types);
gst_pad_set_query_function (pad, gst_ts_demux_srcpad_query);
gst_caps_unref (caps);
}
g_free (name);
return pad;
}
static void
gst_ts_demux_stream_added (MpegTSBase * base, MpegTSBaseStream * bstream,
MpegTSBaseProgram * program)
{
TSDemuxStream *stream = (TSDemuxStream *) bstream;
if (!stream->pad) {
/* Create the pad */
if (bstream->stream_type != 0xff)
stream->pad = create_pad_for_stream (base, bstream, program);
stream->pts = GST_CLOCK_TIME_NONE;
}
stream->flow_return = GST_FLOW_OK;
}
static void
gst_ts_demux_stream_removed (MpegTSBase * base, MpegTSBaseStream * bstream)
{
TSDemuxStream *stream = (TSDemuxStream *) bstream;
if (stream) {
if (stream->pad) {
/* Unref the pad, clear it */
gst_object_unref (stream->pad);
stream->pad = NULL;
}
stream->flow_return = GST_FLOW_NOT_LINKED;
}
}
static void
activate_pad_for_stream (GstTSDemux * tsdemux, TSDemuxStream * stream)
{
if (stream->pad) {
GST_DEBUG_OBJECT (tsdemux, "Activating pad %s:%s for stream %p",
GST_DEBUG_PAD_NAME (stream->pad), stream);
gst_pad_set_active (stream->pad, TRUE);
gst_element_add_pad ((GstElement *) tsdemux, stream->pad);
GST_DEBUG_OBJECT (stream->pad, "done adding pad");
} else
GST_WARNING_OBJECT (tsdemux, "stream %p has no pad", stream);
}
static void
gst_ts_demux_program_started (MpegTSBase * base, MpegTSBaseProgram * program)
{
GstTSDemux *demux = GST_TS_DEMUX (base);
if (demux->program_number == -1 ||
demux->program_number == program->program_number) {
guint i;
GST_LOG ("program %d started", program->program_number);
demux->program_number = program->program_number;
demux->program = program;
/* Activate all stream pads, the pads will already have been created */
/* FIXME : Actually, we don't want to activate *ALL* streams !
* For example, we don't want to expose HDV AUX private streams, we will just
* be using them directly for seeking and metadata. */
if (base->mode != BASE_MODE_SCANNING)
for (i = 0; i < 0x2000; i++)
if (program->streams[i])
activate_pad_for_stream (demux,
(TSDemuxStream *) program->streams[i]);
/* Inform scanner we have got our program */
demux->current_program_number = program->program_number;
}
}
static void
gst_ts_demux_program_stopped (MpegTSBase * base, MpegTSBaseProgram * program)
{
guint i;
GstTSDemux *demux = GST_TS_DEMUX (base);
TSDemuxStream *localstream = NULL;
GST_LOG ("program %d stopped", program->program_number);
if (program != demux->program)
return;
for (i = 0; i < 0x2000; i++) {
if (demux->program->streams[i]) {
localstream = (TSDemuxStream *) program->streams[i];
if (localstream->pad) {
GST_DEBUG ("HAVE PAD %s:%s", GST_DEBUG_PAD_NAME (localstream->pad));
if (gst_pad_is_active (localstream->pad))
gst_element_remove_pad (GST_ELEMENT_CAST (demux), localstream->pad);
else
gst_object_unref (localstream->pad);
localstream->pad = NULL;
}
}
}
demux->program = NULL;
demux->program_number = -1;
}
static gboolean
process_section (MpegTSBase * base)
{
GstTSDemux *demux = GST_TS_DEMUX (base);
gboolean based;
gboolean done = FALSE;
MpegTSPacketizerPacket packet;
MpegTSPacketizerPacketReturn pret;
while ((!done)
&& ((pret =
mpegts_packetizer_next_packet (base->packetizer,
&packet)) != PACKET_NEED_MORE)) {
if (G_UNLIKELY (pret == PACKET_BAD))
/* bad header, skip the packet */
goto next;
/* base PSI data */
if (packet.payload != NULL && mpegts_base_is_psi (base, &packet)) {
MpegTSPacketizerSection section;
based =
mpegts_packetizer_push_section (base->packetizer, &packet, &section);
if (G_UNLIKELY (!based))
/* bad section data */
goto next;
if (G_LIKELY (section.complete)) {
/* section complete */
GST_DEBUG ("Section Complete");
based = mpegts_base_handle_psi (base, &section);
gst_buffer_unref (section.buffer);
if (G_UNLIKELY (!based))
/* bad PSI table */
goto next;
}
if (demux->program != NULL) {
GST_DEBUG ("Got Program");
done = TRUE;
}
}
next:
mpegts_packetizer_clear_packet (base->packetizer, &packet);
}
return done;
}
static GstFlowReturn
find_timestamps (MpegTSBase * base, guint64 initoff, guint64 * offset)
{
GstFlowReturn ret = GST_FLOW_OK;
GstBuffer *buf;
gboolean done = FALSE;
GstFormat format = GST_FORMAT_BYTES;
gint64 total_bytes;
guint64 scan_offset;
guint i = 0;
GstClockTime initial, final;
GstTSDemux *demux = GST_TS_DEMUX (base);
GST_DEBUG ("Scanning for timestamps");
/* Flush what remained from before */
mpegts_packetizer_clear (base->packetizer);
/* Start scanning from know PAT offset */
while (!done) {
ret =
gst_pad_pull_range (base->sinkpad, i * 50 * MPEGTS_MAX_PACKETSIZE,
50 * MPEGTS_MAX_PACKETSIZE, &buf);
if (ret != GST_FLOW_OK)
goto beach;
mpegts_packetizer_push (base->packetizer, buf);
done = process_section (base);
i++;
}
mpegts_packetizer_clear (base->packetizer);
done = FALSE;
i = 1;
*offset = base->seek_offset;
/* Search for the first PCRs */
ret = process_pcr (base, base->first_pat_offset, &initial, 10, TRUE);
mpegts_packetizer_clear (base->packetizer);
/* Remove current program so we ensure looking for a PAT when scanning the
* for the final PCR */
mpegts_base_remove_program (base, demux->current_program_number);
if (ret != GST_FLOW_OK) {
GST_WARNING ("Problem getting initial PCRs");
goto beach;
}
/* Find end position */
if (G_UNLIKELY (!gst_pad_query_peer_duration (base->sinkpad, &format,
&total_bytes) || format != GST_FORMAT_BYTES)) {
GST_WARNING_OBJECT (base, "Couldn't get upstream size in bytes");
ret = GST_FLOW_ERROR;
mpegts_packetizer_clear (base->packetizer);
return ret;
}
GST_DEBUG ("Upstream is %" G_GINT64_FORMAT " bytes", total_bytes);
scan_offset = total_bytes - 4000 * MPEGTS_MAX_PACKETSIZE;
GST_DEBUG ("Scanning for last sync point between:%" G_GINT64_FORMAT
" and the end:%" G_GINT64_FORMAT, scan_offset, total_bytes);
while ((!done) && (scan_offset < total_bytes)) {
ret =
gst_pad_pull_range (base->sinkpad,
scan_offset, 50 * MPEGTS_MAX_PACKETSIZE, &buf);
if (ret != GST_FLOW_OK)
goto beach;
mpegts_packetizer_push (base->packetizer, buf);
done = process_section (base);
scan_offset += 50 * MPEGTS_MAX_PACKETSIZE;
}
mpegts_packetizer_clear (base->packetizer);
GST_DEBUG ("Searching PCR");
ret =
process_pcr (base, total_bytes - 4000 * MPEGTS_MAX_PACKETSIZE, &final, 10,
FALSE);
if (ret != GST_FLOW_OK) {
GST_DEBUG ("Problem getting last PCRs");
goto beach;
}
demux->duration = final - initial;
GST_DEBUG ("Done, duration:%" GST_TIME_FORMAT,
GST_TIME_ARGS (demux->duration));
beach:
mpegts_packetizer_clear (base->packetizer);
/* Remove current program */
mpegts_base_remove_program (base, demux->current_program_number);
return ret;
}
static GstFlowReturn
process_pcr (MpegTSBase * base, guint64 initoff, GstClockTime * pcr,
guint numpcr, gboolean isinitial)
{
GstTSDemux *demux = GST_TS_DEMUX (base);
GstFlowReturn ret = GST_FLOW_OK;
MpegTSBaseProgram *program;
GstBuffer *buf;
guint nbpcr, i = 0;
guint32 pcrmask, pcrpattern;
guint64 pcrs[50];
guint64 pcroffs[50];
GstByteReader br;
GST_DEBUG ("initoff:%" G_GUINT64_FORMAT ", numpcr:%d, isinitial:%d",
initoff, numpcr, isinitial);
/* Get the program */
program = demux->program;
if (G_UNLIKELY (program == NULL))
return GST_FLOW_ERROR;
/* First find the first X PCR */
nbpcr = 0;
/* Mask/pattern is PID:PCR_PID, AFC&0x02 */
/* sync_byte (0x47) : 8bits => 0xff
* transport_error_indicator : 1bit ACTIVATE
* payload_unit_start_indicator : 1bit IGNORE
* transport_priority : 1bit IGNORE
* PID : 13bit => 0x9f 0xff
* transport_scrambling_control : 2bit
* adaptation_field_control : 2bit
* continuity_counter : 4bit => 0x30
*/
pcrmask = 0xff9fff20;
pcrpattern = 0x47000020 | ((program->pcr_pid & 0x1fff) << 8);
for (i = 0; (i < 20) && (nbpcr < numpcr); i++) {
guint offset, size;
ret =
gst_pad_pull_range (base->sinkpad,
initoff + i * 500 * base->packetsize, 500 * base->packetsize, &buf);
if (G_UNLIKELY (ret != GST_FLOW_OK))
goto beach;
gst_byte_reader_init_from_buffer (&br, buf);
offset = 0;
size = GST_BUFFER_SIZE (buf);
/* FIXME : We should jump to next packet instead of scanning everything */
while ((size >= br.size) && (nbpcr < numpcr)
&& (offset =
gst_byte_reader_masked_scan_uint32 (&br, pcrmask, pcrpattern,
offset, size)) != -1) {
/* Potential PCR */
/* GST_DEBUG ("offset %" G_GUINT64_FORMAT, GST_BUFFER_OFFSET (buf) + offset);
GST_MEMDUMP ("something", GST_BUFFER_DATA (buf) + offset, 16);*/
if ((*(br.data + offset + 5)) & 0x10) {
guint16 pcr2;
guint64 pcr, pcr_ext;
pcr = ((guint64) GST_READ_UINT32_BE (br.data + offset + 6)) << 1;
pcr2 = GST_READ_UINT16_BE (br.data + offset + 10);
pcr |= (pcr2 & 0x8000) >> 15;
pcr_ext = (pcr2 & 0x01ff);
pcr = pcr * 300 + pcr_ext % 300;
GST_DEBUG ("Found PCR %" G_GUINT64_FORMAT " %" GST_TIME_FORMAT
" at offset %" G_GUINT64_FORMAT, pcr,
GST_TIME_ARGS (PCRTIME_TO_GSTTIME (pcr)),
GST_BUFFER_OFFSET (buf) + offset);
pcrs[nbpcr] = pcr;
pcroffs[nbpcr] = GST_BUFFER_OFFSET (buf) + offset;
/* Safeguard against bogus PCR (by detecting if it's the same as the
* previous one or wheter the difference with the previous one is
* greater than 10mins */
if (nbpcr > 1) {
if (pcrs[nbpcr] == pcrs[nbpcr - 1]) {
GST_WARNING ("Found same PCR at different offset");
} else if ((pcrs[nbpcr] - pcrs[nbpcr - 1]) >
(guint64) 10 * 60 * 27000000) {
GST_WARNING ("PCR differs with previous PCR by more than 10 mins");
} else
nbpcr += 1;
} else
nbpcr += 1;
}
/* Move offset forward by 1 */
size -= offset + 1;
offset += 1;
}
}
beach:
GST_DEBUG ("Found %d PCR", nbpcr);
if (nbpcr) {
if (isinitial)
*pcr = PCRTIME_TO_GSTTIME (pcrs[0]);
else
*pcr = PCRTIME_TO_GSTTIME (pcrs[nbpcr - 1]);
GST_DEBUG ("pcrdiff:%" GST_TIME_FORMAT " offsetdiff %" G_GUINT64_FORMAT,
GST_TIME_ARGS (PCRTIME_TO_GSTTIME (pcrs[nbpcr - 1] - pcrs[0])),
pcroffs[nbpcr - 1] - pcroffs[0]);
GST_DEBUG ("Estimated bitrate %" G_GUINT64_FORMAT,
gst_util_uint64_scale (GST_SECOND, pcroffs[nbpcr - 1] - pcroffs[0],
PCRTIME_TO_GSTTIME (pcrs[nbpcr - 1] - pcrs[0])));
GST_DEBUG ("Average PCR interval %" G_GUINT64_FORMAT,
(pcroffs[nbpcr - 1] - pcroffs[0]) / nbpcr);
}
/* Swallow any errors if it happened during the end scanning */
if (!isinitial)
ret = GST_FLOW_OK;
return ret;
}
static inline void
gst_ts_demux_record_pcr (GstTSDemux * demux, TSDemuxStream * stream,
guint64 pcr, guint64 offset)
{
MpegTSBaseStream *bs = (MpegTSBaseStream *) stream;
GST_LOG ("pid 0x%04x pcr:%" GST_TIME_FORMAT " at offset %"
G_GUINT64_FORMAT, bs->pid,
GST_TIME_ARGS (PCRTIME_TO_GSTTIME (pcr)), offset);
if (G_UNLIKELY (demux->emit_statistics)) {
GstStructure *st;
st = gst_structure_id_empty_new (QUARK_TSDEMUX);
gst_structure_id_set (st,
QUARK_PID, G_TYPE_UINT, bs->pid,
QUARK_OFFSET, G_TYPE_UINT64, offset, QUARK_PCR, G_TYPE_UINT64, pcr,
NULL);
gst_element_post_message (GST_ELEMENT_CAST (demux),
gst_message_new_element (GST_OBJECT (demux), st));
}
}
static inline void
gst_ts_demux_record_opcr (GstTSDemux * demux, TSDemuxStream * stream,
guint64 opcr, guint64 offset)
{
MpegTSBaseStream *bs = (MpegTSBaseStream *) stream;
GST_LOG ("pid 0x%04x opcr:%" GST_TIME_FORMAT " at offset %"
G_GUINT64_FORMAT, bs->pid,
GST_TIME_ARGS (PCRTIME_TO_GSTTIME (opcr)), offset);
if (G_UNLIKELY (demux->emit_statistics)) {
GstStructure *st;
st = gst_structure_id_empty_new (QUARK_TSDEMUX);
gst_structure_id_set (st,
QUARK_PID, G_TYPE_UINT, bs->pid,
QUARK_OFFSET, G_TYPE_UINT64, offset,
QUARK_OPCR, G_TYPE_UINT64, opcr, NULL);
gst_element_post_message (GST_ELEMENT_CAST (demux),
gst_message_new_element (GST_OBJECT (demux), st));
}
}
static inline void
gst_ts_demux_record_pts (GstTSDemux * demux, TSDemuxStream * stream,
guint64 pts, guint64 offset)
{
MpegTSBaseStream *bs = (MpegTSBaseStream *) stream;
GST_LOG ("pid 0x%04x pts:%" GST_TIME_FORMAT " at offset %"
G_GUINT64_FORMAT, bs->pid,
GST_TIME_ARGS (MPEGTIME_TO_GSTTIME (pts)), offset);
if (G_UNLIKELY (demux->emit_statistics)) {
GstStructure *st;
st = gst_structure_id_empty_new (QUARK_TSDEMUX);
gst_structure_id_set (st,
QUARK_PID, G_TYPE_UINT, bs->pid,
QUARK_OFFSET, G_TYPE_UINT64, offset, QUARK_PTS, G_TYPE_UINT64, pts,
NULL);
gst_element_post_message (GST_ELEMENT_CAST (demux),
gst_message_new_element (GST_OBJECT (demux), st));
}
}
static inline void
gst_ts_demux_record_dts (GstTSDemux * demux, TSDemuxStream * stream,
guint64 dts, guint64 offset)
{
MpegTSBaseStream *bs = (MpegTSBaseStream *) stream;
GST_LOG ("pid 0x%04x dts:%" GST_TIME_FORMAT " at offset %"
G_GUINT64_FORMAT, bs->pid,
GST_TIME_ARGS (MPEGTIME_TO_GSTTIME (dts)), offset);
if (G_UNLIKELY (demux->emit_statistics)) {
GstStructure *st;
st = gst_structure_id_empty_new (QUARK_TSDEMUX);
gst_structure_id_set (st,
QUARK_PID, G_TYPE_UINT, bs->pid,
QUARK_OFFSET, G_TYPE_UINT64, offset, QUARK_DTS, G_TYPE_UINT64, dts,
NULL);
gst_element_post_message (GST_ELEMENT_CAST (demux),
gst_message_new_element (GST_OBJECT (demux), st));
}
}
static GstFlowReturn
gst_ts_demux_parse_pes_header (GstTSDemux * demux, TSDemuxStream * stream)
{
GstFlowReturn res = GST_FLOW_OK;
guint8 *data;
guint32 length;
guint32 psc_stid;
guint8 stid;
guint16 pesplength;
guint8 PES_header_data_length = 0;
data = GST_BUFFER_DATA (stream->pendingbuffers[0]);
length = GST_BUFFER_SIZE (stream->pendingbuffers[0]);
GST_MEMDUMP ("Header buffer", data, MIN (length, 32));
/* packet_start_code_prefix 24
* stream_id 8*/
psc_stid = GST_READ_UINT32_BE (data);
data += 4;
length -= 4;
if (G_UNLIKELY ((psc_stid & 0xffffff00) != 0x00000100)) {
GST_WARNING ("WRONG PACKET START CODE! pid: 0x%x stream_type: 0x%x",
stream->stream.pid, stream->stream.stream_type);
goto discont;
}
stid = psc_stid & 0x000000ff;
GST_LOG ("stream_id:0x%02x", stid);
/* PES_packet_length 16 */
/* FIXME : store the expected pes length somewhere ? */
pesplength = GST_READ_UINT16_BE (data);
data += 2;
length -= 2;
GST_LOG ("PES_packet_length:%d", pesplength);
/* FIXME : Only parse header on streams which require it (see table 2-21) */
if (stid != 0xbf) {
guint8 p1, p2;
guint64 pts, dts;
p1 = *data++;
p2 = *data++;
PES_header_data_length = *data++ + 3;
length -= 3;
GST_LOG ("0x%02x 0x%02x 0x%02x", p1, p2, PES_header_data_length);
GST_LOG ("PES header data length:%d", PES_header_data_length);
/* '10' 2
* PES_scrambling_control 2
* PES_priority 1
* data_alignment_indicator 1
* copyright 1
* original_or_copy 1 */
if (G_UNLIKELY ((p1 & 0xc0) != 0x80)) {
GST_WARNING ("p1 >> 6 != 0x2");
goto discont;
}
/* PTS_DTS_flags 2
* ESCR_flag 1
* ES_rate_flag 1
* DSM_trick_mode_flag 1
* additional_copy_info_flag 1
* PES_CRC_flag 1
* PES_extension_flag 1*/
/* PES_header_data_length 8 */
if (G_UNLIKELY (length < PES_header_data_length)) {
GST_WARNING ("length < PES_header_data_length");
goto discont;
}
/* PTS 32 */
if ((p2 & 0x80)) { /* PTS */
READ_TS (data, pts, discont);
gst_ts_demux_record_pts (demux, stream, pts,
GST_BUFFER_OFFSET (stream->pendingbuffers[0]));
length -= 4;
GST_BUFFER_TIMESTAMP (stream->pendingbuffers[0]) =
MPEGTIME_TO_GSTTIME (pts);
if (!GST_CLOCK_TIME_IS_VALID (stream->pts)) {
stream->pts = GST_BUFFER_TIMESTAMP (stream->pendingbuffers[0]);
}
}
/* DTS 32 */
if ((p2 & 0x40)) { /* DTS */
READ_TS (data, dts, discont);
gst_ts_demux_record_dts (demux, stream, dts,
GST_BUFFER_OFFSET (stream->pendingbuffers[0]));
length -= 4;
}
/* ESCR 48 */
if ((p2 & 0x20)) {
GST_LOG ("ESCR present");
data += 6;
length -= 6;
}
/* ES_rate 24 */
if ((p2 & 0x10)) {
GST_LOG ("ES_rate present");
data += 3;
length -= 3;
}
/* DSM_trick_mode 8 */
if ((p2 & 0x08)) {
GST_LOG ("DSM_trick_mode present");
data += 1;
length -= 1;
}
}
/* Remove PES headers */
GST_BUFFER_DATA (stream->pendingbuffers[0]) += 6 + PES_header_data_length;
GST_BUFFER_SIZE (stream->pendingbuffers[0]) -= 6 + PES_header_data_length;
/* FIXME : responsible for switching to PENDING_PACKET_BUFFER and
* creating the bufferlist */
if (1) {
/* Append to the buffer list */
if (G_UNLIKELY (stream->current == NULL)) {
guint8 i;
/* Create a new bufferlist */
stream->current = gst_buffer_list_new ();
stream->currentit = gst_buffer_list_iterate (stream->current);
stream->currentlist = NULL;
gst_buffer_list_iterator_add_group (stream->currentit);
/* Push pending buffers into the list */
for (i = stream->nbpending; i; i--)
stream->currentlist =
g_list_prepend (stream->currentlist, stream->pendingbuffers[i - 1]);
memset (stream->pendingbuffers, 0, TS_MAX_PENDING_BUFFERS);
stream->nbpending = 0;
}
stream->state = PENDING_PACKET_BUFFER;
}
return res;
discont:
stream->state = PENDING_PACKET_DISCONT;
return res;
}
/* ONLY CALL THIS:
* * WITH packet->payload != NULL
* * WITH pending/current flushed out if beginning of new PES packet
*/
static inline void
gst_ts_demux_queue_data (GstTSDemux * demux, TSDemuxStream * stream,
MpegTSPacketizerPacket * packet)
{
GstBuffer *buf;
GST_DEBUG ("state:%d", stream->state);
buf = packet->buffer;
/* HACK : Instead of creating a new buffer, we just modify the data/size
* of the buffer to point to the payload */
GST_BUFFER_DATA (buf) = packet->payload;
GST_BUFFER_SIZE (buf) = packet->data_end - packet->payload;
if (stream->state == PENDING_PACKET_EMPTY) {
if (G_UNLIKELY (!packet->payload_unit_start_indicator)) {
stream->state = PENDING_PACKET_DISCONT;
GST_WARNING ("Didn't get the first packet of this PES");
} else {
GST_LOG ("EMPTY=>HEADER");
stream->state = PENDING_PACKET_HEADER;
if (stream->pad) {
GST_DEBUG ("Setting pad caps on buffer %p", buf);
gst_buffer_set_caps (buf, GST_PAD_CAPS (stream->pad));
}
}
}
if (stream->state == PENDING_PACKET_HEADER) {
GST_LOG ("HEADER: appending data to array");
/* Append to the array */
stream->pendingbuffers[stream->nbpending++] = buf;
/* parse the header */
gst_ts_demux_parse_pes_header (demux, stream);
} else if (stream->state == PENDING_PACKET_BUFFER) {
GST_LOG ("BUFFER: appending data to bufferlist");
stream->currentlist = g_list_prepend (stream->currentlist, buf);
}
return;
}
static GstFlowReturn
gst_ts_demux_push_pending_data (GstTSDemux * demux, TSDemuxStream * stream)
{
GstFlowReturn res = GST_FLOW_OK;
MpegTSBaseStream *bs = (MpegTSBaseStream *) stream;
guint i;
GstClockTime tinypts = GST_CLOCK_TIME_NONE;
GstClockTime stop = GST_CLOCK_TIME_NONE;
GstEvent *newsegmentevent;
GST_DEBUG ("stream:%p, pid:0x%04x stream_type:%d state:%d pad:%s:%s",
stream, bs->pid, bs->stream_type, stream->state,
GST_DEBUG_PAD_NAME (stream->pad));
if (G_UNLIKELY (stream->current == NULL)) {
GST_LOG ("stream->current == NULL");
goto beach;
}
if (G_UNLIKELY (stream->state == PENDING_PACKET_EMPTY)) {
GST_LOG ("EMPTY: returning");
goto beach;
}
/* We have a confirmed buffer, let's push it out */
if (stream->state == PENDING_PACKET_BUFFER) {
GST_LOG ("BUFFER: pushing out pending data");
stream->currentlist = g_list_reverse (stream->currentlist);
gst_buffer_list_iterator_add_list (stream->currentit, stream->currentlist);
gst_buffer_list_iterator_free (stream->currentit);
if (stream->pad) {
if (demux->need_newsegment) {
for (i = 0; i < 0x2000; i++) {
if (demux->program->streams[i]) {
if ((!GST_CLOCK_TIME_IS_VALID (tinypts))
|| (((TSDemuxStream *) demux->program->streams[i])->pts <
tinypts))
tinypts = ((TSDemuxStream *) demux->program->streams[i])->pts;
}
}
if (GST_CLOCK_TIME_IS_VALID (demux->duration))
stop = tinypts + demux->duration;
GST_DEBUG ("Sending newsegment event");
newsegmentevent =
gst_event_new_new_segment (0, 1.0, GST_FORMAT_TIME, tinypts, stop,
0);
push_event ((MpegTSBase *) demux, newsegmentevent);
demux->need_newsegment = FALSE;
}
GST_DEBUG_OBJECT (stream->pad, "Pushing buffer list ");
res = gst_pad_push_list (stream->pad, stream->current);
GST_DEBUG_OBJECT (stream->pad, "Returned %s", gst_flow_get_name (res));
/* FIXME : combine flow returns */
res = tsdemux_combine_flows (demux, stream, res);
GST_DEBUG_OBJECT (stream->pad, "combined %s", gst_flow_get_name (res));
} else {
gst_buffer_list_unref (stream->current);
}
}
beach:
/* Reset everything */
GST_LOG ("Resetting to EMPTY");
stream->state = PENDING_PACKET_EMPTY;
/* for (i = 0; i < stream->nbpending; i++) */
/* gst_buffer_unref (stream->pendingbuffers[i]); */
memset (stream->pendingbuffers, 0, TS_MAX_PENDING_BUFFERS);
stream->nbpending = 0;
stream->current = NULL;
return res;
}
static GstFlowReturn
gst_ts_demux_handle_packet (GstTSDemux * demux, TSDemuxStream * stream,
MpegTSPacketizerPacket * packet, MpegTSPacketizerSection * section)
{
GstFlowReturn res = GST_FLOW_OK;
GST_DEBUG ("buffer:%p, data:%p", GST_BUFFER_DATA (packet->buffer),
packet->data);
GST_LOG ("pid 0x%04x pusi:%d, afc:%d, cont:%d, payload:%p",
packet->pid,
packet->payload_unit_start_indicator,
packet->adaptation_field_control,
packet->continuity_counter, packet->payload);
if (section) {
GST_DEBUG ("section complete:%d, buffer size %d",
section->complete, GST_BUFFER_SIZE (section->buffer));
gst_buffer_unref (packet->buffer);
return res;
}
if (G_UNLIKELY (packet->payload_unit_start_indicator))
/* Flush previous data */
res = gst_ts_demux_push_pending_data (demux, stream);
if (packet->adaptation_field_control & 0x2) {
if (packet->afc_flags & MPEGTS_AFC_PCR_FLAG)
gst_ts_demux_record_pcr (demux, stream, packet->pcr,
GST_BUFFER_OFFSET (packet->buffer));
if (packet->afc_flags & MPEGTS_AFC_OPCR_FLAG)
gst_ts_demux_record_opcr (demux, stream, packet->opcr,
GST_BUFFER_OFFSET (packet->buffer));
}
if (packet->payload)
gst_ts_demux_queue_data (demux, stream, packet);
else
gst_buffer_unref (packet->buffer);
return res;
}
static GstFlowReturn
gst_ts_demux_push (MpegTSBase * base, MpegTSPacketizerPacket * packet,
MpegTSPacketizerSection * section)
{
GstTSDemux *demux = GST_TS_DEMUX_CAST (base);
TSDemuxStream *stream = NULL;
GstFlowReturn res = GST_FLOW_OK;
if (G_LIKELY (demux->program)) {
stream = (TSDemuxStream *) demux->program->streams[packet->pid];
if (stream) {
res = gst_ts_demux_handle_packet (demux, stream, packet, section);
} else if (packet->buffer)
gst_buffer_unref (packet->buffer);
} else {
if (packet->buffer)
gst_buffer_unref (packet->buffer);
}
return res;
}
gboolean
gst_ts_demux_plugin_init (GstPlugin * plugin)
{
GST_DEBUG_CATEGORY_INIT (ts_demux_debug, "tsdemux", 0,
"MPEG transport stream demuxer");
return gst_element_register (plugin, "tsdemux",
GST_RANK_SECONDARY, GST_TYPE_TS_DEMUX);
}