Logo Search packages:      
Sourcecode: gimp version File versions  Download package

warp.c

/* Warp  --- image filter plug-in for The Gimp image manipulation program
 * Copyright (C) 1997 John P. Beale
 * Much of the 'warp' is from the Displace plug-in: 1996 Stephen Robert Norris
 * Much of the 'displace' code taken in turn  from the pinch plug-in
 *   which is by 1996 Federico Mena Quintero
 *
 * 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 2 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, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 * You can contact me (the warp author) at beale@best.com
 * Please send me any patches or enhancements to this code.
 * You can contact the original The Gimp authors at gimp@xcf.berkeley.edu
 *
 * --------------------------------------------------------------------
 * Warp Program structure: after running the user interface and setting the
 * parameters, warp generates a brand-new image (later to be deleted
 * before the user ever sees it) which contains two grayscale layers,
 * representing the X and Y gradients of the "control" image. For this
 * purpose, all channels of the control image are summed for a scalar
 * value at each pixel coordinate for the gradient operation.
 *
 * The X,Y components of the calculated gradient are then used to displace pixels
 * from the source image into the destination image. The displacement vector is
 * rotated a user-specified amount first. This displacement operation happens
 * iteratively, generating a new displaced image from each prior image.
 * -------------------------------------------------------------------
 *
 * Revision History:
 * Version 0.37  12/19/98 Fixed Tooltips and freeing memory
 * Version 0.36  11/9/97  Changed XY vector layers  back to own image
 *               fixed 'undo' problem (hopefully)
 *
 * Version 0.35  11/3/97  Added vector-map, mag-map, grad-map to
 *               diff vector instead of separate operation
 *               further futzing with drawable updates
 *               starting adding tooltips
 *
 * Version 0.34  10/30/97   'Fixed' drawable update problem
 *               Added 16-bit resolution to differential map
 *             Added substep increments for finer control
 *
 * Version 0.33  10/26/97   Added 'angle increment' to user interface
 *
 * Version 0.32  10/25/97   Added magnitude control map (secondary control)
 *               Changed undo behavior to be one undo-step per warp call.
 *
 * Version 0.31  10/25/97   Fixed src/dest pixregions so program works
 *               with multiple-layer images. Still don't know
 *               exactly what I did to fix it :-/  Also, added 'color' option for
 *               border pixels to use the current selected foreground color.
 *
 * Version 0.3   10/20/97  Initial release for Gimp 0.99.xx
 */

#include "config.h"

#include <stdio.h>
#include <stdlib.h>

#include <gtk/gtk.h>

#include <libgimp/gimp.h>
#include <libgimp/gimpui.h>

#include "libgimp/stdplugins-intl.h"


/* Some useful macros */

#define ENTRY_WIDTH     75
#define TILE_CACHE_SIZE 30  /* was 48. There is a cache flush problem in GIMP preventing sequential updates */
#define MIN_ARGS         6  /* minimum number of arguments required */

enum
{
  WRAP,
  SMEAR,
  BLACK,
  COLOR
};

typedef struct
{
  gdouble amount;
  gint    warp_map;
  gint    iter;
  gdouble dither;
  gdouble angle;
  gint    wrap_type;
  gint    mag_map;
  gint    mag_use;
  gint    substeps;
  gint    grad_map;
  gdouble grad_scale;
  gint    vector_map;
  gdouble vector_scale;
  gdouble vector_angle;
} WarpVals;


/*
 * Function prototypes.
 */

static void      query  (void);
static void      run    (const gchar      *name,
                   gint              nparams,
                   const GimpParam  *param,
                   gint             *nreturn_vals,
                   GimpParam       **return_vals);

static void      blur16           (GimpDrawable *drawable);

static void      diff             (GimpDrawable *drawable,
                           gint32       *xl_id,
                           gint32       *yl_id);

static void      diff_prepare_row (GimpPixelRgn *pixel_rgn,
                           guchar       *data,
                           int           x,
                           int           y,
                           int           w);

static void      warp_one         (GimpDrawable *draw,
                           GimpDrawable *new,
                           GimpDrawable *map_x,
                           GimpDrawable *map_y,
                           GimpDrawable *mag_draw,
                           gint          first_time,
                           gint          step);

static void      warp        (GimpDrawable  *drawable,
                        GimpDrawable **map_x_p,
                        GimpDrawable **map_y_p);

static gboolean  warp_dialog (GimpDrawable *drawable);
static GimpTile *warp_pixel  (GimpDrawable *drawable,
                        GimpTile     *tile,
                        gint          width,
                        gint          height,
                        gint          x1,
                        gint          y1,
                        gint          x2,
                        gint          y2,
                        gint          x,
                        gint          y,
                        gint         *row,
                        gint         *col,
                        guchar       *pixel);

static gboolean  warp_map_constrain       (gint32     image_id,
                                 gint32     drawable_id,
                                 gpointer   data);
static gdouble   warp_map_mag_give_value  (guchar    *pt,
                                 gint       alpha,
                                 gint       bytes);

/* -------------------------------------------------------------------------- */
/*   Variables global over entire plug-in scope                               */
/* -------------------------------------------------------------------------- */

GimpPlugInInfo PLUG_IN_INFO =
{
  NULL,  /* init_proc  */
  NULL,  /* quit_proc  */
  query, /* query_proc */
  run,   /* run_proc   */
};

static WarpVals dvals =
{
  10.0,   /* amount       */
  -1,     /* warp_map     */
  5,      /* iterations   */
  0.0,    /* dither       */
  90.0,   /* angle        */
  WRAP,   /* wrap_type    */
  -1,     /* mag_map      */
  FALSE,  /* mag_use      */
  1,      /* substeps     */
  -1,     /* grad_map     */
  0.0,    /* grad_scale   */
  -1,     /* vector_map   */
  0.0,    /* vector_scale */
  0.0     /* vector_angle */
};

/* -------------------------------------------------------------------------- */

/* static gint         display_diff_map = TRUE;   show 16-bit diff. vectormap */
static gint         progress = 0;              /* progress indicator bar      */
static guint        tile_width, tile_height;   /* size of an image tile       */
static GimpRunMode run_mode;                  /* interactive, non-, etc.     */
static guchar       color_pixel[4] = {0, 0, 0, 255};  /* current fg color     */

/* -------------------------------------------------------------------------- */

/***** Functions *****/

MAIN ()

static void
query (void)
{
  static GimpParamDef args[] =
  {
    { GIMP_PDB_INT32, "run_mode", "Interactive, non-interactive" },
    { GIMP_PDB_IMAGE, "image", "Input image (unused)" },
    { GIMP_PDB_DRAWABLE, "drawable", "Input drawable" },
    { GIMP_PDB_FLOAT, "amount", "Pixel displacement multiplier" },
    { GIMP_PDB_DRAWABLE, "warp_map", "Displacement control map" },
    { GIMP_PDB_INT32, "iter", "Iteration count (last required argument)" },
    { GIMP_PDB_FLOAT, "dither", "Random dither amount (first optional argument)" },
    { GIMP_PDB_FLOAT, "angle", "Angle of gradient vector rotation" },
    { GIMP_PDB_INT32, "wrap_type", "Edge behavior: { WRAP (0), SMEAR (1), BLACK (2), COLOR (3) }" },
    { GIMP_PDB_DRAWABLE, "mag_map", "Magnitude control map" },
    { GIMP_PDB_INT32, "mag_use", "Use magnitude map: { FALSE (0), TRUE (1) }" },
    { GIMP_PDB_INT32, "substeps", "Substeps between image updates" },
    { GIMP_PDB_INT32, "grad_map", "Gradient control map" },
    { GIMP_PDB_FLOAT, "grad_scale", "Scaling factor for gradient map (0=don't use)" },
    { GIMP_PDB_INT32, "vector_map", "Fixed vector control map" },
    { GIMP_PDB_FLOAT, "vector_scale", "Scaling factor for fixed vector map (0=don't use)" },
    { GIMP_PDB_FLOAT, "vector_angle", "Angle for fixed vector map" }
  };

  gimp_install_procedure ("plug_in_warp",
                    "Twist or smear an image. (only first six "
                          "arguments are required)",
                    "Smears an image along vector paths calculated as "
                    "the gradient of a separate control matrix. The "
                    "effect can look like brushstrokes of acrylic or "
                    "watercolor paint, in some cases.",
                    "John P. Beale",
                    "John P. Beale",
                    "1997",
                    N_("_Warp..."),
                    "RGB*, GRAY*",
                    GIMP_PLUGIN,
                    G_N_ELEMENTS (args), 0,
                    args, NULL);

  gimp_plugin_menu_register ("plug_in_warp", "<Image>/Filters/Map");
}

