BPI/BPI-CM9-Linux
6
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
main
BPI-CM9-Linux/aml-comp/multimedia/gst-plugin-aml-demux/aml-tsdemux/amltsbase.c
T
Qubot 818cb714e3 Import complete source tree
2026-08-24 10:03:24 +08:00

2092 lines
64 KiB
C
Raw Permalink Blame History

/*
* mpegtsbase.c -
* Copyright (C) 2007 Alessandro Decina
* 2010 Edward Hervey
* Copyright (C) 2011, Hewlett-Packard Development Company, L.P.
* Author: Youness Alaoui <[email protected]>, Collabora Ltd.
* Author: Sebastian Dröge <[email protected]>, Collabora Ltd.
* Author: Edward Hervey <[email protected]>, Collabora Ltd.
*
* Authors:
* Alessandro Decina <[email protected]>
* Zaheer Abbas Merali <zaheerabbas at merali dot org>
* Edward Hervey <[email protected]>
*
* 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., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <stdlib.h>
#include <string.h>
#include <glib.h>
#include <gst/gst-i18n-plugin.h>
#include "amltsbase.h"
#include "gstamldesc.h"
#define RUNNING_STATUS_RUNNING 4
GST_DEBUG_CATEGORY_STATIC (amlts_base_debug);
#define GST_CAT_DEFAULT amlts_base_debug
static GQuark QUARK_PROGRAMS;
static GQuark QUARK_PROGRAM_NUMBER;
static GQuark QUARK_PID;
static GQuark QUARK_PCR_PID;
static GQuark QUARK_STREAMS;
static GQuark QUARK_STREAM_TYPE;
static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink",
GST_PAD_SINK,
GST_PAD_ALWAYS,
GST_STATIC_CAPS ("video/mpegts, " "systemstream = (boolean) true ")
);
#define DEFAULT_IGNORE_PCR FALSE
enum
{
PROP_0,
PROP_PARSE_PRIVATE_SECTIONS,
PROP_IGNORE_PCR,
/* FILL ME */
};
static void amlts_base_dispose (GObject * object);
static void amlts_base_finalize (GObject * object);
static void amlts_base_set_property (GObject * object, guint prop_id,
const GValue * value, GParamSpec * pspec);
static void amlts_base_get_property (GObject * object, guint prop_id,
GValue * value, GParamSpec * pspec);
static void amlts_base_free_program (AmlTSBaseProgram * program);
static void amlts_base_deactivate_program (AmlTSBase * base,
AmlTSBaseProgram * program);
static gboolean amlts_base_sink_activate (GstPad * pad, GstObject * parent);
static gboolean amlts_base_sink_activate_mode (GstPad * pad,
GstObject * parent, GstPadMode mode, gboolean active);
static GstFlowReturn amlts_base_chain (GstPad * pad, GstObject * parent,
GstBuffer * buf);
static gboolean amlts_base_sink_event (GstPad * pad, GstObject * parent,
GstEvent * event);
static gboolean amlts_base_sink_query (GstPad * pad, GstObject * parent,
GstQuery * query);
static gboolean amlts_base_default_sink_query (AmlTSBase * base,
GstQuery * query);
static GstStateChangeReturn amlts_base_change_state (GstElement * element,
GstStateChange transition);
static gboolean amlts_base_get_tags_from_eit (AmlTSBase * base,
GstAmlMpegtsSection * section);
static gboolean amlts_base_parse_atsc_mgt (AmlTSBase * base,
GstAmlMpegtsSection * section);
static void aml_remove_each_program (AmlTSBaseProgram * program,
AmlTSBase * base);
static void
_aml_extra_init (void)
{
QUARK_PROGRAMS = g_quark_from_string ("programs");
QUARK_PROGRAM_NUMBER = g_quark_from_string ("program-number");
QUARK_PID = g_quark_from_string ("pid");
QUARK_PCR_PID = g_quark_from_string ("pcr-pid");
QUARK_STREAMS = g_quark_from_string ("streams");
QUARK_STREAM_TYPE = g_quark_from_string ("stream-type");
GST_DEBUG_CATEGORY_INIT (amlts_base_debug, "amltsbase", 0,
"MPEG transport stream base class");
gst_amlmpegts_initialize ();
}
#define amlts_base_parent_class parent_class
G_DEFINE_TYPE_WITH_CODE (AmlTSBase, amlts_base, GST_TYPE_ELEMENT,
_aml_extra_init ());
/* Default implementation is that mpegtsbase can remove any program */
static gboolean
amlts_base_can_remove_program (AmlTSBase * base, AmlTSBaseProgram * program)
{
return TRUE;
}
static void
amlts_base_class_init (AmlTSBaseClass * klass)
{
GObjectClass *gobject_class;
GstElementClass *element_class;
klass->can_remove_program = amlts_base_can_remove_program;
element_class = GST_ELEMENT_CLASS (klass);
element_class->change_state = amlts_base_change_state;
gst_element_class_add_static_pad_template (element_class, &sink_template);
gobject_class = G_OBJECT_CLASS (klass);
gobject_class->dispose = amlts_base_dispose;
gobject_class->finalize = amlts_base_finalize;
gobject_class->set_property = amlts_base_set_property;
gobject_class->get_property = amlts_base_get_property;
g_object_class_install_property (gobject_class, PROP_PARSE_PRIVATE_SECTIONS,
g_param_spec_boolean ("parse-private-sections", "Parse private sections",
"Parse private sections", FALSE,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
/**
* GstMpegtsBase:ignore-pcr:
*
* Ignore PCR (Program Clock Reference) data from MPEG-TS PSI.
* This can help with playback of some broken files.
*
* Since: 1.18
*/
g_object_class_install_property (gobject_class, PROP_IGNORE_PCR,
g_param_spec_boolean ("ignore-pcr", "Ignore PCR stream for timing",
"Ignore PCR stream for timing", DEFAULT_IGNORE_PCR,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
klass->sink_query = GST_DEBUG_FUNCPTR (amlts_base_default_sink_query);
klass->handle_psi = NULL;
#if ((GST_VERSION_MAJOR == 1) && (GST_VERSION_MINOR >= 18))
gst_type_mark_as_plugin_api (GST_TYPE_AMLTS_BASE, 0);
#endif
}
static void
amlts_base_set_property (GObject * object, guint prop_id,
const GValue * value, GParamSpec * pspec)
{
AmlTSBase *base = GST_AMLTS_BASE (object);
switch (prop_id) {
case PROP_PARSE_PRIVATE_SECTIONS:
base->parse_private_sections = g_value_get_boolean (value);
break;
case PROP_IGNORE_PCR:
base->ignore_pcr = g_value_get_boolean (value);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
}
}
static void
amlts_base_get_property (GObject * object, guint prop_id,
GValue * value, GParamSpec * pspec)
{
AmlTSBase *base = GST_AMLTS_BASE (object);
switch (prop_id) {
case PROP_PARSE_PRIVATE_SECTIONS:
g_value_set_boolean (value, base->parse_private_sections);
break;
case PROP_IGNORE_PCR:
g_value_set_boolean (value, base->ignore_pcr);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
}
}
static void
amlts_base_reset (AmlTSBase * base)
{
AmlTSBaseClass *klass = GST_AMLTS_BASE_GET_CLASS (base);
amlts_packetizer_clear (base->packetizer);
memset (base->is_pes, 0, 1024);
memset (base->known_psi, 0, 1024);
/* FIXME : Actually these are not *always* know SI streams
* depending on the variant of mpeg-ts being used. */
/* Known PIDs : PAT, TSDT, IPMP CIT */
MPEGTS_BIT_SET (base->known_psi, 0);
MPEGTS_BIT_SET (base->known_psi, 2);
MPEGTS_BIT_SET (base->known_psi, 3);
/* TDT, TOT, ST */
MPEGTS_BIT_SET (base->known_psi, 0x14);
/* network synchronization */
MPEGTS_BIT_SET (base->known_psi, 0x15);
/* ATSC */
MPEGTS_BIT_SET (base->known_psi, 0x1ffb);
if (base->pat) {
g_ptr_array_unref (base->pat);
base->pat = NULL;
}
gst_segment_init (&base->segment, GST_FORMAT_UNDEFINED);
gst_segment_init (&base->out_segment, GST_FORMAT_UNDEFINED);
base->last_seek_seqnum = GST_SEQNUM_INVALID;
base->mode = BASE_MODE_STREAMING;
base->seen_pat = FALSE;
base->seek_offset = -1;
g_ptr_array_foreach (base->programs, (GFunc) aml_remove_each_program, base);
g_ptr_array_remove_range (base->programs, 0, base->programs->len);
base->streams_aware = GST_OBJECT_PARENT (base)
&& GST_OBJECT_FLAG_IS_SET (GST_OBJECT_PARENT (base),
GST_BIN_FLAG_STREAMS_AWARE);
GST_DEBUG_OBJECT (base, "Streams aware : %d", base->streams_aware);
gst_event_replace (&base->seek_event, NULL);
if (klass->reset)
klass->reset (base);
}
static void
amlts_base_init (AmlTSBase * base)
{
base->sinkpad = gst_pad_new_from_static_template (&sink_template, "sink");
gst_pad_set_activate_function (base->sinkpad, amlts_base_sink_activate);
gst_pad_set_activatemode_function (base->sinkpad,
amlts_base_sink_activate_mode);
gst_pad_set_chain_function (base->sinkpad, amlts_base_chain);
gst_pad_set_event_function (base->sinkpad, amlts_base_sink_event);
gst_pad_set_query_function (base->sinkpad, amlts_base_sink_query);
gst_element_add_pad (GST_ELEMENT (base), base->sinkpad);
base->disposed = FALSE;
base->packetizer = amlts_packetizer_new ();
base->programs =
g_ptr_array_new_full (16, (GDestroyNotify) amlts_base_free_program);
base->parse_private_sections = FALSE;
base->is_pes = g_new0 (guint8, 1024);
base->known_psi = g_new0 (guint8, 1024);
base->program_size = sizeof (AmlTSBaseProgram);
base->stream_size = sizeof (AmlTSBaseStream);
base->push_data = TRUE;
base->push_section = TRUE;
base->ignore_pcr = DEFAULT_IGNORE_PCR;
base->probe_buflist = gst_buffer_list_new();
base->av_started = FALSE;
base->push_scandata= FALSE;
amlts_base_reset (base);
}
static void
amlts_base_dispose (GObject * object)
{
AmlTSBase *base = GST_AMLTS_BASE (object);
if (base->probe_buflist) {
gst_buffer_list_unref(base->probe_buflist);
}
if (!base->disposed) {
g_object_unref (base->packetizer);
base->disposed = TRUE;
g_free (base->known_psi);
g_free (base->is_pes);
}
if (G_OBJECT_CLASS (parent_class)->dispose)
G_OBJECT_CLASS (parent_class)->dispose (object);
}
static void
amlts_base_finalize (GObject * object)
{
AmlTSBase *base = GST_AMLTS_BASE (object);
if (base->pat) {
g_ptr_array_unref (base->pat);
base->pat = NULL;
}
g_ptr_array_free (base->programs, TRUE);
gst_event_replace (&base->seek_event, NULL);
if (G_OBJECT_CLASS (parent_class)->finalize)
G_OBJECT_CLASS (parent_class)->finalize (object);
}
/* returns NULL if no matching descriptor found *
* otherwise returns a descriptor that needs to *
* be freed */
const GstAmlMpegtsDescriptor *
amlts_get_descriptor_from_stream (AmlTSBaseStream * stream, guint8 tag)
{
GstAmlMpegtsPMTStream *pmt = stream->stream;
GST_DEBUG ("Searching for tag 0x%02x in stream 0x%04x (stream_type 0x%02x)",
tag, stream->pid, stream->stream_type);
return gst_amlmpegts_find_descriptor (pmt->descriptors, tag);
}
//aml mark: since 1.20
const GstAmlMpegtsDescriptor *
amlts_get_descriptor_from_stream_with_extension (AmlTSBaseStream * stream,
guint8 tag, guint8 tag_extension)
{
GstAmlMpegtsPMTStream *pmt = stream->stream;
GST_DEBUG ("Searching for tag 0x%02x tag_extension 0x%02x "
"in stream 0x%04x (stream_type 0x%02x)",
tag, tag_extension, stream->pid, stream->stream_type);
#if ((GST_VERSION_MAJOR == 1) && (GST_VERSION_MINOR >= 20))
return gst_amlmpegts_find_descriptor_with_extension (pmt->descriptors, tag,
tag_extension);
#else
return NULL;
#endif
}
typedef struct
{
gboolean res;
guint16 pid;
} PIDLookup;
static void
aml_foreach_pid_in_program (AmlTSBaseProgram * program, PIDLookup * lookup)
{
if (!program->active)
return;
if (program->streams[lookup->pid])
lookup->res = TRUE;
}
static gboolean
amlts_pid_in_active_programs (AmlTSBase * base, guint16 pid)
{
PIDLookup lookup;
lookup.res = FALSE;
lookup.pid = pid;
g_ptr_array_foreach (base->programs, (GFunc) aml_foreach_pid_in_program, &lookup);
return lookup.res;
}
/* returns NULL if no matching descriptor found *
* otherwise returns a descriptor that needs to *
* be freed */
const GstAmlMpegtsDescriptor *
amlts_get_descriptor_from_program (AmlTSBaseProgram * program, guint8 tag)
{
const GstAmlMpegtsPMT *pmt = program->pmt;
return gst_amlmpegts_find_descriptor (pmt->descriptors, tag);
}
static gchar *
_aml_get_upstream_id (GstElement * element, GstPad * sinkpad)
{
gchar *upstream_id = gst_pad_get_stream_id (sinkpad);
if (!upstream_id) {
/* Try to create one from the upstream URI, else use a randome number */
GstQuery *query;
gchar *uri = NULL;
/* Try to generate one from the URI query and
* if it fails take a random number instead */
query = gst_query_new_uri ();
if (gst_element_query (element, query)) {
gst_query_parse_uri (query, &uri);
}
if (uri) {
GChecksum *cs;
/* And then generate an SHA256 sum of the URI */
cs = g_checksum_new (G_CHECKSUM_SHA256);
g_checksum_update (cs, (const guchar *) uri, strlen (uri));
g_free (uri);
upstream_id = g_strdup (g_checksum_get_string (cs));
g_checksum_free (cs);
} else {
/* Just get some random number if the URI query fails */
GST_FIXME_OBJECT (element, "Creating random stream-id, consider "
"implementing a deterministic way of creating a stream-id");
upstream_id =
g_strdup_printf ("%08x%08x%08x%08x", g_random_int (), g_random_int (),
g_random_int (), g_random_int ());
}
gst_query_unref (query);
}
return upstream_id;
}
static AmlTSBaseProgram *
amlts_base_new_program (AmlTSBase * base,
gint program_number, guint16 pmt_pid)
{
AmlTSBaseProgram *program;
gchar *upstream_id, *stream_id;
GST_DEBUG_OBJECT (base, "program_number : %d, pmt_pid : 0x%04x",
program_number, pmt_pid);
program = g_malloc0 (base->program_size);
program->program_number = program_number;
program->pmt_pid = pmt_pid;
program->pcr_pid = G_MAXUINT16;
program->streams = g_new0 (AmlTSBaseStream *, 0x2000);
program->patcount = 0;
upstream_id = _aml_get_upstream_id ((GstElement *) base, base->sinkpad);
stream_id = g_strdup_printf ("%s:%d", upstream_id, program_number);
program->collection = gst_stream_collection_new (stream_id);
g_free (stream_id);
g_free (upstream_id);
return program;
}
AmlTSBaseProgram *
amlts_base_add_program (AmlTSBase * base,
gint program_number, guint16 pmt_pid)
{
AmlTSBaseProgram *program;
GST_DEBUG_OBJECT (base, "program_number : %d, pmt_pid : 0x%04x",
program_number, pmt_pid);
program = amlts_base_new_program (base, program_number, pmt_pid);
/* Mark the PMT PID as being a known PSI PID */
if (G_UNLIKELY (MPEGTS_BIT_IS_SET (base->known_psi, pmt_pid))) {
GST_FIXME ("Refcounting. Setting twice a PID (0x%04x) as known PSI",
pmt_pid);
}
MPEGTS_BIT_SET (base->known_psi, pmt_pid);
/* Ensure the PMT PID was not used by some PES stream */
if (G_UNLIKELY (MPEGTS_BIT_IS_SET (base->is_pes, pmt_pid))) {
GST_DEBUG ("New program PMT PID was previously used by a PES stream");
MPEGTS_BIT_UNSET (base->is_pes, pmt_pid);
}
g_ptr_array_add (base->programs, program);
return program;
}
AmlTSBaseProgram *
amlts_base_get_program (AmlTSBase * base, gint program_number)
{
guint i;
for (i = 0; i < base->programs->len; i++) {
AmlTSBaseProgram *program = g_ptr_array_index (base->programs, i);
if (program->program_number == program_number)
return program;
}
return NULL;
}
static AmlTSBaseProgram *
amlts_base_steal_program (AmlTSBase * base, gint program_number)
{
guint i;
for (i = 0; i < base->programs->len; i++) {
AmlTSBaseProgram *program = g_ptr_array_index (base->programs, i);
if (program->program_number == program_number) {
#if GLIB_CHECK_VERSION(2, 58, 0)
return g_ptr_array_steal_index (base->programs, i);
#else
program->recycle = TRUE;
g_ptr_array_remove_index (base->programs, i);
return program;
#endif
}
}
return NULL;
}
static void
amlts_base_free_stream (AmlTSBaseStream * stream)
{
if (stream->stream_object)
gst_object_unref (stream->stream_object);
if (stream->stream_id)
g_free (stream->stream_id);
g_free (stream);
}
static void
amlts_base_free_program (AmlTSBaseProgram * program)
{
GList *tmp;
if (program->recycle) {
program->recycle = FALSE;
return;
}
if (program->pmt) {
gst_amlmpegts_section_unref (program->section);
program->pmt = NULL;
}
/* FIXME FIXME FIXME FREE STREAM OBJECT ! */
for (tmp = program->stream_list; tmp; tmp = tmp->next)
amlts_base_free_stream ((AmlTSBaseStream *) tmp->data);
if (program->stream_list)
g_list_free (program->stream_list);
g_free (program->streams);
if (program->tags)
gst_tag_list_unref (program->tags);
if (program->collection)
gst_object_unref (program->collection);
g_free (program);
}
void
amlts_base_deactivate_and_free_program (AmlTSBase * base,
AmlTSBaseProgram * program)
{
GST_DEBUG_OBJECT (base, "program_number : %d", program->program_number);
amlts_base_deactivate_program (base, program);
amlts_base_free_program (program);
}
static void
amlts_base_remove_program (AmlTSBase * base, AmlTSBaseProgram * program)
{
GST_DEBUG_OBJECT (base, "program_number : %d", program->program_number);
g_ptr_array_remove (base->programs, program);
}
static guint32
aml_get_registration_from_descriptors (GPtrArray * descriptors)
{
const GstAmlMpegtsDescriptor *desc;
if ((desc =
gst_amlmpegts_find_descriptor (descriptors,
GST_MTS_DESC_REGISTRATION))) {
if (G_UNLIKELY (desc->length < 4)) {
GST_WARNING ("Registration descriptor with length < 4. (Corrupted ?)");
} else
return GST_READ_UINT32_BE (desc->data + 2);
}
return 0;
}
//aml mark: since 1.20
static gboolean
aml_find_registration_in_descriptors (GPtrArray * descriptors,
guint32 registration_id)
{
guint i, nb_desc;
if (!descriptors)
return FALSE;
nb_desc = descriptors->len;
for (i = 0; i < nb_desc; i++) {
GstAmlMpegtsDescriptor *desc = g_ptr_array_index (descriptors, i);
if (desc->tag == GST_MTS_DESC_REGISTRATION) {
guint32 reg_desc = GST_READ_UINT32_BE (desc->data + 2);
if (reg_desc == registration_id)
return TRUE;
}
}
return FALSE;
}
static AmlTSBaseStream *
amlts_base_program_add_stream (AmlTSBase * base,
AmlTSBaseProgram * program, guint16 pid, guint8 stream_type,
GstAmlMpegtsPMTStream * stream)
{
AmlTSBaseClass *klass = GST_AMLTS_BASE_GET_CLASS (base);
AmlTSBaseStream *bstream;
GST_DEBUG ("pid:0x%04x, stream_type:0x%03x", pid, stream_type);
/* FIXME : PID information/nature might change through time.
* We therefore *do* want to be able to replace an existing stream
* with updated information */
if (G_UNLIKELY (program->streams[pid])) {
if (stream_type != 0xff)
GST_WARNING ("Stream already present !");
return NULL;
}
bstream = g_malloc0 (base->stream_size);
bstream->stream_id =
g_strdup_printf ("%s/%08x",
gst_stream_collection_get_upstream_id (program->collection), pid);
bstream->pid = pid;
bstream->stream_type = stream_type;
bstream->stream = stream;
/* We don't yet know the stream type, subclasses will fill that */
bstream->stream_object = gst_stream_new (bstream->stream_id, NULL,
GST_STREAM_TYPE_UNKNOWN, GST_STREAM_FLAG_NONE);
if (stream) {
bstream->registration_id =
aml_get_registration_from_descriptors (stream->descriptors);
GST_DEBUG ("PID 0x%04x, registration_id %" SAFE_FOURCC_FORMAT,
bstream->pid, SAFE_FOURCC_ARGS (bstream->registration_id));
}
program->streams[pid] = bstream;
program->stream_list = g_list_append (program->stream_list, bstream);
if (klass->stream_added)
if (klass->stream_added (base, bstream, program)) {
gst_stream_collection_add_stream (program->collection,
(GstStream *) gst_object_ref (bstream->stream_object));
bstream->in_collection = TRUE;
}
return bstream;
}
static void
amlts_base_program_remove_stream (AmlTSBase * base,
AmlTSBaseProgram * program, guint16 pid)
{
AmlTSBaseClass *klass;
AmlTSBaseStream *stream = program->streams[pid];
GST_DEBUG ("pid:0x%04x", pid);
if (G_UNLIKELY (stream == NULL)) {
/* Can happen if the PCR PID is the same as a audio/video PID */
GST_DEBUG ("Stream already removed");
return;
}
klass = GST_AMLTS_BASE_GET_CLASS (base);
/* If subclass needs it, inform it of the stream we are about to remove */
if (klass->stream_removed)
klass->stream_removed (base, stream);
program->stream_list = g_list_remove_all (program->stream_list, stream);
amlts_base_free_stream (stream);
program->streams[pid] = NULL;
}
/* Check if pmtstream is already present in the program */
static inline gboolean
_aml_stream_in_pmt (const GstAmlMpegtsPMT * pmt, AmlTSBaseStream * stream)
{
guint i, nbstreams = pmt->streams->len;
for (i = 0; i < nbstreams; i++) {
GstAmlMpegtsPMTStream *pmt_stream = g_ptr_array_index (pmt->streams, i);
if (pmt_stream->pid == stream->pid &&
pmt_stream->stream_type == stream->stream_type)
return TRUE;
}
return FALSE;
}
static inline gboolean
_aml_pmt_stream_in_program (AmlTSBaseProgram * program,
GstAmlMpegtsPMTStream * stream)
{
AmlTSBaseStream *old_stream = program->streams[stream->pid];
if (!old_stream)
return FALSE;
return old_stream->stream_type == stream->stream_type;
}
static gboolean
amlts_base_update_program (AmlTSBase * base, AmlTSBaseProgram * program,
GstAmlMpegtsSection * section, const GstAmlMpegtsPMT * pmt)
{
AmlTSBaseClass *klass = GST_AMLTS_BASE_GET_CLASS (base);
const gchar *stream_id =
gst_stream_collection_get_upstream_id (program->collection);
GstStreamCollection *collection;
GList *tmp, *toremove;
guint i, nbstreams;
/* Create new collection */
collection = gst_stream_collection_new (stream_id);
gst_object_unref (program->collection);
program->collection = collection;
/* Replace section and pmt with the new one */
gst_amlmpegts_section_unref (program->section);
program->section = gst_amlmpegts_section_ref (section);
program->pmt = pmt;
/* Copy over gststream that still exist into the collection */
for (tmp = program->stream_list; tmp; tmp = tmp->next) {
AmlTSBaseStream *stream = (AmlTSBaseStream *) tmp->data;
if (_aml_stream_in_pmt (pmt, stream) && stream->in_collection) {
gst_stream_collection_add_stream (program->collection,
gst_object_ref (stream->stream_object));
}
}
/* Add new streams (will also create and add gststream to the collection) */
nbstreams = pmt->streams->len;
for (i = 0; i < nbstreams; i++) {
GstAmlMpegtsPMTStream *stream = g_ptr_array_index (pmt->streams, i);
if (!_aml_pmt_stream_in_program (program, stream))
amlts_base_program_add_stream (base, program, stream->pid,
stream->stream_type, stream);
}
/* Call subclass update */
if (klass->update_program)
klass->update_program (base, program);
/* Remove streams no longer present */
toremove = NULL;
for (tmp = program->stream_list; tmp; tmp = tmp->next) {
AmlTSBaseStream *stream = (AmlTSBaseStream *) tmp->data;
if (!_aml_stream_in_pmt (pmt, stream))
toremove = g_list_prepend (toremove, stream);
}
for (tmp = toremove; tmp; tmp = tmp->next) {
AmlTSBaseStream *stream = (AmlTSBaseStream *) tmp->data;
amlts_base_program_remove_stream (base, program, stream->pid);
}
return TRUE;
}
static gboolean
_aml_stream_is_private_section (const GstAmlMpegtsPMT * pmt,
GstAmlMpegtsPMTStream * stream)
{
switch (stream->stream_type) {
case GST_MPEGTS_STREAM_TYPE_SCTE_DSMCC_DCB:
case GST_MPEGTS_STREAM_TYPE_SCTE_SIGNALING:
{
guint32 registration_id =
aml_get_registration_from_descriptors (stream->descriptors);
/* Not a private section stream */
if (registration_id != DRF_ID_CUEI && registration_id != DRF_ID_ETV1)
return FALSE;
}
case GST_MPEGTS_STREAM_TYPE_PRIVATE_SECTIONS:
case GST_MPEGTS_STREAM_TYPE_MHEG:
case GST_MPEGTS_STREAM_TYPE_DSM_CC:
case GST_MPEGTS_STREAM_TYPE_DSMCC_A:
case GST_MPEGTS_STREAM_TYPE_DSMCC_B:
case GST_MPEGTS_STREAM_TYPE_DSMCC_C:
case GST_MPEGTS_STREAM_TYPE_DSMCC_D:
case GST_MPEGTS_STREAM_TYPE_SL_FLEXMUX_SECTIONS:
case GST_MPEGTS_STREAM_TYPE_METADATA_SECTIONS:
/* known PSI streams */
return TRUE;
#if ((GST_VERSION_MAJOR == 1) && (GST_VERSION_MINOR >= 20))
// aml mark: GST_MPEGTS_STREAM_TYPE_SCTE_SIT exist since 1.20
case GST_MPEGTS_STREAM_TYPE_SCTE_SIT:
{
/* Not a private section stream */
if (!aml_find_registration_in_descriptors (pmt->descriptors, DRF_ID_CUEI))
return FALSE;
return TRUE;
}
#endif
default:
return FALSE;
}
}
/* Return TRUE if programs are equal */
static gboolean
amlts_base_is_same_program (AmlTSBase * base, AmlTSBaseProgram * oldprogram,
guint16 new_pmt_pid, const GstAmlMpegtsPMT * new_pmt)
{
guint i, nbstreams;
AmlTSBaseStream *oldstream;
gboolean sawpcrpid = FALSE;
if (oldprogram->pmt_pid != new_pmt_pid) {
GST_DEBUG ("Different pmt_pid (new:0x%04x, old:0x%04x)", new_pmt_pid,
oldprogram->pmt_pid);
return FALSE;
}
if (!base->ignore_pcr && oldprogram->pcr_pid != new_pmt->pcr_pid) {
GST_DEBUG ("Different pcr_pid (new:0x%04x, old:0x%04x)",
new_pmt->pcr_pid, oldprogram->pcr_pid);
return FALSE;
}
/* Check the streams */
nbstreams = new_pmt->streams->len;
for (i = 0; i < nbstreams; ++i) {
GstAmlMpegtsPMTStream *stream = g_ptr_array_index (new_pmt->streams, i);
oldstream = oldprogram->streams[stream->pid];
if (!oldstream) {
GST_DEBUG ("New stream 0x%04x not present in old program", stream->pid);
return FALSE;
}
if (oldstream->stream_type != stream->stream_type) {
GST_DEBUG
("New stream 0x%04x has a different stream type (new:%d, old:%d)",
stream->pid, stream->stream_type, oldstream->stream_type);
return FALSE;
}
if (stream->pid == oldprogram->pcr_pid)
sawpcrpid = TRUE;
}
/* If we have a PCR PID and the pcr is not shared with an existing stream, we'll have one extra stream */
if (!sawpcrpid && !base->ignore_pcr)
nbstreams += 1;
if (nbstreams != g_list_length (oldprogram->stream_list)) {
GST_DEBUG ("Different number of streams (new:%d, old:%d)",
nbstreams, g_list_length (oldprogram->stream_list));
return FALSE;
}
GST_DEBUG ("Programs are equal");
return TRUE;
}
/* Return TRUE if program is an update
*
* A program is equal if:
* * The program number is the same (will be if it enters this function)
* * AND The PMT PID is equal to the old one
* * AND It contains at least one stream from the previous program
*
* Changes that are acceptable are therefore:
* * New streams appearing
* * Old streams going away
* * PCR PID changing
*
* Unclear changes:
* * PMT PID being changed ?
* * Properties of elementary stream being changed ? (new tags ? metadata ?)
*/
static gboolean
amlts_base_is_program_update (AmlTSBase * base,
AmlTSBaseProgram * oldprogram, guint16 new_pmt_pid,
const GstAmlMpegtsPMT * new_pmt)
{
guint i, nbstreams;
AmlTSBaseStream *oldstream;
if (oldprogram->pmt_pid != new_pmt_pid) {
/* FIXME/CHECK: Can a program be updated by just changing its PID
* in the PAT ? */
GST_DEBUG ("Different pmt_pid (new:0x%04x, old:0x%04x)", new_pmt_pid,
oldprogram->pmt_pid);
return FALSE;
}
/* Check if at least one stream from the previous program is still present
* in the new program */
/* Check the streams */
nbstreams = new_pmt->streams->len;
for (i = 0; i < nbstreams; ++i) {
GstAmlMpegtsPMTStream *stream = g_ptr_array_index (new_pmt->streams, i);
oldstream = oldprogram->streams[stream->pid];
if (!oldstream) {
GST_DEBUG ("New stream 0x%04x not present in old program", stream->pid);
} else if (oldstream->stream_type != stream->stream_type) {
GST_DEBUG
("New stream 0x%04x has a different stream type (new:%d, old:%d)",
stream->pid, stream->stream_type, oldstream->stream_type);
} else if (!_aml_stream_is_private_section (new_pmt, stream)) {
/* FIXME : We should actually be checking a bit deeper,
* especially for private streams (where the differentiation is
* done at the registration level) */
GST_DEBUG
("Stream 0x%04x is identical (stream_type %d) ! Program is an update",
stream->pid, stream->stream_type);
return TRUE;
}
}
GST_DEBUG ("Program is not an update of the previous one");
return FALSE;
}
static void
amlts_base_deactivate_program (AmlTSBase * base, AmlTSBaseProgram * program)
{
gint i;
AmlTSBaseClass *klass = GST_AMLTS_BASE_GET_CLASS (base);
if (G_UNLIKELY (program->active == FALSE))
return;
GST_DEBUG_OBJECT (base, "Deactivating PMT");
program->active = FALSE;
if (program->pmt) {
for (i = 0; i < program->pmt->streams->len; ++i) {
GstAmlMpegtsPMTStream *stream = g_ptr_array_index (program->pmt->streams, i);
amlts_base_program_remove_stream (base, program, stream->pid);
/* Only unset the is_pes/known_psi bit if the PID isn't used in any other active
* program */
if (!amlts_pid_in_active_programs (base, stream->pid)) {
if (_aml_stream_is_private_section (program->pmt, stream)) {
if (base->parse_private_sections)
MPEGTS_BIT_UNSET (base->known_psi, stream->pid);
} else {
MPEGTS_BIT_UNSET (base->is_pes, stream->pid);
}
}
}
/* remove pcr stream */
/* FIXME : This might actually be shared with another stream ? */
amlts_base_program_remove_stream (base, program, program->pcr_pid);
if (!amlts_pid_in_active_programs (base, program->pcr_pid))
MPEGTS_BIT_UNSET (base->is_pes, program->pcr_pid);
GST_DEBUG ("program stream_list is now %p", program->stream_list);
}
/* Inform subclasses we're deactivating this program */
if (klass->program_stopped)
klass->program_stopped (base, program);
}
static void
amlts_base_activate_program (AmlTSBase * base, AmlTSBaseProgram * program,
guint16 pmt_pid, GstAmlMpegtsSection * section, const GstAmlMpegtsPMT * pmt,
gboolean initial_program)
{
guint i;
AmlTSBaseClass *klass;
if (G_UNLIKELY (program->active))
return;
GST_DEBUG ("Activating program %d", program->program_number);
/* activate new pmt */
if (program->section)
gst_amlmpegts_section_unref (program->section);
program->section = gst_amlmpegts_section_ref (section);
program->pmt = pmt;
program->pmt_pid = pmt_pid;
if (!base->ignore_pcr)
program->pcr_pid = pmt->pcr_pid;
else
program->pcr_pid = 0x1fff;
/* extract top-level registration_id if present */
program->registration_id =
aml_get_registration_from_descriptors (pmt->descriptors);
GST_DEBUG ("program 0x%04x, registration_id %" SAFE_FOURCC_FORMAT,
program->program_number, SAFE_FOURCC_ARGS (program->registration_id));
for (i = 0; i < pmt->streams->len; ++i) {
GstAmlMpegtsPMTStream *stream = g_ptr_array_index (pmt->streams, i);
if (_aml_stream_is_private_section (pmt, stream)) {
if (base->parse_private_sections)
MPEGTS_BIT_SET (base->known_psi, stream->pid);
} else {
if (G_UNLIKELY (MPEGTS_BIT_IS_SET (base->is_pes, stream->pid)))
GST_FIXME
("Refcounting issue. Setting twice a PID (0x%04x) as known PES",
stream->pid);
if (G_UNLIKELY (MPEGTS_BIT_IS_SET (base->known_psi, stream->pid))) {
GST_FIXME
("Refcounting issue. Setting a known PSI PID (0x%04x) as known PES",
stream->pid);
MPEGTS_BIT_UNSET (base->known_psi, stream->pid);
}
MPEGTS_BIT_SET (base->is_pes, stream->pid);
}
amlts_base_program_add_stream (base, program,
stream->pid, stream->stream_type, stream);
}
/* We add the PCR pid last. If that PID is already used by one of the media
* streams above, no new stream will be created */
amlts_base_program_add_stream (base, program, program->pcr_pid, -1, NULL);
MPEGTS_BIT_SET (base->is_pes, program->pcr_pid);
program->active = TRUE;
program->initial_program = initial_program;
klass = GST_AMLTS_BASE_GET_CLASS (base);
if (klass->program_started != NULL)
klass->program_started (base, program);
GST_DEBUG_OBJECT (base, "new pmt activated");
}
static gboolean
amlts_base_apply_pat (AmlTSBase * base, GstAmlMpegtsSection * section)
{
GPtrArray *pat = gst_amlmpegts_section_get_pat (section);
GPtrArray *old_pat;
AmlTSBaseProgram *program;
gint i;
if (G_UNLIKELY (pat == NULL))
return FALSE;
GST_INFO_OBJECT (base, "PAT");
/* Applying a new PAT does two things:
* * It adds the new programs to the list of programs this element handles
* and increments at the same time the number of times a program is referenced.
*
* * If there was a previously active PAT, It decrements the reference count
* of all program it used. If a program is no longer needed, it is removed.
*/
old_pat = base->pat;
base->pat = pat;
GST_LOG ("Activating new Program Association Table");
/* activate the new table */
for (i = 0; i < pat->len; ++i) {
GstAmlMpegtsPatProgram *patp = g_ptr_array_index (pat, i);
GST_LOG ("Looking for program %d / 0x%04x", patp->program_number,
patp->network_or_program_map_PID);
program = amlts_base_get_program (base, patp->program_number);
if (program) {
GST_LOG ("Program exists on pid 0x%04x", program->pmt_pid);
/* If the new PMT PID clashes with an existing known PES stream, we know
* it is not an update */
if (MPEGTS_BIT_IS_SET (base->is_pes, patp->network_or_program_map_PID)) {
GST_LOG ("Program is not an update");
program =
amlts_base_add_program (base, patp->program_number,
patp->network_or_program_map_PID);
} else if (program->pmt_pid != patp->network_or_program_map_PID) {
/* IF the program already existed, just check if the PMT PID changed */
GST_LOG ("PMT is on a different PID");
if (program->pmt_pid != G_MAXUINT16) {
/* pmt pid changed */
/* FIXME: when this happens it may still be pmt pid of another
* program, so setting to False may make it go through expensive
* path in is_psi unnecessarily */
MPEGTS_BIT_UNSET (base->known_psi, program->pmt_pid);
}
program->pmt_pid = patp->network_or_program_map_PID;
if (G_UNLIKELY (MPEGTS_BIT_IS_SET (base->known_psi, program->pmt_pid)))
GST_FIXME
("Refcounting issue. Setting twice a PMT PID (0x%04x) as know PSI",
program->pmt_pid);
MPEGTS_BIT_SET (base->known_psi, patp->network_or_program_map_PID);
} else {
GST_LOG ("Regular program update");
}
} else {
/* Create a new program */
program =
amlts_base_add_program (base, patp->program_number,
patp->network_or_program_map_PID);
}
/* We mark this program as being referenced by one PAT */
program->patcount += 1;
}
if (old_pat) {
AmlTSBaseClass *klass = GST_AMLTS_BASE_GET_CLASS (base);
/* deactivate the old table */
GST_LOG ("Deactivating old Program Association Table");
for (i = 0; i < old_pat->len; ++i) {
GstAmlMpegtsPatProgram *patp = g_ptr_array_index (old_pat, i);
program = amlts_base_get_program (base, patp->program_number);
if (G_UNLIKELY (program == NULL)) {
GST_DEBUG_OBJECT (base, "broken PAT, duplicated entry for program %d",
patp->program_number);
continue;
}
GST_LOG ("Deactivating program %d / 0x%04x", patp->program_number,
patp->network_or_program_map_PID);
if (--program->patcount > 0) {
GST_LOG ("Referenced by new program, keeping");
/* the program has been referenced by the new pat, keep it */
continue;
}
GST_INFO_OBJECT (base, "PAT removing program 0x%04x 0x%04x",
patp->program_number, patp->network_or_program_map_PID);
if (klass->can_remove_program (base, program)) {
amlts_base_deactivate_program (base, program);
amlts_base_remove_program (base, program);
} else {
/* sub-class now owns the program and must call
* amlts_base_deactivate_and_free_program later */
amlts_base_steal_program (base, patp->program_number);
}
/* FIXME: when this happens it may still be pmt pid of another
* program, so setting to False may make it go through expensive
* path in is_psi unnecessarily */
if (G_UNLIKELY (MPEGTS_BIT_IS_SET (base->known_psi,
patp->network_or_program_map_PID))) {
GST_FIXME
("Program refcounting : Setting twice a pid (0x%04x) as known PSI",
patp->network_or_program_map_PID);
}
MPEGTS_BIT_SET (base->known_psi, patp->network_or_program_map_PID);
amlts_packetizer_remove_stream (base->packetizer,
patp->network_or_program_map_PID);
}
g_ptr_array_unref (old_pat);
}
return TRUE;
}
static gboolean
amlts_base_apply_pmt (AmlTSBase * base, GstAmlMpegtsSection * section)
{
const GstAmlMpegtsPMT *pmt;
AmlTSBaseProgram *program, *old_program;
guint program_number;
gboolean initial_program = TRUE;
pmt = gst_amlmpegts_section_get_pmt (section);
if (G_UNLIKELY (pmt == NULL)) {
GST_ERROR ("Could not get PMT (corrupted ?)");
return FALSE;
}
/* FIXME : not so sure this is valid anymore */
if (G_UNLIKELY (base->seen_pat == FALSE)) {
GST_WARNING ("Got pmt without pat first. Returning");
/* remove the stream since we won't get another PMT otherwise */
amlts_packetizer_remove_stream (base->packetizer, section->pid);
return TRUE;
}
/* Don't attempt to handle pmt without any streams */
if (G_UNLIKELY (pmt->streams->len == 0)) {
GST_WARNING ("Skipping PMT without any entries");
return TRUE;
}
program_number = section->subtable_extension;
GST_DEBUG ("Applying PMT (program_number:%d, pid:0x%04x)",
program_number, section->pid);
/* In order for stream switching to happen properly in decodebin(2),
* we need to first add the new pads (i.e. activate the new program)
* before removing the old ones (i.e. deactivating the old program)
*/
old_program = amlts_base_get_program (base, program_number);
if (G_UNLIKELY (old_program == NULL))
goto no_program;
if (G_UNLIKELY (amlts_base_is_same_program (base, old_program, section->pid,
pmt)))
goto same_program;
if (base->streams_aware
&& amlts_base_is_program_update (base, old_program, section->pid, pmt)) {
GST_FIXME ("We are streams_aware and new program is an update");
/* The program is an update, and we can add/remove pads dynamically */
amlts_base_update_program (base, old_program, section, pmt);
goto beach;
}
/* If the current program is active, this means we have a new program */
if (old_program->active) {
AmlTSBaseClass *klass = GST_AMLTS_BASE_GET_CLASS (base);
old_program = amlts_base_steal_program (base, program_number);
program = amlts_base_new_program (base, program_number, section->pid);
program->patcount = old_program->patcount;
/* Deactivate the old program */
/* FIXME : THIS IS BREAKING THE STREAM SWITCHING LOGIC !
* */
if (klass->can_remove_program (base, old_program)) {
amlts_base_deactivate_program (base, old_program);
amlts_base_free_program (old_program);
} else {
/* sub-class now owns the program and must call
* amlts_base_deactivate_and_free_program later */
amlts_base_steal_program (base, old_program->program_number);
}
/* Add new program to the programs we track */
g_ptr_array_add (base->programs, program);
initial_program = FALSE;
} else {
GST_DEBUG ("Program update, re-using same program");
program = old_program;
}
/* activate program */
/* Ownership of pmt_info is given to the program */
amlts_base_activate_program (base, program, section->pid, section, pmt,
initial_program);
beach:
GST_DEBUG ("Done activating program");
return TRUE;
no_program:
{
GST_ERROR ("Attempted to apply a PMT on a program that wasn't created");
return TRUE;
}
same_program:
{
GST_DEBUG ("Not applying identical program");
return TRUE;
}
}
static void
amlts_base_handle_psi (AmlTSBase * base, GstAmlMpegtsSection * section)
{
gboolean post_message = TRUE;
GST_DEBUG ("Handling PSI (pid: 0x%04x , table_id: 0x%02x)",
section->pid, section->table_id);
switch (section->section_type) {
case GST_MPEGTS_SECTION_PAT:
post_message = amlts_base_apply_pat (base, section);
if (base->seen_pat == FALSE) {
base->seen_pat = TRUE;
GST_DEBUG ("First PAT offset: %" G_GUINT64_FORMAT, section->offset);
amlts_packetizer_set_reference_offset (base->packetizer,
section->offset);
}
break;
case GST_MPEGTS_SECTION_PMT:
post_message = amlts_base_apply_pmt (base, section);
break;
case GST_MPEGTS_SECTION_EIT:
/* some tag xtraction + posting */
post_message = amlts_base_get_tags_from_eit (base, section);
break;
case GST_MPEGTS_SECTION_ATSC_MGT:
post_message = amlts_base_parse_atsc_mgt (base, section);
break;
default:
break;
}
/* Give the subclass a chance to look at the section */
if (GST_AMLTS_BASE_GET_CLASS (base)->handle_psi)
GST_AMLTS_BASE_GET_CLASS (base)->handle_psi (base, section);
/* Finally post message (if it wasn't corrupted) */
if (post_message)
gst_element_post_message (GST_ELEMENT_CAST (base),
gst_message_new_amlmpegts_section (GST_OBJECT (base), section));
gst_amlmpegts_section_unref (section);
}
static gboolean
amlts_base_parse_atsc_mgt (AmlTSBase * base, GstAmlMpegtsSection * section)
{
const GstAmlMpegtsAtscMGT *mgt;
gint i;
mgt = gst_amlmpegts_section_get_atsc_mgt (section);
if (G_UNLIKELY (mgt == NULL))
return FALSE;
for (i = 0; i < mgt->tables->len; ++i) {
GstAmlMpegtsAtscMGTTable *table = g_ptr_array_index (mgt->tables, i);
if ((table->table_type >= GST_MPEGTS_ATSC_MGT_TABLE_TYPE_EIT0 &&
table->table_type <= GST_MPEGTS_ATSC_MGT_TABLE_TYPE_EIT127) ||
(table->table_type >= GST_MPEGTS_ATSC_MGT_TABLE_TYPE_ETT0 &&
table->table_type <= GST_MPEGTS_ATSC_MGT_TABLE_TYPE_ETT127)) {
MPEGTS_BIT_SET (base->known_psi, table->pid);
}
}
return TRUE;
}
static gboolean
amlts_base_get_tags_from_eit (AmlTSBase * base, GstAmlMpegtsSection * section)
{
const GstAmlMpegtsEIT *eit;
guint i;
AmlTSBaseProgram *program;
/* Early exit if it's not from the present/following table_id */
if (section->table_id != GST_MTS_TABLE_ID_EVENT_INFORMATION_ACTUAL_TS_PRESENT
&& section->table_id !=
GST_MTS_TABLE_ID_EVENT_INFORMATION_OTHER_TS_PRESENT)
return TRUE;
eit = gst_amlmpegts_section_get_eit (section);
if (G_UNLIKELY (eit == NULL))
return FALSE;
program = amlts_base_get_program (base, section->subtable_extension);
GST_DEBUG
("program_id:0x%04x, table_id:0x%02x, actual_stream:%d, present_following:%d, program:%p",
section->subtable_extension, section->table_id, eit->actual_stream,
eit->present_following, program);
if (program && eit->present_following) {
for (i = 0; i < eit->events->len; i++) {
GstAmlMpegtsEITEvent *event = g_ptr_array_index (eit->events, i);
const GstAmlMpegtsDescriptor *desc;
if (event->running_status == RUNNING_STATUS_RUNNING) {
program->event_id = event->event_id;
if ((desc =
gst_amlmpegts_find_descriptor (event->descriptors,
GST_MTS_DESC_DVB_SHORT_EVENT))) {
gchar *name = NULL, *text = NULL;
if (gst_amlmpegts_descriptor_parse_dvb_short_event (desc, NULL, &name,
&text)) {
if (!program->tags)
program->tags = gst_tag_list_new_empty ();
if (name) {
gst_tag_list_add (program->tags, GST_TAG_MERGE_APPEND,
GST_TAG_TITLE, name, NULL);
g_free (name);
}
if (text) {
gst_tag_list_add (program->tags, GST_TAG_MERGE_APPEND,
GST_TAG_DESCRIPTION, text, NULL);
g_free (text);
}
/* FIXME : Is it correct to post an event duration as a GST_TAG_DURATION ??? */
gst_tag_list_add (program->tags, GST_TAG_MERGE_APPEND,
GST_TAG_DURATION, event->duration * GST_SECOND, NULL);
return TRUE;
}
}
}
}
}
return TRUE;
}
static void
aml_remove_each_program (AmlTSBaseProgram * program, AmlTSBase * base)
{
/* First deactivate it */
amlts_base_deactivate_program (base, program);
}
static inline GstFlowReturn
amlts_base_drain (AmlTSBase * base)
{
AmlTSBaseClass *klass = GST_AMLTS_BASE_GET_CLASS (base);
/* Call implementation */
if (klass->drain)
return klass->drain (base);
return GST_FLOW_OK;
}
static inline void
amlts_base_flush (AmlTSBase * base, gboolean hard)
{
AmlTSBaseClass *klass = GST_AMLTS_BASE_GET_CLASS (base);
/* Call implementation */
if (klass->flush)
klass->flush (base, hard);
}
static gboolean
amlts_base_sink_event (GstPad * pad, GstObject * parent, GstEvent * event)
{
gboolean res = TRUE;
gboolean hard;
AmlTSBase *base = GST_AMLTS_BASE (parent);
gboolean is_sticky = GST_EVENT_IS_STICKY (event);
GST_DEBUG_OBJECT (base, "Got event %s",
gst_event_type_get_name (GST_EVENT_TYPE (event)));
switch (GST_EVENT_TYPE (event)) {
case GST_EVENT_SEGMENT:
gst_event_copy_segment (event, &base->segment);
GST_DEBUG_OBJECT (base, "Received segment %" GST_SEGMENT_FORMAT,
&base->segment);
/* Check if we need to switch PCR/PTS handling */
if (base->segment.format == GST_FORMAT_TIME) {
base->packetizer->calculate_offset = FALSE;
base->packetizer->calculate_skew = TRUE;
/* Seek was handled upstream */
base->last_seek_seqnum = gst_event_get_seqnum (event);
} else {
base->packetizer->calculate_offset = TRUE;
base->packetizer->calculate_skew = FALSE;
}
res = GST_AMLTS_BASE_GET_CLASS (base)->push_event (base, event);
break;
case GST_EVENT_STREAM_START:
gst_event_unref (event);
break;
case GST_EVENT_CAPS:
/* FIXME, do something */
gst_event_unref (event);
break;
case GST_EVENT_FLUSH_STOP:
res = GST_AMLTS_BASE_GET_CLASS (base)->push_event (base, event);
hard = (base->mode != BASE_MODE_SEEKING);
amlts_packetizer_flush (base->packetizer, hard);
amlts_base_flush (base, hard);
gst_segment_init (&base->segment, GST_FORMAT_UNDEFINED);
base->seen_pat = FALSE;
break;
default:
res = GST_AMLTS_BASE_GET_CLASS (base)->push_event (base, event);
}
/* Always return TRUE for sticky events */
if (is_sticky)
res = TRUE;
return res;
}
static gboolean
amlts_base_default_sink_query (AmlTSBase * base, GstQuery * query)
{
return gst_pad_query_default (base->sinkpad, GST_OBJECT (base), query);
}
static gboolean
amlts_base_sink_query (GstPad * pad, GstObject * parent, GstQuery * query)
{
AmlTSBase *base = GST_AMLTS_BASE (parent);
GST_DEBUG_OBJECT (base, "Got query %s",
gst_query_type_get_name (GST_QUERY_TYPE (query)));
return GST_AMLTS_BASE_GET_CLASS (base)->sink_query (base, query);
}
static GstFlowReturn
amlts_base_chain (GstPad * pad, GstObject * parent, GstBuffer * buf)
{
GstFlowReturn res = GST_FLOW_OK;
AmlTSBase *base;
AmlTSPacketizerPacketReturn pret;
AmlTSPacketizer2 *packetizer;
AmlTSPacketizerPacket packet = {0};
AmlTSBaseClass *klass;
base = GST_AMLTS_BASE (parent);
klass = GST_AMLTS_BASE_GET_CLASS (base);
packetizer = base->packetizer;
if (GST_BUFFER_IS_DISCONT (buf)) {
GST_DEBUG_OBJECT (base, "Got DISCONT buffer, flushing");
res = amlts_base_drain (base);
if (G_UNLIKELY (res != GST_FLOW_OK))
return res;
amlts_base_flush (base, FALSE);
if (base->mode == BASE_MODE_PUSHING) {
if (base->segment.format == GST_FORMAT_TIME) {
/* In the case of discontinuities in push-mode with TIME segment
* we want to drop all previous observations (hard:TRUE) from
* the packetizer */
amlts_packetizer_flush (base->packetizer, TRUE);
}
/* In all cases, we clear observations when we get a discontinuity in
* push-mode to re-check if the sections (PAT/PMT) changed or not */
amlts_packetizer_clear (base->packetizer);
} else {
amlts_packetizer_flush (base->packetizer, FALSE);
}
}
amlts_packetizer_push (base->packetizer, buf);
if (!base->av_started) {
GstBuffer *buf_new = gst_buffer_copy_deep(buf);
if (!base->probe_buflist)
base->probe_buflist = gst_buffer_list_new();
if (buf_new && base->probe_buflist) {
GST_DEBUG_OBJECT (base, "add buflist buf_new=%p, buf=%p", buf_new, buf);
gst_buffer_list_add(base->probe_buflist, buf_new);
gst_buffer_ref(buf_new);
}
}
while (res == GST_FLOW_OK) {
pret = amlts_packetizer_next_packet (base->packetizer, &packet);
/* If we don't have enough data, return */
if (G_UNLIKELY (pret == PACKET_NEED_MORE))
break;
if (G_UNLIKELY (pret == PACKET_BAD)) {
/* bad header, skip the packet */
GST_DEBUG_OBJECT (base, "bad packet, skipping PID 0x%04x", packet.pid);
goto next;
}
if (klass->inspect_packet)
klass->inspect_packet (base, &packet);
/* If it's a known PES, push it */
if (MPEGTS_BIT_IS_SET (base->is_pes, packet.pid)) {
/* push the packet downstream */
if (base->push_data)
res = klass->push (base, &packet, NULL);
} else if (packet.payload
&& MPEGTS_BIT_IS_SET (base->known_psi, packet.pid)) {
/* base PSI data */
GList *others, *tmp;
GstAmlMpegtsSection *section;
section = amlts_packetizer_push_section (packetizer, &packet, &others);
if (section)
amlts_base_handle_psi (base, section);
if (G_UNLIKELY (others)) {
for (tmp = others; tmp; tmp = tmp->next)
amlts_base_handle_psi (base, (GstAmlMpegtsSection *) tmp->data);
g_list_free (others);
}
/* we need to push section packet downstream */
if (base->push_section)
res = klass->push (base, &packet, section);
} else if (base->push_unknown) {
res = klass->push (base, &packet, NULL);
} else if (packet.payload && packet.pid != 0x1fff)
GST_LOG ("PID 0x%04x Saw packet on a pid we don't handle", packet.pid);
if (base->av_started && !base->push_scandata && base->probe_buflist)
{
guint buffer_count;
amlts_packetizer_flush (base->packetizer, FALSE);
buffer_count = gst_buffer_list_length(base->probe_buflist);
GST_DEBUG_OBJECT (base, "re send %d buffers", buffer_count);
for (guint i = 0; i < buffer_count; i++) {
GstBuffer *buffer = gst_buffer_list_get(base->probe_buflist, i);
if (buffer != NULL) {
amlts_packetizer_push (base->packetizer, buffer);
}
}
gst_clear_buffer_list(&base->probe_buflist);
base->push_scandata = TRUE;
}
next:
amlts_packetizer_clear_packet (base->packetizer, &packet);
}
if (res == GST_FLOW_OK && klass->input_done)
res = klass->input_done (base);
return res;
}
static GstFlowReturn
amlts_base_scan (AmlTSBase * base)
{
GstFlowReturn ret = GST_FLOW_OK;
GstBuffer *buf = NULL;
guint i;
gboolean done = FALSE;
AmlTSPacketizerPacketReturn pret;
gint64 tmpval;
gint64 upstream_size, seek_pos, reverse_limit;
GstFormat format;
guint initial_pcr_seen;
GST_DEBUG ("Scanning for initial sync point");
/* Find initial sync point and at least 5 PCR values */
for (i = 0; i < 20 && !done; i++) {
GST_DEBUG ("Grabbing %d => %d", i * 65536, (i + 1) * 65536);
ret = gst_pad_pull_range (base->sinkpad, i * 65536, 65536, &buf);
if (G_UNLIKELY (ret == GST_FLOW_EOS))
break;
if (G_UNLIKELY (ret != GST_FLOW_OK))
goto beach;
/* Push to packetizer */
amlts_packetizer_push (base->packetizer, buf);
buf = NULL;
if (amlts_packetizer_has_packets (base->packetizer)) {
if (base->seek_offset == -1) {
/* Mark the initial sync point and remember the packetsize */
base->seek_offset = base->packetizer->offset;
GST_DEBUG ("Sync point is now %" G_GUINT64_FORMAT, base->seek_offset);
base->packetsize = base->packetizer->packet_size;
}
while (1) {
/* Eat up all packets */
pret = amlts_packetizer_process_next_packet (base->packetizer);
if (pret == PACKET_NEED_MORE)
break;
if (pret != PACKET_BAD && base->packetizer->nb_seen_offsets >= 5) {
GST_DEBUG ("Got enough initial PCR");
done = TRUE;
break;
}
}
}
}
initial_pcr_seen = base->packetizer->nb_seen_offsets;
if (G_UNLIKELY (initial_pcr_seen == 0))
goto no_initial_pcr;
GST_DEBUG ("Seen %d initial PCR", initial_pcr_seen);
/* Now send data from the end */
/* Get the size of upstream */
format = GST_FORMAT_BYTES;
if (!gst_pad_peer_query_duration (base->sinkpad, format, &tmpval))
goto beach;
upstream_size = tmpval;
/* The scanning takes place on the last 2048kB. Considering PCR should
* be present at least every 100ms, this should cope with streams
* up to 160Mbit/s */
reverse_limit = MAX (0, upstream_size - 2097152);
/* Find last PCR value, searching backwards by chunks of 300 MPEG-ts packets */
for (seek_pos = MAX (0, upstream_size - 56400);
seek_pos >= reverse_limit; seek_pos -= 56400) {
amlts_packetizer_clear (base->packetizer);
GST_DEBUG ("Grabbing %" G_GUINT64_FORMAT " => %" G_GUINT64_FORMAT, seek_pos,
seek_pos + 56400);
ret = gst_pad_pull_range (base->sinkpad, seek_pos, 56400, &buf);
if (G_UNLIKELY (ret == GST_FLOW_EOS))
break;
if (G_UNLIKELY (ret != GST_FLOW_OK))
goto beach;
/* Push to packetizer */
amlts_packetizer_push (base->packetizer, buf);
buf = NULL;
if (amlts_packetizer_has_packets (base->packetizer)) {
pret = PACKET_OK;
/* Eat up all packets, really try to get last PCR(s) */
while (pret != PACKET_NEED_MORE)
pret = amlts_packetizer_process_next_packet (base->packetizer);
if (base->packetizer->nb_seen_offsets > initial_pcr_seen) {
GST_DEBUG ("Got last PCR(s) (total seen:%d)",
base->packetizer->nb_seen_offsets);
break;
}
}
}
beach:
amlts_packetizer_clear (base->packetizer);
return ret;
no_initial_pcr:
amlts_packetizer_clear (base->packetizer);
GST_WARNING_OBJECT (base, "Couldn't find any PCR within the first %d bytes",
10 * 65536);
return GST_FLOW_OK;
}
static void
amlts_base_loop (AmlTSBase * base)
{
GstFlowReturn ret = GST_FLOW_ERROR;
switch (base->mode) {
case BASE_MODE_SCANNING:
/* Find first sync point */
ret = amlts_base_scan (base);
if (G_UNLIKELY (ret != GST_FLOW_OK))
goto error;
base->mode = BASE_MODE_STREAMING;
GST_DEBUG ("Changing to Streaming");
break;
case BASE_MODE_SEEKING:
/* FIXME : unclear if we still need mode_seeking... */
base->mode = BASE_MODE_STREAMING;
break;
case BASE_MODE_STREAMING:
{
GstBuffer *buf = NULL;
GST_DEBUG ("Pulling data from %" G_GUINT64_FORMAT, base->seek_offset);
if (G_UNLIKELY (base->last_seek_seqnum == GST_SEQNUM_INVALID)) {
/* No configured seek, set a valid seqnum */
base->last_seek_seqnum = gst_util_seqnum_next ();
}
ret = gst_pad_pull_range (base->sinkpad, base->seek_offset,
100 * base->packetsize, &buf);
if (G_UNLIKELY (ret != GST_FLOW_OK))
goto error;
base->seek_offset += gst_buffer_get_size (buf);
ret = amlts_base_chain (base->sinkpad, GST_OBJECT_CAST (base), buf);
if (G_UNLIKELY (ret != GST_FLOW_OK))
goto error;
}
break;
case BASE_MODE_PUSHING:
GST_WARNING ("wrong BASE_MODE_PUSHING mode in pull loop");
break;
}
return;
error:
{
GST_DEBUG_OBJECT (base, "Pausing task, reason %s", gst_flow_get_name (ret));
if (ret == GST_FLOW_EOS) {
if (!GST_AMLTS_BASE_GET_CLASS (base)->push_event (base,
gst_event_new_eos ()))
GST_ELEMENT_ERROR (base, STREAM, FAILED,
(_("Internal data stream error.")),
("No program activated before EOS"));
} else if (ret == GST_FLOW_NOT_LINKED || ret < GST_FLOW_EOS) {
GST_ELEMENT_FLOW_ERROR (base, ret);
GST_AMLTS_BASE_GET_CLASS (base)->push_event (base, gst_event_new_eos ());
}
gst_pad_pause_task (base->sinkpad);
}
}
gboolean
amlts_base_handle_seek_event (AmlTSBase * base, GstPad * pad,
GstEvent * event)
{
AmlTSBaseClass *klass = GST_AMLTS_BASE_GET_CLASS (base);
GstFlowReturn ret = GST_FLOW_ERROR;
gdouble rate;
gboolean flush, instant_rate_change;
GstFormat format;
GstSeekFlags flags;
GstSeekType start_type, stop_type;
gint64 start, stop;
GstEvent *flush_event = NULL;
gst_event_parse_seek (event, &rate, &format, &flags, &start_type, &start,
&stop_type, &stop);
if (format != GST_FORMAT_TIME)
return FALSE;
if (GST_EVENT_SEQNUM (event) == base->last_seek_seqnum) {
GST_DEBUG_OBJECT (base, "Skipping already handled seek");
return TRUE;
}
if (base->mode == BASE_MODE_PUSHING) {
/* First try if upstream supports seeking in TIME format */
if (gst_pad_push_event (base->sinkpad, gst_event_ref (event))) {
GST_DEBUG ("upstream handled SEEK event");
return TRUE;
}
/* If the subclass can seek, do that */
if (klass->seek) {
ret = klass->seek (base, event);
if (G_UNLIKELY (ret != GST_FLOW_OK))
GST_WARNING ("seeking failed %s", gst_flow_get_name (ret));
else {
GstEvent *new_seek;
if (GST_CLOCK_TIME_IS_VALID (base->seek_offset)) {
base->mode = BASE_MODE_SEEKING;
new_seek = gst_event_new_seek (rate, GST_FORMAT_BYTES, flags,
GST_SEEK_TYPE_SET, base->seek_offset, GST_SEEK_TYPE_NONE, -1);
gst_event_set_seqnum (new_seek, GST_EVENT_SEQNUM (event));
if (!gst_pad_push_event (base->sinkpad, new_seek))
ret = GST_FLOW_ERROR;
else
base->last_seek_seqnum = GST_EVENT_SEQNUM (event);
}
base->mode = BASE_MODE_PUSHING;
}
} else {
GST_WARNING ("subclass has no seek implementation");
}
return ret == GST_FLOW_OK;
}
if (!klass->seek) {
GST_WARNING ("subclass has no seek implementation");
return FALSE;
}
if (rate <= 0.0) {
GST_WARNING ("Negative rate not supported");
return FALSE;
}
GST_DEBUG ("seek event, rate: %f start: %" GST_TIME_FORMAT
" stop: %" GST_TIME_FORMAT, rate, GST_TIME_ARGS (start),
GST_TIME_ARGS (stop));
flush = ! !(flags & GST_SEEK_FLAG_FLUSH);
//aml mark: since 1.20
#if ((GST_VERSION_MAJOR == 1) && (GST_VERSION_MINOR >= 20))
instant_rate_change = ! !(flags & GST_SEEK_FLAG_INSTANT_RATE_CHANGE);
/* Directly send the instant-rate-change event here before taking the
* stream-lock so that it can be applied as soon as possible */
if (base->mode != BASE_MODE_PUSHING && instant_rate_change) {
GstEvent *ev;
/* instant rate change only supported if direction does not change. All
* other requirements are already checked before creating the seek event
* but let's double-check here to be sure */
if ((rate > 0 && base->out_segment.rate < 0) ||
(rate < 0 && base->out_segment.rate > 0) ||
start_type != GST_SEEK_TYPE_NONE ||
stop_type != GST_SEEK_TYPE_NONE || flush) {
GST_ERROR_OBJECT (base,
"Instant rate change seeks only supported in the "
"same direction, without flushing and position change");
return FALSE;
}
ev = gst_event_new_instant_rate_change (rate / base->out_segment.rate,
(GstSegmentFlags) (flags));
gst_event_set_seqnum (ev, GST_EVENT_SEQNUM (event));
GST_AMLTS_BASE_GET_CLASS (base)->push_event (base, ev);
return TRUE;
}
#endif
/* stop streaming, either by flushing or by pausing the task */
base->mode = BASE_MODE_SEEKING;
if (flush) {
GST_DEBUG_OBJECT (base, "sending flush start");
flush_event = gst_event_new_flush_start ();
gst_event_set_seqnum (flush_event, GST_EVENT_SEQNUM (event));
gst_pad_push_event (base->sinkpad, gst_event_ref (flush_event));
GST_AMLTS_BASE_GET_CLASS (base)->push_event (base, flush_event);
} else
gst_pad_pause_task (base->sinkpad);
/* wait for streaming to finish */
GST_PAD_STREAM_LOCK (base->sinkpad);
if (flush) {
/* send a FLUSH_STOP for the sinkpad, since we need data for seeking */
GST_DEBUG_OBJECT (base, "sending flush stop");
flush_event = gst_event_new_flush_stop (TRUE);
gst_event_set_seqnum (flush_event, GST_EVENT_SEQNUM (event));
/* ref for it to be reused later */
gst_pad_push_event (base->sinkpad, gst_event_ref (flush_event));
/* And actually flush our pending data but allow to preserve some info
* to perform the seek */
amlts_base_flush (base, FALSE);
amlts_packetizer_flush (base->packetizer, FALSE);
}
if (flags & (GST_SEEK_FLAG_SEGMENT)) {
GST_WARNING ("seek flags 0x%x are not supported", (int) flags);
goto done;
}
if (flags & GST_SEEK_FLAG_TRICKMODE_NO_AUDIO)
base->no_audio_set = TRUE;
else
base->no_audio_set = FALSE;
/* If the subclass can seek, do that */
ret = klass->seek (base, event);
if (G_UNLIKELY (ret != GST_FLOW_OK))
GST_WARNING ("seeking failed %s", gst_flow_get_name (ret));
else
base->last_seek_seqnum = GST_EVENT_SEQNUM (event);
if (flush_event) {
/* if we sent a FLUSH_START, we now send a FLUSH_STOP */
GST_DEBUG_OBJECT (base, "sending flush stop");
GST_AMLTS_BASE_GET_CLASS (base)->push_event (base, flush_event);
flush_event = NULL;
}
done:
if (flush_event)
gst_event_unref (flush_event);
gst_pad_start_task (base->sinkpad, (GstTaskFunction) amlts_base_loop, base,
NULL);
GST_PAD_STREAM_UNLOCK (base->sinkpad);
return ret == GST_FLOW_OK;
}
static gboolean
amlts_base_sink_activate (GstPad * sinkpad, GstObject * parent)
{
GstQuery *query;
gboolean pull_mode;
query = gst_query_new_scheduling ();
if (!gst_pad_peer_query (sinkpad, query)) {
gst_query_unref (query);
goto activate_push;
}
pull_mode = gst_query_has_scheduling_mode_with_flags (query,
GST_PAD_MODE_PULL, GST_SCHEDULING_FLAG_SEEKABLE);
gst_query_unref (query);
if (!pull_mode)
goto activate_push;
GST_DEBUG_OBJECT (sinkpad, "activating pull");
return gst_pad_activate_mode (sinkpad, GST_PAD_MODE_PULL, TRUE);
activate_push:
{
GST_DEBUG_OBJECT (sinkpad, "activating push");
return gst_pad_activate_mode (sinkpad, GST_PAD_MODE_PUSH, TRUE);
}
}
static gboolean
amlts_base_sink_activate_mode (GstPad * pad, GstObject * parent,
GstPadMode mode, gboolean active)
{
gboolean res;
AmlTSBase *base = GST_AMLTS_BASE (parent);
switch (mode) {
case GST_PAD_MODE_PUSH:
base->mode = BASE_MODE_PUSHING;
res = TRUE;
break;
case GST_PAD_MODE_PULL:
if (active) {
base->mode = BASE_MODE_SCANNING;
/* When working pull-based, we always use offsets for estimation */
base->packetizer->calculate_offset = TRUE;
base->packetizer->calculate_skew = FALSE;
gst_segment_init (&base->segment, GST_FORMAT_BYTES);
res =
gst_pad_start_task (pad, (GstTaskFunction) amlts_base_loop, base,
NULL);
} else
res = gst_pad_stop_task (pad);
break;
default:
res = FALSE;
break;
}
return res;
}
static GstStateChangeReturn
amlts_base_change_state (GstElement * element, GstStateChange transition)
{
AmlTSBase *base;
GstStateChangeReturn ret;
base = GST_AMLTS_BASE (element);
switch (transition) {
case GST_STATE_CHANGE_READY_TO_PAUSED:
amlts_base_reset (base);
break;
default:
break;
}
ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
switch (transition) {
case GST_STATE_CHANGE_PAUSED_TO_READY:
amlts_base_reset (base);
if (base->mode != BASE_MODE_PUSHING)
base->mode = BASE_MODE_SCANNING;
break;
default:
break;
}
return ret;
}
gboolean
gst_amltsbase_plugin_init (GstPlugin * plugin)
{
GST_DEBUG_CATEGORY_INIT (amlts_base_debug, "amltsbase", 0,
"MPEG transport stream base class");
return TRUE;
}
Reference in New Issue View Git Blame Copy Permalink