1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
use core::sync::atomic::Ordering::Relaxed;
use radium::Radium;
use crate::scalar::{OpaqueScalar, ScalarAllocator};
use super::{Pool, PoolMut};
pub struct Sequence<R, P: ?Sized> {
pub index: R,
pub pools: P,
}
impl<A: ScalarAllocator, R: Radium<Item = usize>, P: PoolMut<A>> PoolMut<A> for Sequence<R, [P]> {
fn insert_mut(&mut self, mut scalar: OpaqueScalar<A>) -> Option<OpaqueScalar<A>> {
let index = self.index.get_mut();
let initial_index = *index;
let initial_index = initial_index % self.pools.len();
loop {
let current = *index % self.pools.len();
if current == initial_index {
return Some(scalar)
}
scalar = self.pools[current].insert_mut(scalar)?;
*index = index.wrapping_add(1);
}
}
fn remove_mut(&mut self) -> Option<OpaqueScalar<A>> {
let index = self.index.get_mut();
let initial_index = *index;
let initial_index = initial_index % self.pools.len();
loop {
let current = *index % self.pools.len();
if current == initial_index {
return None
}
if let Some(scalar) = self.pools[current].remove_mut() {
return Some(scalar)
}
*index = index.wrapping_add(1);
}
}
}
impl<A: ScalarAllocator, R: Radium<Item = usize>, P: Pool<A>> Pool<A> for Sequence<R, [P]> {
fn insert(&self, mut scalar: OpaqueScalar<A>) -> Option<OpaqueScalar<A>> {
let mut initial_index = None;
loop {
let current = self.index.fetch_add(1, Relaxed) % self.pools.len();
match initial_index {
None => initial_index = Some(current),
Some(initial_index) => {
if current == initial_index {
return Some(scalar)
}
}
}
scalar = self.pools[current].insert(scalar)?;
}
}
fn remove(&self) -> Option<OpaqueScalar<A>> {
let mut initial_index = None;
loop {
let current = self.index.fetch_add(1, Relaxed) % self.pools.len();
match initial_index {
None => initial_index = Some(current),
Some(initial_index) => {
if current == initial_index {
return None
}
}
}
if let Some(scalar) = self.pools[current].remove() {
return Some(scalar)
}
}
}
}