/****************************************************************************
 * This is a plugin for the GIMP v 0.99.8 or later.  Documentation is
 * available at http://www.rru.com/~meo/gimp/ .
 *
 * Copyright (C) 1997-98 Miles O'Neal  <meo@rru.com>  http://www.rru.com/~meo/
 * Blur code Copyright (C) 1995 Spencer Kimball and Peter Mattis
 * GUI based on GTK code from:
 *    alienmap (Copyright (C) 1996, 1997 Daniel Cotting)
 *    plasma   (Copyright (C) 1996 Stephen Norris),
 *    oilify   (Copyright (C) 1996 Torsten Martinsen),
 *    ripple   (Copyright (C) 1997 Brian Degenhardt) and
 *    whirl    (Copyright (C) 1997 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 *
 ****************************************************************************/

/****************************************************************************
 * Blur:
 *
 * blur version 2.0 (1 May 1998, MEO)
 * history
 *     2.0 -  1 May 1998 MEO
 *         based on randomize 1.7
 *
 * Please send any patches or suggestions to the author: meo@rru.com .
 *
 * Blur applies a 3x3 blurring convolution kernel to the specified drawable.
 *
 * For each pixel in the selection or image,
 * whether to change the pixel is decided by picking a
 * random number, weighted by the user's "randomization" percentage.
 * If the random number is in range, the pixel is modified.  For
 * blurring, an average is determined from the current and adjacent
 * pixels. *(Except for the random factor, the blur code came
 * straight from the original S&P blur plug-in.)*
 *
 * This works only with RGB and grayscale images.
 *
 ****************************************************************************/

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

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

#include <gtk/gtk.h>

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

#include "libgimp/stdplugins-intl.h"

/*********************************
 *
 *  PLUGIN-SPECIFIC CONSTANTS
 *
 ********************************/

/*
 *  progress meter update frequency
 */
#define PROG_UPDATE_TIME ((row % 10) == 0)

#define PLUG_IN_NAME "plug_in_blur"
#define HELP_ID      "plug-in-blur"

#define SEED_DEFAULT 10
#define SEED_USER    11

#define SCALE_WIDTH  64


/*********************************
 *
 *  PLUGIN-SPECIFIC STRUCTURES AND DATA
 *
 ********************************/

typedef struct
{
  gdouble  blur_pct;       /* likelihood of randomization (as %age) */
  gdouble  blur_rcount;    /* repeat count */
  gboolean blur_randomize; /* Generate a random seed value */
  guint    blur_seed;      /* seed value for g_random_set_seed() function */
} BlurVals;

static BlurVals pivals =
{
  100.0,
  1.0,
  FALSE,
  0
};


/*********************************
 *
 *  LOCAL FUNCTIONS
 *
 ********************************/

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

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

static void         blur             (GimpDrawable *drawable);

static inline void  blur_prepare_row (GimpPixelRgn *pixel_rgn,
                                      guchar       *data,
                                      gint          x,
                                      gint          y,
                                      gint          w);

static gboolean     blur_dialog      (void);

/************************************ Guts ***********************************/

MAIN ()

/*********************************
 *
 *  query() - query_proc
 *
 *      called by the GIMP to learn about this plug-in
 *
 ********************************/

static void
query (void)
{
  static GimpParamDef args_ni[] =
  {
    { GIMP_PDB_INT32, "run_mode", "non-interactive" },
    { GIMP_PDB_IMAGE, "image", "Input image (unused)" },
    { GIMP_PDB_DRAWABLE, "drawable", "Input drawable" }
  };

  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, "blur_pct", "Randomization percentage (1 - 100)" },
    { GIMP_PDB_FLOAT, "blur_rcount", "Repeat count(1 - 100)" },
    { GIMP_PDB_INT32, "randomize", "Use a random seed (TRUE, FALSE)" },
    { GIMP_PDB_INT32, "seed", "Seed value (used only if randomize is FALSE)" }
  };

  const gchar *blurb = "Apply a 3x3 blurring convolution kernel to the specified drawable.";
  const gchar *help = "This plug-in randomly blurs the specified drawable, using a 3x3 blur.  You control the percentage of the pixels that are blurred and the number of times blurring is applied.  Indexed images are not supported.";
  const gchar *author = "Miles O'Neal  <meo@rru.com>";
  const gchar *copyrights = "Miles O'Neal, Spencer Kimball, Peter Mattis, Torsten Martinsen, Brian Degenhardt, Federico Mena Quintero, Stephen Norris, Daniel Cotting";
  const gchar *copyright_date = "1995-1998";

  gimp_install_procedure ("plug_in_blur_randomize",
                          blurb,
                          help,
                          author,
                          copyrights,
                          copyright_date,
                          N_("_Blur..."),
                          "RGB*, GRAY*",
                          GIMP_PLUGIN,
                          G_N_ELEMENTS (args), 0,
                          args, NULL);

  gimp_plugin_menu_register ("plug_in_blur_randomize",
                             N_("<Image>/Filters/Blur"));

  gimp_install_procedure (PLUG_IN_NAME,
                          blurb,
                          help,
                          author,
                          copyrights,
                          copyright_date,
                          NULL,
                          "RGB*, GRAY*",
                          GIMP_PLUGIN,
                          G_N_ELEMENTS (args_ni), 0,
                          args_ni, NULL);
}

