Programming Languages Research Group: Git - IRC.git/blob

   1 /*
   2
   3    Class: ParseNodeDOTVisitor
   4    Author: Dan Roy
   5    Purpose: Traverses a ParseNode tree and generates a DOT file that represents the parse
   6           tree.
   7
   8  */
   9
  10 package IR.Tree;
  11
  12 import java.util.*;
  13
  14 public class ParseNodeDOTVisitor {
  15
  16   java.io.PrintWriter output;
  17   int tokennumber;
  18   int color;
  19
  20   private ParseNodeDOTVisitor(java.io.OutputStream output) {
  21     tokennumber = 0;
  22     color = 0;
  23     this.output = new java.io.PrintWriter(output, true);
  24   }
  25
  26   private String getNewID(String name) {
  27     tokennumber = tokennumber + 1;
  28     return new String(name+tokennumber);
  29   }
  30
  31   public static void visit(java.io.OutputStream output, ParseNode root) {
  32     ParseNodeDOTVisitor visitor = new ParseNodeDOTVisitor(output);
  33     visitor.make(root);
  34   }
  35
  36   private void make(ParseNode root) {
  37     output.println("digraph dotvisitor {");
  38     output.println("\tsize=\"7, 10\";");
  39     traverse(root, getNewID("root"));
  40     output.println("}\n");
  41   }
  42
  43   private String newColor() {
  44
  45
  46     if (color == 0) {
  47       color++;
  48       return new String("red");
  49     } else if (color == 1) {
  50       color++;
  51       return new String("green");
  52     } else {
  53       color = 0;
  54       return new String("blue");
  55     }
  56   }
  57
  58   private void traverse(ParseNode node, String nodeid) {
  59     output.println("\t" + nodeid + " [label=\"" + node.getLabel() + "\",shape=box];");
  60     ParseNodeVector children = node.getChildren();
  61     for (int i = 0; i < children.size(); i++) {
  62       ParseNode child = children.elementAt(i);
  63       String childid = getNewID("node");
  64       output.println("\t" + nodeid + " -> " + childid + ";");
  65       if (child.getLabel()=="rule") {
  66         output.println("\tnode [color=" + newColor() + "];");
  67       }
  68       traverse(child, childid);
  69     }
  70   }
  71 }