/* LIBPIKA - The PIKA Library
 * Copyright (C) 1995-1997 Peter Mattis and Spencer Kimball
 *
 * pikaitemcombobox.c
 * Copyright (C) 2004 Sven Neumann <sven@gimp.org>
 * Copyright (C) 2006 Simon Budig <simon@gimp.org>
 *
 * This library is free software: you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library.  If not, see
 * <https://www.gnu.org/licenses/>.
 */

#include "config.h"

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

#include <gegl.h>
#include <gtk/gtk.h>

#include "libpikawidgets/pikawidgets.h"

#include "pika.h"

#include "pikauitypes.h"
#include "pikaitemcombobox.h"
#include "pikapixbuf.h"


/**
 * SECTION: pikaitemcombobox
 * @title: PikaItemComboBox
 * @short_description: Widgets providing popup menus of items.
 *
 * Widgets providing popup menus of items (layers, channels,
 * drawables, vectors).
 **/


#define THUMBNAIL_SIZE  24
#define WIDTH_REQUEST  200


#define GET_PRIVATE(obj) (g_object_get_data (G_OBJECT (obj), "pika-item-combo-box-private"))


typedef struct _PikaItemComboBoxPrivate   PikaItemComboBoxPrivate;

struct _PikaItemComboBoxPrivate
{
  PikaItemConstraintFunc  constraint;
  gpointer                data;
};

typedef struct _PikaDrawableComboBoxClass PikaDrawableComboBoxClass;
typedef struct _PikaChannelComboBoxClass  PikaChannelComboBoxClass;
typedef struct _PikaLayerComboBoxClass    PikaLayerComboBoxClass;
typedef struct _PikaVectorsComboBoxClass  PikaVectorsComboBoxClass;

struct _PikaDrawableComboBox
{
  PikaIntComboBox  parent_instance;
};

struct _PikaDrawableComboBoxClass
{
  PikaIntComboBoxClass  parent_class;
};

struct _PikaChannelComboBox
{
  PikaIntComboBox  parent_instance;
};

struct _PikaChannelComboBoxClass
{
  PikaIntComboBoxClass  parent_class;
};

struct _PikaLayerComboBox
{
  PikaIntComboBox  parent_instance;
};

struct _PikaLayerComboBoxClass
{
  PikaIntComboBoxClass  parent_class;
};

struct _PikaVectorsComboBox
{
  PikaIntComboBox  parent_instance;
};

struct _PikaVectorsComboBoxClass
{
  PikaIntComboBoxClass  parent_class;
};


static GtkWidget * pika_item_combo_box_new (GType                       type,
                                            PikaItemConstraintFunc      constraint,
                                            gpointer                    data,
                                            GDestroyNotify              data_destroy);

static void  pika_item_combo_box_populate  (PikaIntComboBox            *combo_box);
static void  pika_item_combo_box_model_add (PikaIntComboBox            *combo_box,
                                            GtkListStore               *store,
                                            PikaImage                  *image,
                                            GList                      *items,
                                            gint                        tree_level);

static void  pika_item_combo_box_drag_data_received (GtkWidget         *widget,
                                                     GdkDragContext    *context,
                                                     gint               x,
                                                     gint               y,
                                                     GtkSelectionData  *selection,
                                                     guint              info,
                                                     guint              time);

static void  pika_item_combo_box_changed   (PikaIntComboBox *combo_box);


static const GtkTargetEntry targets[] =
{
  { "application/x-pika-channel-id", 0 },
  { "application/x-pika-layer-id",   0 },
  { "application/x-pika-vectors-id", 0 }
};


G_DEFINE_TYPE (PikaDrawableComboBox, pika_drawable_combo_box,
               PIKA_TYPE_INT_COMBO_BOX)

static void
pika_drawable_combo_box_class_init (PikaDrawableComboBoxClass *klass)
{
  GtkWidgetClass *widget_class = GTK_WIDGET_CLASS (klass);

  widget_class->drag_data_received = pika_item_combo_box_drag_data_received;
}

static void
pika_drawable_combo_box_init (PikaDrawableComboBox *combo_box)
{
  gtk_drag_dest_set (GTK_WIDGET (combo_box),
                     GTK_DEST_DEFAULT_HIGHLIGHT |
                     GTK_DEST_DEFAULT_MOTION |
                     GTK_DEST_DEFAULT_DROP,
                     targets, 2,
                     GDK_ACTION_COPY);

  g_object_set_data_full (G_OBJECT (combo_box), "pika-item-combo-box-private",
                          g_new0 (PikaItemComboBoxPrivate, 1),
                          (GDestroyNotify) g_free);
}

