yaml-rt-core 0.2.0

Dependency-free YAML 1.2.2 lossless parser and editor core
Documentation 
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
125
126
127
128
129
130
131
132
133
134

use std::ops::{Index, IndexMut};

/// A small, safe inline vector with a lazily allocated overflow arena.
///
/// Values fill the inline array first. Once it is full, additional values are
/// stored in `spill`, preserving ordinary vector ordering without unsafe code.
#[derive(Debug, Clone, PartialEq, Eq)]
pub(crate) struct InlineVec<T, const N: usize> {
    inline: [Option<T>; N],
    inline_len: usize,
    spill: Vec<T>,
}

impl<T, const N: usize> InlineVec<T, N> {
    pub(crate) fn new() -> Self {
        Self {
            inline: std::array::from_fn(|_| None),
            inline_len: 0,
            spill: Vec::new(),
        }
    }

    pub(crate) fn len(&self) -> usize {
        self.inline_len + self.spill.len()
    }

    pub(crate) fn push(&mut self, value: T) {
        if self.inline_len < N {
            debug_assert!(self.spill.is_empty());
            self.inline[self.inline_len] = Some(value);
            self.inline_len += 1;
        } else {
            self.spill.push(value);
        }
    }

    pub(crate) fn pop(&mut self) -> Option<T> {
        if let Some(value) = self.spill.pop() {
            return Some(value);
        }
        let index = self.inline_len.checked_sub(1)?;
        self.inline_len = index;
        self.inline[index].take()
    }

    pub(crate) fn is_empty(&self) -> bool {
        self.inline_len == 0 && self.spill.is_empty()
    }

    pub(crate) fn last(&self) -> Option<&T> {
        self.spill.last().or_else(|| {
            self.inline_len
                .checked_sub(1)
                .and_then(|index| self.inline[index].as_ref())
        })
    }

    pub(crate) fn last_mut(&mut self) -> Option<&mut T> {
        if self.spill.is_empty() {
            let index = self.inline_len.checked_sub(1)?;
            self.inline[index].as_mut()
        } else {
            self.spill.last_mut()
        }
    }

    pub(crate) fn get(&self, index: usize) -> Option<&T> {
        if index < self.inline_len {
            self.inline[index].as_ref()
        } else {
            self.spill.get(index.checked_sub(self.inline_len)?)
        }
    }

    pub(crate) fn iter(&self) -> impl DoubleEndedIterator<Item = &T> {
        self.inline[..self.inline_len]
            .iter()
            .map(|value| value.as_ref().expect("occupied inline slot"))
            .chain(self.spill.iter())
    }
}

impl<T, const N: usize> Default for InlineVec<T, N> {
    fn default() -> Self {
        Self::new()
    }
}

impl<T, const N: usize> Index<usize> for InlineVec<T, N> {
    type Output = T;

    fn index(&self, index: usize) -> &Self::Output {
        self.get(index).expect("index is outside inline vector")
    }
}

impl<T, const N: usize> IndexMut<usize> for InlineVec<T, N> {
    fn index_mut(&mut self, index: usize) -> &mut Self::Output {
        if index < self.inline_len {
            self.inline[index].as_mut().expect("occupied inline slot")
        } else {
            &mut self.spill[index - self.inline_len]
        }
    }
}

#[cfg(test)]
mod tests {
    use super::InlineVec;

    #[test]
    fn stores_inline_then_spills_in_order() {
        let mut values = InlineVec::<u32, 2>::new();
        assert_eq!(values.len(), 0);
        values.push(1);
        values.push(2);
        values.push(3);
        assert_eq!(values.iter().copied().collect::<Vec<_>>(), vec![1, 2, 3]);
        assert_eq!(values.pop(), Some(3));
        assert_eq!(values.pop(), Some(2));
        assert_eq!(values.pop(), Some(1));
        assert_eq!(values.pop(), None);
    }

    #[test]
    fn mutable_indexing_covers_inline_and_spill() {
        let mut values = InlineVec::<u32, 1>::new();
        values.push(2);
        values.push(4);
        values[0] += 1;
        values[1] += 2;
        assert_eq!(values.iter().copied().collect::<Vec<_>>(), vec![3, 6]);
    }
}