2013-02-20 17:02:21 -06:00
|
|
|
// Copyright 2012-2013 The Rust Project Developers. See the COPYRIGHT
|
2012-12-03 18:48:01 -06:00
|
|
|
// file at the top-level directory of this distribution and at
|
|
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
|
|
// option. This file may not be copied, modified, or distributed
|
|
|
|
// except according to those terms.
|
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
//! Vectors
|
2012-03-15 20:58:14 -05:00
|
|
|
|
2012-09-28 00:20:47 -05:00
|
|
|
#[warn(non_camel_case_types)];
|
|
|
|
|
2012-12-23 16:41:37 -06:00
|
|
|
use cast;
|
2013-07-02 14:47:32 -05:00
|
|
|
use clone::Clone;
|
2013-04-17 16:19:25 -05:00
|
|
|
use container::{Container, Mutable};
|
2013-07-17 14:32:49 -05:00
|
|
|
use cmp::{Eq, TotalOrd, Ordering, Less, Equal, Greater};
|
2013-07-02 14:47:32 -05:00
|
|
|
use cmp;
|
2013-07-10 23:23:59 -05:00
|
|
|
use iterator::*;
|
2013-06-30 21:18:17 -05:00
|
|
|
use libc::c_void;
|
2013-06-14 20:27:52 -05:00
|
|
|
use num::Zero;
|
2013-01-08 21:37:25 -06:00
|
|
|
use option::{None, Option, Some};
|
2013-04-22 16:27:30 -05:00
|
|
|
use ptr::to_unsafe_ptr;
|
2012-12-23 16:41:37 -06:00
|
|
|
use ptr;
|
2013-06-03 12:50:29 -05:00
|
|
|
use ptr::RawPtr;
|
2013-07-11 00:08:50 -05:00
|
|
|
use rt::global_heap::malloc_raw;
|
|
|
|
use rt::global_heap::realloc_raw;
|
2012-12-23 16:41:37 -06:00
|
|
|
use sys;
|
2013-07-15 13:23:42 -05:00
|
|
|
use sys::size_of;
|
2012-12-23 16:41:37 -06:00
|
|
|
use uint;
|
2013-04-22 16:27:30 -05:00
|
|
|
use unstable::intrinsics;
|
2013-07-08 19:22:51 -05:00
|
|
|
use unstable::intrinsics::{get_tydesc, contains_managed};
|
2013-07-21 19:20:52 -05:00
|
|
|
use unstable::raw::{Box, Repr, Slice, Vec};
|
2012-12-23 16:41:37 -06:00
|
|
|
use vec;
|
2013-05-05 23:42:54 -05:00
|
|
|
use util;
|
2011-12-13 18:25:51 -06:00
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Returns true if two vectors have the same length
|
2013-06-23 22:44:11 -05:00
|
|
|
pub fn same_length<T, U>(xs: &[T], ys: &[U]) -> bool {
|
2013-03-09 15:41:43 -06:00
|
|
|
xs.len() == ys.len()
|
2011-12-13 18:25:51 -06:00
|
|
|
}
|
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/**
|
2013-04-30 04:39:16 -05:00
|
|
|
* Creates and initializes an owned vector.
|
2012-07-04 16:53:12 -05:00
|
|
|
*
|
2013-04-30 04:39:16 -05:00
|
|
|
* Creates an owned vector of size `n_elts` and initializes the elements
|
2012-07-04 16:53:12 -05:00
|
|
|
* to the value returned by the function `op`.
|
|
|
|
*/
|
2013-06-23 16:57:39 -05:00
|
|
|
pub fn from_fn<T>(n_elts: uint, op: &fn(uint) -> T) -> ~[T] {
|
2012-10-05 16:58:42 -05:00
|
|
|
unsafe {
|
|
|
|
let mut v = with_capacity(n_elts);
|
2013-07-27 16:30:29 -05:00
|
|
|
let p = raw::to_mut_ptr(v);
|
|
|
|
let mut i: uint = 0u;
|
|
|
|
while i < n_elts {
|
|
|
|
intrinsics::move_val_init(&mut(*ptr::mut_offset(p, i)), op(i));
|
|
|
|
i += 1u;
|
2012-10-05 16:58:42 -05:00
|
|
|
}
|
|
|
|
raw::set_len(&mut v, n_elts);
|
2013-03-09 15:41:43 -06:00
|
|
|
v
|
2012-10-05 16:58:42 -05:00
|
|
|
}
|
|
|
|
}
|
2011-12-13 18:25:51 -06:00
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/**
|
2013-04-30 04:39:16 -05:00
|
|
|
* Creates and initializes an owned vector.
|
2012-07-04 16:53:12 -05:00
|
|
|
*
|
2013-04-30 04:39:16 -05:00
|
|
|
* Creates an owned vector of size `n_elts` and initializes the elements
|
2012-07-04 16:53:12 -05:00
|
|
|
* to the value `t`.
|
|
|
|
*/
|
2013-07-02 14:47:32 -05:00
|
|
|
pub fn from_elem<T:Clone>(n_elts: uint, t: T) -> ~[T] {
|
2013-06-14 18:58:55 -05:00
|
|
|
// FIXME (#7136): manually inline from_fn for 2x plus speedup (sadly very
|
|
|
|
// important, from_elem is a bottleneck in borrowck!). Unfortunately it
|
|
|
|
// still is substantially slower than using the unsafe
|
|
|
|
// vec::with_capacity/ptr::set_memory for primitive types.
|
2013-05-31 23:55:19 -05:00
|
|
|
unsafe {
|
|
|
|
let mut v = with_capacity(n_elts);
|
2013-07-27 16:30:29 -05:00
|
|
|
let p = raw::to_mut_ptr(v);
|
|
|
|
let mut i = 0u;
|
|
|
|
while i < n_elts {
|
|
|
|
intrinsics::move_val_init(&mut(*ptr::mut_offset(p, i)), t.clone());
|
|
|
|
i += 1u;
|
2013-05-31 23:55:19 -05:00
|
|
|
}
|
|
|
|
raw::set_len(&mut v, n_elts);
|
|
|
|
v
|
|
|
|
}
|
2011-12-13 18:25:51 -06:00
|
|
|
}
|
|
|
|
|
2013-04-24 21:33:13 -05:00
|
|
|
/// Creates a new vector with a capacity of `capacity`
|
2013-07-23 11:54:28 -05:00
|
|
|
#[inline]
|
2013-07-09 20:59:30 -05:00
|
|
|
pub fn with_capacity<T>(capacity: uint) -> ~[T] {
|
|
|
|
unsafe {
|
|
|
|
if contains_managed::<T>() {
|
|
|
|
let mut vec = ~[];
|
|
|
|
vec.reserve(capacity);
|
|
|
|
vec
|
|
|
|
} else {
|
|
|
|
let alloc = capacity * sys::nonzero_size_of::<T>();
|
2013-07-21 19:20:52 -05:00
|
|
|
let ptr = malloc_raw(alloc + sys::size_of::<Vec<()>>()) as *mut Vec<()>;
|
2013-07-15 13:23:42 -05:00
|
|
|
(*ptr).alloc = alloc;
|
|
|
|
(*ptr).fill = 0;
|
2013-07-09 20:59:30 -05:00
|
|
|
cast::transmute(ptr)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-07-17 18:31:19 -05:00
|
|
|
/**
|
|
|
|
* Builds a vector by calling a provided function with an argument
|
|
|
|
* function that pushes an element to the back of a vector.
|
2013-03-25 17:34:15 -05:00
|
|
|
* This version takes an initial capacity for the vector.
|
2012-07-17 18:31:19 -05:00
|
|
|
*
|
|
|
|
* # Arguments
|
|
|
|
*
|
|
|
|
* * size - An initial size of the vector to reserve
|
2013-03-25 17:34:15 -05:00
|
|
|
* * builder - A function that will construct the vector. It receives
|
2012-07-17 18:31:19 -05:00
|
|
|
* as an argument a function that will push an element
|
|
|
|
* onto the vector being constructed.
|
|
|
|
*/
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-03-21 23:20:48 -05:00
|
|
|
pub fn build_sized<A>(size: uint, builder: &fn(push: &fn(v: A))) -> ~[A] {
|
2012-09-21 20:43:30 -05:00
|
|
|
let mut vec = with_capacity(size);
|
2013-04-08 15:50:34 -05:00
|
|
|
builder(|x| vec.push(x));
|
2012-12-12 17:38:50 -06:00
|
|
|
vec
|
2012-07-17 18:31:19 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Builds a vector by calling a provided function with an argument
|
|
|
|
* function that pushes an element to the back of a vector.
|
|
|
|
*
|
|
|
|
* # Arguments
|
|
|
|
*
|
2013-03-25 17:34:15 -05:00
|
|
|
* * builder - A function that will construct the vector. It receives
|
2012-07-17 18:31:19 -05:00
|
|
|
* as an argument a function that will push an element
|
|
|
|
* onto the vector being constructed.
|
|
|
|
*/
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-03-21 23:20:48 -05:00
|
|
|
pub fn build<A>(builder: &fn(push: &fn(v: A))) -> ~[A] {
|
2012-07-17 18:31:19 -05:00
|
|
|
build_sized(4, builder)
|
|
|
|
}
|
|
|
|
|
2012-08-23 12:22:14 -05:00
|
|
|
/**
|
|
|
|
* Builds a vector by calling a provided function with an argument
|
|
|
|
* function that pushes an element to the back of a vector.
|
|
|
|
* This version takes an initial size for the vector.
|
|
|
|
*
|
|
|
|
* # Arguments
|
|
|
|
*
|
|
|
|
* * size - An option, maybe containing initial size of the vector to reserve
|
2013-05-08 12:34:47 -05:00
|
|
|
* * builder - A function that will construct the vector. It receives
|
2012-08-23 12:22:14 -05:00
|
|
|
* as an argument a function that will push an element
|
|
|
|
* onto the vector being constructed.
|
|
|
|
*/
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-03-21 23:20:48 -05:00
|
|
|
pub fn build_sized_opt<A>(size: Option<uint>,
|
|
|
|
builder: &fn(push: &fn(v: A)))
|
|
|
|
-> ~[A] {
|
2013-01-04 18:01:26 -06:00
|
|
|
build_sized(size.get_or_default(4), builder)
|
2012-08-23 12:22:14 -05:00
|
|
|
}
|
|
|
|
|
2013-07-02 23:54:11 -05:00
|
|
|
/// An iterator over the slices of a vector separated by elements that
|
|
|
|
/// match a predicate function.
|
2013-07-28 15:37:35 -05:00
|
|
|
pub struct SplitIterator<'self, T> {
|
2013-07-02 23:54:11 -05:00
|
|
|
priv v: &'self [T],
|
|
|
|
priv n: uint,
|
|
|
|
priv pred: &'self fn(t: &T) -> bool,
|
|
|
|
priv finished: bool
|
|
|
|
}
|
2011-12-13 18:25:51 -06:00
|
|
|
|
2013-07-28 15:37:35 -05:00
|
|
|
impl<'self, T> Iterator<&'self [T]> for SplitIterator<'self, T> {
|
2013-07-02 23:54:11 -05:00
|
|
|
fn next(&mut self) -> Option<&'self [T]> {
|
|
|
|
if self.finished { return None; }
|
2013-01-29 00:44:59 -06:00
|
|
|
|
2013-07-02 23:54:11 -05:00
|
|
|
if self.n == 0 {
|
|
|
|
self.finished = true;
|
|
|
|
return Some(self.v);
|
|
|
|
}
|
2012-01-26 20:13:43 -06:00
|
|
|
|
2013-07-04 21:13:26 -05:00
|
|
|
match self.v.iter().position(|x| (self.pred)(x)) {
|
2013-07-02 23:54:11 -05:00
|
|
|
None => {
|
|
|
|
self.finished = true;
|
|
|
|
Some(self.v)
|
|
|
|
}
|
|
|
|
Some(idx) => {
|
|
|
|
let ret = Some(self.v.slice(0, idx));
|
|
|
|
self.v = self.v.slice(idx + 1, self.v.len());
|
|
|
|
self.n -= 1;
|
|
|
|
ret
|
2012-09-26 19:33:34 -05:00
|
|
|
}
|
2012-01-26 20:13:43 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-07-02 23:54:11 -05:00
|
|
|
/// An iterator over the slices of a vector separated by elements that
|
|
|
|
/// match a predicate function, from back to front.
|
2013-07-28 15:37:35 -05:00
|
|
|
pub struct RSplitIterator<'self, T> {
|
2013-07-02 23:54:11 -05:00
|
|
|
priv v: &'self [T],
|
|
|
|
priv n: uint,
|
|
|
|
priv pred: &'self fn(t: &T) -> bool,
|
|
|
|
priv finished: bool
|
2012-01-26 20:13:43 -06:00
|
|
|
}
|
|
|
|
|
2013-07-28 15:37:35 -05:00
|
|
|
impl<'self, T> Iterator<&'self [T]> for RSplitIterator<'self, T> {
|
2013-07-02 23:54:11 -05:00
|
|
|
fn next(&mut self) -> Option<&'self [T]> {
|
|
|
|
if self.finished { return None; }
|
2012-01-26 20:13:43 -06:00
|
|
|
|
2013-07-02 23:54:11 -05:00
|
|
|
if self.n == 0 {
|
|
|
|
self.finished = true;
|
|
|
|
return Some(self.v);
|
2012-01-26 20:13:43 -06:00
|
|
|
}
|
|
|
|
|
2013-07-02 23:54:11 -05:00
|
|
|
match self.v.rposition(|x| (self.pred)(x)) {
|
|
|
|
None => {
|
|
|
|
self.finished = true;
|
|
|
|
Some(self.v)
|
|
|
|
}
|
|
|
|
Some(idx) => {
|
|
|
|
let ret = Some(self.v.slice(idx + 1, self.v.len()));
|
|
|
|
self.v = self.v.slice(0, idx);
|
|
|
|
self.n -= 1;
|
|
|
|
ret
|
2012-09-26 19:33:34 -05:00
|
|
|
}
|
2012-01-26 20:13:43 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2011-12-13 18:25:51 -06:00
|
|
|
|
|
|
|
// Appending
|
2013-05-28 16:35:52 -05:00
|
|
|
|
|
|
|
/// Iterates over the `rhs` vector, copying each element and appending it to the
|
|
|
|
/// `lhs`. Afterwards, the `lhs` is then returned for use again.
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-07-02 14:47:32 -05:00
|
|
|
pub fn append<T:Clone>(lhs: ~[T], rhs: &[T]) -> ~[T] {
|
2012-12-12 17:38:50 -06:00
|
|
|
let mut v = lhs;
|
2013-04-08 15:50:34 -05:00
|
|
|
v.push_all(rhs);
|
2012-12-12 17:38:50 -06:00
|
|
|
v
|
2012-06-13 18:14:01 -05:00
|
|
|
}
|
|
|
|
|
2013-05-28 16:35:52 -05:00
|
|
|
/// Appends one element to the vector provided. The vector itself is then
|
|
|
|
/// returned for use again.
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-03-21 23:20:48 -05:00
|
|
|
pub fn append_one<T>(lhs: ~[T], x: T) -> ~[T] {
|
2012-12-12 17:38:50 -06:00
|
|
|
let mut v = lhs;
|
2013-04-08 15:50:34 -05:00
|
|
|
v.push(x);
|
2012-12-12 17:38:50 -06:00
|
|
|
v
|
2012-06-28 15:52:13 -05:00
|
|
|
}
|
|
|
|
|
2011-12-13 18:25:51 -06:00
|
|
|
// Functional utilities
|
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/**
|
|
|
|
* Apply a function to each element of a vector and return a concatenation
|
|
|
|
* of each result vector
|
|
|
|
*/
|
2013-03-21 23:20:48 -05:00
|
|
|
pub fn flat_map<T, U>(v: &[T], f: &fn(t: &T) -> ~[U]) -> ~[U] {
|
2012-06-29 18:26:56 -05:00
|
|
|
let mut result = ~[];
|
2013-06-24 17:34:20 -05:00
|
|
|
for v.iter().advance |elem| { result.push_all_move(f(elem)); }
|
2012-12-12 17:38:50 -06:00
|
|
|
result
|
2012-03-02 16:36:22 -06:00
|
|
|
}
|
|
|
|
|
2013-06-02 22:19:37 -05:00
|
|
|
/// Flattens a vector of vectors of T into a single vector of T.
|
2013-07-02 14:47:32 -05:00
|
|
|
pub fn concat<T:Clone>(v: &[~[T]]) -> ~[T] { v.concat_vec() }
|
2013-06-02 22:19:37 -05:00
|
|
|
|
|
|
|
/// Concatenate a vector of vectors, placing a given separator between each
|
2013-07-02 14:47:32 -05:00
|
|
|
pub fn connect<T:Clone>(v: &[~[T]], sep: &T) -> ~[T] { v.connect_vec(sep) }
|
2013-06-02 22:19:37 -05:00
|
|
|
|
|
|
|
/// Flattens a vector of vectors of T into a single vector of T.
|
2013-07-02 14:47:32 -05:00
|
|
|
pub fn concat_slices<T:Clone>(v: &[&[T]]) -> ~[T] { v.concat_vec() }
|
2011-12-13 18:25:51 -06:00
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Concatenate a vector of vectors, placing a given separator between each
|
2013-07-02 14:47:32 -05:00
|
|
|
pub fn connect_slices<T:Clone>(v: &[&[T]], sep: &T) -> ~[T] { v.connect_vec(sep) }
|
2013-06-02 22:19:37 -05:00
|
|
|
|
|
|
|
#[allow(missing_doc)]
|
|
|
|
pub trait VectorVector<T> {
|
2013-06-14 21:56:41 -05:00
|
|
|
// FIXME #5898: calling these .concat and .connect conflicts with
|
|
|
|
// StrVector::con{cat,nect}, since they have generic contents.
|
|
|
|
pub fn concat_vec(&self) -> ~[T];
|
|
|
|
pub fn connect_vec(&self, sep: &T) -> ~[T];
|
2013-06-02 22:19:37 -05:00
|
|
|
}
|
|
|
|
|
2013-07-02 14:47:32 -05:00
|
|
|
impl<'self, T:Clone> VectorVector<T> for &'self [~[T]] {
|
2013-06-02 22:19:37 -05:00
|
|
|
/// Flattens a vector of slices of T into a single vector of T.
|
2013-06-14 21:56:41 -05:00
|
|
|
pub fn concat_vec(&self) -> ~[T] {
|
2013-07-02 14:47:32 -05:00
|
|
|
self.flat_map(|inner| (*inner).clone())
|
2013-06-02 22:19:37 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Concatenate a vector of vectors, placing a given separator between each.
|
2013-06-14 21:56:41 -05:00
|
|
|
pub fn connect_vec(&self, sep: &T) -> ~[T] {
|
2013-06-02 22:19:37 -05:00
|
|
|
let mut r = ~[];
|
|
|
|
let mut first = true;
|
2013-06-20 14:14:30 -05:00
|
|
|
for self.iter().advance |inner| {
|
2013-07-02 14:47:32 -05:00
|
|
|
if first { first = false; } else { r.push((*sep).clone()); }
|
|
|
|
r.push_all((*inner).clone());
|
2013-06-02 22:19:37 -05:00
|
|
|
}
|
|
|
|
r
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-07-02 14:47:32 -05:00
|
|
|
impl<'self,T:Clone> VectorVector<T> for &'self [&'self [T]] {
|
2013-06-02 22:19:37 -05:00
|
|
|
/// Flattens a vector of slices of T into a single vector of T.
|
2013-06-14 21:56:41 -05:00
|
|
|
pub fn concat_vec(&self) -> ~[T] {
|
2013-06-02 22:19:37 -05:00
|
|
|
self.flat_map(|&inner| inner.to_owned())
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Concatenate a vector of slices, placing a given separator between each.
|
2013-06-14 21:56:41 -05:00
|
|
|
pub fn connect_vec(&self, sep: &T) -> ~[T] {
|
2013-06-02 22:19:37 -05:00
|
|
|
let mut r = ~[];
|
|
|
|
let mut first = true;
|
2013-06-21 07:29:53 -05:00
|
|
|
for self.iter().advance |&inner| {
|
2013-07-02 14:47:32 -05:00
|
|
|
if first { first = false; } else { r.push((*sep).clone()); }
|
2013-06-02 22:19:37 -05:00
|
|
|
r.push_all(inner);
|
|
|
|
}
|
|
|
|
r
|
2012-01-28 17:41:53 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-12-13 18:25:51 -06:00
|
|
|
// 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)
|
2012-07-04 16:53:12 -05:00
|
|
|
/**
|
2012-08-23 19:51:34 -05:00
|
|
|
* Convert a vector of pairs into a pair of vectors, by reference. As unzip().
|
2012-07-04 16:53:12 -05:00
|
|
|
*/
|
2013-07-02 14:47:32 -05:00
|
|
|
pub fn unzip_slice<T:Clone,U:Clone>(v: &[(T, U)]) -> (~[T], ~[U]) {
|
2013-06-06 20:54:14 -05:00
|
|
|
let mut ts = ~[];
|
|
|
|
let mut us = ~[];
|
2013-06-24 17:34:20 -05:00
|
|
|
for v.iter().advance |p| {
|
2013-07-02 14:47:32 -05:00
|
|
|
let (t, u) = (*p).clone();
|
2013-04-08 15:50:34 -05:00
|
|
|
ts.push(t);
|
|
|
|
us.push(u);
|
2012-06-27 17:21:50 -05:00
|
|
|
}
|
2013-03-09 15:41:43 -06:00
|
|
|
(ts, us)
|
2011-12-13 18:25:51 -06:00
|
|
|
}
|
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/**
|
2012-08-23 19:51:34 -05:00
|
|
|
* Convert a vector of pairs into a pair of vectors.
|
2012-07-04 16:53:12 -05:00
|
|
|
*
|
2012-08-23 19:51:34 -05:00
|
|
|
* 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.
|
|
|
|
*/
|
2013-03-21 23:20:48 -05:00
|
|
|
pub fn unzip<T,U>(v: ~[(T, U)]) -> (~[T], ~[U]) {
|
2013-06-06 20:54:14 -05:00
|
|
|
let mut ts = ~[];
|
|
|
|
let mut us = ~[];
|
2013-07-01 10:26:44 -05:00
|
|
|
for v.consume_iter().advance |p| {
|
2013-04-08 15:50:34 -05:00
|
|
|
let (t, u) = p;
|
|
|
|
ts.push(t);
|
|
|
|
us.push(u);
|
2012-08-23 19:51:34 -05:00
|
|
|
}
|
2012-12-12 17:38:50 -06:00
|
|
|
(ts, us)
|
2012-08-23 19:51:34 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Convert two vectors to a vector of pairs, by reference. As zip().
|
2012-07-04 16:53:12 -05:00
|
|
|
*/
|
2013-07-02 14:47:32 -05:00
|
|
|
pub fn zip_slice<T:Clone,U:Clone>(v: &[T], u: &[U]) -> ~[(T, U)] {
|
2012-06-29 18:26:56 -05:00
|
|
|
let mut zipped = ~[];
|
2013-06-08 20:38:47 -05:00
|
|
|
let sz = v.len();
|
2012-03-06 22:48:40 -06:00
|
|
|
let mut i = 0u;
|
2013-06-08 20:38:47 -05:00
|
|
|
assert_eq!(sz, u.len());
|
2013-01-23 13:43:58 -06:00
|
|
|
while i < sz {
|
2013-07-02 14:47:32 -05:00
|
|
|
zipped.push((v[i].clone(), u[i].clone()));
|
2013-04-08 15:50:34 -05:00
|
|
|
i += 1u;
|
2013-01-23 13:43:58 -06:00
|
|
|
}
|
2012-12-12 17:38:50 -06:00
|
|
|
zipped
|
2011-12-13 18:25:51 -06:00
|
|
|
}
|
|
|
|
|
2012-08-23 19:51:34 -05:00
|
|
|
/**
|
|
|
|
* Convert two vectors to a vector of pairs.
|
|
|
|
*
|
2013-06-06 02:38:41 -05:00
|
|
|
* Returns a vector of tuples, where the i-th tuple contains the
|
2012-08-23 19:51:34 -05:00
|
|
|
* i-th elements from each of the input vectors.
|
|
|
|
*/
|
2013-03-21 23:20:48 -05:00
|
|
|
pub fn zip<T, U>(mut v: ~[T], mut u: ~[U]) -> ~[(T, U)] {
|
2013-06-08 20:38:47 -05:00
|
|
|
let mut i = v.len();
|
|
|
|
assert_eq!(i, u.len());
|
2012-09-21 20:43:30 -05:00
|
|
|
let mut w = with_capacity(i);
|
2012-08-23 19:51:34 -05:00
|
|
|
while i > 0 {
|
2013-04-08 15:50:34 -05:00
|
|
|
w.push((v.pop(),u.pop()));
|
2012-08-23 19:51:34 -05:00
|
|
|
i -= 1;
|
|
|
|
}
|
2013-06-28 11:54:03 -05:00
|
|
|
w.reverse();
|
2012-12-12 17:38:50 -06:00
|
|
|
w
|
2012-08-23 19:51:34 -05:00
|
|
|
}
|
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/**
|
|
|
|
* Iterate over all permutations of vector `v`.
|
|
|
|
*
|
|
|
|
* 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).
|
|
|
|
*
|
2013-06-08 20:38:47 -05:00
|
|
|
* The total number of permutations produced is `v.len()!`. If `v` contains
|
2012-07-04 16:53:12 -05:00
|
|
|
* repeated elements, then some permutations are repeated.
|
2013-02-13 17:52:58 -06:00
|
|
|
*
|
|
|
|
* See [Algorithms to generate
|
|
|
|
* permutations](http://en.wikipedia.org/wiki/Permutation).
|
|
|
|
*
|
|
|
|
* # Arguments
|
|
|
|
*
|
|
|
|
* * `values` - A vector of values from which the permutations are
|
|
|
|
* chosen
|
|
|
|
*
|
|
|
|
* * `fun` - The function to iterate over the combinations
|
2012-07-04 16:53:12 -05:00
|
|
|
*/
|
2013-07-02 14:47:32 -05:00
|
|
|
pub fn each_permutation<T:Clone>(values: &[T], fun: &fn(perm : &[T]) -> bool) -> bool {
|
2013-02-13 17:52:58 -06:00
|
|
|
let length = values.len();
|
2013-07-02 14:47:32 -05:00
|
|
|
let mut permutation = vec::from_fn(length, |i| values[i].clone());
|
2013-02-13 17:52:58 -06:00
|
|
|
if length <= 1 {
|
|
|
|
fun(permutation);
|
2013-05-16 21:57:49 -05:00
|
|
|
return true;
|
2013-02-13 17:52:58 -06:00
|
|
|
}
|
|
|
|
let mut indices = vec::from_fn(length, |i| i);
|
|
|
|
loop {
|
2013-05-16 21:57:49 -05:00
|
|
|
if !fun(permutation) { return true; }
|
2013-02-13 17:52:58 -06:00
|
|
|
// find largest k such that indices[k] < indices[k+1]
|
|
|
|
// if no such k exists, all permutations have been generated
|
|
|
|
let mut k = length - 2;
|
|
|
|
while k > 0 && indices[k] >= indices[k+1] {
|
|
|
|
k -= 1;
|
|
|
|
}
|
2013-05-16 21:57:49 -05:00
|
|
|
if k == 0 && indices[0] > indices[1] { return true; }
|
2013-02-13 17:52:58 -06:00
|
|
|
// find largest l such that indices[k] < indices[l]
|
|
|
|
// k+1 is guaranteed to be such
|
|
|
|
let mut l = length - 1;
|
|
|
|
while indices[k] >= indices[l] {
|
|
|
|
l -= 1;
|
|
|
|
}
|
|
|
|
// swap indices[k] and indices[l]; sort indices[k+1..]
|
|
|
|
// (they're just reversed)
|
2013-06-28 11:54:03 -05:00
|
|
|
indices.swap(k, l);
|
|
|
|
indices.mut_slice(k+1, length).reverse();
|
2013-02-13 17:52:58 -06:00
|
|
|
// fixup permutation based on indices
|
|
|
|
for uint::range(k, length) |i| {
|
2013-07-02 14:47:32 -05:00
|
|
|
permutation[i] = values[indices[i]].clone();
|
2012-06-27 17:21:50 -05:00
|
|
|
}
|
2011-12-13 18:25:51 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-07-03 00:47:58 -05:00
|
|
|
/// An iterator over the (overlapping) slices of length `size` within
|
|
|
|
/// a vector.
|
2013-07-28 15:37:35 -05:00
|
|
|
pub struct WindowIter<'self, T> {
|
2013-07-03 00:47:58 -05:00
|
|
|
priv v: &'self [T],
|
|
|
|
priv size: uint
|
|
|
|
}
|
|
|
|
|
2013-07-28 15:37:35 -05:00
|
|
|
impl<'self, T> Iterator<&'self [T]> for WindowIter<'self, T> {
|
2013-07-03 00:47:58 -05:00
|
|
|
fn next(&mut self) -> Option<&'self [T]> {
|
|
|
|
if self.size > self.v.len() {
|
|
|
|
None
|
|
|
|
} else {
|
|
|
|
let ret = Some(self.v.slice(0, self.size));
|
|
|
|
self.v = self.v.slice(1, self.v.len());
|
|
|
|
ret
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// An iterator over a vector in (non-overlapping) chunks (`size`
|
|
|
|
/// elements at a time).
|
2013-07-28 15:37:35 -05:00
|
|
|
pub struct ChunkIter<'self, T> {
|
2013-07-03 00:47:58 -05:00
|
|
|
priv v: &'self [T],
|
|
|
|
priv size: uint
|
|
|
|
}
|
|
|
|
|
2013-07-28 15:37:35 -05:00
|
|
|
impl<'self, T> Iterator<&'self [T]> for ChunkIter<'self, T> {
|
2013-07-03 00:47:58 -05:00
|
|
|
fn next(&mut self) -> Option<&'self [T]> {
|
|
|
|
if self.size == 0 {
|
|
|
|
None
|
|
|
|
} else if self.size >= self.v.len() {
|
|
|
|
// finished
|
|
|
|
self.size = 0;
|
|
|
|
Some(self.v)
|
|
|
|
} else {
|
|
|
|
let ret = Some(self.v.slice(0, self.size));
|
|
|
|
self.v = self.v.slice(self.size, self.v.len());
|
|
|
|
ret
|
|
|
|
}
|
2013-05-03 15:33:33 -05:00
|
|
|
}
|
|
|
|
}
|
2012-01-23 04:41:40 -06:00
|
|
|
|
2012-08-27 19:15:54 -05:00
|
|
|
// Equality
|
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
#[cfg(not(test))]
|
|
|
|
pub mod traits {
|
|
|
|
use super::Vector;
|
2013-07-02 14:47:32 -05:00
|
|
|
|
|
|
|
use clone::Clone;
|
2013-07-02 21:13:00 -05:00
|
|
|
use cmp::{Eq, Ord, TotalEq, TotalOrd, Ordering, Equal, Equiv};
|
|
|
|
use ops::Add;
|
|
|
|
|
|
|
|
impl<'self,T:Eq> Eq for &'self [T] {
|
|
|
|
fn eq(&self, other: & &'self [T]) -> bool {
|
|
|
|
self.len() == other.len() &&
|
|
|
|
self.iter().zip(other.iter()).all(|(s,o)| *s == *o)
|
|
|
|
}
|
|
|
|
#[inline]
|
|
|
|
fn ne(&self, other: & &'self [T]) -> bool { !self.eq(other) }
|
2012-08-27 19:15:54 -05:00
|
|
|
}
|
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<T:Eq> Eq for ~[T] {
|
|
|
|
#[inline]
|
|
|
|
fn eq(&self, other: &~[T]) -> bool { self.as_slice() == *other }
|
|
|
|
#[inline]
|
|
|
|
fn ne(&self, other: &~[T]) -> bool { !self.eq(other) }
|
2013-03-27 14:20:44 -05:00
|
|
|
}
|
2012-08-27 19:15:54 -05:00
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<T:Eq> Eq for @[T] {
|
|
|
|
#[inline]
|
|
|
|
fn eq(&self, other: &@[T]) -> bool { self.as_slice() == *other }
|
|
|
|
#[inline]
|
|
|
|
fn ne(&self, other: &@[T]) -> bool { !self.eq(other) }
|
|
|
|
}
|
2013-03-27 14:20:44 -05:00
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<'self,T:TotalEq> TotalEq for &'self [T] {
|
|
|
|
fn equals(&self, other: & &'self [T]) -> bool {
|
|
|
|
self.len() == other.len() &&
|
|
|
|
self.iter().zip(other.iter()).all(|(s,o)| s.equals(o))
|
|
|
|
}
|
|
|
|
}
|
2013-03-27 14:20:44 -05:00
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<T:TotalEq> TotalEq for ~[T] {
|
|
|
|
#[inline]
|
|
|
|
fn equals(&self, other: &~[T]) -> bool { self.as_slice().equals(&other.as_slice()) }
|
|
|
|
}
|
2013-03-27 14:20:44 -05:00
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<T:TotalEq> TotalEq for @[T] {
|
|
|
|
#[inline]
|
|
|
|
fn equals(&self, other: &@[T]) -> bool { self.as_slice().equals(&other.as_slice()) }
|
|
|
|
}
|
2012-08-27 19:15:54 -05:00
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<'self,T:Eq, V: Vector<T>> Equiv<V> for &'self [T] {
|
|
|
|
#[inline]
|
|
|
|
fn equiv(&self, other: &V) -> bool { self.as_slice() == other.as_slice() }
|
|
|
|
}
|
2013-03-04 21:43:14 -06:00
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<'self,T:Eq, V: Vector<T>> Equiv<V> for ~[T] {
|
|
|
|
#[inline]
|
|
|
|
fn equiv(&self, other: &V) -> bool { self.as_slice() == other.as_slice() }
|
|
|
|
}
|
2012-08-27 19:15:54 -05:00
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<'self,T:Eq, V: Vector<T>> Equiv<V> for @[T] {
|
|
|
|
#[inline]
|
|
|
|
fn equiv(&self, other: &V) -> bool { self.as_slice() == other.as_slice() }
|
|
|
|
}
|
2013-03-01 21:07:12 -06:00
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<'self,T:TotalOrd> TotalOrd for &'self [T] {
|
|
|
|
fn cmp(&self, other: & &'self [T]) -> Ordering {
|
|
|
|
for self.iter().zip(other.iter()).advance |(s,o)| {
|
|
|
|
match s.cmp(o) {
|
|
|
|
Equal => {},
|
|
|
|
non_eq => { return non_eq; }
|
|
|
|
}
|
|
|
|
}
|
|
|
|
self.len().cmp(&other.len())
|
2013-03-01 21:07:12 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<T: TotalOrd> TotalOrd for ~[T] {
|
|
|
|
#[inline]
|
|
|
|
fn cmp(&self, other: &~[T]) -> Ordering { self.as_slice().cmp(&other.as_slice()) }
|
|
|
|
}
|
2013-03-01 21:07:12 -06:00
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<T: TotalOrd> TotalOrd for @[T] {
|
|
|
|
#[inline]
|
|
|
|
fn cmp(&self, other: &@[T]) -> Ordering { self.as_slice().cmp(&other.as_slice()) }
|
|
|
|
}
|
2013-03-01 21:07:12 -06:00
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<'self,T:Ord> Ord for &'self [T] {
|
|
|
|
fn lt(&self, other: & &'self [T]) -> bool {
|
|
|
|
for self.iter().zip(other.iter()).advance |(s,o)| {
|
|
|
|
if *s < *o { return true; }
|
|
|
|
if *s > *o { return false; }
|
|
|
|
}
|
|
|
|
self.len() < other.len()
|
|
|
|
}
|
|
|
|
#[inline]
|
|
|
|
fn le(&self, other: & &'self [T]) -> bool { !(*other < *self) }
|
|
|
|
#[inline]
|
|
|
|
fn ge(&self, other: & &'self [T]) -> bool { !(*self < *other) }
|
|
|
|
#[inline]
|
|
|
|
fn gt(&self, other: & &'self [T]) -> bool { *other < *self }
|
|
|
|
}
|
2013-03-01 21:07:12 -06:00
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<T:Ord> Ord for ~[T] {
|
|
|
|
#[inline]
|
|
|
|
fn lt(&self, other: &~[T]) -> bool { self.as_slice() < other.as_slice() }
|
|
|
|
#[inline]
|
|
|
|
fn le(&self, other: &~[T]) -> bool { self.as_slice() <= other.as_slice() }
|
|
|
|
#[inline]
|
|
|
|
fn ge(&self, other: &~[T]) -> bool { self.as_slice() >= other.as_slice() }
|
|
|
|
#[inline]
|
|
|
|
fn gt(&self, other: &~[T]) -> bool { self.as_slice() > other.as_slice() }
|
|
|
|
}
|
2012-08-27 19:15:54 -05:00
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<T:Ord> Ord for @[T] {
|
|
|
|
#[inline]
|
|
|
|
fn lt(&self, other: &@[T]) -> bool { self.as_slice() < other.as_slice() }
|
|
|
|
#[inline]
|
|
|
|
fn le(&self, other: &@[T]) -> bool { self.as_slice() <= other.as_slice() }
|
|
|
|
#[inline]
|
|
|
|
fn ge(&self, other: &@[T]) -> bool { self.as_slice() >= other.as_slice() }
|
|
|
|
#[inline]
|
|
|
|
fn gt(&self, other: &@[T]) -> bool { self.as_slice() > other.as_slice() }
|
2012-08-27 19:15:54 -05:00
|
|
|
}
|
|
|
|
|
2013-07-02 14:47:32 -05:00
|
|
|
impl<'self,T:Clone, V: Vector<T>> Add<V, ~[T]> for &'self [T] {
|
2013-07-02 21:13:00 -05:00
|
|
|
#[inline]
|
|
|
|
fn add(&self, rhs: &V) -> ~[T] {
|
|
|
|
let mut res = self.to_owned();
|
|
|
|
res.push_all(rhs.as_slice());
|
|
|
|
res
|
|
|
|
}
|
|
|
|
}
|
2013-07-02 14:47:32 -05:00
|
|
|
impl<T:Clone, V: Vector<T>> Add<V, ~[T]> for ~[T] {
|
2013-07-02 21:13:00 -05:00
|
|
|
#[inline]
|
|
|
|
fn add(&self, rhs: &V) -> ~[T] {
|
|
|
|
let mut res = self.to_owned();
|
|
|
|
res.push_all(rhs.as_slice());
|
|
|
|
res
|
|
|
|
}
|
|
|
|
}
|
2012-08-27 19:15:54 -05:00
|
|
|
}
|
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
#[cfg(test)]
|
|
|
|
pub mod traits {}
|
2012-08-29 21:23:15 -05:00
|
|
|
|
2013-07-02 21:13:00 -05:00
|
|
|
/// Any vector that can be represented as a slice.
|
|
|
|
pub trait Vector<T> {
|
|
|
|
/// Work with `self` as a slice.
|
|
|
|
fn as_slice<'a>(&'a self) -> &'a [T];
|
2012-10-19 08:01:01 -05:00
|
|
|
}
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<'self,T> Vector<T> for &'self [T] {
|
|
|
|
#[inline(always)]
|
|
|
|
fn as_slice<'a>(&'a self) -> &'a [T] { *self }
|
2012-09-19 20:00:26 -05:00
|
|
|
}
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<T> Vector<T> for ~[T] {
|
|
|
|
#[inline(always)]
|
|
|
|
fn as_slice<'a>(&'a self) -> &'a [T] { let v: &'a [T] = *self; v }
|
2012-09-19 20:00:26 -05:00
|
|
|
}
|
2013-07-02 21:13:00 -05:00
|
|
|
impl<T> Vector<T> for @[T] {
|
|
|
|
#[inline(always)]
|
|
|
|
fn as_slice<'a>(&'a self) -> &'a [T] { let v: &'a [T] = *self; v }
|
2012-07-27 16:51:19 -05:00
|
|
|
}
|
|
|
|
|
2013-06-23 22:44:11 -05:00
|
|
|
impl<'self, T> Container for &'self [T] {
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Returns true if a vector contains no elements
|
2012-03-17 20:02:45 -05:00
|
|
|
#[inline]
|
2013-06-15 01:20:06 -05:00
|
|
|
fn is_empty(&self) -> bool {
|
2013-07-03 01:34:17 -05:00
|
|
|
self.as_imm_buf(|_p, len| len == 0u)
|
2013-06-08 20:38:47 -05:00
|
|
|
}
|
2013-01-24 21:19:44 -06:00
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Returns the length of a vector
|
2012-06-02 21:03:28 -05:00
|
|
|
#[inline]
|
2013-06-15 01:20:06 -05:00
|
|
|
fn len(&self) -> uint {
|
2013-07-03 01:34:17 -05:00
|
|
|
self.as_imm_buf(|_p, len| len)
|
2013-06-08 20:38:47 -05:00
|
|
|
}
|
2013-06-15 01:20:06 -05:00
|
|
|
}
|
2013-06-08 20:38:47 -05:00
|
|
|
|
2013-06-15 01:20:06 -05:00
|
|
|
impl<T> Container for ~[T] {
|
|
|
|
/// Returns true if a vector contains no elements
|
|
|
|
#[inline]
|
|
|
|
fn is_empty(&self) -> bool {
|
2013-07-03 01:34:17 -05:00
|
|
|
self.as_imm_buf(|_p, len| len == 0u)
|
2013-06-15 01:20:06 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Returns the length of a vector
|
|
|
|
#[inline]
|
|
|
|
fn len(&self) -> uint {
|
2013-07-03 01:34:17 -05:00
|
|
|
self.as_imm_buf(|_p, len| len)
|
2013-06-15 01:20:06 -05:00
|
|
|
}
|
2012-06-02 21:03:28 -05:00
|
|
|
}
|
|
|
|
|
2013-05-28 16:35:52 -05:00
|
|
|
#[allow(missing_doc)]
|
2012-10-01 17:45:34 -05:00
|
|
|
pub trait CopyableVector<T> {
|
2013-03-21 23:20:48 -05:00
|
|
|
fn to_owned(&self) -> ~[T];
|
2012-07-11 14:45:54 -05:00
|
|
|
}
|
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Extension methods for vectors
|
2013-07-02 14:47:32 -05:00
|
|
|
impl<'self,T:Clone> CopyableVector<T> for &'self [T] {
|
2013-06-13 21:06:47 -05:00
|
|
|
/// Returns a copy of `v`.
|
2012-06-02 21:03:28 -05:00
|
|
|
#[inline]
|
2013-03-21 23:20:48 -05:00
|
|
|
fn to_owned(&self) -> ~[T] {
|
2013-06-27 09:40:47 -05:00
|
|
|
let mut result = with_capacity(self.len());
|
2013-06-21 07:29:53 -05:00
|
|
|
for self.iter().advance |e| {
|
2013-07-02 14:47:32 -05:00
|
|
|
result.push((*e).clone());
|
2013-03-16 13:11:31 -05:00
|
|
|
}
|
2013-03-20 13:13:13 -05:00
|
|
|
result
|
2013-01-07 21:46:45 -06:00
|
|
|
}
|
2012-06-02 21:03:28 -05:00
|
|
|
}
|
|
|
|
|
2013-05-28 16:35:52 -05:00
|
|
|
#[allow(missing_doc)]
|
2013-04-10 15:11:35 -05:00
|
|
|
pub trait ImmutableVector<'self, T> {
|
|
|
|
fn slice(&self, start: uint, end: uint) -> &'self [T];
|
2013-07-21 08:39:01 -05:00
|
|
|
fn slice_from(&self, start: uint) -> &'self [T];
|
|
|
|
fn slice_to(&self, end: uint) -> &'self [T];
|
2013-04-17 18:34:53 -05:00
|
|
|
fn iter(self) -> VecIterator<'self, T>;
|
2013-07-28 15:37:35 -05:00
|
|
|
fn rev_iter(self) -> RevIterator<'self, T>;
|
|
|
|
fn split_iter(self, pred: &'self fn(&T) -> bool) -> SplitIterator<'self, T>;
|
|
|
|
fn splitn_iter(self, n: uint, pred: &'self fn(&T) -> bool) -> SplitIterator<'self, T>;
|
|
|
|
fn rsplit_iter(self, pred: &'self fn(&T) -> bool) -> RSplitIterator<'self, T>;
|
|
|
|
fn rsplitn_iter(self, n: uint, pred: &'self fn(&T) -> bool) -> RSplitIterator<'self, T>;
|
2013-07-02 23:54:11 -05:00
|
|
|
|
2013-07-28 15:37:35 -05:00
|
|
|
fn window_iter(self, size: uint) -> WindowIter<'self, T>;
|
|
|
|
fn chunk_iter(self, size: uint) -> ChunkIter<'self, T>;
|
2013-07-03 00:47:58 -05:00
|
|
|
|
2013-04-10 15:11:35 -05:00
|
|
|
fn head(&self) -> &'self T;
|
|
|
|
fn head_opt(&self) -> Option<&'self T>;
|
|
|
|
fn tail(&self) -> &'self [T];
|
|
|
|
fn tailn(&self, n: uint) -> &'self [T];
|
|
|
|
fn init(&self) -> &'self [T];
|
|
|
|
fn initn(&self, n: uint) -> &'self [T];
|
|
|
|
fn last(&self) -> &'self T;
|
|
|
|
fn last_opt(&self) -> Option<&'self T>;
|
2013-05-14 04:10:50 -05:00
|
|
|
fn rposition(&self, f: &fn(t: &T) -> bool) -> Option<uint>;
|
2013-04-10 15:11:35 -05:00
|
|
|
fn flat_map<U>(&self, f: &fn(t: &T) -> ~[U]) -> ~[U];
|
2013-04-19 17:57:31 -05:00
|
|
|
unsafe fn unsafe_ref(&self, index: uint) -> *T;
|
2013-06-28 22:35:25 -05:00
|
|
|
|
|
|
|
fn bsearch(&self, f: &fn(&T) -> Ordering) -> Option<uint>;
|
2013-06-29 00:05:50 -05:00
|
|
|
|
|
|
|
fn map<U>(&self, &fn(t: &T) -> U) -> ~[U];
|
2013-07-03 01:34:17 -05:00
|
|
|
|
|
|
|
fn as_imm_buf<U>(&self, f: &fn(*T, uint) -> U) -> U;
|
2013-04-10 15:11:35 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Extension methods for vectors
|
|
|
|
impl<'self,T> ImmutableVector<'self, T> for &'self [T] {
|
2013-07-21 08:39:01 -05:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Returns a slice of self between `start` and `end`.
|
|
|
|
*
|
|
|
|
* Fails when `start` or `end` point outside the bounds of self,
|
|
|
|
* or when `start` > `end`.
|
|
|
|
*/
|
2013-04-10 15:11:35 -05:00
|
|
|
#[inline]
|
|
|
|
fn slice(&self, start: uint, end: uint) -> &'self [T] {
|
2013-07-21 08:39:01 -05:00
|
|
|
assert!(start <= end);
|
|
|
|
assert!(end <= self.len());
|
2013-07-03 01:34:17 -05:00
|
|
|
do self.as_imm_buf |p, _len| {
|
2013-06-27 04:48:50 -05:00
|
|
|
unsafe {
|
2013-07-21 19:20:52 -05:00
|
|
|
cast::transmute(Slice {
|
|
|
|
data: ptr::offset(p, start),
|
|
|
|
len: (end - start) * sys::nonzero_size_of::<T>(),
|
|
|
|
})
|
2013-06-27 04:48:50 -05:00
|
|
|
}
|
|
|
|
}
|
2013-04-10 15:11:35 -05:00
|
|
|
}
|
|
|
|
|
2013-07-21 08:39:01 -05:00
|
|
|
/**
|
|
|
|
* Returns a slice of self from `start` to the end of the vec.
|
|
|
|
*
|
|
|
|
* Fails when `start` points outside the bounds of self.
|
|
|
|
*/
|
|
|
|
#[inline]
|
|
|
|
fn slice_from(&self, start: uint) -> &'self [T] {
|
|
|
|
self.slice(start, self.len())
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Returns a slice of self from the start of the vec to `end`.
|
|
|
|
*
|
|
|
|
* Fails when `end` points outside the bounds of self.
|
|
|
|
*/
|
|
|
|
#[inline]
|
|
|
|
fn slice_to(&self, end: uint) -> &'self [T] {
|
|
|
|
self.slice(0, end)
|
|
|
|
}
|
|
|
|
|
2013-04-17 18:34:53 -05:00
|
|
|
#[inline]
|
|
|
|
fn iter(self) -> VecIterator<'self, T> {
|
|
|
|
unsafe {
|
|
|
|
let p = vec::raw::to_ptr(self);
|
2013-07-12 02:59:39 -05:00
|
|
|
VecIterator{ptr: p,
|
2013-07-21 19:20:52 -05:00
|
|
|
end: (p as uint + self.len() *
|
|
|
|
sys::nonzero_size_of::<T>()) as *T,
|
2013-04-17 18:34:53 -05:00
|
|
|
lifetime: cast::transmute(p)}
|
|
|
|
}
|
|
|
|
}
|
2013-07-11 20:10:59 -05:00
|
|
|
|
2013-06-07 21:39:52 -05:00
|
|
|
#[inline]
|
2013-07-28 15:37:35 -05:00
|
|
|
fn rev_iter(self) -> RevIterator<'self, T> {
|
2013-07-10 23:23:59 -05:00
|
|
|
self.iter().invert()
|
2013-06-07 21:39:52 -05:00
|
|
|
}
|
2013-04-17 18:34:53 -05:00
|
|
|
|
2013-07-02 23:54:11 -05:00
|
|
|
/// Returns an iterator over the subslices of the vector which are
|
|
|
|
/// separated by elements that match `pred`.
|
|
|
|
#[inline]
|
2013-07-28 15:37:35 -05:00
|
|
|
fn split_iter(self, pred: &'self fn(&T) -> bool) -> SplitIterator<'self, T> {
|
2013-07-02 23:54:11 -05:00
|
|
|
self.splitn_iter(uint::max_value, pred)
|
|
|
|
}
|
|
|
|
/// Returns an iterator over the subslices of the vector which are
|
|
|
|
/// separated by elements that match `pred`, limited to splitting
|
|
|
|
/// at most `n` times.
|
|
|
|
#[inline]
|
2013-07-28 15:37:35 -05:00
|
|
|
fn splitn_iter(self, n: uint, pred: &'self fn(&T) -> bool) -> SplitIterator<'self, T> {
|
|
|
|
SplitIterator {
|
2013-07-02 23:54:11 -05:00
|
|
|
v: self,
|
|
|
|
n: n,
|
|
|
|
pred: pred,
|
|
|
|
finished: false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
/// Returns an iterator over the subslices of the vector which are
|
|
|
|
/// separated by elements that match `pred`. This starts at the
|
|
|
|
/// end of the vector and works backwards.
|
|
|
|
#[inline]
|
2013-07-28 15:37:35 -05:00
|
|
|
fn rsplit_iter(self, pred: &'self fn(&T) -> bool) -> RSplitIterator<'self, T> {
|
2013-07-02 23:54:11 -05:00
|
|
|
self.rsplitn_iter(uint::max_value, pred)
|
|
|
|
}
|
|
|
|
/// Returns an iterator over the subslices of the vector which are
|
|
|
|
/// separated by elements that match `pred` limited to splitting
|
|
|
|
/// at most `n` times. This starts at the end of the vector and
|
|
|
|
/// works backwards.
|
|
|
|
#[inline]
|
2013-07-28 15:37:35 -05:00
|
|
|
fn rsplitn_iter(self, n: uint, pred: &'self fn(&T) -> bool) -> RSplitIterator<'self, T> {
|
|
|
|
RSplitIterator {
|
2013-07-02 23:54:11 -05:00
|
|
|
v: self,
|
|
|
|
n: n,
|
|
|
|
pred: pred,
|
|
|
|
finished: false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-07-03 00:47:58 -05:00
|
|
|
/**
|
|
|
|
* Returns an iterator over all contiguous windows of length
|
|
|
|
* `size`. The windows overlap. If the vector is shorter than
|
|
|
|
* `size`, the iterator returns no values.
|
|
|
|
*
|
|
|
|
* # Failure
|
|
|
|
*
|
|
|
|
* Fails if `size` is 0.
|
|
|
|
*
|
|
|
|
* # Example
|
|
|
|
*
|
|
|
|
* Print the adjacent pairs of a vector (i.e. `[1,2]`, `[2,3]`,
|
|
|
|
* `[3,4]`):
|
|
|
|
*
|
|
|
|
* ~~~ {.rust}
|
|
|
|
* let v = &[1,2,3,4];
|
|
|
|
* for v.window_iter().advance |win| {
|
2013-07-22 11:03:39 -05:00
|
|
|
* printfln!(win);
|
2013-07-03 00:47:58 -05:00
|
|
|
* }
|
|
|
|
* ~~~
|
|
|
|
*
|
|
|
|
*/
|
2013-07-28 15:37:35 -05:00
|
|
|
fn window_iter(self, size: uint) -> WindowIter<'self, T> {
|
2013-07-03 00:47:58 -05:00
|
|
|
assert!(size != 0);
|
2013-07-28 15:37:35 -05:00
|
|
|
WindowIter { v: self, size: size }
|
2013-07-03 00:47:58 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
*
|
|
|
|
* Returns an iterator over `size` elements of the vector at a
|
|
|
|
* time. The chunks do not overlap. If `size` does not divide the
|
|
|
|
* length of the vector, then the last chunk will not have length
|
|
|
|
* `size`.
|
|
|
|
*
|
|
|
|
* # Failure
|
|
|
|
*
|
|
|
|
* Fails if `size` is 0.
|
|
|
|
*
|
|
|
|
* # Example
|
|
|
|
*
|
|
|
|
* Print the vector two elements at a time (i.e. `[1,2]`,
|
|
|
|
* `[3,4]`, `[5]`):
|
|
|
|
*
|
|
|
|
* ~~~ {.rust}
|
|
|
|
* let v = &[1,2,3,4,5];
|
|
|
|
* for v.chunk_iter().advance |win| {
|
2013-07-22 11:03:39 -05:00
|
|
|
* printfln!(win);
|
2013-07-03 00:47:58 -05:00
|
|
|
* }
|
|
|
|
* ~~~
|
|
|
|
*
|
|
|
|
*/
|
2013-07-28 15:37:35 -05:00
|
|
|
fn chunk_iter(self, size: uint) -> ChunkIter<'self, T> {
|
2013-07-03 00:47:58 -05:00
|
|
|
assert!(size != 0);
|
2013-07-28 15:37:35 -05:00
|
|
|
ChunkIter { v: self, size: size }
|
2013-07-03 00:47:58 -05:00
|
|
|
}
|
|
|
|
|
2013-04-10 15:11:35 -05:00
|
|
|
/// Returns the first element of a vector, failing if the vector is empty.
|
|
|
|
#[inline]
|
2013-06-27 07:36:27 -05:00
|
|
|
fn head(&self) -> &'self T {
|
|
|
|
if self.len() == 0 { fail!("head: empty vector") }
|
|
|
|
&self[0]
|
|
|
|
}
|
2013-04-10 15:11:35 -05:00
|
|
|
|
2013-06-27 07:36:27 -05:00
|
|
|
/// Returns the first element of a vector, or `None` if it is empty
|
2013-04-10 15:11:35 -05:00
|
|
|
#[inline]
|
2013-06-27 07:36:27 -05:00
|
|
|
fn head_opt(&self) -> Option<&'self T> {
|
|
|
|
if self.len() == 0 { None } else { Some(&self[0]) }
|
|
|
|
}
|
2013-04-10 15:11:35 -05:00
|
|
|
|
|
|
|
/// Returns all but the first element of a vector
|
|
|
|
#[inline]
|
2013-06-27 07:36:27 -05:00
|
|
|
fn tail(&self) -> &'self [T] { self.slice(1, self.len()) }
|
2013-04-10 15:11:35 -05:00
|
|
|
|
|
|
|
/// Returns all but the first `n' elements of a vector
|
|
|
|
#[inline]
|
2013-06-27 07:36:27 -05:00
|
|
|
fn tailn(&self, n: uint) -> &'self [T] { self.slice(n, self.len()) }
|
2013-04-10 15:11:35 -05:00
|
|
|
|
2013-06-27 07:36:27 -05:00
|
|
|
/// Returns all but the last element of a vector
|
2013-04-10 15:11:35 -05:00
|
|
|
#[inline]
|
2013-06-27 07:36:27 -05:00
|
|
|
fn init(&self) -> &'self [T] {
|
|
|
|
self.slice(0, self.len() - 1)
|
|
|
|
}
|
2013-04-10 15:11:35 -05:00
|
|
|
|
|
|
|
/// Returns all but the last `n' elemnts of a vector
|
|
|
|
#[inline]
|
2013-06-27 07:36:27 -05:00
|
|
|
fn initn(&self, n: uint) -> &'self [T] {
|
|
|
|
self.slice(0, self.len() - n)
|
|
|
|
}
|
2013-04-10 15:11:35 -05:00
|
|
|
|
2013-06-27 07:36:27 -05:00
|
|
|
/// Returns the last element of a vector, failing if the vector is empty.
|
2013-04-10 15:11:35 -05:00
|
|
|
#[inline]
|
2013-06-27 07:36:27 -05:00
|
|
|
fn last(&self) -> &'self T {
|
|
|
|
if self.len() == 0 { fail!("last: empty vector") }
|
|
|
|
&self[self.len() - 1]
|
|
|
|
}
|
2013-04-10 15:11:35 -05:00
|
|
|
|
2013-06-27 07:36:27 -05:00
|
|
|
/// Returns the last element of a vector, or `None` if it is empty.
|
2013-04-10 15:11:35 -05:00
|
|
|
#[inline]
|
2013-06-27 07:36:27 -05:00
|
|
|
fn last_opt(&self) -> Option<&'self T> {
|
|
|
|
if self.len() == 0 { None } else { Some(&self[self.len() - 1]) }
|
|
|
|
}
|
2013-04-10 15:11:35 -05:00
|
|
|
|
2013-05-14 04:10:50 -05:00
|
|
|
/**
|
|
|
|
* 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
|
2013-06-28 11:08:32 -05:00
|
|
|
* returned. If `f` matches no elements then None is returned.
|
2013-05-14 04:10:50 -05:00
|
|
|
*/
|
|
|
|
#[inline]
|
|
|
|
fn rposition(&self, f: &fn(t: &T) -> bool) -> Option<uint> {
|
2013-06-28 11:08:32 -05:00
|
|
|
for self.rev_iter().enumerate().advance |(i, t)| {
|
|
|
|
if f(t) { return Some(self.len() - i - 1); }
|
2013-04-10 15:11:35 -05:00
|
|
|
}
|
2013-06-28 11:08:32 -05:00
|
|
|
None
|
2013-04-10 15:11:35 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Apply a function to each element of a vector and return a concatenation
|
|
|
|
* of each result vector
|
|
|
|
*/
|
|
|
|
#[inline]
|
|
|
|
fn flat_map<U>(&self, f: &fn(t: &T) -> ~[U]) -> ~[U] {
|
|
|
|
flat_map(*self, f)
|
|
|
|
}
|
2013-07-02 14:47:32 -05:00
|
|
|
|
2013-04-19 17:57:31 -05:00
|
|
|
/// Returns a pointer to the element at the given index, without doing
|
|
|
|
/// bounds checking.
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-04-19 17:57:31 -05:00
|
|
|
unsafe fn unsafe_ref(&self, index: uint) -> *T {
|
2013-07-21 19:20:52 -05:00
|
|
|
self.repr().data.offset(index)
|
2013-04-19 17:57:31 -05:00
|
|
|
}
|
2013-06-28 22:35:25 -05:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Binary search a sorted vector with a comparator function.
|
|
|
|
*
|
|
|
|
* The comparator should implement an order consistent with the sort
|
|
|
|
* order of the underlying vector, returning an order code that indicates
|
|
|
|
* whether its argument is `Less`, `Equal` or `Greater` the desired target.
|
|
|
|
*
|
|
|
|
* Returns the index where the comparator returned `Equal`, or `None` if
|
|
|
|
* not found.
|
|
|
|
*/
|
|
|
|
fn bsearch(&self, f: &fn(&T) -> Ordering) -> Option<uint> {
|
|
|
|
let mut base : uint = 0;
|
|
|
|
let mut lim : uint = self.len();
|
|
|
|
|
|
|
|
while lim != 0 {
|
|
|
|
let ix = base + (lim >> 1);
|
|
|
|
match f(&self[ix]) {
|
|
|
|
Equal => return Some(ix),
|
|
|
|
Less => {
|
|
|
|
base = ix + 1;
|
|
|
|
lim -= 1;
|
|
|
|
}
|
|
|
|
Greater => ()
|
|
|
|
}
|
|
|
|
lim >>= 1;
|
|
|
|
}
|
|
|
|
return None;
|
|
|
|
}
|
2013-06-29 00:05:50 -05:00
|
|
|
|
|
|
|
/// Deprecated, use iterators where possible
|
|
|
|
/// (`self.iter().transform(f)`). Apply a function to each element
|
|
|
|
/// of a vector and return the results.
|
|
|
|
fn map<U>(&self, f: &fn(t: &T) -> U) -> ~[U] {
|
|
|
|
self.iter().transform(f).collect()
|
|
|
|
}
|
2013-07-03 01:34:17 -05:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Work with the buffer of a vector.
|
|
|
|
*
|
|
|
|
* Allows for unsafe manipulation of vector contents, which is useful for
|
|
|
|
* foreign interop.
|
|
|
|
*/
|
|
|
|
#[inline]
|
|
|
|
fn as_imm_buf<U>(&self,
|
|
|
|
/* NB---this CANNOT be const, see below */
|
|
|
|
f: &fn(*T, uint) -> U) -> U {
|
|
|
|
// NB---Do not change the type of s to `&const [T]`. This is
|
|
|
|
// unsound. The reason is that we are going to create immutable pointers
|
|
|
|
// into `s` and pass them to `f()`, but in fact they are potentially
|
|
|
|
// pointing at *mutable memory*. Use `as_mut_buf` instead!
|
|
|
|
|
2013-07-21 19:20:52 -05:00
|
|
|
let s = self.repr();
|
|
|
|
f(s.data, s.len / sys::nonzero_size_of::<T>())
|
2013-07-03 01:34:17 -05:00
|
|
|
}
|
2013-04-10 15:11:35 -05:00
|
|
|
}
|
|
|
|
|
2013-05-28 16:35:52 -05:00
|
|
|
#[allow(missing_doc)]
|
2013-02-20 19:07:17 -06:00
|
|
|
pub trait ImmutableEqVector<T:Eq> {
|
2013-03-21 23:20:48 -05:00
|
|
|
fn position_elem(&self, t: &T) -> Option<uint>;
|
|
|
|
fn rposition_elem(&self, t: &T) -> Option<uint>;
|
2013-06-28 11:08:32 -05:00
|
|
|
fn contains(&self, x: &T) -> bool;
|
2013-01-07 10:38:24 -06:00
|
|
|
}
|
|
|
|
|
2013-03-25 15:21:04 -05:00
|
|
|
impl<'self,T:Eq> ImmutableEqVector<T> for &'self [T] {
|
2012-08-27 18:26:35 -05:00
|
|
|
/// Find the first index containing a matching value
|
|
|
|
#[inline]
|
2013-03-21 23:20:48 -05:00
|
|
|
fn position_elem(&self, x: &T) -> Option<uint> {
|
2013-07-04 21:13:26 -05:00
|
|
|
self.iter().position(|y| *x == *y)
|
2012-09-28 00:20:47 -05:00
|
|
|
}
|
|
|
|
|
2012-08-27 18:26:35 -05:00
|
|
|
/// Find the last index containing a matching value
|
|
|
|
#[inline]
|
2013-03-21 23:20:48 -05:00
|
|
|
fn rposition_elem(&self, t: &T) -> Option<uint> {
|
2013-06-28 11:08:32 -05:00
|
|
|
self.rposition(|x| *x == *t)
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Return true if a vector contains an element with the given value
|
|
|
|
fn contains(&self, x: &T) -> bool {
|
|
|
|
for self.iter().advance |elt| { if *x == *elt { return true; } }
|
|
|
|
false
|
2012-09-28 00:20:47 -05:00
|
|
|
}
|
2012-08-27 18:26:35 -05:00
|
|
|
}
|
|
|
|
|
2013-06-28 22:35:25 -05:00
|
|
|
#[allow(missing_doc)]
|
|
|
|
pub trait ImmutableTotalOrdVector<T: TotalOrd> {
|
|
|
|
fn bsearch_elem(&self, x: &T) -> Option<uint>;
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<'self, T: TotalOrd> ImmutableTotalOrdVector<T> for &'self [T] {
|
|
|
|
/**
|
|
|
|
* Binary search a sorted vector for a given element.
|
|
|
|
*
|
|
|
|
* Returns the index of the element or None if not found.
|
|
|
|
*/
|
|
|
|
fn bsearch_elem(&self, x: &T) -> Option<uint> {
|
|
|
|
self.bsearch(|p| p.cmp(x))
|
2012-09-28 00:20:47 -05:00
|
|
|
}
|
2012-08-27 18:26:35 -05:00
|
|
|
}
|
|
|
|
|
2013-05-28 16:35:52 -05:00
|
|
|
#[allow(missing_doc)]
|
2012-10-01 17:45:34 -05:00
|
|
|
pub trait ImmutableCopyableVector<T> {
|
2013-03-21 23:20:48 -05:00
|
|
|
fn partitioned(&self, f: &fn(&T) -> bool) -> (~[T], ~[T]);
|
2013-04-17 16:19:25 -05:00
|
|
|
unsafe fn unsafe_get(&self, elem: uint) -> T;
|
2012-07-11 14:45:54 -05:00
|
|
|
}
|
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Extension methods for vectors
|
2013-07-02 14:47:32 -05:00
|
|
|
impl<'self,T:Clone> ImmutableCopyableVector<T> for &'self [T] {
|
2013-01-07 10:49:41 -06:00
|
|
|
/**
|
|
|
|
* Partitions the vector into those that satisfies the predicate, and
|
|
|
|
* those that do not.
|
|
|
|
*/
|
|
|
|
#[inline]
|
2013-03-21 23:20:48 -05:00
|
|
|
fn partitioned(&self, f: &fn(&T) -> bool) -> (~[T], ~[T]) {
|
2013-06-27 09:10:18 -05:00
|
|
|
let mut lefts = ~[];
|
|
|
|
let mut rights = ~[];
|
|
|
|
|
|
|
|
for self.iter().advance |elt| {
|
|
|
|
if f(elt) {
|
2013-07-02 14:47:32 -05:00
|
|
|
lefts.push((*elt).clone());
|
2013-06-27 09:10:18 -05:00
|
|
|
} else {
|
2013-07-02 14:47:32 -05:00
|
|
|
rights.push((*elt).clone());
|
2013-06-27 09:10:18 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
(lefts, rights)
|
2013-01-07 10:49:41 -06:00
|
|
|
}
|
2013-04-17 16:19:25 -05:00
|
|
|
|
|
|
|
/// Returns the element at the given index, without doing bounds checking.
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-04-18 17:53:29 -05:00
|
|
|
unsafe fn unsafe_get(&self, index: uint) -> T {
|
2013-07-02 14:47:32 -05:00
|
|
|
(*self.unsafe_ref(index)).clone()
|
2013-04-17 16:19:25 -05:00
|
|
|
}
|
2012-06-02 21:03:28 -05:00
|
|
|
}
|
|
|
|
|
2013-05-28 16:35:52 -05:00
|
|
|
#[allow(missing_doc)]
|
2013-01-07 10:38:24 -06:00
|
|
|
pub trait OwnedVector<T> {
|
2013-07-28 15:37:35 -05:00
|
|
|
fn consume_iter(self) -> ConsumeIterator<T>;
|
|
|
|
fn consume_rev_iter(self) -> ConsumeRevIterator<T>;
|
2013-07-01 10:26:44 -05:00
|
|
|
|
2013-06-27 09:40:47 -05:00
|
|
|
fn reserve(&mut self, n: uint);
|
|
|
|
fn reserve_at_least(&mut self, n: uint);
|
|
|
|
fn capacity(&self) -> uint;
|
|
|
|
|
2012-10-02 13:37:37 -05:00
|
|
|
fn push(&mut self, t: T);
|
2013-06-27 08:53:37 -05:00
|
|
|
unsafe fn push_fast(&mut self, t: T);
|
|
|
|
|
2012-10-02 13:37:37 -05:00
|
|
|
fn push_all_move(&mut self, rhs: ~[T]);
|
2012-09-28 00:20:47 -05:00
|
|
|
fn pop(&mut self) -> T;
|
2013-07-05 13:32:25 -05:00
|
|
|
fn pop_opt(&mut self) -> Option<T>;
|
2012-09-28 00:20:47 -05:00
|
|
|
fn shift(&mut self) -> T;
|
2013-07-05 13:32:25 -05:00
|
|
|
fn shift_opt(&mut self) -> Option<T>;
|
2012-10-02 13:37:37 -05:00
|
|
|
fn unshift(&mut self, x: T);
|
2012-11-25 07:28:16 -06:00
|
|
|
fn insert(&mut self, i: uint, x:T);
|
|
|
|
fn remove(&mut self, i: uint) -> T;
|
2012-09-28 00:20:47 -05:00
|
|
|
fn swap_remove(&mut self, index: uint) -> T;
|
|
|
|
fn truncate(&mut self, newlen: uint);
|
2013-03-21 23:20:48 -05:00
|
|
|
fn retain(&mut self, f: &fn(t: &T) -> bool);
|
|
|
|
fn partition(self, f: &fn(&T) -> bool) -> (~[T], ~[T]);
|
2013-06-23 16:57:39 -05:00
|
|
|
fn grow_fn(&mut self, n: uint, op: &fn(uint) -> T);
|
2012-09-26 19:33:34 -05:00
|
|
|
}
|
|
|
|
|
2013-02-14 13:47:00 -06:00
|
|
|
impl<T> OwnedVector<T> for ~[T] {
|
2013-07-01 10:26:44 -05:00
|
|
|
/// Creates a consuming iterator, that is, one that moves each
|
|
|
|
/// value out of the vector (from start to end). The vector cannot
|
|
|
|
/// be used after calling this.
|
|
|
|
///
|
|
|
|
/// Note that this performs O(n) swaps, and so `consume_rev_iter`
|
|
|
|
/// (which just calls `pop` repeatedly) is more efficient.
|
|
|
|
///
|
|
|
|
/// # Examples
|
|
|
|
///
|
|
|
|
/// ~~~ {.rust}
|
|
|
|
/// let v = ~[~"a", ~"b"];
|
|
|
|
/// for v.consume_iter().advance |s| {
|
|
|
|
/// // s has type ~str, not &~str
|
|
|
|
/// println(s);
|
|
|
|
/// }
|
|
|
|
/// ~~~
|
2013-07-28 15:37:35 -05:00
|
|
|
fn consume_iter(self) -> ConsumeIterator<T> {
|
|
|
|
ConsumeIterator { v: self, idx: 0 }
|
2013-07-01 10:26:44 -05:00
|
|
|
}
|
|
|
|
/// Creates a consuming iterator that moves out of the vector in
|
|
|
|
/// reverse order. Also see `consume_iter`, however note that this
|
|
|
|
/// is more efficient.
|
2013-07-28 15:37:35 -05:00
|
|
|
fn consume_rev_iter(self) -> ConsumeRevIterator<T> {
|
|
|
|
ConsumeRevIterator { v: self }
|
2013-07-01 10:26:44 -05:00
|
|
|
}
|
|
|
|
|
2013-06-27 09:40:47 -05:00
|
|
|
/**
|
|
|
|
* Reserves capacity for exactly `n` elements in the given vector.
|
|
|
|
*
|
|
|
|
* If the capacity for `self` is already equal to or greater than the requested
|
|
|
|
* capacity, then no action is taken.
|
|
|
|
*
|
|
|
|
* # Arguments
|
|
|
|
*
|
|
|
|
* * n - The number of elements to reserve space for
|
|
|
|
*/
|
2013-06-29 21:40:40 -05:00
|
|
|
fn reserve(&mut self, n: uint) {
|
|
|
|
// Only make the (slow) call into the runtime if we have to
|
|
|
|
if self.capacity() < n {
|
|
|
|
unsafe {
|
|
|
|
let td = get_tydesc::<T>();
|
|
|
|
if contains_managed::<T>() {
|
2013-07-21 19:20:52 -05:00
|
|
|
let ptr: *mut *mut Box<Vec<()>> = cast::transmute(self);
|
2013-07-02 19:36:58 -05:00
|
|
|
::at_vec::raw::reserve_raw(td, ptr, n);
|
2013-06-29 21:40:40 -05:00
|
|
|
} else {
|
2013-07-21 19:20:52 -05:00
|
|
|
let ptr: *mut *mut Vec<()> = cast::transmute(self);
|
2013-06-30 21:18:17 -05:00
|
|
|
let alloc = n * sys::nonzero_size_of::<T>();
|
2013-07-21 19:20:52 -05:00
|
|
|
let size = alloc + sys::size_of::<Vec<()>>();
|
2013-07-03 22:59:34 -05:00
|
|
|
if alloc / sys::nonzero_size_of::<T>() != n || size < alloc {
|
|
|
|
fail!("vector size is too large: %u", n);
|
|
|
|
}
|
|
|
|
*ptr = realloc_raw(*ptr as *mut c_void, size)
|
2013-07-21 19:20:52 -05:00
|
|
|
as *mut Vec<()>;
|
2013-07-15 13:23:42 -05:00
|
|
|
(**ptr).alloc = alloc;
|
2013-06-29 21:40:40 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-06-27 09:40:47 -05:00
|
|
|
/**
|
|
|
|
* 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 `self` is already equal to or greater than the requested
|
|
|
|
* capacity, then no action is taken.
|
|
|
|
*
|
|
|
|
* # Arguments
|
|
|
|
*
|
|
|
|
* * n - The number of elements to reserve space for
|
|
|
|
*/
|
|
|
|
fn reserve_at_least(&mut self, n: uint) {
|
|
|
|
self.reserve(uint::next_power_of_two(n));
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Returns the number of elements the vector can hold without reallocating.
|
|
|
|
#[inline]
|
2013-07-15 13:23:42 -05:00
|
|
|
fn capacity(&self) -> uint {
|
|
|
|
unsafe {
|
|
|
|
if contains_managed::<T>() {
|
2013-07-21 19:20:52 -05:00
|
|
|
let repr: **Box<Vec<()>> = cast::transmute(self);
|
|
|
|
(**repr).data.alloc / sys::nonzero_size_of::<T>()
|
2013-07-15 13:23:42 -05:00
|
|
|
} else {
|
2013-07-21 19:20:52 -05:00
|
|
|
let repr: **Vec<()> = cast::transmute(self);
|
2013-07-15 13:23:42 -05:00
|
|
|
(**repr).alloc / sys::nonzero_size_of::<T>()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-06-27 08:53:37 -05:00
|
|
|
/// Append an element to a vector
|
2013-01-07 10:17:03 -06:00
|
|
|
#[inline]
|
2013-07-15 13:23:42 -05:00
|
|
|
fn push(&mut self, t: T) {
|
|
|
|
unsafe {
|
|
|
|
if contains_managed::<T>() {
|
2013-07-21 19:20:52 -05:00
|
|
|
let repr: **Box<Vec<()>> = cast::transmute(&mut *self);
|
|
|
|
let fill = (**repr).data.fill;
|
|
|
|
if (**repr).data.alloc <= fill {
|
2013-07-15 13:23:42 -05:00
|
|
|
let new_len = self.len() + 1;
|
|
|
|
self.reserve_at_least(new_len);
|
|
|
|
}
|
|
|
|
|
|
|
|
self.push_fast(t);
|
|
|
|
} else {
|
2013-07-21 19:20:52 -05:00
|
|
|
let repr: **Vec<()> = cast::transmute(&mut *self);
|
2013-07-15 13:23:42 -05:00
|
|
|
let fill = (**repr).fill;
|
|
|
|
if (**repr).alloc <= fill {
|
|
|
|
let new_len = self.len() + 1;
|
|
|
|
self.reserve_at_least(new_len);
|
|
|
|
}
|
|
|
|
|
|
|
|
self.push_fast(t);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-06-27 08:53:37 -05:00
|
|
|
// This doesn't bother to make sure we have space.
|
|
|
|
#[inline] // really pretty please
|
2013-07-15 13:23:42 -05:00
|
|
|
unsafe fn push_fast(&mut self, t: T) {
|
|
|
|
if contains_managed::<T>() {
|
2013-07-21 19:20:52 -05:00
|
|
|
let repr: **mut Box<Vec<u8>> = cast::transmute(self);
|
|
|
|
let fill = (**repr).data.fill;
|
|
|
|
(**repr).data.fill += sys::nonzero_size_of::<T>();
|
|
|
|
let p = to_unsafe_ptr(&((**repr).data.data));
|
2013-07-15 13:23:42 -05:00
|
|
|
let p = ptr::offset(p, fill) as *mut T;
|
|
|
|
intrinsics::move_val_init(&mut(*p), t);
|
|
|
|
} else {
|
2013-07-21 19:20:52 -05:00
|
|
|
let repr: **mut Vec<u8> = cast::transmute(self);
|
2013-07-15 13:23:42 -05:00
|
|
|
let fill = (**repr).fill;
|
|
|
|
(**repr).fill += sys::nonzero_size_of::<T>();
|
|
|
|
let p = to_unsafe_ptr(&((**repr).data));
|
|
|
|
let p = ptr::offset(p, fill) as *mut T;
|
|
|
|
intrinsics::move_val_init(&mut(*p), t);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-06-27 08:53:37 -05:00
|
|
|
/// Takes ownership of the vector `rhs`, moving all elements into
|
|
|
|
/// the current vector. This does not copy any elements, and it is
|
|
|
|
/// illegal to use the `rhs` vector after calling this method
|
|
|
|
/// (because it is moved here).
|
|
|
|
///
|
|
|
|
/// # Example
|
|
|
|
///
|
|
|
|
/// ~~~ {.rust}
|
|
|
|
/// let mut a = ~[~1];
|
|
|
|
/// a.push_all_move(~[~2, ~3, ~4]);
|
|
|
|
/// assert!(a == ~[~1, ~2, ~3, ~4]);
|
|
|
|
/// ~~~
|
|
|
|
#[inline]
|
|
|
|
fn push_all_move(&mut self, mut rhs: ~[T]) {
|
2013-06-20 14:13:22 -05:00
|
|
|
let self_len = self.len();
|
|
|
|
let rhs_len = rhs.len();
|
|
|
|
let new_len = self_len + rhs_len;
|
2013-06-27 09:40:47 -05:00
|
|
|
self.reserve(new_len);
|
2013-06-20 14:13:22 -05:00
|
|
|
unsafe { // Note: infallible.
|
|
|
|
let self_p = vec::raw::to_mut_ptr(*self);
|
|
|
|
let rhs_p = vec::raw::to_ptr(rhs);
|
|
|
|
ptr::copy_memory(ptr::mut_offset(self_p, self_len), rhs_p, rhs_len);
|
|
|
|
raw::set_len(self, new_len);
|
2013-06-27 08:53:37 -05:00
|
|
|
raw::set_len(&mut rhs, 0);
|
|
|
|
}
|
2012-09-26 19:33:34 -05:00
|
|
|
}
|
2012-09-28 00:20:47 -05:00
|
|
|
|
2013-07-05 13:32:25 -05:00
|
|
|
/// Remove the last element from a vector and return it, or `None` if it is empty
|
|
|
|
fn pop_opt(&mut self) -> Option<T> {
|
|
|
|
match self.len() {
|
|
|
|
0 => None,
|
|
|
|
ln => {
|
|
|
|
let valptr = ptr::to_mut_unsafe_ptr(&mut self[ln - 1u]);
|
|
|
|
unsafe {
|
2013-06-24 12:30:35 -05:00
|
|
|
raw::set_len(self, ln - 1u);
|
2013-07-08 10:05:52 -05:00
|
|
|
Some(ptr::read_ptr(valptr))
|
2013-07-05 13:32:25 -05:00
|
|
|
}
|
|
|
|
}
|
2013-06-27 07:59:52 -05:00
|
|
|
}
|
2012-09-28 00:20:47 -05:00
|
|
|
}
|
|
|
|
|
2013-07-05 13:32:25 -05:00
|
|
|
|
|
|
|
/// Remove the last element from a vector and return it, failing if it is empty
|
|
|
|
#[inline]
|
|
|
|
fn pop(&mut self) -> T {
|
|
|
|
self.pop_opt().expect("pop: empty vector")
|
|
|
|
}
|
|
|
|
|
2013-06-27 07:59:52 -05:00
|
|
|
/// Removes the first element from a vector and return it
|
2013-07-05 13:32:25 -05:00
|
|
|
#[inline]
|
2012-09-28 00:20:47 -05:00
|
|
|
fn shift(&mut self) -> T {
|
2013-07-05 13:32:25 -05:00
|
|
|
self.shift_opt().expect("shift: empty vector")
|
|
|
|
}
|
2013-06-27 07:59:52 -05:00
|
|
|
|
2013-07-05 13:32:25 -05:00
|
|
|
/// Removes the first element from a vector and return it, or `None` if it is empty
|
|
|
|
fn shift_opt(&mut self) -> Option<T> {
|
|
|
|
unsafe {
|
|
|
|
let ln = match self.len() {
|
|
|
|
0 => return None,
|
|
|
|
1 => return self.pop_opt(),
|
|
|
|
2 => {
|
|
|
|
let last = self.pop();
|
|
|
|
let first = self.pop_opt();
|
|
|
|
self.push(last);
|
|
|
|
return first;
|
|
|
|
}
|
|
|
|
x => x
|
|
|
|
};
|
2013-06-27 07:59:52 -05:00
|
|
|
|
|
|
|
let next_ln = self.len() - 1;
|
|
|
|
|
|
|
|
// Save the last element. We're going to overwrite its position
|
|
|
|
let work_elt = self.pop();
|
|
|
|
// We still should have room to work where what last element was
|
2013-06-27 09:40:47 -05:00
|
|
|
assert!(self.capacity() >= ln);
|
2013-06-27 07:59:52 -05:00
|
|
|
// Pretend like we have the original length so we can use
|
|
|
|
// the vector copy_memory to overwrite the hole we just made
|
|
|
|
raw::set_len(self, ln);
|
|
|
|
|
|
|
|
// Memcopy the head element (the one we want) to the location we just
|
|
|
|
// popped. For the moment it unsafely exists at both the head and last
|
|
|
|
// positions
|
|
|
|
{
|
|
|
|
let first_slice = self.slice(0, 1);
|
|
|
|
let last_slice = self.slice(next_ln, ln);
|
2013-07-21 19:20:52 -05:00
|
|
|
raw::copy_memory(cast::transmute(last_slice), first_slice, 1);
|
2013-06-27 07:59:52 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
// Memcopy everything to the left one element
|
|
|
|
{
|
|
|
|
let init_slice = self.slice(0, next_ln);
|
|
|
|
let tail_slice = self.slice(1, ln);
|
2013-07-21 19:20:52 -05:00
|
|
|
raw::copy_memory(cast::transmute(init_slice),
|
2013-06-27 07:59:52 -05:00
|
|
|
tail_slice,
|
|
|
|
next_ln);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Set the new length. Now the vector is back to normal
|
|
|
|
raw::set_len(self, next_ln);
|
|
|
|
|
|
|
|
// Swap out the element we want from the end
|
|
|
|
let vp = raw::to_mut_ptr(*self);
|
|
|
|
let vp = ptr::mut_offset(vp, next_ln - 1);
|
|
|
|
|
2013-07-05 13:32:25 -05:00
|
|
|
Some(ptr::replace_ptr(vp, work_elt))
|
2013-06-27 07:59:52 -05:00
|
|
|
}
|
2012-09-28 00:20:47 -05:00
|
|
|
}
|
|
|
|
|
2013-06-27 07:59:52 -05:00
|
|
|
/// Prepend an element to the vector
|
2012-10-02 13:37:37 -05:00
|
|
|
fn unshift(&mut self, x: T) {
|
2013-06-27 07:59:52 -05:00
|
|
|
let v = util::replace(self, ~[x]);
|
|
|
|
self.push_all_move(v);
|
2012-09-28 00:20:47 -05:00
|
|
|
}
|
|
|
|
|
2013-06-27 07:59:52 -05:00
|
|
|
/// Insert an element at position i within v, shifting all
|
|
|
|
/// elements after position i one position to the right.
|
2012-11-25 07:28:16 -06:00
|
|
|
fn insert(&mut self, i: uint, x:T) {
|
2013-06-27 07:59:52 -05:00
|
|
|
let len = self.len();
|
|
|
|
assert!(i <= len);
|
|
|
|
|
|
|
|
self.push(x);
|
|
|
|
let mut j = len;
|
|
|
|
while j > i {
|
2013-06-28 11:54:03 -05:00
|
|
|
self.swap(j, j - 1);
|
2013-06-27 07:59:52 -05:00
|
|
|
j -= 1;
|
|
|
|
}
|
2012-11-25 07:28:16 -06:00
|
|
|
}
|
|
|
|
|
2013-06-27 07:59:52 -05:00
|
|
|
/// Remove and return the element at position i within v, shifting
|
|
|
|
/// all elements after position i one position to the left.
|
2012-11-25 07:28:16 -06:00
|
|
|
fn remove(&mut self, i: uint) -> T {
|
2013-06-27 07:59:52 -05:00
|
|
|
let len = self.len();
|
|
|
|
assert!(i < len);
|
|
|
|
|
|
|
|
let mut j = i;
|
|
|
|
while j < len - 1 {
|
2013-06-28 11:54:03 -05:00
|
|
|
self.swap(j, j + 1);
|
2013-06-27 07:59:52 -05:00
|
|
|
j += 1;
|
|
|
|
}
|
|
|
|
self.pop()
|
2012-11-25 07:28:16 -06:00
|
|
|
}
|
|
|
|
|
2013-06-27 07:59:52 -05:00
|
|
|
/**
|
|
|
|
* Remove an element from anywhere in the vector and return it, replacing it
|
|
|
|
* with the last element. This does not preserve ordering, but is O(1).
|
|
|
|
*
|
|
|
|
* Fails if index >= length.
|
|
|
|
*/
|
2012-09-28 00:20:47 -05:00
|
|
|
fn swap_remove(&mut self, index: uint) -> T {
|
2013-06-27 07:59:52 -05:00
|
|
|
let ln = self.len();
|
|
|
|
if index >= ln {
|
|
|
|
fail!("vec::swap_remove - index %u >= length %u", index, ln);
|
|
|
|
}
|
|
|
|
if index < ln - 1 {
|
2013-06-28 11:54:03 -05:00
|
|
|
self.swap(index, ln - 1);
|
2013-06-27 07:59:52 -05:00
|
|
|
}
|
|
|
|
self.pop()
|
2012-09-28 00:20:47 -05:00
|
|
|
}
|
|
|
|
|
2013-06-27 08:58:07 -05:00
|
|
|
/// Shorten a vector, dropping excess elements.
|
2012-09-28 00:20:47 -05:00
|
|
|
fn truncate(&mut self, newlen: uint) {
|
2013-07-03 01:34:17 -05:00
|
|
|
do self.as_mut_buf |p, oldlen| {
|
2013-06-27 08:58:07 -05:00
|
|
|
assert!(newlen <= oldlen);
|
|
|
|
unsafe {
|
|
|
|
// This loop is optimized out for non-drop types.
|
|
|
|
for uint::range(newlen, oldlen) |i| {
|
2013-06-24 12:30:35 -05:00
|
|
|
ptr::read_and_zero_ptr(ptr::mut_offset(p, i));
|
2013-06-27 08:58:07 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
unsafe { raw::set_len(self, newlen); }
|
2012-09-28 00:20:47 -05:00
|
|
|
}
|
2012-10-19 08:01:01 -05:00
|
|
|
|
2013-06-27 09:01:21 -05:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Like `filter()`, but in place. Preserves order of `v`. Linear time.
|
|
|
|
*/
|
2013-03-21 23:20:48 -05:00
|
|
|
fn retain(&mut self, f: &fn(t: &T) -> bool) {
|
2013-06-27 09:01:21 -05:00
|
|
|
let len = self.len();
|
|
|
|
let mut deleted: uint = 0;
|
|
|
|
|
|
|
|
for uint::range(0, len) |i| {
|
|
|
|
if !f(&self[i]) {
|
|
|
|
deleted += 1;
|
|
|
|
} else if deleted > 0 {
|
2013-06-28 11:54:03 -05:00
|
|
|
self.swap(i - deleted, i);
|
2013-06-27 09:01:21 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if deleted > 0 {
|
|
|
|
self.truncate(len - deleted);
|
|
|
|
}
|
2012-10-19 08:01:01 -05:00
|
|
|
}
|
2013-01-07 10:17:03 -06:00
|
|
|
|
2013-01-07 10:49:41 -06:00
|
|
|
/**
|
|
|
|
* Partitions the vector into those that satisfies the predicate, and
|
|
|
|
* those that do not.
|
|
|
|
*/
|
|
|
|
#[inline]
|
2013-03-07 16:38:38 -06:00
|
|
|
fn partition(self, f: &fn(&T) -> bool) -> (~[T], ~[T]) {
|
2013-06-27 09:10:18 -05:00
|
|
|
let mut lefts = ~[];
|
|
|
|
let mut rights = ~[];
|
|
|
|
|
2013-07-01 10:26:44 -05:00
|
|
|
for self.consume_iter().advance |elt| {
|
2013-06-27 09:10:18 -05:00
|
|
|
if f(&elt) {
|
|
|
|
lefts.push(elt);
|
|
|
|
} else {
|
|
|
|
rights.push(elt);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
(lefts, rights)
|
2013-01-07 10:49:41 -06:00
|
|
|
}
|
2013-02-17 13:08:04 -06:00
|
|
|
|
2013-06-28 23:02:20 -05:00
|
|
|
/**
|
|
|
|
* Expands a vector in place, initializing the new elements to the result of
|
|
|
|
* a function
|
|
|
|
*
|
|
|
|
* Function `init_op` is called `n` times with the values [0..`n`)
|
|
|
|
*
|
|
|
|
* # Arguments
|
|
|
|
*
|
|
|
|
* * n - The number of elements to add
|
|
|
|
* * init_op - A function to call to retreive each appended element's
|
|
|
|
* value
|
|
|
|
*/
|
2013-06-23 16:57:39 -05:00
|
|
|
fn grow_fn(&mut self, n: uint, op: &fn(uint) -> T) {
|
2013-06-28 23:02:20 -05:00
|
|
|
let new_len = self.len() + n;
|
|
|
|
self.reserve_at_least(new_len);
|
|
|
|
let mut i: uint = 0u;
|
|
|
|
while i < n {
|
|
|
|
self.push(op(i));
|
|
|
|
i += 1u;
|
|
|
|
}
|
2013-02-17 13:08:04 -06:00
|
|
|
}
|
2012-09-26 19:33:34 -05:00
|
|
|
}
|
|
|
|
|
2013-02-14 13:47:00 -06:00
|
|
|
impl<T> Mutable for ~[T] {
|
2013-01-24 22:08:16 -06:00
|
|
|
/// Clear the vector, removing all values.
|
|
|
|
fn clear(&mut self) { self.truncate(0) }
|
|
|
|
}
|
|
|
|
|
2013-05-28 16:35:52 -05:00
|
|
|
#[allow(missing_doc)]
|
2013-07-02 14:47:32 -05:00
|
|
|
pub trait OwnedCopyableVector<T:Clone> {
|
2013-06-23 22:44:11 -05:00
|
|
|
fn push_all(&mut self, rhs: &[T]);
|
2013-01-07 10:38:24 -06:00
|
|
|
fn grow(&mut self, n: uint, initval: &T);
|
|
|
|
fn grow_set(&mut self, index: uint, initval: &T, val: T);
|
|
|
|
}
|
|
|
|
|
2013-07-02 14:47:32 -05:00
|
|
|
impl<T:Clone> OwnedCopyableVector<T> for ~[T] {
|
2013-06-27 08:53:37 -05:00
|
|
|
/// Iterates over the slice `rhs`, copies each element, and then appends it to
|
|
|
|
/// the vector provided `v`. The `rhs` vector is traversed in-order.
|
|
|
|
///
|
|
|
|
/// # Example
|
|
|
|
///
|
|
|
|
/// ~~~ {.rust}
|
|
|
|
/// let mut a = ~[1];
|
|
|
|
/// a.push_all([2, 3, 4]);
|
|
|
|
/// assert!(a == ~[1, 2, 3, 4]);
|
|
|
|
/// ~~~
|
2013-01-07 10:17:03 -06:00
|
|
|
#[inline]
|
2013-06-23 22:44:11 -05:00
|
|
|
fn push_all(&mut self, rhs: &[T]) {
|
2013-06-27 08:53:37 -05:00
|
|
|
let new_len = self.len() + rhs.len();
|
2013-06-27 09:40:47 -05:00
|
|
|
self.reserve(new_len);
|
2013-06-27 08:53:37 -05:00
|
|
|
|
|
|
|
for uint::range(0u, rhs.len()) |i| {
|
|
|
|
self.push(unsafe { raw::get(rhs, i) })
|
|
|
|
}
|
2012-09-26 19:33:34 -05:00
|
|
|
}
|
2012-09-28 00:20:47 -05:00
|
|
|
|
2013-06-28 23:02:20 -05:00
|
|
|
/**
|
|
|
|
* Expands a vector in place, initializing the new elements to a given value
|
|
|
|
*
|
|
|
|
* # Arguments
|
|
|
|
*
|
|
|
|
* * n - The number of elements to add
|
|
|
|
* * initval - The value for the new elements
|
|
|
|
*/
|
2012-09-28 00:20:47 -05:00
|
|
|
fn grow(&mut self, n: uint, initval: &T) {
|
2013-06-28 23:02:20 -05:00
|
|
|
let new_len = self.len() + n;
|
|
|
|
self.reserve_at_least(new_len);
|
|
|
|
let mut i: uint = 0u;
|
|
|
|
|
|
|
|
while i < n {
|
2013-07-02 14:47:32 -05:00
|
|
|
self.push((*initval).clone());
|
2013-06-28 23:02:20 -05:00
|
|
|
i += 1u;
|
|
|
|
}
|
2012-09-28 00:20:47 -05:00
|
|
|
}
|
|
|
|
|
2013-06-28 23:02:20 -05:00
|
|
|
/**
|
|
|
|
* 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.
|
|
|
|
*/
|
2012-10-02 13:37:37 -05:00
|
|
|
fn grow_set(&mut self, index: uint, initval: &T, val: T) {
|
2013-06-28 23:02:20 -05:00
|
|
|
let l = self.len();
|
|
|
|
if index >= l { self.grow(index - l + 1u, initval); }
|
|
|
|
self[index] = val;
|
2012-09-28 00:20:47 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-05-28 16:35:52 -05:00
|
|
|
#[allow(missing_doc)]
|
2013-06-28 22:41:09 -05:00
|
|
|
pub trait OwnedEqVector<T:Eq> {
|
2013-01-07 10:38:24 -06:00
|
|
|
fn dedup(&mut self);
|
|
|
|
}
|
|
|
|
|
2013-02-20 19:07:17 -06:00
|
|
|
impl<T:Eq> OwnedEqVector<T> for ~[T] {
|
2013-06-28 22:41:09 -05:00
|
|
|
/**
|
2013-06-20 14:13:22 -05:00
|
|
|
* Remove consecutive repeated elements from a vector; if the vector is
|
|
|
|
* sorted, this removes all duplicates.
|
|
|
|
*/
|
2013-06-24 12:30:35 -05:00
|
|
|
pub fn dedup(&mut self) {
|
2013-06-28 22:41:09 -05:00
|
|
|
unsafe {
|
2013-06-20 14:13:22 -05:00
|
|
|
// Although we have a mutable reference to `self`, we cannot make
|
|
|
|
// *arbitrary* changes. There exists the possibility that this
|
|
|
|
// vector is contained with an `@mut` box and hence is still
|
|
|
|
// readable by the outside world during the `Eq` comparisons.
|
|
|
|
// Moreover, those comparisons could fail, so we must ensure
|
|
|
|
// that the vector is in a valid state at all time.
|
|
|
|
//
|
|
|
|
// The way that we handle this is by using swaps; we iterate
|
|
|
|
// over all the elements, swapping as we go so that at the end
|
|
|
|
// the elements we wish to keep are in the front, and those we
|
|
|
|
// wish to reject are at the back. We can then truncate the
|
|
|
|
// vector. This operation is still O(n).
|
|
|
|
//
|
|
|
|
// Example: We start in this state, where `r` represents "next
|
|
|
|
// read" and `w` represents "next_write`.
|
|
|
|
//
|
|
|
|
// r
|
|
|
|
// +---+---+---+---+---+---+
|
|
|
|
// | 0 | 1 | 1 | 2 | 3 | 3 |
|
|
|
|
// +---+---+---+---+---+---+
|
|
|
|
// w
|
|
|
|
//
|
|
|
|
// Comparing self[r] against self[w-1], tis is not a duplicate, so
|
|
|
|
// we swap self[r] and self[w] (no effect as r==w) and then increment both
|
|
|
|
// r and w, leaving us with:
|
|
|
|
//
|
|
|
|
// r
|
|
|
|
// +---+---+---+---+---+---+
|
|
|
|
// | 0 | 1 | 1 | 2 | 3 | 3 |
|
|
|
|
// +---+---+---+---+---+---+
|
|
|
|
// w
|
|
|
|
//
|
|
|
|
// Comparing self[r] against self[w-1], this value is a duplicate,
|
|
|
|
// so we increment `r` but leave everything else unchanged:
|
|
|
|
//
|
|
|
|
// r
|
|
|
|
// +---+---+---+---+---+---+
|
|
|
|
// | 0 | 1 | 1 | 2 | 3 | 3 |
|
|
|
|
// +---+---+---+---+---+---+
|
|
|
|
// w
|
|
|
|
//
|
|
|
|
// Comparing self[r] against self[w-1], this is not a duplicate,
|
|
|
|
// so swap self[r] and self[w] and advance r and w:
|
|
|
|
//
|
|
|
|
// r
|
|
|
|
// +---+---+---+---+---+---+
|
|
|
|
// | 0 | 1 | 2 | 1 | 3 | 3 |
|
|
|
|
// +---+---+---+---+---+---+
|
|
|
|
// w
|
|
|
|
//
|
|
|
|
// Not a duplicate, repeat:
|
|
|
|
//
|
|
|
|
// r
|
|
|
|
// +---+---+---+---+---+---+
|
|
|
|
// | 0 | 1 | 2 | 3 | 1 | 3 |
|
|
|
|
// +---+---+---+---+---+---+
|
|
|
|
// w
|
|
|
|
//
|
|
|
|
// Duplicate, advance r. End of vec. Truncate to w.
|
|
|
|
|
|
|
|
let ln = self.len();
|
|
|
|
if ln < 1 { return; }
|
|
|
|
|
|
|
|
// Avoid bounds checks by using unsafe pointers.
|
|
|
|
let p = vec::raw::to_mut_ptr(*self);
|
|
|
|
let mut r = 1;
|
|
|
|
let mut w = 1;
|
|
|
|
|
|
|
|
while r < ln {
|
|
|
|
let p_r = ptr::mut_offset(p, r);
|
|
|
|
let p_wm1 = ptr::mut_offset(p, w - 1);
|
|
|
|
if *p_r != *p_wm1 {
|
|
|
|
if r != w {
|
|
|
|
let p_w = ptr::mut_offset(p_wm1, 1);
|
|
|
|
util::swap(&mut *p_r, &mut *p_w);
|
2013-06-28 22:41:09 -05:00
|
|
|
}
|
2013-06-20 14:13:22 -05:00
|
|
|
w += 1;
|
2013-06-28 22:41:09 -05:00
|
|
|
}
|
2013-06-20 14:13:22 -05:00
|
|
|
r += 1;
|
2013-06-28 22:41:09 -05:00
|
|
|
}
|
2013-06-20 14:13:22 -05:00
|
|
|
|
|
|
|
self.truncate(w);
|
2013-06-28 22:41:09 -05:00
|
|
|
}
|
2012-09-28 00:20:47 -05:00
|
|
|
}
|
2012-09-26 19:33:34 -05:00
|
|
|
}
|
|
|
|
|
2013-05-28 16:35:52 -05:00
|
|
|
#[allow(missing_doc)]
|
2013-05-26 20:40:07 -05:00
|
|
|
pub trait MutableVector<'self, T> {
|
2013-05-29 00:56:21 -05:00
|
|
|
fn mut_slice(self, start: uint, end: uint) -> &'self mut [T];
|
2013-07-26 19:48:56 -05:00
|
|
|
fn mut_slice_from(self, start: uint) -> &'self mut [T];
|
|
|
|
fn mut_slice_to(self, end: uint) -> &'self mut [T];
|
2013-06-07 21:39:52 -05:00
|
|
|
fn mut_iter(self) -> VecMutIterator<'self, T>;
|
2013-07-28 15:37:35 -05:00
|
|
|
fn mut_rev_iter(self) -> MutRevIterator<'self, T>;
|
2013-05-26 20:40:07 -05:00
|
|
|
|
2013-06-28 11:54:03 -05:00
|
|
|
fn swap(self, a: uint, b: uint);
|
|
|
|
|
2013-07-10 08:50:24 -05:00
|
|
|
/**
|
|
|
|
* Divides one `&mut` into two. The first will
|
|
|
|
* contain all indices from `0..mid` (excluding the index `mid`
|
|
|
|
* itself) and the second will contain all indices from
|
|
|
|
* `mid..len` (excluding the index `len` itself).
|
|
|
|
*/
|
|
|
|
fn mut_split(self, mid: uint) -> (&'self mut [T],
|
|
|
|
&'self mut [T]);
|
|
|
|
|
2013-06-28 11:54:03 -05:00
|
|
|
fn reverse(self);
|
|
|
|
|
2013-06-18 01:52:14 -05:00
|
|
|
/**
|
|
|
|
* Consumes `src` and moves as many elements as it can into `self`
|
|
|
|
* from the range [start,end).
|
|
|
|
*
|
|
|
|
* Returns the number of elements copied (the shorter of self.len()
|
|
|
|
* and end - start).
|
|
|
|
*
|
|
|
|
* # Arguments
|
|
|
|
*
|
|
|
|
* * src - A mutable vector of `T`
|
|
|
|
* * start - The index into `src` to start copying from
|
|
|
|
* * end - The index into `str` to stop copying from
|
|
|
|
*/
|
|
|
|
fn move_from(self, src: ~[T], start: uint, end: uint) -> uint;
|
|
|
|
|
2013-07-21 19:20:52 -05:00
|
|
|
unsafe fn unsafe_mut_ref(self, index: uint) -> *mut T;
|
|
|
|
unsafe fn unsafe_set(self, index: uint, val: T);
|
2013-07-03 01:34:17 -05:00
|
|
|
|
2013-07-23 14:31:00 -05:00
|
|
|
fn as_mut_buf<U>(self, f: &fn(*mut T, uint) -> U) -> U;
|
2013-04-17 16:19:25 -05:00
|
|
|
}
|
|
|
|
|
2013-05-26 20:40:07 -05:00
|
|
|
impl<'self,T> MutableVector<'self, T> for &'self mut [T] {
|
2013-06-27 04:48:50 -05:00
|
|
|
/// Return a slice that points into another slice.
|
2013-05-26 20:40:07 -05:00
|
|
|
#[inline]
|
2013-05-29 00:56:21 -05:00
|
|
|
fn mut_slice(self, start: uint, end: uint) -> &'self mut [T] {
|
2013-06-27 04:48:50 -05:00
|
|
|
assert!(start <= end);
|
|
|
|
assert!(end <= self.len());
|
2013-07-03 01:34:17 -05:00
|
|
|
do self.as_mut_buf |p, _len| {
|
2013-06-27 04:48:50 -05:00
|
|
|
unsafe {
|
2013-07-21 19:20:52 -05:00
|
|
|
cast::transmute(Slice {
|
|
|
|
data: ptr::mut_offset(p, start) as *T,
|
|
|
|
len: (end - start) * sys::nonzero_size_of::<T>()
|
|
|
|
})
|
2013-06-27 04:48:50 -05:00
|
|
|
}
|
|
|
|
}
|
2013-05-26 20:40:07 -05:00
|
|
|
}
|
|
|
|
|
2013-07-26 19:48:56 -05:00
|
|
|
/**
|
|
|
|
* Returns a slice of self from `start` to the end of the vec.
|
|
|
|
*
|
|
|
|
* Fails when `start` points outside the bounds of self.
|
|
|
|
*/
|
|
|
|
#[inline]
|
|
|
|
fn mut_slice_from(self, start: uint) -> &'self mut [T] {
|
|
|
|
let len = self.len();
|
|
|
|
self.mut_slice(start, len)
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Returns a slice of self from the start of the vec to `end`.
|
|
|
|
*
|
|
|
|
* Fails when `end` points outside the bounds of self.
|
|
|
|
*/
|
|
|
|
#[inline]
|
|
|
|
fn mut_slice_to(self, end: uint) -> &'self mut [T] {
|
|
|
|
self.mut_slice(0, end)
|
|
|
|
}
|
|
|
|
|
2013-07-10 08:50:24 -05:00
|
|
|
#[inline]
|
|
|
|
fn mut_split(self, mid: uint) -> (&'self mut [T], &'self mut [T]) {
|
|
|
|
unsafe {
|
|
|
|
let len = self.len();
|
|
|
|
let self2: &'self mut [T] = cast::transmute_copy(&self);
|
|
|
|
(self.mut_slice(0, mid), self2.mut_slice(mid, len))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-06-06 00:12:39 -05:00
|
|
|
#[inline]
|
2013-06-07 21:39:52 -05:00
|
|
|
fn mut_iter(self) -> VecMutIterator<'self, T> {
|
2013-06-06 00:12:39 -05:00
|
|
|
unsafe {
|
|
|
|
let p = vec::raw::to_mut_ptr(self);
|
2013-07-12 02:59:39 -05:00
|
|
|
VecMutIterator{ptr: p,
|
2013-07-21 19:20:52 -05:00
|
|
|
end: (p as uint + self.len() *
|
|
|
|
sys::nonzero_size_of::<T>()) as *mut T,
|
2013-06-06 00:12:39 -05:00
|
|
|
lifetime: cast::transmute(p)}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-07-10 23:23:59 -05:00
|
|
|
#[inline]
|
2013-07-28 15:37:35 -05:00
|
|
|
fn mut_rev_iter(self) -> MutRevIterator<'self, T> {
|
2013-07-10 23:23:59 -05:00
|
|
|
self.mut_iter().invert()
|
2013-06-07 21:39:52 -05:00
|
|
|
}
|
|
|
|
|
2013-06-28 11:54:03 -05:00
|
|
|
/**
|
|
|
|
* Swaps two elements in a vector
|
|
|
|
*
|
|
|
|
* # Arguments
|
|
|
|
*
|
|
|
|
* * a - The index of the first element
|
|
|
|
* * b - The index of the second element
|
|
|
|
*/
|
|
|
|
fn swap(self, a: uint, b: uint) {
|
|
|
|
unsafe {
|
|
|
|
// Can't take two mutable loans from one vector, so instead just cast
|
|
|
|
// them to their raw pointers to do the swap
|
|
|
|
let pa: *mut T = &mut self[a];
|
|
|
|
let pb: *mut T = &mut self[b];
|
|
|
|
ptr::swap_ptr(pa, pb);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Reverse the order of elements in a vector, in place
|
|
|
|
fn reverse(self) {
|
|
|
|
let mut i: uint = 0;
|
|
|
|
let ln = self.len();
|
|
|
|
while i < ln / 2 {
|
|
|
|
self.swap(i, ln - i - 1);
|
|
|
|
i += 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-06-18 01:52:14 -05:00
|
|
|
#[inline]
|
|
|
|
fn move_from(self, mut src: ~[T], start: uint, end: uint) -> uint {
|
|
|
|
for self.mut_iter().zip(src.mut_slice(start, end).mut_iter()).advance |(a, b)| {
|
|
|
|
util::swap(a, b);
|
|
|
|
}
|
|
|
|
cmp::min(self.len(), end-start)
|
|
|
|
}
|
|
|
|
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-07-21 19:20:52 -05:00
|
|
|
unsafe fn unsafe_mut_ref(self, index: uint) -> *mut T {
|
|
|
|
ptr::mut_offset(self.repr().data as *mut T, index)
|
2013-04-18 17:53:29 -05:00
|
|
|
}
|
|
|
|
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-07-21 19:20:52 -05:00
|
|
|
unsafe fn unsafe_set(self, index: uint, val: T) {
|
2013-04-18 17:53:29 -05:00
|
|
|
*self.unsafe_mut_ref(index) = val;
|
2013-04-17 16:19:25 -05:00
|
|
|
}
|
2013-07-03 01:34:17 -05:00
|
|
|
|
|
|
|
/// Similar to `as_imm_buf` but passing a `*mut T`
|
|
|
|
#[inline]
|
2013-07-23 14:31:00 -05:00
|
|
|
fn as_mut_buf<U>(self, f: &fn(*mut T, uint) -> U) -> U {
|
2013-07-21 19:20:52 -05:00
|
|
|
let Slice{ data, len } = self.repr();
|
|
|
|
f(data as *mut T, len / sys::nonzero_size_of::<T>())
|
2013-07-03 01:34:17 -05:00
|
|
|
}
|
|
|
|
|
2013-04-17 16:19:25 -05:00
|
|
|
}
|
|
|
|
|
2013-07-06 02:20:08 -05:00
|
|
|
/// Trait for &[T] where T is Cloneable
|
2013-06-18 01:52:14 -05:00
|
|
|
pub trait MutableCloneableVector<T> {
|
|
|
|
/// Copies as many elements from `src` as it can into `self`
|
|
|
|
/// (the shorter of self.len() and src.len()). Returns the number of elements copied.
|
|
|
|
fn copy_from(self, &[T]) -> uint;
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<'self, T:Clone> MutableCloneableVector<T> for &'self mut [T] {
|
|
|
|
#[inline]
|
|
|
|
fn copy_from(self, src: &[T]) -> uint {
|
|
|
|
for self.mut_iter().zip(src.iter()).advance |(a, b)| {
|
|
|
|
*a = b.clone();
|
|
|
|
}
|
|
|
|
cmp::min(self.len(), src.len())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-10-11 17:37:37 -05:00
|
|
|
/**
|
|
|
|
* Constructs a vector from an unsafe pointer to a buffer
|
|
|
|
*
|
|
|
|
* # Arguments
|
|
|
|
*
|
|
|
|
* * ptr - An unsafe pointer to a buffer of `T`
|
|
|
|
* * elts - The number of elements in the buffer
|
|
|
|
*/
|
|
|
|
// Wrapper for fn in raw: needs to be called by net_tcp::on_tcp_read_cb
|
|
|
|
pub unsafe fn from_buf<T>(ptr: *T, elts: uint) -> ~[T] {
|
|
|
|
raw::from_buf_raw(ptr, elts)
|
|
|
|
}
|
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Unsafe operations
|
2012-12-14 17:47:11 -06:00
|
|
|
pub mod raw {
|
2013-07-21 19:20:52 -05:00
|
|
|
use cast;
|
2013-07-02 14:47:32 -05:00
|
|
|
use clone::Clone;
|
2013-07-17 16:41:50 -05:00
|
|
|
use option::Some;
|
2012-12-23 16:41:37 -06:00
|
|
|
use ptr;
|
|
|
|
use sys;
|
2013-04-22 16:27:30 -05:00
|
|
|
use unstable::intrinsics;
|
2013-07-21 19:20:52 -05:00
|
|
|
use vec::{with_capacity, ImmutableVector, MutableVector};
|
2013-07-15 13:23:42 -05:00
|
|
|
use unstable::intrinsics::contains_managed;
|
2013-07-21 19:20:52 -05:00
|
|
|
use unstable::raw::{Box, Vec, Slice};
|
2012-07-12 21:44:00 -05:00
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/**
|
|
|
|
* Sets the length of a vector
|
|
|
|
*
|
|
|
|
* This will 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.
|
|
|
|
*/
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-07-15 13:23:42 -05:00
|
|
|
pub unsafe fn set_len<T>(v: &mut ~[T], new_len: uint) {
|
|
|
|
if contains_managed::<T>() {
|
2013-07-21 19:20:52 -05:00
|
|
|
let repr: **mut Box<Vec<()>> = cast::transmute(v);
|
|
|
|
(**repr).data.fill = new_len * sys::nonzero_size_of::<T>();
|
2013-07-15 13:23:42 -05:00
|
|
|
} else {
|
2013-07-21 19:20:52 -05:00
|
|
|
let repr: **mut Vec<()> = cast::transmute(v);
|
2013-07-15 13:23:42 -05:00
|
|
|
(**repr).fill = new_len * sys::nonzero_size_of::<T>();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/**
|
|
|
|
* 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.
|
|
|
|
*/
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-05-15 23:34:32 -05:00
|
|
|
pub fn to_ptr<T>(v: &[T]) -> *T {
|
2013-07-21 19:20:52 -05:00
|
|
|
v.repr().data
|
2012-07-12 21:44:00 -05:00
|
|
|
}
|
|
|
|
|
2012-09-12 12:38:17 -05:00
|
|
|
/** see `to_ptr()` */
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-05-15 23:34:32 -05:00
|
|
|
pub fn to_mut_ptr<T>(v: &mut [T]) -> *mut T {
|
2013-07-21 19:20:52 -05:00
|
|
|
v.repr().data as *mut T
|
2012-09-12 12:38:17 -05:00
|
|
|
}
|
2012-04-25 19:18:06 -05:00
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/**
|
|
|
|
* Form a slice from a pointer and length (as a number of units,
|
|
|
|
* not bytes).
|
|
|
|
*/
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2012-09-26 23:35:42 -05:00
|
|
|
pub unsafe fn buf_as_slice<T,U>(p: *T,
|
|
|
|
len: uint,
|
2013-03-07 16:38:38 -06:00
|
|
|
f: &fn(v: &[T]) -> U) -> U {
|
2013-07-21 19:20:52 -05:00
|
|
|
f(cast::transmute(Slice {
|
|
|
|
data: p,
|
|
|
|
len: len * sys::nonzero_size_of::<T>()
|
|
|
|
}))
|
2012-04-25 19:18:06 -05:00
|
|
|
}
|
2012-07-20 21:20:13 -05:00
|
|
|
|
2013-03-09 21:14:35 -06:00
|
|
|
/**
|
|
|
|
* Form a slice from a pointer and length (as a number of units,
|
|
|
|
* not bytes).
|
|
|
|
*/
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-03-09 21:14:35 -06:00
|
|
|
pub unsafe fn mut_buf_as_slice<T,U>(p: *mut T,
|
|
|
|
len: uint,
|
|
|
|
f: &fn(v: &mut [T]) -> U) -> U {
|
2013-07-21 19:20:52 -05:00
|
|
|
f(cast::transmute(Slice {
|
|
|
|
data: p as *T,
|
|
|
|
len: len * sys::nonzero_size_of::<T>()
|
|
|
|
}))
|
2013-03-09 21:14:35 -06:00
|
|
|
}
|
|
|
|
|
2012-08-01 14:37:13 -05:00
|
|
|
/**
|
|
|
|
* Unchecked vector indexing.
|
|
|
|
*/
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-07-02 14:47:32 -05:00
|
|
|
pub unsafe fn get<T:Clone>(v: &[T], i: uint) -> T {
|
|
|
|
v.as_imm_buf(|p, _len| (*ptr::offset(p, i)).clone())
|
2012-08-01 14:37:13 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
2012-09-28 00:20:47 -05:00
|
|
|
* Unchecked vector index assignment. Does not drop the
|
|
|
|
* old value and hence is only suitable when the vector
|
|
|
|
* is newly allocated.
|
2012-08-01 14:37:13 -05:00
|
|
|
*/
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-02-12 17:34:48 -06:00
|
|
|
pub unsafe fn init_elem<T>(v: &mut [T], i: uint, val: T) {
|
2012-12-12 17:38:50 -06:00
|
|
|
let mut box = Some(val);
|
2013-07-03 01:34:17 -05:00
|
|
|
do v.as_mut_buf |p, _len| {
|
2013-02-20 10:57:15 -06:00
|
|
|
intrinsics::move_val_init(&mut(*ptr::mut_offset(p, i)),
|
2013-07-17 16:41:50 -05:00
|
|
|
box.take_unwrap());
|
2012-08-01 14:37:13 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-10-11 17:37:37 -05:00
|
|
|
/**
|
|
|
|
* Constructs a vector from an unsafe pointer to a buffer
|
|
|
|
*
|
|
|
|
* # Arguments
|
|
|
|
*
|
|
|
|
* * ptr - An unsafe pointer to a buffer of `T`
|
|
|
|
* * elts - The number of elements in the buffer
|
|
|
|
*/
|
|
|
|
// Was in raw, but needs to be called by net_tcp::on_tcp_read_cb
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2012-10-11 17:37:37 -05:00
|
|
|
pub unsafe fn from_buf_raw<T>(ptr: *T, elts: uint) -> ~[T] {
|
|
|
|
let mut dst = with_capacity(elts);
|
|
|
|
set_len(&mut dst, elts);
|
2013-07-03 01:34:17 -05:00
|
|
|
dst.as_mut_buf(|p_dst, _len_dst| ptr::copy_memory(p_dst, ptr, elts));
|
2012-12-12 17:38:50 -06:00
|
|
|
dst
|
2012-10-11 17:37:37 -05:00
|
|
|
}
|
|
|
|
|
2012-07-20 21:20:13 -05:00
|
|
|
/**
|
|
|
|
* Copies data from one vector to another.
|
|
|
|
*
|
|
|
|
* Copies `count` bytes from `src` to `dst`. The source and destination
|
|
|
|
* may overlap.
|
|
|
|
*/
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-06-23 22:44:11 -05:00
|
|
|
pub unsafe fn copy_memory<T>(dst: &mut [T], src: &[T],
|
2013-01-13 18:34:54 -06:00
|
|
|
count: uint) {
|
2013-03-28 20:39:09 -05:00
|
|
|
assert!(dst.len() >= count);
|
|
|
|
assert!(src.len() >= count);
|
2013-01-05 04:52:37 -06:00
|
|
|
|
2013-07-03 01:34:17 -05:00
|
|
|
do dst.as_mut_buf |p_dst, _len_dst| {
|
|
|
|
do src.as_imm_buf |p_src, _len_src| {
|
2013-01-10 01:11:04 -06:00
|
|
|
ptr::copy_memory(p_dst, p_src, count)
|
2012-07-20 21:20:13 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2011-12-13 18:25:51 -06:00
|
|
|
}
|
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Operations on `[u8]`
|
2012-10-01 17:45:34 -05:00
|
|
|
pub mod bytes {
|
2012-12-23 16:41:37 -06:00
|
|
|
use libc;
|
Replaces the free-standing functions in f32, &c.
The free-standing functions in f32, f64, i8, i16, i32, i64, u8, u16,
u32, u64, float, int, and uint are replaced with generic functions in
num instead.
If you were previously using any of those functions, just replace them
with the corresponding function with the same name in num.
Note: If you were using a function that corresponds to an operator, use
the operator instead.
2013-07-08 11:05:17 -05:00
|
|
|
use num;
|
2013-01-08 21:29:16 -06:00
|
|
|
use vec::raw;
|
2013-01-08 21:37:25 -06:00
|
|
|
use vec;
|
2013-06-18 01:20:53 -05:00
|
|
|
use ptr;
|
|
|
|
|
|
|
|
/// A trait for operations on mutable operations on `[u8]`
|
|
|
|
pub trait MutableByteVector {
|
|
|
|
/// Sets all bytes of the receiver to the given value.
|
|
|
|
pub fn set_memory(self, value: u8);
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<'self> MutableByteVector for &'self mut [u8] {
|
|
|
|
#[inline]
|
|
|
|
fn set_memory(self, value: u8) {
|
2013-07-03 01:34:17 -05:00
|
|
|
do self.as_mut_buf |p, len| {
|
2013-06-18 01:20:53 -05:00
|
|
|
unsafe { ptr::set_memory(p, value, len) };
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2012-01-06 09:36:56 -06:00
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Bytewise string comparison
|
2013-03-21 23:20:48 -05:00
|
|
|
pub fn memcmp(a: &~[u8], b: &~[u8]) -> int {
|
2013-03-09 15:41:43 -06:00
|
|
|
let a_len = a.len();
|
|
|
|
let b_len = b.len();
|
Replaces the free-standing functions in f32, &c.
The free-standing functions in f32, f64, i8, i16, i32, i64, u8, u16,
u32, u64, float, int, and uint are replaced with generic functions in
num instead.
If you were previously using any of those functions, just replace them
with the corresponding function with the same name in num.
Note: If you were using a function that corresponds to an operator, use
the operator instead.
2013-07-08 11:05:17 -05:00
|
|
|
let n = num::min(a_len, b_len) as libc::size_t;
|
2012-06-24 22:18:18 -05:00
|
|
|
let r = unsafe {
|
2012-09-12 19:45:23 -05:00
|
|
|
libc::memcmp(raw::to_ptr(*a) as *libc::c_void,
|
|
|
|
raw::to_ptr(*b) as *libc::c_void, n) as int
|
2012-06-24 22:18:18 -05:00
|
|
|
};
|
2012-01-06 09:36:56 -06:00
|
|
|
|
|
|
|
if r != 0 { r } else {
|
|
|
|
if a_len == b_len {
|
|
|
|
0
|
|
|
|
} else if a_len < b_len {
|
|
|
|
-1
|
|
|
|
} else {
|
|
|
|
1
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Bytewise less than or equal
|
2013-03-21 23:20:48 -05:00
|
|
|
pub fn lt(a: &~[u8], b: &~[u8]) -> bool { memcmp(a, b) < 0 }
|
2012-01-06 09:36:56 -06:00
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Bytewise less than or equal
|
2013-03-21 23:20:48 -05:00
|
|
|
pub fn le(a: &~[u8], b: &~[u8]) -> bool { memcmp(a, b) <= 0 }
|
2012-01-06 09:36:56 -06:00
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Bytewise equality
|
2013-03-21 23:20:48 -05:00
|
|
|
pub fn eq(a: &~[u8], b: &~[u8]) -> bool { memcmp(a, b) == 0 }
|
2012-01-06 09:36:56 -06:00
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Bytewise inequality
|
2013-03-21 23:20:48 -05:00
|
|
|
pub fn ne(a: &~[u8], b: &~[u8]) -> bool { memcmp(a, b) != 0 }
|
2012-01-06 09:36:56 -06:00
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Bytewise greater than or equal
|
2013-03-21 23:20:48 -05:00
|
|
|
pub fn ge(a: &~[u8], b: &~[u8]) -> bool { memcmp(a, b) >= 0 }
|
2012-01-06 09:36:56 -06:00
|
|
|
|
2012-07-04 16:53:12 -05:00
|
|
|
/// Bytewise greater than
|
2013-03-21 23:20:48 -05:00
|
|
|
pub fn gt(a: &~[u8], b: &~[u8]) -> bool { memcmp(a, b) > 0 }
|
2012-01-06 09:36:56 -06:00
|
|
|
|
2012-07-20 21:20:13 -05:00
|
|
|
/**
|
|
|
|
* Copies data from one vector to another.
|
|
|
|
*
|
|
|
|
* Copies `count` bytes from `src` to `dst`. The source and destination
|
2013-01-15 01:36:45 -06:00
|
|
|
* may overlap.
|
2012-07-20 21:20:13 -05:00
|
|
|
*/
|
2013-06-18 16:45:18 -05:00
|
|
|
#[inline]
|
2013-06-23 22:44:11 -05:00
|
|
|
pub fn copy_memory(dst: &mut [u8], src: &[u8], count: uint) {
|
2013-01-10 01:11:04 -06:00
|
|
|
// Bound checks are done at vec::raw::copy_memory.
|
|
|
|
unsafe { vec::raw::copy_memory(dst, src, count) }
|
2012-07-20 21:20:13 -05:00
|
|
|
}
|
2012-01-06 09:36:56 -06:00
|
|
|
}
|
|
|
|
|
2013-03-15 17:26:59 -05:00
|
|
|
impl<A:Clone> Clone for ~[A] {
|
|
|
|
#[inline]
|
|
|
|
fn clone(&self) -> ~[A] {
|
2013-06-29 00:05:50 -05:00
|
|
|
self.iter().transform(|item| item.clone()).collect()
|
2013-03-15 17:26:59 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-06-17 02:05:51 -05:00
|
|
|
// This works because every lifetime is a sub-lifetime of 'static
|
|
|
|
impl<'self, A> Zero for &'self [A] {
|
|
|
|
fn zero() -> &'self [A] { &'self [] }
|
|
|
|
fn is_zero(&self) -> bool { self.is_empty() }
|
|
|
|
}
|
|
|
|
|
2013-06-14 20:27:52 -05:00
|
|
|
impl<A> Zero for ~[A] {
|
|
|
|
fn zero() -> ~[A] { ~[] }
|
|
|
|
fn is_zero(&self) -> bool { self.len() == 0 }
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<A> Zero for @[A] {
|
|
|
|
fn zero() -> @[A] { @[] }
|
|
|
|
fn is_zero(&self) -> bool { self.len() == 0 }
|
|
|
|
}
|
|
|
|
|
2013-06-07 21:39:52 -05:00
|
|
|
macro_rules! iterator {
|
|
|
|
/* FIXME: #4375 Cannot attach documentation/attributes to a macro generated struct.
|
|
|
|
(struct $name:ident -> $ptr:ty, $elem:ty) => {
|
|
|
|
pub struct $name<'self, T> {
|
|
|
|
priv ptr: $ptr,
|
|
|
|
priv end: $ptr,
|
|
|
|
priv lifetime: $elem // FIXME: #5922
|
|
|
|
}
|
|
|
|
};*/
|
2013-07-10 23:23:59 -05:00
|
|
|
(impl $name:ident -> $elem:ty) => {
|
2013-06-07 21:39:52 -05:00
|
|
|
impl<'self, T> Iterator<$elem> for $name<'self, T> {
|
|
|
|
#[inline]
|
|
|
|
fn next(&mut self) -> Option<$elem> {
|
2013-07-10 23:23:59 -05:00
|
|
|
// could be implemented with slices, but this avoids bounds checks
|
2013-06-07 21:39:52 -05:00
|
|
|
unsafe {
|
|
|
|
if self.ptr == self.end {
|
|
|
|
None
|
|
|
|
} else {
|
|
|
|
let old = self.ptr;
|
2013-07-12 02:59:39 -05:00
|
|
|
// purposefully don't use 'ptr.offset' because for
|
|
|
|
// vectors with 0-size elements this would return the
|
|
|
|
// same pointer.
|
|
|
|
self.ptr = cast::transmute(self.ptr as uint +
|
|
|
|
sys::nonzero_size_of::<T>());
|
2013-06-07 21:39:52 -05:00
|
|
|
Some(cast::transmute(old))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2013-06-21 05:12:01 -05:00
|
|
|
|
|
|
|
#[inline]
|
2013-07-02 20:40:46 -05:00
|
|
|
fn size_hint(&self) -> (uint, Option<uint>) {
|
2013-07-22 19:11:24 -05:00
|
|
|
let exact = self.indexable();
|
2013-07-02 20:40:46 -05:00
|
|
|
(exact, Some(exact))
|
2013-06-21 05:12:01 -05:00
|
|
|
}
|
2013-06-07 21:39:52 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-07-10 23:23:59 -05:00
|
|
|
macro_rules! double_ended_iterator {
|
|
|
|
(impl $name:ident -> $elem:ty) => {
|
|
|
|
impl<'self, T> DoubleEndedIterator<$elem> for $name<'self, T> {
|
|
|
|
#[inline]
|
|
|
|
fn next_back(&mut self) -> Option<$elem> {
|
|
|
|
// could be implemented with slices, but this avoids bounds checks
|
|
|
|
unsafe {
|
|
|
|
if self.end == self.ptr {
|
|
|
|
None
|
|
|
|
} else {
|
2013-07-12 02:59:39 -05:00
|
|
|
// See above for why 'ptr.offset' isn't used
|
|
|
|
self.end = cast::transmute(self.end as uint -
|
|
|
|
sys::nonzero_size_of::<T>());
|
2013-07-10 23:23:59 -05:00
|
|
|
Some(cast::transmute(self.end))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-07-22 19:11:24 -05:00
|
|
|
macro_rules! random_access_iterator {
|
|
|
|
(impl $name:ident -> $elem:ty) => {
|
|
|
|
impl<'self, T> RandomAccessIterator<$elem> for $name<'self, T> {
|
|
|
|
#[inline]
|
|
|
|
fn indexable(&self) -> uint {
|
|
|
|
let diff = (self.end as uint) - (self.ptr as uint);
|
|
|
|
diff / sys::nonzero_size_of::<T>()
|
|
|
|
}
|
|
|
|
|
|
|
|
fn idx(&self, index: uint) -> Option<$elem> {
|
|
|
|
unsafe {
|
|
|
|
if index < self.indexable() {
|
|
|
|
cast::transmute(self.ptr.offset(index))
|
|
|
|
} else {
|
|
|
|
None
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-06-07 21:39:52 -05:00
|
|
|
//iterator!{struct VecIterator -> *T, &'self T}
|
2013-07-01 10:26:44 -05:00
|
|
|
/// An iterator for iterating over a vector.
|
2013-04-17 18:34:53 -05:00
|
|
|
pub struct VecIterator<'self, T> {
|
|
|
|
priv ptr: *T,
|
|
|
|
priv end: *T,
|
|
|
|
priv lifetime: &'self T // FIXME: #5922
|
|
|
|
}
|
2013-07-10 23:23:59 -05:00
|
|
|
iterator!{impl VecIterator -> &'self T}
|
|
|
|
double_ended_iterator!{impl VecIterator -> &'self T}
|
2013-07-22 19:11:24 -05:00
|
|
|
random_access_iterator!{impl VecIterator -> &'self T}
|
2013-07-28 15:37:35 -05:00
|
|
|
pub type RevIterator<'self, T> = InvertIterator<VecIterator<'self, T>>;
|
2012-06-02 21:03:28 -05:00
|
|
|
|
2013-07-18 10:38:17 -05:00
|
|
|
impl<'self, T> Clone for VecIterator<'self, T> {
|
|
|
|
fn clone(&self) -> VecIterator<'self, T> { *self }
|
|
|
|
}
|
|
|
|
|
2013-06-07 21:39:52 -05:00
|
|
|
//iterator!{struct VecMutIterator -> *mut T, &'self mut T}
|
2013-07-01 10:26:44 -05:00
|
|
|
/// An iterator for mutating the elements of a vector.
|
2013-06-07 21:39:52 -05:00
|
|
|
pub struct VecMutIterator<'self, T> {
|
2013-06-06 00:12:39 -05:00
|
|
|
priv ptr: *mut T,
|
|
|
|
priv end: *mut T,
|
|
|
|
priv lifetime: &'self mut T // FIXME: #5922
|
|
|
|
}
|
2013-07-10 23:23:59 -05:00
|
|
|
iterator!{impl VecMutIterator -> &'self mut T}
|
|
|
|
double_ended_iterator!{impl VecMutIterator -> &'self mut T}
|
2013-07-22 19:11:24 -05:00
|
|
|
random_access_iterator!{impl VecMutIterator -> &'self mut T}
|
2013-07-28 15:37:35 -05:00
|
|
|
pub type MutRevIterator<'self, T> = InvertIterator<VecMutIterator<'self, T>>;
|
2013-06-06 00:12:39 -05:00
|
|
|
|
2013-07-01 10:26:44 -05:00
|
|
|
/// An iterator that moves out of a vector.
|
2013-07-18 10:38:17 -05:00
|
|
|
#[deriving(Clone)]
|
2013-07-28 15:37:35 -05:00
|
|
|
pub struct ConsumeIterator<T> {
|
2013-07-01 10:26:44 -05:00
|
|
|
priv v: ~[T],
|
|
|
|
priv idx: uint,
|
|
|
|
}
|
|
|
|
|
2013-07-28 15:37:35 -05:00
|
|
|
impl<T> Iterator<T> for ConsumeIterator<T> {
|
2013-07-01 10:26:44 -05:00
|
|
|
fn next(&mut self) -> Option<T> {
|
|
|
|
// this is peculiar, but is required for safety with respect
|
|
|
|
// to dtors. It traverses the first half of the vec, and
|
|
|
|
// removes them by swapping them with the last element (and
|
|
|
|
// popping), which results in the second half in reverse
|
|
|
|
// order, and so these can just be pop'd off. That is,
|
|
|
|
//
|
|
|
|
// [1,2,3,4,5] => 1, [5,2,3,4] => 2, [5,4,3] => 3, [5,4] => 4,
|
|
|
|
// [5] -> 5, []
|
2013-07-05 13:32:25 -05:00
|
|
|
let l = self.v.len();
|
|
|
|
if self.idx < l {
|
|
|
|
self.v.swap(self.idx, l - 1);
|
|
|
|
self.idx += 1;
|
2013-07-01 10:26:44 -05:00
|
|
|
}
|
2013-07-05 13:32:25 -05:00
|
|
|
|
|
|
|
self.v.pop_opt()
|
2013-07-01 10:26:44 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// An iterator that moves out of a vector in reverse order.
|
2013-07-18 10:38:17 -05:00
|
|
|
#[deriving(Clone)]
|
2013-07-28 15:37:35 -05:00
|
|
|
pub struct ConsumeRevIterator<T> {
|
2013-07-01 10:26:44 -05:00
|
|
|
priv v: ~[T]
|
|
|
|
}
|
|
|
|
|
2013-07-28 15:37:35 -05:00
|
|
|
impl<T> Iterator<T> for ConsumeRevIterator<T> {
|
2013-07-01 10:26:44 -05:00
|
|
|
fn next(&mut self) -> Option<T> {
|
2013-07-05 13:32:25 -05:00
|
|
|
self.v.pop_opt()
|
2013-07-01 10:26:44 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-06-21 06:57:22 -05:00
|
|
|
impl<A, T: Iterator<A>> FromIterator<A, T> for ~[A] {
|
2013-07-27 16:41:30 -05:00
|
|
|
fn from_iterator(iterator: &mut T) -> ~[A] {
|
2013-06-21 06:57:22 -05:00
|
|
|
let (lower, _) = iterator.size_hint();
|
2013-07-02 20:40:46 -05:00
|
|
|
let mut xs = with_capacity(lower);
|
2013-06-21 06:57:22 -05:00
|
|
|
for iterator.advance |x| {
|
|
|
|
xs.push(x);
|
|
|
|
}
|
|
|
|
xs
|
|
|
|
}
|
|
|
|
}
|
2013-06-25 16:07:44 -05:00
|
|
|
|
2013-07-27 16:41:30 -05:00
|
|
|
impl<A, T: Iterator<A>> Extendable<A, T> for ~[A] {
|
|
|
|
fn extend(&mut self, iterator: &mut T) {
|
|
|
|
let (lower, _) = iterator.size_hint();
|
|
|
|
let len = self.len();
|
|
|
|
self.reserve(len + lower);
|
|
|
|
for iterator.advance |x| {
|
|
|
|
self.push(x);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-01-17 19:28:21 -06:00
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
2013-01-08 21:37:25 -06:00
|
|
|
use option::{None, Option, Some};
|
2013-02-25 13:11:21 -06:00
|
|
|
use sys;
|
2013-01-08 21:37:25 -06:00
|
|
|
use vec::*;
|
2013-03-01 21:07:12 -06:00
|
|
|
use cmp::*;
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2013-03-09 15:41:43 -06:00
|
|
|
fn square(n: uint) -> uint { n * n }
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2013-03-09 15:41:43 -06:00
|
|
|
fn square_ref(n: &uint) -> uint { square(*n) }
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2013-03-21 23:20:48 -05:00
|
|
|
fn is_three(n: &uint) -> bool { *n == 3u }
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2013-03-21 23:20:48 -05:00
|
|
|
fn is_odd(n: &uint) -> bool { *n % 2u == 1u }
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2013-03-21 23:20:48 -05:00
|
|
|
fn is_equal(x: &uint, y:&uint) -> bool { *x == *y }
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2013-01-31 19:12:29 -06:00
|
|
|
fn square_if_odd_r(n: &uint) -> Option<uint> {
|
2013-03-09 15:41:43 -06:00
|
|
|
if *n % 2u == 1u { Some(*n * *n) } else { None }
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
2013-01-31 19:12:29 -06:00
|
|
|
fn square_if_odd_v(n: uint) -> Option<uint> {
|
2013-03-09 15:41:43 -06:00
|
|
|
if n % 2u == 1u { Some(n * n) } else { None }
|
2013-01-31 19:12:29 -06:00
|
|
|
}
|
|
|
|
|
2013-03-09 15:41:43 -06:00
|
|
|
fn add(x: uint, y: &uint) -> uint { x + *y }
|
2012-01-17 19:28:21 -06:00
|
|
|
|
|
|
|
#[test]
|
2012-06-24 22:18:18 -05:00
|
|
|
fn test_unsafe_ptrs() {
|
|
|
|
unsafe {
|
|
|
|
// Test on-stack copy-from-buf.
|
2012-06-29 18:26:56 -05:00
|
|
|
let a = ~[1, 2, 3];
|
2012-09-12 19:45:23 -05:00
|
|
|
let mut ptr = raw::to_ptr(a);
|
2012-10-11 18:18:36 -05:00
|
|
|
let b = from_buf(ptr, 3u);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(b.len(), 3u);
|
|
|
|
assert_eq!(b[0], 1);
|
|
|
|
assert_eq!(b[1], 2);
|
|
|
|
assert_eq!(b[2], 3);
|
2012-06-24 22:18:18 -05:00
|
|
|
|
|
|
|
// Test on-heap copy-from-buf.
|
2012-06-29 18:26:56 -05:00
|
|
|
let c = ~[1, 2, 3, 4, 5];
|
2012-09-12 19:45:23 -05:00
|
|
|
ptr = raw::to_ptr(c);
|
2012-10-11 18:18:36 -05:00
|
|
|
let d = from_buf(ptr, 5u);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(d.len(), 5u);
|
|
|
|
assert_eq!(d[0], 1);
|
|
|
|
assert_eq!(d[1], 2);
|
|
|
|
assert_eq!(d[2], 3);
|
|
|
|
assert_eq!(d[3], 4);
|
|
|
|
assert_eq!(d[4], 5);
|
2012-06-24 22:18:18 -05:00
|
|
|
}
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2012-03-12 17:52:30 -05:00
|
|
|
fn test_from_fn() {
|
|
|
|
// Test on-stack from_fn.
|
2012-03-22 10:39:41 -05:00
|
|
|
let mut v = from_fn(3u, square);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 3u);
|
|
|
|
assert_eq!(v[0], 0u);
|
|
|
|
assert_eq!(v[1], 1u);
|
|
|
|
assert_eq!(v[2], 4u);
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2012-03-12 17:52:30 -05:00
|
|
|
// Test on-heap from_fn.
|
|
|
|
v = from_fn(5u, square);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 5u);
|
|
|
|
assert_eq!(v[0], 0u);
|
|
|
|
assert_eq!(v[1], 1u);
|
|
|
|
assert_eq!(v[2], 4u);
|
|
|
|
assert_eq!(v[3], 9u);
|
|
|
|
assert_eq!(v[4], 16u);
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2012-03-12 17:52:30 -05:00
|
|
|
fn test_from_elem() {
|
|
|
|
// Test on-stack from_elem.
|
2012-03-22 10:39:41 -05:00
|
|
|
let mut v = from_elem(2u, 10u);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 2u);
|
|
|
|
assert_eq!(v[0], 10u);
|
|
|
|
assert_eq!(v[1], 10u);
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2012-03-12 17:52:30 -05:00
|
|
|
// Test on-heap from_elem.
|
|
|
|
v = from_elem(6u, 20u);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v[0], 20u);
|
|
|
|
assert_eq!(v[1], 20u);
|
|
|
|
assert_eq!(v[2], 20u);
|
|
|
|
assert_eq!(v[3], 20u);
|
|
|
|
assert_eq!(v[4], 20u);
|
|
|
|
assert_eq!(v[5], 20u);
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_is_empty() {
|
2013-06-08 20:38:47 -05:00
|
|
|
let xs: [int, ..0] = [];
|
|
|
|
assert!(xs.is_empty());
|
|
|
|
assert!(![0].is_empty());
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
2013-01-08 02:24:43 -06:00
|
|
|
#[test]
|
|
|
|
fn test_len_divzero() {
|
2013-03-22 20:52:04 -05:00
|
|
|
type Z = [i8, ..0];
|
2013-01-08 02:24:43 -06:00
|
|
|
let v0 : &[Z] = &[];
|
|
|
|
let v1 : &[Z] = &[[]];
|
|
|
|
let v2 : &[Z] = &[[], []];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(sys::size_of::<Z>(), 0);
|
|
|
|
assert_eq!(v0.len(), 0);
|
|
|
|
assert_eq!(v1.len(), 1);
|
|
|
|
assert_eq!(v2.len(), 2);
|
2013-01-08 02:24:43 -06:00
|
|
|
}
|
|
|
|
|
2012-01-17 19:28:21 -06:00
|
|
|
#[test]
|
|
|
|
fn test_head() {
|
2013-03-02 23:49:50 -06:00
|
|
|
let mut a = ~[11];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.head(), &11);
|
2013-03-02 23:49:50 -06:00
|
|
|
a = ~[11, 12];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.head(), &11);
|
2013-03-02 23:49:50 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[should_fail]
|
|
|
|
#[ignore(cfg(windows))]
|
|
|
|
fn test_head_empty() {
|
|
|
|
let a: ~[int] = ~[];
|
|
|
|
a.head();
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_head_opt() {
|
|
|
|
let mut a = ~[];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.head_opt(), None);
|
2013-03-02 23:49:50 -06:00
|
|
|
a = ~[11];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.head_opt().unwrap(), &11);
|
2013-03-02 23:49:50 -06:00
|
|
|
a = ~[11, 12];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.head_opt().unwrap(), &11);
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_tail() {
|
2012-06-29 18:26:56 -05:00
|
|
|
let mut a = ~[11];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.tail(), &[]);
|
2012-06-29 18:26:56 -05:00
|
|
|
a = ~[11, 12];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.tail(), &[12]);
|
2013-03-03 09:22:40 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[should_fail]
|
|
|
|
#[ignore(cfg(windows))]
|
|
|
|
fn test_tail_empty() {
|
|
|
|
let a: ~[int] = ~[];
|
|
|
|
a.tail();
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_tailn() {
|
|
|
|
let mut a = ~[11, 12, 13];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.tailn(0), &[11, 12, 13]);
|
2013-03-03 09:22:40 -06:00
|
|
|
a = ~[11, 12, 13];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.tailn(2), &[13]);
|
2013-03-03 09:22:40 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[should_fail]
|
|
|
|
#[ignore(cfg(windows))]
|
|
|
|
fn test_tailn_empty() {
|
|
|
|
let a: ~[int] = ~[];
|
|
|
|
a.tailn(2);
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
2013-03-03 10:06:31 -06:00
|
|
|
#[test]
|
|
|
|
fn test_init() {
|
|
|
|
let mut a = ~[11];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.init(), &[]);
|
2013-03-03 10:06:31 -06:00
|
|
|
a = ~[11, 12];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.init(), &[11]);
|
2013-03-03 10:06:31 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[init]
|
|
|
|
#[should_fail]
|
|
|
|
#[ignore(cfg(windows))]
|
|
|
|
fn test_init_empty() {
|
|
|
|
let a: ~[int] = ~[];
|
|
|
|
a.init();
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_initn() {
|
|
|
|
let mut a = ~[11, 12, 13];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.initn(0), &[11, 12, 13]);
|
2013-03-03 10:06:31 -06:00
|
|
|
a = ~[11, 12, 13];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.initn(2), &[11]);
|
2013-03-03 10:06:31 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[init]
|
|
|
|
#[should_fail]
|
|
|
|
#[ignore(cfg(windows))]
|
|
|
|
fn test_initn_empty() {
|
|
|
|
let a: ~[int] = ~[];
|
|
|
|
a.initn(2);
|
|
|
|
}
|
|
|
|
|
2012-01-17 19:28:21 -06:00
|
|
|
#[test]
|
|
|
|
fn test_last() {
|
2013-03-05 21:39:18 -06:00
|
|
|
let mut a = ~[11];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.last(), &11);
|
2013-03-05 21:39:18 -06:00
|
|
|
a = ~[11, 12];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.last(), &12);
|
2013-03-05 21:39:18 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[should_fail]
|
2013-03-07 00:02:01 -06:00
|
|
|
#[ignore(cfg(windows))]
|
2013-03-05 21:39:18 -06:00
|
|
|
fn test_last_empty() {
|
|
|
|
let a: ~[int] = ~[];
|
|
|
|
a.last();
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_last_opt() {
|
|
|
|
let mut a = ~[];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.last_opt(), None);
|
2013-03-05 21:39:18 -06:00
|
|
|
a = ~[11];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.last_opt().unwrap(), &11);
|
2013-03-05 21:39:18 -06:00
|
|
|
a = ~[11, 12];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a.last_opt().unwrap(), &12);
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_slice() {
|
2013-02-08 13:28:20 -06:00
|
|
|
// Test fixed length vector.
|
|
|
|
let vec_fixed = [1, 2, 3, 4];
|
2013-06-27 04:48:50 -05:00
|
|
|
let v_a = vec_fixed.slice(1u, vec_fixed.len()).to_owned();
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v_a.len(), 3u);
|
|
|
|
assert_eq!(v_a[0], 2);
|
|
|
|
assert_eq!(v_a[1], 3);
|
|
|
|
assert_eq!(v_a[2], 4);
|
2013-02-08 13:28:20 -06:00
|
|
|
|
|
|
|
// Test on stack.
|
|
|
|
let vec_stack = &[1, 2, 3];
|
2013-06-27 04:48:50 -05:00
|
|
|
let v_b = vec_stack.slice(1u, 3u).to_owned();
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v_b.len(), 2u);
|
|
|
|
assert_eq!(v_b[0], 2);
|
|
|
|
assert_eq!(v_b[1], 3);
|
2013-02-08 13:28:20 -06:00
|
|
|
|
|
|
|
// Test on managed heap.
|
|
|
|
let vec_managed = @[1, 2, 3, 4, 5];
|
2013-06-27 04:48:50 -05:00
|
|
|
let v_c = vec_managed.slice(0u, 3u).to_owned();
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v_c.len(), 3u);
|
|
|
|
assert_eq!(v_c[0], 1);
|
|
|
|
assert_eq!(v_c[1], 2);
|
|
|
|
assert_eq!(v_c[2], 3);
|
2013-02-08 13:28:20 -06:00
|
|
|
|
|
|
|
// Test on exchange heap.
|
|
|
|
let vec_unique = ~[1, 2, 3, 4, 5, 6];
|
2013-06-27 04:48:50 -05:00
|
|
|
let v_d = vec_unique.slice(1u, 6u).to_owned();
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v_d.len(), 5u);
|
|
|
|
assert_eq!(v_d[0], 2);
|
|
|
|
assert_eq!(v_d[1], 3);
|
|
|
|
assert_eq!(v_d[2], 4);
|
|
|
|
assert_eq!(v_d[3], 5);
|
|
|
|
assert_eq!(v_d[4], 6);
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
2013-07-21 08:39:01 -05:00
|
|
|
#[test]
|
|
|
|
fn test_slice_from() {
|
|
|
|
let vec = &[1, 2, 3, 4];
|
|
|
|
assert_eq!(vec.slice_from(0), vec);
|
|
|
|
assert_eq!(vec.slice_from(2), &[3, 4]);
|
|
|
|
assert_eq!(vec.slice_from(4), &[]);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_slice_to() {
|
|
|
|
let vec = &[1, 2, 3, 4];
|
|
|
|
assert_eq!(vec.slice_to(4), vec);
|
|
|
|
assert_eq!(vec.slice_to(2), &[1, 2]);
|
|
|
|
assert_eq!(vec.slice_to(0), &[]);
|
|
|
|
}
|
|
|
|
|
2012-01-17 19:28:21 -06:00
|
|
|
#[test]
|
|
|
|
fn test_pop() {
|
|
|
|
// Test on-heap pop.
|
2013-02-08 13:28:20 -06:00
|
|
|
let mut v = ~[1, 2, 3, 4, 5];
|
|
|
|
let e = v.pop();
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 4u);
|
|
|
|
assert_eq!(v[0], 1);
|
|
|
|
assert_eq!(v[1], 2);
|
|
|
|
assert_eq!(v[2], 3);
|
|
|
|
assert_eq!(v[3], 4);
|
|
|
|
assert_eq!(e, 5);
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
2012-08-22 21:09:34 -05:00
|
|
|
#[test]
|
2013-07-05 13:32:25 -05:00
|
|
|
fn test_pop_opt() {
|
|
|
|
let mut v = ~[5];
|
|
|
|
let e = v.pop_opt();
|
|
|
|
assert_eq!(v.len(), 0);
|
|
|
|
assert_eq!(e, Some(5));
|
|
|
|
let f = v.pop_opt();
|
|
|
|
assert_eq!(f, None);
|
|
|
|
let g = v.pop_opt();
|
|
|
|
assert_eq!(g, None);
|
|
|
|
}
|
|
|
|
|
2012-08-22 21:09:34 -05:00
|
|
|
fn test_swap_remove() {
|
|
|
|
let mut v = ~[1, 2, 3, 4, 5];
|
2012-09-28 00:20:47 -05:00
|
|
|
let mut e = v.swap_remove(0);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 4);
|
|
|
|
assert_eq!(e, 1);
|
|
|
|
assert_eq!(v[0], 5);
|
2012-09-28 00:20:47 -05:00
|
|
|
e = v.swap_remove(3);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 3);
|
|
|
|
assert_eq!(e, 4);
|
|
|
|
assert_eq!(v[0], 5);
|
|
|
|
assert_eq!(v[1], 2);
|
|
|
|
assert_eq!(v[2], 3);
|
2012-08-22 21:09:34 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_swap_remove_noncopyable() {
|
|
|
|
// Tests that we don't accidentally run destructors twice.
|
2013-07-22 15:57:40 -05:00
|
|
|
let mut v = ~[::unstable::sync::Exclusive::new(()),
|
|
|
|
::unstable::sync::Exclusive::new(()),
|
|
|
|
::unstable::sync::Exclusive::new(())];
|
2012-09-28 00:20:47 -05:00
|
|
|
let mut _e = v.swap_remove(0);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 2);
|
2012-09-28 00:20:47 -05:00
|
|
|
_e = v.swap_remove(1);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 1);
|
2012-09-28 00:20:47 -05:00
|
|
|
_e = v.swap_remove(0);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 0);
|
2012-08-22 21:09:34 -05:00
|
|
|
}
|
|
|
|
|
2012-01-17 19:28:21 -06:00
|
|
|
#[test]
|
|
|
|
fn test_push() {
|
|
|
|
// Test on-stack push().
|
2012-06-29 18:26:56 -05:00
|
|
|
let mut v = ~[];
|
2012-09-26 19:33:34 -05:00
|
|
|
v.push(1);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 1u);
|
|
|
|
assert_eq!(v[0], 1);
|
2012-01-17 19:28:21 -06:00
|
|
|
|
|
|
|
// Test on-heap push().
|
2012-09-26 19:33:34 -05:00
|
|
|
v.push(2);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 2u);
|
|
|
|
assert_eq!(v[0], 1);
|
|
|
|
assert_eq!(v[1], 2);
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_grow() {
|
|
|
|
// Test on-stack grow().
|
2012-06-29 18:26:56 -05:00
|
|
|
let mut v = ~[];
|
2012-09-28 00:20:47 -05:00
|
|
|
v.grow(2u, &1);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 2u);
|
|
|
|
assert_eq!(v[0], 1);
|
|
|
|
assert_eq!(v[1], 1);
|
2012-01-17 19:28:21 -06:00
|
|
|
|
|
|
|
// Test on-heap grow().
|
2012-09-28 00:20:47 -05:00
|
|
|
v.grow(3u, &2);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 5u);
|
|
|
|
assert_eq!(v[0], 1);
|
|
|
|
assert_eq!(v[1], 1);
|
|
|
|
assert_eq!(v[2], 2);
|
|
|
|
assert_eq!(v[3], 2);
|
|
|
|
assert_eq!(v[4], 2);
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_grow_fn() {
|
2012-06-29 18:26:56 -05:00
|
|
|
let mut v = ~[];
|
2012-09-28 00:20:47 -05:00
|
|
|
v.grow_fn(3u, square);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 3u);
|
|
|
|
assert_eq!(v[0], 0u);
|
|
|
|
assert_eq!(v[1], 1u);
|
|
|
|
assert_eq!(v[2], 4u);
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_grow_set() {
|
2012-09-21 20:43:30 -05:00
|
|
|
let mut v = ~[1, 2, 3];
|
2012-09-28 00:20:47 -05:00
|
|
|
v.grow_set(4u, &4, 5);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 5u);
|
|
|
|
assert_eq!(v[0], 1);
|
|
|
|
assert_eq!(v[1], 2);
|
|
|
|
assert_eq!(v[2], 3);
|
|
|
|
assert_eq!(v[3], 4);
|
|
|
|
assert_eq!(v[4], 5);
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
2012-08-29 03:21:49 -05:00
|
|
|
#[test]
|
|
|
|
fn test_truncate() {
|
|
|
|
let mut v = ~[@6,@5,@4];
|
2012-09-28 00:20:47 -05:00
|
|
|
v.truncate(1);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 1);
|
|
|
|
assert_eq!(*(v[0]), 6);
|
2012-08-29 03:21:49 -05:00
|
|
|
// If the unsafe block didn't drop things properly, we blow up here.
|
|
|
|
}
|
2013-01-24 22:08:16 -06:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_clear() {
|
|
|
|
let mut v = ~[@6,@5,@4];
|
|
|
|
v.clear();
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 0);
|
2013-01-24 22:08:16 -06:00
|
|
|
// If the unsafe block didn't drop things properly, we blow up here.
|
|
|
|
}
|
2012-08-29 03:21:49 -05:00
|
|
|
|
2012-09-01 14:11:54 -05:00
|
|
|
#[test]
|
|
|
|
fn test_dedup() {
|
2012-10-02 13:37:37 -05:00
|
|
|
fn case(a: ~[uint], b: ~[uint]) {
|
2012-12-12 17:38:50 -06:00
|
|
|
let mut v = a;
|
2012-09-28 00:20:47 -05:00
|
|
|
v.dedup();
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v, b);
|
2012-09-01 14:11:54 -05:00
|
|
|
}
|
|
|
|
case(~[], ~[]);
|
|
|
|
case(~[1], ~[1]);
|
|
|
|
case(~[1,1], ~[1]);
|
|
|
|
case(~[1,2,3], ~[1,2,3]);
|
|
|
|
case(~[1,1,2,3], ~[1,2,3]);
|
|
|
|
case(~[1,2,2,3], ~[1,2,3]);
|
|
|
|
case(~[1,2,3,3], ~[1,2,3]);
|
|
|
|
case(~[1,1,2,2,2,3,3], ~[1,2,3]);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_dedup_unique() {
|
|
|
|
let mut v0 = ~[~1, ~1, ~2, ~3];
|
2012-09-28 00:20:47 -05:00
|
|
|
v0.dedup();
|
2012-09-01 14:11:54 -05:00
|
|
|
let mut v1 = ~[~1, ~2, ~2, ~3];
|
2012-09-28 00:20:47 -05:00
|
|
|
v1.dedup();
|
2012-09-01 14:11:54 -05:00
|
|
|
let mut v2 = ~[~1, ~2, ~3, ~3];
|
2012-09-28 00:20:47 -05:00
|
|
|
v2.dedup();
|
2012-09-01 14:11:54 -05:00
|
|
|
/*
|
|
|
|
* If the ~pointers were leaked or otherwise misused, valgrind and/or
|
|
|
|
* rustrt should raise errors.
|
|
|
|
*/
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_dedup_shared() {
|
|
|
|
let mut v0 = ~[@1, @1, @2, @3];
|
2012-09-28 00:20:47 -05:00
|
|
|
v0.dedup();
|
2012-09-01 14:11:54 -05:00
|
|
|
let mut v1 = ~[@1, @2, @2, @3];
|
2012-09-28 00:20:47 -05:00
|
|
|
v1.dedup();
|
2012-09-01 14:11:54 -05:00
|
|
|
let mut v2 = ~[@1, @2, @3, @3];
|
2012-09-28 00:20:47 -05:00
|
|
|
v2.dedup();
|
2012-09-01 14:11:54 -05:00
|
|
|
/*
|
|
|
|
* If the @pointers were leaked or otherwise misused, valgrind and/or
|
|
|
|
* rustrt should raise errors.
|
|
|
|
*/
|
|
|
|
}
|
|
|
|
|
2012-01-17 19:28:21 -06:00
|
|
|
#[test]
|
|
|
|
fn test_map() {
|
|
|
|
// Test on-stack map.
|
2013-06-29 00:05:50 -05:00
|
|
|
let v = &[1u, 2u, 3u];
|
|
|
|
let mut w = v.map(square_ref);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(w.len(), 3u);
|
|
|
|
assert_eq!(w[0], 1u);
|
|
|
|
assert_eq!(w[1], 4u);
|
|
|
|
assert_eq!(w[2], 9u);
|
2012-01-17 19:28:21 -06:00
|
|
|
|
|
|
|
// Test on-heap map.
|
2013-06-29 00:05:50 -05:00
|
|
|
let v = ~[1u, 2u, 3u, 4u, 5u];
|
|
|
|
w = v.map(square_ref);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(w.len(), 5u);
|
|
|
|
assert_eq!(w[0], 1u);
|
|
|
|
assert_eq!(w[1], 4u);
|
|
|
|
assert_eq!(w[2], 9u);
|
|
|
|
assert_eq!(w[3], 16u);
|
|
|
|
assert_eq!(w[4], 25u);
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
2013-01-14 01:10:54 -06:00
|
|
|
#[test]
|
|
|
|
fn test_retain() {
|
|
|
|
let mut v = ~[1, 2, 3, 4, 5];
|
|
|
|
v.retain(is_odd);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v, ~[1, 3, 5]);
|
2013-01-14 01:10:54 -06:00
|
|
|
}
|
|
|
|
|
2012-01-17 19:28:21 -06:00
|
|
|
#[test]
|
2012-09-28 00:20:47 -05:00
|
|
|
fn test_each_permutation() {
|
2012-06-29 18:26:56 -05:00
|
|
|
let mut results: ~[~[int]];
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2012-06-29 18:26:56 -05:00
|
|
|
results = ~[];
|
2013-07-12 02:59:39 -05:00
|
|
|
for each_permutation([]) |v| { results.push(v.to_owned()); }
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(results, ~[~[]]);
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2012-06-29 18:26:56 -05:00
|
|
|
results = ~[];
|
2013-07-12 02:59:39 -05:00
|
|
|
for each_permutation([7]) |v| { results.push(v.to_owned()); }
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(results, ~[~[7]]);
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2012-06-29 18:26:56 -05:00
|
|
|
results = ~[];
|
2013-07-12 02:59:39 -05:00
|
|
|
for each_permutation([1,1]) |v| { results.push(v.to_owned()); }
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(results, ~[~[1,1],~[1,1]]);
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2012-06-29 18:26:56 -05:00
|
|
|
results = ~[];
|
2013-07-12 02:59:39 -05:00
|
|
|
for each_permutation([5,2,0]) |v| { results.push(v.to_owned()); }
|
2013-03-28 20:39:09 -05:00
|
|
|
assert!(results ==
|
2013-03-06 15:58:02 -06:00
|
|
|
~[~[5,2,0],~[5,0,2],~[2,5,0],~[2,0,5],~[0,5,2],~[0,2,5]]);
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_zip_unzip() {
|
2012-06-29 18:26:56 -05:00
|
|
|
let v1 = ~[1, 2, 3];
|
|
|
|
let v2 = ~[4, 5, 6];
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2012-12-12 17:38:50 -06:00
|
|
|
let z1 = zip(v1, v2);
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!((1, 4), z1[0]);
|
|
|
|
assert_eq!((2, 5), z1[1]);
|
|
|
|
assert_eq!((3, 6), z1[2]);
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2012-12-12 17:38:50 -06:00
|
|
|
let (left, right) = unzip(z1);
|
2012-01-17 19:28:21 -06:00
|
|
|
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!((1, 4), (left[0], right[0]));
|
|
|
|
assert_eq!((2, 5), (left[1], right[1]));
|
|
|
|
assert_eq!((3, 6), (left[2], right[2]));
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2012-03-18 19:14:43 -05:00
|
|
|
fn test_position_elem() {
|
2013-06-28 11:08:32 -05:00
|
|
|
assert!([].position_elem(&1).is_none());
|
2012-01-26 10:39:45 -06:00
|
|
|
|
2012-06-29 18:26:56 -05:00
|
|
|
let v1 = ~[1, 2, 3, 3, 2, 5];
|
2013-06-28 11:08:32 -05:00
|
|
|
assert_eq!(v1.position_elem(&1), Some(0u));
|
|
|
|
assert_eq!(v1.position_elem(&2), Some(1u));
|
|
|
|
assert_eq!(v1.position_elem(&5), Some(5u));
|
|
|
|
assert!(v1.position_elem(&4).is_none());
|
2012-01-26 20:14:27 -06:00
|
|
|
}
|
|
|
|
|
2012-01-26 10:42:50 -06:00
|
|
|
#[test]
|
|
|
|
fn test_rposition() {
|
2012-09-28 00:20:47 -05:00
|
|
|
fn f(xy: &(int, char)) -> bool { let (_x, y) = *xy; y == 'b' }
|
|
|
|
fn g(xy: &(int, char)) -> bool { let (_x, y) = *xy; y == 'd' }
|
2013-05-12 19:34:15 -05:00
|
|
|
let v = ~[(0, 'a'), (1, 'b'), (2, 'c'), (3, 'b')];
|
2012-01-26 10:42:50 -06:00
|
|
|
|
2013-06-28 11:08:32 -05:00
|
|
|
assert_eq!(v.rposition(f), Some(3u));
|
|
|
|
assert!(v.rposition(g).is_none());
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
2013-01-04 15:52:18 -06:00
|
|
|
#[test]
|
|
|
|
fn test_bsearch_elem() {
|
2013-06-28 22:35:25 -05:00
|
|
|
assert_eq!([1,2,3,4,5].bsearch_elem(&5), Some(4));
|
|
|
|
assert_eq!([1,2,3,4,5].bsearch_elem(&4), Some(3));
|
|
|
|
assert_eq!([1,2,3,4,5].bsearch_elem(&3), Some(2));
|
|
|
|
assert_eq!([1,2,3,4,5].bsearch_elem(&2), Some(1));
|
|
|
|
assert_eq!([1,2,3,4,5].bsearch_elem(&1), Some(0));
|
2013-05-18 21:02:45 -05:00
|
|
|
|
2013-06-28 22:35:25 -05:00
|
|
|
assert_eq!([2,4,6,8,10].bsearch_elem(&1), None);
|
|
|
|
assert_eq!([2,4,6,8,10].bsearch_elem(&5), None);
|
|
|
|
assert_eq!([2,4,6,8,10].bsearch_elem(&4), Some(1));
|
|
|
|
assert_eq!([2,4,6,8,10].bsearch_elem(&10), Some(4));
|
2013-05-18 21:02:45 -05:00
|
|
|
|
2013-06-28 22:35:25 -05:00
|
|
|
assert_eq!([2,4,6,8].bsearch_elem(&1), None);
|
|
|
|
assert_eq!([2,4,6,8].bsearch_elem(&5), None);
|
|
|
|
assert_eq!([2,4,6,8].bsearch_elem(&4), Some(1));
|
|
|
|
assert_eq!([2,4,6,8].bsearch_elem(&8), Some(3));
|
2013-05-18 21:02:45 -05:00
|
|
|
|
2013-06-28 22:35:25 -05:00
|
|
|
assert_eq!([2,4,6].bsearch_elem(&1), None);
|
|
|
|
assert_eq!([2,4,6].bsearch_elem(&5), None);
|
|
|
|
assert_eq!([2,4,6].bsearch_elem(&4), Some(1));
|
|
|
|
assert_eq!([2,4,6].bsearch_elem(&6), Some(2));
|
2013-05-18 21:02:45 -05:00
|
|
|
|
2013-06-28 22:35:25 -05:00
|
|
|
assert_eq!([2,4].bsearch_elem(&1), None);
|
|
|
|
assert_eq!([2,4].bsearch_elem(&5), None);
|
|
|
|
assert_eq!([2,4].bsearch_elem(&2), Some(0));
|
|
|
|
assert_eq!([2,4].bsearch_elem(&4), Some(1));
|
2013-05-18 21:02:45 -05:00
|
|
|
|
2013-06-28 22:35:25 -05:00
|
|
|
assert_eq!([2].bsearch_elem(&1), None);
|
|
|
|
assert_eq!([2].bsearch_elem(&5), None);
|
|
|
|
assert_eq!([2].bsearch_elem(&2), Some(0));
|
2013-05-18 21:02:45 -05:00
|
|
|
|
2013-06-28 22:35:25 -05:00
|
|
|
assert_eq!([].bsearch_elem(&1), None);
|
|
|
|
assert_eq!([].bsearch_elem(&5), None);
|
2013-01-08 10:44:31 -06:00
|
|
|
|
2013-06-28 22:35:25 -05:00
|
|
|
assert!([1,1,1,1,1].bsearch_elem(&1) != None);
|
|
|
|
assert!([1,1,1,1,2].bsearch_elem(&1) != None);
|
|
|
|
assert!([1,1,1,2,2].bsearch_elem(&1) != None);
|
|
|
|
assert!([1,1,2,2,2].bsearch_elem(&1) != None);
|
|
|
|
assert_eq!([1,2,2,2,2].bsearch_elem(&1), Some(0));
|
2013-01-08 10:44:31 -06:00
|
|
|
|
2013-06-28 22:35:25 -05:00
|
|
|
assert_eq!([1,2,3,4,5].bsearch_elem(&6), None);
|
|
|
|
assert_eq!([1,2,3,4,5].bsearch_elem(&0), None);
|
2013-01-04 15:52:18 -06:00
|
|
|
}
|
|
|
|
|
2012-01-17 19:28:21 -06:00
|
|
|
#[test]
|
2013-07-01 21:58:23 -05:00
|
|
|
fn test_reverse() {
|
2013-02-12 17:34:48 -06:00
|
|
|
let mut v: ~[int] = ~[10, 20];
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v[0], 10);
|
|
|
|
assert_eq!(v[1], 20);
|
2013-06-28 11:54:03 -05:00
|
|
|
v.reverse();
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v[0], 20);
|
|
|
|
assert_eq!(v[1], 10);
|
2013-07-01 21:58:23 -05:00
|
|
|
|
2013-02-12 17:34:48 -06:00
|
|
|
let mut v3: ~[int] = ~[];
|
2013-06-28 11:54:03 -05:00
|
|
|
v3.reverse();
|
2013-07-01 21:58:23 -05:00
|
|
|
assert!(v3.is_empty());
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
2013-01-07 10:49:41 -06:00
|
|
|
#[test]
|
|
|
|
fn test_partition() {
|
2013-06-27 09:10:18 -05:00
|
|
|
assert_eq!((~[]).partition(|x: &int| *x < 3), (~[], ~[]));
|
|
|
|
assert_eq!((~[1, 2, 3]).partition(|x: &int| *x < 4), (~[1, 2, 3], ~[]));
|
|
|
|
assert_eq!((~[1, 2, 3]).partition(|x: &int| *x < 2), (~[1], ~[2, 3]));
|
|
|
|
assert_eq!((~[1, 2, 3]).partition(|x: &int| *x < 0), (~[], ~[1, 2, 3]));
|
2013-01-07 10:49:41 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_partitioned() {
|
2013-05-23 11:39:17 -05:00
|
|
|
assert_eq!(([]).partitioned(|x: &int| *x < 3), (~[], ~[]))
|
|
|
|
assert_eq!(([1, 2, 3]).partitioned(|x: &int| *x < 4), (~[1, 2, 3], ~[]));
|
|
|
|
assert_eq!(([1, 2, 3]).partitioned(|x: &int| *x < 2), (~[1], ~[2, 3]));
|
|
|
|
assert_eq!(([1, 2, 3]).partitioned(|x: &int| *x < 0), (~[], ~[1, 2, 3]));
|
2013-01-07 10:49:41 -06:00
|
|
|
}
|
|
|
|
|
2012-01-17 19:28:21 -06:00
|
|
|
#[test]
|
|
|
|
fn test_concat() {
|
2013-05-23 11:39:17 -05:00
|
|
|
assert_eq!(concat([~[1], ~[2,3]]), ~[1, 2, 3]);
|
2013-06-14 21:56:41 -05:00
|
|
|
assert_eq!([~[1], ~[2,3]].concat_vec(), ~[1, 2, 3]);
|
2013-06-02 22:19:37 -05:00
|
|
|
|
|
|
|
assert_eq!(concat_slices([&[1], &[2,3]]), ~[1, 2, 3]);
|
2013-06-14 21:56:41 -05:00
|
|
|
assert_eq!([&[1], &[2,3]].concat_vec(), ~[1, 2, 3]);
|
2012-01-17 19:28:21 -06:00
|
|
|
}
|
|
|
|
|
2012-01-28 17:41:53 -06:00
|
|
|
#[test]
|
|
|
|
fn test_connect() {
|
2013-05-23 11:39:17 -05:00
|
|
|
assert_eq!(connect([], &0), ~[]);
|
|
|
|
assert_eq!(connect([~[1], ~[2, 3]], &0), ~[1, 0, 2, 3]);
|
|
|
|
assert_eq!(connect([~[1], ~[2], ~[3]], &0), ~[1, 0, 2, 0, 3]);
|
2013-06-14 21:56:41 -05:00
|
|
|
assert_eq!([~[1], ~[2, 3]].connect_vec(&0), ~[1, 0, 2, 3]);
|
|
|
|
assert_eq!([~[1], ~[2], ~[3]].connect_vec(&0), ~[1, 0, 2, 0, 3]);
|
2013-06-02 22:19:37 -05:00
|
|
|
|
|
|
|
assert_eq!(connect_slices([], &0), ~[]);
|
|
|
|
assert_eq!(connect_slices([&[1], &[2, 3]], &0), ~[1, 0, 2, 3]);
|
|
|
|
assert_eq!(connect_slices([&[1], &[2], &[3]], &0), ~[1, 0, 2, 0, 3]);
|
2013-06-14 21:56:41 -05:00
|
|
|
assert_eq!([&[1], &[2, 3]].connect_vec(&0), ~[1, 0, 2, 3]);
|
|
|
|
assert_eq!([&[1], &[2], &[3]].connect_vec(&0), ~[1, 0, 2, 0, 3]);
|
2012-01-28 17:41:53 -06:00
|
|
|
}
|
|
|
|
|
2013-07-05 13:32:25 -05:00
|
|
|
#[test]
|
|
|
|
fn test_shift() {
|
|
|
|
let mut x = ~[1, 2, 3];
|
|
|
|
assert_eq!(x.shift(), 1);
|
|
|
|
assert_eq!(&x, &~[2, 3]);
|
|
|
|
assert_eq!(x.shift(), 2);
|
|
|
|
assert_eq!(x.shift(), 3);
|
|
|
|
assert_eq!(x.len(), 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_shift_opt() {
|
|
|
|
let mut x = ~[1, 2, 3];
|
|
|
|
assert_eq!(x.shift_opt(), Some(1));
|
|
|
|
assert_eq!(&x, &~[2, 3]);
|
|
|
|
assert_eq!(x.shift_opt(), Some(2));
|
|
|
|
assert_eq!(x.shift_opt(), Some(3));
|
|
|
|
assert_eq!(x.shift_opt(), None);
|
|
|
|
assert_eq!(x.len(), 0);
|
|
|
|
}
|
|
|
|
|
2012-06-22 18:31:57 -05:00
|
|
|
#[test]
|
|
|
|
fn test_unshift() {
|
2012-06-29 18:26:56 -05:00
|
|
|
let mut x = ~[1, 2, 3];
|
2012-09-28 00:20:47 -05:00
|
|
|
x.unshift(0);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(x, ~[0, 1, 2, 3]);
|
2012-06-22 18:31:57 -05:00
|
|
|
}
|
|
|
|
|
2012-11-25 07:28:16 -06:00
|
|
|
#[test]
|
|
|
|
fn test_insert() {
|
|
|
|
let mut a = ~[1, 2, 4];
|
|
|
|
a.insert(2, 3);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a, ~[1, 2, 3, 4]);
|
2012-11-25 07:28:16 -06:00
|
|
|
|
|
|
|
let mut a = ~[1, 2, 3];
|
|
|
|
a.insert(0, 0);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a, ~[0, 1, 2, 3]);
|
2012-11-25 07:28:16 -06:00
|
|
|
|
|
|
|
let mut a = ~[1, 2, 3];
|
|
|
|
a.insert(3, 4);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a, ~[1, 2, 3, 4]);
|
2012-11-25 07:28:16 -06:00
|
|
|
|
|
|
|
let mut a = ~[];
|
|
|
|
a.insert(0, 1);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a, ~[1]);
|
2012-11-25 07:28:16 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2012-11-26 16:48:48 -06:00
|
|
|
#[ignore(cfg(windows))]
|
2012-11-25 07:28:16 -06:00
|
|
|
#[should_fail]
|
|
|
|
fn test_insert_oob() {
|
|
|
|
let mut a = ~[1, 2, 3];
|
|
|
|
a.insert(4, 5);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_remove() {
|
|
|
|
let mut a = ~[1, 2, 3, 4];
|
|
|
|
a.remove(2);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a, ~[1, 2, 4]);
|
2012-11-25 07:28:16 -06:00
|
|
|
|
|
|
|
let mut a = ~[1, 2, 3];
|
|
|
|
a.remove(0);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a, ~[2, 3]);
|
2012-11-25 07:28:16 -06:00
|
|
|
|
|
|
|
let mut a = ~[1];
|
|
|
|
a.remove(0);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(a, ~[]);
|
2012-11-25 07:28:16 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2012-11-26 16:48:48 -06:00
|
|
|
#[ignore(cfg(windows))]
|
2012-11-25 07:28:16 -06:00
|
|
|
#[should_fail]
|
|
|
|
fn test_remove_oob() {
|
|
|
|
let mut a = ~[1, 2, 3];
|
|
|
|
a.remove(3);
|
|
|
|
}
|
|
|
|
|
2012-03-29 01:10:58 -05:00
|
|
|
#[test]
|
|
|
|
fn test_capacity() {
|
2012-06-29 18:26:56 -05:00
|
|
|
let mut v = ~[0u64];
|
2013-06-27 09:40:47 -05:00
|
|
|
v.reserve(10u);
|
|
|
|
assert_eq!(v.capacity(), 10u);
|
2012-06-29 18:26:56 -05:00
|
|
|
let mut v = ~[0u32];
|
2013-06-27 09:40:47 -05:00
|
|
|
v.reserve(10u);
|
|
|
|
assert_eq!(v.capacity(), 10u);
|
2012-03-29 01:10:58 -05:00
|
|
|
}
|
2012-05-18 18:55:22 -05:00
|
|
|
|
|
|
|
#[test]
|
2013-03-21 06:36:21 -05:00
|
|
|
fn test_slice_2() {
|
2012-06-29 18:26:56 -05:00
|
|
|
let v = ~[1, 2, 3, 4, 5];
|
2013-03-21 06:36:21 -05:00
|
|
|
let v = v.slice(1u, 3u);
|
2013-05-18 21:02:45 -05:00
|
|
|
assert_eq!(v.len(), 2u);
|
|
|
|
assert_eq!(v[0], 2);
|
|
|
|
assert_eq!(v[1], 3);
|
2012-05-18 18:55:22 -05:00
|
|
|
}
|
2012-09-27 18:41:38 -05:00
|
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[ignore(windows)]
|
|
|
|
#[should_fail]
|
|
|
|
fn test_from_fn_fail() {
|
|
|
|
do from_fn(100) |v| {
|
2013-02-11 21:26:38 -06:00
|
|
|
if v == 50 { fail!() }
|
2012-09-27 18:41:38 -05:00
|
|
|
(~0, @0)
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[ignore(windows)]
|
|
|
|
#[should_fail]
|
|
|
|
fn test_build_fail() {
|
|
|
|
do build |push| {
|
|
|
|
push((~0, @0));
|
|
|
|
push((~0, @0));
|
|
|
|
push((~0, @0));
|
|
|
|
push((~0, @0));
|
2013-02-11 21:26:38 -06:00
|
|
|
fail!();
|
2012-09-27 18:41:38 -05:00
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[ignore(windows)]
|
|
|
|
#[should_fail]
|
|
|
|
fn test_grow_fn_fail() {
|
|
|
|
let mut v = ~[];
|
2012-09-28 00:20:47 -05:00
|
|
|
do v.grow_fn(100) |i| {
|
2012-09-27 18:41:38 -05:00
|
|
|
if i == 50 {
|
2013-02-11 21:26:38 -06:00
|
|
|
fail!()
|
2012-09-27 18:41:38 -05:00
|
|
|
}
|
|
|
|
(~0, @0)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[ignore(windows)]
|
|
|
|
#[should_fail]
|
|
|
|
fn test_map_fail() {
|
2013-05-12 19:34:15 -05:00
|
|
|
let v = [(~0, @0), (~0, @0), (~0, @0), (~0, @0)];
|
2012-09-27 18:41:38 -05:00
|
|
|
let mut i = 0;
|
2013-06-29 00:05:50 -05:00
|
|
|
do v.map |_elt| {
|
2012-09-27 18:41:38 -05:00
|
|
|
if i == 2 {
|
2013-02-11 21:26:38 -06:00
|
|
|
fail!()
|
2012-09-27 18:41:38 -05:00
|
|
|
}
|
|
|
|
i += 0;
|
|
|
|
~[(~0, @0)]
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[ignore(windows)]
|
|
|
|
#[should_fail]
|
|
|
|
fn test_flat_map_fail() {
|
|
|
|
let v = [(~0, @0), (~0, @0), (~0, @0), (~0, @0)];
|
|
|
|
let mut i = 0;
|
2013-06-29 00:05:50 -05:00
|
|
|
do flat_map(v) |_elt| {
|
2012-09-27 18:41:38 -05:00
|
|
|
if i == 2 {
|
2013-02-11 21:26:38 -06:00
|
|
|
fail!()
|
2012-09-27 18:41:38 -05:00
|
|
|
}
|
|
|
|
i += 0;
|
|
|
|
~[(~0, @0)]
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[ignore(windows)]
|
|
|
|
#[should_fail]
|
|
|
|
fn test_rposition_fail() {
|
|
|
|
let v = [(~0, @0), (~0, @0), (~0, @0), (~0, @0)];
|
|
|
|
let mut i = 0;
|
2013-06-28 11:08:32 -05:00
|
|
|
do v.rposition |_elt| {
|
2012-09-27 18:41:38 -05:00
|
|
|
if i == 2 {
|
2013-02-11 21:26:38 -06:00
|
|
|
fail!()
|
2012-09-27 18:41:38 -05:00
|
|
|
}
|
|
|
|
i += 0;
|
|
|
|
false
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[ignore(windows)]
|
|
|
|
#[should_fail]
|
|
|
|
fn test_permute_fail() {
|
|
|
|
let v = [(~0, @0), (~0, @0), (~0, @0), (~0, @0)];
|
|
|
|
let mut i = 0;
|
2012-09-28 00:20:47 -05:00
|
|
|
for each_permutation(v) |_elt| {
|
2012-09-27 18:41:38 -05:00
|
|
|
if i == 2 {
|
2013-02-11 21:26:38 -06:00
|
|
|
fail!()
|
2012-09-27 18:41:38 -05:00
|
|
|
}
|
|
|
|
i += 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[ignore(windows)]
|
|
|
|
#[should_fail]
|
|
|
|
fn test_as_imm_buf_fail() {
|
|
|
|
let v = [(~0, @0), (~0, @0), (~0, @0), (~0, @0)];
|
2013-07-03 01:34:17 -05:00
|
|
|
do v.as_imm_buf |_buf, _i| {
|
2013-02-11 21:26:38 -06:00
|
|
|
fail!()
|
2012-09-27 18:41:38 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2013-01-09 10:16:31 -06:00
|
|
|
#[ignore(cfg(windows))]
|
2012-09-27 18:41:38 -05:00
|
|
|
#[should_fail]
|
|
|
|
fn test_as_mut_buf_fail() {
|
2013-02-12 17:34:48 -06:00
|
|
|
let mut v = [(~0, @0), (~0, @0), (~0, @0), (~0, @0)];
|
2013-07-03 01:34:17 -05:00
|
|
|
do v.as_mut_buf |_buf, _i| {
|
2013-02-11 21:26:38 -06:00
|
|
|
fail!()
|
2012-09-27 18:41:38 -05:00
|
|
|
}
|
|
|
|
}
|
2013-01-05 04:52:37 -06:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[should_fail]
|
2013-01-09 10:16:31 -06:00
|
|
|
#[ignore(cfg(windows))]
|
2013-01-23 13:43:58 -06:00
|
|
|
fn test_copy_memory_oob() {
|
|
|
|
unsafe {
|
2013-02-12 17:34:48 -06:00
|
|
|
let mut a = [1, 2, 3, 4];
|
2013-01-23 13:43:58 -06:00
|
|
|
let b = [1, 2, 3, 4, 5];
|
|
|
|
raw::copy_memory(a, b, 5);
|
|
|
|
}
|
2013-01-05 04:52:37 -06:00
|
|
|
}
|
|
|
|
|
2013-03-01 21:07:12 -06:00
|
|
|
#[test]
|
|
|
|
fn test_total_ord() {
|
|
|
|
[1, 2, 3, 4].cmp(& &[1, 2, 3]) == Greater;
|
|
|
|
[1, 2, 3].cmp(& &[1, 2, 3, 4]) == Less;
|
|
|
|
[1, 2, 3, 4].cmp(& &[1, 2, 3, 4]) == Equal;
|
|
|
|
[1, 2, 3, 4, 5, 5, 5, 5].cmp(& &[1, 2, 3, 4, 5, 6]) == Less;
|
|
|
|
[2, 2].cmp(& &[1, 2, 3, 4]) == Greater;
|
|
|
|
}
|
2013-04-17 18:34:53 -05:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_iterator() {
|
|
|
|
use iterator::*;
|
|
|
|
let xs = [1, 2, 5, 10, 11];
|
|
|
|
let mut it = xs.iter();
|
2013-07-02 20:40:46 -05:00
|
|
|
assert_eq!(it.size_hint(), (5, Some(5)));
|
2013-06-21 05:12:01 -05:00
|
|
|
assert_eq!(it.next().unwrap(), &1);
|
2013-07-02 20:40:46 -05:00
|
|
|
assert_eq!(it.size_hint(), (4, Some(4)));
|
2013-06-21 05:12:01 -05:00
|
|
|
assert_eq!(it.next().unwrap(), &2);
|
2013-07-02 20:40:46 -05:00
|
|
|
assert_eq!(it.size_hint(), (3, Some(3)));
|
2013-06-21 05:12:01 -05:00
|
|
|
assert_eq!(it.next().unwrap(), &5);
|
2013-07-02 20:40:46 -05:00
|
|
|
assert_eq!(it.size_hint(), (2, Some(2)));
|
2013-06-21 05:12:01 -05:00
|
|
|
assert_eq!(it.next().unwrap(), &10);
|
2013-07-02 20:40:46 -05:00
|
|
|
assert_eq!(it.size_hint(), (1, Some(1)));
|
2013-06-21 05:12:01 -05:00
|
|
|
assert_eq!(it.next().unwrap(), &11);
|
2013-07-02 20:40:46 -05:00
|
|
|
assert_eq!(it.size_hint(), (0, Some(0)));
|
2013-06-21 05:12:01 -05:00
|
|
|
assert!(it.next().is_none());
|
2013-04-17 18:34:53 -05:00
|
|
|
}
|
2013-05-11 17:56:08 -05:00
|
|
|
|
2013-07-22 19:11:24 -05:00
|
|
|
#[test]
|
|
|
|
fn test_random_access_iterator() {
|
|
|
|
use iterator::*;
|
|
|
|
let xs = [1, 2, 5, 10, 11];
|
|
|
|
let mut it = xs.iter();
|
|
|
|
|
|
|
|
assert_eq!(it.indexable(), 5);
|
|
|
|
assert_eq!(it.idx(0).unwrap(), &1);
|
|
|
|
assert_eq!(it.idx(2).unwrap(), &5);
|
|
|
|
assert_eq!(it.idx(4).unwrap(), &11);
|
|
|
|
assert!(it.idx(5).is_none());
|
|
|
|
|
|
|
|
assert_eq!(it.next().unwrap(), &1);
|
|
|
|
assert_eq!(it.indexable(), 4);
|
|
|
|
assert_eq!(it.idx(0).unwrap(), &2);
|
|
|
|
assert_eq!(it.idx(3).unwrap(), &11);
|
|
|
|
assert!(it.idx(4).is_none());
|
|
|
|
|
|
|
|
assert_eq!(it.next().unwrap(), &2);
|
|
|
|
assert_eq!(it.indexable(), 3);
|
|
|
|
assert_eq!(it.idx(1).unwrap(), &10);
|
|
|
|
assert!(it.idx(3).is_none());
|
|
|
|
|
|
|
|
assert_eq!(it.next().unwrap(), &5);
|
|
|
|
assert_eq!(it.indexable(), 2);
|
|
|
|
assert_eq!(it.idx(1).unwrap(), &11);
|
|
|
|
|
|
|
|
assert_eq!(it.next().unwrap(), &10);
|
|
|
|
assert_eq!(it.indexable(), 1);
|
|
|
|
assert_eq!(it.idx(0).unwrap(), &11);
|
|
|
|
assert!(it.idx(1).is_none());
|
|
|
|
|
|
|
|
assert_eq!(it.next().unwrap(), &11);
|
|
|
|
assert_eq!(it.indexable(), 0);
|
|
|
|
assert!(it.idx(0).is_none());
|
|
|
|
|
|
|
|
assert!(it.next().is_none());
|
|
|
|
}
|
|
|
|
|
2013-07-03 07:56:26 -05:00
|
|
|
#[test]
|
|
|
|
fn test_iter_size_hints() {
|
|
|
|
use iterator::*;
|
|
|
|
let mut xs = [1, 2, 5, 10, 11];
|
2013-07-02 20:40:46 -05:00
|
|
|
assert_eq!(xs.iter().size_hint(), (5, Some(5)));
|
|
|
|
assert_eq!(xs.rev_iter().size_hint(), (5, Some(5)));
|
|
|
|
assert_eq!(xs.mut_iter().size_hint(), (5, Some(5)));
|
|
|
|
assert_eq!(xs.mut_rev_iter().size_hint(), (5, Some(5)));
|
2013-07-03 07:56:26 -05:00
|
|
|
}
|
|
|
|
|
2013-07-18 10:38:17 -05:00
|
|
|
#[test]
|
|
|
|
fn test_iter_clone() {
|
|
|
|
let xs = [1, 2, 5];
|
|
|
|
let mut it = xs.iter();
|
|
|
|
it.next();
|
|
|
|
let mut jt = it.clone();
|
|
|
|
assert_eq!(it.next(), jt.next());
|
|
|
|
assert_eq!(it.next(), jt.next());
|
|
|
|
assert_eq!(it.next(), jt.next());
|
|
|
|
}
|
|
|
|
|
2013-06-06 00:12:39 -05:00
|
|
|
#[test]
|
|
|
|
fn test_mut_iterator() {
|
|
|
|
use iterator::*;
|
|
|
|
let mut xs = [1, 2, 3, 4, 5];
|
|
|
|
for xs.mut_iter().advance |x| {
|
|
|
|
*x += 1;
|
|
|
|
}
|
|
|
|
assert_eq!(xs, [2, 3, 4, 5, 6])
|
|
|
|
}
|
|
|
|
|
2013-06-07 21:39:52 -05:00
|
|
|
#[test]
|
|
|
|
fn test_rev_iterator() {
|
|
|
|
use iterator::*;
|
|
|
|
|
|
|
|
let xs = [1, 2, 5, 10, 11];
|
|
|
|
let ys = [11, 10, 5, 2, 1];
|
|
|
|
let mut i = 0;
|
|
|
|
for xs.rev_iter().advance |&x| {
|
|
|
|
assert_eq!(x, ys[i]);
|
|
|
|
i += 1;
|
|
|
|
}
|
|
|
|
assert_eq!(i, 5);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_mut_rev_iterator() {
|
|
|
|
use iterator::*;
|
2013-06-07 23:07:55 -05:00
|
|
|
let mut xs = [1u, 2, 3, 4, 5];
|
2013-06-07 21:39:52 -05:00
|
|
|
for xs.mut_rev_iter().enumerate().advance |(i,x)| {
|
|
|
|
*x += i;
|
|
|
|
}
|
|
|
|
assert_eq!(xs, [5, 5, 5, 5, 5])
|
|
|
|
}
|
|
|
|
|
2013-07-01 10:26:44 -05:00
|
|
|
#[test]
|
|
|
|
fn test_consume_iterator() {
|
|
|
|
use iterator::*;
|
|
|
|
let xs = ~[1u,2,3,4,5];
|
|
|
|
assert_eq!(xs.consume_iter().fold(0, |a: uint, b: uint| 10*a + b), 12345);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_consume_rev_iterator() {
|
|
|
|
use iterator::*;
|
|
|
|
let xs = ~[1u,2,3,4,5];
|
|
|
|
assert_eq!(xs.consume_rev_iter().fold(0, |a: uint, b: uint| 10*a + b), 54321);
|
|
|
|
}
|
|
|
|
|
2013-07-02 23:54:11 -05:00
|
|
|
#[test]
|
|
|
|
fn test_split_iterator() {
|
|
|
|
let xs = &[1i,2,3,4,5];
|
|
|
|
|
|
|
|
assert_eq!(xs.split_iter(|x| *x % 2 == 0).collect::<~[&[int]]>(),
|
|
|
|
~[&[1], &[3], &[5]]);
|
|
|
|
assert_eq!(xs.split_iter(|x| *x == 1).collect::<~[&[int]]>(),
|
|
|
|
~[&[], &[2,3,4,5]]);
|
|
|
|
assert_eq!(xs.split_iter(|x| *x == 5).collect::<~[&[int]]>(),
|
|
|
|
~[&[1,2,3,4], &[]]);
|
|
|
|
assert_eq!(xs.split_iter(|x| *x == 10).collect::<~[&[int]]>(),
|
|
|
|
~[&[1,2,3,4,5]]);
|
|
|
|
assert_eq!(xs.split_iter(|_| true).collect::<~[&[int]]>(),
|
|
|
|
~[&[], &[], &[], &[], &[], &[]]);
|
|
|
|
|
|
|
|
let xs: &[int] = &[];
|
|
|
|
assert_eq!(xs.split_iter(|x| *x == 5).collect::<~[&[int]]>(), ~[&[]]);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_splitn_iterator() {
|
|
|
|
let xs = &[1i,2,3,4,5];
|
|
|
|
|
|
|
|
assert_eq!(xs.splitn_iter(0, |x| *x % 2 == 0).collect::<~[&[int]]>(),
|
|
|
|
~[&[1,2,3,4,5]]);
|
|
|
|
assert_eq!(xs.splitn_iter(1, |x| *x % 2 == 0).collect::<~[&[int]]>(),
|
|
|
|
~[&[1], &[3,4,5]]);
|
|
|
|
assert_eq!(xs.splitn_iter(3, |_| true).collect::<~[&[int]]>(),
|
|
|
|
~[&[], &[], &[], &[4,5]]);
|
|
|
|
|
|
|
|
let xs: &[int] = &[];
|
|
|
|
assert_eq!(xs.splitn_iter(1, |x| *x == 5).collect::<~[&[int]]>(), ~[&[]]);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_rsplit_iterator() {
|
|
|
|
let xs = &[1i,2,3,4,5];
|
|
|
|
|
|
|
|
assert_eq!(xs.rsplit_iter(|x| *x % 2 == 0).collect::<~[&[int]]>(),
|
|
|
|
~[&[5], &[3], &[1]]);
|
|
|
|
assert_eq!(xs.rsplit_iter(|x| *x == 1).collect::<~[&[int]]>(),
|
|
|
|
~[&[2,3,4,5], &[]]);
|
|
|
|
assert_eq!(xs.rsplit_iter(|x| *x == 5).collect::<~[&[int]]>(),
|
|
|
|
~[&[], &[1,2,3,4]]);
|
|
|
|
assert_eq!(xs.rsplit_iter(|x| *x == 10).collect::<~[&[int]]>(),
|
|
|
|
~[&[1,2,3,4,5]]);
|
|
|
|
|
|
|
|
let xs: &[int] = &[];
|
|
|
|
assert_eq!(xs.rsplit_iter(|x| *x == 5).collect::<~[&[int]]>(), ~[&[]]);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_rsplitn_iterator() {
|
|
|
|
let xs = &[1,2,3,4,5];
|
|
|
|
|
|
|
|
assert_eq!(xs.rsplitn_iter(0, |x| *x % 2 == 0).collect::<~[&[int]]>(),
|
|
|
|
~[&[1,2,3,4,5]]);
|
|
|
|
assert_eq!(xs.rsplitn_iter(1, |x| *x % 2 == 0).collect::<~[&[int]]>(),
|
|
|
|
~[&[5], &[1,2,3]]);
|
|
|
|
assert_eq!(xs.rsplitn_iter(3, |_| true).collect::<~[&[int]]>(),
|
|
|
|
~[&[], &[], &[], &[1,2]]);
|
|
|
|
|
|
|
|
let xs: &[int] = &[];
|
|
|
|
assert_eq!(xs.rsplitn_iter(1, |x| *x == 5).collect::<~[&[int]]>(), ~[&[]]);
|
|
|
|
}
|
|
|
|
|
2013-07-03 00:47:58 -05:00
|
|
|
#[test]
|
|
|
|
fn test_window_iterator() {
|
|
|
|
let v = &[1i,2,3,4];
|
|
|
|
|
|
|
|
assert_eq!(v.window_iter(2).collect::<~[&[int]]>(), ~[&[1,2], &[2,3], &[3,4]]);
|
|
|
|
assert_eq!(v.window_iter(3).collect::<~[&[int]]>(), ~[&[1i,2,3], &[2,3,4]]);
|
|
|
|
assert!(v.window_iter(6).next().is_none());
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[should_fail]
|
|
|
|
#[ignore(cfg(windows))]
|
|
|
|
fn test_window_iterator_0() {
|
|
|
|
let v = &[1i,2,3,4];
|
|
|
|
let _it = v.window_iter(0);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_chunk_iterator() {
|
|
|
|
let v = &[1i,2,3,4,5];
|
|
|
|
|
|
|
|
assert_eq!(v.chunk_iter(2).collect::<~[&[int]]>(), ~[&[1i,2], &[3,4], &[5]]);
|
|
|
|
assert_eq!(v.chunk_iter(3).collect::<~[&[int]]>(), ~[&[1i,2,3], &[4,5]]);
|
|
|
|
assert_eq!(v.chunk_iter(6).collect::<~[&[int]]>(), ~[&[1i,2,3,4,5]]);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[should_fail]
|
|
|
|
#[ignore(cfg(windows))]
|
|
|
|
fn test_chunk_iterator_0() {
|
|
|
|
let v = &[1i,2,3,4];
|
|
|
|
let _it = v.chunk_iter(0);
|
|
|
|
}
|
|
|
|
|
2013-06-18 01:52:14 -05:00
|
|
|
#[test]
|
|
|
|
fn test_move_from() {
|
|
|
|
let mut a = [1,2,3,4,5];
|
|
|
|
let b = ~[6,7,8];
|
|
|
|
assert_eq!(a.move_from(b, 0, 3), 3);
|
|
|
|
assert_eq!(a, [6,7,8,4,5]);
|
|
|
|
let mut a = [7,2,8,1];
|
|
|
|
let b = ~[3,1,4,1,5,9];
|
|
|
|
assert_eq!(a.move_from(b, 0, 6), 4);
|
|
|
|
assert_eq!(a, [3,1,4,1]);
|
|
|
|
let mut a = [1,2,3,4];
|
|
|
|
let b = ~[5,6,7,8,9,0];
|
|
|
|
assert_eq!(a.move_from(b, 2, 3), 1);
|
|
|
|
assert_eq!(a, [7,2,3,4]);
|
|
|
|
let mut a = [1,2,3,4,5];
|
|
|
|
let b = ~[5,6,7,8,9,0];
|
|
|
|
assert_eq!(a.mut_slice(2,4).move_from(b,1,6), 2);
|
|
|
|
assert_eq!(a, [1,2,6,7,5]);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_copy_from() {
|
|
|
|
let mut a = [1,2,3,4,5];
|
|
|
|
let b = [6,7,8];
|
|
|
|
assert_eq!(a.copy_from(b), 3);
|
|
|
|
assert_eq!(a, [6,7,8,4,5]);
|
|
|
|
let mut c = [7,2,8,1];
|
|
|
|
let d = [3,1,4,1,5,9];
|
|
|
|
assert_eq!(c.copy_from(d), 4);
|
|
|
|
assert_eq!(c, [3,1,4,1]);
|
|
|
|
}
|
|
|
|
|
2013-02-13 17:52:58 -06:00
|
|
|
#[test]
|
|
|
|
fn test_reverse_part() {
|
|
|
|
let mut values = [1,2,3,4,5];
|
2013-06-28 11:54:03 -05:00
|
|
|
values.mut_slice(1, 4).reverse();
|
2013-05-15 19:35:43 -05:00
|
|
|
assert_eq!(values, [1,4,3,2,5]);
|
2013-02-13 17:52:58 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_permutations0() {
|
|
|
|
let values = [];
|
|
|
|
let mut v : ~[~[int]] = ~[];
|
|
|
|
for each_permutation(values) |p| {
|
2013-05-15 19:35:43 -05:00
|
|
|
v.push(p.to_owned());
|
2013-02-13 17:52:58 -06:00
|
|
|
}
|
2013-05-15 19:35:43 -05:00
|
|
|
assert_eq!(v, ~[~[]]);
|
2013-02-13 17:52:58 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_permutations1() {
|
|
|
|
let values = [1];
|
|
|
|
let mut v : ~[~[int]] = ~[];
|
|
|
|
for each_permutation(values) |p| {
|
2013-05-15 19:35:43 -05:00
|
|
|
v.push(p.to_owned());
|
2013-02-13 17:52:58 -06:00
|
|
|
}
|
2013-05-15 19:35:43 -05:00
|
|
|
assert_eq!(v, ~[~[1]]);
|
2013-02-13 17:52:58 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_permutations2() {
|
|
|
|
let values = [1,2];
|
|
|
|
let mut v : ~[~[int]] = ~[];
|
|
|
|
for each_permutation(values) |p| {
|
2013-05-15 19:35:43 -05:00
|
|
|
v.push(p.to_owned());
|
2013-02-13 17:52:58 -06:00
|
|
|
}
|
2013-05-15 19:35:43 -05:00
|
|
|
assert_eq!(v, ~[~[1,2],~[2,1]]);
|
2013-02-13 17:52:58 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_permutations3() {
|
|
|
|
let values = [1,2,3];
|
|
|
|
let mut v : ~[~[int]] = ~[];
|
|
|
|
for each_permutation(values) |p| {
|
2013-05-15 19:35:43 -05:00
|
|
|
v.push(p.to_owned());
|
2013-02-13 17:52:58 -06:00
|
|
|
}
|
2013-05-15 19:35:43 -05:00
|
|
|
assert_eq!(v, ~[~[1,2,3],~[1,3,2],~[2,1,3],~[2,3,1],~[3,1,2],~[3,2,1]]);
|
2013-02-13 17:52:58 -06:00
|
|
|
}
|
2013-06-17 02:05:51 -05:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_vec_zero() {
|
|
|
|
use num::Zero;
|
|
|
|
macro_rules! t (
|
2013-06-27 10:45:24 -05:00
|
|
|
($ty:ty) => {{
|
2013-06-17 02:05:51 -05:00
|
|
|
let v: $ty = Zero::zero();
|
|
|
|
assert!(v.is_empty());
|
|
|
|
assert!(v.is_zero());
|
2013-06-27 10:45:24 -05:00
|
|
|
}}
|
2013-06-17 02:05:51 -05:00
|
|
|
);
|
|
|
|
|
|
|
|
t!(&[int]);
|
|
|
|
t!(@[int]);
|
|
|
|
t!(~[int]);
|
|
|
|
}
|
2013-06-18 01:20:53 -05:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_bytes_set_memory() {
|
|
|
|
use vec::bytes::MutableByteVector;
|
|
|
|
let mut values = [1u8,2,3,4,5];
|
|
|
|
values.mut_slice(0,5).set_memory(0xAB);
|
|
|
|
assert_eq!(values, [0xAB, 0xAB, 0xAB, 0xAB, 0xAB]);
|
|
|
|
values.mut_slice(2,4).set_memory(0xFF);
|
|
|
|
assert_eq!(values, [0xAB, 0xAB, 0xFF, 0xFF, 0xAB]);
|
|
|
|
}
|
2013-07-03 22:59:34 -05:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[should_fail]
|
|
|
|
fn test_overflow_does_not_cause_segfault() {
|
|
|
|
let mut v = ~[];
|
|
|
|
v.reserve(-1);
|
|
|
|
v.push(1);
|
|
|
|
v.push(2);
|
|
|
|
}
|
2013-07-10 08:50:24 -05:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_mut_split() {
|
|
|
|
let mut values = [1u8,2,3,4,5];
|
|
|
|
{
|
|
|
|
let (left, right) = values.mut_split(2);
|
|
|
|
assert_eq!(left.slice(0, left.len()), [1, 2]);
|
|
|
|
for left.mut_iter().advance |p| {
|
|
|
|
*p += 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
assert_eq!(right.slice(0, right.len()), [3, 4, 5]);
|
|
|
|
for right.mut_iter().advance |p| {
|
|
|
|
*p += 2;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
assert_eq!(values, [2, 3, 5, 6, 7]);
|
|
|
|
}
|
2013-07-12 02:59:39 -05:00
|
|
|
|
2013-07-12 23:05:59 -05:00
|
|
|
#[deriving(Clone, Eq)]
|
2013-07-12 02:59:39 -05:00
|
|
|
struct Foo;
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_iter_zero_sized() {
|
|
|
|
let mut v = ~[Foo, Foo, Foo];
|
|
|
|
assert_eq!(v.len(), 3);
|
|
|
|
let mut cnt = 0;
|
|
|
|
|
|
|
|
for v.iter().advance |f| {
|
|
|
|
assert!(*f == Foo);
|
|
|
|
cnt += 1;
|
|
|
|
}
|
|
|
|
assert_eq!(cnt, 3);
|
|
|
|
|
|
|
|
for v.slice(1, 3).iter().advance |f| {
|
|
|
|
assert!(*f == Foo);
|
|
|
|
cnt += 1;
|
|
|
|
}
|
|
|
|
assert_eq!(cnt, 5);
|
|
|
|
|
|
|
|
for v.mut_iter().advance |f| {
|
|
|
|
assert!(*f == Foo);
|
|
|
|
cnt += 1;
|
|
|
|
}
|
|
|
|
assert_eq!(cnt, 8);
|
|
|
|
|
|
|
|
for v.consume_iter().advance |f| {
|
|
|
|
assert!(f == Foo);
|
|
|
|
cnt += 1;
|
|
|
|
}
|
|
|
|
assert_eq!(cnt, 11);
|
|
|
|
|
|
|
|
let xs = ~[Foo, Foo, Foo];
|
|
|
|
assert_eq!(fmt!("%?", xs.slice(0, 2).to_owned()), ~"~[{}, {}]");
|
|
|
|
|
|
|
|
let xs: [Foo, ..3] = [Foo, Foo, Foo];
|
|
|
|
assert_eq!(fmt!("%?", xs.slice(0, 2).to_owned()), ~"~[{}, {}]");
|
|
|
|
cnt = 0;
|
|
|
|
for xs.iter().advance |f| {
|
|
|
|
assert!(*f == Foo);
|
|
|
|
cnt += 1;
|
|
|
|
}
|
|
|
|
assert!(cnt == 3);
|
|
|
|
}
|
2013-04-18 07:15:40 -05:00
|
|
|
}
|