Permalink
Browse files

Added deinterlace op

  • Loading branch information...
1 parent 69bf923 commit fd8104554b0fa02269615459eddd9f8e85a34aa1 @sasurobert committed Aug 17, 2011
Showing with 302 additions and 0 deletions.
  1. +302 −0 deinterlace.c
View
302 deinterlace.c
@@ -0,0 +1,302 @@
+/* This file is an image processing operation for 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 (C) 1997 Andrew Kieschnick (andrewk@mail.utexas.edu)
+ *
+ * Original deinterlace for GIMP 0.54 API by Federico Mena Quintero
+ *
+ * Copyright (C) 1996 Federico Mena Quintero
+ *
+ * Copyright (C) 2011 Robert Sasu <sasu.robert@gmail.com>
+ */
+
+#include "config.h"
+#include <glib/gi18n-lib.h>
+
+#ifdef GEGL_CHANT_PROPERTIES
+
+gegl_chant_boolean (even, _("Even/Odd"), TRUE, _("Keep even/odd fields"))
+gegl_chant_boolean (horizontal, _("Horizontal/Vertical"), TRUE,
+ _("Choose horizontal or vertical"))
+gegl_chant_int (size, _("Block size"), 0, 100, 1,
+ _("Block size of deinterlacing rows/columns"))
+
+#else
+
+#define GEGL_CHANT_TYPE_AREA_FILTER
+#define GEGL_CHANT_C_FILE "deinterlace.c"
+
+#include "gegl-chant.h"
+#include <stdio.h>
+#include <math.h>
+
+static void prepare (GeglOperation *operation)
+{
+ GeglOperationAreaFilter *op_area = GEGL_OPERATION_AREA_FILTER (operation);
+ GeglChantO *o = GEGL_CHANT_PROPERTIES (operation);
+
+
+ if (o->horizontal)
+ {
+ op_area->left = op_area->right = 0;
+ op_area->top = op_area->bottom = o->size + 1;
+ }
+ else
+ {
+ op_area->left = op_area->right = o->size + 1;
+ op_area->top = op_area->bottom = 0;
+ }
+ gegl_operation_set_format (operation, "input",
+ babl_format ("RGBA float"));
+ gegl_operation_set_format (operation, "output",
+ babl_format ("RGBA float"));
+}
+
+static void
+de_interlace_hor (gfloat *src_buf,
+ gfloat *dest,
+ const GeglRectangle *result,
+ const GeglRectangle *extended,
+ const GeglRectangle *boundary,
+ gint inter,
+ gint y,
+ gint size)
+{
+ gfloat upper[4], lower[4], temp_buf[4];
+ gint x, up_offset, low_offset, offset = 0, i;
+
+ for (x=0; x < result->width; x++)
+ {
+ gfloat ualpha, lalpha, temp;
+ gfloat alpha = 0;
+
+ temp_buf[0] = temp_buf[1] = temp_buf[2] = temp_buf[3] = 0;
+
+ for (i = 0; i < size; i++)
+ {
+ gint b;
+
+ if (y - i> 0)
+ up_offset = (y - i - extended->y) * extended->width * 4;
+ else
+ up_offset = inter * extended->width * 4;
+
+ if (y + i + 1 < boundary->height)
+ low_offset = (y + i + 1 - extended->y) * extended->width * 4;
+ else
+ low_offset = (y - 1 + inter - extended->y) * extended->width * 4;
+
+ offset = (y - result->y) * extended->width * 4;
+
+ for (b=0; b<4; b++)
+ {
+ upper[b] = src_buf[up_offset + x * 4 + b];
+ lower[b] = src_buf[low_offset + x * 4 + b];
+ }
+
+ ualpha = upper[3];
+ lalpha = lower[3];
+ temp = ualpha + lalpha;
+ alpha += temp;
+
+ for (b=0; b < 3; b++)
+ temp_buf[b] += (upper[b] * ualpha +
+ lower[b] * lalpha);
+ }
+
+ if ((dest[offset + x * 4 + 3] = alpha / (2 * size)))
+ {
+ gint b;
+ for (b=0; b < 3; b++)
+ dest[offset + x * 4 + b] = temp_buf[b] / alpha;
+
+ }
+ }
+}
+
+static void
+de_interlace_ver (gfloat *src_buf,
+ gfloat *dest,
+ const GeglRectangle *result,
+ const GeglRectangle *extended,
+ const GeglRectangle *boundary,
+ gint inter,
+ gint x,
+ gint size)
+{
+ gfloat upper[4], lower[4], temp_buf[4];
+ gint y, up_offset, low_offset, offset = 0, i;
+
+ for (y=result->y; y < result->y + result->height; y++)
+ {
+ gfloat ualpha, lalpha, temp;
+ gfloat alpha = 0;
+
+ temp_buf[0] = temp_buf[1] = temp_buf[2] = temp_buf[3] = 0;
+
+ for (i = 0; i < size; i++)
+ {
+ gint b;
+
+ if (x - i > 0)
+ up_offset = (y - extended->y) * extended->width * 4
+ + (x - i - extended->x) * 4;
+ else
+ up_offset = (y - extended->y) * extended->width * 4 + inter * 4;
+
+ if (x + i + 1 < boundary->width)
+ low_offset = (y - extended->y) * extended->width * 4 +
+ (x + i + 1 - extended->x) * 4;
+ else
+ low_offset = (y - extended->y) * extended->width * 4 +
+ (x + i - 1 + inter - extended->x) * 4;
+
+ offset = (y - result->y) * result->width * 4 + (x - result->x) * 4;
+
+ for (b=0; b<4; b++)
+ {
+ upper[b] = src_buf[up_offset + b];
+ lower[b] = src_buf[low_offset + b];
+ }
+
+ ualpha = upper[3];
+ lalpha = lower[3];
+ temp = ualpha + lalpha;
+ alpha += temp;
+
+ for (b=0; b < 3; b++)
+ temp_buf[b] += (upper[b] * ualpha +
+ lower[b] * lalpha);
+ }
+
+ if ((dest[offset + 3] = alpha / (2 * size)))
+ {
+ gint b;
+ for (b=0; b < 3; b++)
+ dest[offset + b] = temp_buf[b] / alpha;
+
+ }
+ }
+}
+
+
+static GeglRectangle
+get_effective_area (GeglOperation *operation)
+{
+ GeglRectangle result = {0,0,0,0};
+ GeglRectangle *in_rect = gegl_operation_source_get_bounding_box (operation, "input");
+
+ gegl_rectangle_copy(&result, in_rect);
+
+ return result;
+}
+
+static gboolean
+process (GeglOperation *operation,
+ GeglBuffer *input,
+ GeglBuffer *output,
+ const GeglRectangle *result)
+{
+ GeglChantO *o = GEGL_CHANT_PROPERTIES (operation);
+ GeglOperationAreaFilter *op_area = GEGL_OPERATION_AREA_FILTER (operation);
+ Babl *format = babl_format ("RGBA float");
+
+ GeglRectangle rect;
+ GeglRectangle boundary = get_effective_area (operation);
+ gint x, y;
+ gfloat *dst_buf, *src_buf;
+
+ rect.x = CLAMP (result->x - op_area->left, boundary.x, boundary.x +
+ boundary.width);
+ rect.width = CLAMP (result->width + op_area->left + op_area->right, 0,
+ boundary.width);
+ rect.y = CLAMP (result->y - op_area->top, boundary.y, boundary.y +
+ boundary.width);
+ rect.height = CLAMP (result->height + op_area->top + op_area->bottom, 0,
+ boundary.height);
+
+ dst_buf = g_new0 (gfloat, result->height * result->width * 4);
+ src_buf = g_new0 (gfloat, rect.height * rect.width * 4);
+
+ gegl_buffer_get (input, 1.0, result, format, dst_buf, GEGL_AUTO_ROWSTRIDE);
+ gegl_buffer_get (input, 1.0, &rect, format, src_buf, GEGL_AUTO_ROWSTRIDE);
+
+ if (o->horizontal)
+ {
+ for (y = result->y; y < result->y + result->height; y++)
+ if ((o->even && y%2==0) || (!o->even && y%2!=0))
+ de_interlace_hor (src_buf, dst_buf, result, &rect, &boundary,
+ o->even ? 0 : 1, y, o->size);
+ }
+ else
+ {
+ for (x = result->x; x < result->x + result->width; x++)
+ if ((o->even && x%2==0) || (!o->even && x%2!=0))
+ de_interlace_ver (src_buf, dst_buf, result, &rect, &boundary,
+ o->even ? 0 : 1, x, o->size);
+
+ }
+
+
+ gegl_buffer_set (output, result, format, dst_buf, GEGL_AUTO_ROWSTRIDE);
+
+ g_free (src_buf);
+ g_free (dst_buf);
+
+ return TRUE;
+}
+
+static GeglRectangle
+get_bounding_box (GeglOperation *operation)
+{
+ GeglRectangle result = {0,0,0,0};
+ GeglRectangle *in_rect;
+
+ in_rect = gegl_operation_source_get_bounding_box (operation, "input");
+ if (!in_rect)
+ return result;
+
+ return *in_rect;
+}
+
+static GeglRectangle
+get_required_for_output (GeglOperation *operation,
+ const gchar *input_pad,
+ const GeglRectangle *roi)
+{
+ return get_bounding_box (operation);
+}
+
+
+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->categories = "enhance";
+ operation_class->name = "gegl:deinterlace";
+ operation_class->description = _("Performs deinterlace on the image.");
+}
+
+#endif

0 comments on commit fd81045

Please sign in to comment.