gst/effectv/gstshagadelic.c

   1 /* GStreamer
   2  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
   3  *
   4  * EffecTV:
   5  * Copyright (C) 2001 FUKUCHI Kentarou
   6  *
   7  * Inspired by Adrian Likin's script for the GIMP.
   8  * EffecTV is free software.  This library is free software;
   9  * you can redistribute it and/or
  10  *  modify it under the terms of the GNU Library General Public
  11  * License as published by the Free Software Foundation; either
  12  * version 2 of the License, or (at your option) any later version.
  13  *
  14  * This library is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  17  * Library General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Library General Public
  20  * License along with this library; if not, write to the
  21  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  22  * Boston, MA 02111-1307, USA.
  23  */
  24
  25 #ifdef HAVE_CONFIG_H
  26 #include "config.h"
  27 #endif
  28
  29 #include <gstvideofilter.h>
  30
  31 #include <math.h>
  32 #include <string.h>
  33
  34 #include <gst/video/video.h>
  35
  36 #define GST_TYPE_SHAGADELICTV \
  37   (gst_shagadelictv_get_type())
  38 #define GST_SHAGADELICTV(obj) \
  39   (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_SHAGADELICTV,GstShagadelicTV))
  40 #define GST_SHAGADELICTV_CLASS(klass) \
  41   (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_SHAGADELICTV,GstShagadelicTVClass))
  42 #define GST_IS_SHAGADELICTV(obj) \
  43   (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_SHAGADELICTV))
  44 #define GST_IS_SHAGADELICTV_CLASS(obj) \
  45   (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_SHAGADELICTV))
  46
  47 typedef struct _GstShagadelicTV GstShagadelicTV;
  48 typedef struct _GstShagadelicTVClass GstShagadelicTVClass;
  49
  50 struct _GstShagadelicTV
  51 {
  52   GstVideofilter videofilter;
  53
  54   gint width, height;
  55   gint stat;
  56   gchar *ripple;
  57   gchar *spiral;
  58   guchar phase;
  59   gint rx, ry;
  60   gint bx, by;
  61   gint rvx, rvy;
  62   gint bvx, bvy;
  63 };
  64
  65 struct _GstShagadelicTVClass
  66 {
  67   GstVideofilterClass parent_class;
  68 };
  69
  70 GType gst_shagadelictv_get_type (void);
  71
  72 static void gst_shagadelic_initialize (GstShagadelicTV * filter);
  73
  74 static GstElementDetails shagadelictv_details =
  75 GST_ELEMENT_DETAILS ("ShagadelicTV",
  76     "Filter/Effect/Video",
  77     "Oh behave, ShagedelicTV makes images shagadelic!",
  78     "Wim Taymans <wim.taymans@chello.be>");
  79
  80 static GstStaticPadTemplate gst_shagadelictv_src_template =
  81 GST_STATIC_PAD_TEMPLATE ("src",
  82     GST_PAD_SRC,
  83     GST_PAD_ALWAYS,
  84     GST_STATIC_CAPS (GST_VIDEO_CAPS_BGRx)
  85     );
  86
  87 static GstStaticPadTemplate gst_shagadelictv_sink_template =
  88 GST_STATIC_PAD_TEMPLATE ("sink",
  89     GST_PAD_SINK,
  90     GST_PAD_ALWAYS,
  91     GST_STATIC_CAPS (GST_VIDEO_CAPS_BGRx)
  92     );
  93
  94 static GstVideofilterClass *parent_class = NULL;
  95
  96 static gboolean
  97 gst_shagadelictv_set_caps (GstBaseTransform * btrans, GstCaps * incaps,
  98     GstCaps * outcaps)
  99 {
 100   GstShagadelicTV *filter = GST_SHAGADELICTV (btrans);
 101   GstStructure *structure;
 102   gboolean ret = FALSE;
 103
 104   structure = gst_caps_get_structure (incaps, 0);
 105
 106   if (gst_structure_get_int (structure, "width", &filter->width) &&
 107       gst_structure_get_int (structure, "height", &filter->height)) {
 108     gint area = filter->width * filter->height;
 109
 110     g_free (filter->ripple);
 111     g_free (filter->spiral);
 112
 113     filter->ripple = (gchar *) g_malloc (area * 4);
 114     filter->spiral = (gchar *) g_malloc (area);
 115
 116     gst_shagadelic_initialize (filter);
 117     ret = TRUE;
 118   }
 119
 120   return ret;
 121 }
 122
 123 static gboolean
 124 gst_shagadelictv_get_unit_size (GstBaseTransform * btrans, GstCaps * caps,
 125     guint * size)
 126 {
 127   GstShagadelicTV *filter;
 128   GstStructure *structure;
 129   gboolean ret = FALSE;
 130   gint width, height;
 131
 132   filter = GST_SHAGADELICTV (btrans);
 133
 134   structure = gst_caps_get_structure (caps, 0);
 135
 136   if (gst_structure_get_int (structure, "width", &width) &&
 137       gst_structure_get_int (structure, "height", &height)) {
 138     *size = width * height * 32 / 8;
 139     ret = TRUE;
 140     GST_DEBUG_OBJECT (filter, "our frame size is %d bytes (%dx%d)", *size,
 141         width, height);
 142   }
 143
 144   return ret;
 145 }
 146
 147 static unsigned int
 148 fastrand (void)
 149 {
 150   static unsigned int fastrand_val;
 151
 152   return (fastrand_val = fastrand_val * 1103515245 + 12345);
 153 }
 154
 155 static void
 156 gst_shagadelic_initialize (GstShagadelicTV * filter)
 157 {
 158   int i, x, y;
 159
 160 #ifdef PS2
 161   float xx, yy;
 162 #else
 163   double xx, yy;
 164 #endif
 165
 166   i = 0;
 167   for (y = 0; y < filter->height * 2; y++) {
 168     yy = y - filter->height;
 169     yy *= yy;
 170
 171     for (x = 0; x < filter->width * 2; x++) {
 172       xx = x - filter->width;
 173 #ifdef PS2
 174       filter->ripple[i++] = ((unsigned int) (sqrtf (xx * xx + yy) * 8)) & 255;
 175 #else
 176       filter->ripple[i++] = ((unsigned int) (sqrt (xx * xx + yy) * 8)) & 255;
 177 #endif
 178     }
 179   }
 180
 181   i = 0;
 182   for (y = 0; y < filter->height; y++) {
 183     yy = y - filter->height / 2;
 184
 185     for (x = 0; x < filter->width; x++) {
 186       xx = x - filter->width / 2;
 187 #ifdef PS2
 188       filter->spiral[i++] = ((unsigned int)
 189           ((atan2f (xx,
 190                       yy) / ((float) M_PI) * 256 * 9) + (sqrtf (xx * xx +
 191                       yy * yy) * 5))) & 255;
 192 #else
 193       filter->spiral[i++] = ((unsigned int)
 194           ((atan2 (xx, yy) / M_PI * 256 * 9) + (sqrt (xx * xx +
 195                       yy * yy) * 5))) & 255;
 196 #endif
 197 /* Here is another Swinger!
 198  * ((atan2(xx, yy)/M_PI*256) + (sqrt(xx*xx+yy*yy)*10))&255;
 199  */
 200     }
 201   }
 202   filter->rx = fastrand () % filter->width;
 203   filter->ry = fastrand () % filter->height;
 204   filter->bx = fastrand () % filter->width;
 205   filter->by = fastrand () % filter->height;
 206   filter->rvx = -2;
 207   filter->rvy = -2;
 208   filter->bvx = 2;
 209   filter->bvy = 2;
 210   filter->phase = 0;
 211 }
 212
 213 static GstFlowReturn
 214 gst_shagadelictv_transform (GstBaseTransform * trans, GstBuffer * in,
 215     GstBuffer * out)
 216 {
 217   GstShagadelicTV *filter;
 218   guint32 *src, *dest;
 219   gint x, y;
 220   guint32 v;
 221   guchar r, g, b;
 222   gint width, height;
 223   GstFlowReturn ret = GST_FLOW_OK;
 224
 225   filter = GST_SHAGADELICTV (trans);
 226
 227   gst_buffer_stamp (out, in);
 228
 229   src = (guint32 *) GST_BUFFER_DATA (in);
 230   dest = (guint32 *) GST_BUFFER_DATA (out);
 231
 232   width = filter->width;
 233   height = filter->height;
 234
 235   for (y = 0; y < height; y++) {
 236     for (x = 0; x < width; x++) {
 237       v = *src++ | 0x1010100;
 238       v = (v - 0x707060) & 0x1010100;
 239       v -= v >> 8;
 240 /* Try another Babe!
 241  * v = *src++;
 242  * *dest++ = v & ((r<<16)|(g<<8)|b);
 243  */
 244       r = (gchar) (filter->ripple[(filter->ry + y) * width * 2 + filter->rx +
 245               x] + filter->phase * 2) >> 7;
 246       g = (gchar) (filter->spiral[y * width + x] + filter->phase * 3) >> 7;
 247       b = (gchar) (filter->ripple[(filter->by + y) * width * 2 + filter->bx +
 248               x] - filter->phase) >> 7;
 249       *dest++ = v & ((r << 16) | (g << 8) | b);
 250     }
 251   }
 252
 253   filter->phase -= 8;
 254   if ((filter->rx + filter->rvx) < 0 || (filter->rx + filter->rvx) >= width)
 255     filter->rvx = -filter->rvx;
 256   if ((filter->ry + filter->rvy) < 0 || (filter->ry + filter->rvy) >= height)
 257     filter->rvy = -filter->rvy;
 258   if ((filter->bx + filter->bvx) < 0 || (filter->bx + filter->bvx) >= width)
 259     filter->bvx = -filter->bvx;
 260   if ((filter->by + filter->bvy) < 0 || (filter->by + filter->bvy) >= height)
 261     filter->bvy = -filter->bvy;
 262   filter->rx += filter->rvx;
 263   filter->ry += filter->rvy;
 264   filter->bx += filter->bvx;
 265   filter->by += filter->bvy;
 266
 267   return ret;
 268 }
 269
 270 static void
 271 gst_shagadelictv_base_init (gpointer g_class)
 272 {
 273   GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
 274
 275   gst_element_class_set_details (element_class, &shagadelictv_details);
 276
 277   gst_element_class_add_pad_template (element_class,
 278       gst_static_pad_template_get (&gst_shagadelictv_sink_template));
 279   gst_element_class_add_pad_template (element_class,
 280       gst_static_pad_template_get (&gst_shagadelictv_src_template));
 281 }
 282
 283 static void
 284 gst_shagadelictv_class_init (gpointer klass, gpointer class_data)
 285 {
 286   GObjectClass *gobject_class;
 287   GstElementClass *element_class;
 288   GstBaseTransformClass *trans_class;
 289
 290   gobject_class = (GObjectClass *) klass;
 291   element_class = (GstElementClass *) klass;
 292   trans_class = (GstBaseTransformClass *) klass;
 293
 294   parent_class = g_type_class_peek_parent (klass);
 295
 296   trans_class->set_caps = GST_DEBUG_FUNCPTR (gst_shagadelictv_set_caps);
 297   trans_class->get_unit_size =
 298       GST_DEBUG_FUNCPTR (gst_shagadelictv_get_unit_size);
 299   trans_class->transform = GST_DEBUG_FUNCPTR (gst_shagadelictv_transform);
 300 }
 301
 302 static void
 303 gst_shagadelictv_init (GTypeInstance * instance, gpointer g_class)
 304 {
 305   GstShagadelicTV *filter = GST_SHAGADELICTV (instance);
 306
 307   filter->ripple = NULL;
 308   filter->spiral = NULL;
 309 }
 310
 311 GType
 312 gst_shagadelictv_get_type (void)
 313 {
 314   static GType shagadelictv_type = 0;
 315
 316   if (!shagadelictv_type) {
 317     static const GTypeInfo shagadelictv_info = {
 318       sizeof (GstShagadelicTVClass),
 319       gst_shagadelictv_base_init,
 320       NULL,
 321       (GClassInitFunc) gst_shagadelictv_class_init,
 322       NULL,
 323       NULL,
 324       sizeof (GstShagadelicTV),
 325       0,
 326       (GInstanceInitFunc) gst_shagadelictv_init,
 327     };
 328
 329     shagadelictv_type =
 330         g_type_register_static (GST_TYPE_VIDEOFILTER, "GstShagadelicTV",
 331         &shagadelictv_info, 0);
 332   }
 333   return shagadelictv_type;
 334 }