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 ();
  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   selection = G_OBJECT (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)));
 264   gtk_tree_selection_set_mode (GTK_TREE_SELECTION (selection), GTK_TREE_SELECTION_SINGLE);
 265
 266   /* Put them together */
 267   gtk_container_add (GTK_CONTAINER (scrolled_window), tree_view);
 268   gtk_box_pack_start (GTK_BOX (vbox), scrolled_window, TRUE, TRUE, 0);
 269   gtk_container_add (GTK_CONTAINER (window), vbox);
 270   gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (scrolled_window),
 271                                   GTK_POLICY_AUTOMATIC,
 272                                   GTK_POLICY_AUTOMATIC);
 273   gtk_signal_connect (GTK_OBJECT (window), "destroy", gtk_main_quit, NULL);
 274
 275   /* buttons */
 276   button = gtk_button_new_with_label ("gtk_tree_store_remove");
 277   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 278   g_signal_connectc (G_OBJECT (selection),
 279                      "changed",
 280                      selection_changed,
 281                      button, FALSE);
 282   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_remove, tree_view);
 283   gtk_widget_set_sensitive (button, FALSE);
 284
 285   button = gtk_button_new_with_label ("gtk_tree_store_insert");
 286   hbox = gtk_hbox_new (FALSE, 8);
 287   entry = gtk_entry_new ();
 288   gtk_box_pack_start (GTK_BOX (vbox), hbox, FALSE, FALSE, 0);
 289   gtk_box_pack_start (GTK_BOX (hbox), button, TRUE, TRUE, 0);
 290   gtk_box_pack_start (GTK_BOX (hbox), entry, FALSE, FALSE, 0);
 291   gtk_object_set_user_data (GTK_OBJECT (button), entry);
 292   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_insert, tree_view);
 293
 294
 295   button = gtk_button_new_with_label ("gtk_tree_store_insert_before");
 296   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 297   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_insert_before, tree_view);
 298   g_signal_connectc (G_OBJECT (selection),
 299                      "changed",
 300                      selection_changed,
 301                      button, FALSE);
 302   gtk_widget_set_sensitive (button, FALSE);
 303
 304   button = gtk_button_new_with_label ("gtk_tree_store_insert_after");
 305   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 306   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_insert_after, tree_view);
 307   g_signal_connectc (G_OBJECT (selection),
 308                      "changed",
 309                      selection_changed,
 310                      button, FALSE);
 311   gtk_widget_set_sensitive (button, FALSE);
 312
 313   button = gtk_button_new_with_label ("gtk_tree_store_prepend");
 314   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 315   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_prepend, tree_view);
 316
 317   button = gtk_button_new_with_label ("gtk_tree_store_append");
 318   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 319   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_append, tree_view);
 320
 321   /* The selected column */
 322   cell = gtk_cell_renderer_text_new ();
 323   column = gtk_tree_view_column_new_with_attributes ("Node ID", cell, "markup", 0, NULL);
 324   gtk_tree_view_column_set_sort_column_id (column, 0);
 325   gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), column);
 326   g_object_unref (G_OBJECT (cell));
 327   g_object_unref (G_OBJECT (column));
 328
 329   cell = gtk_cell_renderer_text_new ();
 330   column = gtk_tree_view_column_new_with_attributes ("Random Number", cell, "text", 1, NULL);
 331   gtk_tree_view_column_set_sort_column_id (column, 1);
 332   gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), column);
 333   g_object_unref (G_OBJECT (cell));
 334   g_object_unref (G_OBJECT (column));
 335
 336   /* A few to start */
 337   if (view_type == 0)
 338     {
 339       iter_append (NULL, GTK_TREE_VIEW (tree_view));
 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       gtk_widget_show_all (window);
 346     }
 347   /* Show it all */
 348 }
 349
 350 int
 351 main (int argc, char *argv[])
 352 {
 353   gtk_init (&argc, &argv);
 354
 355   base_model = gtk_tree_store_new_with_types (2, G_TYPE_STRING, G_TYPE_STRING);
 356
 357   /* FIXME: reverse this */
 358   make_window (0);
 359   //  make_window (1);
 360
 361   gtk_main ();
 362
 363   return 0;
 364 }
 365