Pileus Git - ~andy/gtk/blob - gtk/gtktreemodelsort.c

   1 /* gtktreemodelsort.c
   2  * Copyright (C) 2000,2001  Red Hat, Inc.,  Jonathan Blandford <jrb@redhat.com>
   3  * Copyright (C) 2001,2002  Kristian Rietveld <kris@gtk.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 /* NOTE: There is a potential for confusion in this code as to whether an iter,
  22  * path or value refers to the GtkTreeModelSort model, or the child model being
  23  * sorted.  As a convention, variables referencing the child model will have an
  24  * s_ prefix before them (ie. s_iter, s_value, s_path);
  25  */
  26
  27 /* ITER FORMAT:
  28  *
  29  * iter->stamp = tree_model_sort->stamp
  30  * iter->user_data = SortLevel
  31  * iter->user_data2 = SortElt
  32  */
  33
  34 /* WARNING: this code is dangerous, can cause sleepless nights,
  35  * can cause your dog to die among other bad things
  36  *
  37  * we warned you and we're not liable for any head injuries.
  38  */
  39
  40 #include <string.h>
  41
  42 #include "gtktreemodelsort.h"
  43 #include "gtktreesortable.h"
  44 #include "gtktreestore.h"
  45 #include "gtktreedatalist.h"
  46 #include "gtkintl.h"
  47
  48 typedef struct _SortElt SortElt;
  49 typedef struct _SortLevel SortLevel;
  50 typedef struct _SortData SortData;
  51 typedef struct _SortTuple SortTuple;
  52
  53 struct _SortElt
  54 {
  55   GtkTreeIter  iter;
  56   SortLevel   *children;
  57   gint         offset;
  58   gint         ref_count;
  59   gint         zero_ref_count;
  60 };
  61
  62 struct _SortLevel
  63 {
  64   GArray    *array;
  65   gint       ref_count;
  66   SortElt   *parent_elt;
  67   SortLevel *parent_level;
  68 };
  69
  70 struct _SortData
  71 {
  72   GtkTreeModelSort *tree_model_sort;
  73   GtkTreePath *parent_path;
  74   gint parent_path_depth;
  75   gint *parent_path_indices;
  76   GtkTreeIterCompareFunc sort_func;
  77   gpointer sort_data;
  78 };
  79
  80 struct _SortTuple
  81 {
  82   SortElt   *elt;
  83   gint       offset;
  84 };
  85
  86 /* Properties */
  87 enum {
  88   PROP_0,
  89   /* Construct args */
  90   PROP_MODEL
  91 };
  92
  93
  94
  95 #define GTK_TREE_MODEL_SORT_CACHE_CHILD_ITERS(tree_model_sort) \
  96         (((GtkTreeModelSort *)tree_model_sort)->child_flags&GTK_TREE_MODEL_ITERS_PERSIST)
  97 #define SORT_ELT(sort_elt) ((SortElt *)sort_elt)
  98 #define SORT_LEVEL(sort_level) ((SortLevel *)sort_level)
  99
 100 #define GET_CHILD_ITER(tree_model_sort,child_iter,sort_iter) gtk_tree_model_sort_convert_iter_to_child_iter(GTK_TREE_MODEL_SORT (tree_model_sort), child_iter, sort_iter);
 101
 102 #define NO_SORT_FUNC ((GtkTreeIterCompareFunc) 0x1)
 103
 104 #define VALID_ITER(iter, tree_model_sort) (iter != NULL && iter->user_data != NULL && iter->user_data2 != NULL && tree_model_sort->stamp == iter->stamp)
 105
 106 /* general (object/interface init, etc) */
 107 static void gtk_tree_model_sort_init                  (GtkTreeModelSort      *tree_model_sort);
 108 static void gtk_tree_model_sort_class_init            (GtkTreeModelSortClass *tree_model_sort_class);
 109 static void gtk_tree_model_sort_tree_model_init       (GtkTreeModelIface     *iface);
 110 static void gtk_tree_model_sort_tree_sortable_init    (GtkTreeSortableIface  *iface);
 111 static void gtk_tree_model_sort_finalize              (GObject               *object);
 112 static void gtk_tree_model_sort_set_property          (GObject               *object,
 113                                                        guint                  prop_id,
 114                                                        const GValue          *value,
 115                                                        GParamSpec            *pspec);
 116 static void gtk_tree_model_sort_get_property          (GObject               *object,
 117                                                        guint                  prop_id,
 118                                                        GValue                *value,
 119                                                        GParamSpec            *pspec);
 120
 121 /* our signal handlers */
 122 static void gtk_tree_model_sort_row_changed           (GtkTreeModel          *model,
 123                                                        GtkTreePath           *start_path,
 124                                                        GtkTreeIter           *start_iter,
 125                                                        gpointer               data);
 126 static void gtk_tree_model_sort_row_inserted          (GtkTreeModel          *model,
 127                                                        GtkTreePath           *path,
 128                                                        GtkTreeIter           *iter,
 129                                                        gpointer               data);
 130 static void gtk_tree_model_sort_row_has_child_toggled (GtkTreeModel          *model,
 131                                                        GtkTreePath           *path,
 132                                                        GtkTreeIter           *iter,
 133                                                        gpointer               data);
 134 static void gtk_tree_model_sort_row_deleted           (GtkTreeModel          *model,
 135                                                        GtkTreePath           *path,
 136                                                        gpointer               data);
 137 static void gtk_tree_model_sort_rows_reordered        (GtkTreeModel          *s_model,
 138                                                        GtkTreePath           *s_path,
 139                                                        GtkTreeIter           *s_iter,
 140                                                        gint                  *new_order,
 141                                                        gpointer               data);
 142
 143 /* TreeModel interface */
 144 static GtkTreeModelFlags gtk_tree_model_sort_get_flags     (GtkTreeModel          *tree_model);
 145 static gint         gtk_tree_model_sort_get_n_columns      (GtkTreeModel          *tree_model);
 146 static GType        gtk_tree_model_sort_get_column_type    (GtkTreeModel          *tree_model,
 147                                                             gint                   index);
 148 static gboolean     gtk_tree_model_sort_get_iter           (GtkTreeModel          *tree_model,
 149                                                             GtkTreeIter           *iter,
 150                                                             GtkTreePath           *path);
 151 static GtkTreePath *gtk_tree_model_sort_get_path           (GtkTreeModel          *tree_model,
 152                                                             GtkTreeIter           *iter);
 153 static void         gtk_tree_model_sort_get_value          (GtkTreeModel          *tree_model,
 154                                                             GtkTreeIter           *iter,
 155                                                             gint                   column,
 156                                                             GValue                *value);
 157 static gboolean     gtk_tree_model_sort_iter_next          (GtkTreeModel          *tree_model,
 158                                                             GtkTreeIter           *iter);
 159 static gboolean     gtk_tree_model_sort_iter_children      (GtkTreeModel          *tree_model,
 160                                                             GtkTreeIter           *iter,
 161                                                             GtkTreeIter           *parent);
 162 static gboolean     gtk_tree_model_sort_iter_has_child     (GtkTreeModel          *tree_model,
 163                                                             GtkTreeIter           *iter);
 164 static gint         gtk_tree_model_sort_iter_n_children    (GtkTreeModel          *tree_model,
 165                                                             GtkTreeIter           *iter);
 166 static gboolean     gtk_tree_model_sort_iter_nth_child     (GtkTreeModel          *tree_model,
 167                                                             GtkTreeIter           *iter,
 168                                                             GtkTreeIter           *parent,
 169                                                             gint                   n);
 170 static gboolean     gtk_tree_model_sort_iter_parent        (GtkTreeModel          *tree_model,
 171                                                             GtkTreeIter           *iter,
 172                                                             GtkTreeIter           *child);
 173 static void         gtk_tree_model_sort_ref_node           (GtkTreeModel          *tree_model,
 174                                                             GtkTreeIter           *iter);
 175 static void         gtk_tree_model_sort_real_unref_node    (GtkTreeModel          *tree_model,
 176                                                             GtkTreeIter           *iter,
 177                                                             gboolean               propagate_unref);
 178 static void         gtk_tree_model_sort_unref_node         (GtkTreeModel          *tree_model,
 179                                                             GtkTreeIter           *iter);
 180
 181 /* TreeSortable interface */
 182 static gboolean     gtk_tree_model_sort_get_sort_column_id    (GtkTreeSortable        *sortable,
 183                                                                gint                   *sort_column_id,
 184                                                                GtkSortType            *order);
 185 static void         gtk_tree_model_sort_set_sort_column_id    (GtkTreeSortable        *sortable,
 186                                                                gint                    sort_column_id,
 187                                                                GtkSortType        order);
 188 static void         gtk_tree_model_sort_set_sort_func         (GtkTreeSortable        *sortable,
 189                                                                gint                    sort_column_id,
 190                                                                GtkTreeIterCompareFunc  func,
 191                                                                gpointer                data,
 192                                                                GtkDestroyNotify        destroy);
 193 static void         gtk_tree_model_sort_set_default_sort_func (GtkTreeSortable        *sortable,
 194                                                                GtkTreeIterCompareFunc  func,
 195                                                                gpointer                data,
 196                                                                GtkDestroyNotify        destroy);
 197 static gboolean     gtk_tree_model_sort_has_default_sort_func (GtkTreeSortable     *sortable);
 198
 199 /* Private functions (sort funcs, level handling and other utils) */
 200 static void         gtk_tree_model_sort_build_level       (GtkTreeModelSort *tree_model_sort,
 201                                                            SortLevel        *parent_level,
 202                                                            SortElt          *parent_elt);
 203 static void         gtk_tree_model_sort_free_level        (GtkTreeModelSort *tree_model_sort,
 204                                                            SortLevel        *sort_level);
 205 static void         gtk_tree_model_sort_increment_stamp   (GtkTreeModelSort *tree_model_sort);
 206 static void         gtk_tree_model_sort_sort_level        (GtkTreeModelSort *tree_model_sort,
 207                                                            SortLevel        *level,
 208                                                            gboolean          recurse,
 209                                                            gboolean          emit_reordered);
 210 static void         gtk_tree_model_sort_sort              (GtkTreeModelSort *tree_model_sort);
 211 static gint         gtk_tree_model_sort_level_find_insert (GtkTreeModelSort *tree_model_sort,
 212                                                            SortLevel        *level,
 213                                                            GtkTreeIter      *iter,
 214                                                            gboolean          skip_sort_elt);
 215 static gboolean     gtk_tree_model_sort_insert_value      (GtkTreeModelSort *tree_model_sort,
 216                                                            SortLevel        *level,
 217                                                            GtkTreePath      *s_path,
 218                                                            GtkTreeIter      *s_iter);
 219 static GtkTreePath *gtk_tree_model_sort_elt_get_path      (SortLevel        *level,
 220                                                            SortElt          *elt);
 221 static void         gtk_tree_model_sort_set_model         (GtkTreeModelSort *tree_model_sort,
 222                                                            GtkTreeModel     *child_model);
 223 static GtkTreePath *gtk_real_tree_model_sort_convert_child_path_to_path (GtkTreeModelSort *tree_model_sort,
 224                                                                          GtkTreePath      *child_path,
 225                                                                          gboolean          build_levels);
 226
 227 static GObjectClass *parent_class = NULL;
 228
 229 GType
 230 gtk_tree_model_sort_get_type (void)
 231 {
 232   static GType tree_model_sort_type = 0;
 233
 234   if (!tree_model_sort_type)
 235     {
 236       static const GTypeInfo tree_model_sort_info =
 237       {
 238         sizeof (GtkTreeModelSortClass),
 239         NULL,           /* base_init */
 240         NULL,           /* base_finalize */
 241         (GClassInitFunc) gtk_tree_model_sort_class_init,
 242         NULL,           /* class_finalize */
 243         NULL,           /* class_data */
 244         sizeof (GtkTreeModelSort),
 245         0,              /* n_preallocs */
 246         (GInstanceInitFunc) gtk_tree_model_sort_init
 247       };
 248
 249       static const GInterfaceInfo tree_model_info =
 250       {
 251         (GInterfaceInitFunc) gtk_tree_model_sort_tree_model_init,
 252         NULL,
 253         NULL
 254       };
 255
 256       static const GInterfaceInfo sortable_info =
 257       {
 258         (GInterfaceInitFunc) gtk_tree_model_sort_tree_sortable_init,
 259         NULL,
 260         NULL
 261       };
 262
 263       tree_model_sort_type =
 264         g_type_register_static (G_TYPE_OBJECT, "GtkTreeModelSort",
 265                                 &tree_model_sort_info, 0);
 266
 267       g_type_add_interface_static (tree_model_sort_type,
 268                                    GTK_TYPE_TREE_MODEL,
 269                                    &tree_model_info);
 270
 271       g_type_add_interface_static (tree_model_sort_type,
 272                                    GTK_TYPE_TREE_SORTABLE,
 273                                    &sortable_info);
 274     }
 275
 276   return tree_model_sort_type;
 277 }
 278
 279 static void
 280 gtk_tree_model_sort_init (GtkTreeModelSort *tree_model_sort)
 281 {
 282   tree_model_sort->sort_column_id = GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID;
 283   tree_model_sort->stamp = 0;
 284   tree_model_sort->zero_ref_count = 0;
 285   tree_model_sort->root = NULL;
 286   tree_model_sort->sort_list = NULL;
 287 }
 288
 289 static void
 290 gtk_tree_model_sort_class_init (GtkTreeModelSortClass *class)
 291 {
 292   GObjectClass *object_class;
 293
 294   object_class = (GObjectClass *) class;
 295   parent_class = g_type_class_peek_parent (class);
 296
 297   object_class->set_property = gtk_tree_model_sort_set_property;
 298   object_class->get_property = gtk_tree_model_sort_get_property;
 299
 300   object_class->finalize = gtk_tree_model_sort_finalize;
 301
 302   /* Properties */
 303   g_object_class_install_property (object_class,
 304                                    PROP_MODEL,
 305                                    g_param_spec_object ("model",
 306                                                         _("TreeModelSort Model"),
 307                                                         _("The model for the TreeModelSort to sort"),
 308                                                         GTK_TYPE_TREE_MODEL,
 309                                                         G_PARAM_READWRITE | G_PARAM_CONSTRUCT_ONLY));
 310 }
 311
 312 static void
 313 gtk_tree_model_sort_tree_model_init (GtkTreeModelIface *iface)
 314 {
 315   iface->get_flags = gtk_tree_model_sort_get_flags;
 316   iface->get_n_columns = gtk_tree_model_sort_get_n_columns;
 317   iface->get_column_type = gtk_tree_model_sort_get_column_type;
 318   iface->get_iter = gtk_tree_model_sort_get_iter;
 319   iface->get_path = gtk_tree_model_sort_get_path;
 320   iface->get_value = gtk_tree_model_sort_get_value;
 321   iface->iter_next = gtk_tree_model_sort_iter_next;
 322   iface->iter_children = gtk_tree_model_sort_iter_children;
 323   iface->iter_has_child = gtk_tree_model_sort_iter_has_child;
 324   iface->iter_n_children = gtk_tree_model_sort_iter_n_children;
 325   iface->iter_nth_child = gtk_tree_model_sort_iter_nth_child;
 326   iface->iter_parent = gtk_tree_model_sort_iter_parent;
 327   iface->ref_node = gtk_tree_model_sort_ref_node;
 328   iface->unref_node = gtk_tree_model_sort_unref_node;
 329 }
 330
 331 static void
 332 gtk_tree_model_sort_tree_sortable_init (GtkTreeSortableIface *iface)
 333 {
 334   iface->get_sort_column_id = gtk_tree_model_sort_get_sort_column_id;
 335   iface->set_sort_column_id = gtk_tree_model_sort_set_sort_column_id;
 336   iface->set_sort_func = gtk_tree_model_sort_set_sort_func;
 337   iface->set_default_sort_func = gtk_tree_model_sort_set_default_sort_func;
 338   iface->has_default_sort_func = gtk_tree_model_sort_has_default_sort_func;
 339 }
 340
 341 /**
 342  * gtk_tree_model_sort_new_with_model:
 343  * @child_model: A #GtkTreeModel
 344  *
 345  * Creates a new #GtkTreeModel, with @child_model as the child_model.
 346  *
 347  * Return value: A new #GtkTreeModel.
 348  */
 349 GtkTreeModel *
 350 gtk_tree_model_sort_new_with_model (GtkTreeModel      *child_model)
 351 {
 352   GtkTreeModel *retval;
 353
 354   g_return_val_if_fail (GTK_IS_TREE_MODEL (child_model), NULL);
 355
 356   retval = g_object_new (gtk_tree_model_sort_get_type (), NULL);
 357
 358   gtk_tree_model_sort_set_model (GTK_TREE_MODEL_SORT (retval), child_model);
 359
 360   return retval;
 361 }
 362
 363 /* GObject callbacks */
 364 static void
 365 gtk_tree_model_sort_finalize (GObject *object)
 366 {
 367   GtkTreeModelSort *tree_model_sort = (GtkTreeModelSort *) object;
 368
 369   gtk_tree_model_sort_set_model (tree_model_sort, NULL);
 370
 371   if (tree_model_sort->root)
 372     gtk_tree_model_sort_free_level (tree_model_sort, tree_model_sort->root);
 373
 374   if (tree_model_sort->sort_list)
 375     {
 376       _gtk_tree_data_list_header_free (tree_model_sort->sort_list);
 377       tree_model_sort->sort_list = NULL;
 378     }
 379
 380   /* must chain up */
 381   parent_class->finalize (object);
 382 }
 383
 384 static void
 385 gtk_tree_model_sort_set_property (GObject      *object,
 386                                   guint         prop_id,
 387                                   const GValue *value,
 388                                   GParamSpec   *pspec)
 389 {
 390   GtkTreeModelSort *tree_model_sort = GTK_TREE_MODEL_SORT (object);
 391
 392   switch (prop_id)
 393     {
 394     case PROP_MODEL:
 395       gtk_tree_model_sort_set_model (tree_model_sort, g_value_get_object (value));
 396       break;
 397     default:
 398       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 399       break;
 400     }
 401 }
 402
 403 static void
 404 gtk_tree_model_sort_get_property (GObject    *object,
 405                                   guint       prop_id,
 406                                   GValue     *value,
 407                                   GParamSpec *pspec)
 408 {
 409   GtkTreeModelSort *tree_model_sort = GTK_TREE_MODEL_SORT (object);
 410
 411   switch (prop_id)
 412     {
 413     case PROP_MODEL:
 414       g_value_set_object (value, gtk_tree_model_sort_get_model(tree_model_sort));
 415       break;
 416     default:
 417       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 418       break;
 419     }
 420 }
 421
 422 static void
 423 gtk_tree_model_sort_row_changed (GtkTreeModel *s_model,
 424                                  GtkTreePath  *start_s_path,
 425                                  GtkTreeIter  *start_s_iter,
 426                                  gpointer      data)
 427 {
 428   GtkTreeModelSort *tree_model_sort = GTK_TREE_MODEL_SORT (data);
 429   GtkTreePath *path = NULL;
 430   GtkTreeIter iter;
 431   GtkTreeIter tmpiter;
 432
 433   SortElt tmp;
 434   SortElt *elt;
 435   SortLevel *level;
 436
 437   gboolean free_s_path = FALSE;
 438
 439   gint offset, index = 0, old_index, i;
 440
 441   g_return_if_fail (start_s_path != NULL || start_s_iter != NULL);
 442
 443   if (!start_s_path)
 444     {
 445       free_s_path = TRUE;
 446       start_s_path = gtk_tree_model_get_path (s_model, start_s_iter);
 447     }
 448
 449   path = gtk_real_tree_model_sort_convert_child_path_to_path (tree_model_sort,
 450                                                               start_s_path,
 451                                                               FALSE);
 452   if (!path)
 453     {
 454       if (free_s_path)
 455         gtk_tree_path_free (start_s_path);
 456       return;
 457     }
 458
 459   gtk_tree_model_get_iter (GTK_TREE_MODEL (data), &iter, path);
 460
 461   level = iter.user_data;
 462   elt = iter.user_data2;
 463
 464   if (level->array->len < 2 ||
 465       (tree_model_sort->sort_column_id == GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID &&
 466        tree_model_sort->default_sort_func == NO_SORT_FUNC))
 467     {
 468       if (free_s_path)
 469         gtk_tree_path_free (start_s_path);
 470
 471       gtk_tree_model_row_changed (GTK_TREE_MODEL (data), path, &iter);
 472
 473       gtk_tree_path_free (path);
 474
 475       return;
 476     }
 477
 478   if (!GTK_TREE_MODEL_SORT_CACHE_CHILD_ITERS (tree_model_sort))
 479     {
 480       gtk_tree_model_get_iter (tree_model_sort->child_model,
 481                                &tmpiter, start_s_path);
 482     }
 483
 484   offset = elt->offset;
 485
 486   for (i = 0; i < level->array->len; i++)
 487     if (elt->offset == g_array_index (level->array, SortElt, i).offset)
 488       old_index = i;
 489
 490   memcpy (&tmp, elt, sizeof (SortElt));
 491
 492   if (GTK_TREE_MODEL_SORT_CACHE_CHILD_ITERS (tree_model_sort))
 493     index = gtk_tree_model_sort_level_find_insert (tree_model_sort,
 494                                                    level,
 495                                                    &tmp.iter,
 496                                                    TRUE);
 497   else
 498     index = gtk_tree_model_sort_level_find_insert (tree_model_sort,
 499                                                    level,
 500                                                    &tmpiter,
 501                                                    TRUE);
 502
 503   if (index < old_index)
 504     {
 505       g_memmove (level->array->data + ((index + 1)*sizeof (SortElt)),
 506                  level->array->data + ((index)*sizeof (SortElt)),
 507                  (old_index - index)* sizeof(SortElt));
 508     }
 509   else if (index > old_index)
 510     {
 511       g_memmove (level->array->data + ((old_index)*sizeof (SortElt)),
 512                  level->array->data + ((old_index + 1)*sizeof (SortElt)),
 513                  (index - old_index)* sizeof(SortElt));
 514     }
 515   memcpy (level->array->data + ((index)*sizeof (SortElt)),
 516           &tmp, sizeof (SortElt));
 517
 518   for (i = 0; i < level->array->len; i++)
 519     if (g_array_index (level->array, SortElt, i).children)
 520       g_array_index (level->array, SortElt, i).children->parent_elt = &g_array_index (level->array, SortElt, i);
 521
 522   gtk_tree_path_up (path);
 523   gtk_tree_path_append_index (path, index);
 524
 525   gtk_tree_model_sort_increment_stamp (tree_model_sort);
 526
 527   /* if the item moved, then emit rows_reordered */
 528   if (old_index != index)
 529     {
 530       gint *new_order;
 531       gint j;
 532
 533       GtkTreePath *tmppath;
 534
 535       new_order = g_new (gint, level->array->len);
 536
 537       for (j = 0; j < level->array->len; j++)
 538         {
 539           if (index > old_index)
 540             {
 541               if (j == index)
 542                 new_order[j] = old_index;
 543               else if (j >= old_index && j < index)
 544                 new_order[j] = j + 1;
 545               else
 546                 new_order[j] = j;
 547             }
 548           else if (index < old_index)
 549             {
 550               if (j == index)
 551                 new_order[j] = old_index;
 552               else if (j > index && j <= old_index)
 553                 new_order[j] = j - 1;
 554               else
 555                 new_order[j] = j;
 556             }
 557           /* else? shouldn't really happen */
 558         }
 559
 560       if (level->parent_elt)
 561         {
 562           iter.stamp = tree_model_sort->stamp;
 563           iter.user_data = level->parent_level;
 564           iter.user_data2 = level->parent_elt;
 565
 566           tmppath = gtk_tree_model_get_path (GTK_TREE_MODEL (tree_model_sort), &iter);
 567
 568           gtk_tree_model_rows_reordered (GTK_TREE_MODEL (tree_model_sort),
 569                                          tmppath, &iter, new_order);
 570         }
 571       else
 572         {
 573           /* toplevel */
 574           tmppath = gtk_tree_path_new ();
 575
 576           gtk_tree_model_rows_reordered (GTK_TREE_MODEL (tree_model_sort), tmppath,
 577                                          NULL, new_order);
 578         }
 579
 580       gtk_tree_path_free (tmppath);
 581       g_free (new_order);
 582     }
 583
 584   /* emit row_changed signal (at new location) */
 585   gtk_tree_model_get_iter (GTK_TREE_MODEL (data), &iter, path);
 586   gtk_tree_model_row_changed (GTK_TREE_MODEL (data), path, &iter);
 587
 588   gtk_tree_path_free (path);
 589   if (free_s_path)
 590     gtk_tree_path_free (start_s_path);
 591 }
 592
 593 static void
 594 gtk_tree_model_sort_row_inserted (GtkTreeModel          *s_model,
 595                                   GtkTreePath           *s_path,
 596                                   GtkTreeIter           *s_iter,
 597                                   gpointer               data)
 598 {
 599   GtkTreeModelSort *tree_model_sort = GTK_TREE_MODEL_SORT (data);
 600   GtkTreePath *path;
 601   GtkTreeIter iter;
 602   GtkTreeIter real_s_iter;
 603
 604   gint i = 0;
 605
 606   gboolean free_s_path = FALSE;
 607
 608   SortElt *elt;
 609   SortLevel *level;
 610   SortLevel *parent_level = NULL;
 611
 612   parent_level = level = SORT_LEVEL (tree_model_sort->root);
 613
 614   g_return_if_fail (s_path != NULL || s_iter != NULL);
 615
 616   if (!s_path)
 617     {
 618       s_path = gtk_tree_model_get_path (s_model, s_iter);
 619       free_s_path = TRUE;
 620     }
 621
 622   if (!s_iter)
 623     gtk_tree_model_get_iter (s_model, &real_s_iter, s_path);
 624   else
 625     real_s_iter = *s_iter;
 626
 627   if (!tree_model_sort->root)
 628     {
 629       gtk_tree_model_sort_build_level (tree_model_sort, NULL, NULL);
 630
 631       /* the build level already put the inserted iter in the level,
 632          so no need to handle this signal anymore */
 633
 634       goto done_and_submit;
 635     }
 636
 637   /* find the parent level */
 638   while (i < gtk_tree_path_get_depth (s_path) - 1)
 639     {
 640       gint j;
 641
 642       if (!level)
 643         {
 644           /* level not yet build, we won't cover this signal */
 645           goto done;
 646         }
 647
 648       if (level->array->len < gtk_tree_path_get_indices (s_path)[i])
 649         {
 650           g_warning ("A node was inserted with a parent that's not in the tree.\n"
 651                      "This possibly means that a GtkTreeModel inserted a child node\n"
 652                      "before the parent was inserted.");
 653           goto done;
 654         }
 655
 656       elt = NULL;
 657       for (j = 0; j < level->array->len; j++)
 658         if (g_array_index (level->array, SortElt, j).offset == gtk_tree_path_get_indices (s_path)[i])
 659           {
 660             elt = &g_array_index (level->array, SortElt, j);
 661             break;
 662           }
 663
 664       g_return_if_fail (elt != NULL);
 665
 666       if (!elt->children)
 667         {
 668           GtkTreePath *tmppath;
 669           GtkTreeIter  tmpiter;
 670
 671           tmpiter.stamp = tree_model_sort->stamp;
 672           tmpiter.user_data = level;
 673           tmpiter.user_data2 = elt;
 674
 675           tmppath = gtk_tree_model_get_path (GTK_TREE_MODEL (data), &tmpiter);
 676           if (tmppath)
 677             {
 678               gtk_tree_model_row_has_child_toggled (GTK_TREE_MODEL (data),
 679                                                     tmppath,
 680                                                     &tmpiter);
 681               gtk_tree_path_free (tmppath);
 682             }
 683
 684           /* not covering this signal */
 685           goto done;
 686         }
 687
 688       level = elt->children;
 689       parent_level = level;
 690       i++;
 691     }
 692
 693   if (!parent_level)
 694     goto done;
 695
 696   if (!gtk_tree_model_sort_insert_value (tree_model_sort,
 697                                          parent_level,
 698                                          s_path,
 699                                          &real_s_iter))
 700     goto done;
 701
 702  done_and_submit:
 703   path = gtk_real_tree_model_sort_convert_child_path_to_path (tree_model_sort,
 704                                                               s_path,
 705                                                               FALSE);
 706
 707   if (!path)
 708     return;
 709
 710   gtk_tree_model_sort_increment_stamp (tree_model_sort);
 711
 712   gtk_tree_model_get_iter (GTK_TREE_MODEL (data), &iter, path);
 713   gtk_tree_model_row_inserted (GTK_TREE_MODEL (data), path, &iter);
 714   gtk_tree_path_free (path);
 715
 716  done:
 717   if (free_s_path)
 718     gtk_tree_path_free (s_path);
 719
 720   return;
 721 }
 722
 723 static void
 724 gtk_tree_model_sort_row_has_child_toggled (GtkTreeModel *s_model,
 725                                            GtkTreePath  *s_path,
 726                                            GtkTreeIter  *s_iter,
 727                                            gpointer      data)
 728 {
 729   GtkTreeModelSort *tree_model_sort = GTK_TREE_MODEL_SORT (data);
 730   GtkTreePath *path;
 731   GtkTreeIter iter;
 732
 733   g_return_if_fail (s_path != NULL && s_iter != NULL);
 734
 735   path = gtk_real_tree_model_sort_convert_child_path_to_path (tree_model_sort, s_path, FALSE);
 736   if (path == NULL)
 737     return;
 738
 739   gtk_tree_model_get_iter (GTK_TREE_MODEL (data), &iter, path);
 740   gtk_tree_model_row_has_child_toggled (GTK_TREE_MODEL (data), path, &iter);
 741
 742   gtk_tree_path_free (path);
 743 }
 744
 745 /* FIXME: I still have doubts if this works */
 746 static void
 747 gtk_tree_model_sort_row_deleted (GtkTreeModel *s_model,
 748                                  GtkTreePath  *s_path,
 749                                  gpointer      data)
 750 {
 751   GtkTreeModelSort *tree_model_sort = GTK_TREE_MODEL_SORT (data);
 752   GtkTreePath *path = NULL;
 753   SortElt *elt;
 754   SortLevel *level;
 755   GtkTreeIter iter;
 756   gint offset;
 757   gint i;
 758
 759   g_return_if_fail (s_path != NULL);
 760
 761   path = gtk_real_tree_model_sort_convert_child_path_to_path (tree_model_sort, s_path, FALSE);
 762   if (path == NULL)
 763     return;
 764
 765   gtk_tree_model_get_iter (GTK_TREE_MODEL (data), &iter, path);
 766
 767   level = SORT_LEVEL (iter.user_data);
 768   elt = SORT_ELT (iter.user_data2);
 769   offset = elt->offset;
 770
 771   /* we _need_ to emit ::row_deleted before we start unreffing the node
 772    * itself. This is because of the row refs, which start unreffing nodes
 773    * when we emit ::row_deleted
 774    */
 775   gtk_tree_model_row_deleted (GTK_TREE_MODEL (data), path);
 776
 777   gtk_tree_model_get_iter (GTK_TREE_MODEL (data), &iter, path);
 778
 779   while (elt->ref_count > 0)
 780     gtk_tree_model_sort_real_unref_node (GTK_TREE_MODEL (data), &iter, FALSE);
 781
 782   if (level->ref_count == 0 && level != tree_model_sort->root)
 783     {
 784       /* This will prune the level, so I can just emit the signal and not worry
 785        * about cleaning this level up. */
 786       gtk_tree_model_sort_increment_stamp (tree_model_sort);
 787       gtk_tree_path_free (path);
 788       return;
 789     }
 790
 791   gtk_tree_model_sort_increment_stamp (tree_model_sort);
 792
 793   /* Remove the row */
 794   for (i = 0; i < level->array->len; i++)
 795     if (elt->offset == g_array_index (level->array, SortElt, i).offset)
 796       break;
 797
 798   g_array_remove_index (level->array, i);
 799
 800   /* update all offsets */
 801   for (i = 0; i < level->array->len; i++)
 802     {
 803       elt = & (g_array_index (level->array, SortElt, i));
 804       if (elt->offset > offset)
 805         elt->offset--;
 806       if (elt->children)
 807         elt->children->parent_elt = elt;
 808     }
 809
 810   gtk_tree_path_free (path);
 811 }
 812
 813 static void
 814 gtk_tree_model_sort_rows_reordered (GtkTreeModel *s_model,
 815                                     GtkTreePath  *s_path,
 816                                     GtkTreeIter  *s_iter,
 817                                     gint         *new_order,
 818                                     gpointer      data)
 819 {
 820   SortElt *elt;
 821   SortLevel *level;
 822   GtkTreeIter iter;
 823   gint *tmp_array;
 824   int i, j;
 825   GtkTreePath *path;
 826   GtkTreeModelSort *tree_model_sort = GTK_TREE_MODEL_SORT (data);
 827
 828   g_return_if_fail (new_order != NULL);
 829
 830   if (s_path == NULL || gtk_tree_path_get_indices (s_path) == NULL)
 831     {
 832       if (tree_model_sort->root == NULL)
 833         return;
 834       path = gtk_tree_path_new ();
 835       level = SORT_LEVEL (tree_model_sort->root);
 836     }
 837   else
 838     {
 839       path = gtk_real_tree_model_sort_convert_child_path_to_path (tree_model_sort, s_path, FALSE);
 840       if (path == NULL)
 841         return;
 842       gtk_tree_model_get_iter (GTK_TREE_MODEL (data), &iter, path);
 843
 844       level = SORT_LEVEL (iter.user_data);
 845       elt = SORT_ELT (iter.user_data2);
 846
 847       if (!elt->children)
 848         {
 849           gtk_tree_path_free (path);
 850           return;
 851         }
 852
 853       level = elt->children;
 854     }
 855
 856   if (level->array->len < 2)
 857     {
 858       gtk_tree_path_free (path);
 859       return;
 860     }
 861
 862   tmp_array = g_new (int, level->array->len);
 863   for (i = 0; i < level->array->len; i++)
 864     {
 865       for (j = 0; j < level->array->len; j++)
 866         {
 867           if (g_array_index (level->array, SortElt, i).offset == new_order[j])
 868             tmp_array[i] = j;
 869         }
 870     }
 871
 872   for (i = 0; i < level->array->len; i++)
 873     g_array_index (level->array, SortElt, i).offset = tmp_array[i];
 874   g_free (tmp_array);
 875
 876   if (tree_model_sort->sort_column_id == GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID &&
 877       tree_model_sort->default_sort_func == NO_SORT_FUNC)
 878     {
 879
 880       gtk_tree_model_sort_sort_level (tree_model_sort, level,
 881                                       FALSE, FALSE);
 882       gtk_tree_model_sort_increment_stamp (tree_model_sort);
 883
 884       if (gtk_tree_path_get_depth (path))
 885         {
 886           gtk_tree_model_get_iter (GTK_TREE_MODEL (tree_model_sort),
 887                                    &iter,
 888                                    path);
 889           gtk_tree_model_rows_reordered (GTK_TREE_MODEL (tree_model_sort),
 890                                          path, &iter, new_order);
 891         }
 892       else
 893         {
 894           gtk_tree_model_rows_reordered (GTK_TREE_MODEL (tree_model_sort),
 895                                          path, NULL, new_order);
 896         }
 897     }
 898
 899   gtk_tree_path_free (path);
 900 }
 901
 902 /* Fulfill our model requirements */
 903 static GtkTreeModelFlags
 904 gtk_tree_model_sort_get_flags (GtkTreeModel *tree_model)
 905 {
 906   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model), 0);
 907
 908   return 0;
 909 }
 910
 911 static gint
 912 gtk_tree_model_sort_get_n_columns (GtkTreeModel *tree_model)
 913 {
 914   GtkTreeModelSort *tree_model_sort = (GtkTreeModelSort *) tree_model;
 915
 916   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model), 0);
 917
 918   if (tree_model_sort->child_model == 0)
 919     return 0;
 920
 921   return gtk_tree_model_get_n_columns (tree_model_sort->child_model);
 922 }
 923
 924 static GType
 925 gtk_tree_model_sort_get_column_type (GtkTreeModel *tree_model,
 926                                      gint          index)
 927 {
 928   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model), G_TYPE_INVALID);
 929   g_return_val_if_fail (GTK_TREE_MODEL_SORT (tree_model)->child_model != NULL, G_TYPE_INVALID);
 930
 931   return gtk_tree_model_get_column_type (GTK_TREE_MODEL_SORT (tree_model)->child_model, index);
 932 }
 933
 934 static gboolean
 935 gtk_tree_model_sort_get_iter (GtkTreeModel *tree_model,
 936                               GtkTreeIter  *iter,
 937                               GtkTreePath  *path)
 938 {
 939   GtkTreeModelSort *tree_model_sort;
 940   gint *indices;
 941   SortLevel *level;
 942   gint depth, i;
 943
 944   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model), FALSE);
 945   g_return_val_if_fail (GTK_TREE_MODEL_SORT (tree_model)->child_model != NULL, FALSE);
 946
 947   tree_model_sort = (GtkTreeModelSort *) tree_model;
 948   indices = gtk_tree_path_get_indices (path);
 949
 950   if (tree_model_sort->root == NULL)
 951     gtk_tree_model_sort_build_level (tree_model_sort, NULL, NULL);
 952   level = SORT_LEVEL (tree_model_sort->root);
 953
 954   depth = gtk_tree_path_get_depth (path);
 955   if (depth == 0)
 956     return FALSE;
 957
 958   for (i = 0; i < depth - 1; i++)
 959     {
 960       if ((level == NULL) ||
 961           (indices[i] >= level->array->len))
 962         return FALSE;
 963
 964       if (g_array_index (level->array, SortElt, indices[i]).children == NULL)
 965         gtk_tree_model_sort_build_level (tree_model_sort, level, &g_array_index (level->array, SortElt, indices[i]));
 966       level = g_array_index (level->array, SortElt, indices[i]).children;
 967     }
 968
 969   if (level == NULL)
 970     return FALSE;
 971   iter->stamp = tree_model_sort->stamp;
 972   iter->user_data = level;
 973   iter->user_data2 = &g_array_index (level->array, SortElt, indices[depth - 1]);
 974
 975   return TRUE;
 976 }
 977
 978 static GtkTreePath *
 979 gtk_tree_model_sort_get_path (GtkTreeModel *tree_model,
 980                               GtkTreeIter  *iter)
 981 {
 982   GtkTreePath *retval;
 983   SortLevel *level;
 984   SortElt *elt;
 985
 986   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model), NULL);
 987   g_return_val_if_fail (GTK_TREE_MODEL_SORT (tree_model)->child_model != NULL, NULL);
 988   g_return_val_if_fail (GTK_TREE_MODEL_SORT (tree_model)->stamp == iter->stamp, NULL);
 989
 990   retval = gtk_tree_path_new ();
 991   level = iter->user_data;
 992   elt = iter->user_data2;
 993   while (level != NULL)
 994     {
 995       gtk_tree_path_prepend_index (retval, elt - (SortElt *)level->array->data);
 996
 997       elt = level->parent_elt;
 998       level = level->parent_level;
 999     }
