vec.rs 55.1 KB
Newer Older
B
Brian Anderson 已提交
1 2
#[doc = "Vectors"];

3 4 5
import option::{some, none};
import ptr::addr_of;

6 7 8 9 10
export init_op;
export is_empty;
export is_not_empty;
export same_length;
export reserve;
11
export reserve_at_least;
B
Brian Anderson 已提交
12
export capacity;
13
export len;
14 15
export from_fn;
export from_elem;
16 17 18 19
export to_mut;
export from_mut;
export head;
export tail;
20
export tailn;
21 22 23 24
export init;
export last;
export last_opt;
export slice;
25
export view;
26 27 28 29 30 31 32 33 34 35 36
export split;
export splitn;
export rsplit;
export rsplitn;
export shift;
export pop;
export push;
export grow;
export grow_fn;
export grow_set;
export map;
37
export mapi;
38
export map2;
39
export flat_map;
40 41 42 43 44 45 46 47 48
export filter_map;
export filter;
export concat;
export connect;
export foldl;
export foldr;
export any;
export any2;
export all;
E
Eric Holk 已提交
49
export alli;
50 51 52 53
export all2;
export contains;
export count;
export find;
54
export find_between;
55
export rfind;
56
export rfind_between;
57
export position_elem;
58
export position;
59
export position_between;
60
export position_elem;
61
export rposition;
62
export rposition_between;
63 64 65 66 67
export unzip;
export zip;
export swap;
export reverse;
export reversed;
68
export iter, iter_between, each, eachi;
69 70 71 72 73 74 75 76
export iter2;
export iteri;
export riter;
export riteri;
export permute;
export windowed;
export as_buf;
export as_mut_buf;
77
export unpack_slice;
78 79
export unsafe;
export u8;
80
export extensions;
81

82 83 84 85
#[abi = "cdecl"]
native mod rustrt {
    fn vec_reserve_shared<T>(t: *sys::type_desc,
                             &v: [const T],
86
                             n: libc::size_t);
87 88
    fn vec_from_buf_shared<T>(t: *sys::type_desc,
                              ptr: *T,
89
                              count: libc::size_t) -> [T];
90 91
}

B
Brian Anderson 已提交
92
#[doc = "A function used to initialize the elements of a vector"]
N
Niko Matsakis 已提交
93
type init_op<T> = fn(uint) -> T;
94

B
Brian Anderson 已提交
95
#[doc = "Returns true if a vector contains no elements"]
96
pure fn is_empty<T>(v: [const T]) -> bool {
97
    len(v) == 0u
98 99
}

B
Brian Anderson 已提交
100
#[doc = "Returns true if a vector contains some elements"]
101 102
pure fn is_not_empty<T>(v: [const T]) -> bool { ret !is_empty(v); }

B
Brian Anderson 已提交
103
#[doc = "Returns true if two vectors have the same length"]
104
pure fn same_length<T, U>(xs: [const T], ys: [const U]) -> bool {
105 106 107
    vec::len(xs) == vec::len(ys)
}

B
Brian Anderson 已提交
108
#[doc = "
109
Reserves capacity for exactly `n` elements in the given vector.
110 111 112 113

If the capacity for `v` is already equal to or greater than the requested
capacity, then no action is taken.

B
Brian Anderson 已提交
114
# Arguments
115

B
Brian Anderson 已提交
116 117 118
* v - A vector
* n - The number of elements to reserve space for
"]
119
fn reserve<T>(&v: [const T], n: uint) {
120 121 122 123
    // Only make the (slow) call into the runtime if we have to
    if capacity(v) < n {
        rustrt::vec_reserve_shared(sys::get_type_desc::<T>(), v, n);
    }
124 125
}

126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144
#[doc = "
Reserves capacity for at least `n` elements in the given vector.

This function will over-allocate in order to amortize the allocation costs
in scenarios where the caller may need to repeatedly reserve additional
space.

If the capacity for `v` is already equal to or greater than the requested
capacity, then no action is taken.

# Arguments

* v - A vector
* n - The number of elements to reserve space for
"]
fn reserve_at_least<T>(&v: [const T], n: uint) {
    reserve(v, uint::next_power_of_two(n));
}

B
Brian Anderson 已提交
145 146 147
#[doc = "
Returns the number of elements the vector can hold without reallocating
"]
148
#[inline(always)]
B
Brian Anderson 已提交
149 150 151 152 153
fn capacity<T>(&&v: [const T]) -> uint unsafe {
    let repr: **unsafe::vec_repr = ::unsafe::reinterpret_cast(addr_of(v));
    (**repr).alloc / sys::size_of::<T>()
}

B
Brian Anderson 已提交
154
#[doc = "Returns the length of a vector"]
155
#[inline(always)]
156 157
pure fn len<T>(&&v: [const T]/&) -> uint unsafe {
    unpack_slice(v) {|_p, len| len}
158
}
159

B
Brian Anderson 已提交
160
#[doc = "
161 162 163 164
Creates and initializes an immutable vector.

Creates an immutable vector of size `n_elts` and initializes the elements
to the value returned by the function `op`.
B
Brian Anderson 已提交
165
"]
166
fn from_fn<T>(n_elts: uint, op: init_op<T>) -> [T] {
167
    let mut v = [];
168
    reserve(v, n_elts);
169
    let mut i: uint = 0u;
170 171 172 173
    while i < n_elts { v += [op(i)]; i += 1u; }
    ret v;
}

B
Brian Anderson 已提交
174
#[doc = "
175 176 177 178
Creates and initializes an immutable vector.

Creates an immutable vector of size `n_elts` and initializes the elements
to the value `t`.
B
Brian Anderson 已提交
179
"]
180
fn from_elem<T: copy>(n_elts: uint, t: T) -> [T] {
181
    let mut v = [];
182
    reserve(v, n_elts);
183
    let mut i: uint = 0u;
184 185 186 187
    while i < n_elts { v += [t]; i += 1u; }
    ret v;
}

G
Graydon Hoare 已提交
188 189
#[doc = "Produces a mut vector from an immutable vector."]
fn to_mut<T>(+v: [T]) -> [mut T] unsafe {
M
Marijn Haverbeke 已提交
190
    let r = ::unsafe::reinterpret_cast(v);
191
    ::unsafe::forget(v);
M
Marijn Haverbeke 已提交
192
    r
193 194
}

G
Graydon Hoare 已提交
195 196
#[doc = "Produces an immutable vector from a mut vector."]
fn from_mut<T>(+v: [mut T]) -> [T] unsafe {
M
Marijn Haverbeke 已提交
197
    let r = ::unsafe::reinterpret_cast(v);
198
    ::unsafe::forget(v);
M
Marijn Haverbeke 已提交
199
    r
200 201 202 203
}

// Accessors

B
Brian Anderson 已提交
204
#[doc = "Returns the first element of a vector"]
205
pure fn head<T: copy>(v: [const T]) -> T { v[0] }
206

B
Brian Anderson 已提交
207
#[doc = "Returns all but the first element of a vector"]
208
fn tail<T: copy>(v: [const T]) -> [T] {
209 210 211
    ret slice(v, 1u, len(v));
}

B
Brian Anderson 已提交
212
#[doc = "Returns all but the first `n` elements of a vector"]
213
fn tailn<T: copy>(v: [const T], n: uint) -> [T] {
214 215 216
    slice(v, n, len(v))
}

B
Brian Anderson 已提交
217
#[doc = "Returns all but the last elemnt of a vector"]
218
fn init<T: copy>(v: [const T]) -> [T] {
219 220 221 222
    assert len(v) != 0u;
    slice(v, 0u, len(v) - 1u)
}

B
Brian Anderson 已提交
223
#[doc = "
224
Returns the last element of a `v`, failing if the vector is empty.
B
Brian Anderson 已提交
225
"]
226 227 228
pure fn last<T: copy>(v: [const T]) -> T {
    if len(v) == 0u { fail "last_unsafe: empty vector" }
    v[len(v) - 1u]
229 230
}

B
Brian Anderson 已提交
231
#[doc = "
232 233
Returns some(x) where `x` is the last element of a vector `v`,
or none if the vector is empty.
B
Brian Anderson 已提交
234
"]
235
pure fn last_opt<T: copy>(v: [const T]) -> option<T> {
B
Brian Anderson 已提交
236
   if len(v) == 0u { ret none; }
237
    some(v[len(v) - 1u])
T
Tim Chevalier 已提交
238
}
239

B
Brian Anderson 已提交
240
#[doc = "Returns a copy of the elements from [`start`..`end`) from `v`."]
241
fn slice<T: copy>(v: [const T], start: uint, end: uint) -> [T] {
242 243
    assert (start <= end);
    assert (end <= len(v));
244
    let mut result = [];
245
    reserve(result, end - start);
246
    let mut i = start;
247 248 249 250
    while i < end { result += [v[i]]; i += 1u; }
    ret result;
}

251 252 253 254 255 256 257 258 259 260 261 262
#[doc = "Return a slice that points into another slice."]
fn view<T: copy>(v: [const T]/&, start: uint, end: uint) -> [T]/&a {
    assert (start <= end);
    assert (end <= len(v));
    unpack_slice(v) {|p, _len|
        unsafe {
            ::unsafe::reinterpret_cast(
                (ptr::offset(p, start), (end - start) * sys::size_of::<T>()))
        }
    }
}

B
Brian Anderson 已提交
263
#[doc = "
264
Split the vector `v` by applying each element against the predicate `f`.
B
Brian Anderson 已提交
265
"]
266
fn split<T: copy>(v: [const T], f: fn(T) -> bool) -> [[T]] {
267 268 269
    let ln = len(v);
    if (ln == 0u) { ret [] }

270 271
    let mut start = 0u;
    let mut result = [];
272
    while start < ln {
273
        alt position_between(v, start, ln, f) {
274 275 276 277 278 279 280 281 282 283 284
          none { break }
          some(i) {
            push(result, slice(v, start, i));
            start = i + 1u;
          }
        }
    }
    push(result, slice(v, start, ln));
    result
}

