cl-structures: Sketch out a type-safe generic interning pool

cl-structures/src/intern_pool.rs

@@ -0,0 +1,124 @@
+//! Trivially-copyable, easily comparable typed indices, and a [Pool] to contain them
+//!
+//! # Examples
+//!
+//! ```rust
+//! # use cl_structures::intern_pool::*;
+//! // first, create a new InternKey type (this ensures type safety)
+//! make_intern_key!{
+//!     NumbersKey
+//! }
+//!
+//! // then, create a pool with that type
+//! let mut numbers: Pool<i32, NumbersKey> = Pool::new();
+//! let first = numbers.insert(1);
+//! let second = numbers.insert(2);
+//! let third = numbers.insert(3);
+//!
+//! // You can access elements immutably with `get`
+//! assert_eq!(Some(&3), numbers.get(third));
+//! assert_eq!(Some(&2), numbers.get(second));
+//! // or by indexing
+//! assert_eq!(1, numbers[first]);
+//!
+//! // Or mutably
+//! *numbers.get_mut(first).unwrap() = 100000;
+//!
+//! assert_eq!(Some(&100000), numbers.get(first));
+//! ```
+
+/// Creates newtype indices over [`usize`] for use as [Pool] keys.
+#[macro_export]
+macro_rules! make_intern_key {($($(#[$meta:meta])* $name:ident),*$(,)?) => {$(
+    $(#[$meta])*
+    #[repr(transparent)]
+    #[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
+    pub struct $name(usize);
+
+    impl $crate::intern_pool::InternKey for $name {
+        #[doc = concat!("Constructs a [`", stringify!($name), "`] from a [`usize`] without checking bounds.\n")]
+        /// # Safety
+        ///
+        /// The provided value should be within the bounds of its associated container
+        unsafe fn from_raw_unchecked(value: usize) -> Self {
+            Self(value)
+        }
+        fn get(&self) -> usize {
+            self.0
+        }
+    }
+    impl From< $name > for usize {
+        fn from(value: $name) -> Self {
+            value.0
+        }
+    }
+)*}}
+use std::ops::{Index, IndexMut};
+
+pub use make_intern_key;
+
+/// An index into a [Pool]. For full type-safety,
+/// there should be a unique [InternKey] for each [Pool]
+pub trait InternKey: std::fmt::Debug {
+    /// Constructs an [`InternKey`] from a [`usize`] without checking bounds.
+    ///
+    /// # Safety
+    ///
+    /// The provided value should be within the bounds of its associated container.
+    // ID::from_raw_unchecked here isn't *actually* unsafe, since bounds should always be
+    // checked, however, the function has unverifiable preconditions.
+    unsafe fn from_raw_unchecked(value: usize) -> Self;
+    /// Gets the index of the [`InternKey`] by value
+    fn get(&self) -> usize;
+}
+
+#[derive(Clone, Debug, PartialEq, Eq)]
+pub struct Pool<T, ID: InternKey> {
+    pool: Vec<T>,
+    id_type: std::marker::PhantomData<ID>,
+}
+
+impl<T, ID: InternKey> Pool<T, ID> {
+    pub fn new() -> Self {
+        Self::default()
+    }
+    pub fn get(&self, index: ID) -> Option<&T> {
+        self.pool.get(index.get())
+    }
+    pub fn get_mut(&mut self, index: ID) -> Option<&mut T> {
+        self.pool.get_mut(index.get())
+    }
+
+    pub fn insert(&mut self, value: T) -> ID {
+        let id = self.pool.len();
+        self.pool.push(value);
+
+        // Safety: value was pushed to `self.pool[id]`
+        unsafe { ID::from_raw_unchecked(id) }
+    }
+}
+
+impl<T, ID: InternKey> Default for Pool<T, ID> {
+    fn default() -> Self {
+        Self { pool: vec![], id_type: std::marker::PhantomData }
+    }
+}
+
+impl<T, ID: InternKey> Index<ID> for Pool<T, ID> {
+    type Output = T;
+
+    fn index(&self, index: ID) -> &Self::Output {
+        match self.pool.get(index.get()) {
+            None => panic!("Index {:?} out of bounds in pool!", index),
+            Some(value) => value,
+        }
+    }
+}
+impl<T, ID: InternKey> IndexMut<ID> for Pool<T, ID> {
+    fn index_mut(&mut self, index: ID) -> &mut Self::Output {
+        match self.pool.get_mut(index.get()) {
+            None => panic!("Index {:?} out of bounds in pool!", index),
+            Some(value) => value,
+        }
+    }
+}

cl-structures/src/lib.rs

@@ -10,3 +10,5 @@ pub mod span;
 pub mod tree;
 
 pub mod stack;
+
+pub mod intern_pool;