52 lines
938 B
C#
52 lines
938 B
C#
using System;
|
|
using System.IO;
|
|
using Finn.AST;
|
|
|
|
namespace Finn;
|
|
|
|
class ASTPrinter : IVisitor<string>
|
|
{
|
|
public string print(Expr expr)
|
|
{
|
|
return expr.accept(this);
|
|
}
|
|
|
|
private string parenthesize(string name, params Expr[] exprs)
|
|
{
|
|
var w = new StringWriter();
|
|
w.Write($"({name}");
|
|
foreach (Expr expr in exprs)
|
|
{
|
|
w.Write(" ");
|
|
w.Write(expr.accept(this));
|
|
}
|
|
w.Write(")");
|
|
return w.ToString();
|
|
}
|
|
|
|
public string visitBinaryExpr(Binary expr)
|
|
{
|
|
return parenthesize(expr.Op.lexeme, expr.Left, expr.Right);
|
|
}
|
|
|
|
public string visitGroupingExpr(Grouping expr)
|
|
{
|
|
return parenthesize("group", expr.Expression);
|
|
}
|
|
|
|
public string visitLiteralExpr(Literal expr)
|
|
{
|
|
return expr.Value.ToString() ?? "";
|
|
}
|
|
|
|
public string visitUnaryExpr(Unary expr)
|
|
{
|
|
return parenthesize(expr.Op.lexeme, expr.Right);
|
|
}
|
|
|
|
public string visitIfExpr(If expr)
|
|
{
|
|
return parenthesize("if", expr.Condition, expr.Then, expr.Else);
|
|
}
|
|
}
|