基于ANTLR4语法能否构建表达式树？如何从对象模型生成代码？

你已经搞定了解析和对象模型这两步，接下来要做的反向生成，ANTLR生态里有几种非常合适的方案，完全不用手动拼接字符串，能严格遵循你的自定义语法规则：

1. 写一个「反向Visitor」来遍历对象模型 #

既然你之前用Visitor/Listener把ParseTree转成了对象模型，那反过来也一样：写一个专门的Visitor类，遍历你的对象模型，针对每个模型类（比如AddExpr、NumberExpr）调用对应语法规则的生成逻辑。

举个具体例子，假设你的语法里有：

expr : expr '+' expr | NUMBER;

你的对象模型对应有AddExpr（包含左右两个子表达式）和NumberExpr（包含数值）。那反向Visitor的核心逻辑会是这样：

public class ExprGeneratorVisitor implements ExprVisitor<String> {
    @Override
    public String visitAddExpr(AddExpr expr) {
        return visit(expr.getLeft()) + " + " + visit(expr.getRight());
    }

    @Override
    public String visitNumberExpr(NumberExpr expr) {
        return String.valueOf(expr.getValue());
    }
}

这种方式的好处是完全复用你对语法规则的理解，生成的表达式天然符合语法，还能轻松处理运算符优先级、括号这类容易手动写错的细节。

2. 用StringTemplate 4（ST4）做模板化生成 #

ST4是ANTLR作者开发的模板引擎，天生和ANTLR适配，非常适合做语法驱动的代码生成。你可以给每个语法规则对应写一个模板，把对象模型的数据填充进去就行。

比如针对上面的加法表达式，你可以定义这样的模板文件：

expr(AddExpr) ::= "<visit(left)> + <visit(right)>"
expr(NumberExpr) ::= "<value>"

然后在代码里把你的对象模型传给模板，ST4会自动匹配对应的模板生成正确的表达式字符串。这种方式把生成逻辑和业务代码分离，后期修改语法或者生成格式时，直接改模板就行，维护起来特别方便。

3. 反向构造ParseTree再生成文本 #

如果想完全依赖ANTLR的语法验证能力，你可以从对象模型反向构建ANTLR的ParseTree节点（比如ParserRuleContext、CommonToken），然后调用ParseTree.toStringTree(parser)方法直接生成文本。

比如构造一个加法表达式的ParseTree：

// 假设你有生成的ExprParser和ExprLexer
ExprParser parser = new ExprParser(null); // 不需要输入流，只用来生成上下文
ExprContext addExprCtx = new ExprContext(null, 0);
addExprCtx.addChild(visitNumberExpr(new NumberExpr(5))); // 构造左子节点
addExprCtx.addChild(new CommonToken(ExprParser.PLUS, "+"));
addExprCtx.addChild(visitNumberExpr(new NumberExpr(3))); // 构造右子节点

// 生成表达式文本
String exprText = addExprCtx.toStringTree(parser);

这种方式的优势是生成的文本100%符合你的语法规则（因为是用ANTLR的节点构造的），但实现起来稍微繁琐一点，适合对生成准确性要求极高的场景。

小提示 #

不管选哪种方案，核心都是让生成逻辑和你的ANTLR语法规则绑定，而不是手动拼字符串——这样不仅能避免语法错误，后期语法迭代时，生成逻辑的修改也会更高效。

内容的提问来源于stack exchange，提问作者Ruudjah