aboutsummaryrefslogtreecommitdiff
path: root/src/lib/Vec.rs
diff options
context:
space:
mode:
authorMarijn Haverbeke <[email protected]>2011-05-06 22:13:13 +0200
committerMarijn Haverbeke <[email protected]>2011-05-06 22:51:19 +0200
commita3ec0b1f643d00b9418e4884bd7caa07bf052201 (patch)
tree82000510ac9c9cf3f0c7cf4ae5f3c6b123b559cb /src/lib/Vec.rs
parentRegister new snapshots. (diff)
downloadrust-a3ec0b1f643d00b9418e4884bd7caa07bf052201.tar.xz
rust-a3ec0b1f643d00b9418e4884bd7caa07bf052201.zip
Rename std modules to be camelcased
(Have fun mergining your stuff with this.)
Diffstat (limited to 'src/lib/Vec.rs')
-rw-r--r--src/lib/Vec.rs318
1 files changed, 318 insertions, 0 deletions
diff --git a/src/lib/Vec.rs b/src/lib/Vec.rs
new file mode 100644
index 00000000..28b5d018
--- /dev/null
+++ b/src/lib/Vec.rs
@@ -0,0 +1,318 @@
+import Option.none;
+import Option.some;
+import Util.orb;
+
+type vbuf = rustrt.vbuf;
+
+type operator2[T,U,V] = fn(&T, &U) -> V;
+
+type array[T] = vec[mutable? T];
+
+native "rust" mod rustrt {
+ type vbuf;
+
+ fn vec_buf[T](vec[T] v, uint offset) -> vbuf;
+
+ fn vec_len[T](vec[T] v) -> uint;
+ /**
+ * Sometimes we modify the vec internal data via vec_buf and need to
+ * update the vec's fill length accordingly.
+ */
+ fn vec_len_set[T](vec[T] v, uint n);
+
+ /**
+ * The T in vec_alloc[T, U] is the type of the vec to allocate. The
+ * U is the type of an element in the vec. So to allocate a vec[U] we
+ * want to invoke this as vec_alloc[vec[U], U].
+ */
+ fn vec_alloc[T, U](uint n_elts) -> vec[U];
+ fn vec_alloc_mut[T, U](uint n_elts) -> vec[mutable U];
+
+ fn refcount[T](vec[T] v) -> uint;
+
+ fn vec_print_debug_info[T](vec[T] v);
+
+ fn vec_from_vbuf[T](vbuf v, uint n_elts) -> vec[T];
+
+ fn unsafe_vec_to_mut[T](vec[T] v) -> vec[mutable T];
+}
+
+fn alloc[T](uint n_elts) -> vec[T] {
+ ret rustrt.vec_alloc[vec[T], T](n_elts);
+}
+
+fn alloc_mut[T](uint n_elts) -> vec[mutable T] {
+ ret rustrt.vec_alloc_mut[vec[mutable T], T](n_elts);
+}
+
+fn refcount[T](array[T] v) -> uint {
+ auto r = rustrt.refcount[T](v);
+ if (r == Dbg.const_refcount) {
+ ret r;
+ } else {
+ // -1 because calling this function incremented the refcount.
+ ret r - 1u;
+ }
+}
+
+fn vec_from_vbuf[T](vbuf v, uint n_elts) -> vec[T] {
+ ret rustrt.vec_from_vbuf[T](v, n_elts);
+}
+
+// FIXME: Remove me; this is a botch to get around rustboot's bad typechecker.
+fn empty[T]() -> vec[T] {
+ ret alloc[T](0u);
+}
+
+// FIXME: Remove me; this is a botch to get around rustboot's bad typechecker.
+fn empty_mut[T]() -> vec[mutable T] {
+ ret alloc_mut[T](0u);
+}
+
+type init_op[T] = fn(uint i) -> T;
+
+fn init_fn[T](&init_op[T] op, uint n_elts) -> vec[T] {
+ let vec[T] v = alloc[T](n_elts);
+ let uint i = 0u;
+ while (i < n_elts) {
+ v += vec(op(i));
+ i += 1u;
+ }
+ ret v;
+}
+
+fn init_fn_mut[T](&init_op[T] op, uint n_elts) -> vec[mutable T] {
+ let vec[mutable T] v = alloc_mut[T](n_elts);
+ let uint i = 0u;
+ while (i < n_elts) {
+ v += vec(mutable op(i));
+ i += 1u;
+ }
+ ret v;
+}
+
+fn init_elt[T](&T t, uint n_elts) -> vec[T] {
+ /**
+ * FIXME (issue #81): should be:
+ *
+ * fn elt_op[T](&T x, uint i) -> T { ret x; }
+ * let init_op[T] inner = bind elt_op[T](t, _);
+ * ret init_fn[T](inner, n_elts);
+ */
+ let vec[T] v = alloc[T](n_elts);
+ let uint i = n_elts;
+ while (i > 0u) {
+ i -= 1u;
+ v += vec(t);
+ }
+ ret v;
+}
+
+fn init_elt_mut[T](&T t, uint n_elts) -> vec[mutable T] {
+ let vec[mutable T] v = alloc_mut[T](n_elts);
+ let uint i = n_elts;
+ while (i > 0u) {
+ i -= 1u;
+ v += vec(mutable t);
+ }
+ ret v;
+}
+
+fn buf[T](array[T] v) -> vbuf {
+ ret rustrt.vec_buf[T](v, 0u);
+}
+
+fn len[T](array[T] v) -> uint {
+ ret rustrt.vec_len[T](v);
+}
+
+fn len_set[T](array[T] v, uint n) {
+ rustrt.vec_len_set[T](v, n);
+}
+
+fn buf_off[T](array[T] v, uint offset) -> vbuf {
+ assert (offset < len[T](v));
+ ret rustrt.vec_buf[T](v, offset);
+}
+
+fn print_debug_info[T](array[T] v) {
+ rustrt.vec_print_debug_info[T](v);
+}
+
+// Returns the last element of v.
+fn last[T](array[T] v) -> Option.t[T] {
+ auto l = len[T](v);
+ if (l == 0u) {
+ ret none[T];
+ }
+ ret some[T](v.(l - 1u));
+}
+
+// Returns elements from [start..end) from v.
+
+fn slice[T](array[T] v, uint start, uint end) -> vec[T] {
+ assert (start <= end);
+ assert (end <= len[T](v));
+ auto result = alloc[T](end - start);
+ let uint i = start;
+ while (i < end) {
+ result += vec(v.(i));
+ i += 1u;
+ }
+ ret result;
+}
+
+fn shift[T](&mutable array[T] v) -> T {
+ auto ln = len[T](v);
+ assert (ln > 0u);
+ auto e = v.(0);
+ v = slice[T](v, 1u, ln);
+ ret e;
+}
+
+fn pop[T](&mutable array[T] v) -> T {
+ auto ln = len[T](v);
+ assert (ln > 0u);
+ ln -= 1u;
+ auto e = v.(ln);
+ v = slice[T](v, 0u, ln);
+ ret e;
+}
+
+fn push[T](&mutable array[T] v, &T t) {
+ v += vec(t);
+}
+
+fn unshift[T](&mutable array[T] v, &T t) {
+ auto res = alloc[T](len[T](v) + 1u);
+ res += vec(t);
+ res += v;
+ v = res;
+}
+
+fn grow[T](&array[T] v, uint n, &T initval) {
+ let uint i = n;
+ while (i > 0u) {
+ i -= 1u;
+ v += vec(initval);
+ }
+}
+
+fn grow_set[T](&vec[mutable T] v, uint index, &T initval, &T val) {
+ auto length = Vec.len(v);
+ if (index >= length) {
+ grow(v, index - length + 1u, initval);
+ }
+ v.(index) = val;
+}
+
+fn map[T, U](&Option.operator[T,U] f, &array[T] v) -> vec[U] {
+ let vec[U] u = alloc[U](len[T](v));
+ for (T ve in v) {
+ u += vec(f(ve));
+ }
+ ret u;
+}
+
+fn map2[T,U,V](&operator2[T,U,V] f, &array[T] v0, &array[U] v1) -> vec[V] {
+ auto v0_len = len[T](v0);
+ if (v0_len != len[U](v1)) {
+ fail;
+ }
+
+ let vec[V] u = alloc[V](v0_len);
+ auto i = 0u;
+ while (i < v0_len) {
+ u += vec(f(v0.(i), v1.(i)));
+ i += 1u;
+ }
+
+ ret u;
+}
+
+fn find[T](fn (&T) -> bool f, &array[T] v) -> Option.t[T] {
+ for (T elt in v) {
+ if (f(elt)) {
+ ret some[T](elt);
+ }
+ }
+
+ ret none[T];
+}
+
+fn foldl[T, U](fn (&U, &T) -> U p, &U z, &vec[T] v) -> U {
+ auto sz = len[T](v);
+
+ if (sz == 0u) {
+ ret z;
+ }
+ else {
+ auto rest = slice[T](v, 1u, sz);
+
+ ret (p(foldl[T,U](p, z, rest), v.(0)));
+ }
+}
+
+fn unzip[T, U](&vec[tup(T, U)] v) -> tup(vec[T], vec[U]) {
+ auto sz = len[tup(T, U)](v);
+
+ if (sz == 0u) {
+ ret tup(alloc[T](0u), alloc[U](0u));
+ }
+ else {
+ auto rest = slice[tup(T, U)](v, 1u, sz);
+ auto tl = unzip[T, U](rest);
+ auto a = vec(v.(0)._0);
+ auto b = vec(v.(0)._1);
+ ret tup(a + tl._0, b + tl._1);
+ }
+}
+
+fn or(&vec[bool] v) -> bool {
+ auto f = orb;
+ ret Vec.foldl[bool, bool](f, false, v);
+}
+
+fn clone[T](&vec[T] v) -> vec[T] {
+ ret slice[T](v, 0u, len[T](v));
+}
+
+fn plus_option[T](&vec[T] v, &Option.t[T] o) -> () {
+ alt (o) {
+ case (none[T]) {}
+ case (some[T](?x)) { v += vec(x); }
+ }
+}
+
+fn cat_options[T](&vec[Option.t[T]] v) -> vec[T] {
+ let vec[T] res = vec();
+
+ for (Option.t[T] o in v) {
+ alt (o) {
+ case (none[T]) { }
+ case (some[T](?t)) {
+ res += vec(t);
+ }
+ }
+ }
+
+ ret res;
+}
+
+// TODO: Remove in favor of built-in "freeze" operation when it's implemented.
+fn freeze[T](vec[mutable T] v) -> vec[T] {
+ let vec[T] result = vec();
+ for (T elem in v) {
+ result += vec(elem);
+ }
+ ret result;
+}
+
+// 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: