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 static void
  61 delete (GtkTreeView *treeview)
  62 {
  63   guint n_rows = get_rows (treeview);
  64   guint i = g_random_int_range (0, n_rows);
  65   GtkTreeModel *model;
  66   GtkTreeIter iter, next;
  67
  68   model = gtk_tree_view_get_model (treeview);
  69
  70   if (!gtk_tree_model_get_iter_first (model, &iter))
  71     return;
  72
  73
  74   while (i-- > 0)
  75     {
  76       next = iter;
  77       if (!gtk_tree_model_iter_children (model, &iter, &next) &&
  78           !gtk_tree_model_iter_next (model, &next) &&
  79           !gtk_tree_model_iter_parent (model, &next, &iter))
  80         {
  81           g_assert_not_reached ();
  82           return;
  83         }
  84       iter = next;
  85     }
  86
  87   n_rows -= count_children (model, &iter) + 1;
  88   gtk_tree_store_remove (GTK_TREE_STORE (model), &iter);
  89   set_rows (treeview, n_rows);
  90 }
  91
  92 static void
  93 add_one (GtkTreeModel *model,
  94          GtkTreeIter *iter)
  95 {
  96   guint n = gtk_tree_model_iter_n_children (model, iter);
  97   static guint counter = 0;
  98
  99   if (n > 0 && g_random_boolean ())
 100     {
 101       GtkTreeIter child;
 102       gtk_tree_model_iter_nth_child (model, &child, iter, g_random_int_range (0, n));
 103       add_one (model, &child);
 104       return;
 105     }
 106
 107   gtk_tree_store_insert_with_values (GTK_TREE_STORE (model),
 108                                      NULL,
 109                                      iter,
 110                                      g_random_int_range (-1, n),
 111                                      0, ++counter,
 112                                      -1);
 113 }
 114
 115 static void
 116 add (GtkTreeView *treeview)
 117 {
 118   GtkTreeModel *model;
 119
 120   model = gtk_tree_view_get_model (treeview);
 121   add_one (model, NULL);
 122
 123   set_rows (treeview, get_rows (treeview) + 1);
 124 }
 125
 126 static void
 127 add_or_delete (GtkTreeView *treeview)
 128 {
 129   guint n_rows = get_rows (treeview);
 130
 131   if (g_random_int_range (MIN_ROWS, MAX_ROWS) >= n_rows)
 132     add (treeview);
 133   else
 134     delete (treeview);
 135 }
 136
 137 static gboolean
 138 dance (gpointer treeview)
 139 {
 140   static const DoStuffFunc funcs[] = {
 141     add_or_delete,
 142     add_or_delete
 143   };
 144
 145   funcs[g_random_int_range (0, G_N_ELEMENTS(funcs))] (treeview);
 146
 147   return TRUE;
 148 }
 149
 150 int
 151 main (int    argc,
 152       char **argv)
 153 {
 154   GtkWidget *window;
 155   GtkWidget *sw;
 156   GtkWidget *treeview;
 157   GtkTreeModel *model;
 158   guint i;
 159
 160   gtk_init (&argc, &argv);
 161
 162   if (g_getenv ("RTL"))
 163     gtk_widget_set_default_direction (GTK_TEXT_DIR_RTL);
 164
 165   window = gtk_window_new (GTK_WINDOW_TOPLEVEL);
 166   g_signal_connect (window, "destroy", G_CALLBACK (gtk_main_quit), NULL);
 167   gtk_window_set_default_size (GTK_WINDOW (window), 430, 400);
 168
 169   sw = gtk_scrolled_window_new (NULL, NULL);
 170   gtk_widget_set_hexpand (sw, TRUE);
 171   gtk_widget_set_vexpand (sw, TRUE);
 172   gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (sw),
 173                                   GTK_POLICY_AUTOMATIC,
 174                                   GTK_POLICY_AUTOMATIC);
 175   gtk_container_add (GTK_CONTAINER (window), sw);
 176
 177   model = GTK_TREE_MODEL (gtk_tree_store_new (1, G_TYPE_UINT));
 178   treeview = gtk_tree_view_new_with_model (model);
 179   g_object_unref (model);
 180   gtk_tree_view_insert_column_with_attributes (GTK_TREE_VIEW (treeview),
 181                                                0,
 182                                                "Counter",
 183                                                gtk_cell_renderer_text_new (),
 184                                                "text", 0,
 185                                                NULL);
 186   for (i = 0; i < (MIN_ROWS + MAX_ROWS) / 2; i++)
 187     add (GTK_TREE_VIEW (treeview));
 188   gtk_container_add (GTK_CONTAINER (sw), treeview);
 189
 190   create_prop_editor (G_OBJECT (treeview), GTK_TYPE_TREE_VIEW);
 191   create_prop_editor (G_OBJECT (gtk_tree_view_get_selection (GTK_TREE_VIEW (treeview))), GTK_TYPE_TREE_SELECTION);
 192
 193   gtk_widget_show_all (window);
 194
 195   g_timeout_add (50, dance, treeview);
 196
 197   gtk_main ();
 198
 199   return 0;
 200 }
 201