X-Git-Url: http://git.megacz.com/?p=sbp.git;a=blobdiff_plain;f=src%2Fedu%2Fberkeley%2Fsbp%2Fmisc%2FTreeWalker.java;h=f8a44963011d850d6914941229125cc4ebbf4568;hp=38cbb817f17d36ba18fcc5c3ce0e2764b12a9073;hb=87f214f3da9f43c3ab93923313845c372f9a96be;hpb=0a0227b9180534d2a431f3d6e08a398bde2244c4

diff --git a/src/edu/berkeley/sbp/misc/TreeWalker.java b/src/edu/berkeley/sbp/misc/TreeWalker.java
index 38cbb81..f8a4496 100644
--- a/src/edu/berkeley/sbp/misc/TreeWalker.java
+++ b/src/edu/berkeley/sbp/misc/TreeWalker.java
@@ -10,6 +10,12 @@ import edu.berkeley.sbp.*;
 
 public abstract class TreeWalker<T> {
     public abstract Object walk(T head, Object[] args);
+    public Object walk(Tree<T> tree, int idx) {
+        return walk(tree.child(idx));
+    }
+    public void walkChildren(Tree<T> tree) {
+        for(Tree<T> child : tree.children()) walk(child);        
+    }
     public Object walk(Tree<T> tree) {
         Object[] args = new Object[tree.numChildren()];
         int i = 0;