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