/*********************************
 *
 *  run() - main routine
 *
 *  This handles the main interaction with the GIMP itself,
 *  and invokes the routine that actually does the work.
 *
 ********************************/

static void
run (const gchar      *name,
     gint              nparams,
     const GimpParam  *param,
     gint             *nreturn_vals,
     GimpParam       **return_vals)
{
  GimpDrawable      *drawable;
  GimpRunMode        run_mode;
  GimpPDBStatusType  status = GIMP_PDB_SUCCESS;        /* assume the best! */
  static GimpParam   values[1];

  INIT_I18N ();

  /*
   *  Get the specified drawable, do standard initialization.
   */
  run_mode = param[0].data.d_int32;
  drawable = gimp_drawable_get (param[2].data.d_drawable);

  values[0].type = GIMP_PDB_STATUS;
  values[0].data.d_status = status;
  *nreturn_vals = 1;
  *return_vals = values;

  /*
   *  Make sure the drawable type is appropriate.
   */
  if (gimp_drawable_is_rgb (drawable->drawable_id) ||
      gimp_drawable_is_gray (drawable->drawable_id))
    {
      switch (run_mode)
        {
          /*
           *  If we're running interactively, pop up the dialog box.
           */
        case GIMP_RUN_INTERACTIVE:
          gimp_get_data (PLUG_IN_NAME, &pivals);
          if (! blur_dialog ())        /* return on Cancel */
            return;
          break;

          /*
           *  If we're not interactive (probably scripting), we
           *  get the parameters from the param[] array, since
           *  we don't use the dialog box.  Make sure all
           *  parameters have legitimate values.
           */
        case GIMP_RUN_NONINTERACTIVE:
          if ((strcmp (name, "plug_in_blur_randomize") == 0) &&
              (nparams == 7))
            {
              pivals.blur_pct       = (gdouble) param[3].data.d_float;
              pivals.blur_pct       = (gdouble) MIN (100.0, pivals.blur_pct);
              pivals.blur_pct       = (gdouble) MAX (1.0, pivals.blur_pct);
              pivals.blur_rcount    = (gdouble) param[4].data.d_float;
              pivals.blur_rcount    = (gdouble) MIN (100.0,pivals.blur_rcount);
              pivals.blur_rcount    = (gdouble) MAX (1.0, pivals.blur_rcount);
              pivals.blur_randomize = (gboolean) param[5].data.d_int32;
              pivals.blur_seed      = (gint) param[6].data.d_int32;

              if (pivals.blur_randomize)
                pivals.blur_seed = g_random_int ();
            }
          else if ((strcmp (name, PLUG_IN_NAME) == 0) &&
                   (nparams == 3))
            {
              pivals.blur_pct       = (gdouble) 100.0;
              pivals.blur_rcount    = (gdouble) 1.0;
              pivals.blur_randomize = FALSE;
              pivals.blur_seed      = g_random_int ();
            }
          else
            {
              status = GIMP_PDB_CALLING_ERROR;
            }
          break;

          /*
           *  If we're running with the last set of values, get those values.
           */
        case GIMP_RUN_WITH_LAST_VALS:
          gimp_get_data (PLUG_IN_NAME, &pivals);

          if (pivals.blur_randomize)
            pivals.blur_seed = g_random_int ();
          break;

          /*
           *  Hopefully we never get here!
           */
        default:
          break;
        }

      if (status == GIMP_PDB_SUCCESS)
        {
          /*
           *  JUST DO IT!
           */
          gimp_progress_init (_("Blurring..."));
          gimp_tile_cache_ntiles (2 * (drawable->width / gimp_tile_width ()
                                       + 1));

          g_random_set_seed (pivals.blur_seed);

          blur (drawable);
          /*
           *  If we ran interactively (even repeating) update the display.
           */
          if (run_mode != GIMP_RUN_NONINTERACTIVE)
            {
              gimp_displays_flush ();
            }
          /*
           *  If we use the dialog popup, set the data for future use.
           */
          if (run_mode == GIMP_RUN_INTERACTIVE)
            {
              gimp_set_data (PLUG_IN_NAME, &pivals, sizeof (BlurVals));
            }
        }
    }
  else
    {
      /*
       *  If we got the wrong drawable type, we need to complain.
       */
      status = GIMP_PDB_EXECUTION_ERROR;
    }
  /*
   *  DONE!
   *  Set the status where the GIMP can see it, and let go
   *  of the drawable.
   */
  values[0].data.d_status = status;
  gimp_drawable_detach (drawable);

}

