new trait

tskit-dev · Jul 11, 2024 · ad06d6f · ad06d6f
1 parent d8c4bce
commit ad06d6f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/traits.rs b/src/traits.rs
@@ -128,6 +128,12 @@ pub trait TableAccess {
     fn migrations(&self) -> &crate::MigrationTable;
 }
 
+pub trait TableIteration: TableAccess {
+    fn edges_iter(&self) -> impl Iterator<Item = crate::EdgeTableRow> + '_ {
+        self.edges().iter()
+    }
+}
+
 impl TableAccess for crate::TableCollection {
     fn edges(&self) -> &crate::EdgeTable {
         self.edges()
@@ -171,3 +177,5 @@ impl TableAccess for crate::TreeSequence {
         self.tables().migrations()
     }
 }
+
+impl<T> TableIteration for T where T: TableAccess {}