use alloc::vec::Vec;
use hashbrown::HashMap;
#[derive(Debug, Clone)]
pub struct OrderedMap<K, V> {
values: Vec<V>,
map: HashMap<K, usize>,
}
impl<K, V> Default for OrderedMap<K, V> {
fn default() -> Self {
Self {
values: Default::default(),
map: Default::default(),
}
}
}
impl<K, V> OrderedMap<K, V>
where
K: PartialEq + Eq + core::hash::Hash,
{
pub fn new() -> Self {
Self::default()
}
#[allow(dead_code)]
pub fn len(&self) -> usize {
self.values.len()
}
#[allow(dead_code)]
pub fn is_empty(&self) -> bool {
self.values.is_empty()
}
pub fn push_insert(&mut self, key: K, value: V) {
let idx = self.values.len();
self.values.push(value);
self.map.insert(key, idx);
}
pub fn get_by_key<Q>(&self, key: &Q) -> Option<&V>
where
K: alloc::borrow::Borrow<Q>,
Q: core::hash::Hash + Eq + ?Sized,
{
self.map.get(key).and_then(|&v| self.values.get(v))
}
pub fn get_by_index(&self, i: usize) -> Option<&V> {
self.values.get(i)
}
pub fn values(&self) -> &[V] {
&self.values
}
}
impl<K, V> FromIterator<(K, V)> for OrderedMap<K, V>
where
K: PartialEq + Eq + core::hash::Hash,
{
fn from_iter<T: IntoIterator<Item = (K, V)>>(iter: T) -> Self {
let mut map = OrderedMap::new();
for (k, v) in iter {
map.push_insert(k, v)
}
map
}
}