]> git.proxmox.com Git - rustc.git/blob - library/alloc/src/collections/vec_deque/mod.rs
New upstream version 1.54.0+dfsg1
[rustc.git] / library / alloc / src / collections / vec_deque / mod.rs
1 //! A double-ended queue implemented with a growable ring buffer.
2 //!
3 //! This queue has *O*(1) amortized inserts and removals from both ends of the
4 //! container. It also has *O*(1) indexing like a vector. The contained elements
5 //! are not required to be copyable, and the queue will be sendable if the
6 //! contained type is sendable.
7
8 #![stable(feature = "rust1", since = "1.0.0")]
9
10 use core::cmp::{self, Ordering};
11 use core::fmt;
12 use core::hash::{Hash, Hasher};
13 use core::iter::{repeat_with, FromIterator};
14 use core::marker::PhantomData;
15 use core::mem::{self, ManuallyDrop};
16 use core::ops::{Index, IndexMut, Range, RangeBounds};
17 use core::ptr::{self, NonNull};
18 use core::slice;
19
20 use crate::collections::TryReserveError;
21 use crate::raw_vec::RawVec;
22 use crate::vec::Vec;
23
24 #[macro_use]
25 mod macros;
26
27 #[stable(feature = "drain", since = "1.6.0")]
28 pub use self::drain::Drain;
29
30 mod drain;
31
32 #[stable(feature = "rust1", since = "1.0.0")]
33 pub use self::iter_mut::IterMut;
34
35 mod iter_mut;
36
37 #[stable(feature = "rust1", since = "1.0.0")]
38 pub use self::into_iter::IntoIter;
39
40 mod into_iter;
41
42 #[stable(feature = "rust1", since = "1.0.0")]
43 pub use self::iter::Iter;
44
45 mod iter;
46
47 use self::pair_slices::PairSlices;
48
49 mod pair_slices;
50
51 use self::ring_slices::RingSlices;
52
53 mod ring_slices;
54
55 #[cfg(test)]
56 mod tests;
57
58 const INITIAL_CAPACITY: usize = 7; // 2^3 - 1
59 const MINIMUM_CAPACITY: usize = 1; // 2 - 1
60
61 const MAXIMUM_ZST_CAPACITY: usize = 1 << (usize::BITS - 1); // Largest possible power of two
62
63 /// A double-ended queue implemented with a growable ring buffer.
64 ///
65 /// The "default" usage of this type as a queue is to use [`push_back`] to add to
66 /// the queue, and [`pop_front`] to remove from the queue. [`extend`] and [`append`]
67 /// push onto the back in this manner, and iterating over `VecDeque` goes front
68 /// to back.
69 ///
70 /// Since `VecDeque` is a ring buffer, its elements are not necessarily contiguous
71 /// in memory. If you want to access the elements as a single slice, such as for
72 /// efficient sorting, you can use [`make_contiguous`]. It rotates the `VecDeque`
73 /// so that its elements do not wrap, and returns a mutable slice to the
74 /// now-contiguous element sequence.
75 ///
76 /// [`push_back`]: VecDeque::push_back
77 /// [`pop_front`]: VecDeque::pop_front
78 /// [`extend`]: VecDeque::extend
79 /// [`append`]: VecDeque::append
80 /// [`make_contiguous`]: VecDeque::make_contiguous
81 #[cfg_attr(not(test), rustc_diagnostic_item = "vecdeque_type")]
82 #[stable(feature = "rust1", since = "1.0.0")]
83 pub struct VecDeque<T> {
84 // tail and head are pointers into the buffer. Tail always points
85 // to the first element that could be read, Head always points
86 // to where data should be written.
87 // If tail == head the buffer is empty. The length of the ringbuffer
88 // is defined as the distance between the two.
89 tail: usize,
90 head: usize,
91 buf: RawVec<T>,
92 }
93
94 #[stable(feature = "rust1", since = "1.0.0")]
95 impl<T: Clone> Clone for VecDeque<T> {
96 fn clone(&self) -> VecDeque<T> {
97 self.iter().cloned().collect()
98 }
99
100 fn clone_from(&mut self, other: &Self) {
101 self.truncate(other.len());
102
103 let mut iter = PairSlices::from(self, other);
104 while let Some((dst, src)) = iter.next() {
105 dst.clone_from_slice(&src);
106 }
107
108 if iter.has_remainder() {
109 for remainder in iter.remainder() {
110 self.extend(remainder.iter().cloned());
111 }
112 }
113 }
114 }
115
116 #[stable(feature = "rust1", since = "1.0.0")]
117 unsafe impl<#[may_dangle] T> Drop for VecDeque<T> {
118 fn drop(&mut self) {
119 /// Runs the destructor for all items in the slice when it gets dropped (normally or
120 /// during unwinding).
121 struct Dropper<'a, T>(&'a mut [T]);
122
123 impl<'a, T> Drop for Dropper<'a, T> {
124 fn drop(&mut self) {
125 unsafe {
126 ptr::drop_in_place(self.0);
127 }
128 }
129 }
130
131 let (front, back) = self.as_mut_slices();
132 unsafe {
133 let _back_dropper = Dropper(back);
134 // use drop for [T]
135 ptr::drop_in_place(front);
136 }
137 // RawVec handles deallocation
138 }
139 }
140
141 #[stable(feature = "rust1", since = "1.0.0")]
142 impl<T> Default for VecDeque<T> {
143 /// Creates an empty `VecDeque<T>`.
144 #[inline]
145 fn default() -> VecDeque<T> {
146 VecDeque::new()
147 }
148 }
149
150 impl<T> VecDeque<T> {
151 /// Marginally more convenient
152 #[inline]
153 fn ptr(&self) -> *mut T {
154 self.buf.ptr()
155 }
156
157 /// Marginally more convenient
158 #[inline]
159 fn cap(&self) -> usize {
160 if mem::size_of::<T>() == 0 {
161 // For zero sized types, we are always at maximum capacity
162 MAXIMUM_ZST_CAPACITY
163 } else {
164 self.buf.capacity()
165 }
166 }
167
168 /// Turn ptr into a slice
169 #[inline]
170 unsafe fn buffer_as_slice(&self) -> &[T] {
171 unsafe { slice::from_raw_parts(self.ptr(), self.cap()) }
172 }
173
174 /// Turn ptr into a mut slice
175 #[inline]
176 unsafe fn buffer_as_mut_slice(&mut self) -> &mut [T] {
177 unsafe { slice::from_raw_parts_mut(self.ptr(), self.cap()) }
178 }
179
180 /// Moves an element out of the buffer
181 #[inline]
182 unsafe fn buffer_read(&mut self, off: usize) -> T {
183 unsafe { ptr::read(self.ptr().add(off)) }
184 }
185
186 /// Writes an element into the buffer, moving it.
187 #[inline]
188 unsafe fn buffer_write(&mut self, off: usize, value: T) {
189 unsafe {
190 ptr::write(self.ptr().add(off), value);
191 }
192 }
193
194 /// Returns `true` if the buffer is at full capacity.
195 #[inline]
196 fn is_full(&self) -> bool {
197 self.cap() - self.len() == 1
198 }
199
200 /// Returns the index in the underlying buffer for a given logical element
201 /// index.
202 #[inline]
203 fn wrap_index(&self, idx: usize) -> usize {
204 wrap_index(idx, self.cap())
205 }
206
207 /// Returns the index in the underlying buffer for a given logical element
208 /// index + addend.
209 #[inline]
210 fn wrap_add(&self, idx: usize, addend: usize) -> usize {
211 wrap_index(idx.wrapping_add(addend), self.cap())
212 }
213
214 /// Returns the index in the underlying buffer for a given logical element
215 /// index - subtrahend.
216 #[inline]
217 fn wrap_sub(&self, idx: usize, subtrahend: usize) -> usize {
218 wrap_index(idx.wrapping_sub(subtrahend), self.cap())
219 }
220
221 /// Copies a contiguous block of memory len long from src to dst
222 #[inline]
223 unsafe fn copy(&self, dst: usize, src: usize, len: usize) {
224 debug_assert!(
225 dst + len <= self.cap(),
226 "cpy dst={} src={} len={} cap={}",
227 dst,
228 src,
229 len,
230 self.cap()
231 );
232 debug_assert!(
233 src + len <= self.cap(),
234 "cpy dst={} src={} len={} cap={}",
235 dst,
236 src,
237 len,
238 self.cap()
239 );
240 unsafe {
241 ptr::copy(self.ptr().add(src), self.ptr().add(dst), len);
242 }
243 }
244
245 /// Copies a contiguous block of memory len long from src to dst
246 #[inline]
247 unsafe fn copy_nonoverlapping(&self, dst: usize, src: usize, len: usize) {
248 debug_assert!(
249 dst + len <= self.cap(),
250 "cno dst={} src={} len={} cap={}",
251 dst,
252 src,
253 len,
254 self.cap()
255 );
256 debug_assert!(
257 src + len <= self.cap(),
258 "cno dst={} src={} len={} cap={}",
259 dst,
260 src,
261 len,
262 self.cap()
263 );
264 unsafe {
265 ptr::copy_nonoverlapping(self.ptr().add(src), self.ptr().add(dst), len);
266 }
267 }
268
269 /// Copies a potentially wrapping block of memory len long from src to dest.
270 /// (abs(dst - src) + len) must be no larger than cap() (There must be at
271 /// most one continuous overlapping region between src and dest).
272 unsafe fn wrap_copy(&self, dst: usize, src: usize, len: usize) {
273 #[allow(dead_code)]
274 fn diff(a: usize, b: usize) -> usize {
275 if a <= b { b - a } else { a - b }
276 }
277 debug_assert!(
278 cmp::min(diff(dst, src), self.cap() - diff(dst, src)) + len <= self.cap(),
279 "wrc dst={} src={} len={} cap={}",
280 dst,
281 src,
282 len,
283 self.cap()
284 );
285
286 if src == dst || len == 0 {
287 return;
288 }
289
290 let dst_after_src = self.wrap_sub(dst, src) < len;
291
292 let src_pre_wrap_len = self.cap() - src;
293 let dst_pre_wrap_len = self.cap() - dst;
294 let src_wraps = src_pre_wrap_len < len;
295 let dst_wraps = dst_pre_wrap_len < len;
296
297 match (dst_after_src, src_wraps, dst_wraps) {
298 (_, false, false) => {
299 // src doesn't wrap, dst doesn't wrap
300 //
301 // S . . .
302 // 1 [_ _ A A B B C C _]
303 // 2 [_ _ A A A A B B _]
304 // D . . .
305 //
306 unsafe {
307 self.copy(dst, src, len);
308 }
309 }
310 (false, false, true) => {
311 // dst before src, src doesn't wrap, dst wraps
312 //
313 // S . . .
314 // 1 [A A B B _ _ _ C C]
315 // 2 [A A B B _ _ _ A A]
316 // 3 [B B B B _ _ _ A A]
317 // . . D .
318 //
319 unsafe {
320 self.copy(dst, src, dst_pre_wrap_len);
321 self.copy(0, src + dst_pre_wrap_len, len - dst_pre_wrap_len);
322 }
323 }
324 (true, false, true) => {
325 // src before dst, src doesn't wrap, dst wraps
326 //
327 // S . . .
328 // 1 [C C _ _ _ A A B B]
329 // 2 [B B _ _ _ A A B B]
330 // 3 [B B _ _ _ A A A A]
331 // . . D .
332 //
333 unsafe {
334 self.copy(0, src + dst_pre_wrap_len, len - dst_pre_wrap_len);
335 self.copy(dst, src, dst_pre_wrap_len);
336 }
337 }
338 (false, true, false) => {
339 // dst before src, src wraps, dst doesn't wrap
340 //
341 // . . S .
342 // 1 [C C _ _ _ A A B B]
343 // 2 [C C _ _ _ B B B B]
344 // 3 [C C _ _ _ B B C C]
345 // D . . .
346 //
347 unsafe {
348 self.copy(dst, src, src_pre_wrap_len);
349 self.copy(dst + src_pre_wrap_len, 0, len - src_pre_wrap_len);
350 }
351 }
352 (true, true, false) => {
353 // src before dst, src wraps, dst doesn't wrap
354 //
355 // . . S .
356 // 1 [A A B B _ _ _ C C]
357 // 2 [A A A A _ _ _ C C]
358 // 3 [C C A A _ _ _ C C]
359 // D . . .
360 //
361 unsafe {
362 self.copy(dst + src_pre_wrap_len, 0, len - src_pre_wrap_len);
363 self.copy(dst, src, src_pre_wrap_len);
364 }
365 }
366 (false, true, true) => {
367 // dst before src, src wraps, dst wraps
368 //
369 // . . . S .
370 // 1 [A B C D _ E F G H]
371 // 2 [A B C D _ E G H H]
372 // 3 [A B C D _ E G H A]
373 // 4 [B C C D _ E G H A]
374 // . . D . .
375 //
376 debug_assert!(dst_pre_wrap_len > src_pre_wrap_len);
377 let delta = dst_pre_wrap_len - src_pre_wrap_len;
378 unsafe {
379 self.copy(dst, src, src_pre_wrap_len);
380 self.copy(dst + src_pre_wrap_len, 0, delta);
381 self.copy(0, delta, len - dst_pre_wrap_len);
382 }
383 }
384 (true, true, true) => {
385 // src before dst, src wraps, dst wraps
386 //
387 // . . S . .
388 // 1 [A B C D _ E F G H]
389 // 2 [A A B D _ E F G H]
390 // 3 [H A B D _ E F G H]
391 // 4 [H A B D _ E F F G]
392 // . . . D .
393 //
394 debug_assert!(src_pre_wrap_len > dst_pre_wrap_len);
395 let delta = src_pre_wrap_len - dst_pre_wrap_len;
396 unsafe {
397 self.copy(delta, 0, len - src_pre_wrap_len);
398 self.copy(0, self.cap() - delta, delta);
399 self.copy(dst, src, dst_pre_wrap_len);
400 }
401 }
402 }
403 }
404
405 /// Frobs the head and tail sections around to handle the fact that we
406 /// just reallocated. Unsafe because it trusts old_capacity.
407 #[inline]
408 unsafe fn handle_capacity_increase(&mut self, old_capacity: usize) {
409 let new_capacity = self.cap();
410
411 // Move the shortest contiguous section of the ring buffer
412 // T H
413 // [o o o o o o o . ]
414 // T H
415 // A [o o o o o o o . . . . . . . . . ]
416 // H T
417 // [o o . o o o o o ]
418 // T H
419 // B [. . . o o o o o o o . . . . . . ]
420 // H T
421 // [o o o o o . o o ]
422 // H T
423 // C [o o o o o . . . . . . . . . o o ]
424
425 if self.tail <= self.head {
426 // A
427 // Nop
428 } else if self.head < old_capacity - self.tail {
429 // B
430 unsafe {
431 self.copy_nonoverlapping(old_capacity, 0, self.head);
432 }
433 self.head += old_capacity;
434 debug_assert!(self.head > self.tail);
435 } else {
436 // C
437 let new_tail = new_capacity - (old_capacity - self.tail);
438 unsafe {
439 self.copy_nonoverlapping(new_tail, self.tail, old_capacity - self.tail);
440 }
441 self.tail = new_tail;
442 debug_assert!(self.head < self.tail);
443 }
444 debug_assert!(self.head < self.cap());
445 debug_assert!(self.tail < self.cap());
446 debug_assert!(self.cap().count_ones() == 1);
447 }
448 }
449
450 impl<T> VecDeque<T> {
451 /// Creates an empty `VecDeque`.
452 ///
453 /// # Examples
454 ///
455 /// ```
456 /// use std::collections::VecDeque;
457 ///
458 /// let vector: VecDeque<u32> = VecDeque::new();
459 /// ```
460 #[stable(feature = "rust1", since = "1.0.0")]
461 pub fn new() -> VecDeque<T> {
462 VecDeque::with_capacity(INITIAL_CAPACITY)
463 }
464
465 /// Creates an empty `VecDeque` with space for at least `capacity` elements.
466 ///
467 /// # Examples
468 ///
469 /// ```
470 /// use std::collections::VecDeque;
471 ///
472 /// let vector: VecDeque<u32> = VecDeque::with_capacity(10);
473 /// ```
474 #[stable(feature = "rust1", since = "1.0.0")]
475 pub fn with_capacity(capacity: usize) -> VecDeque<T> {
476 // +1 since the ringbuffer always leaves one space empty
477 let cap = cmp::max(capacity + 1, MINIMUM_CAPACITY + 1).next_power_of_two();
478 assert!(cap > capacity, "capacity overflow");
479
480 VecDeque { tail: 0, head: 0, buf: RawVec::with_capacity(cap) }
481 }
482
483 /// Provides a reference to the element at the given index.
484 ///
485 /// Element at index 0 is the front of the queue.
486 ///
487 /// # Examples
488 ///
489 /// ```
490 /// use std::collections::VecDeque;
491 ///
492 /// let mut buf = VecDeque::new();
493 /// buf.push_back(3);
494 /// buf.push_back(4);
495 /// buf.push_back(5);
496 /// assert_eq!(buf.get(1), Some(&4));
497 /// ```
498 #[stable(feature = "rust1", since = "1.0.0")]
499 pub fn get(&self, index: usize) -> Option<&T> {
500 if index < self.len() {
501 let idx = self.wrap_add(self.tail, index);
502 unsafe { Some(&*self.ptr().add(idx)) }
503 } else {
504 None
505 }
506 }
507
508 /// Provides a mutable reference to the element at the given index.
509 ///
510 /// Element at index 0 is the front of the queue.
511 ///
512 /// # Examples
513 ///
514 /// ```
515 /// use std::collections::VecDeque;
516 ///
517 /// let mut buf = VecDeque::new();
518 /// buf.push_back(3);
519 /// buf.push_back(4);
520 /// buf.push_back(5);
521 /// if let Some(elem) = buf.get_mut(1) {
522 /// *elem = 7;
523 /// }
524 ///
525 /// assert_eq!(buf[1], 7);
526 /// ```
527 #[stable(feature = "rust1", since = "1.0.0")]
528 pub fn get_mut(&mut self, index: usize) -> Option<&mut T> {
529 if index < self.len() {
530 let idx = self.wrap_add(self.tail, index);
531 unsafe { Some(&mut *self.ptr().add(idx)) }
532 } else {
533 None
534 }
535 }
536
537 /// Swaps elements at indices `i` and `j`.
538 ///
539 /// `i` and `j` may be equal.
540 ///
541 /// Element at index 0 is the front of the queue.
542 ///
543 /// # Panics
544 ///
545 /// Panics if either index is out of bounds.
546 ///
547 /// # Examples
548 ///
549 /// ```
550 /// use std::collections::VecDeque;
551 ///
552 /// let mut buf = VecDeque::new();
553 /// buf.push_back(3);
554 /// buf.push_back(4);
555 /// buf.push_back(5);
556 /// assert_eq!(buf, [3, 4, 5]);
557 /// buf.swap(0, 2);
558 /// assert_eq!(buf, [5, 4, 3]);
559 /// ```
560 #[stable(feature = "rust1", since = "1.0.0")]
561 pub fn swap(&mut self, i: usize, j: usize) {
562 assert!(i < self.len());
563 assert!(j < self.len());
564 let ri = self.wrap_add(self.tail, i);
565 let rj = self.wrap_add(self.tail, j);
566 unsafe { ptr::swap(self.ptr().add(ri), self.ptr().add(rj)) }
567 }
568
569 /// Returns the number of elements the `VecDeque` can hold without
570 /// reallocating.
571 ///
572 /// # Examples
573 ///
574 /// ```
575 /// use std::collections::VecDeque;
576 ///
577 /// let buf: VecDeque<i32> = VecDeque::with_capacity(10);
578 /// assert!(buf.capacity() >= 10);
579 /// ```
580 #[inline]
581 #[stable(feature = "rust1", since = "1.0.0")]
582 pub fn capacity(&self) -> usize {
583 self.cap() - 1
584 }
585
586 /// Reserves the minimum capacity for exactly `additional` more elements to be inserted in the
587 /// given `VecDeque`. Does nothing if the capacity is already sufficient.
588 ///
589 /// Note that the allocator may give the collection more space than it requests. Therefore
590 /// capacity can not be relied upon to be precisely minimal. Prefer [`reserve`] if future
591 /// insertions are expected.
592 ///
593 /// # Panics
594 ///
595 /// Panics if the new capacity overflows `usize`.
596 ///
597 /// # Examples
598 ///
599 /// ```
600 /// use std::collections::VecDeque;
601 ///
602 /// let mut buf: VecDeque<i32> = vec![1].into_iter().collect();
603 /// buf.reserve_exact(10);
604 /// assert!(buf.capacity() >= 11);
605 /// ```
606 ///
607 /// [`reserve`]: VecDeque::reserve
608 #[stable(feature = "rust1", since = "1.0.0")]
609 pub fn reserve_exact(&mut self, additional: usize) {
610 self.reserve(additional);
611 }
612
613 /// Reserves capacity for at least `additional` more elements to be inserted in the given
614 /// `VecDeque`. The collection may reserve more space to avoid frequent reallocations.
615 ///
616 /// # Panics
617 ///
618 /// Panics if the new capacity overflows `usize`.
619 ///
620 /// # Examples
621 ///
622 /// ```
623 /// use std::collections::VecDeque;
624 ///
625 /// let mut buf: VecDeque<i32> = vec![1].into_iter().collect();
626 /// buf.reserve(10);
627 /// assert!(buf.capacity() >= 11);
628 /// ```
629 #[stable(feature = "rust1", since = "1.0.0")]
630 pub fn reserve(&mut self, additional: usize) {
631 let old_cap = self.cap();
632 let used_cap = self.len() + 1;
633 let new_cap = used_cap
634 .checked_add(additional)
635 .and_then(|needed_cap| needed_cap.checked_next_power_of_two())
636 .expect("capacity overflow");
637
638 if new_cap > old_cap {
639 self.buf.reserve_exact(used_cap, new_cap - used_cap);
640 unsafe {
641 self.handle_capacity_increase(old_cap);
642 }
643 }
644 }
645
646 /// Tries to reserve the minimum capacity for exactly `additional` more elements to
647 /// be inserted in the given `VecDeque<T>`. After calling `try_reserve_exact`,
648 /// capacity will be greater than or equal to `self.len() + additional`.
649 /// Does nothing if the capacity is already sufficient.
650 ///
651 /// Note that the allocator may give the collection more space than it
652 /// requests. Therefore, capacity can not be relied upon to be precisely
653 /// minimal. Prefer `reserve` if future insertions are expected.
654 ///
655 /// # Errors
656 ///
657 /// If the capacity overflows `usize`, or the allocator reports a failure, then an error
658 /// is returned.
659 ///
660 /// # Examples
661 ///
662 /// ```
663 /// #![feature(try_reserve)]
664 /// use std::collections::TryReserveError;
665 /// use std::collections::VecDeque;
666 ///
667 /// fn process_data(data: &[u32]) -> Result<VecDeque<u32>, TryReserveError> {
668 /// let mut output = VecDeque::new();
669 ///
670 /// // Pre-reserve the memory, exiting if we can't
671 /// output.try_reserve_exact(data.len())?;
672 ///
673 /// // Now we know this can't OOM(Out-Of-Memory) in the middle of our complex work
674 /// output.extend(data.iter().map(|&val| {
675 /// val * 2 + 5 // very complicated
676 /// }));
677 ///
678 /// Ok(output)
679 /// }
680 /// # process_data(&[1, 2, 3]).expect("why is the test harness OOMing on 12 bytes?");
681 /// ```
682 #[unstable(feature = "try_reserve", reason = "new API", issue = "48043")]
683 pub fn try_reserve_exact(&mut self, additional: usize) -> Result<(), TryReserveError> {
684 self.try_reserve(additional)
685 }
686
687 /// Tries to reserve capacity for at least `additional` more elements to be inserted
688 /// in the given `VecDeque<T>`. The collection may reserve more space to avoid
689 /// frequent reallocations. After calling `try_reserve`, capacity will be
690 /// greater than or equal to `self.len() + additional`. Does nothing if
691 /// capacity is already sufficient.
692 ///
693 /// # Errors
694 ///
695 /// If the capacity overflows `usize`, or the allocator reports a failure, then an error
696 /// is returned.
697 ///
698 /// # Examples
699 ///
700 /// ```
701 /// #![feature(try_reserve)]
702 /// use std::collections::TryReserveError;
703 /// use std::collections::VecDeque;
704 ///
705 /// fn process_data(data: &[u32]) -> Result<VecDeque<u32>, TryReserveError> {
706 /// let mut output = VecDeque::new();
707 ///
708 /// // Pre-reserve the memory, exiting if we can't
709 /// output.try_reserve(data.len())?;
710 ///
711 /// // Now we know this can't OOM in the middle of our complex work
712 /// output.extend(data.iter().map(|&val| {
713 /// val * 2 + 5 // very complicated
714 /// }));
715 ///
716 /// Ok(output)
717 /// }
718 /// # process_data(&[1, 2, 3]).expect("why is the test harness OOMing on 12 bytes?");
719 /// ```
720 #[unstable(feature = "try_reserve", reason = "new API", issue = "48043")]
721 pub fn try_reserve(&mut self, additional: usize) -> Result<(), TryReserveError> {
722 let old_cap = self.cap();
723 let used_cap = self.len() + 1;
724 let new_cap = used_cap
725 .checked_add(additional)
726 .and_then(|needed_cap| needed_cap.checked_next_power_of_two())
727 .ok_or(TryReserveError::CapacityOverflow)?;
728
729 if new_cap > old_cap {
730 self.buf.try_reserve_exact(used_cap, new_cap - used_cap)?;
731 unsafe {
732 self.handle_capacity_increase(old_cap);
733 }
734 }
735 Ok(())
736 }
737
738 /// Shrinks the capacity of the `VecDeque` as much as possible.
739 ///
740 /// It will drop down as close as possible to the length but the allocator may still inform the
741 /// `VecDeque` that there is space for a few more elements.
742 ///
743 /// # Examples
744 ///
745 /// ```
746 /// use std::collections::VecDeque;
747 ///
748 /// let mut buf = VecDeque::with_capacity(15);
749 /// buf.extend(0..4);
750 /// assert_eq!(buf.capacity(), 15);
751 /// buf.shrink_to_fit();
752 /// assert!(buf.capacity() >= 4);
753 /// ```
754 #[stable(feature = "deque_extras_15", since = "1.5.0")]
755 pub fn shrink_to_fit(&mut self) {
756 self.shrink_to(0);
757 }
758
759 /// Shrinks the capacity of the `VecDeque` with a lower bound.
760 ///
761 /// The capacity will remain at least as large as both the length
762 /// and the supplied value.
763 ///
764 /// If the current capacity is less than the lower limit, this is a no-op.
765 ///
766 /// # Examples
767 ///
768 /// ```
769 /// #![feature(shrink_to)]
770 /// use std::collections::VecDeque;
771 ///
772 /// let mut buf = VecDeque::with_capacity(15);
773 /// buf.extend(0..4);
774 /// assert_eq!(buf.capacity(), 15);
775 /// buf.shrink_to(6);
776 /// assert!(buf.capacity() >= 6);
777 /// buf.shrink_to(0);
778 /// assert!(buf.capacity() >= 4);
779 /// ```
780 #[unstable(feature = "shrink_to", reason = "new API", issue = "56431")]
781 pub fn shrink_to(&mut self, min_capacity: usize) {
782 let min_capacity = cmp::min(min_capacity, self.capacity());
783 // We don't have to worry about an overflow as neither `self.len()` nor `self.capacity()`
784 // can ever be `usize::MAX`. +1 as the ringbuffer always leaves one space empty.
785 let target_cap = cmp::max(cmp::max(min_capacity, self.len()) + 1, MINIMUM_CAPACITY + 1)
786 .next_power_of_two();
787
788 if target_cap < self.cap() {
789 // There are three cases of interest:
790 // All elements are out of desired bounds
791 // Elements are contiguous, and head is out of desired bounds
792 // Elements are discontiguous, and tail is out of desired bounds
793 //
794 // At all other times, element positions are unaffected.
795 //
796 // Indicates that elements at the head should be moved.
797 let head_outside = self.head == 0 || self.head >= target_cap;
798 // Move elements from out of desired bounds (positions after target_cap)
799 if self.tail >= target_cap && head_outside {
800 // T H
801 // [. . . . . . . . o o o o o o o . ]
802 // T H
803 // [o o o o o o o . ]
804 unsafe {
805 self.copy_nonoverlapping(0, self.tail, self.len());
806 }
807 self.head = self.len();
808 self.tail = 0;
809 } else if self.tail != 0 && self.tail < target_cap && head_outside {
810 // T H
811 // [. . . o o o o o o o . . . . . . ]
812 // H T
813 // [o o . o o o o o ]
814 let len = self.wrap_sub(self.head, target_cap);
815 unsafe {
816 self.copy_nonoverlapping(0, target_cap, len);
817 }
818 self.head = len;
819 debug_assert!(self.head < self.tail);
820 } else if self.tail >= target_cap {
821 // H T
822 // [o o o o o . . . . . . . . . o o ]
823 // H T
824 // [o o o o o . o o ]
825 debug_assert!(self.wrap_sub(self.head, 1) < target_cap);
826 let len = self.cap() - self.tail;
827 let new_tail = target_cap - len;
828 unsafe {
829 self.copy_nonoverlapping(new_tail, self.tail, len);
830 }
831 self.tail = new_tail;
832 debug_assert!(self.head < self.tail);
833 }
834
835 self.buf.shrink_to_fit(target_cap);
836
837 debug_assert!(self.head < self.cap());
838 debug_assert!(self.tail < self.cap());
839 debug_assert!(self.cap().count_ones() == 1);
840 }
841 }
842
843 /// Shortens the `VecDeque`, keeping the first `len` elements and dropping
844 /// the rest.
845 ///
846 /// If `len` is greater than the `VecDeque`'s current length, this has no
847 /// effect.
848 ///
849 /// # Examples
850 ///
851 /// ```
852 /// use std::collections::VecDeque;
853 ///
854 /// let mut buf = VecDeque::new();
855 /// buf.push_back(5);
856 /// buf.push_back(10);
857 /// buf.push_back(15);
858 /// assert_eq!(buf, [5, 10, 15]);
859 /// buf.truncate(1);
860 /// assert_eq!(buf, [5]);
861 /// ```
862 #[stable(feature = "deque_extras", since = "1.16.0")]
863 pub fn truncate(&mut self, len: usize) {
864 /// Runs the destructor for all items in the slice when it gets dropped (normally or
865 /// during unwinding).
866 struct Dropper<'a, T>(&'a mut [T]);
867
868 impl<'a, T> Drop for Dropper<'a, T> {
869 fn drop(&mut self) {
870 unsafe {
871 ptr::drop_in_place(self.0);
872 }
873 }
874 }
875
876 // Safe because:
877 //
878 // * Any slice passed to `drop_in_place` is valid; the second case has
879 // `len <= front.len()` and returning on `len > self.len()` ensures
880 // `begin <= back.len()` in the first case
881 // * The head of the VecDeque is moved before calling `drop_in_place`,
882 // so no value is dropped twice if `drop_in_place` panics
883 unsafe {
884 if len > self.len() {
885 return;
886 }
887 let num_dropped = self.len() - len;
888 let (front, back) = self.as_mut_slices();
889 if len > front.len() {
890 let begin = len - front.len();
891 let drop_back = back.get_unchecked_mut(begin..) as *mut _;
892 self.head = self.wrap_sub(self.head, num_dropped);
893 ptr::drop_in_place(drop_back);
894 } else {
895 let drop_back = back as *mut _;
896 let drop_front = front.get_unchecked_mut(len..) as *mut _;
897 self.head = self.wrap_sub(self.head, num_dropped);
898
899 // Make sure the second half is dropped even when a destructor
900 // in the first one panics.
901 let _back_dropper = Dropper(&mut *drop_back);
902 ptr::drop_in_place(drop_front);
903 }
904 }
905 }
906
907 /// Returns a front-to-back iterator.
908 ///
909 /// # Examples
910 ///
911 /// ```
912 /// use std::collections::VecDeque;
913 ///
914 /// let mut buf = VecDeque::new();
915 /// buf.push_back(5);
916 /// buf.push_back(3);
917 /// buf.push_back(4);
918 /// let b: &[_] = &[&5, &3, &4];
919 /// let c: Vec<&i32> = buf.iter().collect();
920 /// assert_eq!(&c[..], b);
921 /// ```
922 #[stable(feature = "rust1", since = "1.0.0")]
923 pub fn iter(&self) -> Iter<'_, T> {
924 Iter { tail: self.tail, head: self.head, ring: unsafe { self.buffer_as_slice() } }
925 }
926
927 /// Returns a front-to-back iterator that returns mutable references.
928 ///
929 /// # Examples
930 ///
931 /// ```
932 /// use std::collections::VecDeque;
933 ///
934 /// let mut buf = VecDeque::new();
935 /// buf.push_back(5);
936 /// buf.push_back(3);
937 /// buf.push_back(4);
938 /// for num in buf.iter_mut() {
939 /// *num = *num - 2;
940 /// }
941 /// let b: &[_] = &[&mut 3, &mut 1, &mut 2];
942 /// assert_eq!(&buf.iter_mut().collect::<Vec<&mut i32>>()[..], b);
943 /// ```
944 #[stable(feature = "rust1", since = "1.0.0")]
945 pub fn iter_mut(&mut self) -> IterMut<'_, T> {
946 // SAFETY: The internal `IterMut` safety invariant is established because the
947 // `ring` we create is a dereferencable slice for lifetime '_.
948 IterMut {
949 tail: self.tail,
950 head: self.head,
951 ring: ptr::slice_from_raw_parts_mut(self.ptr(), self.cap()),
952 phantom: PhantomData,
953 }
954 }
955
956 /// Returns a pair of slices which contain, in order, the contents of the
957 /// `VecDeque`.
958 ///
959 /// If [`make_contiguous`] was previously called, all elements of the
960 /// `VecDeque` will be in the first slice and the second slice will be empty.
961 ///
962 /// [`make_contiguous`]: VecDeque::make_contiguous
963 ///
964 /// # Examples
965 ///
966 /// ```
967 /// use std::collections::VecDeque;
968 ///
969 /// let mut vector = VecDeque::new();
970 ///
971 /// vector.push_back(0);
972 /// vector.push_back(1);
973 /// vector.push_back(2);
974 ///
975 /// assert_eq!(vector.as_slices(), (&[0, 1, 2][..], &[][..]));
976 ///
977 /// vector.push_front(10);
978 /// vector.push_front(9);
979 ///
980 /// assert_eq!(vector.as_slices(), (&[9, 10][..], &[0, 1, 2][..]));
981 /// ```
982 #[inline]
983 #[stable(feature = "deque_extras_15", since = "1.5.0")]
984 pub fn as_slices(&self) -> (&[T], &[T]) {
985 unsafe {
986 let buf = self.buffer_as_slice();
987 RingSlices::ring_slices(buf, self.head, self.tail)
988 }
989 }
990
991 /// Returns a pair of slices which contain, in order, the contents of the
992 /// `VecDeque`.
993 ///
994 /// If [`make_contiguous`] was previously called, all elements of the
995 /// `VecDeque` will be in the first slice and the second slice will be empty.
996 ///
997 /// [`make_contiguous`]: VecDeque::make_contiguous
998 ///
999 /// # Examples
1000 ///
1001 /// ```
1002 /// use std::collections::VecDeque;
1003 ///
1004 /// let mut vector = VecDeque::new();
1005 ///
1006 /// vector.push_back(0);
1007 /// vector.push_back(1);
1008 ///
1009 /// vector.push_front(10);
1010 /// vector.push_front(9);
1011 ///
1012 /// vector.as_mut_slices().0[0] = 42;
1013 /// vector.as_mut_slices().1[0] = 24;
1014 /// assert_eq!(vector.as_slices(), (&[42, 10][..], &[24, 1][..]));
1015 /// ```
1016 #[inline]
1017 #[stable(feature = "deque_extras_15", since = "1.5.0")]
1018 pub fn as_mut_slices(&mut self) -> (&mut [T], &mut [T]) {
1019 unsafe {
1020 let head = self.head;
1021 let tail = self.tail;
1022 let buf = self.buffer_as_mut_slice();
1023 RingSlices::ring_slices(buf, head, tail)
1024 }
1025 }
1026
1027 /// Returns the number of elements in the `VecDeque`.
1028 ///
1029 /// # Examples
1030 ///
1031 /// ```
1032 /// use std::collections::VecDeque;
1033 ///
1034 /// let mut v = VecDeque::new();
1035 /// assert_eq!(v.len(), 0);
1036 /// v.push_back(1);
1037 /// assert_eq!(v.len(), 1);
1038 /// ```
1039 #[doc(alias = "length")]
1040 #[stable(feature = "rust1", since = "1.0.0")]
1041 pub fn len(&self) -> usize {
1042 count(self.tail, self.head, self.cap())
1043 }
1044
1045 /// Returns `true` if the `VecDeque` is empty.
1046 ///
1047 /// # Examples
1048 ///
1049 /// ```
1050 /// use std::collections::VecDeque;
1051 ///
1052 /// let mut v = VecDeque::new();
1053 /// assert!(v.is_empty());
1054 /// v.push_front(1);
1055 /// assert!(!v.is_empty());
1056 /// ```
1057 #[stable(feature = "rust1", since = "1.0.0")]
1058 pub fn is_empty(&self) -> bool {
1059 self.tail == self.head
1060 }
1061
1062 fn range_tail_head<R>(&self, range: R) -> (usize, usize)
1063 where
1064 R: RangeBounds<usize>,
1065 {
1066 let Range { start, end } = slice::range(range, ..self.len());
1067 let tail = self.wrap_add(self.tail, start);
1068 let head = self.wrap_add(self.tail, end);
1069 (tail, head)
1070 }
1071
1072 /// Creates an iterator that covers the specified range in the `VecDeque`.
1073 ///
1074 /// # Panics
1075 ///
1076 /// Panics if the starting point is greater than the end point or if
1077 /// the end point is greater than the length of the vector.
1078 ///
1079 /// # Examples
1080 ///
1081 /// ```
1082 /// use std::collections::VecDeque;
1083 ///
1084 /// let v: VecDeque<_> = vec![1, 2, 3].into_iter().collect();
1085 /// let range = v.range(2..).copied().collect::<VecDeque<_>>();
1086 /// assert_eq!(range, [3]);
1087 ///
1088 /// // A full range covers all contents
1089 /// let all = v.range(..);
1090 /// assert_eq!(all.len(), 3);
1091 /// ```
1092 #[inline]
1093 #[stable(feature = "deque_range", since = "1.51.0")]
1094 pub fn range<R>(&self, range: R) -> Iter<'_, T>
1095 where
1096 R: RangeBounds<usize>,
1097 {
1098 let (tail, head) = self.range_tail_head(range);
1099 Iter {
1100 tail,
1101 head,
1102 // The shared reference we have in &self is maintained in the '_ of Iter.
1103 ring: unsafe { self.buffer_as_slice() },
1104 }
1105 }
1106
1107 /// Creates an iterator that covers the specified mutable range in the `VecDeque`.
1108 ///
1109 /// # Panics
1110 ///
1111 /// Panics if the starting point is greater than the end point or if
1112 /// the end point is greater than the length of the vector.
1113 ///
1114 /// # Examples
1115 ///
1116 /// ```
1117 /// use std::collections::VecDeque;
1118 ///
1119 /// let mut v: VecDeque<_> = vec![1, 2, 3].into_iter().collect();
1120 /// for v in v.range_mut(2..) {
1121 /// *v *= 2;
1122 /// }
1123 /// assert_eq!(v, vec![1, 2, 6]);
1124 ///
1125 /// // A full range covers all contents
1126 /// for v in v.range_mut(..) {
1127 /// *v *= 2;
1128 /// }
1129 /// assert_eq!(v, vec![2, 4, 12]);
1130 /// ```
1131 #[inline]
1132 #[stable(feature = "deque_range", since = "1.51.0")]
1133 pub fn range_mut<R>(&mut self, range: R) -> IterMut<'_, T>
1134 where
1135 R: RangeBounds<usize>,
1136 {
1137 let (tail, head) = self.range_tail_head(range);
1138
1139 // SAFETY: The internal `IterMut` safety invariant is established because the
1140 // `ring` we create is a dereferencable slice for lifetime '_.
1141 IterMut {
1142 tail,
1143 head,
1144 ring: ptr::slice_from_raw_parts_mut(self.ptr(), self.cap()),
1145 phantom: PhantomData,
1146 }
1147 }
1148
1149 /// Creates a draining iterator that removes the specified range in the
1150 /// `VecDeque` and yields the removed items.
1151 ///
1152 /// Note 1: The element range is removed even if the iterator is not
1153 /// consumed until the end.
1154 ///
1155 /// Note 2: It is unspecified how many elements are removed from the deque,
1156 /// if the `Drain` value is not dropped, but the borrow it holds expires
1157 /// (e.g., due to `mem::forget`).
1158 ///
1159 /// # Panics
1160 ///
1161 /// Panics if the starting point is greater than the end point or if
1162 /// the end point is greater than the length of the vector.
1163 ///
1164 /// # Examples
1165 ///
1166 /// ```
1167 /// use std::collections::VecDeque;
1168 ///
1169 /// let mut v: VecDeque<_> = vec![1, 2, 3].into_iter().collect();
1170 /// let drained = v.drain(2..).collect::<VecDeque<_>>();
1171 /// assert_eq!(drained, [3]);
1172 /// assert_eq!(v, [1, 2]);
1173 ///
1174 /// // A full range clears all contents
1175 /// v.drain(..);
1176 /// assert!(v.is_empty());
1177 /// ```
1178 #[inline]
1179 #[stable(feature = "drain", since = "1.6.0")]
1180 pub fn drain<R>(&mut self, range: R) -> Drain<'_, T>
1181 where
1182 R: RangeBounds<usize>,
1183 {
1184 // Memory safety
1185 //
1186 // When the Drain is first created, the source deque is shortened to
1187 // make sure no uninitialized or moved-from elements are accessible at
1188 // all if the Drain's destructor never gets to run.
1189 //
1190 // Drain will ptr::read out the values to remove.
1191 // When finished, the remaining data will be copied back to cover the hole,
1192 // and the head/tail values will be restored correctly.
1193 //
1194 let (drain_tail, drain_head) = self.range_tail_head(range);
1195
1196 // The deque's elements are parted into three segments:
1197 // * self.tail -> drain_tail
1198 // * drain_tail -> drain_head
1199 // * drain_head -> self.head
1200 //
1201 // T = self.tail; H = self.head; t = drain_tail; h = drain_head
1202 //
1203 // We store drain_tail as self.head, and drain_head and self.head as
1204 // after_tail and after_head respectively on the Drain. This also
1205 // truncates the effective array such that if the Drain is leaked, we
1206 // have forgotten about the potentially moved values after the start of
1207 // the drain.
1208 //
1209 // T t h H
1210 // [. . . o o x x o o . . .]
1211 //
1212 let head = self.head;
1213
1214 // "forget" about the values after the start of the drain until after
1215 // the drain is complete and the Drain destructor is run.
1216 self.head = drain_tail;
1217
1218 Drain {
1219 deque: NonNull::from(&mut *self),
1220 after_tail: drain_head,
1221 after_head: head,
1222 iter: Iter {
1223 tail: drain_tail,
1224 head: drain_head,
1225 // Crucially, we only create shared references from `self` here and read from
1226 // it. We do not write to `self` nor reborrow to a mutable reference.
1227 // Hence the raw pointer we created above, for `deque`, remains valid.
1228 ring: unsafe { self.buffer_as_slice() },
1229 },
1230 }
1231 }
1232
1233 /// Clears the `VecDeque`, removing all values.
1234 ///
1235 /// # Examples
1236 ///
1237 /// ```
1238 /// use std::collections::VecDeque;
1239 ///
1240 /// let mut v = VecDeque::new();
1241 /// v.push_back(1);
1242 /// v.clear();
1243 /// assert!(v.is_empty());
1244 /// ```
1245 #[stable(feature = "rust1", since = "1.0.0")]
1246 #[inline]
1247 pub fn clear(&mut self) {
1248 self.truncate(0);
1249 }
1250
1251 /// Returns `true` if the `VecDeque` contains an element equal to the
1252 /// given value.
1253 ///
1254 /// # Examples
1255 ///
1256 /// ```
1257 /// use std::collections::VecDeque;
1258 ///
1259 /// let mut vector: VecDeque<u32> = VecDeque::new();
1260 ///
1261 /// vector.push_back(0);
1262 /// vector.push_back(1);
1263 ///
1264 /// assert_eq!(vector.contains(&1), true);
1265 /// assert_eq!(vector.contains(&10), false);
1266 /// ```
1267 #[stable(feature = "vec_deque_contains", since = "1.12.0")]
1268 pub fn contains(&self, x: &T) -> bool
1269 where
1270 T: PartialEq<T>,
1271 {
1272 let (a, b) = self.as_slices();
1273 a.contains(x) || b.contains(x)
1274 }
1275
1276 /// Provides a reference to the front element, or `None` if the `VecDeque` is
1277 /// empty.
1278 ///
1279 /// # Examples
1280 ///
1281 /// ```
1282 /// use std::collections::VecDeque;
1283 ///
1284 /// let mut d = VecDeque::new();
1285 /// assert_eq!(d.front(), None);
1286 ///
1287 /// d.push_back(1);
1288 /// d.push_back(2);
1289 /// assert_eq!(d.front(), Some(&1));
1290 /// ```
1291 #[stable(feature = "rust1", since = "1.0.0")]
1292 pub fn front(&self) -> Option<&T> {
1293 self.get(0)
1294 }
1295
1296 /// Provides a mutable reference to the front element, or `None` if the
1297 /// `VecDeque` is empty.
1298 ///
1299 /// # Examples
1300 ///
1301 /// ```
1302 /// use std::collections::VecDeque;
1303 ///
1304 /// let mut d = VecDeque::new();
1305 /// assert_eq!(d.front_mut(), None);
1306 ///
1307 /// d.push_back(1);
1308 /// d.push_back(2);
1309 /// match d.front_mut() {
1310 /// Some(x) => *x = 9,
1311 /// None => (),
1312 /// }
1313 /// assert_eq!(d.front(), Some(&9));
1314 /// ```
1315 #[stable(feature = "rust1", since = "1.0.0")]
1316 pub fn front_mut(&mut self) -> Option<&mut T> {
1317 self.get_mut(0)
1318 }
1319
1320 /// Provides a reference to the back element, or `None` if the `VecDeque` is
1321 /// empty.
1322 ///
1323 /// # Examples
1324 ///
1325 /// ```
1326 /// use std::collections::VecDeque;
1327 ///
1328 /// let mut d = VecDeque::new();
1329 /// assert_eq!(d.back(), None);
1330 ///
1331 /// d.push_back(1);
1332 /// d.push_back(2);
1333 /// assert_eq!(d.back(), Some(&2));
1334 /// ```
1335 #[stable(feature = "rust1", since = "1.0.0")]
1336 pub fn back(&self) -> Option<&T> {
1337 self.get(self.len().wrapping_sub(1))
1338 }
1339
1340 /// Provides a mutable reference to the back element, or `None` if the
1341 /// `VecDeque` is empty.
1342 ///
1343 /// # Examples
1344 ///
1345 /// ```
1346 /// use std::collections::VecDeque;
1347 ///
1348 /// let mut d = VecDeque::new();
1349 /// assert_eq!(d.back(), None);
1350 ///
1351 /// d.push_back(1);
1352 /// d.push_back(2);
1353 /// match d.back_mut() {
1354 /// Some(x) => *x = 9,
1355 /// None => (),
1356 /// }
1357 /// assert_eq!(d.back(), Some(&9));
1358 /// ```
1359 #[stable(feature = "rust1", since = "1.0.0")]
1360 pub fn back_mut(&mut self) -> Option<&mut T> {
1361 self.get_mut(self.len().wrapping_sub(1))
1362 }
1363
1364 /// Removes the first element and returns it, or `None` if the `VecDeque` is
1365 /// empty.
1366 ///
1367 /// # Examples
1368 ///
1369 /// ```
1370 /// use std::collections::VecDeque;
1371 ///
1372 /// let mut d = VecDeque::new();
1373 /// d.push_back(1);
1374 /// d.push_back(2);
1375 ///
1376 /// assert_eq!(d.pop_front(), Some(1));
1377 /// assert_eq!(d.pop_front(), Some(2));
1378 /// assert_eq!(d.pop_front(), None);
1379 /// ```
1380 #[stable(feature = "rust1", since = "1.0.0")]
1381 pub fn pop_front(&mut self) -> Option<T> {
1382 if self.is_empty() {
1383 None
1384 } else {
1385 let tail = self.tail;
1386 self.tail = self.wrap_add(self.tail, 1);
1387 unsafe { Some(self.buffer_read(tail)) }
1388 }
1389 }
1390
1391 /// Removes the last element from the `VecDeque` and returns it, or `None` if
1392 /// it is empty.
1393 ///
1394 /// # Examples
1395 ///
1396 /// ```
1397 /// use std::collections::VecDeque;
1398 ///
1399 /// let mut buf = VecDeque::new();
1400 /// assert_eq!(buf.pop_back(), None);
1401 /// buf.push_back(1);
1402 /// buf.push_back(3);
1403 /// assert_eq!(buf.pop_back(), Some(3));
1404 /// ```
1405 #[stable(feature = "rust1", since = "1.0.0")]
1406 pub fn pop_back(&mut self) -> Option<T> {
1407 if self.is_empty() {
1408 None
1409 } else {
1410 self.head = self.wrap_sub(self.head, 1);
1411 let head = self.head;
1412 unsafe { Some(self.buffer_read(head)) }
1413 }
1414 }
1415
1416 /// Prepends an element to the `VecDeque`.
1417 ///
1418 /// # Examples
1419 ///
1420 /// ```
1421 /// use std::collections::VecDeque;
1422 ///
1423 /// let mut d = VecDeque::new();
1424 /// d.push_front(1);
1425 /// d.push_front(2);
1426 /// assert_eq!(d.front(), Some(&2));
1427 /// ```
1428 #[stable(feature = "rust1", since = "1.0.0")]
1429 pub fn push_front(&mut self, value: T) {
1430 if self.is_full() {
1431 self.grow();
1432 }
1433
1434 self.tail = self.wrap_sub(self.tail, 1);
1435 let tail = self.tail;
1436 unsafe {
1437 self.buffer_write(tail, value);
1438 }
1439 }
1440
1441 /// Appends an element to the back of the `VecDeque`.
1442 ///
1443 /// # Examples
1444 ///
1445 /// ```
1446 /// use std::collections::VecDeque;
1447 ///
1448 /// let mut buf = VecDeque::new();
1449 /// buf.push_back(1);
1450 /// buf.push_back(3);
1451 /// assert_eq!(3, *buf.back().unwrap());
1452 /// ```
1453 #[stable(feature = "rust1", since = "1.0.0")]
1454 pub fn push_back(&mut self, value: T) {
1455 if self.is_full() {
1456 self.grow();
1457 }
1458
1459 let head = self.head;
1460 self.head = self.wrap_add(self.head, 1);
1461 unsafe { self.buffer_write(head, value) }
1462 }
1463
1464 #[inline]
1465 fn is_contiguous(&self) -> bool {
1466 // FIXME: Should we consider `head == 0` to mean
1467 // that `self` is contiguous?
1468 self.tail <= self.head
1469 }
1470
1471 /// Removes an element from anywhere in the `VecDeque` and returns it,
1472 /// replacing it with the first element.
1473 ///
1474 /// This does not preserve ordering, but is *O*(1).
1475 ///
1476 /// Returns `None` if `index` is out of bounds.
1477 ///
1478 /// Element at index 0 is the front of the queue.
1479 ///
1480 /// # Examples
1481 ///
1482 /// ```
1483 /// use std::collections::VecDeque;
1484 ///
1485 /// let mut buf = VecDeque::new();
1486 /// assert_eq!(buf.swap_remove_front(0), None);
1487 /// buf.push_back(1);
1488 /// buf.push_back(2);
1489 /// buf.push_back(3);
1490 /// assert_eq!(buf, [1, 2, 3]);
1491 ///
1492 /// assert_eq!(buf.swap_remove_front(2), Some(3));
1493 /// assert_eq!(buf, [2, 1]);
1494 /// ```
1495 #[stable(feature = "deque_extras_15", since = "1.5.0")]
1496 pub fn swap_remove_front(&mut self, index: usize) -> Option<T> {
1497 let length = self.len();
1498 if length > 0 && index < length && index != 0 {
1499 self.swap(index, 0);
1500 } else if index >= length {
1501 return None;
1502 }
1503 self.pop_front()
1504 }
1505
1506 /// Removes an element from anywhere in the `VecDeque` and returns it, replacing it with the
1507 /// last element.
1508 ///
1509 /// This does not preserve ordering, but is *O*(1).
1510 ///
1511 /// Returns `None` if `index` is out of bounds.
1512 ///
1513 /// Element at index 0 is the front of the queue.
1514 ///
1515 /// # Examples
1516 ///
1517 /// ```
1518 /// use std::collections::VecDeque;
1519 ///
1520 /// let mut buf = VecDeque::new();
1521 /// assert_eq!(buf.swap_remove_back(0), None);
1522 /// buf.push_back(1);
1523 /// buf.push_back(2);
1524 /// buf.push_back(3);
1525 /// assert_eq!(buf, [1, 2, 3]);
1526 ///
1527 /// assert_eq!(buf.swap_remove_back(0), Some(1));
1528 /// assert_eq!(buf, [3, 2]);
1529 /// ```
1530 #[stable(feature = "deque_extras_15", since = "1.5.0")]
1531 pub fn swap_remove_back(&mut self, index: usize) -> Option<T> {
1532 let length = self.len();
1533 if length > 0 && index < length - 1 {
1534 self.swap(index, length - 1);
1535 } else if index >= length {
1536 return None;
1537 }
1538 self.pop_back()
1539 }
1540
1541 /// Inserts an element at `index` within the `VecDeque`, shifting all elements with indices
1542 /// greater than or equal to `index` towards the back.
1543 ///
1544 /// Element at index 0 is the front of the queue.
1545 ///
1546 /// # Panics
1547 ///
1548 /// Panics if `index` is greater than `VecDeque`'s length
1549 ///
1550 /// # Examples
1551 ///
1552 /// ```
1553 /// use std::collections::VecDeque;
1554 ///
1555 /// let mut vec_deque = VecDeque::new();
1556 /// vec_deque.push_back('a');
1557 /// vec_deque.push_back('b');
1558 /// vec_deque.push_back('c');
1559 /// assert_eq!(vec_deque, &['a', 'b', 'c']);
1560 ///
1561 /// vec_deque.insert(1, 'd');
1562 /// assert_eq!(vec_deque, &['a', 'd', 'b', 'c']);
1563 /// ```
1564 #[stable(feature = "deque_extras_15", since = "1.5.0")]
1565 pub fn insert(&mut self, index: usize, value: T) {
1566 assert!(index <= self.len(), "index out of bounds");
1567 if self.is_full() {
1568 self.grow();
1569 }
1570
1571 // Move the least number of elements in the ring buffer and insert
1572 // the given object
1573 //
1574 // At most len/2 - 1 elements will be moved. O(min(n, n-i))
1575 //
1576 // There are three main cases:
1577 // Elements are contiguous
1578 // - special case when tail is 0
1579 // Elements are discontiguous and the insert is in the tail section
1580 // Elements are discontiguous and the insert is in the head section
1581 //
1582 // For each of those there are two more cases:
1583 // Insert is closer to tail
1584 // Insert is closer to head
1585 //
1586 // Key: H - self.head
1587 // T - self.tail
1588 // o - Valid element
1589 // I - Insertion element
1590 // A - The element that should be after the insertion point
1591 // M - Indicates element was moved
1592
1593 let idx = self.wrap_add(self.tail, index);
1594
1595 let distance_to_tail = index;
1596 let distance_to_head = self.len() - index;
1597
1598 let contiguous = self.is_contiguous();
1599
1600 match (contiguous, distance_to_tail <= distance_to_head, idx >= self.tail) {
1601 (true, true, _) if index == 0 => {
1602 // push_front
1603 //
1604 // T
1605 // I H
1606 // [A o o o o o o . . . . . . . . .]
1607 //
1608 // H T
1609 // [A o o o o o o o . . . . . I]
1610 //
1611
1612 self.tail = self.wrap_sub(self.tail, 1);
1613 }
1614 (true, true, _) => {
1615 unsafe {
1616 // contiguous, insert closer to tail:
1617 //
1618 // T I H
1619 // [. . . o o A o o o o . . . . . .]
1620 //
1621 // T H
1622 // [. . o o I A o o o o . . . . . .]
1623 // M M
1624 //
1625 // contiguous, insert closer to tail and tail is 0:
1626 //
1627 //
1628 // T I H
1629 // [o o A o o o o . . . . . . . . .]
1630 //
1631 // H T
1632 // [o I A o o o o o . . . . . . . o]
1633 // M M
1634
1635 let new_tail = self.wrap_sub(self.tail, 1);
1636
1637 self.copy(new_tail, self.tail, 1);
1638 // Already moved the tail, so we only copy `index - 1` elements.
1639 self.copy(self.tail, self.tail + 1, index - 1);
1640
1641 self.tail = new_tail;
1642 }
1643 }
1644 (true, false, _) => {
1645 unsafe {
1646 // contiguous, insert closer to head:
1647 //
1648 // T I H
1649 // [. . . o o o o A o o . . . . . .]
1650 //
1651 // T H
1652 // [. . . o o o o I A o o . . . . .]
1653 // M M M
1654
1655 self.copy(idx + 1, idx, self.head - idx);
1656 self.head = self.wrap_add(self.head, 1);
1657 }
1658 }
1659 (false, true, true) => {
1660 unsafe {
1661 // discontiguous, insert closer to tail, tail section:
1662 //
1663 // H T I
1664 // [o o o o o o . . . . . o o A o o]
1665 //
1666 // H T
1667 // [o o o o o o . . . . o o I A o o]
1668 // M M
1669
1670 self.copy(self.tail - 1, self.tail, index);
1671 self.tail -= 1;
1672 }
1673 }
1674 (false, false, true) => {
1675 unsafe {
1676 // discontiguous, insert closer to head, tail section:
1677 //
1678 // H T I
1679 // [o o . . . . . . . o o o o o A o]
1680 //
1681 // H T
1682 // [o o o . . . . . . o o o o o I A]
1683 // M M M M
1684
1685 // copy elements up to new head
1686 self.copy(1, 0, self.head);
1687
1688 // copy last element into empty spot at bottom of buffer
1689 self.copy(0, self.cap() - 1, 1);
1690
1691 // move elements from idx to end forward not including ^ element
1692 self.copy(idx + 1, idx, self.cap() - 1 - idx);
1693
1694 self.head += 1;
1695 }
1696 }
1697 (false, true, false) if idx == 0 => {
1698 unsafe {
1699 // discontiguous, insert is closer to tail, head section,
1700 // and is at index zero in the internal buffer:
1701 //
1702 // I H T
1703 // [A o o o o o o o o o . . . o o o]
1704 //
1705 // H T
1706 // [A o o o o o o o o o . . o o o I]
1707 // M M M
1708
1709 // copy elements up to new tail
1710 self.copy(self.tail - 1, self.tail, self.cap() - self.tail);
1711
1712 // copy last element into empty spot at bottom of buffer
1713 self.copy(self.cap() - 1, 0, 1);
1714
1715 self.tail -= 1;
1716 }
1717 }
1718 (false, true, false) => {
1719 unsafe {
1720 // discontiguous, insert closer to tail, head section:
1721 //
1722 // I H T
1723 // [o o o A o o o o o o . . . o o o]
1724 //
1725 // H T
1726 // [o o I A o o o o o o . . o o o o]
1727 // M M M M M M
1728
1729 // copy elements up to new tail
1730 self.copy(self.tail - 1, self.tail, self.cap() - self.tail);
1731
1732 // copy last element into empty spot at bottom of buffer
1733 self.copy(self.cap() - 1, 0, 1);
1734
1735 // move elements from idx-1 to end forward not including ^ element
1736 self.copy(0, 1, idx - 1);
1737
1738 self.tail -= 1;
1739 }
1740 }
1741 (false, false, false) => {
1742 unsafe {
1743 // discontiguous, insert closer to head, head section:
1744 //
1745 // I H T
1746 // [o o o o A o o . . . . . . o o o]
1747 //
1748 // H T
1749 // [o o o o I A o o . . . . . o o o]
1750 // M M M
1751
1752 self.copy(idx + 1, idx, self.head - idx);
1753 self.head += 1;
1754 }
1755 }
1756 }
1757
1758 // tail might've been changed so we need to recalculate
1759 let new_idx = self.wrap_add(self.tail, index);
1760 unsafe {
1761 self.buffer_write(new_idx, value);
1762 }
1763 }
1764
1765 /// Removes and returns the element at `index` from the `VecDeque`.
1766 /// Whichever end is closer to the removal point will be moved to make
1767 /// room, and all the affected elements will be moved to new positions.
1768 /// Returns `None` if `index` is out of bounds.
1769 ///
1770 /// Element at index 0 is the front of the queue.
1771 ///
1772 /// # Examples
1773 ///
1774 /// ```
1775 /// use std::collections::VecDeque;
1776 ///
1777 /// let mut buf = VecDeque::new();
1778 /// buf.push_back(1);
1779 /// buf.push_back(2);
1780 /// buf.push_back(3);
1781 /// assert_eq!(buf, [1, 2, 3]);
1782 ///
1783 /// assert_eq!(buf.remove(1), Some(2));
1784 /// assert_eq!(buf, [1, 3]);
1785 /// ```
1786 #[stable(feature = "rust1", since = "1.0.0")]
1787 pub fn remove(&mut self, index: usize) -> Option<T> {
1788 if self.is_empty() || self.len() <= index {
1789 return None;
1790 }
1791
1792 // There are three main cases:
1793 // Elements are contiguous
1794 // Elements are discontiguous and the removal is in the tail section
1795 // Elements are discontiguous and the removal is in the head section
1796 // - special case when elements are technically contiguous,
1797 // but self.head = 0
1798 //
1799 // For each of those there are two more cases:
1800 // Insert is closer to tail
1801 // Insert is closer to head
1802 //
1803 // Key: H - self.head
1804 // T - self.tail
1805 // o - Valid element
1806 // x - Element marked for removal
1807 // R - Indicates element that is being removed
1808 // M - Indicates element was moved
1809
1810 let idx = self.wrap_add(self.tail, index);
1811
1812 let elem = unsafe { Some(self.buffer_read(idx)) };
1813
1814 let distance_to_tail = index;
1815 let distance_to_head = self.len() - index;
1816
1817 let contiguous = self.is_contiguous();
1818
1819 match (contiguous, distance_to_tail <= distance_to_head, idx >= self.tail) {
1820 (true, true, _) => {
1821 unsafe {
1822 // contiguous, remove closer to tail:
1823 //
1824 // T R H
1825 // [. . . o o x o o o o . . . . . .]
1826 //
1827 // T H
1828 // [. . . . o o o o o o . . . . . .]
1829 // M M
1830
1831 self.copy(self.tail + 1, self.tail, index);
1832 self.tail += 1;
1833 }
1834 }
1835 (true, false, _) => {
1836 unsafe {
1837 // contiguous, remove closer to head:
1838 //
1839 // T R H
1840 // [. . . o o o o x o o . . . . . .]
1841 //
1842 // T H
1843 // [. . . o o o o o o . . . . . . .]
1844 // M M
1845
1846 self.copy(idx, idx + 1, self.head - idx - 1);
1847 self.head -= 1;
1848 }
1849 }
1850 (false, true, true) => {
1851 unsafe {
1852 // discontiguous, remove closer to tail, tail section:
1853 //
1854 // H T R
1855 // [o o o o o o . . . . . o o x o o]
1856 //
1857 // H T
1858 // [o o o o o o . . . . . . o o o o]
1859 // M M
1860
1861 self.copy(self.tail + 1, self.tail, index);
1862 self.tail = self.wrap_add(self.tail, 1);
1863 }
1864 }
1865 (false, false, false) => {
1866 unsafe {
1867 // discontiguous, remove closer to head, head section:
1868 //
1869 // R H T
1870 // [o o o o x o o . . . . . . o o o]
1871 //
1872 // H T
1873 // [o o o o o o . . . . . . . o o o]
1874 // M M
1875
1876 self.copy(idx, idx + 1, self.head - idx - 1);
1877 self.head -= 1;
1878 }
1879 }
1880 (false, false, true) => {
1881 unsafe {
1882 // discontiguous, remove closer to head, tail section:
1883 //
1884 // H T R
1885 // [o o o . . . . . . o o o o o x o]
1886 //
1887 // H T
1888 // [o o . . . . . . . o o o o o o o]
1889 // M M M M
1890 //
1891 // or quasi-discontiguous, remove next to head, tail section:
1892 //
1893 // H T R
1894 // [. . . . . . . . . o o o o o x o]
1895 //
1896 // T H
1897 // [. . . . . . . . . o o o o o o .]
1898 // M
1899
1900 // draw in elements in the tail section
1901 self.copy(idx, idx + 1, self.cap() - idx - 1);
1902
1903 // Prevents underflow.
1904 if self.head != 0 {
1905 // copy first element into empty spot
1906 self.copy(self.cap() - 1, 0, 1);
1907
1908 // move elements in the head section backwards
1909 self.copy(0, 1, self.head - 1);
1910 }
1911
1912 self.head = self.wrap_sub(self.head, 1);
1913 }
1914 }
1915 (false, true, false) => {
1916 unsafe {
1917 // discontiguous, remove closer to tail, head section:
1918 //
1919 // R H T
1920 // [o o x o o o o o o o . . . o o o]
1921 //
1922 // H T
1923 // [o o o o o o o o o o . . . . o o]
1924 // M M M M M
1925
1926 // draw in elements up to idx
1927 self.copy(1, 0, idx);
1928
1929 // copy last element into empty spot
1930 self.copy(0, self.cap() - 1, 1);
1931
1932 // move elements from tail to end forward, excluding the last one
1933 self.copy(self.tail + 1, self.tail, self.cap() - self.tail - 1);
1934
1935 self.tail = self.wrap_add(self.tail, 1);
1936 }
1937 }
1938 }
1939
1940 elem
1941 }
1942
1943 /// Splits the `VecDeque` into two at the given index.
1944 ///
1945 /// Returns a newly allocated `VecDeque`. `self` contains elements `[0, at)`,
1946 /// and the returned `VecDeque` contains elements `[at, len)`.
1947 ///
1948 /// Note that the capacity of `self` does not change.
1949 ///
1950 /// Element at index 0 is the front of the queue.
1951 ///
1952 /// # Panics
1953 ///
1954 /// Panics if `at > len`.
1955 ///
1956 /// # Examples
1957 ///
1958 /// ```
1959 /// use std::collections::VecDeque;
1960 ///
1961 /// let mut buf: VecDeque<_> = vec![1, 2, 3].into_iter().collect();
1962 /// let buf2 = buf.split_off(1);
1963 /// assert_eq!(buf, [1]);
1964 /// assert_eq!(buf2, [2, 3]);
1965 /// ```
1966 #[inline]
1967 #[must_use = "use `.truncate()` if you don't need the other half"]
1968 #[stable(feature = "split_off", since = "1.4.0")]
1969 pub fn split_off(&mut self, at: usize) -> Self {
1970 let len = self.len();
1971 assert!(at <= len, "`at` out of bounds");
1972
1973 let other_len = len - at;
1974 let mut other = VecDeque::with_capacity(other_len);
1975
1976 unsafe {
1977 let (first_half, second_half) = self.as_slices();
1978
1979 let first_len = first_half.len();
1980 let second_len = second_half.len();
1981 if at < first_len {
1982 // `at` lies in the first half.
1983 let amount_in_first = first_len - at;
1984
1985 ptr::copy_nonoverlapping(first_half.as_ptr().add(at), other.ptr(), amount_in_first);
1986
1987 // just take all of the second half.
1988 ptr::copy_nonoverlapping(
1989 second_half.as_ptr(),
1990 other.ptr().add(amount_in_first),
1991 second_len,
1992 );
1993 } else {
1994 // `at` lies in the second half, need to factor in the elements we skipped
1995 // in the first half.
1996 let offset = at - first_len;
1997 let amount_in_second = second_len - offset;
1998 ptr::copy_nonoverlapping(
1999 second_half.as_ptr().add(offset),
2000 other.ptr(),
2001 amount_in_second,
2002 );
2003 }
2004 }
2005
2006 // Cleanup where the ends of the buffers are
2007 self.head = self.wrap_sub(self.head, other_len);
2008 other.head = other.wrap_index(other_len);
2009
2010 other
2011 }
2012
2013 /// Moves all the elements of `other` into `self`, leaving `other` empty.
2014 ///
2015 /// # Panics
2016 ///
2017 /// Panics if the new number of elements in self overflows a `usize`.
2018 ///
2019 /// # Examples
2020 ///
2021 /// ```
2022 /// use std::collections::VecDeque;
2023 ///
2024 /// let mut buf: VecDeque<_> = vec![1, 2].into_iter().collect();
2025 /// let mut buf2: VecDeque<_> = vec![3, 4].into_iter().collect();
2026 /// buf.append(&mut buf2);
2027 /// assert_eq!(buf, [1, 2, 3, 4]);
2028 /// assert_eq!(buf2, []);
2029 /// ```
2030 #[inline]
2031 #[stable(feature = "append", since = "1.4.0")]
2032 pub fn append(&mut self, other: &mut Self) {
2033 // naive impl
2034 self.extend(other.drain(..));
2035 }
2036
2037 /// Retains only the elements specified by the predicate.
2038 ///
2039 /// In other words, remove all elements `e` such that `f(&e)` returns false.
2040 /// This method operates in place, visiting each element exactly once in the
2041 /// original order, and preserves the order of the retained elements.
2042 ///
2043 /// # Examples
2044 ///
2045 /// ```
2046 /// use std::collections::VecDeque;
2047 ///
2048 /// let mut buf = VecDeque::new();
2049 /// buf.extend(1..5);
2050 /// buf.retain(|&x| x % 2 == 0);
2051 /// assert_eq!(buf, [2, 4]);
2052 /// ```
2053 ///
2054 /// The exact order may be useful for tracking external state, like an index.
2055 ///
2056 /// ```
2057 /// use std::collections::VecDeque;
2058 ///
2059 /// let mut buf = VecDeque::new();
2060 /// buf.extend(1..6);
2061 ///
2062 /// let keep = [false, true, true, false, true];
2063 /// let mut i = 0;
2064 /// buf.retain(|_| (keep[i], i += 1).0);
2065 /// assert_eq!(buf, [2, 3, 5]);
2066 /// ```
2067 #[stable(feature = "vec_deque_retain", since = "1.4.0")]
2068 pub fn retain<F>(&mut self, mut f: F)
2069 where
2070 F: FnMut(&T) -> bool,
2071 {
2072 let len = self.len();
2073 let mut del = 0;
2074 for i in 0..len {
2075 if !f(&self[i]) {
2076 del += 1;
2077 } else if del > 0 {
2078 self.swap(i - del, i);
2079 }
2080 }
2081 if del > 0 {
2082 self.truncate(len - del);
2083 }
2084 }
2085
2086 // This may panic or abort
2087 #[inline(never)]
2088 fn grow(&mut self) {
2089 if self.is_full() {
2090 let old_cap = self.cap();
2091 // Double the buffer size.
2092 self.buf.reserve_exact(old_cap, old_cap);
2093 assert!(self.cap() == old_cap * 2);
2094 unsafe {
2095 self.handle_capacity_increase(old_cap);
2096 }
2097 debug_assert!(!self.is_full());
2098 }
2099 }
2100
2101 /// Modifies the `VecDeque` in-place so that `len()` is equal to `new_len`,
2102 /// either by removing excess elements from the back or by appending
2103 /// elements generated by calling `generator` to the back.
2104 ///
2105 /// # Examples
2106 ///
2107 /// ```
2108 /// use std::collections::VecDeque;
2109 ///
2110 /// let mut buf = VecDeque::new();
2111 /// buf.push_back(5);
2112 /// buf.push_back(10);
2113 /// buf.push_back(15);
2114 /// assert_eq!(buf, [5, 10, 15]);
2115 ///
2116 /// buf.resize_with(5, Default::default);
2117 /// assert_eq!(buf, [5, 10, 15, 0, 0]);
2118 ///
2119 /// buf.resize_with(2, || unreachable!());
2120 /// assert_eq!(buf, [5, 10]);
2121 ///
2122 /// let mut state = 100;
2123 /// buf.resize_with(5, || { state += 1; state });
2124 /// assert_eq!(buf, [5, 10, 101, 102, 103]);
2125 /// ```
2126 #[stable(feature = "vec_resize_with", since = "1.33.0")]
2127 pub fn resize_with(&mut self, new_len: usize, generator: impl FnMut() -> T) {
2128 let len = self.len();
2129
2130 if new_len > len {
2131 self.extend(repeat_with(generator).take(new_len - len))
2132 } else {
2133 self.truncate(new_len);
2134 }
2135 }
2136
2137 /// Rearranges the internal storage of this deque so it is one contiguous
2138 /// slice, which is then returned.
2139 ///
2140 /// This method does not allocate and does not change the order of the
2141 /// inserted elements. As it returns a mutable slice, this can be used to
2142 /// sort a deque.
2143 ///
2144 /// Once the internal storage is contiguous, the [`as_slices`] and
2145 /// [`as_mut_slices`] methods will return the entire contents of the
2146 /// `VecDeque` in a single slice.
2147 ///
2148 /// [`as_slices`]: VecDeque::as_slices
2149 /// [`as_mut_slices`]: VecDeque::as_mut_slices
2150 ///
2151 /// # Examples
2152 ///
2153 /// Sorting the content of a deque.
2154 ///
2155 /// ```
2156 /// use std::collections::VecDeque;
2157 ///
2158 /// let mut buf = VecDeque::with_capacity(15);
2159 ///
2160 /// buf.push_back(2);
2161 /// buf.push_back(1);
2162 /// buf.push_front(3);
2163 ///
2164 /// // sorting the deque
2165 /// buf.make_contiguous().sort();
2166 /// assert_eq!(buf.as_slices(), (&[1, 2, 3] as &[_], &[] as &[_]));
2167 ///
2168 /// // sorting it in reverse order
2169 /// buf.make_contiguous().sort_by(|a, b| b.cmp(a));
2170 /// assert_eq!(buf.as_slices(), (&[3, 2, 1] as &[_], &[] as &[_]));
2171 /// ```
2172 ///
2173 /// Getting immutable access to the contiguous slice.
2174 ///
2175 /// ```rust
2176 /// use std::collections::VecDeque;
2177 ///
2178 /// let mut buf = VecDeque::new();
2179 ///
2180 /// buf.push_back(2);
2181 /// buf.push_back(1);
2182 /// buf.push_front(3);
2183 ///
2184 /// buf.make_contiguous();
2185 /// if let (slice, &[]) = buf.as_slices() {
2186 /// // we can now be sure that `slice` contains all elements of the deque,
2187 /// // while still having immutable access to `buf`.
2188 /// assert_eq!(buf.len(), slice.len());
2189 /// assert_eq!(slice, &[3, 2, 1] as &[_]);
2190 /// }
2191 /// ```
2192 #[stable(feature = "deque_make_contiguous", since = "1.48.0")]
2193 pub fn make_contiguous(&mut self) -> &mut [T] {
2194 if self.is_contiguous() {
2195 let tail = self.tail;
2196 let head = self.head;
2197 return unsafe { RingSlices::ring_slices(self.buffer_as_mut_slice(), head, tail).0 };
2198 }
2199
2200 let buf = self.buf.ptr();
2201 let cap = self.cap();
2202 let len = self.len();
2203
2204 let free = self.tail - self.head;
2205 let tail_len = cap - self.tail;
2206
2207 if free >= tail_len {
2208 // there is enough free space to copy the tail in one go,
2209 // this means that we first shift the head backwards, and then
2210 // copy the tail to the correct position.
2211 //
2212 // from: DEFGH....ABC
2213 // to: ABCDEFGH....
2214 unsafe {
2215 ptr::copy(buf, buf.add(tail_len), self.head);
2216 // ...DEFGH.ABC
2217 ptr::copy_nonoverlapping(buf.add(self.tail), buf, tail_len);
2218 // ABCDEFGH....
2219
2220 self.tail = 0;
2221 self.head = len;
2222 }
2223 } else if free > self.head {
2224 // FIXME: We currently do not consider ....ABCDEFGH
2225 // to be contiguous because `head` would be `0` in this
2226 // case. While we probably want to change this it
2227 // isn't trivial as a few places expect `is_contiguous`
2228 // to mean that we can just slice using `buf[tail..head]`.
2229
2230 // there is enough free space to copy the head in one go,
2231 // this means that we first shift the tail forwards, and then
2232 // copy the head to the correct position.
2233 //
2234 // from: FGH....ABCDE
2235 // to: ...ABCDEFGH.
2236 unsafe {
2237 ptr::copy(buf.add(self.tail), buf.add(self.head), tail_len);
2238 // FGHABCDE....
2239 ptr::copy_nonoverlapping(buf, buf.add(self.head + tail_len), self.head);
2240 // ...ABCDEFGH.
2241
2242 self.tail = self.head;
2243 self.head = self.wrap_add(self.tail, len);
2244 }
2245 } else {
2246 // free is smaller than both head and tail,
2247 // this means we have to slowly "swap" the tail and the head.
2248 //
2249 // from: EFGHI...ABCD or HIJK.ABCDEFG
2250 // to: ABCDEFGHI... or ABCDEFGHIJK.
2251 let mut left_edge: usize = 0;
2252 let mut right_edge: usize = self.tail;
2253 unsafe {
2254 // The general problem looks like this
2255 // GHIJKLM...ABCDEF - before any swaps
2256 // ABCDEFM...GHIJKL - after 1 pass of swaps
2257 // ABCDEFGHIJM...KL - swap until the left edge reaches the temp store
2258 // - then restart the algorithm with a new (smaller) store
2259 // Sometimes the temp store is reached when the right edge is at the end
2260 // of the buffer - this means we've hit the right order with fewer swaps!
2261 // E.g
2262 // EF..ABCD
2263 // ABCDEF.. - after four only swaps we've finished
2264 while left_edge < len && right_edge != cap {
2265 let mut right_offset = 0;
2266 for i in left_edge..right_edge {
2267 right_offset = (i - left_edge) % (cap - right_edge);
2268 let src: isize = (right_edge + right_offset) as isize;
2269 ptr::swap(buf.add(i), buf.offset(src));
2270 }
2271 let n_ops = right_edge - left_edge;
2272 left_edge += n_ops;
2273 right_edge += right_offset + 1;
2274 }
2275
2276 self.tail = 0;
2277 self.head = len;
2278 }
2279 }
2280
2281 let tail = self.tail;
2282 let head = self.head;
2283 unsafe { RingSlices::ring_slices(self.buffer_as_mut_slice(), head, tail).0 }
2284 }
2285
2286 /// Rotates the double-ended queue `mid` places to the left.
2287 ///
2288 /// Equivalently,
2289 /// - Rotates item `mid` into the first position.
2290 /// - Pops the first `mid` items and pushes them to the end.
2291 /// - Rotates `len() - mid` places to the right.
2292 ///
2293 /// # Panics
2294 ///
2295 /// If `mid` is greater than `len()`. Note that `mid == len()`
2296 /// does _not_ panic and is a no-op rotation.
2297 ///
2298 /// # Complexity
2299 ///
2300 /// Takes `*O*(min(mid, len() - mid))` time and no extra space.
2301 ///
2302 /// # Examples
2303 ///
2304 /// ```
2305 /// use std::collections::VecDeque;
2306 ///
2307 /// let mut buf: VecDeque<_> = (0..10).collect();
2308 ///
2309 /// buf.rotate_left(3);
2310 /// assert_eq!(buf, [3, 4, 5, 6, 7, 8, 9, 0, 1, 2]);
2311 ///
2312 /// for i in 1..10 {
2313 /// assert_eq!(i * 3 % 10, buf[0]);
2314 /// buf.rotate_left(3);
2315 /// }
2316 /// assert_eq!(buf, [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]);
2317 /// ```
2318 #[stable(feature = "vecdeque_rotate", since = "1.36.0")]
2319 pub fn rotate_left(&mut self, mid: usize) {
2320 assert!(mid <= self.len());
2321 let k = self.len() - mid;
2322 if mid <= k {
2323 unsafe { self.rotate_left_inner(mid) }
2324 } else {
2325 unsafe { self.rotate_right_inner(k) }
2326 }
2327 }
2328
2329 /// Rotates the double-ended queue `k` places to the right.
2330 ///
2331 /// Equivalently,
2332 /// - Rotates the first item into position `k`.
2333 /// - Pops the last `k` items and pushes them to the front.
2334 /// - Rotates `len() - k` places to the left.
2335 ///
2336 /// # Panics
2337 ///
2338 /// If `k` is greater than `len()`. Note that `k == len()`
2339 /// does _not_ panic and is a no-op rotation.
2340 ///
2341 /// # Complexity
2342 ///
2343 /// Takes `*O*(min(k, len() - k))` time and no extra space.
2344 ///
2345 /// # Examples
2346 ///
2347 /// ```
2348 /// use std::collections::VecDeque;
2349 ///
2350 /// let mut buf: VecDeque<_> = (0..10).collect();
2351 ///
2352 /// buf.rotate_right(3);
2353 /// assert_eq!(buf, [7, 8, 9, 0, 1, 2, 3, 4, 5, 6]);
2354 ///
2355 /// for i in 1..10 {
2356 /// assert_eq!(0, buf[i * 3 % 10]);
2357 /// buf.rotate_right(3);
2358 /// }
2359 /// assert_eq!(buf, [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]);
2360 /// ```
2361 #[stable(feature = "vecdeque_rotate", since = "1.36.0")]
2362 pub fn rotate_right(&mut self, k: usize) {
2363 assert!(k <= self.len());
2364 let mid = self.len() - k;
2365 if k <= mid {
2366 unsafe { self.rotate_right_inner(k) }
2367 } else {
2368 unsafe { self.rotate_left_inner(mid) }
2369 }
2370 }
2371
2372 // SAFETY: the following two methods require that the rotation amount
2373 // be less than half the length of the deque.
2374 //
2375 // `wrap_copy` requires that `min(x, cap() - x) + copy_len <= cap()`,
2376 // but than `min` is never more than half the capacity, regardless of x,
2377 // so it's sound to call here because we're calling with something
2378 // less than half the length, which is never above half the capacity.
2379
2380 unsafe fn rotate_left_inner(&mut self, mid: usize) {
2381 debug_assert!(mid * 2 <= self.len());
2382 unsafe {
2383 self.wrap_copy(self.head, self.tail, mid);
2384 }
2385 self.head = self.wrap_add(self.head, mid);
2386 self.tail = self.wrap_add(self.tail, mid);
2387 }
2388
2389 unsafe fn rotate_right_inner(&mut self, k: usize) {
2390 debug_assert!(k * 2 <= self.len());
2391 self.head = self.wrap_sub(self.head, k);
2392 self.tail = self.wrap_sub(self.tail, k);
2393 unsafe {
2394 self.wrap_copy(self.tail, self.head, k);
2395 }
2396 }
2397
2398 /// Binary searches this sorted `VecDeque` for a given element.
2399 ///
2400 /// If the value is found then [`Result::Ok`] is returned, containing the
2401 /// index of the matching element. If there are multiple matches, then any
2402 /// one of the matches could be returned. If the value is not found then
2403 /// [`Result::Err`] is returned, containing the index where a matching
2404 /// element could be inserted while maintaining sorted order.
2405 ///
2406 /// See also [`binary_search_by`], [`binary_search_by_key`], and [`partition_point`].
2407 ///
2408 /// [`binary_search_by`]: VecDeque::binary_search_by
2409 /// [`binary_search_by_key`]: VecDeque::binary_search_by_key
2410 /// [`partition_point`]: VecDeque::partition_point
2411 ///
2412 /// # Examples
2413 ///
2414 /// Looks up a series of four elements. The first is found, with a
2415 /// uniquely determined position; the second and third are not
2416 /// found; the fourth could match any position in `[1, 4]`.
2417 ///
2418 /// ```
2419 /// use std::collections::VecDeque;
2420 ///
2421 /// let deque: VecDeque<_> = vec![0, 1, 1, 1, 1, 2, 3, 5, 8, 13, 21, 34, 55].into();
2422 ///
2423 /// assert_eq!(deque.binary_search(&13), Ok(9));
2424 /// assert_eq!(deque.binary_search(&4), Err(7));
2425 /// assert_eq!(deque.binary_search(&100), Err(13));
2426 /// let r = deque.binary_search(&1);
2427 /// assert!(matches!(r, Ok(1..=4)));
2428 /// ```
2429 ///
2430 /// If you want to insert an item to a sorted `VecDeque`, while maintaining
2431 /// sort order:
2432 ///
2433 /// ```
2434 /// use std::collections::VecDeque;
2435 ///
2436 /// let mut deque: VecDeque<_> = vec![0, 1, 1, 1, 1, 2, 3, 5, 8, 13, 21, 34, 55].into();
2437 /// let num = 42;
2438 /// let idx = deque.binary_search(&num).unwrap_or_else(|x| x);
2439 /// deque.insert(idx, num);
2440 /// assert_eq!(deque, &[0, 1, 1, 1, 1, 2, 3, 5, 8, 13, 21, 34, 42, 55]);
2441 /// ```
2442 #[stable(feature = "vecdeque_binary_search", since = "1.54.0")]
2443 #[inline]
2444 pub fn binary_search(&self, x: &T) -> Result<usize, usize>
2445 where
2446 T: Ord,
2447 {
2448 self.binary_search_by(|e| e.cmp(x))
2449 }
2450
2451 /// Binary searches this sorted `VecDeque` with a comparator function.
2452 ///
2453 /// The comparator function should implement an order consistent
2454 /// with the sort order of the underlying `VecDeque`, returning an
2455 /// order code that indicates whether its argument is `Less`,
2456 /// `Equal` or `Greater` than the desired target.
2457 ///
2458 /// If the value is found then [`Result::Ok`] is returned, containing the
2459 /// index of the matching element. If there are multiple matches, then any
2460 /// one of the matches could be returned. If the value is not found then
2461 /// [`Result::Err`] is returned, containing the index where a matching
2462 /// element could be inserted while maintaining sorted order.
2463 ///
2464 /// See also [`binary_search`], [`binary_search_by_key`], and [`partition_point`].
2465 ///
2466 /// [`binary_search`]: VecDeque::binary_search
2467 /// [`binary_search_by_key`]: VecDeque::binary_search_by_key
2468 /// [`partition_point`]: VecDeque::partition_point
2469 ///
2470 /// # Examples
2471 ///
2472 /// Looks up a series of four elements. The first is found, with a
2473 /// uniquely determined position; the second and third are not
2474 /// found; the fourth could match any position in `[1, 4]`.
2475 ///
2476 /// ```
2477 /// use std::collections::VecDeque;
2478 ///
2479 /// let deque: VecDeque<_> = vec![0, 1, 1, 1, 1, 2, 3, 5, 8, 13, 21, 34, 55].into();
2480 ///
2481 /// assert_eq!(deque.binary_search_by(|x| x.cmp(&13)), Ok(9));
2482 /// assert_eq!(deque.binary_search_by(|x| x.cmp(&4)), Err(7));
2483 /// assert_eq!(deque.binary_search_by(|x| x.cmp(&100)), Err(13));
2484 /// let r = deque.binary_search_by(|x| x.cmp(&1));
2485 /// assert!(matches!(r, Ok(1..=4)));
2486 /// ```
2487 #[stable(feature = "vecdeque_binary_search", since = "1.54.0")]
2488 pub fn binary_search_by<'a, F>(&'a self, mut f: F) -> Result<usize, usize>
2489 where
2490 F: FnMut(&'a T) -> Ordering,
2491 {
2492 let (front, back) = self.as_slices();
2493 let cmp_back = back.first().map(|elem| f(elem));
2494
2495 if let Some(Ordering::Equal) = cmp_back {
2496 Ok(front.len())
2497 } else if let Some(Ordering::Less) = cmp_back {
2498 back.binary_search_by(f).map(|idx| idx + front.len()).map_err(|idx| idx + front.len())
2499 } else {
2500 front.binary_search_by(f)
2501 }
2502 }
2503
2504 /// Binary searches this sorted `VecDeque` with a key extraction function.
2505 ///
2506 /// Assumes that the `VecDeque` is sorted by the key, for instance with
2507 /// [`make_contiguous().sort_by_key()`] using the same key extraction function.
2508 ///
2509 /// If the value is found then [`Result::Ok`] is returned, containing the
2510 /// index of the matching element. If there are multiple matches, then any
2511 /// one of the matches could be returned. If the value is not found then
2512 /// [`Result::Err`] is returned, containing the index where a matching
2513 /// element could be inserted while maintaining sorted order.
2514 ///
2515 /// See also [`binary_search`], [`binary_search_by`], and [`partition_point`].
2516 ///
2517 /// [`make_contiguous().sort_by_key()`]: VecDeque::make_contiguous
2518 /// [`binary_search`]: VecDeque::binary_search
2519 /// [`binary_search_by`]: VecDeque::binary_search_by
2520 /// [`partition_point`]: VecDeque::partition_point
2521 ///
2522 /// # Examples
2523 ///
2524 /// Looks up a series of four elements in a slice of pairs sorted by
2525 /// their second elements. The first is found, with a uniquely
2526 /// determined position; the second and third are not found; the
2527 /// fourth could match any position in `[1, 4]`.
2528 ///
2529 /// ```
2530 /// use std::collections::VecDeque;
2531 ///
2532 /// let deque: VecDeque<_> = vec![(0, 0), (2, 1), (4, 1), (5, 1),
2533 /// (3, 1), (1, 2), (2, 3), (4, 5), (5, 8), (3, 13),
2534 /// (1, 21), (2, 34), (4, 55)].into();
2535 ///
2536 /// assert_eq!(deque.binary_search_by_key(&13, |&(a, b)| b), Ok(9));
2537 /// assert_eq!(deque.binary_search_by_key(&4, |&(a, b)| b), Err(7));
2538 /// assert_eq!(deque.binary_search_by_key(&100, |&(a, b)| b), Err(13));
2539 /// let r = deque.binary_search_by_key(&1, |&(a, b)| b);
2540 /// assert!(matches!(r, Ok(1..=4)));
2541 /// ```
2542 #[stable(feature = "vecdeque_binary_search", since = "1.54.0")]
2543 #[inline]
2544 pub fn binary_search_by_key<'a, B, F>(&'a self, b: &B, mut f: F) -> Result<usize, usize>
2545 where
2546 F: FnMut(&'a T) -> B,
2547 B: Ord,
2548 {
2549 self.binary_search_by(|k| f(k).cmp(b))
2550 }
2551
2552 /// Returns the index of the partition point according to the given predicate
2553 /// (the index of the first element of the second partition).
2554 ///
2555 /// The deque is assumed to be partitioned according to the given predicate.
2556 /// This means that all elements for which the predicate returns true are at the start of the deque
2557 /// and all elements for which the predicate returns false are at the end.
2558 /// For example, [7, 15, 3, 5, 4, 12, 6] is a partitioned under the predicate x % 2 != 0
2559 /// (all odd numbers are at the start, all even at the end).
2560 ///
2561 /// If this deque is not partitioned, the returned result is unspecified and meaningless,
2562 /// as this method performs a kind of binary search.
2563 ///
2564 /// See also [`binary_search`], [`binary_search_by`], and [`binary_search_by_key`].
2565 ///
2566 /// [`binary_search`]: VecDeque::binary_search
2567 /// [`binary_search_by`]: VecDeque::binary_search_by
2568 /// [`binary_search_by_key`]: VecDeque::binary_search_by_key
2569 ///
2570 /// # Examples
2571 ///
2572 /// ```
2573 /// use std::collections::VecDeque;
2574 ///
2575 /// let deque: VecDeque<_> = vec![1, 2, 3, 3, 5, 6, 7].into();
2576 /// let i = deque.partition_point(|&x| x < 5);
2577 ///
2578 /// assert_eq!(i, 4);
2579 /// assert!(deque.iter().take(i).all(|&x| x < 5));
2580 /// assert!(deque.iter().skip(i).all(|&x| !(x < 5)));
2581 /// ```
2582 #[stable(feature = "vecdeque_binary_search", since = "1.54.0")]
2583 pub fn partition_point<P>(&self, mut pred: P) -> usize
2584 where
2585 P: FnMut(&T) -> bool,
2586 {
2587 let (front, back) = self.as_slices();
2588
2589 if let Some(true) = back.first().map(|v| pred(v)) {
2590 back.partition_point(pred) + front.len()
2591 } else {
2592 front.partition_point(pred)
2593 }
2594 }
2595 }
2596
2597 impl<T: Clone> VecDeque<T> {
2598 /// Modifies the `VecDeque` in-place so that `len()` is equal to new_len,
2599 /// either by removing excess elements from the back or by appending clones of `value`
2600 /// to the back.
2601 ///
2602 /// # Examples
2603 ///
2604 /// ```
2605 /// use std::collections::VecDeque;
2606 ///
2607 /// let mut buf = VecDeque::new();
2608 /// buf.push_back(5);
2609 /// buf.push_back(10);
2610 /// buf.push_back(15);
2611 /// assert_eq!(buf, [5, 10, 15]);
2612 ///
2613 /// buf.resize(2, 0);
2614 /// assert_eq!(buf, [5, 10]);
2615 ///
2616 /// buf.resize(5, 20);
2617 /// assert_eq!(buf, [5, 10, 20, 20, 20]);
2618 /// ```
2619 #[stable(feature = "deque_extras", since = "1.16.0")]
2620 pub fn resize(&mut self, new_len: usize, value: T) {
2621 self.resize_with(new_len, || value.clone());
2622 }
2623 }
2624
2625 /// Returns the index in the underlying buffer for a given logical element index.
2626 #[inline]
2627 fn wrap_index(index: usize, size: usize) -> usize {
2628 // size is always a power of 2
2629 debug_assert!(size.is_power_of_two());
2630 index & (size - 1)
2631 }
2632
2633 /// Calculate the number of elements left to be read in the buffer
2634 #[inline]
2635 fn count(tail: usize, head: usize, size: usize) -> usize {
2636 // size is always a power of 2
2637 (head.wrapping_sub(tail)) & (size - 1)
2638 }
2639
2640 #[stable(feature = "rust1", since = "1.0.0")]
2641 impl<A: PartialEq> PartialEq for VecDeque<A> {
2642 fn eq(&self, other: &VecDeque<A>) -> bool {
2643 if self.len() != other.len() {
2644 return false;
2645 }
2646 let (sa, sb) = self.as_slices();
2647 let (oa, ob) = other.as_slices();
2648 if sa.len() == oa.len() {
2649 sa == oa && sb == ob
2650 } else if sa.len() < oa.len() {
2651 // Always divisible in three sections, for example:
2652 // self: [a b c|d e f]
2653 // other: [0 1 2 3|4 5]
2654 // front = 3, mid = 1,
2655 // [a b c] == [0 1 2] && [d] == [3] && [e f] == [4 5]
2656 let front = sa.len();
2657 let mid = oa.len() - front;
2658
2659 let (oa_front, oa_mid) = oa.split_at(front);
2660 let (sb_mid, sb_back) = sb.split_at(mid);
2661 debug_assert_eq!(sa.len(), oa_front.len());
2662 debug_assert_eq!(sb_mid.len(), oa_mid.len());
2663 debug_assert_eq!(sb_back.len(), ob.len());
2664 sa == oa_front && sb_mid == oa_mid && sb_back == ob
2665 } else {
2666 let front = oa.len();
2667 let mid = sa.len() - front;
2668
2669 let (sa_front, sa_mid) = sa.split_at(front);
2670 let (ob_mid, ob_back) = ob.split_at(mid);
2671 debug_assert_eq!(sa_front.len(), oa.len());
2672 debug_assert_eq!(sa_mid.len(), ob_mid.len());
2673 debug_assert_eq!(sb.len(), ob_back.len());
2674 sa_front == oa && sa_mid == ob_mid && sb == ob_back
2675 }
2676 }
2677 }
2678
2679 #[stable(feature = "rust1", since = "1.0.0")]
2680 impl<A: Eq> Eq for VecDeque<A> {}
2681
2682 __impl_slice_eq1! { [] VecDeque<A>, Vec<B>, }
2683 __impl_slice_eq1! { [] VecDeque<A>, &[B], }
2684 __impl_slice_eq1! { [] VecDeque<A>, &mut [B], }
2685 __impl_slice_eq1! { [const N: usize] VecDeque<A>, [B; N], }
2686 __impl_slice_eq1! { [const N: usize] VecDeque<A>, &[B; N], }
2687 __impl_slice_eq1! { [const N: usize] VecDeque<A>, &mut [B; N], }
2688
2689 #[stable(feature = "rust1", since = "1.0.0")]
2690 impl<A: PartialOrd> PartialOrd for VecDeque<A> {
2691 fn partial_cmp(&self, other: &VecDeque<A>) -> Option<Ordering> {
2692 self.iter().partial_cmp(other.iter())
2693 }
2694 }
2695
2696 #[stable(feature = "rust1", since = "1.0.0")]
2697 impl<A: Ord> Ord for VecDeque<A> {
2698 #[inline]
2699 fn cmp(&self, other: &VecDeque<A>) -> Ordering {
2700 self.iter().cmp(other.iter())
2701 }
2702 }
2703
2704 #[stable(feature = "rust1", since = "1.0.0")]
2705 impl<A: Hash> Hash for VecDeque<A> {
2706 fn hash<H: Hasher>(&self, state: &mut H) {
2707 self.len().hash(state);
2708 // It's not possible to use Hash::hash_slice on slices
2709 // returned by as_slices method as their length can vary
2710 // in otherwise identical deques.
2711 //
2712 // Hasher only guarantees equivalence for the exact same
2713 // set of calls to its methods.
2714 self.iter().for_each(|elem| elem.hash(state));
2715 }
2716 }
2717
2718 #[stable(feature = "rust1", since = "1.0.0")]
2719 impl<A> Index<usize> for VecDeque<A> {
2720 type Output = A;
2721
2722 #[inline]
2723 fn index(&self, index: usize) -> &A {
2724 self.get(index).expect("Out of bounds access")
2725 }
2726 }
2727
2728 #[stable(feature = "rust1", since = "1.0.0")]
2729 impl<A> IndexMut<usize> for VecDeque<A> {
2730 #[inline]
2731 fn index_mut(&mut self, index: usize) -> &mut A {
2732 self.get_mut(index).expect("Out of bounds access")
2733 }
2734 }
2735
2736 #[stable(feature = "rust1", since = "1.0.0")]
2737 impl<A> FromIterator<A> for VecDeque<A> {
2738 fn from_iter<T: IntoIterator<Item = A>>(iter: T) -> VecDeque<A> {
2739 let iterator = iter.into_iter();
2740 let (lower, _) = iterator.size_hint();
2741 let mut deq = VecDeque::with_capacity(lower);
2742 deq.extend(iterator);
2743 deq
2744 }
2745 }
2746
2747 #[stable(feature = "rust1", since = "1.0.0")]
2748 impl<T> IntoIterator for VecDeque<T> {
2749 type Item = T;
2750 type IntoIter = IntoIter<T>;
2751
2752 /// Consumes the `VecDeque` into a front-to-back iterator yielding elements by
2753 /// value.
2754 fn into_iter(self) -> IntoIter<T> {
2755 IntoIter { inner: self }
2756 }
2757 }
2758
2759 #[stable(feature = "rust1", since = "1.0.0")]
2760 impl<'a, T> IntoIterator for &'a VecDeque<T> {
2761 type Item = &'a T;
2762 type IntoIter = Iter<'a, T>;
2763
2764 fn into_iter(self) -> Iter<'a, T> {
2765 self.iter()
2766 }
2767 }
2768
2769 #[stable(feature = "rust1", since = "1.0.0")]
2770 impl<'a, T> IntoIterator for &'a mut VecDeque<T> {
2771 type Item = &'a mut T;
2772 type IntoIter = IterMut<'a, T>;
2773
2774 fn into_iter(self) -> IterMut<'a, T> {
2775 self.iter_mut()
2776 }
2777 }
2778
2779 #[stable(feature = "rust1", since = "1.0.0")]
2780 impl<A> Extend<A> for VecDeque<A> {
2781 fn extend<T: IntoIterator<Item = A>>(&mut self, iter: T) {
2782 // This function should be the moral equivalent of:
2783 //
2784 // for item in iter.into_iter() {
2785 // self.push_back(item);
2786 // }
2787 let mut iter = iter.into_iter();
2788 while let Some(element) = iter.next() {
2789 if self.len() == self.capacity() {
2790 let (lower, _) = iter.size_hint();
2791 self.reserve(lower.saturating_add(1));
2792 }
2793
2794 let head = self.head;
2795 self.head = self.wrap_add(self.head, 1);
2796 unsafe {
2797 self.buffer_write(head, element);
2798 }
2799 }
2800 }
2801
2802 #[inline]
2803 fn extend_one(&mut self, elem: A) {
2804 self.push_back(elem);
2805 }
2806
2807 #[inline]
2808 fn extend_reserve(&mut self, additional: usize) {
2809 self.reserve(additional);
2810 }
2811 }
2812
2813 #[stable(feature = "extend_ref", since = "1.2.0")]
2814 impl<'a, T: 'a + Copy> Extend<&'a T> for VecDeque<T> {
2815 fn extend<I: IntoIterator<Item = &'a T>>(&mut self, iter: I) {
2816 self.extend(iter.into_iter().cloned());
2817 }
2818
2819 #[inline]
2820 fn extend_one(&mut self, &elem: &T) {
2821 self.push_back(elem);
2822 }
2823
2824 #[inline]
2825 fn extend_reserve(&mut self, additional: usize) {
2826 self.reserve(additional);
2827 }
2828 }
2829
2830 #[stable(feature = "rust1", since = "1.0.0")]
2831 impl<T: fmt::Debug> fmt::Debug for VecDeque<T> {
2832 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
2833 f.debug_list().entries(self).finish()
2834 }
2835 }
2836
2837 #[stable(feature = "vecdeque_vec_conversions", since = "1.10.0")]
2838 impl<T> From<Vec<T>> for VecDeque<T> {
2839 /// Turn a [`Vec<T>`] into a [`VecDeque<T>`].
2840 ///
2841 /// [`Vec<T>`]: crate::vec::Vec
2842 /// [`VecDeque<T>`]: crate::collections::VecDeque
2843 ///
2844 /// This avoids reallocating where possible, but the conditions for that are
2845 /// strict, and subject to change, and so shouldn't be relied upon unless the
2846 /// `Vec<T>` came from `From<VecDeque<T>>` and hasn't been reallocated.
2847 fn from(mut other: Vec<T>) -> Self {
2848 let len = other.len();
2849 if mem::size_of::<T>() == 0 {
2850 // There's no actual allocation for ZSTs to worry about capacity,
2851 // but `VecDeque` can't handle as much length as `Vec`.
2852 assert!(len < MAXIMUM_ZST_CAPACITY, "capacity overflow");
2853 } else {
2854 // We need to resize if the capacity is not a power of two, too small or
2855 // doesn't have at least one free space. We do this while it's still in
2856 // the `Vec` so the items will drop on panic.
2857 let min_cap = cmp::max(MINIMUM_CAPACITY, len) + 1;
2858 let cap = cmp::max(min_cap, other.capacity()).next_power_of_two();
2859 if other.capacity() != cap {
2860 other.reserve_exact(cap - len);
2861 }
2862 }
2863
2864 unsafe {
2865 let (other_buf, len, capacity) = other.into_raw_parts();
2866 let buf = RawVec::from_raw_parts(other_buf, capacity);
2867 VecDeque { tail: 0, head: len, buf }
2868 }
2869 }
2870 }
2871
2872 #[stable(feature = "vecdeque_vec_conversions", since = "1.10.0")]
2873 impl<T> From<VecDeque<T>> for Vec<T> {
2874 /// Turn a [`VecDeque<T>`] into a [`Vec<T>`].
2875 ///
2876 /// [`Vec<T>`]: crate::vec::Vec
2877 /// [`VecDeque<T>`]: crate::collections::VecDeque
2878 ///
2879 /// This never needs to re-allocate, but does need to do *O*(*n*) data movement if
2880 /// the circular buffer doesn't happen to be at the beginning of the allocation.
2881 ///
2882 /// # Examples
2883 ///
2884 /// ```
2885 /// use std::collections::VecDeque;
2886 ///
2887 /// // This one is *O*(1).
2888 /// let deque: VecDeque<_> = (1..5).collect();
2889 /// let ptr = deque.as_slices().0.as_ptr();
2890 /// let vec = Vec::from(deque);
2891 /// assert_eq!(vec, [1, 2, 3, 4]);
2892 /// assert_eq!(vec.as_ptr(), ptr);
2893 ///
2894 /// // This one needs data rearranging.
2895 /// let mut deque: VecDeque<_> = (1..5).collect();
2896 /// deque.push_front(9);
2897 /// deque.push_front(8);
2898 /// let ptr = deque.as_slices().1.as_ptr();
2899 /// let vec = Vec::from(deque);
2900 /// assert_eq!(vec, [8, 9, 1, 2, 3, 4]);
2901 /// assert_eq!(vec.as_ptr(), ptr);
2902 /// ```
2903 fn from(mut other: VecDeque<T>) -> Self {
2904 other.make_contiguous();
2905
2906 unsafe {
2907 let other = ManuallyDrop::new(other);
2908 let buf = other.buf.ptr();
2909 let len = other.len();
2910 let cap = other.cap();
2911
2912 if other.tail != 0 {
2913 ptr::copy(buf.add(other.tail), buf, len);
2914 }
2915 Vec::from_raw_parts(buf, len, cap)
2916 }
2917 }
2918 }