B
Brian Anderson 已提交
285
#[doc = "
286 287
Split the vector `v` by applying each element against the predicate `f` up
to `n` times.
B
Brian Anderson 已提交
288
"]
289
fn splitn<T: copy>(v: [const T], n: uint, f: fn(T) -> bool) -> [[T]] {
290 291 292
    let ln = len(v);
    if (ln == 0u) { ret [] }

293 294 295
    let mut start = 0u;
    let mut count = n;
    let mut result = [];
296
    while start < ln && count > 0u {
297
        alt position_between(v, start, ln, f) {
298 299 300 301 302 303 304 305 306 307 308 309 310
          none { break }
          some(i) {
            push(result, slice(v, start, i));
            // Make sure to skip the separator.
            start = i + 1u;
            count -= 1u;
          }
        }
    }
    push(result, slice(v, start, ln));
    result
}

B
Brian Anderson 已提交
311
#[doc = "
312 313
Reverse split the vector `v` by applying each element against the predicate
`f`.
B
Brian Anderson 已提交
314
"]
315
fn rsplit<T: copy>(v: [const T], f: fn(T) -> bool) -> [[T]] {
316 317 318
    let ln = len(v);
    if (ln == 0u) { ret [] }

319 320
    let mut end = ln;
    let mut result = [];
321
    while end > 0u {
322
        alt rposition_between(v, 0u, end, f) {
323 324 325 326 327 328 329 330 331 332 333
          none { break }
          some(i) {
            push(result, slice(v, i + 1u, end));
            end = i;
          }
        }
    }
    push(result, slice(v, 0u, end));
    reversed(result)
}

