Added corresponding divx/xvid decoders and fixed some first-hand encoder bugs

Original commit message from CVS:
Added corresponding divx/xvid decoders and fixed some first-hand encoder bugs

6 files changed, 512 insertions, 17 deletions

diff --git a/ext/xvid/Makefile.am b/ext/xvid/Makefile.am
index 267b1ccc..50e36859 100644
--- a/ext/xvid/Makefile.am
+++ b/ext/xvid/Makefile.am
@@ -2,9 +2,9 @@ plugindir = $(libdir)/gstreamer-@GST_MAJORMINOR@
 
 plugin_LTLIBRARIES = libgstxvid.la
 
-libgstxvid_la_SOURCES = gstxvidenc.c
+libgstxvid_la_SOURCES = gstxvidenc.c gstxviddec.c gstxvid.c
 libgstxvid_la_CFLAGS = $(GST_CFLAGS) $(XVID_CFLAGS)
 libgstxvid_la_LIBADD = $(XVID_LIBS)
 libgstxvid_la_LDFLAGS = $(GST_PLUGIN_LDFLAGS)
 
-noinst_HEADERS = gstxvidenc.h
+noinst_HEADERS = gstxvidenc.h gstxviddec.h
diff --git a/ext/xvid/gstxvid.c b/ext/xvid/gstxvid.c
new file mode 100644
index 00000000..3683a0f6
--- /dev/null
+++ b/ext/xvid/gstxvid.c
@@ -0,0 +1,40 @@
+/* GStreamer xvid decoder 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 "gstxviddec.h"
+#include "gstxvidenc.h"
+
+static gboolean
+plugin_init (GModule   *module,
+             GstPlugin *plugin)
+{
+  return (gst_xviddec_plugin_init(module, plugin) &&
+          gst_xvidenc_plugin_init(module, plugin));
+}
+
+GstPluginDesc plugin_desc = {
+  GST_VERSION_MAJOR,
+  GST_VERSION_MINOR,
+  "xvid",
+  plugin_init
+};
diff --git a/ext/xvid/gstxviddec.c b/ext/xvid/gstxviddec.c
new file mode 100644
index 00000000..4e5ab26f
--- /dev/null
+++ b/ext/xvid/gstxviddec.c
@@ -0,0 +1,388 @@
+/* GStreamer xvid decoder 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 "gstxviddec.h"
+#include <gst/video/video.h>
+
+/* elementfactory information */
+GstElementDetails gst_xviddec_details = {
+  "Xvid decoder",
+  "Codec/Video/Decoder",
+  "GPL",
+  "Xvid decoder based on xviddecore",
+  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("xviddec_sink",
+               "video/xvid",
+                 NULL)
+)
+
+GST_PAD_TEMPLATE_FACTORY(src_template,
+  "src",
+  GST_PAD_SRC,
+  GST_PAD_ALWAYS,
+  GST_CAPS_NEW("xviddec_src",
+               "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','Y','U')),
+                             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)
+)
+
+
+/* XvidDec signals and args */
+enum {
+  /* FILL ME */
+  LAST_SIGNAL
+};
+
+enum {
+  ARG_0
+  /* FILL ME */
+};
+
+
+static void             gst_xviddec_class_init   (GstXvidDecClass *klass);
+static void             gst_xviddec_init         (GstXvidDec      *xviddec);
+static void             gst_xviddec_dispose      (GObject         *object);
+static void             gst_xviddec_chain        (GstPad          *pad,
+                                                  GstBuffer       *buf);
+static GstPadLinkReturn gst_xviddec_connect      (GstPad          *pad,
+                                                  GstCaps         *vscapslist);
+
+static GstElementClass *parent_class = NULL;
+/* static guint gst_xviddec_signals[LAST_SIGNAL] = { 0 }; */
+
+
+GType
+gst_xviddec_get_type(void)
+{
+  static GType xviddec_type = 0;
+
+  if (!xviddec_type)
+  {
+    static const GTypeInfo xviddec_info = {
+      sizeof(GstXvidDecClass),
+      NULL,
+      NULL,
+      (GClassInitFunc) gst_xviddec_class_init,
+      NULL,
+      NULL,
+      sizeof(GstXvidDec),
+      0,
+      (GInstanceInitFunc) gst_xviddec_init,
+    };
+    xviddec_type = g_type_register_static(GST_TYPE_ELEMENT,
+                                          "GstXvidDec",
+                                          &xviddec_info, 0);
+  }
+  return xviddec_type;
+}
+
+
+static void
+gst_xviddec_class_init (GstXvidDecClass *klass)
+{
+  GObjectClass *gobject_class = (GObjectClass *) klass;
+  XVID_INIT_PARAM xinit;
+
+  /* set up xvid initially (function pointers, CPU flags) */
+  memset(&xinit, 0, sizeof(xinit));
+  xinit.cpu_flags = 0;
+  xvid_init(NULL, 0, &xinit, NULL);
+  if (xinit.api_version != API_VERSION) {
+    g_error("Xvid API version mismatch! %d.%d (that's us) != %d.%d (lib)",
+            (API_VERSION >> 8) & 0xff, API_VERSION & 0xff,
+            (xinit.api_version >> 8) & 0xff, xinit.api_version & 0xff);
+    return;
+  }
+
+  parent_class = g_type_class_ref(GST_TYPE_ELEMENT);
+
+  gobject_class->dispose = gst_xviddec_dispose;
+}
+
+
+static void
+gst_xviddec_init (GstXvidDec *xviddec)
+{
+  /* create the sink pad */
+  xviddec->sinkpad = gst_pad_new_from_template(
+                       GST_PAD_TEMPLATE_GET(sink_template),
+                       "sink");
+  gst_element_add_pad(GST_ELEMENT(xviddec), xviddec->sinkpad);
+
+  gst_pad_set_chain_function(xviddec->sinkpad, gst_xviddec_chain);
+  gst_pad_set_link_function(xviddec->sinkpad, gst_xviddec_connect);
+
+  /* create the src pad */
+  xviddec->srcpad = gst_pad_new_from_template(
+                      GST_PAD_TEMPLATE_GET(src_template),
+                      "src");
+  gst_element_add_pad(GST_ELEMENT(xviddec), xviddec->srcpad);
+
+  /* bitrate, etc. */
+  xviddec->width = xviddec->height = xviddec->csp = -1;
+
+  /* set xvid handle to NULL */
+  xviddec->handle = NULL;
+}
+
+
+static void
+gst_xviddec_unset (GstXvidDec *xviddec)
+{
+  /* unref this instance */
+  xvid_decore(xviddec->handle, XVID_DEC_DESTROY, NULL, NULL);
+  xviddec->handle = NULL;
+}
+
+
+static gboolean
+gst_xviddec_setup (GstXvidDec *xviddec)
+{
+  XVID_DEC_PARAM xdec;
+  int ret;
+
+  /* initialise parameters, see xvid documentation */
+  memset(&xdec, 0, sizeof(XVID_DEC_PARAM));
+  xdec.width = xviddec->width;
+  xdec.height = xviddec->height;
+
+  if ((ret = xvid_decore(NULL, XVID_DEC_CREATE,
+                         &xdec, NULL)) != XVID_ERR_OK) {
+    char *error;
+    switch (ret) {
+      case XVID_ERR_MEMORY:
+        error = "Memory allocation error";
+        break;
+      case XVID_ERR_FORMAT:
+        error = "Bad format";
+        break;
+      default:
+        error = "Internal failure";
+        break;
+    }
+    GST_DEBUG(GST_CAT_PLUGIN_INFO,
+              "Setting parameters %dx%d@%d failed: %s",
+              xviddec->width, xviddec->height, xviddec->csp, error);
+    return FALSE;
+  }
+
+  xviddec->handle = xdec.handle;
+
+  return TRUE;
+}
+
+
+static void
+gst_xviddec_dispose (GObject *object)
+{
+  GstXvidDec *xviddec = GST_XVIDDEC(object);
+
+  gst_xviddec_unset(xviddec);
+}
+
+
+static void
+gst_xviddec_chain (GstPad    *pad,
+                   GstBuffer *buf)
+{
+  GstXvidDec *xviddec;
+  GstBuffer *outbuf;
+  XVID_DEC_FRAME xframe;
+  int ret;
+
+  g_return_if_fail(pad != NULL);
+  g_return_if_fail(GST_IS_PAD(pad));
+  g_return_if_fail(buf != NULL);
+
+  xviddec = GST_XVIDDEC(GST_OBJECT_PARENT(pad));
+
+  if (!xviddec->handle) {
+    gst_element_error(GST_ELEMENT(xviddec),
+                      "No format set - aborting");
+    gst_buffer_unref(buf);
+    return;
+  }
+
+  outbuf = gst_buffer_new_and_alloc(xviddec->width *
+                                    xviddec->height *
+                                    xviddec->bpp / 8);
+  GST_BUFFER_TIMESTAMP(outbuf) = GST_BUFFER_TIMESTAMP(buf);
+  GST_BUFFER_SIZE(outbuf) = xviddec->width *
+                            xviddec->height *
+                            xviddec->bpp / 8;
+
+  /* encode and so ... */
+  xframe.bitstream = (void *) GST_BUFFER_DATA(buf);
+  xframe.image = (void *) GST_BUFFER_DATA(outbuf);
+  xframe.length = GST_BUFFER_SIZE(buf);
+  xframe.stride = xviddec->width * xviddec->bpp / 8;
+  xframe.colorspace = xviddec->csp;
+
+  if ((ret = xvid_decore(xviddec->handle, XVID_DEC_DECODE,
+                         &xframe, NULL))) {
+    gst_element_error(GST_ELEMENT(xviddec),
+                      "Error decoding xvid frame: %d\n", ret);
+    gst_buffer_unref(buf);
+    return;
+  }
+
+  gst_pad_push(xviddec->srcpad, outbuf);
+  gst_buffer_unref(buf);
+}
+
+
+static GstPadLinkReturn
+gst_xviddec_connect (GstPad  *pad,
+                     GstCaps *vscaps)
+{
+  GstXvidDec *xviddec;
+  GstCaps *caps;
+  struct {
+    guint32 fourcc;
+    gint    depth, bpp;
+    gint    csp;
+  } fmt_list[] = {
+    { GST_MAKE_FOURCC('Y','U','Y','V'), 16, 16, XVID_CSP_YUY2   },
+    { GST_MAKE_FOURCC('U','Y','V','Y'), 16, 16, XVID_CSP_UYVY   },
+    { GST_MAKE_FOURCC('Y','V','Y','U'), 16, 16, XVID_CSP_YVYU   },
+    { GST_MAKE_FOURCC('I','4','2','0'), 12, 12, XVID_CSP_I420   },
+    { GST_MAKE_FOURCC('I','Y','U','V'), 12, 12, XVID_CSP_I420   },
+    { GST_MAKE_FOURCC('Y','V','1','2'), 12, 12, XVID_CSP_YV12   },
+    { GST_MAKE_FOURCC('R','G','B',' '), 32, 32, XVID_CSP_RGB32  },
+    { GST_MAKE_FOURCC('R','G','B',' '), 24, 24, XVID_CSP_RGB24  },
+    { GST_MAKE_FOURCC('R','G','B',' '), 16, 16, XVID_CSP_RGB555 },
+    { GST_MAKE_FOURCC('R','G','B',' '), 15, 16, XVID_CSP_RGB565 },
+    { 0, 0, 0 }
+  };
+  gint i;
+
+  xviddec = GST_XVIDDEC(gst_pad_get_parent (pad));
+
+  /* if there's something old around, remove it */
+  if (xviddec->handle) {
+    gst_xviddec_unset(xviddec);
+  }
+
+  /* we are not going to act on variable caps */
+  if (!GST_CAPS_IS_FIXED(vscaps))
+    return GST_PAD_LINK_DELAYED;
+
+  /* if we get here, we know the input is xvid. we
+   * only need to bother with the output colorspace */
+  gst_caps_get_int(vscaps, "width", &xviddec->width);
+  gst_caps_get_int(vscaps, "height", &xviddec->height);
+
+  for (i = 0; fmt_list[i].fourcc != 0; i++) {
+    xviddec->csp = fmt_list[i].csp;
+
+    /* try making a caps to set on the other side */
+    if (fmt_list[i].fourcc == GST_MAKE_FOURCC('R','G','B',' ')) {
+      guint32 r_mask = 0, b_mask = 0, g_mask = 0;
+      switch (fmt_list[i].depth) {
+        case 15:
+          r_mask = 0xf800; g_mask = 0x07c0; b_mask = 0x003e;
+          break;
+        case 16:
+          r_mask = 0xf800; g_mask = 0x07e0; b_mask = 0x001f;
+          break;
+        case 24:
+          r_mask = 0xff0000; g_mask = 0x00ff00; b_mask = 0x0000ff;
+          break;
+        case 32:
+          r_mask = 0xff000000; g_mask = 0x00ff0000; b_mask = 0x0000ff00;
+          break;
+      }
+      caps = GST_CAPS_NEW("xviddec_src_pad_rgb",
+                          "video/raw",
+                            "width",      GST_PROPS_INT(xviddec->width),
+                            "height",     GST_PROPS_INT(xviddec->height),
+                            "format",     GST_PROPS_FOURCC(fmt_list[i].fourcc),
+                            "depth",      GST_PROPS_INT(fmt_list[i].depth),
+                            "bpp",        GST_PROPS_INT(fmt_list[i].bpp),
+                            "endianness", GST_PROPS_INT(G_BYTE_ORDER),
+                            "red_mask",   GST_PROPS_INT(r_mask),
+                            "green_mask", GST_PROPS_INT(g_mask),
+                            "blue_mask",  GST_PROPS_INT(b_mask),
+                            NULL);
+    } else {
+      caps = GST_CAPS_NEW("xviddec_src_pad_yuv",
+                          "video/raw",
+                            "width",      GST_PROPS_INT(xviddec->width),
+                            "height",     GST_PROPS_INT(xviddec->height),
+                            "format",     GST_PROPS_FOURCC(fmt_list[i].fourcc),
+                            NULL);
+    }
+
+    if (gst_pad_try_set_caps(xviddec->srcpad, caps) > 0) {
+      xviddec->csp = fmt_list[i].csp;
+      xviddec->bpp = fmt_list[i].bpp;
+      if (gst_xviddec_setup(xviddec))
+        return GST_PAD_LINK_OK;
+    }
+  }
+
+  /* if we got here - it's not good */
+  return GST_PAD_LINK_REFUSED;
+}
+
+
+gboolean
+gst_xviddec_plugin_init (GModule   *module,
+                         GstPlugin *plugin)
+{
+  GstElementFactory *factory;
+
+  /* create an elementfactory for the v4lmjpegsrcparse element */
+  factory = gst_element_factory_new("xviddec", GST_TYPE_XVIDDEC,
+                                    &gst_xviddec_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;
+}
diff --git a/ext/xvid/gstxviddec.h b/ext/xvid/gstxviddec.h
new file mode 100644
index 00000000..848ad261
--- /dev/null
+++ b/ext/xvid/gstxviddec.h
@@ -0,0 +1,72 @@
+/* GStreamer xvid decoder 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_XVIDDEC_H__
+#define __GST_XVIDDEC_H__
+
+#include <gst/gst.h>
+#include <xvid.h>
+
+#ifdef __cplusplus
+extern "C" {
+#endif /* __cplusplus */
+
+
+#define GST_TYPE_XVIDDEC \
+  (gst_xviddec_get_type())
+#define GST_XVIDDEC(obj) \
+  (G_TYPE_CHECK_INSTANCE_CAST((obj), GST_TYPE_XVIDDEC, GstXvidDec))
+#define GST_XVIDDEC_CLASS(klass) \
+  (G_TYPE_CHECK_CLASS_CAST((klass), GST_TYPE_XVIDDEC, GstXvidDec))
+#define GST_IS_XVIDDEC(obj) \
+  (G_TYPE_CHECK_INSTANCE_TYPE((obj), GST_TYPE_XVIDDEC))
+#define GST_IS_XVIDDEC_CLASS(obj) \
+  (G_TYPE_CHECK_CLASS_TYPE((klass), GST_TYPE_XVIDDEC))
+
+typedef struct _GstXvidDec GstXvidDec;
+typedef struct _GstXvidDecClass GstXvidDecClass;
+
+struct _GstXvidDec {
+  GstElement element;
+
+  /* pads */
+  GstPad *sinkpad, *srcpad;
+
+  /* xvid handle */
+  void *handle;
+
+  /* video (output) settings */
+  int csp, bpp;
+  int width, height;
+};
+
+struct _GstXvidDecClass {
+  GstElementClass parent_class;
+};
+
+GType gst_xviddec_get_type(void);
+
+gboolean gst_xviddec_plugin_init (GModule   *module,
+                                  GstPlugin *plugin);
+
+#ifdef __cplusplus
+}
+#endif /* __cplusplus */
+
+#endif /* __GST_XVIDDEC_H__ */
diff --git a/ext/xvid/gstxvidenc.c b/ext/xvid/gstxvidenc.c
index e916bda1..7efe42f9 100644
--- a/ext/xvid/gstxvidenc.c
+++ b/ext/xvid/gstxvidenc.c
@@ -61,7 +61,7 @@ GST_PAD_TEMPLATE_FACTORY(src_template,
   "src",
   GST_PAD_SRC,
   GST_PAD_ALWAYS,
-  GST_CAPS_NEW("xvidenc_sink",
+  GST_CAPS_NEW("xvidenc_src",
                "video/xvid",
                  NULL)
 )
@@ -191,9 +191,9 @@ gst_xvidenc_init (GstXvidEnc *xvidenc)
   gst_pad_set_link_function(xvidenc->sinkpad, gst_xvidenc_connect);
 
   /* create the src pad */
-  xvidenc->sinkpad = gst_pad_new_from_template(
-                       GST_PAD_TEMPLATE_GET(src_template),
-                       "src");
+  xvidenc->srcpad = gst_pad_new_from_template(
+                      GST_PAD_TEMPLATE_GET(src_template),
+                      "src");
   gst_element_add_pad(GST_ELEMENT(xvidenc), xvidenc->srcpad);
 
   /* bitrate, etc. */
@@ -460,9 +460,9 @@ gst_xvidenc_get_property (GObject    *object,
 }
 
 
-static gboolean
-plugin_init (GModule   *module,
-             GstPlugin *plugin)
+gboolean
+gst_xvidenc_plugin_init (GModule   *module,
+                         GstPlugin *plugin)
 {
   GstElementFactory *factory;
 
@@ -484,11 +484,3 @@ plugin_init (GModule   *module,
 
   return TRUE;
 }
-
-
-GstPluginDesc plugin_desc = {
-  GST_VERSION_MAJOR,
-  GST_VERSION_MINOR,
-  "xvidenc",
-  plugin_init
-};
diff --git a/ext/xvid/gstxvidenc.h b/ext/xvid/gstxvidenc.h
index 24cf6a2a..198cadf3 100644
--- a/ext/xvid/gstxvidenc.h
+++ b/ext/xvid/gstxvidenc.h
@@ -71,6 +71,9 @@ struct _GstXvidEncClass {
 
 GType gst_xvidenc_get_type(void);
 
+gboolean gst_xvidenc_plugin_init (GModule   *module,
+                                  GstPlugin *plugin);
+
 #ifdef __cplusplus
 }
 #endif /* __cplusplus */