1 // Copyright 2014 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
11 //! A growable list type with heap-allocated contents, written `Vec<T>` but pronounced 'vector.'
13 //! Vectors have `O(1)` indexing, push (to the end) and pop (from the end).
17 //! Explicitly creating a `Vec<T>` with `new()`:
20 //! let xs: Vec<i32> = Vec::new();
23 //! Using the `vec!` macro:
26 //! let ys: Vec<i32> = vec![];
28 //! let zs = vec![1i32, 2, 3, 4, 5];
34 //! let mut xs = vec![1i32, 2];
42 //! let mut xs = vec![1i32, 2];
44 //! let two = xs.pop();
51 use alloc::boxed::Box;
52 use alloc::heap::{EMPTY, allocate, reallocate, deallocate};
53 use core::borrow::{Cow, IntoCow};
55 use core::cmp::{Ordering};
56 use core::default::Default;
58 use core::hash::{self, Hash};
59 use core::iter::{repeat, FromIterator};
60 use core::marker::{ContravariantLifetime, InvariantType};
62 use core::nonzero::NonZero;
63 use core::num::{Int, UnsignedInt};
64 use core::ops::{Index, IndexMut, Deref, Add};
67 use core::raw::Slice as RawSlice;
70 /// A growable list type, written `Vec<T>` but pronounced 'vector.'
75 /// let mut vec = Vec::new();
79 /// assert_eq!(vec.len(), 2);
80 /// assert_eq!(vec[0], 1);
82 /// assert_eq!(vec.pop(), Some(2));
83 /// assert_eq!(vec.len(), 1);
86 /// assert_eq!(vec[0], 7);
88 /// vec.push_all(&[1, 2, 3]);
90 /// for x in vec.iter() {
91 /// println!("{}", x);
93 /// assert_eq!(vec, vec![7i, 1, 2, 3]);
96 /// The `vec!` macro is provided to make initialization more convenient:
99 /// let mut vec = vec![1i, 2i, 3i];
101 /// assert_eq!(vec, vec![1, 2, 3, 4]);
104 /// Use a `Vec<T>` as an efficient stack:
107 /// let mut stack = Vec::new();
114 /// let top = match stack.pop() {
115 /// None => break, // empty
118 /// // Prints 3, 2, 1
119 /// println!("{}", top);
123 /// # Capacity and reallocation
125 /// The capacity of a vector is the amount of space allocated for any future elements that will be
126 /// added onto the vector. This is not to be confused with the *length* of a vector, which
127 /// specifies the number of actual elements within the vector. If a vector's length exceeds its
128 /// capacity, its capacity will automatically be increased, but its elements will have to be
131 /// For example, a vector with capacity 10 and length 0 would be an empty vector with space for 10
132 /// more elements. Pushing 10 or fewer elements onto the vector will not change its capacity or
133 /// cause reallocation to occur. However, if the vector's length is increased to 11, it will have
134 /// to reallocate, which can be slow. For this reason, it is recommended to use
135 /// `Vec::with_capacity` whenever possible to specify how big the vector is expected to get.
136 #[unsafe_no_drop_flag]
139 ptr: NonZero<*mut T>,
144 unsafe impl<T: Send> Send for Vec<T> { }
145 unsafe impl<T: Sync> Sync for Vec<T> { }
147 ////////////////////////////////////////////////////////////////////////////////
149 ////////////////////////////////////////////////////////////////////////////////
152 /// Constructs a new, empty `Vec<T>`.
154 /// The vector will not allocate until elements are pushed onto it.
159 /// let mut vec: Vec<int> = Vec::new();
163 pub fn new() -> Vec<T> {
164 // We want ptr to never be NULL so instead we set it to some arbitrary
165 // non-null value which is fine since we never call deallocate on the ptr
166 // if cap is 0. The reason for this is because the pointer of a slice
167 // being NULL would break the null pointer optimization for enums.
168 Vec { ptr: unsafe { NonZero::new(EMPTY as *mut T) }, len: 0, cap: 0 }
171 /// Constructs a new, empty `Vec<T>` with the specified capacity.
173 /// The vector will be able to hold exactly `capacity` elements without reallocating. If
174 /// `capacity` is 0, the vector will not allocate.
176 /// It is important to note that this function does not specify the *length* of the returned
177 /// vector, but only the *capacity*. (For an explanation of the difference between length and
178 /// capacity, see the main `Vec<T>` docs above, 'Capacity and reallocation'.)
183 /// let mut vec: Vec<int> = Vec::with_capacity(10);
185 /// // The vector contains no items, even though it has capacity for more
186 /// assert_eq!(vec.len(), 0);
188 /// // These are all done without reallocating...
189 /// for i in range(0i, 10) {
193 /// // ...but this may make the vector reallocate
198 pub fn with_capacity(capacity: uint) -> Vec<T> {
199 if mem::size_of::<T>() == 0 {
200 Vec { ptr: unsafe { NonZero::new(EMPTY as *mut T) }, len: 0, cap: uint::MAX }
201 } else if capacity == 0 {
204 let size = capacity.checked_mul(mem::size_of::<T>())
205 .expect("capacity overflow");
206 let ptr = unsafe { allocate(size, mem::min_align_of::<T>()) };
207 if ptr.is_null() { ::alloc::oom() }
208 Vec { ptr: unsafe { NonZero::new(ptr as *mut T) }, len: 0, cap: capacity }
212 /// Creates a `Vec<T>` directly from the raw components of another vector.
214 /// This is highly unsafe, due to the number of invariants that aren't checked.
223 /// let mut v = vec![1i, 2, 3];
225 /// // Pull out the various important pieces of information about `v`
226 /// let p = v.as_mut_ptr();
227 /// let len = v.len();
228 /// let cap = v.capacity();
231 /// // Cast `v` into the void: no destructor run, so we are in
232 /// // complete control of the allocation to which `p` points.
235 /// // Overwrite memory with 4, 5, 6
236 /// for i in range(0, len as int) {
237 /// ptr::write(p.offset(i), 4 + i);
240 /// // Put everything back together into a Vec
241 /// let rebuilt = Vec::from_raw_parts(p, len, cap);
242 /// assert_eq!(rebuilt, vec![4i, 5i, 6i]);
247 pub unsafe fn from_raw_parts(ptr: *mut T, length: uint,
248 capacity: uint) -> Vec<T> {
249 Vec { ptr: NonZero::new(ptr), len: length, cap: capacity }
252 /// Creates a vector by copying the elements from a raw pointer.
254 /// This function will copy `elts` contiguous elements starting at `ptr` into a new allocation
255 /// owned by the returned `Vec<T>`. The elements of the buffer are copied into the vector
256 /// without cloning, as if `ptr::read()` were called on them.
258 #[unstable = "may be better expressed via composition"]
259 pub unsafe fn from_raw_buf(ptr: *const T, elts: uint) -> Vec<T> {
260 let mut dst = Vec::with_capacity(elts);
262 ptr::copy_nonoverlapping_memory(dst.as_mut_ptr(), ptr, elts);
266 /// Returns the number of elements the vector can hold without
272 /// let vec: Vec<int> = Vec::with_capacity(10);
273 /// assert_eq!(vec.capacity(), 10);
277 pub fn capacity(&self) -> uint {
281 /// Reserves capacity for at least `additional` more elements to be inserted in the given
282 /// `Vec<T>`. The collection may reserve more space to avoid frequent reallocations.
286 /// Panics if the new capacity overflows `uint`.
291 /// let mut vec: Vec<int> = vec![1];
293 /// assert!(vec.capacity() >= 11);
296 pub fn reserve(&mut self, additional: uint) {
297 if self.cap - self.len < additional {
298 let err_msg = "Vec::reserve: `uint` overflow";
299 let new_cap = self.len.checked_add(additional).expect(err_msg)
300 .checked_next_power_of_two().expect(err_msg);
301 self.grow_capacity(new_cap);
305 /// Reserves the minimum capacity for exactly `additional` more elements to
306 /// be inserted in the given `Vec<T>`. Does nothing if the capacity is already
309 /// Note that the allocator may give the collection more space than it
310 /// requests. Therefore capacity can not be relied upon to be precisely
311 /// minimal. Prefer `reserve` if future insertions are expected.
315 /// Panics if the new capacity overflows `uint`.
320 /// let mut vec: Vec<int> = vec![1];
321 /// vec.reserve_exact(10);
322 /// assert!(vec.capacity() >= 11);
325 pub fn reserve_exact(&mut self, additional: uint) {
326 if self.cap - self.len < additional {
327 match self.len.checked_add(additional) {
328 None => panic!("Vec::reserve: `uint` overflow"),
329 Some(new_cap) => self.grow_capacity(new_cap)
334 /// Shrinks the capacity of the vector as much as possible.
336 /// It will drop down as close as possible to the length but the allocator
337 /// may still inform the vector that there is space for a few more elements.
342 /// let mut vec: Vec<int> = Vec::with_capacity(10);
343 /// vec.push_all(&[1, 2, 3]);
344 /// assert_eq!(vec.capacity(), 10);
345 /// vec.shrink_to_fit();
346 /// assert!(vec.capacity() >= 3);
349 pub fn shrink_to_fit(&mut self) {
350 if mem::size_of::<T>() == 0 { return }
355 dealloc(*self.ptr, self.cap)
361 // Overflow check is unnecessary as the vector is already at
363 let ptr = reallocate(*self.ptr as *mut u8,
364 self.cap * mem::size_of::<T>(),
365 self.len * mem::size_of::<T>(),
366 mem::min_align_of::<T>()) as *mut T;
367 if ptr.is_null() { ::alloc::oom() }
368 self.ptr = NonZero::new(ptr);
374 /// Convert the vector into Box<[T]>.
376 /// Note that this will drop any excess capacity. Calling this and
377 /// converting back to a vector with `into_vec()` is equivalent to calling
378 /// `shrink_to_fit()`.
380 pub fn into_boxed_slice(mut self) -> Box<[T]> {
381 self.shrink_to_fit();
383 let xs: Box<[T]> = mem::transmute(self.as_mut_slice());
389 /// Shorten a vector, dropping excess elements.
391 /// If `len` is greater than the vector's current length, this has no
397 /// let mut vec = vec![1i, 2, 3, 4];
399 /// assert_eq!(vec, vec![1, 2]);
402 pub fn truncate(&mut self, len: uint) {
404 // drop any extra elements
405 while len < self.len {
406 // decrement len before the read(), so a panic on Drop doesn't
407 // re-drop the just-failed value.
409 ptr::read(self.get_unchecked(self.len));
414 /// Returns a mutable slice of the elements of `self`.
419 /// fn foo(slice: &mut [int]) {}
421 /// let mut vec = vec![1i, 2];
422 /// foo(vec.as_mut_slice());
426 pub fn as_mut_slice<'a>(&'a mut self) -> &'a mut [T] {
428 mem::transmute(RawSlice {
435 /// Creates a consuming iterator, that is, one that moves each value out of
436 /// the vector (from start to end). The vector cannot be used after calling
442 /// let v = vec!["a".to_string(), "b".to_string()];
443 /// for s in v.into_iter() {
444 /// // s has type String, not &String
445 /// println!("{}", s);
450 pub fn into_iter(self) -> IntoIter<T> {
454 let begin = ptr as *const T;
455 let end = if mem::size_of::<T>() == 0 {
456 (ptr as uint + self.len()) as *const T
458 ptr.offset(self.len() as int) as *const T
461 IntoIter { allocation: ptr, cap: cap, ptr: begin, end: end }
465 /// Sets the length of a vector.
467 /// This will explicitly set the size of the vector, without actually
468 /// modifying its buffers, so it is up to the caller to ensure that the
469 /// vector is actually the specified size.
474 /// let mut v = vec![1u, 2, 3, 4];
481 pub unsafe fn set_len(&mut self, len: uint) {
485 /// Removes an element from anywhere in the vector and return it, replacing
486 /// it with the last element.
488 /// This does not preserve ordering, but is O(1).
492 /// Panics if `index` is out of bounds.
497 /// let mut v = vec!["foo", "bar", "baz", "qux"];
499 /// assert_eq!(v.swap_remove(1), "bar");
500 /// assert_eq!(v, vec!["foo", "qux", "baz"]);
502 /// assert_eq!(v.swap_remove(0), "foo");
503 /// assert_eq!(v, vec!["baz", "qux"]);
507 pub fn swap_remove(&mut self, index: uint) -> T {
508 let length = self.len();
509 self.swap(index, length - 1);
513 /// Inserts an element at position `index` within the vector, shifting all
514 /// elements after position `i` one position to the right.
518 /// Panics if `index` is not between `0` and the vector's length (both
519 /// bounds inclusive).
524 /// let mut vec = vec![1i, 2, 3];
525 /// vec.insert(1, 4);
526 /// assert_eq!(vec, vec![1, 4, 2, 3]);
527 /// vec.insert(4, 5);
528 /// assert_eq!(vec, vec![1, 4, 2, 3, 5]);
531 pub fn insert(&mut self, index: uint, element: T) {
532 let len = self.len();
533 assert!(index <= len);
534 // space for the new element
537 unsafe { // infallible
538 // The spot to put the new value
540 let p = self.as_mut_ptr().offset(index as int);
541 // Shift everything over to make space. (Duplicating the
542 // `index`th element into two consecutive places.)
543 ptr::copy_memory(p.offset(1), &*p, len - index);
544 // Write it in, overwriting the first copy of the `index`th
546 ptr::write(&mut *p, element);
548 self.set_len(len + 1);
552 /// Removes and returns the element at position `index` within the vector,
553 /// shifting all elements after position `index` one position to the left.
557 /// Panics if `i` is out of bounds.
562 /// let mut v = vec![1i, 2, 3];
563 /// assert_eq!(v.remove(1), 2);
564 /// assert_eq!(v, vec![1, 3]);
567 pub fn remove(&mut self, index: uint) -> T {
568 let len = self.len();
569 assert!(index < len);
570 unsafe { // infallible
573 // the place we are taking from.
574 let ptr = self.as_mut_ptr().offset(index as int);
575 // copy it out, unsafely having a copy of the value on
576 // the stack and in the vector at the same time.
577 ret = ptr::read(ptr);
579 // Shift everything down to fill in that spot.
580 ptr::copy_memory(ptr, &*ptr.offset(1), len - index - 1);
582 self.set_len(len - 1);
587 /// Retains only the elements specified by the predicate.
589 /// In other words, remove all elements `e` such that `f(&e)` returns false.
590 /// This method operates in place and preserves the order of the retained
596 /// let mut vec = vec![1i, 2, 3, 4];
597 /// vec.retain(|&x| x%2 == 0);
598 /// assert_eq!(vec, vec![2, 4]);
601 pub fn retain<F>(&mut self, mut f: F) where F: FnMut(&T) -> bool {
602 let len = self.len();
605 let v = self.as_mut_slice();
607 for i in range(0u, len) {
616 self.truncate(len - del);
620 /// Appends an element to the back of a collection.
624 /// Panics if the number of elements in the vector overflows a `uint`.
629 /// let mut vec = vec!(1i, 2);
631 /// assert_eq!(vec, vec!(1, 2, 3));
635 pub fn push(&mut self, value: T) {
636 if mem::size_of::<T>() == 0 {
637 // zero-size types consume no memory, so we can't rely on the
638 // address space running out
639 self.len = self.len.checked_add(1).expect("length overflow");
640 unsafe { mem::forget(value); }
643 if self.len == self.cap {
644 let old_size = self.cap * mem::size_of::<T>();
645 let size = max(old_size, 2 * mem::size_of::<T>()) * 2;
646 if old_size > size { panic!("capacity overflow") }
648 let ptr = alloc_or_realloc(*self.ptr, old_size, size);
649 if ptr.is_null() { ::alloc::oom() }
650 self.ptr = NonZero::new(ptr);
652 self.cap = max(self.cap, 2) * 2;
656 let end = (*self.ptr).offset(self.len as int);
657 ptr::write(&mut *end, value);
662 /// Removes the last element from a vector and returns it, or `None` if it is empty.
667 /// let mut vec = vec![1i, 2, 3];
668 /// assert_eq!(vec.pop(), Some(3));
669 /// assert_eq!(vec, vec![1, 2]);
673 pub fn pop(&mut self) -> Option<T> {
679 Some(ptr::read(self.get_unchecked(self.len())))
684 /// Moves all the elements of `other` into `Self`, leaving `other` empty.
688 /// Panics if the number of elements in the vector overflows a `uint`.
692 /// let mut vec = vec![1, 2, 3];
693 /// let mut vec2 = vec![4, 5, 6];
694 /// vec.append(&mut vec2);
695 /// assert_eq!(vec, vec![1, 2, 3, 4, 5, 6]);
696 /// assert_eq!(vec2, vec![]);
699 #[unstable = "new API, waiting for dust to settle"]
700 pub fn append(&mut self, other: &mut Self) {
701 if mem::size_of::<T>() == 0 {
702 // zero-size types consume no memory, so we can't rely on the
703 // address space running out
704 self.len = self.len.checked_add(other.len()).expect("length overflow");
705 unsafe { other.set_len(0) }
708 self.reserve(other.len());
709 let len = self.len();
711 ptr::copy_nonoverlapping_memory(
712 self.get_unchecked_mut(len),
717 self.len += other.len();
718 unsafe { other.set_len(0); }
721 /// Creates a draining iterator that clears the `Vec` and iterates over
722 /// the removed items from start to end.
727 /// let mut v = vec!["a".to_string(), "b".to_string()];
728 /// for s in v.drain() {
729 /// // s has type String, not &String
730 /// println!("{}", s);
732 /// assert!(v.is_empty());
735 #[unstable = "matches collection reform specification, waiting for dust to settle"]
736 pub fn drain<'a>(&'a mut self) -> Drain<'a, T> {
738 let begin = *self.ptr as *const T;
739 let end = if mem::size_of::<T>() == 0 {
740 (*self.ptr as uint + self.len()) as *const T
742 (*self.ptr).offset(self.len() as int) as *const T
748 marker: ContravariantLifetime,
753 /// Clears the vector, removing all values.
758 /// let mut v = vec![1i, 2, 3];
762 /// assert!(v.is_empty());
766 pub fn clear(&mut self) {
770 /// Returns the number of elements in the vector.
775 /// let a = vec![1i, 2, 3];
776 /// assert_eq!(a.len(), 3);
780 pub fn len(&self) -> uint { self.len }
782 /// Returns `true` if the vector contains no elements.
787 /// let mut v = Vec::new();
788 /// assert!(v.is_empty());
791 /// assert!(!v.is_empty());
794 pub fn is_empty(&self) -> bool { self.len() == 0 }
796 /// Converts a `Vec<T>` to a `Vec<U>` where `T` and `U` have the same
797 /// size and in case they are not zero-sized the same minimal alignment.
801 /// Panics if `T` and `U` have differing sizes or are not zero-sized and
802 /// have differing minimal alignments.
807 /// let v = vec![0u, 1, 2];
808 /// let w = v.map_in_place(|i| i + 3);
809 /// assert_eq!(w.as_slice(), [3, 4, 5].as_slice());
811 /// #[derive(PartialEq, Show)]
812 /// struct Newtype(u8);
813 /// let bytes = vec![0x11, 0x22];
814 /// let newtyped_bytes = bytes.map_in_place(|x| Newtype(x));
815 /// assert_eq!(newtyped_bytes.as_slice(), [Newtype(0x11), Newtype(0x22)].as_slice());
817 #[unstable = "API may change to provide stronger guarantees"]
818 pub fn map_in_place<U, F>(self, mut f: F) -> Vec<U> where F: FnMut(T) -> U {
819 // FIXME: Assert statically that the types `T` and `U` have the same
821 assert!(mem::size_of::<T>() == mem::size_of::<U>());
825 if mem::size_of::<T>() != 0 {
826 // FIXME: Assert statically that the types `T` and `U` have the
827 // same minimal alignment in case they are not zero-sized.
829 // These asserts are necessary because the `min_align_of` of the
830 // types are passed to the allocator by `Vec`.
831 assert!(mem::min_align_of::<T>() == mem::min_align_of::<U>());
833 // This `as int` cast is safe, because the size of the elements of the
834 // vector is not 0, and:
836 // 1) If the size of the elements in the vector is 1, the `int` may
837 // overflow, but it has the correct bit pattern so that the
838 // `.offset()` function will work.
841 // Address space 0x0-0xF.
842 // `u8` array at: 0x1.
843 // Size of `u8` array: 0x8.
844 // Calculated `offset`: -0x8.
845 // After `array.offset(offset)`: 0x9.
846 // (0x1 + 0x8 = 0x1 - 0x8)
848 // 2) If the size of the elements in the vector is >1, the `uint` ->
849 // `int` conversion can't overflow.
850 let offset = vec.len() as int;
851 let start = vec.as_mut_ptr();
853 let mut pv = PartialVecNonZeroSized {
857 // This points inside the vector, as the vector has length
859 end_t: unsafe { start.offset(offset) },
860 start_u: start as *mut U,
861 end_u: start as *mut U,
872 while pv.end_u as *mut T != pv.end_t {
876 // +-+-+-+-+-+-+-+-+-+
877 // |U|...|U|T|T|...|T|
878 // +-+-+-+-+-+-+-+-+-+
882 let t = ptr::read(pv.start_t);
885 // +-+-+-+-+-+-+-+-+-+
886 // |U|...|U|X|T|...|T|
887 // +-+-+-+-+-+-+-+-+-+
890 // We must not panic here, one cell is marked as `T`
891 // although it is not `T`.
893 pv.start_t = pv.start_t.offset(1);
896 // +-+-+-+-+-+-+-+-+-+
897 // |U|...|U|X|T|...|T|
898 // +-+-+-+-+-+-+-+-+-+
901 // We may panic again.
903 // The function given by the user might panic.
906 ptr::write(pv.end_u, u);
909 // +-+-+-+-+-+-+-+-+-+
910 // |U|...|U|U|T|...|T|
911 // +-+-+-+-+-+-+-+-+-+
914 // We should not panic here, because that would leak the `U`
915 // pointed to by `end_u`.
917 pv.end_u = pv.end_u.offset(1);
920 // +-+-+-+-+-+-+-+-+-+
921 // |U|...|U|U|T|...|T|
922 // +-+-+-+-+-+-+-+-+-+
925 // We may panic again.
937 // Extract `vec` and prevent the destructor of
938 // `PartialVecNonZeroSized` from running. Note that none of the
939 // function calls can panic, thus no resources can be leaked (as the
940 // `vec` member of `PartialVec` is the only one which holds
941 // allocations -- and it is returned from this function. None of
944 let vec_len = pv.vec.len();
945 let vec_cap = pv.vec.capacity();
946 let vec_ptr = pv.vec.as_mut_ptr() as *mut U;
948 Vec::from_raw_parts(vec_ptr, vec_len, vec_cap)
951 // Put the `Vec` into the `PartialVecZeroSized` structure and
952 // prevent the destructor of the `Vec` from running. Since the
953 // `Vec` contained zero-sized objects, it did not allocate, so we
954 // are not leaking memory here.
955 let mut pv = PartialVecZeroSized::<T,U> {
958 marker_t: InvariantType,
959 marker_u: InvariantType,
961 unsafe { mem::forget(vec); }
963 while pv.num_t != 0 {
965 // Create a `T` out of thin air and decrement `num_t`. This
966 // must not panic between these steps, as otherwise a
967 // destructor of `T` which doesn't exist runs.
968 let t = mem::uninitialized();
971 // The function given by the user might panic.
974 // Forget the `U` and increment `num_u`. This increment
975 // cannot overflow the `uint` as we only do this for a
976 // number of times that fits into a `uint` (and start with
977 // `0`). Again, we should not panic between these steps.
982 // Create a `Vec` from our `PartialVecZeroSized` and make sure the
983 // destructor of the latter will not run. None of this can panic.
984 let mut result = Vec::new();
986 result.set_len(pv.num_u);
994 impl<T: Clone> Vec<T> {
995 /// Resizes the `Vec` in-place so that `len()` is equal to `new_len`.
997 /// Calls either `extend()` or `truncate()` depending on whether `new_len`
998 /// is larger than the current value of `len()` or not.
1003 /// let mut vec = vec!["hello"];
1004 /// vec.resize(3, "world");
1005 /// assert_eq!(vec, vec!["hello", "world", "world"]);
1007 /// let mut vec = vec![1i, 2, 3, 4];
1008 /// vec.resize(2, 0);
1009 /// assert_eq!(vec, vec![1, 2]);
1011 #[unstable = "matches collection reform specification; waiting for dust to settle"]
1012 pub fn resize(&mut self, new_len: uint, value: T) {
1013 let len = self.len();
1016 self.extend(repeat(value).take(new_len - len));
1018 self.truncate(new_len);
1022 /// Appends all elements in a slice to the `Vec`.
1024 /// Iterates over the slice `other`, clones each element, and then appends
1025 /// it to this `Vec`. The `other` vector is traversed in-order.
1030 /// let mut vec = vec![1i];
1031 /// vec.push_all(&[2i, 3, 4]);
1032 /// assert_eq!(vec, vec![1, 2, 3, 4]);
1035 #[unstable = "likely to be replaced by a more optimized extend"]
1036 pub fn push_all(&mut self, other: &[T]) {
1037 self.reserve(other.len());
1039 for i in range(0, other.len()) {
1040 let len = self.len();
1042 // Unsafe code so this can be optimised to a memcpy (or something similarly
1043 // fast) when T is Copy. LLVM is easily confused, so any extra operations
1044 // during the loop can prevent this optimisation.
1047 self.get_unchecked_mut(len),
1048 other.get_unchecked(i).clone());
1049 self.set_len(len + 1);
1055 impl<T: PartialEq> Vec<T> {
1056 /// Removes consecutive repeated elements in the vector.
1058 /// If the vector is sorted, this removes all duplicates.
1063 /// let mut vec = vec![1i, 2, 2, 3, 2];
1067 /// assert_eq!(vec, vec![1i, 2, 3, 2]);
1070 pub fn dedup(&mut self) {
1072 // Although we have a mutable reference to `self`, we cannot make
1073 // *arbitrary* changes. The `PartialEq` comparisons could panic, so we
1074 // must ensure that the vector is in a valid state at all time.
1076 // The way that we handle this is by using swaps; we iterate
1077 // over all the elements, swapping as we go so that at the end
1078 // the elements we wish to keep are in the front, and those we
1079 // wish to reject are at the back. We can then truncate the
1080 // vector. This operation is still O(n).
1082 // Example: We start in this state, where `r` represents "next
1083 // read" and `w` represents "next_write`.
1086 // +---+---+---+---+---+---+
1087 // | 0 | 1 | 1 | 2 | 3 | 3 |
1088 // +---+---+---+---+---+---+
1091 // Comparing self[r] against self[w-1], this is not a duplicate, so
1092 // we swap self[r] and self[w] (no effect as r==w) and then increment both
1093 // r and w, leaving us with:
1096 // +---+---+---+---+---+---+
1097 // | 0 | 1 | 1 | 2 | 3 | 3 |
1098 // +---+---+---+---+---+---+
1101 // Comparing self[r] against self[w-1], this value is a duplicate,
1102 // so we increment `r` but leave everything else unchanged:
1105 // +---+---+---+---+---+---+
1106 // | 0 | 1 | 1 | 2 | 3 | 3 |
1107 // +---+---+---+---+---+---+
1110 // Comparing self[r] against self[w-1], this is not a duplicate,
1111 // so swap self[r] and self[w] and advance r and w:
1114 // +---+---+---+---+---+---+
1115 // | 0 | 1 | 2 | 1 | 3 | 3 |
1116 // +---+---+---+---+---+---+
1119 // Not a duplicate, repeat:
1122 // +---+---+---+---+---+---+
1123 // | 0 | 1 | 2 | 3 | 1 | 3 |
1124 // +---+---+---+---+---+---+
1127 // Duplicate, advance r. End of vec. Truncate to w.
1129 let ln = self.len();
1130 if ln < 1 { return; }
1132 // Avoid bounds checks by using unsafe pointers.
1133 let p = self.as_mut_ptr();
1138 let p_r = p.offset(r as int);
1139 let p_wm1 = p.offset((w - 1) as int);
1142 let p_w = p_wm1.offset(1);
1143 mem::swap(&mut *p_r, &mut *p_w);
1155 ////////////////////////////////////////////////////////////////////////////////
1156 // Internal methods and functions
1157 ////////////////////////////////////////////////////////////////////////////////
1160 /// Reserves capacity for exactly `capacity` elements in the given vector.
1162 /// If the capacity for `self` is already equal to or greater than the
1163 /// requested capacity, then no action is taken.
1164 fn grow_capacity(&mut self, capacity: uint) {
1165 if mem::size_of::<T>() == 0 { return }
1167 if capacity > self.cap {
1168 let size = capacity.checked_mul(mem::size_of::<T>())
1169 .expect("capacity overflow");
1171 let ptr = alloc_or_realloc(*self.ptr, self.cap * mem::size_of::<T>(), size);
1172 if ptr.is_null() { ::alloc::oom() }
1173 self.ptr = NonZero::new(ptr);
1175 self.cap = capacity;
1180 // FIXME: #13996: need a way to mark the return value as `noalias`
1182 unsafe fn alloc_or_realloc<T>(ptr: *mut T, old_size: uint, size: uint) -> *mut T {
1184 allocate(size, mem::min_align_of::<T>()) as *mut T
1186 reallocate(ptr as *mut u8, old_size, size, mem::min_align_of::<T>()) as *mut T
1191 unsafe fn dealloc<T>(ptr: *mut T, len: uint) {
1192 if mem::size_of::<T>() != 0 {
1193 deallocate(ptr as *mut u8,
1194 len * mem::size_of::<T>(),
1195 mem::min_align_of::<T>())
1199 ////////////////////////////////////////////////////////////////////////////////
1200 // Common trait implementations for Vec
1201 ////////////////////////////////////////////////////////////////////////////////
1204 impl<T:Clone> Clone for Vec<T> {
1205 fn clone(&self) -> Vec<T> { ::slice::SliceExt::to_vec(self.as_slice()) }
1207 fn clone_from(&mut self, other: &Vec<T>) {
1208 // drop anything in self that will not be overwritten
1209 if self.len() > other.len() {
1210 self.truncate(other.len())
1213 // reuse the contained values' allocations/resources.
1214 for (place, thing) in self.iter_mut().zip(other.iter()) {
1215 place.clone_from(thing)
1218 // self.len <= other.len due to the truncate above, so the
1219 // slice here is always in-bounds.
1220 let slice = &other[self.len()..];
1221 self.push_all(slice);
1225 impl<S: hash::Writer + hash::Hasher, T: Hash<S>> Hash<S> for Vec<T> {
1227 fn hash(&self, state: &mut S) {
1228 self.as_slice().hash(state);
1233 impl<T> Index<uint> for Vec<T> {
1237 fn index<'a>(&'a self, index: &uint) -> &'a T {
1238 &self.as_slice()[*index]
1243 impl<T> IndexMut<uint> for Vec<T> {
1247 fn index_mut<'a>(&'a mut self, index: &uint) -> &'a mut T {
1248 &mut self.as_mut_slice()[*index]
1254 impl<T> ops::Index<ops::Range<uint>> for Vec<T> {
1257 fn index(&self, index: &ops::Range<uint>) -> &[T] {
1258 self.as_slice().index(index)
1262 impl<T> ops::Index<ops::RangeTo<uint>> for Vec<T> {
1265 fn index(&self, index: &ops::RangeTo<uint>) -> &[T] {
1266 self.as_slice().index(index)
1270 impl<T> ops::Index<ops::RangeFrom<uint>> for Vec<T> {
1273 fn index(&self, index: &ops::RangeFrom<uint>) -> &[T] {
1274 self.as_slice().index(index)
1278 impl<T> ops::Index<ops::FullRange> for Vec<T> {
1281 fn index(&self, _index: &ops::FullRange) -> &[T] {
1287 impl<T> ops::IndexMut<ops::Range<uint>> for Vec<T> {
1290 fn index_mut(&mut self, index: &ops::Range<uint>) -> &mut [T] {
1291 self.as_mut_slice().index_mut(index)
1295 impl<T> ops::IndexMut<ops::RangeTo<uint>> for Vec<T> {
1298 fn index_mut(&mut self, index: &ops::RangeTo<uint>) -> &mut [T] {
1299 self.as_mut_slice().index_mut(index)
1303 impl<T> ops::IndexMut<ops::RangeFrom<uint>> for Vec<T> {
1306 fn index_mut(&mut self, index: &ops::RangeFrom<uint>) -> &mut [T] {
1307 self.as_mut_slice().index_mut(index)
1311 impl<T> ops::IndexMut<ops::FullRange> for Vec<T> {
1314 fn index_mut(&mut self, _index: &ops::FullRange) -> &mut [T] {
1320 impl<T> ops::Deref for Vec<T> {
1323 fn deref<'a>(&'a self) -> &'a [T] { self.as_slice() }
1327 impl<T> ops::DerefMut for Vec<T> {
1328 fn deref_mut<'a>(&'a mut self) -> &'a mut [T] { self.as_mut_slice() }
1332 impl<T> FromIterator<T> for Vec<T> {
1334 fn from_iter<I:Iterator<Item=T>>(mut iterator: I) -> Vec<T> {
1335 let (lower, _) = iterator.size_hint();
1336 let mut vector = Vec::with_capacity(lower);
1337 for element in iterator {
1338 vector.push(element)
1344 #[unstable = "waiting on Extend stability"]
1345 impl<T> Extend<T> for Vec<T> {
1347 fn extend<I: Iterator<Item=T>>(&mut self, mut iterator: I) {
1348 let (lower, _) = iterator.size_hint();
1349 self.reserve(lower);
1350 for element in iterator {
1356 impl<A, B> PartialEq<Vec<B>> for Vec<A> where A: PartialEq<B> {
1358 fn eq(&self, other: &Vec<B>) -> bool { PartialEq::eq(&**self, &**other) }
1360 fn ne(&self, other: &Vec<B>) -> bool { PartialEq::ne(&**self, &**other) }
1363 macro_rules! impl_eq {
1364 ($lhs:ty, $rhs:ty) => {
1365 impl<'b, A, B> PartialEq<$rhs> for $lhs where A: PartialEq<B> {
1367 fn eq(&self, other: &$rhs) -> bool { PartialEq::eq(&**self, &**other) }
1369 fn ne(&self, other: &$rhs) -> bool { PartialEq::ne(&**self, &**other) }
1372 impl<'b, A, B> PartialEq<$lhs> for $rhs where B: PartialEq<A> {
1374 fn eq(&self, other: &$lhs) -> bool { PartialEq::eq(&**self, &**other) }
1376 fn ne(&self, other: &$lhs) -> bool { PartialEq::ne(&**self, &**other) }
1381 impl_eq! { Vec<A>, &'b [B] }
1382 impl_eq! { Vec<A>, &'b mut [B] }
1384 impl<'a, A, B> PartialEq<Vec<B>> for CowVec<'a, A> where A: PartialEq<B> + Clone {
1386 fn eq(&self, other: &Vec<B>) -> bool { PartialEq::eq(&**self, &**other) }
1388 fn ne(&self, other: &Vec<B>) -> bool { PartialEq::ne(&**self, &**other) }
1391 impl<'a, A, B> PartialEq<CowVec<'a, A>> for Vec<B> where A: Clone, B: PartialEq<A> {
1393 fn eq(&self, other: &CowVec<'a, A>) -> bool { PartialEq::eq(&**self, &**other) }
1395 fn ne(&self, other: &CowVec<'a, A>) -> bool { PartialEq::ne(&**self, &**other) }
1398 macro_rules! impl_eq_for_cowvec {
1400 impl<'a, 'b, A, B> PartialEq<$rhs> for CowVec<'a, A> where A: PartialEq<B> + Clone {
1402 fn eq(&self, other: &$rhs) -> bool { PartialEq::eq(&**self, &**other) }
1404 fn ne(&self, other: &$rhs) -> bool { PartialEq::ne(&**self, &**other) }
1407 impl<'a, 'b, A, B> PartialEq<CowVec<'a, A>> for $rhs where A: Clone, B: PartialEq<A> {
1409 fn eq(&self, other: &CowVec<'a, A>) -> bool { PartialEq::eq(&**self, &**other) }
1411 fn ne(&self, other: &CowVec<'a, A>) -> bool { PartialEq::ne(&**self, &**other) }
1416 impl_eq_for_cowvec! { &'b [B] }
1417 impl_eq_for_cowvec! { &'b mut [B] }
1419 #[unstable = "waiting on PartialOrd stability"]
1420 impl<T: PartialOrd> PartialOrd for Vec<T> {
1422 fn partial_cmp(&self, other: &Vec<T>) -> Option<Ordering> {
1423 self.as_slice().partial_cmp(other.as_slice())
1427 #[unstable = "waiting on Eq stability"]
1428 impl<T: Eq> Eq for Vec<T> {}
1430 #[unstable = "waiting on Ord stability"]
1431 impl<T: Ord> Ord for Vec<T> {
1433 fn cmp(&self, other: &Vec<T>) -> Ordering {
1434 self.as_slice().cmp(other.as_slice())
1438 impl<T> AsSlice<T> for Vec<T> {
1439 /// Returns a slice into `self`.
1444 /// fn foo(slice: &[int]) {}
1446 /// let vec = vec![1i, 2];
1447 /// foo(vec.as_slice());
1451 fn as_slice<'a>(&'a self) -> &'a [T] {
1453 mem::transmute(RawSlice {
1461 #[unstable = "recent addition, needs more experience"]
1462 impl<'a, T: Clone> Add<&'a [T]> for Vec<T> {
1463 type Output = Vec<T>;
1466 fn add(mut self, rhs: &[T]) -> Vec<T> {
1472 #[unsafe_destructor]
1474 impl<T> Drop for Vec<T> {
1475 fn drop(&mut self) {
1476 // This is (and should always remain) a no-op if the fields are
1477 // zeroed (when moving out, because of #[unsafe_no_drop_flag]).
1480 for x in self.iter() {
1483 dealloc(*self.ptr, self.cap)
1490 impl<T> Default for Vec<T> {
1492 fn default() -> Vec<T> {
1497 #[unstable = "waiting on Show stability"]
1498 impl<T: fmt::Show> fmt::Show for Vec<T> {
1499 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
1500 fmt::Show::fmt(self.as_slice(), f)
1504 impl<'a> fmt::Writer for Vec<u8> {
1505 fn write_str(&mut self, s: &str) -> fmt::Result {
1506 self.push_all(s.as_bytes());
1511 ////////////////////////////////////////////////////////////////////////////////
1513 ////////////////////////////////////////////////////////////////////////////////
1515 #[unstable = "unclear how valuable this alias is"]
1516 /// A clone-on-write vector
1517 pub type CowVec<'a, T> = Cow<'a, Vec<T>, [T]>;
1520 impl<'a, T> FromIterator<T> for CowVec<'a, T> where T: Clone {
1521 fn from_iter<I: Iterator<Item=T>>(it: I) -> CowVec<'a, T> {
1522 Cow::Owned(FromIterator::from_iter(it))
1526 impl<'a, T: 'a> IntoCow<'a, Vec<T>, [T]> for Vec<T> where T: Clone {
1527 fn into_cow(self) -> CowVec<'a, T> {
1532 impl<'a, T> IntoCow<'a, Vec<T>, [T]> for &'a [T] where T: Clone {
1533 fn into_cow(self) -> CowVec<'a, T> {
1538 ////////////////////////////////////////////////////////////////////////////////
1540 ////////////////////////////////////////////////////////////////////////////////
1542 /// An iterator that moves out of a vector.
1544 pub struct IntoIter<T> {
1545 allocation: *mut T, // the block of memory allocated for the vector
1546 cap: uint, // the capacity of the vector
1551 unsafe impl<T: Send> Send for IntoIter<T> { }
1552 unsafe impl<T: Sync> Sync for IntoIter<T> { }
1554 impl<T> IntoIter<T> {
1556 /// Drops all items that have not yet been moved and returns the empty vector.
1558 pub fn into_inner(mut self) -> Vec<T> {
1561 let IntoIter { allocation, cap, ptr: _ptr, end: _end } = self;
1563 Vec { ptr: NonZero::new(allocation), cap: cap, len: 0 }
1569 impl<T> Iterator for IntoIter<T> {
1573 fn next<'a>(&'a mut self) -> Option<T> {
1575 if self.ptr == self.end {
1578 if mem::size_of::<T>() == 0 {
1579 // purposefully don't use 'ptr.offset' because for
1580 // vectors with 0-size elements this would return the
1582 self.ptr = mem::transmute(self.ptr as uint + 1);
1584 // Use a non-null pointer value
1585 Some(ptr::read(mem::transmute(1u)))
1588 self.ptr = self.ptr.offset(1);
1590 Some(ptr::read(old))
1597 fn size_hint(&self) -> (uint, Option<uint>) {
1598 let diff = (self.end as uint) - (self.ptr as uint);
1599 let size = mem::size_of::<T>();
1600 let exact = diff / (if size == 0 {1} else {size});
1601 (exact, Some(exact))
1606 impl<T> DoubleEndedIterator for IntoIter<T> {
1608 fn next_back<'a>(&'a mut self) -> Option<T> {
1610 if self.end == self.ptr {
1613 if mem::size_of::<T>() == 0 {
1614 // See above for why 'ptr.offset' isn't used
1615 self.end = mem::transmute(self.end as uint - 1);
1617 // Use a non-null pointer value
1618 Some(ptr::read(mem::transmute(1u)))
1620 self.end = self.end.offset(-1);
1622 Some(ptr::read(mem::transmute(self.end)))
1630 impl<T> ExactSizeIterator for IntoIter<T> {}
1632 #[unsafe_destructor]
1634 impl<T> Drop for IntoIter<T> {
1635 fn drop(&mut self) {
1636 // destroy the remaining elements
1640 dealloc(self.allocation, self.cap);
1646 /// An iterator that drains a vector.
1647 #[unsafe_no_drop_flag]
1648 #[unstable = "recently added as part of collections reform 2"]
1649 pub struct Drain<'a, T> {
1652 marker: ContravariantLifetime<'a>,
1656 impl<'a, T> Iterator for Drain<'a, T> {
1660 fn next(&mut self) -> Option<T> {
1662 if self.ptr == self.end {
1665 if mem::size_of::<T>() == 0 {
1666 // purposefully don't use 'ptr.offset' because for
1667 // vectors with 0-size elements this would return the
1669 self.ptr = mem::transmute(self.ptr as uint + 1);
1671 // Use a non-null pointer value
1672 Some(ptr::read(mem::transmute(1u)))
1675 self.ptr = self.ptr.offset(1);
1677 Some(ptr::read(old))
1684 fn size_hint(&self) -> (uint, Option<uint>) {
1685 let diff = (self.end as uint) - (self.ptr as uint);
1686 let size = mem::size_of::<T>();
1687 let exact = diff / (if size == 0 {1} else {size});
1688 (exact, Some(exact))
1693 impl<'a, T> DoubleEndedIterator for Drain<'a, T> {
1695 fn next_back(&mut self) -> Option<T> {
1697 if self.end == self.ptr {
1700 if mem::size_of::<T>() == 0 {
1701 // See above for why 'ptr.offset' isn't used
1702 self.end = mem::transmute(self.end as uint - 1);
1704 // Use a non-null pointer value
1705 Some(ptr::read(mem::transmute(1u)))
1707 self.end = self.end.offset(-1);
1709 Some(ptr::read(self.end))
1717 impl<'a, T> ExactSizeIterator for Drain<'a, T> {}
1719 #[unsafe_destructor]
1721 impl<'a, T> Drop for Drain<'a, T> {
1722 fn drop(&mut self) {
1723 // self.ptr == self.end == null if drop has already been called,
1724 // so we can use #[unsafe_no_drop_flag].
1726 // destroy the remaining elements
1731 ////////////////////////////////////////////////////////////////////////////////
1732 // Conversion from &[T] to &Vec<T>
1733 ////////////////////////////////////////////////////////////////////////////////
1735 /// Wrapper type providing a `&Vec<T>` reference via `Deref`.
1737 pub struct DerefVec<'a, T> {
1739 l: ContravariantLifetime<'a>
1743 impl<'a, T> Deref for DerefVec<'a, T> {
1744 type Target = Vec<T>;
1746 fn deref<'b>(&'b self) -> &'b Vec<T> {
1751 // Prevent the inner `Vec<T>` from attempting to deallocate memory.
1752 #[unsafe_destructor]
1754 impl<'a, T> Drop for DerefVec<'a, T> {
1755 fn drop(&mut self) {
1761 /// Convert a slice to a wrapper type providing a `&Vec<T>` reference.
1763 pub fn as_vec<'a, T>(x: &'a [T]) -> DerefVec<'a, T> {
1766 x: Vec::from_raw_parts(x.as_ptr() as *mut T, x.len(), x.len()),
1767 l: ContravariantLifetime::<'a>
1772 ////////////////////////////////////////////////////////////////////////////////
1773 // Partial vec, used for map_in_place
1774 ////////////////////////////////////////////////////////////////////////////////
1776 /// An owned, partially type-converted vector of elements with non-zero size.
1778 /// `T` and `U` must have the same, non-zero size. They must also have the same
1781 /// When the destructor of this struct runs, all `U`s from `start_u` (incl.) to
1782 /// `end_u` (excl.) and all `T`s from `start_t` (incl.) to `end_t` (excl.) are
1783 /// destructed. Additionally the underlying storage of `vec` will be freed.
1784 struct PartialVecNonZeroSized<T,U> {
1793 /// An owned, partially type-converted vector of zero-sized elements.
1795 /// When the destructor of this struct runs, all `num_t` `T`s and `num_u` `U`s
1797 struct PartialVecZeroSized<T,U> {
1800 marker_t: InvariantType<T>,
1801 marker_u: InvariantType<U>,
1804 #[unsafe_destructor]
1805 impl<T,U> Drop for PartialVecNonZeroSized<T,U> {
1806 fn drop(&mut self) {
1808 // `vec` hasn't been modified until now. As it has a length
1809 // currently, this would run destructors of `T`s which might not be
1810 // there. So at first, set `vec`s length to `0`. This must be done
1811 // at first to remain memory-safe as the destructors of `U` or `T`
1812 // might cause unwinding where `vec`s destructor would be executed.
1813 self.vec.set_len(0);
1815 // We have instances of `U`s and `T`s in `vec`. Destruct them.
1816 while self.start_u != self.end_u {
1817 let _ = ptr::read(self.start_u); // Run a `U` destructor.
1818 self.start_u = self.start_u.offset(1);
1820 while self.start_t != self.end_t {
1821 let _ = ptr::read(self.start_t); // Run a `T` destructor.
1822 self.start_t = self.start_t.offset(1);
1824 // After this destructor ran, the destructor of `vec` will run,
1825 // deallocating the underlying memory.
1830 #[unsafe_destructor]
1831 impl<T,U> Drop for PartialVecZeroSized<T,U> {
1832 fn drop(&mut self) {
1834 // Destruct the instances of `T` and `U` this struct owns.
1835 while self.num_t != 0 {
1836 let _: T = mem::uninitialized(); // Run a `T` destructor.
1839 while self.num_u != 0 {
1840 let _: U = mem::uninitialized(); // Run a `U` destructor.
1850 use core::mem::size_of;
1851 use core::iter::repeat;
1852 use core::ops::FullRange;
1856 struct DropCounter<'a> {
1860 #[unsafe_destructor]
1861 impl<'a> Drop for DropCounter<'a> {
1862 fn drop(&mut self) {
1869 let xs = [1u8, 2u8, 3u8];
1870 assert_eq!(as_vec(&xs).as_slice(), xs);
1874 fn test_as_vec_dtor() {
1875 let (mut count_x, mut count_y) = (0, 0);
1877 let xs = &[DropCounter { count: &mut count_x }, DropCounter { count: &mut count_y }];
1878 assert_eq!(as_vec(xs).len(), 2);
1880 assert_eq!(count_x, 1);
1881 assert_eq!(count_y, 1);
1885 fn test_small_vec_struct() {
1886 assert!(size_of::<Vec<u8>>() == size_of::<uint>() * 3);
1890 fn test_double_drop() {
1896 let (mut count_x, mut count_y) = (0, 0);
1898 let mut tv = TwoVec {
1902 tv.x.push(DropCounter {count: &mut count_x});
1903 tv.y.push(DropCounter {count: &mut count_y});
1905 // If Vec had a drop flag, here is where it would be zeroed.
1906 // Instead, it should rely on its internal state to prevent
1907 // doing anything significant when dropped multiple times.
1910 // Here tv goes out of scope, tv.y should be dropped, but not tv.x.
1913 assert_eq!(count_x, 1);
1914 assert_eq!(count_y, 1);
1919 let mut v = Vec::new();
1920 assert_eq!(v.capacity(), 0);
1923 assert!(v.capacity() >= 2);
1925 for i in range(0i, 16) {
1929 assert!(v.capacity() >= 16);
1931 assert!(v.capacity() >= 32);
1936 assert!(v.capacity() >= 33)
1941 let mut v = Vec::new();
1942 let mut w = Vec::new();
1944 v.extend(range(0i, 3));
1945 for i in range(0i, 3) { w.push(i) }
1949 v.extend(range(3i, 10));
1950 for i in range(3i, 10) { w.push(i) }
1956 fn test_slice_from_mut() {
1957 let mut values = vec![1u8,2,3,4,5];
1959 let slice = values.slice_from_mut(2);
1960 assert!(slice == [3, 4, 5]);
1961 for p in slice.iter_mut() {
1966 assert!(values == [1, 2, 5, 6, 7]);
1970 fn test_slice_to_mut() {
1971 let mut values = vec![1u8,2,3,4,5];
1973 let slice = values.slice_to_mut(2);
1974 assert!(slice == [1, 2]);
1975 for p in slice.iter_mut() {
1980 assert!(values == [2, 3, 3, 4, 5]);
1984 fn test_split_at_mut() {
1985 let mut values = vec![1u8,2,3,4,5];
1987 let (left, right) = values.split_at_mut(2);
1989 let left: &[_] = left;
1990 assert!(&left[..left.len()] == &[1, 2][]);
1992 for p in left.iter_mut() {
1997 let right: &[_] = right;
1998 assert!(&right[..right.len()] == &[3, 4, 5][]);
2000 for p in right.iter_mut() {
2005 assert!(values == vec![2u8, 3, 5, 6, 7]);
2010 let v: Vec<int> = vec!();
2011 let w = vec!(1i, 2, 3);
2013 assert_eq!(v, v.clone());
2017 // they should be disjoint in memory.
2018 assert!(w.as_ptr() != z.as_ptr())
2022 fn test_clone_from() {
2024 let three = vec!(box 1i, box 2, box 3);
2025 let two = vec!(box 4i, box 5);
2027 v.clone_from(&three);
2028 assert_eq!(v, three);
2031 v.clone_from(&three);
2032 assert_eq!(v, three);
2039 v.clone_from(&three);
2040 assert_eq!(v, three)
2045 let mut vec = vec![1u, 2, 3, 4];
2046 vec.retain(|&x| x % 2 == 0);
2047 assert!(vec == vec![2u, 4]);
2051 fn zero_sized_values() {
2052 let mut v = Vec::new();
2053 assert_eq!(v.len(), 0);
2055 assert_eq!(v.len(), 1);
2057 assert_eq!(v.len(), 2);
2058 assert_eq!(v.pop(), Some(()));
2059 assert_eq!(v.pop(), Some(()));
2060 assert_eq!(v.pop(), None);
2062 assert_eq!(v.iter().count(), 0);
2064 assert_eq!(v.iter().count(), 1);
2066 assert_eq!(v.iter().count(), 2);
2068 for &() in v.iter() {}
2070 assert_eq!(v.iter_mut().count(), 2);
2072 assert_eq!(v.iter_mut().count(), 3);
2074 assert_eq!(v.iter_mut().count(), 4);
2076 for &mut () in v.iter_mut() {}
2077 unsafe { v.set_len(0); }
2078 assert_eq!(v.iter_mut().count(), 0);
2082 fn test_partition() {
2083 assert_eq!(vec![].into_iter().partition(|x: &int| *x < 3), (vec![], vec![]));
2084 assert_eq!(vec![1i, 2, 3].into_iter().partition(|x: &int| *x < 4), (vec![1, 2, 3], vec![]));
2085 assert_eq!(vec![1i, 2, 3].into_iter().partition(|x: &int| *x < 2), (vec![1], vec![2, 3]));
2086 assert_eq!(vec![1i, 2, 3].into_iter().partition(|x: &int| *x < 0), (vec![], vec![1, 2, 3]));
2090 fn test_zip_unzip() {
2091 let z1 = vec![(1i, 4i), (2, 5), (3, 6)];
2093 let (left, right): (Vec<_>, Vec<_>) = z1.iter().map(|&x| x).unzip();
2095 assert_eq!((1, 4), (left[0], right[0]));
2096 assert_eq!((2, 5), (left[1], right[1]));
2097 assert_eq!((3, 6), (left[2], right[2]));
2101 fn test_unsafe_ptrs() {
2103 // Test on-stack copy-from-buf.
2105 let ptr = a.as_ptr();
2106 let b = Vec::from_raw_buf(ptr, 3u);
2107 assert_eq!(b, vec![1, 2, 3]);
2109 // Test on-heap copy-from-buf.
2110 let c = vec![1i, 2, 3, 4, 5];
2111 let ptr = c.as_ptr();
2112 let d = Vec::from_raw_buf(ptr, 5u);
2113 assert_eq!(d, vec![1, 2, 3, 4, 5]);
2118 fn test_vec_truncate_drop() {
2119 static mut drops: uint = 0;
2121 impl Drop for Elem {
2122 fn drop(&mut self) {
2123 unsafe { drops += 1; }
2127 let mut v = vec![Elem(1), Elem(2), Elem(3), Elem(4), Elem(5)];
2128 assert_eq!(unsafe { drops }, 0);
2130 assert_eq!(unsafe { drops }, 2);
2132 assert_eq!(unsafe { drops }, 5);
2137 fn test_vec_truncate_fail() {
2138 struct BadElem(int);
2139 impl Drop for BadElem {
2140 fn drop(&mut self) {
2141 let BadElem(ref mut x) = *self;
2142 if *x == 0xbadbeef {
2143 panic!("BadElem panic: 0xbadbeef")
2148 let mut v = vec![BadElem(1), BadElem(2), BadElem(0xbadbeef), BadElem(4)];
2154 let vec = vec!(1i, 2, 3);
2155 assert!(vec[1] == 2);
2160 fn test_index_out_of_bounds() {
2161 let vec = vec!(1i, 2, 3);
2167 fn test_slice_out_of_bounds_1() {
2168 let x: Vec<int> = vec![1, 2, 3, 4, 5];
2174 fn test_slice_out_of_bounds_2() {
2175 let x: Vec<int> = vec![1, 2, 3, 4, 5];
2181 fn test_slice_out_of_bounds_3() {
2182 let x: Vec<int> = vec![1, 2, 3, 4, 5];
2188 fn test_slice_out_of_bounds_4() {
2189 let x: Vec<int> = vec![1, 2, 3, 4, 5];
2195 fn test_slice_out_of_bounds_5() {
2196 let x: Vec<int> = vec![1, 2, 3, 4, 5];
2202 fn test_swap_remove_empty() {
2203 let mut vec: Vec<uint> = vec!();
2208 fn test_move_iter_unwrap() {
2209 let mut vec: Vec<uint> = Vec::with_capacity(7);
2212 let ptr = vec.as_ptr();
2213 vec = vec.into_iter().into_inner();
2214 assert_eq!(vec.as_ptr(), ptr);
2215 assert_eq!(vec.capacity(), 7);
2216 assert_eq!(vec.len(), 0);
2221 fn test_map_in_place_incompatible_types_fail() {
2222 let v = vec![0u, 1, 2];
2223 v.map_in_place(|_| ());
2227 fn test_map_in_place() {
2228 let v = vec![0u, 1, 2];
2229 assert_eq!(v.map_in_place(|i: uint| i as int - 1), [-1i, 0, 1]);
2233 fn test_map_in_place_zero_sized() {
2234 let v = vec![(), ()];
2235 #[derive(PartialEq, Show)]
2237 assert_eq!(v.map_in_place(|_| ZeroSized), [ZeroSized, ZeroSized]);
2241 fn test_map_in_place_zero_drop_count() {
2242 use std::sync::atomic::{AtomicUsize, Ordering, ATOMIC_USIZE_INIT};
2244 #[derive(Clone, PartialEq, Show)]
2246 impl Drop for Nothing { fn drop(&mut self) { } }
2248 #[derive(Clone, PartialEq, Show)]
2250 impl Drop for ZeroSized {
2251 fn drop(&mut self) {
2252 DROP_COUNTER.fetch_add(1, Ordering::Relaxed);
2255 const NUM_ELEMENTS: uint = 2;
2256 static DROP_COUNTER: AtomicUsize = ATOMIC_USIZE_INIT;
2258 let v = repeat(Nothing).take(NUM_ELEMENTS).collect::<Vec<_>>();
2260 DROP_COUNTER.store(0, Ordering::Relaxed);
2262 let v = v.map_in_place(|_| ZeroSized);
2263 assert_eq!(DROP_COUNTER.load(Ordering::Relaxed), 0);
2265 assert_eq!(DROP_COUNTER.load(Ordering::Relaxed), NUM_ELEMENTS);
2269 fn test_move_items() {
2270 let vec = vec![1, 2, 3];
2271 let mut vec2 : Vec<i32> = vec![];
2272 for i in vec.into_iter() {
2275 assert!(vec2 == vec![1, 2, 3]);
2279 fn test_move_items_reverse() {
2280 let vec = vec![1, 2, 3];
2281 let mut vec2 : Vec<i32> = vec![];
2282 for i in vec.into_iter().rev() {
2285 assert!(vec2 == vec![3, 2, 1]);
2289 fn test_move_items_zero_sized() {
2290 let vec = vec![(), (), ()];
2291 let mut vec2 : Vec<()> = vec![];
2292 for i in vec.into_iter() {
2295 assert!(vec2 == vec![(), (), ()]);
2299 fn test_drain_items() {
2300 let mut vec = vec![1, 2, 3];
2301 let mut vec2: Vec<i32> = vec![];
2302 for i in vec.drain() {
2305 assert_eq!(vec, []);
2306 assert_eq!(vec2, [ 1, 2, 3 ]);
2310 fn test_drain_items_reverse() {
2311 let mut vec = vec![1, 2, 3];
2312 let mut vec2: Vec<i32> = vec![];
2313 for i in vec.drain().rev() {
2316 assert_eq!(vec, []);
2317 assert_eq!(vec2, [ 3, 2, 1 ]);
2321 fn test_drain_items_zero_sized() {
2322 let mut vec = vec![(), (), ()];
2323 let mut vec2: Vec<()> = vec![];
2324 for i in vec.drain() {
2327 assert_eq!(vec, []);
2328 assert_eq!(vec2, [(), (), ()]);
2332 fn test_into_boxed_slice() {
2333 let xs = vec![1u, 2, 3];
2334 let ys = xs.into_boxed_slice();
2335 assert_eq!(ys.as_slice(), [1u, 2, 3]);
2340 let mut vec = vec![1, 2, 3];
2341 let mut vec2 = vec![4, 5, 6];
2342 vec.append(&mut vec2);
2343 assert_eq!(vec, vec![1, 2, 3, 4, 5, 6]);
2344 assert_eq!(vec2, vec![]);
2348 fn bench_new(b: &mut Bencher) {
2350 let v: Vec<uint> = Vec::new();
2351 assert_eq!(v.len(), 0);
2352 assert_eq!(v.capacity(), 0);
2356 fn do_bench_with_capacity(b: &mut Bencher, src_len: uint) {
2357 b.bytes = src_len as u64;
2360 let v: Vec<uint> = Vec::with_capacity(src_len);
2361 assert_eq!(v.len(), 0);
2362 assert_eq!(v.capacity(), src_len);
2367 fn bench_with_capacity_0000(b: &mut Bencher) {
2368 do_bench_with_capacity(b, 0)
2372 fn bench_with_capacity_0010(b: &mut Bencher) {
2373 do_bench_with_capacity(b, 10)
2377 fn bench_with_capacity_0100(b: &mut Bencher) {
2378 do_bench_with_capacity(b, 100)
2382 fn bench_with_capacity_1000(b: &mut Bencher) {
2383 do_bench_with_capacity(b, 1000)
2386 fn do_bench_from_fn(b: &mut Bencher, src_len: uint) {
2387 b.bytes = src_len as u64;
2390 let dst = range(0, src_len).collect::<Vec<_>>();
2391 assert_eq!(dst.len(), src_len);
2392 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
2397 fn bench_from_fn_0000(b: &mut Bencher) {
2398 do_bench_from_fn(b, 0)
2402 fn bench_from_fn_0010(b: &mut Bencher) {
2403 do_bench_from_fn(b, 10)
2407 fn bench_from_fn_0100(b: &mut Bencher) {
2408 do_bench_from_fn(b, 100)
2412 fn bench_from_fn_1000(b: &mut Bencher) {
2413 do_bench_from_fn(b, 1000)
2416 fn do_bench_from_elem(b: &mut Bencher, src_len: uint) {
2417 b.bytes = src_len as u64;
2420 let dst: Vec<uint> = repeat(5).take(src_len).collect();
2421 assert_eq!(dst.len(), src_len);
2422 assert!(dst.iter().all(|x| *x == 5));
2427 fn bench_from_elem_0000(b: &mut Bencher) {
2428 do_bench_from_elem(b, 0)
2432 fn bench_from_elem_0010(b: &mut Bencher) {
2433 do_bench_from_elem(b, 10)
2437 fn bench_from_elem_0100(b: &mut Bencher) {
2438 do_bench_from_elem(b, 100)
2442 fn bench_from_elem_1000(b: &mut Bencher) {
2443 do_bench_from_elem(b, 1000)
2446 fn do_bench_from_slice(b: &mut Bencher, src_len: uint) {
2447 let src: Vec<uint> = FromIterator::from_iter(range(0, src_len));
2449 b.bytes = src_len as u64;
2452 let dst = src.clone()[].to_vec();
2453 assert_eq!(dst.len(), src_len);
2454 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
2459 fn bench_from_slice_0000(b: &mut Bencher) {
2460 do_bench_from_slice(b, 0)
2464 fn bench_from_slice_0010(b: &mut Bencher) {
2465 do_bench_from_slice(b, 10)
2469 fn bench_from_slice_0100(b: &mut Bencher) {
2470 do_bench_from_slice(b, 100)
2474 fn bench_from_slice_1000(b: &mut Bencher) {
2475 do_bench_from_slice(b, 1000)
2478 fn do_bench_from_iter(b: &mut Bencher, src_len: uint) {
2479 let src: Vec<uint> = FromIterator::from_iter(range(0, src_len));
2481 b.bytes = src_len as u64;
2484 let dst: Vec<uint> = FromIterator::from_iter(src.clone().into_iter());
2485 assert_eq!(dst.len(), src_len);
2486 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
2491 fn bench_from_iter_0000(b: &mut Bencher) {
2492 do_bench_from_iter(b, 0)
2496 fn bench_from_iter_0010(b: &mut Bencher) {
2497 do_bench_from_iter(b, 10)
2501 fn bench_from_iter_0100(b: &mut Bencher) {
2502 do_bench_from_iter(b, 100)
2506 fn bench_from_iter_1000(b: &mut Bencher) {
2507 do_bench_from_iter(b, 1000)
2510 fn do_bench_extend(b: &mut Bencher, dst_len: uint, src_len: uint) {
2511 let dst: Vec<uint> = FromIterator::from_iter(range(0, dst_len));
2512 let src: Vec<uint> = FromIterator::from_iter(range(dst_len, dst_len + src_len));
2514 b.bytes = src_len as u64;
2517 let mut dst = dst.clone();
2518 dst.extend(src.clone().into_iter());
2519 assert_eq!(dst.len(), dst_len + src_len);
2520 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
2525 fn bench_extend_0000_0000(b: &mut Bencher) {
2526 do_bench_extend(b, 0, 0)
2530 fn bench_extend_0000_0010(b: &mut Bencher) {
2531 do_bench_extend(b, 0, 10)
2535 fn bench_extend_0000_0100(b: &mut Bencher) {
2536 do_bench_extend(b, 0, 100)
2540 fn bench_extend_0000_1000(b: &mut Bencher) {
2541 do_bench_extend(b, 0, 1000)
2545 fn bench_extend_0010_0010(b: &mut Bencher) {
2546 do_bench_extend(b, 10, 10)
2550 fn bench_extend_0100_0100(b: &mut Bencher) {
2551 do_bench_extend(b, 100, 100)
2555 fn bench_extend_1000_1000(b: &mut Bencher) {
2556 do_bench_extend(b, 1000, 1000)
2559 fn do_bench_push_all(b: &mut Bencher, dst_len: uint, src_len: uint) {
2560 let dst: Vec<uint> = FromIterator::from_iter(range(0, dst_len));
2561 let src: Vec<uint> = FromIterator::from_iter(range(dst_len, dst_len + src_len));
2563 b.bytes = src_len as u64;
2566 let mut dst = dst.clone();
2567 dst.push_all(src.as_slice());
2568 assert_eq!(dst.len(), dst_len + src_len);
2569 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
2574 fn bench_push_all_0000_0000(b: &mut Bencher) {
2575 do_bench_push_all(b, 0, 0)
2579 fn bench_push_all_0000_0010(b: &mut Bencher) {
2580 do_bench_push_all(b, 0, 10)
2584 fn bench_push_all_0000_0100(b: &mut Bencher) {
2585 do_bench_push_all(b, 0, 100)
2589 fn bench_push_all_0000_1000(b: &mut Bencher) {
2590 do_bench_push_all(b, 0, 1000)
2594 fn bench_push_all_0010_0010(b: &mut Bencher) {
2595 do_bench_push_all(b, 10, 10)
2599 fn bench_push_all_0100_0100(b: &mut Bencher) {
2600 do_bench_push_all(b, 100, 100)
2604 fn bench_push_all_1000_1000(b: &mut Bencher) {
2605 do_bench_push_all(b, 1000, 1000)
2608 fn do_bench_push_all_move(b: &mut Bencher, dst_len: uint, src_len: uint) {
2609 let dst: Vec<uint> = FromIterator::from_iter(range(0u, dst_len));
2610 let src: Vec<uint> = FromIterator::from_iter(range(dst_len, dst_len + src_len));
2612 b.bytes = src_len as u64;
2615 let mut dst = dst.clone();
2616 dst.extend(src.clone().into_iter());
2617 assert_eq!(dst.len(), dst_len + src_len);
2618 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
2623 fn bench_push_all_move_0000_0000(b: &mut Bencher) {
2624 do_bench_push_all_move(b, 0, 0)
2628 fn bench_push_all_move_0000_0010(b: &mut Bencher) {
2629 do_bench_push_all_move(b, 0, 10)
2633 fn bench_push_all_move_0000_0100(b: &mut Bencher) {
2634 do_bench_push_all_move(b, 0, 100)
2638 fn bench_push_all_move_0000_1000(b: &mut Bencher) {
2639 do_bench_push_all_move(b, 0, 1000)
2643 fn bench_push_all_move_0010_0010(b: &mut Bencher) {
2644 do_bench_push_all_move(b, 10, 10)
2648 fn bench_push_all_move_0100_0100(b: &mut Bencher) {
2649 do_bench_push_all_move(b, 100, 100)
2653 fn bench_push_all_move_1000_1000(b: &mut Bencher) {
2654 do_bench_push_all_move(b, 1000, 1000)
2657 fn do_bench_clone(b: &mut Bencher, src_len: uint) {
2658 let src: Vec<uint> = FromIterator::from_iter(range(0, src_len));
2660 b.bytes = src_len as u64;
2663 let dst = src.clone();
2664 assert_eq!(dst.len(), src_len);
2665 assert!(dst.iter().enumerate().all(|(i, x)| i == *x));
2670 fn bench_clone_0000(b: &mut Bencher) {
2671 do_bench_clone(b, 0)
2675 fn bench_clone_0010(b: &mut Bencher) {
2676 do_bench_clone(b, 10)
2680 fn bench_clone_0100(b: &mut Bencher) {
2681 do_bench_clone(b, 100)
2685 fn bench_clone_1000(b: &mut Bencher) {
2686 do_bench_clone(b, 1000)
2689 fn do_bench_clone_from(b: &mut Bencher, times: uint, dst_len: uint, src_len: uint) {
2690 let dst: Vec<uint> = FromIterator::from_iter(range(0, src_len));
2691 let src: Vec<uint> = FromIterator::from_iter(range(dst_len, dst_len + src_len));
2693 b.bytes = (times * src_len) as u64;
2696 let mut dst = dst.clone();
2698 for _ in range(0, times) {
2699 dst.clone_from(&src);
2701 assert_eq!(dst.len(), src_len);
2702 assert!(dst.iter().enumerate().all(|(i, x)| dst_len + i == *x));
2708 fn bench_clone_from_01_0000_0000(b: &mut Bencher) {
2709 do_bench_clone_from(b, 1, 0, 0)
2713 fn bench_clone_from_01_0000_0010(b: &mut Bencher) {
2714 do_bench_clone_from(b, 1, 0, 10)
2718 fn bench_clone_from_01_0000_0100(b: &mut Bencher) {
2719 do_bench_clone_from(b, 1, 0, 100)
2723 fn bench_clone_from_01_0000_1000(b: &mut Bencher) {
2724 do_bench_clone_from(b, 1, 0, 1000)
2728 fn bench_clone_from_01_0010_0010(b: &mut Bencher) {
2729 do_bench_clone_from(b, 1, 10, 10)
2733 fn bench_clone_from_01_0100_0100(b: &mut Bencher) {
2734 do_bench_clone_from(b, 1, 100, 100)
2738 fn bench_clone_from_01_1000_1000(b: &mut Bencher) {
2739 do_bench_clone_from(b, 1, 1000, 1000)
2743 fn bench_clone_from_01_0010_0100(b: &mut Bencher) {
2744 do_bench_clone_from(b, 1, 10, 100)
2748 fn bench_clone_from_01_0100_1000(b: &mut Bencher) {
2749 do_bench_clone_from(b, 1, 100, 1000)
2753 fn bench_clone_from_01_0010_0000(b: &mut Bencher) {
2754 do_bench_clone_from(b, 1, 10, 0)
2758 fn bench_clone_from_01_0100_0010(b: &mut Bencher) {
2759 do_bench_clone_from(b, 1, 100, 10)
2763 fn bench_clone_from_01_1000_0100(b: &mut Bencher) {
2764 do_bench_clone_from(b, 1, 1000, 100)
2768 fn bench_clone_from_10_0000_0000(b: &mut Bencher) {
2769 do_bench_clone_from(b, 10, 0, 0)
2773 fn bench_clone_from_10_0000_0010(b: &mut Bencher) {
2774 do_bench_clone_from(b, 10, 0, 10)
2778 fn bench_clone_from_10_0000_0100(b: &mut Bencher) {
2779 do_bench_clone_from(b, 10, 0, 100)
2783 fn bench_clone_from_10_0000_1000(b: &mut Bencher) {
2784 do_bench_clone_from(b, 10, 0, 1000)
2788 fn bench_clone_from_10_0010_0010(b: &mut Bencher) {
2789 do_bench_clone_from(b, 10, 10, 10)
2793 fn bench_clone_from_10_0100_0100(b: &mut Bencher) {
2794 do_bench_clone_from(b, 10, 100, 100)
2798 fn bench_clone_from_10_1000_1000(b: &mut Bencher) {
2799 do_bench_clone_from(b, 10, 1000, 1000)
2803 fn bench_clone_from_10_0010_0100(b: &mut Bencher) {
2804 do_bench_clone_from(b, 10, 10, 100)
2808 fn bench_clone_from_10_0100_1000(b: &mut Bencher) {
2809 do_bench_clone_from(b, 10, 100, 1000)
2813 fn bench_clone_from_10_0010_0000(b: &mut Bencher) {
2814 do_bench_clone_from(b, 10, 10, 0)
2818 fn bench_clone_from_10_0100_0010(b: &mut Bencher) {
2819 do_bench_clone_from(b, 10, 100, 10)
2823 fn bench_clone_from_10_1000_0100(b: &mut Bencher) {
2824 do_bench_clone_from(b, 10, 1000, 100)