gtk2/tests/testtreechanging.c
Benjamin Otte 070124b0a5 tests: Add test resetting the model
We first set a NULL model and then reset the old model, just to get the
effect of clearing and then resetting.

We reset the cursor and selection afterwards, so the reset doesn't
destroy all the work we did.
2011-12-16 04:53:15 +01:00

523 lines
13 KiB
C

/* testtreeview.c
* Copyright (C) 2011 Red Hat, Inc
* Author: Benjamin Otte <otte@gnome.org>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
#include "prop-editor.h"
#include <gtk/gtk.h>
#define MIN_ROWS 50
#define MAX_ROWS 150
typedef void (* DoStuffFunc) (GtkTreeView *treeview);
static guint
count_children (GtkTreeModel *model,
GtkTreeIter *parent)
{
GtkTreeIter iter;
guint count = 0;
gboolean valid;
for (valid = gtk_tree_model_iter_children (model, &iter, parent);
valid;
valid = gtk_tree_model_iter_next (model, &iter))
{
count += count_children (model, &iter) + 1;
}
return count;
}
static void
set_rows (GtkTreeView *treeview, guint i)
{
g_assert (i == count_children (gtk_tree_view_get_model (treeview), NULL));
g_object_set_data (G_OBJECT (treeview), "rows", GUINT_TO_POINTER (i));
}
static guint
get_rows (GtkTreeView *treeview)
{
return GPOINTER_TO_UINT (g_object_get_data (G_OBJECT (treeview), "rows"));
}
static void
log_operation_for_path (GtkTreePath *path,
const char *operation_name)
{
char *path_string;
path_string = path ? gtk_tree_path_to_string (path) : g_strdup ("");
g_printerr ("%10s %s\n", operation_name, path_string);
g_free (path_string);
}
static void
log_operation (GtkTreeModel *model,
GtkTreeIter *iter,
const char *operation_name)
{
GtkTreePath *path;
path = gtk_tree_model_get_path (model, iter);
log_operation_for_path (path, operation_name);
gtk_tree_path_free (path);
}
/* moves iter to the next iter in the model in the display order
* inside a treeview. Returns FALSE if no more rows exist.
*/
static gboolean
tree_model_iter_step (GtkTreeModel *model,
GtkTreeIter *iter)
{
GtkTreeIter tmp;
if (gtk_tree_model_iter_children (model, &tmp, iter))
{
*iter = tmp;
return TRUE;
}
do {
tmp = *iter;
if (gtk_tree_model_iter_next (model, iter))
return TRUE;
}
while (gtk_tree_model_iter_parent (model, iter, &tmp));
return FALSE;
}
/* NB: may include invisible iters (because they are collapsed) */
static void
tree_view_random_iter (GtkTreeView *treeview,
GtkTreeIter *iter)
{
guint n_rows = get_rows (treeview);
guint i = g_random_int_range (0, n_rows);
GtkTreeModel *model;
model = gtk_tree_view_get_model (treeview);
if (!gtk_tree_model_get_iter_first (model, iter))
return;
while (i-- > 0)
{
if (!tree_model_iter_step (model, iter))
{
g_assert_not_reached ();
return;
}
}
return;
}
static void
delete (GtkTreeView *treeview)
{
guint n_rows = get_rows (treeview);
GtkTreeModel *model;
GtkTreeIter iter;
model = gtk_tree_view_get_model (treeview);
tree_view_random_iter (treeview, &iter);
n_rows -= count_children (model, &iter) + 1;
log_operation (model, &iter, "remove");
gtk_tree_store_remove (GTK_TREE_STORE (model), &iter);
set_rows (treeview, n_rows);
}
static void
add_one (GtkTreeModel *model,
GtkTreeIter *iter)
{
guint n = gtk_tree_model_iter_n_children (model, iter);
GtkTreeIter new_iter;
static guint counter = 0;
if (n > 0 && g_random_boolean ())
{
GtkTreeIter child;
gtk_tree_model_iter_nth_child (model, &child, iter, g_random_int_range (0, n));
add_one (model, &child);
return;
}
gtk_tree_store_insert_with_values (GTK_TREE_STORE (model),
&new_iter,
iter,
g_random_int_range (-1, n),
0, ++counter,
-1);
log_operation (model, &new_iter, "add");
}
static void
add (GtkTreeView *treeview)
{
GtkTreeModel *model;
model = gtk_tree_view_get_model (treeview);
add_one (model, NULL);
set_rows (treeview, get_rows (treeview) + 1);
}
static void
add_or_delete (GtkTreeView *treeview)
{
guint n_rows = get_rows (treeview);
if (g_random_int_range (MIN_ROWS, MAX_ROWS) >= n_rows)
add (treeview);
else
delete (treeview);
}
/* XXX: We only expand/collapse from the top and not randomly */
static void
expand (GtkTreeView *treeview)
{
GtkTreeModel *model;
GtkTreeIter iter;
GtkTreePath *path;
gboolean valid;
model = gtk_tree_view_get_model (treeview);
for (valid = gtk_tree_model_get_iter_first (model, &iter);
valid;
valid = tree_model_iter_step (model, &iter))
{
if (gtk_tree_model_iter_has_child (model, &iter))
{
path = gtk_tree_model_get_path (model, &iter);
if (!gtk_tree_view_row_expanded (treeview, path))
{
log_operation (model, &iter, "expand");
gtk_tree_view_expand_row (treeview, path, FALSE);
gtk_tree_path_free (path);
return;
}
gtk_tree_path_free (path);
}
}
}
static void
collapse (GtkTreeView *treeview)
{
GtkTreeModel *model;
GtkTreeIter iter;
GtkTreePath *last, *path;
gboolean valid;
model = gtk_tree_view_get_model (treeview);
last = NULL;
for (valid = gtk_tree_model_get_iter_first (model, &iter);
valid;
valid = tree_model_iter_step (model, &iter))
{
path = gtk_tree_model_get_path (model, &iter);
if (gtk_tree_view_row_expanded (treeview, path))
{
if (last)
gtk_tree_path_free (last);
last = path;
}
else
gtk_tree_path_free (path);
}
if (last)
{
log_operation_for_path (last, "collapse");
gtk_tree_view_collapse_row (treeview, last);
gtk_tree_path_free (last);
}
}
static void
select_ (GtkTreeView *treeview)
{
GtkTreeIter iter;
tree_view_random_iter (treeview, &iter);
log_operation (gtk_tree_view_get_model (treeview), &iter, "select");
gtk_tree_selection_select_iter (gtk_tree_view_get_selection (treeview),
&iter);
}
static void
unselect (GtkTreeView *treeview)
{
GtkTreeIter iter;
tree_view_random_iter (treeview, &iter);
log_operation (gtk_tree_view_get_model (treeview), &iter, "unselect");
gtk_tree_selection_unselect_iter (gtk_tree_view_get_selection (treeview),
&iter);
}
static void
reset_model (GtkTreeView *treeview)
{
GtkTreeSelection *selection;
GtkTreeModel *model;
GList *list, *selected;
GtkTreePath *cursor;
selection = gtk_tree_view_get_selection (treeview);
model = g_object_ref (gtk_tree_view_get_model (treeview));
log_operation_for_path (NULL, "reset");
selected = gtk_tree_selection_get_selected_rows (selection, NULL);
gtk_tree_view_get_cursor (treeview, &cursor, NULL);
gtk_tree_view_set_model (treeview, NULL);
gtk_tree_view_set_model (treeview, model);
if (cursor)
{
gtk_tree_view_set_cursor (treeview, cursor, NULL, FALSE);
gtk_tree_path_free (cursor);
}
for (list = selected; list; list = list->next)
{
gtk_tree_selection_select_path (selection, list->data);
}
g_list_free_full (selected, (GDestroyNotify) gtk_tree_path_free);
g_object_unref (model);
}
/* sanity checks */
static void
assert_row_reference_is_path (GtkTreeRowReference *ref,
GtkTreePath *path)
{
GtkTreePath *expected;
if (ref == NULL)
{
g_assert (path == NULL);
return;
}
g_assert (path != NULL);
g_assert (gtk_tree_row_reference_valid (ref));
expected = gtk_tree_row_reference_get_path (ref);
g_assert (expected != NULL);
g_assert (gtk_tree_path_compare (expected, path) == 0);
gtk_tree_path_free (expected);
}
static void
check_cursor (GtkTreeView *treeview)
{
GtkTreeRowReference *ref = g_object_get_data (G_OBJECT (treeview), "cursor");
GtkTreePath *cursor;
gtk_tree_view_get_cursor (treeview, &cursor, NULL);
assert_row_reference_is_path (ref, cursor);
if (cursor)
gtk_tree_path_free (cursor);
}
static void
check_selection_item (GtkTreeModel *model,
GtkTreePath *path,
GtkTreeIter *iter,
gpointer listp)
{
GList **list = listp;
g_assert (*list);
assert_row_reference_is_path ((*list)->data, path);
*list = (*list)->next;
}
static void
check_selection (GtkTreeView *treeview)
{
GList *selection = g_object_get_data (G_OBJECT (treeview), "selection");
gtk_tree_selection_selected_foreach (gtk_tree_view_get_selection (treeview),
check_selection_item,
&selection);
}
static void
check_sanity (GtkTreeView *treeview)
{
check_cursor (treeview);
check_selection (treeview);
}
static gboolean
dance (gpointer treeview)
{
static const DoStuffFunc funcs[] = {
add_or_delete,
add_or_delete,
expand,
collapse,
select_,
unselect,
reset_model
};
guint i;
i = g_random_int_range (0, G_N_ELEMENTS(funcs));
funcs[i] (treeview);
check_sanity (treeview);
return TRUE;
}
static void
cursor_changed_cb (GtkTreeView *treeview,
gpointer unused)
{
GtkTreePath *path;
GtkTreeRowReference *ref;
gtk_tree_view_get_cursor (treeview, &path, NULL);
if (path)
{
ref = gtk_tree_row_reference_new (gtk_tree_view_get_model (treeview),
path);
gtk_tree_path_free (path);
}
else
ref = NULL;
g_object_set_data_full (G_OBJECT (treeview), "cursor", ref, (GDestroyNotify) gtk_tree_row_reference_free);
}
static void
selection_list_free (gpointer list)
{
g_list_free_full (list, (GDestroyNotify) gtk_tree_row_reference_free);
}
static void
selection_changed_cb (GtkTreeSelection *tree_selection,
gpointer unused)
{
GList *selected, *list;
GtkTreeModel *model;
selected = gtk_tree_selection_get_selected_rows (tree_selection, &model);
for (list = selected; list; list = list->next)
{
GtkTreePath *path = list->data;
list->data = gtk_tree_row_reference_new (model, path);
gtk_tree_path_free (path);
}
g_object_set_data_full (G_OBJECT (gtk_tree_selection_get_tree_view (tree_selection)),
"selection",
selected,
selection_list_free);
}
static void
setup_sanity_checks (GtkTreeView *treeview)
{
g_signal_connect (treeview, "cursor-changed", G_CALLBACK (cursor_changed_cb), NULL);
cursor_changed_cb (treeview, NULL);
g_signal_connect (gtk_tree_view_get_selection (treeview), "changed", G_CALLBACK (selection_changed_cb), NULL);
selection_changed_cb (gtk_tree_view_get_selection (treeview), NULL);
}
int
main (int argc,
char **argv)
{
GtkWidget *window;
GtkWidget *sw;
GtkWidget *treeview;
GtkTreeModel *model;
guint i;
gtk_init (&argc, &argv);
if (g_getenv ("RTL"))
gtk_widget_set_default_direction (GTK_TEXT_DIR_RTL);
window = gtk_window_new (GTK_WINDOW_TOPLEVEL);
g_signal_connect (window, "destroy", G_CALLBACK (gtk_main_quit), NULL);
gtk_window_set_default_size (GTK_WINDOW (window), 430, 400);
sw = gtk_scrolled_window_new (NULL, NULL);
gtk_widget_set_hexpand (sw, TRUE);
gtk_widget_set_vexpand (sw, TRUE);
gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (sw),
GTK_POLICY_AUTOMATIC,
GTK_POLICY_AUTOMATIC);
gtk_container_add (GTK_CONTAINER (window), sw);
model = GTK_TREE_MODEL (gtk_tree_store_new (1, G_TYPE_UINT));
treeview = gtk_tree_view_new_with_model (model);
g_object_unref (model);
setup_sanity_checks (GTK_TREE_VIEW (treeview));
gtk_tree_view_insert_column_with_attributes (GTK_TREE_VIEW (treeview),
0,
"Counter",
gtk_cell_renderer_text_new (),
"text", 0,
NULL);
for (i = 0; i < (MIN_ROWS + MAX_ROWS) / 2; i++)
add (GTK_TREE_VIEW (treeview));
gtk_container_add (GTK_CONTAINER (sw), treeview);
create_prop_editor (G_OBJECT (treeview), GTK_TYPE_TREE_VIEW);
create_prop_editor (G_OBJECT (gtk_tree_view_get_selection (GTK_TREE_VIEW (treeview))), GTK_TYPE_TREE_SELECTION);
gtk_widget_show_all (window);
g_idle_add (dance, treeview);
gtk_main ();
return 0;
}