-
Notifications
You must be signed in to change notification settings - Fork 5
/
call.rb
121 lines (103 loc) · 3.11 KB
/
call.rb
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
module Reak
module Syntax
class Call < Node
attr_accessor :message, :reciever
unfold
def self.new(*args)
return super if self != Call or args.size != 2
reciever, message = args
# Other messages we may override:
# to:do: == basicAt: basicSize
# to:by:do: timesRepeat: basicAt:put: basicNew:
case message.selector
when 'ifTrue:', 'ifTrue:ifFalse:'
Branch.new(reciever, message.args[0], message.args[1])
when 'ifFalse:', 'ifFalse:ifTrue:'
Branch.new(reciever, message.args[1], message.args[0])
when 'and:'
And.new(reciever, message.args[0])
when 'or:'
Or.new(reciever, message.args[0])
else
Super === reciever ? SuperCall.new(reciever, message) : super
end
end
def initialize(reciever, message)
@message = message
@reciever = reciever
end
def to_sexp
[:call, reciever.to_sexp, message.to_sexp]
end
def visit(visitor)
reciever.visit visitor
message.visit visitor
end
end
class SuperCall < Call
def to_sexp
[:super_call, message.to_sexp]
end
end
class Branch < Call
attr_accessor :condition, :positive, :negative
def self.new(condition, positive, negative)
positive = positive ? positive.unclosure : NilKind.new
negative = negative ? negative.unclosure : NilKind.new
case condition
when TrueKind then positive
when FalseKind then negative
else super(condition, positive, negative)
end
end
def initialize(condition, positive, negative)
@condition = condition
@positive = positive
@negative = negative
end
def visit(visitor)
visitor.branch(condition, positive, negative)
end
def to_sexp
[:if, condition.to_sexp, (positive.to_sexp if positive), (negative.to_sexp if negative)]
end
end
class And < Call
attr_accessor :first, :second
def self.new(first, second)
case [self, first.class]
when [And, TrueKind], [Or, FalseKind] then second.unclosure
when [And, FalseKind], [Or, TrueKind] then first
else super
end
end
def initialize(first, second)
@first = first
@second = second.unclosure if second
end
def to_sexp
[:and, (first.to_sexp if first), (second.to_sexp if second)]
end
end
class Or < And
def to_sexp
[:or, first.to_sexp, second.to_sexp]
end
end
class Cascade < Call
alias messages message
def self.new(reciever, messages)
return super unless Super === reciever or Self === reciever
Body.new nil, messages.map { |m| Call.new(reciever, m) }
end
def visit(visitor)
reciever.visit(visitor)
messages[0..-2].each { |msg| visitor.call_cascade(msg) }
messages.last.visit(visitor)
end
def to_sexp
[:cascade, reciever.to_sexp].concat messages.map { |m| m.to_sexp }
end
end
end
end