X-Git-Url: http://git.megacz.com/?p=sbp.git;a=blobdiff_plain;f=src%2Fedu%2Fberkeley%2Fsbp%2FTree.java;h=f87c5dc7d6761c054c337fb5e8765a7a04e77c73;hp=d2bfc4bbc46a6bf379bd79633fff7746bb1bc834;hb=2c1c0293545f3d12c23220fd05c663e6aa3f3de1;hpb=d86c2344f6f5b958d56e2aa55e84af2dc76ab487 diff --git a/src/edu/berkeley/sbp/Tree.java b/src/edu/berkeley/sbp/Tree.java index d2bfc4b..f87c5dc 100644 --- a/src/edu/berkeley/sbp/Tree.java +++ b/src/edu/berkeley/sbp/Tree.java @@ -6,34 +6,33 @@ import java.io.*; import java.util.*; import java.lang.reflect.*; -/** a tree (or node in a tree); see jargon.txt for details */ -public class Tree - extends PrintableTree> - implements Iterable>, - GraphViz.ToGraphViz { +/** a tree (or node in a tree); see jargon.txt for details */ +public class Tree + extends PrintableTree> + implements Iterable> { private final Input.Region location; - private final T head; - private final Tree[] children; + private final NodeType head; + private final Tree[] children; private final boolean lift; /** the element at the head of the tree */ - public T head() { return head; } + public NodeType head() { return head; } - private Tree lifted() { return children[children.length-1]; } + private Tree lifted() { return children[children.length-1]; } /** the number of children the tree has */ - public int numChildren() { + public int size() { return lift - ? (children.length-1)+lifted().numChildren() + ? (children.length-1)+lifted().size() : children.length; } /** the tree's children */ - public Iterable> children() { return this; } + public Iterable> children() { return this; } /** the tree's children */ - public Iterator> iterator() { + public Iterator> iterator() { return lift ? new ConcatenateIterator(new ArrayIterator(children, 0, children.length-1), children[children.length-1].iterator()) @@ -41,7 +40,7 @@ public class Tree } /** get the ith child */ - public Tree child(int i) { + public Tree child(int i) { return lift && i >= children.length-1 ? children[children.length-1].child(i-(children.length-1)) : children[i]; @@ -50,9 +49,11 @@ public class Tree /** get the input region that this tree was parsed from */ 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[] children) { this(loc, head, children, false); } - Tree(Input.Region loc, T head, Tree[] children, boolean lift) { + public Tree(Input.Region loc, NodeType head) { this(loc, head, null); } + public Tree(Input.Region loc, NodeType head, Tree[] children) { this(loc, head, children, false); } + + /** package-private constructor, allows setting the "lift" bit */ + Tree(Input.Region loc, NodeType head, Tree[] children, boolean lift) { this.location = loc; this.head = head; this.lift = lift && children != null && children.length > 0; @@ -78,22 +79,4 @@ public class Tree protected boolean ignoreSingleton() { return false; } - // ToGraphViz ///////////////////////////////////////////////////////////////////////////// - - public GraphViz.Node toGraphViz(GraphViz gv) { - if (gv.hasNode(this)) return gv.createNode(this); - GraphViz.Node n = gv.createNode(this); - n.label = head()==null ? "" : head().toString(); - //n.color = "red"; - for(Tree t : this) n.edge(t, null); - return n; - } - public boolean isTransparent() { return false; } - public boolean isHidden() { return false; } - - - // TreeFunctor ///////////////////////////////////////////////////////////////////////////// - - public static interface TreeFunctor extends Functor>, R> { } - }