static void
run (const gchar      *name,
     gint              nparams,
     const GimpParam  *param,
     gint             *nreturn_vals,
     GimpParam       **return_vals)
{
  static GimpParam  values[1];
  GimpDrawable     *drawable;
  GimpDrawable     *map_x = NULL;   /* satisfy compiler complaints */
  GimpDrawable     *map_y = NULL;
  gint32            image_ID;       /* image id of drawable */
  GimpRGB           color;

  GimpPDBStatusType status = GIMP_PDB_SUCCESS;
  gint              pcnt;           /* parameter counter for scanning input params. */

  run_mode = param[0].data.d_int32;

  INIT_I18N ();

  tile_width  = gimp_tile_width ();    /* initialize some globals */
  tile_height = gimp_tile_height ();

  /* get currently selected foreground pixel color */
  gimp_context_get_foreground (&color);
  gimp_rgb_get_uchar (&color,
                  &color_pixel[0],
                  &color_pixel[1],
                  &color_pixel[2]);

  /*  Get the specified drawable  */
  drawable = gimp_drawable_get (param[2].data.d_drawable);

  *nreturn_vals = 1;
  *return_vals  = values;

  values[0].type          = GIMP_PDB_STATUS;
  values[0].data.d_status = status;

  switch (run_mode)
    {
    case GIMP_RUN_INTERACTIVE:
      /*  Possibly retrieve data  */
      gimp_get_data ("plug_in_warp", &dvals);

      /*  First acquire information with a dialog  */
      if (! warp_dialog (drawable))
      return;
      break;

    case GIMP_RUN_NONINTERACTIVE:
      /*  Make sure minimum args
       *  (mode, image, draw, amount, warp_map, iter) are there
       */
      if (nparams < MIN_ARGS)
      {
        status = GIMP_PDB_CALLING_ERROR;
      }
      else
      {
        pcnt = MIN_ARGS;                          /* parameter counter */
        dvals.amount   = param[3].data.d_float;
        dvals.warp_map = param[4].data.d_int32;
        dvals.iter     = param[5].data.d_int32;
        if (nparams > pcnt++) dvals.dither       = param[6].data.d_float;
        if (nparams > pcnt++) dvals.angle        = param[7].data.d_float;
        if (nparams > pcnt++) dvals.wrap_type    = param[8].data.d_int32;
        if (nparams > pcnt++) dvals.mag_map      = param[9].data.d_int32;
        if (nparams > pcnt++) dvals.mag_use      = param[10].data.d_int32;
        if (nparams > pcnt++) dvals.substeps     = param[11].data.d_int32;
        if (nparams > pcnt++) dvals.grad_map     = param[12].data.d_int32;
        if (nparams > pcnt++) dvals.grad_scale   = param[13].data.d_float;
        if (nparams > pcnt++) dvals.vector_map   = param[14].data.d_int32;
        if (nparams > pcnt++) dvals.vector_scale = param[15].data.d_float;
        if (nparams > pcnt++) dvals.vector_angle = param[16].data.d_float;
      }
      break;

    case GIMP_RUN_WITH_LAST_VALS:
      /*  Possibly retrieve data  */
      gimp_get_data ("plug_in_warp", &dvals);
      break;

    default:
      break;
    }

  if (status == GIMP_PDB_SUCCESS)
    {
      /*  set the tile cache size  */
      gimp_tile_cache_ntiles (TILE_CACHE_SIZE);

      /*  run the warp effect  */
      warp (drawable, &map_x, &map_y);

      /*  Store data  */
      if (run_mode == GIMP_RUN_INTERACTIVE)
      gimp_set_data ("plug_in_warp", &dvals, sizeof (WarpVals));
    }

  values[0].data.d_status = status;

  image_ID = gimp_drawable_get_image (map_x->drawable_id);

  gimp_drawable_detach (map_x);
  gimp_drawable_detach (map_y);

  gimp_image_delete (image_ID);

  if (run_mode != GIMP_RUN_NONINTERACTIVE)
    gimp_displays_flush ();
}

