orx_linked_list/list/mut_singly.rs
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124
use super::List;
use crate::{iter::SinglyIterMut, variant::Singly, SinglyIdx};
use orx_pinned_vec::PinnedVec;
use orx_selfref_col::{MemoryPolicy, Node, NodeIdx, Refs};
impl<T, M, P> List<Singly<T>, M, P>
where
M: MemoryPolicy<Singly<T>>,
P: PinnedVec<Node<Singly<T>>>,
{
/// ***O(1)*** Sets value of `front` of the list as `new_front` and:
/// * returns value of the front element;
/// * returns None if the list was empty.
///
/// # Examples
///
/// ```rust
/// use orx_linked_list::*;
///
/// let mut list = SinglyList::new();
///
/// assert_eq!(0, list.len());
///
/// let prior_front = list.swap_front('a');
/// assert!(prior_front.is_none());
/// assert_eq!(Some(&'a'), list.front());
///
/// let prior_front = list.swap_front('z');
/// assert_eq!(Some('a'), prior_front);
/// assert_eq!(Some(&'z'), list.front());
/// ```
pub fn swap_front(&mut self, new_front: T) -> Option<T> {
match self.0.ends().get() {
Some(p) => Some(self.0.swap_data(&p, new_front)),
None => {
self.push_front(new_front);
None
}
}
}
/// ***O(1)*** Pushes the `value` to the `front` of the list.
///
/// # Examples
///
/// ```rust
/// use orx_linked_list::*;
///
/// let mut list = DoublyList::new();
///
/// list.push_front('a');
/// list.push_front('b');
///
/// assert_eq!(Some(&'b'), list.front());
/// assert_eq!(Some(&'a'), list.back());
///
/// let popped = list.pop_front();
/// assert_eq!(Some('b'), popped);
/// ```
pub fn push_front(&mut self, value: T) -> SinglyIdx<T> {
let idx = self.0.push(value);
if let Some(front) = self.0.ends().get() {
self.0.node_mut(&idx).next_mut().set_some(&front);
}
self.0.ends_mut().set_some(&idx);
NodeIdx::new(self.0.memory_state(), &idx)
}
/// ***O(1)*** Pops and returns the value at the `front` of the list; returns None if the list is empty.
///
/// # Examples
///
/// ```rust
/// use orx_linked_list::*;
///
/// let mut list = SinglyList::new();
///
/// let popped = list.pop_front();
/// assert!(popped.is_none());
///
/// list.push_front('a');
/// assert_eq!(Some(&'a'), list.front());
///
/// let popped = list.pop_front();
/// assert_eq!(Some('a'), popped);
/// assert!(list.is_empty());
/// ```
pub fn pop_front(&mut self) -> Option<T> {
self.0.ends().get().map(|front| {
match self.0.node(&front).next().get() {
Some(new_front) => self.0.ends_mut().set_some(&new_front),
None => self.0.ends_mut().clear(),
}
self.0.close_and_reclaim(&front)
})
}
/// Returns a forward iterator of mutable references to elements of the list from front to back.
///
/// # Examples
///
/// ```rust
/// use orx_linked_list::*;
///
/// let mut list = SinglyList::new();
///
/// list.push_front(2);
/// list.push_front(1);
/// list.push_front(0);
/// assert!(list.eq_to_iter_vals([0, 1, 2]));
///
/// for x in list.iter_mut() {
/// *x += 40;
/// }
///
/// assert!(list.eq_to_iter_vals([40, 41, 42]));
/// ```
pub fn iter_mut(&mut self) -> SinglyIterMut<T, P> {
SinglyIterMut::new_old(&mut self.0)
}
}