mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-09-08 13:18:52 +00:00
1498 lines
46 KiB
C
1498 lines
46 KiB
C
/* dvb-sub.c - DVB subtitle decoding
|
|
* Copyright (C) Mart Raudsepp 2009 <mart.raudsepp@artecdesign.ee>
|
|
* Copyright (C) 2010 ONELAN Ltd.
|
|
*
|
|
* 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 <string.h> /* memset */
|
|
#include <gst/gstutils.h> /* GST_READ_UINT16_BE */
|
|
#include <gst/base/gstbitreader.h> /* GstBitReader */
|
|
|
|
#include "dvb-sub.h"
|
|
|
|
GST_DEBUG_CATEGORY_STATIC (dvbsub_debug);
|
|
#define GST_CAT_DEFAULT dvbsub_debug
|
|
|
|
static void dvb_sub_init (void);
|
|
|
|
/* 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 AYUV(y,u,v,a) (((a) << 24) | ((y) << 16) | ((u) << 8) | (v))
|
|
#define RGBA_TO_AYUV(r,g,b,a) (((a) << 24) | ((rgb_to_y(r,g,b)) << 16) | ((rgb_to_u(r,g,b)) << 8) | (rgb_to_v(r,g,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;
|
|
|
|
struct _DvbSub
|
|
{
|
|
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;
|
|
DVBSubtitleWindow display_def;
|
|
};
|
|
|
|
typedef enum
|
|
{
|
|
TOP_FIELD = 0,
|
|
BOTTOM_FIELD = 1
|
|
} DvbSubPixelDataSubBlockFieldType;
|
|
|
|
static inline gint
|
|
rgb_to_y (gint r, gint g, gint b)
|
|
{
|
|
gint ret;
|
|
|
|
ret = (gint) (((19595 * r) >> 16) + ((38470 * g) >> 16) + ((7471 * b) >> 16));
|
|
ret = CLAMP (ret, 0, 255);
|
|
return ret;
|
|
}
|
|
|
|
static inline gint
|
|
rgb_to_u (gint r, gint g, gint b)
|
|
{
|
|
gint ret;
|
|
|
|
ret =
|
|
(gint) (-((11059 * r) >> 16) - ((21709 * g) >> 16) + ((32768 * b) >> 16) +
|
|
128);
|
|
ret = CLAMP (ret, 0, 255);
|
|
return ret;
|
|
}
|
|
|
|
static inline gint
|
|
rgb_to_v (gint r, gint g, gint b)
|
|
{
|
|
gint ret;
|
|
|
|
ret =
|
|
(gint) (((32768 * r) >> 16) - ((27439 * g) >> 16) - ((5329 * b) >> 16) +
|
|
128);
|
|
ret = CLAMP (ret, 0, 255);
|
|
return ret;
|
|
}
|
|
|
|
static DVBSubObject *
|
|
get_object (DvbSub * dvb_sub, guint16 object_id)
|
|
{
|
|
DVBSubObject *ptr = dvb_sub->object_list;
|
|
|
|
while (ptr && ptr->id != object_id) {
|
|
ptr = ptr->next;
|
|
}
|
|
|
|
return ptr;
|
|
}
|
|
|
|
static DVBSubCLUT *
|
|
get_clut (DvbSub * dvb_sub, gint clut_id)
|
|
{
|
|
DVBSubCLUT *ptr = dvb_sub->clut_list;
|
|
|
|
while (ptr && ptr->id != clut_id) {
|
|
ptr = ptr->next;
|
|
}
|
|
|
|
return ptr;
|
|
}
|
|
|
|
static DVBSubRegion *
|
|
get_region (DvbSub * dvb_sub, guint8 region_id)
|
|
{
|
|
DVBSubRegion *ptr = dvb_sub->region_list;
|
|
|
|
while (ptr && ptr->id != region_id) {
|
|
ptr = ptr->next;
|
|
}
|
|
|
|
return ptr;
|
|
}
|
|
|
|
static void
|
|
delete_region_display_list (DvbSub * dvb_sub, DVBSubRegion * region)
|
|
{
|
|
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 **) & dvb_sub->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)
|
|
{
|
|
DVBSubRegion *region;
|
|
|
|
while (dvb_sub->region_list) {
|
|
region = dvb_sub->region_list;
|
|
|
|
dvb_sub->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, dvb_sub->clut_list, next);
|
|
dvb_sub->clut_list = NULL;
|
|
|
|
/* Should already be NULL */
|
|
g_warn_if_fail (dvb_sub->object_list == NULL);
|
|
}
|
|
|
|
static void
|
|
dvb_sub_init (void)
|
|
{
|
|
int i, r, g, b, a = 0;
|
|
|
|
GST_DEBUG_CATEGORY_INIT (dvbsub_debug, "dvbsub", 0, "dvbsuboverlay parser");
|
|
|
|
/* 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_TO_AYUV (0, 0, 0, 0);
|
|
default_clut.clut4[1] = RGBA_TO_AYUV (255, 255, 255, 255);
|
|
default_clut.clut4[2] = RGBA_TO_AYUV (0, 0, 0, 255);
|
|
default_clut.clut4[3] = RGBA_TO_AYUV (127, 127, 127, 255);
|
|
|
|
default_clut.clut16[0] = RGBA_TO_AYUV (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_TO_AYUV (r, g, b, 255);
|
|
}
|
|
|
|
default_clut.clut256[0] = RGBA_TO_AYUV (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_TO_AYUV (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? */
|
|
DVBSubRegionDisplay *display;
|
|
DVBSubRegionDisplay *tmp_display_list, **tmp_ptr;
|
|
|
|
const guint8 *buf_end = buf + buf_size;
|
|
guint8 region_id;
|
|
guint8 page_state;
|
|
|
|
if (buf_size < 1)
|
|
return;
|
|
|
|
dvb_sub->page_time_out = *buf++;
|
|
page_state = ((*buf++) >> 2) & 3;
|
|
|
|
#ifndef GST_DISABLE_GST_DEBUG
|
|
{
|
|
static const gchar *page_state_str[4] = {
|
|
"Normal case", "ACQUISITION POINT", "Mode Change", "RESERVED"
|
|
};
|
|
|
|
GST_DEBUG ("PAGE: page_id = %u, length = %d, page_time_out = %u secs, "
|
|
"page_state = %s", page_id, buf_size, dvb_sub->page_time_out,
|
|
page_state_str[page_state]);
|
|
}
|
|
#endif
|
|
|
|
if (page_state == 2) { /* Mode change */
|
|
delete_state (dvb_sub);
|
|
}
|
|
|
|
tmp_display_list = dvb_sub->display_list;
|
|
dvb_sub->display_list = NULL;
|
|
dvb_sub->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 = dvb_sub->display_list;
|
|
dvb_sub->display_list = display;
|
|
dvb_sub->display_list_size++;
|
|
|
|
GST_LOG ("PAGE: REGION information: ID = %u, address = %ux%u", 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)
|
|
{
|
|
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 = dvb_sub->region_list;
|
|
dvb_sub->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) {
|
|
GST_WARNING ("region depth %d is invalid", 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);
|
|
}
|
|
|
|
GST_DEBUG ("REGION: id = %u, (%ux%u)@%u-bit", region_id, region->width,
|
|
region->height, region->depth);
|
|
|
|
if (fill) {
|
|
memset (region->pbuf, region->bgcolor, region->buf_size);
|
|
GST_DEBUG ("REGION: 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 = dvb_sub->object_list;
|
|
dvb_sub->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;
|
|
|
|
GST_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) {
|
|
GST_DEBUG ("REGION DATA: fgcolor = %u, bgcolor = %u",
|
|
object_display->fgcolor, object_display->bgcolor);
|
|
}
|
|
}
|
|
}
|
|
|
|
static void
|
|
_dvb_sub_parse_clut_segment (DvbSub * dvb_sub, guint16 page_id, guint8 * buf,
|
|
gint buf_size)
|
|
{
|
|
const guint8 *buf_end = buf + buf_size;
|
|
guint8 clut_id;
|
|
DVBSubCLUT *clut;
|
|
int entry_id, depth, full_range;
|
|
int y, cr, cb, alpha;
|
|
|
|
GST_MEMDUMP ("DVB clut packet", buf, buf_size);
|
|
|
|
clut_id = *buf++;
|
|
buf += 1;
|
|
|
|
clut = get_clut (dvb_sub, clut_id);
|
|
|
|
if (!clut) {
|
|
clut = g_slice_new (DVBSubCLUT);
|
|
|
|
memcpy (clut, &default_clut, sizeof (DVBSubCLUT));
|
|
|
|
clut->id = clut_id;
|
|
|
|
clut->next = dvb_sub->clut_list;
|
|
dvb_sub->clut_list = clut;
|
|
}
|
|
|
|
while (buf + 4 < buf_end) {
|
|
entry_id = *buf++;
|
|
|
|
depth = (*buf) & 0xe0;
|
|
|
|
if (depth == 0) {
|
|
GST_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;
|
|
|
|
GST_DEBUG ("CLUT DEFINITION: clut %d := (%d,%d,%d,%d)", entry_id, y, cb, cr,
|
|
alpha);
|
|
|
|
if (depth & 0x80)
|
|
clut->clut4[entry_id] = AYUV (y, cb, cr, 255 - alpha);
|
|
if (depth & 0x40)
|
|
clut->clut16[entry_id] = AYUV (y, cb, cr, 255 - alpha);
|
|
if (depth & 0x20)
|
|
clut->clut256[entry_id] = AYUV (y, cb, cr, 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;
|
|
}
|
|
|
|
GST_TRACE ("dbuf_len = %d", dbuf_len);
|
|
|
|
while (!stop_parsing && (gst_bit_reader_get_remaining (&gb) > 0)) {
|
|
guint run_length = 0, clut_index = 0;
|
|
|
|
bits = gst_bit_reader_get_bits_uint32_unchecked (&gb, 2);
|
|
|
|
if (bits) { /* 2-bit_pixel-code */
|
|
run_length = 1;
|
|
clut_index = bits;
|
|
} else { /* 2-bit_zero */
|
|
bits = gst_bit_reader_get_bits_uint32_unchecked (&gb, 1);
|
|
if (bits == 1) { /* switch_1 == '1' */
|
|
run_length = gst_bit_reader_get_bits_uint32_unchecked (&gb, 3);
|
|
run_length += 3;
|
|
clut_index = gst_bit_reader_get_bits_uint32_unchecked (&gb, 2);
|
|
} else { /* switch_1 == '0' */
|
|
bits = gst_bit_reader_get_bits_uint32_unchecked (&gb, 1);
|
|
if (bits == 1) { /* switch_2 == '1' */
|
|
run_length = 1; /* 1x pseudo-colour '00' */
|
|
} else { /* switch_2 == '0' */
|
|
bits = gst_bit_reader_get_bits_uint32_unchecked (&gb, 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 */
|
|
run_length = gst_bit_reader_get_bits_uint32_unchecked (&gb, 4);
|
|
run_length += 12;
|
|
clut_index = gst_bit_reader_get_bits_uint32_unchecked (&gb, 2);
|
|
break;
|
|
case 0x3: /* the following 10 bits contain run length coded pixel data */
|
|
run_length = gst_bit_reader_get_bits_uint32_unchecked (&gb, 8);
|
|
run_length += 29;
|
|
clut_index = gst_bit_reader_get_bits_uint32_unchecked (&gb, 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 */
|
|
GST_TRACE ("RUNLEN: 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 && clut_index == 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;
|
|
|
|
GST_TRACE ("PIXEL: returning, read %u pixels", 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;
|
|
|
|
GST_TRACE ("RUNLEN: srcbuf position %p, 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;
|
|
|
|
bits = gst_bit_reader_get_bits_uint32_unchecked (&gb, 4);
|
|
|
|
if (bits) {
|
|
run_length = 1;
|
|
clut_index = bits;
|
|
} else {
|
|
bits = gst_bit_reader_get_bits_uint32_unchecked (&gb, 1);
|
|
if (bits == 0) { /* switch_1 == '0' */
|
|
run_length = gst_bit_reader_get_bits_uint32_unchecked (&gb, 3);
|
|
if (!run_length) {
|
|
stop_parsing = TRUE;
|
|
} else {
|
|
run_length += 2;
|
|
}
|
|
} else { /* switch_1 == '1' */
|
|
bits = gst_bit_reader_get_bits_uint32_unchecked (&gb, 1);
|
|
if (bits == 0) { /* switch_2 == '0' */
|
|
run_length = gst_bit_reader_get_bits_uint32_unchecked (&gb, 2);
|
|
run_length += 4;
|
|
clut_index = gst_bit_reader_get_bits_uint32_unchecked (&gb, 4);
|
|
} else { /* switch_2 == '1' */
|
|
bits = gst_bit_reader_get_bits_uint32_unchecked (&gb, 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' */
|
|
run_length = gst_bit_reader_get_bits_uint32_unchecked (&gb, 4);
|
|
run_length += 9;
|
|
clut_index = gst_bit_reader_get_bits_uint32_unchecked (&gb, 4);
|
|
break;
|
|
case 0x3: /* switch_3 == '11' */
|
|
run_length = gst_bit_reader_get_bits_uint32_unchecked (&gb, 8);
|
|
run_length += 25;
|
|
clut_index = gst_bit_reader_get_bits_uint32_unchecked (&gb, 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 */
|
|
GST_TRACE ("RUNLEN: 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 && clut_index == 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;
|
|
|
|
GST_LOG ("Returning 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;
|
|
}
|
|
|
|
GST_LOG ("dbuf_len = %d", 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;
|
|
bits = gst_bit_reader_get_bits_uint32_unchecked (&gb, 8);
|
|
|
|
if (bits) { /* 8-bit_pixel-code */
|
|
run_length = 1;
|
|
clut_index = bits;
|
|
} else { /* 8-bit_zero */
|
|
bits = gst_bit_reader_get_bits_uint32_unchecked (&gb, 1);
|
|
if (bits == 0) { /* switch_1 == '0' */
|
|
/* run_length_1-127 for pseudo-colour _entry) '0x00' */
|
|
run_length = gst_bit_reader_get_bits_uint32_unchecked (&gb, 7);
|
|
if (run_length == 0) { /* end_of_string_signal */
|
|
stop_parsing = TRUE;
|
|
}
|
|
} else { /* switch_1 == '1' */
|
|
/* run_length_3-127 */
|
|
run_length = gst_bit_reader_get_bits_uint32_unchecked (&gb, 7);
|
|
clut_index = gst_bit_reader_get_bits_uint32_unchecked (&gb, 8);
|
|
|
|
if (run_length < 3) {
|
|
GST_WARNING ("runlength value was %u, but the spec requires it "
|
|
"must be >=3", run_length);
|
|
}
|
|
}
|
|
}
|
|
|
|
/* 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 */
|
|
GST_TRACE ("RUNLEN: 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 && clut_index == 1))
|
|
memset (destbuf, clut_index, run_length);
|
|
|
|
destbuf += run_length;
|
|
pixels_read += run_length;
|
|
}
|
|
|
|
GST_LOG ("Returning 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;
|
|
|
|
GST_LOG ("DVB pixel block size %d, %s field:", buf_size,
|
|
top_bottom ? "bottom" : "top");
|
|
|
|
GST_MEMDUMP ("packet", buf, buf_size);
|
|
|
|
if (region == NULL) {
|
|
GST_LOG ("Region is NULL, returning");
|
|
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) {
|
|
GST_LOG ("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", (guint) (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) {
|
|
/* FIXME: Be more verbose */
|
|
GST_WARNING ("Invalid object location for data_type 0x%x!",
|
|
*(buf - 1));
|
|
GST_MEMDUMP ("Remaining data after invalid object location:", buf,
|
|
(guint) (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) {
|
|
/* FIXME: Be more verbose */
|
|
GST_WARNING ("Invalid object location for data_type 0x%x!",
|
|
*(buf - 1));
|
|
GST_MEMDUMP ("Remaining data after invalid object location:", 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) {
|
|
GST_WARNING ("4-bit pixel string in %d-bit region!", region->depth);
|
|
return;
|
|
}
|
|
|
|
if (region->depth == 8)
|
|
map_table = map4to8;
|
|
else
|
|
map_table = NULL;
|
|
|
|
GST_LOG ("READ_4BIT_STRING: String data into position %dx%d; "
|
|
"buf before is %p", 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);
|
|
GST_DEBUG ("READ_4BIT_STRING finished: buf pointer now %p", buf);
|
|
break;
|
|
case 0x12:
|
|
if (dest_buf_filled) {
|
|
/* FIXME: Be more verbose */
|
|
GST_WARNING ("Invalid object location for data_type 0x%x!",
|
|
*(buf - 1));
|
|
GST_MEMDUMP ("Remaining data after invalid object location:",
|
|
buf, (guint) (buf_end - buf));
|
|
return;
|
|
}
|
|
|
|
if (region->depth < 8) {
|
|
GST_WARNING ("8-bit pixel string in %d-bit region!", 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:
|
|
GST_DEBUG ("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:
|
|
GST_DEBUG ("handling map2to8 table data");
|
|
for (i = 0; i < 4; i++)
|
|
map2to8[i] = *buf++;
|
|
break;
|
|
case 0x22:
|
|
GST_DEBUG ("handling map4to8 table data");
|
|
for (i = 0; i < 16; i++)
|
|
map4to8[i] = *buf++;
|
|
break;
|
|
case 0xf0:
|
|
GST_DEBUG ("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? */
|
|
GST_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);
|
|
|
|
GST_DEBUG ("OBJECT: a new object segment has occurred for object_id = %u",
|
|
object_id);
|
|
|
|
if (!object) {
|
|
GST_WARNING ("Nothing known about object with ID %u yet, 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) {
|
|
GST_WARNING ("Field data size too large");
|
|
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;
|
|
|
|
GST_DEBUG ("OBJECT: 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) {
|
|
GST_FIXME ("'a string of characters' coding method not supported yet!");
|
|
} else {
|
|
GST_WARNING ("Unknown object coding 0x%x", coding_method);
|
|
}
|
|
}
|
|
|
|
static gint
|
|
_dvb_sub_parse_display_definition_segment (DvbSub * dvb_sub, guint8 * buf,
|
|
gint buf_size)
|
|
{
|
|
int dds_version, info_byte;
|
|
|
|
if (buf_size < 5)
|
|
return -1;
|
|
|
|
info_byte = *buf++;
|
|
dds_version = info_byte >> 4;
|
|
|
|
if (dvb_sub->display_def.version == dds_version)
|
|
return 0; /* already have this display definition version */
|
|
|
|
dvb_sub->display_def.version = dds_version;
|
|
dvb_sub->display_def.display_width = GST_READ_UINT16_BE (buf) + 1;
|
|
buf += 2;
|
|
dvb_sub->display_def.display_height = GST_READ_UINT16_BE (buf) + 1;
|
|
buf += 2;
|
|
|
|
dvb_sub->display_def.window_flag = info_byte & 1 << 3;
|
|
|
|
if (buf_size >= 13 && dvb_sub->display_def.window_flag) {
|
|
dvb_sub->display_def.window_x = GST_READ_UINT16_BE (buf);
|
|
buf += 2;
|
|
dvb_sub->display_def.window_y = GST_READ_UINT16_BE (buf);
|
|
buf += 2;
|
|
dvb_sub->display_def.window_width =
|
|
GST_READ_UINT16_BE (buf) - dvb_sub->display_def.window_x + 1;
|
|
buf += 2;
|
|
dvb_sub->display_def.window_height =
|
|
GST_READ_UINT16_BE (buf) - dvb_sub->display_def.window_y + 1;
|
|
buf += 2;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static gint
|
|
_dvb_sub_parse_end_of_display_set (DvbSub * dvb_sub, guint16 page_id,
|
|
guint64 pts)
|
|
{
|
|
DVBSubRegionDisplay *display;
|
|
DVBSubtitles *sub;
|
|
DVBSubCLUT *clut;
|
|
guint32 *clut_table;
|
|
int i;
|
|
|
|
GST_DEBUG ("DISPLAY SET END: page_id = %u", page_id);
|
|
|
|
sub = g_slice_new0 (DVBSubtitles);
|
|
|
|
#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
|
|
|
|
/* N.B. g_new0() will return NULL if num_rects is 0 */
|
|
sub->num_rects = dvb_sub->display_list_size;
|
|
sub->rects = g_new0 (DVBSubtitleRect, sub->num_rects);
|
|
|
|
i = 0;
|
|
|
|
/* copy subtitle display and window information */
|
|
sub->display_def = dvb_sub->display_def;
|
|
|
|
for (display = dvb_sub->display_list; display; display = display->next) {
|
|
DVBSubtitleRect *rect;
|
|
DVBSubRegion *region;
|
|
|
|
region = get_region (dvb_sub, display->region_id);
|
|
|
|
if (!region)
|
|
continue;
|
|
|
|
rect = &sub->rects[i];
|
|
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));
|
|
|
|
GST_MEMDUMP ("rect->pict.data.palette content",
|
|
(guint8 *) rect->pict.palette, (1 << region->depth) * sizeof (guint32));
|
|
|
|
rect->pict.data = g_malloc (region->buf_size); /* FIXME: Can we use GSlice here? */
|
|
memcpy (rect->pict.data, region->pbuf, region->buf_size);
|
|
|
|
GST_DEBUG ("DISPLAY: an object rect created: iteration %u, "
|
|
"pos: %d:%d, size: %dx%d", i, rect->x, rect->y, rect->w, rect->h);
|
|
|
|
GST_MEMDUMP ("rect->pict.data content", rect->pict.data, region->buf_size);
|
|
|
|
++i;
|
|
}
|
|
|
|
sub->pts = pts;
|
|
sub->page_time_out = dvb_sub->page_time_out;
|
|
sub->num_rects = i;
|
|
|
|
if (dvb_sub->callbacks.new_data) {
|
|
dvb_sub->callbacks.new_data (dvb_sub, sub, dvb_sub->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;
|
|
|
|
if (sub == NULL)
|
|
return;
|
|
|
|
/* Now free up all the temporary memory we allocated */
|
|
for (i = 0; i < sub->num_rects; ++i) {
|
|
g_free (sub->rects[i].pict.palette);
|
|
g_free (sub->rects[i].pict.data);
|
|
}
|
|
g_free (sub->rects);
|
|
g_slice_free (DVBSubtitles, sub);
|
|
}
|
|
|
|
DvbSub *
|
|
dvb_sub_new (void)
|
|
{
|
|
static gsize inited = 0;
|
|
DvbSub *sub;
|
|
|
|
if (g_once_init_enter (&inited)) {
|
|
dvb_sub_init ();
|
|
g_once_init_leave (&inited, TRUE);
|
|
}
|
|
|
|
sub = g_slice_new0 (DvbSub);
|
|
|
|
/* TODO: Add initialization code here */
|
|
/* FIXME: Do we have a reason to initiate the members to zero, or are we guaranteed that anyway? */
|
|
sub->region_list = NULL;
|
|
sub->object_list = NULL;
|
|
sub->page_time_out = 0; /* FIXME: Maybe 255 instead? */
|
|
sub->pes_buffer = g_string_new (NULL);
|
|
|
|
/* display/window information */
|
|
sub->display_def.version = -1;
|
|
sub->display_def.window_flag = 0;
|
|
sub->display_def.display_width = 720;
|
|
sub->display_def.display_height = 576;
|
|
|
|
return sub;
|
|
}
|
|
|
|
void
|
|
dvb_sub_free (DvbSub * sub)
|
|
{
|
|
/* TODO: Add deinitalization code here */
|
|
/* FIXME: Clear up region_list contents */
|
|
delete_state (sub);
|
|
while (sub->display_list) {
|
|
DVBSubRegionDisplay *tmp = sub->display_list->next;
|
|
g_slice_free (DVBSubRegionDisplay, sub->display_list);
|
|
sub->display_list = tmp;
|
|
}
|
|
g_string_free (sub->pes_buffer, TRUE);
|
|
g_slice_free (DvbSub, sub);
|
|
}
|
|
|
|
#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_DISPLAY_DEFINITION 0x14
|
|
#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;
|
|
|
|
GST_DEBUG ("pts=%" G_GUINT64_FORMAT " and length %d", pts, len);
|
|
|
|
g_return_val_if_fail (data != NULL || len == 0, -1);
|
|
|
|
if (G_UNLIKELY (data == NULL)) {
|
|
GST_DEBUG ("no data; forcing end-of-display-set");
|
|
_dvb_sub_parse_end_of_display_set (dvb_sub, 0, pts);
|
|
return 0;
|
|
}
|
|
|
|
if (len <= 3) { /* len(0x20 0x00 end_of_PES_data_field_marker) */
|
|
GST_WARNING ("Data length too short");
|
|
return -1;
|
|
}
|
|
|
|
if (data[pos++] != 0x20) {
|
|
GST_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) {
|
|
GST_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)) {
|
|
GST_WARNING ("Data after SYNC BYTE too short, less than needed to "
|
|
"even get to segment_length");
|
|
return -2;
|
|
}
|
|
segment_type = data[pos++];
|
|
GST_DEBUG ("=== Segment type is 0x%x", segment_type);
|
|
page_id = (data[pos] << 8) | data[pos + 1];
|
|
GST_DEBUG ("page_id is 0x%x", page_id);
|
|
pos += 2;
|
|
segment_len = (data[pos] << 8) | data[pos + 1];
|
|
GST_DEBUG ("segment_length is %d (0x%x 0x%x)", segment_len, data[pos],
|
|
data[pos + 1]);
|
|
pos += 2;
|
|
if ((len - pos) < segment_len) {
|
|
GST_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:
|
|
GST_DEBUG ("Page composition segment at buffer pos %u", 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:
|
|
GST_DEBUG ("Region composition segment at buffer pos %u", 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:
|
|
GST_DEBUG ("CLUT definition segment at buffer pos %u", 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:
|
|
GST_DEBUG ("Object data segment at buffer pos %u", pos);
|
|
_dvb_sub_parse_object_segment (dvb_sub, page_id, data + pos, segment_len); /* FIXME: Not sure about args */
|
|
break;
|
|
case DVB_SUB_SEGMENT_DISPLAY_DEFINITION:
|
|
GST_DEBUG ("display definition segment at buffer pos %u", pos);
|
|
_dvb_sub_parse_display_definition_segment (dvb_sub, data + pos,
|
|
segment_len);
|
|
break;
|
|
case DVB_SUB_SEGMENT_END_OF_DISPLAY_SET:
|
|
GST_DEBUG ("End of display set at buffer pos %u", pos);
|
|
_dvb_sub_parse_end_of_display_set (dvb_sub, page_id, pts); /* FIXME: Not sure about args */
|
|
break;
|
|
default:
|
|
GST_FIXME ("Unhandled segment type 0x%x", segment_type);
|
|
break;
|
|
}
|
|
|
|
pos += segment_len;
|
|
|
|
if (pos == len) {
|
|
GST_WARNING ("Data ended without a PES data end marker");
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
GST_LOG ("Processed %d bytes out of %d", 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)
|
|
{
|
|
g_return_if_fail (dvb_sub != NULL);
|
|
g_return_if_fail (callbacks != NULL);
|
|
|
|
dvb_sub->callbacks = *callbacks;
|
|
dvb_sub->user_data = user_data;
|
|
}
|