add printAST()

dlang · Oct 1, 2015 · e7a6820 · e7a6820
1 parent ae47818
commit e7a6820
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/src/expression.d b/src/expression.d
@@ -2360,6 +2360,18 @@ public:
         return buf.extractString();
     }
 
+    /********************
+     * Print AST data structure in a nice format.
+     * Params:
+     *  indent = indentation level
+     */
+    void printAST(int indent = 0)
+    {
+        foreach (i; 0 .. indent)
+            printf(" ");
+        printf("%s %s\n", Token.toChars(op), type ? type.toChars() : "");
+    }
+
     final void error(const(char)* format, ...)
     {
         if (type != Type.terror)
@@ -7092,6 +7104,12 @@ public:
     {
         v.visit(this);
     }
+
+    override void printAST(int indent)
+    {
+        Expression.printAST(indent);
+        e1.printAST(indent + 2);
+    }
 }
 
 extern (C++) alias fp_t = UnionExp function(Loc loc, Type, Expression, Expression);
@@ -7393,6 +7411,13 @@ public:
     {
         v.visit(this);
     }
+
+    override void printAST(int indent)
+    {
+        Expression.printAST(indent);
+        e1.printAST(indent + 2);
+        e2.printAST(indent + 2);
+    }
 }
 
 /***********************************************************

diff --git a/src/expression.h b/src/expression.h
@@ -143,6 +143,7 @@ class Expression : public RootObject
 
     void print();
     char *toChars();
+    virtual void printAST(int ident = 0);
     void error(const char *format, ...);
     void warning(const char *format, ...);
     void deprecation(const char *format, ...);
@@ -741,6 +742,7 @@ class UnaExp : public Expression
     Expression *resolveLoc(Loc loc, Scope *sc);
 
     void accept(Visitor *v) { v->visit(this); }
+    void printAST(int ident);
 };
 
 typedef UnionExp (*fp_t)(Type *, Expression *, Expression *);
@@ -768,6 +770,7 @@ class BinExp : public Expression
     Expression *reorderSettingAAElem(Scope *sc);
 
     void accept(Visitor *v) { v->visit(this); }
+    void printAST(int ident);
 };
 
 class BinAssignExp : public BinExp