rad

Author: molpopgen

src/table_column.rs

@@ -39,6 +39,9 @@ pub trait TableColumn<I, T>:
     + std::ops::Index<crate::SizeType, Output = T>
 {
     fn as_slice(&self) -> &[T];
+    fn get(&self, at: usize) -> Option<&T>;
+    fn get_with_id(&self, at: I) -> Option<&T>;
+    fn get_with_size_type(&self, at: crate::SizeType) -> Option<&T>;
 }
 
 impl<I, T> TableColumn<I, T> for OpaqueTableColumn<'_, T>
@@ -48,6 +51,18 @@ where
     fn as_slice(&self) -> &[T] {
         self.0
     }
+
+    fn get(&self, at: usize) -> Option<&T> {
+        self.0.get(at)
+    }
+
+    fn get_with_id(&self, at: I) -> Option<&T> {
+        self.0.get(at.try_into().unwrap())
+    }
+
+    fn get_with_size_type(&self, at: crate::SizeType) -> Option<&T> {
+        self.0.get(usize::try_from(at).unwrap())
+    }
 }
 // macro_rules! make_table_column {
 //     ($name: ident, $index: ident) => {

tests/test_tables.rs

@@ -1,3 +1,5 @@
+use tskit::TableColumn;
+
 #[test]
 fn test_empty_table_collection() {
     macro_rules! validate_empty_tables {