more

Author: molpopgen

src/sys/tree.rs

@@ -143,6 +143,10 @@ impl<'treeseq> LLTree<'treeseq> {
     pub fn samples(&self, u: NodeId) -> Result<impl Iterator<Item = NodeId> + '_, TskitError> {
         Ok(NodeIteratorAdapter(SamplesIterator::new(self, u)?))
     }
+
+    pub fn flags(&self) -> TreeFlags {
+        self.flags
+    }
 }
 
 // Trait defining iteration over nodes.

src/trees/tree.rs

@@ -135,6 +135,10 @@ impl<'treeseq> Tree<'treeseq> {
     ) -> Result<impl Iterator<Item = NodeId> + '_, TskitError> {
         self.inner.samples(u.into())
     }
+
+    pub fn flags(&self) -> TreeFlags {
+        self.inner.flags()
+    }
 }
 
 impl<'ts> streaming_iterator::StreamingIterator for Tree<'ts> {