ext/libfame/gstlibfame.c - mtk-gst-plugins-bad - Git at Google

 /* GStreamer
  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
  *
  * 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 <fame.h>
 #include <string.h>
 #include <math.h>

 #include "gstlibfame.h"
 #include <gst/video/video.h>

 #define FAMEENC_BUFFER_SIZE (300 * 1024)

 /* elementfactory information */
 static GstElementDetails gst_fameenc_details = {
   "MPEG1 and MPEG4 video encoder using the libfame library",
   "Codec/Video/Encoder",
   "Uses fameenc to encode MPEG video streams",
   "Thomas Vander Stichele <thomas@apestaart.org>"
 };

 static GQuark fame_object_name;


 /* FameEnc signals and args */
 enum {
   /* FILL ME */
   LAST_SIGNAL
 };

 enum {
   ARG_0,
   ARG_VERSION,
   ARG_BITRATE,
   ARG_QUALITY,
   ARG_PATTERN,
   ARG_FAME_VERBOSE,
   ARG_BUFFER_SIZE,
   ARG_FRAMES_PER_SEQUENCE,
   /* dynamically generated properties start here */
   ARG_FAME_PROPS_START
   /* FILL ME */
 };

 static GstStaticPadTemplate sink_template =
 GST_STATIC_PAD_TEMPLATE (
   "sink",
   GST_PAD_SINK,
   GST_PAD_ALWAYS,
   GST_STATIC_CAPS (
     "video/x-raw-yuv, "
     "format = (fourcc) I420, "
     "width = (int) [ 16, 4096 ], "
     "height = (int) [ 16, 4096 ], "
     "framerate = (double) { 23.976024, 24.0, 25.0, 29.970030, 30.0, "
     "  50.0, 59.940060, 60.0 }"
   )
 );

 static GstStaticPadTemplate src_template =
 GST_STATIC_PAD_TEMPLATE (
   "src",
   GST_PAD_SRC,
   GST_PAD_ALWAYS,
   GST_STATIC_CAPS (
     "video/mpeg, "
     "mpegversion = (int) { 1, 4 }, "
     "systemstream = (boolean) FALSE, "
     "width = (int) [ 16, 4096 ], "
     "height = (int) [ 16, 4096 ], "
     "framerate = (double) { 23.976024, 24.0, 25.0, 29.970030, 30.0, "
     "  50.0, 59.940060, 60.0 }"
   )
 );

 #define MAX_FRAME_RATES  9
 typedef struct
 {
   gint num;
   gint den;
 } frame_rate_entry;

 static const frame_rate_entry frame_rates[] =
 {
   { 0, 0 },
   { 24000, 1001 },
   { 24, 1 },
   { 25, 1 },
   { 30000, 1001 },
   { 30, 1 },
   { 50, 1 },
   { 60000, 1001 },
   { 60, 1 },
 };

 static gint
 framerate_to_index (gfloat fps)
 {
   gint i;
   gint idx = -1;

   for (i = 1; i < MAX_FRAME_RATES; i++) {
     if (idx == -1) {
       idx = i;
     } else {
       gfloat old_diff = fabs((1. * frame_rates[idx].num /
 				frame_rates[idx].den) - fps),
              new_diff = fabs((1. * frame_rates[i].num /
 				frame_rates[i].den) - fps);

       if (new_diff < old_diff) {
         idx = i;
       }
     }
   }

   return idx;
 }

 static void	gst_fameenc_class_init		(GstFameEncClass *klass);
 static void	gst_fameenc_base_init		(GstFameEncClass *klass);
 static void	gst_fameenc_init		(GstFameEnc *fameenc);
 static void	gst_fameenc_dispose		(GObject *object);

 static void	gst_fameenc_set_property	(GObject *object, guint prop_id,
 						 const GValue *value, GParamSpec *pspec);
 static void	gst_fameenc_get_property	(GObject *object, guint prop_id,
 						 GValue *value, GParamSpec *pspec);

 static void	gst_fameenc_chain		(GstPad *pad, GstData *_data);

 static GstElementClass *parent_class = NULL;
 /*static guint gst_fameenc_signals[LAST_SIGNAL] = { 0 };*/

 GType
 gst_fameenc_get_type (void)
 {
   static GType fameenc_type = 0;

   if (!fameenc_type) {
     static const GTypeInfo fameenc_info = {
       sizeof (GstFameEncClass),
       (GBaseInitFunc) gst_fameenc_base_init,
       NULL,
       (GClassInitFunc) gst_fameenc_class_init,
       NULL,
       NULL,
       sizeof (GstFameEnc),
       0,
       (GInstanceInitFunc) gst_fameenc_init,
     };
     fameenc_type = g_type_register_static (GST_TYPE_ELEMENT,
 	                                   "GstFameEnc", &fameenc_info, 0);
   }
   return fameenc_type;
 }

 static int
 gst_fameenc_item_compare (fame_list_t *item1, fame_list_t *item2)
 {
   return strcmp (item1->type, item2->type);
 }

 static void
 gst_fameenc_base_init (GstFameEncClass *klass)
 {
   GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

   gst_element_class_add_pad_template (element_class,
 		gst_static_pad_template_get (&sink_template));
   gst_element_class_add_pad_template (element_class,
 		gst_static_pad_template_get (&src_template));

   gst_element_class_set_details (element_class, &gst_fameenc_details);
 }

 static void
 gst_fameenc_class_init (GstFameEncClass *klass)
 {
   GObjectClass *gobject_class = NULL;
   GstElementClass *gstelement_class = NULL;
   fame_context_t *context;
   fame_list_t *walk;
   GList *props = NULL, *props_walk;
   gint current_prop = ARG_FAME_PROPS_START;

   gobject_class = (GObjectClass*) klass;
   gstelement_class = (GstElementClass*) klass;

   parent_class = g_type_class_ref (GST_TYPE_ELEMENT);

   gobject_class->set_property = gst_fameenc_set_property;
   gobject_class->get_property = gst_fameenc_get_property;
   gobject_class->dispose = gst_fameenc_dispose;

   fame_object_name = g_quark_from_string ("GstFameObjectName");

   context = fame_open ();
   g_assert (context);

   /* first sort the list */
   walk = context->type_list;
   while (walk) {
     props = g_list_insert_sorted (props, walk, (GCompareFunc)gst_fameenc_item_compare);
     walk = walk->next;
   }

   props_walk = props;
   while (props_walk) {
     GArray *array;
     const gchar *current_type;
     gint current_len;
     gint current_value;
     fame_object_t *current_default;
     gint default_index;

     walk = (fame_list_t *)props_walk->data;
     array = g_array_new (TRUE, FALSE, sizeof (GEnumValue));

     current_type = walk->type;
     current_value = 0;
     current_len = strlen (walk->type);
     current_default = fame_get_object (context, current_type);
     default_index = 1;

     do {
       if (strstr (walk->type, "/")) {
 	GEnumValue value;

 	if (current_default == walk->item)
           default_index = current_value;

 	value.value = current_value++;
 	value.value_name = g_strdup (walk->type);
 	value.value_nick = g_strdup (walk->item->name);

 	g_array_append_val (array, value);
       }

       props_walk = g_list_next (props_walk);
       if (props_walk)
 	walk = (fame_list_t *)props_walk->data;

     } while (props_walk && !strncmp (walk->type, current_type, current_len));

     if (array->len > 0) {
       GType type;
       GParamSpec *pspec;

       type = g_enum_register_static (g_strdup_printf ("GstFameEnc_%s", current_type), (GEnumValue *)array->data);

       pspec = g_param_spec_enum (current_type, current_type, g_strdup_printf ("The FAME \"%s\" object", current_type),
                            type, default_index, G_PARAM_READWRITE);

       g_param_spec_set_qdata (pspec, fame_object_name, (gpointer) current_type);

       g_object_class_install_property (G_OBJECT_CLASS (klass), current_prop++, pspec);
     }
   }

   g_object_class_install_property (gobject_class, ARG_BITRATE,
     g_param_spec_int ("bitrate", "Bitrate", "Target bitrate (0 = VBR)",
                       0, 5000000, 0, G_PARAM_READWRITE));
   g_object_class_install_property (gobject_class, ARG_QUALITY,
     g_param_spec_int ("quality", "Quality", "Percentage of quality of compression (versus size)",
                       0, 100, 75, G_PARAM_READWRITE));
   g_object_class_install_property (gobject_class, ARG_PATTERN,
     g_param_spec_string ("pattern", "Pattern", "Encoding pattern of I, P, and B frames",
                          "IPPPPPPPPPPP", G_PARAM_READWRITE));
   g_object_class_install_property (gobject_class, ARG_FRAMES_PER_SEQUENCE,
     g_param_spec_int ("frames_per_sequence", "Frames Per Sequence",
 	              "The number of frames in one sequence",
                       1, G_MAXINT, 12, G_PARAM_READWRITE));
   g_object_class_install_property (gobject_class, ARG_FAME_VERBOSE,
     g_param_spec_boolean ("fame_verbose", "Fame Verbose", "Make FAME produce verbose output",
                          FALSE, G_PARAM_READWRITE));
   g_object_class_install_property (gobject_class, ARG_BUFFER_SIZE,
     g_param_spec_int ("buffer_size", "Buffer Size", "Set the decoding output buffer size",
                       0, 1024*1024, FAMEENC_BUFFER_SIZE, G_PARAM_READWRITE));
 }

 static GstPadLinkReturn
 gst_fameenc_sink_link (GstPad *pad, const GstCaps *caps)
 {
   gint width, height, fps_idx;
   gdouble fps;
   GstFameEnc *fameenc;
   GstStructure *structure;

   fameenc = GST_FAMEENC (gst_pad_get_parent (pad));

   if (fameenc->initialized) {
     GST_DEBUG ("error: fameenc encoder already initialized !");
     return GST_PAD_LINK_REFUSED;
   }

   structure = gst_caps_get_structure (caps, 0);
   gst_structure_get_int (structure, "width", &width);
   gst_structure_get_int (structure, "height", &height);
   gst_structure_get_double (structure, "framerate", &fps);

   /* fameenc requires width and height to be multiples of 16 */
   if (width % 16 != 0 || height % 16 != 0)
     return GST_PAD_LINK_REFUSED;

   fps_idx = framerate_to_index (fps);
   fameenc->fp.frame_rate_num = frame_rates[fps_idx].num;
   fameenc->fp.frame_rate_den = frame_rates[fps_idx].den;
   fameenc->time_interval = 0;

   fameenc->fp.width = width;
   fameenc->fp.height = height;
   fameenc->fp.coding = (const char *) fameenc->pattern;

   /* FIXME: choose good parameters */
   fameenc->fp.slices_per_frame = 1;

   /* FIXME: handle these properly */
   fameenc->fp.shape_quality = 75;
   fameenc->fp.search_range = 0;
   fameenc->fp.total_frames = 0;
   fameenc->fp.retrieve_cb = NULL;

   fame_init (fameenc->fc, &fameenc->fp, fameenc->buffer, fameenc->buffer_size);

   fameenc->initialized = TRUE;
   fameenc->time_interval = 0;

   return GST_PAD_LINK_OK;
 }

 static void
 gst_fameenc_init (GstFameEnc *fameenc)
 {
   g_assert (fameenc != NULL);
   g_assert (GST_IS_FAMEENC (fameenc));

   /* open fameenc */
   fameenc->fc = fame_open ();
   g_assert (fameenc->fc != NULL);

   /* create the sink and src pads */
   fameenc->sinkpad = gst_pad_new_from_template (
 		       gst_static_pad_template_get (&sink_template), "sink");
   gst_element_add_pad (GST_ELEMENT (fameenc), fameenc->sinkpad);
   gst_pad_set_chain_function (fameenc->sinkpad, gst_fameenc_chain);
   gst_pad_set_link_function (fameenc->sinkpad, gst_fameenc_sink_link);

   fameenc->srcpad = gst_pad_new_from_template (
                       gst_static_pad_template_get (&src_template), "src");
   gst_element_add_pad (GST_ELEMENT (fameenc), fameenc->srcpad);
   /* FIXME: set some more handler functions here */

   fameenc->verbose = FALSE;

   /* reset the initial video state */
   fameenc->fp.width = -1;
   fameenc->fp.height = -1;
   fameenc->initialized = FALSE;

   /* defaults */
   fameenc->fp.bitrate = 0;
   fameenc->fp.quality = 75;
   fameenc->fp.frame_rate_num = 25;
   fameenc->fp.frame_rate_den = 1; /* avoid floating point exceptions */
   fameenc->fp.frames_per_sequence = 12;

   fameenc->pattern = g_strdup ("IPPPPPPPPPP");

   /* allocate space for the buffer */
   fameenc->buffer_size = FAMEENC_BUFFER_SIZE; /* FIXME */
   fameenc->buffer = (unsigned char *) g_malloc (fameenc->buffer_size);

   fameenc->next_time = 0;
   fameenc->time_interval = 0;
 }

 static void
 gst_fameenc_dispose (GObject *object)
 {
   GstFameEnc *fameenc = GST_FAMEENC (object);

   G_OBJECT_CLASS (parent_class)->dispose (object);

   g_free (fameenc->buffer);
 }

 static void
 gst_fameenc_chain (GstPad *pad, GstData *_data)
 {
   GstBuffer *buf = GST_BUFFER (_data);
   GstFameEnc *fameenc;
   guchar *data;
   gulong size;
   gint frame_size;
   gint length;

   g_return_if_fail (pad != NULL);
   g_return_if_fail (GST_IS_PAD (pad));
   g_return_if_fail (buf != NULL);
   g_return_if_fail (GST_IS_BUFFER (buf));

   fameenc = GST_FAMEENC (gst_pad_get_parent (pad));

   data = (guchar *) GST_BUFFER_DATA (buf);
   size = GST_BUFFER_SIZE (buf);

   GST_DEBUG ("gst_fameenc_chain: got buffer of %ld bytes in '%s'",
 	     size, GST_OBJECT_NAME (fameenc));

   /* the data contains the three planes side by side, with size w * h, w * h /4,
    * w * h / 4 */
   fameenc->fy.w = fameenc->fp.width;
   fameenc->fy.h = fameenc->fp.height;

   frame_size = fameenc->fp.width * fameenc->fp.height;

   fameenc->fy.p = 0;
   fameenc->fy.y = data;
   fameenc->fy.u = data + frame_size;
   fameenc->fy.v = fameenc->fy.u + (frame_size >> 2);

   fame_start_frame (fameenc->fc, &fameenc->fy, NULL);

   while ((length = fame_encode_slice (fameenc->fc)) != 0) {
     GstBuffer *outbuf;

     outbuf = gst_buffer_new ();

     /* FIXME: safeguard, remove me when a better way is found */
     if (length > FAMEENC_BUFFER_SIZE)
       g_warning ("FAMEENC_BUFFER_SIZE is defined too low, encoded slice has size %d !\n", length);

     if (!fameenc->time_interval) {
   	fameenc->time_interval = GST_SECOND * fameenc->fp.frame_rate_den / fameenc->fp.frame_rate_num;
     }

     fameenc->next_time += fameenc->time_interval;

     GST_BUFFER_SIZE (outbuf) = length;
     GST_BUFFER_TIMESTAMP (outbuf) = fameenc->next_time;
     GST_BUFFER_DATA (outbuf) = g_malloc (length);
     memcpy (GST_BUFFER_DATA(outbuf), fameenc->buffer, length);
     GST_BUFFER_TIMESTAMP (outbuf) = GST_BUFFER_TIMESTAMP (buf);

     GST_DEBUG ("gst_fameenc_chain: pushing buffer of size %d",
                GST_BUFFER_SIZE(outbuf));

     gst_pad_push (fameenc->srcpad, GST_DATA (outbuf));
   }

   fame_end_frame (fameenc->fc, NULL);

   gst_buffer_unref(buf);
 }

 static void
 gst_fameenc_set_property (GObject *object, guint prop_id,
 	                  const GValue *value, GParamSpec *pspec)
 {
   GstFameEnc *fameenc;

   g_return_if_fail (GST_IS_FAMEENC (object));
   fameenc = GST_FAMEENC (object);

   if (fameenc->initialized) {
     GST_DEBUG ("error: fameenc encoder already initialized, cannot set properties !");
     return;
   }

   switch (prop_id) {
     case ARG_BITRATE:
       fameenc->fp.bitrate = g_value_get_int (value);
       break;
     case ARG_QUALITY:
       fameenc->fp.quality = CLAMP (g_value_get_int (value), 0, 100);
       break;
     case ARG_PATTERN:
       g_free (fameenc->pattern);
       fameenc->pattern = g_strdup (g_value_get_string (value));
       break;
     case ARG_FAME_VERBOSE:
       fameenc->verbose = g_value_get_boolean (value);
       break;
     case ARG_BUFFER_SIZE:
       fameenc->buffer_size = g_value_get_int (value);
       break;
     case ARG_FRAMES_PER_SEQUENCE:
       fameenc->fp.frames_per_sequence = g_value_get_int (value);
       break;
     default:
       if (prop_id >= ARG_FAME_PROPS_START) {
 	gchar *name;
 	gint index = g_value_get_enum (value);
 	GEnumValue *values;

 	values = G_ENUM_CLASS (g_type_class_ref (pspec->value_type))->values;
 	name = (gchar *) g_param_spec_get_qdata (pspec, fame_object_name);

         fame_register (fameenc->fc, name, fame_get_object (fameenc->fc, values[index].value_name));
       }
       else
         G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
       break;
   }
 }

 static void
 gst_fameenc_get_property (GObject *object, guint prop_id,
 	                  GValue *value, GParamSpec *pspec)
 {
   GstFameEnc *fameenc;

   g_return_if_fail (GST_IS_FAMEENC (object));
   fameenc = GST_FAMEENC (object);

   switch (prop_id) {
     case ARG_BITRATE:
       g_value_set_int (value, fameenc->fp.bitrate);
       break;
     case ARG_QUALITY:
       g_value_set_int (value, fameenc->fp.quality);
       break;
     case ARG_PATTERN:
       g_value_set_string (value, g_strdup (fameenc->pattern));
       break;
     case ARG_FAME_VERBOSE:
       g_value_set_boolean (value, fameenc->verbose);
       break;
     case ARG_BUFFER_SIZE:
       g_value_set_int (value, fameenc->buffer_size);
       break;
     case ARG_FRAMES_PER_SEQUENCE:
       g_value_set_int (value, fameenc->fp.frames_per_sequence);
       break;
     default:
       if (prop_id >= ARG_FAME_PROPS_START) {
 	gchar *name;
 	gint index = 0;
 	GEnumValue *values;
 	fame_object_t *f_object;

 	values = G_ENUM_CLASS (g_type_class_ref (pspec->value_type))->values;
 	name = (gchar *) g_param_spec_get_qdata (pspec, fame_object_name);

 	f_object = fame_get_object (fameenc->fc, name);

 	while (values[index].value_name) {
 	  if (!strcmp (values[index].value_nick, f_object->name)) {
             g_value_set_enum (value, index);
 	    return;
 	  }
 	  index++;
 	}
       }
       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
       break;
   }
 }

 static gboolean
 plugin_init (GstPlugin *plugin)
 {
   return gst_element_register (plugin, "fameenc",
 			       GST_RANK_NONE, GST_TYPE_FAMEENC);
 }

 GST_PLUGIN_DEFINE (
   GST_VERSION_MAJOR,
   GST_VERSION_MINOR,
   "fameenc",
   "Fast Assembly MPEG Encoder",
   plugin_init,
   LIBFAME_VERSION,
   "LGPL",
   "libfame",
   "http://fame.sourceforge.net/"
 )
	/* GStreamer
	* Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
	*
	* 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 <fame.h>
	#include <string.h>
	#include <math.h>

	#include "gstlibfame.h"
	#include <gst/video/video.h>

	#define FAMEENC_BUFFER_SIZE (300 * 1024)

	/* elementfactory information */
	static GstElementDetails gst_fameenc_details = {
	"MPEG1 and MPEG4 video encoder using the libfame library",
	"Codec/Video/Encoder",
	"Uses fameenc to encode MPEG video streams",
	"Thomas Vander Stichele <thomas@apestaart.org>"
	};

	static GQuark fame_object_name;


	/* FameEnc signals and args */
	enum {
	/* FILL ME */
	LAST_SIGNAL
	};

	enum {
	ARG_0,
	ARG_VERSION,
	ARG_BITRATE,
	ARG_QUALITY,
	ARG_PATTERN,
	ARG_FAME_VERBOSE,
	ARG_BUFFER_SIZE,
	ARG_FRAMES_PER_SEQUENCE,
	/* dynamically generated properties start here */
	ARG_FAME_PROPS_START
	/* FILL ME */
	};

	static GstStaticPadTemplate sink_template =
	GST_STATIC_PAD_TEMPLATE (
	"sink",
	GST_PAD_SINK,
	GST_PAD_ALWAYS,
	GST_STATIC_CAPS (
	"video/x-raw-yuv, "
	"format = (fourcc) I420, "
	"width = (int) [ 16, 4096 ], "
	"height = (int) [ 16, 4096 ], "
	"framerate = (double) { 23.976024, 24.0, 25.0, 29.970030, 30.0, "
	" 50.0, 59.940060, 60.0 }"
	)
	);

	static GstStaticPadTemplate src_template =
	GST_STATIC_PAD_TEMPLATE (
	"src",
	GST_PAD_SRC,
	GST_PAD_ALWAYS,
	GST_STATIC_CAPS (
	"video/mpeg, "
	"mpegversion = (int) { 1, 4 }, "
	"systemstream = (boolean) FALSE, "
	"width = (int) [ 16, 4096 ], "
	"height = (int) [ 16, 4096 ], "
	"framerate = (double) { 23.976024, 24.0, 25.0, 29.970030, 30.0, "
	" 50.0, 59.940060, 60.0 }"
	)
	);

	#define MAX_FRAME_RATES 9
	typedef struct
	{
	gint num;
	gint den;
	} frame_rate_entry;

	static const frame_rate_entry frame_rates[] =
	{
	{ 0, 0 },
	{ 24000, 1001 },
	{ 24, 1 },
	{ 25, 1 },
	{ 30000, 1001 },
	{ 30, 1 },
	{ 50, 1 },
	{ 60000, 1001 },
	{ 60, 1 },
	};

	static gint
	framerate_to_index (gfloat fps)
	{
	gint i;
	gint idx = -1;

	for (i = 1; i < MAX_FRAME_RATES; i++) {
	if (idx == -1) {
	idx = i;
	} else {
	gfloat old_diff = fabs((1. * frame_rates[idx].num /
	frame_rates[idx].den) - fps),
	new_diff = fabs((1. * frame_rates[i].num /
	frame_rates[i].den) - fps);

	if (new_diff < old_diff) {
	idx = i;
	}
	}
	}

	return idx;
	}

	static void gst_fameenc_class_init (GstFameEncClass *klass);
	static void gst_fameenc_base_init (GstFameEncClass *klass);
	static void gst_fameenc_init (GstFameEnc *fameenc);
	static void gst_fameenc_dispose (GObject *object);

	static void gst_fameenc_set_property (GObject *object, guint prop_id,
	const GValue value, GParamSpec pspec);
	static void gst_fameenc_get_property (GObject *object, guint prop_id,
	GValue value, GParamSpec pspec);

	static void gst_fameenc_chain (GstPad pad, GstData _data);

	static GstElementClass *parent_class = NULL;
	/static guint gst_fameenc_signals[LAST_SIGNAL] = { 0 };/

	GType
	gst_fameenc_get_type (void)
	{
	static GType fameenc_type = 0;

	if (!fameenc_type) {
	static const GTypeInfo fameenc_info = {
	sizeof (GstFameEncClass),
	(GBaseInitFunc) gst_fameenc_base_init,
	NULL,
	(GClassInitFunc) gst_fameenc_class_init,
	NULL,
	NULL,
	sizeof (GstFameEnc),
	0,
	(GInstanceInitFunc) gst_fameenc_init,
	};
	fameenc_type = g_type_register_static (GST_TYPE_ELEMENT,
	"GstFameEnc", &fameenc_info, 0);
	}
	return fameenc_type;
	}

	static int
	gst_fameenc_item_compare (fame_list_t item1, fame_list_t item2)
	{
	return strcmp (item1->type, item2->type);
	}

	static void
	gst_fameenc_base_init (GstFameEncClass *klass)
	{
	GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

	gst_element_class_add_pad_template (element_class,
	gst_static_pad_template_get (&sink_template));
	gst_element_class_add_pad_template (element_class,
	gst_static_pad_template_get (&src_template));

	gst_element_class_set_details (element_class, &gst_fameenc_details);
	}

	static void
	gst_fameenc_class_init (GstFameEncClass *klass)
	{
	GObjectClass *gobject_class = NULL;
	GstElementClass *gstelement_class = NULL;
	fame_context_t *context;
	fame_list_t *walk;
	GList props = NULL, props_walk;
	gint current_prop = ARG_FAME_PROPS_START;

	gobject_class = (GObjectClass*) klass;
	gstelement_class = (GstElementClass*) klass;

	parent_class = g_type_class_ref (GST_TYPE_ELEMENT);

	gobject_class->set_property = gst_fameenc_set_property;
	gobject_class->get_property = gst_fameenc_get_property;
	gobject_class->dispose = gst_fameenc_dispose;

	fame_object_name = g_quark_from_string ("GstFameObjectName");

	context = fame_open ();
	g_assert (context);

	/* first sort the list */
	walk = context->type_list;
	while (walk) {
	props = g_list_insert_sorted (props, walk, (GCompareFunc)gst_fameenc_item_compare);
	walk = walk->next;
	}

	props_walk = props;
	while (props_walk) {
	GArray *array;
	const gchar *current_type;
	gint current_len;
	gint current_value;
	fame_object_t *current_default;
	gint default_index;

	walk = (fame_list_t *)props_walk->data;
	array = g_array_new (TRUE, FALSE, sizeof (GEnumValue));

	current_type = walk->type;
	current_value = 0;
	current_len = strlen (walk->type);
	current_default = fame_get_object (context, current_type);
	default_index = 1;

	do {
	if (strstr (walk->type, "/")) {
	GEnumValue value;

	if (current_default == walk->item)
	default_index = current_value;

	value.value = current_value++;
	value.value_name = g_strdup (walk->type);
	value.value_nick = g_strdup (walk->item->name);

	g_array_append_val (array, value);
	}

	props_walk = g_list_next (props_walk);
	if (props_walk)
	walk = (fame_list_t *)props_walk->data;

	} while (props_walk && !strncmp (walk->type, current_type, current_len));

	if (array->len > 0) {
	GType type;
	GParamSpec *pspec;

	type = g_enum_register_static (g_strdup_printf ("GstFameEnc_%s", current_type), (GEnumValue *)array->data);

	pspec = g_param_spec_enum (current_type, current_type, g_strdup_printf ("The FAME \"%s\" object", current_type),
	type, default_index, G_PARAM_READWRITE);

	g_param_spec_set_qdata (pspec, fame_object_name, (gpointer) current_type);

	g_object_class_install_property (G_OBJECT_CLASS (klass), current_prop++, pspec);
	}
	}

	g_object_class_install_property (gobject_class, ARG_BITRATE,
	g_param_spec_int ("bitrate", "Bitrate", "Target bitrate (0 = VBR)",
	0, 5000000, 0, G_PARAM_READWRITE));
	g_object_class_install_property (gobject_class, ARG_QUALITY,
	g_param_spec_int ("quality", "Quality", "Percentage of quality of compression (versus size)",
	0, 100, 75, G_PARAM_READWRITE));
	g_object_class_install_property (gobject_class, ARG_PATTERN,
	g_param_spec_string ("pattern", "Pattern", "Encoding pattern of I, P, and B frames",
	"IPPPPPPPPPPP", G_PARAM_READWRITE));
	g_object_class_install_property (gobject_class, ARG_FRAMES_PER_SEQUENCE,
	g_param_spec_int ("frames_per_sequence", "Frames Per Sequence",
	"The number of frames in one sequence",
	1, G_MAXINT, 12, G_PARAM_READWRITE));
	g_object_class_install_property (gobject_class, ARG_FAME_VERBOSE,
	g_param_spec_boolean ("fame_verbose", "Fame Verbose", "Make FAME produce verbose output",
	FALSE, G_PARAM_READWRITE));
	g_object_class_install_property (gobject_class, ARG_BUFFER_SIZE,
	g_param_spec_int ("buffer_size", "Buffer Size", "Set the decoding output buffer size",
	0, 1024*1024, FAMEENC_BUFFER_SIZE, G_PARAM_READWRITE));
	}

	static GstPadLinkReturn
	gst_fameenc_sink_link (GstPad pad, const GstCaps caps)
	{
	gint width, height, fps_idx;
	gdouble fps;
	GstFameEnc *fameenc;
	GstStructure *structure;

	fameenc = GST_FAMEENC (gst_pad_get_parent (pad));

	if (fameenc->initialized) {
	GST_DEBUG ("error: fameenc encoder already initialized !");
	return GST_PAD_LINK_REFUSED;
	}

	structure = gst_caps_get_structure (caps, 0);
	gst_structure_get_int (structure, "width", &width);
	gst_structure_get_int (structure, "height", &height);
	gst_structure_get_double (structure, "framerate", &fps);

	/* fameenc requires width and height to be multiples of 16 */
	if (width % 16 != 0 \|\| height % 16 != 0)
	return GST_PAD_LINK_REFUSED;

	fps_idx = framerate_to_index (fps);
	fameenc->fp.frame_rate_num = frame_rates[fps_idx].num;
	fameenc->fp.frame_rate_den = frame_rates[fps_idx].den;
	fameenc->time_interval = 0;

	fameenc->fp.width = width;
	fameenc->fp.height = height;
	fameenc->fp.coding = (const char *) fameenc->pattern;

	/* FIXME: choose good parameters */
	fameenc->fp.slices_per_frame = 1;

	/* FIXME: handle these properly */
	fameenc->fp.shape_quality = 75;
	fameenc->fp.search_range = 0;
	fameenc->fp.total_frames = 0;
	fameenc->fp.retrieve_cb = NULL;

	fame_init (fameenc->fc, &fameenc->fp, fameenc->buffer, fameenc->buffer_size);

	fameenc->initialized = TRUE;
	fameenc->time_interval = 0;

	return GST_PAD_LINK_OK;
	}

	static void
	gst_fameenc_init (GstFameEnc *fameenc)
	{
	g_assert (fameenc != NULL);
	g_assert (GST_IS_FAMEENC (fameenc));

	/* open fameenc */
	fameenc->fc = fame_open ();
	g_assert (fameenc->fc != NULL);

	/* create the sink and src pads */
	fameenc->sinkpad = gst_pad_new_from_template (
	gst_static_pad_template_get (&sink_template), "sink");
	gst_element_add_pad (GST_ELEMENT (fameenc), fameenc->sinkpad);
	gst_pad_set_chain_function (fameenc->sinkpad, gst_fameenc_chain);
	gst_pad_set_link_function (fameenc->sinkpad, gst_fameenc_sink_link);

	fameenc->srcpad = gst_pad_new_from_template (
	gst_static_pad_template_get (&src_template), "src");
	gst_element_add_pad (GST_ELEMENT (fameenc), fameenc->srcpad);
	/* FIXME: set some more handler functions here */

	fameenc->verbose = FALSE;

	/* reset the initial video state */
	fameenc->fp.width = -1;
	fameenc->fp.height = -1;
	fameenc->initialized = FALSE;

	/* defaults */
	fameenc->fp.bitrate = 0;
	fameenc->fp.quality = 75;
	fameenc->fp.frame_rate_num = 25;
	fameenc->fp.frame_rate_den = 1; /* avoid floating point exceptions */
	fameenc->fp.frames_per_sequence = 12;

	fameenc->pattern = g_strdup ("IPPPPPPPPPP");

	/* allocate space for the buffer */
	fameenc->buffer_size = FAMEENC_BUFFER_SIZE; /* FIXME */
	fameenc->buffer = (unsigned char *) g_malloc (fameenc->buffer_size);

	fameenc->next_time = 0;
	fameenc->time_interval = 0;
	}

	static void
	gst_fameenc_dispose (GObject *object)
	{
	GstFameEnc *fameenc = GST_FAMEENC (object);

	G_OBJECT_CLASS (parent_class)->dispose (object);

	g_free (fameenc->buffer);
	}

	static void
	gst_fameenc_chain (GstPad pad, GstData _data)
	{
	GstBuffer *buf = GST_BUFFER (_data);
	GstFameEnc *fameenc;
	guchar *data;
	gulong size;
	gint frame_size;
	gint length;

	g_return_if_fail (pad != NULL);
	g_return_if_fail (GST_IS_PAD (pad));
	g_return_if_fail (buf != NULL);
	g_return_if_fail (GST_IS_BUFFER (buf));

	fameenc = GST_FAMEENC (gst_pad_get_parent (pad));

	data = (guchar *) GST_BUFFER_DATA (buf);
	size = GST_BUFFER_SIZE (buf);

	GST_DEBUG ("gst_fameenc_chain: got buffer of %ld bytes in '%s'",
	size, GST_OBJECT_NAME (fameenc));

	/* the data contains the three planes side by side, with size w * h, w * h /4,
	* w * h / 4 */
	fameenc->fy.w = fameenc->fp.width;
	fameenc->fy.h = fameenc->fp.height;

	frame_size = fameenc->fp.width * fameenc->fp.height;

	fameenc->fy.p = 0;
	fameenc->fy.y = data;
	fameenc->fy.u = data + frame_size;
	fameenc->fy.v = fameenc->fy.u + (frame_size >> 2);

	fame_start_frame (fameenc->fc, &fameenc->fy, NULL);

	while ((length = fame_encode_slice (fameenc->fc)) != 0) {
	GstBuffer *outbuf;

	outbuf = gst_buffer_new ();

	/* FIXME: safeguard, remove me when a better way is found */
	if (length > FAMEENC_BUFFER_SIZE)
	g_warning ("FAMEENC_BUFFER_SIZE is defined too low, encoded slice has size %d !\n", length);

	if (!fameenc->time_interval) {
	fameenc->time_interval = GST_SECOND * fameenc->fp.frame_rate_den / fameenc->fp.frame_rate_num;
	}

	fameenc->next_time += fameenc->time_interval;

	GST_BUFFER_SIZE (outbuf) = length;
	GST_BUFFER_TIMESTAMP (outbuf) = fameenc->next_time;
	GST_BUFFER_DATA (outbuf) = g_malloc (length);
	memcpy (GST_BUFFER_DATA(outbuf), fameenc->buffer, length);
	GST_BUFFER_TIMESTAMP (outbuf) = GST_BUFFER_TIMESTAMP (buf);

	GST_DEBUG ("gst_fameenc_chain: pushing buffer of size %d",
	GST_BUFFER_SIZE(outbuf));

	gst_pad_push (fameenc->srcpad, GST_DATA (outbuf));
	}

	fame_end_frame (fameenc->fc, NULL);

	gst_buffer_unref(buf);
	}

	static void
	gst_fameenc_set_property (GObject *object, guint prop_id,
	const GValue value, GParamSpec pspec)
	{
	GstFameEnc *fameenc;

	g_return_if_fail (GST_IS_FAMEENC (object));
	fameenc = GST_FAMEENC (object);

	if (fameenc->initialized) {
	GST_DEBUG ("error: fameenc encoder already initialized, cannot set properties !");
	return;
	}

	switch (prop_id) {
	case ARG_BITRATE:
	fameenc->fp.bitrate = g_value_get_int (value);
	break;
	case ARG_QUALITY:
	fameenc->fp.quality = CLAMP (g_value_get_int (value), 0, 100);
	break;
	case ARG_PATTERN:
	g_free (fameenc->pattern);
	fameenc->pattern = g_strdup (g_value_get_string (value));
	break;
	case ARG_FAME_VERBOSE:
	fameenc->verbose = g_value_get_boolean (value);
	break;
	case ARG_BUFFER_SIZE:
	fameenc->buffer_size = g_value_get_int (value);
	break;
	case ARG_FRAMES_PER_SEQUENCE:
	fameenc->fp.frames_per_sequence = g_value_get_int (value);
	break;
	default:
	if (prop_id >= ARG_FAME_PROPS_START) {
	gchar *name;
	gint index = g_value_get_enum (value);
	GEnumValue *values;

	values = G_ENUM_CLASS (g_type_class_ref (pspec->value_type))->values;
	name = (gchar *) g_param_spec_get_qdata (pspec, fame_object_name);

	fame_register (fameenc->fc, name, fame_get_object (fameenc->fc, values[index].value_name));
	}
	else
	G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
	break;
	}
	}

	static void
	gst_fameenc_get_property (GObject *object, guint prop_id,
	GValue value, GParamSpec pspec)
	{
	GstFameEnc *fameenc;

	g_return_if_fail (GST_IS_FAMEENC (object));
	fameenc = GST_FAMEENC (object);

	switch (prop_id) {
	case ARG_BITRATE:
	g_value_set_int (value, fameenc->fp.bitrate);
	break;
	case ARG_QUALITY:
	g_value_set_int (value, fameenc->fp.quality);
	break;
	case ARG_PATTERN:
	g_value_set_string (value, g_strdup (fameenc->pattern));
	break;
	case ARG_FAME_VERBOSE:
	g_value_set_boolean (value, fameenc->verbose);
	break;
	case ARG_BUFFER_SIZE:
	g_value_set_int (value, fameenc->buffer_size);
	break;
	case ARG_FRAMES_PER_SEQUENCE:
	g_value_set_int (value, fameenc->fp.frames_per_sequence);
	break;
	default:
	if (prop_id >= ARG_FAME_PROPS_START) {
	gchar *name;
	gint index = 0;
	GEnumValue *values;
	fame_object_t *f_object;

	values = G_ENUM_CLASS (g_type_class_ref (pspec->value_type))->values;
	name = (gchar *) g_param_spec_get_qdata (pspec, fame_object_name);

	f_object = fame_get_object (fameenc->fc, name);

	while (values[index].value_name) {
	if (!strcmp (values[index].value_nick, f_object->name)) {
	g_value_set_enum (value, index);
	return;
	}
	index++;
	}
	}
	G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
	break;
	}
	}

	static gboolean
	plugin_init (GstPlugin *plugin)
	{
	return gst_element_register (plugin, "fameenc",
	GST_RANK_NONE, GST_TYPE_FAMEENC);
	}

	GST_PLUGIN_DEFINE (
	GST_VERSION_MAJOR,
	GST_VERSION_MINOR,
	"fameenc",
	"Fast Assembly MPEG Encoder",
	plugin_init,
	LIBFAME_VERSION,
	"LGPL",
	"libfame",
	"http://fame.sourceforge.net/"
	)