aboutsummaryrefslogtreecommitdiffstats
path: root/e-util/e-tree-sorted.c
diff options
context:
space:
mode:
Diffstat (limited to 'e-util/e-tree-sorted.c')
-rw-r--r--e-util/e-tree-sorted.c1433
1 files changed, 1433 insertions, 0 deletions
diff --git a/e-util/e-tree-sorted.c b/e-util/e-tree-sorted.c
new file mode 100644
index 0000000000..25cfceb337
--- /dev/null
+++ b/e-util/e-tree-sorted.c
@@ -0,0 +1,1433 @@
+/*
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2 of the License, or (at your option) version 3.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with the program; if not, see <http://www.gnu.org/licenses/>
+ *
+ *
+ * Authors:
+ * Chris Lahey <clahey@ximian.com>
+ * Chris Toshok <toshok@ximian.com>
+ *
+ * Adapted from the gtree code and ETableModel.
+ *
+ * Copyright (C) 1999-2008 Novell, Inc. (www.novell.com)
+ *
+ */
+
+/* FIXME: Overall e-tree-sorted.c needs to be made more efficient. */
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include "e-tree-sorted.h"
+
+#include <stdio.h>
+#include <errno.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <string.h>
+
+#include <libxml/parser.h>
+#include <libxml/xmlmemory.h>
+
+#include "e-table-sorting-utils.h"
+#include "e-xml-utils.h"
+
+/* maximum insertions between an idle event that we will do without scheduling an idle sort */
+#define ETS_INSERT_MAX (4)
+
+#define d(x)
+
+#define E_TREE_SORTED_GET_PRIVATE(obj) \
+ (G_TYPE_INSTANCE_GET_PRIVATE \
+ ((obj), E_TYPE_TREE_SORTED, ETreeSortedPrivate))
+
+G_DEFINE_TYPE (ETreeSorted, e_tree_sorted, E_TYPE_TREE_MODEL)
+
+enum {
+ NODE_RESORTED,
+ LAST_SIGNAL
+};
+
+static guint signals[LAST_SIGNAL] = {0, };
+
+typedef struct ETreeSortedPath ETreeSortedPath;
+
+struct ETreeSortedPath {
+ ETreePath corresponding;
+
+ /* parent/child/sibling pointers */
+ ETreeSortedPath *parent;
+ gint num_children;
+ ETreeSortedPath **children;
+ gint position;
+ gint orig_position;
+
+ guint needs_resort : 1;
+ guint child_needs_resort : 1;
+ guint resort_all_children : 1;
+ guint needs_regen_to_sort : 1;
+};
+
+struct _ETreeSortedPrivate {
+ ETreeModel *source;
+ ETreeSortedPath *root;
+
+ ETableSortInfo *sort_info;
+ ETableHeader *full_header;
+
+ ETreeSortedPath *last_access;
+
+ gint tree_model_pre_change_id;
+ gint tree_model_no_change_id;
+ gint tree_model_node_changed_id;
+ gint tree_model_node_data_changed_id;
+ gint tree_model_node_col_changed_id;
+ gint tree_model_node_inserted_id;
+ gint tree_model_node_removed_id;
+ gint tree_model_node_deleted_id;
+ gint tree_model_node_request_collapse_id;
+
+ gint sort_info_changed_id;
+ gint sort_idle_id;
+ gint insert_idle_id;
+ gint insert_count;
+
+ guint in_resort_idle : 1;
+ guint nested_resort_idle : 1;
+};
+
+enum {
+ ARG_0,
+
+ ARG_SORT_INFO
+};
+
+static void ets_sort_info_changed (ETableSortInfo *sort_info, ETreeSorted *ets);
+static void resort_node (ETreeSorted *ets, ETreeSortedPath *path, gboolean resort_all_children, gboolean needs_regen, gboolean send_signals);
+static void mark_path_needs_resort (ETreeSorted *ets, ETreeSortedPath *path, gboolean needs_rebuild, gboolean resort_all_children);
+static void schedule_resort (ETreeSorted *ets, ETreeSortedPath *path, gboolean needs_regen, gboolean resort_all_children);
+static void free_path (ETreeSortedPath *path);
+static void generate_children (ETreeSorted *ets, ETreeSortedPath *path);
+static void regenerate_children (ETreeSorted *ets, ETreeSortedPath *path);
+
+/* idle callbacks */
+
+static gboolean
+ets_sort_idle (gpointer user_data)
+{
+ ETreeSorted *ets = user_data;
+ if (ets->priv->in_resort_idle) {
+ ets->priv->nested_resort_idle = TRUE;
+ return FALSE;
+ }
+ ets->priv->in_resort_idle = TRUE;
+ if (ets->priv->root) {
+ do {
+ ets->priv->nested_resort_idle = FALSE;
+ resort_node (ets, ets->priv->root, FALSE, FALSE, TRUE);
+ } while (ets->priv->nested_resort_idle);
+ }
+ ets->priv->in_resort_idle = FALSE;
+ ets->priv->sort_idle_id = 0;
+ return FALSE;
+}
+
+#define ETS_SORT_IDLE_ACTIVATED(ets) ((ets)->priv->sort_idle_id != 0)
+
+inline static void
+ets_stop_sort_idle (ETreeSorted *ets)
+{
+ if (ets->priv->sort_idle_id) {
+ g_source_remove (ets->priv->sort_idle_id);
+ ets->priv->sort_idle_id = 0;
+ }
+}
+
+static gboolean
+ets_insert_idle (ETreeSorted *ets)
+{
+ ets->priv->insert_count = 0;
+ ets->priv->insert_idle_id = 0;
+ return FALSE;
+}
+
+/* Helper functions */
+
+#define CHECK_AROUND_LAST_ACCESS
+
+static inline ETreeSortedPath *
+check_last_access (ETreeSorted *ets,
+ ETreePath corresponding)
+{
+#ifdef CHECK_AROUND_LAST_ACCESS
+ ETreeSortedPath *parent;
+#endif
+
+ if (ets->priv->last_access == NULL)
+ return NULL;
+
+ if (ets->priv->last_access == corresponding) {
+ d (g_print ("Found last access %p at %p.", ets->priv->last_access, ets->priv->last_access));
+ return ets->priv->last_access;
+ }
+
+#ifdef CHECK_AROUND_LAST_ACCESS
+ parent = ets->priv->last_access->parent;
+ if (parent && parent->children) {
+ gint position = ets->priv->last_access->position;
+ gint end = MIN (parent->num_children, position + 10);
+ gint start = MAX (0, position - 10);
+ gint initial = MAX (MIN (position, end), start);
+ gint i;
+
+ for (i = initial; i < end; i++) {
+ if (parent->children[i] && parent->children[i]->corresponding == corresponding) {
+ d (g_print ("Found last access %p at %p.", ets->priv->last_access, parent->children[i]));
+ return parent->children[i];
+ }
+ }
+
+ for (i = initial - 1; i >= start; i--) {
+ if (parent->children[i] && parent->children[i]->corresponding == corresponding) {
+ d (g_print ("Found last access %p at %p.", ets->priv->last_access, parent->children[i]));
+ return parent->children[i];
+ }
+ }
+ }
+#endif
+ return NULL;
+}
+
+static ETreeSortedPath *
+find_path (ETreeSorted *ets,
+ ETreePath corresponding)
+{
+ gint depth;
+ ETreePath *sequence;
+ gint i;
+ ETreeSortedPath *path;
+ ETreeSortedPath *check_last;
+
+ if (corresponding == NULL)
+ return NULL;
+
+ check_last = check_last_access (ets, corresponding);
+ if (check_last) {
+ d (g_print (" (find_path)\n"));
+ return check_last;
+ }
+
+ depth = e_tree_model_node_depth (ets->priv->source, corresponding);
+
+ sequence = g_new (ETreePath, depth + 1);
+
+ sequence[0] = corresponding;
+
+ for (i = 0; i < depth; i++)
+ sequence[i + 1] = e_tree_model_node_get_parent (ets->priv->source, sequence[i]);
+
+ path = ets->priv->root;
+
+ for (i = depth - 1; i >= 0 && path != NULL; i--) {
+ gint j;
+
+ if (path->num_children == -1) {
+ path = NULL;
+ break;
+ }
+
+ for (j = 0; j < path->num_children; j++) {
+ if (path->children[j]->corresponding == sequence[i]) {
+ break;
+ }
+ }
+
+ if (j < path->num_children) {
+ path = path->children[j];
+ } else {
+ path = NULL;
+ }
+ }
+ g_free (sequence);
+
+ d (g_print ("Didn't find last access %p. Setting to %p. (find_path)\n", ets->priv->last_access, path));
+ ets->priv->last_access = path;
+
+ return path;
+}
+
+static ETreeSortedPath *
+find_child_path (ETreeSorted *ets,
+ ETreeSortedPath *parent,
+ ETreePath corresponding)
+{
+ gint i;
+
+ if (corresponding == NULL)
+ return NULL;
+
+ if (parent->num_children == -1) {
+ return NULL;
+ }
+
+ for (i = 0; i < parent->num_children; i++)
+ if (parent->children[i]->corresponding == corresponding)
+ return parent->children[i];
+
+ return NULL;
+}
+
+static ETreeSortedPath *
+find_or_create_path (ETreeSorted *ets,
+ ETreePath corresponding)
+{
+ gint depth;
+ ETreePath *sequence;
+ gint i;
+ ETreeSortedPath *path;
+ ETreeSortedPath *check_last;
+
+ if (corresponding == NULL)
+ return NULL;
+
+ check_last = check_last_access (ets, corresponding);
+ if (check_last) {
+ d (g_print (" (find_or_create_path)\n"));
+ return check_last;
+ }
+
+ depth = e_tree_model_node_depth (ets->priv->source, corresponding);
+
+ sequence = g_new (ETreePath, depth + 1);
+
+ sequence[0] = corresponding;
+
+ for (i = 0; i < depth; i++)
+ sequence[i + 1] = e_tree_model_node_get_parent (ets->priv->source, sequence[i]);
+
+ path = ets->priv->root;
+
+ for (i = depth - 1; i >= 0 && path != NULL; i--) {
+ gint j;
+
+ if (path->num_children == -1) {
+ generate_children (ets, path);
+ }
+
+ for (j = 0; j < path->num_children; j++) {
+ if (path->children[j]->corresponding == sequence[i]) {
+ break;
+ }
+ }
+
+ if (j < path->num_children) {
+ path = path->children[j];
+ } else {
+ path = NULL;
+ }
+ }
+ g_free (sequence);
+
+ d (g_print ("Didn't find last access %p. Setting to %p. (find_or_create_path)\n", ets->priv->last_access, path));
+ ets->priv->last_access = path;
+
+ return path;
+}
+
+static void
+free_children (ETreeSortedPath *path)
+{
+ gint i;
+
+ if (path == NULL)
+ return;
+
+ for (i = 0; i < path->num_children; i++) {
+ free_path (path->children[i]);
+ }
+
+ g_free (path->children);
+ path->children = NULL;
+ path->num_children = -1;
+}
+
+static void
+free_path (ETreeSortedPath *path)
+{
+ free_children (path);
+ g_slice_free (ETreeSortedPath, path);
+}
+
+static ETreeSortedPath *
+new_path (ETreeSortedPath *parent,
+ ETreePath corresponding)
+{
+ ETreeSortedPath *path;
+
+ path = g_slice_new0 (ETreeSortedPath);
+
+ path->corresponding = corresponding;
+ path->parent = parent;
+ path->num_children = -1;
+ path->children = NULL;
+ path->position = -1;
+ path->orig_position = -1;
+ path->child_needs_resort = 0;
+ path->resort_all_children = 0;
+ path->needs_resort = 0;
+ path->needs_regen_to_sort = 0;
+
+ return path;
+}
+
+static gboolean
+reposition_path (ETreeSorted *ets,
+ ETreeSortedPath *path)
+{
+ gint new_index;
+ gint old_index = path->position;
+ ETreeSortedPath *parent = path->parent;
+ gboolean changed = FALSE;
+ if (parent) {
+ if (ets->priv->sort_idle_id == 0) {
+ if (ets->priv->insert_count > ETS_INSERT_MAX) {
+ /* schedule a sort, and append instead */
+ schedule_resort (ets, parent, TRUE, FALSE);
+ } else {
+ /* make sure we have an idle handler to reset the count every now and then */
+ if (ets->priv->insert_idle_id == 0) {
+ ets->priv->insert_idle_id = g_idle_add_full (40, (GSourceFunc) ets_insert_idle, ets, NULL);
+ }
+
+ new_index = e_table_sorting_utils_tree_check_position
+ (E_TREE_MODEL (ets),
+ ets->priv->sort_info,
+ ets->priv->full_header,
+ (ETreePath *) parent->children,
+ parent->num_children,
+ old_index);
+
+ if (new_index > old_index) {
+ gint i;
+ ets->priv->insert_count++;
+ memmove (parent->children + old_index, parent->children + old_index + 1, sizeof (ETreePath) * (new_index - old_index));
+ parent->children[new_index] = path;
+ for (i = old_index; i <= new_index; i++)
+ parent->children[i]->position = i;
+ changed = TRUE;
+ e_tree_model_node_changed (E_TREE_MODEL (ets), parent);
+ e_tree_sorted_node_resorted (ets, parent);
+ } else if (new_index < old_index) {
+ gint i;
+ ets->priv->insert_count++;
+ memmove (parent->children + new_index + 1, parent->children + new_index, sizeof (ETreePath) * (old_index - new_index));
+ parent->children[new_index] = path;
+ for (i = new_index; i <= old_index; i++)
+ parent->children[i]->position = i;
+ changed = TRUE;
+ e_tree_model_node_changed (E_TREE_MODEL (ets), parent);
+ e_tree_sorted_node_resorted (ets, parent);
+ }
+ }
+ } else
+ mark_path_needs_resort (ets, parent, TRUE, FALSE);
+ }
+ return changed;
+}
+
+static void
+regenerate_children (ETreeSorted *ets,
+ ETreeSortedPath *path)
+{
+ ETreeSortedPath **children;
+ gint i;
+
+ children = g_new (ETreeSortedPath *, path->num_children);
+ for (i = 0; i < path->num_children; i++)
+ children[path->children[i]->orig_position] = path->children[i];
+ g_free (path->children);
+ path->children = children;
+}
+
+static void
+generate_children (ETreeSorted *ets,
+ ETreeSortedPath *path)
+{
+ ETreePath child;
+ gint i;
+ gint count;
+
+ free_children (path);
+
+ count = 0;
+ for (child = e_tree_model_node_get_first_child (ets->priv->source, path->corresponding);
+ child;
+ child = e_tree_model_node_get_next (ets->priv->source, child)) {
+ count++;
+ }
+
+ path->num_children = count;
+ path->children = g_new (ETreeSortedPath *, count);
+ for (child = e_tree_model_node_get_first_child (ets->priv->source, path->corresponding), i = 0;
+ child;
+ child = e_tree_model_node_get_next (ets->priv->source, child), i++) {
+ path->children[i] = new_path (path, child);
+ path->children[i]->position = i;
+ path->children[i]->orig_position = i;
+ }
+ if (path->num_children > 0)
+ schedule_resort (ets, path, FALSE, TRUE);
+}
+
+static void
+resort_node (ETreeSorted *ets,
+ ETreeSortedPath *path,
+ gboolean resort_all_children,
+ gboolean needs_regen,
+ gboolean send_signals)
+{
+ gboolean needs_resort;
+ if (path) {
+ needs_resort = path->needs_resort || resort_all_children;
+ needs_regen = path->needs_regen_to_sort || needs_regen;
+ if (path->num_children > 0) {
+ if (needs_resort && send_signals)
+ e_tree_model_pre_change (E_TREE_MODEL (ets));
+ if (needs_resort) {
+ gint i;
+ d (g_print ("Start sort of node %p\n", path));
+ if (needs_regen)
+ regenerate_children (ets, path);
+ d (g_print ("Regened sort of node %p\n", path));
+ e_table_sorting_utils_tree_sort (
+ E_TREE_MODEL (ets),
+ ets->priv->sort_info,
+ ets->priv->full_header,
+ (ETreePath *) path->children,
+ path->num_children);
+ d (g_print ("Renumbering sort of node %p\n", path));
+ for (i = 0; i < path->num_children; i++) {
+ path->children[i]->position = i;
+ }
+ d (g_print ("End sort of node %p\n", path));
+ }
+ if (path->resort_all_children)
+ resort_all_children = TRUE;
+ if ((resort_all_children || path->child_needs_resort) && path->num_children >= 0) {
+ gint i;
+ for (i = 0; i < path->num_children; i++) {
+ resort_node (ets, path->children[i], resort_all_children, needs_regen, send_signals && !needs_resort);
+ }
+ path->child_needs_resort = 0;
+ }
+ }
+ path->needs_resort = 0;
+ path->child_needs_resort = 0;
+ path->needs_regen_to_sort = 0;
+ path->resort_all_children = 0;
+ if (needs_resort && send_signals && path->num_children > 0) {
+ e_tree_model_node_changed (E_TREE_MODEL (ets), path);
+ e_tree_sorted_node_resorted (ets, path);
+ }
+ }
+}
+
+static void
+mark_path_child_needs_resort (ETreeSorted *ets,
+ ETreeSortedPath *path)
+{
+ if (path == NULL)
+ return;
+ if (!path->child_needs_resort) {
+ path->child_needs_resort = 1;
+ mark_path_child_needs_resort (ets, path->parent);
+ }
+}
+
+static void
+mark_path_needs_resort (ETreeSorted *ets,
+ ETreeSortedPath *path,
+ gboolean needs_regen,
+ gboolean resort_all_children)
+{
+ if (path == NULL)
+ return;
+ if (path->num_children == 0)
+ return;
+ path->needs_resort = 1;
+ path->needs_regen_to_sort = needs_regen;
+ path->resort_all_children = resort_all_children;
+ mark_path_child_needs_resort (ets, path->parent);
+}
+
+static void
+schedule_resort (ETreeSorted *ets,
+ ETreeSortedPath *path,
+ gboolean needs_regen,
+ gboolean resort_all_children)
+{
+ ets->priv->insert_count = 0;
+ if (ets->priv->insert_idle_id != 0) {
+ g_source_remove (ets->priv->insert_idle_id);
+ ets->priv->insert_idle_id = 0;
+ }
+
+ if (path == NULL)
+ return;
+ if (path->num_children == 0)
+ return;
+
+ mark_path_needs_resort (ets, path, needs_regen, resort_all_children);
+ if (ets->priv->sort_idle_id == 0) {
+ ets->priv->sort_idle_id = g_idle_add_full (50, (GSourceFunc) ets_sort_idle, ets, NULL);
+ } else if (ets->priv->in_resort_idle) {
+ ets->priv->nested_resort_idle = TRUE;
+ }
+}
+
+/* virtual methods */
+
+static void
+ets_dispose (GObject *object)
+{
+ ETreeSortedPrivate *priv;
+
+ priv = E_TREE_SORTED_GET_PRIVATE (object);
+
+ if (priv->source) {
+ g_signal_handler_disconnect (
+ priv->source, priv->tree_model_pre_change_id);
+ g_signal_handler_disconnect (
+ priv->source, priv->tree_model_no_change_id);
+ g_signal_handler_disconnect (
+ priv->source, priv->tree_model_node_changed_id);
+ g_signal_handler_disconnect (
+ priv->source, priv->tree_model_node_data_changed_id);
+ g_signal_handler_disconnect (
+ priv->source, priv->tree_model_node_col_changed_id);
+ g_signal_handler_disconnect (
+ priv->source, priv->tree_model_node_inserted_id);
+ g_signal_handler_disconnect (
+ priv->source, priv->tree_model_node_removed_id);
+ g_signal_handler_disconnect (
+ priv->source, priv->tree_model_node_deleted_id);
+ g_signal_handler_disconnect (
+ priv->source, priv->tree_model_node_request_collapse_id);
+
+ g_object_unref (priv->source);
+ priv->source = NULL;
+
+ priv->tree_model_pre_change_id = 0;
+ priv->tree_model_no_change_id = 0;
+ priv->tree_model_node_changed_id = 0;
+ priv->tree_model_node_data_changed_id = 0;
+ priv->tree_model_node_col_changed_id = 0;
+ priv->tree_model_node_inserted_id = 0;
+ priv->tree_model_node_removed_id = 0;
+ priv->tree_model_node_deleted_id = 0;
+ priv->tree_model_node_request_collapse_id = 0;
+ }
+
+ if (priv->sort_info) {
+ g_signal_handler_disconnect (
+ priv->sort_info, priv->sort_info_changed_id);
+ priv->sort_info_changed_id = 0;
+
+ g_object_unref (priv->sort_info);
+ priv->sort_info = NULL;
+ }
+
+ ets_stop_sort_idle (E_TREE_SORTED (object));
+
+ if (priv->insert_idle_id) {
+ g_source_remove (priv->insert_idle_id);
+ priv->insert_idle_id = 0;
+ }
+
+ if (priv->full_header) {
+ g_object_unref (priv->full_header);
+ priv->full_header = NULL;
+ }
+
+ /* Chain up to parent's dispose() method. */
+ G_OBJECT_CLASS (e_tree_sorted_parent_class)->dispose (object);
+}
+
+static void
+ets_finalize (GObject *object)
+{
+ ETreeSortedPrivate *priv;
+
+ priv = E_TREE_SORTED_GET_PRIVATE (object);
+
+ if (priv->root)
+ free_path (priv->root);
+
+ /* Chain up to parent's finalize() method. */
+ G_OBJECT_CLASS (e_tree_sorted_parent_class)->finalize (object);
+}
+
+static ETreePath
+ets_get_root (ETreeModel *etm)
+{
+ ETreeSortedPrivate *priv = E_TREE_SORTED (etm)->priv;
+ if (priv->root == NULL) {
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+ ETreePath corresponding = e_tree_model_get_root (ets->priv->source);
+
+ if (corresponding) {
+ priv->root = new_path (NULL, corresponding);
+ }
+ }
+ if (priv->root && priv->root->num_children == -1) {
+ generate_children (E_TREE_SORTED (etm), priv->root);
+ }
+
+ return priv->root;
+}
+
+static ETreePath
+ets_get_parent (ETreeModel *etm,
+ ETreePath node)
+{
+ ETreeSortedPath *path = node;
+ return path->parent;
+}
+
+static ETreePath
+ets_get_first_child (ETreeModel *etm,
+ ETreePath node)
+{
+ ETreeSortedPath *path = node;
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+
+ if (path->num_children == -1)
+ generate_children (ets, path);
+
+ if (path->num_children > 0)
+ return path->children[0];
+ else
+ return NULL;
+}
+
+static ETreePath
+ets_get_last_child (ETreeModel *etm,
+ ETreePath node)
+{
+ ETreeSortedPath *path = node;
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+
+ if (path->num_children == -1)
+ generate_children (ets, path);
+
+ if (path->num_children > 0)
+ return path->children[path->num_children - 1];
+ else
+ return NULL;
+}
+
+static ETreePath
+ets_get_next (ETreeModel *etm,
+ ETreePath node)
+{
+ ETreeSortedPath *path = node;
+ ETreeSortedPath *parent = path->parent;
+ if (parent) {
+ if (parent->num_children > path->position + 1)
+ return parent->children[path->position + 1];
+ else
+ return NULL;
+ } else
+ return NULL;
+}
+
+static ETreePath
+ets_get_prev (ETreeModel *etm,
+ ETreePath node)
+{
+ ETreeSortedPath *path = node;
+ ETreeSortedPath *parent = path->parent;
+ if (parent) {
+ if (path->position - 1 >= 0)
+ return parent->children[path->position - 1];
+ else
+ return NULL;
+ } else
+ return NULL;
+}
+
+static gboolean
+ets_is_root (ETreeModel *etm,
+ ETreePath node)
+{
+ ETreeSortedPath *path = node;
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+
+ return e_tree_model_node_is_root (ets->priv->source, path->corresponding);
+}
+
+static gboolean
+ets_is_expandable (ETreeModel *etm,
+ ETreePath node)
+{
+ ETreeSortedPath *path = node;
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+ gboolean expandable = e_tree_model_node_is_expandable (ets->priv->source, path->corresponding);
+
+ if (path->num_children == -1) {
+ generate_children (ets, node);
+ }
+
+ return expandable;
+}
+
+static guint
+ets_get_children (ETreeModel *etm,
+ ETreePath node,
+ ETreePath **nodes)
+{
+ ETreeSortedPath *path = node;
+ guint n_children;
+
+ if (path->num_children == -1) {
+ generate_children (E_TREE_SORTED (etm), node);
+ }
+
+ n_children = path->num_children;
+
+ if (nodes) {
+ gint i;
+
+ (*nodes) = g_malloc (sizeof (ETreePath) * n_children);
+ for (i = 0; i < n_children; i++) {
+ (*nodes)[i] = path->children[i];
+ }
+ }
+
+ return n_children;
+}
+
+static guint
+ets_depth (ETreeModel *etm,
+ ETreePath node)
+{
+ ETreeSortedPath *path = node;
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+
+ return e_tree_model_node_depth (ets->priv->source, path->corresponding);
+}
+
+static GdkPixbuf *
+ets_icon_at (ETreeModel *etm,
+ ETreePath node)
+{
+ ETreeSortedPath *path = node;
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+
+ return e_tree_model_icon_at (ets->priv->source, path->corresponding);
+}
+
+static gboolean
+ets_get_expanded_default (ETreeModel *etm)
+{
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+
+ return e_tree_model_get_expanded_default (ets->priv->source);
+}
+
+static gint
+ets_column_count (ETreeModel *etm)
+{
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+
+ return e_tree_model_column_count (ets->priv->source);
+}
+
+static gboolean
+ets_has_save_id (ETreeModel *etm)
+{
+ return TRUE;
+}
+
+static gchar *
+ets_get_save_id (ETreeModel *etm,
+ ETreePath node)
+{
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+ ETreeSortedPath *path = node;
+
+ if (e_tree_model_has_save_id (ets->priv->source))
+ return e_tree_model_get_save_id (ets->priv->source, path->corresponding);
+ else
+ return g_strdup_printf ("%p", path->corresponding);
+}
+
+static gboolean
+ets_has_get_node_by_id (ETreeModel *etm)
+{
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+ return e_tree_model_has_get_node_by_id (ets->priv->source);
+}
+
+static ETreePath
+ets_get_node_by_id (ETreeModel *etm,
+ const gchar *save_id)
+{
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+ ETreePath node;
+
+ node = e_tree_model_get_node_by_id (ets->priv->source, save_id);
+
+ return find_path (ets, node);
+}
+
+static gboolean
+ets_has_change_pending (ETreeModel *etm)
+{
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+
+ return ets->priv->sort_idle_id != 0;
+}
+
+static gpointer
+ets_value_at (ETreeModel *etm,
+ ETreePath node,
+ gint col)
+{
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+ ETreeSortedPath *path = node;
+
+ return e_tree_model_value_at (ets->priv->source, path->corresponding, col);
+}
+
+static void
+ets_set_value_at (ETreeModel *etm,
+ ETreePath node,
+ gint col,
+ gconstpointer val)
+{
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+ ETreeSortedPath *path = node;
+
+ e_tree_model_set_value_at (ets->priv->source, path->corresponding, col, val);
+}
+
+static gboolean
+ets_is_editable (ETreeModel *etm,
+ ETreePath node,
+ gint col)
+{
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+ ETreeSortedPath *path = node;
+
+ return e_tree_model_node_is_editable (ets->priv->source, path->corresponding, col);
+}
+
+/* The default for ets_duplicate_value is to return the raw value. */
+static gpointer
+ets_duplicate_value (ETreeModel *etm,
+ gint col,
+ gconstpointer value)
+{
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+
+ return e_tree_model_duplicate_value (ets->priv->source, col, value);
+}
+
+static void
+ets_free_value (ETreeModel *etm,
+ gint col,
+ gpointer value)
+{
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+
+ e_tree_model_free_value (ets->priv->source, col, value);
+}
+
+static gpointer
+ets_initialize_value (ETreeModel *etm,
+ gint col)
+{
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+
+ return e_tree_model_initialize_value (ets->priv->source, col);
+}
+
+static gboolean
+ets_value_is_empty (ETreeModel *etm,
+ gint col,
+ gconstpointer value)
+{
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+
+ return e_tree_model_value_is_empty (ets->priv->source, col, value);
+}
+
+static gchar *
+ets_value_to_string (ETreeModel *etm,
+ gint col,
+ gconstpointer value)
+{
+ ETreeSorted *ets = E_TREE_SORTED (etm);
+
+ return e_tree_model_value_to_string (ets->priv->source, col, value);
+}
+
+/* Proxy functions */
+
+static void
+ets_proxy_pre_change (ETreeModel *etm,
+ ETreeSorted *ets)
+{
+ e_tree_model_pre_change (E_TREE_MODEL (ets));
+}
+
+static void
+ets_proxy_no_change (ETreeModel *etm,
+ ETreeSorted *ets)
+{
+ e_tree_model_no_change (E_TREE_MODEL (ets));
+}
+
+static void
+ets_proxy_node_changed (ETreeModel *etm,
+ ETreePath node,
+ ETreeSorted *ets)
+{
+ ets->priv->last_access = NULL;
+ d (g_print ("Setting last access %p. (ets_proxy_node_changed)\n", ets->priv->last_access));
+
+ if (e_tree_model_node_is_root (ets->priv->source, node)) {
+ ets_stop_sort_idle (ets);
+
+ if (ets->priv->root) {
+ free_path (ets->priv->root);
+ }
+ ets->priv->root = new_path (NULL, node);
+ e_tree_model_node_changed (E_TREE_MODEL (ets), ets->priv->root);
+ return;
+ } else {
+ ETreeSortedPath *path = find_path (ets, node);
+
+ if (path) {
+ free_children (path);
+ if (!reposition_path (ets, path)) {
+ e_tree_model_node_changed (E_TREE_MODEL (ets), path);
+ } else {
+ e_tree_model_no_change (E_TREE_MODEL (ets));
+ }
+ } else {
+ e_tree_model_no_change (E_TREE_MODEL (ets));
+ }
+ }
+}
+
+static void
+ets_proxy_node_data_changed (ETreeModel *etm,
+ ETreePath node,
+ ETreeSorted *ets)
+{
+ ETreeSortedPath *path = find_path (ets, node);
+
+ if (path) {
+ if (!reposition_path (ets, path))
+ e_tree_model_node_data_changed (E_TREE_MODEL (ets), path);
+ else
+ e_tree_model_no_change (E_TREE_MODEL (ets));
+ } else
+ e_tree_model_no_change (E_TREE_MODEL (ets));
+}
+
+static void
+ets_proxy_node_col_changed (ETreeModel *etm,
+ ETreePath node,
+ gint col,
+ ETreeSorted *ets)
+{
+ ETreeSortedPath *path = find_path (ets, node);
+
+ if (path) {
+ gboolean changed = FALSE;
+ if (e_table_sorting_utils_affects_sort (ets->priv->sort_info, ets->priv->full_header, col))
+ changed = reposition_path (ets, path);
+ if (!changed)
+ e_tree_model_node_col_changed (E_TREE_MODEL (ets), path, col);
+ else
+ e_tree_model_no_change (E_TREE_MODEL (ets));
+ } else
+ e_tree_model_no_change (E_TREE_MODEL (ets));
+}
+
+static void
+ets_proxy_node_inserted (ETreeModel *etm,
+ ETreePath parent,
+ ETreePath child,
+ ETreeSorted *ets)
+{
+ ETreeSortedPath *parent_path = find_path (ets, parent);
+
+ if (parent_path && parent_path->num_children != -1) {
+ gint i;
+ gint j;
+ ETreeSortedPath *path;
+ gint position = parent_path->num_children;
+ ETreePath counter;
+
+ for (counter = e_tree_model_node_get_next (etm, child);
+ counter;
+ counter = e_tree_model_node_get_next (etm, counter))
+ position--;
+
+ if (position != parent_path->num_children) {
+ for (i = 0; i < parent_path->num_children; i++) {
+ if (parent_path->children[i]->orig_position >= position)
+ parent_path->children[i]->orig_position++;
+ }
+ }
+
+ i = parent_path->num_children;
+ path = new_path (parent_path, child);
+ path->orig_position = position;
+ if (!ETS_SORT_IDLE_ACTIVATED (ets)) {
+ ets->priv->insert_count++;
+ if (ets->priv->insert_count > ETS_INSERT_MAX) {
+ /* schedule a sort, and append instead */
+ schedule_resort (ets, parent_path, TRUE, FALSE);
+ } else {
+ /* make sure we have an idle handler to reset the count every now and then */
+ if (ets->priv->insert_idle_id == 0) {
+ ets->priv->insert_idle_id = g_idle_add_full (40, (GSourceFunc) ets_insert_idle, ets, NULL);
+ }
+ i = e_table_sorting_utils_tree_insert
+ (ets->priv->source,
+ ets->priv->sort_info,
+ ets->priv->full_header,
+ (ETreePath *) parent_path->children,
+ parent_path->num_children,
+ path);
+ }
+ } else {
+ mark_path_needs_resort (ets, parent_path, TRUE, FALSE);
+ }
+ parent_path->num_children++;
+ parent_path->children = g_renew (ETreeSortedPath *, parent_path->children, parent_path->num_children);
+ memmove (parent_path->children + i + 1, parent_path->children + i, (parent_path->num_children - 1 - i) * sizeof (gint));
+ parent_path->children[i] = path;
+ for (j = i; j < parent_path->num_children; j++) {
+ parent_path->children[j]->position = j;
+ }
+ e_tree_model_node_inserted (E_TREE_MODEL (ets), parent_path, parent_path->children[i]);
+ } else if (ets->priv->root == NULL && parent == NULL) {
+ if (child) {
+ ets->priv->root = new_path (NULL, child);
+ e_tree_model_node_inserted (E_TREE_MODEL (ets), NULL, ets->priv->root);
+ } else {
+ e_tree_model_no_change (E_TREE_MODEL (ets));
+ }
+ } else {
+ e_tree_model_no_change (E_TREE_MODEL (ets));
+ }
+}
+
+static void
+ets_proxy_node_removed (ETreeModel *etm,
+ ETreePath parent,
+ ETreePath child,
+ gint old_position,
+ ETreeSorted *ets)
+{
+ ETreeSortedPath *parent_path = find_path (ets, parent);
+ ETreeSortedPath *path;
+
+ if (parent_path)
+ path = find_child_path (ets, parent_path, child);
+ else
+ path = find_path (ets, child);
+
+ d (g_print ("Setting last access %p. (ets_proxy_node_removed)\n ", ets->priv->last_access));
+ ets->priv->last_access = NULL;
+
+ if (path && parent_path && parent_path->num_children != -1) {
+ gint i;
+ for (i = 0; i < parent_path->num_children; i++) {
+ if (parent_path->children[i]->orig_position > old_position)
+ parent_path->children[i]->orig_position--;
+ }
+
+ i = path->position;
+
+ parent_path->num_children--;
+ memmove (parent_path->children + i, parent_path->children + i + 1, sizeof (ETreeSortedPath *) * (parent_path->num_children - i));
+ for (; i < parent_path->num_children; i++) {
+ parent_path->children[i]->position = i;
+ }
+ e_tree_model_node_removed (E_TREE_MODEL (ets), parent_path, path, path->position);
+ free_path (path);
+ } else if (path && path == ets->priv->root) {
+ ets->priv->root = NULL;
+ e_tree_model_node_removed (E_TREE_MODEL (ets), NULL, path, -1);
+ free_path (path);
+ }
+}
+
+static void
+ets_proxy_node_deleted (ETreeModel *etm,
+ ETreePath child,
+ ETreeSorted *ets)
+{
+ e_tree_model_node_deleted (E_TREE_MODEL (ets), NULL);
+}
+
+static void
+ets_proxy_node_request_collapse (ETreeModel *etm,
+ ETreePath node,
+ ETreeSorted *ets)
+{
+ ETreeSortedPath *path = find_path (ets, node);
+ if (path) {
+ e_tree_model_node_request_collapse (E_TREE_MODEL (ets), path);
+ }
+}
+
+static void
+ets_sort_info_changed (ETableSortInfo *sort_info,
+ ETreeSorted *ets)
+{
+ schedule_resort (ets, ets->priv->root, TRUE, TRUE);
+}
+
+/* Initialization and creation */
+
+static void
+e_tree_sorted_class_init (ETreeSortedClass *class)
+{
+ GObjectClass *object_class;
+ ETreeModelClass *tree_model_class;
+
+ g_type_class_add_private (class, sizeof (ETreeSortedPrivate));
+
+ object_class = G_OBJECT_CLASS (class);
+ object_class->dispose = ets_dispose;
+ object_class->finalize = ets_finalize;
+
+ tree_model_class = E_TREE_MODEL_CLASS (class);
+ tree_model_class->get_root = ets_get_root;
+ tree_model_class->get_parent = ets_get_parent;
+ tree_model_class->get_first_child = ets_get_first_child;
+ tree_model_class->get_last_child = ets_get_last_child;
+ tree_model_class->get_prev = ets_get_prev;
+ tree_model_class->get_next = ets_get_next;
+
+ tree_model_class->is_root = ets_is_root;
+ tree_model_class->is_expandable = ets_is_expandable;
+ tree_model_class->get_children = ets_get_children;
+ tree_model_class->depth = ets_depth;
+
+ tree_model_class->icon_at = ets_icon_at;
+
+ tree_model_class->get_expanded_default = ets_get_expanded_default;
+ tree_model_class->column_count = ets_column_count;
+
+ tree_model_class->has_save_id = ets_has_save_id;
+ tree_model_class->get_save_id = ets_get_save_id;
+
+ tree_model_class->has_get_node_by_id = ets_has_get_node_by_id;
+ tree_model_class->get_node_by_id = ets_get_node_by_id;
+
+ tree_model_class->has_change_pending = ets_has_change_pending;
+
+ tree_model_class->value_at = ets_value_at;
+ tree_model_class->set_value_at = ets_set_value_at;
+ tree_model_class->is_editable = ets_is_editable;
+
+ tree_model_class->duplicate_value = ets_duplicate_value;
+ tree_model_class->free_value = ets_free_value;
+ tree_model_class->initialize_value = ets_initialize_value;
+ tree_model_class->value_is_empty = ets_value_is_empty;
+ tree_model_class->value_to_string = ets_value_to_string;
+
+ signals[NODE_RESORTED] = g_signal_new (
+ "node_resorted",
+ G_TYPE_FROM_CLASS (object_class),
+ G_SIGNAL_RUN_LAST,
+ G_STRUCT_OFFSET (ETreeSortedClass, node_resorted),
+ (GSignalAccumulator) NULL, NULL,
+ g_cclosure_marshal_VOID__POINTER,
+ G_TYPE_NONE, 1,
+ G_TYPE_POINTER);
+}
+
+static void
+e_tree_sorted_init (ETreeSorted *ets)
+{
+ ets->priv = E_TREE_SORTED_GET_PRIVATE (ets);
+}
+
+/**
+ * e_tree_sorted_construct:
+ * @etree:
+ *
+ *
+ **/
+void
+e_tree_sorted_construct (ETreeSorted *ets,
+ ETreeModel *source,
+ ETableHeader *full_header,
+ ETableSortInfo *sort_info)
+{
+ ets->priv->source = source;
+ if (source)
+ g_object_ref (source);
+
+ ets->priv->full_header = full_header;
+ if (full_header)
+ g_object_ref (full_header);
+
+ e_tree_sorted_set_sort_info (ets, sort_info);
+
+ ets->priv->tree_model_pre_change_id = g_signal_connect (
+ source, "pre_change",
+ G_CALLBACK (ets_proxy_pre_change), ets);
+ ets->priv->tree_model_no_change_id = g_signal_connect (
+ source, "no_change",
+ G_CALLBACK (ets_proxy_no_change), ets);
+ ets->priv->tree_model_node_changed_id = g_signal_connect (
+ source, "node_changed",
+ G_CALLBACK (ets_proxy_node_changed), ets);
+ ets->priv->tree_model_node_data_changed_id = g_signal_connect (
+ source, "node_data_changed",
+ G_CALLBACK (ets_proxy_node_data_changed), ets);
+ ets->priv->tree_model_node_col_changed_id = g_signal_connect (
+ source, "node_col_changed",
+ G_CALLBACK (ets_proxy_node_col_changed), ets);
+ ets->priv->tree_model_node_inserted_id = g_signal_connect (
+ source, "node_inserted",
+ G_CALLBACK (ets_proxy_node_inserted), ets);
+ ets->priv->tree_model_node_removed_id = g_signal_connect (
+ source, "node_removed",
+ G_CALLBACK (ets_proxy_node_removed), ets);
+ ets->priv->tree_model_node_deleted_id = g_signal_connect (
+ source, "node_deleted",
+ G_CALLBACK (ets_proxy_node_deleted), ets);
+ ets->priv->tree_model_node_request_collapse_id = g_signal_connect (
+ source, "node_request_collapse",
+ G_CALLBACK (ets_proxy_node_request_collapse), ets);
+
+}
+
+/**
+ * e_tree_sorted_new
+ *
+ * FIXME docs here.
+ *
+ * return values: a newly constructed ETreeSorted.
+ */
+ETreeSorted *
+e_tree_sorted_new (ETreeModel *source,
+ ETableHeader *full_header,
+ ETableSortInfo *sort_info)
+{
+ ETreeSorted *ets = g_object_new (E_TYPE_TREE_SORTED, NULL);
+
+ e_tree_sorted_construct (ets, source, full_header, sort_info);
+
+ return ets;
+}
+
+ETreePath
+e_tree_sorted_view_to_model_path (ETreeSorted *ets,
+ ETreePath view_path)
+{
+ ETreeSortedPath *path = view_path;
+ if (path) {
+ ets->priv->last_access = path;
+ d (g_print ("Setting last access %p. (e_tree_sorted_view_to_model_path)\n", ets->priv->last_access));
+ return path->corresponding;
+ } else
+ return NULL;
+}
+
+ETreePath
+e_tree_sorted_model_to_view_path (ETreeSorted *ets,
+ ETreePath model_path)
+{
+ return find_or_create_path (ets, model_path);
+}
+
+gint
+e_tree_sorted_orig_position (ETreeSorted *ets,
+ ETreePath path)
+{
+ ETreeSortedPath *sorted_path = path;
+ return sorted_path->orig_position;
+}
+
+gint
+e_tree_sorted_node_num_children (ETreeSorted *ets,
+ ETreePath path)
+{
+ ETreeSortedPath *sorted_path = path;
+
+ if (sorted_path->num_children == -1) {
+ generate_children (ets, sorted_path);
+ }
+
+ return sorted_path->num_children;
+}
+
+void
+e_tree_sorted_node_resorted (ETreeSorted *sorted,
+ ETreePath node)
+{
+ g_return_if_fail (sorted != NULL);
+ g_return_if_fail (E_IS_TREE_SORTED (sorted));
+
+ g_signal_emit (sorted, signals[NODE_RESORTED], 0, node);
+}
+
+void
+e_tree_sorted_set_sort_info (ETreeSorted *ets,
+ ETableSortInfo *sort_info)
+{
+
+ g_return_if_fail (ets != NULL);
+
+ if (ets->priv->sort_info) {
+ if (ets->priv->sort_info_changed_id != 0)
+ g_signal_handler_disconnect (
+ ets->priv->sort_info,
+ ets->priv->sort_info_changed_id);
+ ets->priv->sort_info_changed_id = 0;
+ g_object_unref (ets->priv->sort_info);
+ }
+
+ ets->priv->sort_info = sort_info;
+ if (sort_info) {
+ g_object_ref (sort_info);
+ ets->priv->sort_info_changed_id = g_signal_connect (
+ ets->priv->sort_info, "sort_info_changed",
+ G_CALLBACK (ets_sort_info_changed), ets);
+ }
+
+ if (ets->priv->root)
+ schedule_resort (ets, ets->priv->root, TRUE, TRUE);
+}
+
+ETableSortInfo *
+e_tree_sorted_get_sort_info (ETreeSorted *ets)
+{
+ return ets->priv->sort_info;
+}
+