Add ability to access child node by index

1d10bd0cdbf9f4d93ad572b50e3ff98ffbee595c · ruwaq David Flemström · 2019-12-13T18:43:35Z (6 wata ñawpaq) · tayta b152b204d · kay commitpi sach'ata qhaway

Semántico sayk'uy · 1

src/node.rs

musuq fn child_at_index pub

1 qillqakuna · +5 −0

src/node.rs +5 −0

diff --git a/src/node.rs b/src/node.rs
index fb40d0b9f..7f3693712 100644
--- a/src/node.rs
+++ b/src/node.rs
@@ -186,6 +186,11 @@ impl Stretch {
         Ok(self.forest.children[id].iter().map(|child| self.ids_to_nodes[child]).collect())
     }
 
+    pub fn child_at_index(&self, node: Node, index: usize) -> Result<Node, Error> {
+        let id = self.find_node(node)?;
+        Ok(self.ids_to_nodes[&self.forest.children[id][index]])
+    }
+
     pub fn child_count(&self, node: Node) -> Result<usize, Error> {
         let id = self.find_node(node)?;
         Ok(self.forest.children[id].len())