3 //! Provides `P<T>`, a frozen owned smart pointer, as a replacement for `@T` in
6 //! # Motivations and benefits
8 //! * **Identity**: sharing AST nodes is problematic for the various analysis
9 //! passes (e.g., one may be able to bypass the borrow checker with a shared
10 //! `ExprKind::AddrOf` node taking a mutable borrow). The only reason `@T` in the
11 //! AST hasn't caused issues is because of inefficient folding passes which
12 //! would always deduplicate any such shared nodes. Even if the AST were to
13 //! switch to an arena, this would still hold, i.e., it couldn't use `&'a T`,
14 //! but rather a wrapper like `P<'a, T>`.
16 //! * **Immutability**: `P<T>` disallows mutating its inner `T`, unlike `Box<T>`
17 //! (unless it contains an `Unsafe` interior, but that may be denied later).
18 //! This mainly prevents mistakes, but can also enforces a kind of "purity".
20 //! * **Efficiency**: folding can reuse allocation space for `P<T>` and `Vec<T>`,
21 //! the latter even when the input and output types differ (as it would be the
22 //! case with arenas or a GADT AST using type parameters to toggle features).
24 //! * **Maintainability**: `P<T>` provides a fixed interface - `Deref`,
25 //! `and_then` and `map` - which can remain fully functional even if the
26 //! implementation changes (using a special thread-local heap, for example).
27 //! Moreover, a switch to, e.g., `P<'a, T>` would be easy and mostly automated.
29 use std::fmt::{self, Display, Debug};
30 use std::iter::FromIterator;
31 use std::ops::{Deref, DerefMut};
32 use std::{slice, vec};
34 use rustc_serialize::{Encodable, Decodable, Encoder, Decoder};
36 use rustc_data_structures::stable_hasher::{StableHasher, HashStable};
37 /// An owned smart pointer.
38 #[derive(Hash, PartialEq, Eq)]
39 pub struct P<T: ?Sized> {
43 /// Construct a `P<T>` from a `T` value.
44 #[allow(non_snake_case)]
45 pub fn P<T: 'static>(value: T) -> P<T> {
51 impl<T: 'static> P<T> {
52 /// Move out of the pointer.
53 /// Intended for chaining transformations not covered by `map`.
54 pub fn and_then<U, F>(self, f: F) -> U where
60 /// Equivalent to `and_then(|x| x)`.
61 pub fn into_inner(self) -> T {
65 /// Produce a new `P<T>` from `self` without reallocating.
66 pub fn map<F>(mut self, f: F) -> P<T> where
75 /// Optionally produce a new `P<T>` from `self` without reallocating.
76 pub fn filter_map<F>(mut self, f: F) -> Option<P<T>> where
77 F: FnOnce(T) -> Option<T>,
79 *self.ptr = f(*self.ptr)?;
84 impl<T: ?Sized> Deref for P<T> {
87 fn deref(&self) -> &T {
92 impl<T: ?Sized> DerefMut for P<T> {
93 fn deref_mut(&mut self) -> &mut T {
98 impl<T: 'static + Clone> Clone for P<T> {
99 fn clone(&self) -> P<T> {
104 impl<T: ?Sized + Debug> Debug for P<T> {
105 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
106 Debug::fmt(&self.ptr, f)
110 impl<T: Display> Display for P<T> {
111 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
112 Display::fmt(&**self, f)
116 impl<T> fmt::Pointer for P<T> {
117 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
118 fmt::Pointer::fmt(&self.ptr, f)
122 impl<T: 'static + Decodable> Decodable for P<T> {
123 fn decode<D: Decoder>(d: &mut D) -> Result<P<T>, D::Error> {
124 Decodable::decode(d).map(P)
128 impl<T: Encodable> Encodable for P<T> {
129 fn encode<S: Encoder>(&self, s: &mut S) -> Result<(), S::Error> {
135 pub const fn new() -> P<[T]> {
136 // HACK(eddyb) bypass the lack of a `const fn` to create an empty `Box<[T]>`
137 // (as trait methods, `default` in this case, can't be `const fn` yet).
140 use std::ptr::NonNull;
141 std::mem::transmute(NonNull::<[T; 0]>::dangling() as NonNull<[T]>)
147 pub fn from_vec(v: Vec<T>) -> P<[T]> {
148 P { ptr: v.into_boxed_slice() }
152 pub fn into_vec(self) -> Vec<T> {
157 impl<T> Default for P<[T]> {
158 /// Creates an empty `P<[T]>`.
159 fn default() -> P<[T]> {
164 impl<T: Clone> Clone for P<[T]> {
165 fn clone(&self) -> P<[T]> {
166 P::from_vec(self.to_vec())
170 impl<T> From<Vec<T>> for P<[T]> {
171 fn from(v: Vec<T>) -> Self {
176 impl<T> Into<Vec<T>> for P<[T]> {
177 fn into(self) -> Vec<T> {
182 impl<T> FromIterator<T> for P<[T]> {
183 fn from_iter<I: IntoIterator<Item=T>>(iter: I) -> P<[T]> {
184 P::from_vec(iter.into_iter().collect())
188 impl<T> IntoIterator for P<[T]> {
190 type IntoIter = vec::IntoIter<T>;
192 fn into_iter(self) -> Self::IntoIter {
193 self.into_vec().into_iter()
197 impl<'a, T> IntoIterator for &'a P<[T]> {
199 type IntoIter = slice::Iter<'a, T>;
200 fn into_iter(self) -> Self::IntoIter {
205 impl<T: Encodable> Encodable for P<[T]> {
206 fn encode<S: Encoder>(&self, s: &mut S) -> Result<(), S::Error> {
207 Encodable::encode(&**self, s)
211 impl<T: Decodable> Decodable for P<[T]> {
212 fn decode<D: Decoder>(d: &mut D) -> Result<P<[T]>, D::Error> {
213 Ok(P::from_vec(Decodable::decode(d)?))
217 impl<CTX, T> HashStable<CTX> for P<T>
218 where T: ?Sized + HashStable<CTX>
220 fn hash_stable(&self, hcx: &mut CTX, hasher: &mut StableHasher) {
221 (**self).hash_stable(hcx, hasher);