static gboolean
warp_dialog (GimpDrawable *drawable)
{
  GtkWidget    *dlg;
  GtkWidget    *vbox;
  GtkWidget    *label;
  GtkWidget    *toggle;
  GtkWidget    *toggle_hbox;
  GtkWidget    *frame;
  GtkWidget    *table;
  GtkWidget    *spinbutton;
  GtkObject    *adj;
  GtkWidget    *combo;
  GtkWidget    *ebox;
  GtkSizeGroup *label_group;
  GtkSizeGroup *spin_group;
  GSList       *group = NULL;
  gboolean      run;

  gimp_ui_init ("warp", FALSE);

  dlg = gimp_dialog_new (_("Warp"), "warp",
                         NULL, 0,
                   gimp_standard_help_func, "plug-in-warp",

                         GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL,
                         GTK_STOCK_OK,     GTK_RESPONSE_OK,

                         NULL);

  vbox = gtk_vbox_new (FALSE, 12);
  gtk_container_set_border_width (GTK_CONTAINER (vbox), 12);
  gtk_box_pack_start (GTK_BOX (GTK_DIALOG (dlg)->vbox), vbox, TRUE, TRUE, 0);
  gtk_widget_show (vbox);

  frame = gimp_frame_new (_("Basic Options"));
  gtk_box_pack_start (GTK_BOX (vbox), frame, FALSE, FALSE, 0);
  gtk_widget_show (frame);

  table = gtk_table_new (3, 3, FALSE);
  gtk_table_set_row_spacings (GTK_TABLE (table), 6);
  gtk_table_set_col_spacings (GTK_TABLE (table), 6);
  gtk_table_set_col_spacing (GTK_TABLE (table), 1, 12);
  gtk_container_add (GTK_CONTAINER (frame), table);
  gtk_widget_show (table);

  spin_group = gtk_size_group_new (GTK_SIZE_GROUP_HORIZONTAL);
  label_group = gtk_size_group_new (GTK_SIZE_GROUP_HORIZONTAL);

  /*  amount, iter */
  spinbutton = gimp_spin_button_new (&adj, dvals.amount,
                             -1000, 1000, /* ??? */
                             1, 10, 0, 1, 2);
  gtk_size_group_add_widget (spin_group, spinbutton);
  g_object_unref (spin_group);

  label = gimp_table_attach_aligned (GTK_TABLE (table), 0, 0,
                                     _("Step size:"), 0.0, 0.5,
                                     spinbutton, 1, FALSE);
  gtk_size_group_add_widget (label_group, label);
  g_object_unref (label_group);

  g_signal_connect (adj, "value_changed",
                    G_CALLBACK (gimp_double_adjustment_update),
                    &dvals.amount);

  spinbutton = gimp_spin_button_new (&adj, dvals.iter,
                             1, 100, 1, 5, 0, 1, 0);
  gtk_size_group_add_widget (spin_group, spinbutton);

  label = gimp_table_attach_aligned (GTK_TABLE (table), 0, 1,
                                     _("Iterations:"), 0.0, 0.5,
                                     spinbutton, 1, FALSE);
  gtk_size_group_add_widget (label_group, label);

  g_signal_connect (adj, "value_changed",
                    G_CALLBACK (gimp_int_adjustment_update),
                    &dvals.iter);

  /*  Displacement map menu  */
  label = gtk_label_new (_("Displacement map:"));
  gtk_misc_set_alignment (GTK_MISC (label), 0.0, 1.0);
  gtk_table_attach (GTK_TABLE (table), label, 2, 3, 0, 1,
                GTK_FILL, GTK_FILL, 0, 0);
  gtk_widget_show (label);

  combo = gimp_drawable_combo_box_new (warp_map_constrain, drawable);
  gimp_int_combo_box_connect (GIMP_INT_COMBO_BOX (combo), dvals.warp_map,
                              G_CALLBACK (gimp_int_combo_box_get_active),
                              &dvals.warp_map);

  gtk_table_attach (GTK_TABLE (table), combo, 2, 3, 1, 2,
                GTK_EXPAND | GTK_FILL, GTK_EXPAND | GTK_FILL, 0, 0);
  gtk_widget_show (combo);

  /* ======================================================================= */

  /*  Displacement Type  */
  label = gtk_label_new (_("On edges:"));
  gtk_misc_set_alignment (GTK_MISC (label), 0.0, 0.0);
  gtk_table_attach (GTK_TABLE (table), label, 0, 1, 2, 3,
                GTK_FILL, GTK_FILL, 0, 0);
  gtk_widget_show (label);

  toggle_hbox = gtk_hbox_new (FALSE, 4);
  gtk_table_attach (GTK_TABLE (table), toggle_hbox, 1, 3, 2, 3,
                GTK_FILL, GTK_FILL, 0, 0);
  gtk_widget_show (toggle_hbox);

  toggle = gtk_radio_button_new_with_label (group, _("Wrap"));
  group = gtk_radio_button_get_group (GTK_RADIO_BUTTON (toggle));
  gtk_box_pack_start (GTK_BOX (toggle_hbox), toggle, FALSE, FALSE, 0);
  gtk_widget_show (toggle);

  g_object_set_data (G_OBJECT (toggle), "gimp-item-data",
                     GINT_TO_POINTER (WRAP));

  g_signal_connect (toggle, "toggled",
                    G_CALLBACK (gimp_radio_button_update),
                    &dvals.wrap_type);

  gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (toggle),
                        dvals.wrap_type == WRAP);

  toggle = gtk_radio_button_new_with_label (group, _("Smear"));
  group = gtk_radio_button_get_group (GTK_RADIO_BUTTON (toggle));
  gtk_box_pack_start (GTK_BOX (toggle_hbox), toggle, FALSE, FALSE, 0);
  gtk_widget_show (toggle);

  g_object_set_data (G_OBJECT (toggle), "gimp-item-data",
                     GINT_TO_POINTER (SMEAR));

  g_signal_connect (toggle, "toggled",
                    G_CALLBACK (gimp_radio_button_update),
                    &dvals.wrap_type);

  gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (toggle),
                        dvals.wrap_type == SMEAR);

  toggle = gtk_radio_button_new_with_label (group, _("Black"));
  group = gtk_radio_button_get_group (GTK_RADIO_BUTTON (toggle));
  gtk_box_pack_start (GTK_BOX (toggle_hbox), toggle, FALSE, FALSE, 0);
  gtk_widget_show (toggle);

  g_object_set_data (G_OBJECT (toggle), "gimp-item-data",
                     GINT_TO_POINTER (BLACK));

  g_signal_connect (toggle, "toggled",
                    G_CALLBACK (gimp_radio_button_update),
                    &dvals.wrap_type);

  gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (toggle),
                        dvals.wrap_type == BLACK);

  toggle = gtk_radio_button_new_with_label (group, _("FG color"));
  group = gtk_radio_button_get_group (GTK_RADIO_BUTTON (toggle));
  gtk_box_pack_start (GTK_BOX (toggle_hbox), toggle, FALSE, FALSE, 0);
  gtk_widget_show (toggle);

  g_object_set_data (G_OBJECT (toggle), "gimp-item-data",
                     GINT_TO_POINTER (COLOR));

  g_signal_connect (toggle, "toggled",
                    G_CALLBACK (gimp_radio_button_update),
                    &dvals.wrap_type);

  gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (toggle),
                        dvals.wrap_type == COLOR);



  /* -------------------------------------------------------------------- */
  /* ---------    The secondary table         --------------------------  */

  frame = gimp_frame_new (_("Advanced Options"));
  gtk_box_pack_start (GTK_BOX (vbox), frame, FALSE, FALSE, 0);
  gtk_widget_show (frame);

  table = gtk_table_new (3, 3, FALSE);
  gtk_table_set_row_spacings (GTK_TABLE (table), 6);
  gtk_table_set_col_spacings (GTK_TABLE (table), 6);
  gtk_table_set_col_spacing (GTK_TABLE (table), 1, 12);
  gtk_container_add (GTK_CONTAINER (frame), table);
  gtk_widget_show (table);

  spinbutton = gimp_spin_button_new (&adj, dvals.dither,
                             0, 100, 1, 10, 0, 1, 2);
  gtk_size_group_add_widget (spin_group, spinbutton);

  label = gimp_table_attach_aligned (GTK_TABLE (table), 0, 0,
                                     _("Dither size:"), 0.0, 0.5,
                                     spinbutton, 1, FALSE);
  gtk_size_group_add_widget (label_group, label);

  g_signal_connect (adj, "value_changed",
                    G_CALLBACK (gimp_double_adjustment_update),
                    &dvals.dither);

  spinbutton = gimp_spin_button_new (&adj, dvals.angle,
                             0, 360, 1, 15, 0, 1, 1);
  gtk_size_group_add_widget (spin_group, spinbutton);

  label = gimp_table_attach_aligned (GTK_TABLE (table), 0, 1,
                                     _("Rotation angle:"), 0.0, 0.5,
                                     spinbutton, 1, FALSE);
  gtk_size_group_add_widget (label_group, label);

  g_signal_connect (adj, "value_changed",
                    G_CALLBACK (gimp_double_adjustment_update),
                    &dvals.angle);

  spinbutton = gimp_spin_button_new (&adj, dvals.substeps,
                             1, 100, 1, 5, 0, 1, 0);
  gtk_size_group_add_widget (spin_group, spinbutton);

  label = gimp_table_attach_aligned (GTK_TABLE (table), 0, 2,
                                     _("Substeps:"), 0.0, 0.5,
                                     spinbutton, 1, FALSE);
  gtk_size_group_add_widget (label_group, label);

  g_signal_connect (adj, "value_changed",
                    G_CALLBACK (gimp_int_adjustment_update),
                    &dvals.substeps);

  /*  Magnitude map menu  */
  label = gtk_label_new (_("Magnitude map:"));
  gtk_misc_set_alignment (GTK_MISC (label), 0.0, 1.0);
  gtk_table_attach (GTK_TABLE (table), label, 2, 3, 0, 1,
                GTK_FILL, GTK_FILL, 0, 0);
  gtk_widget_show (label);

  combo = gimp_drawable_combo_box_new (warp_map_constrain, drawable);
  gimp_int_combo_box_connect (GIMP_INT_COMBO_BOX (combo), dvals.mag_map,
                              G_CALLBACK (gimp_int_combo_box_get_active),
                              &dvals.mag_map);

  gtk_table_attach (GTK_TABLE (table), combo, 2, 3, 1, 2,
                GTK_EXPAND | GTK_FILL, GTK_EXPAND | GTK_FILL, 0, 0);
  gtk_widget_show (combo);

  /*  Magnitude Usage  */
  toggle_hbox = gtk_hbox_new (FALSE, 4);
  gtk_container_set_border_width (GTK_CONTAINER (toggle_hbox), 1);
  gtk_table_attach (GTK_TABLE (table), toggle_hbox, 2, 3, 2, 3,
                GTK_FILL, GTK_FILL, 0, 0);
  gtk_widget_show (toggle_hbox);

  toggle = gtk_check_button_new_with_label (_("Use magnitude map"));
  gtk_box_pack_start (GTK_BOX (toggle_hbox), toggle, FALSE, FALSE, 0);
  gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (toggle), dvals.mag_use);
  gtk_widget_show (toggle);

  g_signal_connect (toggle, "toggled",
                    G_CALLBACK (gimp_toggle_button_update),
                    &dvals.mag_use);


  /* -------------------------------------------------------------------- */
  /* ---------    The "other" table         --------------------------  */

  frame = gimp_frame_new (_("More Advanced Options"));
  gtk_box_pack_start (GTK_BOX (vbox), frame, FALSE, FALSE, 0);
  gtk_widget_show (frame);

  table = gtk_table_new (3, 3, FALSE);
  gtk_table_set_row_spacings (GTK_TABLE (table), 6);
  gtk_table_set_col_spacings (GTK_TABLE (table), 6);
  gtk_table_set_col_spacing (GTK_TABLE (table), 1, 12);
  gtk_container_add (GTK_CONTAINER (frame), table);
  gtk_widget_show (table);

  spinbutton = gimp_spin_button_new (&adj, dvals.grad_scale,
                             -1000, 1000, /* ??? */
                             0.01, 0.1, 0, 1, 3);
  gtk_size_group_add_widget (spin_group, spinbutton);

  label = gimp_table_attach_aligned (GTK_TABLE (table), 0, 0,
                                     _("Gradient scale:"), 0.0, 0.5,
                                     spinbutton, 1, FALSE);
  gtk_size_group_add_widget (label_group, label);

  g_signal_connect (adj, "value_changed",
                    G_CALLBACK (gimp_double_adjustment_update),
                    &dvals.grad_scale);

  /* ---------  Gradient map menu ----------------  */

  combo = gimp_drawable_combo_box_new (warp_map_constrain, drawable);
  gimp_int_combo_box_connect (GIMP_INT_COMBO_BOX (combo), dvals.grad_map,
                              G_CALLBACK (gimp_int_combo_box_get_active),
                              &dvals.grad_map);

  ebox = gtk_event_box_new ();
  gtk_container_add (GTK_CONTAINER (ebox), combo);
  gtk_widget_show (combo);

  gtk_table_attach (GTK_TABLE (table), ebox, 2, 3, 0, 1,
                GTK_EXPAND | GTK_FILL, GTK_EXPAND | GTK_FILL, 0, 0);
  gtk_widget_show (ebox);

  gimp_help_set_help_data (ebox, _("Gradient map selection menu"), NULL);

  /* ---------------------------------------------- */

  spinbutton = gimp_spin_button_new (&adj, dvals.vector_scale,
                             -1000, 1000, /* ??? */
                             0.01, 0.1, 0, 1, 3);
  gtk_size_group_add_widget (spin_group, spinbutton);

  label = gimp_table_attach_aligned (GTK_TABLE (table), 0, 1,
                                     _("Vector mag:"), 0.0, 0.5,
                                     spinbutton, 1, FALSE);
  gtk_size_group_add_widget (label_group, label);

  g_signal_connect (adj, "value_changed",
                    G_CALLBACK (gimp_double_adjustment_update),
                    &dvals.vector_scale);

  /* -------------------------------------------------------- */

  spinbutton = gimp_spin_button_new (&adj, dvals.vector_angle,
                             0, 360, 1, 15, 0, 1, 1);
  gtk_size_group_add_widget (spin_group, spinbutton);

  label = gimp_table_attach_aligned (GTK_TABLE (table), 0, 2,
                                     _("Angle:"), 0.0, 0.5,
                                     spinbutton, 1, FALSE);
  gtk_size_group_add_widget (label_group, label);

  g_signal_connect (adj, "value_changed",
                    G_CALLBACK (gimp_double_adjustment_update),
                    &dvals.vector_angle);

  /* ---------  Vector map menu ----------------  */
  combo = gimp_drawable_combo_box_new (warp_map_constrain, drawable);
  gimp_int_combo_box_connect (GIMP_INT_COMBO_BOX (combo), dvals.vector_map,
                              G_CALLBACK (gimp_int_combo_box_get_active),
                              &dvals.vector_map);

  ebox = gtk_event_box_new ();
  gtk_container_add (GTK_CONTAINER (ebox), combo);
  gtk_widget_show (combo);

  gtk_table_attach (GTK_TABLE (table), ebox, 2, 3, 1, 2,
                GTK_EXPAND | GTK_FILL, GTK_EXPAND | GTK_FILL, 0, 0);
  gtk_widget_show (ebox);

  gimp_help_set_help_data (ebox,
                     _("Fixed-direction-vector map selection menu"),
                     NULL);

  gtk_widget_show (dlg);

  run = (gimp_dialog_run (GIMP_DIALOG (dlg)) == GTK_RESPONSE_OK);

  gtk_widget_destroy (dlg);

  return run;
}

