use indexmap::Equivalent;
use core::cmp::Ordering;
use core::fmt;
use core::hash::BuildHasher;
use core::hash::Hash;
use core::hash::Hasher;
use core::ops::RangeBounds;
use indexmap::IndexMap;
pub use indexmap::map::Drain;
pub use indexmap::map::Entry;
pub use indexmap::map::IntoIter;
pub use indexmap::map::IntoKeys;
pub use indexmap::map::IntoValues;
pub use indexmap::map::Iter;
pub use indexmap::map::IterMut;
pub use indexmap::map::Keys;
pub use indexmap::map::Values;
pub use indexmap::map::ValuesMut;
pub struct OrderedMap<K, V, S = crate::BuildHasher> {
pub(crate) inner: IndexMap<K, V, S>,
}
impl<K, V> OrderedMap<K, V, crate::BuildHasher> {
#[inline]
pub fn new() -> Self {
Self::default()
}
#[inline]
pub fn with_capacity(capacity: usize) -> Self {
Self::with_capacity_and_hasher(capacity, crate::BuildHasher::default())
}
}
impl<K, V, S> OrderedMap<K, V, S> {
#[inline]
pub fn with_capacity_and_hasher(capacity: usize, hasher: S) -> Self {
Self {
inner: IndexMap::<K, V, S>::with_capacity_and_hasher(capacity, hasher),
}
}
#[inline]
pub fn len(&self) -> usize {
self.inner.len()
}
#[inline]
pub fn capacity(&self) -> usize {
self.inner.capacity()
}
#[inline]
pub fn keys(&self) -> Keys<'_, K, V> {
self.inner.keys()
}
#[inline]
pub fn into_keys(self) -> IntoKeys<K, V> {
self.inner.into_keys()
}
#[inline]
pub fn values(&self) -> Values<'_, K, V> {
self.inner.values()
}
#[inline]
pub fn values_mut(&mut self) -> ValuesMut<'_, K, V> {
self.inner.values_mut()
}
#[inline]
pub fn into_values(self) -> IntoValues<K, V> {
self.inner.into_values()
}
#[inline]
pub fn iter(&self) -> Iter<'_, K, V> {
self.inner.iter()
}
#[inline]
pub fn iter_mut(&mut self) -> IterMut<'_, K, V> {
self.inner.iter_mut()
}
#[inline]
pub fn is_empty(&self) -> bool {
self.inner.is_empty()
}
#[inline]
pub fn drain<R>(&mut self, range: R) -> Drain<'_, K, V>
where
R: RangeBounds<usize>,
{
self.inner.drain(range)
}
#[inline]
pub fn clear_no_shrink(&mut self) {
self.inner.clear();
}
#[inline]
pub fn hasher(&self) -> &S {
self.inner.hasher()
}
#[inline]
pub fn inner(&self) -> &IndexMap<K, V, S> {
&self.inner
}
#[inline]
pub fn inner_mut(&mut self) -> &mut IndexMap<K, V, S> {
&mut self.inner
}
#[inline]
pub fn into_inner(self) -> IndexMap<K, V, S> {
self.inner
}
}
impl<K, V, const N: usize> From<[(K, V); N]> for OrderedMap<K, V, crate::BuildHasher>
where
K: Hash + Eq,
{
fn from(arr: [(K, V); N]) -> Self {
Self {
inner: IndexMap::<K, V, crate::BuildHasher>::from_iter(arr),
}
}
}
impl<K, V, S> OrderedMap<K, V, S>
where
K: Hash + Eq,
S: BuildHasher,
{
#[inline]
pub fn retain<F>(&mut self, keep: F)
where
F: FnMut(&K, &mut V) -> bool,
{
self.inner.retain(keep);
}
#[inline]
pub fn reserve(&mut self, additional: usize) {
self.inner.reserve(additional);
}
#[inline]
pub fn shrink_to_fit(&mut self) {
self.inner.shrink_to_fit();
}
#[inline]
pub fn shrink_to(&mut self, min_capacity: usize) {
self.inner.shrink_to(min_capacity);
}
#[inline]
pub fn clear(&mut self) {
self.clear_and_shrink();
}
#[inline]
pub fn clear_and_shrink(&mut self) {
if self.capacity() > 128 {
let new_cap = 128usize.max((self.len() as f64 * 1.1) as usize);
self.clear_and_shrink_to(new_cap);
} else {
self.clear_no_shrink();
}
}
#[inline]
pub fn clear_and_shrink_to(&mut self, capacity: usize) {
self.clear_no_shrink();
self.shrink_to(capacity);
}
#[inline]
pub fn entry(&mut self, key: K) -> Entry<'_, K, V> {
self.inner.entry(key)
}
#[inline]
pub fn get<Q>(&self, k: &Q) -> Option<&V>
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.get(k)
}
#[inline]
pub fn get_key_value<Q>(&self, k: &Q) -> Option<(&K, &V)>
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.get_key_value(k)
}
pub fn get_full<Q>(&self, key: &Q) -> Option<(usize, &K, &V)>
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.get_full(key)
}
#[inline]
pub fn get_mut<Q>(&mut self, k: &Q) -> Option<&mut V>
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.get_mut(k)
}
pub fn get_full_mut<Q>(&mut self, key: &Q) -> Option<(usize, &K, &mut V)>
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.get_full_mut(key)
}
#[inline]
pub fn contains_key<Q>(&self, key: &Q) -> bool
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.contains_key(key)
}
#[inline]
pub fn get_index_of<Q>(&self, key: &Q) -> Option<usize>
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.get_index_of(key)
}
#[inline]
pub fn insert(&mut self, k: K, v: V) -> Option<V> {
self.inner.insert(k, v)
}
#[inline]
pub fn insert_unique(&mut self, k: K, v: V) {
if self.inner.insert(k, v).is_some() {
crate::panic_key_already_existed();
}
}
#[inline]
pub fn insert_full(&mut self, key: K, value: V) -> (usize, Option<V>) {
self.inner.insert_full(key, value)
}
#[inline]
pub fn remove<Q>(&mut self, k: &Q) -> Option<V>
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.swap_remove(k)
}
#[inline]
pub fn remove_entry<Q>(&mut self, key: &Q) -> Option<(K, V)>
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.swap_remove_entry(key)
}
#[inline]
pub fn swap_remove<Q>(&mut self, key: &Q) -> Option<V>
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.swap_remove(key)
}
#[inline]
pub fn swap_remove_entry<Q>(&mut self, key: &Q) -> Option<(K, V)>
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.swap_remove_entry(key)
}
#[inline]
pub fn swap_remove_full<Q>(&mut self, key: &Q) -> Option<(usize, K, V)>
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.swap_remove_full(key)
}
#[inline]
pub fn shift_remove<Q>(&mut self, key: &Q) -> Option<V>
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.shift_remove(key)
}
#[inline]
pub fn shift_remove_entry<Q>(&mut self, key: &Q) -> Option<(K, V)>
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.shift_remove_entry(key)
}
#[inline]
pub fn shift_remove_full<Q>(&mut self, key: &Q) -> Option<(usize, K, V)>
where
Q: ?Sized + Hash + Equivalent<K>,
{
self.inner.shift_remove_full(key)
}
#[inline]
pub fn pop(&mut self) -> Option<(K, V)> {
self.inner.pop()
}
#[inline]
pub fn sort_keys(&mut self)
where
K: Ord,
{
self.inner.sort_keys();
}
#[inline]
pub fn sort_by<F>(&mut self, cmp: F)
where
F: FnMut(&K, &V, &K, &V) -> Ordering,
{
self.inner.sort_by(cmp);
}
#[inline]
pub fn sorted_by<F>(self, cmp: F) -> IntoIter<K, V>
where
F: FnMut(&K, &V, &K, &V) -> Ordering,
{
self.inner.sorted_by(cmp)
}
#[inline]
pub fn sort_unstable_keys(&mut self)
where
K: Ord,
{
self.inner.sort_unstable_keys();
}
#[inline]
pub fn sort_unstable_by<F>(&mut self, cmp: F)
where
F: FnMut(&K, &V, &K, &V) -> Ordering,
{
self.inner.sort_unstable_by(cmp);
}
#[inline]
pub fn sorted_unstable_by<F>(self, cmp: F) -> IntoIter<K, V>
where
F: FnMut(&K, &V, &K, &V) -> Ordering,
{
self.inner.sorted_unstable_by(cmp)
}
#[inline]
pub fn reverse(&mut self) {
self.inner.reverse();
}
}
impl<K, V, S> OrderedMap<K, V, S> {
#[inline]
pub fn get_index(&self, index: usize) -> Option<(&K, &V)> {
self.inner.get_index(index)
}
#[inline]
pub fn get_index_mut(&mut self, index: usize) -> Option<(&K, &mut V)> {
self.inner.get_index_mut(index)
}
#[inline]
pub fn first(&self) -> Option<(&K, &V)> {
self.inner.first()
}
#[inline]
pub fn first_mut(&mut self) -> Option<(&K, &mut V)> {
self.inner.first_mut()
}
#[inline]
pub fn last(&self) -> Option<(&K, &V)> {
self.inner.last()
}
#[inline]
pub fn last_mut(&mut self) -> Option<(&K, &mut V)> {
self.inner.last_mut()
}
#[inline]
pub fn swap_remove_index(&mut self, index: usize) -> Option<(K, V)> {
self.inner.swap_remove_index(index)
}
#[inline]
pub fn shift_remove_index(&mut self, index: usize) -> Option<(K, V)> {
self.inner.shift_remove_index(index)
}
#[inline]
pub fn move_index(&mut self, from: usize, to: usize) {
self.inner.move_index(from, to);
}
#[inline]
pub fn swap_indices(&mut self, a: usize, b: usize) {
self.inner.swap_indices(a, b);
}
}
impl<K, V, S> Clone for OrderedMap<K, V, S>
where
K: Clone,
V: Clone,
S: Clone,
{
#[inline]
fn clone(&self) -> Self {
Self {
inner: self.inner.clone(),
}
}
#[inline]
fn clone_from(&mut self, other: &Self) {
self.inner.clone_from(&other.inner);
}
}
impl<K, V, S> Default for OrderedMap<K, V, S>
where
S: BuildHasher + Default,
{
#[inline]
fn default() -> Self {
Self {
inner: IndexMap::with_hasher(S::default()),
}
}
}
impl<K, V, S> fmt::Debug for OrderedMap<K, V, S>
where
K: fmt::Debug,
V: fmt::Debug,
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("OrderedMap")
.field("inner", &self.inner)
.finish()
}
}
impl<K, V, S> PartialEq for OrderedMap<K, V, S>
where
K: Eq + Hash,
V: PartialEq,
S: BuildHasher,
{
#[inline]
fn eq(&self, other: &Self) -> bool {
if self.len() != other.len() {
return false;
}
self.inner
.iter()
.zip(other.inner.iter())
.all(|(self_elt, other_elt)| self_elt == other_elt)
}
}
impl<K, V, S> Eq for OrderedMap<K, V, S>
where
K: Eq + Hash,
V: Eq,
S: BuildHasher,
{
}
impl<K, V, S> PartialOrd for OrderedMap<K, V, S>
where
K: Eq + Hash + Ord,
V: Ord,
S: BuildHasher,
{
#[inline]
fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
Some(self.cmp(other))
}
}
impl<K, V, S> Ord for OrderedMap<K, V, S>
where
K: Eq + Hash + Ord,
V: Ord,
S: BuildHasher,
{
fn cmp(&self, other: &Self) -> Ordering {
match self.len().cmp(&other.len()) {
Ordering::Less => return Ordering::Less,
Ordering::Greater => return Ordering::Greater,
Ordering::Equal => (),
}
self.iter().cmp(other.iter())
}
}
impl<K, V, S> Hash for OrderedMap<K, V, S>
where
K: Hash,
V: Hash,
S: BuildHasher,
{
fn hash<H: Hasher>(&self, state: &mut H) {
for elt in self.inner.iter() {
elt.hash(state);
}
}
}
impl<K, V, S> FromIterator<(K, V)> for OrderedMap<K, V, S>
where
K: Eq + Hash,
S: BuildHasher + Default,
{
#[inline]
fn from_iter<T: IntoIterator<Item = (K, V)>>(iter: T) -> Self {
Self {
inner: IndexMap::<K, V, S>::from_iter(iter),
}
}
}
impl<K, V, S> IntoIterator for OrderedMap<K, V, S> {
type Item = (K, V);
type IntoIter = indexmap::map::IntoIter<K, V>;
#[inline]
fn into_iter(self) -> Self::IntoIter {
self.inner.into_iter()
}
}
impl<'a, K, V, S> IntoIterator for &'a OrderedMap<K, V, S> {
type Item = (&'a K, &'a V);
type IntoIter = indexmap::map::Iter<'a, K, V>;
#[inline]
fn into_iter(self) -> Self::IntoIter {
self.inner.iter()
}
}
impl<'a, K, V, S> IntoIterator for &'a mut OrderedMap<K, V, S> {
type Item = (&'a K, &'a mut V);
type IntoIter = indexmap::map::IterMut<'a, K, V>;
#[inline]
fn into_iter(self) -> Self::IntoIter {
self.inner.iter_mut()
}
}
impl<K, V, S> Extend<(K, V)> for OrderedMap<K, V, S>
where
K: Eq + Hash,
S: BuildHasher,
{
#[inline]
fn extend<T: IntoIterator<Item = (K, V)>>(&mut self, iter: T) {
self.inner.extend(iter);
}
}
impl<'a, K, V, S> Extend<(&'a K, &'a V)> for OrderedMap<K, V, S>
where
K: Eq + Hash + Copy,
V: Copy,
S: BuildHasher,
{
#[inline]
fn extend<T: IntoIterator<Item = (&'a K, &'a V)>>(&mut self, iter: T) {
self.inner.extend(iter);
}
}
#[cfg(feature = "serde")]
pub mod serde_as_map {
use super::*;
pub fn serialize<K, V, RS, S>(
map: &OrderedMap<K, V, RS>,
serializer: S,
) -> Result<S::Ok, S::Error>
where
K: serde::Serialize + Hash + Eq,
V: serde::Serialize,
RS: BuildHasher,
S: serde::Serializer,
{
use serde::ser::Serialize;
map.inner.serialize(serializer)
}
pub fn deserialize<'de, K, V, S, D>(deserializer: D) -> Result<OrderedMap<K, V, S>, D::Error>
where
K: serde::Deserialize<'de> + Eq + Hash,
V: serde::Deserialize<'de>,
S: BuildHasher + Default,
D: serde::Deserializer<'de>,
{
use serde::Deserialize;
Ok(OrderedMap {
inner: IndexMap::<K, V, S>::deserialize(deserializer)?,
})
}
}
#[cfg(feature = "serde")]
impl<K, V, RS> serde::Serialize for OrderedMap<K, V, RS>
where
K: serde::Serialize + Eq + Hash,
V: serde::Serialize,
RS: BuildHasher + Default,
{
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
use serde::ser::SerializeSeq;
let mut seq = serializer.serialize_seq(Some(self.inner.len()))?;
for elt in self.inner.iter() {
seq.serialize_element(&elt)?;
}
seq.end()
}
}
#[cfg(feature = "serde")]
impl<'de, K, V, S> serde::Deserialize<'de> for OrderedMap<K, V, S>
where
K: serde::Deserialize<'de> + Eq + Hash,
V: serde::Deserialize<'de>,
S: BuildHasher + Default,
{
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>,
{
use core::marker::PhantomData;
use serde::de::SeqAccess;
use serde::de::Visitor;
struct IndexMapSeqVisitor<K, V, S>(PhantomData<(K, V, S)>);
impl<'de, K, V, S> Visitor<'de> for IndexMapSeqVisitor<K, V, S>
where
K: serde::Deserialize<'de> + Eq + Hash,
V: serde::Deserialize<'de>,
S: BuildHasher + Default,
{
type Value = IndexMap<K, V, S>;
fn expecting(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
formatter.write_str("a sequence of (k, v) pairs")
}
fn visit_seq<A>(self, mut seq: A) -> Result<Self::Value, A::Error>
where
A: SeqAccess<'de>,
{
let mut map =
IndexMap::with_capacity_and_hasher(seq.size_hint().unwrap_or(0), S::default());
while let Some((k, v)) = seq.next_element::<(K, V)>()? {
map.insert(k, v);
}
Ok(map)
}
}
let map = deserializer.deserialize_seq(IndexMapSeqVisitor::<K, V, S>(PhantomData))?;
Ok(Self { inner: map })
}
}
#[cfg(feature = "speedy")]
impl<'a, C, K, V, S> speedy::Readable<'a, C> for OrderedMap<K, V, S>
where
C: speedy::Context,
K: speedy::Readable<'a, C> + Eq + Hash,
V: speedy::Readable<'a, C>,
S: BuildHasher + Default,
{
fn read_from<R: speedy::Reader<'a, C>>(reader: &mut R) -> Result<Self, C::Error> {
let length = reader.read_u32()? as usize;
let map = (0..length)
.map(|_| -> Result<_, <C as speedy::Context>::Error> {
let key = K::read_from(reader)?;
let value = V::read_from(reader)?;
Ok((key, value))
})
.collect::<Result<_, _>>()?;
Ok(Self { inner: map })
}
}
#[cfg(feature = "speedy")]
impl<C, K, V, S> speedy::Writable<C> for OrderedMap<K, V, S>
where
C: speedy::Context,
K: speedy::Writable<C>,
V: speedy::Writable<C>,
{
#[inline]
fn write_to<T: ?Sized + speedy::Writer<C>>(
&self,
writer: &mut T,
) -> Result<(), <C as speedy::Context>::Error> {
writer.write_u32(self.inner.len() as u32)?;
writer.write_collection(self.iter())
}
#[inline]
fn bytes_needed(&self) -> Result<usize, C::Error> {
crate::internal_macros::unsafe_is_length!(self.len());
let mut count = core::mem::size_of::<u32>(); for elt in self {
count += elt.bytes_needed()?;
}
Ok(count)
}
}