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

   1 /* testtreeview.c
   2  * Copyright (C) 2001 Red Hat, Inc
   3  * Author: Jonathan Blandford
   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 <string.h>
  22 #include "prop-editor.h"
  23 #include <gtk/gtk.h>
  24 #include <stdlib.h>
  25
  26 /* Don't copy this bad example; inline RGB data is always a better
  27  * idea than inline XPMs.
  28  */
  29 static char  *book_closed_xpm[] = {
  30 "16 16 6 1",
  31 "       c None s None",
  32 ".      c black",
  33 "X      c red",
  34 "o      c yellow",
  35 "O      c #808080",
  36 "#      c white",
  37 "                ",
  38 "       ..       ",
  39 "     ..XX.      ",
  40 "   ..XXXXX.     ",
  41 " ..XXXXXXXX.    ",
  42 ".ooXXXXXXXXX.   ",
  43 "..ooXXXXXXXXX.  ",
  44 ".X.ooXXXXXXXXX. ",
  45 ".XX.ooXXXXXX..  ",
  46 " .XX.ooXXX..#O  ",
  47 "  .XX.oo..##OO. ",
  48 "   .XX..##OO..  ",
  49 "    .X.#OO..    ",
  50 "     ..O..      ",
  51 "      ..        ",
  52 "                "
  53 };
  54
  55 static void run_automated_tests (void);
  56
  57 /* This custom model is to test custom model use. */
  58
  59 #define GTK_TYPE_MODEL_TYPES                            (gtk_tree_model_types_get_type ())
  60 #define GTK_TREE_MODEL_TYPES(obj)                       (G_TYPE_CHECK_INSTANCE_CAST ((obj), GTK_TYPE_MODEL_TYPES, GtkTreeModelTypes))
  61 #define GTK_TREE_MODEL_TYPES_CLASS(klass)               (G_TYPE_CHECK_CLASS_CAST ((klass), GTK_TYPE_MODEL_TYPES, GtkTreeModelTypesClass))
  62 #define GTK_IS_TREE_MODEL_TYPES(obj)                    (G_TYPE_CHECK_INSTANCE_TYPE ((obj), GTK_TYPE_MODEL_TYPES))
  63 #define GTK_IS_TREE_MODEL_TYPES_GET_CLASS(klass)        (G_TYPE_INSTANCE_GET_CLASS ((obj), GTK_TYPE_MODEL_TYPES))
  64
  65 typedef struct _GtkTreeModelTypes       GtkTreeModelTypes;
  66 typedef struct _GtkTreeModelTypesClass  GtkTreeModelTypesClass;
  67
  68 struct _GtkTreeModelTypes
  69 {
  70   GObject parent;
  71
  72   gint stamp;
  73 };
  74
  75 struct _GtkTreeModelTypesClass
  76 {
  77   GObjectClass parent_class;
  78
  79   guint        (* get_flags)       (GtkTreeModel *tree_model);
  80   gint         (* get_n_columns)   (GtkTreeModel *tree_model);
  81   GType        (* get_column_type) (GtkTreeModel *tree_model,
  82                                     gint          index);
  83   gboolean     (* get_iter)        (GtkTreeModel *tree_model,
  84                                     GtkTreeIter  *iter,
  85                                     GtkTreePath  *path);
  86   GtkTreePath *(* get_path)        (GtkTreeModel *tree_model,
  87                                     GtkTreeIter  *iter);
  88   void         (* get_value)       (GtkTreeModel *tree_model,
  89                                     GtkTreeIter  *iter,
  90                                     gint          column,
  91                                     GValue       *value);
  92   gboolean     (* iter_next)       (GtkTreeModel *tree_model,
  93                                     GtkTreeIter  *iter);
  94   gboolean     (* iter_children)   (GtkTreeModel *tree_model,
  95                                     GtkTreeIter  *iter,
  96                                     GtkTreeIter  *parent);
  97   gboolean     (* iter_has_child)  (GtkTreeModel *tree_model,
  98                                     GtkTreeIter  *iter);
  99   gint         (* iter_n_children) (GtkTreeModel *tree_model,
 100                                     GtkTreeIter  *iter);
 101   gboolean     (* iter_nth_child)  (GtkTreeModel *tree_model,
 102                                     GtkTreeIter  *iter,
 103                                     GtkTreeIter  *parent,
 104                                     gint          n);
 105   gboolean     (* iter_parent)     (GtkTreeModel *tree_model,
 106                                     GtkTreeIter  *iter,
 107                                     GtkTreeIter  *child);
 108   void         (* ref_iter)        (GtkTreeModel *tree_model,
 109                                     GtkTreeIter  *iter);
 110   void         (* unref_iter)      (GtkTreeModel *tree_model,
 111                                     GtkTreeIter  *iter);
 112
 113   /* These will be moved into the GtkTreeModelIface eventually */
 114   void         (* changed)         (GtkTreeModel *tree_model,
 115                                     GtkTreePath  *path,
 116                                     GtkTreeIter  *iter);
 117   void         (* inserted)        (GtkTreeModel *tree_model,
 118                                     GtkTreePath  *path,
 119                                     GtkTreeIter  *iter);
 120   void         (* child_toggled)   (GtkTreeModel *tree_model,
 121                                     GtkTreePath  *path,
 122                                     GtkTreeIter  *iter);
 123   void         (* deleted)         (GtkTreeModel *tree_model,
 124                                     GtkTreePath  *path);
 125 };
 126
 127 GType              gtk_tree_model_types_get_type      (void) G_GNUC_CONST;
 128 GtkTreeModelTypes *gtk_tree_model_types_new           (void);
 129
 130 typedef enum
 131 {
 132   COLUMNS_NONE,
 133   COLUMNS_ONE,
 134   COLUMNS_LOTS,
 135   COLUMNS_LAST
 136 } ColumnsType;
 137
 138 static gchar *column_type_names[] = {
 139   "No columns",
 140   "One column",
 141   "Many columns"
 142 };
 143
 144 #define N_COLUMNS 9
 145
 146 static GType*
 147 get_model_types (void)
 148 {
 149   static GType column_types[N_COLUMNS] = { 0 };
 150
 151   if (column_types[0] == 0)
 152     {
 153       column_types[0] = G_TYPE_STRING;
 154       column_types[1] = G_TYPE_STRING;
 155       column_types[2] = GDK_TYPE_PIXBUF;
 156       column_types[3] = G_TYPE_FLOAT;
 157       column_types[4] = G_TYPE_UINT;
 158       column_types[5] = G_TYPE_UCHAR;
 159       column_types[6] = G_TYPE_CHAR;
 160 #define BOOL_COLUMN 7
 161       column_types[BOOL_COLUMN] = G_TYPE_BOOLEAN;
 162       column_types[8] = G_TYPE_INT;
 163     }
 164
 165   return column_types;
 166 }
 167
 168 static void
 169 col_clicked_cb (GtkTreeViewColumn *col, gpointer data)
 170 {
 171   GtkWindow *win;
 172
 173   win = GTK_WINDOW (create_prop_editor (G_OBJECT (col), GTK_TYPE_TREE_VIEW_COLUMN));
 174
 175   gtk_window_set_title (win, gtk_tree_view_column_get_title (col));
 176 }
 177
 178 static void
 179 setup_column (GtkTreeViewColumn *col)
 180 {
 181   gtk_tree_view_column_set_clickable (col, TRUE);
 182   g_signal_connect (col,
 183                     "clicked",
 184                     G_CALLBACK (col_clicked_cb),
 185                     NULL);
 186 }
 187
 188 static void
 189 toggled_callback (GtkCellRendererToggle *celltoggle,
 190                   gchar                 *path_string,
 191                   GtkTreeView           *tree_view)
 192 {
 193   GtkTreeModel *model = NULL;
 194   GtkTreeModelSort *sort_model = NULL;
 195   GtkTreePath *path;
 196   GtkTreeIter iter;
 197   gboolean active = FALSE;
 198
 199   g_return_if_fail (GTK_IS_TREE_VIEW (tree_view));
 200
 201   model = gtk_tree_view_get_model (tree_view);
 202
 203   if (GTK_IS_TREE_MODEL_SORT (model))
 204     {
 205       sort_model = GTK_TREE_MODEL_SORT (model);
 206       model = gtk_tree_model_sort_get_model (sort_model);
 207     }
 208
 209   if (model == NULL)
 210     return;
 211
 212   if (sort_model)
 213     {
 214       g_warning ("FIXME implement conversion from TreeModelSort iter to child model iter");
 215       return;
 216     }
 217
 218   path = gtk_tree_path_new_from_string (path_string);
 219   if (!gtk_tree_model_get_iter (model,
 220                                 &iter, path))
 221     {
 222       g_warning ("%s: bad path?", G_STRLOC);
 223       return;
 224     }
 225   gtk_tree_path_free (path);
 226
 227   if (GTK_IS_LIST_STORE (model))
 228     {
 229       gtk_tree_model_get (GTK_TREE_MODEL (model),
 230                           &iter,
 231                           BOOL_COLUMN,
 232                           &active,
 233                           -1);
 234
 235       gtk_list_store_set (GTK_LIST_STORE (model),
 236                           &iter,
 237                           BOOL_COLUMN,
 238                           !active,
 239                           -1);
 240     }
 241   else if (GTK_IS_TREE_STORE (model))
 242     {
 243       gtk_tree_model_get (GTK_TREE_MODEL (model),
 244                           &iter,
 245                           BOOL_COLUMN,
 246                           &active,
 247                           -1);
 248
 249       gtk_tree_store_set (GTK_TREE_STORE (model),
 250                           &iter,
 251                           BOOL_COLUMN,
 252                           !active,
 253                           -1);
 254     }
 255   else
 256     g_warning ("don't know how to actually toggle value for model type %s",
 257                g_type_name (G_TYPE_FROM_INSTANCE (model)));
 258 }
 259
 260 static void
 261 edited_callback (GtkCellRendererText *renderer,
 262                  const gchar   *path_string,
 263                  const gchar   *new_text,
 264                  GtkTreeView  *tree_view)
 265 {
 266   GtkTreeModel *model = NULL;
 267   GtkTreeModelSort *sort_model = NULL;
 268   GtkTreePath *path;
 269   GtkTreeIter iter;
 270   guint value = atoi (new_text);
 271
 272   g_return_if_fail (GTK_IS_TREE_VIEW (tree_view));
 273
 274   model = gtk_tree_view_get_model (tree_view);
 275
 276   if (GTK_IS_TREE_MODEL_SORT (model))
 277     {
 278       sort_model = GTK_TREE_MODEL_SORT (model);
 279       model = gtk_tree_model_sort_get_model (sort_model);
 280     }
 281
 282   if (model == NULL)
 283     return;
 284
 285   if (sort_model)
 286     {
 287       g_warning ("FIXME implement conversion from TreeModelSort iter to child model iter");
 288       return;
 289     }
 290
 291   path = gtk_tree_path_new_from_string (path_string);
 292   if (!gtk_tree_model_get_iter (model,
 293                                 &iter, path))
 294     {
 295       g_warning ("%s: bad path?", G_STRLOC);
 296       return;
 297     }
 298   gtk_tree_path_free (path);
 299
 300   if (GTK_IS_LIST_STORE (model))
 301     {
 302       gtk_list_store_set (GTK_LIST_STORE (model),
 303                           &iter,
 304                           4,
 305                           value,
 306                           -1);
 307     }
 308   else if (GTK_IS_TREE_STORE (model))
 309     {
 310       gtk_tree_store_set (GTK_TREE_STORE (model),
 311                           &iter,
 312                           4,
 313                           value,
 314                           -1);
 315     }
 316   else
 317     g_warning ("don't know how to actually toggle value for model type %s",
 318                g_type_name (G_TYPE_FROM_INSTANCE (model)));
 319 }
 320
 321 static ColumnsType current_column_type = COLUMNS_LOTS;
 322
 323 static void
 324 set_columns_type (GtkTreeView *tree_view, ColumnsType type)
 325 {
 326   GtkTreeViewColumn *col;
 327   GtkCellRenderer *rend;
 328   GdkPixbuf *pixbuf;
 329   GtkWidget *image;
 330   GtkAdjustment *adjustment;
 331
 332   current_column_type = type;
 333
 334   col = gtk_tree_view_get_column (tree_view, 0);
 335   while (col)
 336     {
 337       gtk_tree_view_remove_column (tree_view, col);
 338
 339       col = gtk_tree_view_get_column (tree_view, 0);
 340     }
 341
 342   gtk_tree_view_set_rules_hint (tree_view, FALSE);
 343
 344   switch (type)
 345     {
 346     case COLUMNS_NONE:
 347       break;
 348
 349     case COLUMNS_LOTS:
 350       /* with lots of columns we need to turn on rules */
 351       gtk_tree_view_set_rules_hint (tree_view, TRUE);
 352
 353       rend = gtk_cell_renderer_text_new ();
 354
 355       col = gtk_tree_view_column_new_with_attributes ("Column 1",
 356                                                       rend,
 357                                                       "text", 1,
 358                                                       NULL);
 359       setup_column (col);
 360
 361       gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), col);
 362
 363       col = gtk_tree_view_column_new();
 364       gtk_tree_view_column_set_title (col, "Column 2");
 365
 366       rend = gtk_cell_renderer_pixbuf_new ();
 367       gtk_tree_view_column_pack_start (col, rend, FALSE);
 368       gtk_tree_view_column_add_attribute (col, rend, "pixbuf", 2);
 369       rend = gtk_cell_renderer_text_new ();
 370       gtk_tree_view_column_pack_start (col, rend, TRUE);
 371       gtk_tree_view_column_add_attribute (col, rend, "text", 0);
 372
 373       setup_column (col);
 374
 375
 376       gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), col);
 377       gtk_tree_view_set_expander_column (tree_view, col);
 378
 379       rend = gtk_cell_renderer_toggle_new ();
 380
 381       g_signal_connect (rend, "toggled",
 382                         G_CALLBACK (toggled_callback), tree_view);
 383
 384       col = gtk_tree_view_column_new_with_attributes ("Column 3",
 385                                                       rend,
 386                                                       "active", BOOL_COLUMN,
 387                                                       NULL);
 388
 389       setup_column (col);
 390
 391       gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), col);
 392
 393       pixbuf = gdk_pixbuf_new_from_xpm_data ((const char **)book_closed_xpm);
 394
 395       image = gtk_image_new_from_pixbuf (pixbuf);
 396
 397       g_object_unref (pixbuf);
 398
 399       gtk_widget_show (image);
 400
 401       gtk_tree_view_column_set_widget (col, image);
 402
 403       rend = gtk_cell_renderer_toggle_new ();
 404
 405       /* you could also set this per-row by tying it to a column
 406        * in the model of course.
 407        */
 408       g_object_set (rend, "radio", TRUE, NULL);
 409
 410       g_signal_connect (rend, "toggled",
 411                         G_CALLBACK (toggled_callback), tree_view);
 412
 413       col = gtk_tree_view_column_new_with_attributes ("Column 4",
 414                                                       rend,
 415                                                       "active", BOOL_COLUMN,
 416                                                       NULL);
 417
 418       setup_column (col);
 419
 420       gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), col);
 421
 422       rend = gtk_cell_renderer_spin_new ();
 423
 424       adjustment = gtk_adjustment_new (0, 0, 10000, 100, 100, 100);
 425       g_object_set (rend, "editable", TRUE, NULL);
 426       g_object_set (rend, "adjustment", adjustment, NULL);
 427
 428       g_signal_connect (rend, "edited",
 429                         G_CALLBACK (edited_callback), tree_view);
 430
 431       col = gtk_tree_view_column_new_with_attributes ("Column 5",
 432                                                       rend,
 433                                                       "text", 4,
 434                                                       NULL);
 435
 436       setup_column (col);
 437
 438       gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), col);
 439 #if 0
 440
 441       rend = gtk_cell_renderer_text_new ();
 442
 443       col = gtk_tree_view_column_new_with_attributes ("Column 6",
 444                                                       rend,
 445                                                       "text", 4,
 446                                                       NULL);
 447
 448       setup_column (col);
 449
 450       gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), col);
 451
 452       rend = gtk_cell_renderer_text_new ();
 453
 454       col = gtk_tree_view_column_new_with_attributes ("Column 7",
 455                                                       rend,
 456                                                       "text", 5,
 457                                                       NULL);
 458
 459       setup_column (col);
 460
 461       gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), col);
 462
 463       rend = gtk_cell_renderer_text_new ();
 464
 465       col = gtk_tree_view_column_new_with_attributes ("Column 8",
 466                                                       rend,
 467                                                       "text", 6,
 468                                                       NULL);
 469
 470       setup_column (col);
 471
 472       gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), col);
 473
 474       rend = gtk_cell_renderer_text_new ();
 475
 476       col = gtk_tree_view_column_new_with_attributes ("Column 9",
 477                                                       rend,
 478                                                       "text", 7,
 479                                                       NULL);
 480
 481       setup_column (col);
 482
 483       gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), col);
 484
 485       rend = gtk_cell_renderer_text_new ();
 486
 487       col = gtk_tree_view_column_new_with_attributes ("Column 10",
 488                                                       rend,
 489                                                       "text", 8,
 490                                                       NULL);
 491
 492       setup_column (col);
 493
 494       gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), col);
 495
 496 #endif
 497
 498       /* FALL THRU */
 499
 500     case COLUMNS_ONE:
 501       rend = gtk_cell_renderer_text_new ();
 502
 503       col = gtk_tree_view_column_new_with_attributes ("Column 0",
 504                                                       rend,
 505                                                       "text", 0,
 506                                                       NULL);
 507
 508       setup_column (col);
 509
 510       gtk_tree_view_insert_column (GTK_TREE_VIEW (tree_view), col, 0);
 511     default:
 512       break;
 513     }
 514 }
 515
 516 static ColumnsType
 517 get_columns_type (void)
 518 {
 519   return current_column_type;
 520 }
 521
 522 static GdkPixbuf *our_pixbuf;
 523
 524 typedef enum
 525 {
 526   /*   MODEL_TYPES, */
 527   MODEL_TREE,
 528   MODEL_LIST,
 529   MODEL_SORTED_TREE,
 530   MODEL_SORTED_LIST,
 531   MODEL_EMPTY_LIST,
 532   MODEL_EMPTY_TREE,
 533   MODEL_NULL,
 534   MODEL_LAST
 535 } ModelType;
 536
 537 /* FIXME add a custom model to test */
 538 static GtkTreeModel *models[MODEL_LAST];
 539 static const char *model_names[MODEL_LAST] = {
 540   "GtkTreeStore",
 541   "GtkListStore",
 542   "GtkTreeModelSort wrapping GtkTreeStore",
 543   "GtkTreeModelSort wrapping GtkListStore",
 544   "Empty GtkListStore",
 545   "Empty GtkTreeStore",
 546   "NULL (no model)"
 547 };
 548
 549 static GtkTreeModel*
 550 create_list_model (void)
 551 {
 552   GtkListStore *store;
 553   GtkTreeIter iter;
 554   gint i;
 555   GType *t;
 556
 557   t = get_model_types ();
 558
 559   store = gtk_list_store_new (N_COLUMNS,
 560                               t[0], t[1], t[2],
 561                               t[3], t[4], t[5],
 562                               t[6], t[7], t[8]);
 563
 564   i = 0;
 565   while (i < 200)
 566     {
 567       char *msg;
 568
 569       gtk_list_store_append (store, &iter);
 570
 571       msg = g_strdup_printf ("%d", i);
 572
 573       gtk_list_store_set (store, &iter, 0, msg, 1, "Foo! Foo! Foo!",
 574                           2, our_pixbuf,
 575                           3, 7.0, 4, (guint) 9000,
 576                           5, 'f', 6, 'g',
 577                           7, TRUE, 8, 23245454,
 578                           -1);
 579
 580       g_free (msg);
 581
 582       ++i;
 583     }
 584
 585   return GTK_TREE_MODEL (store);
 586 }
 587
 588 static void
 589 typesystem_recurse (GType        type,
 590                     GtkTreeIter *parent_iter,
 591                     GtkTreeStore *store)
 592 {
 593   GType* children;
 594   guint n_children = 0;
 595   gint i;
 596   GtkTreeIter iter;
 597   gchar *str;
 598
 599   gtk_tree_store_append (store, &iter, parent_iter);
 600
 601   str = g_strdup_printf ("%ld", (glong)type);
 602   gtk_tree_store_set (store, &iter, 0, str, 1, g_type_name (type),
 603                       2, our_pixbuf,
 604                       3, 7.0, 4, (guint) 9000,
 605                       5, 'f', 6, 'g',
 606                       7, TRUE, 8, 23245454,
 607                       -1);
 608   g_free (str);
 609
 610   children = g_type_children (type, &n_children);
 611
 612   i = 0;
 613   while (i < n_children)
 614     {
 615       typesystem_recurse (children[i], &iter, store);
 616
 617       ++i;
 618     }
 619
 620   g_free (children);
 621 }
 622
 623 static GtkTreeModel*
 624 create_tree_model (void)
 625 {
 626   GtkTreeStore *store;
 627   gint i;
 628   GType *t;
 629
 630   /* Make the tree more interesting */
 631   /* - we need this magic here so we are sure the type ends up being
 632    * registered and gcc doesn't optimize away the code */
 633   g_type_class_unref (g_type_class_ref (gtk_scrolled_window_get_type ()));
 634   g_type_class_unref (g_type_class_ref (gtk_label_get_type ()));
 635   g_type_class_unref (g_type_class_ref (gtk_hscrollbar_get_type ()));
 636   g_type_class_unref (g_type_class_ref (gtk_vscrollbar_get_type ()));
 637   g_type_class_unref (g_type_class_ref (pango_layout_get_type ()));
 638
 639   t = get_model_types ();
 640
 641   store = gtk_tree_store_new (N_COLUMNS,
 642                               t[0], t[1], t[2],
 643                               t[3], t[4], t[5],
 644                               t[6], t[7], t[8]);
 645
 646   i = 0;
 647   while (i < G_TYPE_FUNDAMENTAL_MAX)
 648     {
 649       typesystem_recurse (i, NULL, store);
 650
 651       ++i;
 652     }
 653
 654   return GTK_TREE_MODEL (store);
 655 }
 656
 657 static void
 658 model_selected (GtkComboBox *combo_box, gpointer data)
 659 {
 660   GtkTreeView *tree_view = GTK_TREE_VIEW (data);
 661   gint hist;
 662
 663   hist = gtk_combo_box_get_active (combo_box);
 664
 665   if (models[hist] != gtk_tree_view_get_model (tree_view))
 666     {
 667       gtk_tree_view_set_model (tree_view, models[hist]);
 668     }
 669 }
 670
 671 static void
 672 columns_selected (GtkComboBox *combo_box, gpointer data)
 673 {
 674   GtkTreeView *tree_view = GTK_TREE_VIEW (data);
 675   gint hist;
 676
 677   hist = gtk_combo_box_get_active (combo_box);
 678
 679   if (hist != get_columns_type ())
 680     {
 681       set_columns_type (tree_view, hist);
 682     }
 683 }
 684
 685
 686 enum
 687 {
 688   TARGET_GTK_TREE_MODEL_ROW
 689 };
 690
 691 static GtkTargetEntry row_targets[] = {
 692   { "GTK_TREE_MODEL_ROW", GTK_TARGET_SAME_APP,
 693     TARGET_GTK_TREE_MODEL_ROW }
 694 };
 695
 696 int
 697 main (int    argc,
 698       char **argv)
 699 {
 700   GtkWidget *window;
 701   GtkWidget *sw;
 702   GtkWidget *tv;
 703   GtkWidget *table;
 704   GtkWidget *combo_box;
 705   GtkTreeModel *model;
 706   gint i;
 707
 708   gtk_init (&argc, &argv);
 709
 710   if (g_getenv ("RTL"))
 711     gtk_widget_set_default_direction (GTK_TEXT_DIR_RTL);
 712
 713   our_pixbuf = gdk_pixbuf_new_from_xpm_data ((const char **) book_closed_xpm);
 714
 715 #if 0
 716   models[MODEL_TYPES] = GTK_TREE_MODEL (gtk_tree_model_types_new ());
 717 #endif
 718   models[MODEL_LIST] = create_list_model ();
 719   models[MODEL_TREE] = create_tree_model ();
 720
 721   model = create_list_model ();
 722   models[MODEL_SORTED_LIST] = gtk_tree_model_sort_new_with_model (model);
 723   g_object_unref (model);
 724
 725   model = create_tree_model ();
 726   models[MODEL_SORTED_TREE] = gtk_tree_model_sort_new_with_model (model);
 727   g_object_unref (model);
 728
 729   models[MODEL_EMPTY_LIST] = GTK_TREE_MODEL (gtk_list_store_new (1, G_TYPE_INT));
 730   models[MODEL_EMPTY_TREE] = GTK_TREE_MODEL (gtk_tree_store_new (1, G_TYPE_INT));
 731
 732   models[MODEL_NULL] = NULL;
 733
 734   run_automated_tests ();
 735
 736   window = gtk_window_new (GTK_WINDOW_TOPLEVEL);
 737   g_signal_connect (window, "destroy", G_CALLBACK (gtk_main_quit), NULL);
 738   gtk_window_set_default_size (GTK_WINDOW (window), 430, 400);
 739
 740   table = gtk_table_new (3, 1, FALSE);
 741
 742   gtk_container_add (GTK_CONTAINER (window), table);
 743
 744   tv = gtk_tree_view_new_with_model (models[0]);
 745
 746   gtk_tree_view_enable_model_drag_source (GTK_TREE_VIEW (tv),
 747                                           GDK_BUTTON1_MASK,
 748                                           row_targets,
 749                                           G_N_ELEMENTS (row_targets),
 750                                           GDK_ACTION_MOVE | GDK_ACTION_COPY);
 751
 752   gtk_tree_view_enable_model_drag_dest (GTK_TREE_VIEW (tv),
 753                                         row_targets,
 754                                         G_N_ELEMENTS (row_targets),
 755                                         GDK_ACTION_MOVE | GDK_ACTION_COPY);
 756
 757   /* Model menu */
 758   combo_box = gtk_combo_box_text_new ();
 759   for (i = 0; i < MODEL_LAST; i++)
 760       gtk_combo_box_text_append_text (GTK_COMBO_BOX_TEXT (combo_box), model_names[i]);
 761
 762   gtk_table_attach (GTK_TABLE (table), combo_box,
 763                     0, 1, 0, 1,
 764                     0, 0,
 765                     0, 0);
 766
 767   g_signal_connect (combo_box,
 768                     "changed",
 769                     G_CALLBACK (model_selected),
 770                     tv);
 771
 772   /* Columns menu */
 773   combo_box = gtk_combo_box_text_new ();
 774   for (i = 0; i < COLUMNS_LAST; i++)
 775       gtk_combo_box_text_append_text (GTK_COMBO_BOX_TEXT (combo_box), column_type_names[i]);
 776
 777   gtk_table_attach (GTK_TABLE (table), combo_box,
 778                     0, 1, 1, 2,
 779                     0, 0,
 780                     0, 0);
 781
 782   set_columns_type (GTK_TREE_VIEW (tv), COLUMNS_LOTS);
 783   gtk_combo_box_set_active (GTK_COMBO_BOX (combo_box), COLUMNS_LOTS);
 784
 785   g_signal_connect (combo_box,
 786                     "changed",
 787                     G_CALLBACK (columns_selected),
 788                     tv);
 789
 790   sw = gtk_scrolled_window_new (NULL, NULL);
 791   gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (sw),
 792                                   GTK_POLICY_AUTOMATIC,
 793                                   GTK_POLICY_AUTOMATIC);
 794
 795   gtk_table_attach (GTK_TABLE (table), sw,
 796                     0, 1, 2, 3,
 797                     GTK_EXPAND | GTK_FILL,
 798                     GTK_EXPAND | GTK_FILL,
 799                     0, 0);
 800
 801   gtk_container_add (GTK_CONTAINER (sw), tv);
 802
 803   gtk_widget_show_all (window);
 804
 805   gtk_main ();
 806
 807   return 0;
 808 }
 809
 810 /*
 811  * GtkTreeModelTypes
 812  */
 813
 814 static void         gtk_tree_model_types_init                 (GtkTreeModelTypes      *model_types);
 815 static void         gtk_tree_model_types_tree_model_init      (GtkTreeModelIface   *iface);
 816 static gint         gtk_real_model_types_get_n_columns   (GtkTreeModel        *tree_model);
 817 static GType        gtk_real_model_types_get_column_type (GtkTreeModel        *tree_model,
 818                                                            gint                 index);
 819 static GtkTreePath *gtk_real_model_types_get_path        (GtkTreeModel        *tree_model,
 820                                                            GtkTreeIter         *iter);
 821 static void         gtk_real_model_types_get_value       (GtkTreeModel        *tree_model,
 822                                                            GtkTreeIter         *iter,
 823                                                            gint                 column,
 824                                                            GValue              *value);
 825 static gboolean     gtk_real_model_types_iter_next       (GtkTreeModel        *tree_model,
 826                                                            GtkTreeIter         *iter);
 827 static gboolean     gtk_real_model_types_iter_children   (GtkTreeModel        *tree_model,
 828                                                            GtkTreeIter         *iter,
 829                                                            GtkTreeIter         *parent);
 830 static gboolean     gtk_real_model_types_iter_has_child  (GtkTreeModel        *tree_model,
 831                                                            GtkTreeIter         *iter);
 832 static gint         gtk_real_model_types_iter_n_children (GtkTreeModel        *tree_model,
 833                                                            GtkTreeIter         *iter);
 834 static gboolean     gtk_real_model_types_iter_nth_child  (GtkTreeModel        *tree_model,
 835                                                            GtkTreeIter         *iter,
 836                                                            GtkTreeIter         *parent,
 837                                                            gint                 n);
 838 static gboolean     gtk_real_model_types_iter_parent     (GtkTreeModel        *tree_model,
 839                                                            GtkTreeIter         *iter,
 840                                                            GtkTreeIter         *child);
 841
 842
 843 GType
 844 gtk_tree_model_types_get_type (void)
 845 {
 846   static GType model_types_type = 0;
 847
 848   if (!model_types_type)
 849     {
 850       const GTypeInfo model_types_info =
 851       {
 852         sizeof (GtkTreeModelTypesClass),
 853         NULL,           /* base_init */
 854         NULL,           /* base_finalize */
 855         NULL,           /* class_init */
 856         NULL,           /* class_finalize */
 857         NULL,           /* class_data */
 858         sizeof (GtkTreeModelTypes),
 859         0,
 860         (GInstanceInitFunc) gtk_tree_model_types_init
 861       };
 862
 863       const GInterfaceInfo tree_model_info =
 864       {
 865         (GInterfaceInitFunc) gtk_tree_model_types_tree_model_init,
 866         NULL,
 867         NULL
 868       };
 869
 870       model_types_type = g_type_register_static (G_TYPE_OBJECT,
 871                                                  "GtkTreeModelTypes",
 872                                                  &model_types_info, 0);
 873       g_type_add_interface_static (model_types_type,
 874                                    GTK_TYPE_TREE_MODEL,
 875                                    &tree_model_info);
 876     }
 877
 878   return model_types_type;
 879 }
 880
 881 GtkTreeModelTypes *
 882 gtk_tree_model_types_new (void)
 883 {
 884   GtkTreeModelTypes *retval;
 885
 886   retval = g_object_new (GTK_TYPE_MODEL_TYPES, NULL);
 887
 888   return retval;
 889 }
 890
 891 static void
 892 gtk_tree_model_types_tree_model_init (GtkTreeModelIface *iface)
 893 {
 894   iface->get_n_columns = gtk_real_model_types_get_n_columns;
 895   iface->get_column_type = gtk_real_model_types_get_column_type;
 896   iface->get_path = gtk_real_model_types_get_path;
 897   iface->get_value = gtk_real_model_types_get_value;
 898   iface->iter_next = gtk_real_model_types_iter_next;
 899   iface->iter_children = gtk_real_model_types_iter_children;
 900   iface->iter_has_child = gtk_real_model_types_iter_has_child;
 901   iface->iter_n_children = gtk_real_model_types_iter_n_children;
 902   iface->iter_nth_child = gtk_real_model_types_iter_nth_child;
 903   iface->iter_parent = gtk_real_model_types_iter_parent;
 904 }
 905
 906 static void
 907 gtk_tree_model_types_init (GtkTreeModelTypes *model_types)
 908 {
 909   model_types->stamp = g_random_int ();
 910 }
 911
 912 static GType column_types[] = {
 913   G_TYPE_STRING, /* GType */
 914   G_TYPE_STRING  /* type name */
 915 };
 916
 917 static gint
 918 gtk_real_model_types_get_n_columns (GtkTreeModel *tree_model)
 919 {
 920   return G_N_ELEMENTS (column_types);
 921 }
 922
 923 static GType
 924 gtk_real_model_types_get_column_type (GtkTreeModel *tree_model,
 925                                       gint          index)
 926 {
 927   g_return_val_if_fail (index < G_N_ELEMENTS (column_types), G_TYPE_INVALID);
 928
 929   return column_types[index];
 930 }
 931
 932 #if 0
 933 /* Use default implementation of this */
 934 static gboolean
 935 gtk_real_model_types_get_iter (GtkTreeModel *tree_model,
 936                                GtkTreeIter  *iter,
 937                                GtkTreePath  *path)
 938 {
 939
 940 }
 941 #endif
 942
 943 /* The toplevel nodes of the tree are the reserved types, G_TYPE_NONE through
 944  * G_TYPE_RESERVED_FUNDAMENTAL.
 945  */
 946
 947 static GtkTreePath *
 948 gtk_real_model_types_get_path (GtkTreeModel *tree_model,
 949                                GtkTreeIter  *iter)
 950 {
 951   GtkTreePath *retval;
 952   GType type;
 953   GType parent;
 954
 955   g_return_val_if_fail (GTK_IS_TREE_MODEL_TYPES (tree_model), NULL);
 956   g_return_val_if_fail (iter != NULL, NULL);
 957
 958   type = GPOINTER_TO_INT (iter->user_data);
 959
 960   retval = gtk_tree_path_new ();
 961
 962   parent = g_type_parent (type);
 963   while (parent != G_TYPE_INVALID)
 964     {
 965       GType* children = g_type_children (parent, NULL);
 966       gint i = 0;
 967
 968       if (!children || children[0] == G_TYPE_INVALID)
 969         {
 970           g_warning ("bad iterator?");
 971           return NULL;
 972         }
 973
 974       while (children[i] != type)
 975         ++i;
 976
 977       gtk_tree_path_prepend_index (retval, i);
 978
 979       g_free (children);
 980
 981       type = parent;
 982       parent = g_type_parent (parent);
 983     }
 984
 985   /* The fundamental type itself is the index on the toplevel */
 986   gtk_tree_path_prepend_index (retval, type);
 987
 988   return retval;
 989 }
 990
 991 static void
 992 gtk_real_model_types_get_value (GtkTreeModel *tree_model,
 993                                 GtkTreeIter  *iter,
 994                                 gint          column,
 995                                 GValue       *value)
 996 {
 997   GType type;
 998
 999   type = GPOINTER_TO_INT (iter->user_data);
1000
1001   switch (column)
1002     {
1003     case 0:
1004       {
1005         gchar *str;
1006
1007         g_value_init (value, G_TYPE_STRING);
1008
1009         str = g_strdup_printf ("%ld", (long int) type);
1010         g_value_set_string (value, str);
1011         g_free (str);
1012       }
1013       break;
1014
1015     case 1:
1016       g_value_init (value, G_TYPE_STRING);
1017       g_value_set_string (value, g_type_name (type));
1018       break;
1019
1020     default:
1021       g_warning ("Bad column %d requested", column);
1022     }
1023 }
1024
1025 static gboolean
1026 gtk_real_model_types_iter_next (GtkTreeModel  *tree_model,
1027                                 GtkTreeIter   *iter)
1028 {
1029
1030   GType parent;
1031   GType type;
1032
1033   type = GPOINTER_TO_INT (iter->user_data);
1034
1035   parent = g_type_parent (type);
1036
1037   if (parent == G_TYPE_INVALID)
1038     {
1039       /* find next _valid_ fundamental type */
1040       do
1041         type++;
1042       while (!g_type_name (type) && type <= G_TYPE_FUNDAMENTAL_MAX);
1043       if (type <= G_TYPE_FUNDAMENTAL_MAX)
1044         {
1045           /* found one */
1046           iter->user_data = GINT_TO_POINTER (type);
1047           return TRUE;
1048         }
1049       else
1050         return FALSE;
1051     }
1052   else
1053     {
1054       GType* children = g_type_children (parent, NULL);
1055       gint i = 0;
1056
1057       g_assert (children != NULL);
1058
1059       while (children[i] != type)
1060         ++i;
1061
1062       ++i;
1063
1064       if (children[i] != G_TYPE_INVALID)
1065         {
1066           g_free (children);
1067           iter->user_data = GINT_TO_POINTER (children[i]);
1068           return TRUE;
1069         }
1070       else
1071         {
1072           g_free (children);
1073           return FALSE;
1074         }
1075     }
1076 }
1077
1078 static gboolean
1079 gtk_real_model_types_iter_children (GtkTreeModel *tree_model,
1080                                     GtkTreeIter  *iter,
1081                                     GtkTreeIter  *parent)
1082 {
1083   GType type;
1084   GType* children;
1085
1086   type = GPOINTER_TO_INT (parent->user_data);
1087
1088   children = g_type_children (type, NULL);
1089
1090   if (!children || children[0] == G_TYPE_INVALID)
1091     {
1092       g_free (children);
1093       return FALSE;
1094     }
1095   else
1096     {
1097       iter->user_data = GINT_TO_POINTER (children[0]);
1098       g_free (children);
1099       return TRUE;
1100     }
1101 }
1102
1103 static gboolean
1104 gtk_real_model_types_iter_has_child (GtkTreeModel *tree_model,
1105                                      GtkTreeIter  *iter)
1106 {
1107   GType type;
1108   GType* children;
1109
1110   type = GPOINTER_TO_INT (iter->user_data);
1111
1112   children = g_type_children (type, NULL);
1113
1114   if (!children || children[0] == G_TYPE_INVALID)
1115     {
1116       g_free (children);
1117       return FALSE;
1118     }
1119   else
1120     {
1121       g_free (children);
1122       return TRUE;
1123     }
1124 }
1125
1126 static gint
1127 gtk_real_model_types_iter_n_children (GtkTreeModel *tree_model,
1128                                       GtkTreeIter  *iter)
1129 {
1130   if (iter == NULL)
1131     {
1132       return G_TYPE_FUNDAMENTAL_MAX;
1133     }
1134   else
1135     {
1136       GType type;
1137       GType* children;
1138       guint n_children = 0;
1139
1140       type = GPOINTER_TO_INT (iter->user_data);
1141
1142       children = g_type_children (type, &n_children);
1143
1144       g_free (children);
1145
1146       return n_children;
1147     }
1148 }
1149
1150 static gboolean
1151 gtk_real_model_types_iter_nth_child (GtkTreeModel *tree_model,
1152                                      GtkTreeIter  *iter,
1153                                      GtkTreeIter  *parent,
1154                                      gint          n)
1155 {
1156   if (parent == NULL)
1157     {
1158       /* fundamental type */
1159       if (n < G_TYPE_FUNDAMENTAL_MAX)
1160         {
1161           iter->user_data = GINT_TO_POINTER (n);
1162           return TRUE;
1163         }
1164       else
1165         return FALSE;
1166     }
1167   else
1168     {
1169       GType type = GPOINTER_TO_INT (parent->user_data);
1170       guint n_children = 0;
1171       GType* children = g_type_children (type, &n_children);
1172
1173       if (n_children == 0)
1174         {
1175           g_free (children);
1176           return FALSE;
1177         }
1178       else if (n >= n_children)
1179         {
1180           g_free (children);
1181           return FALSE;
1182         }
1183       else
1184         {
1185           iter->user_data = GINT_TO_POINTER (children[n]);
1186           g_free (children);
1187
1188           return TRUE;
1189         }
1190     }
1191 }
1192
1193 static gboolean
1194 gtk_real_model_types_iter_parent (GtkTreeModel *tree_model,
1195                                   GtkTreeIter  *iter,
1196                                   GtkTreeIter  *child)
1197 {
1198   GType type;
1199   GType parent;
1200
1201   type = GPOINTER_TO_INT (child->user_data);
1202
1203   parent = g_type_parent (type);
1204
1205   if (parent == G_TYPE_INVALID)
1206     {
1207       if (type > G_TYPE_FUNDAMENTAL_MAX)
1208         g_warning ("no parent for %ld %s\n",
1209                    (long int) type,
1210                    g_type_name (type));
1211       return FALSE;
1212     }
1213   else
1214     {
1215       iter->user_data = GINT_TO_POINTER (parent);
1216
1217       return TRUE;
1218     }
1219 }
1220
1221 /*
1222  * Automated testing
1223  */
1224
1225 #if 0
1226
1227 static void
1228 treestore_torture_recurse (GtkTreeStore *store,
1229                            GtkTreeIter  *root,
1230                            gint          depth)
1231 {
1232   GtkTreeModel *model;
1233   gint i;
1234   GtkTreeIter iter;
1235
1236   model = GTK_TREE_MODEL (store);
1237
1238   if (depth > 2)
1239     return;
1240
1241   ++depth;
1242
1243   gtk_tree_store_append (store, &iter, root);
1244
1245   gtk_tree_model_iter_children (model, &iter, root);
1246
1247   i = 0;
1248   while (i < 100)
1249     {
1250       gtk_tree_store_append (store, &iter, root);
1251       ++i;
1252     }
1253
1254   while (gtk_tree_model_iter_children (model, &iter, root))
1255     gtk_tree_store_remove (store, &iter);
1256
1257   gtk_tree_store_append (store, &iter, root);
1258
1259   /* inserts before last node in tree */
1260   i = 0;
1261   while (i < 100)
1262     {
1263       gtk_tree_store_insert_before (store, &iter, root, &iter);
1264       ++i;
1265     }
1266
1267   /* inserts after the node before the last node */
1268   i = 0;
1269   while (i < 100)
1270     {
1271       gtk_tree_store_insert_after (store, &iter, root, &iter);
1272       ++i;
1273     }
1274
1275   /* inserts after the last node */
1276   gtk_tree_store_append (store, &iter, root);
1277
1278   i = 0;
1279   while (i < 100)
1280     {
1281       gtk_tree_store_insert_after (store, &iter, root, &iter);
1282       ++i;
1283     }
1284
1285   /* remove everything again */
1286   while (gtk_tree_model_iter_children (model, &iter, root))
1287     gtk_tree_store_remove (store, &iter);
1288
1289
1290     /* Prepends */
1291   gtk_tree_store_prepend (store, &iter, root);
1292
1293   i = 0;
1294   while (i < 100)
1295     {
1296       gtk_tree_store_prepend (store, &iter, root);
1297       ++i;
1298     }
1299
1300   /* remove everything again */
1301   while (gtk_tree_model_iter_children (model, &iter, root))
1302     gtk_tree_store_remove (store, &iter);
1303
1304   gtk_tree_store_append (store, &iter, root);
1305   gtk_tree_store_append (store, &iter, root);
1306   gtk_tree_store_append (store, &iter, root);
1307   gtk_tree_store_append (store, &iter, root);
1308
1309   while (gtk_tree_model_iter_children (model, &iter, root))
1310     {
1311       treestore_torture_recurse (store, &iter, depth);
1312       gtk_tree_store_remove (store, &iter);
1313     }
1314 }
1315
1316 #endif
1317
1318 static void
1319 run_automated_tests (void)
1320 {
1321   g_print ("Running automated tests...\n");
1322
1323   /* FIXME TreePath basic verification */
1324
1325   /* FIXME generic consistency checks on the models */
1326
1327   {
1328     /* Make sure list store mutations don't crash anything */
1329     GtkListStore *store;
1330     GtkTreeModel *model;
1331     gint i;
1332     GtkTreeIter iter;
1333
1334     store = gtk_list_store_new (1, G_TYPE_INT);
1335
1336     model = GTK_TREE_MODEL (store);
1337
1338     i = 0;
1339     while (i < 100)
1340       {
1341         gtk_list_store_append (store, &iter);
1342         ++i;
1343       }
1344
1345     while (gtk_tree_model_get_iter_first (model, &iter))
1346       gtk_list_store_remove (store, &iter);
1347
1348     gtk_list_store_append (store, &iter);
1349
1350     /* inserts before last node in list */
1351     i = 0;
1352     while (i < 100)
1353       {
1354         gtk_list_store_insert_before (store, &iter, &iter);
1355         ++i;
1356       }
1357
1358     /* inserts after the node before the last node */
1359     i = 0;
1360     while (i < 100)
1361       {
1362         gtk_list_store_insert_after (store, &iter, &iter);
1363         ++i;
1364       }
1365
1366     /* inserts after the last node */
1367     gtk_list_store_append (store, &iter);
1368
1369     i = 0;
1370     while (i < 100)
1371       {
1372         gtk_list_store_insert_after (store, &iter, &iter);
1373         ++i;
1374       }
1375
1376     /* remove everything again */
1377     while (gtk_tree_model_get_iter_first (model, &iter))
1378       gtk_list_store_remove (store, &iter);
1379
1380
1381     /* Prepends */
1382     gtk_list_store_prepend (store, &iter);
1383
1384     i = 0;
1385     while (i < 100)
1386       {
1387         gtk_list_store_prepend (store, &iter);
1388         ++i;
1389       }
1390
1391     /* remove everything again */
1392     while (gtk_tree_model_get_iter_first (model, &iter))
1393       gtk_list_store_remove (store, &iter);
1394
1395     g_object_unref (store);
1396   }
1397
1398   {
1399     /* Make sure tree store mutations don't crash anything */
1400     GtkTreeStore *store;
1401     GtkTreeIter root;
1402
1403     store = gtk_tree_store_new (1, G_TYPE_INT);
1404     gtk_tree_store_append (GTK_TREE_STORE (store), &root, NULL);
1405     /* Remove test until it is rewritten to work */
1406     /*    treestore_torture_recurse (store, &root, 0);*/
1407
1408     g_object_unref (store);
1409   }
1410
1411   g_print ("Passed.\n");
1412 }