gstreamer/ext/libfame/gstlibfame.c

600 lines
17 KiB
C
Raw Normal View History

/* 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 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",
VERSION,
"fameenc: (C) 2000-2001, Vivien Chappelier\n"
"Thomas Vander Stichele <thomas@apestaart.org>",
"(C) 2002",
};
static GQuark fame_object_name;
/* FameEnc signals and args */
enum {
/* FILL ME */
LAST_SIGNAL
};
enum {
ARG_0,
ARG_VERSION,
ARG_FRAMERATE,
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 */
};
GST_PAD_TEMPLATE_FACTORY (sink_template_factory,
"sink",
GST_PAD_SINK,
GST_PAD_ALWAYS,
GST_CAPS_NEW (
"fameenc_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 (
"fameenc_src_caps",
"video/mpeg",
"mpegversion", GST_PROPS_LIST (
GST_PROPS_INT (1), GST_PROPS_INT (4)),
"systemstream", GST_PROPS_BOOLEAN (FALSE)
)
);
#define MAX_FRAME_RATES 16
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 },
{ 0, 0 },
{ 0, 0 },
{ 0, 0 },
{ 0, 0 },
{ 0, 0 },
{ 0, 0 },
{ 0, 0 },
};
static gint
framerate_to_index (num, den)
{
gint i;
for (i = 0; i < MAX_FRAME_RATES; i++) {
if (frame_rates[i].num == num && frame_rates[i].den == den)
return i;
}
return 0;
}
#define GST_TYPE_FAMEENC_FRAMERATE (gst_fameenc_framerate_get_type())
static GType
gst_fameenc_framerate_get_type(void) {
static GType fameenc_framerate_type = 0;
static GEnumValue fameenc_framerate[] = {
{1, "1", "24000/1001 (23.97)"},
{2, "2", "24"},
{3, "3", "25"},
{4, "4", "30000/1001 (29.97)"},
{5, "5", "30"},
{6, "6", "50"},
{7, "7", "60000/1001 (59.94)"},
{8, "8", "60"},
{0, NULL, NULL},
};
if (!fameenc_framerate_type) {
fameenc_framerate_type = g_enum_register_static("GstFameEncFrameRate", fameenc_framerate);
}
return fameenc_framerate_type;
}
static void gst_fameenc_class_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, GstBuffer *buf);
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),
NULL,
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_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_FRAMERATE,
g_param_spec_enum ("framerate", "Frame Rate", "Number of frames per second",
GST_TYPE_FAMEENC_FRAMERATE, 3, G_PARAM_READWRITE));
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 GstPadConnectReturn
gst_fameenc_sinkconnect (GstPad *pad, GstCaps *caps)
{
gint width, height;
GstFameEnc *fameenc;
fameenc = GST_FAMEENC (gst_pad_get_parent (pad));
if (!GST_CAPS_IS_FIXED (caps))
return GST_PAD_CONNECT_DELAYED;
if (fameenc->initialized) {
GST_DEBUG(0, "error: fameenc encoder already initialized !");
return GST_PAD_CONNECT_REFUSED;
}
gst_caps_get_int (caps, "width", &width);
gst_caps_get_int (caps, "height", &height);
/* fameenc requires width and height to be multiples of 16 */
if (width % 16 != 0 || height % 16 != 0)
return GST_PAD_CONNECT_REFUSED;
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;
return GST_PAD_CONNECT_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_PAD_TEMPLATE_GET (sink_template_factory), "sink");
gst_element_add_pad (GST_ELEMENT (fameenc), fameenc->sinkpad);
gst_pad_set_chain_function (fameenc->sinkpad, gst_fameenc_chain);
gst_pad_set_connect_function (fameenc->sinkpad, gst_fameenc_sinkconnect);
fameenc->srcpad = gst_pad_new_from_template (
GST_PAD_TEMPLATE_GET (src_template_factory), "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);
}
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, GstBuffer *buf)
{
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 (0,"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);
GST_BUFFER_SIZE (outbuf) = length;
GST_BUFFER_DATA (outbuf) = g_malloc (length);
memcpy (GST_BUFFER_DATA(outbuf), fameenc->buffer, length);
GST_DEBUG (0,"gst_fameenc_chain: pushing buffer of size %d",
GST_BUFFER_SIZE(outbuf));
gst_pad_push (fameenc->srcpad, 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(0, "error: fameenc encoder already initialized, cannot set properties !");
return;
}
switch (prop_id) {
case ARG_FRAMERATE:
{
gint index = g_value_get_enum (value);
fameenc->fp.frame_rate_num = frame_rates[index].num;
fameenc->fp.frame_rate_den = frame_rates[index].den;
break;
}
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_FRAMERATE:
{
gint index = framerate_to_index (fameenc->fp.frame_rate_num,
fameenc->fp.frame_rate_den);
g_value_set_enum (value, index);
break;
}
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 (GModule *module, GstPlugin *plugin)
{
GstElementFactory *factory;
/* create an elementfactory for the fameenc element */
factory = gst_element_factory_new ("fameenc", GST_TYPE_FAMEENC,
&gst_fameenc_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,
"fameenc",
plugin_init
};