gstreamer/gst/mpegdemux/gstmpegtsdemux.c
Sebastian Dröge b1a12c3a2d mpegtsdemux: Don't use liboil functions
glibc memcpy() will be faster than liboil's/orc's for now anyway
and we can use orc's later, after orc has support for loop unrolling.
2010-06-14 14:48:02 +02:00

3269 lines
106 KiB
C

/*
* This library is licensed under 2 different licenses and you
* can choose to use it under the terms of either one of them. The
* two licenses are the MPL 1.1 and the LGPL.
*
* MPL:
*
* The contents of this file are subject to the Mozilla Public License
* Version 1.1 (the "License"); you may not use this file except in
* compliance with the License. You may obtain a copy of the License at
* http://www.mozilla.org/MPL/.
*
* Software distributed under the License is distributed on an "AS IS"
* basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
* License for the specific language governing rights and limitations
* under the License.
*
* LGPL:
*
* 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.
*
* The Original Code is Fluendo MPEG Demuxer plugin.
*
* The Initial Developer of the Original Code is Fluendo, S.A.
* Portions created by Fluendo, S.L. are Copyright (C) 2005,2006,2007,2008,2009
* Fluendo, S.A. All Rights Reserved.
*
* Contributor(s): Wim Taymans <wim@fluendo.com>
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <string.h>
#include <stdlib.h>
#include <gst/tag/tag.h>
#include "gstmpegdefs.h"
#include "gstmpegtsdemux.h"
#include "flutspatinfo.h"
#include "flutspmtinfo.h"
GST_DEBUG_CATEGORY_STATIC (gstmpegtsdemux_debug);
#define GST_CAT_DEFAULT (gstmpegtsdemux_debug)
/* elementfactory information */
#ifndef __always_inline
#if (__GNUC__ > 3) || (__GNUC__ == 3 && __GNUC_MINOR__ >= 1)
#define __always_inline inline __attribute__((always_inline))
#else
#define __always_inline inline
#endif
#endif
#ifndef DISABLE_INLINE
#define FORCE_INLINE __always_inline
#else
#define FORCE_INLINE
#endif
/* MPEG2Demux signals and args */
enum
{
/* FILL ME */
LAST_SIGNAL
};
#define DEFAULT_PROP_ES_PIDS ""
#define DEFAULT_PROP_CHECK_CRC TRUE
#define DEFAULT_PROP_PROGRAM_NUMBER -1
/* latency in mseconds */
#define TS_LATENCY 700
enum
{
PROP_0,
PROP_ES_PIDS,
PROP_CHECK_CRC,
PROP_PROGRAM_NUMBER,
PROP_PAT_INFO,
PROP_PMT_INFO,
};
#define GSTTIME_TO_BYTES(time) \
((time != -1) ? gst_util_uint64_scale (MAX(0,(gint64) ((time))), \
demux->bitrate, GST_SECOND) : -1)
#define BYTES_TO_GSTTIME(bytes) \
((bytes != -1) ? (gst_util_uint64_scale (bytes, GST_SECOND, \
demux->bitrate)) : -1)
#define VIDEO_CAPS \
GST_STATIC_CAPS (\
"video/mpeg, " \
"mpegversion = (int) { 1, 2, 4 }, " \
"systemstream = (boolean) FALSE; " \
"video/x-h264;" \
"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 sink_template = GST_STATIC_PAD_TEMPLATE ("sink",
GST_PAD_SINK,
GST_PAD_ALWAYS,
GST_STATIC_CAPS ("video/mpegts")
);
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);
static void gst_mpegts_demux_base_init (GstMpegTSDemuxClass * klass);
static void gst_mpegts_demux_class_init (GstMpegTSDemuxClass * klass);
static void gst_mpegts_demux_init (GstMpegTSDemux * demux);
static void gst_mpegts_demux_finalize (GstMpegTSDemux * demux);
static void gst_mpegts_demux_reset (GstMpegTSDemux * demux);
//static void gst_mpegts_demux_remove_pads (GstMpegTSDemux * demux);
static void gst_mpegts_demux_set_property (GObject * object, guint prop_id,
const GValue * value, GParamSpec * pspec);
static void gst_mpegts_demux_get_property (GObject * object, guint prop_id,
GValue * value, GParamSpec * pspec);
static gboolean gst_mpegts_demux_is_PMT (GstMpegTSDemux * demux, guint16 PID);
static gboolean gst_mpegts_demux_sink_event (GstPad * pad, GstEvent * event);
static gboolean gst_mpegts_demux_src_event (GstPad * pad, GstEvent * event);
static GstFlowReturn gst_mpegts_demux_chain (GstPad * pad, GstBuffer * buffer);
static gboolean gst_mpegts_demux_sink_setcaps (GstPad * pad, GstCaps * caps);
static GstClock *gst_mpegts_demux_provide_clock (GstElement * element);
static gboolean gst_mpegts_demux_src_pad_query (GstPad * pad, GstQuery * query);
static const GstQueryType *gst_mpegts_demux_src_pad_query_type (GstPad * pad);
static GstStateChangeReturn gst_mpegts_demux_change_state (GstElement * element,
GstStateChange transition);
static MpegTsPmtInfo *mpegts_demux_build_pmt_info (GstMpegTSDemux * demux,
guint16 pmt_pid);
static GstElementClass *parent_class = NULL;
/*static guint gst_mpegts_demux_signals[LAST_SIGNAL] = { 0 };*/
GType
gst_mpegts_demux_get_type (void)
{
static GType mpegts_demux_type = 0;
if (G_UNLIKELY (!mpegts_demux_type)) {
static const GTypeInfo mpegts_demux_info = {
sizeof (GstMpegTSDemuxClass),
(GBaseInitFunc) gst_mpegts_demux_base_init,
NULL,
(GClassInitFunc) gst_mpegts_demux_class_init,
NULL,
NULL,
sizeof (GstMpegTSDemux),
0,
(GInstanceInitFunc) gst_mpegts_demux_init,
};
mpegts_demux_type =
g_type_register_static (GST_TYPE_ELEMENT, "GstMpegTSDemux",
&mpegts_demux_info, 0);
GST_DEBUG_CATEGORY_INIT (gstmpegtsdemux_debug, "mpegtsdemux", 0,
"MPEG program stream demultiplexer element");
}
return mpegts_demux_type;
}
static void
gst_mpegts_demux_base_init (GstMpegTSDemuxClass * klass)
{
GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
klass->sink_template = gst_static_pad_template_get (&sink_template);
klass->video_template = gst_static_pad_template_get (&video_template);
klass->audio_template = gst_static_pad_template_get (&audio_template);
klass->subpicture_template =
gst_static_pad_template_get (&subpicture_template);
klass->private_template = gst_static_pad_template_get (&private_template);
gst_element_class_add_pad_template (element_class, klass->video_template);
gst_element_class_add_pad_template (element_class, klass->audio_template);
gst_element_class_add_pad_template (element_class,
klass->subpicture_template);
gst_element_class_add_pad_template (element_class, klass->private_template);
gst_element_class_add_pad_template (element_class, klass->sink_template);
gst_element_class_set_details_simple (element_class,
"The Fluendo MPEG Transport stream demuxer", "Codec/Demuxer",
"Demultiplexes MPEG2 Transport Streams", "Wim Taymans <wim@fluendo.com>");
}
static void
gst_mpegts_demux_class_init (GstMpegTSDemuxClass * klass)
{
GObjectClass *gobject_class;
GstElementClass *gstelement_class;
parent_class = g_type_class_peek_parent (klass);
gobject_class = (GObjectClass *) klass;
gstelement_class = (GstElementClass *) klass;
gobject_class->finalize = (GObjectFinalizeFunc) gst_mpegts_demux_finalize;
gobject_class->set_property = gst_mpegts_demux_set_property;
gobject_class->get_property = gst_mpegts_demux_get_property;
g_object_class_install_property (gobject_class, PROP_ES_PIDS,
g_param_spec_string ("es-pids",
"Colon separated list of PIDs containing Elementary Streams",
"PIDs to treat as Elementary Streams in the absence of a PMT, "
"eg 0x10:0x11:0x20", DEFAULT_PROP_ES_PIDS, G_PARAM_READWRITE));
g_object_class_install_property (gobject_class, PROP_CHECK_CRC,
g_param_spec_boolean ("check-crc", "Check CRC",
"Enable CRC checking", DEFAULT_PROP_CHECK_CRC, G_PARAM_READWRITE));
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,
DEFAULT_PROP_PROGRAM_NUMBER, G_PARAM_READWRITE));
g_object_class_install_property (gobject_class, PROP_PAT_INFO,
g_param_spec_value_array ("pat-info",
"GValueArray containing GObjects with properties",
"Array of GObjects containing information from the TS PAT "
"about all programs listed in the current Program Association "
"Table (PAT)",
g_param_spec_object ("flu-pat-streaminfo", "FluPATStreamInfo",
"Fluendo TS Demuxer PAT Stream info object",
MPEGTS_TYPE_PAT_INFO, G_PARAM_READABLE), G_PARAM_READABLE));
g_object_class_install_property (gobject_class, PROP_PMT_INFO,
g_param_spec_object ("pmt-info",
"Information about the current program",
"GObject with properties containing information from the TS PMT "
"about the currently selected program and its streams",
MPEGTS_TYPE_PMT_INFO, G_PARAM_READABLE));
gstelement_class->change_state = gst_mpegts_demux_change_state;
gstelement_class->provide_clock = gst_mpegts_demux_provide_clock;
}
static void
gst_mpegts_demux_init (GstMpegTSDemux * demux)
{
GstMpegTSDemuxClass *klass = GST_MPEGTS_DEMUX_GET_CLASS (demux);
demux->streams =
g_malloc0 (sizeof (GstMpegTSStream *) * (MPEGTS_MAX_PID + 1));
demux->sinkpad = gst_pad_new_from_template (klass->sink_template, "sink");
gst_pad_set_chain_function (demux->sinkpad, gst_mpegts_demux_chain);
gst_pad_set_event_function (demux->sinkpad, gst_mpegts_demux_sink_event);
gst_pad_set_setcaps_function (demux->sinkpad, gst_mpegts_demux_sink_setcaps);
gst_element_add_pad (GST_ELEMENT (demux), demux->sinkpad);
demux->elementary_pids = NULL;
demux->nb_elementary_pids = 0;
demux->check_crc = DEFAULT_PROP_CHECK_CRC;
demux->program_number = DEFAULT_PROP_PROGRAM_NUMBER;
demux->sync_lut = NULL;
demux->sync_lut_len = 0;
demux->bitrate = -1;
demux->num_packets = 0;
demux->pcr[0] = -1;
demux->pcr[1] = -1;
demux->cache_duration = GST_CLOCK_TIME_NONE;
demux->base_pts = GST_CLOCK_TIME_NONE;
}
static void
gst_mpegts_demux_finalize (GstMpegTSDemux * demux)
{
gst_mpegts_demux_reset (demux);
g_free (demux->streams);
G_OBJECT_CLASS (parent_class)->finalize (G_OBJECT (demux));
}
static void
gst_mpegts_demux_reset (GstMpegTSDemux * demux)
{
/* Clean up the streams and pads we allocated */
gint i;
for (i = 0; i < MPEGTS_MAX_PID + 1; i++) {
GstMpegTSStream *stream = demux->streams[i];
if (stream != NULL) {
if (stream->pad)
gst_element_remove_pad (GST_ELEMENT_CAST (demux), stream->pad);
if (stream->ES_info)
gst_mpeg_descriptor_free (stream->ES_info);
if (stream->PMT.entries)
g_array_free (stream->PMT.entries, TRUE);
if (stream->PMT.program_info)
gst_mpeg_descriptor_free (stream->PMT.program_info);
if (stream->PAT.entries)
g_array_free (stream->PAT.entries, TRUE);
switch (stream->PID_type) {
case PID_TYPE_ELEMENTARY:
gst_pes_filter_uninit (&stream->filter);
break;
case PID_TYPE_PROGRAM_ASSOCIATION:
case PID_TYPE_CONDITIONAL_ACCESS:
case PID_TYPE_PROGRAM_MAP:
gst_section_filter_uninit (&stream->section_filter);
break;
}
if (stream->pes_buffer) {
gst_buffer_unref (stream->pes_buffer);
stream->pes_buffer = NULL;
}
g_free (stream);
demux->streams[i] = NULL;
}
}
if (demux->clock) {
g_object_unref (demux->clock);
demux->clock = NULL;
}
}
#if 0
static void
gst_mpegts_demux_remove_pads (GstMpegTSDemux * demux)
{
/* remove pads we added in preparation for adding new ones */
/* FIXME: instead of walking all streams, we should retain a list only
* of streams that have added pads */
gint i;
if (demux->need_no_more_pads) {
gst_element_no_more_pads ((GstElement *) demux);
demux->need_no_more_pads = FALSE;
}
for (i = 0; i < MPEGTS_MAX_PID + 1; i++) {
GstMpegTSStream *stream = demux->streams[i];
if (stream != NULL) {
if (GST_IS_PAD (stream->pad)) {
gst_pad_push_event (stream->pad, gst_event_new_eos ());
gst_element_remove_pad (GST_ELEMENT_CAST (demux), stream->pad);
}
stream->pad = NULL;
if (stream->PID_type == PID_TYPE_ELEMENTARY)
gst_pes_filter_drain (&stream->filter);
}
}
}
#endif
static guint32 crc_tab[256] = {
0x00000000, 0x04c11db7, 0x09823b6e, 0x0d4326d9, 0x130476dc, 0x17c56b6b,
0x1a864db2, 0x1e475005, 0x2608edb8, 0x22c9f00f, 0x2f8ad6d6, 0x2b4bcb61,
0x350c9b64, 0x31cd86d3, 0x3c8ea00a, 0x384fbdbd, 0x4c11db70, 0x48d0c6c7,
0x4593e01e, 0x4152fda9, 0x5f15adac, 0x5bd4b01b, 0x569796c2, 0x52568b75,
0x6a1936c8, 0x6ed82b7f, 0x639b0da6, 0x675a1011, 0x791d4014, 0x7ddc5da3,
0x709f7b7a, 0x745e66cd, 0x9823b6e0, 0x9ce2ab57, 0x91a18d8e, 0x95609039,
0x8b27c03c, 0x8fe6dd8b, 0x82a5fb52, 0x8664e6e5, 0xbe2b5b58, 0xbaea46ef,
0xb7a96036, 0xb3687d81, 0xad2f2d84, 0xa9ee3033, 0xa4ad16ea, 0xa06c0b5d,
0xd4326d90, 0xd0f37027, 0xddb056fe, 0xd9714b49, 0xc7361b4c, 0xc3f706fb,
0xceb42022, 0xca753d95, 0xf23a8028, 0xf6fb9d9f, 0xfbb8bb46, 0xff79a6f1,
0xe13ef6f4, 0xe5ffeb43, 0xe8bccd9a, 0xec7dd02d, 0x34867077, 0x30476dc0,
0x3d044b19, 0x39c556ae, 0x278206ab, 0x23431b1c, 0x2e003dc5, 0x2ac12072,
0x128e9dcf, 0x164f8078, 0x1b0ca6a1, 0x1fcdbb16, 0x018aeb13, 0x054bf6a4,
0x0808d07d, 0x0cc9cdca, 0x7897ab07, 0x7c56b6b0, 0x71159069, 0x75d48dde,
0x6b93dddb, 0x6f52c06c, 0x6211e6b5, 0x66d0fb02, 0x5e9f46bf, 0x5a5e5b08,
0x571d7dd1, 0x53dc6066, 0x4d9b3063, 0x495a2dd4, 0x44190b0d, 0x40d816ba,
0xaca5c697, 0xa864db20, 0xa527fdf9, 0xa1e6e04e, 0xbfa1b04b, 0xbb60adfc,
0xb6238b25, 0xb2e29692, 0x8aad2b2f, 0x8e6c3698, 0x832f1041, 0x87ee0df6,
0x99a95df3, 0x9d684044, 0x902b669d, 0x94ea7b2a, 0xe0b41de7, 0xe4750050,
0xe9362689, 0xedf73b3e, 0xf3b06b3b, 0xf771768c, 0xfa325055, 0xfef34de2,
0xc6bcf05f, 0xc27dede8, 0xcf3ecb31, 0xcbffd686, 0xd5b88683, 0xd1799b34,
0xdc3abded, 0xd8fba05a, 0x690ce0ee, 0x6dcdfd59, 0x608edb80, 0x644fc637,
0x7a089632, 0x7ec98b85, 0x738aad5c, 0x774bb0eb, 0x4f040d56, 0x4bc510e1,
0x46863638, 0x42472b8f, 0x5c007b8a, 0x58c1663d, 0x558240e4, 0x51435d53,
0x251d3b9e, 0x21dc2629, 0x2c9f00f0, 0x285e1d47, 0x36194d42, 0x32d850f5,
0x3f9b762c, 0x3b5a6b9b, 0x0315d626, 0x07d4cb91, 0x0a97ed48, 0x0e56f0ff,
0x1011a0fa, 0x14d0bd4d, 0x19939b94, 0x1d528623, 0xf12f560e, 0xf5ee4bb9,
0xf8ad6d60, 0xfc6c70d7, 0xe22b20d2, 0xe6ea3d65, 0xeba91bbc, 0xef68060b,
0xd727bbb6, 0xd3e6a601, 0xdea580d8, 0xda649d6f, 0xc423cd6a, 0xc0e2d0dd,
0xcda1f604, 0xc960ebb3, 0xbd3e8d7e, 0xb9ff90c9, 0xb4bcb610, 0xb07daba7,
0xae3afba2, 0xaafbe615, 0xa7b8c0cc, 0xa379dd7b, 0x9b3660c6, 0x9ff77d71,
0x92b45ba8, 0x9675461f, 0x8832161a, 0x8cf30bad, 0x81b02d74, 0x857130c3,
0x5d8a9099, 0x594b8d2e, 0x5408abf7, 0x50c9b640, 0x4e8ee645, 0x4a4ffbf2,
0x470cdd2b, 0x43cdc09c, 0x7b827d21, 0x7f436096, 0x7200464f, 0x76c15bf8,
0x68860bfd, 0x6c47164a, 0x61043093, 0x65c52d24, 0x119b4be9, 0x155a565e,
0x18197087, 0x1cd86d30, 0x029f3d35, 0x065e2082, 0x0b1d065b, 0x0fdc1bec,
0x3793a651, 0x3352bbe6, 0x3e119d3f, 0x3ad08088, 0x2497d08d, 0x2056cd3a,
0x2d15ebe3, 0x29d4f654, 0xc5a92679, 0xc1683bce, 0xcc2b1d17, 0xc8ea00a0,
0xd6ad50a5, 0xd26c4d12, 0xdf2f6bcb, 0xdbee767c, 0xe3a1cbc1, 0xe760d676,
0xea23f0af, 0xeee2ed18, 0xf0a5bd1d, 0xf464a0aa, 0xf9278673, 0xfde69bc4,
0x89b8fd09, 0x8d79e0be, 0x803ac667, 0x84fbdbd0, 0x9abc8bd5, 0x9e7d9662,
0x933eb0bb, 0x97ffad0c, 0xafb010b1, 0xab710d06, 0xa6322bdf, 0xa2f33668,
0xbcb4666d, 0xb8757bda, 0xb5365d03, 0xb1f740b4
};
/*This function fills the value of negotiated packetsize at sinkpad*/
static gboolean
gst_mpegts_demux_sink_setcaps (GstPad * pad, GstCaps * caps)
{
GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (gst_pad_get_parent (pad));
GstStructure *structure = NULL;
structure = gst_caps_get_structure (caps, 0);
GST_DEBUG_OBJECT (demux, "setcaps called with %" GST_PTR_FORMAT, caps);
if (!gst_structure_get_int (structure, "packetsize", &demux->packetsize)) {
GST_DEBUG_OBJECT (demux, "packetsize parameter not found in sink caps");
}
gst_object_unref (demux);
return TRUE;
}
static FORCE_INLINE guint32
gst_mpegts_demux_calc_crc32 (guint8 * data, guint datalen)
{
gint i;
guint32 crc = 0xffffffff;
for (i = 0; i < datalen; i++) {
crc = (crc << 8) ^ crc_tab[((crc >> 24) ^ *data++) & 0xff];
}
return crc;
}
static FORCE_INLINE gboolean
gst_mpegts_is_dirac_stream (GstMpegTSStream * stream)
{
gboolean is_dirac = FALSE;
if (stream->stream_type != ST_VIDEO_DIRAC)
return FALSE;
if (stream->ES_info != NULL) {
guint8 *dirac_desc;
/* Check for a Registration Descriptor to confirm this is dirac */
dirac_desc = gst_mpeg_descriptor_find (stream->ES_info, DESC_REGISTRATION);
if (dirac_desc != NULL && DESC_LENGTH (dirac_desc) >= 4) {
if (DESC_REGISTRATION_format_identifier (dirac_desc) == 0x64726163) { /* 'drac' in hex */
is_dirac = TRUE;
}
} else {
/* Check for old mapping as originally specified too */
dirac_desc = gst_mpeg_descriptor_find (stream->ES_info,
DESC_DIRAC_TC_PRIVATE);
if (dirac_desc != NULL && DESC_LENGTH (dirac_desc) == 0)
is_dirac = TRUE;
}
}
return is_dirac;
}
static FORCE_INLINE gboolean
gst_mpegts_stream_is_video (GstMpegTSStream * stream)
{
switch (stream->stream_type) {
case ST_VIDEO_MPEG1:
case ST_VIDEO_MPEG2:
case ST_VIDEO_MPEG4:
case ST_VIDEO_H264:
return TRUE;
case ST_VIDEO_DIRAC:
return gst_mpegts_is_dirac_stream (stream);
}
return FALSE;
}
static gboolean
gst_mpegts_demux_is_reserved_PID (GstMpegTSDemux * demux, guint16 PID)
{
return (PID >= PID_RESERVED_FIRST) && (PID < PID_RESERVED_LAST);
}
/* This function assumes that provided PID never will be greater than
* MPEGTS_MAX_PID (13 bits), this is currently guaranteed as everywhere in
* the code recovered PID at maximum is 13 bits long.
*/
static FORCE_INLINE GstMpegTSStream *
gst_mpegts_demux_get_stream_for_PID (GstMpegTSDemux * demux, guint16 PID)
{
GstMpegTSStream *stream = NULL;
stream = demux->streams[PID];
if (G_UNLIKELY (stream == NULL)) {
stream = g_new0 (GstMpegTSStream, 1);
stream->demux = demux;
stream->PID = PID;
stream->pad = NULL;
stream->base_PCR = -1;
stream->last_PCR = -1;
stream->last_PCR_difference = -1;
stream->PMT.version_number = -1;
stream->PAT.version_number = -1;
stream->PMT_pid = MPEGTS_MAX_PID + 1;
stream->flags |= MPEGTS_STREAM_FLAG_STREAM_TYPE_UNKNOWN;
stream->pes_buffer_in_sync = FALSE;
switch (PID) {
/* check for fixed mapping */
case PID_PROGRAM_ASSOCIATION_TABLE:
stream->PID_type = PID_TYPE_PROGRAM_ASSOCIATION;
/* initialise section filter */
gst_section_filter_init (&stream->section_filter);
break;
case PID_CONDITIONAL_ACCESS_TABLE:
stream->PID_type = PID_TYPE_CONDITIONAL_ACCESS;
/* initialise section filter */
gst_section_filter_init (&stream->section_filter);
break;
case PID_NULL_PACKET:
stream->PID_type = PID_TYPE_NULL_PACKET;
break;
default:
/* mark reserved PIDs */
if (gst_mpegts_demux_is_reserved_PID (demux, PID)) {
stream->PID_type = PID_TYPE_RESERVED;
} else {
/* check if PMT found in PAT */
if (gst_mpegts_demux_is_PMT (demux, PID)) {
stream->PID_type = PID_TYPE_PROGRAM_MAP;
/* initialise section filter */
gst_section_filter_init (&stream->section_filter);
} else
stream->PID_type = PID_TYPE_UNKNOWN;
}
break;
}
GST_DEBUG_OBJECT (demux, "creating stream %p for PID 0x%04x, PID_type %d",
stream, PID, stream->PID_type);
demux->streams[PID] = stream;
}
return stream;
}
static gboolean
gst_mpegts_demux_fill_stream (GstMpegTSStream * stream, guint8 id,
guint8 stream_type)
{
GstPadTemplate *template;
gchar *name;
GstMpegTSDemuxClass *klass;
GstMpegTSDemux *demux;
GstCaps *caps;
if (stream->stream_type && stream->stream_type != stream_type)
goto wrong_type;
demux = stream->demux;
klass = GST_MPEGTS_DEMUX_GET_CLASS (demux);
name = NULL;
template = NULL;
caps = NULL;
switch (stream_type) {
case ST_VIDEO_MPEG1:
case ST_VIDEO_MPEG2:
template = klass->video_template;
name = g_strdup_printf ("video_%04x", stream->PID);
caps = gst_caps_new_simple ("video/mpeg",
"mpegversion", G_TYPE_INT, stream_type == ST_VIDEO_MPEG1 ? 1 : 2,
"systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
break;
case ST_AUDIO_MPEG1:
case ST_AUDIO_MPEG2:
template = klass->audio_template;
name = g_strdup_printf ("audio_%04x", stream->PID);
caps = gst_caps_new_simple ("audio/mpeg",
"mpegversion", G_TYPE_INT, 1, NULL);
break;
case ST_PRIVATE_DATA:
/* check if there is an AC3 descriptor associated with this stream
* from the PMT */
if (gst_mpeg_descriptor_find (stream->ES_info, DESC_DVB_AC3)) {
template = klass->audio_template;
name = g_strdup_printf ("audio_%04x", stream->PID);
caps = gst_caps_new_simple ("audio/x-ac3", NULL);
} else if (gst_mpeg_descriptor_find (stream->ES_info,
DESC_DVB_ENHANCED_AC3)) {
template = klass->private_template;
name = g_strdup_printf ("audio_%04x", stream->PID);
caps = gst_caps_new_simple ("audio/x-eac3", NULL);
} else if (gst_mpeg_descriptor_find (stream->ES_info, DESC_DVB_TELETEXT)) {
template = klass->private_template;
name = g_strdup_printf ("private_%04x", stream->PID);
caps = gst_caps_new_simple ("private/teletext", NULL);
} else if (gst_mpeg_descriptor_find (stream->ES_info,
DESC_DVB_SUBTITLING)) {
template = klass->private_template;
name = g_strdup_printf ("private_%04x", stream->PID);
caps = gst_caps_new_simple ("private/x-dvbsub", NULL);
}
break;
case ST_HDV_AUX_V:
template = klass->private_template;
name = g_strdup_printf ("private_%04x", stream->PID);
caps = gst_caps_new_simple ("hdv/aux-v", NULL);
break;
case ST_HDV_AUX_A:
template = klass->private_template;
name = g_strdup_printf ("private_%04x", stream->PID);
caps = gst_caps_new_simple ("hdv/aux-a", NULL);
break;
case ST_PRIVATE_SECTIONS:
case ST_MHEG:
case ST_DSMCC:
break;
case ST_AUDIO_AAC:
template = klass->audio_template;
name = g_strdup_printf ("audio_%04x", stream->PID);
caps = gst_caps_new_simple ("audio/mpeg",
"mpegversion", G_TYPE_INT, 4, NULL);
break;
case ST_VIDEO_MPEG4:
template = klass->video_template;
name = g_strdup_printf ("video_%04x", stream->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 = klass->video_template;
name = g_strdup_printf ("video_%04x", stream->PID);
caps = gst_caps_new_simple ("video/x-h264", NULL);
break;
case ST_VIDEO_DIRAC:
if (gst_mpegts_is_dirac_stream (stream)) {
template = klass->video_template;
name = g_strdup_printf ("video_%04x", stream->PID);
caps = gst_caps_new_simple ("video/x-dirac", NULL);
}
break;
case ST_PRIVATE_EA: /* Try to detect a VC1 stream */
{
guint8 *desc = NULL;
if (stream->ES_info)
desc = gst_mpeg_descriptor_find (stream->ES_info, DESC_REGISTRATION);
if (!(desc && DESC_REGISTRATION_format_identifier (desc) == DRF_ID_VC1)) {
GST_WARNING ("0xea private stream type found but no descriptor "
"for VC1. Assuming plain VC1.");
}
template = klass->video_template;
name = g_strdup_printf ("video_%04x", stream->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);
break;
}
case ST_BD_AUDIO_AC3:
{
GstMpegTSStream *PMT_stream =
gst_mpegts_demux_get_stream_for_PID (stream->demux, stream->PMT_pid);
GstMPEGDescriptor *program_info = PMT_stream->PMT.program_info;
guint8 *desc = NULL;
if (program_info)
desc = gst_mpeg_descriptor_find (program_info, DESC_REGISTRATION);
if (desc && DESC_REGISTRATION_format_identifier (desc) == DRF_ID_HDMV) {
template = klass->audio_template;
name = g_strdup_printf ("audio_%04x", stream->PID);
caps = gst_caps_new_simple ("audio/x-eac3", NULL);
} else if (stream->ES_info && gst_mpeg_descriptor_find (stream->ES_info,
DESC_DVB_ENHANCED_AC3)) {
template = klass->audio_template;
name = g_strdup_printf ("audio_%04x", stream->PID);
caps = gst_caps_new_simple ("audio/x-eac3", NULL);
} else {
if (!stream->ES_info ||
!gst_mpeg_descriptor_find (stream->ES_info, DESC_DVB_AC3)) {
GST_WARNING ("AC3 stream type found but no corresponding "
"descriptor to differentiate between AC3 and EAC3. "
"Assuming plain AC3.");
}
template = klass->audio_template;
name = g_strdup_printf ("audio_%04x", stream->PID);
caps = gst_caps_new_simple ("audio/x-ac3", NULL);
}
break;
}
case ST_BD_AUDIO_EAC3:
template = klass->audio_template;
name = g_strdup_printf ("audio_%04x", stream->PID);
caps = gst_caps_new_simple ("audio/x-eac3", NULL);
break;
case ST_PS_AUDIO_DTS:
case ST_BD_AUDIO_DTS:
case ST_BD_AUDIO_DTS_HD:
case ST_BD_AUDIO_DTS_HD_MASTER_AUDIO:
template = klass->audio_template;
name = g_strdup_printf ("audio_%04x", stream->PID);
caps = gst_caps_new_simple ("audio/x-dts", NULL);
break;
case ST_PS_AUDIO_LPCM:
template = klass->audio_template;
name = g_strdup_printf ("audio_%04x", stream->PID);
caps = gst_caps_new_simple ("audio/x-lpcm", NULL);
break;
case ST_BD_AUDIO_LPCM:
template = klass->audio_template;
name = g_strdup_printf ("audio_%04x", stream->PID);
caps = gst_caps_new_simple ("audio/x-private-ts-lpcm", NULL);
break;
case ST_PS_DVD_SUBPICTURE:
template = klass->subpicture_template;
name = g_strdup_printf ("subpicture_%04x", stream->PID);
caps = gst_caps_new_simple ("video/x-dvd-subpicture", NULL);
break;
case ST_BD_PGS_SUBPICTURE:
template = klass->subpicture_template;
name = g_strdup_printf ("subpicture_%04x", stream->PID);
caps = gst_caps_new_simple ("subpicture/x-pgs", NULL);
break;
default:
break;
}
if (name == NULL || template == NULL || caps == NULL)
return FALSE;
stream->stream_type = stream_type;
stream->id = id;
GST_DEBUG ("creating new pad %s", name);
stream->pad = gst_pad_new_from_template (template, name);
gst_pad_use_fixed_caps (stream->pad);
gst_pad_set_caps (stream->pad, caps);
gst_caps_unref (caps);
gst_pad_set_query_function (stream->pad,
GST_DEBUG_FUNCPTR (gst_mpegts_demux_src_pad_query));
gst_pad_set_query_type_function (stream->pad,
GST_DEBUG_FUNCPTR (gst_mpegts_demux_src_pad_query_type));
gst_pad_set_event_function (stream->pad,
GST_DEBUG_FUNCPTR (gst_mpegts_demux_src_event));
g_free (name);
return TRUE;
wrong_type:
{
return FALSE;
}
}
static FORCE_INLINE gboolean
mpegts_is_elem_pid (GstMpegTSDemux * demux, guint16 PID)
{
int i;
/* check if it's in our partial ts pid list */
for (i = 0; i < demux->nb_elementary_pids; i++) {
if (demux->elementary_pids[i] == PID) {
return TRUE;
}
}
return FALSE;
}
static gboolean
gst_mpegts_demux_setup_base_pts (GstMpegTSDemux * demux, gint64 pts)
{
GstMpegTSStream *PCR_stream;
GstMpegTSStream *PMT_stream;
guint64 base_PCR;
/* for the reference start time we need to consult the PCR_PID of the
* current PMT */
if (demux->current_PMT == 0)
goto no_pmt_stream;
PMT_stream = demux->streams[demux->current_PMT];
if (PMT_stream == NULL)
goto no_pmt_stream;
PCR_stream = demux->streams[PMT_stream->PMT.PCR_PID];
if (PCR_stream == NULL)
goto no_pcr_stream;
if (PCR_stream->base_PCR == -1) {
GST_DEBUG_OBJECT (demux, "no base PCR, using last PCR %" G_GUINT64_FORMAT,
PCR_stream->last_PCR);
PCR_stream->base_PCR = PCR_stream->last_PCR;
} else {
GST_DEBUG_OBJECT (demux, "using base PCR %" G_GUINT64_FORMAT,
PCR_stream->base_PCR);
}
if (PCR_stream->last_PCR == -1) {
GST_DEBUG_OBJECT (demux, "no last PCR, using PTS %" G_GUINT64_FORMAT, pts);
PCR_stream->base_PCR = pts;
PCR_stream->last_PCR = pts;
}
base_PCR = PCR_stream->base_PCR;
demux->base_pts = MPEGTIME_TO_GSTTIME (base_PCR);
if (demux->base_pts == GST_CLOCK_TIME_NONE)
return FALSE;
return TRUE;
no_pmt_stream:
{
GST_DEBUG_OBJECT (demux, "no PMT stream found");
return FALSE;
}
no_pcr_stream:
{
GST_DEBUG_OBJECT (demux, "no PCR stream found");
return FALSE;
}
}
static gboolean
gst_mpegts_demux_send_new_segment (GstMpegTSDemux * demux,
GstMpegTSStream * stream, gint64 pts)
{
GstClockTime time;
/* base_pts needs to have been set up by a call to
* gst_mpegts_demux_setup_base_pts() before calling this function */
if (demux->base_pts == GST_CLOCK_TIME_NONE)
goto no_base_time;
time = demux->base_pts;
GST_DEBUG_OBJECT (demux, "segment PTS to time: %"
GST_TIME_FORMAT, GST_TIME_ARGS (time));
if (demux->clock && demux->clock_base == GST_CLOCK_TIME_NONE) {
demux->clock_base = gst_clock_get_time (demux->clock);
gst_clock_set_calibration (demux->clock,
gst_clock_get_internal_time (demux->clock), demux->clock_base, 1, 1);
}
gst_pad_push_event (stream->pad,
gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_TIME, time, -1, 0));
return TRUE;
/* ERRORS */
no_base_time:
{
/* check if it's in our partial ts pid list */
if (mpegts_is_elem_pid (demux, stream->PID)) {
GST_DEBUG_OBJECT (demux,
"Elementary PID, using pts %" G_GUINT64_FORMAT, pts);
time = MPEGTIME_TO_GSTTIME (pts) + stream->base_time;
GST_DEBUG_OBJECT (demux, "segment PTS to (%" G_GUINT64_FORMAT ") time: %"
G_GUINT64_FORMAT, pts, time);
gst_pad_push_event (stream->pad,
gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_TIME, time, -1, 0));
return TRUE;
}
}
return FALSE;
}
static void
gst_mpegts_demux_send_tags_for_stream (GstMpegTSDemux * demux,
GstMpegTSStream * stream)
{
GstTagList *list = NULL;
if (stream->ES_info) {
guint8 *iso639_languages =
gst_mpeg_descriptor_find (stream->ES_info, DESC_ISO_639_LANGUAGE);
if (iso639_languages) {
if (DESC_ISO_639_LANGUAGE_codes_n (iso639_languages)) {
const gchar *lc;
gchar lang_code[4];
gchar *language_n;
language_n = (gchar *)
DESC_ISO_639_LANGUAGE_language_code_nth (iso639_languages, 0);
lang_code[0] = language_n[0];
lang_code[1] = language_n[1];
lang_code[2] = language_n[2];
lang_code[3] = 0;
if (!list)
list = gst_tag_list_new ();
/* descriptor contains ISO 639-2 code, we want the ISO 639-1 code */
lc = gst_tag_get_language_code (lang_code);
gst_tag_list_add (list, GST_TAG_MERGE_REPLACE,
GST_TAG_LANGUAGE_CODE, (lc) ? lc : lang_code, NULL);
}
}
}
if (list) {
GST_DEBUG_OBJECT (demux, "Sending tags %p for pad %s:%s",
list, GST_DEBUG_PAD_NAME (stream->pad));
gst_element_found_tags_for_pad (GST_ELEMENT (demux), stream->pad, list);
}
}
static GstFlowReturn
gst_mpegts_demux_combine_flows (GstMpegTSDemux * demux,
GstMpegTSStream * stream, GstFlowReturn ret)
{
gint i;
/* store the value */
stream->last_ret = ret;
/* if it's success we can return the value right away */
if (GST_FLOW_IS_SUCCESS (ret))
goto done;
/* 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 < MPEGTS_MAX_PID + 1; i++) {
if (!(stream = demux->streams[i]))
continue;
/* some other return value (must be SUCCESS but we can return
* other values as well) */
ret = stream->last_ret;
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 GstFlowReturn
gst_mpegts_demux_data_cb (GstPESFilter * filter, gboolean first,
GstBuffer * buffer, GstMpegTSStream * stream)
{
GstMpegTSDemux *demux;
GstFlowReturn ret;
GstPad *srcpad;
gint64 pts;
GstClockTime time;
demux = stream->demux;
srcpad = stream->pad;
GST_DEBUG_OBJECT (demux, "got data on PID 0x%04x", stream->PID);
if (first && filter->pts != -1) {
pts = filter->pts;
time = MPEGTIME_TO_GSTTIME (pts) + stream->base_time;
if ((stream->last_time > 0 && stream->last_time < time &&
time - stream->last_time > GST_SECOND * 60 * 10)
|| (stream->last_time > time
&& stream->last_time - time > GST_SECOND * 60 * 10)) {
/* check first to see if we're in middle of detecting a discont in PCR.
* if we are we're not sure what timestamp the buffer should have, best
* to drop. */
if (stream->PMT_pid <= MPEGTS_MAX_PID && demux->streams[stream->PMT_pid]
&& demux->streams[demux->streams[stream->PMT_pid]->PMT.PCR_PID]
&& demux->streams[demux->streams[stream->PMT_pid]->PMT.
PCR_PID]->discont_PCR) {
GST_WARNING_OBJECT (demux, "middle of discont, dropping");
goto bad_timestamp;
}
/* check for wraparounds */
else if (stream->last_time > 0 && time < stream->last_time &&
stream->last_time - time > MPEGTIME_TO_GSTTIME (G_MAXUINT32)) {
/* wrap around occurred */
if (stream->base_time + MPEGTIME_TO_GSTTIME ((guint64) (1) << 33) +
MPEGTIME_TO_GSTTIME (pts) >
stream->last_time + GST_SECOND * 60 * 10) {
GST_DEBUG_OBJECT (demux,
"looks like we have a corrupt packet because its pts is a lot lower than"
" the previous pts but not a wraparound");
goto bad_timestamp;
}
/* wraparound has occured but before we have detected in the pcr,
* so check we're actually getting pcr's...if we are, don't update
* the base time..just set the time and last_time correctly
*/
if (stream->PMT_pid <= MPEGTS_MAX_PID && demux->streams[stream->PMT_pid]
&& demux->streams[demux->streams[stream->PMT_pid]->PMT.PCR_PID]
&& demux->streams[demux->streams[stream->PMT_pid]->PMT.
PCR_PID]->last_PCR > 0) {
GST_DEBUG_OBJECT (demux, "timestamps wrapped before noticed in PCR");
time = MPEGTIME_TO_GSTTIME (pts) + stream->base_time +
MPEGTIME_TO_GSTTIME ((guint64) (1) << 33);
stream->last_time = time;
} else {
stream->base_time = stream->base_time +
MPEGTIME_TO_GSTTIME ((guint64) (1) << 33);
time = MPEGTIME_TO_GSTTIME (pts) + stream->base_time;
GST_DEBUG_OBJECT (demux,
"timestamps wrapped around, compensating with new base time: %"
GST_TIME_FORMAT "last time: %" GST_TIME_FORMAT " time: %"
GST_TIME_FORMAT, GST_TIME_ARGS (stream->base_time),
GST_TIME_ARGS (stream->last_time), GST_TIME_ARGS (time));
stream->last_time = time;
}
} else if (stream->last_time > 0 && time > stream->last_time &&
time - stream->last_time > MPEGTIME_TO_GSTTIME (G_MAXUINT32) &&
stream->base_time > 0) {
/* had a previous wrap around */
if (time - MPEGTIME_TO_GSTTIME ((guint64) (1) << 33) +
GST_SECOND * 60 * 10 < stream->last_time) {
GST_DEBUG_OBJECT (demux,
"looks like we have a corrupt packet because its pts is a lot higher than"
" the previous pts but not because of a wraparound or pcr discont");
goto bad_timestamp;
}
if (ABS ((time - MPEGTIME_TO_GSTTIME ((guint64) (1) << 33)) -
stream->last_time) < GST_SECOND) {
GST_DEBUG_OBJECT (demux,
"timestamps wrapped around earlier but we have an out of pts: %"
G_GUINT64_FORMAT ", as %" GST_TIME_FORMAT " translated to: %"
GST_TIME_FORMAT " and last_time of %" GST_TIME_FORMAT, pts,
GST_TIME_ARGS (time),
GST_TIME_ARGS (time - MPEGTIME_TO_GSTTIME ((guint64) (1) << 33)),
GST_TIME_ARGS (stream->last_time));
time = time - MPEGTIME_TO_GSTTIME ((guint64) (1) << 33);
} else {
GST_DEBUG_OBJECT (demux,
"timestamp may have wrapped around recently but not sure and pts"
" is very different, dropping it timestamp of this packet: %"
GST_TIME_FORMAT " compared to last timestamp: %" GST_TIME_FORMAT,
GST_TIME_ARGS (time -
MPEGTIME_TO_GSTTIME ((guint64) (1) << (33))),
GST_TIME_ARGS (stream->last_time));
goto bad_timestamp;
}
} else {
/* we must have a corrupt packet */
GST_WARNING_OBJECT (demux, "looks like we have a corrupt packet because"
" its timestamp is buggered timestamp: %" GST_TIME_FORMAT
" compared to" " last timestamp: %" GST_TIME_FORMAT,
GST_TIME_ARGS (time), GST_TIME_ARGS (stream->last_time));
goto bad_timestamp;
}
} else { /* do not set last_time if a packet with pts from before wrap
around arrived after the wrap around occured */
stream->last_time = time;
}
} else {
time = GST_CLOCK_TIME_NONE;
pts = -1;
}
GST_DEBUG_OBJECT (demux, "setting PTS to (%" G_GUINT64_FORMAT ") time: %"
GST_TIME_FORMAT " on buffer %p first buffer: %d base_time: %"
GST_TIME_FORMAT, pts, GST_TIME_ARGS (time), buffer, first,
GST_TIME_ARGS (stream->base_time));
GST_BUFFER_TIMESTAMP (buffer) = time;
/* check if we have a pad already */
if (srcpad == NULL) {
/* When adding a stream, require either a valid base PCR, or a valid PTS */
if (!gst_mpegts_demux_setup_base_pts (demux, pts))
goto bad_timestamp;
/* fill in the last bits of the stream */
/* if no stream type, then assume it based on the PES start code,
* needed for partial ts streams without PMT */
if (G_UNLIKELY (stream->flags & MPEGTS_STREAM_FLAG_STREAM_TYPE_UNKNOWN)) {
if ((filter->start_code & 0xFFFFFFF0) == PACKET_VIDEO_START_CODE) {
/* it is mpeg2 video */
stream->stream_type = ST_VIDEO_MPEG2;
stream->flags &= ~MPEGTS_STREAM_FLAG_STREAM_TYPE_UNKNOWN;
stream->flags |= MPEGTS_STREAM_FLAG_IS_VIDEO;
GST_DEBUG_OBJECT (demux, "Found stream 0x%04x without PMT with video "
"start_code. Treating as video", stream->PID);
} else if ((filter->start_code & 0xFFFFFFE0) == PACKET_AUDIO_START_CODE) {
/* it is mpeg audio */
stream->stream_type = ST_AUDIO_MPEG2;
stream->flags &= ~MPEGTS_STREAM_FLAG_STREAM_TYPE_UNKNOWN;
GST_DEBUG_OBJECT (demux, "Found stream 0x%04x without PMT with audio "
"start_code. Treating as audio", stream->PID);
} else {
GST_LOG_OBJECT (demux, "Stream start code on pid 0x%04x is: 0x%x",
stream->PID, filter->start_code);
}
}
if (!gst_mpegts_demux_fill_stream (stream, filter->id, stream->stream_type))
goto unknown_type;
GST_DEBUG_OBJECT (demux,
"New stream 0x%04x of type 0x%02x with caps %" GST_PTR_FORMAT,
stream->PID, stream->stream_type, GST_PAD_CAPS (stream->pad));
srcpad = stream->pad;
/* activate and add */
gst_pad_set_active (srcpad, TRUE);
gst_element_add_pad (GST_ELEMENT_CAST (demux), srcpad);
demux->need_no_more_pads = TRUE;
stream->discont = TRUE;
/* send new_segment */
gst_mpegts_demux_send_new_segment (demux, stream, pts);
/* send tags */
gst_mpegts_demux_send_tags_for_stream (demux, stream);
}
GST_DEBUG_OBJECT (srcpad, "pushing buffer");
gst_buffer_set_caps (buffer, GST_PAD_CAPS (srcpad));
if (stream->discont) {
GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_FLAG_DISCONT);
stream->discont = FALSE;
}
ret = gst_pad_push (srcpad, buffer);
ret = gst_mpegts_demux_combine_flows (demux, stream, ret);
return ret;
/* ERROR */
unknown_type:
{
GST_DEBUG_OBJECT (demux, "got unknown stream id 0x%02x, type 0x%02x",
filter->id, stream->stream_type);
gst_buffer_unref (buffer);
return gst_mpegts_demux_combine_flows (demux, stream, GST_FLOW_NOT_LINKED);
}
bad_timestamp:
{
gst_buffer_unref (buffer);
return gst_mpegts_demux_combine_flows (demux, stream, GST_FLOW_OK);
}
}
static void
gst_mpegts_demux_resync_cb (GstPESFilter * filter, GstMpegTSStream * stream)
{
/* does nothing for now */
}
/*
* CA_section() {
* table_id 8 uimsbf == 0x01
* section_syntax_indicator 1 bslbf == 1
* '0' 1 bslbf == 0
* reserved 2 bslbf
* section_length 12 uimsbf == 00xxxxx...
* reserved 18 bslbf
* version_number 5 uimsbf
* current_next_indicator 1 bslbf
* section_number 8 uimsbf
* last_section_number 8 uimsbf
* for (i=0; i<N;i++) {
* descriptor()
* }
* CRC_32 32 rpchof
* }
*/
static FORCE_INLINE gboolean
gst_mpegts_stream_parse_cat (GstMpegTSStream * stream,
guint8 * data, guint datalen)
{
GstMpegTSDemux *demux;
demux = stream->demux;
GST_DEBUG_OBJECT (demux, "parsing CA section");
return TRUE;
}
static void
gst_mpegts_activate_pmt (GstMpegTSDemux * demux, GstMpegTSStream * stream)
{
GST_DEBUG_OBJECT (demux, "activating PMT 0x%08x", stream->PID);
/* gst_mpegts_demux_remove_pads (demux); */
demux->current_PMT = stream->PID;
/* PMT has been updated, signal the change */
if (demux->current_PMT == stream->PID)
g_object_notify ((GObject *) (demux), "pmt-info");
}
/*
* TS_program_map_section() {
* table_id 8 uimsbf == 0x02
* section_syntax_indicator 1 bslbf == 1
* '0' 1 bslbf == 0
* reserved 2 bslbf
* section_length 12 uimsbf == 00xxxxx...
* program_number 16 uimsbf
* reserved 2 bslbf
* version_number 5 uimsbf
* current_next_indicator 1 bslbf
* section_number 8 uimsbf
* last_section_number 8 uimsbf
* reserved 3 bslbf
* PCR_PID 13 uimsbf
* reserved 4 bslbf
* program_info_length 12 uimsbf == 00xxxxx...
* for (i=0; i<N; i++) {
* descriptor()
* }
* for (i=0;i<N1;i++) {
* stream_type 8 uimsbf
* reserved 3 bslbf
* elementary_PID 13 uimsnf
* reserved 4 bslbf
* ES_info_length 12 uimsbf == 00xxxxx...
* for (i=0; i<N2; i++) {
* descriptor()
* }
* }
* CRC_32 32 rpchof
* }
*/
static FORCE_INLINE gboolean
gst_mpegts_stream_parse_pmt (GstMpegTSStream * stream,
guint8 * data, guint datalen)
{
GstMpegTSDemux *demux;
gint entries;
guint32 CRC;
GstMpegTSPMT *PMT;
guint version_number;
guint8 current_next_indicator;
guint16 program_number;
demux = stream->demux;
if (G_UNLIKELY (*data++ != 0x02))
goto wrong_id;
if ((data[0] & 0xc0) != 0x80)
goto wrong_sync;
if ((data[0] & 0x0c) != 0x00)
goto wrong_seclen;
data += 2;
if (demux->check_crc)
if (G_UNLIKELY (gst_mpegts_demux_calc_crc32 (data - 3, datalen) != 0))
goto wrong_crc;
GST_LOG_OBJECT (demux, "PMT section_length: %d", datalen - 3);
PMT = &stream->PMT;
/* check if version number changed */
version_number = (data[2] & 0x3e) >> 1;
GST_LOG_OBJECT (demux, "PMT version_number: %d", version_number);
current_next_indicator = (data[2] & 0x01);
GST_LOG_OBJECT (demux, "PMT current_next_indicator %d",
current_next_indicator);
if (current_next_indicator == 0)
goto not_yet_applicable;
program_number = GST_READ_UINT16_BE (data);
if (demux->program_number != -1 && demux->program_number != program_number) {
goto wrong_program_number;
}
if (demux->program_number == -1) {
GST_INFO_OBJECT (demux, "No program number set, so using first parsed PMT"
"'s program number: %d", program_number);
demux->program_number = program_number;
}
if (version_number == PMT->version_number)
goto same_version;
PMT->version_number = version_number;
PMT->current_next_indicator = current_next_indicator;
stream->PMT.program_number = program_number;
data += 3;
GST_DEBUG_OBJECT (demux, "PMT program_number: %d", PMT->program_number);
PMT->section_number = *data++;
GST_DEBUG_OBJECT (demux, "PMT section_number: %d", PMT->section_number);
PMT->last_section_number = *data++;
GST_DEBUG_OBJECT (demux, "PMT last_section_number: %d",
PMT->last_section_number);
PMT->PCR_PID = GST_READ_UINT16_BE (data);
PMT->PCR_PID &= 0x1fff;
data += 2;
GST_DEBUG_OBJECT (demux, "PMT PCR_PID: 0x%04x", PMT->PCR_PID);
/* create or get stream, not much we can say about it except that when we get
* a data stream and we need a PCR, we can use the stream to get/store the
* base_PCR. */
gst_mpegts_demux_get_stream_for_PID (demux, PMT->PCR_PID);
if ((data[0] & 0x0c) != 0x00)
goto wrong_pilen;
PMT->program_info_length = GST_READ_UINT16_BE (data);
PMT->program_info_length &= 0x0fff;
/* FIXME: validate value of program_info_length */
data += 2;
/* FIXME: validate value of program_info_length, before using */
/* parse descriptor */
if (G_UNLIKELY (PMT->program_info))
gst_mpeg_descriptor_free (PMT->program_info);
PMT->program_info =
gst_mpeg_descriptor_parse (data, PMT->program_info_length);
/* skip descriptor */
data += PMT->program_info_length;
GST_DEBUG_OBJECT (demux, "PMT program_info_length: %d",
PMT->program_info_length);
entries = datalen - 3 - PMT->program_info_length - 9 - 4;
if (G_UNLIKELY (PMT->entries))
g_array_free (PMT->entries, TRUE);
PMT->entries = g_array_new (FALSE, TRUE, sizeof (GstMpegTSPMTEntry));
while (entries > 0) {
GstMpegTSPMTEntry entry;
GstMpegTSStream *ES_stream;
guint8 stream_type;
guint16 ES_info_length;
stream_type = *data++;
entry.PID = GST_READ_UINT16_BE (data);
entry.PID &= 0x1fff;
data += 2;
if ((data[0] & 0x0c) != 0x00)
goto wrong_esilen;
ES_info_length = GST_READ_UINT16_BE (data);
ES_info_length &= 0x0fff;
data += 2;
/* get/create elementary stream */
ES_stream = gst_mpegts_demux_get_stream_for_PID (demux, entry.PID);
/* check if PID unknown */
if (ES_stream->PID_type == PID_TYPE_UNKNOWN) {
/* set as elementary */
ES_stream->PID_type = PID_TYPE_ELEMENTARY;
/* set stream type */
/* hack for ITV HD (sid 10510, video pid 3401 */
if (program_number == 10510 && entry.PID == 3401 &&
stream_type == ST_PRIVATE_DATA)
stream_type = ST_VIDEO_H264;
ES_stream->stream_type = stream_type;
ES_stream->flags &= ~MPEGTS_STREAM_FLAG_STREAM_TYPE_UNKNOWN;
/* init base and last time */
ES_stream->base_time = 0;
ES_stream->last_time = 0;
/* parse descriptor */
ES_stream->ES_info = gst_mpeg_descriptor_parse (data, ES_info_length);
if (stream_type == ST_PRIVATE_SECTIONS) {
/* not really an ES, so use section filter not pes filter */
/* initialise section filter */
GstCaps *caps;
gst_section_filter_init (&ES_stream->section_filter);
ES_stream->PID_type = PID_TYPE_PRIVATE_SECTION;
ES_stream->pad = gst_pad_new_from_static_template (&private_template,
g_strdup_printf ("private_%04x", entry.PID));
gst_pad_set_active (ES_stream->pad, TRUE);
caps = gst_caps_new_simple ("application/x-mpegts-private-section",
NULL);
gst_pad_use_fixed_caps (ES_stream->pad);
gst_pad_set_caps (ES_stream->pad, caps);
gst_caps_unref (caps);
gst_element_add_pad (GST_ELEMENT_CAST (demux), ES_stream->pad);
} else {
/* Recognise video streams based on stream_type */
if (gst_mpegts_stream_is_video (ES_stream))
ES_stream->flags |= MPEGTS_STREAM_FLAG_IS_VIDEO;
/* set adaptor */
gst_pes_filter_init (&ES_stream->filter, NULL, NULL);
gst_pes_filter_set_callbacks (&ES_stream->filter,
(GstPESFilterData) gst_mpegts_demux_data_cb,
(GstPESFilterResync) gst_mpegts_demux_resync_cb, ES_stream);
if (ES_stream->flags & MPEGTS_STREAM_FLAG_IS_VIDEO)
ES_stream->filter.allow_unbounded = TRUE;
ES_stream->PMT_pid = stream->PID;
}
}
/* skip descriptor */
data += ES_info_length;
GST_DEBUG_OBJECT (demux,
" PMT stream_type: %02x, PID: 0x%04x (ES_info_len %d)", stream_type,
entry.PID, ES_info_length);
g_array_append_val (PMT->entries, entry);
entries -= 5 + ES_info_length;
}
CRC = GST_READ_UINT32_BE (data);
GST_DEBUG_OBJECT (demux, "PMT CRC: 0x%08x", CRC);
if (demux->program_number == -1) {
/* No program specified, take the first PMT */
if (demux->current_PMT == 0 || demux->current_PMT == stream->PID)
gst_mpegts_activate_pmt (demux, stream);
} else {
/* Program specified, activate this if it matches */
if (demux->program_number == PMT->program_number)
gst_mpegts_activate_pmt (demux, stream);
}
return TRUE;
/* ERRORS */
wrong_crc:
{
GST_DEBUG_OBJECT (demux, "wrong crc");
return FALSE;
}
same_version:
{
GST_DEBUG_OBJECT (demux, "same version as existing PMT");
return TRUE;
}
wrong_program_number:
{
GST_DEBUG_OBJECT (demux, "PMT is for program number we don't care about");
return TRUE;
}
not_yet_applicable:
{
GST_DEBUG_OBJECT (demux, "Ignoring PMT with current_next_indicator = 0");
return TRUE;
}
wrong_id:
{
GST_DEBUG_OBJECT (demux, "expected table_id == 0, got 0x%02x", data[0]);
return FALSE;
}
wrong_sync:
{
GST_DEBUG_OBJECT (demux, "expected sync 10, got %02x", data[0]);
return FALSE;
}
wrong_seclen:
{
GST_DEBUG_OBJECT (demux,
"first two bits of section length must be 0, got %02x", data[0]);
return FALSE;
}
wrong_pilen:
{
GST_DEBUG_OBJECT (demux,
"first two bits of program_info length must be 0, got %02x", data[0]);
return FALSE;
}
wrong_esilen:
{
GST_DEBUG_OBJECT (demux,
"first two bits of ES_info length must be 0, got %02x", data[0]);
g_array_free (stream->PMT.entries, TRUE);
stream->PMT.entries = NULL;
gst_mpeg_descriptor_free (stream->PMT.program_info);
stream->PMT.program_info = NULL;
return FALSE;
}
}
/*
* private_section() {
* table_id 8 uimsbf
* section_syntax_indicator 1 bslbf
* private_indicator 1 bslbf
* reserved 2 bslbf
* private_section_length 12 uimsbf
* if (section_syntax_indicator == '0') {
* for ( i=0;i<N;i++) {
* private_data_byte 8 bslbf
* }
* }
* else {
* table_id_extension 16 uimsbf
* reserved 2 bslbf
* version_number 5 uimsbf
* current_next_indicator 1 bslbf
* section_number 8 uimsbf
* last_section_number 8 uimsbf
* for ( i=0;i<private_section_length-9;i++) {
* private_data_byte 8 bslbf
* }
* CRC_32 32 rpchof
* }
* }
*/
static FORCE_INLINE gboolean
gst_mpegts_stream_parse_private_section (GstMpegTSStream * stream,
guint8 * data, guint datalen)
{
GstMpegTSDemux *demux;
GstBuffer *buffer;
demux = stream->demux;
if (demux->check_crc)
if (gst_mpegts_demux_calc_crc32 (data, datalen) != 0)
goto wrong_crc;
/* just dump this down the pad */
if (gst_pad_alloc_buffer (stream->pad, 0, datalen, NULL, &buffer) ==
GST_FLOW_OK) {
memcpy (buffer->data, data, datalen);
gst_pad_push (stream->pad, buffer);
}
GST_DEBUG_OBJECT (demux, "parsing private section");
return TRUE;
wrong_crc:
{
GST_DEBUG_OBJECT (demux, "wrong crc");
return FALSE;
}
}
/*
* adaptation_field() {
* adaptation_field_length 8 uimsbf
* if(adaptation_field_length >0) {
* discontinuity_indicator 1 bslbf
* random_access_indicator 1 bslbf
* elementary_stream_priority_indicator 1 bslbf
* PCR_flag 1 bslbf
* OPCR_flag 1 bslbf
* splicing_point_flag 1 bslbf
* transport_private_data_flag 1 bslbf
* adaptation_field_extension_flag 1 bslbf
* if(PCR_flag == '1') {
* program_clock_reference_base 33 uimsbf
* reserved 6 bslbf
* program_clock_reference_extension 9 uimsbf
* }
* if(OPCR_flag == '1') {
* original_program_clock_reference_base 33 uimsbf
* reserved 6 bslbf
* original_program_clock_reference_extension 9 uimsbf
* }
* if (splicing_point_flag == '1') {
* splice_countdown 8 tcimsbf
* }
* if(transport_private_data_flag == '1') {
* transport_private_data_length 8 uimsbf
* for (i=0; i<transport_private_data_length;i++){
* private_data_byte 8 bslbf
* }
* }
* if (adaptation_field_extension_flag == '1' ) {
* adaptation_field_extension_length 8 uimsbf
* ltw_flag 1 bslbf
* piecewise_rate_flag 1 bslbf
* seamless_splice_flag 1 bslbf
* reserved 5 bslbf
* if (ltw_flag == '1') {
* ltw_valid_flag 1 bslbf
* ltw_offset 15 uimsbf
* }
* if (piecewise_rate_flag == '1') {
* reserved 2 bslbf
* piecewise_rate 22 uimsbf
* }
* if (seamless_splice_flag == '1'){
* splice_type 4 bslbf
* DTS_next_AU[32..30] 3 bslbf
* marker_bit 1 bslbf
* DTS_next_AU[29..15] 15 bslbf
* marker_bit 1 bslbf
* DTS_next_AU[14..0] 15 bslbf
* marker_bit 1 bslbf
* }
* for ( i=0;i<N;i++) {
* reserved 8 bslbf
* }
* }
* for (i=0;i<N;i++){
* stuffing_byte 8 bslbf
* }
* }
* }
*/
static FORCE_INLINE gboolean
gst_mpegts_demux_parse_adaptation_field (GstMpegTSStream * stream,
const guint8 * data, guint data_len, guint * consumed)
{
GstMpegTSDemux *demux;
guint8 length;
guint8 *data_end;
gint i;
GstMpegTSStream *pmt_stream;
demux = stream->demux;
data_end = ((guint8 *) data) + data_len;
length = *data++;
if (G_UNLIKELY (length > data_len))
goto wrong_length;
GST_DEBUG_OBJECT (demux, "parsing adaptation field, length %d", length);
if (length > 0) {
guint8 flags = *data++;
GST_LOG_OBJECT (demux, "flags 0x%02x", flags);
/* discontinuity flag */
if (flags & 0x80) {
GST_DEBUG_OBJECT (demux, "discontinuity flag set");
}
/* PCR_flag */
if (flags & 0x10) {
guint32 pcr1;
guint16 pcr2;
guint64 pcr, pcr_ext;
gboolean valid_pcr = TRUE;
pcr1 = GST_READ_UINT32_BE (data);
pcr2 = GST_READ_UINT16_BE (data + 4);
pcr = ((guint64) pcr1) << 1;
pcr |= (pcr2 & 0x8000) >> 15;
pcr_ext = (pcr2 & 0x01ff);
if (pcr_ext)
pcr = (pcr * 300 + pcr_ext % 300) / 300;
GST_DEBUG_OBJECT (demux,
"have PCR %" G_GUINT64_FORMAT "(%" GST_TIME_FORMAT ") on PID 0x%04x "
"and last pcr is %" G_GUINT64_FORMAT " (%" GST_TIME_FORMAT ")", pcr,
GST_TIME_ARGS (MPEGTIME_TO_GSTTIME (pcr)), stream->PID,
stream->last_PCR,
GST_TIME_ARGS (MPEGTIME_TO_GSTTIME (stream->last_PCR)));
/* pcr has been converted into units of 90Khz ticks
* so assume discont if last pcr was > 900000 (10 second) lower */
if (stream->last_PCR != -1 &&
(pcr - stream->last_PCR > 900000 || pcr < stream->last_PCR)) {
GstClockTimeDiff base_time_difference;
GST_DEBUG_OBJECT (demux,
"looks like we have a discont, this pcr should really be approx: %"
G_GUINT64_FORMAT, stream->last_PCR + stream->last_PCR_difference);
if (stream->discont_PCR == FALSE) {
if (pcr > stream->last_PCR) {
base_time_difference = -MPEGTIME_TO_GSTTIME ((pcr -
(stream->last_PCR + stream->last_PCR_difference)));
} else {
base_time_difference = MPEGTIME_TO_GSTTIME ((stream->last_PCR +
stream->last_PCR_difference) - pcr);
}
stream->discont_PCR = TRUE;
stream->discont_difference = base_time_difference;
valid_pcr = FALSE;
} else {
GstClockTimeDiff base_time_difference;
/* need to update all pmt streams in case this pcr is pcr
* for multiple programs */
int j;
gboolean *pmts_checked = (gboolean *) & demux->pmts_checked;
memset (pmts_checked, 0, sizeof (gboolean) * (MPEGTS_MAX_PID + 1));
for (j = 0; j < MPEGTS_MAX_PID + 1; j++) {
if (demux->streams[j]
&& demux->streams[j]->PMT_pid <= MPEGTS_MAX_PID) {
if (!pmts_checked[demux->streams[j]->PMT_pid]) {
/* check if this is correct pcr for pmt */
if (demux->streams[demux->streams[j]->PMT_pid] &&
stream->PID ==
demux->streams[demux->streams[j]->PMT_pid]->PMT.PCR_PID) {
/* checking the pcr discont is similar this second time
* if similar, update the es pids
* if not, assume it's a false discont due to corruption
* or other */
if (pcr > stream->last_PCR) {
base_time_difference = -MPEGTIME_TO_GSTTIME ((pcr -
(stream->last_PCR + stream->last_PCR_difference)));
} else {
base_time_difference =
MPEGTIME_TO_GSTTIME ((stream->last_PCR +
stream->last_PCR_difference) - pcr);
}
if ((base_time_difference - stream->discont_difference > 0 &&
base_time_difference - stream->discont_difference <
GST_SECOND * 10) ||
(stream->discont_difference - base_time_difference > 0 &&
stream->discont_difference - base_time_difference <
GST_SECOND * 10)) {
pmt_stream = demux->streams[demux->streams[j]->PMT_pid];
GST_DEBUG_OBJECT (demux, "Updating base_time on all es "
"pids belonging to PMT 0x%02x", stream->PMT_pid);
for (i = 0; i < pmt_stream->PMT.entries->len; i++) {
GstMpegTSPMTEntry *cur_entry =
&g_array_index (pmt_stream->PMT.entries,
GstMpegTSPMTEntry, i);
GST_DEBUG_OBJECT (demux,
"Updating base time on " "pid 0x%02x by %"
G_GINT64_FORMAT, cur_entry->PID,
stream->discont_difference);
if (cur_entry->PID <= MPEGTS_MAX_PID
&& demux->streams[cur_entry->PID]) {
demux->streams[cur_entry->PID]->base_time +=
stream->discont_difference;
}
}
} else {
GST_DEBUG_OBJECT (demux, "last PCR discont looked to be "
"bogus: previous discont difference %" G_GINT64_FORMAT
" now %" G_GINT64_FORMAT, stream->discont_difference,
base_time_difference);
valid_pcr = FALSE;
}
}
}
pmts_checked[demux->streams[j]->PMT_pid] = TRUE;
}
}
stream->discont_PCR = FALSE;
stream->discont_difference = 0;
}
} else if (stream->last_PCR != -1) {
if (stream->discont_PCR) {
GST_DEBUG_OBJECT (demux, "last PCR discont looked to be bogus");
stream->discont_PCR = FALSE;
stream->discont_difference = 0;
}
stream->last_PCR_difference = pcr - stream->last_PCR;
}
GST_DEBUG_OBJECT (demux,
"valid pcr: %d last PCR difference: %" G_GUINT64_FORMAT, valid_pcr,
stream->last_PCR_difference);
if (valid_pcr) {
GstMpegTSStream *PMT_stream = demux->streams[demux->current_PMT];
if (PMT_stream && PMT_stream->PMT.PCR_PID == stream->PID) {
if (demux->pcr[0] == -1) {
GST_DEBUG ("RECORDING pcr[0]:%" G_GUINT64_FORMAT, pcr);
demux->pcr[0] = pcr;
demux->num_packets = 0;
} /* Considering a difference of 1 sec ie 90000 ticks */
else if (G_UNLIKELY (demux->pcr[1] == -1
&& ((pcr - demux->pcr[0]) >= 90000))) {
GST_DEBUG ("RECORDING pcr[1]:%" G_GUINT64_FORMAT, pcr);
demux->pcr[1] = pcr;
}
}
stream->last_PCR = pcr;
if (demux->clock && demux->clock_base != GST_CLOCK_TIME_NONE) {
gdouble r_squared;
GstMpegTSStream *PMT_stream;
/* for the reference start time we need to consult the PCR_PID of the
* current PMT */
PMT_stream = demux->streams[demux->current_PMT];
if (PMT_stream->PMT.PCR_PID == stream->PID) {
GST_LOG_OBJECT (demux,
"internal %" GST_TIME_FORMAT " observation %" GST_TIME_FORMAT
" pcr: %" G_GUINT64_FORMAT " base_pcr: %" G_GUINT64_FORMAT
"pid: %d",
GST_TIME_ARGS (gst_clock_get_internal_time (demux->clock)),
GST_TIME_ARGS (MPEGTIME_TO_GSTTIME (pcr) -
MPEGTIME_TO_GSTTIME (stream->base_PCR) + stream->base_time +
demux->clock_base), pcr, stream->base_PCR, stream->PID);
gst_clock_add_observation (demux->clock,
gst_clock_get_internal_time (demux->clock),
demux->clock_base + stream->base_time +
MPEGTIME_TO_GSTTIME (pcr) -
MPEGTIME_TO_GSTTIME (stream->base_PCR), &r_squared);
}
}
}
data += 6;
}
/* OPCR_flag */
if (flags & 0x08) {
guint32 opcr1;
guint16 opcr2;
guint64 opcr, opcr_ext;
opcr1 = GST_READ_UINT32_BE (data);
opcr2 = GST_READ_UINT16_BE (data + 4);
opcr = ((guint64) opcr1) << 1;
opcr |= (opcr2 & 0x8000) >> 15;
opcr_ext = (opcr2 & 0x01ff);
if (opcr_ext)
opcr = (opcr * 300 + opcr_ext % 300) / 300;
GST_DEBUG_OBJECT (demux, "have OPCR %" G_GUINT64_FORMAT " on PID 0x%04x",
opcr, stream->PID);
stream->last_OPCR = opcr;
data += 6;
}
/* splicing_point_flag */
if (flags & 0x04) {
guint8 splice_countdown;
splice_countdown = *data++;
GST_DEBUG_OBJECT (demux, "have splicing point, countdown %d",
splice_countdown);
}
/* transport_private_data_flag */
if (flags & 0x02) {
guint8 plength = *data++;
if (data + plength > data_end)
goto private_data_too_large;
GST_DEBUG_OBJECT (demux, "have private data, length: %d", plength);
data += plength;
}
/* adaptation_field_extension_flag */
if (flags & 0x01) {
GST_DEBUG_OBJECT (demux, "have field extension");
}
}
*consumed = length + 1;
return TRUE;
/* ERRORS */
wrong_length:
{
GST_DEBUG_OBJECT (demux, "length %d > %d", length, data_len);
return FALSE;
}
private_data_too_large:
{
GST_DEBUG_OBJECT (demux, "have too large a private data length");
return FALSE;
}
}
/*
* program_association_section() {
* table_id 8 uimsbf == 0x00
* section_syntax_indicator 1 bslbf == 1
* '0' 1 bslbf == 0
* reserved 2 bslbf
* section_length 12 uimsbf == 00xxxxx...
* transport_stream_id 16 uimsbf
* reserved 2 bslbf
* version_number 5 uimsbf
* current_next_indicator 1 bslbf
* section_number 8 uimsbf
* last_section_number 8 uimsbf
* for (i=0; i<N;i++) {
* program_number 16 uimsbf
* reserved 3 bslbf
* if(program_number == '0') {
* network_PID 13 uimsbf
* }
* else {
* program_map_PID 13 uimsbf
* }
* }
* CRC_32 32 rpchof
* }
*/
static FORCE_INLINE gboolean
gst_mpegts_stream_parse_pat (GstMpegTSStream * stream,
guint8 * data, guint datalen)
{
GstMpegTSDemux *demux;
gint entries;
guint32 CRC;
guint version_number;
guint8 current_next_indicator;
GstMpegTSPAT *PAT;
demux = stream->demux;
if (datalen < 8)
return FALSE;
if (*data++ != 0x00)
goto wrong_id;
if ((data[0] & 0xc0) != 0x80)
goto wrong_sync;
if (G_UNLIKELY ((data[0] & 0x0c) != 0x00))
goto wrong_seclen;
data += 2;
GST_DEBUG_OBJECT (demux, "PAT section_length: %d", datalen - 3);
if (demux->check_crc)
if (gst_mpegts_demux_calc_crc32 (data - 3, datalen) != 0)
goto wrong_crc;
PAT = &stream->PAT;
version_number = (data[2] & 0x3e) >> 1;
GST_DEBUG_OBJECT (demux, "PAT version_number: %d", version_number);
if (G_UNLIKELY (version_number == PAT->version_number))
goto same_version;
current_next_indicator = (data[2] & 0x01);
GST_DEBUG_OBJECT (demux, "PAT current_next_indicator %d",
current_next_indicator);
if (current_next_indicator == 0)
goto not_yet_applicable;
PAT->version_number = version_number;
PAT->current_next_indicator = current_next_indicator;
PAT->transport_stream_id = GST_READ_UINT16_BE (data);
data += 3;
GST_DEBUG_OBJECT (demux, "PAT stream_id: %d", PAT->transport_stream_id);
PAT->section_number = *data++;
PAT->last_section_number = *data++;
GST_DEBUG_OBJECT (demux, "PAT current_next_indicator: %d",
PAT->current_next_indicator);
GST_DEBUG_OBJECT (demux, "PAT section_number: %d", PAT->section_number);
GST_DEBUG_OBJECT (demux, "PAT last_section_number: %d",
PAT->last_section_number);
/* 5 bytes after section length and a 4 bytes CRC,
* the rest is 4 byte entries */
entries = (datalen - 3 - 9) / 4;
if (PAT->entries)
g_array_free (PAT->entries, TRUE);
PAT->entries =
g_array_sized_new (FALSE, TRUE, sizeof (GstMpegTSPATEntry), entries);
while (entries--) {
GstMpegTSPATEntry entry;
GstMpegTSStream *PMT_stream;
entry.program_number = GST_READ_UINT16_BE (data);
data += 2;
entry.PID = GST_READ_UINT16_BE (data);
entry.PID &= 0x1fff;
data += 2;
/* get/create stream for PMT */
PMT_stream = gst_mpegts_demux_get_stream_for_PID (demux, entry.PID);
if (PMT_stream->PID_type != PID_TYPE_PROGRAM_MAP) {
/* set as program map */
PMT_stream->PID_type = PID_TYPE_PROGRAM_MAP;
/* initialise section filter */
gst_section_filter_init (&PMT_stream->section_filter);
}
g_array_append_val (PAT->entries, entry);
GST_DEBUG_OBJECT (demux, " PAT program: %d, PID 0x%04x",
entry.program_number, entry.PID);
}
CRC = GST_READ_UINT32_BE (data);
GST_DEBUG_OBJECT (demux, "PAT CRC: 0x%08x", CRC);
/* PAT has been updated, signal the change */
g_object_notify ((GObject *) (demux), "pat-info");
return TRUE;
/* ERRORS */
wrong_crc:
{
GST_DEBUG_OBJECT (demux, "wrong crc");
return FALSE;
}
same_version:
{
GST_DEBUG_OBJECT (demux, "same version as existing PAT");
return TRUE;
}
not_yet_applicable:
{
GST_DEBUG_OBJECT (demux, "Ignoring PAT with current_next_indicator = 0");
return TRUE;
}
wrong_id:
{
GST_DEBUG_OBJECT (demux, "expected table_id == 0, got %02x", data[0]);
return FALSE;
}
wrong_sync:
{
GST_DEBUG_OBJECT (demux, "expected sync 10, got %02x", data[0]);
return FALSE;
}
wrong_seclen:
{
GST_DEBUG_OBJECT (demux,
"first two bits of section length must be 0, got %02x", data[0]);
return FALSE;
}
}
static gboolean
gst_mpegts_demux_is_PMT (GstMpegTSDemux * demux, guint16 PID)
{
GstMpegTSStream *stream;
GstMpegTSPAT *PAT;
gint i;
/* get the PAT */
stream = demux->streams[PID_PROGRAM_ASSOCIATION_TABLE];
if (stream == NULL || stream->PAT.entries == NULL)
return FALSE;
PAT = &stream->PAT;
for (i = 0; i < PAT->entries->len; i++) {
GstMpegTSPATEntry *entry;
entry = &g_array_index (PAT->entries, GstMpegTSPATEntry, i);
if (!entry)
continue;
if (entry->PID == PID)
return TRUE;
}
return FALSE;
}
static FORCE_INLINE GstFlowReturn
gst_mpegts_stream_pes_buffer_flush (GstMpegTSStream * stream, gboolean discard)
{
GstFlowReturn ret = GST_FLOW_OK;
if (stream->pes_buffer) {
if (discard) {
gst_buffer_unref (stream->pes_buffer);
stream->pes_buffer_in_sync = FALSE;
} else {
GST_BUFFER_SIZE (stream->pes_buffer) = stream->pes_buffer_used;
ret = gst_pes_filter_push (&stream->filter, stream->pes_buffer);
if (ret == GST_FLOW_LOST_SYNC)
stream->pes_buffer_in_sync = FALSE;
}
stream->pes_buffer = NULL;
}
return ret;
}
static FORCE_INLINE GstFlowReturn
gst_mpegts_stream_pes_buffer_push (GstMpegTSStream * stream,
const guint8 * in_data, guint in_size)
{
GstFlowReturn ret = GST_FLOW_OK;
guint8 *out_data;
if (G_UNLIKELY (stream->pes_buffer
&& stream->pes_buffer_used + in_size > stream->pes_buffer_size)) {
GST_DEBUG ("stream with PID 0x%04x have PES buffer full at %u bytes."
" Flushing and growing the buffer",
stream->PID, stream->pes_buffer_size);
stream->pes_buffer_overflow = TRUE;
if (stream->pes_buffer_size < (MPEGTS_MAX_PES_BUFFER_SIZE >> 1))
stream->pes_buffer_size <<= 1;
ret = gst_mpegts_stream_pes_buffer_flush (stream, FALSE);
if (ret == GST_FLOW_LOST_SYNC)
goto done;
}
if (G_UNLIKELY (!stream->pes_buffer)) {
/* set initial size of PES buffer */
if (G_UNLIKELY (stream->pes_buffer_size == 0))
stream->pes_buffer_size = MPEGTS_MIN_PES_BUFFER_SIZE;
stream->pes_buffer = gst_buffer_new_and_alloc (stream->pes_buffer_size);
stream->pes_buffer_used = 0;
}
out_data = GST_BUFFER_DATA (stream->pes_buffer) + stream->pes_buffer_used;
memcpy (out_data, in_data, in_size);
stream->pes_buffer_used += in_size;
done:
return ret;
}
static FORCE_INLINE GstFlowReturn
gst_mpegts_demux_pes_buffer_flush (GstMpegTSDemux * demux, gboolean discard)
{
gint i;
GstFlowReturn ret = GST_FLOW_OK;
for (i = 0; i < MPEGTS_MAX_PID + 1; i++) {
GstMpegTSStream *stream = demux->streams[i];
if (stream && stream->pad) {
gst_mpegts_stream_pes_buffer_flush (stream, discard);
stream->pes_buffer_in_sync = FALSE;
}
}
return ret;
}
static FORCE_INLINE GstFlowReturn
gst_mpegts_demux_push_fragment (GstMpegTSStream * stream,
const guint8 * in_data, guint in_size)
{
GstFlowReturn ret;
GstBuffer *es_buf = gst_buffer_new_and_alloc (in_size);
memcpy (GST_BUFFER_DATA (es_buf), in_data, in_size);
ret = gst_pes_filter_push (&stream->filter, es_buf);
/* If PES filter return ok then PES fragment buffering
* can be enabled */
if (ret == GST_FLOW_OK)
stream->pes_buffer_in_sync = TRUE;
else if (ret == GST_FLOW_LOST_SYNC)
stream->pes_buffer_in_sync = FALSE;
return ret;
}
/*
* transport_packet(){
* sync_byte 8 bslbf == 0x47
* transport_error_indicator 1 bslbf
* payload_unit_start_indicator 1 bslbf
* transport _priority 1 bslbf
* PID 13 uimsbf
* transport_scrambling_control 2 bslbf
* adaptation_field_control 2 bslbf
* continuity_counter 4 uimsbf
* if(adaptation_field_control=='10' || adaptation_field_control=='11'){
* adaptation_field()
* }
* if(adaptation_field_control=='01' || adaptation_field_control=='11') {
* for (i=0;i<N;i++){
* data_byte 8 bslbf
* }
* }
* }
*/
static FORCE_INLINE GstFlowReturn
gst_mpegts_demux_parse_stream (GstMpegTSDemux * demux, GstMpegTSStream * stream,
const guint8 * in_data, guint in_size)
{
GstFlowReturn ret;
gboolean transport_error_indicator;
gboolean payload_unit_start_indicator;
gboolean transport_priority;
guint16 PID;
guint8 transport_scrambling_control;
guint8 adaptation_field_control;
guint8 continuity_counter;
const guint8 *data = in_data;
guint datalen = in_size;
transport_error_indicator = (data[0] & 0x80) == 0x80;
payload_unit_start_indicator = (data[0] & 0x40) == 0x40;
transport_priority = (data[0] & 0x20) == 0x20;
PID = stream->PID;
transport_scrambling_control = (data[2] & 0xc0) >> 6;
adaptation_field_control = (data[2] & 0x30) >> 4;
continuity_counter = data[2] & 0x0f;
data += 3;
datalen -= 3;
GST_LOG_OBJECT (demux, "afc 0x%x, pusi %d, PID 0x%04x datalen %u",
adaptation_field_control, payload_unit_start_indicator, PID, datalen);
ret = GST_FLOW_OK;
/* packets with adaptation_field_control == 0 must be skipped */
if (adaptation_field_control == 0)
goto skip;
/* parse adaption field if any */
if (adaptation_field_control & 0x2) {
guint consumed;
if (!gst_mpegts_demux_parse_adaptation_field (stream, data,
datalen, &consumed))
goto done;
if (datalen <= consumed)
goto too_small;
data += consumed;
datalen -= consumed;
GST_LOG_OBJECT (demux, "consumed: %u datalen: %u", consumed, datalen);
}
/* If this packet has a payload, handle it */
if (adaptation_field_control & 0x1) {
GST_LOG_OBJECT (demux, "Packet payload %d bytes, PID 0x%04x", datalen, PID);
/* For unknown streams, check if the PID is in the partial PIDs
* list as an elementary stream and override the type if so
*/
if (G_UNLIKELY (stream->PID_type == PID_TYPE_UNKNOWN)) {
if (mpegts_is_elem_pid (demux, PID)) {
GST_DEBUG_OBJECT (demux,
"PID 0x%04x is an elementary stream in the PID list", PID);
stream->PID_type = PID_TYPE_ELEMENTARY;
stream->flags |= MPEGTS_STREAM_FLAG_STREAM_TYPE_UNKNOWN;
stream->base_time = 0;
stream->last_time = 0;
/* Clear any existing descriptor */
if (stream->ES_info) {
gst_mpeg_descriptor_free (stream->ES_info);
stream->ES_info = NULL;
}
/* Initialise our PES filter */
gst_pes_filter_init (&stream->filter, NULL, NULL);
gst_pes_filter_set_callbacks (&stream->filter,
(GstPESFilterData) gst_mpegts_demux_data_cb,
(GstPESFilterResync) gst_mpegts_demux_resync_cb, stream);
}
}
/* now parse based on the stream type */
switch (stream->PID_type) {
case PID_TYPE_PROGRAM_ASSOCIATION:
case PID_TYPE_CONDITIONAL_ACCESS:
case PID_TYPE_PROGRAM_MAP:
case PID_TYPE_PRIVATE_SECTION:
{
GstBuffer *sec_buf;
guint8 *section_data;
guint16 section_length;
guint8 pointer;
/* do stuff with our section */
if (payload_unit_start_indicator) {
pointer = *data++;
datalen -= 1;
if (pointer >= datalen) {
GST_DEBUG_OBJECT (demux, "pointer: 0x%02x too large", pointer);
return GST_FLOW_OK;
}
data += pointer;
datalen -= pointer;
}
/* FIXME: try to use data directly instead of creating a buffer and
pushing in into adapter at section filter */
sec_buf = gst_buffer_new_and_alloc (datalen);
memcpy (GST_BUFFER_DATA (sec_buf), data, datalen);
if (gst_section_filter_push (&stream->section_filter,
payload_unit_start_indicator, continuity_counter, sec_buf)) {
GST_DEBUG_OBJECT (demux, "section finished");
/* section ready */
section_length = stream->section_filter.section_length;
section_data =
(guint8 *) gst_adapter_peek (stream->section_filter.adapter,
section_length + 3);
switch (stream->PID_type) {
case PID_TYPE_PROGRAM_ASSOCIATION:
gst_mpegts_stream_parse_pat (stream, section_data,
section_length + 3);
break;
case PID_TYPE_CONDITIONAL_ACCESS:
gst_mpegts_stream_parse_cat (stream, section_data,
section_length + 3);
break;
case PID_TYPE_PROGRAM_MAP:
gst_mpegts_stream_parse_pmt (stream, section_data,
section_length + 3);
break;
case PID_TYPE_PRIVATE_SECTION:
gst_mpegts_stream_parse_private_section (stream, section_data,
section_length + 3);
break;
}
gst_section_filter_clear (&stream->section_filter);
} else {
/* section still going, don't parse left */
GST_DEBUG_OBJECT (demux, "section still going for PID 0x%04x", PID);
}
break;
}
case PID_TYPE_NULL_PACKET:
GST_DEBUG_OBJECT (demux,
"skipping PID 0x%04x, type 0x%04x (NULL packet)", PID,
stream->PID_type);
break;
case PID_TYPE_UNKNOWN:
GST_DEBUG_OBJECT (demux, "skipping unknown PID 0x%04x, type 0x%04x",
PID, stream->PID_type);
break;
case PID_TYPE_ELEMENTARY:
{
if (payload_unit_start_indicator) {
GST_DEBUG_OBJECT (demux, "new PES start for PID 0x%04x, used %u "
"bytes of %u bytes in the PES buffer",
PID, stream->pes_buffer_used, stream->pes_buffer_size);
/* Flush buffered PES data */
gst_mpegts_stream_pes_buffer_flush (stream, FALSE);
gst_pes_filter_drain (&stream->filter);
/* Resize the buffer to half if no overflow detected and
* had been used less than half of it */
if (stream->pes_buffer_overflow == FALSE
&& stream->pes_buffer_used < (stream->pes_buffer_size >> 1)) {
stream->pes_buffer_size >>= 1;
if (stream->pes_buffer_size < MPEGTS_MIN_PES_BUFFER_SIZE)
stream->pes_buffer_size = MPEGTS_MIN_PES_BUFFER_SIZE;
GST_DEBUG_OBJECT (demux, "PES buffer size reduced to %u bytes",
stream->pes_buffer_size);
}
/* mark the stream not in sync to give a chance on PES filter to
* detect lost sync */
stream->pes_buffer_in_sync = FALSE;
stream->pes_buffer_overflow = FALSE;
}
GST_LOG_OBJECT (demux, "Elementary packet of size %u for PID 0x%04x",
datalen, PID);
if (datalen > 0) {
if (!stream->pes_buffer_in_sync) {
/* Push the first fragment to PES filter to have a chance to
* detect GST_FLOW_LOST_SYNC.
*/
GST_LOG_OBJECT (demux, "fragment directly pushed to PES filter");
ret = gst_mpegts_demux_push_fragment (stream, data, datalen);
} else {
/* Otherwhise we buffer the PES fragment */
ret = gst_mpegts_stream_pes_buffer_push (stream, data, datalen);
/* If sync is lost here is due a pes_buffer_flush and we can try
* to resync in the PES filter with the current fragment
*/
if (ret == GST_FLOW_LOST_SYNC) {
GST_LOG_OBJECT (demux, "resync, fragment pushed to PES filter");
ret = gst_mpegts_demux_push_fragment (stream, data, datalen);
}
}
break;
} else {
GST_WARNING_OBJECT (demux, "overflow of datalen: %u so skipping",
datalen);
return GST_FLOW_OK;
}
}
}
}
done:
return ret;
skip:
{
GST_DEBUG_OBJECT (demux, "skipping, adaptation_field_control == 0");
return GST_FLOW_OK;
}
too_small:
{
GST_DEBUG_OBJECT (demux, "skipping, adaptation_field consumed all data");
return GST_FLOW_OK;
}
}
static FORCE_INLINE GstFlowReturn
gst_mpegts_demux_parse_transport_packet (GstMpegTSDemux * demux,
const guint8 * data)
{
GstFlowReturn ret = GST_FLOW_OK;
guint16 PID;
GstMpegTSStream *stream;
/* skip sync byte */
data++;
/* get PID */
PID = ((data[0] & 0x1f) << 8) | data[1];
/* Skip NULL packets */
if (G_UNLIKELY (PID == 0x1fff))
goto beach;
/* get the stream. */
stream = gst_mpegts_demux_get_stream_for_PID (demux, PID);
/* parse the stream */
ret = gst_mpegts_demux_parse_stream (demux, stream, data,
MPEGTS_NORMAL_TS_PACKETSIZE - 1);
if (demux->pcr[1] != -1 && demux->bitrate == -1) {
guint64 bitrate;
GST_DEBUG_OBJECT (demux, "pcr[0]:%" G_GUINT64_FORMAT, demux->pcr[0]);
GST_DEBUG_OBJECT (demux, "pcr[1]:%" G_GUINT64_FORMAT, demux->pcr[1]);
GST_DEBUG_OBJECT (demux, "diff in time %" GST_TIME_FORMAT,
GST_TIME_ARGS (MPEGTIME_TO_GSTTIME (demux->pcr[1] - demux->pcr[0])));
GST_DEBUG_OBJECT (demux, "stream->last_PCR_difference: %" G_GUINT64_FORMAT
", demux->num_packets %" G_GUINT64_FORMAT,
demux->pcr[1] - demux->pcr[0], demux->num_packets);
bitrate = gst_util_uint64_scale (GST_SECOND,
MPEGTS_NORMAL_TS_PACKETSIZE * demux->num_packets,
MPEGTIME_TO_GSTTIME (demux->pcr[1] - demux->pcr[0]));
/* somehow... I doubt a bitrate below one packet per second is valid */
if (bitrate > MPEGTS_NORMAL_TS_PACKETSIZE - 1) {
demux->bitrate = bitrate;
GST_DEBUG_OBJECT (demux, "bitrate is %" G_GINT64_FORMAT
" bytes per second", demux->bitrate);
} else {
GST_WARNING_OBJECT (demux, "Couldn't compute valid bitrate, recomputing");
demux->pcr[0] = demux->pcr[1] = -1;
demux->num_packets = -1;
}
}
beach:
demux->num_packets++;
return ret;
/* ERRORS */
}
static gboolean
gst_mpegts_demux_handle_seek_push (GstMpegTSDemux * demux, GstEvent * event)
{
gboolean res = FALSE;
gdouble rate;
GstFormat format;
GstSeekFlags flags;
GstSeekType start_type, stop_type;
gint64 start, stop, bstart, bstop;
GstEvent *bevent;
gst_event_parse_seek (event, &rate, &format, &flags, &start_type, &start,
&stop_type, &stop);
GST_DEBUG_OBJECT (demux, "seek event, rate: %f start: %" GST_TIME_FORMAT
" stop: %" GST_TIME_FORMAT, rate, GST_TIME_ARGS (start),
GST_TIME_ARGS (stop));
if (format == GST_FORMAT_BYTES) {
GST_DEBUG_OBJECT (demux, "seek not supported on format %d", format);
goto beach;
}
GST_DEBUG_OBJECT (demux, "seek - trying directly upstream first");
/* first try original format seek */
res = gst_pad_push_event (demux->sinkpad, gst_event_ref (event));
if (res == TRUE)
goto beach;
GST_DEBUG_OBJECT (demux, "seek - no upstream");
if (format != GST_FORMAT_TIME) {
/* From here down, we only support time based seeks */
GST_DEBUG_OBJECT (demux, "seek not supported on format %d", format);
goto beach;
}
/* We need to convert to byte based seek and we need a scr_rate for that. */
if (demux->bitrate == -1) {
GST_DEBUG_OBJECT (demux, "seek not possible, no bitrate");
goto beach;
}
GST_DEBUG_OBJECT (demux, "try with bitrate");
bstart = GSTTIME_TO_BYTES (start);
bstop = GSTTIME_TO_BYTES (stop);
GST_DEBUG_OBJECT (demux, "in bytes bstart %" G_GINT64_FORMAT " bstop %"
G_GINT64_FORMAT, bstart, bstop);
bevent = gst_event_new_seek (rate, GST_FORMAT_BYTES, flags, start_type,
bstart, stop_type, bstop);
res = gst_pad_push_event (demux->sinkpad, bevent);
beach:
gst_event_unref (event);
return res;
}
static gboolean
gst_mpegts_demux_src_event (GstPad * pad, GstEvent * event)
{
GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (gst_pad_get_parent (pad));
gboolean res = FALSE;
GST_DEBUG_OBJECT (demux, "got event %s",
gst_event_type_get_name (GST_EVENT_TYPE (event)));
switch (GST_EVENT_TYPE (event)) {
case GST_EVENT_SEEK:
res = gst_mpegts_demux_handle_seek_push (demux, event);
break;
default:
res = gst_pad_push_event (demux->sinkpad, event);
break;
}
gst_object_unref (demux);
return res;
}
static void
gst_mpegts_demux_flush (GstMpegTSDemux * demux, gboolean discard)
{
gint i;
GstMpegTSStream *PCR_stream;
GstMpegTSStream *PMT_stream;
GST_DEBUG_OBJECT (demux, "flushing MPEG TS demuxer (discard %d)", discard);
/* Start by flushing internal buffers */
gst_mpegts_demux_pes_buffer_flush (demux, discard);
/* Clear adapter */
gst_adapter_clear (demux->adapter);
/* Try resetting the last_PCR value as we will have a discont */
if (demux->current_PMT == 0)
goto beach;
PMT_stream = demux->streams[demux->current_PMT];
if (PMT_stream == NULL)
goto beach;
PCR_stream = demux->streams[PMT_stream->PMT.PCR_PID];
if (PCR_stream == NULL)
goto beach;
PCR_stream->last_PCR = -1;
/* Reset last time of all streams */
for (i = 0; i < MPEGTS_MAX_PID + 1; i++) {
GstMpegTSStream *stream = demux->streams[i];
if (stream) {
stream->last_time = 0;
stream->discont = TRUE;
}
}
beach:
return;
}
static gboolean
gst_mpegts_demux_send_event (GstMpegTSDemux * demux, GstEvent * event)
{
gint i;
gboolean have_stream = FALSE, res = TRUE;
for (i = 0; i < MPEGTS_MAX_PID + 1; i++) {
GstMpegTSStream *stream = demux->streams[i];
if (stream && stream->pad) {
res &= gst_pad_push_event (stream->pad, gst_event_ref (event));
have_stream = TRUE;
}
}
gst_event_unref (event);
return have_stream;
}
static gboolean
gst_mpegts_demux_sink_event (GstPad * pad, GstEvent * event)
{
GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (gst_pad_get_parent (pad));
gboolean res = FALSE;
GST_DEBUG_OBJECT (demux, "got event %s",
gst_event_type_get_name (GST_EVENT_TYPE (event)));
switch (GST_EVENT_TYPE (event)) {
case GST_EVENT_FLUSH_START:
res = gst_mpegts_demux_send_event (demux, event);
break;
case GST_EVENT_FLUSH_STOP:
gst_adapter_clear (demux->adapter);
gst_mpegts_demux_flush (demux, TRUE);
res = gst_mpegts_demux_send_event (demux, event);
break;
case GST_EVENT_EOS:
gst_mpegts_demux_flush (demux, FALSE);
/* Send the EOS event on each stream */
if (!(res = gst_mpegts_demux_send_event (demux, event))) {
/* we have no streams */
GST_ELEMENT_ERROR (demux, STREAM, TYPE_NOT_FOUND,
(NULL), ("No valid streams found at EOS"));
}
break;
case GST_EVENT_NEWSEGMENT:
{
gboolean update;
gdouble rate;
GstFormat format;
gint64 start, stop, time;
gst_event_parse_new_segment (event, &update, &rate, &format,
&start, &stop, &time);
gst_event_unref (event);
GST_INFO_OBJECT (demux, "received new segment: rate %g "
"format %d, start: %" G_GINT64_FORMAT ", stop: %" G_GINT64_FORMAT
", time: %" G_GINT64_FORMAT, rate, format, start, stop, time);
if (format == GST_FORMAT_BYTES && demux->bitrate != -1) {
gint64 tstart = 0, tstop = 0, pos = 0;
if (demux->base_pts != GST_CLOCK_TIME_NONE) {
tstart = tstop = demux->base_pts;
}
tstart += BYTES_TO_GSTTIME (start);
tstop += BYTES_TO_GSTTIME (stop);
pos = BYTES_TO_GSTTIME (time);
event = gst_event_new_new_segment (update, rate,
GST_FORMAT_TIME, tstart, tstop, pos);
GST_DEBUG_OBJECT (demux, "pushing time newsegment from %"
GST_TIME_FORMAT " to %" GST_TIME_FORMAT " pos %" GST_TIME_FORMAT,
GST_TIME_ARGS (tstart), GST_TIME_ARGS (tstop), GST_TIME_ARGS (pos));
res = gst_mpegts_demux_send_event (demux, event);
}
break;
}
default:
res = gst_mpegts_demux_send_event (demux, event);
break;
}
gst_object_unref (demux);
return res;
}
static gboolean
gst_mpegts_demux_provides_clock (GstElement * element)
{
GstMpegTSDemux *demux;
GstQuery *query;
gboolean is_live = FALSE;
GstPad *peer;
demux = GST_MPEGTS_DEMUX (element);
query = gst_query_new_latency ();
peer = gst_pad_get_peer (demux->sinkpad);
if (peer) {
if (gst_pad_query (peer, query))
gst_query_parse_latency (query, &is_live, NULL, NULL);
gst_object_unref (peer);
}
gst_query_unref (query);
return is_live;
}
static GstClock *
gst_mpegts_demux_provide_clock (GstElement * element)
{
GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (element);
if (gst_mpegts_demux_provides_clock (element)) {
if (demux->clock == NULL) {
demux->clock = g_object_new (GST_TYPE_SYSTEM_CLOCK, "name",
"MpegTSClock", NULL);
demux->clock_base = GST_CLOCK_TIME_NONE;
}
return gst_object_ref (demux->clock);
}
return NULL;
}
static const GstQueryType *
gst_mpegts_demux_src_pad_query_type (GstPad * pad)
{
static const GstQueryType types[] = {
GST_QUERY_LATENCY,
GST_QUERY_DURATION,
GST_QUERY_SEEKING,
0
};
return types;
}
static gboolean
gst_mpegts_demux_src_pad_query (GstPad * pad, GstQuery * query)
{
GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (gst_pad_get_parent (pad));
gboolean res = FALSE;
GstPad *peer;
switch (GST_QUERY_TYPE (query)) {
case GST_QUERY_LATENCY:
{
peer = gst_pad_get_peer (demux->sinkpad);
if (peer) {
res = gst_pad_query (peer, query);
if (res) {
gboolean is_live;
GstClockTime min_latency, max_latency;
gst_query_parse_latency (query, &is_live, &min_latency, &max_latency);
if (is_live) {
min_latency += TS_LATENCY * GST_MSECOND;
if (max_latency != GST_CLOCK_TIME_NONE)
max_latency += TS_LATENCY * GST_MSECOND;
}
gst_query_set_latency (query, is_live, min_latency, max_latency);
}
gst_object_unref (peer);
}
break;
}
case GST_QUERY_DURATION:
{
GstFormat format;
GstPad *peer;
gst_query_parse_duration (query, &format, NULL);
/* Try query upstream first */
peer = gst_pad_get_peer (demux->sinkpad);
if (peer) {
res = gst_pad_query (peer, query);
/* Try doing something with that query if it failed */
if (!res && format == GST_FORMAT_TIME && demux->bitrate != -1) {
/* Try using cache first */
if (GST_CLOCK_TIME_IS_VALID (demux->cache_duration)) {
GST_LOG_OBJECT (demux, "replying duration query from cache %"
GST_TIME_FORMAT, GST_TIME_ARGS (demux->cache_duration));
gst_query_set_duration (query, GST_FORMAT_TIME,
demux->cache_duration);
res = TRUE;
} else { /* Query upstream and approximate */
GstQuery *bquery = gst_query_new_duration (GST_FORMAT_BYTES);
gint64 duration = 0;
/* Query peer for duration in bytes */
res = gst_pad_query (peer, bquery);
if (res) {
/* Convert to time format */
gst_query_parse_duration (bquery, &format, &duration);
GST_DEBUG_OBJECT (demux, "query on peer pad reported bytes %"
G_GUINT64_FORMAT, duration);
demux->cache_duration = BYTES_TO_GSTTIME (duration);
GST_DEBUG_OBJECT (demux, "converted to time %" GST_TIME_FORMAT,
GST_TIME_ARGS (demux->cache_duration));
gst_query_set_duration (query, GST_FORMAT_TIME,
demux->cache_duration);
}
gst_query_unref (bquery);
}
} else {
GST_WARNING_OBJECT (demux, "unsupported query format or no bitrate "
"yet to approximate duration from bytes");
}
gst_object_unref (peer);
}
break;
}
case GST_QUERY_SEEKING:{
GstFormat fmt;
gst_query_parse_seeking (query, &fmt, NULL, NULL, NULL);
if (fmt == GST_FORMAT_BYTES) {
/* Seeking in BYTES format not supported at all */
gst_query_set_seeking (query, fmt, FALSE, -1, -1);
} else {
GstQuery *peerquery;
gboolean seekable;
/* Then ask upstream */
res = gst_pad_peer_query (demux->sinkpad, query);
if (res) {
/* If upstream can handle seeks we're done, if it
* can't we still have our TIME->BYTES conversion seek
*/
gst_query_parse_seeking (query, NULL, &seekable, NULL, NULL);
if (seekable || fmt != GST_FORMAT_TIME)
goto beach;
}
/* We can't say anything about seekability if we didn't
* have a second PCR yet because the bitrate is calculated
* from this
*/
if (demux->bitrate == -1 && demux->pcr[1] == -1)
goto beach;
/* We can seek if upstream supports BYTES seeks and we
* have a bitrate
*/
peerquery = gst_query_new_seeking (GST_FORMAT_BYTES);
res = gst_pad_peer_query (demux->sinkpad, peerquery);
if (!res || demux->bitrate == -1) {
gst_query_set_seeking (query, fmt, FALSE, -1, -1);
} else {
gst_query_parse_seeking (peerquery, NULL, &seekable, NULL, NULL);
if (seekable)
gst_query_set_seeking (query, GST_FORMAT_TIME, TRUE, 0, -1);
else
gst_query_set_seeking (query, fmt, FALSE, -1, -1);
}
gst_query_unref (peerquery);
res = TRUE;
}
break;
}
default:
res = gst_pad_query_default (pad, query);
break;
}
beach:
gst_object_unref (demux);
return res;
}
static FORCE_INLINE gint
is_mpegts_sync (const guint8 * in_data, const guint8 * end_data,
guint packetsize)
{
guint ret = 0;
if (G_LIKELY (IS_MPEGTS_SYNC (in_data)))
return 100;
if (in_data + packetsize < end_data - 5) {
if (G_LIKELY (IS_MPEGTS_SYNC (in_data + packetsize)))
ret += 50;
}
if (in_data[0] == 0x47) {
ret += 25;
if ((in_data[1] & 0x80) == 0x00)
ret += 10;
if ((in_data[3] & 0x10) == 0x10)
ret += 5;
}
return ret;
}
static inline void
gst_mpegts_demux_detect_packet_size (GstMpegTSDemux * demux, guint len)
{
guint i, packetsize = 0;
for (i = 1; i < len; i++) {
packetsize = demux->sync_lut[i] - demux->sync_lut[i - 1];
if (packetsize == MPEGTS_NORMAL_TS_PACKETSIZE ||
packetsize == MPEGTS_M2TS_TS_PACKETSIZE ||
packetsize == MPEGTS_DVB_ASI_TS_PACKETSIZE ||
packetsize == MPEGTS_ATSC_TS_PACKETSIZE)
goto done;
else
packetsize = 0;
}
done:
demux->packetsize = (packetsize ? packetsize : MPEGTS_NORMAL_TS_PACKETSIZE);
GST_DEBUG_OBJECT (demux, "packet_size set to %d bytes", demux->packetsize);
}
static FORCE_INLINE guint
gst_mpegts_demux_sync_scan (GstMpegTSDemux * demux, const guint8 * in_data,
guint size, guint * flush)
{
guint sync_count = 0;
const guint8 *end_scan = in_data + size - demux->packetsize;
guint8 *ptr_data = (guint8 *) in_data;
guint packetsize =
(demux->packetsize ? demux->packetsize : MPEGTS_NORMAL_TS_PACKETSIZE);
/* Check if the LUT table is big enough */
if (G_UNLIKELY (demux->sync_lut_len < (size / packetsize))) {
demux->sync_lut_len = size / packetsize;
if (demux->sync_lut)
g_free (demux->sync_lut);
demux->sync_lut = g_new0 (guint8 *, demux->sync_lut_len);
GST_DEBUG_OBJECT (demux, "created sync LUT table with %u entries",
demux->sync_lut_len);
}
while (ptr_data <= end_scan && sync_count < demux->sync_lut_len) {
/* if sync code is found try to store it in the LUT */
guint chance = is_mpegts_sync (ptr_data, end_scan, packetsize);
if (G_LIKELY (chance > 50)) {
/* skip paketsize bytes and try find next */
guint8 *next_sync = ptr_data + packetsize;
if (next_sync < end_scan) {
demux->sync_lut[sync_count] = ptr_data;
sync_count++;
ptr_data += packetsize;
} else
goto done;
} else {
ptr_data++;
}
}
done:
if (G_UNLIKELY (!demux->packetsize))
gst_mpegts_demux_detect_packet_size (demux, sync_count);
*flush = MIN (ptr_data - in_data, size);
return sync_count;
}
static GstFlowReturn
gst_mpegts_demux_chain (GstPad * pad, GstBuffer * buffer)
{
GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (gst_pad_get_parent (pad));
GstFlowReturn ret = GST_FLOW_OK;
const guint8 *data;
guint avail;
guint flush = 0;
gint i;
guint sync_count;
if (GST_BUFFER_IS_DISCONT (buffer)) {
gst_mpegts_demux_flush (demux, FALSE);
}
/* first push the new buffer into the adapter */
gst_adapter_push (demux->adapter, buffer);
/* check if there's enough data to parse a packet */
avail = gst_adapter_available (demux->adapter);
if (G_UNLIKELY (avail < demux->packetsize))
goto done;
/* recover all data from adapter */
data = gst_adapter_peek (demux->adapter, avail);
/* scan for sync codes */
sync_count = gst_mpegts_demux_sync_scan (demux, data, avail, &flush);
/* process all packets */
for (i = 0; i < sync_count; i++) {
ret = gst_mpegts_demux_parse_transport_packet (demux, demux->sync_lut[i]);
if (G_UNLIKELY (ret == GST_FLOW_LOST_SYNC
|| ret == GST_FLOW_NEED_MORE_DATA)) {
ret = GST_FLOW_OK;
continue;
}
if (G_UNLIKELY (ret != GST_FLOW_OK)) {
flush = demux->sync_lut[i] - data + demux->packetsize;
flush = MIN (avail, flush);
goto done;
}
}
done:
/* flush processed data */
if (flush) {
GST_DEBUG_OBJECT (demux, "flushing %d/%d", flush, avail);
gst_adapter_flush (demux->adapter, flush);
}
gst_object_unref (demux);
return ret;
}
static GstStateChangeReturn
gst_mpegts_demux_change_state (GstElement * element, GstStateChange transition)
{
GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (element);
GstStateChangeReturn result;
switch (transition) {
case GST_STATE_CHANGE_NULL_TO_READY:
demux->adapter = gst_adapter_new ();
break;
case GST_STATE_CHANGE_READY_TO_PAUSED:
break;
default:
break;
}
result = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
switch (transition) {
case GST_STATE_CHANGE_PAUSED_TO_READY:
gst_mpegts_demux_reset (demux);
break;
case GST_STATE_CHANGE_READY_TO_NULL:
g_object_unref (demux->adapter);
if (demux->sync_lut)
g_free (demux->sync_lut);
demux->sync_lut = NULL;
demux->sync_lut_len = 0;
break;
default:
break;
}
return result;
}
static GValueArray *
mpegts_demux_build_pat_info (GstMpegTSDemux * demux)
{
GValueArray *vals = NULL;
GstMpegTSPAT *PAT;
gint i;
g_return_val_if_fail (demux->streams[0] != NULL, NULL);
g_return_val_if_fail (demux->streams[0]->PID_type ==
PID_TYPE_PROGRAM_ASSOCIATION, NULL);
PAT = &(demux->streams[0]->PAT);
vals = g_value_array_new (PAT->entries->len);
for (i = 0; i < PAT->entries->len; i++) {
GstMpegTSPATEntry *cur_entry =
&g_array_index (PAT->entries, GstMpegTSPATEntry, i);
GValue v = { 0, };
MpegTsPatInfo *info_obj;
info_obj = mpegts_pat_info_new (cur_entry->program_number, cur_entry->PID);
g_value_init (&v, G_TYPE_OBJECT);
g_value_take_object (&v, info_obj);
g_value_array_append (vals, &v);
}
return vals;
}
static MpegTsPmtInfo *
mpegts_demux_build_pmt_info (GstMpegTSDemux * demux, guint16 pmt_pid)
{
MpegTsPmtInfo *info_obj;
GstMpegTSPMT *PMT;
gint i;
g_return_val_if_fail (demux->streams[pmt_pid] != NULL, NULL);
g_return_val_if_fail (demux->streams[pmt_pid]->PID_type ==
PID_TYPE_PROGRAM_MAP, NULL);
PMT = &(demux->streams[pmt_pid]->PMT);
info_obj = mpegts_pmt_info_new (PMT->program_number, PMT->PCR_PID,
PMT->version_number);
for (i = 0; i < PMT->entries->len; i++) {
GstMpegTSStream *stream;
MpegTsPmtStreamInfo *stream_info;
GstMpegTSPMTEntry *cur_entry =
&g_array_index (PMT->entries, GstMpegTSPMTEntry, i);
stream = demux->streams[cur_entry->PID];
stream_info =
mpegts_pmt_stream_info_new (cur_entry->PID, stream->stream_type);
if (stream->ES_info) {
int i;
/* add languages */
guint8 *iso639_languages =
gst_mpeg_descriptor_find (stream->ES_info, DESC_ISO_639_LANGUAGE);
if (iso639_languages) {
for (i = 0; i < DESC_ISO_639_LANGUAGE_codes_n (iso639_languages); i++) {
gchar *language_n = (gchar *)
DESC_ISO_639_LANGUAGE_language_code_nth (iso639_languages, i);
mpegts_pmt_stream_info_add_language (stream_info,
g_strndup (language_n, 3));
}
}
for (i = 0; i < gst_mpeg_descriptor_n_desc (stream->ES_info); ++i) {
guint8 *desc = gst_mpeg_descriptor_nth (stream->ES_info, i);
/* add the whole descriptor, tag + length + DESC_LENGTH bytes */
mpegts_pmt_stream_info_add_descriptor (stream_info,
(gchar *) desc, 2 + DESC_LENGTH (desc));
}
}
mpegts_pmt_info_add_stream (info_obj, stream_info);
}
return info_obj;
}
static void
gst_mpegts_demux_set_property (GObject * object, guint prop_id,
const GValue * value, GParamSpec * pspec)
{
GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (object);
gchar **pids;
guint num_pids;
int i;
switch (prop_id) {
case PROP_ES_PIDS:
pids = g_strsplit (g_value_get_string (value), ":", -1);
num_pids = g_strv_length (pids);
if (num_pids > 0) {
demux->elementary_pids = g_new0 (guint16, num_pids);
demux->nb_elementary_pids = num_pids;
for (i = 0; i < num_pids; i++) {
demux->elementary_pids[i] = strtol (pids[i], NULL, 0);
GST_INFO ("partial TS ES pid %d", demux->elementary_pids[i]);
}
}
g_strfreev (pids);
break;
case PROP_CHECK_CRC:
demux->check_crc = g_value_get_boolean (value);
break;
case PROP_PROGRAM_NUMBER:
demux->program_number = g_value_get_int (value);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
static void
gst_mpegts_demux_get_property (GObject * object, guint prop_id,
GValue * value, GParamSpec * pspec)
{
GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (object);
int i;
switch (prop_id) {
case PROP_ES_PIDS:
if (demux->nb_elementary_pids == 0) {
g_value_set_static_string (value, "");
} else {
GString *ts_pids;
ts_pids = g_string_sized_new (32);
/* FIXME: align with property description which uses hex numbers? */
g_string_append_printf (ts_pids, "%d", demux->elementary_pids[0]);
for (i = 1; i < demux->nb_elementary_pids; i++) {
g_string_append_printf (ts_pids, ":%d", demux->elementary_pids[i]);
}
g_value_take_string (value, g_string_free (ts_pids, FALSE));
}
break;
case PROP_CHECK_CRC:
g_value_set_boolean (value, demux->check_crc);
break;
case PROP_PROGRAM_NUMBER:
g_value_set_int (value, demux->program_number);
break;
case PROP_PAT_INFO:
{
if (demux->streams[0] != NULL) {
g_value_take_boxed (value, mpegts_demux_build_pat_info (demux));
}
break;
}
case PROP_PMT_INFO:
{
if (demux->current_PMT != 0 && demux->streams[demux->current_PMT] != NULL) {
g_value_take_object (value, mpegts_demux_build_pmt_info (demux,
demux->current_PMT));
}
break;
}
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
gboolean
gst_mpegts_demux_plugin_init (GstPlugin * plugin)
{
if (!gst_element_register (plugin, "mpegtsdemux",
GST_RANK_PRIMARY, GST_TYPE_MPEGTS_DEMUX))
return FALSE;
return TRUE;
}