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

   1 /* testtreeview.c
   2  * Copyright (C) 2011 Red Hat, Inc
   3  * Author: Benjamin Otte <otte@gnome.org>
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Library General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Library General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Library General Public
  16  * License along with this library; if not, write to the
  17  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18  * Boston, MA 02111-1307, USA.
  19  */
  20
  21 #include "prop-editor.h"
  22 #include <gtk/gtk.h>
  23
  24 #define MIN_ROWS 50
  25 #define MAX_ROWS 150
  26
  27 typedef void (* DoStuffFunc) (GtkTreeView *treeview);
  28
  29 static guint
  30 count_children (GtkTreeModel *model,
  31                 GtkTreeIter  *parent)
  32 {
  33   GtkTreeIter iter;
  34   guint count = 0;
  35   gboolean valid;
  36
  37   for (valid = gtk_tree_model_iter_children (model, &iter, parent);
  38        valid;
  39        valid = gtk_tree_model_iter_next (model, &iter))
  40     {
  41       count += count_children (model, &iter) + 1;
  42     }
  43
  44   return count;
  45 }
  46
  47 static void
  48 set_rows (GtkTreeView *treeview, guint i)
  49 {
  50   g_assert (i == count_children (gtk_tree_view_get_model (treeview), NULL));
  51   g_object_set_data (G_OBJECT (treeview), "rows", GUINT_TO_POINTER (i));
  52 }
  53
  54 static guint
  55 get_rows (GtkTreeView *treeview)
  56 {
  57   return GPOINTER_TO_UINT (g_object_get_data (G_OBJECT (treeview), "rows"));
  58 }
  59
  60 /* moves iter to the next iter in the model in the display order
  61  * inside a treeview. Returns FALSE if no more rows exist.
  62  */
  63 static gboolean
  64 tree_model_iter_step (GtkTreeModel *model,
  65                       GtkTreeIter *iter)
  66 {
  67   GtkTreeIter tmp;
  68
  69   if (gtk_tree_model_iter_children (model, &tmp, iter))
  70     {
  71       *iter = tmp;
  72       return TRUE;
  73     }
  74
  75   do {
  76     tmp = *iter;
  77
  78     if (gtk_tree_model_iter_next (model, iter))
  79       return TRUE;
  80     }
  81   while (gtk_tree_model_iter_parent (model, iter, &tmp));
  82
  83   return FALSE;
  84 }
  85
  86 static void
  87 delete (GtkTreeView *treeview)
  88 {
  89   guint n_rows = get_rows (treeview);
  90   guint i = g_random_int_range (0, n_rows);
  91   GtkTreeModel *model;
  92   GtkTreeIter iter;
  93
  94   model = gtk_tree_view_get_model (treeview);
  95
  96   if (!gtk_tree_model_get_iter_first (model, &iter))
  97     return;
  98
  99   while (i-- > 0)
 100     {
 101       if (!tree_model_iter_step (model, &iter))
 102         {
 103           g_assert_not_reached ();
 104           return;
 105         }
 106     }
 107
 108   n_rows -= count_children (model, &iter) + 1;
 109   gtk_tree_store_remove (GTK_TREE_STORE (model), &iter);
 110   set_rows (treeview, n_rows);
 111 }
 112
 113 static void
 114 add_one (GtkTreeModel *model,
 115          GtkTreeIter *iter)
 116 {
 117   guint n = gtk_tree_model_iter_n_children (model, iter);
 118   static guint counter = 0;
 119
 120   if (n > 0 && g_random_boolean ())
 121     {
 122       GtkTreeIter child;
 123       gtk_tree_model_iter_nth_child (model, &child, iter, g_random_int_range (0, n));
 124       add_one (model, &child);
 125       return;
 126     }
 127
 128   gtk_tree_store_insert_with_values (GTK_TREE_STORE (model),
 129                                      NULL,
 130                                      iter,
 131                                      g_random_int_range (-1, n),
 132                                      0, ++counter,
 133                                      -1);
 134 }
 135
 136 static void
 137 add (GtkTreeView *treeview)
 138 {
 139   GtkTreeModel *model;
 140
 141   model = gtk_tree_view_get_model (treeview);
 142   add_one (model, NULL);
 143
 144   set_rows (treeview, get_rows (treeview) + 1);
 145 }
 146
 147 static void
 148 add_or_delete (GtkTreeView *treeview)
 149 {
 150   guint n_rows = get_rows (treeview);
 151
 152   if (g_random_int_range (MIN_ROWS, MAX_ROWS) >= n_rows)
 153     add (treeview);
 154   else
 155     delete (treeview);
 156 }
 157
 158 /* XXX: We only expand/collapse from the top and not randomly */
 159 static void
 160 expand (GtkTreeView *treeview)
 161 {
 162   GtkTreeModel *model;
 163   GtkTreeIter iter;
 164   GtkTreePath *path;
 165   gboolean valid;
 166
 167   model = gtk_tree_view_get_model (treeview);
 168
 169   for (valid = gtk_tree_model_get_iter_first (model, &iter);
 170        valid;
 171        valid = tree_model_iter_step (model, &iter))
 172     {
 173       if (gtk_tree_model_iter_has_child (model, &iter))
 174         {
 175           path = gtk_tree_model_get_path (model, &iter);
 176           if (!gtk_tree_view_row_expanded (treeview, path))
 177             {
 178               gtk_tree_view_expand_row (treeview, path, FALSE);
 179               gtk_tree_path_free (path);
 180               return;
 181             }
 182           gtk_tree_path_free (path);
 183         }
 184     }
 185 }
 186
 187 static void
 188 collapse (GtkTreeView *treeview)
 189 {
 190   GtkTreeModel *model;
 191   GtkTreeIter iter;
 192   GtkTreePath *last, *path;
 193   gboolean valid;
 194
 195   model = gtk_tree_view_get_model (treeview);
 196   last = NULL;
 197
 198   for (valid = gtk_tree_model_get_iter_first (model, &iter);
 199        valid;
 200        valid = tree_model_iter_step (model, &iter))
 201     {
 202       path = gtk_tree_model_get_path (model, &iter);
 203       if (gtk_tree_view_row_expanded (treeview, path))
 204         {
 205           if (last)
 206             gtk_tree_path_free (last);
 207           last = path;
 208         }
 209       else
 210         gtk_tree_path_free (path);
 211     }
 212
 213   if (last)
 214     {
 215       gtk_tree_view_collapse_row (treeview, last);
 216       gtk_tree_path_free (last);
 217     }
 218 }
 219
 220 static void
 221 check_cursor (GtkTreeView *treeview)
 222 {
 223   GtkTreeRowReference *ref = g_object_get_data (G_OBJECT (treeview), "cursor");
 224   GtkTreePath *expected, *cursor;
 225
 226   gtk_tree_view_get_cursor (treeview, &cursor, NULL);
 227   if (ref == NULL)
 228     {
 229       g_assert (cursor == NULL);
 230     }
 231   else
 232     {
 233       g_assert (cursor != NULL);
 234       g_assert (gtk_tree_row_reference_valid (ref));
 235
 236       expected = gtk_tree_row_reference_get_path (ref);
 237       g_assert (expected != NULL);
 238       g_assert (gtk_tree_path_compare (expected, cursor) == 0);
 239
 240       gtk_tree_path_free (expected);
 241     }
 242
 243   if (cursor)
 244     gtk_tree_path_free (cursor);
 245 }
 246
 247 static void
 248 check_sanity (GtkTreeView *treeview)
 249 {
 250   check_cursor (treeview);
 251 }
 252
 253 static gboolean
 254 dance (gpointer treeview)
 255 {
 256   static const DoStuffFunc funcs[] = {
 257     add_or_delete,
 258     add_or_delete,
 259     expand,
 260     collapse
 261   };
 262   guint i;
 263
 264   i = g_random_int_range (0, G_N_ELEMENTS(funcs));
 265
 266   funcs[i] (treeview);
 267
 268   check_sanity (treeview);
 269
 270   return TRUE;
 271 }
 272
 273 static void
 274 cursor_changed_cb (GtkTreeView *treeview,
 275                    gpointer     unused)
 276 {
 277   GtkTreePath *path;
 278   GtkTreeRowReference *ref;
 279
 280   gtk_tree_view_get_cursor (treeview, &path, NULL);
 281   if (path)
 282     {
 283       ref = gtk_tree_row_reference_new (gtk_tree_view_get_model (treeview),
 284                                         path);
 285       gtk_tree_path_free (path);
 286     }
 287   else
 288     ref = NULL;
 289   g_object_set_data_full (G_OBJECT (treeview), "cursor", ref, (GDestroyNotify) gtk_tree_row_reference_free);
 290 }
 291
 292 static void
 293 setup_sanity_checks (GtkTreeView *treeview)
 294 {
 295   g_signal_connect (treeview, "cursor-changed", G_CALLBACK (cursor_changed_cb), NULL);
 296   cursor_changed_cb (treeview, NULL);
 297 }
 298
 299 int
 300 main (int    argc,
 301       char **argv)
 302 {
 303   GtkWidget *window;
 304   GtkWidget *sw;
 305   GtkWidget *treeview;
 306   GtkTreeModel *model;
 307   guint i;
 308
 309   gtk_init (&argc, &argv);
 310
 311   if (g_getenv ("RTL"))
 312     gtk_widget_set_default_direction (GTK_TEXT_DIR_RTL);
 313
 314   window = gtk_window_new (GTK_WINDOW_TOPLEVEL);
 315   g_signal_connect (window, "destroy", G_CALLBACK (gtk_main_quit), NULL);
 316   gtk_window_set_default_size (GTK_WINDOW (window), 430, 400);
 317
 318   sw = gtk_scrolled_window_new (NULL, NULL);
 319   gtk_widget_set_hexpand (sw, TRUE);
 320   gtk_widget_set_vexpand (sw, TRUE);
 321   gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (sw),
 322                                   GTK_POLICY_AUTOMATIC,
 323                                   GTK_POLICY_AUTOMATIC);
 324   gtk_container_add (GTK_CONTAINER (window), sw);
 325
 326   model = GTK_TREE_MODEL (gtk_tree_store_new (1, G_TYPE_UINT));
 327   treeview = gtk_tree_view_new_with_model (model);
 328   g_object_unref (model);
 329   setup_sanity_checks (GTK_TREE_VIEW (treeview));
 330   gtk_tree_view_insert_column_with_attributes (GTK_TREE_VIEW (treeview),
 331                                                0,
 332                                                "Counter",
 333                                                gtk_cell_renderer_text_new (),
 334                                                "text", 0,
 335                                                NULL);
 336   for (i = 0; i < (MIN_ROWS + MAX_ROWS) / 2; i++)
 337     add (GTK_TREE_VIEW (treeview));
 338   gtk_container_add (GTK_CONTAINER (sw), treeview);
 339
 340   create_prop_editor (G_OBJECT (treeview), GTK_TYPE_TREE_VIEW);
 341   create_prop_editor (G_OBJECT (gtk_tree_view_get_selection (GTK_TREE_VIEW (treeview))), GTK_TYPE_TREE_SELECTION);
 342
 343   gtk_widget_show_all (window);
 344
 345   g_idle_add (dance, treeview);
 346
 347   gtk_main ();
 348
 349   return 0;
 350 }
 351