use super::QueueError;
#[derive(Clone, Debug)]
pub struct Queue<T> {
items: Vec<T>,
max_size: Option<usize>,
reverse: bool,
}
impl<T> Queue<T> {
pub fn new(reverse: bool) -> Self {
Self { items: vec![], max_size: None, reverse }
}
pub fn new_sized(size: usize, reverse: bool) -> Self {
Self { items: vec![], max_size: Some(size), reverse }
}
pub fn clear(&mut self) {
self.items = vec![];
}
pub fn len(&self) -> usize {
self.items.len()
}
pub fn is_full(&self) -> bool {
if let Some(max_size) = self.max_size {
return self.len() >= max_size;
}
false
}
pub fn is_reversed(&self) -> bool {
self.reverse
}
pub fn add(&mut self, item: T) {
if self.reverse {
self.items.push(item);
} else {
self.items.insert(0, item);
}
}
pub fn try_add(&mut self, item: T) -> Result<(), QueueError> {
if self.is_full() {
Err(QueueError::Full)
} else {
self.add(item);
Ok(())
}
}
pub fn pop(&mut self) -> Option<T> {
self.items.pop()
}
pub fn first(&self) -> Option<&T> {
self.items.last()
}
}
impl<T> Iterator for Queue<T> {
type Item = T;
fn next(&mut self) -> Option<Self::Item> {
self.pop()
}
}