/*

 *  Copyright © 2010 Nokia Corporation.

 *  This material, including documentation and any related

 *  computer progrs, is protected by copyright controlled by

 *  Nokia Corporation. All rights are reserved. Copying,

 *  including reproducing, storing, adapting or translating, any

 *  or all of this material requires the prior written consent of

 *  Nokia Corporation. This material also contains confidential

 *  information which may not be disclosed to others without the

 *  prior written consent of Nokia Corporation.

 * ============================================================================

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include "gstamrmux.h"

#include <stdlib.h>

#include <string.h>

#ifdef __SYMBIAN32__

#include <gst/gstinfo.h>

#endif

/* AMR nb header value */

const char* amrnbmagicnumber = "#!AMR\n";

static void gst_amrmux_base_init (gpointer g_class);

static void gst_amrmux_class_init (GstAmrMuxClass * klass);

static void gst_amrmux_init ( GstAmrMux * filter, GstAmrMuxClass *filter_klass);

static GstFlowReturn gst_amrmux_chain (GstPad * pad, GstBuffer * buf);

static void gst_amrmux_get_property (GObject * object, guint prop_id,

    GValue * value, GParamSpec * pspec);

static void gst_amrmux_set_property (GObject *object, guint prop_id,

                                     const GValue *value, GParamSpec *pspec);

static GstStateChangeReturn gst_amrmux_change_state (GstElement * element,

    GstStateChange transition);

static void gst_amrmux_dispose(GObject * object);

static GstElementClass *parent_class= NULL;

static const GstElementDetails gst_amrmux_details =

GST_ELEMENT_DETAILS ("AMR MUX Details",

        "Codec/Muxer/Audio",

        "Adaptive Multi-Rate Narrow-Band Muxer",

        "");

static GstStaticPadTemplate sink_factory =

GST_STATIC_PAD_TEMPLATE (

  "sink",

  GST_PAD_SINK,

  GST_PAD_ALWAYS,

  GST_STATIC_CAPS ("audio/amr")

);

static GstStaticPadTemplate source_factory =

GST_STATIC_PAD_TEMPLATE (

  "src",

  GST_PAD_SRC,

  GST_PAD_ALWAYS,

  GST_STATIC_CAPS ("audio/amr")

);

enum {

  ARG_0,

  ARG_HEADER

};

GType

gst_amrmux_get_type (void)

{

  static GType gst_amrmux_type = 0;

  if (!gst_amrmux_type) {

    static const GTypeInfo amrmux_info = {

      sizeof (GstAmrMuxClass),

      gst_amrmux_base_init,

      NULL,

      (GClassInitFunc) gst_amrmux_class_init,

      NULL,

      NULL,

      sizeof (GstAmrMux),

      0,

      (GInstanceInitFunc) gst_amrmux_init,

    };

    gst_amrmux_type =

        g_type_register_static (GST_TYPE_ELEMENT, "GstAmrMux",

        &amrmux_info, (GTypeFlags)0);

  }

  return gst_amrmux_type;

}

static void

gst_amrmux_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(&sink_factory)); 

  gst_element_class_add_pad_template(element_class,

  gst_static_pad_template_get(&source_factory));

  gst_element_class_set_details (element_class, &gst_amrmux_details);

}

static void

gst_amrmux_class_init ( GstAmrMuxClass * klass )

{

  GObjectClass *object_class;

  GstElementClass *gstelement_class;

  gstelement_class = (GstElementClass *) klass;

  object_class = (GObjectClass *) klass;

  object_class->set_property = gst_amrmux_set_property;

  object_class->get_property = gst_amrmux_get_property;

  gstelement_class->change_state = gst_amrmux_change_state;

  object_class->dispose = gst_amrmux_dispose;

  parent_class = g_type_class_peek_parent(klass);

  g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_HEADER,

        g_param_spec_boolean ("header", "header",

            "For writing AMR header", TRUE, (GParamFlags)G_PARAM_READWRITE));

 }

static void gst_amrmux_dispose(GObject * object)

{

    G_OBJECT_CLASS (parent_class)->dispose (object);

}

static void

