使用 ANTLR 实现基本的脚本语言

ANTLR（Another Tool for Language Recognition）是一个用于构建语言识别器的工具。它可以使用语法文件生成语法分析器，从而轻松地对输入的文本进行分析和处理。在这里，我们将使用 ANTLR 来实现一个基本的脚本语言。

首先，我们需要定义我们的脚本语言的语法。在这个例子中，我们将使用类似于 BASIC 的语法，包括变量赋值、算术操作、条件语句和循环语句。

以下是我们的语法文件：

grammar BasicScript;

program : statement+ ;

statement : assignStmt | printStmt | ifStmt | whileStmt ;

assignStmt : ID '=' expr ';' ;

printStmt : 'PRINT' expr ';' ;

ifStmt : 'IF' expr 'THEN' statement+ ('ELSE' statement+)? 'END IF' ;

whileStmt : 'WHILE' expr 'DO' statement+ 'END WHILE' ;

expr : ID | INT | expr op expr | '(' expr ')' ;

op : '+' | '-' | '*' | '/' ;

ID : [a-zA-Z]+ ;

INT : [0-9]+ ;

这个语法文件定义了一个简单的程序，由一系列语句组成。语句可以是赋值语句、打印语句、条件语句或循环语句。

赋值语句由一个标识符、一个等号和一个表达式组成。打印语句以 PRINT 关键字开头，后面跟着一个表达式。条件语句以 IF 关键字开头，后面跟着一个表达式，然后是一个或多个语句。ELSE 子句是可选的。循环语句以 WHILE 关键字开头，后面跟着一个表达式，然后是一个或多个语句。

表达式可以是一个标识符、一个整数或一个算术表达式。算术表达式可以使用加、减、乘、除四种运算符。括号可以用于改变优先级。

现在，我们可以使用 ANTLR 生成我们的语法分析器和词法分析器。我们可以使用以下命令来生成 Java 代码：

antlr4 BasicScript.g4

这将生成 BasicScriptLexer.java 和 BasicScriptParser.java 两个文件。

现在，我们可以使用这些文件来编写一个简单的程序。以下是一个计算 1 到 10 之和的示例程序：

10 LET sum = 0;
20 FOR i = 1 TO 10
30 LET sum = sum + i;
40 NEXT i
50 PRINT sum;

我们可以使用以下 Java 代码来解析和执行这个程序：

import org.antlr.v4.runtime.*;
import org.antlr.v4.runtime.tree.*;

public class BasicScriptMain {

    public static void main(String[] args) throws Exception {
        // create a CharStream that reads from standard input
        CharStream input = CharStreams.fromStream(System.in);
        // create a lexer that feeds off of input CharStream
        BasicScriptLexer lexer = new BasicScriptLexer(input);
        // create a buffer of tokens pulled from the lexer
        CommonTokenStream tokens = new CommonTokenStream(lexer);
        // create a parser that feeds off the tokens buffer
        BasicScriptParser parser = new BasicScriptParser(tokens);
        // create a parse tree from the program rule
        ParseTree tree = parser.program();
        // create a visitor to evaluate the program
        BasicScriptVisitorImpl visitor = new BasicScriptVisitorImpl();
        // evaluate the program
        visitor.visit(tree);
    }
}

这个程序首先创建一个 CharStream，然后使用它来创建一个词法分析器。词法分析器将输入的文本分成标记（tokens）。然后，它创建一个语法分析器，并使用标记流来解析输入的文本。它生成一个 ParseTree，代表输入的程序。然后，它创建一个 BasicScriptVisitorImpl 实例，并使用它来遍历 ParseTree 并计算程序。

BasicScriptVisitorImpl 实现了语法分析器生成的接口，用于遍历 ParseTree 并计算程序。以下是这个类的示例代码：

import java.util.HashMap;
import java.util.Map;

public class BasicScriptVisitorImpl extends BasicScriptBaseVisitor<Void> {

    private Map<String, Integer> variables = new HashMap<>();

    @Override
    public Void visitAssignStmt(BasicScriptParser.AssignStmtContext ctx) {
        String id = ctx.ID().getText();
        int value = visit(ctx.expr());
        variables.put(id, value);
        return null;
    }

    @Override
    public Void visitPrintStmt(BasicScriptParser.PrintStmtContext ctx) {
        int value = visit(ctx.expr());
        System.out.println(value);
        return null;
    }

    @Override
    public Void visitIfStmt(BasicScriptParser.IfStmtContext ctx) {
        int condition = visit(ctx.expr());
        if (condition != 0) {
            for (BasicScriptParser.StatementContext stmt : ctx.statement()) {
                visit(stmt);
            }
        } else if (ctx.ELSE() != null) {
            for (BasicScriptParser.StatementContext stmt : ctx.statement(1)) {
                visit(stmt);
            }
        }
        return null;
    }

    @Override
    public Void visitWhileStmt(BasicScriptParser.WhileStmtContext ctx) {
        while (visit(ctx.expr()) != 0) {
            for (BasicScriptParser.StatementContext stmt : ctx.statement()) {
                visit(stmt);
            }
        }
        return null;
    }

    @Override
    public Integer visitID(BasicScriptParser.IDContext ctx) {
        String id = ctx.getText();
        Integer value = variables.get(id);
        if (value == null) {
            throw new RuntimeException("Undefined variable: " + id);
        }
        return value;
    }

    @Override
    public Integer visitINT(BasicScriptParser.INTContext ctx) {
        return Integer.valueOf(ctx.getText());
    }

    @Override
    public Integer visitExpr(BasicScriptParser.ExprContext ctx) {
        if (ctx.ID() != null) {
            return visitID(ctx.ID());
        } else if (ctx.INT() != null) {
            return visitINT(ctx.INT());
        } else {
            int left = visit(ctx.expr(0));
            int right = visit(ctx.expr(1));
            String op = ctx.op.getText();
            switch (op) {
                case "+":
                    return left + right;
                case "-":
                    return left - right;
                case "*":
                    return left * right;
                case "/":
                    return left / right;
                default:
                    throw new RuntimeException("Unknown operator: " + op);
            }
        }
    }
}

这个类遍历 ParseTree 中的各个节点，并执行相应的操作。visitAssignStmt 方法处理赋值语句，visitPrintStmt 方法处理打印语句，visitIfStmt 方法处理条件语句，visitWhileStmt 方法处理循环语句。visitID 和 visitINT 方法分别处理标识符和整数。visitExpr 方法处理算术表达式。

现在，我们可以使用以下命令来运行程序：

java BasicScriptMain < program.bas

其中，program.bas 是包含我们示例程序的文件。该程序将输出 1 到 10 之和。