1use crate::arena::{Arena, Handle, UniqueArena};
4
5#[derive(Debug)]
7pub struct HandleSet<T> {
8 len: usize,
10
11 members: bit_set::BitSet,
13
14 as_keys: core::marker::PhantomData<T>,
16}
17
18impl<T> HandleSet<T> {
19 pub fn new() -> Self {
21 Self {
22 len: 0,
23 members: bit_set::BitSet::new(),
24 as_keys: core::marker::PhantomData,
25 }
26 }
27
28 pub fn is_empty(&self) -> bool {
29 self.members.is_empty()
30 }
31
32 pub fn for_arena(arena: &impl ArenaType<T>) -> Self {
34 let len = arena.len();
35 Self {
36 len,
37 members: bit_set::BitSet::with_capacity(len),
38 as_keys: core::marker::PhantomData,
39 }
40 }
41
42 pub fn clear(&mut self) {
44 self.members.clear();
45 }
46
47 pub fn clear_for_arena(&mut self, arena: &impl ArenaType<T>) {
49 self.members.clear();
50 self.members.reserve_len(arena.len());
51 }
52
53 pub fn all_possible(&self) -> impl Iterator<Item = Handle<T>> {
56 super::Range::full_range_from_size(self.len)
57 }
58
59 pub fn insert(&mut self, handle: Handle<T>) -> bool {
63 self.members.insert(handle.index())
64 }
65
66 pub fn remove(&mut self, handle: Handle<T>) -> bool {
70 self.members.remove(handle.index())
71 }
72
73 pub fn insert_iter(&mut self, iter: impl IntoIterator<Item = Handle<T>>) {
75 for handle in iter {
76 self.insert(handle);
77 }
78 }
79
80 pub fn add_all(&mut self) {
82 self.members.get_mut().set_all();
83 }
84
85 pub fn contains(&self, handle: Handle<T>) -> bool {
86 self.members.contains(handle.index())
87 }
88
89 pub fn iter(&self) -> impl '_ + Iterator<Item = Handle<T>> {
91 self.members.iter().map(Handle::from_usize)
92 }
93
94 pub fn pop(&mut self) -> Option<Handle<T>> {
97 let members = core::mem::take(&mut self.members);
98 let mut vec = members.into_bit_vec();
99 let result = vec.iter_mut().enumerate().rev().find_map(|(i, mut bit)| {
100 if *bit {
101 *bit = false;
102 Some(i)
103 } else {
104 None
105 }
106 });
107 self.members = bit_set::BitSet::from_bit_vec(vec);
108 result.map(Handle::from_usize)
109 }
110}
111
112impl<T> Default for HandleSet<T> {
113 fn default() -> Self {
114 Self::new()
115 }
116}
117
118pub trait ArenaType<T> {
119 fn len(&self) -> usize;
120}
121
122impl<T> ArenaType<T> for Arena<T> {
123 fn len(&self) -> usize {
124 self.len()
125 }
126}
127
128impl<T: core::hash::Hash + Eq> ArenaType<T> for UniqueArena<T> {
129 fn len(&self) -> usize {
130 self.len()
131 }
132}