Implement full decoding support

2 files changed, 142 insertions, 2 deletions

diff --git a/audio/gstsbcdec.c b/audio/gstsbcdec.c
index 9d547eda..ceea4383 100644
--- a/audio/gstsbcdec.c
+++ b/audio/gstsbcdec.c
@@ -25,7 +25,7 @@
 #include <config.h>
 #endif
 
-#include "sbc.h"
+#include <string.h>
 
 #include "gstsbcdec.h"
 
@@ -40,21 +40,155 @@ static const GstElementDetails sbc_dec_details =
 				"Decode a SBC audio stream",
 				"Marcel Holtmann <marcel@holtmann.org>");
 
+static GstStaticPadTemplate sbc_dec_sink_factory =
+	GST_STATIC_PAD_TEMPLATE("sink", GST_PAD_SINK, GST_PAD_ALWAYS,
+		GST_STATIC_CAPS("audio/x-sbc"));
+
+static GstStaticPadTemplate sbc_dec_src_factory =
+	GST_STATIC_PAD_TEMPLATE("src", GST_PAD_SRC, GST_PAD_ALWAYS,
+		GST_STATIC_CAPS("audio/x-raw-int, "
+				"rate = (int) [ 6000, 48000 ], "
+				"channels = (int) [ 1, 2 ], "
+				"endianness = (int) BYTE_ORDER, "
+				"signed = (boolean) true, "
+				"width = (int) 16, "
+				"depth = (int) 16"));
+
+static GstFlowReturn sbc_dec_chain(GstPad *pad, GstBuffer *buffer)
+{
+	GstSbcDec *dec = GST_SBC_DEC(gst_pad_get_parent(pad));
+	GstFlowReturn res = GST_FLOW_OK;
+	guint size, offset = 0;
+	guint8 *data;
+	GstClockTime timestamp;
+
+	timestamp = GST_BUFFER_TIMESTAMP(buffer);
+
+	if (dec->buffer) {
+		GstBuffer *temp = buffer;
+		buffer = gst_buffer_span(dec->buffer, 0, buffer,
+			GST_BUFFER_SIZE(dec->buffer) + GST_BUFFER_SIZE(buffer));
+		gst_buffer_unref(temp);
+		gst_buffer_unref(dec->buffer);
+		dec->buffer = NULL;
+	}
+
+	data = GST_BUFFER_DATA(buffer);
+	size = GST_BUFFER_SIZE(buffer);
+
+	while (offset < size) {
+		GstBuffer *output;
+		GstPadTemplate *template;
+		GstCaps *caps, *temp;
+		int consumed;
+
+		consumed = sbc_decode(&dec->sbc, data + offset, size - offset);
+		if (consumed <= 0)
+			break;
+
+		caps = gst_caps_new_simple("audio/x-raw-int",
+				"rate", G_TYPE_INT, dec->sbc.rate,
+				"channels", G_TYPE_INT, dec->sbc.channels,
+				NULL);
+
+		template = gst_static_pad_template_get(&sbc_dec_src_factory);
+
+		temp = gst_caps_intersect(caps,
+					gst_pad_template_get_caps(template));
+
+		gst_caps_unref(caps);
+
+		res = gst_pad_alloc_buffer_and_set_caps(dec->srcpad,
+						GST_BUFFER_OFFSET_NONE,
+						dec->sbc.len, temp, &output);
+
+		gst_caps_unref(temp);
+
+		if (res != GST_FLOW_OK)
+			goto done;
+
+		memcpy(GST_BUFFER_DATA(output), dec->sbc.data, dec->sbc.len);
+
+		res = gst_pad_push(dec->srcpad, output);
+		if (res != GST_FLOW_OK)
+			goto done;
+
+		offset += consumed;
+	}
+
+	if (offset < size)
+		dec->buffer = gst_buffer_create_sub(buffer,
+							offset, size - offset);
+
+done:
+	gst_buffer_unref(buffer);
+	gst_object_unref(dec);
+
+	return res;
+}
+
+static GstStateChangeReturn sbc_dec_change_state(GstElement *element,
+						GstStateChange transition)
+{
+	GstSbcDec *dec = GST_SBC_DEC(element);
+
+	switch (transition) {
+	case GST_STATE_CHANGE_READY_TO_PAUSED:
+		GST_DEBUG("Setup subband codec");
+		if (dec->buffer) {
+			gst_buffer_unref(dec->buffer);
+			dec->buffer = NULL;
+		}
+		sbc_init(&dec->sbc, 0);
+		break;
+
+	case GST_STATE_CHANGE_PAUSED_TO_READY:
+		GST_DEBUG("Finish subband codec");
+		if (dec->buffer) {
+			gst_buffer_unref(dec->buffer);
+			dec->buffer = NULL;
+		}
+		sbc_finish(&dec->sbc);
+		break;
+
+	default:
+		break;
+	}
+
+	return parent_class->change_state(element, transition);
+}
+
 static void gst_sbc_dec_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_dec_sink_factory));
+
+	gst_element_class_add_pad_template(element_class,
+		gst_static_pad_template_get(&sbc_dec_src_factory));
+
 	gst_element_class_set_details(element_class, &sbc_dec_details);
 }
 
 static void gst_sbc_dec_class_init(GstSbcDecClass *klass)
 {
+	GstElementClass *gstelement_class = GST_ELEMENT_CLASS(klass);
+
 	parent_class = g_type_class_peek_parent(klass);
 
+	gstelement_class->change_state = GST_DEBUG_FUNCPTR(sbc_dec_change_state);
+
 	GST_DEBUG_CATEGORY_INIT(sbc_dec_debug, "sbcdec", 0,
 						"SBC decoding element");
 }
 
-static void gst_sbc_dec_init(GstSbcDec *sbcdec, GstSbcDecClass *klass)
+static void gst_sbc_dec_init(GstSbcDec *self, GstSbcDecClass *klass)
 {
+	self->sinkpad = gst_pad_new_from_static_template(&sbc_dec_sink_factory, "sink");
+	gst_pad_set_chain_function(self->sinkpad, GST_DEBUG_FUNCPTR(sbc_dec_chain));
+	gst_element_add_pad(GST_ELEMENT(self), self->sinkpad);
+
+	self->srcpad = gst_pad_new_from_static_template(&sbc_dec_src_factory, "src");
+	gst_element_add_pad(GST_ELEMENT(self), self->srcpad);
 }
diff --git a/audio/gstsbcdec.h b/audio/gstsbcdec.h
index 0737b9d9..4a6922a0 100644
--- a/audio/gstsbcdec.h
+++ b/audio/gstsbcdec.h
@@ -23,6 +23,8 @@
 
 #include <gst/gst.h>
 
+#include "sbc.h"
+
 G_BEGIN_DECLS
 
 #define GST_TYPE_SBC_DEC \
@@ -44,6 +46,10 @@ struct _GstSbcDec {
 
 	GstPad *sinkpad;
 	GstPad *srcpad;
+
+	GstBuffer *buffer;
+
+	sbc_t sbc;
 };
 
 struct _GstSbcDecClass {