gimp/plug-ins/common/destripe.c

504 lines
13 KiB
C

/* GIMP - The GNU Image Manipulation Program
* Copyright (C) 1995 Spencer Kimball and Peter Mattis
*
* Destripe filter
*
* Copyright 1997 Marc Lehmann, heavily modified from a filter by
* Michael Sweet.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* This program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
#include "config.h"
#include <string.h>
#include <libgimp/gimp.h>
#include <libgimp/gimpui.h>
#include "libgimp/stdplugins-intl.h"
/*
* Constants...
*/
#define PLUG_IN_PROC "plug-in-destripe"
#define PLUG_IN_BINARY "destripe"
#define PLUG_IN_VERSION "0.2"
#define SCALE_WIDTH 140
#define MAX_AVG 100
/*
* Local functions...
*/
static void query (void);
static void run (const gchar *name,
gint nparams,
const GimpParam *param,
gint *nreturn_vals,
GimpParam **return_vals);
static void destripe (GimpDrawable *drawable,
GimpPreview *preview);
static gboolean destripe_dialog (GimpDrawable *drawable);
/*
* Globals...
*/
const GimpPlugInInfo PLUG_IN_INFO =
{
NULL, /* init_proc */
NULL, /* quit_proc */
query, /* query_proc */
run /* run_proc */
};
typedef struct
{
gboolean histogram;
gint avg_width;
gboolean preview;
} DestripeValues;
static DestripeValues vals =
{
FALSE, /* histogram */
36, /* average width */
TRUE /* preview */
};
MAIN ()
static void
query (void)
{
static const GimpParamDef args[] =
{
{ GIMP_PDB_INT32, "run-mode", "The run mode { RUN-INTERACTIVE (0), RUN-NONINTERACTIVE (1) }" },
{ GIMP_PDB_IMAGE, "image", "Input image" },
{ GIMP_PDB_DRAWABLE, "drawable", "Input drawable" },
{ GIMP_PDB_INT32, "avg-width", "Averaging filter width (default = 36)" }
};
gimp_install_procedure (PLUG_IN_PROC,
N_("Remove vertical stripe artifacts from the image"),
"This plug-in tries to remove vertical stripes from "
"an image.",
"Marc Lehmann <pcg@goof.com>",
"Marc Lehmann <pcg@goof.com>",
PLUG_IN_VERSION,
N_("Des_tripe..."),
"RGB*, GRAY*",
GIMP_PLUGIN,
G_N_ELEMENTS (args), 0,
args, NULL);
gimp_plugin_menu_register (PLUG_IN_PROC, "<Image>/Filters/Enhance");
}
static void
run (const gchar *name,
gint nparams,
const GimpParam *param,
gint *nreturn_vals,
GimpParam **return_vals)
{
GimpRunMode run_mode; /* Current run mode */
GimpPDBStatusType status; /* Return status */
static GimpParam values[1]; /* Return values */
GimpDrawable *drawable;
INIT_I18N ();
/*
* Initialize parameter data...
*/
status = GIMP_PDB_SUCCESS;
run_mode = param[0].data.d_int32;
values[0].type = GIMP_PDB_STATUS;
values[0].data.d_status = status;
*nreturn_vals = 1;
*return_vals = values;
/*
* Get drawable information...
*/
drawable = gimp_drawable_get (param[2].data.d_drawable);
/*
* See how we will run
*/
switch (run_mode)
{
case GIMP_RUN_INTERACTIVE:
/*
* Possibly retrieve data...
*/
gimp_get_data (PLUG_IN_PROC, &vals);
/*
* Get information from the dialog...
*/
if (!destripe_dialog (drawable))
return;
break;
case GIMP_RUN_NONINTERACTIVE:
/*
* Make sure all the arguments are present...
*/
if (nparams != 4)
status = GIMP_PDB_CALLING_ERROR;
else
vals.avg_width = param[3].data.d_int32;
break;
case GIMP_RUN_WITH_LAST_VALS :
/*
* Possibly retrieve data...
*/
gimp_get_data (PLUG_IN_PROC, &vals);
break;
default :
status = GIMP_PDB_CALLING_ERROR;
break;
};
/*
* Destripe the image...
*/
if (status == GIMP_PDB_SUCCESS)
{
if ((gimp_drawable_is_rgb (drawable->drawable_id) ||
gimp_drawable_is_gray (drawable->drawable_id)))
{
/*
* Set the tile cache size...
*/
gimp_tile_cache_ntiles ((drawable->width + gimp_tile_width () - 1) /
gimp_tile_width ());
/*
* Run!
*/
destripe (drawable, NULL);
/*
* If run mode is interactive, flush displays...
*/
if (run_mode != GIMP_RUN_NONINTERACTIVE)
gimp_displays_flush ();
/*
* Store data...
*/
if (run_mode == GIMP_RUN_INTERACTIVE)
gimp_set_data (PLUG_IN_PROC, &vals, sizeof (vals));
}
else
status = GIMP_PDB_EXECUTION_ERROR;
};
/*
* Reset the current run status...
*/
values[0].data.d_status = status;
/*
* Detach from the drawable...
*/
gimp_drawable_detach (drawable);
}
static void
destripe (GimpDrawable *drawable,
GimpPreview *preview)
{
GimpPixelRgn src_rgn; /* source image region */
GimpPixelRgn dst_rgn; /* destination image region */
guchar *src_rows; /* image data */
gdouble progress, progress_inc;
gint x1, x2, y1, y2;
gint width, height;
gint bpp;
glong *hist, *corr; /* "histogram" data */
gint tile_width = gimp_tile_width ();
gint i, x, y, ox, cols;
/* initialize */
progress = 0.0;
progress_inc = 0.0;
/*
* Let the user know what we're doing...
*/
bpp = gimp_drawable_bpp (drawable->drawable_id);
if (preview)
{
gimp_preview_get_position (preview, &x1, &y1);
gimp_preview_get_size (preview, &width, &height);
x2 = x1 + width;
y2 = y1 + height;
}
else
{
gimp_progress_init (_("Destriping"));
gimp_drawable_mask_bounds (drawable->drawable_id, &x1, &y1, &x2, &y2);
width = x2 - x1;
height = y2 - y1;
progress = 0;
progress_inc = 0.5 * tile_width / width;
}
/*
* Setup for filter...
*/
gimp_pixel_rgn_init (&src_rgn, drawable,
x1, y1, width, height, FALSE, FALSE);
gimp_pixel_rgn_init (&dst_rgn, drawable,
x1, y1, width, height, (preview == NULL), TRUE);
hist = g_new (long, width * bpp);
corr = g_new (long, width * bpp);
src_rows = g_new (guchar, tile_width * height * bpp);
memset (hist, 0, width * bpp * sizeof (long));
/*
* collect "histogram" data.
*/
for (ox = x1; ox < x2; ox += tile_width)
{
guchar *rows = src_rows;
cols = x2 - ox;
if (cols > tile_width)
cols = tile_width;
gimp_pixel_rgn_get_rect (&src_rgn, rows, ox, y1, cols, height);
for (y = 0; y < height; y++)
{
long *h = hist + (ox - x1) * bpp;
guchar *row_end = rows + cols * bpp;
while (rows < row_end)
*h++ += *rows++;
}
if (!preview)
gimp_progress_update (progress += progress_inc);
}
/*
* average out histogram
*/
{
gint extend = (vals.avg_width / 2) * bpp;
for (i = 0; i < MIN (3, bpp); i++)
{
long *h = hist - extend + i;
long *c = corr - extend + i;
long sum = 0;
gint cnt = 0;
for (x = -extend; x < width * bpp; x += bpp)
{
if (x + extend < width * bpp)
{
sum += h[ extend]; cnt++;
}
if (x - extend >= 0)
{
sum -= h[-extend]; cnt--;
}
if (x >= 0)
{
if (*h)
*c = ((sum / cnt - *h) << 10) / *h;
else
*c = G_MAXINT;
}
h += bpp;
c += bpp;
}
}
}
/*
* remove stripes.
*/
for (ox = x1; ox < x2; ox += tile_width)
{
guchar *rows = src_rows;
cols = x2 - ox;
if (cols > tile_width)
cols = tile_width;
gimp_pixel_rgn_get_rect (&src_rgn, rows, ox, y1, cols, height);
if (!preview)
gimp_progress_update (progress += progress_inc);
for (y = 0; y < height; y++)
{
long *c = corr + (ox - x1) * bpp;
guchar *row_end = rows + cols * bpp;
if (vals.histogram)
while (rows < row_end)
{
*rows = MIN (255, MAX (0, 128 + (*rows * *c >> 10)));
c++; rows++;
}
else
while (rows < row_end)
{
*rows = MIN (255, MAX (0, *rows + (*rows * *c >> 10) ));
c++; rows++;
}
}
gimp_pixel_rgn_set_rect (&dst_rgn, src_rows,
ox, y1, cols, height);
if (!preview)
gimp_progress_update (progress += progress_inc);
}
g_free (src_rows);
/*
* Update the screen...
*/
if (preview)
{
gimp_drawable_preview_draw_region (GIMP_DRAWABLE_PREVIEW (preview),
&dst_rgn);
}
else
{
gimp_drawable_flush (drawable);
gimp_drawable_merge_shadow (drawable->drawable_id, TRUE);
gimp_drawable_update (drawable->drawable_id,
x1, y1, width, height);
}
g_free (hist);
g_free (corr);
}
static gboolean
destripe_dialog (GimpDrawable *drawable)
{
GtkWidget *dialog;
GtkWidget *main_vbox;
GtkWidget *preview;
GtkWidget *table;
GtkWidget *button;
GtkObject *adj;
gboolean run;
gimp_ui_init (PLUG_IN_BINARY, TRUE);
dialog = gimp_dialog_new (_("Destripe"), PLUG_IN_BINARY,
NULL, 0,
gimp_standard_help_func, PLUG_IN_PROC,
GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL,
GTK_STOCK_OK, GTK_RESPONSE_OK,
NULL);
gtk_dialog_set_alternative_button_order (GTK_DIALOG (dialog),
GTK_RESPONSE_OK,
GTK_RESPONSE_CANCEL,
-1);
gimp_window_set_transient (GTK_WINDOW (dialog));
main_vbox = gtk_vbox_new (FALSE, 12);
gtk_container_set_border_width (GTK_CONTAINER (main_vbox), 12);
gtk_container_add (GTK_CONTAINER (gtk_dialog_get_content_area (GTK_DIALOG (dialog))),
main_vbox);
gtk_widget_show (main_vbox);
preview = gimp_drawable_preview_new (drawable, NULL);
gtk_box_pack_start (GTK_BOX (main_vbox), preview, TRUE, TRUE, 0);
gtk_widget_show (preview);
g_signal_connect_swapped (preview, "invalidated",
G_CALLBACK (destripe),
drawable);
table = gtk_table_new (1, 3, FALSE);
gtk_table_set_col_spacings (GTK_TABLE (table), 6);
gtk_box_pack_start (GTK_BOX (main_vbox), table, FALSE, FALSE, 0);
gtk_widget_show (table);
adj = gimp_scale_entry_new (GTK_TABLE (table), 0, 1,
_("_Width:"), SCALE_WIDTH, 0,
vals.avg_width, 2, MAX_AVG, 1, 10, 0,
TRUE, 0, 0,
NULL, NULL);
g_signal_connect (adj, "value-changed",
G_CALLBACK (gimp_int_adjustment_update),
&vals.avg_width);
g_signal_connect_swapped (adj, "value-changed",
G_CALLBACK (gimp_preview_invalidate),
preview);
button = gtk_check_button_new_with_mnemonic (_("Create _histogram"));
gtk_box_pack_start (GTK_BOX (main_vbox), button, FALSE, FALSE, 0);
gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (button), vals.histogram);
gtk_widget_show (button);
g_signal_connect (button, "toggled",
G_CALLBACK (gimp_toggle_button_update),
&vals.histogram);
g_signal_connect_swapped (button, "toggled",
G_CALLBACK (gimp_preview_invalidate),
preview);
gtk_widget_show (dialog);
run = (gimp_dialog_run (GIMP_DIALOG (dialog)) == GTK_RESPONSE_OK);
gtk_widget_destroy (dialog);
return run;
}