Merge remote-tracking branch 'origin/master' into 0.11
[platform/upstream/gstreamer.git] / gst / effectv / gstwarp.c
index c098ab6..80161a7 100644 (file)
@@ -1,11 +1,11 @@
 /* GStreamer
  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
+ * Copyright (C) <2003> David Schleef <ds@schleef.org>
  *
  * EffecTV - Realtime Digital Video Effector
  * Copyright (C) 2001 FUKUCHI Kentarou
- * 
- * EffecTV is free software. This library is free software;
- * you can redistribute it and/or
+ *
+ * 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.
  * License along with this library; if not, write to the
  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  * Boston, MA 02111-1307, USA.
- * From main.c of warp-1.1:
+ */
+
+/*
+ * This file was (probably) generated from gstvideotemplate.c,
+ * gstvideotemplate.c,v 1.11 2004/01/07 08:56:45 ds Exp 
+ */
+
+/* From main.c of warp-1.1:
  *
  *      Simple DirectMedia Layer demo
  *      Realtime picture 'gooing'
  *      by sam lantinga slouken@devolution.com
  */
 
+/**
+ * SECTION:element-warptv
+ *
+ * WarpTV does realtime goo'ing of the video input.
+ *
+ * <refsect2>
+ * <title>Example launch line</title>
+ * |[
+ * gst-launch -v videotestsrc ! warptv ! videoconvert ! autovideosink
+ * ]| This pipeline shows the effect of warptv on a test stream.
+ * </refsect2>
+ */
+
 #ifdef HAVE_CONFIG_H
 #include "config.h"
 #endif
+
 #include <string.h>
 #include <math.h>
-#include <gst/gst.h>
-#include "gsteffectv.h"
+
+#include "gstwarp.h"
+#include <gst/video/gstvideometa.h>
+#include <gst/video/gstvideopool.h>
 
 #ifndef M_PI
-#define M_PI   3.14159265358979323846
+#define M_PI    3.14159265358979323846
 #endif
 
-#define GST_TYPE_WARPTV \
-  (gst_warptv_get_type())
-#define GST_WARPTV(obj) \
-  (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_WARPTV,GstWarpTV))
-#define GST_WARPTV_CLASS(klass) \
-  (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_ULAW,GstWarpTV))
-#define GST_IS_WARPTV(obj) \
-  (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_WARPTV))
-#define GST_IS_WARPTV_CLASS(obj) \
-  (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_WARPTV))
-
-typedef struct _GstWarpTV GstWarpTV;
-typedef struct _GstWarpTVClass GstWarpTVClass;
-
-struct _GstWarpTV
+#define gst_warptv_parent_class parent_class
+G_DEFINE_TYPE (GstWarpTV, gst_warptv, GST_TYPE_VIDEO_FILTER);
+
+static void initSinTable ();
+static void initDistTable (GstWarpTV * filter, gint width, gint height);
+
+static GstStaticPadTemplate gst_warptv_src_template =
+GST_STATIC_PAD_TEMPLATE ("src",
+    GST_PAD_SRC,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("{ RGBx, xRGB, BGRx, xBGR }"))
+    );
+
+static GstStaticPadTemplate gst_warptv_sink_template =
+GST_STATIC_PAD_TEMPLATE ("sink",
+    GST_PAD_SINK,
+    GST_PAD_ALWAYS,
+    GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("{ RGBx, xRGB, BGRx, xBGR }"))
+    );
+
+static gboolean
+gst_warptv_set_caps (GstBaseTransform * btrans, GstCaps * incaps,
+    GstCaps * outcaps)
 {
-  GstElement element;
-
-  GstPad *sinkpad, *srcpad;
-
+  GstWarpTV *filter = GST_WARPTV (btrans);
+  GstVideoInfo info;
   gint width, height;
-  gint *offstable;
-  gint32 *disttable;
-  gint32 ctable[1024];
-  gint32 sintable[1024+256];
-  gint tval;
-};
-
-struct _GstWarpTVClass
-{
-  GstElementClass parent_class;
-};
-
-/* elementfactory information */
-GstElementDetails gst_warptv_details = {
-  "WarpTV",
-  "Filter/Video/Effect",
-  "LGPL",
-  "WarpTV does realtime goo'ing of the video input",
-  VERSION,
-  "Sam Lantinga <slouken@devolution.com>",
-  "Wim Taymans <wim.taymans@chello.be>, "
-  "(C) 2001 FUKUCHI Kentarou",
-};
-
-
-/* Filter signals and args */
-enum
-{
-  /* FILL ME */
-  LAST_SIGNAL
-};
-
-enum
-{
-  ARG_0,
-};
 
