diff --git a/ext/sbc/gstsbcenc.c b/ext/sbc/gstsbcenc.c index 07f1628c23..05596ef2e6 100644 --- a/ext/sbc/gstsbcenc.c +++ b/ext/sbc/gstsbcenc.c @@ -25,7 +25,7 @@ #include #endif -#include "sbc.h" +#include #include "gstsbcenc.h" @@ -40,23 +40,133 @@ GST_ELEMENT_DETAILS ("Bluetooth SBC encoder", "Encode a SBC audio stream", "Marcel Holtmann "); +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) BYTE_ORDER, " + "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")); + +static GstFlowReturn +sbc_enc_chain (GstPad * pad, GstBuffer * buffer) +{ + GstSbcEnc *enc = GST_SBC_ENC (gst_pad_get_parent (pad)); + GstFlowReturn res = GST_FLOW_OK; + GstStructure *structure; + gint rate, channels; + guint size, offset = 0; + guint8 *data; + + data = GST_BUFFER_DATA (buffer); + size = GST_BUFFER_SIZE (buffer); + + structure = gst_caps_get_structure (GST_PAD_CAPS (pad), 0); + gst_structure_get_int (structure, "rate", &rate); + gst_structure_get_int (structure, "channels", &channels); + + enc->sbc.rate = rate; + enc->sbc.channels = channels; + enc->sbc.subbands = 8; + enc->sbc.joint = 0; + + 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_class_init (GstSbcEncClass * klass) { + GstElementClass *element_class = GST_ELEMENT_CLASS (klass); + parent_class = g_type_class_peek_parent (klass); + element_class->change_state = GST_DEBUG_FUNCPTR (sbc_enc_change_state); + GST_DEBUG_CATEGORY_INIT (sbc_enc_debug, "sbcenc", 0, "SBC encoding element"); } static void -gst_sbc_enc_init (GstSbcEnc * sbcenc, GstSbcEncClass * klass) +gst_sbc_enc_init (GstSbcEnc * self, GstSbcEncClass * klass) { + self->sinkpad = + gst_pad_new_from_static_template (&sbc_enc_sink_factory, "sink"); + 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); } diff --git a/ext/sbc/gstsbcenc.h b/ext/sbc/gstsbcenc.h index 64457d2131..491135c16a 100644 --- a/ext/sbc/gstsbcenc.h +++ b/ext/sbc/gstsbcenc.h @@ -23,6 +23,8 @@ #include +#include "sbc.h" + G_BEGIN_DECLS #define GST_TYPE_SBC_ENC \ @@ -44,6 +46,8 @@ struct _GstSbcEnc { GstPad *sinkpad; GstPad *srcpad; + + sbc_t sbc; }; struct _GstSbcEncClass {