New plugin for generating a test video stream

Original commit message from CVS:
New plugin for generating a test video stream

gst/videotestsrc/Makefile.am

@@ -0,0 +1,12 @@
+plugindir = $(libdir)/gst
+
+plugin_LTLIBRARIES = libgstvideotestsrc.la
+
+libgstvideotestsrc_la_SOURCES = \
+			gstvideotestsrc.c
+
+libgstvideotestsrc_la_CFLAGS = -O2 $(FOMIT_FRAME_POINTER) -funroll-all-loops -finline-functions -ffast-math $(GST_CFLAGS)
+libgstvideotestsrc_la_LIBADD =
+libgstvideotestsrc_la_LDFLAGS = $(GST_PLUGIN_LDFLAGS)
+
+noinst_HEADERS = gstvideotestsrc.h

gst/videotestsrc/gstvideotestsrc.c

@@ -0,0 +1,352 @@
+/* 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.
+ */
+
+
+/*#define DEBUG_ENABLED */
+#include <gstvideotestsrc.h>
+
+#include <stdlib.h>
+
+
+
+/* elementfactory information */
+static GstElementDetails videotestsrc_details = {
+  "Video test source",
+  "Source/Video",
+  "Creates a test video stream",
+  VERSION,
+  "David A. Schleef <ds@schleef.org>",
+  "(C) 2002",
+};
+
+/* GstVideotestsrc signals and args */
+enum {
+  /* FILL ME */
+  LAST_SIGNAL
+};
+
+enum {
+  ARG_0,
+  ARG_WIDTH,
+  ARG_HEIGHT,
+  ARG_METHOD,
+  /* FILL ME */
+};
+
+GST_PAD_TEMPLATE_FACTORY (src_templ,
+  "src",
+  GST_PAD_SRC,
+  GST_PAD_ALWAYS,
+  GST_CAPS_NEW (
+    "videotestsrc_caps",
+    "video/raw",
+      "format",		GST_PROPS_FOURCC (GST_MAKE_FOURCC ('I','4','2','0')),
+      "width",		GST_PROPS_INT(640),
+      "height",		GST_PROPS_INT(480)
+  )
+)
+
+#define GST_TYPE_VIDEOTESTSRC_METHOD (gst_videotestsrc_method_get_type())
+static GType
+gst_videotestsrc_method_get_type (void)
+{
+  static GType videotestsrc_method_type = 0;
+  static GEnumValue videotestsrc_methods[] = {
+    { GST_VIDEOTESTSRC_POINT_SAMPLE, "0", "Point Sample" },
+    { GST_VIDEOTESTSRC_NEAREST,      "1", "Nearest" },
+    { GST_VIDEOTESTSRC_BILINEAR,     "2", "Bilinear" },
+    { GST_VIDEOTESTSRC_BICUBIC,      "3", "Bicubic" },
+    { 0, NULL, NULL },
+  };
+  if (!videotestsrc_method_type) {
+    videotestsrc_method_type = g_enum_register_static ("GstVideotestsrcMethod", videotestsrc_methods);
+  }
+  return videotestsrc_method_type;
+}
+
+static void	gst_videotestsrc_class_init	(GstVideotestsrcClass *klass);
+static void	gst_videotestsrc_init		(GstVideotestsrc *videotestsrc);
+
+static void	gst_videotestsrc_set_property		(GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec);
+static void	gst_videotestsrc_get_property		(GObject *object, guint prop_id, GValue *value, GParamSpec *pspec);
+
+static GstBuffer * gst_videotestsrc_get (GstPad *pad);
+
+static GstElementClass *parent_class = NULL;
+
+void gst_videotestsrc_setup (GstVideotestsrc *v);
+static void random_chars(unsigned char *dest, int nbytes);
+static void gst_videotestsrc_random_yuv (GstVideotestsrc *v, unsigned char *dest, int w, int h);
+
+
+GType
+gst_videotestsrc_get_type (void)
+{
+  static GType videotestsrc_type = 0;
+
+  if (!videotestsrc_type) {
+    static const GTypeInfo videotestsrc_info = {
+      sizeof(GstVideotestsrcClass),      NULL,
+      NULL,
+      (GClassInitFunc)gst_videotestsrc_class_init,
+      NULL,
+      NULL,
+      sizeof(GstVideotestsrc),
+      0,
+      (GInstanceInitFunc)gst_videotestsrc_init,
+    };
+    videotestsrc_type = g_type_register_static(GST_TYPE_ELEMENT, "GstVideotestsrc", &videotestsrc_info, 0);
+  }
+  return videotestsrc_type;
+}
+
+static void
+gst_videotestsrc_class_init (GstVideotestsrcClass *klass)
+{
+  GObjectClass *gobject_class;
+  GstElementClass *gstelement_class;
+
+  gobject_class = (GObjectClass*)klass;
+  gstelement_class = (GstElementClass*)klass;
+
+  g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_WIDTH,
+    g_param_spec_int("width","width","width",
+                     G_MININT,G_MAXINT,0,G_PARAM_READWRITE)); /* CHECKME */
+  g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_HEIGHT,
+    g_param_spec_int("height","height","height",
+                     G_MININT,G_MAXINT,0,G_PARAM_READWRITE)); /* CHECKME */
+  g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_METHOD,
+    g_param_spec_enum("method","method","method",
+                      GST_TYPE_VIDEOTESTSRC_METHOD,0,G_PARAM_READWRITE)); /* CHECKME! */
+
+  parent_class = g_type_class_ref(GST_TYPE_ELEMENT);
+
+  gobject_class->set_property = gst_videotestsrc_set_property;
+  gobject_class->get_property = gst_videotestsrc_get_property;
+
+}
+
+static GstPadConnectReturn
+gst_videotestsrc_srcconnect (GstPad *pad, GstCaps *caps)
+{
+  GstVideotestsrc *videotestsrc;
+
+  GST_DEBUG(0,"gst_videotestsrc_srcconnect");
+  videotestsrc = GST_VIDEOTESTSRC (gst_pad_get_parent (pad));
+
+#if 0
+  if (!GST_CAPS_IS_FIXED (caps)) {
+    return GST_PAD_CONNECT_DELAYED;
+  }
+#endif
+
+  gst_caps_get_fourcc_int (caps, "format", &videotestsrc->format);
+  gst_caps_get_int (caps, "width", &videotestsrc->width);
+  gst_caps_get_int (caps, "height", &videotestsrc->height);
+
+  gst_videotestsrc_setup(videotestsrc);
+
+  GST_DEBUG (0,"size %d x %d",videotestsrc->width, videotestsrc->height);
+
+  return GST_PAD_CONNECT_OK;
+}
+
+static void
+gst_videotestsrc_init (GstVideotestsrc *videotestsrc)
+{
+  GST_DEBUG(0,"gst_videotestsrc_init");
+
+  /*gst_pad_set_negotiate_function(videotestsrc->sinkpad,videotestsrc_negotiate_sink); */
+  //gst_element_add_pad(GST_ELEMENT(videotestsrc),videotestsrc->sinkpad);
+  //gst_pad_set_chain_function(videotestsrc->sinkpad,gst_videotestsrc_chain);
+
+  videotestsrc->srcpad = gst_pad_new_from_template (
+		  GST_PAD_TEMPLATE_GET (src_templ), "src");
+  /*gst_pad_set_negotiate_function(videotestsrc->srcpad,videotestsrc_negotiate_src); */
+  gst_element_add_pad(GST_ELEMENT(videotestsrc),videotestsrc->srcpad);
+  gst_pad_set_get_function(videotestsrc->srcpad,gst_videotestsrc_get);
+  gst_pad_set_connect_function(videotestsrc->srcpad,gst_videotestsrc_srcconnect);
+
+  videotestsrc->width = 640;
+  videotestsrc->height = 480;
+}
+
+
+static GstBuffer *
+gst_videotestsrc_get (GstPad *pad)
+{
+  GstVideotestsrc *videotestsrc;
+  gulong newsize;
+  GstBuffer *buf;
+
+  GST_DEBUG (0,"gst_videotestsrc_get");
+
+  g_return_val_if_fail (pad != NULL, NULL);
+  g_return_val_if_fail (GST_IS_PAD (pad), NULL);
+
+  videotestsrc = GST_VIDEOTESTSRC (gst_pad_get_parent (pad));
+
+  newsize = videotestsrc->width*videotestsrc->height + 
+	  videotestsrc->width*videotestsrc->height/2;
+
+  GST_DEBUG(0,"size=%ld %dx%d",newsize,
+	videotestsrc->width, videotestsrc->height);
+
+  buf = gst_buffer_new();
+  /* XXX this is wrong for anything but I420 */
+  GST_BUFFER_SIZE(buf) = newsize;
+  GST_BUFFER_DATA(buf) = g_malloc (newsize);
+  g_return_val_if_fail(GST_BUFFER_DATA(buf) != NULL, NULL);
+  //GST_BUFFER_TIMESTAMP(buf) = GST_BUFFER_TIMESTAMP(buf);
+
+  gst_videotestsrc_random_yuv(videotestsrc, (void *)GST_BUFFER_DATA(buf),
+		  videotestsrc->width, videotestsrc->height);
+
+  return buf;
+}
+
+static void
+gst_videotestsrc_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec)
+{
+  GstVideotestsrc *src;
+
+  /* it's not null if we got it, but it might not be ours */
+  g_return_if_fail(GST_IS_VIDEOTESTSRC(object));
+  src = GST_VIDEOTESTSRC(object);
+
+  GST_DEBUG(0,"gst_videotestsrc_set_property");
+  switch (prop_id) {
+    case ARG_WIDTH:
+      src->width = g_value_get_int (value);
+      break;
+    case ARG_HEIGHT:
+      src->height = g_value_get_int (value);
+      break;
+    case ARG_METHOD:
+      src->method = g_value_get_enum (value);
+      break;
+    default:
+      break;
+  }
+}
+
+static void
+gst_videotestsrc_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
+{
+  GstVideotestsrc *src;
+
+  /* it's not null if we got it, but it might not be ours */
+  g_return_if_fail(GST_IS_VIDEOTESTSRC(object));
+  src = GST_VIDEOTESTSRC(object);
+
+  switch (prop_id) {
+    case ARG_WIDTH:
+      g_value_set_int (value, src->width);
+      break;
+    case ARG_HEIGHT:
+      g_value_set_int (value, src->height);
+      break;
+    case ARG_METHOD:
+      g_value_set_enum (value, src->method);
+      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 videotestsrc element */
+  factory = gst_element_factory_new("videotestsrc",GST_TYPE_VIDEOTESTSRC,
+                                   &videotestsrc_details);
+  g_return_val_if_fail(factory != NULL, FALSE);
+
+  gst_element_factory_add_pad_template (factory, GST_PAD_TEMPLATE_GET (src_templ));
+
+  gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory));
+
+  return TRUE;
+}
+
+GstPluginDesc plugin_desc = {
+  GST_VERSION_MAJOR,
+  GST_VERSION_MINOR,
+  "videotestsrc",
+  plugin_init
+};
+
+
+
+/* Non-GST specific stuff */
+
+void
+gst_videotestsrc_setup (GstVideotestsrc *v)
+{
+
+}
+
+
+static void
+random_chars(unsigned char *dest, int nbytes)
+{
+	int i;
+
+	for(i=0;i<nbytes;i++){
+		dest[i] = random();
+	}
+}
+
+static int u_colors[] = { 128,   0, 192,   0, 255,  64, 255 };
+static int v_colors[] = { 128, 192,   0,   0, 255, 255,  64 };
+
+static void
+gst_videotestsrc_random_yuv (GstVideotestsrc *v, unsigned char *dest, int w, int h)
+{
+	unsigned char *up = dest + w*h;
+	unsigned char *vp = up + w*h/4;
+	int w7;
+	int i,j;
+
+	memset(dest,255,w*h/2);
+	random_chars(dest + w*h/2,w*h/2);
+
+	//random_chars(dest + w*h, w*h/4);
+	memset(dest + w*h, 128, w*h/4);
+	
+	//random_chars(dest + w*h + w*h/4, w*h/4);
+	memset(dest + w*h + w*h/4, 128, w*h/4);
+
+	w7 = w/2/7 + 1;
+	for(i=0;i<7;i++){
+		for(j=0;j<w7;j++){
+			up[i*w7 + j] = u_colors[i];
+			vp[i*w7 + j] = v_colors[i];
+		}
+	}
+	for(i=0;i<h/4;i++){
+		memcpy(up+(w/2)*i, up, w/2);
+		memcpy(vp+(w/2)*i, vp, w/2);
+	}
+}
+