/**
 * pika_drawable_combo_box_new:
 * @constraint: (nullable):       A #PikaItemConstraintFunc or %NULL
 * @data: (closure constraint):   A pointer that is passed to @constraint
 * @data_destroy: (destroy data): Destroy function for @data
 *
 * Creates a new #PikaIntComboBox filled with all currently opened
 * drawables. If a @constraint function is specified, it is called for
 * each drawable and only if the function returns %TRUE, the drawable
 * is added to the combobox.
 *
 * You should use pika_int_combo_box_connect() to initialize and connect
 * the combo.  Use pika_int_combo_box_set_active() to get the active
 * drawable ID and pika_int_combo_box_get_active() to retrieve the ID
 * of the selected drawable.
 *
 * Returns: a new #PikaIntComboBox.
 *
 * Since: 2.2
 **/
GtkWidget *
pika_drawable_combo_box_new (PikaItemConstraintFunc constraint,
                             gpointer               data,
                             GDestroyNotify         data_destroy)
{
  return pika_item_combo_box_new (PIKA_TYPE_DRAWABLE_COMBO_BOX,
                                  constraint, data, data_destroy);
}


G_DEFINE_TYPE (PikaChannelComboBox, pika_channel_combo_box,
               PIKA_TYPE_INT_COMBO_BOX)

static void
pika_channel_combo_box_class_init (PikaChannelComboBoxClass *klass)
{
  GtkWidgetClass *widget_class = GTK_WIDGET_CLASS (klass);

  widget_class->drag_data_received = pika_item_combo_box_drag_data_received;
}

static void
pika_channel_combo_box_init (PikaChannelComboBox *combo_box)
{
  gtk_drag_dest_set (GTK_WIDGET (combo_box),
                     GTK_DEST_DEFAULT_HIGHLIGHT |
                     GTK_DEST_DEFAULT_MOTION |
                     GTK_DEST_DEFAULT_DROP,
                     targets, 1,
                     GDK_ACTION_COPY);

  g_object_set_data_full (G_OBJECT (combo_box), "pika-item-combo-box-private",
                          g_new0 (PikaItemComboBoxPrivate, 1),
                          (GDestroyNotify) g_free);
}

/**
 * pika_channel_combo_box_new:
 * @constraint: (nullable):       A #PikaItemConstraintFunc or %NULL
 * @data: (closure constraint):   A pointer that is passed to @constraint
 * @data_destroy: (destroy data): Destroy function for @data
 *
 * Creates a new #PikaIntComboBox filled with all currently opened
 * channels. See pika_drawable_combo_box_new() for more information.
 *
 * Returns: a new #PikaIntComboBox.
 *
 * Since: 2.2
 **/
GtkWidget *
pika_channel_combo_box_new (PikaItemConstraintFunc constraint,
                            gpointer               data,
                            GDestroyNotify         data_destroy)
{
  return pika_item_combo_box_new (PIKA_TYPE_CHANNEL_COMBO_BOX,
                                  constraint, data, data_destroy);
}


G_DEFINE_TYPE (PikaLayerComboBox, pika_layer_combo_box,
               PIKA_TYPE_INT_COMBO_BOX)

static void
pika_layer_combo_box_class_init (PikaLayerComboBoxClass *klass)
{
  GtkWidgetClass *widget_class = GTK_WIDGET_CLASS (klass);

  widget_class->drag_data_received = pika_item_combo_box_drag_data_received;
}

static void
pika_layer_combo_box_init (PikaLayerComboBox *combo_box)
{
  gtk_drag_dest_set (GTK_WIDGET (combo_box),
                     GTK_DEST_DEFAULT_HIGHLIGHT |
                     GTK_DEST_DEFAULT_MOTION |
                     GTK_DEST_DEFAULT_DROP,
                     targets + 1, 1,
                     GDK_ACTION_COPY);

  g_object_set_data_full (G_OBJECT (combo_box), "pika-item-combo-box-private",
                          g_new0 (PikaItemComboBoxPrivate, 1),
                          (GDestroyNotify) g_free);
}

/**
 * pika_layer_combo_box_new:
 * @constraint: (nullable):       A #PikaItemConstraintFunc or %NULL
 * @data: (closure constraint):   A pointer that is passed to @constraint
 * @data_destroy: (destroy data): Destroy function for @data
 *
 * Creates a new #PikaIntComboBox filled with all currently opened
 * layers. See pika_drawable_combo_box_new() for more information.
 *
 * Returns: a new #PikaIntComboBox.
 *
 * Since: 2.2
 **/
GtkWidget *
pika_layer_combo_box_new (PikaItemConstraintFunc constraint,
                          gpointer               data,
                          GDestroyNotify         data_destroy)
{
  return pika_item_combo_box_new (PIKA_TYPE_LAYER_COMBO_BOX,
                                  constraint, data, data_destroy);
}


