1 package edu.berkeley.sbp;
2 import edu.berkeley.sbp.*;
3 import edu.berkeley.sbp.util.*;
4 import edu.berkeley.sbp.bind.*;
7 import java.lang.reflect.*;
9 /** a tree (or node in a tree); see jargon.txt for details */
11 extends PrintableTree<Tree<T>>
12 implements Iterable<Tree<T>>,
16 private final Tree<T>[] children;
17 private final Input.Region location;
18 private final boolean lift;
20 public T head() { return head; }
22 private Tree<T> lifted() { return children[children.length-1]; }
24 public int numChildren() {
25 if (lift && children.length > 0)
26 return (children.length-1)+lifted().numChildren();
27 return children.length;
29 public Iterable<Tree<T>> children() { return this; }
30 public Iterator<Tree<T>> iterator() {
32 return new ConcatenateIterator(new ArrayIterator(children, 0, children.length-1),
33 children[children.length-1].iterator());
34 return new ArrayIterator(children);
36 public Tree<T> child(int i) {
37 if (lift && i >= children.length-1)
38 return children[children.length-1].child(i-(children.length-1));
42 public Input.Region getRegion() { return location; }
44 public Tree(Input.Region loc, T head) { this(loc, head, null); }
45 public Tree(Input.Region loc, T head, Tree<T>[] children) { this(loc, head, children, false); }
46 public Tree(Input.Region loc, T head, Tree<T>[] children, boolean lift) {
51 Tree<T>[] children2 = children==null ? new Tree[0] : new Tree[children.length];
52 if (children != null) System.arraycopy(children, 0, children2, 0, children.length);
53 this.children = children2;
56 protected String headToString() { return head()==null?null:head().toString(); }
57 protected String headToJava() {
59 if (head()==null) return null;
60 if (head() instanceof ToJava) {
61 StringBuffer sb = new StringBuffer();
62 ((ToJava)head()).toJava(sb);
65 return (head()==null?"null":("\""+StringUtil.toJavaString(head().toString())+"\""));
67 protected String left() { return "{"; }
68 protected String right() { return "}"; }
69 protected boolean ignoreSingleton() { return false; }
71 public GraphViz.Node toGraphViz(GraphViz gv) {
72 if (gv.hasNode(this)) return gv.createNode(this);
73 GraphViz.Node n = gv.createNode(this);
74 n.label = head()==null ? "" : head().toString();
76 for(Tree t : this) n.edge(t, null);
79 public boolean isTransparent() { return false; }
80 public boolean isHidden() { return false; }
82 public static interface TreeFunctor<T,R> extends Functor<Iterable<Tree<T>>, R> {
85 public static class ArrayBuildingTreeFunctor<T> implements TreeFunctor<T,T[]>, ToJava {
87 public void toJava(StringBuffer sb) { sb.append("new Tree.ArrayBuildingTreeFunctor()"); }
88 public String toString() { return ""; }
90 public T[] invoke(Iterable<Tree<T>> t) {
91 ArrayList ret = new ArrayList();
93 if (tc.head() != null && tc.head() instanceof Functor)
94 ret.add(((Functor<Iterable<Tree>,Object>)tc.head()).invoke(tc.children()));
95 else if (tc.numChildren() == 0)
98 System.err.println("FIXME: don't know what to do about " + tc);
102 return (T[])ret.toArray(new Object[0]);
106 public static class RawBindingFunctor<T> implements TreeFunctor<T,Object>, ToJava {
107 private Binding _binding;
108 private String _toString;
109 public RawBindingFunctor(String toString, Binding binding) {
110 this._binding = binding;
111 this._toString = toString;
113 public String toString() {
116 public void toJava(StringBuffer sb) {
117 sb.append("new Tree.RawBindingFunctor(");
119 sb.append(StringUtil.toJavaString(_toString));
124 public Object invoke(Iterable<Tree<T>> t) {
125 return _binding.invoke(new Object[] { t });
129 public static class BindingFunctor<T> implements TreeFunctor<T,Object>, ToJava {
130 private Binding _binding;
131 private String _toString;
132 public String toString() {
135 public BindingFunctor(String toString, Binding binding) {
136 this._binding = binding;
137 this._toString = toString;
139 public void toJava(StringBuffer sb) {
140 sb.append("new Tree.BindingFunctor(");
142 sb.append(StringUtil.toJavaString(_toString));
147 public Object invoke(Iterable<Tree<T>> t) {
148 ArrayList ret = new ArrayList();
150 if (tc.head() != null && tc.head() instanceof Functor)
151 ret.add(((Tree.TreeFunctor<Object,Object>)tc.head()).invoke(tc.children()));
152 else if (tc.numChildren() == 0)
155 System.err.println("FIXME: don't know what to do about " + tc);
159 Object[] o = (Object[])ret.toArray(new Object[0]);
160 return _binding.invoke(o);