Pileus Git - ~andy/gtk/blob - tests/treestoretest.c

   1 #include <gtk/gtk.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4
   5 GtkTreeStore *base_model;
   6
   7 static void
   8 selection_changed (GtkTreeSelection *selection,
   9                    GtkWidget        *button)
  10 {
  11   if (gtk_tree_selection_get_selected (selection, NULL, NULL))
  12     gtk_widget_set_sensitive (button, TRUE);
  13   else
  14     gtk_widget_set_sensitive (button, FALSE);
  15 }
  16
  17 static void
  18 node_set (GtkTreeIter *iter)
  19 {
  20   static gint i = 0;
  21   gint n;
  22   gchar *str;
  23
  24   str = g_strdup_printf ("Row (<span color=\"red\">%d</span>)", i++);
  25   gtk_tree_store_set (base_model, iter, 0, str, -1);
  26   g_free (str);
  27
  28   n = g_random_int_range (10000,99999);
  29   if (n < 0)
  30     n *= -1;
  31   str = g_strdup_printf ("%d", n);
  32
  33   gtk_tree_store_set (base_model, iter, 1, str, -1);
  34   g_free (str);
  35 }
  36
  37 static void
  38 iter_remove (GtkWidget *button, GtkTreeView *tree_view)
  39 {
  40   GtkTreeIter selected;
  41   GtkTreeIter s_selected;
  42   GtkTreeModel *model;
  43
  44   model = gtk_tree_view_get_model (tree_view);
  45
  46   if (gtk_tree_selection_get_selected (gtk_tree_view_get_selection (tree_view),
  47                                        NULL,
  48                                        &selected))
  49     {
  50       if (GTK_IS_TREE_STORE (model))
  51         {
  52           gtk_tree_store_remove (GTK_TREE_STORE (model), &selected);
  53         }
  54     }
  55 }
  56
  57 static void
  58 iter_insert (GtkWidget *button, GtkTreeView *tree_view)
  59 {
  60   GtkWidget *entry;
  61   GtkTreeIter iter;
  62   GtkTreeIter selected;
  63   GtkTreeStore *model = gtk_tree_view_get_model (tree_view);
  64
  65   entry = gtk_object_get_user_data (GTK_OBJECT (button));
  66   if (gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)),
  67                                        NULL,
  68                                        &selected))
  69     {
  70       gtk_tree_store_insert (model,
  71                              &iter,
  72                              &selected,
  73                              atoi (gtk_entry_get_text (GTK_ENTRY (entry))));
  74     }
  75   else
  76     {
  77       gtk_tree_store_insert (model,
  78                              &iter,
  79                              NULL,
  80                              atoi (gtk_entry_get_text (GTK_ENTRY (entry))));
  81     }
  82
  83   node_set (&iter);
  84 }
  85
  86 static void
  87 iter_insert_before  (GtkWidget *button, GtkTreeView *tree_view)
  88 {
  89   GtkTreeIter iter;
  90   GtkTreeIter selected;
  91   GtkTreeStore *model = gtk_tree_view_get_model (tree_view);
  92
  93   if (gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)),
  94                                        NULL,
  95                                        &selected))
  96     {
  97       gtk_tree_store_insert_before (model,
  98                                     &iter,
  99                                     NULL,
 100                                     &selected);
 101     }
 102   else
 103     {
 104       gtk_tree_store_insert_before (model,
 105                                     &iter,
 106                                     NULL,
 107                                     NULL);
 108     }
 109
 110   node_set (&iter);
 111 }
 112
 113 static void
 114 iter_insert_after (GtkWidget *button, GtkTreeView *tree_view)
 115 {
 116   GtkTreeIter iter;
 117   GtkTreeIter selected;
 118   GtkTreeStore *model = gtk_tree_view_get_model (tree_view);
 119
 120   if (gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)),
 121                                        NULL,
 122                                        &selected))
 123     {
 124       if (GTK_IS_TREE_STORE (model))
 125         {
 126           gtk_tree_store_insert_after (model,
 127                                        &iter,
 128                                        NULL,
 129                                        &selected);
 130           node_set (&iter);
 131         }
 132     }
 133   else
 134     {
 135       if (GTK_IS_TREE_STORE (model))
 136         {
 137           gtk_tree_store_insert_after (model,
 138                                        &iter,
 139                                        NULL,
 140                                        NULL);
 141           node_set (&iter);
 142         }
 143     }
 144 }
 145
 146 static void
 147 iter_prepend (GtkWidget *button, GtkTreeView *tree_view)
 148 {
 149   GtkTreeIter iter;
 150   GtkTreeIter selected;
 151   GtkTreeModel *model = gtk_tree_view_get_model (tree_view);
 152   GtkTreeSelection *selection = gtk_tree_view_get_selection (tree_view);
 153
 154   if (gtk_tree_selection_get_selected (selection, NULL, &selected))
 155     {
 156       if (GTK_IS_TREE_STORE (model))
 157         {
 158           gtk_tree_store_prepend (GTK_TREE_STORE (model),
 159                                   &iter,
 160                                   &selected);
 161           node_set (&iter);
 162         }
 163     }
 164   else
 165     {
 166       if (GTK_IS_TREE_STORE (model))
 167         {
 168           gtk_tree_store_prepend (GTK_TREE_STORE (model),
 169                                   &iter,
 170                                   NULL);
 171           node_set (&iter);
 172         }
 173     }
 174 }
 175
 176 static void
 177 iter_append (GtkWidget *button, GtkTreeView *tree_view)
 178 {
 179   GtkTreeIter iter;
 180   GtkTreeIter selected;
 181   GtkTreeStore *model = gtk_tree_view_get_model (tree_view);
 182
 183   if (gtk_tree_selection_get_selected (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)),
 184                                        NULL,
 185                                        &selected))
 186     {
 187       if (GTK_IS_TREE_STORE (model))
 188         {
 189           gtk_tree_store_append (model, &iter, &selected);
 190           node_set (&iter);
 191         }
 192     }
 193   else
 194     {
 195       if (GTK_IS_TREE_STORE (model))
 196         {
 197           gtk_tree_store_append (model, &iter, NULL);
 198           node_set (&iter);
 199         }
 200     }
 201 }
 202
 203 static void
 204 uppercase_value (const GValue *src, GValue *dest, gpointer data)
 205 {
 206   gchar *str;
 207
 208   g_value_init (dest, G_TYPE_STRING);
 209   str = g_strdup (g_value_get_string (src));
 210   if (str)
 211     g_strup (str);
 212   g_value_set_string (dest, str);
 213   g_free (str);
 214 }
 215
 216 static void
 217 make_window (gint view_type)
 218 {
 219   GtkWidget *window;
 220   GtkWidget *vbox;
 221   GtkWidget *hbox, *entry;
 222   GtkWidget *button;
 223   GtkWidget *scrolled_window;
 224   GtkWidget *tree_view;
 225   GtkTreeViewColumn *column;
 226   GtkCellRenderer *cell;
 227   GObject *selection;
 228
 229   /* Make the Widgets/Objects */
 230   window = gtk_window_new (GTK_WINDOW_TOPLEVEL);
 231   switch (view_type)
 232     {
 233     case 0:
 234       gtk_window_set_title (GTK_WINDOW (window), "Unsorted list");
 235       break;
 236     case 1:
 237       gtk_window_set_title (GTK_WINDOW (window), "Sorted list");
 238       break;
 239     }
 240
 241   vbox = gtk_vbox_new (FALSE, 8);
 242   gtk_container_set_border_width (GTK_CONTAINER (vbox), 8);
 243   gtk_window_set_default_size (GTK_WINDOW (window), 300, 350);
 244   scrolled_window = gtk_scrolled_window_new (NULL, NULL);
 245   switch (view_type)
 246     {
 247     case 0:
 248       tree_view = gtk_tree_view_new_with_model (GTK_TREE_MODEL (base_model));
 249       break;
 250     case 1:
 251       {
 252         GtkTreeModel *sort_model;
 253
 254         sort_model = gtk_tree_model_sort_new_with_model (GTK_TREE_MODEL (base_model));
 255         tree_view = gtk_tree_view_new_with_model (GTK_TREE_MODEL (sort_model));
 256       }
 257       break;
 258     default:
 259       g_assert_not_reached ();
 260       break;
 261     }
 262
 263   gtk_tree_view_set_rules_hint (GTK_TREE_VIEW (tree_view), TRUE);
 264   selection = G_OBJECT (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)));
 265   gtk_tree_selection_set_mode (GTK_TREE_SELECTION (selection), GTK_TREE_SELECTION_SINGLE);
 266
 267   /* Put them together */
 268   gtk_container_add (GTK_CONTAINER (scrolled_window), tree_view);
 269   gtk_box_pack_start (GTK_BOX (vbox), scrolled_window, TRUE, TRUE, 0);
 270   gtk_container_add (GTK_CONTAINER (window), vbox);
 271   gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (scrolled_window),
 272                                   GTK_POLICY_AUTOMATIC,
 273                                   GTK_POLICY_AUTOMATIC);
 274   gtk_signal_connect (GTK_OBJECT (window), "destroy", gtk_main_quit, NULL);
 275
 276   /* buttons */
 277   button = gtk_button_new_with_label ("gtk_tree_store_remove");
 278   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 279   g_signal_connect (G_OBJECT (selection),
 280                     "changed",
 281                     selection_changed,
 282                     button);
 283   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_remove, tree_view);
 284   gtk_widget_set_sensitive (button, FALSE);
 285
 286   button = gtk_button_new_with_label ("gtk_tree_store_insert");
 287   hbox = gtk_hbox_new (FALSE, 8);
 288   entry = gtk_entry_new ();
 289   gtk_box_pack_start (GTK_BOX (vbox), hbox, FALSE, FALSE, 0);
 290   gtk_box_pack_start (GTK_BOX (hbox), button, TRUE, TRUE, 0);
 291   gtk_box_pack_start (GTK_BOX (hbox), entry, FALSE, FALSE, 0);
 292   gtk_object_set_user_data (GTK_OBJECT (button), entry);
 293   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_insert, tree_view);
 294
 295
 296   button = gtk_button_new_with_label ("gtk_tree_store_insert_before");
 297   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 298   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_insert_before, tree_view);
 299   g_signal_connect (G_OBJECT (selection),
 300                     "changed",
 301                     selection_changed,
 302                     button);
 303   gtk_widget_set_sensitive (button, FALSE);
 304
 305   button = gtk_button_new_with_label ("gtk_tree_store_insert_after");
 306   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 307   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_insert_after, tree_view);
 308   g_signal_connect (G_OBJECT (selection),
 309                     "changed",
 310                     selection_changed,
 311                     button);
 312   gtk_widget_set_sensitive (button, FALSE);
 313
 314   button = gtk_button_new_with_label ("gtk_tree_store_prepend");
 315   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 316   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_prepend, tree_view);
 317
 318   button = gtk_button_new_with_label ("gtk_tree_store_append");
 319   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 320   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_append, tree_view);
 321
 322   /* The selected column */
 323   cell = gtk_cell_renderer_text_new ();
 324   column = gtk_tree_view_column_new_with_attributes ("Node ID", cell, "markup", 0, NULL);
 325   gtk_tree_view_column_set_sort_column_id (column, 0);
 326   gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), column);
 327   g_object_unref (G_OBJECT (cell));
 328   g_object_unref (G_OBJECT (column));
 329
 330   cell = gtk_cell_renderer_text_new ();
 331   column = gtk_tree_view_column_new_with_attributes ("Random Number", cell, "text", 1, NULL);
 332   gtk_tree_view_column_set_sort_column_id (column, 1);
 333   gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), column);
 334   g_object_unref (G_OBJECT (cell));
 335   g_object_unref (G_OBJECT (column));
 336
 337   /* A few to start */
 338   if (view_type == 0)
 339     {
 340       iter_append (NULL, GTK_TREE_VIEW (tree_view));
 341       iter_append (NULL, GTK_TREE_VIEW (tree_view));
 342       iter_append (NULL, GTK_TREE_VIEW (tree_view));
 343       iter_append (NULL, GTK_TREE_VIEW (tree_view));
 344       iter_append (NULL, GTK_TREE_VIEW (tree_view));
 345       iter_append (NULL, GTK_TREE_VIEW (tree_view));
 346     }
 347   /* Show it all */
 348   gtk_widget_show_all (window);
 349 }
 350
 351 int
 352 main (int argc, char *argv[])
 353 {
 354   gtk_init (&argc, &argv);
 355
 356   base_model = gtk_tree_store_new_with_types (2, G_TYPE_STRING, G_TYPE_STRING);
 357
 358   /* FIXME: reverse this */
 359   make_window (0);
 360   make_window (1);
 361
 362   gtk_main ();
 363
 364   return 0;
 365 }
 366