gst_amrmux_init( GstAmrMux * amrmux, GstAmrMuxClass *amrmux_klass)

{  

   GstElementClass *klass = GST_ELEMENT_CLASS (amrmux_klass);

   //By default we have to write header   

   amrmux->writeheader= TRUE;

   //Add Sink pad to this element

   amrmux->sinkpad = gst_pad_new_from_template (

    gst_element_class_get_pad_template (klass, "sink"), "sink");  

  gst_element_add_pad (GST_ELEMENT (amrmux), amrmux->sinkpad);

  //Add Src pad to this element

  amrmux->srcpad = gst_pad_new_from_template (

    gst_element_class_get_pad_template (klass, "src"), "src");

  gst_element_add_pad (GST_ELEMENT (amrmux), amrmux->srcpad);  

  gst_pad_set_chain_function (amrmux->sinkpad, gst_amrmux_chain);

}

static void

gst_amrmux_get_property (GObject * object,

    guint prop_id, GValue * value, GParamSpec * pspec)

{

  GstAmrMux *amrmux = GST_AMRMUX( object );

  switch (prop_id) {

      case ARG_HEADER:

            g_value_set_boolean (value, amrmux->writeheader);

            break;

          default:

            G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);

            break;

  }

}

static void

gst_amrmux_set_property (GObject * object,

    guint prop_id, const GValue * value, GParamSpec * pspec)

{

GstAmrMux *amrmux = GST_AMRMUX( object );

  switch (prop_id) {

      case ARG_HEADER:

          amrmux->writeheader = g_value_get_boolean (value);

          break;

        default:

          G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);

          break;

  }

}

static GstFlowReturn

gst_amrmux_chain (GstPad * pad, GstBuffer * buf)

{    

  gint8 *codecdata;

  guint8* amrdata;

  guint8* amrheaderstart; 

  int codecdatasize = GST_BUFFER_SIZE(buf); 

  int magicnumberlength = strlen( amrnbmagicnumber );   

  GstAmrMux *amrmux = GST_AMRMUX (GST_PAD_PARENT (pad)); 

  if ( amrmux->writeheader )

   {

      buf = gst_buffer_make_writable( buf );  

      //Allocate a buffer which will hold codec data + magic number

      amrdata = ( guint8* )g_malloc( GST_BUFFER_SIZE(buf) + magicnumberlength );

      //To save the starting address of amr data

      amrheaderstart = amrdata;

      //Copy magic number to newly allocated buffer

      memcpy( amrdata, amrnbmagicnumber, magicnumberlength);

      codecdata = (gint8*)GST_BUFFER_DATA (buf);  

      //Move the pointer to the end of magic number

      amrdata += magicnumberlength;

      //Copy codec data to newly allocated buffer

      memcpy( amrdata , codecdata, codecdatasize );

      // free codec data from GstBuffer

      g_free(  buf->malloc_data );

      //Copying the newly allocated buffer and size in GstBuffer

      buf->data = amrheaderstart;

      buf->malloc_data = amrheaderstart;

      buf->size+= magicnumberlength; 

      //Do it only first time

      amrmux->writeheader = FALSE;  

   }  

  return gst_pad_push( amrmux->srcpad , buf );

}

static GstStateChangeReturn

gst_amrmux_change_state (GstElement * element, GstStateChange transition)

{

  GstStateChangeReturn ret;

  GstAmrMux *amrmux = GST_AMRMUX (element);

  ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);

  if (G_UNLIKELY (ret == GST_STATE_CHANGE_FAILURE))

      return ret; 

  switch (transition) {

        case GST_STATE_CHANGE_READY_TO_NULL: 

         {

            amrmux->writeheader = TRUE; 

            break;

         }               

         default:

            break;

  }

  return ret;    

}

static gboolean

plugin_init (GstPlugin * plugin)

{

  return gst_element_register (plugin, "amrmux", GST_RANK_PRIMARY,

          GST_TYPE_AMRMUX);

}

GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,

    GST_VERSION_MINOR,

    "amrmux",

    "Add header to amr-nb encoded stream",

    plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)

EXPORT_C GstPluginDesc* _GST_PLUGIN_DESC()

{

	return &gst_plugin_desc;

}