gegl r2503 - in branches/branch2_zhangjb: . operations operations/frequency

[zhangjb svn gnome org]

Author: zhangjb
Date: Fri Jun 27 20:16:58 2008
New Revision: 2503
URL: http://svn.gnome.org/viewvc/gegl?rev=2503&view=rev

Log:
Moved to the new format successfully.
* operations/Makefile.am: added the path: frequency.
* operations/frequency/dft.c: the first step to move to the new format.
* operations/frequency/idft.c: the first step to move to the new format.


Added:
   branches/branch2_zhangjb/operations/frequency/idft.c
Modified:
   branches/branch2_zhangjb/ChangeLog
   branches/branch2_zhangjb/operations/Makefile.am
   branches/branch2_zhangjb/operations/frequency/dft.c

Modified: branches/branch2_zhangjb/operations/Makefile.am
==============================================================================
--- branches/branch2_zhangjb/operations/Makefile.am	(original)
+++ branches/branch2_zhangjb/operations/Makefile.am	Fri Jun 27 20:16:58 2008
@@ -3,6 +3,7 @@
 	core		\
 	common		\
 	external	\
+	frequency	\
 	generated
 
 if ENABLE_WORKSHOP

Modified: branches/branch2_zhangjb/operations/frequency/dft.c
==============================================================================
--- branches/branch2_zhangjb/operations/frequency/dft.c	(original)
+++ branches/branch2_zhangjb/operations/frequency/dft.c	Fri Jun 27 20:16:58 2008
@@ -82,8 +82,8 @@
 
   src_buf = g_new0(gdouble, 4*width*height);
   tmp_src_buf = g_new0(gdouble, width*height);
-  dst_buf = g_new0(gdouble, 8*width*FFT_HALF(width));
-  tmp_dst_buf = g_new0(gdouble, width*FFT_HALF(width));
+  dst_buf = g_new0(gdouble, 4*2*height*FFT_HALF(width));
+  tmp_dst_buf = g_new0(gdouble, 2*height*FFT_HALF(width));
 
   gegl_buffer_get(input, 1.0, NULL, babl_format ("RGBA double"), src_buf,
                   GEGL_AUTO_ROWSTRIDE);
@@ -92,7 +92,7 @@
     {
       get_rgba_component(src_buf, tmp_src_buf, i, width*height);
       dft(tmp_src_buf, (fftw_complex *)tmp_dst_buf, width, height);
-      set_freq_component(tmp_dst_buf, dst_buf, i, FFT_HALF(width)*height);
+      set_rgba_component(tmp_dst_buf, dst_buf, i, 2*FFT_HALF(width)*height);
     }
   
   gegl_buffer_set(output, NULL, babl_format ("frequency double"), dst_buf,

Added: branches/branch2_zhangjb/operations/frequency/idft.c
==============================================================================
--- (empty file)
+++ branches/branch2_zhangjb/operations/frequency/idft.c	Fri Jun 27 20:16:58 2008
@@ -0,0 +1,126 @@
+/* This file is a part of GEGL
+ *
+ * GEGL is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 3 of the License, or (at your option) any later version.
+ *
+ * GEGL 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
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with GEGL; if not, see <http://www.gnu.org/licenses/>.
+ *
+ * Copyright 2008 Zhang Junbo  <zhangjb svn gnome org>
+ */
+
+#ifdef GEGL_CHANT_PROPERTIES
+
+/* no properties */
+
+#else
+
+#define GEGL_CHANT_TYPE_FILTER
+#define GEGL_CHANT_C_FILE       "idft.c"
+
+#include "gegl-chant.h"
+#include "tools/fourier.c"
+#include "tools/component.c"
+#include <fftw3.h>
+
+static GeglRectangle
+get_bounding_box (GeglOperation *operation)
+{
+  GeglRectangle *in_rect = gegl_operation_source_get_bounding_box (operation, "input");
+  GeglRectangle  result  = *in_rect;
+
+  result.width = 2*(result.width-1);
+  return result;
+}
+
+static GeglRectangle
+get_cached_region(GeglOperation *operation,
+                  const GeglRectangle *roi)
+{
+  return get_bounding_box(operation);
+}
+
+static GeglRectangle
+get_required_for_output(GeglOperation *operation,
+                        const gchar *input_pad,
+                        const GeglRectangle *roi)
+{
+  GeglRectangle result = *gegl_operation_source_get_bounding_box(operation,
+                                                                 "input");
+  return result;
+}
+
+static void
+prepare(GeglOperation *operation)
+{
+  gegl_operation_set_format(operation, "input",
+                            babl_format("frequency double"));
+  gegl_operation_set_format(operation, "output", babl_format("RGBA double"));
+}
+
+static gboolean 
+process(GeglOperation *operation,
+                        GeglBuffer *input,
+                        GeglBuffer *output,
+                        const GeglRectangle *result)
+{
+  gint width = 2*(gegl_buffer_get_width(input)-1); /* width always refers to the image's (not buffer's) width. */
+  gint height = gegl_buffer_get_height(input);
+  gdouble *src_buf;
+  gdouble *dst_buf;
+  gdouble *tmp_src_buf;
+  gdouble *tmp_dst_buf;
+  gint i;
+
+  src_buf = g_new0(gdouble, 8*height*FFT_HALF(width));
+  dst_buf = g_new0(gdouble, 4*width*height);
+  tmp_src_buf = g_new0(gdouble, 2*height*FFT_HALF(width));
+  tmp_dst_buf = g_new0(gdouble, width*height);
+
+  gegl_buffer_get(input, 1.0, NULL, babl_format("frequency double"), (gdouble *)src_buf, GEGL_AUTO_ROWSTRIDE);
+  for (i=0; i<4; i++)
+    {
+      get_rgba_component(src_buf, tmp_src_buf, i, 2*height*FFT_HALF(width));
+      idft((fftw_complex *)tmp_src_buf, tmp_dst_buf, width, height);
+      set_rgba_component(tmp_dst_buf, dst_buf, i, width*height);
+    }
+
+  gegl_buffer_set(output, NULL, babl_format("RGBA double"), dst_buf,
+                  GEGL_AUTO_ROWSTRIDE);
+  
+  g_free(src_buf);
+  g_free(dst_buf);
+  g_free(tmp_src_buf);
+  g_free(tmp_dst_buf);
+  return TRUE;
+}
+
+static void
+gegl_chant_class_init(GeglChantClass *klass)
+{
+  GeglOperationClass *operation_class;
+  GeglOperationFilterClass *filter_class;
+
+  operation_class = GEGL_OPERATION_CLASS(klass);
+  filter_class = GEGL_OPERATION_FILTER_CLASS(klass);
+
+  filter_class->process = process;
+  operation_class->prepare = prepare;
+  operation_class->get_bounding_box = get_bounding_box;
+  operation_class->get_required_for_output= get_required_for_output;
+  operation_class->get_cached_region = get_cached_region;
+
+  operation_class->name = "idft";
+  operation_class->categories = "frequency";
+  operation_class->description
+    = "Perform 2-D inverse Discrete Fourier Transform for the image.\n";
+}
+
+#endif