/* ---------------------------------------------------------------------- */

static void
blur16 (GimpDrawable *drawable)
{
  /*  blur a 2-or-more byte-per-pixel drawable,
   *  1st 2 bytes interpreted as a 16-bit height field.
   */
  GimpPixelRgn srcPR, destPR;
  gint width, height;
  gint src_bytes;
  gint dest_bytes;
  gint dest_bytes_inc;
  gint offb, off1;

  guchar *dest, *d;  /* pointers to rows of X and Y diff. data */
  guchar *prev_row, *pr;
  guchar *cur_row, *cr;
  guchar *next_row, *nr;
  guchar *tmp;
  gint row, col;  /* relating to indexing into pixel row arrays */
  gint x1, y1, x2, y2;
  gdouble pval;          /* average pixel value of pixel & neighbors */

  /* --------------------------------------- */

  gimp_drawable_mask_bounds (drawable->drawable_id, &x1, &y1, &x2, &y2);

  width = drawable->width;     /* size of input drawable*/
  height = drawable->height;
  src_bytes = drawable->bpp;   /* bytes per pixel in SOURCE drawable, must be 2 or more  */
  dest_bytes = drawable->bpp;   /* bytes per pixel in SOURCE drawable, >= 2  */
  dest_bytes_inc = dest_bytes - 2;  /* this is most likely zero, but I guess it's more conservative... */

  /*  allocate row buffers for source & dest. data  */

  prev_row = (guchar *) malloc ((x2 - x1 + 2) * src_bytes);
  cur_row = (guchar *) malloc ((x2 - x1 + 2) * src_bytes);
  next_row = (guchar *) malloc ((x2 - x1 + 2) * src_bytes);
  dest = (guchar *) malloc ((x2 - x1) * src_bytes);

  /* initialize the pixel regions (read from source, write into dest)  */
  gimp_pixel_rgn_init (&srcPR, drawable, 0, 0, width, height, FALSE, FALSE);
  gimp_pixel_rgn_init (&destPR, drawable, 0, 0, width, height, TRUE, TRUE);

  pr = prev_row + src_bytes;   /* row arrays are prepared for indexing to -1 (!) */
  cr = cur_row + src_bytes;
  nr = next_row + src_bytes;

  diff_prepare_row (&srcPR, pr, x1, y1, (x2 - x1));
  diff_prepare_row (&srcPR, cr, x1, y1+1, (x2 - x1));

  /*  loop through the rows, applying the smoothing function  */
  for (row = y1; row < y2; row++)
    {
      /*  prepare the next row  */
      diff_prepare_row (&srcPR, nr, x1, row + 1, (x2 - x1));

      d = dest;
      for (col = 0; col < (x2 - x1); col++) /* over columns of pixels */
      {
        offb = col*src_bytes;    /* base of byte pointer offset */
        off1 = offb+1;                 /* offset into row arrays */

        pval = (256.0*pr[offb - src_bytes] + pr[off1 - src_bytes] +
              256.0*pr[offb] + pr[off1] +
              256.0*pr[offb + src_bytes] + pr[off1 + src_bytes] +
              256.0*cr[offb - src_bytes] + cr[off1 - src_bytes] +
              256.0*cr[offb]  + cr[off1] +
              256.0*cr[offb + src_bytes] + cr[off1 + src_bytes] +
              256.0*nr[offb - src_bytes] + nr[off1 - src_bytes] +
              256.0*nr[offb] + nr[off1] +
              256.0*nr[offb + src_bytes]) + nr[off1 + src_bytes];

        pval /= 9.0;  /* take the average */
        *d++ = (guchar) (((gint)pval)>>8);   /* high-order byte */
        *d++ = (guchar) (((gint)pval)%256);  /* low-order byte */
        d += dest_bytes_inc;       /* move data pointer on to next destination pixel */

      }
      /*  store the dest  */
      gimp_pixel_rgn_set_row (&destPR, dest, x1, row, (x2 - x1));

      /*  shuffle the row pointers  */
      tmp = pr;
      pr = cr;
      cr = nr;
      nr = tmp;

      if ((row % 5) == 0)
      gimp_progress_update ((double) row / (double) (y2 - y1));
    }

  /*  update the region  */
  gimp_drawable_flush (drawable);
  gimp_drawable_merge_shadow (drawable->drawable_id, TRUE);
  gimp_drawable_update (drawable->drawable_id, x1, y1, (x2 - x1), (y2 - y1));

  free (prev_row);  /* row buffers allocated at top of fn. */
  free (cur_row);
  free (next_row);
  free (dest);

} /* end blur16() */


/* ====================================================================== */
/* Get one row of pixels from the PixelRegion and put them in 'data'      */

