blob: 628d15c94797792be169235de7f232c31e923b3f [file] [log] [blame]
/*
*
* BlueZ - Bluetooth protocol stack for Linux
*
* Copyright (C) 2004-2007 Marcel Holtmann <marcel@holtmann.org>
*
*
* 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 St, Fifth Floor, Boston, MA 02110-1301 USA
*
*/
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <string.h>
#include "gstsbcenc.h"
#include "gstsbcutil.h"
#define SBC_ENC_DEFAULT_MODE CFG_MODE_AUTO
#define SBC_ENC_DEFAULT_BLOCKS 16
#define SBC_ENC_DEFAULT_SUB_BANDS 8
#define SBC_ENC_DEFAULT_ALLOCATION CFG_ALLOCATION_AUTO
GST_DEBUG_CATEGORY_STATIC(sbc_enc_debug);
#define GST_CAT_DEFAULT sbc_enc_debug
#define GST_TYPE_SBC_MODE (gst_sbc_mode_get_type())
static GType gst_sbc_mode_get_type(void)
{
static GType sbc_mode_type = 0;
static GEnumValue sbc_modes[] = {
{ 0, "Auto", "auto" },
{ 1, "Mono", "mono" },
{ 2, "Dual Channel", "dual" },
{ 3, "Stereo", "stereo" },
{ 4, "Joint Stereo", "joint" },
{ -1, NULL, NULL}
};
if (!sbc_mode_type)
sbc_mode_type = g_enum_register_static("GstSbcMode", sbc_modes);
return sbc_mode_type;
}
#define GST_TYPE_SBC_ALLOCATION (gst_sbc_allocation_get_type())
static GType gst_sbc_allocation_get_type(void)
{
static GType sbc_allocation_type = 0;
static GEnumValue sbc_allocations[] = {
{ CFG_ALLOCATION_AUTO, "Auto", "auto" },
{ CFG_ALLOCATION_LOUDNESS, "Loudness", "loudness" },
{ CFG_ALLOCATION_SNR, "SNR", "snr" },
{ -1, NULL, NULL}
};
if (!sbc_allocation_type)
sbc_allocation_type = g_enum_register_static(
"GstSbcAllocation", sbc_allocations);
return sbc_allocation_type;
}
enum {
PROP_0,
PROP_MODE,
PROP_ALLOCATION,
PROP_BLOCKS,
PROP_SUBBANDS
};
GST_BOILERPLATE(GstSbcEnc, gst_sbc_enc, GstElement, GST_TYPE_ELEMENT);
static const GstElementDetails sbc_enc_details =
GST_ELEMENT_DETAILS("Bluetooth SBC encoder",
"Codec/Encoder/Audio",
"Encode a SBC audio stream",
"Marcel Holtmann <marcel@holtmann.org>");
static GstStaticPadTemplate sbc_enc_sink_factory =
GST_STATIC_PAD_TEMPLATE("sink", GST_PAD_SINK, GST_PAD_ALWAYS,
GST_STATIC_CAPS("audio/x-raw-int, "
"rate = (int) { 16000, 32000, 44100, 48000 }, "
"channels = (int) [ 1, 2 ], "
"endianness = (int) LITTLE_ENDIAN, "
"signed = (boolean) true, "
"width = (int) 16, "
"depth = (int) 16"));
static GstStaticPadTemplate sbc_enc_src_factory =
GST_STATIC_PAD_TEMPLATE("src", GST_PAD_SRC, GST_PAD_ALWAYS,
GST_STATIC_CAPS("audio/x-sbc, "
"rate = (int) { 16000, 32000, 44100, 48000 }, "
"channels = (int) [ 1, 2 ], "
"mode = (string) { mono, dual, stereo, joint }, "
"blocks = (int) { 4, 8, 12, 16 }, "
"subbands = (int) { 4, 8 }, "
"allocation = (string) { snr, loudness }"));
static GstCaps* sbc_enc_generate_srcpad_caps(GstSbcEnc *enc, GstCaps *caps)
{
gint rate;
gint channels;
GstCaps* src_caps;
GstStructure *structure;
const gchar *mode;
const gchar *allocation;
structure = gst_caps_get_structure(caps, 0);
if (!gst_structure_get_int (structure, "rate", &rate))
return NULL;
if (!gst_structure_get_int (structure, "channels", &channels))
return NULL;
enc->sbc.rate = rate;
enc->sbc.channels = channels;
if (enc->mode == 0)
enc->sbc.joint = CFG_MODE_JOINT_STEREO;
else
enc->sbc.joint = enc->mode;
enc->sbc.blocks = enc->blocks;
enc->sbc.subbands = enc->subbands;
if (enc->allocation == 0)
enc->sbc.allocation = CFG_ALLOCATION_LOUDNESS;
else
enc->sbc.allocation = enc->allocation;
mode = gst_sbc_get_mode_string(enc->sbc.joint);
allocation = gst_sbc_get_allocation_string(enc->sbc.allocation);
src_caps = gst_caps_new_simple("audio/x-sbc",
"rate", G_TYPE_INT, enc->sbc.rate,
"channels", G_TYPE_INT, enc->sbc.channels,
"mode", G_TYPE_STRING, mode,
"subbands", G_TYPE_INT, enc->sbc.subbands,
"blocks", G_TYPE_INT, enc->sbc.blocks,
"allocation", G_TYPE_STRING, allocation,
NULL);
return src_caps;
}
static gboolean sbc_enc_sink_setcaps (GstPad * pad, GstCaps * caps)
{
GstSbcEnc *enc;
GstStructure *structure;
GstCaps *othercaps;
enc = GST_SBC_ENC(GST_PAD_PARENT (pad));
structure = gst_caps_get_structure(caps, 0);
othercaps = sbc_enc_generate_srcpad_caps(enc, caps);
if (othercaps == NULL)
goto error;
gst_pad_set_caps (enc->srcpad, othercaps);
gst_caps_unref(othercaps);
return TRUE;
error:
GST_ERROR_OBJECT (enc, "invalid input caps");
return FALSE;
}
static GstFlowReturn sbc_enc_chain(GstPad *pad, GstBuffer *buffer)
{
GstSbcEnc *enc = GST_SBC_ENC(gst_pad_get_parent(pad));
GstFlowReturn res = GST_FLOW_OK;
guint size, offset = 0;
guint8 *data;
data = GST_BUFFER_DATA(buffer);
size = GST_BUFFER_SIZE(buffer);
while (offset < size) {
GstBuffer *output;
GstCaps *caps;
int consumed;
consumed = sbc_encode(&enc->sbc, data + offset, size - offset);
if (consumed <= 0)
break;
caps = GST_PAD_CAPS(enc->srcpad);
res = gst_pad_alloc_buffer_and_set_caps(enc->srcpad,
GST_BUFFER_OFFSET_NONE,
enc->sbc.len, caps, &output);
if (res != GST_FLOW_OK)
goto done;
memcpy(GST_BUFFER_DATA(output), enc->sbc.data, enc->sbc.len);
res = gst_pad_push(enc->srcpad, output);
if (res != GST_FLOW_OK)
goto done;
offset += consumed;
}
if (offset < size)
res = GST_FLOW_ERROR;
done:
gst_buffer_unref(buffer);
gst_object_unref(enc);
return res;
}
static GstStateChangeReturn sbc_enc_change_state(GstElement *element,
GstStateChange transition)
{
GstSbcEnc *enc = GST_SBC_ENC(element);
switch (transition) {
case GST_STATE_CHANGE_READY_TO_PAUSED:
GST_DEBUG("Setup subband codec");
sbc_init(&enc->sbc, 0);
break;
case GST_STATE_CHANGE_PAUSED_TO_READY:
GST_DEBUG("Finish subband codec");
sbc_finish(&enc->sbc);
break;
default:
break;
}
return parent_class->change_state(element, transition);
}
static void gst_sbc_enc_base_init(gpointer g_class)
{
GstElementClass *element_class = GST_ELEMENT_CLASS(g_class);
gst_element_class_add_pad_template(element_class,
gst_static_pad_template_get(&sbc_enc_sink_factory));
gst_element_class_add_pad_template(element_class,
gst_static_pad_template_get(&sbc_enc_src_factory));
gst_element_class_set_details(element_class, &sbc_enc_details);
}
static void gst_sbc_enc_set_property(GObject *object, guint prop_id,
const GValue *value, GParamSpec *pspec)
{
GstSbcEnc *enc = GST_SBC_ENC(object);
/* TODO - CAN ONLY BE CHANGED ON READY AND BELOW */
switch (prop_id) {
case PROP_MODE:
enc->mode = g_value_get_enum(value);
break;
case PROP_ALLOCATION:
enc->allocation = g_value_get_enum(value);
break;
case PROP_BLOCKS:
/* TODO - verify consistency */
enc->blocks = g_value_get_int(value);
break;
case PROP_SUBBANDS:
/* TODO - verify consistency */
enc->subbands = g_value_get_int(value);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID(object, prop_id, pspec);
break;
}
}
static void gst_sbc_enc_get_property(GObject *object, guint prop_id,
GValue *value, GParamSpec *pspec)
{
GstSbcEnc *enc = GST_SBC_ENC(object);
switch (prop_id) {
case PROP_MODE:
g_value_set_enum(value, enc->mode);
break;
case PROP_ALLOCATION:
g_value_set_enum(value, enc->allocation);
break;
case PROP_BLOCKS:
g_value_set_int(value, enc->blocks);
break;
case PROP_SUBBANDS:
g_value_set_int(value, enc->subbands);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID(object, prop_id, pspec);
break;
}
}
static void gst_sbc_enc_class_init(GstSbcEncClass *klass)
{
GObjectClass *object_class = G_OBJECT_CLASS(klass);
GstElementClass *element_class = GST_ELEMENT_CLASS(klass);
parent_class = g_type_class_peek_parent(klass);
object_class->set_property = GST_DEBUG_FUNCPTR(gst_sbc_enc_set_property);
object_class->get_property = GST_DEBUG_FUNCPTR(gst_sbc_enc_get_property);
element_class->change_state = GST_DEBUG_FUNCPTR(sbc_enc_change_state);
g_object_class_install_property(object_class, PROP_MODE,
g_param_spec_enum("mode", "Mode",
"Encoding mode", GST_TYPE_SBC_MODE,
SBC_ENC_DEFAULT_MODE, G_PARAM_READWRITE));
g_object_class_install_property(object_class, PROP_ALLOCATION,
g_param_spec_enum("allocation", "Allocation",
"Allocation mode", GST_TYPE_SBC_ALLOCATION,
SBC_ENC_DEFAULT_ALLOCATION, G_PARAM_READWRITE));
g_object_class_install_property(object_class, PROP_BLOCKS,
g_param_spec_int("blocks", "Blocks",
"Blocks", 0, G_MAXINT,
SBC_ENC_DEFAULT_BLOCKS, G_PARAM_READWRITE));
g_object_class_install_property(object_class, PROP_SUBBANDS,
g_param_spec_int("subbands", "Sub Bands",
"Sub Bands", 0, G_MAXINT,
SBC_ENC_DEFAULT_SUB_BANDS, G_PARAM_READWRITE));
GST_DEBUG_CATEGORY_INIT(sbc_enc_debug, "sbcenc", 0,
"SBC encoding element");
}
static void gst_sbc_enc_init(GstSbcEnc *self, GstSbcEncClass *klass)
{
self->sinkpad = gst_pad_new_from_static_template(&sbc_enc_sink_factory, "sink");
gst_pad_set_setcaps_function (self->sinkpad,
GST_DEBUG_FUNCPTR (sbc_enc_sink_setcaps));
gst_element_add_pad(GST_ELEMENT(self), self->sinkpad);
self->srcpad = gst_pad_new_from_static_template(&sbc_enc_src_factory, "src");
gst_pad_set_chain_function(self->sinkpad, GST_DEBUG_FUNCPTR(sbc_enc_chain));
gst_element_add_pad(GST_ELEMENT(self), self->srcpad);
self->subbands = SBC_ENC_DEFAULT_SUB_BANDS;
self->blocks = SBC_ENC_DEFAULT_BLOCKS;
self->mode = SBC_ENC_DEFAULT_MODE;
self->allocation = SBC_ENC_DEFAULT_ALLOCATION;
}