mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-09-09 05:38:58 +00:00
1150 lines
33 KiB
C
1150 lines
33 KiB
C
/*
|
|
* GStreamer
|
|
* Copyright (C) 2009 Sebastian Pölsterl <sebp@k-d-w.org>
|
|
* Copyright (C) 2010 Andoni Morales Alastruey <ylatuya@gmail.com>
|
|
*
|
|
* This library 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.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor,
|
|
* Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
/**
|
|
* SECTION:element-teletextdec
|
|
* @title: teletextdec
|
|
*
|
|
* Decode a stream of raw VBI packets containing teletext information to a RGBA
|
|
* stream.
|
|
*
|
|
* ## Example launch line
|
|
* |[
|
|
* gst-launch-1.0 -v -m filesrc location=recording.mpeg ! tsdemux ! teletextdec ! videoconvert ! ximagesink
|
|
* ]|
|
|
*
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <gst/gst.h>
|
|
#include <gst/video/video.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "gstteletextdec.h"
|
|
|
|
GST_DEBUG_CATEGORY_STATIC (gst_teletextdec_debug);
|
|
#define GST_CAT_DEFAULT gst_teletextdec_debug
|
|
|
|
#define parent_class gst_teletextdec_parent_class
|
|
|
|
#define SUBTITLES_PAGE 888
|
|
#define MAX_SLICES 32
|
|
#define DEFAULT_FONT_DESCRIPTION "verdana 12"
|
|
#define PANGO_TEMPLATE "<span font_desc=\"%s\" foreground=\"%s\"> %s \n</span>"
|
|
|
|
/* Filter signals and args */
|
|
enum
|
|
{
|
|
LAST_SIGNAL
|
|
};
|
|
|
|
enum
|
|
{
|
|
PROP_0,
|
|
PROP_PAGENO,
|
|
PROP_SUBNO,
|
|
PROP_SUBTITLES_MODE,
|
|
PROP_SUBS_TEMPLATE,
|
|
PROP_FONT_DESCRIPTION
|
|
};
|
|
|
|
enum
|
|
{
|
|
VBI_ERROR = -1,
|
|
VBI_SUCCESS = 0,
|
|
VBI_NEW_FRAME = 1
|
|
};
|
|
|
|
typedef enum
|
|
{
|
|
DATA_UNIT_EBU_TELETEXT_NON_SUBTITLE = 0x02,
|
|
DATA_UNIT_EBU_TELETEXT_SUBTITLE = 0x03,
|
|
DATA_UNIT_EBU_TELETEXT_INVERTED = 0x0C,
|
|
|
|
DATA_UNIT_ZVBI_WSS_CPR1204 = 0xB4,
|
|
DATA_UNIT_ZVBI_CLOSED_CAPTION_525 = 0xB5,
|
|
DATA_UNIT_ZVBI_MONOCHROME_SAMPLES_525 = 0xB6,
|
|
|
|
DATA_UNIT_VPS = 0xC3,
|
|
DATA_UNIT_WSS = 0xC4,
|
|
DATA_UNIT_CLOSED_CAPTION = 0xC5,
|
|
DATA_UNIT_MONOCHROME_SAMPLES = 0xC6,
|
|
|
|
DATA_UNIT_STUFFING = 0xFF,
|
|
} data_unit_id;
|
|
|
|
typedef struct
|
|
{
|
|
int pgno;
|
|
int subno;
|
|
} page_info;
|
|
|
|
typedef enum
|
|
{
|
|
SYSTEM_525 = 0,
|
|
SYSTEM_625
|
|
} systems;
|
|
|
|
/*
|
|
* ETS 300 706 Table 30: Colour Map
|
|
*/
|
|
static const gchar *default_color_map[40] = {
|
|
"#000000", "#FF0000", "#00FF00", "#FFFF00", "#0000FF",
|
|
"#FF00FF", "#00FFFF", "#FFFFFF", "#000000", "#770000",
|
|
"#007700", "#777700", "#000077", "#770077", "#007777",
|
|
"#777777", "#FF0055", "#FF7700", "#00FF77", "#FFFFBB",
|
|
"#00CCAA", "#550000", "#665522", "#CC7777", "#333333",
|
|
"#FF7777", "#77FF77", "#FFFF77", "#7777FF", "#FF77FF",
|
|
"#77FFFF", "#DDD0DD",
|
|
|
|
/* Private colors */
|
|
"#000000", "#FFAA99", "#44EE00", "#FFDD00", "#FFAA99",
|
|
"#FF00FF", "#00FFFF", "#EEEEEE"
|
|
};
|
|
|
|
/* in RGBA mode, one character occupies 12 x 10 pixels. */
|
|
#define COLUMNS_TO_WIDTH(cols) ((cols) * 12)
|
|
#define ROWS_TO_HEIGHT(rows) ((rows) * 10)
|
|
|
|
static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink",
|
|
GST_PAD_SINK,
|
|
GST_PAD_ALWAYS,
|
|
GST_STATIC_CAPS ("application/x-teletext;")
|
|
);
|
|
|
|
static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src",
|
|
GST_PAD_SRC,
|
|
GST_PAD_ALWAYS,
|
|
GST_STATIC_CAPS
|
|
(GST_VIDEO_CAPS_MAKE ("RGBA") ";"
|
|
"text/x-raw, format={utf-8,pango-markup} ;")
|
|
);
|
|
|
|
G_DEFINE_TYPE (GstTeletextDec, gst_teletextdec, GST_TYPE_ELEMENT);
|
|
|
|
static void gst_teletextdec_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec);
|
|
static void gst_teletextdec_get_property (GObject * object, guint prop_id,
|
|
GValue * value, GParamSpec * pspec);
|
|
static void gst_teletextdec_finalize (GObject * object);
|
|
|
|
static GstStateChangeReturn gst_teletextdec_change_state (GstElement *
|
|
element, GstStateChange transition);
|
|
|
|
static GstFlowReturn gst_teletextdec_chain (GstPad * pad, GstObject * parent,
|
|
GstBuffer * buffer);
|
|
static gboolean gst_teletextdec_sink_event (GstPad * pad, GstObject * parent,
|
|
GstEvent * event);
|
|
static gboolean gst_teletextdec_src_event (GstPad * pad, GstObject * parent,
|
|
GstEvent * event);
|
|
|
|
static void gst_teletextdec_event_handler (vbi_event * ev, void *user_data);
|
|
|
|
static GstFlowReturn gst_teletextdec_push_page (GstTeletextDec * teletext);
|
|
static GstFlowReturn gst_teletextdec_export_text_page (GstTeletextDec *
|
|
teletext, vbi_page * page, GstBuffer ** buf);
|
|
static GstFlowReturn gst_teletextdec_export_rgba_page (GstTeletextDec *
|
|
teletext, vbi_page * page, GstBuffer ** buf);
|
|
static GstFlowReturn gst_teletextdec_export_pango_page (GstTeletextDec *
|
|
teletext, vbi_page * page, GstBuffer ** buf);
|
|
|
|
static void gst_teletextdec_process_telx_buffer (GstTeletextDec * teletext,
|
|
GstBuffer * buf);
|
|
static gboolean gst_teletextdec_extract_data_units (GstTeletextDec *
|
|
teletext, GstTeletextFrame * f, const guint8 * packet, guint * offset,
|
|
gsize size);
|
|
|
|
static void gst_teletextdec_zvbi_init (GstTeletextDec * teletext);
|
|
static void gst_teletextdec_zvbi_clear (GstTeletextDec * teletext);
|
|
static void gst_teletextdec_reset_frame (GstTeletextDec * teletext);
|
|
|
|
/* initialize the gstteletext's class */
|
|
static void
|
|
gst_teletextdec_class_init (GstTeletextDecClass * klass)
|
|
{
|
|
GObjectClass *gobject_class;
|
|
GstElementClass *gstelement_class;
|
|
|
|
gobject_class = G_OBJECT_CLASS (klass);
|
|
gobject_class->set_property = gst_teletextdec_set_property;
|
|
gobject_class->get_property = gst_teletextdec_get_property;
|
|
gobject_class->finalize = gst_teletextdec_finalize;
|
|
|
|
gstelement_class = GST_ELEMENT_CLASS (klass);
|
|
gstelement_class->change_state =
|
|
GST_DEBUG_FUNCPTR (gst_teletextdec_change_state);
|
|
|
|
g_object_class_install_property (gobject_class, PROP_PAGENO,
|
|
g_param_spec_int ("page", "Page number",
|
|
"Number of page that should displayed",
|
|
100, 999, 100, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
g_object_class_install_property (gobject_class, PROP_SUBNO,
|
|
g_param_spec_int ("subpage", "Sub-page number",
|
|
"Number of sub-page that should displayed (-1 for all)",
|
|
-1, 0x99, -1, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
g_object_class_install_property (gobject_class, PROP_SUBTITLES_MODE,
|
|
g_param_spec_boolean ("subtitles-mode", "Enable subtitles mode",
|
|
"Enables subtitles mode for text output stripping the blank lines and "
|
|
"the teletext state lines", FALSE,
|
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
g_object_class_install_property (gobject_class, PROP_SUBS_TEMPLATE,
|
|
g_param_spec_string ("subtitles-template", "Subtitles output template",
|
|
"Output template used to print each one of the subtitles lines",
|
|
g_strescape ("%s\n", NULL),
|
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
g_object_class_install_property (gobject_class, PROP_FONT_DESCRIPTION,
|
|
g_param_spec_string ("font-description", "Pango font description",
|
|
"Font description used for the pango output.",
|
|
DEFAULT_FONT_DESCRIPTION,
|
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
|
|
|
|
gst_element_class_set_static_metadata (gstelement_class,
|
|
"Teletext decoder",
|
|
"Decoder",
|
|
"Decode a raw VBI stream containing teletext information to RGBA and text",
|
|
"Sebastian Pölsterl <sebp@k-d-w.org>, "
|
|
"Andoni Morales Alastruey <ylatuya@gmail.com>");
|
|
|
|
gst_element_class_add_static_pad_template (gstelement_class, &src_template);
|
|
gst_element_class_add_static_pad_template (gstelement_class, &sink_template);
|
|
}
|
|
|
|
/* initialize the new element
|
|
* initialize instance structure
|
|
*/
|
|
static void
|
|
gst_teletextdec_init (GstTeletextDec * teletext)
|
|
{
|
|
/* Create sink pad */
|
|
teletext->sinkpad = gst_pad_new_from_static_template (&sink_template, "sink");
|
|
gst_pad_set_chain_function (teletext->sinkpad,
|
|
GST_DEBUG_FUNCPTR (gst_teletextdec_chain));
|
|
gst_pad_set_event_function (teletext->sinkpad,
|
|
GST_DEBUG_FUNCPTR (gst_teletextdec_sink_event));
|
|
gst_element_add_pad (GST_ELEMENT (teletext), teletext->sinkpad);
|
|
|
|
/* Create src pad */
|
|
teletext->srcpad = gst_pad_new_from_static_template (&src_template, "src");
|
|
gst_pad_set_event_function (teletext->srcpad,
|
|
GST_DEBUG_FUNCPTR (gst_teletextdec_src_event));
|
|
gst_element_add_pad (GST_ELEMENT (teletext), teletext->srcpad);
|
|
|
|
teletext->segment = NULL;
|
|
teletext->decoder = NULL;
|
|
teletext->pageno = 0x100;
|
|
teletext->subno = -1;
|
|
teletext->subtitles_mode = FALSE;
|
|
teletext->subtitles_template = g_strescape ("%s\n", NULL);
|
|
teletext->font_description = g_strdup (DEFAULT_FONT_DESCRIPTION);
|
|
|
|
teletext->in_timestamp = GST_CLOCK_TIME_NONE;
|
|
teletext->in_duration = GST_CLOCK_TIME_NONE;
|
|
|
|
teletext->rate_numerator = 0;
|
|
teletext->rate_denominator = 1;
|
|
|
|
teletext->queue = NULL;
|
|
g_mutex_init (&teletext->queue_lock);
|
|
|
|
gst_teletextdec_reset_frame (teletext);
|
|
|
|
teletext->last_ts = 0;
|
|
|
|
teletext->export_func = NULL;
|
|
teletext->buf_pool = NULL;
|
|
}
|
|
|
|
static void
|
|
gst_teletextdec_finalize (GObject * object)
|
|
{
|
|
GstTeletextDec *teletext = GST_TELETEXTDEC (object);
|
|
|
|
g_mutex_clear (&teletext->queue_lock);
|
|
|
|
g_free (teletext->font_description);
|
|
g_free (teletext->subtitles_template);
|
|
g_free (teletext->frame);
|
|
|
|
G_OBJECT_CLASS (parent_class)->finalize (object);
|
|
}
|
|
|
|
static void
|
|
gst_teletextdec_zvbi_init (GstTeletextDec * teletext)
|
|
{
|
|
g_return_if_fail (teletext != NULL);
|
|
|
|
GST_LOG_OBJECT (teletext, "Initializing structures");
|
|
|
|
teletext->decoder = vbi_decoder_new ();
|
|
|
|
vbi_event_handler_register (teletext->decoder,
|
|
VBI_EVENT_TTX_PAGE | VBI_EVENT_CAPTION,
|
|
gst_teletextdec_event_handler, teletext);
|
|
|
|
g_mutex_lock (&teletext->queue_lock);
|
|
teletext->queue = g_queue_new ();
|
|
g_mutex_unlock (&teletext->queue_lock);
|
|
}
|
|
|
|
static void
|
|
gst_teletextdec_zvbi_clear (GstTeletextDec * teletext)
|
|
{
|
|
g_return_if_fail (teletext != NULL);
|
|
|
|
GST_LOG_OBJECT (teletext, "Clearing structures");
|
|
|
|
if (teletext->decoder != NULL) {
|
|
vbi_decoder_delete (teletext->decoder);
|
|
teletext->decoder = NULL;
|
|
}
|
|
if (teletext->frame != NULL) {
|
|
if (teletext->frame->sliced_begin)
|
|
g_free (teletext->frame->sliced_begin);
|
|
g_free (teletext->frame);
|
|
teletext->frame = NULL;
|
|
}
|
|
|
|
g_mutex_lock (&teletext->queue_lock);
|
|
if (teletext->queue != NULL) {
|
|
g_queue_free (teletext->queue);
|
|
teletext->queue = NULL;
|
|
}
|
|
g_mutex_unlock (&teletext->queue_lock);
|
|
|
|
teletext->in_timestamp = GST_CLOCK_TIME_NONE;
|
|
teletext->in_duration = GST_CLOCK_TIME_NONE;
|
|
teletext->pageno = 0x100;
|
|
teletext->subno = -1;
|
|
teletext->last_ts = 0;
|
|
}
|
|
|
|
static void
|
|
gst_teletextdec_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec)
|
|
{
|
|
GstTeletextDec *teletext = GST_TELETEXTDEC (object);
|
|
|
|
switch (prop_id) {
|
|
case PROP_PAGENO:
|
|
teletext->pageno = (gint) vbi_bin2bcd (g_value_get_int (value));
|
|
break;
|
|
case PROP_SUBNO:
|
|
teletext->subno = g_value_get_int (value);
|
|
break;
|
|
case PROP_SUBTITLES_MODE:
|
|
teletext->subtitles_mode = g_value_get_boolean (value);
|
|
break;
|
|
case PROP_SUBS_TEMPLATE:
|
|
g_free (teletext->subtitles_template);
|
|
teletext->subtitles_template = g_value_dup_string (value);
|
|
break;
|
|
case PROP_FONT_DESCRIPTION:
|
|
g_free (teletext->font_description);
|
|
teletext->font_description = g_value_dup_string (value);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void
|
|
gst_teletextdec_get_property (GObject * object, guint prop_id,
|
|
GValue * value, GParamSpec * pspec)
|
|
{
|
|
GstTeletextDec *teletext = GST_TELETEXTDEC (object);
|
|
|
|
switch (prop_id) {
|
|
case PROP_PAGENO:
|
|
g_value_set_int (value, (gint) vbi_bcd2dec (teletext->pageno));
|
|
break;
|
|
case PROP_SUBNO:
|
|
g_value_set_int (value, teletext->subno);
|
|
break;
|
|
case PROP_SUBTITLES_MODE:
|
|
g_value_set_boolean (value, teletext->subtitles_mode);
|
|
break;
|
|
case PROP_SUBS_TEMPLATE:
|
|
g_value_set_string (value, teletext->subtitles_template);
|
|
break;
|
|
case PROP_FONT_DESCRIPTION:
|
|
g_value_set_string (value, teletext->font_description);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static gboolean
|
|
gst_teletextdec_sink_event (GstPad * pad, GstObject * parent, GstEvent * event)
|
|
{
|
|
gboolean ret;
|
|
GstTeletextDec *teletext = GST_TELETEXTDEC (parent);
|
|
|
|
GST_DEBUG_OBJECT (teletext, "got event %s",
|
|
gst_event_type_get_name (GST_EVENT_TYPE (event)));
|
|
|
|
switch (GST_EVENT_TYPE (event)) {
|
|
case GST_EVENT_SEGMENT:
|
|
/* maybe save and/or update the current segment (e.g. for output
|
|
* clipping) or convert the event into one in a different format
|
|
* (e.g. BYTES to TIME) or drop it and set a flag to send a newsegment
|
|
* event in a different format later */
|
|
if (NULL == teletext->export_func) {
|
|
/* save the segment event and send it after sending caps. replace the
|
|
* old event if present. */
|
|
if (teletext->segment) {
|
|
gst_event_unref (teletext->segment);
|
|
}
|
|
teletext->segment = event;
|
|
ret = TRUE;
|
|
} else {
|
|
ret = gst_pad_push_event (teletext->srcpad, event);
|
|
}
|
|
break;
|
|
case GST_EVENT_EOS:
|
|
/* end-of-stream, we should close down all stream leftovers here */
|
|
gst_teletextdec_zvbi_clear (teletext);
|
|
ret = gst_pad_push_event (teletext->srcpad, event);
|
|
break;
|
|
case GST_EVENT_FLUSH_STOP:
|
|
gst_teletextdec_zvbi_clear (teletext);
|
|
gst_teletextdec_zvbi_init (teletext);
|
|
ret = gst_pad_push_event (teletext->srcpad, event);
|
|
break;
|
|
default:
|
|
ret = gst_pad_event_default (pad, parent, event);
|
|
break;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static gboolean
|
|
gst_teletextdec_src_event (GstPad * pad, GstObject * parent, GstEvent * event)
|
|
{
|
|
gboolean ret;
|
|
GstTeletextDec *teletext = GST_TELETEXTDEC (parent);
|
|
|
|
switch (GST_EVENT_TYPE (event)) {
|
|
case GST_EVENT_RECONFIGURE:
|
|
/* setting export_func to NULL will cause the element to renegotiate caps
|
|
* before pushing a buffer. */
|
|
teletext->export_func = NULL;
|
|
ret = TRUE;
|
|
break;
|
|
|
|
default:
|
|
ret = gst_pad_event_default (pad, parent, event);
|
|
break;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static GstStateChangeReturn
|
|
gst_teletextdec_change_state (GstElement * element, GstStateChange transition)
|
|
{
|
|
GstStateChangeReturn ret;
|
|
GstTeletextDec *teletext;
|
|
|
|
teletext = GST_TELETEXTDEC (element);
|
|
|
|
switch (transition) {
|
|
case GST_STATE_CHANGE_READY_TO_PAUSED:
|
|
gst_teletextdec_zvbi_init (teletext);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
|
|
if (ret != GST_STATE_CHANGE_SUCCESS)
|
|
return ret;
|
|
|
|
switch (transition) {
|
|
case GST_STATE_CHANGE_PAUSED_TO_READY:
|
|
gst_teletextdec_zvbi_clear (teletext);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static void
|
|
gst_teletextdec_reset_frame (GstTeletextDec * teletext)
|
|
{
|
|
if (teletext->frame == NULL)
|
|
teletext->frame = g_new0 (GstTeletextFrame, 1);
|
|
if (teletext->frame->sliced_begin == NULL)
|
|
teletext->frame->sliced_begin = g_new (vbi_sliced, MAX_SLICES);
|
|
teletext->frame->current_slice = teletext->frame->sliced_begin;
|
|
teletext->frame->sliced_end = teletext->frame->sliced_begin + MAX_SLICES;
|
|
teletext->frame->last_field = 0;
|
|
teletext->frame->last_field_line = 0;
|
|
teletext->frame->last_frame_line = 0;
|
|
}
|
|
|
|
static void
|
|
gst_teletextdec_process_telx_buffer (GstTeletextDec * teletext, GstBuffer * buf)
|
|
{
|
|
GstMapInfo buf_map;
|
|
guint offset = 0;
|
|
gint res;
|
|
gst_buffer_map (buf, &buf_map, GST_MAP_READ);
|
|
|
|
teletext->in_timestamp = GST_BUFFER_TIMESTAMP (buf);
|
|
teletext->in_duration = GST_BUFFER_DURATION (buf);
|
|
|
|
if (teletext->frame == NULL)
|
|
gst_teletextdec_reset_frame (teletext);
|
|
|
|
while (offset < buf_map.size) {
|
|
res =
|
|
gst_teletextdec_extract_data_units (teletext, teletext->frame,
|
|
buf_map.data, &offset, buf_map.size);
|
|
|
|
if (res == VBI_NEW_FRAME) {
|
|
/* We have a new frame, it's time to feed the decoder */
|
|
vbi_sliced *s;
|
|
gint n_lines;
|
|
|
|
n_lines = teletext->frame->current_slice - teletext->frame->sliced_begin;
|
|
GST_LOG_OBJECT (teletext, "Completed frame, decoding new %d lines",
|
|
n_lines);
|
|
s = g_memdup (teletext->frame->sliced_begin,
|
|
n_lines * sizeof (vbi_sliced));
|
|
vbi_decode (teletext->decoder, s, n_lines, teletext->last_ts);
|
|
/* From vbi_decode():
|
|
* timestamp shall advance by 1/30 to 1/25 seconds whenever calling this
|
|
* function. Failure to do so will be interpreted as frame dropping, which
|
|
* starts a resynchronization cycle, eventually a channel switch may be assumed
|
|
* which resets even more decoder state. So even if a frame did not contain
|
|
* any useful data this function must be called, with lines set to zero.
|
|
*/
|
|
teletext->last_ts += 0.04;
|
|
|
|
g_free (s);
|
|
gst_teletextdec_reset_frame (teletext);
|
|
} else if (res == VBI_ERROR) {
|
|
gst_teletextdec_reset_frame (teletext);
|
|
goto beach;
|
|
}
|
|
}
|
|
beach:
|
|
gst_buffer_unmap (buf, &buf_map);
|
|
return;
|
|
}
|
|
|
|
static void
|
|
gst_teletextdec_event_handler (vbi_event * ev, void *user_data)
|
|
{
|
|
page_info *pi;
|
|
vbi_pgno pgno;
|
|
vbi_subno subno;
|
|
|
|
GstTeletextDec *teletext = GST_TELETEXTDEC (user_data);
|
|
|
|
switch (ev->type) {
|
|
case VBI_EVENT_TTX_PAGE:
|
|
pgno = ev->ev.ttx_page.pgno;
|
|
subno = ev->ev.ttx_page.subno;
|
|
|
|
if (pgno != teletext->pageno
|
|
|| (teletext->subno != -1 && subno != teletext->subno))
|
|
return;
|
|
|
|
GST_DEBUG_OBJECT (teletext, "Received teletext page %03d.%02d",
|
|
(gint) vbi_bcd2dec (pgno), (gint) vbi_bcd2dec (subno));
|
|
|
|
pi = g_new (page_info, 1);
|
|
pi->pgno = pgno;
|
|
pi->subno = subno;
|
|
|
|
g_mutex_lock (&teletext->queue_lock);
|
|
g_queue_push_tail (teletext->queue, pi);
|
|
g_mutex_unlock (&teletext->queue_lock);
|
|
break;
|
|
case VBI_EVENT_CAPTION:
|
|
/* TODO: Handle subtitles in caption teletext pages */
|
|
GST_DEBUG_OBJECT (teletext, "Received caption page. Not implemented");
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
return;
|
|
}
|
|
|
|
static void
|
|
gst_teletextdec_try_get_buffer_pool (GstTeletextDec * teletext, GstCaps * caps,
|
|
gssize size)
|
|
{
|
|
guint pool_bufsize, min_bufs, max_bufs;
|
|
GstStructure *poolcfg;
|
|
GstBufferPool *new_pool;
|
|
GstQuery *alloc = gst_query_new_allocation (caps, TRUE);
|
|
|
|
if (teletext->buf_pool) {
|
|
/* this function is called only on a caps/size change, so it's practically
|
|
* impossible that we'll be able to reuse the old pool. */
|
|
gst_buffer_pool_set_active (teletext->buf_pool, FALSE);
|
|
gst_object_unref (teletext->buf_pool);
|
|
}
|
|
|
|
if (!gst_pad_peer_query (teletext->srcpad, alloc)) {
|
|
GST_DEBUG_OBJECT (teletext, "Failed to query peer pad for allocation "
|
|
"parameters");
|
|
teletext->buf_pool = NULL;
|
|
goto beach;
|
|
}
|
|
|
|
if (gst_query_get_n_allocation_pools (alloc) > 0) {
|
|
gst_query_parse_nth_allocation_pool (alloc, 0, &new_pool, &pool_bufsize,
|
|
&min_bufs, &max_bufs);
|
|
} else {
|
|
new_pool = gst_buffer_pool_new ();
|
|
max_bufs = 0;
|
|
min_bufs = 1;
|
|
}
|
|
|
|
poolcfg = gst_buffer_pool_get_config (new_pool);
|
|
gst_buffer_pool_config_set_params (poolcfg, gst_caps_copy (caps), size,
|
|
min_bufs, max_bufs);
|
|
if (!gst_buffer_pool_set_config (new_pool, poolcfg)) {
|
|
GST_DEBUG_OBJECT (teletext, "Failed to configure the buffer pool");
|
|
gst_object_unref (new_pool);
|
|
teletext->buf_pool = NULL;
|
|
goto beach;
|
|
}
|
|
if (!gst_buffer_pool_set_active (new_pool, TRUE)) {
|
|
GST_DEBUG_OBJECT (teletext, "Failed to make the buffer pool active");
|
|
gst_object_unref (new_pool);
|
|
teletext->buf_pool = NULL;
|
|
goto beach;
|
|
}
|
|
|
|
teletext->buf_pool = new_pool;
|
|
|
|
beach:
|
|
gst_query_unref (alloc);
|
|
}
|
|
|
|
static gboolean
|
|
gst_teletextdec_negotiate_caps (GstTeletextDec * teletext, guint width,
|
|
guint height)
|
|
{
|
|
gboolean rv = FALSE;
|
|
/* get the peer's caps filtered by our own ones. */
|
|
GstCaps *ourcaps = gst_pad_query_caps (teletext->srcpad, NULL);
|
|
GstCaps *peercaps = gst_pad_peer_query_caps (teletext->srcpad, ourcaps);
|
|
GstStructure *caps_struct;
|
|
const gchar *caps_name, *caps_fmt;
|
|
|
|
gst_caps_unref (ourcaps);
|
|
|
|
if (gst_caps_is_empty (peercaps)) {
|
|
goto beach;
|
|
}
|
|
|
|
/* make them writable in case we need to fixate them (video/x-raw). */
|
|
peercaps = gst_caps_make_writable (peercaps);
|
|
caps_struct = gst_caps_get_structure (peercaps, 0);
|
|
caps_name = gst_structure_get_name (caps_struct);
|
|
caps_fmt = gst_structure_get_string (caps_struct, "format");
|
|
|
|
if (!g_strcmp0 (caps_name, "video/x-raw")) {
|
|
teletext->width = width;
|
|
teletext->height = height;
|
|
teletext->export_func = gst_teletextdec_export_rgba_page;
|
|
gst_structure_set (caps_struct,
|
|
"width", G_TYPE_INT, width,
|
|
"height", G_TYPE_INT, height,
|
|
"framerate", GST_TYPE_FRACTION, 0, 1, NULL);
|
|
} else if (!g_strcmp0 (caps_name, "text/x-raw") &&
|
|
!g_strcmp0 (caps_fmt, "utf-8")) {
|
|
teletext->export_func = gst_teletextdec_export_text_page;
|
|
} else if (!g_strcmp0 (caps_name, "text/x-raw") &&
|
|
!g_strcmp0 (caps_fmt, "pango-markup")) {
|
|
teletext->export_func = gst_teletextdec_export_pango_page;
|
|
} else {
|
|
goto beach;
|
|
}
|
|
|
|
if (!gst_pad_push_event (teletext->srcpad, gst_event_new_caps (peercaps))) {
|
|
goto beach;
|
|
}
|
|
|
|
/* try to get a bufferpool from the peer pad in case of RGBA output. */
|
|
if (gst_teletextdec_export_rgba_page == teletext->export_func) {
|
|
gst_teletextdec_try_get_buffer_pool (teletext, peercaps,
|
|
width * height * sizeof (vbi_rgba));
|
|
}
|
|
|
|
/* we can happily return a success now. */
|
|
rv = TRUE;
|
|
|
|
beach:
|
|
gst_caps_unref (peercaps);
|
|
return rv;
|
|
}
|
|
|
|
/* this function does the actual processing
|
|
*/
|
|
static GstFlowReturn
|
|
gst_teletextdec_chain (GstPad * pad, GstObject * parent, GstBuffer * buf)
|
|
{
|
|
GstTeletextDec *teletext = GST_TELETEXTDEC (parent);
|
|
GstFlowReturn ret = GST_FLOW_OK;
|
|
|
|
teletext->in_timestamp = GST_BUFFER_TIMESTAMP (buf);
|
|
teletext->in_duration = GST_BUFFER_DURATION (buf);
|
|
|
|
gst_teletextdec_process_telx_buffer (teletext, buf);
|
|
gst_buffer_unref (buf);
|
|
|
|
g_mutex_lock (&teletext->queue_lock);
|
|
if (!g_queue_is_empty (teletext->queue)) {
|
|
ret = gst_teletextdec_push_page (teletext);
|
|
if (ret != GST_FLOW_OK) {
|
|
g_mutex_unlock (&teletext->queue_lock);
|
|
goto error;
|
|
}
|
|
}
|
|
g_mutex_unlock (&teletext->queue_lock);
|
|
|
|
return ret;
|
|
|
|
/* ERRORS */
|
|
error:
|
|
{
|
|
if (ret != GST_FLOW_OK && ret != GST_FLOW_NOT_LINKED
|
|
&& ret != GST_FLOW_FLUSHING) {
|
|
GST_ELEMENT_FLOW_ERROR (teletext, ret);
|
|
return GST_FLOW_ERROR;
|
|
}
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
static GstFlowReturn
|
|
gst_teletextdec_push_page (GstTeletextDec * teletext)
|
|
{
|
|
GstFlowReturn ret = GST_FLOW_OK;
|
|
GstBuffer *buf;
|
|
vbi_page page;
|
|
page_info *pi;
|
|
gint pgno, subno;
|
|
gboolean success;
|
|
guint width, height;
|
|
|
|
pi = g_queue_pop_head (teletext->queue);
|
|
pgno = vbi_bcd2dec (pi->pgno);
|
|
subno = vbi_bcd2dec (pi->subno);
|
|
|
|
GST_INFO_OBJECT (teletext, "Fetching teletext page %03d.%02d", pgno, subno);
|
|
|
|
success = vbi_fetch_vt_page (teletext->decoder, &page, pi->pgno, pi->subno,
|
|
VBI_WST_LEVEL_3p5, 25, FALSE);
|
|
g_free (pi);
|
|
if (G_UNLIKELY (!success))
|
|
goto fetch_page_failed;
|
|
|
|
width = COLUMNS_TO_WIDTH (page.columns);
|
|
height = ROWS_TO_HEIGHT (page.rows);
|
|
|
|
/* if output_func is NULL, we need to (re-)negotiate. also, it is possible
|
|
* (though unlikely) that we received a page of a different size. */
|
|
if (G_UNLIKELY (NULL == teletext->export_func ||
|
|
teletext->width != width || teletext->height != height)) {
|
|
/* if negotiate_caps returns FALSE, that means we weren't able to
|
|
* negotiate. */
|
|
if (G_UNLIKELY (!gst_teletextdec_negotiate_caps (teletext, width, height))) {
|
|
ret = GST_FLOW_NOT_NEGOTIATED;
|
|
goto push_failed;
|
|
}
|
|
if (G_UNLIKELY (teletext->segment)) {
|
|
gst_pad_push_event (teletext->srcpad, teletext->segment);
|
|
teletext->segment = NULL;
|
|
}
|
|
}
|
|
|
|
teletext->export_func (teletext, &page, &buf);
|
|
vbi_unref_page (&page);
|
|
|
|
GST_BUFFER_TIMESTAMP (buf) = teletext->in_timestamp;
|
|
GST_BUFFER_DURATION (buf) = teletext->in_duration;
|
|
|
|
GST_INFO_OBJECT (teletext, "Pushing buffer of size %" G_GSIZE_FORMAT,
|
|
gst_buffer_get_size (buf));
|
|
|
|
ret = gst_pad_push (teletext->srcpad, buf);
|
|
if (ret != GST_FLOW_OK)
|
|
goto push_failed;
|
|
|
|
return GST_FLOW_OK;
|
|
|
|
fetch_page_failed:
|
|
{
|
|
GST_ELEMENT_ERROR (teletext, RESOURCE, READ, (NULL), (NULL));
|
|
return GST_FLOW_ERROR;
|
|
}
|
|
|
|
push_failed:
|
|
{
|
|
GST_ERROR_OBJECT (teletext, "Pushing buffer failed, reason %s",
|
|
gst_flow_get_name (ret));
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
static gchar **
|
|
gst_teletextdec_vbi_page_to_text_lines (guint start, guint stop, vbi_page *
|
|
page)
|
|
{
|
|
const guint lines_count = stop - start + 1;
|
|
const guint line_length = page->columns;
|
|
gchar **lines;
|
|
guint i;
|
|
|
|
/* allocate a new NULL-terminated array of strings */
|
|
lines = (gchar **) g_malloc (sizeof (gchar *) * (lines_count + 1));
|
|
lines[lines_count] = NULL;
|
|
|
|
/* export each line in the range of the teletext page in text format */
|
|
for (i = start; i <= stop; i++) {
|
|
lines[i - start] = (gchar *) g_malloc (sizeof (gchar) * (line_length + 1));
|
|
vbi_print_page_region (page, lines[i - start], line_length + 1, "UTF-8",
|
|
TRUE, 0, 0, i, line_length, 1);
|
|
/* Add the null character */
|
|
lines[i - start][line_length] = '\0';
|
|
}
|
|
|
|
return lines;
|
|
}
|
|
|
|
static GstFlowReturn
|
|
gst_teletextdec_export_text_page (GstTeletextDec * teletext, vbi_page * page,
|
|
GstBuffer ** buf)
|
|
{
|
|
gchar *text;
|
|
guint size;
|
|
|
|
if (teletext->subtitles_mode) {
|
|
gchar **lines;
|
|
GString *subs;
|
|
guint i;
|
|
|
|
lines = gst_teletextdec_vbi_page_to_text_lines (1, 23, page);
|
|
subs = g_string_new ("");
|
|
/* Strip white spaces and squash blank lines */
|
|
for (i = 0; i < 23; i++) {
|
|
g_strstrip (lines[i]);
|
|
if (g_strcmp0 (lines[i], ""))
|
|
g_string_append_printf (subs, teletext->subtitles_template, lines[i]);
|
|
}
|
|
/* if the page is blank and doesn't contain any line of text, just add a
|
|
* line break */
|
|
if (!g_strcmp0 (subs->str, ""))
|
|
g_string_append (subs, "\n");
|
|
|
|
text = subs->str;
|
|
size = subs->len + 1;
|
|
g_string_free (subs, FALSE);
|
|
g_strfreev (lines);
|
|
} else {
|
|
size = page->columns * page->rows;
|
|
text = g_malloc (size);
|
|
vbi_print_page (page, text, size, "UTF-8", FALSE, TRUE);
|
|
}
|
|
|
|
/* Allocate new buffer */
|
|
*buf = gst_buffer_new_wrapped (text, size);
|
|
|
|
return GST_FLOW_OK;
|
|
}
|
|
|
|
static GstFlowReturn
|
|
gst_teletextdec_export_rgba_page (GstTeletextDec * teletext, vbi_page * page,
|
|
GstBuffer ** buf)
|
|
{
|
|
guint size;
|
|
GstBuffer *lbuf;
|
|
GstMapInfo buf_map;
|
|
|
|
size = teletext->width * teletext->height * sizeof (vbi_rgba);
|
|
|
|
/* Allocate new buffer, using the negotiated pool if available. */
|
|
if (teletext->buf_pool) {
|
|
GstFlowReturn acquire_rv =
|
|
gst_buffer_pool_acquire_buffer (teletext->buf_pool, &lbuf, NULL);
|
|
if (acquire_rv != GST_FLOW_OK) {
|
|
return acquire_rv;
|
|
}
|
|
} else {
|
|
lbuf = gst_buffer_new_allocate (NULL, size, NULL);
|
|
if (NULL == lbuf)
|
|
return GST_FLOW_ERROR;
|
|
}
|
|
|
|
if (!gst_buffer_map (lbuf, &buf_map, GST_MAP_WRITE)) {
|
|
gst_buffer_unref (lbuf);
|
|
return GST_FLOW_ERROR;
|
|
}
|
|
|
|
vbi_draw_vt_page (page, VBI_PIXFMT_RGBA32_LE, buf_map.data, FALSE, TRUE);
|
|
gst_buffer_unmap (lbuf, &buf_map);
|
|
*buf = lbuf;
|
|
|
|
return GST_FLOW_OK;
|
|
}
|
|
|
|
static GstFlowReturn
|
|
gst_teletextdec_export_pango_page (GstTeletextDec * teletext, vbi_page * page,
|
|
GstBuffer ** buf)
|
|
{
|
|
vbi_char *acp;
|
|
const guint rows = page->rows;
|
|
gchar **colors;
|
|
gchar **lines;
|
|
GString *subs;
|
|
guint start, stop, k;
|
|
gint i, j;
|
|
|
|
colors = (gchar **) g_malloc (sizeof (gchar *) * (rows + 1));
|
|
colors[rows] = NULL;
|
|
|
|
/* parse all the lines and approximate it's foreground color using the first
|
|
* non null character */
|
|
for (acp = page->text, i = 0; i < page->rows; acp += page->columns, i++) {
|
|
for (j = 0; j < page->columns; j++) {
|
|
colors[i] = g_strdup (default_color_map[7]);
|
|
if (acp[j].unicode != 0x20) {
|
|
colors[i] = g_strdup (default_color_map[acp[j].foreground]);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* get an array of strings with each line of the telext page */
|
|
start = teletext->subtitles_mode ? 1 : 0;
|
|
stop = teletext->subtitles_mode ? rows - 2 : rows - 1;
|
|
lines = gst_teletextdec_vbi_page_to_text_lines (start, stop, page);
|
|
|
|
/* format each line in pango markup */
|
|
subs = g_string_new ("");
|
|
for (k = start; k <= stop; k++) {
|
|
g_string_append_printf (subs, PANGO_TEMPLATE,
|
|
teletext->font_description, colors[k], lines[k - start]);
|
|
}
|
|
|
|
/* Allocate new buffer */
|
|
*buf = gst_buffer_new_wrapped (subs->str, subs->len + 1);
|
|
|
|
g_strfreev (lines);
|
|
g_strfreev (colors);
|
|
g_string_free (subs, FALSE);
|
|
return GST_FLOW_OK;
|
|
}
|
|
|
|
/* Converts the line_offset / field_parity byte of a VBI data unit. */
|
|
static void
|
|
gst_teletextdec_lofp_to_line (guint * field, guint * field_line,
|
|
guint * frame_line, guint lofp, systems system)
|
|
{
|
|
guint line_offset;
|
|
|
|
/* field_parity */
|
|
*field = !(lofp & (1 << 5));
|
|
|
|
line_offset = lofp & 31;
|
|
|
|
if (line_offset > 0) {
|
|
static const guint field_start[2][2] = {
|
|
{0, 263},
|
|
{0, 313},
|
|
};
|
|
|
|
*field_line = line_offset;
|
|
*frame_line = field_start[system][*field] + line_offset;
|
|
} else {
|
|
*field_line = 0;
|
|
*frame_line = 0;
|
|
}
|
|
}
|
|
|
|
static int
|
|
gst_teletextdec_line_address (GstTeletextDec * teletext,
|
|
GstTeletextFrame * frame, vbi_sliced ** spp, guint lofp, systems system)
|
|
{
|
|
guint field;
|
|
guint field_line;
|
|
guint frame_line;
|
|
|
|
if (G_UNLIKELY (frame->current_slice >= frame->sliced_end)) {
|
|
GST_LOG_OBJECT (teletext, "Out of sliced VBI buffer space (%d lines).",
|
|
(int) (frame->sliced_end - frame->sliced_begin));
|
|
return VBI_ERROR;
|
|
}
|
|
|
|
gst_teletextdec_lofp_to_line (&field, &field_line, &frame_line, lofp, system);
|
|
|
|
GST_LOG_OBJECT (teletext, "Line %u/%u=%u.", field, field_line, frame_line);
|
|
|
|
if (frame_line != 0) {
|
|
GST_LOG_OBJECT (teletext, "Last frame Line %u.", frame->last_frame_line);
|
|
if (frame_line <= frame->last_frame_line) {
|
|
GST_LOG_OBJECT (teletext, "New frame");
|
|
return VBI_NEW_FRAME;
|
|
}
|
|
|
|
/* FIXME : This never happens, since lofp is a guint8 */
|
|
#if 0
|
|
/* new segment flag */
|
|
if (lofp < 0) {
|
|
GST_LOG_OBJECT (teletext, "New frame");
|
|
return VBI_NEW_FRAME;
|
|
}
|
|
#endif
|
|
|
|
frame->last_field = field;
|
|
frame->last_field_line = field_line;
|
|
frame->last_frame_line = frame_line;
|
|
|
|
*spp = frame->current_slice++;
|
|
(*spp)->line = frame_line;
|
|
} else {
|
|
/* Undefined line. */
|
|
return VBI_ERROR;
|
|
}
|
|
|
|
return VBI_SUCCESS;
|
|
}
|
|
|
|
static gboolean
|
|
gst_teletextdec_extract_data_units (GstTeletextDec * teletext,
|
|
GstTeletextFrame * f, const guint8 * packet, guint * offset, gsize size)
|
|
{
|
|
const guint8 *data_unit;
|
|
guint i;
|
|
|
|
while (*offset < size) {
|
|
vbi_sliced *s = NULL;
|
|
gint data_unit_id, data_unit_length;
|
|
|
|
data_unit = packet + *offset;
|
|
data_unit_id = data_unit[0];
|
|
data_unit_length = data_unit[1];
|
|
GST_LOG_OBJECT (teletext, "vbi header %02x %02x %02x", data_unit[0],
|
|
data_unit[1], data_unit[2]);
|
|
|
|
switch (data_unit_id) {
|
|
case DATA_UNIT_STUFFING:
|
|
{
|
|
*offset += 2 + data_unit_length;
|
|
break;
|
|
}
|
|
|
|
case DATA_UNIT_EBU_TELETEXT_NON_SUBTITLE:
|
|
case DATA_UNIT_EBU_TELETEXT_SUBTITLE:
|
|
{
|
|
gint res;
|
|
|
|
if (G_UNLIKELY (data_unit_length != 1 + 1 + 42)) {
|
|
/* Skip this data unit */
|
|
GST_WARNING_OBJECT (teletext, "The data unit length is not 44 bytes");
|
|
*offset += 2 + data_unit_length;
|
|
break;
|
|
}
|
|
|
|
res =
|
|
gst_teletextdec_line_address (teletext, f, &s, data_unit[2],
|
|
SYSTEM_625);
|
|
if (G_UNLIKELY (res == VBI_ERROR)) {
|
|
/* Can't retrieve line address, skip this data unit */
|
|
GST_WARNING_OBJECT (teletext,
|
|
"Could not retrieve line address for this data unit");
|
|
return VBI_ERROR;
|
|
}
|
|
if (G_UNLIKELY (f->last_field_line > 0
|
|
&& (f->last_field_line - 7 >= 23 - 7))) {
|
|
GST_WARNING_OBJECT (teletext, "Bad line: %d", f->last_field_line - 7);
|
|
return VBI_ERROR;
|
|
}
|
|
if (res == VBI_NEW_FRAME) {
|
|
/* New frame */
|
|
return VBI_NEW_FRAME;
|
|
}
|
|
s->id = VBI_SLICED_TELETEXT_B;
|
|
for (i = 0; i < 42; i++)
|
|
s->data[i] = vbi_rev8 (data_unit[4 + i]);
|
|
*offset += 46;
|
|
break;
|
|
}
|
|
|
|
case DATA_UNIT_ZVBI_WSS_CPR1204:
|
|
case DATA_UNIT_ZVBI_CLOSED_CAPTION_525:
|
|
case DATA_UNIT_ZVBI_MONOCHROME_SAMPLES_525:
|
|
case DATA_UNIT_VPS:
|
|
case DATA_UNIT_WSS:
|
|
case DATA_UNIT_CLOSED_CAPTION:
|
|
case DATA_UNIT_MONOCHROME_SAMPLES:
|
|
{
|
|
/*Not supported yet */
|
|
*offset += 2 + data_unit_length;
|
|
break;
|
|
}
|
|
|
|
default:
|
|
{
|
|
/* corrupted stream, increase the offset by one until we sync */
|
|
GST_LOG_OBJECT (teletext, "Corrupted, increasing offset by one");
|
|
*offset += 1;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
return VBI_SUCCESS;
|
|
}
|
|
|
|
static gboolean
|
|
teletext_init (GstPlugin * teletext)
|
|
{
|
|
GST_DEBUG_CATEGORY_INIT (gst_teletextdec_debug, "teletext", 0,
|
|
"Teletext decoder");
|
|
return gst_element_register (teletext, "teletextdec", GST_RANK_NONE,
|
|
GST_TYPE_TELETEXTDEC);
|
|
}
|
|
|
|
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
|
|
GST_VERSION_MINOR,
|
|
teletext,
|
|
"Teletext plugin",
|
|
teletext_init, VERSION, "LGPL", GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)
|