use indexed_set::IdxSetBuf;
use indexed_vec::Idx;
use std::collections::VecDeque;
pub struct WorkQueue<T: Idx> {
deque: VecDeque<T>,
set: IdxSetBuf<T>,
}
impl<T: Idx> WorkQueue<T> {
#[inline]
pub fn with_all(len: usize) -> Self {
WorkQueue {
deque: (0..len).map(T::new).collect(),
set: IdxSetBuf::new_filled(len),
}
}
#[inline]
pub fn with_none(len: usize) -> Self {
WorkQueue {
deque: VecDeque::with_capacity(len),
set: IdxSetBuf::new_empty(len),
}
}
#[inline]
pub fn insert(&mut self, element: T) -> bool {
if self.set.add(&element) {
self.deque.push_back(element);
true
} else {
false
}
}
#[inline]
pub fn pop(&mut self) -> Option<T> {
if let Some(element) = self.deque.pop_front() {
self.set.remove(&element);
Some(element)
} else {
None
}
}
#[inline]
pub fn is_empty(&self) -> bool {
self.deque.is_empty()
}
}