static void
diff_prepare_row (GimpPixelRgn *pixel_rgn,
              guchar    *data,
              int        x,
              int        y,
              int        w)
{
  int b;

  if (y == 0)    /* wrap around */
    gimp_pixel_rgn_get_row (pixel_rgn, data, x, (pixel_rgn->h - 1), w);
  else if (y == pixel_rgn->h)
    gimp_pixel_rgn_get_row (pixel_rgn, data, x, 1, w);
  else
    gimp_pixel_rgn_get_row (pixel_rgn, data, x, y, w);

  /*  Fill in edge pixels  */
  for (b = 0; b < pixel_rgn->bpp; b++)
    {
      data[-pixel_rgn->bpp + b] = data[b];
      data[w * pixel_rgn->bpp + b] = data[(w - 1) * pixel_rgn->bpp + b];
    }
}

/* -------------------------------------------------------------------------- */
/*  'diff' combines the input drawables to prepare the two                    */
/*  16-bit (X,Y) vector displacement maps                                     */
/* -------------------------------------------------------------------------- */

static void
diff (GimpDrawable *drawable,
      gint32    *xl_id,
      gint32    *yl_id)
{
  GimpDrawable *draw_xd, *draw_yd;  /* vector disp. drawables */
  GimpDrawable *mdraw, *vdraw, *gdraw;
  gint32 image_id;           /* image holding X and Y diff. arrays */
  gint32 new_image_id;       /* image holding X and Y diff. layers */
  gint32 layer_active;       /* currently active layer */
  gint32 xlayer_id, ylayer_id;   /* individual X and Y layer ID numbers */
  GimpPixelRgn srcPR, destxPR, destyPR;
  GimpPixelRgn vecPR, magPR, gradPR;
  gint width, height;
  gint src_bytes;
  gint mbytes=0;
  gint vbytes=0;
  gint gbytes=0;   /* bytes-per-pixel of various source drawables */
  gint dest_bytes;
  gint dest_bytes_inc;
  gint do_gradmap = FALSE;          /* whether to add in gradient of gradmap to final diff. map */
  gint do_vecmap = FALSE;          /* whether to add in a fixed vector scaled by the vector map */
  gint do_magmap = FALSE;          /* whether to multiply result by the magnitude map */

  guchar *destx, *dx, *desty, *dy;  /* pointers to rows of X and Y diff. data */
  guchar *tmp;
  guchar *prev_row, *pr;
  guchar *cur_row, *cr;
  guchar *next_row, *nr;
  guchar *prev_row_g, *prg=NULL;          /* pointers to gradient map data */
  guchar *cur_row_g, *crg=NULL;
  guchar *next_row_g, *nrg=NULL;
  guchar *cur_row_v, *crv=NULL;      /* pointers to vector map data */
  guchar *cur_row_m, *crm=NULL;      /* pointers to magnitude map data */
  gint row, col, offb, off, bytes;   /* relating to indexing into pixel row arrays */
  gint x1, y1, x2, y2;
  gint dvalx, dvaly;                  /* differential value at particular pixel */
  gdouble tx, ty;                     /* temporary x,y differential value increments from gradmap, etc. */
  gdouble rdx, rdy;                     /* x,y differential values: real #s */
  gdouble rscalefac;                  /* scaling factor for x,y differential of 'curl' map */
  gdouble gscalefac;                  /* scaling factor for x,y differential of 'gradient' map */
  gdouble r, theta, dtheta;           /* rectangular<-> spherical coordinate transform for vector rotation */
  gdouble scale_vec_x, scale_vec_y;   /* fixed vector X,Y component scaling factors */
  gint has_alpha, ind;

  /* ----------------------------------------------------------------------- */

  if (dvals.grad_scale != 0.0)
      do_gradmap = TRUE;    /* add in gradient of gradmap if scale != 0.000 */
  if (dvals.vector_scale != 0.0)   /* add in gradient of vectormap if scale != 0.000 */
      do_vecmap = TRUE;
  do_magmap = (dvals.mag_use == TRUE);   /* multiply by magnitude map if so requested */

  /* Get the input area. This is the bounding box of the selection in
   *  the image (or the entire image if there is no selection). Only
   *  operating on the input area is simply an optimization. It doesn't
   *  need to be done for correct operation. (It simply makes it go
   *  faster, since fewer pixels need to be operated on).
   */
  gimp_drawable_mask_bounds (drawable->drawable_id, &x1, &y1, &x2, &y2);

  /* Get the size of the input image. (This will/must be the same
   *  as the size of the output image.
   */
  width     = drawable->width;
  height    = drawable->height;
  src_bytes = drawable->bpp;
  has_alpha = gimp_drawable_has_alpha (drawable->drawable_id);

  /* -- Add two layers: X and Y Displacement vectors -- */
  /* -- I'm using a RGB  drawable and using the first two bytes for a
        16-bit pixel value. This is either clever, or a kluge,
        depending on your point of view.  */

  image_id = gimp_drawable_get_image (drawable->drawable_id);
  layer_active = gimp_image_get_active_layer (image_id);

  new_image_id = gimp_image_new (width, height, GIMP_RGB); /* create new image for X,Y diff */

  xlayer_id = gimp_layer_new (new_image_id, "Warp_X_Vectors",
                              width, height,
                              GIMP_RGB_IMAGE, 100.0, GIMP_NORMAL_MODE);

  ylayer_id = gimp_layer_new (new_image_id, "Warp_Y_Vectors",
                              width, height,
                              GIMP_RGB_IMAGE, 100.0, GIMP_NORMAL_MODE);

  draw_yd = gimp_drawable_get (ylayer_id);
  draw_xd = gimp_drawable_get (xlayer_id);

  gimp_image_add_layer (new_image_id, xlayer_id, 1);
  gimp_image_add_layer (new_image_id, ylayer_id, 1);
  gimp_drawable_fill (xlayer_id, GIMP_BACKGROUND_FILL);
  gimp_drawable_fill (ylayer_id, GIMP_BACKGROUND_FILL);
  gimp_image_set_active_layer (image_id, layer_active);

  dest_bytes = draw_xd->bpp;
  /* for a GRAYA drawable, I would expect this to be two bytes; any more would be excess */
  dest_bytes_inc = dest_bytes - 2;

  /*  allocate row buffers for source & dest. data  */

  prev_row = g_new (guchar, (x2 - x1 + 2) * src_bytes);
  cur_row  = g_new (guchar, (x2 - x1 + 2) * src_bytes);
  next_row = g_new (guchar, (x2 - x1 + 2) * src_bytes);

  prev_row_g = g_new (guchar, (x2 - x1 + 2) * src_bytes);
  cur_row_g  = g_new (guchar, (x2 - x1 + 2) * src_bytes);
  next_row_g = g_new (guchar, (x2 - x1 + 2) * src_bytes);

  cur_row_v = g_new (guchar, (x2 - x1 + 2) * src_bytes);  /* vector map */
  cur_row_m = g_new (guchar, (x2 - x1 + 2) * src_bytes);  /* magnitude map */

  destx = g_new (guchar, (x2 - x1) * dest_bytes);
  desty = g_new (guchar, (x2 - x1) * dest_bytes);

  if ((desty==NULL) || (destx==NULL) || (cur_row_m==NULL) || (cur_row_v==NULL)
      || (next_row_g==NULL) || (cur_row_g==NULL) || (prev_row_g==NULL)
      || (next_row==NULL) || (cur_row==NULL) || (prev_row==NULL))
    {
      g_printerr ("Warp diff: error allocating memory.\n");
      exit (EXIT_FAILURE);
    }

  /*  initialize the source and destination pixel regions  */
  gimp_pixel_rgn_init (&srcPR, drawable, 0, 0, width, height, FALSE, FALSE);  /* 'curl' vector-rotation input */
  gimp_pixel_rgn_init (&destxPR, draw_xd, 0, 0, width, height, TRUE, FALSE);  /* destination: X diff output */
  gimp_pixel_rgn_init (&destyPR, draw_yd, 0, 0, width, height, TRUE, FALSE);  /* Y diff output */

  pr = prev_row + src_bytes;
  cr = cur_row + src_bytes;
  nr = next_row + src_bytes;

  diff_prepare_row (&srcPR, pr, x1, y1, (x2 - x1));
  diff_prepare_row (&srcPR, cr, x1, y1+1, (x2 - x1));

 /* fixed-vector (x,y) component scale factors */
  scale_vec_x = dvals.vector_scale*cos((90-dvals.vector_angle)*G_PI/180.0)*256.0/10;
  scale_vec_y = dvals.vector_scale*sin((90-dvals.vector_angle)*G_PI/180.0)*256.0/10;

  if (do_vecmap) {
    /*    fprintf(stderr,"%f %f  x,y vector components.\n",scale_vec_x,scale_vec_y); */

    vdraw = gimp_drawable_get(dvals.vector_map);
    vbytes = vdraw->bpp;                /* bytes per pixel in SOURCE drawable */
    gimp_pixel_rgn_init (&vecPR, vdraw, 0, 0, width, height, FALSE, FALSE);          /* fixed-vector scale-map */
    crv = cur_row_v + vbytes;
    diff_prepare_row (&vecPR, crv, x1, y1, (x2 - x1));
  }
  if (do_gradmap) {
    gdraw = gimp_drawable_get(dvals.grad_map);
    gbytes = gdraw->bpp;
    gimp_pixel_rgn_init (&gradPR, gdraw, 0, 0, width, height, FALSE, FALSE);          /* fixed-vector scale-map */
    prg = prev_row_g + gbytes;
    crg = cur_row_g + gbytes;
    nrg = next_row_g + gbytes;
    diff_prepare_row (&gradPR, prg, x1, y1 - 1, (x2 - x1));
    diff_prepare_row (&gradPR, crg, x1, y1, (x2 - x1));
  }
  if (do_magmap) {
    mdraw = gimp_drawable_get(dvals.mag_map);
    mbytes = mdraw->bpp;
    gimp_pixel_rgn_init (&magPR, mdraw, 0, 0, width, height, FALSE, FALSE);          /* fixed-vector scale-map */
    crm = cur_row_m + mbytes;
    diff_prepare_row (&magPR, crm, x1, y1, (x2 - x1));
  }

  dtheta = dvals.angle * G_PI / 180.0;
  rscalefac = 256.0 / (3*src_bytes);         /* note that '3' is rather arbitrary here. */
  gscalefac = dvals.grad_scale* 256.0 / (3*gbytes);            /* scale factor for gradient map components */

  /*  loop through the rows, applying the differential convolution  */
  for (row = y1; row < y2; row++)
    {
      /*  prepare the next row  */
      diff_prepare_row (&srcPR, nr, x1, row + 1, (x2 - x1));

      if (do_magmap)
        diff_prepare_row (&magPR, crm, x1, row + 1, (x2 - x1));
      if (do_vecmap)
        diff_prepare_row (&vecPR, crv, x1, row + 1, (x2 - x1));
      if (do_gradmap)
        diff_prepare_row (&gradPR, crg, x1, row + 1, (x2 - x1));

      dx = destx;
      dy = desty;
      ind = 0;

      for (col = 0; col < (x2 - x1); col++) /* over columns of pixels */
        {
           rdx = 0.0;
           rdy = 0.0;
           ty = 0.0;
           tx = 0.0;

           offb = col*src_bytes;    /* base of byte pointer offset */
           for (bytes=0; bytes < src_bytes; bytes++) /* add all channels together */
             {
              off = offb+bytes;                 /* offset into row arrays */
              rdx += ((gint) -pr[off - src_bytes]   + (gint) pr[off + src_bytes] +
                    (gint) -2*cr[off - src_bytes] + (gint) 2*cr[off + src_bytes] +
                    (gint) -nr[off - src_bytes]   + (gint) nr[off + src_bytes]);

              rdy += ((gint) -pr[off - src_bytes] - (gint)2*pr[off] - (gint) pr[off + src_bytes] +
                    (gint) nr[off - src_bytes] + (gint)2*nr[off] + (gint) nr[off + src_bytes]);
             }

           rdx *= rscalefac;   /* take average, then reduce. Assume max. rdx now 65535 */
           rdy *= rscalefac;   /* take average, then reduce */

           theta = atan2(rdy,rdx);          /* convert to polar, then back to rectang. coords */
           r = sqrt(rdy*rdy + rdx*rdx);
           theta += dtheta;              /* rotate gradient vector by this angle (radians) */
           rdx = r * cos(theta);
           rdy = r * sin(theta);

             if (do_gradmap) {

             offb = col*gbytes;     /* base of byte pointer offset into pixel values (R,G,B,Alpha, etc.) */
             for (bytes=0; bytes < src_bytes; bytes++) /* add all channels together */
              {
              off = offb+bytes;                 /* offset into row arrays */
              tx += ((gint) -prg[off - gbytes]   + (gint) prg[off + gbytes] +
                    (gint) -2*crg[off - gbytes] + (gint) 2*crg[off + gbytes] +
                    (gint) -nrg[off - gbytes]   + (gint) nrg[off + gbytes]);

              ty += ((gint) -prg[off - gbytes] - (gint)2*prg[off] - (gint) prg[off + gbytes] +
                    (gint) nrg[off - gbytes] + (gint)2*nrg[off] + (gint) nrg[off + gbytes]);
              }
             tx *= gscalefac;
             ty *= gscalefac;

             rdx += tx;         /* add gradient component in to the other one */
             rdy += ty;

             } /* if (do_gradmap) */


           if (do_vecmap) {  /* add in fixed vector scaled by  vec. map data */
             tx = (gdouble) crv[col*vbytes];       /* use first byte only */
             rdx += scale_vec_x * tx;
             rdy += scale_vec_y * tx;

           } /* if (do_vecmap) */

           if (do_magmap) {  /* multiply result by mag. map data */
             tx = (gdouble) crm[col*mbytes];
             rdx = (rdx * tx)/(255.0);
             rdy = (rdy * tx)/(255.0);

           } /* if do_magmap */


           dvalx = rdx + (2<<14);         /* take zero point to be 2^15, since this is two bytes */
           dvaly = rdy + (2<<14);

             if (dvalx<0) dvalx=0;
           if (dvalx>65535) dvalx=65535;
           *dx++ = (guchar) (dvalx >> 8);    /* store high order byte in value channel */
             *dx++ = (guchar) (dvalx % 256);   /* store low order byte in alpha channel */
           dx += dest_bytes_inc;       /* move data pointer on to next destination pixel */

           if (dvaly<0) dvaly=0;
           if (dvaly>65535) dvaly=65535;
           *dy++ = (guchar) (dvaly >> 8);
             *dy++ = (guchar) (dvaly % 256);
           dy += dest_bytes_inc;

        } /* ------------------------------- for (col...) ----------------  */

      /*  store the dest  */
      gimp_pixel_rgn_set_row (&destxPR, destx, x1, row, (x2 - x1));
      gimp_pixel_rgn_set_row (&destyPR, desty, x1, row, (x2 - x1));

      /*  swap around the pointers to row buffers  */
      tmp = pr;
      pr = cr;
      cr = nr;
      nr = tmp;

      if (do_gradmap) {
        tmp = prg;
        prg = crg;
        crg = nrg;
        nrg = tmp;
      }

      if ((row % 5) == 0)
      gimp_progress_update ((double) row / (double) (y2 - y1));

    } /* for (row..) */

  /*  update the region  */
  gimp_drawable_flush (draw_xd);
  gimp_drawable_flush (draw_yd);

  gimp_drawable_update (draw_xd->drawable_id, x1, y1, (x2 - x1), (y2 - y1));
  gimp_drawable_update (draw_yd->drawable_id, x1, y1, (x2 - x1), (y2 - y1));

  /*
  if (display_diff_map) {
    gimp_display_new(new_image_id);
  }
  */

  gimp_displays_flush();  /* make sure layer is visible */

  gimp_progress_init (_("Smoothing X gradient..."));
  blur16(draw_xd);
  gimp_progress_init (_("Smoothing Y gradient..."));
  blur16(draw_yd);

  g_free (prev_row);  /* row buffers allocated at top of fn. */
  g_free (cur_row);
  g_free (next_row);
  g_free (prev_row_g);  /* row buffers allocated at top of fn. */
  g_free (cur_row_g);
  g_free (next_row_g);
  g_free (cur_row_v);
  g_free (cur_row_m);

  g_free (destx);
  g_free (desty);

  *xl_id = xlayer_id;  /* pass back the X and Y layer ID numbers */
  *yl_id = ylayer_id;

} /* end diff() */

