closer

Author: molpopgen

src/trees/tree.rs

@@ -312,6 +312,10 @@ impl<'treeseq> Tree<'treeseq> {
     pub fn parents<N: Into<NodeId> + Copy>(&self, u: N) -> impl Iterator<Item = NodeId> + '_ {
         self.inner.parents(u.into())
     }
+
+    pub fn as_ref(&self) -> &ll_bindings::tsk_tree_t {
+        self.inner.as_ref()
+    }
 }
 
 impl<'ts> streaming_iterator::StreamingIterator for Tree<'ts> {

tests/test_trees.rs

@@ -326,14 +326,14 @@ fn test_iterate_samples_two_trees() {
         {
             let mut nodes: Vec<NodeId> = vec![
                 NodeId::NULL;
-                unsafe { tskit::bindings::tsk_tree_get_size_bound(tree.as_ptr()) }
+                unsafe { tskit::bindings::tsk_tree_get_size_bound(tree.as_ref()) }
                     as usize
             ];
             let mut num_nodes: tskit::bindings::tsk_size_t = 0;
             let ptr = std::ptr::addr_of_mut!(num_nodes);
             unsafe {
                 tskit::bindings::tsk_tree_preorder(
-                    tree.as_ptr(),
+                    tree.as_ref(),
                     nodes.as_mut_ptr() as *mut tskit::bindings::tsk_id_t,
                     ptr,
                 );