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 (%d)", 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 make_window (gboolean use_sort)
 205 {
 206   GtkWidget *window;
 207   GtkWidget *vbox;
 208   GtkWidget *hbox, *entry;
 209   GtkWidget *button;
 210   GtkWidget *scrolled_window;
 211   GtkWidget *tree_view;
 212   GtkTreeViewColumn *column;
 213   GtkCellRenderer *cell;
 214   GtkObject *selection;
 215
 216   /* Make the Widgets/Objects */
 217   window = gtk_window_new (GTK_WINDOW_TOPLEVEL);
 218   if (use_sort)
 219     gtk_window_set_title (GTK_WINDOW (window), "Sorted list");
 220   else
 221     gtk_window_set_title (GTK_WINDOW (window), "Unsorted list");
 222
 223   vbox = gtk_vbox_new (FALSE, 8);
 224   gtk_container_set_border_width (GTK_CONTAINER (vbox), 8);
 225   gtk_window_set_default_size (GTK_WINDOW (window), 300, 350);
 226   scrolled_window = gtk_scrolled_window_new (NULL, NULL);
 227   if (use_sort)
 228     {
 229       GtkTreeModel *sort_model;
 230
 231       sort_model = gtk_tree_model_sort_new_with_model (GTK_TREE_MODEL (base_model),
 232                                                        NULL, 1);
 233       tree_view = gtk_tree_view_new_with_model (GTK_TREE_MODEL (sort_model));
 234     }
 235   else
 236     {
 237       tree_view = gtk_tree_view_new_with_model (GTK_TREE_MODEL (base_model));
 238     }
 239
 240   selection = GTK_OBJECT (gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view)));
 241   gtk_tree_selection_set_type (GTK_TREE_SELECTION (selection), GTK_TREE_SELECTION_SINGLE);
 242
 243   /* Put them together */
 244   gtk_container_add (GTK_CONTAINER (scrolled_window), tree_view);
 245   gtk_box_pack_start (GTK_BOX (vbox), scrolled_window, TRUE, TRUE, 0);
 246   gtk_container_add (GTK_CONTAINER (window), vbox);
 247   gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (scrolled_window),
 248                                   GTK_POLICY_AUTOMATIC,
 249                                   GTK_POLICY_AUTOMATIC);
 250   gtk_signal_connect (GTK_OBJECT (window), "destroy", gtk_main_quit, NULL);
 251
 252   /* buttons */
 253   button = gtk_button_new_with_label ("gtk_tree_store_remove");
 254   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 255   gtk_signal_connect (GTK_OBJECT (selection),
 256                       "selection_changed",
 257                       selection_changed,
 258                       button);
 259   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_remove, tree_view);
 260   gtk_widget_set_sensitive (button, FALSE);
 261
 262   button = gtk_button_new_with_label ("gtk_tree_store_insert");
 263   hbox = gtk_hbox_new (FALSE, 8);
 264   entry = gtk_entry_new ();
 265   gtk_box_pack_start (GTK_BOX (vbox), hbox, FALSE, FALSE, 0);
 266   gtk_box_pack_start (GTK_BOX (hbox), button, TRUE, TRUE, 0);
 267   gtk_box_pack_start (GTK_BOX (hbox), entry, FALSE, FALSE, 0);
 268   gtk_object_set_user_data (GTK_OBJECT (button), entry);
 269   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_insert, tree_view);
 270
 271
 272   button = gtk_button_new_with_label ("gtk_tree_store_insert_before");
 273   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 274   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_insert_before, tree_view);
 275   gtk_signal_connect (GTK_OBJECT (selection),
 276                       "selection_changed",
 277                       selection_changed,
 278                       button);
 279   gtk_widget_set_sensitive (button, FALSE);
 280
 281   button = gtk_button_new_with_label ("gtk_tree_store_insert_after");
 282   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 283   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_insert_after, tree_view);
 284   gtk_signal_connect (GTK_OBJECT (selection),
 285                       "selection_changed",
 286                       selection_changed,
 287                       button);
 288   gtk_widget_set_sensitive (button, FALSE);
 289
 290   button = gtk_button_new_with_label ("gtk_tree_store_prepend");
 291   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 292   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_prepend, tree_view);
 293
 294   button = gtk_button_new_with_label ("gtk_tree_store_append");
 295   gtk_box_pack_start (GTK_BOX (vbox), button, FALSE, FALSE, 0);
 296   gtk_signal_connect (GTK_OBJECT (button), "clicked", iter_append, tree_view);
 297
 298   /* The selected column */
 299   cell = gtk_cell_renderer_text_new ();
 300   column = gtk_tree_view_column_new_with_attributes ("Node ID", cell, "text", 0, NULL);
 301   gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), column);
 302
 303   cell = gtk_cell_renderer_text_new ();
 304   column = gtk_tree_view_column_new_with_attributes ("Random Number", cell, "text", 1, NULL);
 305   gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), column);
 306
 307   /* A few to start */
 308   if (!use_sort)
 309     {
 310       iter_append (NULL, GTK_TREE_VIEW (tree_view));
 311       iter_append (NULL, GTK_TREE_VIEW (tree_view));
 312       iter_append (NULL, GTK_TREE_VIEW (tree_view));
 313       iter_append (NULL, GTK_TREE_VIEW (tree_view));
 314       iter_append (NULL, GTK_TREE_VIEW (tree_view));
 315       iter_append (NULL, GTK_TREE_VIEW (tree_view));
 316     }
 317   /* Show it all */
 318   gtk_widget_show_all (window);
 319 }
 320
 321 int
 322 main (int argc, char *argv[])
 323 {
 324   gtk_init (&argc, &argv);
 325
 326   base_model = gtk_tree_store_new_with_types (2, G_TYPE_STRING, G_TYPE_STRING);
 327
 328   /* FIXME: reverse this */
 329   make_window (FALSE);
 330   make_window (TRUE);
 331
 332   gtk_main ();
 333
 334   return 0;
 335 }