progress, but names TBD

Author: molpopgen

src/edge_differences.rs

@@ -1,3 +1,4 @@
+use crate::edge_differences::private::EdgeDifferenceIteration;
 use crate::EdgeId;
 use crate::NodeId;
 use crate::Position;
@@ -116,6 +117,46 @@ pub struct EdgeDifferences<'ts> {
     right: f64,
 }
 
+pub struct X<'ts, T: EdgeDifferenceIteration> {
+    edges_left: &'ts [Position],
+    edges_right: &'ts [Position],
+    edges_parent: &'ts [NodeId],
+    edges_child: &'ts [NodeId],
+    removals: (usize, usize),
+    marker: std::marker::PhantomData<T>,
+}
+
+impl<'ts, T: EdgeDifferenceIteration> Iterator for X<'ts, T> {
+    type Item = EdgeDifference<T>;
+    fn next(&mut self) -> Option<Self::Item> {
+        todo!()
+    }
+}
+
+impl<'ts> EdgeDifferences<'ts> {
+    pub fn removals(&self) -> impl Iterator<Item = EdgeRemoval> + '_ {
+        X {
+            edges_left: self.edges_left,
+            edges_right: self.edges_right,
+            edges_parent: self.edges_parent,
+            edges_child: self.edges_child,
+            removals: self.removals,
+            marker: std::marker::PhantomData::<Removal> {},
+        }
+    }
+
+    pub fn insertions(&self) -> impl Iterator<Item = EdgeInsertion> + '_ {
+        X {
+            edges_left: self.edges_left,
+            edges_right: self.edges_right,
+            edges_parent: self.edges_parent,
+            edges_child: self.edges_child,
+            removals: self.removals,
+            marker: std::marker::PhantomData::<Insertion> {},
+        }
+    }
+}
+
 fn update_right(
     right: f64,
     index: usize,