]> Pileus Git - ~andy/gtk/blobdiff - tests/treestoretest.c
Use g_setenv() instead of putenv().
[~andy/gtk] / tests / treestoretest.c
index 04d3ac4b4271dcb3a7d4f19538ac12de4badd1f3..1946d9e04e51e38c801b35abc29b96daaaef9ee6 100644 (file)
 #include <gtk/gtk.h>
 #include <stdlib.h>
+#include <string.h>
 
-GtkObject *model;
+GtkTreeStore *base_model;
 
 static void
 selection_changed (GtkTreeSelection *selection,
-                  GtkWidget         *button)
+                  GtkWidget        *button)
 {
-  if (gtk_tree_selection_get_selected (selection))
+  if (gtk_tree_selection_get_selected (selection, NULL, NULL))
     gtk_widget_set_sensitive (button, TRUE);
   else
     gtk_widget_set_sensitive (button, FALSE);
 }
 
-static GtkTreeNode *
-node_new ()
+static void
+node_set (GtkTreeIter *iter)
 {
-  static GValue value = {0, };
   static gint i = 0;
+  gint n;
   gchar *str;
-  GtkTreeNode *node = gtk_tree_store_node_new ();
 
-  g_value_init (&value, G_TYPE_STRING);
-  str = g_strdup_printf ("FOO: %d", i++);
-  g_value_set_string (&value, str);
+  str = g_strdup_printf ("Row (<span color=\"red\">%d</span>)", i++);
+  gtk_tree_store_set (base_model, iter, 0, str, -1);
   g_free (str);
-  gtk_tree_store_node_set_cell (GTK_TREE_STORE (model), node, 0, &value);
-  g_value_unset (&value);
 
-  return node;
+  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
-node_remove (GtkWidget *button, GtkTreeView *tree_view)
+iter_remove (GtkWidget *button, GtkTreeView *tree_view)
 {
-  GtkTreeNode *selected = gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)));
-  gtk_tree_store_node_remove (GTK_TREE_STORE (model),
-                             selected);
+  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
-node_insert (GtkWidget *button, GtkTreeView *tree_view)
+iter_insert (GtkWidget *button, GtkTreeView *tree_view)
 {
   GtkWidget *entry;
-  GtkTreeNode *selected = gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)));
+  GtkTreeIter iter;
+  GtkTreeIter selected;
+  GtkTreeModel *model = gtk_tree_view_get_model (tree_view);
 
-  entry = gtk_object_get_user_data (GTK_OBJECT (button));
-  gtk_tree_store_node_insert (GTK_TREE_STORE (model),
-                             selected,
-                             atoi (gtk_entry_get_text (GTK_ENTRY (entry))),
-                             node_new ());
+  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
-node_insert_before  (GtkWidget *button, GtkTreeView *tree_view)
+iter_change (GtkWidget *button, GtkTreeView *tree_view)
 {
-  GtkTreeNode *selected = gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)));
-  gtk_tree_store_node_insert_before (GTK_TREE_STORE (model),
-                                   NULL,
-                                   selected,
-                                   node_new ());
+  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
-node_insert_after (GtkWidget *button, GtkTreeView *tree_view)
+iter_insert_before  (GtkWidget *button, GtkTreeView *tree_view)
 {
-  GtkTreeNode *selected = gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)));
-  gtk_tree_store_node_insert_after (GTK_TREE_STORE (model),
+  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,
-                                   selected,
-                                   node_new ());
+                                   NULL);
+    }
+
+  node_set (&iter);
 }
 
 static void
-node_prepend (GtkWidget *button, GtkTreeView *tree_view)
+iter_insert_after (GtkWidget *button, GtkTreeView *tree_view)
 {
-  GtkTreeNode *selected = gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)));
-  gtk_tree_store_node_prepend (GTK_TREE_STORE (model),
-                              selected,
-                              node_new ());
+  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
-node_append (GtkWidget *button, GtkTreeView *tree_view)
+iter_prepend (GtkWidget *button, GtkTreeView *tree_view)
 {
-  GtkTreeNode *selected = gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)));
-  gtk_tree_store_node_append (GTK_TREE_STORE (model),
-                             selected,
-                             node_new ());
+  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
-make_window ()
+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;
@@ -99,19 +227,48 @@ make_window ()
   GtkWidget *button;
   GtkWidget *scrolled_window;
   GtkWidget *tree_view;
