antlr4做一个计算器

1.环境准备

在IDEA中安装使用Antlr

2.建立G4文件

/Users/lcc/IdeaProjects/JdkSource/src/main/java/com/antlr/label/CalExpr.g4

内容

grammar CalExpr;

prog : stat+;

stat :expr NEWLINE        #printExpr
    |ID '=' expr NEWLINE  #assign
    |NEWLINE              #blank
    ;

expr : expr op=('+'|'-') expr   # AddSub
    | expr op=('*'|'/') expr    # MulDiv
    | ID                        # id
    | INT                       # int
    | '(' expr ')'              # parens
    ;

ID : [a-zA-Z]+ ;
INT : [0-9]+ ;
NEWLINE : '\r' ? '\n';
WS : [ \t]+ -> skip;

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

建立测试文件

/Users/lcc/IdeaProjects/JdkSource/src/main/java/com/antlr/label/MyVisitor.java

内容

package com.antlr.label;

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

import org.stringtemplate.v4.compiler.STParser.expr_return;



public class MyVisitor extends CalExprBaseVisitor<Integer>{

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

    @Override
    public Integer visitParens(CalExprParser.ParensContext ctx) {
        return super.visit(ctx.expr());
    }

    @Override
    public Integer visitBlank(CalExprParser.BlankContext ctx) {
        return super.visitBlank(ctx);
    }

    @Override
    public Integer visitAddSub(CalExprParser.AddSubContext ctx) {

        Integer left=visit(ctx.expr(0));        //获取左边表达式最终值
        Integer right=visit(ctx.expr(1));       //获取右边表达式最终值

        if(ctx.op.getType()==CalExprLexer.ADD) return left+right;   //如果是加法
        else return left-right;                                     //如果是减法
    }

    @Override
    public Integer visitMulDiv(CalExprParser.MulDivContext ctx) {
        Integer left=visit(ctx.expr(0));        //获取左边表达式最终值
        Integer right=visit(ctx.expr(1));       //获取右边表达式最终值

        if(ctx.op.getType()==CalExprLexer.DIV) return left/right;   //如果是除法
        else return left*right;                                     //如果是乘法
    }

    @Override
    public Integer visitId(CalExprParser.IdContext ctx) {
        String key=ctx.ID().getText();

        if(map.containsKey(key)){   //如果变量被赋值
            return map.get(key);
        }
        return 0;
    }

    @Override
    public Integer visitInt(CalExprParser.IntContext ctx) {
        return Integer.parseInt(ctx.INT().getText());
    }

    @Override
    public Integer visitPrintExpr(CalExprParser.PrintExprContext ctx) {
        Integer value=visit(ctx.expr());
        System.out.println(value);
        return 0;
    }

    @Override
    public Integer visitAssign(CalExprParser.AssignContext ctx) {

        String key=ctx.ID().getText();
        Integer value=visit(ctx.expr());
        map.put(key, value);
        return value;                   // 返回 value ：a=b=6 则 a==6
    }

}

建立主文件

package com.antlr.label;

import java.io.FileInputStream;
import java.io.InputStream;
import org.antlr.v4.runtime.*;
import org.antlr.v4.runtime.tree.*;

public class MyMain {
    public static void main(String[] args) throws Exception {
        String inputFile = "/Users/lcc/IdeaProjects/JdkSource/src/main/java/com/antlr/label/data.ini";                      //文件读取数据
        InputStream is = System.in;

        if ( inputFile!=null ) is = new FileInputStream(inputFile);
        ANTLRInputStream input = new ANTLRInputStream(is);

        CalExprLexer lexer = new CalExprLexer(input);
        CommonTokenStream tokens = new CommonTokenStream(lexer);
        CalExprParser parser = new CalExprParser(tokens);
        ParseTree tree = parser.prog();                      // 生成语法树
        MyVisitor visitor = new MyVisitor();
        visitor.visit(tree);
    }
}

建立数据测试文件

/Users/lcc/IdeaProjects/JdkSource/src/main/java/com/antlr/label/data.ini

1+1
2+2
3+5

最后要加上一个换行

运行主文件,结果

2
4
8

Process finished with exit code 0

参考:https://blog.csdn.net/zjq_1314520/article/details/65938532