X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;ds=sidebyside;f=src%2Fedu%2Fberkeley%2Fsbp%2Fmisc%2FDemo.java;h=2b99ffbe3bf33c5c1ffee506e08aaddc8d42cf04;hb=03e1f20f869819893205ec771571b2c62ab00421;hp=63849398c4b76384b4f4f60d1c02d3e218dea432;hpb=ac0f7d32be59d02b1f8d48d2b23232adf8871b46;p=sbp.git

diff --git a/src/edu/berkeley/sbp/misc/Demo.java b/src/edu/berkeley/sbp/misc/Demo.java
index 6384939..2b99ffb 100644
--- a/src/edu/berkeley/sbp/misc/Demo.java
+++ b/src/edu/berkeley/sbp/misc/Demo.java
@@ -91,20 +91,19 @@ public class Demo {
         public Object repeatTag() {
             return new Reducer() {
                     public String toString() { return ""; }
-                    public Object reduce(Tree t) {
-                        Object[] ret = new Object[t.numChildren()];
-                        for(int i=0; i<t.numChildren(); i++) {
-                            Tree tc = t.child(i);
+                    public Object reduce(Iterable<Tree> t) {
+                        ArrayList ret = new ArrayList();
+                        for(Tree tc : t) {
                             if (tc.head() != null && tc.head() instanceof Reducer)
-                                ret[i] = ((Reducer)tc.head()).reduce(tc);
+                                ret.add(((Reducer)tc.head()).reduce(tc.children()));
                             else if (tc.numChildren() == 0)
-                                ret[i] = tc.head();
+                                ret.add(tc.head());
                             else {
                                 System.err.println("FIXME: don't know what to do about " + tc);
-                                ret[i] = null;
+                                ret.add(null);
                             }
                         }
-                        return ret;
+                        return ret.toArray(new Object[0]);
                     }
                 };
         }
@@ -238,41 +237,42 @@ public class Demo {
             }
         }
         public Sequence makeSequence(Production p) {
-            return Sequence.rewritingSequence(new TargetReducer(p), p.elements, p.labels, p.drops);
+            return Sequence.rewritingSequence(new TargetReducer(p, buildSequence(p), "reducer-"+this), p.elements, p.labels, p.drops);
         }
         public abstract Object plant(Object[] fields, int[] map);
         public boolean isRaw() { return false; }
-        public Object invokeRaw(Tree t) { return null; }
+        public Object invokeRaw(Iterable<Tree> t) { return null; }
         public class TargetReducer implements Reducer {
             private Production p;
             private int[] map;
-            public TargetReducer(Production p) {
+            private String name;
+            public TargetReducer(Production p, int[] map, String name) {
                 this.p = p;
-                this.map = buildSequence(p);
+                this.map = map;
+                this.name = name;
             }
-            public String toString() { return "reducer-"+Target.this; }
-            public Object reduce(Tree t) {
+            public String toString() { return name; }
+            public Object reduce(Iterable<Tree> t) {
                 if (isRaw()) return invokeRaw(t);
-                Object[] objects = new Object[t.numChildren()];
-                for(int i=0; i<t.numChildren(); i++) {
-                    Tree tc = t.child(i);
+                ArrayList ret = new ArrayList();
+                for(Tree tc : t) {
                     if (tc.head() != null && tc.head() instanceof Reducer)
-                        objects[i] = ((Reducer)tc.head()).reduce(tc);
+                        ret.add(((Reducer)tc.head()).reduce(tc.children()));
                     else if (tc.numChildren() == 0)
-                        objects[i] = tc.head();
+                        ret.add(tc.head());
                     else {
                         System.err.println("FIXME: don't know what to do about " + tc);
-                        objects[i] = null;
+                        ret.add(null);
                     }
                 }
                 System.err.println("input tree: " + t);
-                return plant(objects, map);
+                return plant(ret.toArray(new Object[0]), map);
             }
         }
     }
 
     public static interface Reducer {
-        public Object reduce(Tree t);
+        public Object reduce(Iterable<Tree> t);
     }
 
     public static class TargetClass extends Target {
@@ -394,7 +394,7 @@ public class Demo {
             return ret;
         }
         public boolean isRaw() { return _method.isAnnotationPresent(raw.class); }
-        public Object invokeRaw(Tree t) {
+        public Object invokeRaw(Iterable<Tree> t) {
             try {
                 return _method.invoke(null, new Object[] { t });
             } catch (Exception e) {
@@ -486,7 +486,7 @@ public class Demo {
     public static Union make(Tree t, String s) { return make(t, s, new ReflectiveMeta()); }
     public static Union make(Tree t, String s, ReflectiveMeta rm) {
         Reducer red = (Reducer)t.head();
-        MG.Grammar g = (MG.Grammar)red.reduce(t);
+        MG.Grammar g = (MG.Grammar)red.reduce(t.children());
         Context cx = new Context(g,rm);
         Union u = null;
         for(MG.NonTerminal nt : g.nonterminals) {
@@ -786,7 +786,7 @@ public class Demo {
         public Context(Tree t, ReflectiveMeta rm) {
             this.rm = rm;
             Reducer red = (Reducer)t.head();
-            this.grammar = (MG.Grammar)red.reduce(t);
+            this.grammar = (MG.Grammar)red.reduce(t.children());
         }
         public Union peek(String name) { return map.get(name); }
         public void  put(String name, Union u) { map.put(name, u); }