From 017808ef8d1e3abaa5a6a05eb7411f277569504c Mon Sep 17 00:00:00 2001 From: Thiago Santos Date: Sun, 6 Jun 2010 22:34:55 -0300 Subject: [PATCH] marble: Adds 'marble' element to geometrictransform Adds a new element that does a marbling effect to geometrictransform plugin --- gst/geometrictransform/Makefile.am | 2 + gst/geometrictransform/geometricmath.c | 116 +++++++++++ gst/geometrictransform/geometricmath.h | 6 + gst/geometrictransform/gstmarble.c | 266 +++++++++++++++++++++++++ gst/geometrictransform/gstmarble.h | 95 +++++++++ gst/geometrictransform/plugin.c | 4 + 6 files changed, 489 insertions(+) create mode 100644 gst/geometrictransform/gstmarble.c create mode 100644 gst/geometrictransform/gstmarble.h diff --git a/gst/geometrictransform/Makefile.am b/gst/geometrictransform/Makefile.am index 0c317429c5..b88932a4c7 100644 --- a/gst/geometrictransform/Makefile.am +++ b/gst/geometrictransform/Makefile.am @@ -7,6 +7,7 @@ libgstgeometrictransform_la_SOURCES = plugin.c \ gstcircle.c \ gstdiffuse.c \ gstkaleidoscope.c \ + gstmarble.c \ gstpinch.c \ gstsphere.c \ gsttwirl.c \ @@ -24,6 +25,7 @@ noinst_HEADERS = gstgeometrictransform.h \ gstcircle.h \ gstdiffuse.h \ gstkaleidoscope.h \ + gstmarble.h \ gstpinch.h \ gstsphere.h \ gsttwirl.h \ diff --git a/gst/geometrictransform/geometricmath.c b/gst/geometrictransform/geometricmath.c index 6679438cc9..dbdfe97bf1 100644 --- a/gst/geometrictransform/geometricmath.c +++ b/gst/geometrictransform/geometricmath.c @@ -48,6 +48,122 @@ */ #include "geometricmath.h" +#include + +#define N 0x1000 +#define B 0x100 +#define BM 0xff + +struct _Noise +{ + gdouble p[2 * B + 2]; + gdouble g2[2 * B + 2][2]; +}; + +static void +normalize_2 (gdouble * v) +{ + gdouble s = sqrt (v[0] * v[0] + v[1] * v[1]); + + v[0] = v[0] / s; + v[1] = v[1] / s; +} + +Noise * +noise_new (void) +{ + Noise *noise = g_new0 (Noise, 1); + gint i, j, k; + + for (i = 0; i < B; i++) { + noise->p[i] = i; + for (j = 0; j < 2; j++) { + noise->g2[i][j] = ((g_random_int () % (2 * B)) + - B) / (gdouble) B; + } + normalize_2 (noise->g2[i]); + } + + for (i = B - 1; i >= 0; i--) { + k = noise->p[i]; + j = g_random_int () % B; + noise->p[i] = noise->p[j]; + noise->p[j] = k; + } + + for (i = 0; i < B + 2; i++) { + noise->p[B + i] = noise->p[i]; + for (j = 0; j < 2; j++) { + noise->g2[B + i][j] = noise->g2[i][j]; + } + } + + return noise; +} + +void +noise_free (Noise * noise) +{ + g_free (noise); +} + +static gdouble +s_curve (gdouble x) +{ + return x * x * (3.0 - 2.0 * x); +} + +static gdouble +lerp (gdouble t, gdouble a, gdouble b) +{ + return a + t * (b - a); +} + +gdouble +noise_2 (Noise * noise, gdouble x, gdouble y) +{ + gint bx0, bx1, by0, by1, b00, b10, b01, b11; + gdouble rx0, rx1, ry0, ry1, sx, sy, a, b, t, u, v; + gdouble *q; + gint i, j; + + t = x + N; + bx0 = ((gint) t) & BM; + bx1 = (bx0 + 1) & BM; + rx0 = t - (gint) t; + rx1 = rx0 - 1.0; + + t = y + N; + by0 = ((gint) t) & BM; + by1 = (by0 + 1) & BM; + ry0 = t - (gint) t; + ry1 = ry0 - 1.0; + + i = noise->p[bx0]; + j = noise->p[bx1]; + + b00 = noise->p[i + by0]; + b10 = noise->p[j + by0]; + b01 = noise->p[i + by1]; + b11 = noise->p[j + by1]; + + sx = s_curve (rx0); + sy = s_curve (ry0); + + q = noise->g2[b00]; + u = rx0 * q[0] + ry0 * q[1]; + q = noise->g2[b10]; + v = rx1 * q[0] + ry0 * q[1]; + a = lerp (sx, u, v); + + q = noise->g2[b01]; + u = rx0 * q[0] + ry1 * q[1]; + q = noise->g2[b11]; + v = rx1 * q[0] + ry1 * q[1]; + b = lerp (sx, u, v); + + return 1.5 * lerp (sy, a, b); +} /* * This differs from the % operator with respect to negative numbers diff --git a/gst/geometrictransform/geometricmath.h b/gst/geometrictransform/geometricmath.h index 7b8fa8614b..dfd58009be 100644 --- a/gst/geometrictransform/geometricmath.h +++ b/gst/geometrictransform/geometricmath.h @@ -54,6 +54,12 @@ G_BEGIN_DECLS +typedef struct _Noise Noise; + +Noise * noise_new (void); +void noise_free (Noise * noise); +gdouble noise_2 (Noise * noise, gdouble x, gdouble y); + gdouble mod_float (gdouble a, gdouble b); gdouble geometric_math_triangle (gdouble x); diff --git a/gst/geometrictransform/gstmarble.c b/gst/geometrictransform/gstmarble.c new file mode 100644 index 0000000000..fd6e31ca27 --- /dev/null +++ b/gst/geometrictransform/gstmarble.c @@ -0,0 +1,266 @@ +/* + * GStreamer + * Copyright (C) 2010 Thiago Santos + * + * Permission is hereby granted, free of charge, to any person obtaining a + * copy of this software and associated documentation files (the "Software"), + * to deal in the Software without restriction, including without limitation + * the rights to use, copy, modify, merge, publish, distribute, sublicense, + * and/or sell copies of the Software, and to permit persons to whom the + * Software is furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in + * all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING + * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER + * DEALINGS IN THE SOFTWARE. + * + * Alternatively, the contents of this file may be used under the + * GNU Lesser General Public License Version 2.1 (the "LGPL"), in + * which case the following provisions apply instead of the ones + * mentioned above: + * + * 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. + */ + +/* + * Thanks to Jerry Huxtable work on its java + * image editor and filters. The algorithms here were extracted from + * his code. + */ + + +#ifdef HAVE_CONFIG_H +# include +#endif + +#include +#include + +#include "gstmarble.h" + +GST_DEBUG_CATEGORY_STATIC (gst_marble_debug); +#define GST_CAT_DEFAULT gst_marble_debug + +enum +{ + PROP_0, + PROP_XSCALE, + PROP_YSCALE, + PROP_AMOUNT, + PROP_TURBULENCE +}; + +#define DEFAULT_XSCALE 4 +#define DEFAULT_YSCALE 4 +#define DEFAULT_AMOUNT 1 +#define DEFAULT_TURBULENCE 1 + +GST_BOILERPLATE (GstMarble, gst_marble, GstGeometricTransform, + GST_TYPE_GEOMETRIC_TRANSFORM); + +static void +gst_marble_set_property (GObject * object, guint prop_id, const GValue * value, + GParamSpec * pspec) +{ + GstMarble *marble; + + marble = GST_MARBLE_CAST (object); + + switch (prop_id) { + case PROP_XSCALE: + marble->xscale = g_value_get_double (value); + break; + case PROP_YSCALE: + marble->yscale = g_value_get_double (value); + break; + case PROP_AMOUNT: + marble->amount = g_value_get_double (value); + break; + case PROP_TURBULENCE: + marble->turbulence = g_value_get_double (value); + break; + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +static void +gst_marble_get_property (GObject * object, guint prop_id, + GValue * value, GParamSpec * pspec) +{ + GstMarble *marble; + + marble = GST_MARBLE_CAST (object); + + switch (prop_id) { + case PROP_XSCALE: + g_value_set_double (value, marble->xscale); + break; + case PROP_YSCALE: + g_value_set_double (value, marble->yscale); + break; + case PROP_AMOUNT: + g_value_set_double (value, marble->amount); + break; + case PROP_TURBULENCE: + g_value_set_double (value, marble->turbulence); + break; + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +/* Clean up */ +static void +gst_marble_finalize (GObject * obj) +{ + GstMarble *marble = GST_MARBLE_CAST (obj); + + noise_free (marble->noise); + g_free (marble->sin_table); + g_free (marble->cos_table); + + G_OBJECT_CLASS (parent_class)->finalize (obj); +} + +/* GObject vmethod implementations */ + +static void +gst_marble_base_init (gpointer gclass) +{ + GstElementClass *element_class = GST_ELEMENT_CLASS (gclass); + + gst_element_class_set_details_simple (element_class, + "marble", + "Transform/Effect/Video", + "Applies a marbling effect to the image", + "Thiago Santos"); +} + +static gboolean +marble_prepare (GstGeometricTransform * trans) +{ + GstMarble *marble = GST_MARBLE_CAST (trans); + gint i; + + if (!marble->noise) { + marble->noise = noise_new (); + } + + g_free (marble->sin_table); + g_free (marble->cos_table); + + marble->sin_table = g_malloc0 (sizeof (gdouble) * 256); + marble->cos_table = g_malloc0 (sizeof (gdouble) * 256); + + for (i = 0; i < 256; i++) { + gdouble angle = (G_PI * 2 * i) / 256.0 * marble->turbulence; + + marble->sin_table[i] = -marble->yscale * sin (angle); + marble->cos_table[i] = marble->yscale * cos (angle); + } + return TRUE; +} + +static gboolean +marble_map (GstGeometricTransform * gt, gint x, gint y, gdouble * in_x, + gdouble * in_y) +{ + GstMarble *marble = GST_MARBLE_CAST (gt); + gint displacement; + + displacement = 127 * (1 + noise_2 (marble->noise, x / marble->xscale, + y / marble->xscale)); + displacement = CLAMP (displacement, 0, 255); + + *in_x = x + marble->sin_table[displacement]; + *in_y = y + marble->cos_table[displacement]; + + GST_DEBUG_OBJECT (marble, "Inversely mapped %d %d into %lf %lf", + x, y, *in_x, *in_y); + + return TRUE; +} + +static void +gst_marble_class_init (GstMarbleClass * klass) +{ + GObjectClass *gobject_class; + GstGeometricTransformClass *gstgt_class; + + gobject_class = (GObjectClass *) klass; + gstgt_class = (GstGeometricTransformClass *) klass; + + parent_class = g_type_class_peek_parent (klass); + + gobject_class->finalize = GST_DEBUG_FUNCPTR (gst_marble_finalize); + gobject_class->set_property = GST_DEBUG_FUNCPTR (gst_marble_set_property); + gobject_class->get_property = GST_DEBUG_FUNCPTR (gst_marble_get_property); + + g_object_class_install_property (gobject_class, PROP_XSCALE, + g_param_spec_double ("x-scale", "x-scale", + "X scale of the texture", + 0, G_MAXDOUBLE, DEFAULT_XSCALE, + G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); + g_object_class_install_property (gobject_class, PROP_YSCALE, + g_param_spec_double ("y-scale", "y-scale", + "Y scale of the texture", + 0, G_MAXDOUBLE, DEFAULT_YSCALE, + G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); + g_object_class_install_property (gobject_class, PROP_AMOUNT, + g_param_spec_double ("amount", "amount", + "Amount of effect", + 0.0, 1.0, DEFAULT_AMOUNT, + G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); + g_object_class_install_property (gobject_class, PROP_YSCALE, + g_param_spec_double ("turbulence", "turbulence", + "Turbulence of the effect", + 0.0, 1.0, DEFAULT_TURBULENCE, + G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); + + gstgt_class->prepare_func = marble_prepare; + gstgt_class->map_func = marble_map; +} + +static void +gst_marble_init (GstMarble * filter, GstMarbleClass * gclass) +{ + GstGeometricTransform *gt = GST_GEOMETRIC_TRANSFORM (filter); + + gt->precalc_map = TRUE; + gt->off_edge_pixels = GST_GT_OFF_EDGES_PIXELS_CLAMP; + filter->xscale = DEFAULT_XSCALE; + filter->yscale = DEFAULT_YSCALE; + filter->amount = DEFAULT_AMOUNT; + filter->turbulence = DEFAULT_TURBULENCE; +} + +gboolean +gst_marble_plugin_init (GstPlugin * plugin) +{ + GST_DEBUG_CATEGORY_INIT (gst_marble_debug, "marble", 0, "marble"); + + return gst_element_register (plugin, "marble", GST_RANK_NONE, + GST_TYPE_MARBLE); +} diff --git a/gst/geometrictransform/gstmarble.h b/gst/geometrictransform/gstmarble.h new file mode 100644 index 0000000000..6aa525ba21 --- /dev/null +++ b/gst/geometrictransform/gstmarble.h @@ -0,0 +1,95 @@ +/* + * GStreamer + * Copyright (C) 2010 Thiago Santos + * + * Permission is hereby granted, free of charge, to any person obtaining a + * copy of this software and associated documentation files (the "Software"), + * to deal in the Software without restriction, including without limitation + * the rights to use, copy, modify, merge, publish, distribute, sublicense, + * and/or sell copies of the Software, and to permit persons to whom the + * Software is furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in + * all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING + * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER + * DEALINGS IN THE SOFTWARE. + * + * Alternatively, the contents of this file may be used under the + * GNU Lesser General Public License Version 2.1 (the "LGPL"), in + * which case the following provisions apply instead of the ones + * mentioned above: + * + * 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_MARBLE_H__ +#define __GST_MARBLE_H__ + +#include +#include "gstgeometrictransform.h" +#include "geometricmath.h" + +G_BEGIN_DECLS + +/* #defines don't like whitespacey bits */ +#define GST_TYPE_MARBLE \ + (gst_marble_get_type()) +#define GST_MARBLE(obj) \ + (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_MARBLE,GstMarble)) +#define GST_MARBLE_CAST(obj) \ + ((GstMarble *)(obj)) +#define GST_MARBLE_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_MARBLE,GstMarbleClass)) +#define GST_IS_MARBLE(obj) \ + (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_MARBLE)) +#define GST_IS_MARBLE_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_MARBLE)) + +typedef struct _GstMarble GstMarble; +typedef struct _GstMarbleClass GstMarbleClass; + +struct _GstMarble +{ + GstGeometricTransform element; + + gdouble xscale; + gdouble yscale; + gdouble turbulence; + gdouble amount; + + Noise *noise; + gdouble *sin_table; + gdouble *cos_table; +}; + +struct _GstMarbleClass +{ + GstGeometricTransformClass parent_class; +}; + +GType gst_marble_get_type (void); + +gboolean gst_marble_plugin_init (GstPlugin * plugin); + +G_END_DECLS + +#endif /* __GST_MARBLE_H__ */ diff --git a/gst/geometrictransform/plugin.c b/gst/geometrictransform/plugin.c index d0a6ca7f62..2f655dc585 100644 --- a/gst/geometrictransform/plugin.c +++ b/gst/geometrictransform/plugin.c @@ -24,6 +24,7 @@ #include "gstcircle.h" #include "gstdiffuse.h" #include "gstkaleidoscope.h" +#include "gstmarble.h" #include "gstpinch.h" #include "gstsphere.h" #include "gsttwirl.h" @@ -41,6 +42,9 @@ plugin_init (GstPlugin * plugin) if (!gst_kaleidoscope_plugin_init (plugin)) return FALSE; + if (!gst_marble_plugin_init (plugin)) + return FALSE; + if (!gst_pinch_plugin_init (plugin)) return FALSE;