/*********************************
 *
 *  blur_prepare_row()
 *
 *  Get a row of pixels.  If the requested row
 *  is off the edge, clone the edge row.
 *
 ********************************/

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

  y = CLAMP (y, 1, pixel_rgn->h - 1);
  gimp_pixel_rgn_get_row (pixel_rgn, data, x, y, w);

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

/*********************************
 *
 *  blur()
 *
 *  Actually mess with the image.
 *
 ********************************/

static void
blur (GimpDrawable *drawable)
{
  GimpPixelRgn  srcPR, destPR, destPR2, *sp, *dp, *tp;
  gint          width, height;
  gint          bytes;
  guchar       *dest, *d;
  guchar       *prev_row, *pr;
  guchar       *cur_row, *cr;
  guchar       *next_row, *nr;
  guchar       *tmp;
  gint          row, col;
  gint          x1, y1, x2, y2;
  gint          cnt, ind;
  gint          repeat_count;
  gboolean      has_alpha;

  /*
   *  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.  Also get alpha info.
   */
  width = drawable->width;
  height = drawable->height;
  bytes = drawable->bpp;
  has_alpha = gimp_drawable_has_alpha (drawable->drawable_id);
  /*
   *  allocate row buffers
   */
  prev_row = g_new (guchar, (x2 - x1 + 2) * bytes);
  cur_row  = g_new (guchar, (x2 - x1 + 2) * bytes);
  next_row = g_new (guchar, (x2 - x1 + 2) * bytes);
  dest     = g_new (guchar, (x2 - x1) * bytes);

  /*
   *  initialize the pixel regions
   */
  gimp_pixel_rgn_init (&srcPR, drawable, 0, 0, width, height, FALSE, FALSE);
  gimp_pixel_rgn_init (&destPR, drawable, 0, 0, width, height, TRUE, TRUE);
  gimp_pixel_rgn_init (&destPR2, drawable, 0, 0, width, height, TRUE, TRUE);
  sp = &srcPR;
  dp = &destPR;
  tp = NULL;

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

  repeat_count = (gint) pivals.blur_rcount;

  for (cnt = 1; cnt <= repeat_count; cnt++)
    {
      /*
       *  prepare the first row and previous row
       */
      blur_prepare_row (sp, pr, x1, y1 - 1, (x2 - x1));
      blur_prepare_row (dp, cr, x1, y1, (x2 - x1));
      /*
       *  loop through the rows, applying the selected convolution
       */
      for (row = y1; row < y2; row++)
        {
          /*  prepare the next row  */
          blur_prepare_row (sp, nr, x1, row + 1, (x2 - x1));

          d = dest;
          ind = 0;
          for (col = 0; col < (x2 - x1) * bytes; col++)
            {
              if (g_random_int_range (0, 100) <= (gint) pivals.blur_pct)
                {
                  ind++;
                  if (ind == bytes || !has_alpha)
                    {
                      /*
                       *  If no alpha channel,
                       *   or if there is one and this is it...
                       */
                      *d++ = ((gint) pr[col - bytes] + (gint) pr[col] +
                              (gint) pr[col + bytes] +
                              (gint) cr[col - bytes] + (gint) cr[col] +
                              (gint) cr[col + bytes] +
                              (gint) nr[col - bytes] + (gint) nr[col] +
                              (gint) nr[col + bytes]) / 9;
                      ind = 0;
                    }
                  else
                    {
                      /*
                       *  otherwise we have an alpha channel,
                       *   but this is a color channel
                       */
                      *d++ = ((gint)
                              (((gdouble) (pr[col - bytes] * pr[col - ind])
                                + (gdouble) (pr[col] * pr[col + bytes - ind])
                                + (gdouble) (pr[col + bytes] * pr[col + 2*bytes - ind])
                                + (gdouble) (cr[col - bytes] * cr[col - ind])
                                + (gdouble) (cr[col] * cr[col + bytes - ind])
                                + (gdouble) (cr[col + bytes] * cr[col + 2*bytes - ind])
                                + (gdouble) (nr[col - bytes] * nr[col - ind])
                                + (gdouble) (nr[col] * nr[col + bytes - ind])
                                + (gdouble) (nr[col + bytes] * nr[col + 2*bytes - ind]))
                             / ((gdouble) pr[col - ind]
                                + (gdouble) pr[col + bytes - ind]
                                + (gdouble) pr[col + 2*bytes - ind]
                                + (gdouble) cr[col - ind]
                                + (gdouble) cr[col + bytes - ind]
                                + (gdouble) cr[col + 2*bytes - ind]
                                + (gdouble) nr[col - ind]
                                + (gdouble) nr[col + bytes - ind]
                                + (gdouble) nr[col + 2*bytes - ind])));
                    }
                  /*
                   *  Otherwise, this pixel was not selected for randomization,
                   *  so use the current value.
                   */
                }
              else
                {
                  *d++ = (gint) cr[col];
                }
            }
          /*
           *  Save the modified row, shuffle the row pointers, and every
           *  so often, update the progress meter.
           */
          gimp_pixel_rgn_set_row (dp, dest, x1, row, (x2 - x1));

          tmp = pr;
          pr = cr;
          cr = nr;
          nr = tmp;

          if (PROG_UPDATE_TIME)
            gimp_progress_update ((double) row / (double) (y2 - y1));
        }
      /*
       *  if we have more cycles to perform, swap the src and dest Pixel Regions
       */
      if (cnt < repeat_count)
        {
          if (tp != NULL)
            {
              tp = dp;
              dp = sp;
              sp = tp;
            }
          else
            {
              tp = &srcPR;
              sp = &destPR;
              dp = &destPR2;
            }
        }
    }
  gimp_progress_update ((double) 100);
  /*
   *  update the blurred 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));
  /*
   *  clean up after ourselves.
   */
  g_free (prev_row);
  g_free (cur_row);
  g_free (next_row);
  g_free (dest);
}

