--- /dev/null
+/* 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.
+ */
+
+#include <fame.h>
+#include <string.h>
+
+#include "gstlibfame.h"
+
+#define LIBFAME_BUFFER_SIZE (1024 * 1024) /* FIXME: do this properly */
+
+/* elementfactory information */
+static GstElementDetails gst_libfame_details = {
+ "MPEG1 and MPEG4 video encoder using the libfame library",
+ "Codec/Video/Encoder",
+ "Uses libfame to encode MPEG video streams",
+ VERSION,
+ "(C) 1996, MPEG Software Simulation Group\n"
+ "Thomas Vander Stichele <thomas@apestaart.org>",
+ "(C) 2002",
+};
+
+/* Libfame signals and args */
+enum {
+ /* FILL ME */
+ LAST_SIGNAL
+};
+
+enum {
+ ARG_0,
+ ARG_VERSION,
+ ARG_FPS,
+ ARG_QUALITY,
+ ARG_BITRATE,
+ /* FILL ME */
+};
+
+/* FIXME: either use or delete this */
+/*
+static double video_rates[16] =
+{
+ 0.0,
+ 24000.0/1001.,
+ 24.0,
+ 25.0,
+ 30000.0/1001.,
+ 30.0,
+ 50.0,
+ 60000.0/1001.,
+ 60.0,
+ 1,
+ 5,
+ 10,
+ 12,
+ 15,
+ 0,
+ 0
+};
+*/
+
+GST_PAD_TEMPLATE_FACTORY (sink_template_factory,
+ "sink",
+ GST_PAD_SINK,
+ GST_PAD_ALWAYS,
+ GST_CAPS_NEW (
+ "libfame_sink_caps",
+ "video/raw",
+ "format", GST_PROPS_FOURCC (GST_MAKE_FOURCC ('I','4','2','0')),
+ "width", GST_PROPS_INT_RANGE (16, 4096),
+ "height", GST_PROPS_INT_RANGE (16, 4096)
+ )
+)
+
+GST_PAD_TEMPLATE_FACTORY (src_template_factory,
+ "src",
+ GST_PAD_SRC,
+ GST_PAD_ALWAYS,
+ GST_CAPS_NEW (
+ "libfame_src_caps",
+ "video/mpeg",
+ "mpegversion", GST_PROPS_LIST (
+ GST_PROPS_INT (1), GST_PROPS_INT (4)),
+ "systemstream", GST_PROPS_BOOLEAN (FALSE)
+ )
+);
+
+static void gst_libfame_class_init (GstLibfameClass *klass);
+static void gst_libfame_init (GstLibfame *libfame);
+static void gst_libfame_dispose (GObject *object);
+
+static void gst_libfame_set_property (GObject *object, guint prop_id,
+ const GValue *value, GParamSpec *pspec);
+static void gst_libfame_get_property (GObject *object, guint prop_id,
+ GValue *value, GParamSpec *pspec);
+
+static void gst_libfame_chain (GstPad *pad, GstBuffer *buf);
+
+static GstElementClass *parent_class = NULL;
+/*static guint gst_libfame_signals[LAST_SIGNAL] = { 0 };*/
+
+GType
+gst_libfame_get_type (void)
+{
+ static GType libfame_type = 0;
+
+ if (!libfame_type) {
+ static const GTypeInfo libfame_info = {
+ sizeof (GstLibfameClass),
+ NULL,
+ NULL,
+ (GClassInitFunc) gst_libfame_class_init,
+ NULL,
+ NULL,
+ sizeof (GstLibfame),
+ 0,
+ (GInstanceInitFunc) gst_libfame_init,
+ };
+ libfame_type = g_type_register_static (GST_TYPE_ELEMENT,
+ "GstLibfame", &libfame_info, 0);
+ }
+ return libfame_type;
+}
+
+static void
+gst_libfame_class_init (GstLibfameClass *klass)
+{
+ GObjectClass *gobject_class = NULL;
+ GstElementClass *gstelement_class = NULL;
+
+ gobject_class = (GObjectClass*) klass;
+ gstelement_class = (GstElementClass*) klass;
+
+ parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
+
+ gobject_class->set_property = gst_libfame_set_property;
+ gobject_class->get_property = gst_libfame_get_property;
+ gobject_class->dispose = gst_libfame_dispose;
+
+ g_object_class_install_property (gobject_class, ARG_VERSION,
+ g_param_spec_int ("mpeg_version", "MPEG Version",
+ "MPEG Codec Version (1 or 4)",
+ 1, 4, 1, G_PARAM_READWRITE));
+ g_object_class_install_property (gobject_class, ARG_FPS,
+ g_param_spec_double ("frames_per_second", "Frames per second",
+ "Number of frames per second",
+ -G_MAXDOUBLE, G_MAXDOUBLE, 25.0, G_PARAM_READWRITE));
+/*
+ g_object_class_install_property (gobject_class, ARG_BITRATE,
+ g_param_spec_int ("bitrate", "bitrate", "bitrate",
+ 0, 1500000, 10, G_PARAM_READWRITE)); */
+ /* CHECKME */
+ 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)); /* CHECKME */
+}
+
+static GstPadConnectReturn
+gst_libfame_sinkconnect (GstPad *pad, GstCaps *caps)
+{
+ gint width, height;
+ GstLibfame *libfame;
+
+ libfame = GST_LIBFAME (gst_pad_get_parent (pad));
+
+ if (!GST_CAPS_IS_FIXED (caps))
+ return GST_PAD_CONNECT_DELAYED;
+
+ if (libfame->initialized)
+ {
+ GST_DEBUG(0, "error: libfame encoder already initialized !");
+ return GST_PAD_CONNECT_REFUSED;
+ }
+
+ gst_caps_get_int (caps, "width", &width);
+ gst_caps_get_int (caps, "height", &height);
+
+ /* FIXME: do better error handling here */
+ /* libfame requires width and height to be multiples of 16 */
+ g_assert (width % 16 == 0);
+ g_assert (height % 16 == 0);
+
+ libfame->fp.width = width;
+ libfame->fp.height = height;
+ libfame->fp.coding = "I";
+
+ /* FIXME: choose good parameters */
+ libfame->fp.slices_per_frame = 1;
+ libfame->fp.frames_per_sequence = 0xffffffff; /* infinite */
+ /* FIXME: 25 fps */
+ libfame->fp.frame_rate_num = 25;
+ libfame->fp.frame_rate_den = 1;
+
+ /* FIXME: handle these properly */
+ libfame->fp.shape_quality = 100;
+ libfame->fp.search_range = 0;
+ libfame->fp.verbose = 1;
+ libfame->fp.profile = "mpeg1";
+ libfame->fp.total_frames = 0;
+ libfame->fp.retrieve_cb = NULL;
+
+ fame_init (libfame->fc, &libfame->fp, libfame->buffer, libfame->buffer_size);
+
+ g_print ("libfame: init done.\n");
+ g_assert (libfame->fc != NULL); /* FIXME */
+ libfame->initialized = TRUE;
+ return GST_PAD_CONNECT_OK;
+}
+
+static void
+gst_libfame_init (GstLibfame *libfame)
+{
+ g_assert (libfame != NULL);
+ g_assert (GST_IS_LIBFAME (libfame));
+
+ /* open libfame */
+ libfame->fc = fame_open ();
+ g_assert (libfame->fc != NULL);
+
+ /* create the sink and src pads */
+ libfame->sinkpad = gst_pad_new_from_template (
+ GST_PAD_TEMPLATE_GET (sink_template_factory), "sink");
+ gst_element_add_pad (GST_ELEMENT (libfame), libfame->sinkpad);
+ gst_pad_set_chain_function (libfame->sinkpad, gst_libfame_chain);
+ gst_pad_set_connect_function (libfame->sinkpad, gst_libfame_sinkconnect);
+
+ libfame->srcpad = gst_pad_new_from_template (
+ GST_PAD_TEMPLATE_GET (src_template_factory), "src");
+ gst_element_add_pad (GST_ELEMENT (libfame), libfame->srcpad);
+ /* FIXME: set some more handler functions here */
+
+ /* reset the initial video state */
+ libfame->width = -1;
+ libfame->height = -1;
+ libfame->initialized = FALSE;
+
+ /* defaults */
+ libfame->fp.quality = 75;
+ libfame->fp.frame_rate_num = 25;
+ libfame->fp.frame_rate_den = 1; /* avoid floating point exceptions */
+ libfame->fp.profile = g_strdup_printf ("mpeg1");
+
+ /* allocate space for the buffer */
+ libfame->buffer_size = LIBFAME_BUFFER_SIZE; /* FIXME */
+ libfame->buffer = (unsigned char *) g_malloc (libfame->buffer_size);
+}
+
+static void
+gst_libfame_dispose (GObject *object)
+{
+ GstLibfame *libfame = GST_LIBFAME (object);
+
+ G_OBJECT_CLASS (parent_class)->dispose (object);
+
+ g_free (libfame->buffer);
+}
+
+static void
+gst_libfame_chain (GstPad *pad, GstBuffer *buf)
+{
+ GstLibfame *libfame = NULL;
+ guchar *data = NULL;
+ gulong size;
+ GstBuffer *outbuf = NULL;
+ int 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));
+
+ libfame = GST_LIBFAME (gst_pad_get_parent (pad));
+
+ data = (guchar *) GST_BUFFER_DATA (buf);
+ size = GST_BUFFER_SIZE (buf);
+ GST_DEBUG (0,"gst_libfame_chain: got buffer of %ld bytes in '%s'",
+ size, GST_OBJECT_NAME (libfame));
+
+ /* the data contains the three planes side by side, with size w * h, w * h /4,
+ * w * h / 4 */
+ libfame->fy.w = libfame->fp.width;
+ libfame->fy.h = libfame->fp.height;
+ libfame->fy.p = 0; /* FIXME: is this pointing to previous data ? */
+ libfame->fy.y = data;
+ libfame->fy.u = data + libfame->width * libfame->height;
+ libfame->fy.v = data + 5 * libfame->width * libfame->height / 4;
+ fame_start_frame (libfame->fc, &libfame->fy, NULL);
+
+ while ((length = fame_encode_slice (libfame->fc)) != 0)
+ {
+ outbuf = gst_buffer_new ();
+
+ /* FIXME: safeguard, remove me when a better way is found */
+ if (length > LIBFAME_BUFFER_SIZE)
+ g_warning ("LIBFAME_BUFFER_SIZE is defined too low, encoded slice has size %d !\n", length);
+ GST_BUFFER_SIZE (outbuf) = length;
+ GST_BUFFER_DATA (outbuf) = g_malloc (length);
+ memcpy (GST_BUFFER_DATA(outbuf), libfame->buffer, length);
+ GST_DEBUG (0,"gst_libfame_chain: pushing buffer of size %d",
+ GST_BUFFER_SIZE(outbuf));
+ gst_pad_push (libfame->srcpad, outbuf);
+ gst_buffer_unref(buf);
+ }
+
+ fame_end_frame (libfame->fc, NULL); /* FIXME: get stats */
+}
+
+static void
+gst_libfame_set_property (GObject *object, guint prop_id,
+ const GValue *value, GParamSpec *pspec)
+{
+ GstLibfame *src;
+
+ g_return_if_fail (GST_IS_LIBFAME (object));
+ src = GST_LIBFAME (object);
+
+ if (src->initialized)
+ {
+ GST_DEBUG(0, "error: libfame encoder already initialized, cannot set properties !");
+ return;
+ }
+
+ switch (prop_id) {
+ case ARG_VERSION:
+ {
+ int version = g_value_get_int (value);
+ if (version != 1 && version != 4)
+ {
+ g_warning ("libfame: only use MPEG version 1 or 4 !");
+ break;
+ }
+ /*
+ if (src->fp.profile)
+ { g_free (src->fp.profile); src->fp.profile = NULL; }
+ src->fp.profile = g_strdup_printf ("mpeg%d", version);
+ */
+ /* FIXME: this should be done using fame_register */
+ break;
+ }
+ case ARG_FPS:
+ /* FIXME: we could do a much better job of finding num and den here */
+ src->fp.frame_rate_num = g_value_get_double (value);
+ src->fp.frame_rate_den = 1;
+ gst_info ("libfame: setting framerate for encoding to %f (%d/%d)\n",
+ (float) src->fp.frame_rate_num / src->fp.frame_rate_den,
+ src->fp.frame_rate_num, src->fp.frame_rate_den);
+ break;
+
+ case ARG_QUALITY:
+ {
+ int quality;
+
+ quality = g_value_get_int (value);
+ /* quality is a percentage */
+ if (quality < 0) quality = 0;
+ if (quality > 100) quality = 100;
+ src->fp.quality = quality;
+ gst_info ("libfame: setting quality for encoding to %d\n", quality);
+ g_warning ("libfame: setting quality for encoding to %d\n", quality);
+ break;
+ }
+ case ARG_BITRATE:
+ g_warning ("bitrate not implemented yet !\n");
+ /*
+ src->encoder->seq.bit_rate = g_value_get_int (value);
+ */
+ break;
+ default:
+ G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
+ break;
+ }
+}
+
+static void
+gst_libfame_get_property (GObject *object, guint prop_id,
+ GValue *value, GParamSpec *pspec)
+{
+ GstLibfame *src;
+
+ g_return_if_fail (GST_IS_LIBFAME (object));
+ src = GST_LIBFAME (object);
+
+ switch (prop_id) {
+ case ARG_FPS:
+ g_value_set_double (value, (src->fp.frame_rate_num /
+ src->fp.frame_rate_den));
+ break;
+ case ARG_QUALITY:
+ g_value_set_int (value, src->fp.quality);
+ break;
+ case ARG_BITRATE:
+ g_warning ("You think you WANT to know bitrate ? Think AGAIN !\n");
+ /* g_value_set_int (value, src->encoder->seq.bit_rate); */
+ break;
+ default:
+ G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
+ break;
+ }
+}
+
+static gboolean
+plugin_init (GModule *module, GstPlugin *plugin)
+{
+ GstElementFactory *factory;
+
+ /* create an elementfactory for the libfame element */
+ factory = gst_element_factory_new ("libfame", GST_TYPE_LIBFAME,
+ &gst_libfame_details);
+ g_return_val_if_fail (factory != NULL, FALSE);
+
+ gst_element_factory_add_pad_template (factory,
+ GST_PAD_TEMPLATE_GET (sink_template_factory));
+ gst_element_factory_add_pad_template (factory,
+ GST_PAD_TEMPLATE_GET (src_template_factory));
+
+ gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory));
+
+ return TRUE;
+}
+
+GstPluginDesc plugin_desc = {
+ GST_VERSION_MAJOR,
+ GST_VERSION_MINOR,
+ "libfame",
+ plugin_init
+};
projects / platform / upstream / gstreamer.git / commitdiff