1000
1001   return retval;
1002 }
1003
1004 static void
1005 gtk_tree_model_sort_get_value (GtkTreeModel *tree_model,
1006                                GtkTreeIter  *iter,
1007                                gint          column,
1008                                GValue       *value)
1009 {
1010   GtkTreeIter child_iter;
1011
1012   g_return_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model));
1013   g_return_if_fail (GTK_TREE_MODEL_SORT (tree_model)->child_model != NULL);
1014   g_return_if_fail (GTK_TREE_MODEL_SORT (tree_model)->stamp == iter->stamp);
1015
1016   GET_CHILD_ITER (tree_model, &child_iter, iter);
1017   gtk_tree_model_get_value (GTK_TREE_MODEL_SORT (tree_model)->child_model,
1018                             &child_iter, column, value);
1019 }
1020
1021 static gboolean
1022 gtk_tree_model_sort_iter_next (GtkTreeModel *tree_model,
1023                                GtkTreeIter  *iter)
1024 {
1025   SortLevel *level;
1026   SortElt *elt;
1027
1028   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model), FALSE);
1029   g_return_val_if_fail (GTK_TREE_MODEL_SORT (tree_model)->child_model != NULL, FALSE);
1030   g_return_val_if_fail (GTK_TREE_MODEL_SORT (tree_model)->stamp == iter->stamp, FALSE);
1031
1032   level = iter->user_data;
1033   elt = iter->user_data2;
1034
1035   if (elt - (SortElt *)level->array->data >= level->array->len - 1)
1036     {
1037       iter->stamp = 0;
1038       return FALSE;
1039     }
1040   iter->user_data2 = elt + 1;
1041
1042   return TRUE;
1043 }
1044
1045 static gboolean
1046 gtk_tree_model_sort_iter_children (GtkTreeModel *tree_model,
1047                                    GtkTreeIter  *iter,
1048                                    GtkTreeIter  *parent)
1049 {
1050   GtkTreeModelSort *tree_model_sort = (GtkTreeModelSort *) tree_model;
1051   SortLevel *level;
1052
1053   iter->stamp = 0;
1054   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model), FALSE);
1055   g_return_val_if_fail (tree_model_sort->child_model != NULL, FALSE);
1056   if (parent) g_return_val_if_fail (tree_model_sort->stamp == parent->stamp, FALSE);
1057
1058   if (parent == NULL)
1059     {
1060       if (tree_model_sort->root == NULL)
1061         gtk_tree_model_sort_build_level (tree_model_sort, NULL, NULL);
1062       if (tree_model_sort->root == NULL)
1063         return FALSE;
1064
1065       level = tree_model_sort->root;
1066       iter->stamp = tree_model_sort->stamp;
1067       iter->user_data = level;
1068       iter->user_data2 = level->array->data;
1069     }
1070   else
1071     {
1072       if (((SortElt *)parent->user_data2)->children == NULL)
1073         gtk_tree_model_sort_build_level (tree_model_sort,
1074                                          (SortLevel *)parent->user_data,
1075                                          (SortElt *)parent->user_data2);
1076       if (((SortElt *)parent->user_data2)->children == NULL)
1077         return FALSE;
1078       iter->stamp = tree_model_sort->stamp;
1079       iter->user_data = ((SortElt *)parent->user_data2)->children;
1080       iter->user_data2 = ((SortLevel *)iter->user_data)->array->data;
1081     }
1082
1083   return TRUE;
1084 }
1085
1086 static gboolean
1087 gtk_tree_model_sort_iter_has_child (GtkTreeModel *tree_model,
1088                                     GtkTreeIter  *iter)
1089 {
1090   GtkTreeIter child_iter;
1091
1092   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model), FALSE);
1093   g_return_val_if_fail (GTK_TREE_MODEL_SORT (tree_model)->child_model != NULL, FALSE);
1094   g_return_val_if_fail (GTK_TREE_MODEL_SORT (tree_model)->stamp == iter->stamp, FALSE);
1095
1096   GET_CHILD_ITER (tree_model, &child_iter, iter);
1097
1098   return gtk_tree_model_iter_has_child (GTK_TREE_MODEL_SORT (tree_model)->child_model, &child_iter);
1099 }
1100
1101 static gint
1102 gtk_tree_model_sort_iter_n_children (GtkTreeModel *tree_model,
1103                                      GtkTreeIter  *iter)
1104 {
1105   GtkTreeIter child_iter;
1106
1107   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model), 0);
1108   g_return_val_if_fail (GTK_TREE_MODEL_SORT (tree_model)->child_model != NULL, 0);
1109   if (iter) g_return_val_if_fail (GTK_TREE_MODEL_SORT (tree_model)->stamp == iter->stamp, 0);
1110
1111   if (iter == NULL)
1112     return gtk_tree_model_iter_n_children (GTK_TREE_MODEL_SORT (tree_model)->child_model, NULL);
1113
1114   GET_CHILD_ITER (tree_model, &child_iter, iter);
1115
1116   return gtk_tree_model_iter_n_children (GTK_TREE_MODEL_SORT (tree_model)->child_model, &child_iter);
1117 }
1118
1119 static gboolean
1120 gtk_tree_model_sort_iter_nth_child (GtkTreeModel *tree_model,
1121                                     GtkTreeIter  *iter,
1122                                     GtkTreeIter  *parent,
1123                                     gint          n)
1124 {
1125   SortLevel *level;
1126   /* We have this for the iter == parent case */
1127   GtkTreeIter children;
1128
1129   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model), FALSE);
1130   if (parent) g_return_val_if_fail (GTK_TREE_MODEL_SORT (tree_model)->stamp == parent->stamp, FALSE);
1131
1132   /* Use this instead of has_child to force us to build the level, if needed */
1133   if (gtk_tree_model_sort_iter_children (tree_model, &children, parent) == FALSE)
1134     {
1135       iter->stamp = 0;
1136       return FALSE;
1137     }
1138
1139   level = children.user_data;
1140   if (n >= level->array->len)
1141     {
1142       iter->stamp = 0;
1143       return FALSE;
1144     }
1145
1146   iter->stamp = GTK_TREE_MODEL_SORT (tree_model)->stamp;
1147   iter->user_data = level;
1148   iter->user_data2 = &g_array_index (level->array, SortElt, n);
1149
1150   return TRUE;
1151 }
1152
1153 static gboolean
1154 gtk_tree_model_sort_iter_parent (GtkTreeModel *tree_model,
1155                                  GtkTreeIter  *iter,
1156                                  GtkTreeIter  *child)
1157 {
1158   SortLevel *level;
1159
1160   iter->stamp = 0;
1161   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model), FALSE);
1162   g_return_val_if_fail (GTK_TREE_MODEL_SORT (tree_model)->child_model != NULL, FALSE);
1163   g_return_val_if_fail (GTK_TREE_MODEL_SORT (tree_model)->stamp == child->stamp, FALSE);
1164
1165   level = child->user_data;
1166
1167   if (level->parent_level)
1168     {
1169       iter->stamp = GTK_TREE_MODEL_SORT (tree_model)->stamp;
1170       iter->user_data = level->parent_level;
1171       iter->user_data2 = level->parent_elt;
1172
1173       return TRUE;
1174     }
1175   return FALSE;
1176 }
1177
1178 static void
1179 gtk_tree_model_sort_ref_node (GtkTreeModel *tree_model,
1180                               GtkTreeIter  *iter)
1181 {
1182   GtkTreeModelSort *tree_model_sort = (GtkTreeModelSort *) tree_model;
1183   GtkTreeIter child_iter;
1184   SortLevel *level;
1185   SortElt *elt;
1186
1187   g_return_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model));
1188   g_return_if_fail (GTK_TREE_MODEL_SORT (tree_model)->child_model != NULL);
1189   g_return_if_fail (GTK_TREE_MODEL_SORT (tree_model)->stamp == iter->stamp);
1190
1191   GET_CHILD_ITER (tree_model, &child_iter, iter);
1192
1193   gtk_tree_model_ref_node (GTK_TREE_MODEL_SORT (tree_model)->child_model, &child_iter);
1194
1195   level = iter->user_data;
1196   elt = iter->user_data2;
1197
1198   elt->ref_count++;
1199   level->ref_count++;
1200   if (level->ref_count == 1)
1201     {
1202       SortLevel *parent_level = level->parent_level;
1203       SortElt *parent_elt = level->parent_elt;
1204       /* We were at zero -- time to decrement the zero_ref_count val */
1205       do
1206         {
1207           if (parent_elt)
1208             parent_elt->zero_ref_count--;
1209           else
1210             tree_model_sort->zero_ref_count--;
1211
1212           if (parent_level)
1213             {
1214               parent_elt = parent_level->parent_elt;
1215               parent_level = parent_level->parent_level;
1216             }
1217         }
1218       while (parent_level);
1219     }
1220 }
1221
1222 static void
1223 gtk_tree_model_sort_real_unref_node (GtkTreeModel *tree_model,
1224                                      GtkTreeIter  *iter,
1225                                      gboolean      propagate_unref)
1226 {
1227   GtkTreeModelSort *tree_model_sort = (GtkTreeModelSort *) tree_model;
1228   GtkTreeIter child_iter;
1229   SortLevel *level;
1230   SortElt *elt;
1231
1232   g_return_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model));
1233   g_return_if_fail (GTK_TREE_MODEL_SORT (tree_model)->child_model != NULL);
1234   g_return_if_fail (GTK_TREE_MODEL_SORT (tree_model)->stamp == iter->stamp);
1235
1236   GET_CHILD_ITER (tree_model, &child_iter, iter);
1237
1238   if (propagate_unref)
1239     gtk_tree_model_unref_node (GTK_TREE_MODEL_SORT (tree_model)->child_model, &child_iter);
1240
1241   level = iter->user_data;
1242   elt = iter->user_data2;
1243
1244   g_return_if_fail (elt->ref_count > 0);
1245
1246   elt->ref_count--;
1247   level->ref_count--;
1248
1249   if (level->ref_count == 0)
1250     {
1251       SortLevel *parent_level = level->parent_level;
1252       SortElt *parent_elt = level->parent_elt;
1253
1254       /* We are at zero -- time to increment the zero_ref_count val */
1255       while (parent_level)
1256         {
1257           parent_elt->zero_ref_count++;
1258
1259           parent_elt = parent_level->parent_elt;
1260           parent_level = parent_level->parent_level;
1261         }
1262       tree_model_sort->zero_ref_count++;
1263     }
1264 }
1265
1266 static void
1267 gtk_tree_model_sort_unref_node (GtkTreeModel *tree_model,
1268                                 GtkTreeIter  *iter)
1269 {
1270   gtk_tree_model_sort_real_unref_node (tree_model, iter, TRUE);
1271 }
1272
1273 /* Sortable interface */
1274 static gboolean
1275 gtk_tree_model_sort_get_sort_column_id (GtkTreeSortable *sortable,
1276                                         gint            *sort_column_id,
1277                                         GtkSortType     *order)
1278 {
1279   GtkTreeModelSort *tree_model_sort = (GtkTreeModelSort *)sortable;
1280
1281   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (sortable), FALSE);
1282
1283   if (tree_model_sort->sort_column_id == GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID)
1284     return FALSE;
1285
1286   if (sort_column_id)
1287     *sort_column_id = tree_model_sort->sort_column_id;
1288   if (order)
1289     *order = tree_model_sort->order;
1290
1291   return TRUE;
1292 }
1293
1294 static void
1295 gtk_tree_model_sort_set_sort_column_id (GtkTreeSortable *sortable,
1296                                         gint             sort_column_id,
1297                                         GtkSortType      order)
1298 {
1299   GtkTreeModelSort *tree_model_sort = (GtkTreeModelSort *)sortable;
1300
1301   g_return_if_fail (GTK_IS_TREE_MODEL_SORT (sortable));
1302
1303   if (sort_column_id != GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID)
1304     {
1305       GtkTreeDataSortHeader *header = NULL;
1306
1307       header = _gtk_tree_data_list_get_header (tree_model_sort->sort_list,
1308                                                sort_column_id);
1309
1310       /* we want to make sure that we have a function */
1311       g_return_if_fail (header != NULL);
1312       g_return_if_fail (header->func != NULL);
1313     }
1314   else
1315     g_return_if_fail (tree_model_sort->default_sort_func != NULL);
1316
1317   if (tree_model_sort->sort_column_id == sort_column_id)
1318     {
1319       if (sort_column_id != GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID)
1320         {
1321           if (tree_model_sort->order == order)
1322             return;
1323         }
1324       else
1325         return;
1326     }
1327
1328   tree_model_sort->sort_column_id = sort_column_id;
1329   tree_model_sort->order = order;
1330
1331   gtk_tree_sortable_sort_column_changed (sortable);
1332
1333   gtk_tree_model_sort_sort (tree_model_sort);
1334 }
1335
1336 static void
1337 gtk_tree_model_sort_set_sort_func (GtkTreeSortable        *sortable,
1338                                    gint                    sort_column_id,
1339                                    GtkTreeIterCompareFunc  func,
1340                                    gpointer                data,
1341                                    GtkDestroyNotify        destroy)
1342 {
1343   GtkTreeModelSort *tree_model_sort = (GtkTreeModelSort *) sortable;
1344   GtkTreeDataSortHeader *header = NULL;
1345   GList *list;
1346
1347   g_return_if_fail (GTK_IS_TREE_MODEL_SORT (sortable));
1348   g_return_if_fail (func != NULL);
1349
1350   for (list = tree_model_sort->sort_list; list; list = list->next)
1351     {
1352       GtkTreeDataSortHeader *list_header;
1353
1354       list_header = (GtkTreeDataSortHeader*) list->data;
1355       if (list_header->sort_column_id == sort_column_id)
1356         {
1357           header = list_header;
1358           break;
1359         }
1360     }
1361
1362   if (header == NULL)
1363     {
1364       header = g_new0 (GtkTreeDataSortHeader, 1);
1365       header->sort_column_id = sort_column_id;
1366       tree_model_sort->sort_list = g_list_append (tree_model_sort->sort_list,
1367                                                   header);
1368     }
1369
1370   if (header->destroy)
1371     {
1372       GtkDestroyNotify d = header->destroy;
1373
1374       header->destroy = NULL;
1375       d (header->data);
1376     }
1377
1378   header->func = func;
1379   header->data = data;
1380   header->destroy = destroy;
1381
1382   if (tree_model_sort->sort_column_id == sort_column_id)
1383     gtk_tree_model_sort_sort (tree_model_sort);
1384 }
1385
1386 static void
1387 gtk_tree_model_sort_set_default_sort_func (GtkTreeSortable        *sortable,
1388                                            GtkTreeIterCompareFunc  func,
1389                                            gpointer                data,
1390                                            GtkDestroyNotify        destroy)
1391 {
1392   GtkTreeModelSort *tree_model_sort = (GtkTreeModelSort *)sortable;
1393
1394   g_return_if_fail (GTK_IS_TREE_MODEL_SORT (sortable));
1395
1396   if (tree_model_sort->default_sort_destroy)
1397     {
1398       GtkDestroyNotify d = tree_model_sort->default_sort_destroy;
1399
1400       tree_model_sort->default_sort_destroy = NULL;
1401       d (tree_model_sort->default_sort_data);
1402     }
1403
1404   tree_model_sort->default_sort_func = func;
1405   tree_model_sort->default_sort_data = data;
1406   tree_model_sort->default_sort_destroy = destroy;
1407
1408   if (tree_model_sort->sort_column_id == GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID)
1409     gtk_tree_model_sort_sort (tree_model_sort);
1410 }
1411
1412 static gboolean
1413 gtk_tree_model_sort_has_default_sort_func (GtkTreeSortable *sortable)
1414 {
1415   GtkTreeModelSort *tree_model_sort = (GtkTreeModelSort *)sortable;
1416
1417   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (sortable), FALSE);
1418
1419   return (tree_model_sort->default_sort_func != NULL);
1420 }
1421
1422 /* sorting code - private */
1423 static gint
1424 gtk_tree_model_sort_compare_func (gconstpointer a,
1425                                   gconstpointer b,
1426                                   gpointer      user_data)
1427 {
1428   SortData *data = (SortData *)user_data;
1429   GtkTreeModelSort *tree_model_sort = data->tree_model_sort;
1430   SortTuple *sa = (SortTuple *)a;
1431   SortTuple *sb = (SortTuple *)b;
1432
1433   GtkTreeIter iter_a, iter_b;
1434   gint retval;
1435
1436   /* shortcut, if we've the same offsets here, they should be equal */
1437   if (sa->offset == sb->offset)
1438     return 0;
1439
1440   if (GTK_TREE_MODEL_SORT_CACHE_CHILD_ITERS (tree_model_sort))
1441     {
1442       iter_a = sa->elt->iter;
1443       iter_b = sb->elt->iter;
1444     }
1445   else
1446     {
1447       data->parent_path_indices [data->parent_path_depth-1] = sa->elt->offset;
1448       gtk_tree_model_get_iter (GTK_TREE_MODEL (tree_model_sort->child_model), &iter_a, data->parent_path);
1449       data->parent_path_indices [data->parent_path_depth-1] = sb->elt->offset;
1450       gtk_tree_model_get_iter (GTK_TREE_MODEL (tree_model_sort->child_model), &iter_b, data->parent_path);
1451     }
1452
1453   retval = (* data->sort_func) (GTK_TREE_MODEL (tree_model_sort->child_model),
1454                                 &iter_a, &iter_b,
1455                                 data->sort_data);
1456
1457   if (tree_model_sort->order == GTK_SORT_DESCENDING)
1458     {
1459       if (retval > 0)
1460         retval = -1;
1461       else if (retval < 0)
1462         retval = 1;
1463     }
1464
1465   return retval;
1466 }
1467
1468 static gint
1469 gtk_tree_model_sort_offset_compare_func (gconstpointer a,
1470                                          gconstpointer b,
1471                                          gpointer      user_data)
1472 {
1473   gint retval;
1474
1475   SortTuple *sa = (SortTuple *)a;
1476   SortTuple *sb = (SortTuple *)b;
1477
1478   SortData *data = (SortData *)user_data;
1479
1480   if (sa->elt->offset < sb->elt->offset)
1481     retval = -1;
1482   else if (sa->elt->offset > sb->elt->offset)
1483     retval = 1;
1484   else
1485     retval = 0;
1486
1487   if (data->tree_model_sort->order == GTK_SORT_DESCENDING)
1488     {
1489       if (retval > 0)
1490         retval = -1;
1491       else if (retval < 0)
1492         retval = 1;
1493     }
1494
1495   return retval;
1496 }
1497
1498 static void
1499 gtk_tree_model_sort_sort_level (GtkTreeModelSort *tree_model_sort,
1500                                 SortLevel        *level,
1501                                 gboolean          recurse,
1502                                 gboolean          emit_reordered)
1503 {
1504   gint i;
1505   GArray *sort_array;
1506   GArray *new_array;
1507   gint *new_order;
1508
1509   GtkTreeIter iter;
1510   GtkTreePath *path;
1511
1512   SortData data;
1513
1514   g_return_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model_sort));
1515   g_return_if_fail (level != NULL);
1516
1517   if (level->array->len < 1 && !((SortElt *)level->array->data)->children)
1518     return;
1519
1520   /* Set up data */
1521   data.tree_model_sort = tree_model_sort;
1522   if (level->parent_elt)
1523     {
1524       data.parent_path = gtk_tree_model_sort_elt_get_path (level->parent_level,
1525                                                            level->parent_elt);
1526       gtk_tree_path_append_index (data.parent_path, 0);
1527     }
1528   else
1529     {
1530       data.parent_path = gtk_tree_path_new_first ();
1531     }
1532   data.parent_path_depth = gtk_tree_path_get_depth (data.parent_path);
1533   data.parent_path_indices = gtk_tree_path_get_indices (data.parent_path);
1534
1535   /* make the array to be sorted */
1536   sort_array = g_array_sized_new (FALSE, FALSE, sizeof (SortTuple), level->array->len);
1537   for (i = 0; i < level->array->len; i++)
1538     {
1539       SortTuple tuple;
1540
1541       tuple.elt = &g_array_index (level->array, SortElt, i);
1542       tuple.offset = i;
1543
1544       g_array_append_val (sort_array, tuple);
1545     }
1546
1547     if (tree_model_sort->sort_column_id != GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID)
1548       {
1549         GtkTreeDataSortHeader *header = NULL;
1550
1551         header = _gtk_tree_data_list_get_header (tree_model_sort->sort_list,
1552                                                  tree_model_sort->sort_column_id);
1553
1554         g_return_if_fail (header != NULL);
1555         g_return_if_fail (header->func != NULL);
1556
1557         data.sort_func = header->func;
1558         data.sort_data = header->data;
1559       }
1560     else
1561       {
1562         /* absolutely SHOULD NOT happen: */
1563         g_return_if_fail (tree_model_sort->default_sort_func != NULL);
1564
1565         data.sort_func = tree_model_sort->default_sort_func;
1566         data.sort_data = tree_model_sort->default_sort_data;
1567       }
1568
1569   if (data.sort_func == NO_SORT_FUNC)
1570     g_array_sort_with_data (sort_array,
1571                             gtk_tree_model_sort_offset_compare_func,
1572                             &data);
1573   else
1574     g_array_sort_with_data (sort_array,
1575                             gtk_tree_model_sort_compare_func,
1576                             &data);
1577
1578   gtk_tree_path_free (data.parent_path);
1579
1580   new_array = g_array_sized_new (FALSE, FALSE, sizeof (SortElt), level->array->len);
1581   new_order = g_new (gint, level->array->len);
1582
1583   for (i = 0; i < level->array->len; i++)
1584     {
1585       SortElt *elt;
1586
1587       elt = g_array_index (sort_array, SortTuple, i).elt;
1588       new_order[i] = g_array_index (sort_array, SortTuple, i).offset;
1589
1590       g_array_append_val (new_array, *elt);
1591       elt = &g_array_index (new_array, SortElt, i);
1592       if (elt->children)
1593         elt->children->parent_elt = elt;
1594     }
1595
1596   g_array_free (level->array, TRUE);
1597   level->array = new_array;
1598   g_array_free (sort_array, TRUE);
1599
1600   if (emit_reordered)
1601     {
1602       gtk_tree_model_sort_increment_stamp (tree_model_sort);
1603       if (level->parent_elt)
1604         {
1605           iter.stamp = tree_model_sort->stamp;
1606           iter.user_data = level->parent_level;
1607           iter.user_data2 = level->parent_elt;
1608
1609           path = gtk_tree_model_get_path (GTK_TREE_MODEL (tree_model_sort),
1610                                           &iter);
1611
1612           gtk_tree_model_rows_reordered (GTK_TREE_MODEL (tree_model_sort), path,
1613                                          &iter, new_order);
1614         }
1615       else
1616         {
1617           /* toplevel list */
1618           path = gtk_tree_path_new ();
1619           gtk_tree_model_rows_reordered (GTK_TREE_MODEL (tree_model_sort), path,
1620                                          NULL, new_order);
1621         }
1622
1623       gtk_tree_path_free (path);
1624     }
1625
1626   /* recurse, if possible */
1627   if (recurse)
1628     {
1629       for (i = 0; i < level->array->len; i++)
1630         {
1631           SortElt *elt = &g_array_index (level->array, SortElt, i);
1632
1633           if (elt->children)
1634             gtk_tree_model_sort_sort_level (tree_model_sort,
1635                                             elt->children,
1636                                             TRUE, emit_reordered);
1637         }
1638     }
1639
1640   g_free (new_order);
1641 }
1642
1643 static void
1644 gtk_tree_model_sort_sort (GtkTreeModelSort *tree_model_sort)
1645 {
1646   g_return_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model_sort));
1647
1648   if (!tree_model_sort->root)
1649     return;
1650
1651   if (tree_model_sort->sort_column_id != GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID)
1652     {
1653       GtkTreeDataSortHeader *header = NULL;
1654
1655       header = _gtk_tree_data_list_get_header (tree_model_sort->sort_list,
1656                                                tree_model_sort->sort_column_id);
1657
1658       /* we want to make sure that we have a function */
1659       g_return_if_fail (header != NULL);
1660       g_return_if_fail (header->func != NULL);
1661     }
1662   else
1663     g_return_if_fail (tree_model_sort->default_sort_func != NULL);
1664
1665   gtk_tree_model_sort_sort_level (tree_model_sort, tree_model_sort->root,
1666                                   TRUE, TRUE);
1667 }
1668
1669 /* signal helpers */
1670 static gint
1671 gtk_tree_model_sort_level_find_insert (GtkTreeModelSort *tree_model_sort,
1672                                        SortLevel        *level,
1673                                        GtkTreeIter      *iter,
1674                                        gboolean          skip_sort_elt)
1675 {
1676   gint start, middle, end;
1677   gint skip_index;
1678   gint cmp;
1679   SortElt *tmp_elt;
1680   GtkTreeIter tmp_iter;
1681
1682   GtkTreeIterCompareFunc func;
1683   gpointer data;
1684
1685   if (tree_model_sort->sort_column_id != GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID)
1686     {
1687       GtkTreeDataSortHeader *header;
1688
1689       header = _gtk_tree_data_list_get_header (tree_model_sort->sort_list,
1690                                                tree_model_sort->sort_column_id);
1691
1692       g_return_val_if_fail (header != NULL, 0);
1693
1694       func = header->func;
1695       data = header->data;
1696     }
1697   else
1698     {
1699       func = tree_model_sort->default_sort_func;
1700       data = tree_model_sort->default_sort_data;
1701
1702       g_return_val_if_fail (func != NO_SORT_FUNC, 0);
1703     }
1704
1705   g_return_val_if_fail (func != NULL, 0);
1706
1707   start = 0;
1708   end = level->array->len;
1709   if (skip_sort_elt)
1710     {
1711       skip_index = SORT_ELT (iter->user_data2) - SORT_ELT (level->array->data);
1712       end--;
1713     }
1714   else
1715     skip_index = end;
1716
1717   if (start == end)
1718     return 0;
1719
1720   while (start != end)
1721     {
1722       middle = (start + end) / 2;
1723
1724       if (middle < skip_index)
1725         tmp_elt = &(g_array_index (level->array, SortElt, middle));
1726       else
1727         tmp_elt = &(g_array_index (level->array, SortElt, middle + 1));
1728
1729       if (!GTK_TREE_MODEL_SORT_CACHE_CHILD_ITERS (tree_model_sort))
1730         {
1731           GtkTreePath *path = gtk_tree_model_sort_elt_get_path (level, tmp_elt);
1732           gtk_tree_model_get_iter (tree_model_sort->child_model,
1733                                    &tmp_iter, path);
1734           gtk_tree_path_free (path);
1735         }
1736       else
1737         tmp_iter = tmp_elt->iter;
1738
1739       if (tree_model_sort->order == GTK_SORT_ASCENDING)
1740         cmp = (* func) (GTK_TREE_MODEL (tree_model_sort->child_model),
1741                         &tmp_iter, iter, data);
1742       else
1743         cmp = (* func) (GTK_TREE_MODEL (tree_model_sort->child_model),
1744                         iter, &tmp_iter, data);
1745
1746       if (cmp <= 0)
1747         start = middle + 1;
1748       else
1749         end = middle;
1750     }
1751
1752   if (cmp <= 0)
1753     return middle + 1;
1754   else
1755     return middle;
1756 }
1757
1758 static gboolean
1759 gtk_tree_model_sort_insert_value (GtkTreeModelSort *tree_model_sort,
1760                                   SortLevel        *level,
1761                                   GtkTreePath      *s_path,
1762                                   GtkTreeIter      *s_iter)
1763 {
1764   gint offset, index, i;
1765
1766   SortElt elt;
1767   SortElt *tmp_elt;
1768
1769   offset = gtk_tree_path_get_indices (s_path)[gtk_tree_path_get_depth (s_path) - 1];
1770
1771   if (GTK_TREE_MODEL_SORT_CACHE_CHILD_ITERS (tree_model_sort))
1772     elt.iter = *s_iter;
1773   elt.offset = offset;
1774   elt.zero_ref_count = 0;
1775   elt.ref_count = 0;
1776   elt.children = NULL;
1777
1778   /* update all larger offsets */
1779   tmp_elt = SORT_ELT (level->array->data);
1780   for (i = 0; i < level->array->len; i++, tmp_elt++)
1781     if (tmp_elt->offset >= offset)
1782       tmp_elt->offset++;
1783
1784   if (tree_model_sort->sort_column_id == GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID &&
1785       tree_model_sort->default_sort_func == NO_SORT_FUNC)
1786     index = offset;
1787   else
1788     index = gtk_tree_model_sort_level_find_insert (tree_model_sort,
1789                                                    level, s_iter,
1790                                                    FALSE);
1791
1792   g_array_insert_vals (level->array, index, &elt, 1);
1793   tmp_elt = SORT_ELT (level->array->data);
1794   for (i = 0; i < level->array->len; i++, tmp_elt++)
1795     if (tmp_elt->children)
1796       tmp_elt->children->parent_elt = tmp_elt;
1797
1798   return TRUE;
1799 }
1800
1801 /* sort elt stuff */
1802 static GtkTreePath *
1803 gtk_tree_model_sort_elt_get_path (SortLevel *level,
1804                                   SortElt *elt)
1805 {
1806   SortLevel *walker = level;
1807   SortElt *walker2 = elt;
1808   GtkTreePath *path;
1809
1810   g_return_val_if_fail (level != NULL, NULL);
1811   g_return_val_if_fail (elt != NULL, NULL);
1812
1813   path = gtk_tree_path_new ();
1814
1815   while (walker)
1816     {
1817       gtk_tree_path_prepend_index (path, walker2->offset);
1818
1819       walker2 = walker->parent_elt;
1820       walker = walker->parent_level;
1821     }
1822
1823   return path;
1824 }
1825
1826 /**
1827  * gtk_tree_model_sort_set_model:
1828  * @tree_model_sort: The #GtkTreeModelSort.
1829  * @child_model: A #GtkTreeModel, or NULL.
1830  *
1831  * Sets the model of @tree_model_sort to be @model.  If @model is NULL, then the
1832  * old model is unset.  The sort function is unset as a result of this call.
1833  * The model will be in an unsorted state until a sort function is set.
1834  **/
1835 static void
1836 gtk_tree_model_sort_set_model (GtkTreeModelSort *tree_model_sort,
1837                                GtkTreeModel     *child_model)
1838 {
1839   g_return_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model_sort));
1840
1841   if (child_model)
1842     g_object_ref (child_model);
1843
1844   if (tree_model_sort->child_model)
1845     {
1846       g_signal_handler_disconnect (tree_model_sort->child_model,
1847                                    tree_model_sort->changed_id);
1848       g_signal_handler_disconnect (tree_model_sort->child_model,
1849                                    tree_model_sort->inserted_id);
1850       g_signal_handler_disconnect (tree_model_sort->child_model,
1851                                    tree_model_sort->has_child_toggled_id);
1852       g_signal_handler_disconnect (tree_model_sort->child_model,
1853                                    tree_model_sort->deleted_id);
1854       g_signal_handler_disconnect (tree_model_sort->child_model,
1855                                    tree_model_sort->reordered_id);
1856
1857       /* reset our state */
1858       if (tree_model_sort->root)
1859         gtk_tree_model_sort_free_level (tree_model_sort, tree_model_sort->root);
1860       tree_model_sort->root = NULL;
1861       _gtk_tree_data_list_header_free (tree_model_sort->sort_list);
1862       tree_model_sort->sort_list = NULL;
1863       g_object_unref (tree_model_sort->child_model);
1864     }
1865
1866   tree_model_sort->child_model = child_model;
1867
1868   if (child_model)
1869     {
1870       GType *types;
1871       gint i, n_columns;
1872
1873       tree_model_sort->changed_id =
1874         g_signal_connect (child_model, "row_changed",
1875                           G_CALLBACK (gtk_tree_model_sort_row_changed),
1876                           tree_model_sort);
1877       tree_model_sort->inserted_id =
1878         g_signal_connect (child_model, "row_inserted",
1879                           G_CALLBACK (gtk_tree_model_sort_row_inserted),
1880                           tree_model_sort);
1881       tree_model_sort->has_child_toggled_id =
1882         g_signal_connect (child_model, "row_has_child_toggled",
1883                           G_CALLBACK (gtk_tree_model_sort_row_has_child_toggled),
1884                           tree_model_sort);
1885       tree_model_sort->deleted_id =
1886         g_signal_connect (child_model, "row_deleted",
1887                           G_CALLBACK (gtk_tree_model_sort_row_deleted),
1888                           tree_model_sort);
1889       tree_model_sort->reordered_id =
1890         g_signal_connect (child_model, "rows_reordered",
1891                           G_CALLBACK (gtk_tree_model_sort_rows_reordered),
1892                           tree_model_sort);
1893
1894       tree_model_sort->child_flags = gtk_tree_model_get_flags (child_model);
1895       n_columns = gtk_tree_model_get_n_columns (child_model);
1896
1897       types = g_new (GType, n_columns);
1898       for (i = 0; i < n_columns; i++)
1899         types[i] = gtk_tree_model_get_column_type (child_model, i);
1900
1901       tree_model_sort->sort_list = _gtk_tree_data_list_header_new (n_columns, types);
1902       g_free (types);
1903
1904       tree_model_sort->default_sort_func = NO_SORT_FUNC;
1905       tree_model_sort->stamp = g_random_int ();
1906     }
1907 }
1908
1909 /**
1910  * gtk_tree_model_sort_get_model:
1911  * @tree_model: a #GtkTreeModelSort
1912  *
1913  * Returns the model the #GtkTreeModelSort is sorting.
1914  *
1915  * Return value: the "child model" being sorted
1916  **/
1917 GtkTreeModel *
1918 gtk_tree_model_sort_get_model (GtkTreeModelSort  *tree_model)
1919 {
1920   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model), NULL);
1921
1922   return tree_model->child_model;
1923 }
1924
1925
1926 static GtkTreePath *
1927 gtk_real_tree_model_sort_convert_child_path_to_path (GtkTreeModelSort *tree_model_sort,
1928                                                      GtkTreePath      *child_path,
1929                                                      gboolean          build_levels)
1930 {
1931   gint *child_indices;
1932   GtkTreePath *retval;
1933   SortLevel *level;
1934   gint i;
1935
1936   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model_sort), NULL);
1937   g_return_val_if_fail (tree_model_sort->child_model != NULL, NULL);
1938   g_return_val_if_fail (child_path != NULL, NULL);
1939
1940   retval = gtk_tree_path_new ();
1941   child_indices = gtk_tree_path_get_indices (child_path);
1942
1943   if (tree_model_sort->root == NULL && build_levels)
1944     gtk_tree_model_sort_build_level (tree_model_sort, NULL, NULL);
1945   level = SORT_LEVEL (tree_model_sort->root);
1946
1947   for (i = 0; i < gtk_tree_path_get_depth (child_path); i++)
1948     {
1949       gint j;
1950       gboolean found_child = FALSE;
1951
1952       if (!level)
1953         {
1954           gtk_tree_path_free (retval);
1955           return NULL;
1956         }
1957
1958       if (child_indices[i] >= level->array->len)
1959         {
1960           gtk_tree_path_free (retval);
1961           return NULL;
1962         }
1963       for (j = 0; j < level->array->len; j++)
1964         {
1965           if ((g_array_index (level->array, SortElt, j)).offset == child_indices[i])
1966             {
1967               gtk_tree_path_append_index (retval, j);
1968               if (g_array_index (level->array, SortElt, j).children == NULL && build_levels)
1969                 {
1970                   gtk_tree_model_sort_build_level (tree_model_sort, level, &g_array_index (level->array, SortElt, j));
1971                 }
1972               level = g_array_index (level->array, SortElt, j).children;
1973               found_child = TRUE;
1974               break;
1975             }
1976         }
1977       if (! found_child)
1978         {
1979           gtk_tree_path_free (retval);
1980           return NULL;
1981         }
1982     }
1983
1984   return retval;
1985 }
1986
1987
1988 /**
1989  * gtk_tree_model_sort_convert_child_path_to_path:
1990  * @tree_model_sort: A #GtkTreeModelSort
1991  * @child_path: A #GtkTreePath to convert
1992  *
1993  * Converts @child_path to a path relative to @tree_model_sort.  That is,
1994  * @child_path points to a path in the child model.  The returned path will
1995  * point to the same row in the sorted model.  If @child_path isn't a valid path
1996  * on the child model, then %NULL is returned.
1997  *
1998  * Return value: A newly allocated #GtkTreePath, or %NULL
1999  **/
2000 GtkTreePath *
2001 gtk_tree_model_sort_convert_child_path_to_path (GtkTreeModelSort *tree_model_sort,
2002                                                 GtkTreePath      *child_path)
2003 {
2004   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model_sort), NULL);
2005   g_return_val_if_fail (tree_model_sort->child_model != NULL, NULL);
2006   g_return_val_if_fail (child_path != NULL, NULL);
2007
2008   return gtk_real_tree_model_sort_convert_child_path_to_path (tree_model_sort, child_path, TRUE);
2009 }
2010
2011 /**
2012  * gtk_tree_model_sort_convert_child_iter_to_iter:
2013  * @tree_model_sort: A #GtkTreeModelSort
2014  * @sort_iter: An uninitialized #GtkTreeIter.
2015  * @child_iter: A valid #GtkTreeIter pointing to a row on the child model
2016  *
2017  * Sets @sort_iter to point to the row in @tree_model_sort that corresponds to
2018  * the row pointed at by @child_iter.
2019  **/
2020 void
2021 gtk_tree_model_sort_convert_child_iter_to_iter (GtkTreeModelSort *tree_model_sort,
2022                                                 GtkTreeIter      *sort_iter,
2023                                                 GtkTreeIter      *child_iter)
2024 {
2025   GtkTreePath *child_path, *path;
2026
2027   g_return_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model_sort));
2028   g_return_if_fail (tree_model_sort->child_model != NULL);
2029   g_return_if_fail (sort_iter != NULL);
2030   g_return_if_fail (child_iter != NULL);
2031
2032   sort_iter->stamp = 0;
2033
2034   child_path = gtk_tree_model_get_path (tree_model_sort->child_model, child_iter);
2035   g_return_if_fail (child_path != NULL);
2036
2037   path = gtk_tree_model_sort_convert_child_path_to_path (tree_model_sort, child_path);
2038   gtk_tree_path_free (child_path);
2039   g_return_if_fail (path != NULL);
2040
2041   gtk_tree_model_get_iter (GTK_TREE_MODEL (tree_model_sort), sort_iter, path);
2042   gtk_tree_path_free (path);
2043 }
2044
2045 /**
2046  * gtk_tree_model_sort_convert_path_to_child_path:
2047  * @tree_model_sort: A #GtkTreeModelSort
2048  * @sorted_path: A #GtkTreePath to convert
2049  *
2050  * Converts @sorted_path to a path on the child model of @tree_model_sort.  That
2051  * is, @sorted_path points to a location in @tree_model_sort.  The returned path
2052  * will point to the same location in the model not being sorted.  If @sorted_path
2053  * does not point to a location in the child model, %NULL is returned.
2054  *
2055  * Return value: A newly allocated #GtkTreePath, or %NULL
2056  **/
2057 GtkTreePath *
2058 gtk_tree_model_sort_convert_path_to_child_path (GtkTreeModelSort *tree_model_sort,
2059                                                 GtkTreePath      *sorted_path)
2060 {
2061   gint *sorted_indices;
2062   GtkTreePath *retval;
2063   SortLevel *level;
2064   gint i;
2065
2066   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model_sort), NULL);
2067   g_return_val_if_fail (tree_model_sort->child_model != NULL, NULL);
2068   g_return_val_if_fail (sorted_path != NULL, NULL);
2069
2070   retval = gtk_tree_path_new ();
2071   sorted_indices = gtk_tree_path_get_indices (sorted_path);
2072   if (tree_model_sort->root == NULL)
2073     gtk_tree_model_sort_build_level (tree_model_sort, NULL, NULL);
2074   level = SORT_LEVEL (tree_model_sort->root);
2075
2076   for (i = 0; i < gtk_tree_path_get_depth (sorted_path); i++)
2077     {
2078       gint count = sorted_indices[i];
2079
2080       if ((level == NULL) ||
2081           (level->array->len <= count))
2082         {
2083           gtk_tree_path_free (retval);
2084           return NULL;
2085         }
2086
2087       if (g_array_index (level->array, SortElt, count).children == NULL)
2088         gtk_tree_model_sort_build_level (tree_model_sort, level, &g_array_index (level->array, SortElt, count));
2089
2090       if (level == NULL)
2091         {
2092           gtk_tree_path_free (retval);
2093           break;
2094         }
2095
2096       gtk_tree_path_append_index (retval, g_array_index (level->array, SortElt, count).offset);
2097       level = g_array_index (level->array, SortElt, count).children;
2098     }
2099
2100   return retval;
2101 }
2102
2103 /**
2104  * gtk_tree_model_sort_convert_iter_to_child_iter:
2105  * @tree_model_sort: A #GtkTreeModelSort
2106  * @child_iter: An uninitialized #GtkTreeIter
2107  * @sorted_iter: A valid #GtkTreeIter pointing to a row on @tree_model_sort.
2108  *
2109  * Sets @child_iter to point to the row pointed to by @sorted_iter.
2110  **/
2111 void
2112 gtk_tree_model_sort_convert_iter_to_child_iter (GtkTreeModelSort *tree_model_sort,
2113                                                 GtkTreeIter      *child_iter,
2114                                                 GtkTreeIter      *sorted_iter)
2115 {
2116   g_return_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model_sort));
2117   g_return_if_fail (tree_model_sort->child_model != NULL);
2118   g_return_if_fail (child_iter != NULL);
2119   g_return_if_fail (sorted_iter != NULL);
2120   g_return_if_fail (sorted_iter->stamp == tree_model_sort->stamp);
2121
2122   if (GTK_TREE_MODEL_SORT_CACHE_CHILD_ITERS (tree_model_sort))
2123     {
2124       *child_iter = SORT_ELT (sorted_iter->user_data2)->iter;
2125     }
2126   else
2127     {
2128       GtkTreePath *path;
2129
2130       path = gtk_tree_model_sort_elt_get_path (sorted_iter->user_data,
2131                                                sorted_iter->user_data2);
2132       gtk_tree_model_get_iter (tree_model_sort->child_model, child_iter, path);
2133       gtk_tree_path_free (path);
2134     }
2135 }
2136
2137 static void
2138 gtk_tree_model_sort_build_level (GtkTreeModelSort *tree_model_sort,
2139                                  SortLevel        *parent_level,
2140                                  SortElt          *parent_elt)
2141 {
2142   GtkTreeIter iter;
2143   SortLevel *new_level;
2144   gint length = 0;
2145   gint i;
2146
2147   g_assert (tree_model_sort->child_model != NULL);
2148
2149   if (parent_level == NULL)
2150     {
2151       if (gtk_tree_model_get_iter_first (tree_model_sort->child_model, &iter) == FALSE)
2152         return;
2153       length = gtk_tree_model_iter_n_children (tree_model_sort->child_model, NULL);
2154     }
2155   else
2156     {
2157       GtkTreeIter parent_iter;
2158       GtkTreeIter child_parent_iter;
2159
2160       parent_iter.stamp = tree_model_sort->stamp;
2161       parent_iter.user_data = parent_level;
2162       parent_iter.user_data2 = parent_elt;
2163
2164       gtk_tree_model_sort_convert_iter_to_child_iter (tree_model_sort,
2165                                                       &child_parent_iter,
2166                                                       &parent_iter);
2167       if (gtk_tree_model_iter_children (tree_model_sort->child_model,
2168                                         &iter,
2169                                         &child_parent_iter) == FALSE)
2170         return;
2171
2172       /* stamp may have changed */
2173       gtk_tree_model_sort_convert_iter_to_child_iter (tree_model_sort,
2174                                                       &child_parent_iter,
2175                                                       &parent_iter);
2176
2177       length = gtk_tree_model_iter_n_children (tree_model_sort->child_model, &child_parent_iter);
2178     }
2179
2180   g_return_if_fail (length > 0);
2181
2182   new_level = g_new (SortLevel, 1);
2183   new_level->array = g_array_sized_new (FALSE, FALSE, sizeof (SortElt), length);
2184   new_level->ref_count = 0;
2185   new_level->parent_elt = parent_elt;
2186   new_level->parent_level = parent_level;
2187
2188   if (parent_elt)
2189     parent_elt->children = new_level;
2190   else
2191     tree_model_sort->root = new_level;
2192
2193   /* increase the count of zero ref_counts.*/
2194   while (parent_level)
2195     {
2196       parent_elt->zero_ref_count++;
2197
2198       parent_elt = parent_level->parent_elt;
2199       parent_level = parent_level->parent_level;
2200     }
2201   if (new_level != tree_model_sort->root)
2202     tree_model_sort->zero_ref_count++;
2203
2204   for (i = 0; i < length; i++)
2205     {
2206       SortElt sort_elt;
2207       sort_elt.offset = i;
2208       sort_elt.zero_ref_count = 0;
2209       sort_elt.ref_count = 0;
2210       sort_elt.children = NULL;
2211
2212       if (GTK_TREE_MODEL_SORT_CACHE_CHILD_ITERS (tree_model_sort))
2213         {
2214           sort_elt.iter = iter;
2215           if (gtk_tree_model_iter_next (tree_model_sort->child_model, &iter) == FALSE &&
2216               i < length - 1)
2217             {
2218               g_warning ("There is a discrepency between the sort model and the child model.");
2219               return;
2220             }
2221         }
2222       g_array_append_val (new_level->array, sort_elt);
2223     }
2224
2225   /* sort level */
2226   gtk_tree_model_sort_sort_level (tree_model_sort, new_level,
2227                                   FALSE, FALSE);
2228 }
2229
2230 static void
2231 gtk_tree_model_sort_free_level (GtkTreeModelSort *tree_model_sort,
2232                                 SortLevel        *sort_level)
2233 {
2234   gint i;
2235
2236   g_assert (sort_level);
2237
2238   if (sort_level->ref_count == 0)
2239     {
2240       SortLevel *parent_level = sort_level->parent_level;
2241       SortElt *parent_elt = sort_level->parent_elt;
2242
2243       do
2244         {
2245           if (parent_elt)
2246             parent_elt->zero_ref_count--;
2247           else
2248             tree_model_sort->zero_ref_count--;
2249
2250           if (parent_level)
2251             {
2252               parent_elt = parent_level->parent_elt;
2253               parent_level = parent_level->parent_level;
2254             }
2255         }
2256       while (parent_level);
2257     }
2258
2259   for (i = 0; i < sort_level->array->len; i++)
2260     {
2261       if (g_array_index (sort_level->array, SortElt, i).children)
2262         gtk_tree_model_sort_free_level (tree_model_sort,
2263                                         SORT_LEVEL(g_array_index (sort_level->array, SortElt, i).children));
2264     }
2265
2266   if (sort_level->parent_elt)
2267     sort_level->parent_elt->children = NULL;
2268   else
2269     tree_model_sort->root = NULL;
2270
2271   g_array_free (sort_level->array, TRUE);
2272   sort_level->array = NULL;
2273
2274   g_free (sort_level);
2275   sort_level = NULL;
2276 }
2277
2278 static void
2279 gtk_tree_model_sort_increment_stamp (GtkTreeModelSort *tree_model_sort)
2280 {
2281   do
2282     {
2283       tree_model_sort->stamp++;
2284     }
2285   while (tree_model_sort->stamp == 0);
2286
2287   gtk_tree_model_sort_clear_cache (tree_model_sort);
2288 }
2289
2290 static void
2291 gtk_tree_model_sort_clear_cache_helper (GtkTreeModelSort *tree_model_sort,
2292                                         SortLevel        *level)
2293 {
2294   gint i;
2295
2296   g_assert (level != NULL);
2297
2298   for (i = 0; i < level->array->len; i++)
2299     {
2300       if (g_array_index (level->array, SortElt, i).zero_ref_count > 0)
2301         gtk_tree_model_sort_clear_cache_helper (tree_model_sort, g_array_index (level->array, SortElt, i).children);
2302     }
2303
2304   if (level->ref_count == 0 && level != tree_model_sort->root)
2305     gtk_tree_model_sort_free_level (tree_model_sort, level);
2306 }
2307
2308 /**
2309  * gtk_tree_model_sort_reset_default_sort_func:
2310  * @tree_model_sort: A #GtkTreeModelSort
2311  *
2312  * This resets the default sort function to be in the 'unsorted' state.  That
2313  * is, it is in the same order as the child model. It will re-sort the model
2314  * to be in the same order as the child model only if the #GtkTreeModelSort
2315  * is in 'unsorted' state.
2316  **/
2317 void
2318 gtk_tree_model_sort_reset_default_sort_func (GtkTreeModelSort *tree_model_sort)
2319 {
2320   g_return_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model_sort));
2321
2322   if (tree_model_sort->default_sort_destroy)
2323     {
2324       GtkDestroyNotify d = tree_model_sort->default_sort_destroy;
2325
2326       tree_model_sort->default_sort_destroy = NULL;
2327       d (tree_model_sort->default_sort_data);
2328     }
2329
2330   tree_model_sort->default_sort_func = NO_SORT_FUNC;
2331   tree_model_sort->default_sort_data = NULL;
2332   tree_model_sort->default_sort_destroy = NULL;
2333
2334   if (tree_model_sort->sort_column_id == GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID)
2335     gtk_tree_model_sort_sort (tree_model_sort);
2336   tree_model_sort->sort_column_id = GTK_TREE_SORTABLE_DEFAULT_SORT_COLUMN_ID;
2337 }
2338
2339 /**
2340  * gtk_tree_model_sort_clear_cache:
2341  * @tree_model_sort: A #GtkTreeModelSort
2342  *
2343  * This function should almost never be called.  It clears the @tree_model_sort
2344  * of any cached iterators that haven't been reffed with
2345  * gtk_tree_model_ref_node().  This might be useful if the child model being
2346  * sorted is static (and doesn't change often) and there has been a lot of
2347  * unreffed access to nodes.  As a side effect of this function, all unreffed
2348  * iters will be invalid.
2349  **/
2350 void
2351 gtk_tree_model_sort_clear_cache (GtkTreeModelSort *tree_model_sort)
2352 {
2353   g_return_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model_sort));
2354
2355   if (tree_model_sort->zero_ref_count)
2356     gtk_tree_model_sort_clear_cache_helper (tree_model_sort, (SortLevel *)tree_model_sort->root);
2357 }
2358
2359 static gboolean
2360 gtk_tree_model_sort_iter_is_valid_helper (GtkTreeIter *iter,
2361                                           SortLevel   *level)
2362 {
2363   gint i;
2364
2365   for (i = 0; i < level->array->len; i++)
2366     {
2367       SortElt *elt = &g_array_index (level->array, SortElt, i);
2368
2369       if (iter->user_data == level && iter->user_data2 == elt)
2370         return TRUE;
2371
2372       if (elt->children)
2373         if (gtk_tree_model_sort_iter_is_valid_helper (iter, elt->children))
2374           return TRUE;
2375     }
2376
2377   return FALSE;
2378 }
2379
2380 /**
2381  * gtk_tree_model_sort_iter_is_valid:
2382  * @tree_model_sort: A #GtkTreeModelSort.
2383  * @iter: A #GtkTreeIter.
2384  *
2385  * WARNING: This function is slow. Only use it for debugging and/or testing
2386  * purposes.
2387  *
2388  * Checks if the given iter is a valid iter for this #GtkTreeModelSort.
2389  *
2390  * Return value: %TRUE if the iter is valid, %FALSE if the iter is invalid.
2391  *
2392  * Since: 2.2
2393  **/
2394 gboolean
2395 gtk_tree_model_sort_iter_is_valid (GtkTreeModelSort *tree_model_sort,
2396                                    GtkTreeIter      *iter)
2397 {
2398   g_return_val_if_fail (GTK_IS_TREE_MODEL_SORT (tree_model_sort), FALSE);
2399   g_return_val_if_fail (iter != NULL, FALSE);
2400
2401   if (!VALID_ITER (iter, tree_model_sort))
2402     return FALSE;
2403
2404   return gtk_tree_model_sort_iter_is_valid_helper (iter,
2405                                                    tree_model_sort->root);
2406 }