/* treestoretest.c

 * Copyright (C) 2001 Red Hat, Inc

 * Author: Jonathan Blandford

 *

 * This library is free software; you can redistribute it and/or

 * modify it under the terms of the GNU Library General Public

 * License as published by the Free Software Foundation; either

 * version 2 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

 * Library General Public License for more details.

 *

 * You should have received a copy of the GNU Library General Public

 * License along with this library; if not, write to the

 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,

 * Boston, MA 02111-1307, USA.

 */

#include "config.h"

#include <gtk/gtk.h>

#include <stdlib.h>

#include <string.h>

GtkTreeStore *base_model;

static gint node_count = 0;

static void

selection_changed (GtkTreeSelection *selection,

		   GtkWidget        *button)

{

  if (gtk_tree_selection_get_selected (selection, NULL, NULL))

    gtk_widget_set_sensitive (button, TRUE);

  else

    gtk_widget_set_sensitive (button, FALSE);

}

static void

node_set (GtkTreeIter *iter)

{

  gint n;

  gchar *str;

  str = g_strdup_printf ("Row (%d)", node_count++);

  gtk_tree_store_set (base_model, iter, 0, str, -1);

  g_free (str);

  n = g_random_int_range (10000,99999);

  if (n < 0)

    n *= -1;

  str = g_strdup_printf ("%d", n);

  gtk_tree_store_set (base_model, iter, 1, str, -1);

  g_free (str);

}

static void

iter_remove (GtkWidget *button, GtkTreeView *tree_view)

{

  GtkTreeIter selected;

  GtkTreeModel *model;

  model = gtk_tree_view_get_model (tree_view);

  if (gtk_tree_selection_get_selected (gtk_tree_view_get_selection (tree_view),

				       NULL,

				       &selected))

    {

      if (GTK_IS_TREE_STORE (model))

	{

	  gtk_tree_store_remove (GTK_TREE_STORE (model), &selected);

	}

    }

}

static void

iter_insert (GtkWidget *button, GtkTreeView *tree_view)

{

  GtkWidget *entry;

  GtkTreeIter iter;

  GtkTreeIter selected;

  GtkTreeModel *model = gtk_tree_view_get_model (tree_view);

  entry = g_object_get_data (G_OBJECT (button), "user_data");

  if (gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)),

				       NULL,

				       &selected))

    {

      gtk_tree_store_insert (GTK_TREE_STORE (model),

			     &iter,

			     &selected,

			     atoi (gtk_entry_get_text (GTK_ENTRY (entry))));

    }

  else

    {

      gtk_tree_store_insert (GTK_TREE_STORE (model),

			     &iter,

			     NULL,

			     atoi (gtk_entry_get_text (GTK_ENTRY (entry))));

    }

  node_set (&iter);

}

static void

iter_change (GtkWidget *button, GtkTreeView *tree_view)

{

  GtkWidget *entry;

  GtkTreeIter selected;

  GtkTreeModel *model = gtk_tree_view_get_model (tree_view);

  entry = g_object_get_data (G_OBJECT (button), "user_data");

  if (gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)),

				       NULL, &selected))

    {

      gtk_tree_store_set (GTK_TREE_STORE (model),

			  &selected,

			  1,

			  gtk_entry_get_text (GTK_ENTRY (entry)),

			  -1);

    }

}

static void

iter_insert_with_values (GtkWidget *button, GtkTreeView *tree_view)

{

  GtkWidget *entry;

  GtkTreeIter iter;

  GtkTreeIter selected;

  GtkTreeModel *model = gtk_tree_view_get_model (tree_view);

  gchar *str1, *str2;

  entry = g_object_get_data (G_OBJECT (button), "user_data");

  str1 = g_strdup_printf ("Row (%d)", node_count++);

  str2 = g_strdup_printf ("%d", atoi (gtk_entry_get_text (GTK_ENTRY (entry))));

  if (gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)),

				       NULL,

				       &selected))

    {

      gtk_tree_store_insert_with_values (GTK_TREE_STORE (model),

					 &iter,

					 &selected,

					 -1,

					 0, str1,

					 1, str2,

					 -1);

    }

  else

    {

      gtk_tree_store_insert_with_values (GTK_TREE_STORE (model),

					 &iter,

					 NULL,

					 -1,

					 0, str1,

					 1, str2,

					 -1);

    }

  g_free (str1);

  g_free (str2);

}

static void

iter_insert_before  (GtkWidget *button, GtkTreeView *tree_view)

{

  GtkTreeIter iter;

  GtkTreeIter selected;

  GtkTreeModel *model = gtk_tree_view_get_model (tree_view);

  if (gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)),

				       NULL,

				       &selected))

    {

      gtk_tree_store_insert_before (GTK_TREE_STORE (model),

				    &iter,

				    NULL,

				    &selected);

    }

  else

    {

      gtk_tree_store_insert_before (GTK_TREE_STORE (model),

				    &iter,

				    NULL,

				    NULL);

    }

  node_set (&iter);

}

