diff options
| author | Marijn Haverbeke <[email protected]> | 2011-05-12 17:24:54 +0200 |
|---|---|---|
| committer | Marijn Haverbeke <[email protected]> | 2011-05-12 21:30:44 +0200 |
| commit | 3816e57fd2a8ab19e4ac6d4b3ddd5b49d5973ff2 (patch) | |
| tree | 508982ed2f789aedd89eebd529343d9dc88b8e01 /src/lib/deque.rs | |
| parent | Transitional change to make extfmt output lowercase module name (diff) | |
| download | rust-3816e57fd2a8ab19e4ac6d4b3ddd5b49d5973ff2.tar.xz rust-3816e57fd2a8ab19e4ac6d4b3ddd5b49d5973ff2.zip | |
Downcase std modules again, move to :: for module dereferencing
This should be a snapshot transition.
Diffstat (limited to 'src/lib/deque.rs')
| -rw-r--r-- | src/lib/deque.rs | 145 |
1 files changed, 145 insertions, 0 deletions
diff --git a/src/lib/deque.rs b/src/lib/deque.rs new file mode 100644 index 00000000..c0ded5bf --- /dev/null +++ b/src/lib/deque.rs @@ -0,0 +1,145 @@ +/** + * A deque, for fun. Untested as of yet. Likely buggy. + */ + +type t[T] = obj { + fn size() -> uint; + + fn add_front(&T t); + fn add_back(&T t); + + fn pop_front() -> T; + fn pop_back() -> T; + + fn peek_front() -> T; + fn peek_back() -> T; + + fn get(int i) -> T; +}; + +fn create[T]() -> t[T] { + + type cell[T] = option::t[T]; + + let uint initial_capacity = 32u; // 2^5 + + /** + * Grow is only called on full elts, so nelts is also len(elts), unlike + * elsewhere. + */ + fn grow[T](uint nelts, uint lo, vec[cell[T]] elts) -> vec[cell[T]] { + assert (nelts == _vec::len[cell[T]](elts)); + + fn fill[T](uint i, uint nelts, uint lo, + vec[cell[T]] old) -> cell[T] { + if (i < nelts) { + ret old.((lo + i) % nelts); + } else { + ret option::none[T]; + } + } + + let uint nalloc = _uint::next_power_of_two(nelts + 1u); + let _vec::init_op[cell[T]] copy_op = bind fill[T](_, nelts, lo, elts); + ret _vec::init_fn[cell[T]](copy_op, nalloc); + } + + fn get[T](vec[cell[T]] elts, uint i) -> T { + alt (elts.(i)) { + case (option::some[T](?t)) { ret t; } + case (_) { fail; } + } + fail; // FIXME: remove me when exhaustiveness checking works + } + + obj deque[T](mutable uint nelts, + mutable uint lo, + mutable uint hi, + mutable vec[cell[T]] elts) + { + fn size() -> uint { ret nelts; } + + fn add_front(&T t) { + let uint oldlo = lo; + + if (lo == 0u) { + lo = _vec::len[cell[T]](elts) - 1u; + } else { + lo -= 1u; + } + + if (lo == hi) { + elts = grow[T](nelts, oldlo, elts); + lo = _vec::len[cell[T]](elts) - 1u; + hi = nelts; + } + + elts.(lo) = option::some[T](t); + nelts += 1u; + } + + fn add_back(&T t) { + if (lo == hi && nelts != 0u) { + elts = grow[T](nelts, lo, elts); + lo = 0u; + hi = nelts; + } + + elts.(hi) = option::some[T](t); + hi = (hi + 1u) % _vec::len[cell[T]](elts); + nelts += 1u; + } + + /** + * We actually release (turn to none()) the T we're popping so + * that we don't keep anyone's refcount up unexpectedly. + */ + fn pop_front() -> T { + let T t = get[T](elts, lo); + elts.(lo) = option::none[T]; + lo = (lo + 1u) % _vec::len[cell[T]](elts); + nelts -= 1u; + ret t; + } + + fn pop_back() -> T { + if (hi == 0u) { + hi = _vec::len[cell[T]](elts) - 1u; + } else { + hi -= 1u; + } + + let T t = get[T](elts, hi); + elts.(hi) = option::none[T]; + nelts -= 1u; + ret t; + } + + fn peek_front() -> T { + ret get[T](elts, lo); + } + + fn peek_back() -> T { + ret get[T](elts, hi - 1u); + } + + fn get(int i) -> T { + let uint idx = (lo + (i as uint)) % _vec::len[cell[T]](elts); + ret get[T](elts, idx); + } + + } + let vec[cell[T]] v = _vec::init_elt[cell[T]](option::none[T], + initial_capacity); + + ret deque[T](0u, 0u, 0u, v); +} + +// 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 .. 2>&1 | sed -e 's/\\/x\\//x:\\//g'"; +// End: |