-static void    gst_warptv_class_init           (GstWarpTVClass * klass);
-static void    gst_warptv_init                 (GstWarpTV * filter);
+  if (!gst_video_info_from_caps (&info, incaps))
+    goto invalid_caps;
 
-static void    gst_warptv_initialize           (GstWarpTV *filter);
+  filter->info = info;
 
-static void    gst_warptv_set_property         (GObject * object, guint prop_id,
-                                                const GValue * value, GParamSpec * pspec);
-static void    gst_warptv_get_property         (GObject * object, guint prop_id,
-                                                GValue * value, GParamSpec * pspec);
+  width = GST_VIDEO_INFO_WIDTH (&info);
+  height = GST_VIDEO_INFO_HEIGHT (&info);
 
-static void    gst_warptv_chain                (GstPad * pad, GstBuffer * buf);
+  g_free (filter->disttable);
+  filter->disttable = g_malloc (width * height * sizeof (guint32));
+  initDistTable (filter, width, height);
 
-static GstElementClass *parent_class = NULL;
-/*static guint gst_warptv_signals[LAST_SIGNAL] = { 0 }; */
+  return TRUE;
 
-GType gst_warptv_get_type (void)
-{
-  static GType warptv_type = 0;
-
-  if (!warptv_type) {
-    static const GTypeInfo warptv_info = {
-      sizeof (GstWarpTVClass), NULL,
-      NULL,
-      (GClassInitFunc) gst_warptv_class_init,
-      NULL,
-      NULL,
-      sizeof (GstWarpTV),
-      0,
-      (GInstanceInitFunc) gst_warptv_init,
-    };
-
-    warptv_type = g_type_register_static (GST_TYPE_ELEMENT, "GstWarpTV", &warptv_info, 0);
+  /* ERRORS */
+invalid_caps:
+  {
+    GST_DEBUG_OBJECT (filter, "invalid caps received");
+    return FALSE;
   }
-  return warptv_type;
 }
 
-static void
-gst_warptv_class_init (GstWarpTVClass * klass)
-{
-  GObjectClass *gobject_class;
-  GstElementClass *gstelement_class;
-
-  gobject_class = (GObjectClass *) klass;
-  gstelement_class = (GstElementClass *) klass;
-
-  parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
-
-  gobject_class->set_property = gst_warptv_set_property;
-  gobject_class->get_property = gst_warptv_get_property;
-}
-
-static GstPadLinkReturn
-gst_warptv_sinkconnect (GstPad * pad, GstCaps * caps)
-{
-  GstWarpTV *filter;
-
-  filter = GST_WARPTV (gst_pad_get_parent (pad));
-
-  if (!GST_CAPS_IS_FIXED (caps))
-    return GST_PAD_LINK_DELAYED;
-
-  gst_caps_get_int (caps, "width", &filter->width);
-  gst_caps_get_int (caps, "height", &filter->height);
-
-  gst_warptv_initialize (filter);
-
-  return gst_pad_try_set_caps (filter->srcpad, gst_caps_ref (caps));
-}
+static gint32 sintable[1024 + 256];
 
 static void
-gst_warptv_init (GstWarpTV * filter)
-{
-  filter->sinkpad = gst_pad_new_from_template (gst_effectv_sink_factory (), "sink");
-  gst_pad_set_chain_function (filter->sinkpad, gst_warptv_chain);
-  gst_pad_set_link_function (filter->sinkpad, gst_warptv_sinkconnect);
-  gst_element_add_pad (GST_ELEMENT (filter), filter->sinkpad);
-
-  filter->srcpad = gst_pad_new_from_template (gst_effectv_src_factory (), "src");
-  gst_element_add_pad (GST_ELEMENT (filter), filter->srcpad);
-
-  filter->tval = 0;
-  filter->disttable = NULL;
-  filter->offstable = NULL;
-}
-
-
-static void 
-initSinTable (GstWarpTV *filter) 
+initSinTable (void)
 {
-  gint32       *tptr, *tsinptr;
-  double       i;
+  gint32 *tptr, *tsinptr;
+  gint i;
 
-  tsinptr = tptr = filter->sintable;
+  tsinptr = tptr = sintable;
 
   for (i = 0; i < 1024; i++)
     *tptr++ = (int) (sin (i * M_PI / 512) * 32767);
@@ -198,35 +128,22 @@ initSinTable (GstWarpTV *filter)
     *tptr++ = *tsinptr++;
 }
 
-static void 
-initOffsTable (GstWarpTV *filter) 
-{
-  int y;
-       
-  for (y = 0; y < filter->height; y++) {
-    filter->offstable[y] = y * filter->width;
-  }
-}
-      
-static void 
-initDistTable (GstWarpTV *filter) 
+static void
+initDistTable (GstWarpTV * filter, gint width, gint height)
 {
   gint32 halfw, halfh, *distptr;
-#ifdef PS2
-  float x,y,m;
-#else
-  double x,y,m;
-#endif
+  gint x, y;
+  float m;
 
-  halfw = filter->width>> 1;
-  halfh = filter->height >> 1;
+  halfw = width >> 1;
+  halfh = height >> 1;
 
   distptr = filter->disttable;
 
-  m = sqrt ((double)(halfw * halfw + halfh * halfh));
+  m = sqrt ((double) (halfw * halfw + halfh * halfh));
 
   for (y = -halfh; y < halfh; y++)
-    for (x= -halfw; x < halfw; x++)
+    for (x = -halfw; x < halfw; x++)
 #ifdef PS2
       *distptr++ = ((int) ((sqrtf (x * x + y * y) * 511.9999) / m)) << 1;
 #else
@@ -234,54 +151,42 @@ initDistTable (GstWarpTV *filter)
 #endif
 }
 
-static void 
-gst_warptv_initialize (GstWarpTV *filter) 
+static GstFlowReturn
+gst_warptv_transform (GstBaseTransform * trans, GstBuffer * in, GstBuffer * out)
 {
-  g_free (filter->disttable);
-  g_free (filter->offstable);
+  GstWarpTV *warptv = GST_WARPTV (trans);
+  gint width, height;
+  gint xw, yw, cw;
+  gint32 c, i, x, y, dx, dy, maxx, maxy;
+  gint32 *ctptr, *distptr;
+  gint32 *ctable;
+  guint32 *src, *dest;
+  gint sstride, dstride;
+  GstVideoFrame in_frame, out_frame;
 
-  filter->offstable = (guint32 *) g_malloc (filter->height * sizeof (guint32));      
-  filter->disttable = g_malloc (filter->width * filter->height * sizeof (guint32));
+  gst_video_frame_map (&in_frame, &warptv->info, in, GST_MAP_READ);
+  gst_video_frame_map (&out_frame, &warptv->info, out, GST_MAP_WRITE);
 
-  initSinTable (filter);
-  initOffsTable (filter);
-  initDistTable (filter);
-}
+  src = GST_VIDEO_FRAME_PLANE_DATA (&in_frame, 0);
+  dest = GST_VIDEO_FRAME_PLANE_DATA (&out_frame, 0);
 
-static void
-gst_warptv_chain (GstPad * pad, GstBuffer * buf)
-{
-  GstWarpTV *filter;
-  guint32 *src, *dest;
-  gint xw,yw,cw;
-  GstBuffer *outbuf;
-  gint32 c,i, x,y, dx,dy, maxx, maxy;
-  gint32 width, height, skip, *ctptr, *distptr;
-  gint32 *sintable, *ctable;
-
-  filter = GST_WARPTV (gst_pad_get_parent (pad));
-
-  src = (guint32 *) GST_BUFFER_DATA (buf);
-
-  outbuf = gst_buffer_new ();
-  GST_BUFFER_SIZE (outbuf) = (filter->width * filter->height * sizeof(guint32));
-  dest = (guint32 *) GST_BUFFER_DATA (outbuf) = g_malloc (GST_BUFFER_SIZE (outbuf));
-  GST_BUFFER_TIMESTAMP (outbuf) = GST_BUFFER_TIMESTAMP (buf);
-  
-  xw  = (gint) (sin ((filter->tval + 100) * M_PI / 128) * 30);
-  yw  = (gint) (sin ((filter->tval) * M_PI / 256) * -35);
-  cw  = (gint) (sin ((filter->tval - 70) * M_PI / 64) * 50);
-  xw += (gint) (sin ((filter->tval - 10) * M_PI / 512) * 40);
-  yw += (gint) (sin ((filter->tval + 30) * M_PI / 512) * 40);    
-
-  ctptr = filter->ctable;
-  distptr = filter->disttable;
-  width = filter->width;
-  height = filter->height;
-  sintable = filter->sintable;
-  ctable = filter->ctable;
+  sstride = GST_VIDEO_FRAME_PLANE_STRIDE (&in_frame, 0) / 4;
+  dstride = GST_VIDEO_FRAME_PLANE_STRIDE (&out_frame, 0) / 4;
+
+  width = GST_VIDEO_FRAME_WIDTH (&in_frame);
+  height = GST_VIDEO_FRAME_HEIGHT (&in_frame);
+
+  GST_OBJECT_LOCK (warptv);
+  xw = (gint) (sin ((warptv->tval + 100) * M_PI / 128) * 30);
+  yw = (gint) (sin ((warptv->tval) * M_PI / 256) * -35);
+  cw = (gint) (sin ((warptv->tval - 70) * M_PI / 64) * 50);
+  xw += (gint) (sin ((warptv->tval - 10) * M_PI / 512) * 40);
+  yw += (gint) (sin ((warptv->tval + 30) * M_PI / 512) * 40);
+
+  ctptr = warptv->ctable;
+  distptr = warptv->disttable;
+  ctable = warptv->ctable;
 
-  skip = 0 ; /* video_width*sizeof(RGB32)/4 - video_width;; */
   c = 0;
 
   for (x = 0; x < 512; x++) {
@@ -290,60 +195,110 @@ gst_warptv_chain (GstPad * pad, GstBuffer * buf)
     *ctptr++ = ((sintable[i + 256] * xw) >> 15);
     c += cw;
   }
-  maxx = width - 2; maxy = height - 2;
+  maxx = width - 2;
+  maxy = height - 2;
 
   for (y = 0; y < height - 1; y++) {
     for (x = 0; x < width; x++) {
-      i = *distptr++; 
-      dx = ctable [i + 1] + x; 
-      dy = ctable [i] + y;      
-
-      if (dx < 0) dx = 0; 
-      else if (dx > maxx) dx = maxx; 
-   
-      if (dy < 0) dy = 0; 
-      else if (dy > maxy) dy = maxy; 
-      *dest++ = src[filter->offstable[dy] + dx]; 
+      i = *distptr++;
+      dx = ctable[i + 1] + x;
+      dy = ctable[i] + y;
+
+      if (dx < 0)
+        dx = 0;
+      else if (dx > maxx)
+        dx = maxx;
+
+      if (dy < 0)
+        dy = 0;
+      else if (dy > maxy)
+        dy = maxy;
+
+      dest[x] = src[dy * sstride + dx];
     }
-    dest += skip;
+    dest += dstride;
   }
 
-  filter->tval = (filter->tval + 1) & 511;
+  warptv->tval = (warptv->tval + 1) & 511;
+  GST_OBJECT_UNLOCK (warptv);
 
-  gst_buffer_unref (buf);
+  gst_video_frame_unmap (&in_frame);
+  gst_video_frame_unmap (&out_frame);
 
-  gst_pad_push (filter->srcpad, outbuf);
+  return GST_FLOW_OK;
 }
 
-static void
-gst_warptv_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec)
+static gboolean
+gst_warptv_start (GstBaseTransform * trans)
+{
+  GstWarpTV *warptv = GST_WARPTV (trans);
+
+  warptv->tval = 0;
+
+  return TRUE;
+}
+
+static gboolean
+gst_wraptv_decide_allocation (GstBaseTransform * trans, GstQuery * query)
 {
-  GstWarpTV *filter;
+  GstBufferPool *pool = NULL;
+  guint size, min, max, prefix, alignment;
 
-  /* it's not null if we got it, but it might not be ours */
-  g_return_if_fail (GST_IS_WARPTV (object));
+  gst_query_parse_allocation_params (query, &size, &min, &max, &prefix,
+      &alignment, &pool);
 
-  filter = GST_WARPTV (object);
+  if (pool) {
+    GstStructure *config;
 
-  switch (prop_id) {
-    default:
-      break;
+    config = gst_buffer_pool_get_config (pool);
+    gst_buffer_pool_config_add_option (config,
+        GST_BUFFER_POOL_OPTION_VIDEO_META);
+    gst_buffer_pool_set_config (pool, config);
   }
+  return TRUE;
 }
 
 static void
-gst_warptv_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec)
+gst_warptv_finalize (GObject * object)
 {
-  GstWarpTV *filter;
+  GstWarpTV *warptv = GST_WARPTV (object);
 
-  /* it's not null if we got it, but it might not be ours */
-  g_return_if_fail (GST_IS_WARPTV (object));
+  g_free (warptv->disttable);
+  warptv->disttable = NULL;
 
-  filter = GST_WARPTV (object);
+  G_OBJECT_CLASS (parent_class)->finalize (object);
+}
 
