sbc: Implement full decoding support

diff --git a/ext/sbc/gstsbcdec.c b/ext/sbc/gstsbcdec.c
index f7067a9..1f75e38 100644 (file)
--- a/ext/sbc/gstsbcdec.c
+++ b/ext/sbc/gstsbcdec.c
@@ -25,7 +25,7 @@
 #include <config.h>
 #endif
 
-#include "sbc.h"
+#include <string.h>
 
 #include "gstsbcdec.h"
 
@@ -40,23 +40,153 @@ GST_ELEMENT_DETAILS ("Bluetooth SBC decoder",
     "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)
+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/ext/sbc/gstsbcdec.h b/ext/sbc/gstsbcdec.h
index 0737b9d..4a6922a 100644 (file)
--- a/ext/sbc/gstsbcdec.h
+++ b/ext/sbc/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 {