From 722a6ca9a69cdd0a463d48af5380af19fd9779fc Mon Sep 17 00:00:00 2001 From: Kaz Kylheku Date: Wed, 12 May 2021 06:21:58 -0700 Subject: tree: let tree-iter be iterable via generic iteration. * lib.c (seq_iter_init_with_info): Recognize tree_iter object, and treat using tree iterator function. * tests/010/tree.tl: test case for tree subrange iteration with collect-each. * txr.1: Updated. --- lib.c | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'lib.c') diff --git a/lib.c b/lib.c index 45ec1417..3f2a9616 100644 --- a/lib.c +++ b/lib.c @@ -857,6 +857,16 @@ void seq_iter_init_with_info(val self, seq_iter_t *it, } } } + if (it->inf.obj->co.cls == tree_iter_s) + { + it->ui.iter = if3(support_rewind, + copy_tree_iter(it->inf.obj), + it->inf.obj); + it->ul.len = 0; + it->get = seq_iter_get_tree; + it->peek = seq_iter_peek_tree; + break; + } /* fallthrough */ default: switch (it->inf.kind) { -- cgit v1.2.3