Antlr4学习【开发一个计算器】

1、eclipse安装antlr插件

工具栏【help】-【Eclipse Marketplace】

2、新建一个【antlr】项目

 

3、新建一个【firstAntlr4】项目，新建一个【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 : '/' ;

4、编译CalExpr.g4，生成自动文件

 

5、创建一个java项目

MyVisitor.java源代码

package com.zuk.cal;

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
    }

}

MyMain.java

package com.zuk.cal;

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 = "D:\\development-java\\workspace\\workspace-flink\\firstAntlr4Demo\\src\\test.txt";                      //文件读取数据
        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);
    }
}

test.txt

1+1
2+2
3+5
9*5
9/3
a=19
b=20
b-a