-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathVisitor.java
More file actions
95 lines (75 loc) · 2.13 KB
/
Visitor.java
File metadata and controls
95 lines (75 loc) · 2.13 KB
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
import static org.junit.Assert.assertEquals;
abstract class ExpressionVisitor {
abstract void visit(Value value);
abstract void visit(AdditionExpression ae);
abstract void visit(MultiplicationExpression me);
}
abstract class Expression {
abstract void accept(ExpressionVisitor ev);
}
class Value extends Expression {
public int value;
public Value(int value) {
this.value = value;
}
@Override
void accept(ExpressionVisitor ev) {
ev.visit(this);
}
}
class AdditionExpression extends Expression {
public Expression lhs, rhs;
public AdditionExpression(Expression lhs, Expression rhs) {
this.lhs = lhs;
this.rhs = rhs;
}
@Override
void accept(ExpressionVisitor ev) {
ev.visit(this);
}
}
class MultiplicationExpression extends Expression {
public Expression lhs, rhs;
public MultiplicationExpression(Expression lhs, Expression rhs) {
this.lhs = lhs;
this.rhs = rhs;
}
@Override
void accept(ExpressionVisitor ev) {
ev.visit(this);
}
}
class ExpressionPrinter extends ExpressionVisitor {
private StringBuilder sb = new StringBuilder();
@Override
public String toString() {
return sb.toString();
}
@Override
void visit(Value value) {
sb.append(value.value);
}
@Override
void visit(AdditionExpression ae) {
visitBinary(ae.lhs, "+", ae.rhs);
}
@Override
void visit(MultiplicationExpression me) {
visitBinary(me.lhs, "*", me.rhs);
}
private void visitBinary(Expression lhs, String operator, Expression rhs) {
sb.append("(");
lhs.accept(this);
sb.append(operator);
rhs.accept(this);
sb.append(")");
}
}
class DemoVisitor {
public static void main(String[] args) {
AdditionExpression simple = new AdditionExpression(new Value(2), new Value(3));
ExpressionPrinter ep = new ExpressionPrinter();
ep.visit(simple);
assertEquals("(2+3)", ep.toString());
}
}