Source code
Revision control
Copy as Markdown
Other Tools
use alloc::vec::Vec;
/// An item in a [`Table`].
pub trait Item {
/// The type of identifier for the item.
type Id: Id;
/// Return `True` if the item is deleted.
fn is_deleted(&self) -> bool;
}
/// An identifier for referring to an item in a [`Table`].
pub trait Id: IdPrivate {
/// Return the index of the item in the table.
fn index(&self) -> usize;
}
mod id_private {
pub trait IdPrivate {
fn new(id: usize) -> Self;
}
}
pub(super) use id_private::IdPrivate;
/// A table of items.
///
/// Each item has a unique identifier.
/// Items can be deleted without changing the identifiers of other items.
#[derive(Debug)]
pub struct Table<T>(Vec<T>);
impl<T> Table<T> {
pub(super) fn new() -> Self {
Table(Vec::new())
}
}
impl<T: Item> Table<T> {
pub(super) fn next_id(&self) -> T::Id {
T::Id::new(self.0.len())
}
pub(super) fn push(&mut self, item: T) -> &mut T {
self.0.push(item);
self.0.last_mut().unwrap()
}
/// Number of items, including deleted items.
pub(super) fn len(&self) -> usize {
self.0.len()
}
/// Return `True` if there are no non-deleted items.
pub fn is_empty(&self) -> bool {
self.into_iter().next().is_none()
}
/// Number of non-deleted items.
pub fn count(&self) -> usize {
self.into_iter().count()
}
/// Return a reference to an item.
pub fn get(&self, id: T::Id) -> &T {
self.0.get(id.index()).unwrap()
}
/// Return a mutable reference to a segment.
pub fn get_mut(&mut self, id: T::Id) -> &mut T {
self.0.get_mut(id.index()).unwrap()
}
/// Return an iterator for the segments.
pub fn iter(&self) -> TableIter<'_, T> {
self.into_iter()
}
/// Return a mutable iterator for the segments.
pub fn iter_mut(&mut self) -> TableIterMut<'_, T> {
self.into_iter()
}
}
impl<'a, T: Item> IntoIterator for &'a Table<T> {
type Item = &'a T;
type IntoIter = TableIter<'a, T>;
fn into_iter(self) -> TableIter<'a, T> {
TableIter {
iter: self.0.iter(),
}
}
}
impl<'a, T: Item> IntoIterator for &'a mut Table<T> {
type Item = &'a mut T;
type IntoIter = TableIterMut<'a, T>;
fn into_iter(self) -> TableIterMut<'a, T> {
TableIterMut {
iter: self.0.iter_mut(),
}
}
}
/// An iterator for non-deleted items in a [`Table`].
#[derive(Debug)]
pub struct TableIter<'a, T> {
iter: core::slice::Iter<'a, T>,
}
impl<'a, T: Item> Iterator for TableIter<'a, T> {
type Item = &'a T;
fn next(&mut self) -> Option<&'a T> {
self.iter.find(|item| !item.is_deleted())
}
}
/// An iterator for non-deleted items in a [`Table`].
#[derive(Debug)]
pub struct TableIterMut<'a, T> {
iter: core::slice::IterMut<'a, T>,
}
impl<'a, T: Item> Iterator for TableIterMut<'a, T> {
type Item = &'a mut T;
fn next(&mut self) -> Option<&'a mut T> {
self.iter.find(|item| !item.is_deleted())
}
}