/* -------------------------------------------------------------------------- */
/*            The Warp displacement is done here.                             */
/* -------------------------------------------------------------------------- */

static void
warp (GimpDrawable  *orig_draw,
      GimpDrawable **map_x,
      GimpDrawable **map_y)
{
  GimpDrawable *disp_map;    /* Displacement map, ie, control array */
  GimpDrawable *mag_draw;    /* Magnitude multiplier factor map */

  gchar *string;          /* string to hold title of progress bar window */

  gint    first_time = TRUE;
  gint    width;
  gint    height;
  gint    bytes;
  gint orig_image_id;
  gint image_type;


  gint    x1, y1, x2, y2;

  gint32 xdlayer = -1;
  gint32 ydlayer = -1;

  gint warp_iter;      /* index var. over all "warp" Displacement iterations */


  disp_map = gimp_drawable_get(dvals.warp_map);
  mag_draw = gimp_drawable_get(dvals.mag_map);

  /* calculate new X,Y Displacement image maps */

  gimp_progress_init (_("Finding XY gradient..."));

  diff (disp_map, &xdlayer, &ydlayer);    /* generate x,y differential images (arrays) */

  /* Get selection area */
  gimp_drawable_mask_bounds (orig_draw->drawable_id, &x1, &y1, &x2, &y2);

  width  = orig_draw->width;
  height = orig_draw->height;
  bytes  = orig_draw->bpp;
  image_type = gimp_drawable_type (orig_draw->drawable_id);

  *map_x = gimp_drawable_get (xdlayer);
  *map_y = gimp_drawable_get (ydlayer);

  orig_image_id = gimp_drawable_get_image (orig_draw->drawable_id);

  /*   gimp_image_lower_layer(orig_image_id, new_layer_id); */ /* hide it! */

  /*   gimp_layer_set_opacity(new_layer_id, 0.0); */

  for (warp_iter = 0; warp_iter < dvals.iter; warp_iter++)
    {
      string = g_strdup_printf (_("Flow Step %d..."), warp_iter+1);
      gimp_progress_init (string);
      g_free (string);
      progress = 0;
      gimp_progress_update (0);

      warp_one (orig_draw, orig_draw,
                *map_x, *map_y, mag_draw, first_time, warp_iter);

      gimp_drawable_update (orig_draw->drawable_id,
                            x1, y1, (x2 - x1), (y2 - y1));

      if (run_mode != GIMP_RUN_NONINTERACTIVE)
        gimp_displays_flush ();

      first_time = FALSE;
    }

   /* gimp_image_add_layer (orig_image_id, new_layer_id, 1); */  /* make layer visible in 'layers' dialog */

} /* Warp */

