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
|
from rpython.rtyper.lltypesystem import lltype
from rpython.translator.unsimplify import varoftype
from rpython.flowspace.model import Constant, SpaceOperation
from rpython.jit.codewriter.jtransform import Transformer, NotSupported
from rpython.jit.codewriter.flatten import GraphFlattener
from rpython.jit.codewriter.format import assert_format
from rpython.jit.codewriter.test.test_flatten import fake_regallocs
from rpython.jit.metainterp.history import AbstractDescr
# ____________________________________________________________
FIXEDLIST = lltype.Ptr(lltype.GcArray(lltype.Signed))
FIXEDPTRLIST = lltype.Ptr(lltype.GcArray(FIXEDLIST))
VARLIST = lltype.Ptr(lltype.GcStruct('VARLIST',
('length', lltype.Signed),
('items', FIXEDLIST),
adtmeths={"ITEM": lltype.Signed}))
class FakeCPU:
class arraydescrof(AbstractDescr):
def __init__(self, ARRAY):
assert ARRAY.OF != lltype.Void
self.ARRAY = ARRAY
def __repr__(self):
return '<ArrayDescr>'
class fielddescrof(AbstractDescr):
def __init__(self, STRUCT, fieldname):
self.STRUCT = STRUCT
self.fieldname = fieldname
def __repr__(self):
return '<FieldDescr %s>' % self.fieldname
class sizeof(AbstractDescr):
def __init__(self, STRUCT, vtable=None):
self.STRUCT = STRUCT
def __repr__(self):
return '<SizeDescr>'
class FakeCallControl:
class getcalldescr(AbstractDescr):
def __init__(self, op, oopspecindex=0, extraeffect=None,
extradescr=None, calling_graph=None):
self.op = op
self.oopspecindex = oopspecindex
def __repr__(self):
if self.oopspecindex == 0:
return '<CallDescr>'
else:
return '<CallDescrOS%d>' % self.oopspecindex
def calldescr_canraise(self, calldescr):
return False
def builtin_test(oopspec_name, args, RESTYPE, expected):
v_result = varoftype(RESTYPE)
tr = Transformer(FakeCPU(), FakeCallControl())
tr.immutable_arrays = {}
tr.vable_array_vars = {}
if '/' in oopspec_name:
oopspec_name, property = oopspec_name.split('/')
def force_flags(op):
if property == 'NONNEG': return True
if property == 'NEG': return False
raise ValueError(property)
tr._get_list_nonneg_canraise_flags = force_flags
op = SpaceOperation('direct_call',
[Constant("myfunc", lltype.Void)] + args,
v_result)
try:
oplist = tr._handle_list_call(op, oopspec_name, args)
except NotSupported:
assert expected is NotSupported
else:
assert expected is not NotSupported
assert oplist is not None
flattener = GraphFlattener(None, fake_regallocs())
if not isinstance(oplist, list):
oplist = [oplist]
for op1 in oplist:
flattener.serialize_op(op1)
assert_format(flattener.ssarepr, expected)
# ____________________________________________________________
# Fixed lists
def test_newlist():
builtin_test('newlist', [], FIXEDLIST,
"""new_array $0, <ArrayDescr> -> %r0""")
builtin_test('newlist', [Constant(5, lltype.Signed)], FIXEDLIST,
"""new_array $5, <ArrayDescr> -> %r0""")
builtin_test('newlist', [varoftype(lltype.Signed)], FIXEDLIST,
"""new_array %i0, <ArrayDescr> -> %r0""")
builtin_test('newlist_clear', [Constant(5, lltype.Signed)], FIXEDLIST,
"""new_array_clear $5, <ArrayDescr> -> %r0""")
builtin_test('newlist', [], FIXEDPTRLIST,
"""new_array_clear $0, <ArrayDescr> -> %r0""")
def test_fixed_ll_arraycopy():
builtin_test('list.ll_arraycopy',
[varoftype(FIXEDLIST),
varoftype(FIXEDLIST),
varoftype(lltype.Signed),
varoftype(lltype.Signed),
varoftype(lltype.Signed)],
lltype.Void, """
residual_call_ir_v $'myfunc', I[%i0, %i1, %i2], R[%r0, %r1], <CallDescrOS1>
""")
def test_fixed_getitem():
builtin_test('list.getitem/NONNEG',
[varoftype(FIXEDLIST), varoftype(lltype.Signed)],
lltype.Signed, """
getarrayitem_gc_i %r0, %i0, <ArrayDescr> -> %i1
""")
builtin_test('list.getitem/NEG',
[varoftype(FIXEDLIST), varoftype(lltype.Signed)],
lltype.Signed, """
-live-
check_neg_index %r0, %i0, <ArrayDescr> -> %i1
getarrayitem_gc_i %r0, %i1, <ArrayDescr> -> %i2
""")
def test_fixed_getitem_foldable():
builtin_test('list.getitem_foldable/NONNEG',
[varoftype(FIXEDLIST), varoftype(lltype.Signed)],
lltype.Signed, """
getarrayitem_gc_i_pure %r0, %i0, <ArrayDescr> -> %i1
""")
builtin_test('list.getitem_foldable/NEG',
[varoftype(FIXEDLIST), varoftype(lltype.Signed)],
lltype.Signed, """
-live-
check_neg_index %r0, %i0, <ArrayDescr> -> %i1
getarrayitem_gc_i_pure %r0, %i1, <ArrayDescr> -> %i2
""")
def test_fixed_setitem():
builtin_test('list.setitem/NONNEG', [varoftype(FIXEDLIST),
varoftype(lltype.Signed),
varoftype(lltype.Signed)],
lltype.Void, """
setarrayitem_gc_i %r0, %i0, %i1, <ArrayDescr>
""")
builtin_test('list.setitem/NEG', [varoftype(FIXEDLIST),
varoftype(lltype.Signed),
varoftype(lltype.Signed)],
lltype.Void, """
-live-
check_neg_index %r0, %i0, <ArrayDescr> -> %i1
setarrayitem_gc_i %r0, %i1, %i2, <ArrayDescr>
""")
def test_fixed_len():
builtin_test('list.len', [varoftype(FIXEDLIST)], lltype.Signed,
"""arraylen_gc %r0, <ArrayDescr> -> %i0""")
def test_fixed_len_foldable():
builtin_test('list.len_foldable', [varoftype(FIXEDLIST)], lltype.Signed,
"""arraylen_gc %r0, <ArrayDescr> -> %i0""")
# ____________________________________________________________
# Resizable lists
def test_resizable_newlist():
alldescrs = ("<SizeDescr>, <FieldDescr length>,"
" <FieldDescr items>, <ArrayDescr>")
builtin_test('newlist', [], VARLIST,
"""newlist $0, """+alldescrs+""" -> %r0""")
builtin_test('newlist', [Constant(5, lltype.Signed)], VARLIST,
"""newlist $5, """+alldescrs+""" -> %r0""")
builtin_test('newlist', [varoftype(lltype.Signed)], VARLIST,
"""newlist %i0, """+alldescrs+""" -> %r0""")
builtin_test('newlist_clear', [Constant(5, lltype.Signed)], VARLIST,
"""newlist_clear $5, """+alldescrs+""" -> %r0""")
def test_resizable_getitem():
builtin_test('list.getitem/NONNEG',
[varoftype(VARLIST), varoftype(lltype.Signed)],
lltype.Signed, """
getlistitem_gc_i %r0, %i0, <FieldDescr items>, <ArrayDescr> -> %i1
""")
builtin_test('list.getitem/NEG',
[varoftype(VARLIST), varoftype(lltype.Signed)],
lltype.Signed, """
-live-
check_resizable_neg_index %r0, %i0, <FieldDescr length> -> %i1
getlistitem_gc_i %r0, %i1, <FieldDescr items>, <ArrayDescr> -> %i2
""")
def test_resizable_setitem():
builtin_test('list.setitem/NONNEG', [varoftype(VARLIST),
varoftype(lltype.Signed),
varoftype(lltype.Signed)],
lltype.Void, """
setlistitem_gc_i %r0, %i0, %i1, <FieldDescr items>, <ArrayDescr>
""")
builtin_test('list.setitem/NEG', [varoftype(VARLIST),
varoftype(lltype.Signed),
varoftype(lltype.Signed)],
lltype.Void, """
-live-
check_resizable_neg_index %r0, %i0, <FieldDescr length> -> %i1
setlistitem_gc_i %r0, %i1, %i2, <FieldDescr items>, <ArrayDescr>
""")
def test_resizable_len():
builtin_test('list.len', [varoftype(VARLIST)], lltype.Signed,
"""getfield_gc_i %r0, <FieldDescr length> -> %i0""")
def test_resizable_unsupportedop():
builtin_test('list.foobar', [varoftype(VARLIST)], lltype.Signed,
NotSupported)
|