/*********************************
 *  GUI ROUTINES
 ********************************/

static gboolean
blur_dialog (void)
{
  GtkWidget *dlg;
  GtkWidget *table;
  GtkWidget *label;
  GtkWidget *seed_hbox;
  GtkObject *adj;
  gboolean   run;

  gimp_ui_init ("blur", FALSE);

  dlg = gimp_dialog_new (_("Blur"), "blur",
                         NULL, 0,
                         gimp_standard_help_func, HELP_ID,

                         GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL,
                         GTK_STOCK_OK,     GTK_RESPONSE_OK,

                         NULL);

  /*
   *  Parameter settings
   *
   *  First set up the basic containers, label them, etc.
   */
  table = gtk_table_new (3, 3, FALSE);
  gtk_table_set_col_spacings (GTK_TABLE (table), 6);
  gtk_table_set_row_spacings (GTK_TABLE (table), 6);
  gtk_container_set_border_width (GTK_CONTAINER (table), 12);
  gtk_box_pack_start (GTK_BOX (GTK_DIALOG(dlg)->vbox), table, TRUE, TRUE, 0);
  gtk_widget_show (table);

  /*  Random Seed  */
  seed_hbox = gimp_random_seed_new (&pivals.blur_seed, &pivals.blur_randomize);
  label = gimp_table_attach_aligned (GTK_TABLE (table), 0, 0,
                                     _("_Random Seed:"), 0.0, 0.5,
                                     seed_hbox, 2, TRUE);
  gtk_label_set_mnemonic_widget (GTK_LABEL (label),
                                 GIMP_RANDOM_SEED_SPINBUTTON (seed_hbox));

  /*
   *  Randomization percentage label & scale (1 to 100)
   */
  adj = gimp_scale_entry_new (GTK_TABLE (table), 0, 1,
                              _("R_andomization %:"), SCALE_WIDTH, 0,
                              pivals.blur_pct, 1.0, 100.0, 1.0, 10.0, 0,
                              TRUE, 0, 0,
                              _("Percentage of pixels to be filtered"), NULL);
  g_signal_connect (adj, "value_changed",
                    G_CALLBACK (gimp_double_adjustment_update),
                    &pivals.blur_pct);

  /*
   *  Repeat count label & scale (1 to 100)
   */
  adj = gimp_scale_entry_new (GTK_TABLE (table), 0, 2,
                              _("R_epeat:"), SCALE_WIDTH, 0,
                              pivals.blur_rcount, 1.0, 100.0, 1.0, 10.0, 0,
                              TRUE, 0, 0,
                              _("Number of times to apply filter"), NULL);
  g_signal_connect (adj, "value_changed",
                    G_CALLBACK (gimp_double_adjustment_update),
                    &pivals.blur_rcount);

  gtk_widget_show (dlg);

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

  gtk_widget_destroy (dlg);

  return run;
}
