/* PIKA - Photo and Image Kooker Application
 * a rebranding of The GNU Image Manipulation Program (created with heckimp)
 * A derived work which may be trivial. However, any changes may be (C)2023 by Aldercone Studio
 *
 * Original copyright, applying to most contents (license remains unchanged): 
 * 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 <gdk-pixbuf/gdk-pixbuf.h>
#include <gegl.h>

#include "libpikamath/pikamath.h"

#include "paint-types.h"

#include "core/pikaboundary.h"
#include "core/pikadrawable.h"
#include "core/pikaerror.h"
#include "core/pikacoords.h"

#include "vectors/pikastroke.h"
#include "vectors/pikavectors.h"

#include "pikapaintcore.h"
#include "pikapaintcore-stroke.h"
#include "pikapaintoptions.h"

#include "pika-intl.h"


static void pika_paint_core_stroke_emulate_dynamics (PikaCoords *coords,
                                                     gint        length);


static const PikaCoords default_coords = PIKA_COORDS_DEFAULT_VALUES;


gboolean
pika_paint_core_stroke (PikaPaintCore     *core,
                        PikaDrawable      *drawable,
                        PikaPaintOptions  *paint_options,
                        PikaCoords        *strokes,
                        gint               n_strokes,
                        gboolean           push_undo,
                        GError           **error)
{
  GList    *drawables;
  gboolean  success = FALSE;

  g_return_val_if_fail (PIKA_IS_PAINT_CORE (core), FALSE);
  g_return_val_if_fail (PIKA_IS_DRAWABLE (drawable), FALSE);
  g_return_val_if_fail (pika_item_is_attached (PIKA_ITEM (drawable)), FALSE);
  g_return_val_if_fail (PIKA_IS_PAINT_OPTIONS (paint_options), FALSE);
  g_return_val_if_fail (strokes != NULL, FALSE);
  g_return_val_if_fail (n_strokes > 0, FALSE);
  g_return_val_if_fail (error == NULL || *error == NULL, FALSE);

  drawables = g_list_prepend (NULL, drawable);

  if (pika_paint_core_start (core, drawables, paint_options, &strokes[0],
                             error))
    {
      gint i;

      core->last_coords = strokes[0];

      pika_paint_core_paint (core, drawables, paint_options,
                             PIKA_PAINT_STATE_INIT, 0);

      pika_paint_core_paint (core, drawables, paint_options,
                             PIKA_PAINT_STATE_MOTION, 0);

      for (i = 1; i < n_strokes; i++)
        {
          pika_paint_core_interpolate (core, drawables, paint_options,
                                       &strokes[i], 0);
        }

      pika_paint_core_paint (core, drawables, paint_options,
                             PIKA_PAINT_STATE_FINISH, 0);

      pika_paint_core_finish (core, drawables, push_undo);

      pika_paint_core_cleanup (core);

      success = TRUE;
    }

  g_list_free (drawables);

  return success;
}

gboolean
pika_paint_core_stroke_boundary (PikaPaintCore      *core,
                                 PikaDrawable       *drawable,
                                 PikaPaintOptions   *paint_options,
                                 gboolean            emulate_dynamics,
                                 const PikaBoundSeg *bound_segs,
                                 gint                n_bound_segs,
                                 gint                offset_x,
                                 gint                offset_y,
                                 gboolean            push_undo,
                                 GError            **error)
{
  GList        *drawables;
  PikaBoundSeg *stroke_segs;
  gint          n_stroke_segs;
  PikaCoords   *coords;
  gboolean      initialized = FALSE;
  gint          n_coords;
  gint          seg;
  gint          s;

  g_return_val_if_fail (PIKA_IS_PAINT_CORE (core), FALSE);
  g_return_val_if_fail (PIKA_IS_DRAWABLE (drawable), FALSE);
  g_return_val_if_fail (pika_item_is_attached (PIKA_ITEM (drawable)), FALSE);
  g_return_val_if_fail (PIKA_IS_PAINT_OPTIONS (paint_options), FALSE);
  g_return_val_if_fail (bound_segs != NULL && n_bound_segs > 0, FALSE);
  g_return_val_if_fail (error == NULL || *error == NULL, FALSE);

  stroke_segs = pika_boundary_sort (bound_segs, n_bound_segs,
                                    &n_stroke_segs);

  if (n_stroke_segs == 0)
    return TRUE;

  coords = g_new0 (PikaCoords, n_bound_segs + 4);

  seg      = 0;
  n_coords = 0;

  /* we offset all coordinates by 0.5 to align the brush with the path */

  coords[n_coords]   = default_coords;
  coords[n_coords].x = (gdouble) (stroke_segs[0].x1 + offset_x + 0.5);
  coords[n_coords].y = (gdouble) (stroke_segs[0].y1 + offset_y + 0.5);

  n_coords++;

  drawables = g_list_prepend (NULL, drawable);

  for (s = 0; s < n_stroke_segs; s++)
    {
      while (stroke_segs[seg].x1 != -1 ||
             stroke_segs[seg].x2 != -1 ||
             stroke_segs[seg].y1 != -1 ||
             stroke_segs[seg].y2 != -1)
        {
          coords[n_coords]   = default_coords;
          coords[n_coords].x = (gdouble) (stroke_segs[seg].x1 + offset_x + 0.5);
          coords[n_coords].y = (gdouble) (stroke_segs[seg].y1 + offset_y + 0.5);

          n_coords++;
          seg++;
        }

      /* Close the stroke points up */
      coords[n_coords] = coords[0];

      n_coords++;

      if (emulate_dynamics)
        pika_paint_core_stroke_emulate_dynamics (coords, n_coords);

      if (initialized ||
          pika_paint_core_start (core, drawables, paint_options, &coords[0],
                                 error))
        {
          gint i;

          initialized = TRUE;

          core->cur_coords  = coords[0];
          core->last_coords = coords[0];

          pika_paint_core_paint (core, drawables, paint_options,
                                 PIKA_PAINT_STATE_INIT, 0);

          pika_paint_core_paint (core, drawables, paint_options,
                                 PIKA_PAINT_STATE_MOTION, 0);

          for (i = 1; i < n_coords; i++)
            {
              pika_paint_core_interpolate (core, drawables, paint_options,
                                           &coords[i], 0);
            }

          pika_paint_core_paint (core, drawables, paint_options,
                                 PIKA_PAINT_STATE_FINISH, 0);
        }
      else
        {
          break;
        }

      n_coords = 0;
      seg++;

      coords[n_coords]   = default_coords;
      coords[n_coords].x = (gdouble) (stroke_segs[seg].x1 + offset_x + 0.5);
      coords[n_coords].y = (gdouble) (stroke_segs[seg].y1 + offset_y + 0.5);

      n_coords++;
    }

  if (initialized)
    {
      pika_paint_core_finish (core, drawables, push_undo);

      pika_paint_core_cleanup (core);
    }

  g_list_free (drawables);
  g_free (coords);
  g_free (stroke_segs);

  return initialized;
}

