alex/agb
1
0
Fork 0
mirror of https://github.com/italicsjenga/agb.git synced 2025-01-11 09:31:34 +11:00
Code Issues Packages Projects Releases Wiki Activity

Add with_capacity method

Browse source
This commit is contained in:
Gwilym Kuiper 2022-03-17 23:54:42 +00:00
parent c0d9f0ab41
commit 03f5cd0953
1 changed files with 16 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
agb/src/hash_map.rs
View file

@ -43,10 +43,16 @@ pub struct HashMap<K, V> {
impl<K, V> HashMap<K, V> {
pub fn new() -> Self {
Self::with_capacity(16)
}
pub fn with_capacity(capacity: usize) -> Self {
let next_power_of_2 = find_next_power_of_2(capacity);
Self {
number_of_elements: 0,
max_distance_to_initial_bucket: 0,
nodes: iter::repeat_with(|| None).take(16).collect(),
nodes: iter::repeat_with(|| None).take(next_power_of_2).collect(),
hasher: Default::default(),
}
}
@ -61,6 +67,15 @@ const fn fast_mod(len: usize, hash: HashType) -> usize {
(hash as usize) & (len - 1)
}
const fn find_next_power_of_2(n: usize) -> usize {
let mut next_power_of_2 = 1;
while next_power_of_2 <= n {
next_power_of_2 *= 2;
}
next_power_of_2
}
impl<K, V> HashMap<K, V>
where
K: Eq,