summaryrefslogtreecommitdiffstats
path: root/ext/divx
diff options
context:
space:
mode:
Diffstat (limited to 'ext/divx')
-rw-r--r--ext/divx/Makefile.am10
-rw-r--r--ext/divx/gstdivxenc.c469
-rw-r--r--ext/divx/gstdivxenc.h78
3 files changed, 557 insertions, 0 deletions
diff --git a/ext/divx/Makefile.am b/ext/divx/Makefile.am
new file mode 100644
index 00000000..e6a6eaa5
--- /dev/null
+++ b/ext/divx/Makefile.am
@@ -0,0 +1,10 @@
+plugindir = $(libdir)/gstreamer-@GST_MAJORMINOR@
+
+plugin_LTLIBRARIES = libgstdivx.la
+
+libgstdivx_la_SOURCES = gstdivxenc.c
+libgstdivx_la_CFLAGS = $(GST_CFLAGS) $(DIVX_CFLAGS)
+libgstdivx_la_LIBADD = $(DIVX_LIBS)
+libgstdivx_la_LDFLAGS = $(GST_PLUGIN_LDFLAGS)
+
+noinst_HEADERS = gstdivxenc.h
diff --git a/ext/divx/gstdivxenc.c b/ext/divx/gstdivxenc.c
new file mode 100644
index 00000000..207d1e19
--- /dev/null
+++ b/ext/divx/gstdivxenc.c
@@ -0,0 +1,469 @@
+/* GStreamer divx encoder plugin
+ * Copyright (C) 2003 Ronald Bultje <rbultje@ronald.bitfreak.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 <string.h>
+#include "gstdivxenc.h"
+#include <gst/video/video.h>
+#include <encore2.h>
+
+/* elementfactory information */
+GstElementDetails gst_divxenc_details = {
+ "Divx encoder",
+ "Codec/Video/Encoder",
+ "Commercial",
+ "Divx encoder based on divxencore",
+ VERSION,
+ "Ronald Bultje <rbultje@ronald.bitfreak.net>",
+ "(C) 2003",
+};
+
+GST_PAD_TEMPLATE_FACTORY(sink_template,
+ "sink",
+ GST_PAD_SINK,
+ GST_PAD_ALWAYS,
+ GST_CAPS_NEW("divxenc_sink",
+ "video/raw",
+ "format", GST_PROPS_LIST(
+ GST_PROPS_FOURCC(GST_MAKE_FOURCC('R','G','B',' ')),
+ GST_PROPS_FOURCC(GST_MAKE_FOURCC('I','4','2','0')),
+ GST_PROPS_FOURCC(GST_MAKE_FOURCC('I','Y','U','V')),
+ GST_PROPS_FOURCC(GST_MAKE_FOURCC('Y','U','Y','2')),
+ GST_PROPS_FOURCC(GST_MAKE_FOURCC('Y','V','1','2')),
+ GST_PROPS_FOURCC(GST_MAKE_FOURCC('U','Y','V','Y'))
+ ),
+ "width", GST_PROPS_INT_RANGE(0, G_MAXINT),
+ "height", GST_PROPS_INT_RANGE(0, G_MAXINT),
+ NULL)
+)
+
+GST_PAD_TEMPLATE_FACTORY(src_template,
+ "src",
+ GST_PAD_SRC,
+ GST_PAD_ALWAYS,
+ GST_CAPS_NEW("divxenc_sink",
+ "video/divx",
+ NULL)
+)
+
+
+/* DivxEnc signals and args */
+enum {
+ FRAME_ENCODED,
+ LAST_SIGNAL
+};
+
+enum {
+ ARG_0,
+ ARG_BITRATE,
+ ARG_MAXKEYINTERVAL,
+ ARG_BUFSIZE
+};
+
+
+static void gst_divxenc_class_init (GstDivxEncClass *klass);
+static void gst_divxenc_init (GstDivxEnc *divxenc);
+static void gst_divxenc_chain (GstPad *pad,
+ GstBuffer *buf);
+static GstPadLinkReturn gst_divxenc_connect (GstPad *pad,
+ GstCaps *vscapslist);
+
+/* properties */
+static void gst_divxenc_set_property (GObject *object,
+ guint prop_id,
+ const GValue *value,
+ GParamSpec *pspec);
+static void gst_divxenc_get_property (GObject *object,
+ guint prop_id,
+ GValue *value,
+ GParamSpec *pspec);
+
+static GstElementClass *parent_class = NULL;
+static guint gst_divxenc_signals[LAST_SIGNAL] = { 0 };
+
+
+GType
+gst_divxenc_get_type(void)
+{
+ static GType divxenc_type = 0;
+
+ if (!divxenc_type)
+ {
+ static const GTypeInfo divxenc_info = {
+ sizeof(GstDivxEncClass),
+ NULL,
+ NULL,
+ (GClassInitFunc) gst_divxenc_class_init,
+ NULL,
+ NULL,
+ sizeof(GstDivxEnc),
+ 0,
+ (GInstanceInitFunc) gst_divxenc_init,
+ };
+ divxenc_type = g_type_register_static(GST_TYPE_ELEMENT,
+ "GstDivxEnc",
+ &divxenc_info, 0);
+ }
+ return divxenc_type;
+}
+
+
+static void
+gst_divxenc_class_init (GstDivxEncClass *klass)
+{
+ GstElementClass *gstelement_class;
+ GObjectClass *gobject_class;
+
+ gobject_class = (GObjectClass *) klass;
+ gstelement_class = (GstElementClass *) klass;
+
+ parent_class = g_type_class_ref(GST_TYPE_ELEMENT);
+
+ g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_BITRATE,
+ g_param_spec_ulong("bitrate","Bitrate",
+ "Target video bitrate",
+ 0,G_MAXULONG,0,G_PARAM_READWRITE));
+
+ g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_MAXKEYINTERVAL,
+ g_param_spec_int("max_key_interval","Max. Key Interval",
+ "Maximum number of frames between two keyframes",
+ 0,G_MAXINT,0,G_PARAM_READWRITE));
+
+ g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_BUFSIZE,
+ g_param_spec_ulong("buffer_size", "Buffer Size",
+ "Size of the video buffers",
+ 0,G_MAXULONG,0,G_PARAM_READWRITE));
+
+ gobject_class->set_property = gst_divxenc_set_property;
+ gobject_class->get_property = gst_divxenc_get_property;
+
+ gst_divxenc_signals[FRAME_ENCODED] =
+ g_signal_new ("frame_encoded", G_TYPE_FROM_CLASS(klass),
+ G_SIGNAL_RUN_LAST,
+ G_STRUCT_OFFSET (GstDivxEncClass, frame_encoded),
+ NULL, NULL,
+ g_cclosure_marshal_VOID__VOID, G_TYPE_NONE, 0);
+}
+
+
+static void
+gst_divxenc_init (GstDivxEnc *divxenc)
+{
+ /* create the sink pad */
+ divxenc->sinkpad = gst_pad_new_from_template(
+ GST_PAD_TEMPLATE_GET(sink_template),
+ "sink");
+ gst_element_add_pad(GST_ELEMENT(divxenc), divxenc->sinkpad);
+
+ gst_pad_set_chain_function(divxenc->sinkpad, gst_divxenc_chain);
+ gst_pad_set_link_function(divxenc->sinkpad, gst_divxenc_connect);
+
+ /* create the src pad */
+ divxenc->sinkpad = gst_pad_new_from_template(
+ GST_PAD_TEMPLATE_GET(src_template),
+ "src");
+ gst_element_add_pad(GST_ELEMENT(divxenc), divxenc->srcpad);
+
+ /* bitrate, etc. */
+ divxenc->width = divxenc->height = divxenc->csp = -1;
+ divxenc->bitrate = 512 * 1024;
+ divxenc->max_key_interval = -1; /* default - 2*fps */
+ divxenc->buffer_size = 512 * 1024;
+
+ /* set divx handle to NULL */
+ divxenc->handle = NULL;
+}
+
+
+static gboolean
+gst_divxenc_setup (GstDivxEnc *divxenc)
+{
+ ENC_PARAM xenc;
+ gdouble fps;
+ int ret;
+
+ fps = gst_video_frame_rate(GST_PAD_PEER(divxenc->sinkpad));
+
+ /* set it up */
+ memset(&xenc, 0, sizeof(ENC_PARAM));
+ xenc.x_dim = divxenc->width;
+ xenc.y_dim = divxenc->height;
+ xenc.framerate = fps;
+
+ xenc.rc_period = 2000;
+ xenc.rc_reaction_period = 10;
+ xenc.rc_reaction_ratio = 20;
+
+ xenc.max_quantizer = 31;
+ xenc.min_quantizer = 1;
+
+ xenc.quality = 3;
+ xenc.bitrate = divxenc->bitrate;
+
+ xenc.deinterlace = 0;
+
+ xenc.max_key_interval = (divxenc->max_key_interval == -1) ?
+ (2 * xenc.framerate) :
+ divxenc->max_key_interval;
+ xenc.handle = NULL;
+
+ if ((ret = encore(NULL, ENC_OPT_INIT, &xenc, NULL))) {
+ gst_element_error(GST_ELEMENT(divxenc),
+ "Error setting up divx encoder: %d\n",
+ ret);
+ return FALSE;
+ }
+
+ divxenc->handle = xenc.handle;
+
+ return TRUE;
+}
+
+
+static void
+gst_divxenc_chain (GstPad *pad,
+ GstBuffer *buf)
+{
+ GstDivxEnc *divxenc;
+ GstBuffer *outbuf;
+ ENC_FRAME xframe;
+ ENC_RESULT xres;
+ int ret;
+
+ g_return_if_fail(pad != NULL);
+ g_return_if_fail(GST_IS_PAD(pad));
+ g_return_if_fail(buf != NULL);
+
+ divxenc = GST_DIVXENC(GST_OBJECT_PARENT(pad));
+
+ if (!divxenc->handle) {
+ if (!gst_divxenc_setup(divxenc)) {
+ gst_buffer_unref(buf);
+ return;
+ }
+ }
+
+ outbuf = gst_buffer_new_and_alloc(divxenc->buffer_size);
+ GST_BUFFER_TIMESTAMP(outbuf) = GST_BUFFER_TIMESTAMP(buf);
+
+ /* encode and so ... */
+ xframe.image = GST_BUFFER_DATA(buf);
+ xframe.bitstream = (void *) GST_BUFFER_DATA(outbuf);
+ xframe.length = GST_BUFFER_MAXSIZE(outbuf);
+ xframe.mvs = NULL;
+ xframe.colorspace = divxenc->csp;
+
+ if ((ret = encore(divxenc->handle, ENC_OPT_ENCODE,
+ &xframe, &xres))) {
+ gst_element_error(GST_ELEMENT(divxenc),
+ "Error encoding divx frame: %d\n", ret);
+ gst_buffer_unref(buf);
+ return;
+ }
+
+ GST_BUFFER_SIZE(outbuf) = xframe.length;
+ if (xres.is_key_frame)
+ GST_BUFFER_FLAG_SET(outbuf, GST_BUFFER_KEY_UNIT);
+
+ /* go out, multiply! */
+ gst_pad_push(divxenc->srcpad, outbuf);
+
+ /* proclaim destiny */
+ g_signal_emit(G_OBJECT(divxenc),gst_divxenc_signals[FRAME_ENCODED], 0);
+
+ /* until the final judgement */
+ gst_buffer_unref(buf);
+}
+
+
+static GstPadLinkReturn
+gst_divxenc_connect (GstPad *pad,
+ GstCaps *vscaps)
+{
+ GstDivxEnc *divxenc;
+ GstCaps *caps;
+
+ divxenc = GST_DIVXENC(gst_pad_get_parent (pad));
+
+ /* if there's something old around, remove it */
+ if (divxenc->handle) {
+ encore(divxenc->handle, ENC_OPT_RELEASE, NULL, NULL);
+ divxenc->handle = NULL;
+ }
+
+ /* we are not going to act on variable caps */
+ if (!GST_CAPS_IS_FIXED(vscaps))
+ return GST_PAD_LINK_DELAYED;
+
+ for (caps = vscaps; caps != NULL; caps = caps->next)
+ {
+ int w,h,d;
+ guint32 fourcc;
+ gint divx_cs;
+ gst_caps_get_int(caps, "width", &w);
+ gst_caps_get_int(caps, "height", &h);
+ gst_caps_get_fourcc_int(caps, "format", &fourcc);
+
+ switch (fourcc)
+ {
+ case GST_MAKE_FOURCC('I','4','2','0'):
+ case GST_MAKE_FOURCC('I','Y','U','V'):
+ divx_cs = ENC_CSP_I420;
+ break;
+ case GST_MAKE_FOURCC('Y','U','Y','2'):
+ divx_cs = ENC_CSP_YUY2;
+ break;
+ case GST_MAKE_FOURCC('Y','V','1','2'):
+ divx_cs = ENC_CSP_YV12;
+ break;
+ case GST_MAKE_FOURCC('U','Y','V','Y'):
+ divx_cs = ENC_CSP_UYVY;
+ break;
+ case GST_MAKE_FOURCC('R','G','B',' '):
+ gst_caps_get_int(caps, "depth", &d);
+ switch (d) {
+ case 24:
+ divx_cs = ENC_CSP_RGB24;
+ break;
+ case 32:
+ divx_cs = ENC_CSP_RGB32;
+ break;
+ default:
+ goto trynext;
+ }
+ break;
+ default:
+ goto trynext;
+ }
+
+ /* grmbl, we only know the peer pad *after*
+ * linking, so we accept here, get the fps on
+ * the first cycle and set it all up then */
+ divxenc->csp = divx_cs;
+ divxenc->width = w;
+ divxenc->height = h;
+ return gst_pad_try_set_caps(divxenc->srcpad,
+ GST_CAPS_NEW("divxenc_src_caps",
+ "video/divx",
+ "width", GST_PROPS_INT(w),
+ "height", GST_PROPS_INT(h)));
+
+trynext:
+ continue;
+ }
+
+ /* if we got here - it's not good */
+ return GST_PAD_LINK_REFUSED;
+}
+
+
+static void
+gst_divxenc_set_property (GObject *object,
+ guint prop_id,
+ const GValue *value,
+ GParamSpec *pspec)
+{
+ GstDivxEnc *divxenc;
+
+ /* it's not null if we got it, but it might not be ours */
+ g_return_if_fail (GST_IS_DIVXENC (object));
+ divxenc = GST_DIVXENC(object);
+
+ switch (prop_id)
+ {
+ case ARG_BITRATE:
+ divxenc->bitrate = g_value_get_ulong(value);
+ break;
+ case ARG_BUFSIZE:
+ divxenc->buffer_size = g_value_get_ulong(value);
+ break;
+ case ARG_MAXKEYINTERVAL:
+ divxenc->max_key_interval = g_value_get_int(value);
+ break;
+ default:
+ G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
+ break;
+ }
+}
+
+
+static void
+gst_divxenc_get_property (GObject *object,
+ guint prop_id,
+ GValue *value,
+ GParamSpec *pspec)
+{
+ GstDivxEnc *divxenc;
+
+ /* it's not null if we got it, but it might not be ours */
+ g_return_if_fail (GST_IS_DIVXENC (object));
+ divxenc = GST_DIVXENC(object);
+
+ switch (prop_id) {
+ case ARG_BITRATE:
+ g_value_set_ulong(value, divxenc->bitrate);
+ break;
+ case ARG_BUFSIZE:
+ g_value_set_ulong(value, divxenc->buffer_size);
+ break;
+ case ARG_MAXKEYINTERVAL:
+ g_value_set_int(value, divxenc->max_key_interval);
+ break;
+ default:
+ G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
+ break;
+ }
+}
+
+
+static gboolean
+plugin_init (GModule *module,
+ GstPlugin *plugin)
+{
+ GstElementFactory *factory;
+
+ if (!gst_library_load("gstvideo"))
+ return FALSE;
+
+ /* create an elementfactory for the v4lmjpegsrcparse element */
+ factory = gst_element_factory_new("divxenc", GST_TYPE_DIVXENC,
+ &gst_divxenc_details);
+ g_return_val_if_fail(factory != NULL, FALSE);
+
+ /* add pad templates */
+ gst_element_factory_add_pad_template(factory,
+ GST_PAD_TEMPLATE_GET(sink_template));
+ gst_element_factory_add_pad_template(factory,
+ GST_PAD_TEMPLATE_GET(src_template));
+
+ gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory));
+
+ return TRUE;
+}
+
+
+GstPluginDesc plugin_desc = {
+ GST_VERSION_MAJOR,
+ GST_VERSION_MINOR,
+ "divxenc",
+ plugin_init
+};
diff --git a/ext/divx/gstdivxenc.h b/ext/divx/gstdivxenc.h
new file mode 100644
index 00000000..fac3b421
--- /dev/null
+++ b/ext/divx/gstdivxenc.h
@@ -0,0 +1,78 @@
+/* GStreamer divx encoder plugin
+ * Copyright (C) 2003 Ronald Bultje <rbultje@ronald.bitfreak.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.
+ */
+
+#ifndef __GST_DIVXENC_H__
+#define __GST_DIVXENC_H__
+
+#include <gst/gst.h>
+
+#ifdef __cplusplus
+extern "C" {
+#endif /* __cplusplus */
+
+
+#define GST_TYPE_DIVXENC \
+ (gst_divxenc_get_type())
+#define GST_DIVXENC(obj) \
+ (G_TYPE_CHECK_INSTANCE_CAST((obj), GST_TYPE_DIVXENC, GstDivxEnc))
+#define GST_DIVXENC_CLASS(klass) \
+ (G_TYPE_CHECK_CLASS_CAST((klass), GST_TYPE_DIVXENC, GstDivxEnc))
+#define GST_IS_DIVXENC(obj) \
+ (G_TYPE_CHECK_INSTANCE_TYPE((obj), GST_TYPE_DIVXENC))
+#define GST_IS_DIVXENC_CLASS(obj) \
+ (G_TYPE_CHECK_CLASS_TYPE((klass), GST_TYPE_DIVXENC))
+
+typedef struct _GstDivxEnc GstDivxEnc;
+typedef struct _GstDivxEncClass GstDivxEncClass;
+
+struct _GstDivxEnc {
+ GstElement element;
+
+ /* pads */
+ GstPad *sinkpad, *srcpad;
+
+ /* quality of encoded image */
+ gulong bitrate;
+
+ /* size of the buffers */
+ gulong buffer_size;
+
+ /* max key interval */
+ gint max_key_interval;
+
+ /* divx handle */
+ void *handle;
+ int csp;
+ int width, height;
+};
+
+struct _GstDivxEncClass {
+ GstElementClass parent_class;
+
+ /* signals */
+ void (*frame_encoded) (GstElement *element);
+};
+
+GType gst_divxenc_get_type(void);
+
+#ifdef __cplusplus
+}
+#endif /* __cplusplus */
+
+#endif /* __GST_DIVXENC_H__ */