mod iter;
mod mutable;
mod slice;
#[cfg(test)]
mod tests;
pub use self::mutable::MutableValues;
pub use indexmap::set::{
Difference, Drain, ExtractIf, Intersection, IntoIter, Iter, Slice, Splice, SymmetricDifference,
Union,
};
#[cfg(feature = "rayon")]
#[cfg_attr(docsrs, doc(cfg(feature = "rayon")))]
pub mod rayon;
use alloc::boxed::Box;
use core::cmp::Ordering;
use core::fmt;
use core::hash::{BuildHasher, Hash, Hasher};
use core::ops::{BitAnd, BitOr, BitXor, Index, RangeBounds, Sub};
use indexmap::IndexSet;
#[cfg(doc)]
use alloc::vec::Vec;
#[cfg(feature = "std")]
use std::collections::hash_map::RandomState;
use crate::{Equivalent, TryReserveError};
#[cfg(feature = "std")]
pub struct OrderSet<T, S = RandomState> {
pub(crate) inner: IndexSet<T, S>,
}
#[cfg(not(feature = "std"))]
pub struct OrderSet<T, S> {
pub(crate) inner: IndexSet<T, S>,
}
impl<T, S> Clone for OrderSet<T, S>
where
T: Clone,
S: Clone,
{
fn clone(&self) -> Self {
Self {
inner: self.inner.clone(),
}
}
fn clone_from(&mut self, other: &Self) {
self.inner.clone_from(&other.inner);
}
}
impl<T, S> fmt::Debug for OrderSet<T, S>
where
T: fmt::Debug,
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_set().entries(self.iter()).finish()
}
}
#[cfg(feature = "std")]
#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
impl<T> OrderSet<T> {
pub fn new() -> Self {
Self {
inner: IndexSet::new(),
}
}
pub fn with_capacity(n: usize) -> Self {
Self {
inner: IndexSet::with_capacity(n),
}
}
}
impl<T, S> OrderSet<T, S> {
pub fn with_capacity_and_hasher(n: usize, hash_builder: S) -> Self {
Self {
inner: IndexSet::with_capacity_and_hasher(n, hash_builder),
}
}
pub const fn with_hasher(hash_builder: S) -> Self {
Self {
inner: IndexSet::with_hasher(hash_builder),
}
}
pub fn capacity(&self) -> usize {
self.inner.capacity()
}
pub fn hasher(&self) -> &S {
self.inner.hasher()
}
pub fn len(&self) -> usize {
self.inner.len()
}
pub fn is_empty(&self) -> bool {
self.inner.is_empty()
}
pub fn iter(&self) -> Iter<'_, T> {
self.inner.iter()
}
pub fn clear(&mut self) {
self.inner.clear();
}
pub fn truncate(&mut self, len: usize) {
self.inner.truncate(len);
}
#[track_caller]
pub fn drain<R>(&mut self, range: R) -> Drain<'_, T>
where
R: RangeBounds<usize>,
{
self.inner.drain(range)
}
#[track_caller]
pub fn extract_if<F, R>(&mut self, range: R, pred: F) -> ExtractIf<'_, T, F>
where
F: FnMut(&T) -> bool,
R: RangeBounds<usize>,
{
self.inner.extract_if(range, pred)
}
#[track_caller]
pub fn split_off(&mut self, at: usize) -> Self
where
S: Clone,
{
Self {
inner: self.inner.split_off(at),
}
}
pub fn reserve(&mut self, additional: usize) {
self.inner.reserve(additional);
}
pub fn reserve_exact(&mut self, additional: usize) {
self.inner.reserve_exact(additional);
}
pub fn try_reserve(&mut self, additional: usize) -> Result<(), TryReserveError> {
self.inner.try_reserve(additional)
}
pub fn try_reserve_exact(&mut self, additional: usize) -> Result<(), TryReserveError> {
self.inner.try_reserve_exact(additional)
}
pub fn shrink_to_fit(&mut self) {
self.inner.shrink_to_fit();
}
pub fn shrink_to(&mut self, min_capacity: usize) {
self.inner.shrink_to(min_capacity);
}
}
impl<T, S> OrderSet<T, S>
where
T: Hash + Eq,
S: BuildHasher,
{
pub fn insert(&mut self, value: T) -> bool {
self.inner.insert(value)
}
pub fn insert_full(&mut self, value: T) -> (usize, bool) {
self.inner.insert_full(value)
}
pub fn insert_sorted(&mut self, value: T) -> (usize, bool)
where
T: Ord,
{
self.inner.insert_sorted(value)
}
pub fn insert_sorted_by<F>(&mut self, value: T, cmp: F) -> (usize, bool)
where
F: FnMut(&T, &T) -> Ordering,
{
self.inner.insert_sorted_by(value, cmp)
}
pub fn insert_sorted_by_key<B, F>(&mut self, value: T, sort_key: F) -> (usize, bool)
where
B: Ord,
F: FnMut(&T) -> B,
{
self.inner.insert_sorted_by_key(value, sort_key)
}
#[track_caller]
pub fn insert_before(&mut self, index: usize, value: T) -> (usize, bool) {
self.inner.insert_before(index, value)
}
#[track_caller]
pub fn shift_insert(&mut self, index: usize, value: T) -> bool {
self.inner.shift_insert(index, value)
}
pub fn replace(&mut self, value: T) -> Option<T> {
self.inner.replace(value)
}
pub fn replace_full(&mut self, value: T) -> (usize, Option<T>) {
self.inner.replace_full(value)
}
#[track_caller]
pub fn replace_index(&mut self, index: usize, value: T) -> Result<T, (usize, T)> {
self.inner.replace_index(index, value)
}
pub fn difference<'a, S2>(&'a self, other: &'a OrderSet<T, S2>) -> Difference<'a, T, S2>
where
S2: BuildHasher,
{
self.inner.difference(&other.inner)
}
pub fn symmetric_difference<'a, S2>(
&'a self,
other: &'a OrderSet<T, S2>,
) -> SymmetricDifference<'a, T, S, S2>
where
S2: BuildHasher,
{
self.inner.symmetric_difference(&other.inner)
}
pub fn intersection<'a, S2>(&'a self, other: &'a OrderSet<T, S2>) -> Intersection<'a, T, S2>
where
S2: BuildHasher,
{
self.inner.intersection(&other.inner)
}
pub fn union<'a, S2>(&'a self, other: &'a OrderSet<T, S2>) -> Union<'a, T, S>
where
S2: BuildHasher,
{
self.inner.union(&other.inner)
}
#[track_caller]
pub fn splice<R, I>(&mut self, range: R, replace_with: I) -> Splice<'_, I::IntoIter, T, S>
where
R: RangeBounds<usize>,
I: IntoIterator<Item = T>,
{
self.inner.splice(range, replace_with)
}
pub fn append<S2>(&mut self, other: &mut OrderSet<T, S2>) {
self.inner.append(&mut other.inner);
}
}
impl<T, S> OrderSet<T, S>
where
S: BuildHasher,
{
pub fn contains<Q>(&self, value: &Q) -> bool
where
Q: ?Sized + Hash + Equivalent<T>,
{
self.inner.contains(value)
}
pub fn get<Q>(&self, value: &Q) -> Option<&T>
where
Q: ?Sized + Hash + Equivalent<T>,
{
self.inner.get(value)
}
pub fn get_full<Q>(&self, value: &Q) -> Option<(usize, &T)>
where
Q: ?Sized + Hash + Equivalent<T>,
{
self.inner.get_full(value)
}
pub fn get_index_of<Q>(&self, value: &Q) -> Option<usize>
where
Q: ?Sized + Hash + Equivalent<T>,
{
self.inner.get_index_of(value)
}
pub fn remove<Q>(&mut self, value: &Q) -> bool
where
Q: ?Sized + Hash + Equivalent<T>,
{
self.inner.shift_remove(value)
}
pub fn swap_remove<Q>(&mut self, value: &Q) -> bool
where
Q: ?Sized + Hash + Equivalent<T>,
{
self.inner.swap_remove(value)
}
pub fn take<Q>(&mut self, value: &Q) -> Option<T>
where
Q: ?Sized + Hash + Equivalent<T>,
{
self.inner.shift_take(value)
}
pub fn swap_take<Q>(&mut self, value: &Q) -> Option<T>
where
Q: ?Sized + Hash + Equivalent<T>,
{
self.inner.swap_take(value)
}
pub fn remove_full<Q>(&mut self, value: &Q) -> Option<(usize, T)>
where
Q: ?Sized + Hash + Equivalent<T>,
{
self.inner.shift_remove_full(value)
}
pub fn swap_remove_full<Q>(&mut self, value: &Q) -> Option<(usize, T)>
where
Q: ?Sized + Hash + Equivalent<T>,
{
self.inner.swap_remove_full(value)
}
}
impl<T, S> OrderSet<T, S> {
#[doc(alias = "pop_last")] pub fn pop(&mut self) -> Option<T> {
self.inner.pop()
}
pub fn retain<F>(&mut self, keep: F)
where
F: FnMut(&T) -> bool,
{
self.inner.retain(keep)
}
pub fn sort(&mut self)
where
T: Ord,
{
self.inner.sort()
}
pub fn sort_by<F>(&mut self, cmp: F)
where
F: FnMut(&T, &T) -> Ordering,
{
self.inner.sort_by(cmp);
}
pub fn sorted_by<F>(self, cmp: F) -> IntoIter<T>
where
F: FnMut(&T, &T) -> Ordering,
{
self.inner.sorted_by(cmp)
}
pub fn sort_by_key<K, F>(&mut self, sort_key: F)
where
K: Ord,
F: FnMut(&T) -> K,
{
self.inner.sort_by_key(sort_key)
}
pub fn sort_unstable(&mut self)
where
T: Ord,
{
self.inner.sort_unstable()
}
pub fn sort_unstable_by<F>(&mut self, cmp: F)
where
F: FnMut(&T, &T) -> Ordering,
{
self.inner.sort_unstable_by(cmp)
}
pub fn sorted_unstable_by<F>(self, cmp: F) -> IntoIter<T>
where
F: FnMut(&T, &T) -> Ordering,
{
self.inner.sorted_unstable_by(cmp)
}
pub fn sort_unstable_by_key<K, F>(&mut self, sort_key: F)
where
K: Ord,
F: FnMut(&T) -> K,
{
self.inner.sort_unstable_by_key(sort_key)
}
pub fn sort_by_cached_key<K, F>(&mut self, sort_key: F)
where
K: Ord,
F: FnMut(&T) -> K,
{
self.inner.sort_by_cached_key(sort_key)
}
pub fn binary_search(&self, x: &T) -> Result<usize, usize>
where
T: Ord,
{
self.inner.binary_search(x)
}
#[inline]
pub fn binary_search_by<'a, F>(&'a self, f: F) -> Result<usize, usize>
where
F: FnMut(&'a T) -> Ordering,
{
self.inner.binary_search_by(f)
}
#[inline]
pub fn binary_search_by_key<'a, B, F>(&'a self, b: &B, f: F) -> Result<usize, usize>
where
F: FnMut(&'a T) -> B,
B: Ord,
{
self.inner.binary_search_by_key(b, f)
}
#[inline]
pub fn is_sorted(&self) -> bool
where
T: PartialOrd,
{
self.inner.is_sorted()
}
#[inline]
pub fn is_sorted_by<'a, F>(&'a self, cmp: F) -> bool
where
F: FnMut(&'a T, &'a T) -> bool,
{
self.inner.is_sorted_by(cmp)
}
#[inline]
pub fn is_sorted_by_key<'a, F, K>(&'a self, sort_key: F) -> bool
where
F: FnMut(&'a T) -> K,
K: PartialOrd,
{
self.inner.is_sorted_by_key(sort_key)
}
#[must_use]
pub fn partition_point<P>(&self, pred: P) -> usize
where
P: FnMut(&T) -> bool,
{
self.inner.partition_point(pred)
}
pub fn reverse(&mut self) {
self.inner.reverse()
}
pub fn as_slice(&self) -> &Slice<T> {
self.inner.as_slice()
}
pub fn into_boxed_slice(self) -> Box<Slice<T>> {
self.inner.into_boxed_slice()
}
pub fn get_index(&self, index: usize) -> Option<&T> {
self.inner.get_index(index)
}
pub fn get_range<R: RangeBounds<usize>>(&self, range: R) -> Option<&Slice<T>> {
self.inner.get_range(range)
}
pub fn first(&self) -> Option<&T> {
self.inner.first()
}
pub fn last(&self) -> Option<&T> {
self.inner.last()
}
pub fn remove_index(&mut self, index: usize) -> Option<T> {
self.inner.shift_remove_index(index)
}
pub fn swap_remove_index(&mut self, index: usize) -> Option<T> {
self.inner.swap_remove_index(index)
}
#[track_caller]
pub fn move_index(&mut self, from: usize, to: usize) {
self.inner.move_index(from, to)
}
#[track_caller]
pub fn swap_indices(&mut self, a: usize, b: usize) {
self.inner.swap_indices(a, b)
}
}
impl<T, S> Index<usize> for OrderSet<T, S> {
type Output = T;
fn index(&self, index: usize) -> &T {
if let Some(value) = self.get_index(index) {
value
} else {
panic!(
"index out of bounds: the len is {len} but the index is {index}",
len = self.len()
);
}
}
}
impl<T, S> FromIterator<T> for OrderSet<T, S>
where
T: Hash + Eq,
S: BuildHasher + Default,
{
fn from_iter<I: IntoIterator<Item = T>>(iterable: I) -> Self {
Self {
inner: IndexSet::from_iter(iterable),
}
}
}
#[cfg(feature = "std")]
#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
impl<T, const N: usize> From<[T; N]> for OrderSet<T, RandomState>
where
T: Eq + Hash,
{
fn from(arr: [T; N]) -> Self {
Self::from_iter(arr)
}
}
impl<T, S> Extend<T> for OrderSet<T, S>
where
T: Hash + Eq,
S: BuildHasher,
{
fn extend<I: IntoIterator<Item = T>>(&mut self, iterable: I) {
self.inner.extend(iterable);
}
}
impl<'a, T, S> Extend<&'a T> for OrderSet<T, S>
where
T: Hash + Eq + Copy + 'a,
S: BuildHasher,
{
fn extend<I: IntoIterator<Item = &'a T>>(&mut self, iterable: I) {
self.inner.extend(iterable);
}
}
impl<T, S> Default for OrderSet<T, S>
where
S: Default,
{
fn default() -> Self {
OrderSet {
inner: IndexSet::default(),
}
}
}
impl<T, S1, S2> PartialEq<OrderSet<T, S2>> for OrderSet<T, S1>
where
T: PartialEq,
{
fn eq(&self, other: &OrderSet<T, S2>) -> bool {
self.len() == other.len() && self.iter().eq(other)
}
}
impl<T, S> Eq for OrderSet<T, S> where T: Eq {}
impl<T, S1, S2> PartialOrd<OrderSet<T, S2>> for OrderSet<T, S1>
where
T: PartialOrd,
{
fn partial_cmp(&self, other: &OrderSet<T, S2>) -> Option<Ordering> {
self.iter().partial_cmp(other)
}
}
impl<T, S> Ord for OrderSet<T, S>
where
T: Ord,
{
fn cmp(&self, other: &Self) -> Ordering {
self.iter().cmp(other)
}
}
impl<T, S> Hash for OrderSet<T, S>
where
T: Hash,
{
fn hash<H: Hasher>(&self, state: &mut H) {
self.len().hash(state);
for value in self {
value.hash(state);
}
}
}
impl<T, S> OrderSet<T, S>
where
T: Eq + Hash,
S: BuildHasher,
{
pub fn is_disjoint<S2>(&self, other: &OrderSet<T, S2>) -> bool
where
S2: BuildHasher,
{
self.inner.is_disjoint(&other.inner)
}
pub fn is_subset<S2>(&self, other: &OrderSet<T, S2>) -> bool
where
S2: BuildHasher,
{
self.inner.is_subset(&other.inner)
}
pub fn is_superset<S2>(&self, other: &OrderSet<T, S2>) -> bool
where
S2: BuildHasher,
{
self.inner.is_superset(&other.inner)
}
pub fn set_eq<S2>(&self, other: &OrderSet<T, S2>) -> bool
where
S2: BuildHasher,
{
self.inner == other.inner
}
}
impl<T, S1, S2> BitAnd<&OrderSet<T, S2>> for &OrderSet<T, S1>
where
T: Eq + Hash + Clone,
S1: BuildHasher + Default,
S2: BuildHasher,
{
type Output = OrderSet<T, S1>;
fn bitand(self, other: &OrderSet<T, S2>) -> Self::Output {
OrderSet {
inner: &self.inner & &other.inner,
}
}
}
impl<T, S1, S2> BitOr<&OrderSet<T, S2>> for &OrderSet<T, S1>
where
T: Eq + Hash + Clone,
S1: BuildHasher + Default,
S2: BuildHasher,
{
type Output = OrderSet<T, S1>;
fn bitor(self, other: &OrderSet<T, S2>) -> Self::Output {
OrderSet {
inner: &self.inner | &other.inner,
}
}
}
impl<T, S1, S2> BitXor<&OrderSet<T, S2>> for &OrderSet<T, S1>
where
T: Eq + Hash + Clone,
S1: BuildHasher + Default,
S2: BuildHasher,
{
type Output = OrderSet<T, S1>;
fn bitxor(self, other: &OrderSet<T, S2>) -> Self::Output {
OrderSet {
inner: &self.inner ^ &other.inner,
}
}
}
impl<T, S1, S2> Sub<&OrderSet<T, S2>> for &OrderSet<T, S1>
where
T: Eq + Hash + Clone,
S1: BuildHasher + Default,
S2: BuildHasher,
{
type Output = OrderSet<T, S1>;
fn sub(self, other: &OrderSet<T, S2>) -> Self::Output {
OrderSet {
inner: &self.inner - &other.inner,
}
}
}