more

Author: molpopgen

src/sys/tree.rs

@@ -144,6 +144,14 @@ impl<'treeseq> LLTree<'treeseq> {
         Ok(NodeIteratorAdapter(SamplesIterator::new(self, u)?))
     }
 
+    pub fn parent(&self, u: NodeId) -> Option<NodeId> {
+        super::tsk_column_access::<NodeId, _, _, _>(
+            u,
+            self.as_ref().parent,
+            self.treeseq.num_nodes_raw() + 1,
+        )
+    }
+
     pub fn flags(&self) -> TreeFlags {
         self.flags
     }

src/trees/tree.rs

@@ -139,6 +139,13 @@ impl<'treeseq> Tree<'treeseq> {
     pub fn flags(&self) -> TreeFlags {
         self.inner.flags()
     }
+
+    /// Get the parent of node `u`.
+    ///
+    /// Returns `None` if `u` is out of range.
+    pub fn parent<N: Into<NodeId> + Copy + std::fmt::Debug>(&self, u: N) -> Option<NodeId> {
+        self.inner.parent(u.into())
+    }
 }
 
 impl<'ts> streaming_iterator::StreamingIterator for Tree<'ts> {