311 lines
9.9 KiB
Rust
311 lines
9.9 KiB
Rust
//! A deque. Untested as of yet. Likely buggy
|
|
|
|
use option::{Some, None};
|
|
use dvec::DVec;
|
|
|
|
trait Deque<T> {
|
|
fn size() -> uint;
|
|
fn add_front(T);
|
|
fn add_back(T);
|
|
fn pop_front() -> T;
|
|
fn pop_back() -> T;
|
|
fn peek_front() -> T;
|
|
fn peek_back() -> T;
|
|
fn get(int) -> T;
|
|
}
|
|
|
|
// FIXME (#2343) eventually, a proper datatype plus an exported impl would
|
|
// be preferrable.
|
|
fn create<T: Copy>() -> Deque<T> {
|
|
type Cell<T> = Option<T>;
|
|
|
|
let initial_capacity: uint = 32u; // 2^5
|
|
/**
|
|
* Grow is only called on full elts, so nelts is also len(elts), unlike
|
|
* elsewhere.
|
|
*/
|
|
fn grow<T: Copy>(nelts: uint, lo: uint, -elts: ~[mut Cell<T>]) ->
|
|
~[mut Cell<T>] {
|
|
assert (nelts == vec::len(elts));
|
|
let mut rv = ~[mut];
|
|
|
|
let mut i = 0u;
|
|
let nalloc = uint::next_power_of_two(nelts + 1u);
|
|
while i < nalloc {
|
|
if i < nelts {
|
|
vec::push(rv, elts[(lo + i) % nelts]);
|
|
} else { vec::push(rv, None); }
|
|
i += 1u;
|
|
}
|
|
|
|
return rv;
|
|
}
|
|
fn get<T: Copy>(elts: DVec<Cell<T>>, i: uint) -> T {
|
|
match elts.get_elt(i) { Some(t) => t, _ => fail }
|
|
}
|
|
|
|
type Repr<T> = {mut nelts: uint,
|
|
mut lo: uint,
|
|
mut hi: uint,
|
|
elts: DVec<Cell<T>>};
|
|
|
|
impl <T: Copy> Repr<T>: Deque<T> {
|
|
fn size() -> uint { return self.nelts; }
|
|
fn add_front(t: T) {
|
|
let oldlo: uint = self.lo;
|
|
if self.lo == 0u {
|
|
self.lo = self.elts.len() - 1u;
|
|
} else { self.lo -= 1u; }
|
|
if self.lo == self.hi {
|
|
self.elts.swap(|v| grow(self.nelts, oldlo, v));
|
|
self.lo = self.elts.len() - 1u;
|
|
self.hi = self.nelts;
|
|
}
|
|
self.elts.set_elt(self.lo, Some(t));
|
|
self.nelts += 1u;
|
|
}
|
|
fn add_back(t: T) {
|
|
if self.lo == self.hi && self.nelts != 0u {
|
|
self.elts.swap(|v| grow(self.nelts, self.lo, v));
|
|
self.lo = 0u;
|
|
self.hi = self.nelts;
|
|
}
|
|
self.elts.set_elt(self.hi, Some(t));
|
|
self.hi = (self.hi + 1u) % self.elts.len();
|
|
self.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(self.elts, self.lo);
|
|
self.elts.set_elt(self.lo, None);
|
|
self.lo = (self.lo + 1u) % self.elts.len();
|
|
self.nelts -= 1u;
|
|
return t;
|
|
}
|
|
fn pop_back() -> T {
|
|
if self.hi == 0u {
|
|
self.hi = self.elts.len() - 1u;
|
|
} else { self.hi -= 1u; }
|
|
let t: T = get(self.elts, self.hi);
|
|
self.elts.set_elt(self.hi, None);
|
|
self.nelts -= 1u;
|
|
return t;
|
|
}
|
|
fn peek_front() -> T { return get(self.elts, self.lo); }
|
|
fn peek_back() -> T { return get(self.elts, self.hi - 1u); }
|
|
fn get(i: int) -> T {
|
|
let idx = (self.lo + (i as uint)) % self.elts.len();
|
|
return get(self.elts, idx);
|
|
}
|
|
}
|
|
|
|
let repr: Repr<T> = {
|
|
mut nelts: 0u,
|
|
mut lo: 0u,
|
|
mut hi: 0u,
|
|
elts:
|
|
dvec::from_vec(
|
|
vec::to_mut(
|
|
vec::from_elem(initial_capacity, None)))
|
|
};
|
|
repr as Deque::<T>
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
#[test]
|
|
fn test_simple() {
|
|
let d: deque::Deque<int> = deque::create::<int>();
|
|
assert (d.size() == 0u);
|
|
d.add_front(17);
|
|
d.add_front(42);
|
|
d.add_back(137);
|
|
assert (d.size() == 3u);
|
|
d.add_back(137);
|
|
assert (d.size() == 4u);
|
|
log(debug, d.peek_front());
|
|
assert (d.peek_front() == 42);
|
|
log(debug, d.peek_back());
|
|
assert (d.peek_back() == 137);
|
|
let mut i: int = d.pop_front();
|
|
log(debug, i);
|
|
assert (i == 42);
|
|
i = d.pop_back();
|
|
log(debug, i);
|
|
assert (i == 137);
|
|
i = d.pop_back();
|
|
log(debug, i);
|
|
assert (i == 137);
|
|
i = d.pop_back();
|
|
log(debug, i);
|
|
assert (i == 17);
|
|
assert (d.size() == 0u);
|
|
d.add_back(3);
|
|
assert (d.size() == 1u);
|
|
d.add_front(2);
|
|
assert (d.size() == 2u);
|
|
d.add_back(4);
|
|
assert (d.size() == 3u);
|
|
d.add_front(1);
|
|
assert (d.size() == 4u);
|
|
log(debug, d.get(0));
|
|
log(debug, d.get(1));
|
|
log(debug, d.get(2));
|
|
log(debug, d.get(3));
|
|
assert (d.get(0) == 1);
|
|
assert (d.get(1) == 2);
|
|
assert (d.get(2) == 3);
|
|
assert (d.get(3) == 4);
|
|
}
|
|
|
|
fn test_boxes(a: @int, b: @int, c: @int, d: @int) {
|
|
let deq: deque::Deque<@int> = deque::create::<@int>();
|
|
assert (deq.size() == 0u);
|
|
deq.add_front(a);
|
|
deq.add_front(b);
|
|
deq.add_back(c);
|
|
assert (deq.size() == 3u);
|
|
deq.add_back(d);
|
|
assert (deq.size() == 4u);
|
|
assert (deq.peek_front() == b);
|
|
assert (deq.peek_back() == d);
|
|
assert (deq.pop_front() == b);
|
|
assert (deq.pop_back() == d);
|
|
assert (deq.pop_back() == c);
|
|
assert (deq.pop_back() == a);
|
|
assert (deq.size() == 0u);
|
|
deq.add_back(c);
|
|
assert (deq.size() == 1u);
|
|
deq.add_front(b);
|
|
assert (deq.size() == 2u);
|
|
deq.add_back(d);
|
|
assert (deq.size() == 3u);
|
|
deq.add_front(a);
|
|
assert (deq.size() == 4u);
|
|
assert (deq.get(0) == a);
|
|
assert (deq.get(1) == b);
|
|
assert (deq.get(2) == c);
|
|
assert (deq.get(3) == d);
|
|
}
|
|
|
|
type EqFn<T> = fn@(T, T) -> bool;
|
|
|
|
fn test_parameterized<T: Copy Owned>(
|
|
e: EqFn<T>, a: T, b: T, c: T, d: T) {
|
|
|
|
let deq: deque::Deque<T> = deque::create::<T>();
|
|
assert (deq.size() == 0u);
|
|
deq.add_front(a);
|
|
deq.add_front(b);
|
|
deq.add_back(c);
|
|
assert (deq.size() == 3u);
|
|
deq.add_back(d);
|
|
assert (deq.size() == 4u);
|
|
assert (e(deq.peek_front(), b));
|
|
assert (e(deq.peek_back(), d));
|
|
assert (e(deq.pop_front(), b));
|
|
assert (e(deq.pop_back(), d));
|
|
assert (e(deq.pop_back(), c));
|
|
assert (e(deq.pop_back(), a));
|
|
assert (deq.size() == 0u);
|
|
deq.add_back(c);
|
|
assert (deq.size() == 1u);
|
|
deq.add_front(b);
|
|
assert (deq.size() == 2u);
|
|
deq.add_back(d);
|
|
assert (deq.size() == 3u);
|
|
deq.add_front(a);
|
|
assert (deq.size() == 4u);
|
|
assert (e(deq.get(0), a));
|
|
assert (e(deq.get(1), b));
|
|
assert (e(deq.get(2), c));
|
|
assert (e(deq.get(3), d));
|
|
}
|
|
|
|
enum Taggy { One(int), Two(int, int), Three(int, int, int), }
|
|
|
|
enum Taggypar<T> {
|
|
Onepar(int), Twopar(int, int), Threepar(int, int, int),
|
|
}
|
|
|
|
type RecCy = {x: int, y: int, t: Taggy};
|
|
|
|
#[test]
|
|
fn test() {
|
|
fn inteq(&&a: int, &&b: int) -> bool { return a == b; }
|
|
fn intboxeq(&&a: @int, &&b: @int) -> bool { return a == b; }
|
|
fn taggyeq(a: Taggy, b: Taggy) -> bool {
|
|
match a {
|
|
One(a1) => match b {
|
|
One(b1) => return a1 == b1,
|
|
_ => return false
|
|
},
|
|
Two(a1, a2) => match b {
|
|
Two(b1, b2) => return a1 == b1 && a2 == b2,
|
|
_ => return false
|
|
},
|
|
Three(a1, a2, a3) => match b {
|
|
Three(b1, b2, b3) => return a1 == b1 && a2 == b2 && a3 == b3,
|
|
_ => return false
|
|
}
|
|
}
|
|
}
|
|
fn taggypareq<T>(a: Taggypar<T>, b: Taggypar<T>) -> bool {
|
|
match a {
|
|
Onepar::<T>(a1) => match b {
|
|
Onepar::<T>(b1) => return a1 == b1,
|
|
_ => return false
|
|
},
|
|
Twopar::<T>(a1, a2) => match b {
|
|
Twopar::<T>(b1, b2) => return a1 == b1 && a2 == b2,
|
|
_ => return false
|
|
},
|
|
Threepar::<T>(a1, a2, a3) => match b {
|
|
Threepar::<T>(b1, b2, b3) => {
|
|
return a1 == b1 && a2 == b2 && a3 == b3
|
|
}
|
|
_ => return false
|
|
}
|
|
}
|
|
}
|
|
fn reccyeq(a: RecCy, b: RecCy) -> bool {
|
|
return a.x == b.x && a.y == b.y && taggyeq(a.t, b.t);
|
|
}
|
|
debug!("*** test boxes");
|
|
test_boxes(@5, @72, @64, @175);
|
|
debug!("*** end test boxes");
|
|
debug!("test parameterized: int");
|
|
let eq1: EqFn<int> = inteq;
|
|
test_parameterized::<int>(eq1, 5, 72, 64, 175);
|
|
debug!("*** test parameterized: @int");
|
|
let eq2: EqFn<@int> = intboxeq;
|
|
test_parameterized::<@int>(eq2, @5, @72, @64, @175);
|
|
debug!("*** end test parameterized @int");
|
|
debug!("test parameterized: taggy");
|
|
let eq3: EqFn<Taggy> = taggyeq;
|
|
test_parameterized::<Taggy>(eq3, One(1), Two(1, 2), Three(1, 2, 3),
|
|
Two(17, 42));
|
|
|
|
debug!("*** test parameterized: taggypar<int>");
|
|
let eq4: EqFn<Taggypar<int>> = |x,y| taggypareq::<int>(x, y);
|
|
test_parameterized::<Taggypar<int>>(eq4, Onepar::<int>(1),
|
|
Twopar::<int>(1, 2),
|
|
Threepar::<int>(1, 2, 3),
|
|
Twopar::<int>(17, 42));
|
|
debug!("*** end test parameterized: taggypar::<int>");
|
|
|
|
debug!("*** test parameterized: reccy");
|
|
let reccy1: RecCy = {x: 1, y: 2, t: One(1)};
|
|
let reccy2: RecCy = {x: 345, y: 2, t: Two(1, 2)};
|
|
let reccy3: RecCy = {x: 1, y: 777, t: Three(1, 2, 3)};
|
|
let reccy4: RecCy = {x: 19, y: 252, t: Two(17, 42)};
|
|
let eq5: EqFn<RecCy> = reccyeq;
|
|
test_parameterized::<RecCy>(eq5, reccy1, reccy2, reccy3, reccy4);
|
|
debug!("*** end test parameterized: reccy");
|
|
debug!("*** done");
|
|
}
|
|
}
|