ANTLR 4 的 If/else 語句

ANTLR 4 預設使用監聽器，但它也支援訪問者。訪客提供了對解析樹遍歷的更多控制，使它們更適合實現 if/else 語句。要啟用訪客，請執行以下命令：

java -cp antlr-4.0-complete.jar org.antlr.v4.Tool Mu.g4 -visitor

這將產生一個名為MuBaseVisitor 的類別，您可以擴展它來實現您想要的功能。以下是 EvalVisitor 類別的範例：

public class EvalVisitor extends MuBaseVisitor {

    // Override visit methods for each rule that needs to be implemented

    // Example: visitIf_stat for handling if/else statements
    @Override
    public Value visitIf_stat(MuParser.If_statContext ctx) {
        List conditions = ctx.condition_block();
        boolean evaluatedBlock = false;

        for (MuParser.Condition_blockContext condition : conditions) {
            Value evaluated = this.visit(condition.expr());
            if (evaluated.asBoolean()) {
                evaluatedBlock = true;
                this.visit(condition.stat_block()); // Evaluate the true block
                break;
            }
        }

        if (!evaluatedBlock && ctx.stat_block() != null) {
            this.visit(ctx.stat_block()); // Evaluate the else block
        }

        return Value.VOID;
    }
}

在這裡，我們迭代條件並評估第一個為真的條件。如果沒有條件為 true 並且存在 else 區塊，我們將對其進行評估。

要使用此訪問者，請建立一個 Main 類別來解析和評估輸入：

public class Main {
    public static void main(String[] args) throws Exception {
        MuLexer lexer = new MuLexer(new ANTLRFileStream("test.mu"));
        MuParser parser = new MuParser(new CommonTokenStream(lexer));
        ParseTree tree = parser.parse();
        EvalVisitor visitor = new EvalVisitor();
        visitor.visit(tree); // Start the evaluation process
    }
}