more

Author: molpopgen

src/sys/tree.rs

@@ -1,6 +1,8 @@
+use super::bindings::tsk_id_t;
 use super::bindings::tsk_size_t;
 use super::bindings::tsk_tree_t;
 use super::flags::TreeFlags;
+use super::newtypes::NodeId;
 use super::tskbox::TskBox;
 use super::TreeSequence;
 use super::TskitError;
@@ -56,6 +58,11 @@ impl<'treeseq> LLTree<'treeseq> {
         )
     }
 
+    /// Return the virtual root of the tree.
+    pub fn virtual_root(&self) -> tsk_id_t {
+        self.as_ref().virtual_root
+    }
+
     pub fn as_mut_ptr(&mut self) -> *mut tsk_tree_t {
         self.inner.as_mut()
     }

src/trees/tree.rs

@@ -96,6 +96,11 @@ impl<'treeseq> Tree<'treeseq> {
     pub fn samples_array(&self) -> Result<&[NodeId], TskitError> {
         self.inner.samples_array()
     }
+
+    /// Return the virtual root of the tree.
+    pub fn virtual_root(&self) -> NodeId {
+        self.inner.virtual_root().into()
+    }
 }
 
 impl<'ts> streaming_iterator::StreamingIterator for Tree<'ts> {