gboolean
pika_paint_core_stroke_vectors (PikaPaintCore     *core,
                                PikaDrawable      *drawable,
                                PikaPaintOptions  *paint_options,
                                gboolean           emulate_dynamics,
                                PikaVectors       *vectors,
                                gboolean           push_undo,
                                GError           **error)
{
  GList    *drawables;
  GList    *stroke;
  gboolean  initialized = FALSE;
  gboolean  due_to_lack_of_points = FALSE;
  gint      off_x, off_y;

  g_return_val_if_fail (PIKA_IS_PAINT_CORE (core), FALSE);
  g_return_val_if_fail (PIKA_IS_DRAWABLE (drawable), FALSE);
  g_return_val_if_fail (pika_item_is_attached (PIKA_ITEM (drawable)), FALSE);
  g_return_val_if_fail (PIKA_IS_PAINT_OPTIONS (paint_options), FALSE);
  g_return_val_if_fail (PIKA_IS_VECTORS (vectors), FALSE);
  g_return_val_if_fail (error == NULL || *error == NULL, FALSE);

  pika_item_get_offset (PIKA_ITEM (vectors),  &off_x, &off_y);

  drawables = g_list_prepend (NULL, drawable);

  for (stroke = vectors->strokes->head;
       stroke;
       stroke = stroke->next)
    {
      GArray   *coords;
      gboolean  closed;

      coords = pika_stroke_interpolate (PIKA_STROKE (stroke->data),
                                        1.0, &closed);

      if (coords && coords->len)
        {
          gint i;

          for (i = 0; i < coords->len; i++)
            {
              g_array_index (coords, PikaCoords, i).x += off_x;
              g_array_index (coords, PikaCoords, i).y += off_y;
            }

          if (emulate_dynamics)
            pika_paint_core_stroke_emulate_dynamics ((PikaCoords *) coords->data,
                                                     coords->len);

          if (initialized ||
              pika_paint_core_start (core, drawables, paint_options,
                                     &g_array_index (coords, PikaCoords, 0),
                                     error))
            {
              initialized = TRUE;

              core->cur_coords  = g_array_index (coords, PikaCoords, 0);
              core->last_coords = g_array_index (coords, PikaCoords, 0);

              pika_paint_core_paint (core, drawables, paint_options,
                                     PIKA_PAINT_STATE_INIT, 0);

              pika_paint_core_paint (core, drawables, paint_options,
                                     PIKA_PAINT_STATE_MOTION, 0);

              for (i = 1; i < coords->len; i++)
                {
                  pika_paint_core_interpolate (core, drawables, paint_options,
                                               &g_array_index (coords, PikaCoords, i),
                                               0);
                }

              pika_paint_core_paint (core, drawables, paint_options,
                                     PIKA_PAINT_STATE_FINISH, 0);
            }
          else
            {
              if (coords)
                g_array_free (coords, TRUE);

              break;
            }
        }
      else
        {
          due_to_lack_of_points = TRUE;
        }

      if (coords)
        g_array_free (coords, TRUE);
    }

  if (initialized)
    {
      pika_paint_core_finish (core, drawables, push_undo);

      pika_paint_core_cleanup (core);
    }

  if (! initialized && due_to_lack_of_points && *error == NULL)
    {
      g_set_error_literal (error, PIKA_ERROR, PIKA_FAILED,
                           _("Not enough points to stroke"));
    }

  g_list_free (drawables);

  return initialized;
}

static void
pika_paint_core_stroke_emulate_dynamics (PikaCoords *coords,
                                         gint        length)
{
  const gint ramp_length = length / 3;

  /* Calculate and create pressure ramp parameters */
  if (ramp_length > 0)
    {
      gdouble slope = 1.0 / (gdouble) (ramp_length);
      gint    i;

      /* Calculate pressure start ramp */
      for (i = 0; i < ramp_length; i++)
        {
          coords[i].pressure =  i * slope;
        }

      /* Calculate pressure end ramp */
      for (i = length - ramp_length; i < length; i++)
        {
          coords[i].pressure = 1.0 - (i - (length - ramp_length)) * slope;
        }
    }

  /* Calculate and create velocity ramp parameters */
  if (length > 0)
    {
      gdouble slope = 1.0 / length;
      gint    i;

      /* Calculate velocity end ramp */
      for (i = 0; i < length; i++)
        {
          coords[i].velocity = i * slope;
        }
    }

  if (length > 1)
    {
      gint i;
      /* Fill in direction */
      for (i = 1; i < length; i++)
        {
          coords[i].direction = pika_coords_direction (&coords[i-1], &coords[i]);
        }

      coords[0].direction = coords[1].direction;
    }
}