-  switch (prop_id) {
-    default:
-      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
-      break;
-  }
+static void
+gst_warptv_class_init (GstWarpTVClass * klass)
+{
+  GObjectClass *gobject_class = (GObjectClass *) klass;
+  GstElementClass *gstelement_class = (GstElementClass *) klass;
+  GstBaseTransformClass *trans_class = (GstBaseTransformClass *) klass;
+
+  gobject_class->finalize = gst_warptv_finalize;
+
+  gst_element_class_set_details_simple (gstelement_class, "WarpTV effect",
+      "Filter/Effect/Video",
+      "WarpTV does realtime goo'ing of the video input",
+      "Sam Lantinga <slouken@devolution.com>");
+
+  gst_element_class_add_pad_template (gstelement_class,
+      gst_static_pad_template_get (&gst_warptv_sink_template));
+  gst_element_class_add_pad_template (gstelement_class,
+      gst_static_pad_template_get (&gst_warptv_src_template));
+
+  trans_class->start = GST_DEBUG_FUNCPTR (gst_warptv_start);
+  trans_class->set_caps = GST_DEBUG_FUNCPTR (gst_warptv_set_caps);
+  trans_class->decide_allocation =
+      GST_DEBUG_FUNCPTR (gst_wraptv_decide_allocation);
+  trans_class->transform = GST_DEBUG_FUNCPTR (gst_warptv_transform);
+
+  initSinTable ();
+}
+
+static void
+gst_warptv_init (GstWarpTV * warptv)
+{
+  /* nothing to do */
 }