G_DEFINE_TYPE (PikaVectorsComboBox, pika_vectors_combo_box,
               PIKA_TYPE_INT_COMBO_BOX)

static void
pika_vectors_combo_box_class_init (PikaVectorsComboBoxClass *klass)
{
  GtkWidgetClass *widget_class = GTK_WIDGET_CLASS (klass);

  widget_class->drag_data_received = pika_item_combo_box_drag_data_received;
}

static void
pika_vectors_combo_box_init (PikaVectorsComboBox *combo_box)
{
  gtk_drag_dest_set (GTK_WIDGET (combo_box),
                     GTK_DEST_DEFAULT_HIGHLIGHT |
                     GTK_DEST_DEFAULT_MOTION |
                     GTK_DEST_DEFAULT_DROP,
                     targets + 2, 1,
                     GDK_ACTION_COPY);

  g_object_set_data_full (G_OBJECT (combo_box), "pika-item-combo-box-private",
                          g_new0 (PikaItemComboBoxPrivate, 1),
                          (GDestroyNotify) g_free);
}


/**
 * pika_vectors_combo_box_new:
 * @constraint: (nullable):       A #PikaItemConstraintFunc or %NULL
 * @data: (closure constraint):   A pointer that is passed to @constraint
 * @data_destroy: (destroy data): Destroy function for @data
 *
 * Creates a new #PikaIntComboBox filled with all currently opened
 * vectors objects. If a @constraint function is specified, it is called for
 * each vectors object and only if the function returns %TRUE, the vectors
 * object is added to the combobox.
 *
 * You should use pika_int_combo_box_connect() to initialize and connect
 * the combo.  Use pika_int_combo_box_set_active() to set the active
 * vectors ID and pika_int_combo_box_get_active() to retrieve the ID
 * of the selected vectors object.
 *
 * Returns: a new #PikaIntComboBox.
 *
 * Since: 2.4
 **/
GtkWidget *
pika_vectors_combo_box_new (PikaItemConstraintFunc constraint,
                            gpointer               data,
                            GDestroyNotify         data_destroy)
{
  return pika_item_combo_box_new (PIKA_TYPE_VECTORS_COMBO_BOX,
                                  constraint, data, data_destroy);
}


static GtkWidget *
pika_item_combo_box_new (GType                  type,
                         PikaItemConstraintFunc constraint,
                         gpointer               data,
                         GDestroyNotify         data_destroy)
{
  PikaIntComboBox         *combo_box;
  PikaItemComboBoxPrivate *private;

  combo_box = g_object_new (type,
                            "width-request", WIDTH_REQUEST,
                            "ellipsize",     PANGO_ELLIPSIZE_MIDDLE,
                            NULL);

  private = GET_PRIVATE (combo_box);

  private->constraint = constraint;
  private->data       = data;

  if (data_destroy)
    g_object_weak_ref (G_OBJECT (combo_box), (GWeakNotify) data_destroy, data);

  pika_item_combo_box_populate (combo_box);

  g_signal_connect (combo_box, "changed",
                    G_CALLBACK (pika_item_combo_box_changed),
                    NULL);

  return GTK_WIDGET (combo_box);
}

static void
pika_item_combo_box_populate (PikaIntComboBox *combo_box)
{
  GtkTreeModel *model;
  GtkTreeIter   iter;
  GList        *images;
  GList        *list;

  model = gtk_combo_box_get_model (GTK_COMBO_BOX (combo_box));

  images = pika_list_images ();

  for (list = images; list; list = g_list_next (list))
    {
      PikaImage *image = list->data;
      GList     *items;

      if (PIKA_IS_DRAWABLE_COMBO_BOX (combo_box) ||
          PIKA_IS_LAYER_COMBO_BOX (combo_box))
        {
          items = pika_image_list_layers (image);
          pika_item_combo_box_model_add (combo_box, GTK_LIST_STORE (model),
                                         image, items, 0);
          g_list_free (items);
        }

      if (PIKA_IS_DRAWABLE_COMBO_BOX (combo_box) ||
          PIKA_IS_CHANNEL_COMBO_BOX (combo_box))
        {
          items = pika_image_list_channels (image);
          pika_item_combo_box_model_add (combo_box, GTK_LIST_STORE (model),
                                         image, items, 0);
          g_list_free (items);
        }

      if (PIKA_IS_VECTORS_COMBO_BOX (combo_box))
        {
          items = pika_image_list_vectors (image);
          pika_item_combo_box_model_add (combo_box, GTK_LIST_STORE (model),
                                         image, items, 0);
          g_list_free (items);
        }
    }

  g_list_free (images);

  if (gtk_tree_model_get_iter_first (model, &iter))
    gtk_combo_box_set_active_iter (GTK_COMBO_BOX (combo_box), &iter);
}

