mq_tree_foreach_from(): New function

1 files changed, 32 insertions, 0 deletions

diff --git a/src/tree.c b/src/tree.c
index 90a7af7..b64bc5e 100644
--- a/src/tree.c
+++ b/src/tree.c
@@ -201,3 +201,35 @@ mq_tree_foreach_child(MqTree *node,
 {
 	foreach_child(node->root->first_child, cb, user_data);
 }
+
+static gboolean
+foreach_from(MqTree *node, gboolean (*cb)(MqTree *node, gpointer user_data),
+	gpointer user_data)
+{
+	for (; node; node = node->next) {
+		if (cb(node, user_data) == MQ_TREE_STOP) {
+			return MQ_TREE_STOP;
+		}
+		if (node->first_child) {
+			if (foreach_from(node->first_child, cb, user_data) ==
+					MQ_TREE_STOP) {
+				return MQ_TREE_STOP;
+			}
+		}
+		if (node->parent && node->parent->next) {
+			if (foreach_from(node->parent->next, cb, user_data) ==
+					MQ_TREE_STOP) {
+				return MQ_TREE_STOP;
+			}
+		}
+	}
+
+	return MQ_TREE_CONTINUE;
+}
+
+void
+mq_tree_foreach_from(MqTree *node,
+	gboolean (*cb)(MqTree *node, gpointer user_data), gpointer user_data)
+{
+	foreach_from(node, cb, user_data);
+}