/* -------------------------------------------------------------------------- */

static void
warp_one (GimpDrawable *draw,
        GimpDrawable *new,
        GimpDrawable *map_x,
        GimpDrawable *map_y,
        GimpDrawable *mag_draw,
        gint       first_time,
        gint       step)
{
  GimpPixelRgn src_rgn;
  GimpPixelRgn dest_rgn;
  GimpPixelRgn map_x_rgn;
  GimpPixelRgn map_y_rgn;
  GimpPixelRgn mag_rgn;
  GimpTile   * tile = NULL;
  GimpTile   * xtile = NULL;
  GimpTile   * ytile = NULL;
  gint row=-1;
  gint xrow=-1;
  gint yrow=-1;
  gint col=-1;
  gint xcol=-1;
  gint ycol=-1;

  gpointer  pr;

  gint    width = -1;
  gint    height = -1;
  gint    dest_bytes=-1;
  gint    dmap_bytes=-1;

  guchar *destrow, *dest;
  guchar *srcrow, *src;
  guchar *mxrow=NULL, *mx;  /* NULL ptr. to make gcc's -Wall fn. happy */
  guchar *myrow=NULL, *my;

  guchar *mmagrow=NULL, *mmag=NULL;
  guchar  pixel[4][4];
  gint    x1, y1, x2, y2;
  gint    x, y;
  gint    max_progress;

  gdouble needx, needy;
  gdouble xval=0;      /* initialize to quiet compiler grumbles */
  gdouble yval=0;      /* interpolated vector displacement */
  gdouble scalefac;        /* multiplier for vector displacement scaling */
  gdouble dscalefac;       /* multiplier for incremental displacement vectors */
  gint    xi, yi;
  gint    substep;         /* loop variable counting displacement vector substeps */

  guchar  values[4];
  guint32 ivalues[4];
  guchar  val;

  gint k;

  gdouble dx, dy;           /* X and Y Displacement, integer from GRAY map */

  gint    xm_alpha = 0;
  gint    ym_alpha = 0;
  gint    mmag_alpha = 0;
  gint    xm_bytes = 1;
  gint    ym_bytes = 1;
  gint    mmag_bytes = 1;

  GRand  *gr;

  gr = g_rand_new (); /* Seed Pseudo Random Number Generator */

  /* ================ Outer Loop calculation ================================ */

  /* Get selection area */

   gimp_drawable_mask_bounds (draw->drawable_id, &x1, &y1, &x2, &y2);
   width  = draw->width;
   height = draw->height;
   dest_bytes  = draw->bpp;

   dmap_bytes = map_x->bpp;

   max_progress = (x2 - x1) * (y2 - y1);


   /*  --------- Register the (many) pixel regions ----------  */

   gimp_pixel_rgn_init (&src_rgn, draw, x1, y1, (x2 - x1), (y2 - y1), FALSE, FALSE);

   /* only push undo-stack the first time through. Thanks Spencer! */
   if (first_time==TRUE)
     gimp_pixel_rgn_init (&dest_rgn, new, x1, y1, (x2 - x1), (y2 - y1), TRUE, TRUE);
   else
     /*     gimp_pixel_rgn_init (&dest_rgn, new, x1, y1, (x2 - x1), (y2 - y1), TRUE, FALSE); */
     gimp_pixel_rgn_init (&dest_rgn, new, x1, y1, (x2 - x1), (y2 - y1), TRUE, TRUE);


   gimp_pixel_rgn_init (&map_x_rgn, map_x, x1, y1, (x2 - x1), (y2 - y1), FALSE, FALSE);
   if (gimp_drawable_has_alpha(map_x->drawable_id))
      xm_alpha = 1;
   xm_bytes = gimp_drawable_bpp(map_x->drawable_id);

   gimp_pixel_rgn_init (&map_y_rgn, map_y, x1, y1, (x2 - x1), (y2 - y1), FALSE, FALSE);
   if (gimp_drawable_has_alpha(map_y->drawable_id))
      ym_alpha = 1;
   ym_bytes = gimp_drawable_bpp(map_y->drawable_id);


   if (dvals.mag_use == TRUE) {
     gimp_pixel_rgn_init (&mag_rgn, mag_draw, x1, y1, (x2 - x1), (y2 - y1), FALSE, FALSE);
     if (gimp_drawable_has_alpha(mag_draw->drawable_id))
       mmag_alpha = 1;
     mmag_bytes = gimp_drawable_bpp(mag_draw->drawable_id);

     pr = gimp_pixel_rgns_register (5, &src_rgn, &dest_rgn, &map_x_rgn, &map_y_rgn, &mag_rgn);
   } else {
     pr = gimp_pixel_rgns_register (4, &src_rgn, &dest_rgn, &map_x_rgn, &map_y_rgn);
   }

   dscalefac = (dvals.amount) / (256* 127.5 * dvals.substeps);  /* substep displacement vector scale factor */


   for (pr = pr; pr != NULL; pr = gimp_pixel_rgns_process (pr))
    {

      srcrow = src_rgn.data;
      destrow = dest_rgn.data;
      mxrow = map_x_rgn.data;
      myrow = map_y_rgn.data;
      if (dvals.mag_use == TRUE)
      mmagrow = mag_rgn.data;

      /* loop over destination pixels */
      for (y = dest_rgn.y; y < (dest_rgn.y + dest_rgn.h); y++)
      {
        src = srcrow;
        dest = destrow;
        mx = mxrow;
        my = myrow;

          if (dvals.mag_use == TRUE)
          mmag = mmagrow;

        for (x = dest_rgn.x; x < (dest_rgn.x + dest_rgn.w); x++)
          {
            /* ----- Find displacement vector (amnt_x, amnt_y) ------------ */

              dx = dscalefac * ((256.0*mx[0])+mx[1] -32768);  /* 16-bit values */
            dy = dscalefac * ((256.0*my[0])+my[1] -32768);

            if (dvals.mag_use == TRUE) {
            scalefac = warp_map_mag_give_value(mmag, mmag_alpha, mmag_bytes)/255.0;
            dx *= scalefac;
            dy *= scalefac;
            }

            if (dvals.dither != 0.0) {       /* random dither is +/- dvals.dither pixels */
            dx += g_rand_double_range (gr, -dvals.dither, dvals.dither);
            dy += g_rand_double_range (gr, -dvals.dither, dvals.dither);
            }

            if (dvals.substeps != 1) {   /* trace (substeps) iterations of displacement vector */

            for (substep = 1; substep < dvals.substeps; substep++) {

                needx = x + dx;   /* In this (substep) loop, (x,y) remain fixed. (dx,dy) vary each step. */
                needy = y + dy;

                if (needx >= 0.0)    xi = (int) needx;
                else           xi = -((int) -needx + 1);

              if (needy >= 0.0)    yi = (int) needy;
                else               yi = -((int) -needy + 1);

                 /* get 4 neighboring DX values from DiffX drawable for linear interpolation */
                xtile = warp_pixel (map_x, xtile, width, height, x1, y1, x2, y2, xi, yi, &xrow, &xcol, pixel[0]);
                xtile = warp_pixel (map_x, xtile, width, height, x1, y1, x2, y2, xi + 1, yi, &xrow, &xcol, pixel[1]);
                xtile = warp_pixel (map_x, xtile, width, height, x1, y1, x2, y2, xi, yi + 1, &xrow, &xcol, pixel[2]);
                xtile = warp_pixel (map_x, xtile, width, height, x1, y1, x2, y2, xi + 1, yi + 1, &xrow, &xcol, pixel[3]);

                ivalues[0] = 256*pixel[0][0] + pixel[0][1];
              ivalues[1] = 256*pixel[1][0] + pixel[1][1];
              ivalues[2] = 256*pixel[2][0] + pixel[2][1];
              ivalues[3] = 256*pixel[3][0] + pixel[3][1];
              xval = gimp_bilinear_32(needx, needy, ivalues);

                 /* get 4 neighboring DY values from DiffY drawable for linear interpolation */
                ytile = warp_pixel (map_y, ytile, width, height, x1, y1, x2, y2, xi, yi, &yrow, &ycol, pixel[0]);
                ytile = warp_pixel (map_y, ytile, width, height, x1, y1, x2, y2, xi + 1, yi, &yrow, &ycol, pixel[1]);
                ytile = warp_pixel (map_y, ytile, width, height, x1, y1, x2, y2, xi, yi + 1, &yrow, &ycol, pixel[2]);
                ytile = warp_pixel (map_y, ytile, width, height, x1, y1, x2, y2, xi + 1, yi + 1, &yrow, &ycol, pixel[3]);

                ivalues[0] = 256*pixel[0][0] + pixel[0][1];
              ivalues[1] = 256*pixel[1][0] + pixel[1][1];
              ivalues[2] = 256*pixel[2][0] + pixel[2][1];
              ivalues[3] = 256*pixel[3][0] + pixel[3][1];
              yval = gimp_bilinear_32(needx, needy, ivalues);

              dx += dscalefac * (xval-32768);      /* move displacement vector to this new value */
              dy += dscalefac * (yval-32768);

            } /* for (substep) */
            } /* if (substeps != 0) */


            /* --------------------------------------------------------- */

            needx = x + dx;
            needy = y + dy;

            mx += xm_bytes;         /* pointers into x,y displacement maps */
            my += ym_bytes;

            if (dvals.mag_use == TRUE)
            mmag += mmag_bytes;

            /* Calculations complete; now copy the proper pixel */

            if (needx >= 0.0)
                  xi = (int) needx;
            else
                  xi = -((int) -needx + 1);

            if (needy >= 0.0)
                  yi = (int) needy;
            else
                  yi = -((int) -needy + 1);

            /* get 4 neighboring pixel values from source drawable for linear interpolation */
            tile = warp_pixel (draw, tile, width, height, x1, y1, x2, y2, xi, yi, &row, &col, pixel[0]);
            tile = warp_pixel (draw, tile, width, height, x1, y1, x2, y2, xi + 1, yi, &row, &col, pixel[1]);
            tile = warp_pixel (draw, tile, width, height, x1, y1, x2, y2, xi, yi + 1, &row, &col, pixel[2]);
            tile = warp_pixel (draw, tile, width, height, x1, y1, x2, y2, xi + 1, yi + 1, &row, &col, pixel[3]);

            for (k = 0; k < dest_bytes; k++)
            {
              values[0] = pixel[0][k];
              values[1] = pixel[1][k];
              values[2] = pixel[2][k];
              values[3] = pixel[3][k];
              val = gimp_bilinear_8(needx, needy, values);

              *dest++ = val;
            } /* for k */

          } /* for x */

        /*    srcrow += src_rgn.rowstride; */
        srcrow += src_rgn.rowstride;
        destrow += dest_rgn.rowstride;
        mxrow += map_x_rgn.rowstride;
        myrow += map_y_rgn.rowstride;
          if (dvals.mag_use == TRUE)
          mmagrow += mag_rgn.rowstride;

      } /* for y */

      progress += (dest_rgn.w * dest_rgn.h);
      gimp_progress_update ((double) progress / (double) max_progress);


    } /* for pr */

   if (tile != NULL)
    gimp_tile_unref (tile, FALSE);

   if (xtile != NULL)
    gimp_tile_unref (xtile, FALSE);

   if (ytile != NULL)
    gimp_tile_unref (ytile, FALSE);

     /*  update the region  */
   gimp_drawable_flush (new);

   gimp_drawable_merge_shadow(draw->drawable_id, (first_time == TRUE));

   g_rand_free (gr);

} /* warp_one */

