Jonathan S | 03609e5 | 2014-04-21 04:59:12 | [diff] [blame] | 1 | // Copyright 2012-2014 The Rust Project Developers. See the COPYRIGHT |
Graydon Hoare | 00c856c | 2012-12-04 00:48:01 | [diff] [blame] | 2 | // file at the top-level directory of this distribution and at |
| 3 | // http://rust-lang.org/COPYRIGHT. |
| 4 | // |
| 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. |
| 10 | |
Steve Klabnik | 84030fd | 2014-08-30 21:11:22 | [diff] [blame] | 11 | //! This crate implements a double-ended queue with `O(1)` amortized inserts and removals from both |
| 12 | //! ends of the container. It also has `O(1)` indexing like a vector. The contained elements are |
| 13 | //! not required to be copyable, and the queue will be sendable if the contained type is sendable. |
| 14 | //! Its interface `Deque` is defined in `collections`. |
Patrick Walton | f3723cf | 2013-05-17 22:28:44 | [diff] [blame] | 15 | |
Alex Crichton | 6a58537 | 2014-05-30 01:50:12 | [diff] [blame] | 16 | use core::prelude::*; |
| 17 | |
| 18 | use core::cmp; |
Tom Jakubowski | d6a3941 | 2014-06-09 07:30:04 | [diff] [blame] | 19 | use core::default::Default; |
Alex Crichton | 6a58537 | 2014-05-30 01:50:12 | [diff] [blame] | 20 | use core::fmt; |
nham | 6361577 | 2014-07-27 03:18:56 | [diff] [blame] | 21 | use core::iter; |
Alex Crichton | 9d5d97b | 2014-10-15 06:05:01 | [diff] [blame] | 22 | use core::slice; |
nham | 1cfa656 | 2014-07-27 02:33:47 | [diff] [blame] | 23 | use std::hash::{Writer, Hash}; |
Alex Crichton | 998fece | 2013-05-06 04:42:54 | [diff] [blame] | 24 | |
Alex Crichton | 6a58537 | 2014-05-30 01:50:12 | [diff] [blame] | 25 | use vec::Vec; |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 26 | |
blake2-ppc | 0ff5c17 | 2013-07-06 03:42:45 | [diff] [blame] | 27 | static INITIAL_CAPACITY: uint = 8u; // 2^3 |
| 28 | static MINIMUM_CAPACITY: uint = 2u; |
Daniel Micay | b47e1e9 | 2013-02-16 22:55:55 | [diff] [blame] | 29 | |
P1start | f2aa88c | 2014-08-04 10:48:39 | [diff] [blame] | 30 | /// `RingBuf` is a circular buffer that implements `Deque`. |
blake2-ppc | 10c7698 | 2013-07-06 13:27:32 | [diff] [blame] | 31 | #[deriving(Clone)] |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 32 | pub struct RingBuf<T> { |
Alex Crichton | 8ad7e54 | 2014-03-27 22:10:04 | [diff] [blame] | 33 | nelts: uint, |
| 34 | lo: uint, |
Huon Wilson | 4b9a7a2 | 2014-04-05 05:45:42 | [diff] [blame] | 35 | elts: Vec<Option<T>> |
Marijn Haverbeke | 26610db | 2012-01-11 11:49:33 | [diff] [blame] | 36 | } |
Roy Frostig | 9c81889 | 2010-07-21 01:03:09 | [diff] [blame] | 37 | |
Tom Jakubowski | d6a3941 | 2014-06-09 07:30:04 | [diff] [blame] | 38 | impl<T> Default for RingBuf<T> { |
| 39 | #[inline] |
| 40 | fn default() -> RingBuf<T> { RingBuf::new() } |
| 41 | } |
| 42 | |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 43 | impl<T> RingBuf<T> { |
P1start | f2aa88c | 2014-08-04 10:48:39 | [diff] [blame] | 44 | /// Creates an empty `RingBuf`. |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 45 | pub fn new() -> RingBuf<T> { |
| 46 | RingBuf::with_capacity(INITIAL_CAPACITY) |
| 47 | } |
| 48 | |
P1start | f2aa88c | 2014-08-04 10:48:39 | [diff] [blame] | 49 | /// Creates an empty `RingBuf` with space for at least `n` elements. |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 50 | pub fn with_capacity(n: uint) -> RingBuf<T> { |
| 51 | RingBuf{nelts: 0, lo: 0, |
Huon Wilson | 4b9a7a2 | 2014-04-05 05:45:42 | [diff] [blame] | 52 | elts: Vec::from_fn(cmp::max(MINIMUM_CAPACITY, n), |_| None)} |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 53 | } |
| 54 | |
P1start | f2aa88c | 2014-08-04 10:48:39 | [diff] [blame] | 55 | /// Retrieves an element in the `RingBuf` by index. |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 56 | /// |
P1start | f2aa88c | 2014-08-04 10:48:39 | [diff] [blame] | 57 | /// Fails if there is no element with the given index. |
nham | ebe8097 | 2014-07-17 23:19:51 | [diff] [blame] | 58 | /// |
| 59 | /// # Example |
| 60 | /// |
| 61 | /// ```rust |
Alex Crichton | 1d35662 | 2014-10-23 15:42:21 | [diff] [blame] | 62 | /// # #![allow(deprecated)] |
nham | ebe8097 | 2014-07-17 23:19:51 | [diff] [blame] | 63 | /// use std::collections::RingBuf; |
| 64 | /// |
| 65 | /// let mut buf = RingBuf::new(); |
| 66 | /// buf.push(3i); |
| 67 | /// buf.push(4); |
| 68 | /// buf.push(5); |
| 69 | /// *buf.get_mut(1) = 7; |
P1start | fd10d20 | 2014-08-02 06:39:39 | [diff] [blame] | 70 | /// assert_eq!(buf[1], 7); |
nham | ebe8097 | 2014-07-17 23:19:51 | [diff] [blame] | 71 | /// ``` |
Alex Crichton | 1d35662 | 2014-10-23 15:42:21 | [diff] [blame] | 72 | #[deprecated = "use indexing instead: `buf[index] = value`"] |
Alex Crichton | 21ac985 | 2014-10-30 20:43:24 | [diff] [blame^] | 73 | pub fn get_mut(&mut self, i: uint) -> &mut T { |
Alex Crichton | 1d35662 | 2014-10-23 15:42:21 | [diff] [blame] | 74 | &mut self[i] |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 75 | } |
| 76 | |
P1start | f2aa88c | 2014-08-04 10:48:39 | [diff] [blame] | 77 | /// Swaps elements at indices `i` and `j`. |
blake2-ppc | 57757a8 | 2013-09-26 07:19:26 | [diff] [blame] | 78 | /// |
| 79 | /// `i` and `j` may be equal. |
| 80 | /// |
P1start | f2aa88c | 2014-08-04 10:48:39 | [diff] [blame] | 81 | /// Fails if there is no element with either index. |
nham | ebe8097 | 2014-07-17 23:19:51 | [diff] [blame] | 82 | /// |
| 83 | /// # Example |
| 84 | /// |
| 85 | /// ```rust |
| 86 | /// use std::collections::RingBuf; |
| 87 | /// |
| 88 | /// let mut buf = RingBuf::new(); |
| 89 | /// buf.push(3i); |
| 90 | /// buf.push(4); |
| 91 | /// buf.push(5); |
| 92 | /// buf.swap(0, 2); |
P1start | fd10d20 | 2014-08-02 06:39:39 | [diff] [blame] | 93 | /// assert_eq!(buf[0], 5); |
| 94 | /// assert_eq!(buf[2], 3); |
nham | ebe8097 | 2014-07-17 23:19:51 | [diff] [blame] | 95 | /// ``` |
blake2-ppc | 57757a8 | 2013-09-26 07:19:26 | [diff] [blame] | 96 | pub fn swap(&mut self, i: uint, j: uint) { |
| 97 | assert!(i < self.len()); |
| 98 | assert!(j < self.len()); |
| 99 | let ri = self.raw_index(i); |
| 100 | let rj = self.raw_index(j); |
Huon Wilson | 4b9a7a2 | 2014-04-05 05:45:42 | [diff] [blame] | 101 | self.elts.as_mut_slice().swap(ri, rj); |
blake2-ppc | 57757a8 | 2013-09-26 07:19:26 | [diff] [blame] | 102 | } |
| 103 | |
P1start | f2aa88c | 2014-08-04 10:48:39 | [diff] [blame] | 104 | /// Returns the index in the underlying `Vec` for a given logical element |
| 105 | /// index. |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 106 | fn raw_index(&self, idx: uint) -> uint { |
| 107 | raw_index(self.lo, self.elts.len(), idx) |
| 108 | } |
| 109 | |
P1start | f2aa88c | 2014-08-04 10:48:39 | [diff] [blame] | 110 | /// Reserves capacity for exactly `n` elements in the given `RingBuf`, |
Tim Chevalier | 77de84b | 2013-05-27 18:47:38 | [diff] [blame] | 111 | /// doing nothing if `self`'s capacity is already equal to or greater |
P1start | f2aa88c | 2014-08-04 10:48:39 | [diff] [blame] | 112 | /// than the requested capacity. |
David Manescu | 65f3578 | 2014-01-31 13:03:20 | [diff] [blame] | 113 | pub fn reserve_exact(&mut self, n: uint) { |
| 114 | self.elts.reserve_exact(n); |
Tim Chevalier | 77de84b | 2013-05-27 18:47:38 | [diff] [blame] | 115 | } |
| 116 | |
P1start | f2aa88c | 2014-08-04 10:48:39 | [diff] [blame] | 117 | /// Reserves capacity for at least `n` elements in the given `RingBuf`, |
Tim Chevalier | 77de84b | 2013-05-27 18:47:38 | [diff] [blame] | 118 | /// over-allocating in case the caller needs to reserve additional |
| 119 | /// space. |
| 120 | /// |
| 121 | /// Do nothing if `self`'s capacity is already equal to or greater |
| 122 | /// than the requested capacity. |
David Manescu | 65f3578 | 2014-01-31 13:03:20 | [diff] [blame] | 123 | pub fn reserve(&mut self, n: uint) { |
| 124 | self.elts.reserve(n); |
Tim Chevalier | 77de84b | 2013-05-27 18:47:38 | [diff] [blame] | 125 | } |
Jed Estep | 4f7a742 | 2013-06-25 19:08:47 | [diff] [blame] | 126 | |
P1start | f2aa88c | 2014-08-04 10:48:39 | [diff] [blame] | 127 | /// Returns a front-to-back iterator. |
nham | ebe8097 | 2014-07-17 23:19:51 | [diff] [blame] | 128 | /// |
| 129 | /// # Example |
| 130 | /// |
| 131 | /// ```rust |
| 132 | /// use std::collections::RingBuf; |
| 133 | /// |
| 134 | /// let mut buf = RingBuf::new(); |
| 135 | /// buf.push(5i); |
| 136 | /// buf.push(3); |
| 137 | /// buf.push(4); |
Nick Cameron | 52ef462 | 2014-08-06 09:59:40 | [diff] [blame] | 138 | /// let b: &[_] = &[&5, &3, &4]; |
| 139 | /// assert_eq!(buf.iter().collect::<Vec<&int>>().as_slice(), b); |
nham | ebe8097 | 2014-07-17 23:19:51 | [diff] [blame] | 140 | /// ``` |
Alex Crichton | 21ac985 | 2014-10-30 20:43:24 | [diff] [blame^] | 141 | pub fn iter(&self) -> Items<T> { |
Huon Wilson | 4b9a7a2 | 2014-04-05 05:45:42 | [diff] [blame] | 142 | Items{index: 0, rindex: self.nelts, lo: self.lo, elts: self.elts.as_slice()} |
blake2-ppc | 3385e79 | 2013-07-15 23:13:26 | [diff] [blame] | 143 | } |
| 144 | |
P1start | f2aa88c | 2014-08-04 10:48:39 | [diff] [blame] | 145 | /// Returns a front-to-back iterator which returns mutable references. |
nham | ebe8097 | 2014-07-17 23:19:51 | [diff] [blame] | 146 | /// |
| 147 | /// # Example |
| 148 | /// |
| 149 | /// ```rust |
| 150 | /// use std::collections::RingBuf; |
| 151 | /// |
| 152 | /// let mut buf = RingBuf::new(); |
| 153 | /// buf.push(5i); |
| 154 | /// buf.push(3); |
| 155 | /// buf.push(4); |
Aaron Turon | fc525ee | 2014-09-15 03:27:36 | [diff] [blame] | 156 | /// for num in buf.iter_mut() { |
nham | ebe8097 | 2014-07-17 23:19:51 | [diff] [blame] | 157 | /// *num = *num - 2; |
| 158 | /// } |
Nick Cameron | 52ef462 | 2014-08-06 09:59:40 | [diff] [blame] | 159 | /// let b: &[_] = &[&mut 3, &mut 1, &mut 2]; |
Nick Cameron | 5997694 | 2014-09-24 11:41:09 | [diff] [blame] | 160 | /// assert_eq!(buf.iter_mut().collect::<Vec<&mut int>>()[], b); |
nham | ebe8097 | 2014-07-17 23:19:51 | [diff] [blame] | 161 | /// ``` |
Alex Crichton | 21ac985 | 2014-10-30 20:43:24 | [diff] [blame^] | 162 | pub fn iter_mut(&mut self) -> MutItems<T> { |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 163 | let start_index = raw_index(self.lo, self.elts.len(), 0); |
| 164 | let end_index = raw_index(self.lo, self.elts.len(), self.nelts); |
| 165 | |
| 166 | // Divide up the array |
| 167 | if end_index <= start_index { |
| 168 | // Items to iterate goes from: |
| 169 | // start_index to self.elts.len() |
| 170 | // and then |
| 171 | // 0 to end_index |
Aaron Turon | fc525ee | 2014-09-15 03:27:36 | [diff] [blame] | 172 | let (temp, remaining1) = self.elts.split_at_mut(start_index); |
| 173 | let (remaining2, _) = temp.split_at_mut(end_index); |
Alex Crichton | 9d5d97b | 2014-10-15 06:05:01 | [diff] [blame] | 174 | MutItems { |
| 175 | remaining1: remaining1.iter_mut(), |
| 176 | remaining2: remaining2.iter_mut(), |
| 177 | nelts: self.nelts, |
| 178 | } |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 179 | } else { |
| 180 | // Items to iterate goes from start_index to end_index: |
Aaron Turon | fc525ee | 2014-09-15 03:27:36 | [diff] [blame] | 181 | let (empty, elts) = self.elts.split_at_mut(0); |
Nick Cameron | 5997694 | 2014-09-24 11:41:09 | [diff] [blame] | 182 | let remaining1 = elts[mut start_index..end_index]; |
Alex Crichton | 9d5d97b | 2014-10-15 06:05:01 | [diff] [blame] | 183 | MutItems { |
| 184 | remaining1: remaining1.iter_mut(), |
| 185 | remaining2: empty.iter_mut(), |
| 186 | nelts: self.nelts, |
| 187 | } |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 188 | } |
Jed Estep | 4f7a742 | 2013-06-25 19:08:47 | [diff] [blame] | 189 | } |
Alex Crichton | 21ac985 | 2014-10-30 20:43:24 | [diff] [blame^] | 190 | |
| 191 | /// Returns the number of elements in the `RingBuf`. |
| 192 | /// |
| 193 | /// # Example |
| 194 | /// |
| 195 | /// ``` |
| 196 | /// use std::collections::RingBuf; |
| 197 | /// |
| 198 | /// let mut v = RingBuf::new(); |
| 199 | /// assert_eq!(v.len(), 0); |
| 200 | /// v.push(1i); |
| 201 | /// assert_eq!(v.len(), 1); |
| 202 | /// ``` |
| 203 | pub fn len(&self) -> uint { self.nelts } |
| 204 | |
| 205 | /// Returns true if the buffer contains no elements |
| 206 | /// |
| 207 | /// # Example |
| 208 | /// |
| 209 | /// ``` |
| 210 | /// use std::collections::RingBuf; |
| 211 | /// |
| 212 | /// let mut v = RingBuf::new(); |
| 213 | /// assert!(v.is_empty()); |
| 214 | /// v.push_front(1i); |
| 215 | /// assert!(!v.is_empty()); |
| 216 | /// ``` |
| 217 | pub fn is_empty(&self) -> bool { self.len() == 0 } |
| 218 | |
| 219 | /// Clears the buffer, removing all values. |
| 220 | /// |
| 221 | /// # Example |
| 222 | /// |
| 223 | /// ``` |
| 224 | /// use std::collections::RingBuf; |
| 225 | /// |
| 226 | /// let mut v = RingBuf::new(); |
| 227 | /// v.push(1i); |
| 228 | /// v.clear(); |
| 229 | /// assert!(v.is_empty()); |
| 230 | /// ``` |
| 231 | pub fn clear(&mut self) { |
| 232 | for x in self.elts.iter_mut() { *x = None } |
| 233 | self.nelts = 0; |
| 234 | self.lo = 0; |
| 235 | } |
| 236 | |
| 237 | /// Provides a reference to the front element, or `None` if the sequence is |
| 238 | /// empty. |
| 239 | /// |
| 240 | /// # Example |
| 241 | /// |
| 242 | /// ``` |
| 243 | /// use std::collections::RingBuf; |
| 244 | /// |
| 245 | /// let mut d = RingBuf::new(); |
| 246 | /// assert_eq!(d.front(), None); |
| 247 | /// |
| 248 | /// d.push(1i); |
| 249 | /// d.push(2i); |
| 250 | /// assert_eq!(d.front(), Some(&1i)); |
| 251 | /// ``` |
| 252 | pub fn front(&self) -> Option<&T> { |
| 253 | if self.nelts > 0 { Some(&self[0]) } else { None } |
| 254 | } |
| 255 | |
| 256 | /// Provides a mutable reference to the front element, or `None` if the |
| 257 | /// sequence is empty. |
| 258 | /// |
| 259 | /// # Example |
| 260 | /// |
| 261 | /// ``` |
| 262 | /// use std::collections::RingBuf; |
| 263 | /// |
| 264 | /// let mut d = RingBuf::new(); |
| 265 | /// assert_eq!(d.front_mut(), None); |
| 266 | /// |
| 267 | /// d.push(1i); |
| 268 | /// d.push(2i); |
| 269 | /// match d.front_mut() { |
| 270 | /// Some(x) => *x = 9i, |
| 271 | /// None => (), |
| 272 | /// } |
| 273 | /// assert_eq!(d.front(), Some(&9i)); |
| 274 | /// ``` |
| 275 | pub fn front_mut(&mut self) -> Option<&mut T> { |
| 276 | if self.nelts > 0 { Some(&mut self[0]) } else { None } |
| 277 | } |
| 278 | |
| 279 | /// Provides a reference to the back element, or `None` if the sequence is |
| 280 | /// empty. |
| 281 | /// |
| 282 | /// # Example |
| 283 | /// |
| 284 | /// ``` |
| 285 | /// use std::collections::RingBuf; |
| 286 | /// |
| 287 | /// let mut d = RingBuf::new(); |
| 288 | /// assert_eq!(d.back(), None); |
| 289 | /// |
| 290 | /// d.push(1i); |
| 291 | /// d.push(2i); |
| 292 | /// assert_eq!(d.back(), Some(&2i)); |
| 293 | /// ``` |
| 294 | pub fn back(&self) -> Option<&T> { |
| 295 | if self.nelts > 0 { Some(&self[self.nelts - 1]) } else { None } |
| 296 | } |
| 297 | |
| 298 | /// Provides a mutable reference to the back element, or `None` if the |
| 299 | /// sequence is empty. |
| 300 | /// |
| 301 | /// # Example |
| 302 | /// |
| 303 | /// ``` |
| 304 | /// use std::collections::RingBuf; |
| 305 | /// |
| 306 | /// let mut d = RingBuf::new(); |
| 307 | /// assert_eq!(d.back(), None); |
| 308 | /// |
| 309 | /// d.push(1i); |
| 310 | /// d.push(2i); |
| 311 | /// match d.back_mut() { |
| 312 | /// Some(x) => *x = 9i, |
| 313 | /// None => (), |
| 314 | /// } |
| 315 | /// assert_eq!(d.back(), Some(&9i)); |
| 316 | /// ``` |
| 317 | pub fn back_mut(&mut self) -> Option<&mut T> { |
| 318 | let nelts = self.nelts; |
| 319 | if nelts > 0 { Some(&mut self[nelts - 1]) } else { None } |
| 320 | } |
| 321 | |
| 322 | /// Removes the first element and returns it, or `None` if the sequence is |
| 323 | /// empty. |
| 324 | /// |
| 325 | /// # Example |
| 326 | /// |
| 327 | /// ``` |
| 328 | /// use std::collections::RingBuf; |
| 329 | /// |
| 330 | /// let mut d = RingBuf::new(); |
| 331 | /// d.push(1i); |
| 332 | /// d.push(2i); |
| 333 | /// |
| 334 | /// assert_eq!(d.pop_front(), Some(1i)); |
| 335 | /// assert_eq!(d.pop_front(), Some(2i)); |
| 336 | /// assert_eq!(d.pop_front(), None); |
| 337 | /// ``` |
| 338 | pub fn pop_front(&mut self) -> Option<T> { |
| 339 | let result = self.elts[self.lo].take(); |
| 340 | if result.is_some() { |
| 341 | self.lo = (self.lo + 1u) % self.elts.len(); |
| 342 | self.nelts -= 1u; |
| 343 | } |
| 344 | result |
| 345 | } |
| 346 | |
| 347 | /// Inserts an element first in the sequence. |
| 348 | /// |
| 349 | /// # Example |
| 350 | /// |
| 351 | /// ``` |
| 352 | /// use std::collections::RingBuf; |
| 353 | /// |
| 354 | /// let mut d = RingBuf::new(); |
| 355 | /// d.push_front(1i); |
| 356 | /// d.push_front(2i); |
| 357 | /// assert_eq!(d.front(), Some(&2i)); |
| 358 | /// ``` |
| 359 | pub fn push_front(&mut self, t: T) { |
| 360 | if self.nelts == self.elts.len() { |
| 361 | grow(self.nelts, &mut self.lo, &mut self.elts); |
| 362 | } |
| 363 | if self.lo == 0u { |
| 364 | self.lo = self.elts.len() - 1u; |
| 365 | } else { self.lo -= 1u; } |
| 366 | self.elts[self.lo] = Some(t); |
| 367 | self.nelts += 1u; |
| 368 | } |
| 369 | |
| 370 | /// Appends an element to the back of a buffer |
| 371 | /// |
| 372 | /// # Example |
| 373 | /// |
| 374 | /// ```rust |
| 375 | /// use std::collections::RingBuf; |
| 376 | /// |
| 377 | /// let mut buf = RingBuf::new(); |
| 378 | /// buf.push(1i); |
| 379 | /// buf.push(3); |
| 380 | /// assert_eq!(3, *buf.back().unwrap()); |
| 381 | /// ``` |
| 382 | pub fn push(&mut self, t: T) { |
| 383 | if self.nelts == self.elts.len() { |
| 384 | grow(self.nelts, &mut self.lo, &mut self.elts); |
| 385 | } |
| 386 | let hi = self.raw_index(self.nelts); |
| 387 | self.elts[hi] = Some(t); |
| 388 | self.nelts += 1u; |
| 389 | } |
| 390 | |
| 391 | /// Removes the last element from a buffer and returns it, or `None` if |
| 392 | /// it is empty. |
| 393 | /// |
| 394 | /// # Example |
| 395 | /// |
| 396 | /// ```rust |
| 397 | /// use std::collections::RingBuf; |
| 398 | /// |
| 399 | /// let mut buf = RingBuf::new(); |
| 400 | /// assert_eq!(buf.pop(), None); |
| 401 | /// buf.push(1i); |
| 402 | /// buf.push(3); |
| 403 | /// assert_eq!(buf.pop(), Some(3)); |
| 404 | /// ``` |
| 405 | pub fn pop(&mut self) -> Option<T> { |
| 406 | if self.nelts > 0 { |
| 407 | self.nelts -= 1; |
| 408 | let hi = self.raw_index(self.nelts); |
| 409 | self.elts[hi].take() |
| 410 | } else { |
| 411 | None |
| 412 | } |
| 413 | } |
Jed Estep | 4f7a742 | 2013-06-25 19:08:47 | [diff] [blame] | 414 | } |
| 415 | |
Niko Matsakis | 1b487a8 | 2014-08-28 01:46:52 | [diff] [blame] | 416 | /// `RingBuf` iterator. |
Niko Matsakis | 1b487a8 | 2014-08-28 01:46:52 | [diff] [blame] | 417 | pub struct Items<'a, T:'a> { |
| 418 | lo: uint, |
| 419 | index: uint, |
| 420 | rindex: uint, |
| 421 | elts: &'a [Option<T>], |
| 422 | } |
| 423 | |
Palmer Cox | 3fd8c8b | 2014-01-15 03:32:24 | [diff] [blame] | 424 | impl<'a, T> Iterator<&'a T> for Items<'a, T> { |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 425 | #[inline] |
Erik Price | 5731ca3 | 2013-12-10 07:16:18 | [diff] [blame] | 426 | fn next(&mut self) -> Option<&'a T> { |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 427 | if self.index == self.rindex { |
| 428 | return None; |
| 429 | } |
| 430 | let raw_index = raw_index(self.lo, self.elts.len(), self.index); |
| 431 | self.index += 1; |
Aaron Turon | 276b8b1 | 2014-08-19 00:52:38 | [diff] [blame] | 432 | Some(self.elts[raw_index].as_ref().unwrap()) |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 433 | } |
| 434 | |
| 435 | #[inline] |
| 436 | fn size_hint(&self) -> (uint, Option<uint>) { |
| 437 | let len = self.rindex - self.index; |
| 438 | (len, Some(len)) |
| 439 | } |
| 440 | } |
| 441 | |
Palmer Cox | 3fd8c8b | 2014-01-15 03:32:24 | [diff] [blame] | 442 | impl<'a, T> DoubleEndedIterator<&'a T> for Items<'a, T> { |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 443 | #[inline] |
Erik Price | 5731ca3 | 2013-12-10 07:16:18 | [diff] [blame] | 444 | fn next_back(&mut self) -> Option<&'a T> { |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 445 | if self.index == self.rindex { |
| 446 | return None; |
| 447 | } |
| 448 | self.rindex -= 1; |
| 449 | let raw_index = raw_index(self.lo, self.elts.len(), self.rindex); |
Aaron Turon | 276b8b1 | 2014-08-19 00:52:38 | [diff] [blame] | 450 | Some(self.elts[raw_index].as_ref().unwrap()) |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 451 | } |
| 452 | } |
Jed Estep | 35314c9 | 2013-06-26 15:38:29 | [diff] [blame] | 453 | |
Palmer Cox | 3fd8c8b | 2014-01-15 03:32:24 | [diff] [blame] | 454 | impl<'a, T> ExactSize<&'a T> for Items<'a, T> {} |
blake2-ppc | 7c369ee7 | 2013-09-01 16:20:24 | [diff] [blame] | 455 | |
Palmer Cox | 3fd8c8b | 2014-01-15 03:32:24 | [diff] [blame] | 456 | impl<'a, T> RandomAccessIterator<&'a T> for Items<'a, T> { |
blake2-ppc | f686213 | 2013-07-29 18:16:26 | [diff] [blame] | 457 | #[inline] |
| 458 | fn indexable(&self) -> uint { self.rindex - self.index } |
| 459 | |
| 460 | #[inline] |
Alex Crichton | f4083a2 | 2014-04-22 05:15:42 | [diff] [blame] | 461 | fn idx(&mut self, j: uint) -> Option<&'a T> { |
blake2-ppc | f686213 | 2013-07-29 18:16:26 | [diff] [blame] | 462 | if j >= self.indexable() { |
| 463 | None |
| 464 | } else { |
| 465 | let raw_index = raw_index(self.lo, self.elts.len(), self.index + j); |
Aaron Turon | 276b8b1 | 2014-08-19 00:52:38 | [diff] [blame] | 466 | Some(self.elts[raw_index].as_ref().unwrap()) |
blake2-ppc | f686213 | 2013-07-29 18:16:26 | [diff] [blame] | 467 | } |
| 468 | } |
| 469 | } |
| 470 | |
Niko Matsakis | 1b487a8 | 2014-08-28 01:46:52 | [diff] [blame] | 471 | /// `RingBuf` mutable iterator. |
Niko Matsakis | 1b487a8 | 2014-08-28 01:46:52 | [diff] [blame] | 472 | pub struct MutItems<'a, T:'a> { |
Alex Crichton | 9d5d97b | 2014-10-15 06:05:01 | [diff] [blame] | 473 | remaining1: slice::MutItems<'a, Option<T>>, |
| 474 | remaining2: slice::MutItems<'a, Option<T>>, |
Niko Matsakis | 1b487a8 | 2014-08-28 01:46:52 | [diff] [blame] | 475 | nelts: uint, |
| 476 | } |
| 477 | |
Palmer Cox | 3fd8c8b | 2014-01-15 03:32:24 | [diff] [blame] | 478 | impl<'a, T> Iterator<&'a mut T> for MutItems<'a, T> { |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 479 | #[inline] |
Erik Price | 5731ca3 | 2013-12-10 07:16:18 | [diff] [blame] | 480 | fn next(&mut self) -> Option<&'a mut T> { |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 481 | if self.nelts == 0 { |
| 482 | return None; |
| 483 | } |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 484 | self.nelts -= 1; |
Alex Crichton | 9d5d97b | 2014-10-15 06:05:01 | [diff] [blame] | 485 | match self.remaining1.next() { |
| 486 | Some(ptr) => return Some(ptr.as_mut().unwrap()), |
| 487 | None => {} |
| 488 | } |
| 489 | match self.remaining2.next() { |
| 490 | Some(ptr) => return Some(ptr.as_mut().unwrap()), |
| 491 | None => unreachable!(), |
| 492 | } |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 493 | } |
| 494 | |
| 495 | #[inline] |
| 496 | fn size_hint(&self) -> (uint, Option<uint>) { |
| 497 | (self.nelts, Some(self.nelts)) |
| 498 | } |
| 499 | } |
| 500 | |
Palmer Cox | 3fd8c8b | 2014-01-15 03:32:24 | [diff] [blame] | 501 | impl<'a, T> DoubleEndedIterator<&'a mut T> for MutItems<'a, T> { |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 502 | #[inline] |
Erik Price | 5731ca3 | 2013-12-10 07:16:18 | [diff] [blame] | 503 | fn next_back(&mut self) -> Option<&'a mut T> { |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 504 | if self.nelts == 0 { |
| 505 | return None; |
| 506 | } |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 507 | self.nelts -= 1; |
Alex Crichton | 9d5d97b | 2014-10-15 06:05:01 | [diff] [blame] | 508 | match self.remaining2.next_back() { |
| 509 | Some(ptr) => return Some(ptr.as_mut().unwrap()), |
| 510 | None => {} |
| 511 | } |
| 512 | match self.remaining1.next_back() { |
| 513 | Some(ptr) => return Some(ptr.as_mut().unwrap()), |
| 514 | None => unreachable!(), |
| 515 | } |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 516 | } |
| 517 | } |
Daniel Micay | b47e1e9 | 2013-02-16 22:55:55 | [diff] [blame] | 518 | |
Palmer Cox | 3fd8c8b | 2014-01-15 03:32:24 | [diff] [blame] | 519 | impl<'a, T> ExactSize<&'a mut T> for MutItems<'a, T> {} |
blake2-ppc | 7c369ee7 | 2013-09-01 16:20:24 | [diff] [blame] | 520 | |
Daniel Micay | b47e1e9 | 2013-02-16 22:55:55 | [diff] [blame] | 521 | /// Grow is only called on full elts, so nelts is also len(elts), unlike |
| 522 | /// elsewhere. |
Huon Wilson | 4b9a7a2 | 2014-04-05 05:45:42 | [diff] [blame] | 523 | fn grow<T>(nelts: uint, loptr: &mut uint, elts: &mut Vec<Option<T>>) { |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 524 | assert_eq!(nelts, elts.len()); |
blake2-ppc | 8a32676 | 2013-07-06 03:42:45 | [diff] [blame] | 525 | let lo = *loptr; |
Kevin Butler | 64896d6 | 2014-08-07 01:11:13 | [diff] [blame] | 526 | elts.reserve(nelts * 2); |
| 527 | let newlen = elts.capacity(); |
Daniel Micay | b47e1e9 | 2013-02-16 22:55:55 | [diff] [blame] | 528 | |
blake2-ppc | 40ce0b7 | 2013-07-06 03:42:45 | [diff] [blame] | 529 | /* fill with None */ |
Kevin Butler | 64896d6 | 2014-08-07 01:11:13 | [diff] [blame] | 530 | for _ in range(elts.len(), newlen) { |
blake2-ppc | 40ce0b7 | 2013-07-06 03:42:45 | [diff] [blame] | 531 | elts.push(None); |
Daniel Micay | b47e1e9 | 2013-02-16 22:55:55 | [diff] [blame] | 532 | } |
blake2-ppc | 8a32676 | 2013-07-06 03:42:45 | [diff] [blame] | 533 | |
| 534 | /* |
| 535 | Move the shortest half into the newly reserved area. |
| 536 | lo ---->| |
| 537 | nelts ----------->| |
| 538 | [o o o|o o o o o] |
| 539 | A [. . .|o o o o o o o o|. . . . .] |
| 540 | B [o o o|. . . . . . . .|o o o o o] |
| 541 | */ |
| 542 | |
| 543 | assert!(newlen - nelts/2 >= nelts); |
| 544 | if lo <= (nelts - lo) { // A |
Daniel Micay | 10089455 | 2013-08-03 16:45:23 | [diff] [blame] | 545 | for i in range(0u, lo) { |
Huon Wilson | 4b9a7a2 | 2014-04-05 05:45:42 | [diff] [blame] | 546 | elts.as_mut_slice().swap(i, nelts + i); |
blake2-ppc | 8a32676 | 2013-07-06 03:42:45 | [diff] [blame] | 547 | } |
| 548 | } else { // B |
Daniel Micay | 10089455 | 2013-08-03 16:45:23 | [diff] [blame] | 549 | for i in range(lo, nelts) { |
Huon Wilson | 4b9a7a2 | 2014-04-05 05:45:42 | [diff] [blame] | 550 | elts.as_mut_slice().swap(i, newlen - nelts + i); |
blake2-ppc | 8a32676 | 2013-07-06 03:42:45 | [diff] [blame] | 551 | } |
| 552 | *loptr += newlen - nelts; |
blake2-ppc | 40ce0b7 | 2013-07-06 03:42:45 | [diff] [blame] | 553 | } |
Daniel Micay | b47e1e9 | 2013-02-16 22:55:55 | [diff] [blame] | 554 | } |
| 555 | |
P1start | f2aa88c | 2014-08-04 10:48:39 | [diff] [blame] | 556 | /// Returns the index in the underlying `Vec` for a given logical element index. |
blake2-ppc | 75015c3 | 2013-07-06 03:42:45 | [diff] [blame] | 557 | fn raw_index(lo: uint, len: uint, index: uint) -> uint { |
| 558 | if lo >= len - index { |
| 559 | lo + index - len |
| 560 | } else { |
| 561 | lo + index |
| 562 | } |
| 563 | } |
| 564 | |
Alex Crichton | 748bc3c | 2014-05-30 00:45:07 | [diff] [blame] | 565 | impl<A: PartialEq> PartialEq for RingBuf<A> { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 566 | fn eq(&self, other: &RingBuf<A>) -> bool { |
blake2-ppc | 10c7698 | 2013-07-06 13:27:32 | [diff] [blame] | 567 | self.nelts == other.nelts && |
| 568 | self.iter().zip(other.iter()).all(|(a, b)| a.eq(b)) |
| 569 | } |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 570 | fn ne(&self, other: &RingBuf<A>) -> bool { |
blake2-ppc | 10c7698 | 2013-07-06 13:27:32 | [diff] [blame] | 571 | !self.eq(other) |
| 572 | } |
| 573 | } |
| 574 | |
nham | 25acfde | 2014-08-01 20:05:03 | [diff] [blame] | 575 | impl<A: Eq> Eq for RingBuf<A> {} |
| 576 | |
nham | 6361577 | 2014-07-27 03:18:56 | [diff] [blame] | 577 | impl<A: PartialOrd> PartialOrd for RingBuf<A> { |
| 578 | fn partial_cmp(&self, other: &RingBuf<A>) -> Option<Ordering> { |
| 579 | iter::order::partial_cmp(self.iter(), other.iter()) |
| 580 | } |
| 581 | } |
| 582 | |
nham | 3737c53 | 2014-08-01 20:22:48 | [diff] [blame] | 583 | impl<A: Ord> Ord for RingBuf<A> { |
| 584 | #[inline] |
| 585 | fn cmp(&self, other: &RingBuf<A>) -> Ordering { |
| 586 | iter::order::cmp(self.iter(), other.iter()) |
| 587 | } |
| 588 | } |
| 589 | |
nham | 1cfa656 | 2014-07-27 02:33:47 | [diff] [blame] | 590 | impl<S: Writer, A: Hash<S>> Hash<S> for RingBuf<A> { |
| 591 | fn hash(&self, state: &mut S) { |
nham | 9fa4424 | 2014-07-27 16:37:32 | [diff] [blame] | 592 | self.len().hash(state); |
nham | 1cfa656 | 2014-07-27 02:33:47 | [diff] [blame] | 593 | for elt in self.iter() { |
| 594 | elt.hash(state); |
| 595 | } |
| 596 | } |
| 597 | } |
| 598 | |
P1start | fd10d20 | 2014-08-02 06:39:39 | [diff] [blame] | 599 | impl<A> Index<uint, A> for RingBuf<A> { |
| 600 | #[inline] |
| 601 | fn index<'a>(&'a self, i: &uint) -> &'a A { |
Alex Crichton | 9d5d97b | 2014-10-15 06:05:01 | [diff] [blame] | 602 | let idx = self.raw_index(*i); |
| 603 | match self.elts[idx] { |
Steve Klabnik | 7828c3d | 2014-10-09 19:17:22 | [diff] [blame] | 604 | None => panic!(), |
Alex Crichton | 9d5d97b | 2014-10-15 06:05:01 | [diff] [blame] | 605 | Some(ref v) => v, |
| 606 | } |
P1start | fd10d20 | 2014-08-02 06:39:39 | [diff] [blame] | 607 | } |
| 608 | } |
| 609 | |
Alex Crichton | 1d35662 | 2014-10-23 15:42:21 | [diff] [blame] | 610 | impl<A> IndexMut<uint, A> for RingBuf<A> { |
P1start | fd10d20 | 2014-08-02 06:39:39 | [diff] [blame] | 611 | #[inline] |
Alex Crichton | 1d35662 | 2014-10-23 15:42:21 | [diff] [blame] | 612 | fn index_mut<'a>(&'a mut self, i: &uint) -> &'a mut A { |
| 613 | let idx = self.raw_index(*i); |
| 614 | match *(&mut self.elts[idx]) { |
| 615 | None => panic!(), |
| 616 | Some(ref mut v) => v |
| 617 | } |
P1start | fd10d20 | 2014-08-02 06:39:39 | [diff] [blame] | 618 | } |
Alex Crichton | 1d35662 | 2014-10-23 15:42:21 | [diff] [blame] | 619 | } |
P1start | fd10d20 | 2014-08-02 06:39:39 | [diff] [blame] | 620 | |
Huon Wilson | 53487a0 | 2013-08-13 13:08:14 | [diff] [blame] | 621 | impl<A> FromIterator<A> for RingBuf<A> { |
Brian Anderson | ee05219 | 2014-03-31 04:45:55 | [diff] [blame] | 622 | fn from_iter<T: Iterator<A>>(iterator: T) -> RingBuf<A> { |
blake2-ppc | f8ae526 | 2013-07-30 00:06:49 | [diff] [blame] | 623 | let (lower, _) = iterator.size_hint(); |
| 624 | let mut deq = RingBuf::with_capacity(lower); |
| 625 | deq.extend(iterator); |
blake2-ppc | 08dc72f | 2013-07-06 03:42:45 | [diff] [blame] | 626 | deq |
| 627 | } |
| 628 | } |
| 629 | |
Huon Wilson | 53487a0 | 2013-08-13 13:08:14 | [diff] [blame] | 630 | impl<A> Extendable<A> for RingBuf<A> { |
Marvin Löbel | 6200e76 | 2014-03-20 13:12:56 | [diff] [blame] | 631 | fn extend<T: Iterator<A>>(&mut self, mut iterator: T) { |
| 632 | for elt in iterator { |
Aaron Turon | f77cabe | 2014-08-11 23:47:46 | [diff] [blame] | 633 | self.push(elt); |
blake2-ppc | f8ae526 | 2013-07-30 00:06:49 | [diff] [blame] | 634 | } |
| 635 | } |
| 636 | } |
| 637 | |
Alex Crichton | 6a58537 | 2014-05-30 01:50:12 | [diff] [blame] | 638 | impl<T: fmt::Show> fmt::Show for RingBuf<T> { |
Adolfo Ochagavía | 8e4e3ab | 2014-06-04 14:15:04 | [diff] [blame] | 639 | fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { |
| 640 | try!(write!(f, "[")); |
| 641 | |
| 642 | for (i, e) in self.iter().enumerate() { |
| 643 | if i != 0 { try!(write!(f, ", ")); } |
| 644 | try!(write!(f, "{}", *e)); |
| 645 | } |
| 646 | |
| 647 | write!(f, "]") |
| 648 | } |
| 649 | } |
| 650 | |
Brian Anderson | 6e27b27 | 2012-01-18 03:05:07 | [diff] [blame] | 651 | #[cfg(test)] |
| 652 | mod tests { |
Alex Crichton | 02882fb | 2014-02-28 09:23:06 | [diff] [blame] | 653 | use std::fmt::Show; |
Alex Crichton | 760b93a | 2014-05-30 02:03:06 | [diff] [blame] | 654 | use std::prelude::*; |
nham | 1cfa656 | 2014-07-27 02:33:47 | [diff] [blame] | 655 | use std::hash; |
Alex Crichton | 760b93a | 2014-05-30 02:03:06 | [diff] [blame] | 656 | use test::Bencher; |
| 657 | use test; |
| 658 | |
Alex Crichton | f47e4b2 | 2014-01-07 06:33:50 | [diff] [blame] | 659 | use super::RingBuf; |
Alex Crichton | 760b93a | 2014-05-30 02:03:06 | [diff] [blame] | 660 | use vec::Vec; |
Patrick Walton | fa5ee93 | 2012-12-28 02:24:18 | [diff] [blame] | 661 | |
Brian Anderson | 6e27b27 | 2012-01-18 03:05:07 | [diff] [blame] | 662 | #[test] |
Victor Berger | 52ea83d | 2014-09-22 17:30:06 | [diff] [blame] | 663 | #[allow(deprecated)] |
Brian Anderson | 6e27b27 | 2012-01-18 03:05:07 | [diff] [blame] | 664 | fn test_simple() { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 665 | let mut d = RingBuf::new(); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 666 | assert_eq!(d.len(), 0u); |
Niko Matsakis | 9e3d0b0 | 2014-04-21 21:58:52 | [diff] [blame] | 667 | d.push_front(17i); |
| 668 | d.push_front(42i); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 669 | d.push(137); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 670 | assert_eq!(d.len(), 3u); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 671 | d.push(137); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 672 | assert_eq!(d.len(), 4u); |
Luqman Aden | 3ef9aa0 | 2014-10-15 07:22:55 | [diff] [blame] | 673 | debug!("{}", d.front()); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 674 | assert_eq!(*d.front().unwrap(), 42); |
Luqman Aden | 3ef9aa0 | 2014-10-15 07:22:55 | [diff] [blame] | 675 | debug!("{}", d.back()); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 676 | assert_eq!(*d.back().unwrap(), 137); |
| 677 | let mut i = d.pop_front(); |
Luqman Aden | 3ef9aa0 | 2014-10-15 07:22:55 | [diff] [blame] | 678 | debug!("{}", i); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 679 | assert_eq!(i, Some(42)); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 680 | i = d.pop(); |
Luqman Aden | 3ef9aa0 | 2014-10-15 07:22:55 | [diff] [blame] | 681 | debug!("{}", i); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 682 | assert_eq!(i, Some(137)); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 683 | i = d.pop(); |
Luqman Aden | 3ef9aa0 | 2014-10-15 07:22:55 | [diff] [blame] | 684 | debug!("{}", i); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 685 | assert_eq!(i, Some(137)); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 686 | i = d.pop(); |
Luqman Aden | 3ef9aa0 | 2014-10-15 07:22:55 | [diff] [blame] | 687 | debug!("{}", i); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 688 | assert_eq!(i, Some(17)); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 689 | assert_eq!(d.len(), 0u); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 690 | d.push(3); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 691 | assert_eq!(d.len(), 1u); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 692 | d.push_front(2); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 693 | assert_eq!(d.len(), 2u); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 694 | d.push(4); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 695 | assert_eq!(d.len(), 3u); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 696 | d.push_front(1); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 697 | assert_eq!(d.len(), 4u); |
Alex Crichton | 9d5d97b | 2014-10-15 06:05:01 | [diff] [blame] | 698 | debug!("{}", d[0]); |
| 699 | debug!("{}", d[1]); |
| 700 | debug!("{}", d[2]); |
| 701 | debug!("{}", d[3]); |
| 702 | assert_eq!(d[0], 1); |
| 703 | assert_eq!(d[1], 2); |
| 704 | assert_eq!(d[2], 3); |
| 705 | assert_eq!(d[3], 4); |
Brian Anderson | 6e27b27 | 2012-01-18 03:05:07 | [diff] [blame] | 706 | } |
| 707 | |
Felix S. Klock II | a636f51 | 2013-05-01 23:32:37 | [diff] [blame] | 708 | #[cfg(test)] |
Alex Crichton | 748bc3c | 2014-05-30 00:45:07 | [diff] [blame] | 709 | fn test_parameterized<T:Clone + PartialEq + Show>(a: T, b: T, c: T, d: T) { |
Patrick Walton | dc4bf17 | 2013-07-13 04:05:59 | [diff] [blame] | 710 | let mut deq = RingBuf::new(); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 711 | assert_eq!(deq.len(), 0); |
Patrick Walton | dc4bf17 | 2013-07-13 04:05:59 | [diff] [blame] | 712 | deq.push_front(a.clone()); |
| 713 | deq.push_front(b.clone()); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 714 | deq.push(c.clone()); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 715 | assert_eq!(deq.len(), 3); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 716 | deq.push(d.clone()); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 717 | assert_eq!(deq.len(), 4); |
Marvin Löbel | 0ac7a21 | 2013-08-03 23:59:24 | [diff] [blame] | 718 | assert_eq!((*deq.front().unwrap()).clone(), b.clone()); |
| 719 | assert_eq!((*deq.back().unwrap()).clone(), d.clone()); |
| 720 | assert_eq!(deq.pop_front().unwrap(), b.clone()); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 721 | assert_eq!(deq.pop().unwrap(), d.clone()); |
| 722 | assert_eq!(deq.pop().unwrap(), c.clone()); |
| 723 | assert_eq!(deq.pop().unwrap(), a.clone()); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 724 | assert_eq!(deq.len(), 0); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 725 | deq.push(c.clone()); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 726 | assert_eq!(deq.len(), 1); |
Patrick Walton | dc4bf17 | 2013-07-13 04:05:59 | [diff] [blame] | 727 | deq.push_front(b.clone()); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 728 | assert_eq!(deq.len(), 2); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 729 | deq.push(d.clone()); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 730 | assert_eq!(deq.len(), 3); |
Patrick Walton | dc4bf17 | 2013-07-13 04:05:59 | [diff] [blame] | 731 | deq.push_front(a.clone()); |
Corey Richardson | cc57ca0 | 2013-05-19 02:02:45 | [diff] [blame] | 732 | assert_eq!(deq.len(), 4); |
NODA, Kai | f27ad3d | 2014-10-05 10:11:17 | [diff] [blame] | 733 | assert_eq!(deq[0].clone(), a.clone()); |
| 734 | assert_eq!(deq[1].clone(), b.clone()); |
| 735 | assert_eq!(deq[2].clone(), c.clone()); |
| 736 | assert_eq!(deq[3].clone(), d.clone()); |
Brian Anderson | 6e27b27 | 2012-01-18 03:05:07 | [diff] [blame] | 737 | } |
| 738 | |
blake2-ppc | 81933ed | 2013-07-06 03:42:45 | [diff] [blame] | 739 | #[test] |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 740 | fn test_push_front_grow() { |
| 741 | let mut deq = RingBuf::new(); |
Daniel Micay | 10089455 | 2013-08-03 16:45:23 | [diff] [blame] | 742 | for i in range(0u, 66) { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 743 | deq.push_front(i); |
blake2-ppc | 81933ed | 2013-07-06 03:42:45 | [diff] [blame] | 744 | } |
| 745 | assert_eq!(deq.len(), 66); |
| 746 | |
Daniel Micay | 10089455 | 2013-08-03 16:45:23 | [diff] [blame] | 747 | for i in range(0u, 66) { |
NODA, Kai | f27ad3d | 2014-10-05 10:11:17 | [diff] [blame] | 748 | assert_eq!(deq[i], 65 - i); |
blake2-ppc | 81933ed | 2013-07-06 03:42:45 | [diff] [blame] | 749 | } |
| 750 | |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 751 | let mut deq = RingBuf::new(); |
Daniel Micay | 10089455 | 2013-08-03 16:45:23 | [diff] [blame] | 752 | for i in range(0u, 66) { |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 753 | deq.push(i); |
blake2-ppc | 81933ed | 2013-07-06 03:42:45 | [diff] [blame] | 754 | } |
| 755 | |
Daniel Micay | 10089455 | 2013-08-03 16:45:23 | [diff] [blame] | 756 | for i in range(0u, 66) { |
NODA, Kai | f27ad3d | 2014-10-05 10:11:17 | [diff] [blame] | 757 | assert_eq!(deq[i], i); |
blake2-ppc | 81933ed | 2013-07-06 03:42:45 | [diff] [blame] | 758 | } |
| 759 | } |
| 760 | |
P1start | fd10d20 | 2014-08-02 06:39:39 | [diff] [blame] | 761 | #[test] |
| 762 | fn test_index() { |
| 763 | let mut deq = RingBuf::new(); |
| 764 | for i in range(1u, 4) { |
| 765 | deq.push_front(i); |
| 766 | } |
| 767 | assert_eq!(deq[1], 2); |
| 768 | } |
| 769 | |
| 770 | #[test] |
| 771 | #[should_fail] |
| 772 | fn test_index_out_of_bounds() { |
| 773 | let mut deq = RingBuf::new(); |
| 774 | for i in range(1u, 4) { |
| 775 | deq.push_front(i); |
| 776 | } |
| 777 | deq[3]; |
| 778 | } |
| 779 | |
blake2-ppc | 81933ed | 2013-07-06 03:42:45 | [diff] [blame] | 780 | #[bench] |
Liigo Zhuang | 408f484 | 2014-04-01 01:16:35 | [diff] [blame] | 781 | fn bench_new(b: &mut test::Bencher) { |
Patrick Walton | 38efa17 | 2013-11-22 03:20:48 | [diff] [blame] | 782 | b.iter(|| { |
Patrick Walton | 8693943 | 2013-08-08 18:38:10 | [diff] [blame] | 783 | let _: RingBuf<u64> = RingBuf::new(); |
Patrick Walton | 38efa17 | 2013-11-22 03:20:48 | [diff] [blame] | 784 | }) |
blake2-ppc | 81933ed | 2013-07-06 03:42:45 | [diff] [blame] | 785 | } |
| 786 | |
| 787 | #[bench] |
Liigo Zhuang | 408f484 | 2014-04-01 01:16:35 | [diff] [blame] | 788 | fn bench_push_back(b: &mut test::Bencher) { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 789 | let mut deq = RingBuf::new(); |
Patrick Walton | 38efa17 | 2013-11-22 03:20:48 | [diff] [blame] | 790 | b.iter(|| { |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 791 | deq.push(0i); |
Patrick Walton | 38efa17 | 2013-11-22 03:20:48 | [diff] [blame] | 792 | }) |
blake2-ppc | 81933ed | 2013-07-06 03:42:45 | [diff] [blame] | 793 | } |
| 794 | |
| 795 | #[bench] |
Liigo Zhuang | 408f484 | 2014-04-01 01:16:35 | [diff] [blame] | 796 | fn bench_push_front(b: &mut test::Bencher) { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 797 | let mut deq = RingBuf::new(); |
Patrick Walton | 38efa17 | 2013-11-22 03:20:48 | [diff] [blame] | 798 | b.iter(|| { |
Patrick Walton | a5bb0a3 | 2014-06-27 19:30:25 | [diff] [blame] | 799 | deq.push_front(0i); |
Patrick Walton | 38efa17 | 2013-11-22 03:20:48 | [diff] [blame] | 800 | }) |
blake2-ppc | 81933ed | 2013-07-06 03:42:45 | [diff] [blame] | 801 | } |
| 802 | |
| 803 | #[bench] |
Liigo Zhuang | 408f484 | 2014-04-01 01:16:35 | [diff] [blame] | 804 | fn bench_grow(b: &mut test::Bencher) { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 805 | let mut deq = RingBuf::new(); |
Patrick Walton | 38efa17 | 2013-11-22 03:20:48 | [diff] [blame] | 806 | b.iter(|| { |
Niko Matsakis | 9e3d0b0 | 2014-04-21 21:58:52 | [diff] [blame] | 807 | for _ in range(0i, 65) { |
Patrick Walton | a5bb0a3 | 2014-06-27 19:30:25 | [diff] [blame] | 808 | deq.push_front(1i); |
Brendan Zabarauskas | 729060d | 2014-01-30 00:20:34 | [diff] [blame] | 809 | } |
Patrick Walton | 38efa17 | 2013-11-22 03:20:48 | [diff] [blame] | 810 | }) |
blake2-ppc | 81933ed | 2013-07-06 03:42:45 | [diff] [blame] | 811 | } |
| 812 | |
Alex Crichton | 748bc3c | 2014-05-30 00:45:07 | [diff] [blame] | 813 | #[deriving(Clone, PartialEq, Show)] |
Patrick Walton | 99b33f7 | 2013-07-02 19:47:32 | [diff] [blame] | 814 | enum Taggy { |
| 815 | One(int), |
| 816 | Two(int, int), |
| 817 | Three(int, int, int), |
Brian Anderson | 6e27b27 | 2012-01-18 03:05:07 | [diff] [blame] | 818 | } |
| 819 | |
Alex Crichton | 748bc3c | 2014-05-30 00:45:07 | [diff] [blame] | 820 | #[deriving(Clone, PartialEq, Show)] |
Patrick Walton | 99b33f7 | 2013-07-02 19:47:32 | [diff] [blame] | 821 | enum Taggypar<T> { |
| 822 | Onepar(int), |
| 823 | Twopar(int, int), |
| 824 | Threepar(int, int, int), |
| 825 | } |
| 826 | |
Alex Crichton | 748bc3c | 2014-05-30 00:45:07 | [diff] [blame] | 827 | #[deriving(Clone, PartialEq, Show)] |
Erick Tryzelaar | e84576b | 2013-01-22 16:44:24 | [diff] [blame] | 828 | struct RecCy { |
| 829 | x: int, |
| 830 | y: int, |
Patrick Walton | eb4d39e | 2013-01-26 00:57:39 | [diff] [blame] | 831 | t: Taggy |
Patrick Walton | 9117dcb | 2012-09-20 01:00:26 | [diff] [blame] | 832 | } |
Kevin Cantu | c43426e | 2012-09-13 05:09:55 | [diff] [blame] | 833 | |
| 834 | #[test] |
| 835 | fn test_param_int() { |
| 836 | test_parameterized::<int>(5, 72, 64, 175); |
| 837 | } |
| 838 | |
| 839 | #[test] |
Kevin Cantu | c43426e | 2012-09-13 05:09:55 | [diff] [blame] | 840 | fn test_param_taggy() { |
Corey Richardson | f8ae9b0 | 2013-06-26 22:14:35 | [diff] [blame] | 841 | test_parameterized::<Taggy>(One(1), Two(1, 2), Three(1, 2, 3), Two(17, 42)); |
Kevin Cantu | c43426e | 2012-09-13 05:09:55 | [diff] [blame] | 842 | } |
| 843 | |
| 844 | #[test] |
| 845 | fn test_param_taggypar() { |
| 846 | test_parameterized::<Taggypar<int>>(Onepar::<int>(1), |
Ben Striegel | a605fd0 | 2012-08-11 14:08:42 | [diff] [blame] | 847 | Twopar::<int>(1, 2), |
| 848 | Threepar::<int>(1, 2, 3), |
| 849 | Twopar::<int>(17, 42)); |
Kevin Cantu | c43426e | 2012-09-13 05:09:55 | [diff] [blame] | 850 | } |
Brian Anderson | 6e27b27 | 2012-01-18 03:05:07 | [diff] [blame] | 851 | |
Kevin Cantu | c43426e | 2012-09-13 05:09:55 | [diff] [blame] | 852 | #[test] |
| 853 | fn test_param_reccy() { |
Erick Tryzelaar | e84576b | 2013-01-22 16:44:24 | [diff] [blame] | 854 | let reccy1 = RecCy { x: 1, y: 2, t: One(1) }; |
| 855 | let reccy2 = RecCy { x: 345, y: 2, t: Two(1, 2) }; |
| 856 | let reccy3 = RecCy { x: 1, y: 777, t: Three(1, 2, 3) }; |
| 857 | let reccy4 = RecCy { x: 19, y: 252, t: Two(17, 42) }; |
Kevin Cantu | c43426e | 2012-09-13 05:09:55 | [diff] [blame] | 858 | test_parameterized::<RecCy>(reccy1, reccy2, reccy3, reccy4); |
Brian Anderson | 6e27b27 | 2012-01-18 03:05:07 | [diff] [blame] | 859 | } |
Erick Tryzelaar | 909d8f0 | 2013-03-30 01:02:44 | [diff] [blame] | 860 | |
| 861 | #[test] |
blake2-ppc | 0ff5c17 | 2013-07-06 03:42:45 | [diff] [blame] | 862 | fn test_with_capacity() { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 863 | let mut d = RingBuf::with_capacity(0); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 864 | d.push(1i); |
blake2-ppc | 0ff5c17 | 2013-07-06 03:42:45 | [diff] [blame] | 865 | assert_eq!(d.len(), 1); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 866 | let mut d = RingBuf::with_capacity(50); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 867 | d.push(1i); |
blake2-ppc | 0ff5c17 | 2013-07-06 03:42:45 | [diff] [blame] | 868 | assert_eq!(d.len(), 1); |
| 869 | } |
| 870 | |
| 871 | #[test] |
Kevin Butler | 64896d6 | 2014-08-07 01:11:13 | [diff] [blame] | 872 | fn test_with_capacity_non_power_two() { |
| 873 | let mut d3 = RingBuf::with_capacity(3); |
| 874 | d3.push(1i); |
| 875 | |
| 876 | // X = None, | = lo |
| 877 | // [|1, X, X] |
| 878 | assert_eq!(d3.pop_front(), Some(1)); |
| 879 | // [X, |X, X] |
| 880 | assert_eq!(d3.front(), None); |
| 881 | |
| 882 | // [X, |3, X] |
| 883 | d3.push(3); |
| 884 | // [X, |3, 6] |
| 885 | d3.push(6); |
| 886 | // [X, X, |6] |
| 887 | assert_eq!(d3.pop_front(), Some(3)); |
| 888 | |
| 889 | // Pushing the lo past half way point to trigger |
| 890 | // the 'B' scenario for growth |
| 891 | // [9, X, |6] |
| 892 | d3.push(9); |
| 893 | // [9, 12, |6] |
| 894 | d3.push(12); |
| 895 | |
| 896 | d3.push(15); |
| 897 | // There used to be a bug here about how the |
| 898 | // RingBuf made growth assumptions about the |
| 899 | // underlying Vec which didn't hold and lead |
| 900 | // to corruption. |
| 901 | // (Vec grows to next power of two) |
| 902 | //good- [9, 12, 15, X, X, X, X, |6] |
| 903 | //bug- [15, 12, X, X, X, |6, X, X] |
| 904 | assert_eq!(d3.pop_front(), Some(6)); |
| 905 | |
| 906 | // Which leads us to the following state which |
| 907 | // would be a failure case. |
| 908 | //bug- [15, 12, X, X, X, X, |X, X] |
| 909 | assert_eq!(d3.front(), Some(&9)); |
| 910 | } |
| 911 | |
| 912 | #[test] |
David Manescu | 65f3578 | 2014-01-31 13:03:20 | [diff] [blame] | 913 | fn test_reserve_exact() { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 914 | let mut d = RingBuf::new(); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 915 | d.push(0u64); |
David Manescu | 65f3578 | 2014-01-31 13:03:20 | [diff] [blame] | 916 | d.reserve_exact(50); |
Huon Wilson | 32d6559 | 2013-06-27 14:40:47 | [diff] [blame] | 917 | assert_eq!(d.elts.capacity(), 50); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 918 | let mut d = RingBuf::new(); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 919 | d.push(0u32); |
David Manescu | 65f3578 | 2014-01-31 13:03:20 | [diff] [blame] | 920 | d.reserve_exact(50); |
Huon Wilson | 32d6559 | 2013-06-27 14:40:47 | [diff] [blame] | 921 | assert_eq!(d.elts.capacity(), 50); |
Tim Chevalier | 77de84b | 2013-05-27 18:47:38 | [diff] [blame] | 922 | } |
| 923 | |
| 924 | #[test] |
David Manescu | 65f3578 | 2014-01-31 13:03:20 | [diff] [blame] | 925 | fn test_reserve() { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 926 | let mut d = RingBuf::new(); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 927 | d.push(0u64); |
David Manescu | 65f3578 | 2014-01-31 13:03:20 | [diff] [blame] | 928 | d.reserve(50); |
Huon Wilson | 32d6559 | 2013-06-27 14:40:47 | [diff] [blame] | 929 | assert_eq!(d.elts.capacity(), 64); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 930 | let mut d = RingBuf::new(); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 931 | d.push(0u32); |
David Manescu | 65f3578 | 2014-01-31 13:03:20 | [diff] [blame] | 932 | d.reserve(50); |
Huon Wilson | 32d6559 | 2013-06-27 14:40:47 | [diff] [blame] | 933 | assert_eq!(d.elts.capacity(), 64); |
Tim Chevalier | 77de84b | 2013-05-27 18:47:38 | [diff] [blame] | 934 | } |
| 935 | |
Jed Estep | 096fb79 | 2013-06-26 14:04:44 | [diff] [blame] | 936 | #[test] |
blake2-ppc | 57757a8 | 2013-09-26 07:19:26 | [diff] [blame] | 937 | fn test_swap() { |
Niko Matsakis | 9e3d0b0 | 2014-04-21 21:58:52 | [diff] [blame] | 938 | let mut d: RingBuf<int> = range(0i, 5).collect(); |
blake2-ppc | 57757a8 | 2013-09-26 07:19:26 | [diff] [blame] | 939 | d.pop_front(); |
| 940 | d.swap(0, 3); |
Huon Wilson | 4b9a7a2 | 2014-04-05 05:45:42 | [diff] [blame] | 941 | assert_eq!(d.iter().map(|&x|x).collect::<Vec<int>>(), vec!(4, 2, 3, 1)); |
blake2-ppc | 57757a8 | 2013-09-26 07:19:26 | [diff] [blame] | 942 | } |
| 943 | |
| 944 | #[test] |
Jed Estep | 096fb79 | 2013-06-26 14:04:44 | [diff] [blame] | 945 | fn test_iter() { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 946 | let mut d = RingBuf::new(); |
blake2-ppc | f88d532 | 2013-07-06 03:42:45 | [diff] [blame] | 947 | assert_eq!(d.iter().next(), None); |
blake2-ppc | 9ccf443 | 2013-07-14 20:30:22 | [diff] [blame] | 948 | assert_eq!(d.iter().size_hint(), (0, Some(0))); |
blake2-ppc | f88d532 | 2013-07-06 03:42:45 | [diff] [blame] | 949 | |
Niko Matsakis | 9e3d0b0 | 2014-04-21 21:58:52 | [diff] [blame] | 950 | for i in range(0i, 5) { |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 951 | d.push(i); |
Jed Estep | 096fb79 | 2013-06-26 14:04:44 | [diff] [blame] | 952 | } |
Nick Cameron | 37a94b8 | 2014-08-04 12:19:02 | [diff] [blame] | 953 | { |
| 954 | let b: &[_] = &[&0,&1,&2,&3,&4]; |
| 955 | assert_eq!(d.iter().collect::<Vec<&int>>().as_slice(), b); |
| 956 | } |
Corey Richardson | f8ae9b0 | 2013-06-26 22:14:35 | [diff] [blame] | 957 | |
Niko Matsakis | 9e3d0b0 | 2014-04-21 21:58:52 | [diff] [blame] | 958 | for i in range(6i, 9) { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 959 | d.push_front(i); |
Jed Estep | 096fb79 | 2013-06-26 14:04:44 | [diff] [blame] | 960 | } |
Nick Cameron | 37a94b8 | 2014-08-04 12:19:02 | [diff] [blame] | 961 | { |
| 962 | let b: &[_] = &[&8,&7,&6,&0,&1,&2,&3,&4]; |
| 963 | assert_eq!(d.iter().collect::<Vec<&int>>().as_slice(), b); |
| 964 | } |
blake2-ppc | 9ccf443 | 2013-07-14 20:30:22 | [diff] [blame] | 965 | |
| 966 | let mut it = d.iter(); |
| 967 | let mut len = d.len(); |
| 968 | loop { |
| 969 | match it.next() { |
| 970 | None => break, |
| 971 | _ => { len -= 1; assert_eq!(it.size_hint(), (len, Some(len))) } |
| 972 | } |
| 973 | } |
Jed Estep | 096fb79 | 2013-06-26 14:04:44 | [diff] [blame] | 974 | } |
| 975 | |
| 976 | #[test] |
| 977 | fn test_rev_iter() { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 978 | let mut d = RingBuf::new(); |
Jonathan S | 03609e5 | 2014-04-21 04:59:12 | [diff] [blame] | 979 | assert_eq!(d.iter().rev().next(), None); |
blake2-ppc | f88d532 | 2013-07-06 03:42:45 | [diff] [blame] | 980 | |
Niko Matsakis | 9e3d0b0 | 2014-04-21 21:58:52 | [diff] [blame] | 981 | for i in range(0i, 5) { |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 982 | d.push(i); |
Jed Estep | 096fb79 | 2013-06-26 14:04:44 | [diff] [blame] | 983 | } |
Nick Cameron | 37a94b8 | 2014-08-04 12:19:02 | [diff] [blame] | 984 | { |
| 985 | let b: &[_] = &[&4,&3,&2,&1,&0]; |
| 986 | assert_eq!(d.iter().rev().collect::<Vec<&int>>().as_slice(), b); |
| 987 | } |
Corey Richardson | f8ae9b0 | 2013-06-26 22:14:35 | [diff] [blame] | 988 | |
Niko Matsakis | 9e3d0b0 | 2014-04-21 21:58:52 | [diff] [blame] | 989 | for i in range(6i, 9) { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 990 | d.push_front(i); |
Jed Estep | 096fb79 | 2013-06-26 14:04:44 | [diff] [blame] | 991 | } |
Nick Cameron | 37a94b8 | 2014-08-04 12:19:02 | [diff] [blame] | 992 | let b: &[_] = &[&4,&3,&2,&1,&0,&6,&7,&8]; |
| 993 | assert_eq!(d.iter().rev().collect::<Vec<&int>>().as_slice(), b); |
Jed Estep | 096fb79 | 2013-06-26 14:04:44 | [diff] [blame] | 994 | } |
blake2-ppc | 08dc72f | 2013-07-06 03:42:45 | [diff] [blame] | 995 | |
| 996 | #[test] |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 997 | fn test_mut_rev_iter_wrap() { |
| 998 | let mut d = RingBuf::with_capacity(3); |
Aaron Turon | fc525ee | 2014-09-15 03:27:36 | [diff] [blame] | 999 | assert!(d.iter_mut().rev().next().is_none()); |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 1000 | |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 1001 | d.push(1i); |
| 1002 | d.push(2); |
| 1003 | d.push(3); |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 1004 | assert_eq!(d.pop_front(), Some(1)); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 1005 | d.push(4); |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 1006 | |
Aaron Turon | fc525ee | 2014-09-15 03:27:36 | [diff] [blame] | 1007 | assert_eq!(d.iter_mut().rev().map(|x| *x).collect::<Vec<int>>(), |
Huon Wilson | 4b9a7a2 | 2014-04-05 05:45:42 | [diff] [blame] | 1008 | vec!(4, 3, 2)); |
Niko Matsakis | bc4164d | 2013-11-16 22:29:39 | [diff] [blame] | 1009 | } |
| 1010 | |
| 1011 | #[test] |
blake2-ppc | f88d532 | 2013-07-06 03:42:45 | [diff] [blame] | 1012 | fn test_mut_iter() { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 1013 | let mut d = RingBuf::new(); |
Aaron Turon | fc525ee | 2014-09-15 03:27:36 | [diff] [blame] | 1014 | assert!(d.iter_mut().next().is_none()); |
blake2-ppc | f88d532 | 2013-07-06 03:42:45 | [diff] [blame] | 1015 | |
Daniel Micay | 10089455 | 2013-08-03 16:45:23 | [diff] [blame] | 1016 | for i in range(0u, 3) { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 1017 | d.push_front(i); |
blake2-ppc | f88d532 | 2013-07-06 03:42:45 | [diff] [blame] | 1018 | } |
| 1019 | |
Aaron Turon | fc525ee | 2014-09-15 03:27:36 | [diff] [blame] | 1020 | for (i, elt) in d.iter_mut().enumerate() { |
blake2-ppc | f88d532 | 2013-07-06 03:42:45 | [diff] [blame] | 1021 | assert_eq!(*elt, 2 - i); |
| 1022 | *elt = i; |
| 1023 | } |
| 1024 | |
| 1025 | { |
Aaron Turon | fc525ee | 2014-09-15 03:27:36 | [diff] [blame] | 1026 | let mut it = d.iter_mut(); |
blake2-ppc | f88d532 | 2013-07-06 03:42:45 | [diff] [blame] | 1027 | assert_eq!(*it.next().unwrap(), 0); |
| 1028 | assert_eq!(*it.next().unwrap(), 1); |
| 1029 | assert_eq!(*it.next().unwrap(), 2); |
| 1030 | assert!(it.next().is_none()); |
| 1031 | } |
| 1032 | } |
| 1033 | |
| 1034 | #[test] |
| 1035 | fn test_mut_rev_iter() { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 1036 | let mut d = RingBuf::new(); |
Aaron Turon | fc525ee | 2014-09-15 03:27:36 | [diff] [blame] | 1037 | assert!(d.iter_mut().rev().next().is_none()); |
blake2-ppc | f88d532 | 2013-07-06 03:42:45 | [diff] [blame] | 1038 | |
Daniel Micay | 10089455 | 2013-08-03 16:45:23 | [diff] [blame] | 1039 | for i in range(0u, 3) { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 1040 | d.push_front(i); |
blake2-ppc | f88d532 | 2013-07-06 03:42:45 | [diff] [blame] | 1041 | } |
| 1042 | |
Aaron Turon | fc525ee | 2014-09-15 03:27:36 | [diff] [blame] | 1043 | for (i, elt) in d.iter_mut().rev().enumerate() { |
blake2-ppc | f88d532 | 2013-07-06 03:42:45 | [diff] [blame] | 1044 | assert_eq!(*elt, i); |
| 1045 | *elt = i; |
| 1046 | } |
| 1047 | |
| 1048 | { |
Aaron Turon | fc525ee | 2014-09-15 03:27:36 | [diff] [blame] | 1049 | let mut it = d.iter_mut().rev(); |
blake2-ppc | f88d532 | 2013-07-06 03:42:45 | [diff] [blame] | 1050 | assert_eq!(*it.next().unwrap(), 0); |
| 1051 | assert_eq!(*it.next().unwrap(), 1); |
| 1052 | assert_eq!(*it.next().unwrap(), 2); |
| 1053 | assert!(it.next().is_none()); |
| 1054 | } |
| 1055 | } |
| 1056 | |
| 1057 | #[test] |
Brian Anderson | ee05219 | 2014-03-31 04:45:55 | [diff] [blame] | 1058 | fn test_from_iter() { |
Daniel Micay | 6919cf5 | 2013-09-08 15:01:16 | [diff] [blame] | 1059 | use std::iter; |
Niko Matsakis | 9e3d0b0 | 2014-04-21 21:58:52 | [diff] [blame] | 1060 | let v = vec!(1i,2,3,4,5,6,7); |
Erick Tryzelaar | 68f40d2 | 2013-08-10 03:09:47 | [diff] [blame] | 1061 | let deq: RingBuf<int> = v.iter().map(|&x| x).collect(); |
Huon Wilson | 4b9a7a2 | 2014-04-05 05:45:42 | [diff] [blame] | 1062 | let u: Vec<int> = deq.iter().map(|&x| x).collect(); |
blake2-ppc | 08dc72f | 2013-07-06 03:42:45 | [diff] [blame] | 1063 | assert_eq!(u, v); |
| 1064 | |
Daniel Micay | 6919cf5 | 2013-09-08 15:01:16 | [diff] [blame] | 1065 | let mut seq = iter::count(0u, 2).take(256); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 1066 | let deq: RingBuf<uint> = seq.collect(); |
Daniel Micay | 10089455 | 2013-08-03 16:45:23 | [diff] [blame] | 1067 | for (i, &x) in deq.iter().enumerate() { |
blake2-ppc | 08dc72f | 2013-07-06 03:42:45 | [diff] [blame] | 1068 | assert_eq!(2*i, x); |
| 1069 | } |
| 1070 | assert_eq!(deq.len(), 256); |
| 1071 | } |
blake2-ppc | 10c7698 | 2013-07-06 13:27:32 | [diff] [blame] | 1072 | |
| 1073 | #[test] |
| 1074 | fn test_clone() { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 1075 | let mut d = RingBuf::new(); |
Niko Matsakis | 9e3d0b0 | 2014-04-21 21:58:52 | [diff] [blame] | 1076 | d.push_front(17i); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 1077 | d.push_front(42); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 1078 | d.push(137); |
| 1079 | d.push(137); |
blake2-ppc | 10c7698 | 2013-07-06 13:27:32 | [diff] [blame] | 1080 | assert_eq!(d.len(), 4u); |
| 1081 | let mut e = d.clone(); |
| 1082 | assert_eq!(e.len(), 4u); |
| 1083 | while !d.is_empty() { |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 1084 | assert_eq!(d.pop(), e.pop()); |
blake2-ppc | 10c7698 | 2013-07-06 13:27:32 | [diff] [blame] | 1085 | } |
| 1086 | assert_eq!(d.len(), 0u); |
| 1087 | assert_eq!(e.len(), 0u); |
| 1088 | } |
| 1089 | |
| 1090 | #[test] |
| 1091 | fn test_eq() { |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 1092 | let mut d = RingBuf::new(); |
Alex Crichton | 02882fb | 2014-02-28 09:23:06 | [diff] [blame] | 1093 | assert!(d == RingBuf::with_capacity(0)); |
Niko Matsakis | 9e3d0b0 | 2014-04-21 21:58:52 | [diff] [blame] | 1094 | d.push_front(137i); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 1095 | d.push_front(17); |
| 1096 | d.push_front(42); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 1097 | d.push(137); |
blake2-ppc | 7052371 | 2013-07-10 13:27:14 | [diff] [blame] | 1098 | let mut e = RingBuf::with_capacity(0); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 1099 | e.push(42); |
| 1100 | e.push(17); |
| 1101 | e.push(137); |
| 1102 | e.push(137); |
Alex Crichton | 02882fb | 2014-02-28 09:23:06 | [diff] [blame] | 1103 | assert!(&e == &d); |
nham | 7b31058 | 2014-08-26 19:45:55 | [diff] [blame] | 1104 | e.pop(); |
| 1105 | e.push(0); |
blake2-ppc | 10c7698 | 2013-07-06 13:27:32 | [diff] [blame] | 1106 | assert!(e != d); |
| 1107 | e.clear(); |
Alex Crichton | 02882fb | 2014-02-28 09:23:06 | [diff] [blame] | 1108 | assert!(e == RingBuf::new()); |
blake2-ppc | 10c7698 | 2013-07-06 13:27:32 | [diff] [blame] | 1109 | } |
Adolfo Ochagavía | 8e4e3ab | 2014-06-04 14:15:04 | [diff] [blame] | 1110 | |
| 1111 | #[test] |
nham | 1cfa656 | 2014-07-27 02:33:47 | [diff] [blame] | 1112 | fn test_hash() { |
| 1113 | let mut x = RingBuf::new(); |
| 1114 | let mut y = RingBuf::new(); |
| 1115 | |
| 1116 | x.push(1i); |
| 1117 | x.push(2); |
| 1118 | x.push(3); |
| 1119 | |
| 1120 | y.push(0i); |
| 1121 | y.push(1i); |
| 1122 | y.pop_front(); |
| 1123 | y.push(2); |
| 1124 | y.push(3); |
| 1125 | |
| 1126 | assert!(hash::hash(&x) == hash::hash(&y)); |
| 1127 | } |
| 1128 | |
| 1129 | #[test] |
nham | 6361577 | 2014-07-27 03:18:56 | [diff] [blame] | 1130 | fn test_ord() { |
| 1131 | let x = RingBuf::new(); |
| 1132 | let mut y = RingBuf::new(); |
| 1133 | y.push(1i); |
| 1134 | y.push(2); |
| 1135 | y.push(3); |
| 1136 | assert!(x < y); |
| 1137 | assert!(y > x); |
| 1138 | assert!(x <= x); |
| 1139 | assert!(x >= x); |
| 1140 | } |
| 1141 | |
| 1142 | #[test] |
Adolfo Ochagavía | 8e4e3ab | 2014-06-04 14:15:04 | [diff] [blame] | 1143 | fn test_show() { |
Niko Matsakis | 9e3d0b0 | 2014-04-21 21:58:52 | [diff] [blame] | 1144 | let ringbuf: RingBuf<int> = range(0i, 10).collect(); |
Adolfo Ochagavía | 8e4e3ab | 2014-06-04 14:15:04 | [diff] [blame] | 1145 | assert!(format!("{}", ringbuf).as_slice() == "[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]"); |
| 1146 | |
| 1147 | let ringbuf: RingBuf<&str> = vec!["just", "one", "test", "more"].iter() |
| 1148 | .map(|&s| s) |
| 1149 | .collect(); |
| 1150 | assert!(format!("{}", ringbuf).as_slice() == "[just, one, test, more]"); |
| 1151 | } |
Michael Sullivan | c854d6e | 2012-07-03 17:52:32 | [diff] [blame] | 1152 | } |