diff options
Diffstat (limited to 'vendor/indexmap/src/set/slice.rs')
| -rw-r--r-- | vendor/indexmap/src/set/slice.rs | 379 |
1 files changed, 0 insertions, 379 deletions
diff --git a/vendor/indexmap/src/set/slice.rs b/vendor/indexmap/src/set/slice.rs deleted file mode 100644 index faa9041a..00000000 --- a/vendor/indexmap/src/set/slice.rs +++ /dev/null @@ -1,379 +0,0 @@ -use super::{Bucket, Entries, IndexSet, IntoIter, Iter}; -use crate::util::{slice_eq, try_simplify_range}; - -use alloc::boxed::Box; -use alloc::vec::Vec; -use core::cmp::Ordering; -use core::fmt; -use core::hash::{Hash, Hasher}; -use core::ops::{self, Bound, Index, RangeBounds}; - -/// A dynamically-sized slice of values in an [`IndexSet`]. -/// -/// This supports indexed operations much like a `[T]` slice, -/// but not any hashed operations on the values. -/// -/// Unlike `IndexSet`, `Slice` does consider the order for [`PartialEq`] -/// and [`Eq`], and it also implements [`PartialOrd`], [`Ord`], and [`Hash`]. -#[repr(transparent)] -pub struct Slice<T> { - pub(crate) entries: [Bucket<T>], -} - -// SAFETY: `Slice<T>` is a transparent wrapper around `[Bucket<T>]`, -// and reference lifetimes are bound together in function signatures. -#[allow(unsafe_code)] -impl<T> Slice<T> { - pub(super) const fn from_slice(entries: &[Bucket<T>]) -> &Self { - unsafe { &*(entries as *const [Bucket<T>] as *const Self) } - } - - pub(super) fn from_boxed(entries: Box<[Bucket<T>]>) -> Box<Self> { - unsafe { Box::from_raw(Box::into_raw(entries) as *mut Self) } - } - - fn into_boxed(self: Box<Self>) -> Box<[Bucket<T>]> { - unsafe { Box::from_raw(Box::into_raw(self) as *mut [Bucket<T>]) } - } -} - -impl<T> Slice<T> { - pub(crate) fn into_entries(self: Box<Self>) -> Vec<Bucket<T>> { - self.into_boxed().into_vec() - } - - /// Returns an empty slice. - pub const fn new<'a>() -> &'a Self { - Self::from_slice(&[]) - } - - /// Return the number of elements in the set slice. - pub const fn len(&self) -> usize { - self.entries.len() - } - - /// Returns true if the set slice contains no elements. - pub const fn is_empty(&self) -> bool { - self.entries.is_empty() - } - - /// Get a value by index. - /// - /// Valid indices are `0 <= index < self.len()`. - pub fn get_index(&self, index: usize) -> Option<&T> { - self.entries.get(index).map(Bucket::key_ref) - } - - /// Returns a slice of values in the given range of indices. - /// - /// Valid indices are `0 <= index < self.len()`. - pub fn get_range<R: RangeBounds<usize>>(&self, range: R) -> Option<&Self> { - let range = try_simplify_range(range, self.entries.len())?; - self.entries.get(range).map(Self::from_slice) - } - - /// Get the first value. - pub fn first(&self) -> Option<&T> { - self.entries.first().map(Bucket::key_ref) - } - - /// Get the last value. - pub fn last(&self) -> Option<&T> { - self.entries.last().map(Bucket::key_ref) - } - - /// Divides one slice into two at an index. - /// - /// ***Panics*** if `index > len`. - pub fn split_at(&self, index: usize) -> (&Self, &Self) { - let (first, second) = self.entries.split_at(index); - (Self::from_slice(first), Self::from_slice(second)) - } - - /// Returns the first value and the rest of the slice, - /// or `None` if it is empty. - pub fn split_first(&self) -> Option<(&T, &Self)> { - if let [first, rest @ ..] = &self.entries { - Some((&first.key, Self::from_slice(rest))) - } else { - None - } - } - - /// Returns the last value and the rest of the slice, - /// or `None` if it is empty. - pub fn split_last(&self) -> Option<(&T, &Self)> { - if let [rest @ .., last] = &self.entries { - Some((&last.key, Self::from_slice(rest))) - } else { - None - } - } - - /// Return an iterator over the values of the set slice. - pub fn iter(&self) -> Iter<'_, T> { - Iter::new(&self.entries) - } - - /// Search over a sorted set for a value. - /// - /// Returns the position where that value is present, or the position where it can be inserted - /// to maintain the sort. See [`slice::binary_search`] for more details. - /// - /// Computes in **O(log(n))** time, which is notably less scalable than looking the value up in - /// the set this is a slice from using [`IndexSet::get_index_of`], but this can also position - /// missing values. - pub fn binary_search(&self, x: &T) -> Result<usize, usize> - where - T: Ord, - { - self.binary_search_by(|p| p.cmp(x)) - } - - /// Search over a sorted set with a comparator function. - /// - /// Returns the position where that value is present, or the position where it can be inserted - /// to maintain the sort. See [`slice::binary_search_by`] for more details. - /// - /// Computes in **O(log(n))** time. - #[inline] - pub fn binary_search_by<'a, F>(&'a self, mut f: F) -> Result<usize, usize> - where - F: FnMut(&'a T) -> Ordering, - { - self.entries.binary_search_by(move |a| f(&a.key)) - } - - /// Search over a sorted set with an extraction function. - /// - /// Returns the position where that value is present, or the position where it can be inserted - /// to maintain the sort. See [`slice::binary_search_by_key`] for more details. - /// - /// Computes in **O(log(n))** time. - #[inline] - pub fn binary_search_by_key<'a, B, F>(&'a self, b: &B, mut f: F) -> Result<usize, usize> - where - F: FnMut(&'a T) -> B, - B: Ord, - { - self.binary_search_by(|k| f(k).cmp(b)) - } - - /// Returns the index of the partition point of a sorted set according to the given predicate - /// (the index of the first element of the second partition). - /// - /// See [`slice::partition_point`] for more details. - /// - /// Computes in **O(log(n))** time. - #[must_use] - pub fn partition_point<P>(&self, mut pred: P) -> usize - where - P: FnMut(&T) -> bool, - { - self.entries.partition_point(move |a| pred(&a.key)) - } -} - -impl<'a, T> IntoIterator for &'a Slice<T> { - type IntoIter = Iter<'a, T>; - type Item = &'a T; - - fn into_iter(self) -> Self::IntoIter { - self.iter() - } -} - -impl<T> IntoIterator for Box<Slice<T>> { - type IntoIter = IntoIter<T>; - type Item = T; - - fn into_iter(self) -> Self::IntoIter { - IntoIter::new(self.into_entries()) - } -} - -impl<T> Default for &'_ Slice<T> { - fn default() -> Self { - Slice::from_slice(&[]) - } -} - -impl<T> Default for Box<Slice<T>> { - fn default() -> Self { - Slice::from_boxed(Box::default()) - } -} - -impl<T: Clone> Clone for Box<Slice<T>> { - fn clone(&self) -> Self { - Slice::from_boxed(self.entries.to_vec().into_boxed_slice()) - } -} - -impl<T: Copy> From<&Slice<T>> for Box<Slice<T>> { - fn from(slice: &Slice<T>) -> Self { - Slice::from_boxed(Box::from(&slice.entries)) - } -} - -impl<T: fmt::Debug> fmt::Debug for Slice<T> { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - f.debug_list().entries(self).finish() - } -} - -impl<T, U> PartialEq<Slice<U>> for Slice<T> -where - T: PartialEq<U>, -{ - fn eq(&self, other: &Slice<U>) -> bool { - slice_eq(&self.entries, &other.entries, |b1, b2| b1.key == b2.key) - } -} - -impl<T, U> PartialEq<[U]> for Slice<T> -where - T: PartialEq<U>, -{ - fn eq(&self, other: &[U]) -> bool { - slice_eq(&self.entries, other, |b, o| b.key == *o) - } -} - -impl<T, U> PartialEq<Slice<U>> for [T] -where - T: PartialEq<U>, -{ - fn eq(&self, other: &Slice<U>) -> bool { - slice_eq(self, &other.entries, |o, b| *o == b.key) - } -} - -impl<T, U, const N: usize> PartialEq<[U; N]> for Slice<T> -where - T: PartialEq<U>, -{ - fn eq(&self, other: &[U; N]) -> bool { - <Self as PartialEq<[U]>>::eq(self, other) - } -} - -impl<T, const N: usize, U> PartialEq<Slice<U>> for [T; N] -where - T: PartialEq<U>, -{ - fn eq(&self, other: &Slice<U>) -> bool { - <[T] as PartialEq<Slice<U>>>::eq(self, other) - } -} - -impl<T: Eq> Eq for Slice<T> {} - -impl<T: PartialOrd> PartialOrd for Slice<T> { - fn partial_cmp(&self, other: &Self) -> Option<Ordering> { - self.iter().partial_cmp(other) - } -} - -impl<T: Ord> Ord for Slice<T> { - fn cmp(&self, other: &Self) -> Ordering { - self.iter().cmp(other) - } -} - -impl<T: Hash> Hash for Slice<T> { - fn hash<H: Hasher>(&self, state: &mut H) { - self.len().hash(state); - for value in self { - value.hash(state); - } - } -} - -impl<T> Index<usize> for Slice<T> { - type Output = T; - - fn index(&self, index: usize) -> &Self::Output { - &self.entries[index].key - } -} - -// We can't have `impl<I: RangeBounds<usize>> Index<I>` because that conflicts with `Index<usize>`. -// Instead, we repeat the implementations for all the core range types. -macro_rules! impl_index { - ($($range:ty),*) => {$( - impl<T, S> Index<$range> for IndexSet<T, S> { - type Output = Slice<T>; - - fn index(&self, range: $range) -> &Self::Output { - Slice::from_slice(&self.as_entries()[range]) - } - } - - impl<T> Index<$range> for Slice<T> { - type Output = Self; - - fn index(&self, range: $range) -> &Self::Output { - Slice::from_slice(&self.entries[range]) - } - } - )*} -} -impl_index!( - ops::Range<usize>, - ops::RangeFrom<usize>, - ops::RangeFull, - ops::RangeInclusive<usize>, - ops::RangeTo<usize>, - ops::RangeToInclusive<usize>, - (Bound<usize>, Bound<usize>) -); - -#[cfg(test)] -mod tests { - use super::*; - - #[test] - fn slice_index() { - fn check(vec_slice: &[i32], set_slice: &Slice<i32>, sub_slice: &Slice<i32>) { - assert_eq!(set_slice as *const _, sub_slice as *const _); - itertools::assert_equal(vec_slice, set_slice); - } - - let vec: Vec<i32> = (0..10).map(|i| i * i).collect(); - let set: IndexSet<i32> = vec.iter().cloned().collect(); - let slice = set.as_slice(); - - // RangeFull - check(&vec[..], &set[..], &slice[..]); - - for i in 0usize..10 { - // Index - assert_eq!(vec[i], set[i]); - assert_eq!(vec[i], slice[i]); - - // RangeFrom - check(&vec[i..], &set[i..], &slice[i..]); - - // RangeTo - check(&vec[..i], &set[..i], &slice[..i]); - - // RangeToInclusive - check(&vec[..=i], &set[..=i], &slice[..=i]); - - // (Bound<usize>, Bound<usize>) - let bounds = (Bound::Excluded(i), Bound::Unbounded); - check(&vec[i + 1..], &set[bounds], &slice[bounds]); - - for j in i..=10 { - // Range - check(&vec[i..j], &set[i..j], &slice[i..j]); - } - - for j in i..10 { - // RangeInclusive - check(&vec[i..=j], &set[i..=j], &slice[i..=j]); - } - } - } -} |
