diff options
Diffstat (limited to 'gst/siren/gstsirenenc.c')
-rw-r--r-- | gst/siren/gstsirenenc.c | 330 |
1 files changed, 330 insertions, 0 deletions
diff --git a/gst/siren/gstsirenenc.c b/gst/siren/gstsirenenc.c new file mode 100644 index 00000000..5ffb6641 --- /dev/null +++ b/gst/siren/gstsirenenc.c @@ -0,0 +1,330 @@ +/* + * Siren Encoder Gst Element + * + * @author: Youness Alaoui <kakaroto@kakaroto.homelinux.net> + * + * 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 "gstsirenenc.h" + +#include <string.h> + +GST_DEBUG_CATEGORY (sirenenc_debug); +#define GST_CAT_DEFAULT (sirenenc_debug) + +/* elementfactory information */ +static const GstElementDetails gst_siren_enc_details = +GST_ELEMENT_DETAILS ( + "Siren Encoder element", + "Codec/Encoder/Audio ", + "Encode 16bit PCM streams into the Siren7 codec", + "Youness Alaoui <kakaroto@kakaroto.homelinux.net>"); + + +static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src", + GST_PAD_SRC, + GST_PAD_ALWAYS, + GST_STATIC_CAPS ("audio/x-siren, " + "dct-length = (int) 320")); + +static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink", + GST_PAD_SINK, + GST_PAD_ALWAYS, + GST_STATIC_CAPS ("audio/x-raw-int, " + "width = (int) 16, " + "depth = (int) 16, " + "endianness = (int) 1234, " + "signed = (boolean) true, " + "rate = (int) 16000, " + "channels = (int) 1")); + +/* signals and args */ +enum +{ + /* FILL ME */ + LAST_SIGNAL +}; + +enum +{ + ARG_0, +}; + + + +static void gst_siren_enc_dispose (GObject *object); + +static GstFlowReturn gst_siren_enc_transform (GstBaseTransform *trans, + GstBuffer *inbuf, GstBuffer *outbuf); +static gboolean gst_siren_enc_transform_size (GstBaseTransform *trans, + GstPadDirection direction, GstCaps *caps, guint size, + GstCaps *othercaps, guint *othersize); +static GstCaps * gst_siren_enc_transform_caps (GstBaseTransform * base, + GstPadDirection direction, GstCaps * caps); +static gboolean gst_siren_enc_start (GstBaseTransform *trans); +static gboolean gst_siren_enc_stop (GstBaseTransform *trans); + + +static void +_do_init (GType type) +{ + GST_DEBUG_CATEGORY_INIT + (sirenenc_debug, "sirenenc", 0, "sirenenc"); +} + +GST_BOILERPLATE_FULL (GstSirenEnc, gst_siren_enc, GstBaseTransform, + GST_TYPE_BASE_TRANSFORM, _do_init); + +static void +gst_siren_enc_base_init (gpointer klass) +{ + GstElementClass *element_class = GST_ELEMENT_CLASS (klass); + + gst_element_class_add_pad_template (element_class, + gst_static_pad_template_get (&srctemplate)); + gst_element_class_add_pad_template (element_class, + gst_static_pad_template_get (&sinktemplate)); + + gst_element_class_set_details (element_class, &gst_siren_enc_details); +} + +static void +gst_siren_enc_class_init (GstSirenEncClass *klass) +{ + GObjectClass *gobject_class; + GstBaseTransformClass *gstbasetransform_class; + + gobject_class = (GObjectClass *) klass; + gstbasetransform_class = (GstBaseTransformClass *) klass; + + GST_DEBUG ("Initializing Class"); + + gobject_class->dispose = GST_DEBUG_FUNCPTR (gst_siren_enc_dispose); + + gstbasetransform_class->transform = + GST_DEBUG_FUNCPTR (gst_siren_enc_transform); + gstbasetransform_class->transform_caps = + GST_DEBUG_FUNCPTR (gst_siren_enc_transform_caps); + gstbasetransform_class->transform_size = + GST_DEBUG_FUNCPTR (gst_siren_enc_transform_size); + gstbasetransform_class->start = + GST_DEBUG_FUNCPTR (gst_siren_enc_start); + gstbasetransform_class->stop = + GST_DEBUG_FUNCPTR (gst_siren_enc_stop); + + parent_class = g_type_class_peek_parent (klass); + + GST_DEBUG ("Class Init done"); +} + +static void +gst_siren_enc_init (GstSirenEnc *enc, GstSirenEncClass *klass) +{ + + GST_DEBUG ("Initializing"); + enc->encoder = NULL; + enc->adapter = gst_adapter_new (); + + GST_DEBUG ("Init done"); +} + +static void +gst_siren_enc_dispose (GObject *object) +{ + GstSirenEnc *enc = GST_SIREN_ENC (object); + + GST_DEBUG ("Disposing"); + + if (enc->encoder) { + Siren7_CloseEncoder (enc->encoder); + enc->encoder = NULL; + } + if (enc->adapter) { + g_object_unref (enc->adapter); + enc->adapter = NULL; + } + + G_OBJECT_CLASS (parent_class)->dispose (object); + + GST_DEBUG ("Dispose done"); + +} + +static gboolean +gst_siren_enc_start (GstBaseTransform *trans) +{ + GstSirenEnc *enc = GST_SIREN_ENC (trans); + + GST_DEBUG ("Start"); + + if (enc->encoder) { + Siren7_CloseEncoder (enc->encoder); + enc->encoder = NULL; + } + enc->encoder = Siren7_NewEncoder (16000); + gst_adapter_clear (enc->adapter); + + return enc->encoder != NULL; +} + + + +static gboolean +gst_siren_enc_stop (GstBaseTransform *trans) +{ + GstSirenEnc *enc = GST_SIREN_ENC (trans); + + GST_DEBUG ("Stop"); + + if (enc->encoder) { + Siren7_CloseEncoder (enc->encoder); + enc->encoder = NULL; + } + gst_adapter_clear (enc->adapter); + return TRUE; +} + + +static gboolean gst_siren_enc_transform_size (GstBaseTransform *trans, + GstPadDirection direction, GstCaps *caps, guint size, + GstCaps *othercaps, guint *othersize) +{ + GstSirenEnc *enc = GST_SIREN_ENC (trans); + GstStructure *structure; + const gchar *in_name; + const gchar *out_name; + + if (caps == NULL || othercaps == NULL) { + GST_WARNING ("caps NULL"); + return FALSE; + } + + structure = gst_caps_get_structure (caps, 0); + in_name = gst_structure_get_name (structure); + structure = gst_caps_get_structure (othercaps, 0); + out_name = gst_structure_get_name (structure); + + GST_DEBUG ("Transform size from caps '%s' to '%s'", in_name, out_name); + + if (in_name == NULL || out_name == NULL) { + return FALSE; + } else if (strcmp (in_name, "audio/x-raw-int") == 0 && + strcmp (out_name, "audio/x-siren") == 0 ) { + size += gst_adapter_available (enc->adapter); + *othersize = size / 16; + *othersize -= *othersize % 40; + if ((size / 16) % 40 > 0) + *othersize+=40; + } else if (strcmp (in_name, "audio/x-siren") == 0 && + strcmp (out_name, "audio/x-raw-int") == 0 ) { + *othersize = size * 16; + *othersize -= *othersize % 640; + if ((size * 16) % 640 > 0) + *othersize+= 640; + } else { + GST_DEBUG ("Unknown in/out caps"); + return FALSE; + } + + GST_DEBUG ("Transform size %d to %d", size, *othersize); + + return TRUE; +} + +static GstCaps * gst_siren_enc_transform_caps (GstBaseTransform * base, + GstPadDirection direction, GstCaps * caps) +{ + GST_DEBUG ("Transforming caps"); + + if (direction == GST_PAD_SRC) { + return gst_static_pad_template_get_caps (&sinktemplate); + } else if (direction == GST_PAD_SINK) { + return gst_static_pad_template_get_caps (&srctemplate); + } + + /* Make gcc happy */ + return NULL; +} + +static GstFlowReturn +gst_siren_enc_transform (GstBaseTransform *trans, GstBuffer *inbuf, + GstBuffer *outbuf) +{ + + GstSirenEnc *enc = GST_SIREN_ENC (trans); + GstFlowReturn ret = GST_FLOW_OK; + guint8 *data = NULL; + gint offset = 0; + gint encode_ret = 0; + + GST_DEBUG ("Transform"); + + if (enc->encoder == NULL) { + GST_DEBUG ("Siren encoder not set"); + return GST_FLOW_WRONG_STATE; + } + if (enc->adapter == NULL) { + GST_DEBUG ("Adapter not set"); + return GST_FLOW_UNEXPECTED; + } + + gst_buffer_ref (inbuf); + gst_adapter_push (enc->adapter, inbuf); + + GST_DEBUG ("Received buffer of size %d with adapter of size : %d", + GST_BUFFER_SIZE (inbuf), gst_adapter_available (enc->adapter)); + + data = GST_BUFFER_DATA (outbuf); + offset = 0; + while(gst_adapter_available (enc->adapter) >= 640 && + offset + 40 <= GST_BUFFER_SIZE (outbuf) && + ret == GST_FLOW_OK) { + + GST_DEBUG ("Encoding frame"); + + encode_ret = Siren7_EncodeFrame (enc->encoder, + (guint8 *)gst_adapter_peek (enc->adapter, 640), + data + offset); + if (encode_ret != 0) { + GST_DEBUG ("Siren7_EncodeFrame returned %d", encode_ret); + ret = GST_FLOW_ERROR; + } + + gst_adapter_flush (enc->adapter, 640); + offset += 40; + } + + GST_DEBUG ("Finished encoding : %d", offset); + + GST_BUFFER_SIZE (outbuf) = offset; + + return ret; +} + + + +gboolean +gst_siren_enc_plugin_init (GstPlugin *plugin) +{ + return gst_element_register (plugin, "sirenenc", + GST_RANK_MARGINAL, GST_TYPE_SIREN_ENC); +} |