gst/videotestsrc/gstvideotestsrc.h

@@ -0,0 +1,85 @@
+/* GStreamer
+ * Copyright (C) <2002> David A. Schleef <ds@schleef.org>
+ * 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.
+ */
+
+
+#ifndef __GST_VIDEOTESTSRC_H__
+#define __GST_VIDEOTESTSRC_H__
+
+
+#include <config.h>
+#include <gst/gst.h>
+
+
+#ifdef __cplusplus
+extern "C" {
+#endif /* __cplusplus */
+
+
+#define GST_TYPE_VIDEOTESTSRC \
+  (gst_videotestsrc_get_type())
+#define GST_VIDEOTESTSRC(obj) \
+  (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_VIDEOTESTSRC,GstVideotestsrc))
+#define GST_VIDEOTESTSRC_CLASS(klass) \
+  (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_VIDEOTESTSRC,GstVideotestsrc))
+#define GST_IS_VIDEOTESTSRC(obj) \
+  (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_VIDEOTESTSRC))
+#define GST_IS_VIDEOTESTSRC_CLASS(obj) \
+  (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_VIDEOTESTSRC))
+
+typedef enum {
+  GST_VIDEOTESTSRC_POINT_SAMPLE,
+  GST_VIDEOTESTSRC_NEAREST,
+  GST_VIDEOTESTSRC_BILINEAR,
+  GST_VIDEOTESTSRC_BICUBIC
+} GstVideoTestSrcMethod;
+
+typedef struct _GstVideotestsrc GstVideotestsrc;
+typedef struct _GstVideotestsrcClass GstVideotestsrcClass;
+
+struct _GstVideotestsrc {
+  GstElement element;
+
+  GstPad *sinkpad,*srcpad;
+
+  /* video state */
+  guint32 format;
+  gint width;
+  gint height;
+  GstVideoTestSrcMethod method;
+  
+  /* private */
+  guchar *temp;
+};
+
+struct _GstVideotestsrcClass {
+  GstElementClass parent_class;
+};
+
+GType gst_videotestsrc_get_type(void);
+
+void gst_videotestsrc_setup(GstVideotestsrc *);
+#define gst_videotestsrc_scale(scale, src, dest) (scale)->scale_cc((scale), (src), (dest))
+
+#ifdef __cplusplus
+}
+#endif /* __cplusplus */
+
+
+#endif /* __GST_VIDEOTESTSRC_H__ */