/* ------------------------------------------------------------------------- */

static gdouble
warp_map_mag_give_value (guchar *pt,
                   gint    alpha,
                   gint    bytes)
{
  gdouble ret, val_alpha;

  if (bytes >= 3)
    ret =  (pt[0] + pt[1] + pt[2])/3.0;
  else
    ret = (gdouble) *pt;

  if (alpha)
    {
      val_alpha = pt[bytes - 1];
      ret = (ret * val_alpha / 255.0);
    };

  return (ret);
}


static GimpTile *
warp_pixel (GimpDrawable *drawable,
          GimpTile     *tile,
          gint       width,
          gint       height,
          gint       x1,
          gint       y1,
          gint       x2,
          gint       y2,
          gint       x,
          gint       y,
          gint      *row,
          gint      *col,
          guchar    *pixel)
{
  static guchar empty_pixel[4] = {0, 0, 0, 0};
  guchar *data;
  gint b;

  /* Tile the image. */
  if (dvals.wrap_type == WRAP)
    {
      if (x < 0)
      x = width - (-x % width);
      else
      x %= width;

      if (y < 0)
      y = height - (-y % height);
      else
      y %= height;
    }
  /* Smear out the edges of the image by repeating pixels. */
  else if (dvals.wrap_type == SMEAR)
    {
      if (x < 0)
      x = 0;
      else if (x > width - 1)
      x = width - 1;

      if (y < 0)
      y = 0;
      else if (y > height - 1)
      y = height - 1;
    }

  if (x >= x1 && y >= y1 && x < x2 && y < y2)
    {
      if ((( (guint) (x / tile_width)) != *col) || (( (guint) (y / tile_height)) != *row))
      {
        *col = x / tile_width;
        *row = y / tile_height;
        if (tile)
          gimp_tile_unref (tile, FALSE);
        tile = gimp_drawable_get_tile (drawable, FALSE, *row, *col);
        gimp_tile_ref (tile);
      }

      data = tile->data + tile->bpp * (tile->ewidth * (y % tile_height) + (x % tile_width));
    }
  else
    {
      if (dvals.wrap_type == BLACK)
        data = empty_pixel;
      else
        data = color_pixel;      /* must have selected COLOR type */
    }

  for (b = 0; b < drawable->bpp; b++)
    pixel[b] = data[b];

  return tile;
}

/*  Warp interface functions  */

static gboolean
warp_map_constrain (gint32     image_id,
                gint32     drawable_id,
                gpointer   data)
{
  GimpDrawable *drawable = data;

  return (gimp_drawable_width (drawable_id)  == drawable->width &&
        gimp_drawable_height (drawable_id) == drawable->height);
}

Generated by  Doxygen 1.6.0   Back to index