From 783be711f51e4334693caba8b526335bdee95dd4 Mon Sep 17 00:00:00 2001 From: Graydon Hoare Date: Mon, 18 Oct 2010 14:35:44 -0700 Subject: [PATCH] Flesh out the std.list module a touch. --- src/lib/list.rs | 49 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 49 insertions(+) diff --git a/src/lib/list.rs b/src/lib/list.rs index e73d26255fb..60c23a9ef80 100644 --- a/src/lib/list.rs +++ b/src/lib/list.rs @@ -1,9 +1,58 @@ +import util.option; +import util.some; +import util.none; + +// FIXME: It would probably be more appealing to define this as +// type list[T] = rec(T hd, option[@list[T]] tl), but at the moment +// our recursion rules do not permit that. + tag list[T] { cons(T, @list[T]); nil; } +fn foldl[T,U](&list[T] ls, U u, fn(&T t, U u) -> U f) -> U { + alt(ls) { + case (cons[T](?hd, ?tl)) { + auto u_ = f(hd, u); + // FIXME: should use 'be' here, not 'ret'. But parametric + // tail calls currently don't work. + ret foldl[T,U](*tl, u_, f); + } + case (nil[T]) { + ret u; + } + } +} + +fn find[T](&list[T] ls, + (fn(&T) -> option[T]) f) -> option[T] { + alt(ls) { + case (cons[T](?hd, ?tl)) { + alt (f(hd)) { + case (none[T]) { + // FIXME: should use 'be' here, not 'ret'. But parametric tail + // calls currently don't work. + ret find[T](*tl, f); + } + case (some[T](?res)) { + ret some[T](res); + } + } + } + case (nil[T]) { + ret none[T]; + } + } +} + +fn length[T](&list[T] ls) -> uint { + fn count[T](&T t, uint u) -> uint { + ret u + 1u; + } + ret foldl[T,uint](ls, 0u, bind count[T](_, _)); +} // Local Variables: // mode: rust;