static void

iter_insert_after (GtkWidget *button, GtkTreeView *tree_view)

{

  GtkTreeIter iter;

  GtkTreeIter selected;

  GtkTreeModel *model = gtk_tree_view_get_model (tree_view);

  if (gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)),

				       NULL,

				       &selected))

    {

      if (GTK_IS_TREE_STORE (model))

	{

	  gtk_tree_store_insert_after (GTK_TREE_STORE (model),

				       &iter,

				       NULL,

				       &selected);

	  node_set (&iter);

	}

    }

  else

    {

      if (GTK_IS_TREE_STORE (model))

	{

	  gtk_tree_store_insert_after (GTK_TREE_STORE (model),

				       &iter,

				       NULL,

				       NULL);

	  node_set (&iter);

	}

    }

}

static void

iter_prepend (GtkWidget *button, GtkTreeView *tree_view)

{

  GtkTreeIter iter;

  GtkTreeIter selected;

  GtkTreeModel *model = gtk_tree_view_get_model (tree_view);

  GtkTreeSelection *selection = gtk_tree_view_get_selection (tree_view);

  if (gtk_tree_selection_get_selected (selection, NULL, &selected))

    {

      if (GTK_IS_TREE_STORE (model))

	{

	  gtk_tree_store_prepend (GTK_TREE_STORE (model),

				  &iter,

				  &selected);

	  node_set (&iter);

	}

    }

  else

    {

      if (GTK_IS_TREE_STORE (model))

	{

	  gtk_tree_store_prepend (GTK_TREE_STORE (model),

				  &iter,

				  NULL);

	  node_set (&iter);

	}

    }

}

static void

iter_append (GtkWidget *button, GtkTreeView *tree_view)

{

  GtkTreeIter iter;

  GtkTreeIter selected;

  GtkTreeModel *model = gtk_tree_view_get_model (tree_view);

  if (gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)),

				       NULL,

				       &selected))

    {

      if (GTK_IS_TREE_STORE (model))

	{

	  gtk_tree_store_append (GTK_TREE_STORE (model), &iter, &selected);

	  node_set (&iter);

	}

    }

  else

    {

      if (GTK_IS_TREE_STORE (model))

	{

	  gtk_tree_store_append (GTK_TREE_STORE (model), &iter, NULL);

	  node_set (&iter);

	}

    }

}

static void

make_window (gint view_type)