static void
pika_item_combo_box_model_add (PikaIntComboBox *combo_box,
                               GtkListStore    *store,
                               PikaImage       *image,
                               GList           *items,
                               gint             tree_level)
{
  PikaItemComboBoxPrivate *private = GET_PRIVATE (combo_box);
  GtkTreeIter              iter;
  GList                   *list;
  gchar                   *indent;

  if (tree_level > 0)
    {
      indent = g_new (gchar, tree_level + 2);
      memset (indent, '-', tree_level);
      indent[tree_level] = ' ';
      indent[tree_level + 1] = '\0';
    }
  else
    {
      indent = g_strdup ("");
    }

  for (list = items; list; list = g_list_next (list))
    {
      PikaItem *item    = list->data;
      gint32    item_id = pika_item_get_id (item);

      if (! private->constraint ||
          private->constraint (image, item, private->data))
        {
          gchar     *image_name = pika_image_get_name (image);
          gchar     *item_name  = pika_item_get_name (item);
          gchar     *label;
          GdkPixbuf *thumb;

          label = g_strdup_printf ("%s%s-%d / %s-%d",
                                   indent, image_name,
                                   pika_image_get_id (image),
                                   item_name, item_id);

          g_free (item_name);
          g_free (image_name);

          if (PIKA_IS_VECTORS_COMBO_BOX (combo_box))
            thumb = NULL;
          else
            thumb = pika_drawable_get_thumbnail (PIKA_DRAWABLE (item),
                                                 THUMBNAIL_SIZE, THUMBNAIL_SIZE,
                                                 PIKA_PIXBUF_SMALL_CHECKS);

          gtk_list_store_append (store, &iter);
          gtk_list_store_set (store, &iter,
                              PIKA_INT_STORE_VALUE,  item_id,
                              PIKA_INT_STORE_LABEL,  label,
                              PIKA_INT_STORE_PIXBUF, thumb,
                              -1);

          if (thumb)
            g_object_unref (thumb);

          g_free (label);
        }

      if (pika_item_is_group (item))
        {
          GList *children;

          children = pika_item_list_children (item);
          pika_item_combo_box_model_add (combo_box, store,
                                         image, children,
                                         tree_level + 1);
          g_list_free (children);
        }
    }

  g_free (indent);
}

static void
pika_item_combo_box_drag_data_received (GtkWidget        *widget,
                                        GdkDragContext   *context,
                                        gint              x,
                                        gint              y,
                                        GtkSelectionData *selection,
                                        guint             info,
                                        guint             time)
{
  gint   length = gtk_selection_data_get_length (selection);
  gchar *str;

  if (gtk_selection_data_get_format (selection) != 8 || length < 1)
    {
      g_warning ("%s: received invalid item ID data", G_STRFUNC);
      return;
    }

  str = g_strndup ((const gchar *) gtk_selection_data_get_data (selection),
                   length);

  if (g_utf8_validate (str, -1, NULL))
    {
      gint pid;
      gint ID;

      if (sscanf (str, "%i:%i", &pid, &ID) == 2 &&
          pid == pika_getpid ())
        {
          pika_int_combo_box_set_active (PIKA_INT_COMBO_BOX (widget), ID);
        }
    }

  g_free (str);
}

static gboolean
pika_item_combo_box_remove_items (GtkTreeModel *model,
                                  GtkTreePath  *path,
                                  GtkTreeIter  *iter,
                                  gpointer      data)
{
  gint    item_ID;
  GList **remove = data;

  gtk_tree_model_get (model, iter,
                      PIKA_INT_STORE_VALUE, &item_ID,
                      -1);

  if (item_ID > 0)
    *remove = g_list_prepend (*remove, g_memdup2 (iter, sizeof (GtkTreeIter)));

  return FALSE;
}

static void
pika_item_combo_box_changed (PikaIntComboBox *combo_box)
{
  gint item_ID;

  if (pika_int_combo_box_get_active (combo_box, &item_ID))
    {
      if (item_ID > 0 && ! pika_item_get_by_id (item_ID))
        {
          GtkTreeModel *model;
          GList        *remove = NULL;
          GList        *list;

          model = gtk_combo_box_get_model (GTK_COMBO_BOX (combo_box));

          g_signal_stop_emission_by_name (combo_box, "changed");

          gtk_tree_model_foreach (model,
                                  pika_item_combo_box_remove_items,
                                  &remove);

          for (list = remove; list; list = g_list_next (list))
            gtk_list_store_remove (GTK_LIST_STORE (model), list->data);

          g_list_free_full (remove, (GDestroyNotify) g_free);

          pika_item_combo_box_populate (combo_box);
        }
    }
}