-
Notifications
You must be signed in to change notification settings - Fork 4
/
llvmutil.rb
262 lines (232 loc) · 7.16 KB
/
llvmutil.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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
module YARV2LLVM
module LLVMUtil
include LLVM
include RubyHelpers
def get_or_create_block(ln, b, context)
if context.blocks[ln] then
context.blocks[ln]
else
context.blocks[ln] = context.builder.create_block
end
end
def check_same_type_2arg_static(p1, p2)
p1[0].add_same_type(p2[0])
p2[0].add_same_type(p1[0])
end
def check_same_type_2arg_gencode(b, context, p1, p2)
RubyType.resolve
if p1[0].type == nil then
if p2[0].type == nil then
print "ambious type #{p2[1].call(b, context).org}\n"
else
p1[0].type = p2[0].type.dup_type
end
else
if p2[0].type and p1[0].type.llvm != p2[0].type.llvm then
print "diff type #{p1[1].call(b, context).org}(#{p1[0].inspect2}) and #{p2[1].call(b, context).org}(#{p2[0].inspect2}) \n"
else
p2[0].type = p1[0].type.dup_type
end
end
end
def gen_common_opt_2arg(b, context, s1, s2)
check_same_type_2arg_gencode(b, context, s1, s2)
context = s1[1].call(b, context)
s1val = context.rc
# pppp s1[0]
context = s2[1].call(b, context)
s2val = context.rc
[s1val, s2val, context]
end
def make_frame_struct(local)
member = []
local.each do |ele|
if ele[:type].type then
member.push ele[:type].type.llvm
else
member.push VALUE
end
end
Type.struct(member)
end
def gen_array_size(b, context, arr)
recval = context.rc
aptr = b.int_to_ptr(recval, P_RARRAY)
lenptr = b.struct_gep(aptr, 1)
b.load(lenptr)
end
def gen_call_var_args_and_self(para, fname, rtype, slf)
ftype = Type.function(VALUE, [Type::Int32Ty, P_VALUE, VALUE])
gen_call_var_args_common(para, fname, rtype, ftype) {
|b, context, func, nele, argarea|
b.call(func, nele.llvm, argarea, slf)
}
end
def gen_call_var_args(para, fname, rtype)
ftype = Type.function(VALUE, [Type::Int32Ty, P_VALUE])
gen_call_var_args_common(para, fname, rtype, ftype) {
|b, context, func, nele, argarea|
b.call(func, nele.llvm, argarea)
}
end
def gen_call_var_args_common(para, fname, rtype, ftype)
args = para[:args].reverse
nele = para[:args].size
if @array_alloca_size == nil or @array_alloca_size < nele then
@array_alloca_size = nele
end
@expstack.push [rtype,
lambda {|b, context|
func = @builder.external_function(fname, ftype)
argarea = context.array_alloca_area
args.each_with_index {|pterm, i|
context = pterm[1].call(b, context)
srcval = context.rc
src = pterm[0].type.to_value(srcval, b, context)
dst = b.gep(argarea, i.llvm)
b.store(src, dst)
}
context.rc = yield(b, context, func, nele, argarea)
context}]
end
def add_global_variable(name, type, init)
@global_malloc_area_tab[name] = [type, init]
@global_malloc_area_tab.size - 1
end
end
module SendUtil
include LLVM
include RubyHelpers
MaxSmallPolymotphicNum = 4
def gen_method_select(recklass, mname)
minfo = MethodDefinition::RubyMethod[mname][recklass]
if minfo then
# recklass == nil -> functional method
return [minfo, lambda { minfo[:func]}]
end
# recklass == nil -> need dynamic dispatch
# inheritance search
if recklass then
sup = Object.const_get(recklass, true)
while sup do
minfo = MethodDefinition::RubyMethod[mname][sup.name]
if minfo then
return [minfo, lambda { minfo[:func]}]
end
if sup.is_a?(Class) then
sup = sup.superclass
else
break
end
end
end
candidatenum = MethodDefinition::RubyMethod[mname].size
if candidatenum == 0 then
return [nil, lambda {nil}]
elsif candidatenum == 1 then
minfo = MethodDefinition::RubyMethod[mname].values[0]
return [minfo, lambda {minfo[:func]}]
elsif candidatenum < MaxSmallPolymotphicNum then
# TODO : Use inline hash function generation
raise('Not implimented polymorphic methed call yet')
else
# TODO : Use cukko-hasing and inline hash function generation
raise('Not implimented polymorphic methed call yet')
end
end
def gen_call(func, arg, b, context)
args = []
arg.each do |pe|
args.push pe[1].call(b, context).rc
end
context.rc = b.call(func, *args)
context
end
=begin
def gen_get_framaddress(fstruct, b, context)
ftype = Type.function(P_CHAR, [Type::Int32Ty])
func = context.builder.external_function('llvm.frameaddress', ftype)
fraw = b.call(func, 0.llvm)
fraw2 = b.bit_cast(fraw, fstruct)
fraw2 = b.gep(fraw2, -1.llvm)
fraw = b.bit_cast(fraw2, P_CHAR)
fraw = b.gep(fraw, -4.llvm)
context.rc = fraw
context
end
=end
def gen_get_block_ptr(recklass, info, blk, b, context)
blab = (info[1].to_s + '+blk+' + blk[1].to_s).to_sym
minfo = MethodDefinition::RubyMethod[blab][recklass]
func2 = minfo[:func]
if func2 == nil then
argtype = minfo[:argtype].map {|ele|
ele.type.llvm
}
rett = minfo[:rettype]
rettllvm = rett.type
if rettllvm == nil then
rettllvm = VALUE
else
rettllvm = rettllvm.llvm
end
ftype = Type.function(rettllvm, argtype)
func2 = context.builder.get_or_insert_function(recklass, blab.to_s, ftype)
end
context.rc = b.ptr_to_int(func2, MACHINE_WORD)
context
end
def gen_arg_eval(args, receiver, ins, local, info, minfo)
blk = ins[3]
para = []
nargs = ins[2]
args.each_with_index do |pe, n|
if minfo then
pe[0].add_same_type(minfo[:argtype][nargs - n - 1])
minfo[:argtype][nargs - n - 1].add_same_value(pe[0])
end
para[n] = pe
end
para.reverse!
v = nil
if receiver then
v = receiver
else
v = [local[2][:type],
lambda {|b, context|
context.rc = b.load(context.local_vars[2][:area])
context}]
end
if receiver then
para.push [local[2][:type], lambda {|b, context|
context = v[1].call(b, context)
if v[0].type then
rc = v[0].type.to_value(context.rc, b, context)
context.rc = rc
end
context
}]
end
if blk[0] then
para.push [local[0][:type], lambda {|b, context|
# gen_get_framaddress(@frame_struct[code], b, context)
fm = context.current_frame
context.rc = b.bit_cast(fm, P_CHAR)
context
}]
para.push [local[1][:type], lambda {|b, context|
# Send with block may break local frame, so must clear local
# value cache
local.each do |le|
if le[:type].type then
le[:type].type.content = nil
end
end
# receiver of block is parent class
gen_get_block_ptr(info[0], info, blk, b, context)
}]
end
para
end
end
end