AlderconeStudio/PIKApp
AlderconeStudio
/
PIKApp
0
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases 1 Wiki Activity
PIKApp/plug-ins/common/smooth-palette.c

539 lines
16 KiB
C
Raw Blame History

/*
* smooth palette - derive smooth palette from image
* Copyright (C) 1997 Scott Draves <spot@cs.cmu.edu>
*
* PIKA - Photo and Image Kooker Application
* Copyright (C) 1995 Spencer Kimball and Peter Mattis
*
* 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 <https://www.gnu.org/licenses/>.
*/
#include "config.h"
#include <string.h>
#include <libpika/pika.h>
#include <libpika/pikaui.h>
#include "libpika/stdplugins-intl.h"
#define PLUG_IN_PROC "plug-in-smooth-palette"
#define PLUG_IN_BINARY "smooth-palette"
#define PLUG_IN_ROLE "pika-smooth-palette"
typedef struct _Palette Palette;
typedef struct _PaletteClass PaletteClass;
struct _Palette
{
PikaPlugIn parent_instance;
};
struct _PaletteClass
{
PikaPlugInClass parent_class;
};
#define PALETTE_TYPE (palette_get_type ())
#define PALETTE(obj) (G_TYPE_CHECK_INSTANCE_CAST ((obj), PALETTE_TYPE, Palette))
GType palette_get_type (void) G_GNUC_CONST;
static GList * palette_query_procedures (PikaPlugIn *plug_in);
static PikaProcedure * palette_create_procedure (PikaPlugIn *plug_in,
const gchar *name);
static PikaValueArray * palette_run (PikaProcedure *procedure,
PikaRunMode run_mode,
PikaImage *image,
gint n_drawables,
PikaDrawable **drawables,
PikaProcedureConfig *config,
gpointer run_data);
static gboolean dialog (PikaProcedure *procedure,
PikaProcedureConfig *config,
PikaDrawable *drawable);
static PikaImage * smooth_palette (PikaProcedureConfig *config,
PikaDrawable *drawable,
PikaLayer **layer);
G_DEFINE_TYPE (Palette, palette, PIKA_TYPE_PLUG_IN)
PIKA_MAIN (PALETTE_TYPE)
DEFINE_STD_SET_I18N
#define TRY_SIZE 10000
static void
palette_class_init (PaletteClass *klass)
{
PikaPlugInClass *plug_in_class = PIKA_PLUG_IN_CLASS (klass);
plug_in_class->query_procedures = palette_query_procedures;
plug_in_class->create_procedure = palette_create_procedure;
plug_in_class->set_i18n = STD_SET_I18N;
}
static void
palette_init (Palette *palette)
{
}
static GList *
palette_query_procedures (PikaPlugIn *plug_in)
{
return g_list_append (NULL, g_strdup (PLUG_IN_PROC));
}
static PikaProcedure *
palette_create_procedure (PikaPlugIn *plug_in,
const gchar *name)
{
PikaProcedure *procedure = NULL;
if (! strcmp (name, PLUG_IN_PROC))
{
procedure = pika_image_procedure_new (plug_in, name,
PIKA_PDB_PROC_TYPE_PLUGIN,
palette_run, NULL, NULL);
pika_procedure_set_image_types (procedure, "RGB*");
pika_procedure_set_sensitivity_mask (procedure,
PIKA_PROCEDURE_SENSITIVE_DRAWABLE);
pika_procedure_set_menu_label (procedure, _("Smoo_th Palette..."));
pika_procedure_add_menu_path (procedure, "<Image>/Colors/Info");
pika_procedure_set_documentation (procedure,
_("Derive a smooth color palette "
"from the image"),
"help!",
name);
pika_procedure_set_attribution (procedure,
"Scott Draves",
"Scott Draves",
"1997");
PIKA_PROC_ARG_INT (procedure, "width",
_("_Width"),
_("Width"),
2, PIKA_MAX_IMAGE_SIZE, 256,
G_PARAM_READWRITE);
PIKA_PROC_ARG_INT (procedure, "height",
_("_Height"),
_("Height"),
2, PIKA_MAX_IMAGE_SIZE, 64,
G_PARAM_READWRITE);
PIKA_PROC_ARG_INT (procedure, "n-tries",
_("Search _depth"),
_("Search depth"),
1, 1024, 50,
G_PARAM_READWRITE);
PIKA_PROC_ARG_BOOLEAN (procedure, "show-image",
_("Show image"),
_("Show image"),
TRUE,
G_PARAM_READWRITE);
PIKA_PROC_VAL_IMAGE (procedure, "new-image",
_("New image"),
_("Output image"),
FALSE,
G_PARAM_READWRITE);
PIKA_PROC_VAL_LAYER (procedure, "new-layer",
_("New layer"),
_("Output layer"),
FALSE,
G_PARAM_READWRITE);
}
return procedure;
}
static PikaValueArray *
palette_run (PikaProcedure *procedure,
PikaRunMode run_mode,
PikaImage *image,
gint n_drawables,
PikaDrawable **drawables,
PikaProcedureConfig *config,
gpointer run_data)
{
PikaImage *new_image;
PikaLayer *new_layer;
PikaDrawable *drawable;
gboolean show_image;
gegl_init (NULL, NULL);
if (n_drawables != 1)
{
GError *error = NULL;
g_set_error (&error, PIKA_PLUG_IN_ERROR, 0,
_("Procedure '%s' only works with one drawable."),
PLUG_IN_PROC);
return pika_procedure_new_return_values (procedure,
PIKA_PDB_CALLING_ERROR,
error);
}
else
{
drawable = drawables[0];
}
if (run_mode == PIKA_RUN_INTERACTIVE && ! dialog (procedure, config, drawable))
{
return pika_procedure_new_return_values (procedure,
PIKA_PDB_CANCEL,
NULL);
}
if (pika_drawable_is_rgb (drawable))
{
pika_progress_init (_("Deriving smooth palette"));
new_image = smooth_palette (config, drawable, &new_layer);
g_object_get (config, "show-image", &show_image, NULL);
if (show_image)
pika_display_new (new_image);
}
else
{
return pika_procedure_new_return_values (procedure,
PIKA_PDB_EXECUTION_ERROR,
NULL);
}
return pika_procedure_new_return_values (procedure, PIKA_PDB_SUCCESS, NULL);
}
static gfloat
pix_diff (gfloat *pal,
guint bpp,
gint i,
gint j)
{
gfloat r = 0.f;
guint k;
for (k = 0; k < bpp; k++)
{
gfloat p1 = pal[j * bpp + k];
gfloat p2 = pal[i * bpp + k];
r += (p1 - p2) * (p1 - p2);
}
return r;
}
static void
pix_swap (gfloat *pal,
guint bpp,
gint i,
gint j)
{
guint k;
for (k = 0; k < bpp; k++)
{
gfloat t = pal[j * bpp + k];
pal[j * bpp + k] = pal[i * bpp + k];
pal[i * bpp + k] = t;
}
}
static PikaImage *
smooth_palette (PikaProcedureConfig *config,
PikaDrawable *drawable,
PikaLayer **layer)
{
PikaImage *new_image;
gint psize, i, j;
guint bpp;
gint sel_x1, sel_y1;
gint width, height;
gint config_width;
gint config_height;
gint config_n_tries;
GeglBuffer *buffer;
GeglSampler *sampler;
gfloat *pal;
GRand *gr;
const Babl *format = babl_format ("RGB float");
g_object_get (config,
"width", &config_width,
"height", &config_height,
"n-tries", &config_n_tries,
NULL);
new_image = pika_image_new_with_precision (config_width,
config_height,
PIKA_RGB,
PIKA_PRECISION_FLOAT_LINEAR);
pika_image_undo_disable (new_image);
*layer = pika_layer_new (new_image, _("Background"),
config_width, config_height,
pika_drawable_type (drawable),
100,
pika_image_get_default_new_layer_mode (new_image));
pika_image_insert_layer (new_image, *layer, NULL, 0);
if (! pika_drawable_mask_intersect (drawable,
&sel_x1, &sel_y1, &width, &height))
return new_image;
gr = g_rand_new ();
psize = config_width;
buffer = pika_drawable_get_buffer (drawable);
sampler = gegl_buffer_sampler_new (buffer, format, GEGL_SAMPLER_NEAREST);
bpp = babl_format_get_n_components (gegl_buffer_get_format (buffer));
pal = g_new (gfloat, psize * bpp);
/* get initial palette */
for (i = 0; i < psize; i++)
{
gint x = sel_x1 + g_rand_int_range (gr, 0, width);
gint y = sel_y1 + g_rand_int_range (gr, 0, height);
gegl_sampler_get (sampler,
(gdouble) x, (gdouble) y, NULL, pal + i * bpp,
GEGL_ABYSS_NONE);
}
g_object_unref (sampler);
g_object_unref (buffer);
/* reorder */
if (1)
{
gfloat *pal_best;
gfloat *original;
gdouble len_best = 0;
gint try;
pal_best = g_memdup2 (pal, bpp * psize);
original = g_memdup2 (pal, bpp * psize);
for (try = 0; try < config_n_tries; try++)
{
gdouble len;
if (! (try % 5))
pika_progress_update (try / (double) config_n_tries);
memcpy (pal, original, bpp * psize);
/* scramble */
for (i = 1; i < psize; i++)
pix_swap (pal, bpp, i, g_rand_int_range (gr, 0, psize));
/* measure */
len = 0.0;
for (i = 1; i < psize; i++)
len += pix_diff (pal, bpp, i, i-1);
/* improve */
for (i = 0; i < TRY_SIZE; i++)
{
gint i0 = 1 + g_rand_int_range (gr, 0, psize-2);
gint i1 = 1 + g_rand_int_range (gr, 0, psize-2);
gfloat as_is, swapd;
if (1 == (i0 - i1))
{
as_is = (pix_diff (pal, bpp, i1 - 1, i1) +
pix_diff (pal, bpp, i0, i0 + 1));
swapd = (pix_diff (pal, bpp, i1 - 1, i0) +
pix_diff (pal, bpp, i1, i0 + 1));
}
else if (1 == (i1 - i0))
{
as_is = (pix_diff (pal, bpp, i0 - 1, i0) +
pix_diff (pal, bpp, i1, i1 + 1));
swapd = (pix_diff (pal, bpp, i0 - 1, i1) +
pix_diff (pal, bpp, i0, i1 + 1));
}
else
{
as_is = (pix_diff (pal, bpp, i0, i0 + 1) +
pix_diff (pal, bpp, i0, i0 - 1) +
pix_diff (pal, bpp, i1, i1 + 1) +
pix_diff (pal, bpp, i1, i1 - 1));
swapd = (pix_diff (pal, bpp, i1, i0 + 1) +
pix_diff (pal, bpp, i1, i0 - 1) +
pix_diff (pal, bpp, i0, i1 + 1) +
pix_diff (pal, bpp, i0, i1 - 1));
}
if (swapd < as_is)
{
pix_swap (pal, bpp, i0, i1);
len += swapd - as_is;
}
}
/* best? */
if (0 == try || len < len_best)
{
memcpy (pal_best, pal, bpp * psize);
len_best = len;
}
}
pika_progress_update (1.0);
memcpy (pal, pal_best, bpp * psize);
g_free (pal_best);
g_free (original);
/* clean */
for (i = 1; i < 4 * psize; i++)
{
gfloat as_is, swapd;
gint i0 = 1 + g_rand_int_range (gr, 0, psize - 2);
gint i1 = i0 + 1;
as_is = (pix_diff (pal, bpp, i0 - 1, i0) +
pix_diff (pal, bpp, i1, i1 + 1));
swapd = (pix_diff (pal, bpp, i0 - 1, i1) +
pix_diff (pal, bpp, i0, i1 + 1));
if (swapd < as_is)
{
pix_swap (pal, bpp, i0, i1);
len_best += swapd - as_is;
}
}
}
/* store smooth palette */
buffer = pika_drawable_get_buffer (PIKA_DRAWABLE (*layer));
for (j = 0; j < config_height; j++)
{
GeglRectangle row = {0, j, config_width, 1};
gegl_buffer_set (buffer, &row, 0, format, pal, GEGL_AUTO_ROWSTRIDE);
}
gegl_buffer_flush (buffer);
pika_drawable_update (PIKA_DRAWABLE (*layer), 0, 0,
config_width, config_height);
pika_image_undo_enable (new_image);
g_object_unref (buffer);
g_free (pal);
g_rand_free (gr);
return new_image;
}
static gboolean
dialog (PikaProcedure *procedure,
PikaProcedureConfig *config,
PikaDrawable *drawable)
{
GtkWidget *dlg;
GtkWidget *sizeentry;
PikaImage *image;
PikaUnit unit;
gdouble xres, yres;
gint width;
gint height;
gboolean run;
g_object_get (config,
"width", &width,
"height", &height,
NULL);
pika_ui_init (PLUG_IN_BINARY);
dlg = pika_procedure_dialog_new (procedure,
PIKA_PROCEDURE_CONFIG (config),
_("Smooth Palette"));
pika_dialog_set_alternative_button_order (GTK_DIALOG (dlg),
GTK_RESPONSE_OK,
GTK_RESPONSE_CANCEL,
-1);
pika_window_set_transient (GTK_WINDOW (dlg));
image = pika_item_get_image (PIKA_ITEM (drawable));
unit = pika_image_get_unit (image);
pika_image_get_resolution (image, &xres, &yres);
sizeentry = pika_coordinates_new (unit, "%a", TRUE, FALSE, 6,
PIKA_SIZE_ENTRY_UPDATE_SIZE,
FALSE, FALSE,
_("_Width:"),
width, xres,
2, PIKA_MAX_IMAGE_SIZE,
2, PIKA_MAX_IMAGE_SIZE,
_("_Height:"),
height, yres,
1, PIKA_MAX_IMAGE_SIZE,
1, PIKA_MAX_IMAGE_SIZE);
gtk_container_set_border_width (GTK_CONTAINER (sizeentry), 12);
gtk_box_pack_start (GTK_BOX (gtk_dialog_get_content_area (GTK_DIALOG (dlg))),
sizeentry, FALSE, FALSE, 0);
gtk_widget_show (sizeentry);
/* We don't want to have the "Show Image" option in the GUI */
pika_procedure_dialog_fill (PIKA_PROCEDURE_DIALOG (dlg), "n-tries", NULL);
gtk_widget_show (dlg);
run = (pika_dialog_run (PIKA_DIALOG (dlg)) == GTK_RESPONSE_OK);
if (run)
{
width = pika_size_entry_get_refval (PIKA_SIZE_ENTRY (sizeentry), 0);
height = pika_size_entry_get_refval (PIKA_SIZE_ENTRY (sizeentry), 1);
g_object_set (config,
"width", width,
"height", height,
NULL);
}
gtk_widget_destroy (dlg);
return run;
}
Reference in New Issue View Git Blame Copy Permalink