cool

Author: molpopgen

src/sys/tree.rs

@@ -43,4 +43,8 @@ impl<'treeseq> LLTree<'treeseq> {
     pub fn as_ptr(&mut self) -> *const tsk_tree_t {
         self.inner.as_ptr()
     }
+
+    pub fn as_ref(&self) -> &tsk_tree_t {
+        self.inner.as_ref()
+    }
 }

src/trees/tree.rs

@@ -19,10 +19,7 @@ impl<'treeseq> Tree<'treeseq> {
     ) -> Result<Self, TskitError> {
         let flags = flags.into();
         let inner = LLTree::new(ts, flags)?;
-        Ok(Self {
-            inner,
-            advanced: 0,
-        })
+        Ok(Self { inner, advanced: 0 })
     }
 
     pub(crate) fn new_at_position<F: Into<TreeFlags>, P: Into<Position>>(
@@ -56,6 +53,14 @@ impl<'treeseq> Tree<'treeseq> {
         };
         Ok(tree)
     }
+
+    /// Return the `[left, right)` coordinates of the tree.
+    pub fn interval(&self) -> (Position, Position) {
+        (
+            self.inner.as_ref().interval.left.into(),
+            self.inner.as_ref().interval.right.into(),
+        )
+    }
 }
 
 impl<'ts> streaming_iterator::StreamingIterator for Tree<'ts> {