Add converstions toTree in ASTs

romildo · Oct 10, 2016 · 49e6533 · 49e6533
1 parent 08d2091
commit 49e6533
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 4 deletions.
diff --git a/src/main/java/absyn/AST.java b/src/main/java/absyn/AST.java
@@ -1,9 +1,10 @@
 package absyn;
 
+import javaslang.render.ToTree;
 import org.apache.commons.lang3.builder.ToStringBuilder;
 import org.apache.commons.lang3.builder.ToStringStyle;
 
-public abstract class AST {
+public abstract class AST implements ToTree<String> {
 
    @Override
    public String toString() {

diff --git a/src/main/java/absyn/ExpBinOp.java b/src/main/java/absyn/ExpBinOp.java
@@ -1,5 +1,7 @@
 package absyn;
 
+import javaslang.collection.Tree;
+
 public class ExpBinOp extends Exp {
 
    public enum Op {PLUS, MINUS, TIMES, DIV}
@@ -14,4 +16,10 @@ public ExpBinOp(Op op, Exp left, Exp right) {
       this.right = right;
    }
 
+   @Override
+   public Tree.Node<String> toTree() {
+      return Tree.of("ExpBinOp: " + op,
+                     left.toTree(),
+                     right.toTree());
+   }
 }
diff --git a/src/main/java/absyn/ExpNum.java b/src/main/java/absyn/ExpNum.java
@@ -1,5 +1,7 @@
 package absyn;
 
+import javaslang.collection.Tree;
+
 public class ExpNum extends Exp {
 
    public final Double value;
@@ -8,4 +10,8 @@ public ExpNum(Double value) {
       this.value = value;
    }
 
+   @Override
+   public Tree.Node<String> toTree() {
+      return Tree.of("ExpNum: " + value);
+   }
 }
diff --git a/src/main/java/main/Driver.java b/src/main/java/main/Driver.java
@@ -7,10 +7,14 @@
 import java.util.ArrayList;
 import java.util.List;
 
+import absyn.Exp;
 import com.beust.jcommander.JCommander;
 import com.beust.jcommander.Parameter;
 import com.beust.jcommander.ParameterException;
 import java_cup.runtime.Symbol;
+import javaslang.render.dot.DotFile;
+import javaslang.render.text.Boxes;
+import javaslang.render.text.PrettyPrinter;
 import parse.SymbolConstants;
 import parse.Lexer;
 import parse.parser;
@@ -107,10 +111,19 @@ public static void lexicalAnalysis(Reader input) throws IOException {
    }
 
    public static void syntaxAnalysis(Reader input) throws Exception {
-      Lexer lexer = new Lexer(input);
-      parser parser = new parser(lexer);
-      Symbol result = parser.parse();
+      final Lexer lexer = new Lexer(input);
+      final parser parser = new parser(lexer);
+      final Symbol result = parser.parse();
+      final Exp parseTree = (Exp) result.value;
       System.out.println(result.value);
+      System.out.println();
+      System.out.println("===Abstract syntax tree:===========");
+      System.out.println();
+      System.out.println(PrettyPrinter.pp(parseTree.toTree()));
+      System.out.println();
+      System.out.println(Boxes.box(parseTree.toTree()));
+      DotFile.write(parseTree.toTree(), "test1.dot");
+      System.out.println();
    }
 
 }