1use alloc::rc::Rc;
2use alloc::vec;
3use core::mem;
4use core::panic::RefUnwindSafe;
5use core::slice;
6
7pub(crate) struct RcVec<T> {
8 inner: Rc<Vec<T>>,
9}
10
11pub(crate) struct RcVecBuilder<T> {
12 inner: Vec<T>,
13}
14
15pub(crate) struct RcVecMut<'a, T> {
16 inner: &'a mut Vec<T>,
17}
18
19#[derive(Clone)]
20pub(crate) struct RcVecIntoIter<T> {
21 inner: vec::IntoIter<T>,
22}
23
24impl<T> RcVec<T> {
25 pub(crate) fn is_empty(&self) -> bool {
26 self.inner.is_empty()
27 }
28
29 pub(crate) fn len(&self) -> usize {
30 self.inner.len()
31 }
32
33 pub(crate) fn iter(&self) -> slice::Iter<T> {
34 self.inner.iter()
35 }
36
37 pub(crate) fn make_mut(&mut self) -> RcVecMut<T>
38 where
39 T: Clone,
40 {
41 RcVecMut {
42 inner: Rc::make_mut(&mut self.inner),
43 }
44 }
45
46 pub(crate) fn get_mut(&mut self) -> Option<RcVecMut<T>> {
47 let inner = Rc::get_mut(&mut self.inner)?;
48 Some(RcVecMut { inner })
49 }
50
51 pub(crate) fn make_owned(mut self) -> RcVecBuilder<T>
52 where
53 T: Clone,
54 {
55 let vec = if let Some(owned) = Rc::get_mut(&mut self.inner) {
56 mem::take(owned)
57 } else {
58 Vec::clone(&self.inner)
59 };
60 RcVecBuilder { inner: vec }
61 }
62}
63
64impl<T> RcVecBuilder<T> {
65 pub(crate) fn new() -> Self {
66 RcVecBuilder { inner: Vec::new() }
67 }
68
69 pub(crate) fn with_capacity(cap: usize) -> Self {
70 RcVecBuilder {
71 inner: Vec::with_capacity(cap),
72 }
73 }
74
75 pub(crate) fn push(&mut self, element: T) {
76 self.inner.push(element);
77 }
78
79 pub(crate) fn extend(&mut self, iter: impl IntoIterator<Item = T>) {
80 self.inner.extend(iter);
81 }
82
83 pub(crate) fn as_mut(&mut self) -> RcVecMut<T> {
84 RcVecMut {
85 inner: &mut self.inner,
86 }
87 }
88
89 pub(crate) fn build(self) -> RcVec<T> {
90 RcVec {
91 inner: Rc::new(self.inner),
92 }
93 }
94}
95
96impl<'a, T> RcVecMut<'a, T> {
97 pub(crate) fn push(&mut self, element: T) {
98 self.inner.push(element);
99 }
100
101 pub(crate) fn extend(&mut self, iter: impl IntoIterator<Item = T>) {
102 self.inner.extend(iter);
103 }
104
105 pub(crate) fn as_mut(&mut self) -> RcVecMut<T> {
106 RcVecMut { inner: self.inner }
107 }
108
109 pub(crate) fn take(self) -> RcVecBuilder<T> {
110 let vec: Vec = mem::take(self.inner);
111 RcVecBuilder { inner: vec }
112 }
113}
114
115impl<T> Clone for RcVec<T> {
116 fn clone(&self) -> Self {
117 RcVec {
118 inner: Rc::clone(&self.inner),
119 }
120 }
121}
122
123impl<T> IntoIterator for RcVecBuilder<T> {
124 type Item = T;
125 type IntoIter = RcVecIntoIter<T>;
126
127 fn into_iter(self) -> Self::IntoIter {
128 RcVecIntoIter {
129 inner: self.inner.into_iter(),
130 }
131 }
132}
133
134impl<T> Iterator for RcVecIntoIter<T> {
135 type Item = T;
136
137 fn next(&mut self) -> Option<Self::Item> {
138 self.inner.next()
139 }
140
141 fn size_hint(&self) -> (usize, Option<usize>) {
142 self.inner.size_hint()
143 }
144}
145
146impl<T> RefUnwindSafe for RcVec<T> where T: RefUnwindSafe {}
147