mirror of https://github.com/GNOME/gimp.git
419 lines
14 KiB
C
419 lines
14 KiB
C
/*
|
|
* Wavelet decompose plug-in by Miroslav Talasek, miroslav.talasek@seznam.cz
|
|
*/
|
|
|
|
/*
|
|
* 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 <libgimp/gimp.h>
|
|
#include <libgimp/gimpui.h>
|
|
|
|
#include "libgimp/stdplugins-intl.h"
|
|
|
|
|
|
#define PLUG_IN_PROC "plug-in-wavelet-decompose"
|
|
#define PLUG_IN_ROLE "gimp-wavelet-decompose"
|
|
#define PLUG_IN_BINARY "wavelet-decompose"
|
|
|
|
#define SCALE_WIDTH 120
|
|
#define ENTRY_WIDTH 5
|
|
|
|
|
|
typedef struct
|
|
{
|
|
gint scales;
|
|
gint create_group;
|
|
gint create_masks;
|
|
} WaveletDecomposeParams;
|
|
|
|
|
|
/* Declare 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 wavelet_blur (gint32 drawable_id,
|
|
gint radius);
|
|
|
|
|
|
static gboolean wavelet_decompose_dialog (void);
|
|
|
|
|
|
/* create a few globals, set default values */
|
|
static WaveletDecomposeParams wavelet_params =
|
|
{
|
|
5, /* default scales */
|
|
1, /* create group */
|
|
0 /* do not add mask by default */
|
|
};
|
|
|
|
const GimpPlugInInfo PLUG_IN_INFO =
|
|
{
|
|
NULL, /* init_proc */
|
|
NULL, /* quit_proc */
|
|
query, /* query_proc */
|
|
run, /* run_proc */
|
|
};
|
|
|
|
|
|
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 (unused)" },
|
|
{ GIMP_PDB_DRAWABLE, "drawable", "Input drawable" },
|
|
{ GIMP_PDB_INT32, "scales", "Number of scales (1-7)" },
|
|
{ GIMP_PDB_INT32, "create-group", "Create a layer group to store the "
|
|
"decomposition" },
|
|
{ GIMP_PDB_INT32, "create-masks", "Add a layer mask to each scales "
|
|
"layers" }
|
|
};
|
|
|
|
gimp_install_procedure (PLUG_IN_PROC,
|
|
N_("Wavelet decompose"),
|
|
"Compute and render wavelet scales",
|
|
"Miroslav Talasek <miroslav.talasek@seznam.cz>",
|
|
"Miroslav Talasek <miroslav.talasek@seznam.cz>",
|
|
"19january 2017",
|
|
N_("_Wavelet-decompose..."),
|
|
"RGB*",
|
|
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)
|
|
{
|
|
static GimpParam values[1];
|
|
GimpPDBStatusType status = GIMP_PDB_SUCCESS;
|
|
gint32 image_id;
|
|
gint32 drawable_id;
|
|
GimpRunMode run_mode;
|
|
|
|
run_mode = param[0].data.d_int32;
|
|
|
|
INIT_I18N();
|
|
gegl_init (NULL, NULL);
|
|
|
|
*nreturn_vals = 1;
|
|
*return_vals = values;
|
|
|
|
values[0].type = GIMP_PDB_STATUS;
|
|
values[0].data.d_status = status;
|
|
|
|
image_id = param[1].data.d_image;
|
|
drawable_id = param[2].data.d_drawable;
|
|
|
|
switch (run_mode)
|
|
{
|
|
case GIMP_RUN_INTERACTIVE:
|
|
gimp_get_data (PLUG_IN_PROC, &wavelet_params);
|
|
|
|
if (! wavelet_decompose_dialog ())
|
|
return;
|
|
break;
|
|
|
|
case GIMP_RUN_NONINTERACTIVE:
|
|
if (nparams != 6)
|
|
{
|
|
status = GIMP_PDB_CALLING_ERROR;
|
|
}
|
|
else
|
|
{
|
|
wavelet_params.scales = param[3].data.d_int32;
|
|
wavelet_params.create_group = param[4].data.d_int32;
|
|
wavelet_params.create_masks = param[5].data.d_int32;
|
|
}
|
|
break;
|
|
|
|
case GIMP_RUN_WITH_LAST_VALS:
|
|
gimp_get_data (PLUG_IN_PROC, &wavelet_params);
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
if (status == GIMP_PDB_SUCCESS)
|
|
{
|
|
gint32 *scale_ids;
|
|
gint32 new_scale_id;
|
|
gint32 parent_id;
|
|
GimpLayerMode grain_extract_mode = GIMP_LAYER_MODE_GRAIN_EXTRACT;
|
|
GimpLayerMode grain_merge_mode = GIMP_LAYER_MODE_GRAIN_MERGE;
|
|
gint id;
|
|
|
|
gimp_progress_init (_("Wavelet-Decompose"));
|
|
|
|
gimp_image_undo_group_start (image_id);
|
|
|
|
if (wavelet_params.create_group)
|
|
{
|
|
gint32 group_id = gimp_layer_group_new (image_id);
|
|
gimp_item_set_name (group_id, _("Decomposition"));
|
|
gimp_item_set_visible (group_id, FALSE);
|
|
gimp_image_insert_layer (image_id, group_id,
|
|
gimp_item_get_parent (drawable_id),
|
|
gimp_image_get_item_position (image_id,
|
|
drawable_id));
|
|
parent_id = group_id;
|
|
}
|
|
else
|
|
parent_id = -1;
|
|
|
|
scale_ids = g_new (gint32, wavelet_params.scales);
|
|
new_scale_id = gimp_layer_copy (drawable_id);
|
|
gimp_image_insert_layer (image_id, new_scale_id, parent_id,
|
|
gimp_image_get_item_position (image_id,
|
|
drawable_id));
|
|
|
|
/* the exact result of the grain-extract and grain-merge modes depends on
|
|
* the choice of (gamma-corrected) midpoint intensity value. for the
|
|
* non-legacy modes, the midpoint value is 0.5, which isn't representable
|
|
* exactly using integer precision. for the legacy modes, the midpoint
|
|
* value is 128/255 (i.e., 0x80), which is representable exactly using
|
|
* (gamma-corrected) integer precision. we therefore use the legacy
|
|
* modes when the input image precision is integer, and only use the
|
|
* (preferable) non-legacy modes when the input image precision is
|
|
* floating point.
|
|
*
|
|
* this avoids imperfect reconstruction of the image when using integer
|
|
* precision. see bug #786844.
|
|
*/
|
|
switch (gimp_image_get_precision (image_id))
|
|
{
|
|
case GIMP_PRECISION_U8_LINEAR:
|
|
case GIMP_PRECISION_U8_GAMMA:
|
|
case GIMP_PRECISION_U16_LINEAR:
|
|
case GIMP_PRECISION_U16_GAMMA:
|
|
case GIMP_PRECISION_U32_LINEAR:
|
|
case GIMP_PRECISION_U32_GAMMA:
|
|
grain_extract_mode = GIMP_LAYER_MODE_GRAIN_EXTRACT_LEGACY;
|
|
grain_merge_mode = GIMP_LAYER_MODE_GRAIN_MERGE_LEGACY;
|
|
break;
|
|
|
|
case GIMP_PRECISION_HALF_LINEAR:
|
|
case GIMP_PRECISION_HALF_GAMMA:
|
|
case GIMP_PRECISION_FLOAT_LINEAR:
|
|
case GIMP_PRECISION_FLOAT_GAMMA:
|
|
case GIMP_PRECISION_DOUBLE_LINEAR:
|
|
case GIMP_PRECISION_DOUBLE_GAMMA:
|
|
grain_extract_mode = GIMP_LAYER_MODE_GRAIN_EXTRACT;
|
|
grain_merge_mode = GIMP_LAYER_MODE_GRAIN_MERGE;
|
|
break;
|
|
}
|
|
|
|
for (id = 0 ; id < wavelet_params.scales; ++id)
|
|
{
|
|
gint32 blur_id, tmp_id;
|
|
gchar scale_name[20];
|
|
|
|
gimp_progress_update ((gdouble) id / (gdouble) wavelet_params.scales);
|
|
|
|
scale_ids[id] = new_scale_id;
|
|
|
|
g_snprintf (scale_name, sizeof (scale_name), _("Scale %d"), id + 1);
|
|
gimp_item_set_name (new_scale_id, scale_name);
|
|
|
|
tmp_id = gimp_layer_copy (new_scale_id);
|
|
gimp_image_insert_layer (image_id, tmp_id, parent_id,
|
|
gimp_image_get_item_position (image_id,
|
|
new_scale_id));
|
|
wavelet_blur (tmp_id, pow(2.0, id));
|
|
|
|
blur_id = gimp_layer_copy (tmp_id);
|
|
gimp_image_insert_layer (image_id, blur_id, parent_id,
|
|
gimp_image_get_item_position (image_id,
|
|
tmp_id));
|
|
|
|
gimp_layer_set_mode (tmp_id, grain_extract_mode);
|
|
new_scale_id = gimp_image_merge_down (image_id, tmp_id,
|
|
GIMP_EXPAND_AS_NECESSARY);
|
|
scale_ids[id] = new_scale_id;
|
|
|
|
gimp_item_set_visible (new_scale_id, FALSE);
|
|
|
|
new_scale_id = blur_id;
|
|
}
|
|
|
|
gimp_item_set_name (new_scale_id, _("Residual"));
|
|
|
|
for (id = 0; id < wavelet_params.scales; id++)
|
|
{
|
|
gimp_image_reorder_item (image_id, scale_ids[id], parent_id,
|
|
gimp_image_get_item_position (image_id,
|
|
new_scale_id));
|
|
gimp_layer_set_mode (scale_ids[id], grain_merge_mode);
|
|
|
|
if (wavelet_params.create_masks)
|
|
{
|
|
gint32 mask_id = gimp_layer_create_mask (scale_ids[id],
|
|
GIMP_ADD_MASK_WHITE);
|
|
gimp_layer_add_mask (scale_ids[id], mask_id);
|
|
}
|
|
|
|
gimp_item_set_visible (scale_ids[id], TRUE);
|
|
}
|
|
|
|
if (wavelet_params.create_group)
|
|
gimp_item_set_visible (parent_id, TRUE);
|
|
|
|
g_free (scale_ids);
|
|
|
|
gimp_image_undo_group_end (image_id);
|
|
|
|
gimp_progress_update (1.0);
|
|
|
|
values[0].data.d_status = status;
|
|
gimp_displays_flush ();
|
|
|
|
/* set data for next use of filter */
|
|
if (run_mode == GIMP_RUN_INTERACTIVE)
|
|
gimp_set_data (PLUG_IN_PROC,
|
|
&wavelet_params, sizeof (WaveletDecomposeParams));
|
|
}
|
|
|
|
gegl_exit ();
|
|
}
|
|
|
|
static void
|
|
wavelet_blur (gint32 drawable_id,
|
|
gint radius)
|
|
{
|
|
gint x, y, width, height;
|
|
|
|
if (gimp_drawable_mask_intersect (drawable_id, &x, &y, &width, &height))
|
|
{
|
|
GeglBuffer *buffer = gimp_drawable_get_buffer (drawable_id);
|
|
GeglBuffer *shadow = gimp_drawable_get_shadow_buffer (drawable_id);
|
|
|
|
gegl_render_op (buffer, shadow,
|
|
"gegl:wavelet-blur",
|
|
"radius", (gdouble) radius,
|
|
NULL);
|
|
|
|
gegl_buffer_flush (shadow);
|
|
gimp_drawable_merge_shadow (drawable_id, FALSE);
|
|
gimp_drawable_update (drawable_id, x, y, width, height);
|
|
g_object_unref (buffer);
|
|
g_object_unref (shadow);
|
|
}
|
|
}
|
|
|
|
static gboolean
|
|
wavelet_decompose_dialog (void)
|
|
{
|
|
GtkWidget *dialog;
|
|
GtkWidget *main_vbox;
|
|
GtkWidget *table;
|
|
GtkWidget *button;
|
|
GtkObject *adj;
|
|
gboolean run;
|
|
|
|
gimp_ui_init (PLUG_IN_BINARY, TRUE);
|
|
|
|
dialog = gimp_dialog_new (_("Wavelet decompose"), PLUG_IN_ROLE,
|
|
NULL, 0,
|
|
gimp_standard_help_func, PLUG_IN_PROC,
|
|
|
|
_("_Cancel"), GTK_RESPONSE_CANCEL,
|
|
_("_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_box_new (GTK_ORIENTATION_VERTICAL, 12);
|
|
gtk_container_set_border_width (GTK_CONTAINER (main_vbox), 12);
|
|
gtk_box_pack_start (GTK_BOX (gtk_dialog_get_content_area (GTK_DIALOG (dialog))),
|
|
main_vbox, TRUE, TRUE, 0);
|
|
gtk_widget_show (main_vbox);
|
|
|
|
table = gtk_table_new (3, 1, FALSE);
|
|
gtk_table_set_col_spacings (GTK_TABLE (table), 6);
|
|
gtk_table_set_row_spacings (GTK_TABLE (table), 6);
|
|
gtk_box_pack_start (GTK_BOX (main_vbox), table, FALSE, FALSE, 0);
|
|
gtk_widget_show (table);
|
|
|
|
/* scales */
|
|
|
|
adj = gimp_scale_entry_new (GTK_TABLE (table), 0, 0,
|
|
_("Scales:"), SCALE_WIDTH, ENTRY_WIDTH,
|
|
wavelet_params.scales,
|
|
1.0, 7.0, 1.0, 1.0, 0,
|
|
TRUE, 0, 0,
|
|
NULL, NULL);
|
|
|
|
g_signal_connect (adj, "value-changed",
|
|
G_CALLBACK (gimp_int_adjustment_update),
|
|
&wavelet_params.scales);
|
|
|
|
/* create group layer */
|
|
|
|
button = gtk_check_button_new_with_mnemonic (_("Create a layer group to store the decomposition"));
|
|
gtk_box_pack_start (GTK_BOX (main_vbox), button, FALSE, FALSE, 0);
|
|
gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (button),
|
|
wavelet_params.create_group);
|
|
gtk_widget_show (button);
|
|
|
|
g_signal_connect (button, "toggled",
|
|
G_CALLBACK (gimp_toggle_button_update),
|
|
&wavelet_params.create_group);
|
|
|
|
/* create layer masks */
|
|
|
|
button = gtk_check_button_new_with_mnemonic (_("Add a layer mask to each scales layers"));
|
|
gtk_box_pack_start (GTK_BOX (main_vbox), button, FALSE, FALSE, 0);
|
|
gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (button),
|
|
wavelet_params.create_masks);
|
|
gtk_widget_show (button);
|
|
|
|
g_signal_connect (button, "toggled",
|
|
G_CALLBACK (gimp_toggle_button_update),
|
|
&wavelet_params.create_masks);
|
|
|
|
gtk_widget_show (dialog);
|
|
|
|
run = (gimp_dialog_run (GIMP_DIALOG (dialog)) == GTK_RESPONSE_OK);
|
|
|
|
gtk_widget_destroy (dialog);
|
|
|
|
return run;
|
|
}
|