-  GtkObject *column;
+  GtkTreeViewColumn *column;
   GtkCellRenderer *cell;
-  GtkObject *selection;
+  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);
-  tree_view = gtk_tree_view_new_with_model (GTK_TREE_MODEL (model));
-  selection = GTK_OBJECT (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)));
-  gtk_tree_selection_set_type (GTK_TREE_SELECTION (selection), GTK_TREE_SELECTION_SINGLE);
+  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);
@@ -120,59 +277,94 @@ make_window ()
   gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (scrolled_window),
                                  GTK_POLICY_AUTOMATIC,
                                  GTK_POLICY_AUTOMATIC);
-  gtk_signal_connect (GTK_OBJECT (window), "destroy", gtk_main_quit, NULL);
+  g_signal_connect (window, "destroy", gtk_main_quit, NULL);
 
   /* buttons */
-  button = gtk_button_new_with_label ("gtk_tree_store_node_remove");
+  button = gtk_button_new_with_label ("gtk_tree_store_remove");
   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
-  gtk_signal_connect (GTK_OBJECT (selection),
-                     "selection_changed",
-                     selection_changed,
-                     button);
-  gtk_signal_connect (GTK_OBJECT (button), "clicked", node_remove, tree_view);
+  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_node_insert");
+  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);
-  gtk_object_set_user_data (GTK_OBJECT (button), entry);
-  gtk_signal_connect (GTK_OBJECT (button), "clicked", node_insert, tree_view);
-
+  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_node_insert_before");
+  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_before");
   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
-  gtk_signal_connect (GTK_OBJECT (button), "clicked", node_insert_before, tree_view);
-  gtk_signal_connect (GTK_OBJECT (selection),
-                     "selection_changed",
-                     selection_changed,
-                     button);
+  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_node_insert_after");
+  button = gtk_button_new_with_label ("gtk_tree_store_insert_after");
   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
-  gtk_signal_connect (GTK_OBJECT (button), "clicked", node_insert_after, tree_view);
-  gtk_signal_connect (GTK_OBJECT (selection),
-                     "selection_changed",
-                     selection_changed,
-                     button);
+  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_node_prepend");
+  button = gtk_button_new_with_label ("gtk_tree_store_prepend");
   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
-  gtk_signal_connect (GTK_OBJECT (button), "clicked", node_prepend, tree_view);
+  g_signal_connect (button, "clicked", 
+                    G_CALLBACK (iter_prepend), 
+                    tree_view);
 
-  button = gtk_button_new_with_label ("gtk_tree_store_node_append");
+  button = gtk_button_new_with_label ("gtk_tree_store_append");
   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
-  gtk_signal_connect (GTK_OBJECT (button), "clicked", node_append, tree_view);
+  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 ("nodes", cell, "text", 0, NULL);
-  gtk_tree_view_add_column (GTK_TREE_VIEW (tree_view), GTK_TREE_VIEW_COLUMN (column));
+  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);
 }
@@ -182,17 +374,14 @@ main (int argc, char *argv[])
 {
   gtk_init (&argc, &argv);
 
-  model = gtk_tree_store_new_with_values (2, G_TYPE_STRING, G_TYPE_STRING);
-
-  make_window ();
-  make_window ();
+  base_model = gtk_tree_store_new (2, G_TYPE_STRING, G_TYPE_STRING);
 
-  /* A few to start */
-  gtk_tree_store_node_append (GTK_TREE_STORE (model), NULL, node_new ());
-  gtk_tree_store_node_append (GTK_TREE_STORE (model), NULL, node_new ());
-  gtk_tree_store_node_append (GTK_TREE_STORE (model), NULL, node_new ());
+  /* FIXME: reverse this */
+  make_window (0);
+  make_window (1);
 
   gtk_main ();
 
   return 0;
 }
+