mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2024-07-19 04:48:22 +00:00
Code Issues 1 Projects Releases Wiki Activity
gstreamer/gst/dvbsuboverlay/dvb-sub.c
Mart Raudsepp 7e2e52cab8 dvbsuboverlay: Pass page_time_out and pts with the DVBSubtitles structure 
Eases holding onto the information in gst plugins side queue of
DVBSubtitles, so we won't need to create yet another temporary struct
to keep the pts and page_time_out too.
And this really logically belongs at the toplevel information set anyway
and in that struct...
2010-12-08 16:30:09 +01:00

1559 lines
50 KiB
C
Raw Blame History

/* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 4; tab-width: 4 -*- */
/*
* libdvbsub - DVB subtitle decoding
* Copyright (C) Mart Raudsepp 2009 <mart.raudsepp@artecdesign.ee>
*
* Heavily uses code algorithms ported from ffmpeg's libavcodec/dvbsubdec.c,
* especially the segment parsers. The original license applies to this
* ported code and the whole code in this file as well.
*
* Original copyright information follows:
*/
/*
* DVB subtitle decoding for ffmpeg
* Copyright (c) 2005 Ian Caulfield
*
* This file is part of FFmpeg.
*
* FFmpeg is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* FFmpeg 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with FFmpeg; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
#include "dvb-sub.h"
#include <string.h> /* memset */
#include <gst/gstutils.h> /* GST_READ_UINT16_BE */
#include <gst/base/gstbitreader.h> /* GstBitReader */
#include "ffmpeg-colorspace.h" /* YUV_TO_RGB1_CCIR */ /* FIXME: Just give YUV data to gstreamer then? */
/* FIXME: Convert to GST_LOG and clean up */
void (*g_log_callback) (GLogLevelFlags log_level, const gchar * format,
va_list args, gpointer user_data) = NULL;
gpointer g_log_callback_user_data = NULL;
#define DEBUG
#ifdef DEBUG
#define dvb_log(log_type, log_level, format...) real_dvb_log(log_type, log_level, ## format)
typedef enum
{
/* dvb_log types // DVB_LOG environment variable string */
DVB_LOG_GENERAL, /* GENERAL */
DVB_LOG_PAGE, /* PAGE */
DVB_LOG_REGION, /* REGION */
DVB_LOG_CLUT, /* CLUT */
DVB_LOG_OBJECT, /* OBJECT */
DVB_LOG_PIXEL, /* PIXEL */
DVB_LOG_RUNLEN, /* RUNLEN */
DVB_LOG_DISPLAY, /* DISPLAY */
DVB_LOG_STREAM, /* STREAM - issues in the encoded stream (TV service provider encoder problem) */
DVB_LOG_PACKET, /* PACKET - messages during raw demuxer data packet handling */
DVB_LOG_LAST /* sentinel use only */
} DvbLogTypes;
static void
real_dvb_log (const gint log_type, GLogLevelFlags log_level,
const gchar * format, ...)
{
if (g_log_callback) {
va_list va;
va_start (va, format);
switch (log_type) {
default:
g_log_callback (log_level, format, va, g_log_callback_user_data);
break;
case DVB_LOG_PIXEL:
case DVB_LOG_RUNLEN:
break;
}
va_end (va);
}
}
#else
#define dvb_log(log_type, log_level, format...)
#endif
/* FIXME: Are we waiting for an acquisition point before trying to do things? */
/* FIXME: In the end convert some of the guint8/16 (especially stack variables) back to gint for access efficiency */
/**
* SECTION:dvb-sub
* @short_description: a DVB subtitle parsing class
* @stability: Unstable
*
* The #DvbSub represents an object used for parsing a DVB subpicture,
* and signalling the API user for new bitmaps to show on screen.
*/
#define MAX_NEG_CROP 1024
static guint8 ff_cropTbl[256 + 2 * MAX_NEG_CROP] = { 0, };
#define cm (ff_cropTbl + MAX_NEG_CROP)
/* FIXME: This is really ARGB... We might need this configurable for performant
* FIXME: use in GStreamer as well if that likes RGBA more (Qt prefers ARGB) */
#define RGBA(r,g,b,a) (((a) << 24) | ((r) << 16) | ((g) << 8) | (b))
typedef struct DVBSubCLUT
{
int id; /* default_clut uses -1 for this, so guint8 isn't fine without adaptations first */
guint32 clut4[4];
guint32 clut16[16];
guint32 clut256[256];
struct DVBSubCLUT *next;
} DVBSubCLUT;
static DVBSubCLUT default_clut;
typedef struct DVBSubObjectDisplay
{
/* FIXME: Use more correct sizes */
int object_id;
int region_id;
int x_pos;
int y_pos;
int fgcolor;
int bgcolor;
/* FIXME: Should we use GSList? The relating interaction and pointer assigment is quite complex and perhaps unsuited for a plain GSList anyway */
struct DVBSubObjectDisplay *region_list_next;
struct DVBSubObjectDisplay *object_list_next;
} DVBSubObjectDisplay;
typedef struct DVBSubObject
{
/* FIXME: Use more correct sizes */
int id; /* FIXME: Use guint8 after checking it's fine in all code using it */
int type;
/* FIXME: Should we use GSList? */
DVBSubObjectDisplay *display_list;
struct DVBSubObject *next;
} DVBSubObject;
typedef struct DVBSubRegionDisplay
{ /* FIXME: Figure out if this structure is only used temporarily in page_segment parser, or also more */
int region_id;
int x_pos;
int y_pos;
struct DVBSubRegionDisplay *next;
} DVBSubRegionDisplay;
typedef struct DVBSubRegion
{
guint8 id;
guint16 width;
guint16 height;
guint8 depth; /* If we want to make this a guint8, then need to ensure it isn't wrap around with reserved values in region handling code */
guint8 clut;
guint8 bgcolor;
/* FIXME: Validate these fields existence and exact types */
guint8 *pbuf;
int buf_size;
DVBSubObjectDisplay *display_list;
struct DVBSubRegion *next;
} DVBSubRegion;
typedef struct _DvbSubPrivate DvbSubPrivate;
struct _DvbSubPrivate
{
int fd;
DvbSubCallbacks callbacks;
gpointer user_data;
guint8 page_time_out;
DVBSubRegion *region_list;
DVBSubCLUT *clut_list;
DVBSubObject *object_list;
/* FIXME... */
int display_list_size;
DVBSubRegionDisplay *display_list;
GString *pes_buffer;
};
#define DVB_SUB_GET_PRIVATE(obj) (G_TYPE_INSTANCE_GET_PRIVATE ((obj), DVB_TYPE_SUB, DvbSubPrivate))
G_DEFINE_TYPE (DvbSub, dvb_sub, G_TYPE_OBJECT);
typedef enum
{
TOP_FIELD = 0,
BOTTOM_FIELD = 1
} DvbSubPixelDataSubBlockFieldType;
/* FIXME: It might make sense to pass DvbSubPrivate for all the get_* functions, instead of public DvbSub */
static DVBSubObject *
get_object (DvbSub * dvb_sub, guint16 object_id)
{
const DvbSubPrivate *priv = (DvbSubPrivate *) dvb_sub->private_data;
DVBSubObject *ptr = priv->object_list;
while (ptr && ptr->id != object_id) {
ptr = ptr->next;
}
return ptr;
}
static DVBSubCLUT *
get_clut (DvbSub * dvb_sub, gint clut_id)
{
const DvbSubPrivate *priv = (DvbSubPrivate *) dvb_sub->private_data;
DVBSubCLUT *ptr = priv->clut_list;
while (ptr && ptr->id != clut_id) {
ptr = ptr->next;
}
return ptr;
}
// FIXME: Just pass private_data pointer directly here and in other get_* helper functions?
static DVBSubRegion *
get_region (DvbSub * dvb_sub, guint8 region_id)
{
const DvbSubPrivate *priv = (DvbSubPrivate *) dvb_sub->private_data;
DVBSubRegion *ptr = priv->region_list;
while (ptr && ptr->id != region_id) {
ptr = ptr->next;
}
return ptr;
}
static void
delete_region_display_list (DvbSub * dvb_sub, DVBSubRegion * region)
{
const DvbSubPrivate *priv = (DvbSubPrivate *) dvb_sub->private_data;
DVBSubObject *object, *obj2;
DVBSubObject **obj2_ptr;
DVBSubObjectDisplay *display, *obj_disp, **obj_disp_ptr;
while (region->display_list) {
display = region->display_list;
object = get_object (dvb_sub, display->object_id);
if (object) {
obj_disp_ptr = &object->display_list;
obj_disp = *obj_disp_ptr;
while (obj_disp && obj_disp != display) {
obj_disp_ptr = &obj_disp->object_list_next;
obj_disp = *obj_disp_ptr;
}
if (obj_disp) {
*obj_disp_ptr = obj_disp->object_list_next;
if (!object->display_list) {
obj2_ptr = (DVBSubObject **) & priv->object_list; /* FIXME: Evil casting */
obj2 = *obj2_ptr;
while (obj2 != object) {
g_assert (obj2);
obj2_ptr = &obj2->next;
obj2 = *obj2_ptr;
}
*obj2_ptr = obj2->next;
g_slice_free (DVBSubObject, obj2);
}
}
}
region->display_list = display->region_list_next;
g_slice_free (DVBSubObjectDisplay, display);
}
}
static void
delete_state (DvbSub * dvb_sub)
{
DvbSubPrivate *priv = (DvbSubPrivate *) dvb_sub->private_data;
DVBSubRegion *region;
while (priv->region_list) {
region = priv->region_list;
priv->region_list = region->next;
delete_region_display_list (dvb_sub, region);
if (region->pbuf)
g_free (region->pbuf);
g_slice_free (DVBSubRegion, region);
}
g_slice_free_chain (DVBSubCLUT, priv->clut_list, next);
priv->clut_list = NULL;
/* Should already be null */
if (priv->object_list)
g_warning ("Memory deallocation error!");
}
static void
dvb_sub_init (DvbSub * self)
{
DvbSubPrivate *priv;
self->private_data = priv = DVB_SUB_GET_PRIVATE (self);
/* TODO: Add initialization code here */
/* FIXME: Do we have a reason to initiate the members to zero, or are we guaranteed that anyway? */
priv->region_list = NULL;
priv->object_list = NULL;
priv->page_time_out = 0; /* FIXME: Maybe 255 instead? */
priv->pes_buffer = g_string_new (NULL);
}
static void
dvb_sub_finalize (GObject * object)
{
DvbSub *self = DVB_SUB (object);
DvbSubPrivate *priv = (DvbSubPrivate *) self->private_data;
/* TODO: Add deinitalization code here */
/* FIXME: Clear up region_list contents */
delete_state (self); /* close_pid should have called this, but lets be sure */
g_string_free (priv->pes_buffer, TRUE);
G_OBJECT_CLASS (dvb_sub_parent_class)->finalize (object);
}
/* init static data necessary for ffmpeg-colorspace conversion */
static void
dsputil_static_init (void)
{
int i;
for (i = 0; i < 256; i++)
ff_cropTbl[i + MAX_NEG_CROP] = i;
for (i = 0; i < MAX_NEG_CROP; i++) {
ff_cropTbl[i] = 0;
ff_cropTbl[i + MAX_NEG_CROP + 256] = 255;
}
}
static void
dvb_sub_class_init (DvbSubClass * klass)
{
int i, r, g, b, a = 0;
GObjectClass *object_class = (GObjectClass *) klass;
object_class->finalize = dvb_sub_finalize;
g_type_class_add_private (klass, sizeof (DvbSubPrivate));
dsputil_static_init (); /* Initializes ff_cropTbl table, used in YUV_TO_RGB conversion */
/* Initialize the static default_clut structure, from which other clut
* structures are initialized from (to start off with default CLUTs
* as defined in the specification). */
default_clut.id = -1;
default_clut.clut4[0] = RGBA (0, 0, 0, 0);
default_clut.clut4[1] = RGBA (255, 255, 255, 255);
default_clut.clut4[2] = RGBA (0, 0, 0, 255);
default_clut.clut4[3] = RGBA (127, 127, 127, 255);
default_clut.clut16[0] = RGBA (0, 0, 0, 0);
for (i = 1; i < 16; i++) {
if (i < 8) {
r = (i & 1) ? 255 : 0;
g = (i & 2) ? 255 : 0;
b = (i & 4) ? 255 : 0;
} else {
r = (i & 1) ? 127 : 0;
g = (i & 2) ? 127 : 0;
b = (i & 4) ? 127 : 0;
}
default_clut.clut16[i] = RGBA (r, g, b, 255);
}
default_clut.clut256[0] = RGBA (0, 0, 0, 0);
for (i = 1; i < 256; i++) {
if (i < 8) {
r = (i & 1) ? 255 : 0;
g = (i & 2) ? 255 : 0;
b = (i & 4) ? 255 : 0;
a = 63;
} else {
switch (i & 0x88) {
case 0x00:
r = ((i & 1) ? 85 : 0) + ((i & 0x10) ? 170 : 0);
g = ((i & 2) ? 85 : 0) + ((i & 0x20) ? 170 : 0);
b = ((i & 4) ? 85 : 0) + ((i & 0x40) ? 170 : 0);
a = 255;
break;
case 0x08:
r = ((i & 1) ? 85 : 0) + ((i & 0x10) ? 170 : 0);
g = ((i & 2) ? 85 : 0) + ((i & 0x20) ? 170 : 0);
b = ((i & 4) ? 85 : 0) + ((i & 0x40) ? 170 : 0);
a = 127;
break;
case 0x80:
r = 127 + ((i & 1) ? 43 : 0) + ((i & 0x10) ? 85 : 0);
g = 127 + ((i & 2) ? 43 : 0) + ((i & 0x20) ? 85 : 0);
b = 127 + ((i & 4) ? 43 : 0) + ((i & 0x40) ? 85 : 0);
a = 255;
break;
case 0x88:
r = ((i & 1) ? 43 : 0) + ((i & 0x10) ? 85 : 0);
g = ((i & 2) ? 43 : 0) + ((i & 0x20) ? 85 : 0);
b = ((i & 4) ? 43 : 0) + ((i & 0x40) ? 85 : 0);
a = 255;
break;
}
}
default_clut.clut256[i] = RGBA (r, g, b, a);
}
}
static void
_dvb_sub_parse_page_segment (DvbSub * dvb_sub, guint16 page_id, guint8 * buf,
gint buf_size)
{ /* FIXME: Use guint for buf_size here and in many other places? */
DvbSubPrivate *priv = (DvbSubPrivate *) dvb_sub->private_data;
DVBSubRegionDisplay *display;
DVBSubRegionDisplay *tmp_display_list, **tmp_ptr;
const guint8 *buf_end = buf + buf_size;
guint8 region_id;
guint8 page_state;
#ifdef DEBUG
static int counter = 0;
static const gchar *page_state_str[] = {
"Normal case",
"ACQUISITION POINT",
"Mode Change",
"RESERVED"
};
#endif
if (buf_size < 1)
return;
priv->page_time_out = *buf++;
page_state = ((*buf++) >> 2) & 3;
#ifdef DEBUG
++counter;
dvb_log (DVB_LOG_PAGE, G_LOG_LEVEL_DEBUG,
"%d: page_id = %u, length = %d, page_time_out = %u seconds, page_state = %s",
counter, page_id, buf_size, priv->page_time_out,
page_state_str[page_state]);
#endif
if (page_state == 2) { /* Mode change */
delete_state (dvb_sub);
}
tmp_display_list = priv->display_list;
priv->display_list = NULL;
priv->display_list_size = 0;
while (buf + 5 < buf_end) {
region_id = *buf++;
buf += 1;
display = tmp_display_list;
tmp_ptr = &tmp_display_list;
while (display && display->region_id != region_id) {
tmp_ptr = &display->next;
display = display->next;
}
if (!display)
display = g_slice_new0 (DVBSubRegionDisplay);
display->region_id = region_id;
display->x_pos = GST_READ_UINT16_BE (buf);
buf += 2;
display->y_pos = GST_READ_UINT16_BE (buf);
buf += 2;
*tmp_ptr = display->next;
display->next = priv->display_list;
priv->display_list = display;
priv->display_list_size++;
dvb_log (DVB_LOG_PAGE, G_LOG_LEVEL_DEBUG,
"%d: REGION information: ID = %u, address = %ux%u",
counter, region_id, display->x_pos, display->y_pos);
}
while (tmp_display_list) {
display = tmp_display_list;
tmp_display_list = display->next;
g_slice_free (DVBSubRegionDisplay, display);
}
}
static void
_dvb_sub_parse_region_segment (DvbSub * dvb_sub, guint16 page_id, guint8 * buf,
gint buf_size)
{
DvbSubPrivate *priv = (DvbSubPrivate *) dvb_sub->private_data;
const guint8 *buf_end = buf + buf_size;
guint8 region_id;
guint16 object_id;
DVBSubRegion *region;
DVBSubObject *object;
DVBSubObjectDisplay *object_display;
gboolean fill;
if (buf_size < 10)
return;
region_id = *buf++;
region = get_region (dvb_sub, region_id);
if (!region) { /* Create a new region */
region = g_slice_new0 (DVBSubRegion);
region->id = region_id;
region->next = priv->region_list;
priv->region_list = region;
}
fill = ((*buf++) >> 3) & 1;
region->width = GST_READ_UINT16_BE (buf);
buf += 2;
region->height = GST_READ_UINT16_BE (buf);
buf += 2;
if (region->width * region->height != region->buf_size) { /* FIXME: Read closer from spec what happens when dimensions change */
if (region->pbuf)
g_free (region->pbuf);
region->buf_size = region->width * region->height;
region->pbuf = g_malloc (region->buf_size); /* TODO: We can probably use GSlice here if careful about freeing while buf_size still records the correct size */
fill = 1; /* FIXME: Validate from spec that fill is forced on (in the following codes context) when dimensions change */
}
region->depth = 1 << (((*buf++) >> 2) & 7);
if (region->depth < 2 || region->depth > 8) {
g_warning ("region depth %d is invalid\n", region->depth);
region->depth = 4; /* FIXME: Check from spec this is the default? */
}
region->clut = *buf++;
if (region->depth == 8)
region->bgcolor = *buf++;
else {
buf += 1;
if (region->depth == 4)
region->bgcolor = (((*buf++) >> 4) & 15);
else
region->bgcolor = (((*buf++) >> 2) & 3);
}
dvb_log (DVB_LOG_REGION, G_LOG_LEVEL_DEBUG,
"id = %u, (%ux%u)@%u-bit",
region_id, region->width, region->height, region->depth);
if (fill) {
memset (region->pbuf, region->bgcolor, region->buf_size);
dvb_log (DVB_LOG_REGION, G_LOG_LEVEL_DEBUG,
"Filling region (%u) with bgcolor = %u", region->id, region->bgcolor);
}
delete_region_display_list (dvb_sub, region); /* Delete the region display list for current region - FIXME: why? */
while (buf + 6 <= buf_end) {
object_id = GST_READ_UINT16_BE (buf);
buf += 2;
object = get_object (dvb_sub, object_id);
if (!object) {
object = g_slice_new0 (DVBSubObject);
object->id = object_id;
object->next = priv->object_list;
priv->object_list = object;
}
object->type = (*buf) >> 6;
object_display = g_slice_new0 (DVBSubObjectDisplay);
object_display->object_id = object_id;
object_display->region_id = region_id;
object_display->x_pos = GST_READ_UINT16_BE (buf) & 0xfff;
buf += 2;
object_display->y_pos = GST_READ_UINT16_BE (buf) & 0xfff;
buf += 2;
if ((object->type == 1 || object->type == 2) && buf + 2 <= buf_end) {
object_display->fgcolor = *buf++;
object_display->bgcolor = *buf++;
}
object_display->region_list_next = region->display_list;
region->display_list = object_display;
object_display->object_list_next = object->display_list;
object->display_list = object_display;
dvb_log (DVB_LOG_REGION, G_LOG_LEVEL_DEBUG,
"REGION DATA: object_id = %u, region_id = %u, pos = %ux%u, obj_type = %u",
object->id, region->id, object_display->x_pos, object_display->y_pos,
object->type);
if (object->type == 1 || object->type == 2)
dvb_log (DVB_LOG_REGION, G_LOG_LEVEL_DEBUG,
"REGION DATA: fgcolor = %u, bgcolor = %u\n", object_display->fgcolor,
object_display->bgcolor);
}
}
static void
_dvb_sub_parse_clut_segment (DvbSub * dvb_sub, guint16 page_id, guint8 * buf,
gint buf_size)
{
DvbSubPrivate *priv = (DvbSubPrivate *) dvb_sub->private_data;
const guint8 *buf_end = buf + buf_size;
guint8 clut_id;
DVBSubCLUT *clut;
int entry_id, depth, full_range;
int y, cr, cb, alpha;
int r, g, b, r_add, g_add, b_add;
#ifdef DEBUG_PACKET_CONTENTS
g_print ("DVB clut packet:\n");
gst_util_dump_mem (buf, buf_size);
#endif
clut_id = *buf++;
buf += 1;
clut = get_clut (dvb_sub, clut_id);
if (!clut) {
clut = g_slice_new (DVBSubCLUT); /* FIXME-MEMORY-LEAK: This seems to leak per valgrind */
memcpy (clut, &default_clut, sizeof (DVBSubCLUT));
clut->id = clut_id;
clut->next = priv->clut_list;
priv->clut_list = clut;
}
while (buf + 4 < buf_end) {
entry_id = *buf++;
depth = (*buf) & 0xe0;
if (depth == 0) {
g_warning ("Invalid clut depth 0x%x!", *buf);
return;
}
full_range = (*buf++) & 1;
if (full_range) {
y = *buf++;
cr = *buf++;
cb = *buf++;
alpha = *buf++;
} else {
y = buf[0] & 0xfc;
cr = (((buf[0] & 3) << 2) | ((buf[1] >> 6) & 3)) << 4;
cb = (buf[1] << 2) & 0xf0;
alpha = (buf[1] << 6) & 0xc0;
buf += 2;
}
if (y == 0)
alpha = 0xff;
YUV_TO_RGB1_CCIR (cb, cr);
YUV_TO_RGB2_CCIR (r, g, b, y);
dvb_log (DVB_LOG_CLUT, G_LOG_LEVEL_DEBUG,
"CLUT DEFINITION: clut %d := (%d,%d,%d,%d)", entry_id, r, g, b, alpha);
if (depth & 0x80)
clut->clut4[entry_id] = RGBA (r, g, b, 255 - alpha);
if (depth & 0x40)
clut->clut16[entry_id] = RGBA (r, g, b, 255 - alpha);
if (depth & 0x20)
clut->clut256[entry_id] = RGBA (r, g, b, 255 - alpha);
}
}
// FFMPEG-FIXME: The same code in ffmpeg is much more complex, it could use the same
// FFMPEG-FIXME: refactoring as done here
static int
_dvb_sub_read_2bit_string (guint8 * destbuf, gint dbuf_len,
const guint8 ** srcbuf, gint buf_size, guint8 non_mod, guint8 * map_table)
{
GstBitReader gb = GST_BIT_READER_INIT (*srcbuf, buf_size);
/* FIXME: Handle FALSE returns from gst_bit_reader_get_* calls? */
gboolean stop_parsing = FALSE;
guint32 bits = 0;
guint32 pixels_read = 0;
static gboolean warning_shown = FALSE;
if (!warning_shown) {
g_warning
("Parsing 2bit color DVB sub-picture. This is not tested at all. If you see this message, "
"please provide the developers with sample media with these subtitles, if possible.");
warning_shown = TRUE;
}
dvb_log (DVB_LOG_PIXEL, G_LOG_LEVEL_DEBUG,
"(n=2): Inside %s with dbuf_len = %d", __PRETTY_FUNCTION__, dbuf_len);
while (!stop_parsing && (gst_bit_reader_get_remaining (&gb) > 0)) {
guint run_length = 0, clut_index = 0;
gst_bit_reader_get_bits_uint32 (&gb, &bits, 2);
if (bits) { /* 2-bit_pixel-code */
run_length = 1;
clut_index = bits;
} else { /* 2-bit_zero */
gst_bit_reader_get_bits_uint32 (&gb, &bits, 1);
if (bits == 1) { /* switch_1 == '1' */
gst_bit_reader_get_bits_uint32 (&gb, &run_length, 3);
run_length += 3;
gst_bit_reader_get_bits_uint32 (&gb, &clut_index, 2);
} else { /* switch_1 == '0' */
gst_bit_reader_get_bits_uint32 (&gb, &bits, 1);
if (bits == 1) { /* switch_2 == '1' */
run_length = 1; /* 1x pseudo-colour '00' */
} else { /* switch_2 == '0' */
gst_bit_reader_get_bits_uint32 (&gb, &bits, 2);
switch (bits) { /* switch_3 */
case 0x0: /* end of 2-bit/pixel_code_string */
stop_parsing = TRUE;
break;
case 0x1: /* two pixels shall be set to pseudo colour (entry) '00' */
run_length = 2;
break;
case 0x2: /* the following 6 bits contain run length coded pixel data */
gst_bit_reader_get_bits_uint32 (&gb, &run_length, 4);
run_length += 12;
gst_bit_reader_get_bits_uint32 (&gb, &clut_index, 2);
break;
case 0x3: /* the following 10 bits contain run length coded pixel data */
gst_bit_reader_get_bits_uint32 (&gb, &run_length, 8);
run_length += 29;
gst_bit_reader_get_bits_uint32 (&gb, &clut_index, 2);
break;
}
}
}
}
/* If run_length is zero, continue. Only case happening is when
* stop_parsing is TRUE too, so next cycle shouldn't run */
if (run_length == 0)
continue;
/* Trim the run_length to not go beyond the line end and consume
* it from remaining length of dest line */
run_length = MIN (run_length, dbuf_len);
dbuf_len -= run_length;
/* Make clut_index refer to the index into the desired bit depths
* CLUT definition table */
if (map_table)
clut_index = map_table[clut_index]; /* now clut_index signifies the index into map_table dest */
/* Now we can simply memset run_length count of destination bytes
* to clut_index, but only if not non_modifying */
dvb_log (DVB_LOG_RUNLEN, G_LOG_LEVEL_DEBUG,
"Setting %u pixels to color 0x%x in destination buffer; dbuf_len left is %d pixels",
run_length, clut_index, dbuf_len);
if (!(non_mod == 1 && bits == 1))
memset (destbuf, clut_index, run_length);
destbuf += run_length;
pixels_read += run_length;
}
// FIXME: Test skip_to_byte instead of adding 7 bits, once everything else is working good
//gst_bit_reader_skip_to_byte (&gb);
*srcbuf += (gst_bit_reader_get_pos (&gb) + 7) >> 3;
dvb_log (DVB_LOG_PIXEL, G_LOG_LEVEL_DEBUG,
"Returning from 2bit_string parser with %u pixels read", pixels_read);
// FIXME: Shouldn't need this variable if tracking things in the loop better
return pixels_read;
}
// FFMPEG-FIXME: The same code in ffmpeg is much more complex, it could use the same
// FFMPEG-FIXME: refactoring as done here, explained in commit 895296c3
static int
_dvb_sub_read_4bit_string (guint8 * destbuf, gint dbuf_len,
const guint8 ** srcbuf, gint buf_size, guint8 non_mod, guint8 * map_table)
{
GstBitReader gb = GST_BIT_READER_INIT (*srcbuf, buf_size);
/* FIXME: Handle FALSE returns from gst_bit_reader_get_* calls? */
gboolean stop_parsing = FALSE;
guint32 bits = 0;
guint32 pixels_read = 0;
dvb_log (DVB_LOG_RUNLEN, G_LOG_LEVEL_DEBUG,
"Entering 4bit_string parser at srcbuf position %p with buf_size = %d; destination buffer size is %d @ %p",
*srcbuf, buf_size, dbuf_len, destbuf);
while (!stop_parsing && (gst_bit_reader_get_remaining (&gb) > 0)) {
guint run_length = 0, clut_index = 0;
gst_bit_reader_get_bits_uint32 (&gb, &bits, 4);
if (bits) {
run_length = 1;
clut_index = bits;
} else {
gst_bit_reader_get_bits_uint32 (&gb, &bits, 1);
if (bits == 0) { /* switch_1 == '0' */
gst_bit_reader_get_bits_uint32 (&gb, &run_length, 3);
if (!run_length) {
stop_parsing = TRUE;
} else {
run_length += 2;
}
} else { /* switch_1 == '1' */
gst_bit_reader_get_bits_uint32 (&gb, &bits, 1);
if (bits == 0) { /* switch_2 == '0' */
gst_bit_reader_get_bits_uint32 (&gb, &run_length, 2);
run_length += 4;
gst_bit_reader_get_bits_uint32 (&gb, &clut_index, 4);
} else { /* switch_2 == '1' */
gst_bit_reader_get_bits_uint32 (&gb, &bits, 2);
switch (bits) {
case 0x0: /* switch_3 == '00' */
run_length = 1; /* 1 pixel of pseudo-color 0 */
break;
case 0x1: /* switch_3 == '01' */
run_length = 2; /* 2 pixels of pseudo-color 0 */
break;
case 0x2: /* switch_3 == '10' */
gst_bit_reader_get_bits_uint32 (&gb, &run_length, 4);
run_length += 9;
gst_bit_reader_get_bits_uint32 (&gb, &clut_index, 4);
break;
case 0x3: /* switch_3 == '11' */
gst_bit_reader_get_bits_uint32 (&gb, &run_length, 8);
run_length += 25;
gst_bit_reader_get_bits_uint32 (&gb, &clut_index, 4);
break;
}
}
}
}
/* If run_length is zero, continue. Only case happening is when
* stop_parsing is TRUE too, so next cycle shouldn't run */
if (run_length == 0)
continue;
/* Trim the run_length to not go beyond the line end and consume
* it from remaining length of dest line */
run_length = MIN (run_length, dbuf_len);
dbuf_len -= run_length;
/* Make clut_index refer to the index into the desired bit depths
* CLUT definition table */
if (map_table)
clut_index = map_table[clut_index]; /* now clut_index signifies the index into map_table dest */
/* Now we can simply memset run_length count of destination bytes
* to clut_index, but only if not non_modifying */
dvb_log (DVB_LOG_RUNLEN, G_LOG_LEVEL_DEBUG,
"Setting %u pixels to color 0x%x in destination buffer; dbuf_len left is %d pixels",
run_length, clut_index, dbuf_len);
if (!(non_mod == 1 && bits == 1))
memset (destbuf, clut_index, run_length);
destbuf += run_length;
pixels_read += run_length;
}
// FIXME: Test skip_to_byte instead of adding 7 bits, once everything else is working good
//gst_bit_reader_skip_to_byte (&gb);
*srcbuf += (gst_bit_reader_get_pos (&gb) + 7) >> 3;
dvb_log (DVB_LOG_PIXEL, G_LOG_LEVEL_DEBUG,
"Returning from 4bit_string parser with %u pixels read", pixels_read);
// FIXME: Shouldn't need this variable if tracking things in the loop better
return pixels_read;
}
static int
_dvb_sub_read_8bit_string (guint8 * destbuf, gint dbuf_len,
const guint8 ** srcbuf, gint buf_size, guint8 non_mod, guint8 * map_table)
{
GstBitReader gb = GST_BIT_READER_INIT (*srcbuf, buf_size);
/* FIXME: Handle FALSE returns from gst_bit_reader_get_* calls? */
gboolean stop_parsing = FALSE;
guint32 bits = 0;
guint32 pixels_read = 0;
static gboolean warning_shown = FALSE;
if (!warning_shown) {
g_warning
("Parsing 8bit color DVB sub-picture. This is not tested at all. If you see this message, "
"please provide the developers with sample media with these subtitles, if possible.");
warning_shown = TRUE;
}
dvb_log (DVB_LOG_PIXEL, G_LOG_LEVEL_DEBUG,
"(n=8): Inside %s with dbuf_len = %d", __PRETTY_FUNCTION__, dbuf_len);
/* FFMPEG-FIXME: ffmpeg uses a manual byte walking algorithm, which might be more performant,
* FFMPEG-FIXME: but it does almost absolutely no buffer length checking, so could walk over
* FFMPEG-FIXME: memory boundaries. While we don't check gst_bit_reader_get_bits_uint32
* FFMPEG-FIXME: return values either and therefore might get some pixels corrupted, we at
* FFMPEG-FIXME: lest have no chance of reading memory we don't own and visual corruption
* FFMPEG-FIXME: is guaranteed anyway when not all bytes are present */
/* Rephrased - it's better to work with bytes with default value '0' instead of reading from memory we don't own. */
while (!stop_parsing && (gst_bit_reader_get_remaining (&gb) > 0)) {
guint run_length = 0, clut_index = 0;
gst_bit_reader_get_bits_uint32 (&gb, &bits, 8);
if (bits) { /* 8-bit_pixel-code */
run_length = 1;
clut_index = bits;
} else { /* 8-bit_zero */
gst_bit_reader_get_bits_uint32 (&gb, &bits, 1);
if (bits == 0) { /* switch_1 == '0' */
/* run_length_1-127 for pseudo-colour _entry) '0x00' */
gst_bit_reader_get_bits_uint32 (&gb, &run_length, 7);
if (run_length == 0) { /* end_of_string_signal */
stop_parsing = TRUE;
}
} else { /* switch_1 == '1' */
/* run_length_3-127 */
gst_bit_reader_get_bits_uint32 (&gb, &run_length, 7);
gst_bit_reader_get_bits_uint32 (&gb, &clut_index, 8);
#ifdef DEBUG
/* Emit a debugging message about stream not following specification */
if (run_length < 3) {
dvb_log (DVB_LOG_STREAM, G_LOG_LEVEL_WARNING,
"8-bit/pixel_code_string::run_length_3-127 value was %u, but the spec requires it must be >=3",
run_length);
}
#endif
}
}
/* If run_length is zero, continue. Only case happening is when
* stop_parsing is TRUE too, so next cycle shouldn't run */
if (run_length == 0)
continue;
/* Trim the run_length to not go beyond the line end and consume
* it from remaining length of dest line */
run_length = MIN (run_length, dbuf_len);
dbuf_len -= run_length;
/* Make clut_index refer to the index into the desired bit depths
* CLUT definition table */
if (map_table)
clut_index = map_table[clut_index]; /* now clut_index signifies the index into map_table dest */
/* Now we can simply memset run_length count of destination bytes
* to clut_index, but only if not non_modifying */
dvb_log (DVB_LOG_RUNLEN, G_LOG_LEVEL_DEBUG,
"Setting %u pixels to color 0x%x in destination buffer; dbuf_len left is %d pixels",
run_length, clut_index, dbuf_len);
if (!(non_mod == 1 && bits == 1))
memset (destbuf, clut_index, run_length);
destbuf += run_length;
pixels_read += run_length;
}
dvb_log (DVB_LOG_PIXEL, G_LOG_LEVEL_DEBUG,
"Returning from 8bit_string parser with %u pixels read", pixels_read);
// FIXME: Shouldn't need this variable if tracking things in the loop better
return pixels_read;
}
static void
_dvb_sub_parse_pixel_data_block (DvbSub * dvb_sub,
DVBSubObjectDisplay * display, const guint8 * buf, gint buf_size,
DvbSubPixelDataSubBlockFieldType top_bottom, guint8 non_mod)
{
DVBSubRegion *region = get_region (dvb_sub, display->region_id);
const guint8 *buf_end = buf + buf_size;
guint8 *pbuf;
int x_pos, y_pos;
int i;
gboolean dest_buf_filled = FALSE;
guint8 map2to4[] = { 0x0, 0x7, 0x8, 0xf };
guint8 map2to8[] = { 0x00, 0x77, 0x88, 0xff };
guint8 map4to8[] = { 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff
};
guint8 *map_table;
dvb_log (DVB_LOG_PIXEL, G_LOG_LEVEL_DEBUG,
"(parse_block): DVB pixel block size %d, %s field:",
buf_size, top_bottom ? "bottom" : "top");
#ifdef DEBUG_PACKET_CONTENTS
gst_util_dump_mem (buf, buf_size);
#endif
if (region == NULL) {
g_print ("Region is NULL, returning\n");
return;
}
pbuf = region->pbuf;
x_pos = display->x_pos;
y_pos = display->y_pos;
if ((y_pos & 1) != top_bottom)
y_pos++;
while (buf < buf_end) {
dvb_log (DVB_LOG_PIXEL, G_LOG_LEVEL_DEBUG,
"Iteration start, %u bytes missing from end; buf = %p, buf_end = %p; "
"Region is number %u, with a dimension of %dx%d; We are at position %dx%d",
buf_end - buf, buf, buf_end,
region->id, region->width, region->height, x_pos, y_pos);
// FFMPEG-FIXME: ffmpeg doesn't check for equality and so can overflow destination buffer later on with bad input data
// FFMPEG-FIXME: However that makes it warn on end_of_object_line and map tables as well, so we add the dest_buf_filled tracking
// FIXME: Removed x_pos checking here, because we don't want to turn dest_buf_filled to TRUE permanently in that case
// FIXME: We assume that region->width - x_pos as dbuf_len to read_nbit_string will take care of that case nicely;
// FIXME: That is, that read_nbit_string never scribbles anything if dbuf_len passed to it is zero due to this.
if (y_pos >= region->height) {
dest_buf_filled = TRUE;
}
switch (*buf++) {
case 0x10:
if (dest_buf_filled) {
g_warning ("Invalid object location for data_type 0x%x!\n", *(buf - 1)); /* FIXME: Be more verbose */
g_print ("Remaining data after invalid object location:\n");
gst_util_dump_mem (buf, buf_end - buf);
return;
}
if (region->depth == 8)
map_table = map2to8;
else if (region->depth == 4)
map_table = map2to4;
else
map_table = NULL;
// FFMPEG-FIXME: ffmpeg code passes buf_size instead of buf_end - buf, and could
// FFMPEG-FIXME: therefore potentially walk over the memory area we own
x_pos +=
_dvb_sub_read_2bit_string (pbuf + (y_pos * region->width) + x_pos,
region->width - x_pos, &buf, buf_end - buf, non_mod, map_table);
break;
case 0x11:
if (dest_buf_filled) {
g_warning ("Invalid object location for data_type 0x%x!\n", *(buf - 1)); /* FIXME: Be more verbose */
g_print ("Remaining data after invalid object location:\n");
gst_util_dump_mem (buf, buf_end - buf);
return; // FIXME: Perhaps tell read_nbit_string that dbuf_len is zero and let it walk the bytes regardless? (Same FIXME for 2bit and 8bit)
}
if (region->depth < 4) {
g_warning ("4-bit pixel string in %d-bit region!\n", region->depth);
return;
}
if (region->depth == 8)
map_table = map4to8;
else
map_table = NULL;
dvb_log (DVB_LOG_PIXEL, G_LOG_LEVEL_DEBUG,
"READ_nBIT_STRING (4): String data into position %dx%d; buf before is %p\n",
x_pos, y_pos, buf);
// FFMPEG-FIXME: ffmpeg code passes buf_size instead of buf_end - buf, and could
// FFMPEG-FIXME: therefore potentially walk over the memory area we own
x_pos +=
_dvb_sub_read_4bit_string (pbuf + (y_pos * region->width) + x_pos,
region->width - x_pos, &buf, buf_end - buf, non_mod, map_table);
dvb_log (DVB_LOG_PIXEL, G_LOG_LEVEL_DEBUG,
"READ_nBIT_STRING (4) finished: buf pointer now %p", buf);
break;
case 0x12:
if (dest_buf_filled) {
g_warning ("Invalid object location for data_type 0x%x!\n", *(buf - 1)); /* FIXME: Be more verbose */
g_print ("Remaining data after invalid object location:\n");
gst_util_dump_mem (buf, buf_end - buf);
return;
}
if (region->depth < 8) {
g_warning ("8-bit pixel string in %d-bit region!\n", region->depth);
return;
}
// FFMPEG-FIXME: ffmpeg code passes buf_size instead of buf_end - buf, and could
// FFMPEG-FIXME: therefore potentially walk over the memory area we own
x_pos +=
_dvb_sub_read_8bit_string (pbuf + (y_pos * region->width) + x_pos,
region->width - x_pos, &buf, buf_end - buf, non_mod, NULL);
break;
case 0x20:
dvb_log (DVB_LOG_PIXEL, G_LOG_LEVEL_DEBUG,
"(parse_block): handling map2to4 table data");
/* FIXME: I don't see any guards about buffer size here - buf++ happens with the switch, but
* FIXME: buffer is walked without length checks? Same deal in other map table cases */
map2to4[0] = (*buf) >> 4;
map2to4[1] = (*buf++) & 0xf;
map2to4[2] = (*buf) >> 4;
map2to4[3] = (*buf++) & 0xf;
break;
case 0x21:
dvb_log (DVB_LOG_PIXEL, G_LOG_LEVEL_DEBUG,
"(parse_block): handling map2to8 table data");
for (i = 0; i < 4; i++)
map2to8[i] = *buf++;
break;
case 0x22:
dvb_log (DVB_LOG_PIXEL, G_LOG_LEVEL_DEBUG,
"(parse_block): handling map4to8 table data");
for (i = 0; i < 16; i++)
map4to8[i] = *buf++;
break;
case 0xf0:
dvb_log (DVB_LOG_PIXEL, G_LOG_LEVEL_DEBUG,
"(parse_block): end of object line code encountered");
x_pos = display->x_pos;
y_pos += 2;
break;
default:
/* FIXME: Do we consume word align stuffing byte that could follow top/bottom data? */
g_warning ("Unknown/unsupported pixel block 0x%x", *(buf - 1));
}
}
}
static void
_dvb_sub_parse_object_segment (DvbSub * dvb_sub, guint16 page_id, guint8 * buf,
gint buf_size)
{
const guint8 *buf_end = buf + buf_size;
guint object_id;
DVBSubObject *object;
guint8 coding_method, non_modifying_color;
object_id = GST_READ_UINT16_BE (buf);
buf += 2;
object = get_object (dvb_sub, object_id);
dvb_log (DVB_LOG_OBJECT, G_LOG_LEVEL_DEBUG,
"parse_object_segment: A new object segment has occurred for object_id = %u",
object_id);
if (!object) {
g_warning
("Nothing known about object with ID %u yet inside parse_object_segment, bailing out",
object_id);
return;
}
coding_method = ((*buf) >> 2) & 3;
non_modifying_color = ((*buf++) >> 1) & 1;
if (coding_method == 0) {
const guint8 *block;
DVBSubObjectDisplay *display;
guint16 top_field_len, bottom_field_len;
top_field_len = GST_READ_UINT16_BE (buf);
buf += 2;
bottom_field_len = GST_READ_UINT16_BE (buf);
buf += 2;
if (buf + top_field_len + bottom_field_len > buf_end) {
g_warning ("%s: Field data size too large\n", __PRETTY_FUNCTION__);
return;
}
/* FIXME: Potential optimization opportunity here - parse the object pixmap only once, and copy it to all the
* FIXME: regions that need it. One object being in multiple regions is a rare occurrence in real life, however */
for (display = object->display_list; display;
display = display->object_list_next) {
block = buf;
dvb_log (DVB_LOG_OBJECT, G_LOG_LEVEL_DEBUG,
"Parsing top and bottom part of object id %d; top_field_len = %u, bottom_field_len = %u",
display->object_id, top_field_len, bottom_field_len);
_dvb_sub_parse_pixel_data_block (dvb_sub, display, block, top_field_len,
TOP_FIELD, non_modifying_color);
if (bottom_field_len > 0)
block = buf + top_field_len;
else
bottom_field_len = top_field_len;
_dvb_sub_parse_pixel_data_block (dvb_sub, display, block,
bottom_field_len, BOTTOM_FIELD, non_modifying_color);
}
} else if (coding_method == 1) {
g_warning ("'a string of characters' coding method not supported (yet?)!");
} else {
g_warning ("%s: Unknown object coding 0x%x\n", __PRETTY_FUNCTION__,
coding_method);
}
}
static gint
_dvb_sub_parse_end_of_display_set (DvbSub * dvb_sub, guint16 page_id,
guint8 * buf, gint buf_size, guint64 pts)
{
DvbSubPrivate *priv = (DvbSubPrivate *) dvb_sub->private_data;
DVBSubtitles *sub = g_slice_new0 (DVBSubtitles);
DVBSubRegion *region;
DVBSubRegionDisplay *display;
DVBSubtitleRect *rect;
DVBSubCLUT *clut;
guint32 *clut_table;
int i;
static unsigned counter = 0; /* DEBUG use only */
dvb_log (DVB_LOG_DISPLAY, G_LOG_LEVEL_DEBUG,
"END OF DISPLAY SET: page_id = %u, length = %d\n", page_id, buf_size);
sub->rects = NULL;
#if 0 /* FIXME: PTS stuff not figured out yet */
sub->start_display_time = 0;
sub->end_display_time = priv->page_time_out * 1000;
sub->format = 0; /* 0 = graphics */
#endif
sub->num_rects = priv->display_list_size;
if (sub->num_rects > 0) {
// FIXME-MEMORY-LEAK: This structure is not freed up yet
sub->rects = g_malloc0 (sizeof (*sub->rects) * sub->num_rects); /* GSlice? */
for (i = 0; i < sub->num_rects; i++)
sub->rects[i] = g_malloc0 (sizeof (*sub->rects[i])); /* GSlice? */
}
i = 0;
for (display = priv->display_list; display; display = display->next) {
region = get_region (dvb_sub, display->region_id);
rect = sub->rects[i];
if (!region)
continue;
rect->x = display->x_pos;
rect->y = display->y_pos;
rect->w = region->width;
rect->h = region->height;
#if 0 /* FIXME: Don't think we need to save the number of colors in the palette when we are saving as RGBA? */
rect->nb_colors = 16;
#endif
#if 0 /* FIXME: Needed to be specified once we support strings of characters based subtitles */
rect->type = SUBTITLE_BITMAP;
#endif
rect->pict.rowstride = region->width;
rect->pict.palette_bits_count = region->depth;
clut = get_clut (dvb_sub, region->clut);
if (!clut)
clut = &default_clut;
switch (region->depth) {
case 2:
clut_table = clut->clut4;
break;
case 8:
clut_table = clut->clut256;
break;
case 4:
default:
clut_table = clut->clut16;
break;
}
/* FIXME: Tweak this to be saved in a format most suitable for Qt and GStreamer instead.
* Currently kept in AVPicture for quick save_display_set testing */
rect->pict.palette = g_malloc ((1 << region->depth) * sizeof (guint32)); /* FIXME: Can we use GSlice here? */
memcpy (rect->pict.palette, clut_table,
(1 << region->depth) * sizeof (guint32));
#if 0
g_print ("rect->pict.data.palette content:\n");
gst_util_dump_mem (rect->pict.palette,
(1 << region->depth) * sizeof (guint32));
#endif
rect->pict.data = g_malloc (region->buf_size); /* FIXME: Can we use GSlice here? */
memcpy (rect->pict.data, region->pbuf, region->buf_size);
++counter;
dvb_log (DVB_LOG_DISPLAY, G_LOG_LEVEL_DEBUG,
"An object rect created: number %u, iteration %u, pos: %d:%d, size: %dx%d",
counter, i, rect->x, rect->y, rect->w, rect->h);
#if 0
g_print ("rect->pict.data content:\n");
gst_util_dump_mem (rect->pict.data, region->buf_size);
#endif
++i;
}
sub->pts = pts;
sub->page_time_out = priv->page_time_out;
sub->num_rects = i;
if (priv->callbacks.new_data) {
priv->callbacks.new_data (dvb_sub, sub, priv->user_data);
} else {
/* No-one responsible to clean up memory, so do it ourselves */
/* FIXME: Just don't bother with all this palette image creation in the first place then... */
dvb_subtitles_free (sub);
}
return 1; /* FIXME: The caller of this function is probably supposed to do something with the return value */
}
void
dvb_subtitles_free (DVBSubtitles * sub)
{
int i;
DVBSubtitleRect *rect;
/* Now free up all the temporary memory we allocated */
for (i = 0; i < sub->num_rects; ++i) {
rect = sub->rects[i];
g_free (rect->pict.palette);
g_free (rect->pict.data);
g_free (rect);
}
g_free (sub->rects);
g_slice_free (DVBSubtitles, sub);
}
/**
* dvb_sub_new:
*
* Creates a new #DvbSub.
*
* Return value: a newly created #DvbSub
*/
DvbSub *
dvb_sub_new (void)
{
DvbSub *dvbsub = g_object_new (DVB_TYPE_SUB, NULL);
return dvbsub;
}
#define DVB_SUB_SEGMENT_PAGE_COMPOSITION 0x10
#define DVB_SUB_SEGMENT_REGION_COMPOSITION 0x11
#define DVB_SUB_SEGMENT_CLUT_DEFINITION 0x12
#define DVB_SUB_SEGMENT_OBJECT_DATA 0x13
#define DVB_SUB_SEGMENT_END_OF_DISPLAY_SET 0x80
#define DVB_SUB_SEGMENT_STUFFING 0xFF
#define DVB_SUB_SYNC_BYTE 0x0f
/**
* dvb_sub_feed_with_pts:
* @dvb_sub: a #DvbSub
* @pts: The PTS of the data
* @data: The data to feed to the parser
* @len: Length of the data
*
* Feeds the DvbSub parser with new binary data to parse,
* with an associated PTS value. E.g, data left after PES
* packet header has been already parsed, which contains
* the PTS information).
*
* Return value: -1 if data was unhandled (e.g, not a subtitle packet),
* -2 if data parsing was unsuccesful (e.g, length was invalid),
* 0 or positive if data was handled. If positive, then amount of data consumed on success. FIXME: List the positive return values.
*/
gint
dvb_sub_feed_with_pts (DvbSub * dvb_sub, guint64 pts, guint8 * data, gint len)
{
unsigned int pos = 0;
guint8 segment_type;
guint16 segment_len;
guint16 page_id;
dvb_log (DVB_LOG_PACKET, G_LOG_LEVEL_DEBUG,
"Inside dvb_sub_feed_with_pts with pts=%" G_GUINT64_FORMAT
" and length %d", pts, len);
g_return_val_if_fail (data != NULL, -1);
if (len <= 3) { /* len(0x20 0x00 end_of_PES_data_field_marker) */
g_warning ("Data length too short");
return -1;
}
if (data[pos++] != 0x20) {
g_warning
("Tried to handle a PES packet private data that isn't a subtitle packet (does not start with 0x20)");
return -1;
}
if (data[pos++] != 0x00) {
g_warning
("'Subtitle stream in this PES packet' was not 0x00, so this is in theory not a DVB subtitle stream (but some other subtitle standard?); bailing out");
return -1;
}
while (data[pos++] == DVB_SUB_SYNC_BYTE) {
if ((len - pos) < (2 * 2 + 1)) {
g_warning
("Data after SYNC BYTE too short, less than needed to even get to segment_length");
return -2;
}
segment_type = data[pos++];
dvb_log (DVB_LOG_PACKET, G_LOG_LEVEL_DEBUG,
"=== Segment type is 0x%x", segment_type);
page_id = (data[pos] << 8) | data[pos + 1];
dvb_log (DVB_LOG_PACKET, G_LOG_LEVEL_DEBUG, "page_id is 0x%x", page_id);
pos += 2;
segment_len = (data[pos] << 8) | data[pos + 1];
dvb_log (DVB_LOG_PACKET, G_LOG_LEVEL_DEBUG,
"segment_length is %d (0x%x 0x%x)", segment_len, data[pos],
data[pos + 1]);
pos += 2;
if ((len - pos) < segment_len) {
g_warning
("segment_length was told to be %u, but we only have %d bytes left",
segment_len, len - pos);
return -2;
}
// TODO: Parse the segment per type (this is probably a leftover TODO that is now done?)
/* FIXME: Handle differing PTS values - all segments of a given display set must be with the same PTS,
* FIXME: but we let it slip and just take it for granted in end_of_display_set */
switch (segment_type) {
case DVB_SUB_SEGMENT_PAGE_COMPOSITION:
dvb_log (DVB_LOG_PACKET, G_LOG_LEVEL_DEBUG,
"Page composition segment at buffer pos %u\n", pos);
_dvb_sub_parse_page_segment (dvb_sub, page_id, data + pos, segment_len); /* FIXME: Not sure about args */
break;
case DVB_SUB_SEGMENT_REGION_COMPOSITION:
dvb_log (DVB_LOG_PACKET, G_LOG_LEVEL_DEBUG,
"Region composition segment at buffer pos %u\n", pos);
_dvb_sub_parse_region_segment (dvb_sub, page_id, data + pos, segment_len); /* FIXME: Not sure about args */
break;
case DVB_SUB_SEGMENT_CLUT_DEFINITION:
dvb_log (DVB_LOG_PACKET, G_LOG_LEVEL_DEBUG,
"CLUT definition segment at buffer pos %u\n", pos);
_dvb_sub_parse_clut_segment (dvb_sub, page_id, data + pos, segment_len); /* FIXME: Not sure about args */
break;
case DVB_SUB_SEGMENT_OBJECT_DATA:
dvb_log (DVB_LOG_PACKET, G_LOG_LEVEL_DEBUG,
"Object data segment at buffer pos %u\n", pos);
_dvb_sub_parse_object_segment (dvb_sub, page_id, data + pos, segment_len); /* FIXME: Not sure about args */
break;
case DVB_SUB_SEGMENT_END_OF_DISPLAY_SET:
dvb_log (DVB_LOG_PACKET, G_LOG_LEVEL_DEBUG,
"End of display set at buffer pos %u\n", pos);
_dvb_sub_parse_end_of_display_set (dvb_sub, page_id, data + pos, segment_len, pts); /* FIXME: Not sure about args */
break;
default:
g_warning ("Unhandled segment type 0x%x", segment_type);
break;
}
pos += segment_len;
if (pos == len) {
g_warning ("Data ended without a PES data end marker");
return 1;
}
}
g_warning ("Processed %d bytes out of %d\n", pos, len);
return pos;
}
/**
* dvb_sub_set_callbacks:
* @dvb_sub: a #DvbSub
* @callbacks: the callbacks to install
* @user_data: a user_data argument for the callback
*
* Set callback which will be executed when new subpictures are available.
*/
void
dvb_sub_set_callbacks (DvbSub * dvb_sub, DvbSubCallbacks * callbacks,
gpointer user_data)
{
DvbSubPrivate *priv;
g_return_if_fail (dvb_sub != NULL);
g_return_if_fail (DVB_IS_SUB (dvb_sub));
g_return_if_fail (callbacks != NULL);
priv = (DvbSubPrivate *) dvb_sub->private_data;
priv->callbacks = *callbacks;
priv->user_data = user_data;
}
void
dvb_sub_set_global_log_cb (void (*log_cb) (GLogLevelFlags log_level,
const gchar * format, va_list args, gpointer user_data),
gpointer user_data)
{
if (log_cb) {
g_log_callback = log_cb;
g_log_callback_user_data = user_data;
}
}
Reference in a new issue View git blame Copy permalink