gstreamer/gst/iec958/ac3iec.c

463 lines
12 KiB
C

/* GStreamer
* Copyright (C) 2004 Martin Soto <martinsoto@users.sourceforge.net>
2005 Michael Smith <msmith@fluendo.com>
*
* ac3iec.c: Pad AC3 frames into IEC958 frames for the S/PDIF interface.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <string.h>
#include <gst/gst.h>
#include "ac3iec.h"
GST_DEBUG_CATEGORY_STATIC (ac3iec_debug);
#define GST_CAT_DEFAULT (ac3iec_debug)
/* The duration of a single IEC958 frame. */
#define IEC958_FRAME_DURATION (32 * GST_MSECOND)
/* AC3IEC signals and args */
enum
{
LAST_SIGNAL,
};
enum
{
PROP_0,
PROP_RAW_AUDIO,
};
static GstStaticPadTemplate ac3iec_sink_template =
GST_STATIC_PAD_TEMPLATE ("sink",
GST_PAD_SINK,
GST_PAD_ALWAYS,
GST_STATIC_CAPS ("audio/x-private1-ac3; audio/x-ac3; audio/ac3")
);
/* Two different output caps are possible. */
#define NORMAL_CAPS_DEF "audio/x-iec958, rate = (int){32000, 44100, 48000}"
#define RAW_AUDIO_CAPS_DEF "audio/x-raw-int, " \
"endianness = (int) " G_STRINGIFY (G_BIG_ENDIAN) ", " \
"signed = (boolean) true, " \
"width = (int) 16, " \
"depth = (int) 16, " \
"rate = (int){32000, 44100, 48000}, " \
"channels = (int) 2"
static GstStaticCaps normal_caps = GST_STATIC_CAPS (NORMAL_CAPS_DEF);
static GstStaticCaps raw_audio_caps = GST_STATIC_CAPS (RAW_AUDIO_CAPS_DEF);
static GstStaticPadTemplate ac3iec_src_template =
GST_STATIC_PAD_TEMPLATE ("src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_STATIC_CAPS (NORMAL_CAPS_DEF ";" RAW_AUDIO_CAPS_DEF));
static void ac3iec_base_init (gpointer g_class);
static void ac3iec_class_init (AC3IECClass * klass);
static void ac3iec_init (AC3IEC * ac3iec);
static void ac3iec_finalize (GObject * object);
static void ac3iec_set_property (GObject * object,
guint prop_id, const GValue * value, GParamSpec * pspec);
static void ac3iec_get_property (GObject * object,
guint prop_id, GValue * value, GParamSpec * pspec);
static gboolean ac3iec_setcaps (GstPad * pad, GstCaps * caps);
static GstFlowReturn ac3iec_chain_dvd (GstPad * pad, GstBuffer * buf);
static GstFlowReturn ac3iec_chain_raw (GstPad * pad, GstBuffer * buf);
static GstStateChangeReturn ac3iec_change_state (GstElement * element,
GstStateChange transition);
static GstElementClass *parent_class = NULL;
/* static guint ac3iec_signals[LAST_SIGNAL] = { 0 }; */
GType
ac3iec_get_type (void)
{
static GType ac3iec_type = 0;
if (!ac3iec_type) {
static const GTypeInfo ac3iec_info = {
sizeof (AC3IECClass),
ac3iec_base_init,
NULL,
(GClassInitFunc) ac3iec_class_init,
NULL,
NULL,
sizeof (AC3IEC),
0,
(GInstanceInitFunc) ac3iec_init,
};
ac3iec_type = g_type_register_static (GST_TYPE_ELEMENT,
"AC3IEC", &ac3iec_info, 0);
GST_DEBUG_CATEGORY_INIT (ac3iec_debug, "ac3iec", 0,
"AC3 to IEC958 padding element");
}
return ac3iec_type;
}
static void
ac3iec_base_init (gpointer g_class)
{
GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
gst_element_class_set_details_simple (element_class, "AC3 to IEC958 filter",
"Codec/Muxer/Audio",
"Pads AC3 frames into IEC958 frames suitable for a raw S/PDIF interface",
"Martin Soto <martinsoto@users.sourceforge.net>");
gst_element_class_add_pad_template (element_class,
gst_static_pad_template_get (&ac3iec_sink_template));
gst_element_class_add_pad_template (element_class,
gst_static_pad_template_get (&ac3iec_src_template));
}
static void
ac3iec_class_init (AC3IECClass * klass)
{
GObjectClass *gobject_class;
GstElementClass *gstelement_class;
gobject_class = (GObjectClass *) klass;
gstelement_class = (GstElementClass *) klass;
parent_class = g_type_class_peek_parent (klass);
gobject_class->set_property = ac3iec_set_property;
gobject_class->get_property = ac3iec_get_property;
gobject_class->finalize = ac3iec_finalize;
g_object_class_install_property (gobject_class, PROP_RAW_AUDIO,
g_param_spec_boolean ("raw-audio", "raw-audio",
"If true, source pad caps are set to raw audio.",
FALSE, G_PARAM_READWRITE));
gstelement_class->change_state = ac3iec_change_state;
}
static void
ac3iec_init (AC3IEC * ac3iec)
{
ac3iec->sink =
gst_pad_new_from_static_template (&ac3iec_sink_template, "sink");
gst_pad_set_setcaps_function (ac3iec->sink, ac3iec_setcaps);
gst_pad_set_chain_function (ac3iec->sink, ac3iec_chain_dvd);
gst_element_add_pad (GST_ELEMENT (ac3iec), ac3iec->sink);
ac3iec->src = gst_pad_new_from_static_template (&ac3iec_src_template, "src");
gst_pad_use_fixed_caps (ac3iec->src);
gst_element_add_pad (GST_ELEMENT (ac3iec), ac3iec->src);
ac3iec->cur_ts = GST_CLOCK_TIME_NONE;
ac3iec->padder = g_malloc (sizeof (ac3_padder));
}
static void
ac3iec_finalize (GObject * object)
{
AC3IEC *ac3iec = AC3IEC (object);
g_free (ac3iec->padder);
G_OBJECT_CLASS (parent_class)->finalize (object);
}
static void
ac3iec_set_property (GObject * object, guint prop_id,
const GValue * value, GParamSpec * pspec)
{
AC3IEC *ac3iec = AC3IEC (object);
switch (prop_id) {
case PROP_RAW_AUDIO:
{
ac3iec->raw_audio = g_value_get_boolean (value);
break;
}
default:
break;
}
}
static gboolean
ac3iec_setcaps (GstPad * pad, GstCaps * caps)
{
AC3IEC *ac3iec = AC3IEC (gst_pad_get_parent (pad));
GstStructure *structure = gst_caps_get_structure (caps, 0);
if (structure && gst_structure_has_name (structure, "audio/x-private1-ac3"))
ac3iec->dvdmode = TRUE;
else
ac3iec->dvdmode = FALSE;
gst_object_unref (ac3iec);
return TRUE;
}
static void
ac3iec_get_property (GObject * object, guint prop_id,
GValue * value, GParamSpec * pspec)
{
AC3IEC *ac3iec = AC3IEC (object);
switch (prop_id) {
case PROP_RAW_AUDIO:
g_value_set_boolean (value, ac3iec->raw_audio);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
static GstFlowReturn
ac3iec_chain_dvd (GstPad * pad, GstBuffer * buf)
{
guint first_access;
guint8 *data;
guint size;
gint offset;
gint len;
GstBuffer *subbuf;
GstFlowReturn ret;
AC3IEC *ac3iec = AC3IEC (gst_pad_get_parent (pad));
if (ac3iec->dvdmode) {
size = GST_BUFFER_SIZE (buf);
data = GST_BUFFER_DATA (buf);
first_access = (data[0] << 8) | data[1];
/* Skip the first_access header */
offset = 2;
if (first_access > 1) {
/* Length of data before first_access */
len = first_access - 1;
/* Ensure we don't crash if fed totally invalid data */
if (offset + len > size) {
ret = GST_FLOW_ERROR;
goto done;
}
if (len > 0) {
subbuf = gst_buffer_create_sub (buf, offset, len);
GST_BUFFER_TIMESTAMP (subbuf) = GST_CLOCK_TIME_NONE;
ret = ac3iec_chain_raw (pad, subbuf);
if (ret != GST_FLOW_OK)
goto done;
}
offset += len;
len = size - offset;
subbuf = gst_buffer_create_sub (buf, offset, len);
GST_BUFFER_TIMESTAMP (subbuf) = GST_BUFFER_TIMESTAMP (buf);
ret = ac3iec_chain_raw (pad, subbuf);
} else {
/* Ensure we don't crash if fed totally invalid data */
if (size < 2) {
ret = GST_FLOW_ERROR;
goto done;
}
/* No first_access, so no timestamp */
subbuf = gst_buffer_create_sub (buf, offset, size - offset);
GST_BUFFER_TIMESTAMP (subbuf) = GST_CLOCK_TIME_NONE;
ret = ac3iec_chain_raw (pad, subbuf);
}
} else {
ret = ac3iec_chain_raw (pad, buf);
gst_object_unref (ac3iec);
return ret;
}
done:
gst_object_unref (ac3iec);
gst_buffer_unref (buf);
return ret;
}
static GstFlowReturn
ac3iec_chain_raw (GstPad * pad, GstBuffer * buf)
{
GstBuffer *new;
AC3IEC *ac3iec;
int event;
GstFlowReturn ret = GST_FLOW_OK;
g_return_val_if_fail (pad != NULL, GST_FLOW_ERROR);
g_return_val_if_fail (GST_IS_PAD (pad), GST_FLOW_ERROR);
g_return_val_if_fail (buf != NULL, GST_FLOW_ERROR);
ac3iec = AC3IEC (gst_pad_get_parent (pad));
if (GST_BUFFER_TIMESTAMP (buf) != GST_CLOCK_TIME_NONE) {
/* Whoever tells me why it is necessary to add a frame in order to
get synchronized sound will get a beer from me. */
ac3iec->cur_ts = GST_BUFFER_TIMESTAMP (buf) + IEC958_FRAME_DURATION;
}
/* Push the new data into the padder. */
ac3p_push_data (ac3iec->padder, GST_BUFFER_DATA (buf), GST_BUFFER_SIZE (buf));
/* Parse the data. */
event = ac3p_parse (ac3iec->padder);
while (event != AC3P_EVENT_PUSH) {
if (event == AC3P_EVENT_FRAME) {
GstCaps *bufcaps = NULL;
if (ac3iec->caps == NULL) {
gint rate = ac3iec->padder->rate;
if (ac3iec->raw_audio) {
ac3iec->caps =
gst_caps_make_writable (gst_static_caps_get (&raw_audio_caps));
} else {
ac3iec->caps =
gst_caps_make_writable (gst_static_caps_get (&normal_caps));
}
gst_structure_set (gst_caps_get_structure (ac3iec->caps, 0), "rate",
G_TYPE_INT, rate, NULL);
gst_pad_set_caps (ac3iec->src, ac3iec->caps);
}
/* We have a new frame: */
bufcaps = GST_PAD_CAPS (ac3iec->src);
/* Create a new buffer, and copy the frame data into it. */
ret =
gst_pad_alloc_buffer_and_set_caps (ac3iec->src, 0,
AC3P_IEC_FRAME_SIZE, bufcaps, &new);
if (ret != GST_FLOW_OK)
goto buffer_alloc_failed;
memcpy (GST_BUFFER_DATA (new), ac3p_frame (ac3iec->padder),
AC3P_IEC_FRAME_SIZE);
/* Set the timestamp. */
GST_BUFFER_TIMESTAMP (new) = ac3iec->cur_ts;
GST_BUFFER_DURATION (new) = IEC958_FRAME_DURATION;
ac3iec->cur_ts = GST_CLOCK_TIME_NONE;
GST_LOG_OBJECT (ac3iec, "Pushing IEC958 buffer of size %d",
GST_BUFFER_SIZE (new));
/* Push the buffer to the source pad. */
ret = gst_pad_push (ac3iec->src, new);
}
event = ac3p_parse (ac3iec->padder);
}
gst_buffer_unref (buf);
done:
gst_object_unref (ac3iec);
return ret;
buffer_alloc_failed:
gst_buffer_unref (buf);
goto done;
}
static GstStateChangeReturn
ac3iec_change_state (GstElement * element, GstStateChange transition)
{
AC3IEC *ac3iec;
GstStateChangeReturn ret;
g_return_val_if_fail (GST_IS_AC3IEC (element), GST_STATE_CHANGE_FAILURE);
ac3iec = AC3IEC (element);
switch (transition) {
case GST_STATE_CHANGE_NULL_TO_READY:
break;
case GST_STATE_CHANGE_READY_TO_PAUSED:
ac3p_init (ac3iec->padder);
break;
case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
break;
default:
break;
}
ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
switch (transition) {
case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
break;
case GST_STATE_CHANGE_PAUSED_TO_READY:
ac3p_clear (ac3iec->padder);
if (ac3iec->caps) {
gst_caps_unref (ac3iec->caps);
ac3iec->caps = NULL;
}
break;
case GST_STATE_CHANGE_READY_TO_NULL:
break;
default:
break;
}
return ret;
}
static gboolean
plugin_init (GstPlugin * plugin)
{
if (!gst_element_register (plugin, "ac3iec958", GST_RANK_NONE,
GST_TYPE_AC3IEC)) {
return FALSE;
}
return TRUE;
}
GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
GST_VERSION_MINOR,
"iec958",
"Convert raw AC3 into IEC958 (S/PDIF) frames",
plugin_init, VERSION, "LGPL", GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN);