rust/src/lib/list.rs

84 lines
2.1 KiB
Rust

import option::some;
import option::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 from_vec[@T](v: vec[T]) -> list[T] {
let l = nil[T];
// FIXME: This would be faster and more space efficient if it looped over
// a reverse vector iterator. Unfortunately generic iterators seem not to
// work yet.
for item: T in vec::reversed(v) { l = cons[T](item, @l); }
ret l;
}
fn foldl[@T, @U](ls_: &list[T], u: &U, f: fn(&T, &U) -> U ) -> U {
let accum: U = u;
let ls = ls_;
while true {
alt ls {
cons(hd, tl) { accum = f(hd, accum); ls = *tl; }
nil. { break; }
}
}
ret accum;
}
fn find[@T, @U](ls_: &list[T], f: fn(&T) -> option::t[U] ) -> option::t[U] {
let ls = ls_;
while true {
alt ls {
cons(hd, tl) {
alt f(hd) { none. { ls = *tl; } some(rs) { ret some(rs); } }
}
nil. { break; }
}
}
ret none;
}
fn has[@T](ls_: &list[T], elt: &T) -> bool {
let ls = ls_;
while true {
alt ls {
cons(hd, tl) { if elt == hd { ret true; } else { ls = *tl; } }
nil. { ret false; }
}
}
ret false; // Typestate checker doesn't understand infinite loops
}
fn length[T](ls: &list[T]) -> uint {
fn count[T](t: &T, u: &uint) -> uint { ret u + 1u; }
ret foldl[T, uint](ls, 0u, bind count[T](_, _));
}
fn cdr[@T](ls: &list[T]) -> list[T] { alt ls { cons(_, tl) { ret *tl; } } }
fn car[@T](ls: &list[T]) -> T { alt ls { cons(hd, _) { ret hd; } } }
fn append[@T](l: &list[T], m: &list[T]) -> list[T] {
alt l {
nil. { ret m; }
cons(x, xs) {
let rest: list[T] = append[T](*xs, m);
ret cons[T](x, @rest);
}
}
}
// Local Variables:
// mode: rust;
// fill-column: 78;
// indent-tabs-mode: nil
// c-basic-offset: 4
// buffer-file-coding-system: utf-8-unix
// compile-command: "make -k -C $RBUILD 2>&1 | sed -e 's/\\/x\\//x:\\//g'";
// End: