mirror of
https://github.com/italicsjenga/agb.git
synced 2024-12-23 16:21:33 +11:00
Extract node module
This commit is contained in:
parent
a0ead37eb5
commit
5e04e8c03f
|
@ -23,13 +23,16 @@ use core::{
|
||||||
borrow::Borrow,
|
borrow::Borrow,
|
||||||
hash::{BuildHasher, BuildHasherDefault, Hash, Hasher},
|
hash::{BuildHasher, BuildHasherDefault, Hash, Hasher},
|
||||||
iter::FromIterator,
|
iter::FromIterator,
|
||||||
mem::{self, MaybeUninit},
|
mem,
|
||||||
ops::Index,
|
ops::Index,
|
||||||
ptr,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
use rustc_hash::FxHasher;
|
use rustc_hash::FxHasher;
|
||||||
|
|
||||||
|
mod node;
|
||||||
|
|
||||||
|
use node::Node;
|
||||||
|
|
||||||
type HashType = u32;
|
type HashType = u32;
|
||||||
|
|
||||||
// # Robin Hood Hash Tables
|
// # Robin Hood Hash Tables
|
||||||
|
@ -835,7 +838,7 @@ impl<K, V, ALLOCATOR: ClonableAllocator> NodeStorage<K, V, ALLOCATOR> {
|
||||||
loop {
|
loop {
|
||||||
let location = fast_mod(
|
let location = fast_mod(
|
||||||
self.backing_vec_size(),
|
self.backing_vec_size(),
|
||||||
new_node.hash + new_node.distance() as HashType,
|
new_node.hash() + new_node.distance() as HashType,
|
||||||
);
|
);
|
||||||
let current_node = &mut self.nodes[location];
|
let current_node = &mut self.nodes[location];
|
||||||
|
|
||||||
|
@ -950,142 +953,6 @@ impl<K, V, ALLOCATOR: ClonableAllocator> NodeStorage<K, V, ALLOCATOR> {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
struct Node<K, V> {
|
|
||||||
hash: HashType,
|
|
||||||
|
|
||||||
// distance_to_initial_bucket = -1 => key and value are uninit.
|
|
||||||
// distance_to_initial_bucket >= 0 => key and value are init
|
|
||||||
distance_to_initial_bucket: i32,
|
|
||||||
key: MaybeUninit<K>,
|
|
||||||
value: MaybeUninit<V>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<K, V> Node<K, V> {
|
|
||||||
fn new() -> Self {
|
|
||||||
Self {
|
|
||||||
hash: 0,
|
|
||||||
distance_to_initial_bucket: -1,
|
|
||||||
key: MaybeUninit::uninit(),
|
|
||||||
value: MaybeUninit::uninit(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn new_with(key: K, value: V, hash: HashType) -> Self {
|
|
||||||
Self {
|
|
||||||
hash,
|
|
||||||
distance_to_initial_bucket: 0,
|
|
||||||
key: MaybeUninit::new(key),
|
|
||||||
value: MaybeUninit::new(value),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn value_ref(&self) -> Option<&V> {
|
|
||||||
if self.has_value() {
|
|
||||||
Some(unsafe { self.value.assume_init_ref() })
|
|
||||||
} else {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn value_mut(&mut self) -> Option<&mut V> {
|
|
||||||
if self.has_value() {
|
|
||||||
Some(unsafe { self.value.assume_init_mut() })
|
|
||||||
} else {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn key_ref(&self) -> Option<&K> {
|
|
||||||
if self.distance_to_initial_bucket >= 0 {
|
|
||||||
Some(unsafe { self.key.assume_init_ref() })
|
|
||||||
} else {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn key_value_ref(&self) -> Option<(&K, &V)> {
|
|
||||||
if self.has_value() {
|
|
||||||
Some(unsafe { (self.key.assume_init_ref(), self.value.assume_init_ref()) })
|
|
||||||
} else {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn key_value_mut(&mut self) -> Option<(&K, &mut V)> {
|
|
||||||
if self.has_value() {
|
|
||||||
Some(unsafe { (self.key.assume_init_ref(), self.value.assume_init_mut()) })
|
|
||||||
} else {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn has_value(&self) -> bool {
|
|
||||||
self.distance_to_initial_bucket >= 0
|
|
||||||
}
|
|
||||||
|
|
||||||
fn take_key_value(&mut self) -> Option<(K, V, HashType)> {
|
|
||||||
if self.has_value() {
|
|
||||||
let key = mem::replace(&mut self.key, MaybeUninit::uninit());
|
|
||||||
let value = mem::replace(&mut self.value, MaybeUninit::uninit());
|
|
||||||
self.distance_to_initial_bucket = -1;
|
|
||||||
|
|
||||||
Some(unsafe { (key.assume_init(), value.assume_init(), self.hash) })
|
|
||||||
} else {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn replace_value(&mut self, value: V) -> V {
|
|
||||||
if self.has_value() {
|
|
||||||
let old_value = mem::replace(&mut self.value, MaybeUninit::new(value));
|
|
||||||
unsafe { old_value.assume_init() }
|
|
||||||
} else {
|
|
||||||
panic!("Cannot replace an uninitialised node");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn replace(&mut self, key: K, value: V) -> (K, V) {
|
|
||||||
if self.has_value() {
|
|
||||||
let old_key = mem::replace(&mut self.key, MaybeUninit::new(key));
|
|
||||||
let old_value = mem::replace(&mut self.value, MaybeUninit::new(value));
|
|
||||||
|
|
||||||
unsafe { (old_key.assume_init(), old_value.assume_init()) }
|
|
||||||
} else {
|
|
||||||
panic!("Cannot replace an uninitialised node");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn increment_distance(&mut self) {
|
|
||||||
self.distance_to_initial_bucket += 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
fn decrement_distance(&mut self) {
|
|
||||||
self.distance_to_initial_bucket -= 1;
|
|
||||||
if self.distance_to_initial_bucket < 0 {
|
|
||||||
panic!("Cannot decrement distance to below 0");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn distance(&self) -> i32 {
|
|
||||||
self.distance_to_initial_bucket
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<K, V> Drop for Node<K, V> {
|
|
||||||
fn drop(&mut self) {
|
|
||||||
if self.has_value() {
|
|
||||||
unsafe { ptr::drop_in_place(self.key.as_mut_ptr()) };
|
|
||||||
unsafe { ptr::drop_in_place(self.value.as_mut_ptr()) };
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<K, V> Default for Node<K, V> {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self::new()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod test {
|
||||||
use core::cell::RefCell;
|
use core::cell::RefCell;
|
||||||
|
|
146
agb-hashmap/src/node.rs
Normal file
146
agb-hashmap/src/node.rs
Normal file
|
@ -0,0 +1,146 @@
|
||||||
|
use core::{
|
||||||
|
mem::{self, MaybeUninit},
|
||||||
|
ptr,
|
||||||
|
};
|
||||||
|
|
||||||
|
use crate::HashType;
|
||||||
|
|
||||||
|
pub(crate) struct Node<K, V> {
|
||||||
|
hash: HashType,
|
||||||
|
|
||||||
|
// distance_to_initial_bucket = -1 => key and value are uninit.
|
||||||
|
// distance_to_initial_bucket >= 0 => key and value are init
|
||||||
|
distance_to_initial_bucket: i32,
|
||||||
|
key: MaybeUninit<K>,
|
||||||
|
value: MaybeUninit<V>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<K, V> Node<K, V> {
|
||||||
|
fn new() -> Self {
|
||||||
|
Self {
|
||||||
|
hash: 0,
|
||||||
|
distance_to_initial_bucket: -1,
|
||||||
|
key: MaybeUninit::uninit(),
|
||||||
|
value: MaybeUninit::uninit(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn new_with(key: K, value: V, hash: HashType) -> Self {
|
||||||
|
Self {
|
||||||
|
hash,
|
||||||
|
distance_to_initial_bucket: 0,
|
||||||
|
key: MaybeUninit::new(key),
|
||||||
|
value: MaybeUninit::new(value),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn value_ref(&self) -> Option<&V> {
|
||||||
|
if self.has_value() {
|
||||||
|
Some(unsafe { self.value.assume_init_ref() })
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn value_mut(&mut self) -> Option<&mut V> {
|
||||||
|
if self.has_value() {
|
||||||
|
Some(unsafe { self.value.assume_init_mut() })
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn key_ref(&self) -> Option<&K> {
|
||||||
|
if self.distance_to_initial_bucket >= 0 {
|
||||||
|
Some(unsafe { self.key.assume_init_ref() })
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn key_value_ref(&self) -> Option<(&K, &V)> {
|
||||||
|
if self.has_value() {
|
||||||
|
Some(unsafe { (self.key.assume_init_ref(), self.value.assume_init_ref()) })
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn key_value_mut(&mut self) -> Option<(&K, &mut V)> {
|
||||||
|
if self.has_value() {
|
||||||
|
Some(unsafe { (self.key.assume_init_ref(), self.value.assume_init_mut()) })
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn has_value(&self) -> bool {
|
||||||
|
self.distance_to_initial_bucket >= 0
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn take_key_value(&mut self) -> Option<(K, V, HashType)> {
|
||||||
|
if self.has_value() {
|
||||||
|
let key = mem::replace(&mut self.key, MaybeUninit::uninit());
|
||||||
|
let value = mem::replace(&mut self.value, MaybeUninit::uninit());
|
||||||
|
self.distance_to_initial_bucket = -1;
|
||||||
|
|
||||||
|
Some(unsafe { (key.assume_init(), value.assume_init(), self.hash) })
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn replace_value(&mut self, value: V) -> V {
|
||||||
|
if self.has_value() {
|
||||||
|
let old_value = mem::replace(&mut self.value, MaybeUninit::new(value));
|
||||||
|
unsafe { old_value.assume_init() }
|
||||||
|
} else {
|
||||||
|
panic!("Cannot replace an uninitialised node");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn replace(&mut self, key: K, value: V) -> (K, V) {
|
||||||
|
if self.has_value() {
|
||||||
|
let old_key = mem::replace(&mut self.key, MaybeUninit::new(key));
|
||||||
|
let old_value = mem::replace(&mut self.value, MaybeUninit::new(value));
|
||||||
|
|
||||||
|
unsafe { (old_key.assume_init(), old_value.assume_init()) }
|
||||||
|
} else {
|
||||||
|
panic!("Cannot replace an uninitialised node");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn increment_distance(&mut self) {
|
||||||
|
self.distance_to_initial_bucket += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn decrement_distance(&mut self) {
|
||||||
|
self.distance_to_initial_bucket -= 1;
|
||||||
|
if self.distance_to_initial_bucket < 0 {
|
||||||
|
panic!("Cannot decrement distance to below 0");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn distance(&self) -> i32 {
|
||||||
|
self.distance_to_initial_bucket
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn hash(&self) -> HashType {
|
||||||
|
self.hash
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<K, V> Drop for Node<K, V> {
|
||||||
|
fn drop(&mut self) {
|
||||||
|
if self.has_value() {
|
||||||
|
unsafe { ptr::drop_in_place(self.key.as_mut_ptr()) };
|
||||||
|
unsafe { ptr::drop_in_place(self.value.as_mut_ptr()) };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<K, V> Default for Node<K, V> {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self::new()
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue