src/librustc_data_structures/thin_vec.rs

   1 use crate::stable_hasher::{StableHasher, HashStable};
   2
   3 /// A vector type optimized for cases where this size is usually 0 (cf. `SmallVector`).
   4 /// The `Option<Box<..>>` wrapping allows us to represent a zero sized vector with `None`,
   5 /// which uses only a single (null) pointer.
   6 #[derive(Clone, RustcEncodable, RustcDecodable, Debug)]
   7 pub struct ThinVec<T>(Option<Box<Vec<T>>>);
   8
   9 impl<T> ThinVec<T> {
  10     pub fn new() -> Self {
  11         ThinVec(None)
  12     }
  13 }
  14
  15 impl<T> From<Vec<T>> for ThinVec<T> {
  16     fn from(vec: Vec<T>) -> Self {
  17         if vec.is_empty() {
  18             ThinVec(None)
  19         } else {
  20             ThinVec(Some(Box::new(vec)))
  21         }
  22     }
  23 }
  24
  25 impl<T> Into<Vec<T>> for ThinVec<T> {
  26     fn into(self) -> Vec<T> {
  27         match self {
  28             ThinVec(None) => Vec::new(),
  29             ThinVec(Some(vec)) => *vec,
  30         }
  31     }
  32 }
  33
  34 impl<T> ::std::ops::Deref for ThinVec<T> {
  35     type Target = [T];
  36     fn deref(&self) -> &[T] {
  37         match *self {
  38             ThinVec(None) => &[],
  39             ThinVec(Some(ref vec)) => vec,
  40         }
  41     }
  42 }
  43
  44 impl<T> ::std::ops::DerefMut for ThinVec<T> {
  45     fn deref_mut(&mut self) -> &mut [T] {
  46         match *self {
  47             ThinVec(None) => &mut [],
  48             ThinVec(Some(ref mut vec)) => vec,
  49         }
  50     }
  51 }
  52
  53 impl<T> Extend<T> for ThinVec<T> {
  54     fn extend<I: IntoIterator<Item = T>>(&mut self, iter: I) {
  55         match *self {
  56             ThinVec(Some(ref mut vec)) => vec.extend(iter),
  57             ThinVec(None) => *self = iter.into_iter().collect::<Vec<_>>().into(),
  58         }
  59     }
  60 }
  61
  62 impl<T: HashStable<CTX>, CTX> HashStable<CTX> for ThinVec<T> {
  63     fn hash_stable(&self, hcx: &mut CTX, hasher: &mut StableHasher) {
  64         (**self).hash_stable(hcx, hasher)
  65     }
  66 }
  67
  68 impl<T> Default for ThinVec<T> {
  69     fn default() -> Self {
  70         Self(None)
  71     }
  72 }