-
Notifications
You must be signed in to change notification settings - Fork 0
/
dewCustomVisitor.java
66 lines (61 loc) · 2.45 KB
/
dewCustomVisitor.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import org.antlr.v4.runtime.tree.AbstractParseTreeVisitor;
import java.util.HashMap;
import java.util.Map;
public class dewCustomVisitor extends dewBaseVisitor<Object> {
private Map<String, Object> variables = new HashMap<>();
@Override
public Object visitAssignment(dewParser.AssignmentContext ctx) {
String variableName = ctx.variableName().getText();
Object value = visit(ctx.expression());
variables.put(variableName, value);
return value;
}
@Override
public Object visitVariableName(dewParser.VariableNameContext ctx) {
String variableName = ctx.getText();
if (variables.containsKey(variableName)) {
return variables.get(variableName);
} else {
throw new RuntimeException("Variable " + variableName + " not defined");
}
}
@Override
public Object visitExpression(dewParser.ExpressionContext ctx) {
if (ctx.op != null) {
Object left = visit(ctx.expression(0));
Object right = visit(ctx.expression(1));
switch (ctx.op.getType()) {
case dewParser.ADD:
return (Integer) left + (Integer) right;
case dewParser.SUB:
return (Integer) left - (Integer) right;
case dewParser.MUL:
return (Integer) left * (Integer) right;
case dewParser.DIV:
return (Integer) left / (Integer) right;
default:
throw new RuntimeException("Unknown operator: " + ctx.op.getText());
}
} else if (ctx.methodCall() != null) {
return visit(ctx.methodCall());
} else if (ctx.variableName() != null) {
return visit(ctx.variableName());
} else if (ctx.NUMBER() != null) {
return Integer.valueOf(ctx.NUMBER().getText());
} else if (ctx.STRING() != null) {
return ctx.STRING().getText();
} else {
throw new RuntimeException("Unknown expression: " + ctx.getText());
}
}
@Override
public Object visitMethodCall(dewParser.MethodCallContext ctx) {
String methodName = ctx.methodName().getText();
if (methodName.equals("output")) {
Object argument = visit(ctx.methodCallArguments().expression(0));
System.out.println(argument);
return Integer.valueOf(1);
}
return Integer.valueOf(0);
}
}