{

  GtkWidget *window;

  GtkWidget *vbox;

  GtkWidget *hbox, *entry;

  GtkWidget *button;

  GtkWidget *scrolled_window;

  GtkWidget *tree_view;

  GtkTreeViewColumn *column;

  GtkCellRenderer *cell;

  GObject *selection;

  /* Make the Widgets/Objects */

  window = gtk_window_new (GTK_WINDOW_TOPLEVEL);

  switch (view_type)

    {

    case 0:

      gtk_window_set_title (GTK_WINDOW (window), "Unsorted list");

      break;

    case 1:

      gtk_window_set_title (GTK_WINDOW (window), "Sorted list");

      break;

    }

  vbox = gtk_vbox_new (FALSE, 8);

  gtk_container_set_border_width (GTK_CONTAINER (vbox), 8);

  gtk_window_set_default_size (GTK_WINDOW (window), 300, 350);

  scrolled_window = gtk_scrolled_window_new (NULL, NULL);

  switch (view_type)

    {

    case 0:

      tree_view = gtk_tree_view_new_with_model (GTK_TREE_MODEL (base_model));

      break;

    case 1:

      {

	GtkTreeModel *sort_model;

	sort_model = gtk_tree_model_sort_new_with_model (GTK_TREE_MODEL (base_model));

	tree_view = gtk_tree_view_new_with_model (GTK_TREE_MODEL (sort_model));

      }

      break;

    default:

      g_assert_not_reached ();

      tree_view = NULL; /* Quiet compiler */

      break;

    }

  gtk_tree_view_set_rules_hint (GTK_TREE_VIEW (tree_view), TRUE);

  selection = G_OBJECT (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)));

  gtk_tree_selection_set_mode (GTK_TREE_SELECTION (selection), GTK_SELECTION_SINGLE);

  /* Put them together */

  gtk_container_add (GTK_CONTAINER (scrolled_window), tree_view);

  gtk_box_pack_start (GTK_BOX (vbox), scrolled_window, TRUE, TRUE, 0);

  gtk_container_add (GTK_CONTAINER (window), vbox);

  gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (scrolled_window),

				  GTK_POLICY_AUTOMATIC,

				  GTK_POLICY_AUTOMATIC);

  g_signal_connect (window, "destroy", gtk_main_quit, NULL);

  /* buttons */

  button = gtk_button_new_with_label ("gtk_tree_store_remove");

  gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);

  g_signal_connect (selection, "changed",

                    G_CALLBACK (selection_changed),

                    button);

  g_signal_connect (button, "clicked", 

                    G_CALLBACK (iter_remove), 

                    tree_view);

  gtk_widget_set_sensitive (button, FALSE);

  button = gtk_button_new_with_label ("gtk_tree_store_insert");

  hbox = gtk_hbox_new (FALSE, 8);

  entry = gtk_entry_new ();

  gtk_box_pack_start (GTK_BOX (vbox), hbox, FALSE, FALSE, 0);

  gtk_box_pack_start (GTK_BOX (hbox), button, TRUE, TRUE, 0);

  gtk_box_pack_start (GTK_BOX (hbox), entry, FALSE, FALSE, 0);

  g_object_set_data (G_OBJECT (button), "user_data", entry);

  g_signal_connect (button, "clicked", 

                    G_CALLBACK (iter_insert), 

                    tree_view);

  button = gtk_button_new_with_label ("gtk_tree_store_set");

  hbox = gtk_hbox_new (FALSE, 8);

  entry = gtk_entry_new ();

  gtk_box_pack_start (GTK_BOX (vbox), hbox, FALSE, FALSE, 0);

  gtk_box_pack_start (GTK_BOX (hbox), button, TRUE, TRUE, 0);

  gtk_box_pack_start (GTK_BOX (hbox), entry, FALSE, FALSE, 0);

  g_object_set_data (G_OBJECT (button), "user_data", entry);

  g_signal_connect (button, "clicked",

		    G_CALLBACK (iter_change),

		    tree_view);

  button = gtk_button_new_with_label ("gtk_tree_store_insert_with_values");

  hbox = gtk_hbox_new (FALSE, 8);

  entry = gtk_entry_new ();

  gtk_box_pack_start (GTK_BOX (vbox), hbox, FALSE, FALSE, 0);

  gtk_box_pack_start (GTK_BOX (hbox), button, TRUE, TRUE, 0);

  gtk_box_pack_start (GTK_BOX (hbox), entry, FALSE, FALSE, 0);

  g_object_set_data (G_OBJECT (button), "user_data", entry);

  g_signal_connect (button, "clicked",

		    G_CALLBACK (iter_insert_with_values),

		    tree_view);

  button = gtk_button_new_with_label ("gtk_tree_store_insert_before");

  gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);

  g_signal_connect (button, "clicked", 

                    G_CALLBACK (iter_insert_before), 

                    tree_view);

  g_signal_connect (selection, "changed",

                    G_CALLBACK (selection_changed),

                    button);

  gtk_widget_set_sensitive (button, FALSE);

  button = gtk_button_new_with_label ("gtk_tree_store_insert_after");

  gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);

  g_signal_connect (button, "clicked", 

                    G_CALLBACK (iter_insert_after), 

                    tree_view);

  g_signal_connect (selection, "changed",

                    G_CALLBACK (selection_changed),

                    button);

  gtk_widget_set_sensitive (button, FALSE);

  button = gtk_button_new_with_label ("gtk_tree_store_prepend");

  gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);

  g_signal_connect (button, "clicked", 

                    G_CALLBACK (iter_prepend), 

                    tree_view);

  button = gtk_button_new_with_label ("gtk_tree_store_append");

  gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);

  g_signal_connect (button, "clicked", 

                    G_CALLBACK (iter_append), 

                    tree_view);

  /* The selected column */

  cell = gtk_cell_renderer_text_new ();

  column = gtk_tree_view_column_new_with_attributes ("Node ID", cell, "markup", 0, NULL);

  gtk_tree_view_column_set_sort_column_id (column, 0);

  gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), column);

  cell = gtk_cell_renderer_text_new ();

  column = gtk_tree_view_column_new_with_attributes ("Random Number", cell, "text", 1, NULL);

  gtk_tree_view_column_set_sort_column_id (column, 1);

  gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), column);

  /* A few to start */

  if (view_type == 0)

    {

      iter_append (NULL, GTK_TREE_VIEW (tree_view));

      iter_append (NULL, GTK_TREE_VIEW (tree_view));

      iter_append (NULL, GTK_TREE_VIEW (tree_view));

      iter_append (NULL, GTK_TREE_VIEW (tree_view));

      iter_append (NULL, GTK_TREE_VIEW (tree_view));

      iter_append (NULL, GTK_TREE_VIEW (tree_view));

    }

  /* Show it all */

  gtk_widget_show_all (window);

}

int

main (int argc, char *argv[])

{

  gtk_init (&argc, &argv);

  base_model = gtk_tree_store_new (2, G_TYPE_STRING, G_TYPE_STRING);

  /* FIXME: reverse this */

  make_window (0);

  make_window (1);

  gtk_main ();

  return 0;

}