ANTLR4(十) 内嵌代码之计算器

写在之前

采用访问器的计算器

语法文件：

grammar Expr; 

prog:   stat+ ;

stat:   expr NEWLINE                
    |   ID '=' expr NEWLINE         
    |   CLEAR NEWLINE               
    |   NEWLINE                     
    ;

expr:   expr op=('*'|'/') expr      
    |   expr op=('+'|'-') expr      
    |   INT                         
    |   ID                          
    |   '(' expr ')'                
    ;

MUL :   '*' ; 
DIV :   '/' ;
ADD :   '+' ;
SUB :   '-' ;

ID  :   [a-zA-Z]+ ;      // match identifiers
INT :   [0-9]+ ;         // match integers
NEWLINE:'\r'? '\n' ;     // return newlines to parser (is end-statement signal)
WS  :   [ \t]+ -> skip ; // toss out whitespace

自定义Visitor：

//EvalVistor.java
import java.util.HashMap;
import java.util.Map;

public class EvalVisitor extends LabeledExprBaseVisitor<Integer> {
   
    /** "memory" for our calculator; variable/value pairs go here */
    Map<String, Integer> memory = new HashMap<String, Integer>();

    /** ID '=' expr NEWLINE */
    @Override
    public Integer visitAssign(LabeledExprParser.AssignContext ctx) {
   
        String id = ctx.ID().getText();  // id is left-hand side of '='
        int value = visit(ctx.expr());   // compute value of expression on right
        memory.put(id, value);           // store it in our memory
        return value;
    }

    /** expr NEWLINE */
    @Override