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
|
local insert, remove, concat
do
local _obj_0 = table
insert, remove, concat = _obj_0.insert, _obj_0.remove, _obj_0.concat
end
local repr, stringify, equivalent, nth_to_last, size
do
local _obj_0 = require('utils')
repr, stringify, equivalent, nth_to_last, size = _obj_0.repr, _obj_0.stringify, _obj_0.equivalent, _obj_0.nth_to_last, _obj_0.size
end
local list, dict
local _list_mt = {
__eq = equivalent,
__tostring = function(self)
return "[" .. concat((function()
local _accum_0 = { }
local _len_0 = 1
for _index_0 = 1, #self do
local b = self[_index_0]
_accum_0[_len_0] = repr(b)
_len_0 = _len_0 + 1
end
return _accum_0
end)(), ", ") .. "]"
end,
__lt = function(self, other)
assert(type(self) == 'table' and type(other) == 'table', "Incompatible types for comparison")
for i = 1, math.max(#self, #other) do
if not self[i] and other[i] then
return true
elseif self[i] and not other[i] then
return false
elseif self[i] < other[i] then
return true
elseif self[i] > other[i] then
return false
end
end
return false
end,
__le = function(self, other)
assert(type(self) == 'table' and type(other) == 'table', "Incompatible types for comparison")
for i = 1, math.max(#self, #other) do
if not self[i] and other[i] then
return true
elseif self[i] and not other[i] then
return false
elseif self[i] < other[i] then
return true
elseif self[i] > other[i] then
return false
end
end
return true
end,
__add = function(self, other)
local ret = list((function()
local _accum_0 = { }
local _len_0 = 1
for _index_0 = 1, #self do
local x = self[_index_0]
_accum_0[_len_0] = x
_len_0 = _len_0 + 1
end
return _accum_0
end)())
for _index_0 = 1, #other do
local x = other[_index_0]
insert(ret, x)
end
return ret
end,
__index = {
add_1 = insert,
append_1 = insert,
add_1_at_index_2 = function(t, x, i)
return insert(t, i, x)
end,
at_index_1_add_2 = insert,
pop = remove,
remove_last = remove,
remove_index_1 = remove,
last = (function(self)
return self[#self]
end),
first = (function(self)
return self[1]
end),
_1_st_to_last = nth_to_last,
_1_nd_to_last = nth_to_last,
_1_rd_to_last = nth_to_last,
_1_th_to_last = nth_to_last,
has_1 = function(self, item)
for _index_0 = 1, #self do
local x = self[_index_0]
if x == item then
return true
end
end
return false
end,
index_of_1 = function(self, item)
for i, x in ipairs(self) do
if x == item then
return i
end
end
return nil
end
}
}
list = function(t)
return setmetatable(t, _list_mt)
end
local walk_items
walk_items = function(self, i)
i = i + 1
local k, v = next(self.table, self.key)
if k ~= nil then
self.key = k
return i, dict({
key = k,
value = v
})
end
end
local _dict_mt = {
__eq = equivalent,
__len = size,
__tostring = function(self)
return "{" .. concat((function()
local _accum_0 = { }
local _len_0 = 1
for k, v in pairs(self) do
_accum_0[_len_0] = tostring(repr(k)) .. ": " .. tostring(repr(v))
_len_0 = _len_0 + 1
end
return _accum_0
end)(), ", ") .. "}"
end,
__ipairs = function(self)
return walk_items, {
table = self,
key = nil
}, 0
end,
__band = function(self, other)
return dict((function()
local _tbl_0 = { }
for k, v in pairs(self) do
if other[k] ~= nil then
_tbl_0[k] = v
end
end
return _tbl_0
end)())
end,
__bor = function(self, other)
local ret
do
local _tbl_0 = { }
for k, v in pairs(self) do
_tbl_0[k] = v
end
ret = _tbl_0
end
for k, v in pairs(other) do
if ret[k] == nil then
ret[k] = v
end
end
return dict(ret)
end,
__bxor = function(self, other)
local ret
do
local _tbl_0 = { }
for k, v in pairs(self) do
_tbl_0[k] = v
end
ret = _tbl_0
end
for k, v in pairs(other) do
if ret[k] == nil then
ret[k] = v
else
ret[k] = nil
end
end
return dict(ret)
end,
__add = function(self, other)
local ret
do
local _tbl_0 = { }
for k, v in pairs(self) do
_tbl_0[k] = v
end
ret = _tbl_0
end
for k, v in pairs(other) do
if ret[k] == nil then
ret[k] = v
else
ret[k] = ret[k] + v
end
end
return dict(ret)
end,
__sub = function(self, other)
local ret
do
local _tbl_0 = { }
for k, v in pairs(self) do
_tbl_0[k] = v
end
ret = _tbl_0
end
for k, v in pairs(other) do
if ret[k] == nil then
ret[k] = -v
else
ret[k] = ret[k] - v
end
end
return dict(ret)
end
}
dict = function(t)
return setmetatable(t, _dict_mt)
end
for i, entry in ipairs(dict({
x = 99
})) do
assert(i == 1 and entry.key == "x" and entry.value == 99, "ipairs compatibility issue")
end
return {
list = list,
dict = dict
}
|
