package edu.berkeley.sbp;
import edu.berkeley.sbp.*;
-import edu.berkeley.sbp.*;
import edu.berkeley.sbp.util.*;
+import edu.berkeley.sbp.bind.*;
import java.io.*;
import java.util.*;
import java.lang.reflect.*;
implements Iterable<Tree<T>>,
GraphViz.ToGraphViz {
- final T head;
- Tree<T>[] children;
- final Input.Region location;
+ private final T head;
+ private final Tree<T>[] children;
+ private final Input.Region location;
+ private final boolean lift;
public T head() { return head; }
- public int numChildren() { return children.length; }
- public Iterable<Tree<T>> children() { return new ArrayIterator(children); }
- public Iterator<Tree<T>> iterator() { return new ArrayIterator(children); }
- public Tree<T> child(int i) { return children[i]; }
+
+ private Tree<T> lifted() { return children[children.length-1]; }
+
+ public int numChildren() {
+ if (lift && children.length > 0)
+ return (children.length-1)+lifted().numChildren();
+ return children.length;
+ }
+ public Iterable<Tree<T>> children() { return this; }
+ public Iterator<Tree<T>> iterator() {
+ if (lift)
+ return new ConcatenateIterator(new ArrayIterator(children, 0, children.length-1),
+ children[children.length-1].iterator());
+ return new ArrayIterator(children);
+ }
+ public Tree<T> child(int i) {
+ if (lift && i >= children.length-1)
+ return children[children.length-1].child(i-(children.length-1));
+ return children[i];
+ }
public Input.Region getRegion() { return location; }
public Tree(Input.Region loc, T head) { this(loc, head, null); }
- public Tree(Input.Region loc, T head, Tree<T>[] children) {
+ public Tree(Input.Region loc, T head, Tree<T>[] children) { this(loc, head, children, false); }
+ public Tree(Input.Region loc, T head, Tree<T>[] children, boolean lift) {
this.location = loc;
this.head = head;
+ this.lift = lift;
Tree<T>[] children2 = children==null ? new Tree[0] : new Tree[children.length];
if (children != null) System.arraycopy(children, 0, children2, 0, children.length);
this.children = children2;
}
- protected String headToString() { return head==null?null:head.toString(); }
+ protected String headToString() { return head()==null?null:head().toString(); }
protected String headToJava() {
// FIXME
- if (head==null) return null;
- if (head instanceof ToJava) {
+ if (head()==null) return null;
+ if (head() instanceof ToJava) {
StringBuffer sb = new StringBuffer();
- ((ToJava)head).toJava(sb);
+ ((ToJava)head()).toJava(sb);
return sb.toString();
}
- return (head==null?"null":("\""+StringUtil.toJavaString(head.toString())+"\""));
+ return (head()==null?"null":("\""+StringUtil.toJavaString(head().toString())+"\""));
}
protected String left() { return "{"; }
protected String right() { return "}"; }
}
}
+ public static class RawBindingFunctor<T> implements TreeFunctor<T,Object>, ToJava {
+ private Binding _binding;
+ private String _toString;
+ public RawBindingFunctor(String toString, Binding binding) {
+ this._binding = binding;
+ this._toString = toString;
+ }
+ public String toString() {
+ return _toString;
+ }
+ public void toJava(StringBuffer sb) {
+ sb.append("new Tree.RawBindingFunctor(");
+ sb.append("\"");
+ sb.append(StringUtil.toJavaString(_toString));
+ sb.append("\", ");
+ _binding.toJava(sb);
+ sb.append(")");
+ }
+ public Object invoke(Iterable<Tree<T>> t) {
+ return _binding.invoke(new Object[] { t });
+ }
+ }
+
+ public static class BindingFunctor<T> implements TreeFunctor<T,Object>, ToJava {
+ private Binding _binding;
+ private String _toString;
+ public String toString() {
+ return _toString;
+ }
+ public BindingFunctor(String toString, Binding binding) {
+ this._binding = binding;
+ this._toString = toString;
+ }
+ public void toJava(StringBuffer sb) {
+ sb.append("new Tree.BindingFunctor(");
+ sb.append("\"");
+ sb.append(StringUtil.toJavaString(_toString));
+ sb.append("\", ");
+ _binding.toJava(sb);
+ sb.append(")");
+ }
+ public Object invoke(Iterable<Tree<T>> t) {
+ ArrayList ret = new ArrayList();
+ for(Tree tc : t) {
+ if (tc.head() != null && tc.head() instanceof Functor)
+ ret.add(((Tree.TreeFunctor<Object,Object>)tc.head()).invoke(tc.children()));
+ else if (tc.numChildren() == 0)
+ ret.add(tc.head());
+ else {
+ System.err.println("FIXME: don't know what to do about " + tc);
+ ret.add(null);
+ }
+ }
+ Object[] o = (Object[])ret.toArray(new Object[0]);
+ return _binding.invoke(o);
+ }
+ }
+
}