B
Brian Anderson 已提交
334
#[doc = "
335 336
Reverse split the vector `v` by applying each element against the predicate
`f` up to `n times.
B
Brian Anderson 已提交
337
"]
338
fn rsplitn<T: copy>(v: [const T], n: uint, f: fn(T) -> bool) -> [[T]] {
339 340 341
    let ln = len(v);
    if (ln == 0u) { ret [] }

342 343 344
    let mut end = ln;
    let mut count = n;
    let mut result = [];
345
    while end > 0u && count > 0u {
346
        alt rposition_between(v, 0u, end, f) {
347 348 349 350 351 352 353 354 355 356 357 358
          none { break }
          some(i) {
            push(result, slice(v, i + 1u, end));
            // Make sure to skip the separator.
            end = i;
            count -= 1u;
          }
        }
    }
    push(result, slice(v, 0u, end));
    reversed(result)
}
359 360 361

// Mutators

B
Brian Anderson 已提交
362
#[doc = "Removes the first element from a vector and return it"]
363
fn shift<T: copy>(&v: [T]) -> T {
364 365 366 367 368 369 370
    let ln = len::<T>(v);
    assert (ln > 0u);
    let e = v[0];
    v = slice::<T>(v, 1u, ln);
    ret e;
}

B
Brian Anderson 已提交
371 372
#[doc = "Prepend an element to a vector"]
fn unshift<T: copy>(&v: [const T], +t: T) {
373 374 375
    // n.b.---for most callers, using unshift() ought not to type check, but
    // it does. It's because the type system is unaware of the mutability of
    // `v` and so allows the vector to be covariant.
B
Brian Anderson 已提交
376 377 378
    v = [const t] + v;
}

B
Brian Anderson 已提交
379
#[doc = "Remove the last element from a vector and return it"]
M
Marijn Haverbeke 已提交
380
fn pop<T>(&v: [const T]) -> T unsafe {
381
    let ln = len(v);
M
Marijn Haverbeke 已提交
382 383
    assert ln > 0u;
    let valptr = ptr::mut_addr_of(v[ln - 1u]);
384
    let val <- *valptr;
M
Marijn Haverbeke 已提交
385
    unsafe::set_len(v, ln - 1u);
386
    val
387 388
}

B
Brian Anderson 已提交
389
#[doc = "Append an element to a vector"]
390
fn push<T>(&v: [const T], +initval: T) {
B
Brian Anderson 已提交
391
    v += [initval];
E
Erick Tryzelaar 已提交
392 393
}

394 395 396

// Appending

B
Brian Anderson 已提交
397
#[doc = "
398 399
Expands a vector in place, initializing the new elements to a given value

B
Brian Anderson 已提交
400
# Arguments
401

B
Brian Anderson 已提交
402 403 404 405
* v - The vector to grow
* n - The number of elements to add
* initval - The value for the new elements
"]
406
fn grow<T: copy>(&v: [const T], n: uint, initval: T) {
407
    reserve_at_least(v, len(v) + n);
408
    let mut i: uint = 0u;
409 410 411
    while i < n { v += [initval]; i += 1u; }
}

B
Brian Anderson 已提交
412 413 414
#[doc = "
Expands a vector in place, initializing the new elements to the result of
a function
415

416
Function `init_op` is called `n` times with the values [0..`n`)
417

B
Brian Anderson 已提交
418
# Arguments
419

B
Brian Anderson 已提交
420 421 422 423 424
* v - The vector to grow
* n - The number of elements to add
* init_op - A function to call to retreive each appended element's
            value
"]
425
fn grow_fn<T>(&v: [const T], n: uint, op: init_op<T>) {
426
    reserve_at_least(v, len(v) + n);
427
    let mut i: uint = 0u;
428 429 430
    while i < n { v += [op(i)]; i += 1u; }
}

B
Brian Anderson 已提交
431
#[doc = "
432 433 434 435 436 437
Sets the value of a vector element at a given index, growing the vector as
needed

Sets the element at position `index` to `val`. If `index` is past the end
of the vector, expands the vector by replicating `initval` to fill the
intervening space.
B
Brian Anderson 已提交
438
"]
G
Graydon Hoare 已提交
439
fn grow_set<T: copy>(&v: [mut T], index: uint, initval: T, val: T) {
440
    if index >= len(v) { grow(v, index - len(v) + 1u, initval); }
441 442 443 444 445 446
    v[index] = val;
}


// Functional utilities

B
Brian Anderson 已提交
447
#[doc = "
448
Apply a function to each element of a vector and return the results
B
Brian Anderson 已提交
449
"]
450
fn map<T, U>(v: [const T]/&, f: fn(T) -> U) -> [U] {
451
    let mut result = [];
452
    reserve(result, len(v));
453
    for each(v) {|elem| result += [f(elem)]; }
454 455 456
    ret result;
}

457 458 459 460 461 462 463 464 465 466
#[doc = "
Apply a function to each element of a vector and return the results
"]
fn mapi<T, U>(v: [const T]/&, f: fn(uint, T) -> U) -> [U] {
    let mut result = [];
    reserve(result, len(v));
    for eachi(v) {|i, elem| result += [f(i, elem)]; }
    ret result;
}

B
Brian Anderson 已提交
467
#[doc = "
468
Apply a function to each element of a vector and return a concatenation
B
Brian Anderson 已提交
469 470
of each result vector
"]
471
fn flat_map<T, U>(v: [T], f: fn(T) -> [U]) -> [U] {
472
    let mut result = [];
473
    for each(v) {|elem| result += f(elem); }
474 475 476
    ret result;
}

B
Brian Anderson 已提交
477
#[doc = "
478
Apply a function to each pair of elements and return the results
B
Brian Anderson 已提交
479
"]
480 481
fn map2<T: copy, U: copy, V>(v0: [const T], v1: [const U],
                             f: fn(T, U) -> V) -> [V] {
482 483
    let v0_len = len(v0);
    if v0_len != len(v1) { fail; }
484 485
    let mut u: [V] = [];
    let mut i = 0u;
486 487 488 489
    while i < v0_len { u += [f(copy v0[i], copy v1[i])]; i += 1u; }
    ret u;
}

B
Brian Anderson 已提交
490
#[doc = "
491 492 493 494
Apply a function to each element of a vector and return the results

If function `f` returns `none` then that element is excluded from
the resulting vector.
B
Brian Anderson 已提交
495
"]
496
fn filter_map<T, U: copy>(v: [T], f: fn(T) -> option<U>)
497
    -> [U] {
498
    let mut result = [];
499
    for each(v) {|elem|
500
        alt f(elem) {
501
          none {/* no-op */ }
502 503 504 505 506 507
          some(result_elem) { result += [result_elem]; }
        }
    }
    ret result;
}

B
Brian Anderson 已提交
508
#[doc = "
509 510 511 512 513
Construct a new vector from the elements of a vector for which some predicate
holds.

Apply function `f` to each element of `v` and return a vector containing
only those elements for which `f` returned true.
B
Brian Anderson 已提交
514
"]
515
fn filter<T: copy>(v: [const T], f: fn(T) -> bool) -> [T] {
516
    let mut result = [];
517
    for each(v) {|elem|
518 519 520 521 522
        if f(elem) { result += [elem]; }
    }
    ret result;
}

B
Brian Anderson 已提交
523 524
#[doc = "
Concatenate a vector of vectors.
525

B
Brian Anderson 已提交
526 527
Flattens a vector of vectors of T into a single vector of T.
"]
528
fn concat<T: copy>(v: [const [const T]]) -> [T] {
529
    let mut r = [];
530
    for each(v) {|inner| r += inner; }
531
    ret r;
532 533
}

B
Brian Anderson 已提交
534
#[doc = "
535
Concatenate a vector of vectors, placing a given separator between each
B
Brian Anderson 已提交
536
"]
537
fn connect<T: copy>(v: [const [const T]], sep: T) -> [T] {
538
    let mut r: [T] = [];
539
    let mut first = true;
540
    for each(v) {|inner|
541
        if first { first = false; } else { push(r, sep); }
542
        r += inner;
543
    }
544
    ret r;
545 546
}

B
Brian Anderson 已提交
547
#[doc = "Reduce a vector from left to right"]
N
Niko Matsakis 已提交
548
fn foldl<T: copy, U>(z: T, v: [const U], p: fn(T, U) -> T) -> T {
549
    let mut accum = z;
550 551 552 553 554 555
    iter(v) { |elt|
        accum = p(accum, elt);
    }
    ret accum;
}

B
Brian Anderson 已提交
556
#[doc = "Reduce a vector from right to left"]
557
fn foldr<T, U: copy>(v: [const T]/&, z: U, p: fn(T, U) -> U) -> U {
558
    let mut accum = z;
559 560 561 562 563 564
    riter(v) { |elt|
        accum = p(elt, accum);
    }
    ret accum;
}

B
Brian Anderson 已提交
565
#[doc = "
566 567 568
Return true if a predicate matches any elements

If the vector contains no elements then false is returned.
B
Brian Anderson 已提交
569
"]
570
fn any<T>(v: [const T]/&, f: fn(T) -> bool) -> bool {
571
    for each(v) {|elem| if f(elem) { ret true; } }
572 573 574
    ret false;
}

B
Brian Anderson 已提交
575
#[doc = "
576 577 578
Return true if a predicate matches any elements in both vectors.

If the vectors contains no elements then false is returned.
B
Brian Anderson 已提交
579
"]
580
fn any2<T, U>(v0: [const T]/&, v1: [const U]/&, f: fn(T, U) -> bool) -> bool {
581 582
    let v0_len = len(v0);
    let v1_len = len(v1);
583
    let mut i = 0u;
584 585 586 587 588 589 590
    while i < v0_len && i < v1_len {
        if f(v0[i], v1[i]) { ret true; };
        i += 1u;
    }
    ret false;
}

B
Brian Anderson 已提交
591
#[doc = "
592 593 594
Return true if a predicate matches all elements

If the vector contains no elements then true is returned.
B
Brian Anderson 已提交
595
"]
596
fn all<T>(v: [const T]/&, f: fn(T) -> bool) -> bool {
597
    for each(v) {|elem| if !f(elem) { ret false; } }
598 599 600
    ret true;
}

601 602 603 604 605
#[doc = "
Return true if a predicate matches all elements

If the vector contains no elements then true is returned.
"]
606
fn alli<T>(v: [const T]/&, f: fn(uint, T) -> bool) -> bool {
607 608 609 610
    for eachi(v) {|i, elem| if !f(i, elem) { ret false; } }
    ret true;
}

B
Brian Anderson 已提交
611
#[doc = "
612 613 614
Return true if a predicate matches all elements in both vectors.

If the vectors are not the same size then false is returned.
B
Brian Anderson 已提交
615
"]
616
fn all2<T, U>(v0: [const T]/&, v1: [const U]/&, f: fn(T, U) -> bool) -> bool {
617 618
    let v0_len = len(v0);
    if v0_len != len(v1) { ret false; }
619
    let mut i = 0u;
620 621 622 623
    while i < v0_len { if !f(v0[i], v1[i]) { ret false; }; i += 1u; }
    ret true;
}

B
Brian Anderson 已提交
624
#[doc = "Return true if a vector contains an element with the given value"]
625
fn contains<T>(v: [const T], x: T) -> bool {
626
    for each(v) {|elt| if x == elt { ret true; } }
627 628 629
    ret false;
}

B
Brian Anderson 已提交
630
#[doc = "Returns the number of elements that are equal to a given value"]
631
fn count<T>(v: [const T], x: T) -> uint {
632
    let mut cnt = 0u;
633
    for each(v) {|elt| if x == elt { cnt += 1u; } }
634 635 636
    ret cnt;
}

B
Brian Anderson 已提交
637
#[doc = "
638
Search for the first element that matches a given predicate
639 640 641 642

Apply function `f` to each element of `v`, starting from the first.
When function `f` returns true then an option containing the element
is returned. If `f` matches no elements then none is returned.
B
Brian Anderson 已提交
643
"]
644
fn find<T: copy>(v: [const T], f: fn(T) -> bool) -> option<T> {
645
    find_between(v, 0u, len(v), f)
646 647
}

B
Brian Anderson 已提交
648
#[doc = "
649 650 651 652 653
Search for the first element that matches a given predicate within a range

Apply function `f` to each element of `v` within the range [`start`, `end`).
When function `f` returns true then an option containing the element
is returned. If `f` matches no elements then none is returned.
B
Brian Anderson 已提交
654
"]
655
fn find_between<T: copy>(v: [const T], start: uint, end: uint,
656
                      f: fn(T) -> bool) -> option<T> {
657
    option::map(position_between(v, start, end, f)) { |i| v[i] }
658 659
}

B
Brian Anderson 已提交
660
#[doc = "
661 662 663 664 665
Search for the last element that matches a given predicate

Apply function `f` to each element of `v` in reverse order. When function `f`
returns true then an option containing the element is returned. If `f`
matches no elements then none is returned.
B
Brian Anderson 已提交
666
"]
667
fn rfind<T: copy>(v: [const T], f: fn(T) -> bool) -> option<T> {
668
    rfind_between(v, 0u, len(v), f)
669 670
}

B
Brian Anderson 已提交
671
#[doc = "
672 673 674 675 676
Search for the last element that matches a given predicate within a range

Apply function `f` to each element of `v` in reverse order within the range
[`start`, `end`). When function `f` returns true then an option containing
the element is returned. If `f` matches no elements then none is returned.
B
Brian Anderson 已提交
677
"]
678
fn rfind_between<T: copy>(v: [const T], start: uint, end: uint,
679
                       f: fn(T) -> bool) -> option<T> {
680
    option::map(rposition_between(v, start, end, f)) { |i| v[i] }
681 682
}

B
Brian Anderson 已提交
683
#[doc = "Find the first index containing a matching value"]
684
fn position_elem<T>(v: [const T], x: T) -> option<uint> {
685
    position(v) { |y| x == y }
686 687
}

B
Brian Anderson 已提交
688
#[doc = "
689 690 691 692 693
Find the first index matching some predicate

Apply function `f` to each element of `v`.  When function `f` returns true
then an option containing the index is returned. If `f` matches no elements
then none is returned.
B
Brian Anderson 已提交
694
"]
695
fn position<T>(v: [const T], f: fn(T) -> bool) -> option<uint> {
696
    position_between(v, 0u, len(v), f)
697 698
}

B
Brian Anderson 已提交
699
#[doc = "
700 701 702 703 704
Find the first index matching some predicate within a range

Apply function `f` to each element of `v` between the range [`start`, `end`).
When function `f` returns true then an option containing the index is
returned. If `f` matches no elements then none is returned.
B
Brian Anderson 已提交
705
"]
706
fn position_between<T>(v: [const T], start: uint, end: uint,
707
                    f: fn(T) -> bool) -> option<uint> {
708 709
    assert start <= end;
    assert end <= len(v);
710
    let mut i = start;
711
    while i < end { if f(v[i]) { ret some::<uint>(i); } i += 1u; }
712 713 714
    ret none;
}

B
Brian Anderson 已提交
715
#[doc = "Find the last index containing a matching value"]
716
fn rposition_elem<T>(v: [const T], x: T) -> option<uint> {
717 718 719
    rposition(v) { |y| x == y }
}

B
Brian Anderson 已提交
720
#[doc = "
721
Find the last index matching some predicate
722

723 724 725
Apply function `f` to each element of `v` in reverse order.  When function
`f` returns true then an option containing the index is returned. If `f`
matches no elements then none is returned.
B
Brian Anderson 已提交
726
"]
727
fn rposition<T>(v: [const T], f: fn(T) -> bool) -> option<uint> {
728
    rposition_between(v, 0u, len(v), f)
729 730
}

B
Brian Anderson 已提交
731
#[doc = "
732 733 734 735 736
Find the last index matching some predicate within a range

Apply function `f` to each element of `v` in reverse order between the range
[`start`, `end`). When function `f` returns true then an option containing
the index is returned. If `f` matches no elements then none is returned.
B
Brian Anderson 已提交
737
"]
738
fn rposition_between<T>(v: [const T], start: uint, end: uint,
739
                     f: fn(T) -> bool) -> option<uint> {
740 741
    assert start <= end;
    assert end <= len(v);
742
    let mut i = end;
743 744 745 746
    while i > start {
        if f(v[i - 1u]) { ret some::<uint>(i - 1u); }
        i -= 1u;
    }
747 748 749 750 751 752 753
    ret none;
}

// FIXME: if issue #586 gets implemented, could have a postcondition
// saying the two result lists have the same length -- or, could
// return a nominal record with a constraint saying that, instead of
// returning a tuple (contingent on issue #869)
B
Brian Anderson 已提交
754
#[doc = "
755 756 757 758 759 760
Convert a vector of pairs into a pair of vectors

Returns a tuple containing two vectors where the i-th element of the first
vector contains the first element of the i-th tuple of the input vector,
and the i-th element of the second vector contains the second element
of the i-th tuple of the input vector.
B
Brian Anderson 已提交
761
"]
762
fn unzip<T: copy, U: copy>(v: [const (T, U)]) -> ([T], [U]) {
763
    let mut as = [], bs = [];
764
    for each(v) {|p| let (a, b) = p; as += [a]; bs += [b]; }
765 766 767
    ret (as, bs);
}

B
Brian Anderson 已提交
768
#[doc = "
769 770 771 772
Convert two vectors to a vector of pairs

Returns a vector of tuples, where the i-th tuple contains contains the
i-th elements from each of the input vectors.
B
Brian Anderson 已提交
773
"]
774
fn zip<T: copy, U: copy>(v: [const T], u: [const U]) -> [(T, U)] {
775 776 777
    let mut zipped = [];
    let sz = len(v);
    let mut i = 0u;
778
    assert sz == len(u);
779 780 781 782
    while i < sz { zipped += [(v[i], u[i])]; i += 1u; }
    ret zipped;
}

B
Brian Anderson 已提交
783
#[doc = "
784 785
Swaps two elements in a vector

B
Brian Anderson 已提交
786 787 788 789 790 791
# Arguments

* v  The input vector
* a - The index of the first element
* b - The index of the second element
"]
G
Graydon Hoare 已提交
792
fn swap<T>(v: [mut T], a: uint, b: uint) {
793 794 795
    v[a] <-> v[b];
}

B
Brian Anderson 已提交
796
#[doc = "Reverse the order of elements in a vector, in place"]
G
Graydon Hoare 已提交
797
fn reverse<T>(v: [mut T]) {
798
    let mut i: uint = 0u;
799 800 801 802 803
    let ln = len::<T>(v);
    while i < ln / 2u { v[i] <-> v[ln - i - 1u]; i += 1u; }
}


B
Brian Anderson 已提交
804
#[doc = "Returns a vector with the order of elements reversed"]
805
fn reversed<T: copy>(v: [const T]) -> [T] {
806 807
    let mut rs: [T] = [];
    let mut i = len::<T>(v);
808 809 810 811 812 813
    if i == 0u { ret rs; } else { i -= 1u; }
    while i != 0u { rs += [v[i]]; i -= 1u; }
    rs += [v[0]];
    ret rs;
}

B
Brian Anderson 已提交
814
#[doc = "
815 816 817 818
Iterates over a vector

Iterates over vector `v` and, for each element, calls function `f` with the
element's value.
B
Brian Anderson 已提交
819
"]
820
#[inline(always)]
N
Niko Matsakis 已提交
821
fn iter<T>(v: [const T], f: fn(T)) {
822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837
    iter_between(v, 0u, vec::len(v), f)
}

/*
Function: iter_between

Iterates over a vector

Iterates over vector `v` and, for each element, calls function `f` with the
element's value.

*/
#[inline(always)]
fn iter_between<T>(v: [const T], start: uint, end: uint, f: fn(T)) {
    assert start <= end;
    assert end <= vec::len(v);
838
    unsafe {
839 840 841
        let mut n = end;
        let mut p = ptr::offset(unsafe::to_ptr(v), start);
        while n > start {
842 843 844 845 846
            f(*p);
            p = ptr::offset(p, 1u);
            n -= 1u;
        }
    }
847 848
}

849 850
#[doc = "
Iterates over a vector, with option to break
851 852

Return true to continue, false to break.
853 854
"]
#[inline(always)]
855 856 857 858 859 860 861 862 863
fn each<T>(v: [const T]/&, f: fn(T) -> bool) unsafe {
    vec::unpack_slice(v) {|p, n|
        let mut n = n;
        let mut p = p;
        while n > 0u {
            if !f(*p) { break; }
            p = ptr::offset(p, 1u);
            n -= 1u;
        }
864 865 866 867 868
    }
}

#[doc = "
Iterates over a vector's elements and indices
869 870

Return true to continue, false to break.
871 872
"]
#[inline(always)]
873 874 875 876 877 878 879 880 881
fn eachi<T>(v: [const T]/&, f: fn(uint, T) -> bool) unsafe {
    vec::unpack_slice(v) {|p, n|
        let mut i = 0u;
        let mut p = p;
        while i < n {
            if !f(i, *p) { break; }
            p = ptr::offset(p, 1u);
            i += 1u;
        }
882 883 884
    }
}

885 886 887 888 889 890 891
#[doc = "
Iterates over two vectors simultaneously

# Failure

Both vectors must have the same length
"]
892
#[inline]
893 894
fn iter2<U, T>(v1: [const U], v2: [const T], f: fn(U, T)) {
    assert len(v1) == len(v2);
895
    for uint::range(0u, len(v1)) {|i|
896 897
        f(v1[i], v2[i])
    }
898 899
}

B
Brian Anderson 已提交
900
#[doc = "
901 902 903 904
Iterates over a vector's elements and indexes

Iterates over vector `v` and, for each element, calls function `f` with the
element's value and index.
B
Brian Anderson 已提交
905
"]
906
#[inline(always)]
N
Niko Matsakis 已提交
907
fn iteri<T>(v: [const T], f: fn(uint, T)) {
908 909
    let mut i = 0u;
    let l = len(v);
910 911 912
    while i < l { f(i, v[i]); i += 1u; }
}

B
Brian Anderson 已提交
913
#[doc = "
914 915 916 917
Iterates over a vector in reverse

Iterates over vector `v` and, for each element, calls function `f` with the
element's value.
B
Brian Anderson 已提交
918
"]
919
fn riter<T>(v: [const T]/&, f: fn(T)) {
920
    riteri(v) { |_i, v| f(v) }
921 922
}

B
Brian Anderson 已提交
923
#[doc ="
924 925 926 927
Iterates over a vector's elements and indexes in reverse

Iterates over vector `v` and, for each element, calls function `f` with the
element's value and index.
B
Brian Anderson 已提交
928
"]
929
fn riteri<T>(v: [const T]/&, f: fn(uint, T)) {
930
    let mut i = len(v);
931 932 933 934 935 936
    while 0u < i {
        i -= 1u;
        f(i, v[i]);
    };
}

B
Brian Anderson 已提交
937 938
#[doc = "
Iterate over all permutations of vector `v`.
939

B
Brian Anderson 已提交
940 941 942
Permutations are produced in lexicographic order with respect to the order of
elements in `v` (so if `v` is sorted then the permutations are
lexicographically sorted).
943 944 945

The total number of permutations produced is `len(v)!`.  If `v` contains
repeated elements, then some permutations are repeated.
B
Brian Anderson 已提交
946
"]
947
fn permute<T: copy>(v: [T], put: fn([T])) {
948 949 950 951
  let ln = len(v);
  if ln == 0u {
    put([]);
  } else {
952
    let mut i = 0u;
953 954 955 956 957 958 959 960 961
    while i < ln {
      let elt = v[i];
      let rest = slice(v, 0u, i) + slice(v, i+1u, ln);
      permute(rest) {|permutation| put([elt] + permutation)}
      i += 1u;
    }
  }
}

962
fn windowed <TT: copy> (nn: uint, xx: [const TT]) -> [[TT]] {
963
   let mut ww = [];
964 965 966 967 968 969 970 971 972 973 974 975 976 977 978

   assert 1u <= nn;

   vec::iteri (xx, {|ii, _x|
      let len = vec::len(xx);

      if ii+nn <= len {
         let w = vec::slice ( xx, ii, ii+nn );
         vec::push (ww, w);
      }
   });

   ret ww;
}

B
Brian Anderson 已提交
979 980
#[doc = "
Work with the buffer of a vector.
981

B
Brian Anderson 已提交
982 983 984
Allows for unsafe manipulation of vector contents, which is useful for native
interop.
"]
985 986 987 988
fn as_buf<E,T>(v: [const E], f: fn(*E) -> T) -> T unsafe {
    let buf = unsafe::to_ptr(v); f(buf)
}

G
Graydon Hoare 已提交
989 990
fn as_mut_buf<E,T>(v: [mut E], f: fn(*mut E) -> T) -> T unsafe {
    let buf = unsafe::to_ptr(v) as *mut E; f(buf)
991 992
}

993 994 995
#[doc = "
Work with the buffer and length of a slice.
"]
996
#[inline(always)]
997 998 999 1000 1001 1002
fn unpack_slice<T,U>(s: [const T]/&, f: fn(*T, uint) -> U) -> U unsafe {
    let v : *(*T,uint) = ::unsafe::reinterpret_cast(ptr::addr_of(s));
    let (buf,len) = *v;
    f(buf, len / sys::size_of::<T>())
}

1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031
#[doc = "Extension methods for vectors"]
impl extensions<T> for [const T] {
    #[doc = "Reduce a vector from right to left"]
    #[inline]
    fn foldr<U: copy>(z: U, p: fn(T, U) -> U) -> U { foldr(self, z, p) }
    #[doc = "Returns true if a vector contains no elements"]
    #[inline]
    fn is_empty() -> bool { is_empty(self) }
    #[doc = "Returns true if a vector contains some elements"]
    #[inline]
    fn is_not_empty() -> bool { is_not_empty(self) }
    #[doc = "
    Iterates over a vector

    Iterates over vector `v` and, for each element, calls function `f` with
    the element's value.
    "]
    #[inline]
    fn iter(f: fn(T)) { iter(self, f) }
    #[doc = "
    Iterates over a vector's elements and indexes

    Iterates over vector `v` and, for each element, calls function `f` with
    the element's value and index.
    "]
    #[inline]
    fn iteri(f: fn(uint, T)) { iteri(self, f) }
    #[doc = "Returns the length of a vector"]
    #[inline]
1032
    pure fn len() -> uint { len(self) }
1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076
    #[doc = "
    Find the first index matching some predicate

    Apply function `f` to each element of `v`.  When function `f` returns true
    then an option containing the index is returned. If `f` matches no
    elements then none is returned.
    "]
    #[inline]
    fn position(f: fn(T) -> bool) -> option<uint> { position(self, f) }
    #[doc = "Find the first index containing a matching value"]
    #[inline]
    fn position_elem(x: T) -> option<uint> { position_elem(self, x) }
    #[doc = "
    Iterates over a vector in reverse

    Iterates over vector `v` and, for each element, calls function `f` with
    the element's value.
    "]
    #[inline]
    fn riter(f: fn(T)) { riter(self, f) }
    #[doc ="
    Iterates over a vector's elements and indexes in reverse

    Iterates over vector `v` and, for each element, calls function `f` with
    the element's value and index.
    "]
    #[inline]
    fn riteri(f: fn(uint, T)) { riteri(self, f) }
    #[doc = "
    Find the last index matching some predicate

    Apply function `f` to each element of `v` in reverse order.  When function
    `f` returns true then an option containing the index is returned. If `f`
    matches no elements then none is returned.
    "]
    #[inline]
    fn rposition(f: fn(T) -> bool) -> option<uint> { rposition(self, f) }
    #[doc = "Find the last index containing a matching value"]
    #[inline]
    fn rposition_elem(x: T) -> option<uint> { rposition_elem(self, x) }
}

#[doc = "Extension methods for vectors"]
impl extensions<T: copy> for [const T] {
1077 1078 1079 1080 1081 1082 1083 1084 1085
    #[doc = "
    Construct a new vector from the elements of a vector for which some
    predicate holds.

    Apply function `f` to each element of `v` and return a vector containing
    only those elements for which `f` returned true.
    "]
    #[inline]
    fn filter(f: fn(T) -> bool) -> [T] { filter(self, f) }
1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125
    #[doc = "
    Search for the first element that matches a given predicate

    Apply function `f` to each element of `v`, starting from the first.
    When function `f` returns true then an option containing the element
    is returned. If `f` matches no elements then none is returned.
    "]
    #[inline]
    fn find(f: fn(T) -> bool) -> option<T> { find(self, f) }
    #[doc = "Returns the first element of a vector"]
    #[inline]
    fn head() -> T { head(self) }
    #[doc = "Returns all but the last elemnt of a vector"]
    #[inline]
    fn init() -> [T] { init(self) }
    #[doc = "
    Returns the last element of a `v`, failing if the vector is empty.
    "]
    #[inline]
    fn last() -> T { last(self) }
    #[doc = "
    Search for the last element that matches a given predicate

    Apply function `f` to each element of `v` in reverse order. When function
    `f` returns true then an option containing the element is returned. If `f`
    matches no elements then none is returned.
    "]
    #[inline]
    fn rfind(f: fn(T) -> bool) -> option<T> { rfind(self, f) }
    #[doc = "Returns a copy of the elements from [`start`..`end`) from `v`."]
    #[inline]
    fn slice(start: uint, end: uint) -> [T] { slice(self, start, end) }
    #[doc = "Returns all but the first element of a vector"]
    #[inline]
    fn tail() -> [T] { tail(self) }
}

#[doc = "Extension methods for vectors"]
impl extensions<T> for [T] {
    #[doc = "
1126
    Apply a function to each element of a vector and return the results
1127 1128
    "]
    #[inline]
1129
    fn map<U>(f: fn(T) -> U) -> [U] { map(self, f) }
1130
    #[doc = "
N
Niko Matsakis 已提交
1131 1132 1133 1134
    Apply a function to the index and value of each element in the vector
    and return the results
    "]
    fn mapi<U>(f: fn(uint, T) -> U) -> [U] {
1135
        mapi(self, f)
N
Niko Matsakis 已提交
1136
    }
1137 1138 1139 1140 1141 1142
    #[doc = "Returns true if the function returns true for all elements.

    If the vector is empty, true is returned."]
    fn alli(f: fn(uint, T) -> bool) -> bool {
        alli(self, f)
    }
N
Niko Matsakis 已提交
1143
    #[doc = "
1144 1145
    Apply a function to each element of a vector and return a concatenation
    of each result vector
1146 1147
    "]
    #[inline]
1148
    fn flat_map<U>(f: fn(T) -> [U]) -> [U] { flat_map(self, f) }
1149 1150 1151 1152 1153 1154 1155 1156 1157 1158
    #[doc = "
    Apply a function to each element of a vector and return the results

    If function `f` returns `none` then that element is excluded from
    the resulting vector.
    "]
    #[inline]
    fn filter_map<U: copy>(f: fn(T) -> option<U>) -> [U] {
        filter_map(self, f)
    }
1159
}
1160

B
Brian Anderson 已提交
1161
#[doc = "Unsafe operations"]
1162
mod unsafe {
T
Tim Chevalier 已提交
1163
    // FIXME: This should have crate visibility (#1893 blocks that)
B
Brian Anderson 已提交
1164
    #[doc = "The internal representation of a vector"]
1165 1166 1167 1168 1169 1170
    type vec_repr = {
        box_header: (uint, uint, uint, uint),
        mut fill: uint,
        mut alloc: uint,
        data: u8
    };
1171

B
Brian Anderson 已提交
1172
    #[doc = "
1173 1174
    Constructs a vector from an unsafe pointer to a buffer

B
Brian Anderson 已提交
1175
    # Arguments
1176

B
Brian Anderson 已提交
1177 1178 1179
    * ptr - An unsafe pointer to a buffer of `T`
    * elts - The number of elements in the buffer
    "]
1180
    #[inline(always)]
1181 1182 1183 1184 1185
    unsafe fn from_buf<T>(ptr: *T, elts: uint) -> [T] {
        ret rustrt::vec_from_buf_shared(sys::get_type_desc::<T>(),
                                        ptr, elts);
    }

B
Brian Anderson 已提交
1186
    #[doc = "
1187 1188 1189 1190 1191
    Sets the length of a vector

    This well explicitly set the size of the vector, without actually
    modifing its buffers, so it is up to the caller to ensure that
    the vector is actually the specified size.
B
Brian Anderson 已提交
1192
    "]
1193
    #[inline(always)]
1194 1195 1196 1197 1198
    unsafe fn set_len<T>(&v: [const T], new_len: uint) {
        let repr: **vec_repr = ::unsafe::reinterpret_cast(addr_of(v));
        (**repr).fill = new_len * sys::size_of::<T>();
    }

B
Brian Anderson 已提交
1199
    #[doc = "
1200 1201 1202 1203 1204 1205 1206
    Returns an unsafe pointer to the vector's buffer

    The caller must ensure that the vector outlives the pointer this
    function returns, or else it will end up pointing to garbage.

    Modifying the vector may cause its buffer to be reallocated, which
    would also make any pointers to it invalid.
B
Brian Anderson 已提交
1207
    "]
1208
    #[inline(always)]
1209 1210 1211 1212
    unsafe fn to_ptr<T>(v: [const T]) -> *T {
        let repr: **vec_repr = ::unsafe::reinterpret_cast(addr_of(v));
        ret ::unsafe::reinterpret_cast(addr_of((**repr).data));
    }
1213 1214 1215 1216 1217 1218 1219 1220


    #[doc = "
    Form a slice from a pointer and length (as a number of units, not bytes).
    "]
    #[inline(always)]
    unsafe fn form_slice<T,U>(p: *T, len: uint, f: fn([T]/&) -> U) -> U {
        let pair = (p, len * sys::size_of::<T>());
1221 1222 1223
        // FIXME: should use &blk not &static here, but a snapshot is req'd
        let v : *([T]/&static) =
            ::unsafe::reinterpret_cast(ptr::addr_of(pair));
1224 1225
        f(*v)
    }
1226 1227
}

B
Brian Anderson 已提交
1228
#[doc = "Operations on `[u8]`"]
1229 1230 1231 1232 1233
mod u8 {
    export cmp;
    export lt, le, eq, ne, ge, gt;
    export hash;

B
Brian Anderson 已提交
1234
    #[doc = "Bytewise string comparison"]
1235 1236 1237
    pure fn cmp(&&a: [u8], &&b: [u8]) -> int unsafe {
        let a_len = len(a);
        let b_len = len(b);
1238
        let n = uint::min(a_len, b_len) as libc::size_t;
1239 1240
        let r = libc::memcmp(unsafe::to_ptr(a) as *libc::c_void,
                             unsafe::to_ptr(b) as *libc::c_void, n) as int;
1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252

        if r != 0 { r } else {
            if a_len == b_len {
                0
            } else if a_len < b_len {
                -1
            } else {
                1
            }
        }
    }

B
Brian Anderson 已提交
1253
    #[doc = "Bytewise less than or equal"]
1254 1255
    pure fn lt(&&a: [u8], &&b: [u8]) -> bool { cmp(a, b) < 0 }

B
Brian Anderson 已提交
1256
    #[doc = "Bytewise less than or equal"]
1257 1258
    pure fn le(&&a: [u8], &&b: [u8]) -> bool { cmp(a, b) <= 0 }

B
Brian Anderson 已提交
1259
    #[doc = "Bytewise equality"]
1260 1261
    pure fn eq(&&a: [u8], &&b: [u8]) -> bool unsafe { cmp(a, b) == 0 }

B
Brian Anderson 已提交
1262
    #[doc = "Bytewise inequality"]
1263 1264
    pure fn ne(&&a: [u8], &&b: [u8]) -> bool unsafe { cmp(a, b) != 0 }

B
Brian Anderson 已提交
1265
    #[doc ="Bytewise greater than or equal"]
1266 1267
    pure fn ge(&&a: [u8], &&b: [u8]) -> bool { cmp(a, b) >= 0 }

B
Brian Anderson 已提交
1268
    #[doc = "Bytewise greater than"]
1269 1270
    pure fn gt(&&a: [u8], &&b: [u8]) -> bool { cmp(a, b) > 0 }

B
Brian Anderson 已提交
1271
    #[doc = "String hash function"]
1272
    fn hash(&&s: [u8]) -> uint {
T
Tim Chevalier 已提交
1273 1274 1275
        /* Seems to have been tragically copy/pasted from str.rs,
           or vice versa. But I couldn't figure out how to abstract
           it out. -- tjc */
1276

1277
        let mut u: uint = 5381u;
1278 1279 1280 1281 1282
        vec::iter(s, { |c| u *= 33u; u += c as uint; });
        ret u;
    }
}

1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295
#[cfg(test)]
mod tests {

    fn square(n: uint) -> uint { ret n * n; }

    fn square_ref(&&n: uint) -> uint { ret n * n; }

    pure fn is_three(&&n: uint) -> bool { ret n == 3u; }

    pure fn is_odd(&&n: uint) -> bool { ret n % 2u == 1u; }

    pure fn is_equal(&&x: uint, &&y:uint) -> bool { ret x == y; }

1296
    fn square_if_odd(&&n: uint) -> option<uint> {
1297 1298 1299 1300 1301 1302 1303 1304 1305
        ret if n % 2u == 1u { some(n * n) } else { none };
    }

    fn add(&&x: uint, &&y: uint) -> uint { ret x + y; }

    #[test]
    fn test_unsafe_ptrs() unsafe {
        // Test on-stack copy-from-buf.
        let a = [1, 2, 3];
1306
        let mut ptr = unsafe::to_ptr(a);
1307 1308 1309 1310 1311 1312 1313 1314
        let b = unsafe::from_buf(ptr, 3u);
        assert (len(b) == 3u);
        assert (b[0] == 1);
        assert (b[1] == 2);
        assert (b[2] == 3);

        // Test on-heap copy-from-buf.
        let c = [1, 2, 3, 4, 5];
1315
        ptr = unsafe::to_ptr(c);
1316 1317 1318 1319 1320 1321 1322 1323 1324 1325
        let d = unsafe::from_buf(ptr, 5u);
        assert (len(d) == 5u);
        assert (d[0] == 1);
        assert (d[1] == 2);
        assert (d[2] == 3);
        assert (d[3] == 4);
        assert (d[4] == 5);
    }

    #[test]
1326 1327
    fn test_from_fn() {
        // Test on-stack from_fn.
1328
        let mut v = from_fn(3u, square);
1329 1330 1331 1332 1333
        assert (len(v) == 3u);
        assert (v[0] == 0u);
        assert (v[1] == 1u);
        assert (v[2] == 4u);

1334 1335
        // Test on-heap from_fn.
        v = from_fn(5u, square);
1336 1337 1338 1339 1340 1341 1342 1343 1344
        assert (len(v) == 5u);
        assert (v[0] == 0u);
        assert (v[1] == 1u);
        assert (v[2] == 4u);
        assert (v[3] == 9u);
        assert (v[4] == 16u);
    }

    #[test]
1345 1346
    fn test_from_elem() {
        // Test on-stack from_elem.
1347
        let mut v = from_elem(2u, 10u);
1348 1349 1350 1351
        assert (len(v) == 2u);
        assert (v[0] == 10u);
        assert (v[1] == 10u);

1352 1353
        // Test on-heap from_elem.
        v = from_elem(6u, 20u);
1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381
        assert (v[0] == 20u);
        assert (v[1] == 20u);
        assert (v[2] == 20u);
        assert (v[3] == 20u);
        assert (v[4] == 20u);
        assert (v[5] == 20u);
    }

    #[test]
    fn test_is_empty() {
        assert (is_empty::<int>([]));
        assert (!is_empty([0]));
    }

    #[test]
    fn test_is_not_empty() {
        assert (is_not_empty([0]));
        assert (!is_not_empty::<int>([]));
    }

    #[test]
    fn test_head() {
        let a = [11, 12];
        assert (head(a) == 11);
    }

    #[test]
    fn test_tail() {
1382
        let mut a = [11];
1383 1384 1385 1386 1387 1388 1389 1390
        assert (tail(a) == []);

        a = [11, 12];
        assert (tail(a) == [12]);
    }

    #[test]
    fn test_last() {
1391
        let mut n = last_opt([]);
1392
        assert (n == none);
1393
        n = last_opt([1, 2, 3]);
1394
        assert (n == some(3));
1395
        n = last_opt([1, 2, 3, 4, 5]);
1396 1397 1398 1399 1400 1401
        assert (n == some(5));
    }

    #[test]
    fn test_slice() {
        // Test on-stack -> on-stack slice.
1402
        let mut v = slice([1, 2, 3], 1u, 3u);
1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426
        assert (len(v) == 2u);
        assert (v[0] == 2);
        assert (v[1] == 3);

        // Test on-heap -> on-stack slice.
        v = slice([1, 2, 3, 4, 5], 0u, 3u);
        assert (len(v) == 3u);
        assert (v[0] == 1);
        assert (v[1] == 2);
        assert (v[2] == 3);

        // Test on-heap -> on-heap slice.
        v = slice([1, 2, 3, 4, 5, 6], 1u, 6u);
        assert (len(v) == 5u);
        assert (v[0] == 2);
        assert (v[1] == 3);
        assert (v[2] == 4);
        assert (v[3] == 5);
        assert (v[4] == 6);
    }

    #[test]
    fn test_pop() {
        // Test on-stack pop.
1427 1428
        let mut v = [1, 2, 3];
        let mut e = pop(v);
1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447
        assert (len(v) == 2u);
        assert (v[0] == 1);
        assert (v[1] == 2);
        assert (e == 3);

        // Test on-heap pop.
        v = [1, 2, 3, 4, 5];
        e = pop(v);
        assert (len(v) == 4u);
        assert (v[0] == 1);
        assert (v[1] == 2);
        assert (v[2] == 3);
        assert (v[3] == 4);
        assert (e == 5);
    }

    #[test]
    fn test_push() {
        // Test on-stack push().
1448
        let mut v = [];
1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462
        push(v, 1);
        assert (len(v) == 1u);
        assert (v[0] == 1);

        // Test on-heap push().
        push(v, 2);
        assert (len(v) == 2u);
        assert (v[0] == 1);
        assert (v[1] == 2);
    }

    #[test]
    fn test_grow() {
        // Test on-stack grow().
1463
        let mut v = [];
1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480
        grow(v, 2u, 1);
        assert (len(v) == 2u);
        assert (v[0] == 1);
        assert (v[1] == 1);

        // Test on-heap grow().
        grow(v, 3u, 2);
        assert (len(v) == 5u);
        assert (v[0] == 1);
        assert (v[1] == 1);
        assert (v[2] == 2);
        assert (v[3] == 2);
        assert (v[4] == 2);
    }

    #[test]
    fn test_grow_fn() {
1481
        let mut v = [];
1482 1483 1484 1485 1486 1487 1488 1489 1490
        grow_fn(v, 3u, square);
        assert (len(v) == 3u);
        assert (v[0] == 0u);
        assert (v[1] == 1u);
        assert (v[2] == 4u);
    }

    #[test]
    fn test_grow_set() {
G
Graydon Hoare 已提交
1491
        let mut v = [mut 1, 2, 3];
1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503
        grow_set(v, 4u, 4, 5);
        assert (len(v) == 5u);
        assert (v[0] == 1);
        assert (v[1] == 2);
        assert (v[2] == 3);
        assert (v[3] == 4);
        assert (v[4] == 5);
    }

    #[test]
    fn test_map() {
        // Test on-stack map.
1504 1505
        let mut v = [1u, 2u, 3u];
        let mut w = map(v, square_ref);
1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528
        assert (len(w) == 3u);
        assert (w[0] == 1u);
        assert (w[1] == 4u);
        assert (w[2] == 9u);

        // Test on-heap map.
        v = [1u, 2u, 3u, 4u, 5u];
        w = map(v, square_ref);
        assert (len(w) == 5u);
        assert (w[0] == 1u);
        assert (w[1] == 4u);
        assert (w[2] == 9u);
        assert (w[3] == 16u);
        assert (w[4] == 25u);
    }

    #[test]
    fn test_map2() {
        fn times(&&x: int, &&y: int) -> int { ret x * y; }
        let f = times;
        let v0 = [1, 2, 3, 4, 5];
        let v1 = [5, 4, 3, 2, 1];
        let u = map2::<int, int, int>(v0, v1, f);
1529
        let mut i = 0;
1530 1531 1532 1533 1534 1535
        while i < 5 { assert (v0[i] * v1[i] == u[i]); i += 1; }
    }

    #[test]
    fn test_filter_map() {
        // Test on-stack filter-map.
1536 1537
        let mut v = [1u, 2u, 3u];
        let mut w = filter_map(v, square_if_odd);
1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549
        assert (len(w) == 2u);
        assert (w[0] == 1u);
        assert (w[1] == 9u);

        // Test on-heap filter-map.
        v = [1u, 2u, 3u, 4u, 5u];
        w = filter_map(v, square_if_odd);
        assert (len(w) == 3u);
        assert (w[0] == 1u);
        assert (w[1] == 9u);
        assert (w[2] == 25u);

1550
        fn halve(&&i: int) -> option<int> {
1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575
            if i % 2 == 0 {
                ret option::some::<int>(i / 2);
            } else { ret option::none::<int>; }
        }
        fn halve_for_sure(&&i: int) -> int { ret i / 2; }
        let all_even: [int] = [0, 2, 8, 6];
        let all_odd1: [int] = [1, 7, 3];
        let all_odd2: [int] = [];
        let mix: [int] = [9, 2, 6, 7, 1, 0, 0, 3];
        let mix_dest: [int] = [1, 3, 0, 0];
        assert (filter_map(all_even, halve) == map(all_even, halve_for_sure));
        assert (filter_map(all_odd1, halve) == []);
        assert (filter_map(all_odd2, halve) == []);
        assert (filter_map(mix, halve) == mix_dest);
    }

    #[test]
    fn test_filter() {
        assert filter([1u, 2u, 3u], is_odd) == [1u, 3u];
        assert filter([1u, 2u, 4u, 8u, 16u], is_three) == [];
    }

    #[test]
    fn test_foldl() {
        // Test on-stack fold.
1576 1577
        let mut v = [1u, 2u, 3u];
        let mut sum = foldl(0u, v, add);
1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590
        assert (sum == 6u);

        // Test on-heap fold.
        v = [1u, 2u, 3u, 4u, 5u];
        sum = foldl(0u, v, add);
        assert (sum == 15u);
    }

    #[test]
    fn test_foldl2() {
        fn sub(&&a: int, &&b: int) -> int {
            a - b
        }
1591
        let mut v = [1, 2, 3, 4];
1592 1593 1594 1595 1596 1597 1598 1599 1600
        let sum = foldl(0, v, sub);
        assert sum == -10;
    }

    #[test]
    fn test_foldr() {
        fn sub(&&a: int, &&b: int) -> int {
            a - b
        }
1601
        let mut v = [1, 2, 3, 4];
1602 1603 1604 1605 1606 1607
        let sum = foldr(v, 0, sub);
        assert sum == -2;
    }

    #[test]
    fn test_iter_empty() {
1608
        let mut i = 0;
1609 1610 1611 1612 1613 1614
        iter::<int>([], { |_v| i += 1 });
        assert i == 0;
    }

    #[test]
    fn test_iter_nonempty() {
1615
        let mut i = 0;
1616 1617 1618 1619 1620 1621
        iter([1, 2, 3], { |v| i += v });
        assert i == 6;
    }

    #[test]
    fn test_iteri() {
1622
        let mut i = 0;
1623 1624 1625 1626 1627 1628 1629 1630 1631 1632
        iteri([1, 2, 3], { |j, v|
            if i == 0 { assert v == 1; }
            assert j + 1u == v as uint;
            i += v;
        });
        assert i == 6;
    }

    #[test]
    fn test_riter_empty() {
1633
        let mut i = 0;
1634 1635 1636 1637 1638 1639
        riter::<int>([], { |_v| i += 1 });
        assert i == 0;
    }

    #[test]
    fn test_riter_nonempty() {
1640
        let mut i = 0;
1641 1642 1643 1644 1645 1646 1647 1648 1649
        riter([1, 2, 3], { |v|
            if i == 0 { assert v == 3; }
            i += v
        });
        assert i == 6;
    }

    #[test]
    fn test_riteri() {
1650
        let mut i = 0;
1651 1652 1653 1654 1655 1656 1657 1658 1659 1660
        riteri([0, 1, 2], { |j, v|
            if i == 0 { assert v == 2; }
            assert j == v as uint;
            i += v;
        });
        assert i == 3;
    }

    #[test]
    fn test_permute() {
1661
        let mut results: [[int]];
1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725

        results = [];
        permute([]) {|v| results += [v]; }
        assert results == [[]];

        results = [];
        permute([7]) {|v| results += [v]; }
        assert results == [[7]];

        results = [];
        permute([1,1]) {|v| results += [v]; }
        assert results == [[1,1],[1,1]];

        results = [];
        permute([5,2,0]) {|v| results += [v]; }
        assert results == [[5,2,0],[5,0,2],[2,5,0],[2,0,5],[0,5,2],[0,2,5]];
    }

    #[test]
    fn test_any_and_all() {
        assert (any([1u, 2u, 3u], is_three));
        assert (!any([0u, 1u, 2u], is_three));
        assert (any([1u, 2u, 3u, 4u, 5u], is_three));
        assert (!any([1u, 2u, 4u, 5u, 6u], is_three));

        assert (all([3u, 3u, 3u], is_three));
        assert (!all([3u, 3u, 2u], is_three));
        assert (all([3u, 3u, 3u, 3u, 3u], is_three));
        assert (!all([3u, 3u, 0u, 1u, 2u], is_three));
    }

    #[test]
    fn test_any2_and_all2() {

        assert (any2([2u, 4u, 6u], [2u, 4u, 6u], is_equal));
        assert (any2([1u, 2u, 3u], [4u, 5u, 3u], is_equal));
        assert (!any2([1u, 2u, 3u], [4u, 5u, 6u], is_equal));
        assert (any2([2u, 4u, 6u], [2u, 4u], is_equal));

        assert (all2([2u, 4u, 6u], [2u, 4u, 6u], is_equal));
        assert (!all2([1u, 2u, 3u], [4u, 5u, 3u], is_equal));
        assert (!all2([1u, 2u, 3u], [4u, 5u, 6u], is_equal));
        assert (!all2([2u, 4u, 6u], [2u, 4u], is_equal));
    }

    #[test]
    fn test_zip_unzip() {
        let v1 = [1, 2, 3];
        let v2 = [4, 5, 6];

        let z1 = zip(v1, v2);

        assert ((1, 4) == z1[0]);
        assert ((2, 5) == z1[1]);
        assert ((3, 6) == z1[2]);

        let (left, right) = unzip(z1);

        assert ((1, 4) == (left[0], right[0]));
        assert ((2, 5) == (left[1], right[1]));
        assert ((3, 6) == (left[2], right[2]));
    }

    #[test]
1726 1727
    fn test_position_elem() {
        assert position_elem([], 1) == none;
1728 1729

        let v1 = [1, 2, 3, 3, 2, 5];
1730 1731 1732 1733
        assert position_elem(v1, 1) == some(0u);
        assert position_elem(v1, 2) == some(1u);
        assert position_elem(v1, 5) == some(5u);
        assert position_elem(v1, 4) == none;
1734 1735 1736
    }

    #[test]
1737
    fn test_position() {
1738 1739
        fn less_than_three(&&i: int) -> bool { ret i < 3; }
        fn is_eighteen(&&i: int) -> bool { ret i == 18; }
1740 1741 1742 1743 1744 1745

        assert position([], less_than_three) == none;

        let v1 = [5, 4, 3, 2, 1];
        assert position(v1, less_than_three) == some(3u);
        assert position(v1, is_eighteen) == none;
1746 1747 1748
    }

    #[test]
1749 1750
    fn test_position_between() {
        assert position_between([], 0u, 0u, f) == none;
1751 1752

        fn f(xy: (int, char)) -> bool { let (_x, y) = xy; y == 'b' }
1753
        let mut v = [(0, 'a'), (1, 'b'), (2, 'c'), (3, 'b')];
1754

1755 1756 1757 1758 1759
        assert position_between(v, 0u, 0u, f) == none;
        assert position_between(v, 0u, 1u, f) == none;
        assert position_between(v, 0u, 2u, f) == some(1u);
        assert position_between(v, 0u, 3u, f) == some(1u);
        assert position_between(v, 0u, 4u, f) == some(1u);
1760

1761 1762 1763 1764
        assert position_between(v, 1u, 1u, f) == none;
        assert position_between(v, 1u, 2u, f) == some(1u);
        assert position_between(v, 1u, 3u, f) == some(1u);
        assert position_between(v, 1u, 4u, f) == some(1u);
1765

1766 1767 1768
        assert position_between(v, 2u, 2u, f) == none;
        assert position_between(v, 2u, 3u, f) == none;
        assert position_between(v, 2u, 4u, f) == some(3u);
1769

1770 1771
        assert position_between(v, 3u, 3u, f) == none;
        assert position_between(v, 3u, 4u, f) == some(3u);
1772

1773
        assert position_between(v, 4u, 4u, f) == none;
1774 1775
    }

1776 1777 1778 1779 1780 1781
    #[test]
    fn test_find() {
        assert find([], f) == none;

        fn f(xy: (int, char)) -> bool { let (_x, y) = xy; y == 'b' }
        fn g(xy: (int, char)) -> bool { let (_x, y) = xy; y == 'd' }
1782
        let mut v = [(0, 'a'), (1, 'b'), (2, 'c'), (3, 'b')];
1783 1784 1785 1786 1787 1788

        assert find(v, f) == some((1, 'b'));
        assert find(v, g) == none;
    }

    #[test]
1789 1790
    fn test_find_between() {
        assert find_between([], 0u, 0u, f) == none;
1791 1792

        fn f(xy: (int, char)) -> bool { let (_x, y) = xy; y == 'b' }
1793
        let mut v = [(0, 'a'), (1, 'b'), (2, 'c'), (3, 'b')];
1794

1795 1796 1797 1798 1799
        assert find_between(v, 0u, 0u, f) == none;
        assert find_between(v, 0u, 1u, f) == none;
        assert find_between(v, 0u, 2u, f) == some((1, 'b'));
        assert find_between(v, 0u, 3u, f) == some((1, 'b'));
        assert find_between(v, 0u, 4u, f) == some((1, 'b'));
1800

1801 1802 1803 1804
        assert find_between(v, 1u, 1u, f) == none;
        assert find_between(v, 1u, 2u, f) == some((1, 'b'));
        assert find_between(v, 1u, 3u, f) == some((1, 'b'));
        assert find_between(v, 1u, 4u, f) == some((1, 'b'));
1805

1806 1807 1808
        assert find_between(v, 2u, 2u, f) == none;
        assert find_between(v, 2u, 3u, f) == none;
        assert find_between(v, 2u, 4u, f) == some((3, 'b'));
1809

1810 1811
        assert find_between(v, 3u, 3u, f) == none;
        assert find_between(v, 3u, 4u, f) == some((3, 'b'));
1812

1813
        assert find_between(v, 4u, 4u, f) == none;
1814 1815
    }

1816 1817 1818 1819 1820 1821
    #[test]
    fn test_rposition() {
        assert find([], f) == none;

        fn f(xy: (int, char)) -> bool { let (_x, y) = xy; y == 'b' }
        fn g(xy: (int, char)) -> bool { let (_x, y) = xy; y == 'd' }
1822
        let mut v = [(0, 'a'), (1, 'b'), (2, 'c'), (3, 'b')];
1823 1824 1825 1826 1827 1828

        assert position(v, f) == some(1u);
        assert position(v, g) == none;
    }

    #[test]
1829 1830
    fn test_rposition_between() {
        assert rposition_between([], 0u, 0u, f) == none;
1831 1832

        fn f(xy: (int, char)) -> bool { let (_x, y) = xy; y == 'b' }
1833
        let mut v = [(0, 'a'), (1, 'b'), (2, 'c'), (3, 'b')];
1834

1835 1836 1837 1838 1839
        assert rposition_between(v, 0u, 0u, f) == none;
        assert rposition_between(v, 0u, 1u, f) == none;
        assert rposition_between(v, 0u, 2u, f) == some(1u);
        assert rposition_between(v, 0u, 3u, f) == some(1u);
        assert rposition_between(v, 0u, 4u, f) == some(3u);
1840

1841 1842 1843 1844
        assert rposition_between(v, 1u, 1u, f) == none;
        assert rposition_between(v, 1u, 2u, f) == some(1u);
        assert rposition_between(v, 1u, 3u, f) == some(1u);
        assert rposition_between(v, 1u, 4u, f) == some(3u);
1845

1846 1847 1848
        assert rposition_between(v, 2u, 2u, f) == none;
        assert rposition_between(v, 2u, 3u, f) == none;
        assert rposition_between(v, 2u, 4u, f) == some(3u);
1849

1850 1851
        assert rposition_between(v, 3u, 3u, f) == none;
        assert rposition_between(v, 3u, 4u, f) == some(3u);
1852

1853
        assert rposition_between(v, 4u, 4u, f) == none;
1854
    }
1855 1856 1857 1858 1859 1860 1861

    #[test]
    fn test_rfind() {
        assert rfind([], f) == none;

        fn f(xy: (int, char)) -> bool { let (_x, y) = xy; y == 'b' }
        fn g(xy: (int, char)) -> bool { let (_x, y) = xy; y == 'd' }
1862
        let mut v = [(0, 'a'), (1, 'b'), (2, 'c'), (3, 'b')];
1863 1864 1865 1866 1867 1868

        assert rfind(v, f) == some((3, 'b'));
        assert rfind(v, g) == none;
    }

    #[test]
1869 1870
    fn test_rfind_between() {
        assert rfind_between([], 0u, 0u, f) == none;
1871 1872

        fn f(xy: (int, char)) -> bool { let (_x, y) = xy; y == 'b' }
1873
        let mut v = [(0, 'a'), (1, 'b'), (2, 'c'), (3, 'b')];
1874

1875 1876 1877 1878 1879
        assert rfind_between(v, 0u, 0u, f) == none;
        assert rfind_between(v, 0u, 1u, f) == none;
        assert rfind_between(v, 0u, 2u, f) == some((1, 'b'));
        assert rfind_between(v, 0u, 3u, f) == some((1, 'b'));
        assert rfind_between(v, 0u, 4u, f) == some((3, 'b'));
1880

1881 1882 1883 1884
        assert rfind_between(v, 1u, 1u, f) == none;
        assert rfind_between(v, 1u, 2u, f) == some((1, 'b'));
        assert rfind_between(v, 1u, 3u, f) == some((1, 'b'));
        assert rfind_between(v, 1u, 4u, f) == some((3, 'b'));
1885

1886 1887 1888
        assert rfind_between(v, 2u, 2u, f) == none;
        assert rfind_between(v, 2u, 3u, f) == none;
        assert rfind_between(v, 2u, 4u, f) == some((3, 'b'));
1889

1890 1891
        assert rfind_between(v, 3u, 3u, f) == none;
        assert rfind_between(v, 3u, 4u, f) == some((3, 'b'));
1892

1893
        assert rfind_between(v, 4u, 4u, f) == none;
1894 1895 1896 1897
    }

    #[test]
    fn reverse_and_reversed() {
G
Graydon Hoare 已提交
1898
        let v: [mut int] = [mut 10, 20];
1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912
        assert (v[0] == 10);
        assert (v[1] == 20);
        reverse(v);
        assert (v[0] == 20);
        assert (v[1] == 10);
        let v2 = reversed::<int>([10, 20]);
        assert (v2[0] == 20);
        assert (v2[1] == 10);
        v[0] = 30;
        assert (v2[0] == 20);
        // Make sure they work with 0-length vectors too.

        let v4 = reversed::<int>([]);
        assert (v4 == []);
G
Graydon Hoare 已提交
1913
        let v3: [mut int] = [mut];
1914 1915 1916 1917 1918
        reverse::<int>(v3);
    }

    #[test]
    fn reversed_mut() {
G
Graydon Hoare 已提交
1919
        let v2 = reversed::<int>([mut 10, 20]);
1920 1921 1922 1923 1924 1925 1926 1927 1928 1929
        assert (v2[0] == 20);
        assert (v2[1] == 10);
    }

    #[test]
    fn test_init() {
        let v = init([1, 2, 3]);
        assert v == [1, 2];
    }

1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971
    #[test]
    fn test_split() {
        fn f(&&x: int) -> bool { x == 3 }

        assert split([], f) == [];
        assert split([1, 2], f) == [[1, 2]];
        assert split([3, 1, 2], f) == [[], [1, 2]];
        assert split([1, 2, 3], f) == [[1, 2], []];
        assert split([1, 2, 3, 4, 3, 5], f) == [[1, 2], [4], [5]];
    }

    #[test]
    fn test_splitn() {
        fn f(&&x: int) -> bool { x == 3 }

        assert splitn([], 1u, f) == [];
        assert splitn([1, 2], 1u, f) == [[1, 2]];
        assert splitn([3, 1, 2], 1u, f) == [[], [1, 2]];
        assert splitn([1, 2, 3], 1u, f) == [[1, 2], []];
        assert splitn([1, 2, 3, 4, 3, 5], 1u, f) == [[1, 2], [4, 3, 5]];
    }

    #[test]
    fn test_rsplit() {
        fn f(&&x: int) -> bool { x == 3 }

        assert rsplit([], f) == [];
        assert rsplit([1, 2], f) == [[1, 2]];
        assert rsplit([1, 2, 3], f) == [[1, 2], []];
        assert rsplit([1, 2, 3, 4, 3, 5], f) == [[1, 2], [4], [5]];
    }

    #[test]
    fn test_rsplitn() {
        fn f(&&x: int) -> bool { x == 3 }

        assert rsplitn([], 1u, f) == [];
        assert rsplitn([1, 2], 1u, f) == [[1, 2]];
        assert rsplitn([1, 2, 3], 1u, f) == [[1, 2], []];
        assert rsplitn([1, 2, 3, 4, 3, 5], 1u, f) == [[1, 2, 3, 4], [5]];
    }

1972
    #[test]
B
Brian Anderson 已提交
1973
    #[should_fail]
1974 1975
    #[ignore(cfg(target_os = "win32"))]
    fn test_init_empty() {
B
Brian Anderson 已提交
1976
        init::<int>([]);
1977 1978 1979 1980 1981 1982 1983
    }

    #[test]
    fn test_concat() {
        assert concat([[1], [2,3]]) == [1, 2, 3];
    }

1984 1985 1986 1987 1988 1989 1990
    #[test]
    fn test_connect() {
        assert connect([], 0) == [];
        assert connect([[1], [2, 3]], 0) == [1, 0, 2, 3];
        assert connect([[1], [2], [3]], 0) == [1, 0, 2, 0, 3];
    }

1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003
    #[test]
    fn test_windowed () {
        assert [[1u,2u,3u],[2u,3u,4u],[3u,4u,5u],[4u,5u,6u]]
              == windowed (3u, [1u,2u,3u,4u,5u,6u]);

        assert [[1u,2u,3u,4u],[2u,3u,4u,5u],[3u,4u,5u,6u]]
              == windowed (4u, [1u,2u,3u,4u,5u,6u]);

        assert [] == windowed (7u, [1u,2u,3u,4u,5u,6u]);
    }

    #[test]
    #[should_fail]
2004
    #[ignore(cfg(target_os = "win32"))]
2005 2006 2007
    fn test_windowed_() {
        let _x = windowed (0u, [1u,2u,3u,4u,5u,6u]);
    }
2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025

    #[test]
    fn to_mut_no_copy() unsafe {
        let x = [1, 2, 3];
        let addr = unsafe::to_ptr(x);
        let x_mut = to_mut(x);
        let addr_mut = unsafe::to_ptr(x_mut);
        assert addr == addr_mut;
    }

    #[test]
    fn from_mut_no_copy() unsafe {
        let x = [mut 1, 2, 3];
        let addr = unsafe::to_ptr(x);
        let x_imm = from_mut(x);
        let addr_imm = unsafe::to_ptr(x_imm);
        assert addr == addr_imm;
    }
B
Brian Anderson 已提交
2026 2027 2028

    #[test]
    fn test_unshift() {
2029
        let mut x = [1, 2, 3];
B
Brian Anderson 已提交
2030 2031 2032
        unshift(x, 0);
        assert x == [0, 1, 2, 3];
    }
B
Brian Anderson 已提交
2033 2034 2035 2036 2037 2038 2039 2040 2041 2042

    #[test]
    fn test_capacity() {
        let mut v = [0u64];
        reserve(v, 10u);
        assert capacity(v) == 10u;
        let mut v = [0u32];
        reserve(v, 10u);
        assert capacity(v) == 10u;
    }
2043 2044 2045 2046 2047 2048 2049 2050 2051

    #[test]
    fn test_view() {
        let v = [1, 2, 3, 4, 5];
        let v = view(v, 1u, 3u);
        assert(len(v) == 2u);
        assert(v[0] == 2);
        assert(v[1] == 3);
    }
2052 2053
}

2054 2055 2056 2057 2058 2059 2060
// Local Variables:
// mode: rust;
// fill-column: 78;
// indent-tabs-mode: nil
// c-basic-offset: 4
// buffer-file-coding-system: utf-8-unix
// End: