auto-editor 25.0.1__py3-none-any.whl → 25.1.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- auto_editor/__init__.py +1 -1
- auto_editor/__main__.py +3 -0
- auto_editor/edit.py +22 -18
- auto_editor/formats/fcp11.py +16 -26
- auto_editor/formats/fcp7.py +144 -74
- auto_editor/lang/libintrospection.py +10 -0
- auto_editor/lang/palet.py +22 -1183
- auto_editor/lang/stdenv.py +1139 -0
- auto_editor/make_layers.py +14 -0
- auto_editor/render/subtitle.py +3 -1
- auto_editor/subcommands/palet.py +2 -0
- auto_editor/subcommands/repl.py +3 -2
- auto_editor/subcommands/test.py +3 -0
- auto_editor/utils/types.py +1 -0
- {auto_editor-25.0.1.dist-info → auto_editor-25.1.0.dist-info}/METADATA +1 -1
- {auto_editor-25.0.1.dist-info → auto_editor-25.1.0.dist-info}/RECORD +21 -18
- {auto_editor-25.0.1.dist-info → auto_editor-25.1.0.dist-info}/WHEEL +1 -1
- {auto_editor-25.0.1.dist-info → auto_editor-25.1.0.dist-info}/top_level.txt +1 -0
- docs/build.py +53 -0
- {auto_editor-25.0.1.dist-info → auto_editor-25.1.0.dist-info}/LICENSE +0 -0
- {auto_editor-25.0.1.dist-info → auto_editor-25.1.0.dist-info}/entry_points.txt +0 -0
@@ -0,0 +1,1139 @@
|
|
1
|
+
from __future__ import annotations
|
2
|
+
|
3
|
+
from typing import TYPE_CHECKING
|
4
|
+
|
5
|
+
from auto_editor.analyze import mut_remove_large, mut_remove_small
|
6
|
+
from auto_editor.lib.contracts import *
|
7
|
+
from auto_editor.lib.data_structs import *
|
8
|
+
|
9
|
+
from .palet import Syntax, env, is_boolarr, is_iterable, my_eval, p_slice, raise_, ref
|
10
|
+
|
11
|
+
if TYPE_CHECKING:
|
12
|
+
from typing import Any, Literal
|
13
|
+
|
14
|
+
from numpy.typing import NDArray
|
15
|
+
|
16
|
+
Number = int | float | complex | Fraction
|
17
|
+
Real = int | float | Fraction
|
18
|
+
BoolList = NDArray[np.bool_]
|
19
|
+
Node = tuple
|
20
|
+
|
21
|
+
|
22
|
+
def make_standard_env() -> dict[str, Any]:
|
23
|
+
from cmath import sqrt as complex_sqrt
|
24
|
+
from functools import reduce
|
25
|
+
from operator import add, ge, gt, is_, le, lt, mod, mul
|
26
|
+
from subprocess import run
|
27
|
+
from time import sleep
|
28
|
+
|
29
|
+
from numpy import logical_and, logical_not, logical_or, logical_xor
|
30
|
+
|
31
|
+
from auto_editor.utils.func import boolop, mut_margin
|
32
|
+
|
33
|
+
# fmt: off
|
34
|
+
|
35
|
+
class UserProc(Proc):
|
36
|
+
"""A user-defined procedure."""
|
37
|
+
|
38
|
+
__slots__ = ("env", "name", "parms", "body", "contracts", "arity")
|
39
|
+
|
40
|
+
def __init__(
|
41
|
+
self,
|
42
|
+
env: Env,
|
43
|
+
name: str,
|
44
|
+
parms: list[str],
|
45
|
+
contracts: tuple[Any, ...],
|
46
|
+
body: Node,
|
47
|
+
):
|
48
|
+
self.env = env
|
49
|
+
self.name = name
|
50
|
+
self.parms = parms
|
51
|
+
self.body = body
|
52
|
+
self.contracts = contracts
|
53
|
+
|
54
|
+
if parms and parms[-1] == "...":
|
55
|
+
parms.pop()
|
56
|
+
self.arity: tuple[int, int | None] = len(parms) - 1, None
|
57
|
+
else:
|
58
|
+
self.arity = len(parms), len(parms)
|
59
|
+
|
60
|
+
def __call__(self, *args: Any) -> Any:
|
61
|
+
check_args(self.name, args, self.arity, self.contracts)
|
62
|
+
|
63
|
+
if self.arity[1] is None:
|
64
|
+
args = tuple(
|
65
|
+
list(args[: len(self.parms) - 1]) + [list(args[len(self.parms) - 1 :])]
|
66
|
+
)
|
67
|
+
|
68
|
+
inner_env = Env(dict(zip(self.parms, args)), self.env)
|
69
|
+
|
70
|
+
for item in self.body[0:-1]:
|
71
|
+
my_eval(inner_env, item)
|
72
|
+
|
73
|
+
return my_eval(inner_env, self.body[-1])
|
74
|
+
|
75
|
+
@dataclass(slots=True)
|
76
|
+
class KeywordUserProc:
|
77
|
+
env: Env
|
78
|
+
name: str
|
79
|
+
parms: list[str]
|
80
|
+
kw_parms: list[str]
|
81
|
+
body: Node
|
82
|
+
arity: tuple[int, None]
|
83
|
+
contracts: list[Any] | None = None
|
84
|
+
|
85
|
+
def __call__(self, *args: Any, **kwargs: Any) -> Any:
|
86
|
+
env = {}
|
87
|
+
all_parms = self.parms + self.kw_parms
|
88
|
+
for i, arg in enumerate(args):
|
89
|
+
if i >= len(all_parms):
|
90
|
+
raise MyError("Too many arguments")
|
91
|
+
env[all_parms[i]] = arg
|
92
|
+
|
93
|
+
for key, val in kwargs.items():
|
94
|
+
if key in env:
|
95
|
+
raise MyError(
|
96
|
+
f"Keyword: {key} already fulfilled by positional argument."
|
97
|
+
)
|
98
|
+
else:
|
99
|
+
env[key] = val
|
100
|
+
|
101
|
+
inner_env = Env(env, self.env)
|
102
|
+
|
103
|
+
for item in self.body[0:-1]:
|
104
|
+
my_eval(inner_env, item)
|
105
|
+
|
106
|
+
return my_eval(inner_env, self.body[-1])
|
107
|
+
|
108
|
+
def __str__(self) -> str:
|
109
|
+
return self.name
|
110
|
+
|
111
|
+
def __repr__(self) -> str:
|
112
|
+
return f"#<kw-proc:{self.name}>"
|
113
|
+
|
114
|
+
is_cont = Contract("contract?", is_contract)
|
115
|
+
is_sequence = Contract(
|
116
|
+
"sequence?",
|
117
|
+
lambda v: type(v) in (str, range, Quoted) or isinstance(v, list | np.ndarray),
|
118
|
+
)
|
119
|
+
bool_or_barr = Contract(
|
120
|
+
"(or/c bool? bool-array?)",
|
121
|
+
lambda v: type(v) is bool or is_boolarr(v),
|
122
|
+
)
|
123
|
+
is_keyw = Contract("keyword?", lambda v: type(v) is QuotedKeyword)
|
124
|
+
dtype_map = {
|
125
|
+
Sym("bool"): np.bool_,
|
126
|
+
Sym("int8"): np.int8,
|
127
|
+
Sym("int16"): np.int16,
|
128
|
+
Sym("int32"): np.int32,
|
129
|
+
Sym("int64"): np.int64,
|
130
|
+
Sym("uint8"): np.uint8,
|
131
|
+
Sym("uint16"): np.uint16,
|
132
|
+
Sym("uint32"): np.uint32,
|
133
|
+
Sym("uint64"): np.uint64,
|
134
|
+
Sym("float32"): np.float32,
|
135
|
+
Sym("float64"): np.float64,
|
136
|
+
}
|
137
|
+
|
138
|
+
@dataclass(slots=True)
|
139
|
+
class OutputPort:
|
140
|
+
name: str
|
141
|
+
port: Any
|
142
|
+
write: Any
|
143
|
+
closed: bool
|
144
|
+
|
145
|
+
def close(self) -> None:
|
146
|
+
if not self.closed:
|
147
|
+
self.closed = True
|
148
|
+
self.port.close()
|
149
|
+
|
150
|
+
def __str__(self) -> str:
|
151
|
+
return f"#<output-port:{self.name}>"
|
152
|
+
|
153
|
+
__repr__ = __str__
|
154
|
+
|
155
|
+
def initOutPort(name: str) -> OutputPort | Literal[False]:
|
156
|
+
try:
|
157
|
+
port = open(name, "w", encoding="utf-8")
|
158
|
+
except Exception:
|
159
|
+
return False
|
160
|
+
return OutputPort(name, port, port.write, False)
|
161
|
+
|
162
|
+
def is_equal(a: object, b: object) -> bool:
|
163
|
+
if isinstance(a, np.ndarray) and isinstance(b, np.ndarray):
|
164
|
+
return np.array_equal(a, b)
|
165
|
+
return type(a) == type(b) and a == b
|
166
|
+
|
167
|
+
# Syntax definitions
|
168
|
+
def check_for_syntax(env: Env, node: Node) -> tuple[Sym, Any]:
|
169
|
+
name = node[0]
|
170
|
+
if len(node) < 2:
|
171
|
+
raise MyError(f"{name}: bad syntax")
|
172
|
+
|
173
|
+
if len(node) == 2:
|
174
|
+
raise MyError(f"{name}: missing body")
|
175
|
+
|
176
|
+
assert isinstance(node[1], tuple)
|
177
|
+
assert isinstance(node[1][0], tuple)
|
178
|
+
|
179
|
+
var = node[1][0][0]
|
180
|
+
if type(var) is not Sym:
|
181
|
+
raise MyError(f"{name}: binding must be an identifier")
|
182
|
+
my_iter = my_eval(env, node[1][0][1])
|
183
|
+
|
184
|
+
if not is_iterable(my_iter):
|
185
|
+
if type(my_iter) is int:
|
186
|
+
return var, range(my_iter)
|
187
|
+
raise MyError(f"{name}: got non-iterable in iter slot")
|
188
|
+
|
189
|
+
return var, my_iter
|
190
|
+
|
191
|
+
def syn_lambda(env: Env, node: Node) -> UserProc:
|
192
|
+
if len(node) < 3:
|
193
|
+
raise MyError(f"{node[0]}: too few terms")
|
194
|
+
|
195
|
+
if type(node[1]) is not tuple:
|
196
|
+
raise MyError(f"{node[0]}: bad syntax")
|
197
|
+
|
198
|
+
parms: list[str] = []
|
199
|
+
for item in node[1]:
|
200
|
+
if type(item) is not Sym:
|
201
|
+
raise MyError(f"{node[0]}: must be an identifier")
|
202
|
+
|
203
|
+
parms.append(f"{item}")
|
204
|
+
|
205
|
+
return UserProc(env, "", parms, (), node[2:])
|
206
|
+
|
207
|
+
def syn_define(env: Env, node: Node) -> None:
|
208
|
+
if len(node) < 3:
|
209
|
+
raise MyError(f"{node[0]}: too few terms")
|
210
|
+
|
211
|
+
if type(node[1]) is tuple:
|
212
|
+
term = node[1]
|
213
|
+
body = node[2:]
|
214
|
+
|
215
|
+
if not term or type(term[0]) is not Sym:
|
216
|
+
raise MyError(f"{node[0]}: proc-binding must be an identifier")
|
217
|
+
|
218
|
+
n = term[0].val
|
219
|
+
parms: list[str] = []
|
220
|
+
kparms: list[str] = []
|
221
|
+
kw_only = False
|
222
|
+
|
223
|
+
for item in term[1:]:
|
224
|
+
if kw_only:
|
225
|
+
if type(item) is Sym:
|
226
|
+
raise MyError(f"{node[0]}: {item} must be a keyword")
|
227
|
+
if type(item) is not Keyword:
|
228
|
+
raise MyError(f"{node[0]}: must be a keyword")
|
229
|
+
kparms.append(item.val)
|
230
|
+
else:
|
231
|
+
if type(item) is Keyword:
|
232
|
+
kw_only = True
|
233
|
+
kparms.append(item.val)
|
234
|
+
elif type(item) is Sym:
|
235
|
+
parms.append(item.val)
|
236
|
+
else:
|
237
|
+
raise MyError(f"{node[0]}: must be an identifier")
|
238
|
+
|
239
|
+
if kw_only:
|
240
|
+
env[n] = KeywordUserProc(env, n, parms, kparms, body, (len(parms), None))
|
241
|
+
else:
|
242
|
+
env[n] = UserProc(env, n, parms, (), body)
|
243
|
+
return None
|
244
|
+
|
245
|
+
elif type(node[1]) is not Sym:
|
246
|
+
raise MyError(f"{node[0]}: must be an identifier")
|
247
|
+
|
248
|
+
if len(node) > 3:
|
249
|
+
raise MyError(f"{node[0]}: multiple expressions after identifier")
|
250
|
+
|
251
|
+
n = node[1].val
|
252
|
+
|
253
|
+
if (
|
254
|
+
type(node[2]) is tuple
|
255
|
+
and node[2]
|
256
|
+
and type(node[2][0]) is Sym
|
257
|
+
and node[2][0].val in ("lambda", "λ")
|
258
|
+
):
|
259
|
+
terms = node[2][1]
|
260
|
+
body = node[2][2:]
|
261
|
+
|
262
|
+
parms = []
|
263
|
+
for item in terms:
|
264
|
+
if type(item) is not Sym:
|
265
|
+
raise MyError(f"{node[0]}: must be an identifier")
|
266
|
+
|
267
|
+
parms.append(f"{item}")
|
268
|
+
|
269
|
+
env[n] = UserProc(env, n, parms, (), body)
|
270
|
+
|
271
|
+
else:
|
272
|
+
for item in node[2:-1]:
|
273
|
+
my_eval(env, item)
|
274
|
+
env[n] = my_eval(env, node[-1])
|
275
|
+
|
276
|
+
def syn_definec(env: Env, node: Node) -> None:
|
277
|
+
if len(node) < 3:
|
278
|
+
raise MyError(f"{node[0]}: too few terms")
|
279
|
+
|
280
|
+
if type(node[1]) is not tuple:
|
281
|
+
raise MyError(f"{node[0]} only allows procedure declarations")
|
282
|
+
|
283
|
+
if not node[1] or type(node[1][0]) is not Sym:
|
284
|
+
raise MyError(f"{node[0]}: bad proc-binding syntax")
|
285
|
+
|
286
|
+
n = node[1][0].val
|
287
|
+
|
288
|
+
contracts: list[Any] = []
|
289
|
+
parms: list[str] = []
|
290
|
+
for item in node[1][1:]:
|
291
|
+
if item == Sym("->"):
|
292
|
+
break
|
293
|
+
if type(item) is not tuple or len(item) != 2:
|
294
|
+
raise MyError(f"{node[0]}: bad var-binding syntax")
|
295
|
+
if type(item[0]) is not Sym:
|
296
|
+
raise MyError(f"{node[0]}: binding must be identifier")
|
297
|
+
|
298
|
+
con = my_eval(env, item[1])
|
299
|
+
if not is_cont(con):
|
300
|
+
raise MyError(f"{node[0]}: {print_str(con)} is not a valid contract")
|
301
|
+
|
302
|
+
parms.append(f"{item[0]}")
|
303
|
+
contracts.append(con)
|
304
|
+
|
305
|
+
env[n] = UserProc(env, n, parms, tuple(contracts), node[2:])
|
306
|
+
return None
|
307
|
+
|
308
|
+
def guard_term(node: Node, n: int, u: int) -> None:
|
309
|
+
if n == u:
|
310
|
+
if len(node) != n:
|
311
|
+
raise MyError(
|
312
|
+
f"{node[0]}: Expects exactly {n-1} term{'s' if n > 2 else ''}"
|
313
|
+
)
|
314
|
+
return None
|
315
|
+
if len(node) < n:
|
316
|
+
raise MyError(f"{node[0]}: Expects at least {n-1} term{'s' if n > 2 else ''}")
|
317
|
+
if len(node) > u:
|
318
|
+
raise MyError(f"{node[0]}: Expects at most {u-1} term{'s' if u > 2 else ''}")
|
319
|
+
|
320
|
+
def syn_set(env: Env, node: Node) -> None:
|
321
|
+
guard_term(node, 3, 3)
|
322
|
+
|
323
|
+
if type(node[1]) is Sym:
|
324
|
+
name = node[1].val
|
325
|
+
if name not in env:
|
326
|
+
raise MyError(f"{node[0]}: Can't set variable `{name}` before definition")
|
327
|
+
env[name] = my_eval(env, node[2])
|
328
|
+
return None
|
329
|
+
|
330
|
+
if type(node[1]) is tuple and len(node[1]) == 3 and node[1][0] == Sym("@r"):
|
331
|
+
base = my_eval(env, node[1][1])
|
332
|
+
name = node[1][2].val
|
333
|
+
for i, item in enumerate(base.attrs[0::2]):
|
334
|
+
if name == item:
|
335
|
+
result = my_eval(env, node[2])
|
336
|
+
check_args(item, (result,), (1, 1), (base.attrs[i * 2 + 1],))
|
337
|
+
base.values[i] = result
|
338
|
+
return None
|
339
|
+
|
340
|
+
raise MyError(f"{node[0]}: {base.name} has no attribute `{name}`")
|
341
|
+
|
342
|
+
raise MyError(f"{node[0]}: Expected identifier, got {print_str(node[1])}")
|
343
|
+
|
344
|
+
def syn_incf(env: Env, node: Node) -> None:
|
345
|
+
guard_term(node, 2, 3)
|
346
|
+
|
347
|
+
incre_by = 1
|
348
|
+
if len(node) == 3:
|
349
|
+
incre_by = my_eval(env, node[2])
|
350
|
+
if not is_num(incre_by):
|
351
|
+
raise MyError(f"{node[0]}: Expected number? got: {print_str(incre_by)}")
|
352
|
+
|
353
|
+
if type(node[1]) is Sym:
|
354
|
+
name = node[1].val
|
355
|
+
|
356
|
+
if type(env[name]) is NotFound:
|
357
|
+
raise MyError(f"{node[0]}: `{name}` is not defined")
|
358
|
+
if not is_num(env[name]):
|
359
|
+
raise MyError(f"{node[0]}: `{name}` is not a number?")
|
360
|
+
env[name] += incre_by
|
361
|
+
return None
|
362
|
+
|
363
|
+
if type(node[1]) is tuple and len(node[1]) == 3 and node[1][0] == Sym("@r"):
|
364
|
+
base = my_eval(env, node[1][1])
|
365
|
+
if type(base) is not PaletClass:
|
366
|
+
raise MyError(f"{node[0]}: must be a class instance")
|
367
|
+
if type(node[1][2]) is not Sym:
|
368
|
+
raise MyError(f"{node[0]}: class attribute must be an identifier")
|
369
|
+
name = node[1][2].val
|
370
|
+
for i, item in enumerate(base.attrs[0::2]):
|
371
|
+
if name == item:
|
372
|
+
if not is_num(base.values[i]):
|
373
|
+
raise MyError(f"{node[0]}: `{name}` is not a number?")
|
374
|
+
|
375
|
+
check_args(
|
376
|
+
name, (base.values[i] + incre_by,), (1, 1), (base.attrs[i * 2 + 1],)
|
377
|
+
)
|
378
|
+
base.values[i] += incre_by
|
379
|
+
return None
|
380
|
+
raise MyError(f"{node[0]}: {base.name} has no attribute `{name}`")
|
381
|
+
|
382
|
+
raise MyError(f"{node[0]}: Expected identifier, got {print_str(node[1])}")
|
383
|
+
|
384
|
+
def syn_decf(env: Env, node: Node) -> None:
|
385
|
+
guard_term(node, 2, 3)
|
386
|
+
|
387
|
+
incre_by = 1
|
388
|
+
if len(node) == 3:
|
389
|
+
incre_by = my_eval(env, node[2])
|
390
|
+
if not is_num(incre_by):
|
391
|
+
raise MyError(f"{node[0]}: Expected number? got: {print_str(incre_by)}")
|
392
|
+
|
393
|
+
if type(node[1]) is Sym:
|
394
|
+
name = node[1].val
|
395
|
+
|
396
|
+
if type(env[name]) is NotFound:
|
397
|
+
raise MyError(f"{node[0]}: `{name}` is not defined")
|
398
|
+
if not is_num(env[name]):
|
399
|
+
raise MyError(f"{node[0]}: `{name}` is not a number?")
|
400
|
+
env[name] -= incre_by
|
401
|
+
return None
|
402
|
+
|
403
|
+
if type(node[1]) is tuple and len(node[1]) == 3 and node[1][0] == Sym("@r"):
|
404
|
+
base = my_eval(env, node[1][1])
|
405
|
+
if type(base) is not PaletClass:
|
406
|
+
raise MyError(f"{node[0]}: must be a class instance")
|
407
|
+
if type(node[1][2]) is not Sym:
|
408
|
+
raise MyError(f"{node[0]}: class attribute must be an identifier")
|
409
|
+
name = node[1][2].val
|
410
|
+
for i, item in enumerate(base.attrs[0::2]):
|
411
|
+
if name == item:
|
412
|
+
if not is_num(base.values[i]):
|
413
|
+
raise MyError(f"{node[0]}: `{name}` is not a number?")
|
414
|
+
|
415
|
+
check_args(
|
416
|
+
name, (base.values[i] - incre_by,), (1, 1), (base.attrs[i * 2 + 1],)
|
417
|
+
)
|
418
|
+
base.values[i] -= incre_by
|
419
|
+
return None
|
420
|
+
raise MyError(f"{node[0]}: {base.name} has no attribute `{name}`")
|
421
|
+
|
422
|
+
raise MyError(f"{node[0]}: Expected identifier, got {print_str(node[1])}")
|
423
|
+
|
424
|
+
def syn_strappend(env: Env, node: Node) -> None:
|
425
|
+
guard_term(node, 3, 3)
|
426
|
+
|
427
|
+
if type(node[1]) is not Sym:
|
428
|
+
raise MyError(f"{node[0]}: Expected identifier, got {print_str(node[1])}")
|
429
|
+
name = node[1].val
|
430
|
+
|
431
|
+
if type(env[name]) is NotFound:
|
432
|
+
raise MyError(f"{node[0]}: `{name}` is not defined")
|
433
|
+
if not is_str(env[name]):
|
434
|
+
raise MyError(f"{node[0]}: `{name}` is not a string?")
|
435
|
+
|
436
|
+
if not is_str(num := my_eval(env, node[2])):
|
437
|
+
raise MyError(f"{node[0]}: Expected string? got: {print_str(num)}")
|
438
|
+
env[name] += num
|
439
|
+
|
440
|
+
def syn_for(env: Env, node: Node) -> None:
|
441
|
+
var, my_iter = check_for_syntax(env, node)
|
442
|
+
|
443
|
+
if isinstance(my_iter, np.ndarray) and my_iter.dtype == np.bool_:
|
444
|
+
for item in my_iter:
|
445
|
+
env[var.val] = int(item)
|
446
|
+
for c in node[2:]:
|
447
|
+
my_eval(env, c)
|
448
|
+
else:
|
449
|
+
for item in my_iter:
|
450
|
+
env[var.val] = item
|
451
|
+
for c in node[2:]:
|
452
|
+
my_eval(env, c)
|
453
|
+
|
454
|
+
def syn_for_items(env: Env, node: Node) -> None:
|
455
|
+
if len(node) < 2:
|
456
|
+
raise MyError(f"{node[0]}: bad syntax")
|
457
|
+
|
458
|
+
if type(node[1]) is not tuple or len(node[1]) != 3:
|
459
|
+
raise MyError(f"{node[0]}: Invalid id body")
|
460
|
+
|
461
|
+
key, val, dic = node[1]
|
462
|
+
if type(key) is not Sym or type(val) is not Sym:
|
463
|
+
raise MyError(f"{node[0]}: key and val must be identifiers")
|
464
|
+
|
465
|
+
dic = my_eval(env, dic)
|
466
|
+
if type(dic) is not dict:
|
467
|
+
raise MyError(f"{node[0]}: dict must be a hash?")
|
468
|
+
|
469
|
+
for k, v in dic.items():
|
470
|
+
env[key.val] = k
|
471
|
+
env[val.val] = v
|
472
|
+
for c in node[2:]:
|
473
|
+
my_eval(env, c)
|
474
|
+
|
475
|
+
def syn_quote(env: Env, node: Node) -> Any:
|
476
|
+
guard_term(node, 2, 2)
|
477
|
+
if type(node[1]) is Keyword:
|
478
|
+
return QuotedKeyword(node[1])
|
479
|
+
if type(node[1]) is tuple:
|
480
|
+
return Quoted(node[1])
|
481
|
+
return node[1]
|
482
|
+
|
483
|
+
def syn_if(env: Env, node: Node) -> Any:
|
484
|
+
guard_term(node, 4, 4)
|
485
|
+
test_expr = my_eval(env, node[1])
|
486
|
+
|
487
|
+
if type(test_expr) is not bool:
|
488
|
+
raise MyError(
|
489
|
+
f"{node[0]} test-expr: expected bool?, got {print_str(test_expr)}"
|
490
|
+
)
|
491
|
+
|
492
|
+
return my_eval(env, node[2] if test_expr else node[3])
|
493
|
+
|
494
|
+
def syn_when(env: Env, node: Node) -> Any:
|
495
|
+
if len(node) < 3:
|
496
|
+
raise MyError(f"{node[0]}: Expected at least 2 terms")
|
497
|
+
test_expr = my_eval(env, node[1])
|
498
|
+
|
499
|
+
if type(test_expr) is not bool:
|
500
|
+
raise MyError(
|
501
|
+
f"{node[0]} test-expr: expected bool?, got {print_str(test_expr)}"
|
502
|
+
)
|
503
|
+
|
504
|
+
if test_expr:
|
505
|
+
for item in node[2:-1]:
|
506
|
+
my_eval(env, item)
|
507
|
+
return my_eval(env, node[-1])
|
508
|
+
return None
|
509
|
+
|
510
|
+
def syn_and(env: Env, node: Node) -> Any:
|
511
|
+
if len(node) == 1:
|
512
|
+
raise MyError(f"{node[0]}: Expected at least 1 term")
|
513
|
+
|
514
|
+
first = my_eval(env, node[1])
|
515
|
+
if first is False:
|
516
|
+
return False
|
517
|
+
if first is True:
|
518
|
+
for n in node[2:]:
|
519
|
+
val = my_eval(env, n)
|
520
|
+
if val is False:
|
521
|
+
return False
|
522
|
+
if val is not True:
|
523
|
+
raise MyError(f"{node[0]} args must be bool?")
|
524
|
+
return True
|
525
|
+
|
526
|
+
if is_boolarr(first):
|
527
|
+
vals = [first] + [my_eval(env, n) for n in node[2:]]
|
528
|
+
check_args(node[0], vals, (2, None), (is_boolarr,))
|
529
|
+
return reduce(lambda a, b: boolop(a, b, logical_and), vals)
|
530
|
+
|
531
|
+
raise MyError(f"{node[0]} expects (or/c bool? bool-array?)")
|
532
|
+
|
533
|
+
def syn_or(env: Env, node: Node) -> Any:
|
534
|
+
if len(node) == 1:
|
535
|
+
raise MyError(f"{node[0]}: Expected at least 1 term")
|
536
|
+
|
537
|
+
first = my_eval(env, node[1])
|
538
|
+
if first is True:
|
539
|
+
return True
|
540
|
+
if first is False:
|
541
|
+
for n in node[2:]:
|
542
|
+
val = my_eval(env, n)
|
543
|
+
if val is True:
|
544
|
+
return True
|
545
|
+
if val is not False:
|
546
|
+
raise MyError(f"{node[0]} args must be bool?")
|
547
|
+
return False
|
548
|
+
|
549
|
+
if is_boolarr(first):
|
550
|
+
vals = [first] + [my_eval(env, n) for n in node[2:]]
|
551
|
+
check_args(node[0], vals, (2, None), (is_boolarr,))
|
552
|
+
return reduce(lambda a, b: boolop(a, b, logical_or), vals)
|
553
|
+
|
554
|
+
raise MyError(f"{node[0]} expects (or/c bool? bool-array?)")
|
555
|
+
|
556
|
+
def syn_delete(env: Env, node: Node) -> None:
|
557
|
+
guard_term(node, 2, 2)
|
558
|
+
if type(node[1]) is not Sym:
|
559
|
+
raise MyError(f"{node[0]}: Expected identifier for first term")
|
560
|
+
|
561
|
+
del env[node[1].val]
|
562
|
+
|
563
|
+
def syn_rename(env: Env, node: Node) -> None:
|
564
|
+
guard_term(node, 3, 3)
|
565
|
+
|
566
|
+
first = node[1]
|
567
|
+
if type(first) is not Sym:
|
568
|
+
raise MyError(f"{node[0]}: Expected identifier for first term")
|
569
|
+
|
570
|
+
sec = node[2]
|
571
|
+
if type(sec) is not Sym:
|
572
|
+
raise MyError(f"{node[0]}: Expected identifier for second term")
|
573
|
+
|
574
|
+
if first.val not in env:
|
575
|
+
raise MyError(f"{node[0]}: Original identifier does not exist")
|
576
|
+
|
577
|
+
env[sec.val] = env[first.val]
|
578
|
+
del env[first.val]
|
579
|
+
|
580
|
+
def syn_cond(env: Env, node: Node) -> Any:
|
581
|
+
for test_expr in node[1:]:
|
582
|
+
if type(test_expr) is not tuple or not test_expr:
|
583
|
+
raise MyError(f"{node[0]}: bad syntax, clause is not a test-value pair")
|
584
|
+
|
585
|
+
if test_expr[0] == Sym("else"):
|
586
|
+
if len(test_expr) == 1:
|
587
|
+
raise MyError(f"{node[0]}: missing expression in else clause")
|
588
|
+
test_clause = True
|
589
|
+
else:
|
590
|
+
test_clause = my_eval(env, test_expr[0])
|
591
|
+
if type(test_clause) is not bool:
|
592
|
+
raise MyError(
|
593
|
+
f"{node[0]} test-expr: expected bool?, got {print_str(test_clause)}"
|
594
|
+
)
|
595
|
+
|
596
|
+
if test_clause:
|
597
|
+
if len(test_expr) == 1:
|
598
|
+
return True
|
599
|
+
|
600
|
+
for rest_clause in test_expr[1:-1]:
|
601
|
+
my_eval(env, rest_clause)
|
602
|
+
return my_eval(env, test_expr[-1])
|
603
|
+
|
604
|
+
return None
|
605
|
+
|
606
|
+
def syn_case(env: Env, node: Node) -> Any:
|
607
|
+
val_expr = my_eval(env, node[1])
|
608
|
+
for case_clause in node[2:]:
|
609
|
+
if type(case_clause) is not tuple or len(case_clause) != 2:
|
610
|
+
raise MyError("case: bad syntax")
|
611
|
+
if type(case_clause[0]) is tuple:
|
612
|
+
for case in case_clause[0]:
|
613
|
+
if is_equal(case, val_expr):
|
614
|
+
return my_eval(env, case_clause[1])
|
615
|
+
elif type(case_clause[0]) is Sym and case_clause[0].val == "else":
|
616
|
+
return my_eval(env, case_clause[1])
|
617
|
+
else:
|
618
|
+
raise MyError("case: bad syntax")
|
619
|
+
return None
|
620
|
+
|
621
|
+
def syn_let(env: Env, node: Node) -> Any:
|
622
|
+
if len(node) < 2:
|
623
|
+
raise MyError(f"{node[0]}: Expected at least 1 term")
|
624
|
+
|
625
|
+
if type(node[1]) is Sym:
|
626
|
+
raise MyError(f"{node[0]}: Named-let form is not supported")
|
627
|
+
|
628
|
+
for var_ids in node[1]:
|
629
|
+
if type(var_ids) is not tuple or len(var_ids) != 2:
|
630
|
+
raise MyError(f"{node[0]}: Expected two terms: `id` and `val-expr`")
|
631
|
+
|
632
|
+
new_maps: dict[str, Any] = {}
|
633
|
+
for var, val in node[1]:
|
634
|
+
if type(var) is not Sym:
|
635
|
+
raise MyError(f"{node[0]}: Expected symbol for `id` term")
|
636
|
+
new_maps[var.val] = my_eval(env, val)
|
637
|
+
|
638
|
+
inner_env = Env(new_maps, env)
|
639
|
+
for item in node[2:-1]:
|
640
|
+
my_eval(inner_env, item)
|
641
|
+
return my_eval(inner_env, node[-1])
|
642
|
+
|
643
|
+
def syn_let_star(env: Env, node: Node) -> Any:
|
644
|
+
if len(node) < 2:
|
645
|
+
raise MyError(f"{node[0]}: Expected at least 1 term")
|
646
|
+
|
647
|
+
for var_ids in node[1]:
|
648
|
+
if len(var_ids) != 2:
|
649
|
+
raise MyError(f"{node[0]}: Expected two terms: `id` and `val-expr`")
|
650
|
+
|
651
|
+
inner_env = Env({}, env)
|
652
|
+
|
653
|
+
for var, val in node[1]:
|
654
|
+
if type(var) is not Sym:
|
655
|
+
raise MyError(f"{node[0]}: Expected symbol for `id` term")
|
656
|
+
inner_env[var.val] = my_eval(inner_env, val)
|
657
|
+
|
658
|
+
for item in node[2:-1]:
|
659
|
+
my_eval(inner_env, item)
|
660
|
+
return my_eval(inner_env, node[-1])
|
661
|
+
|
662
|
+
def syn_import(env: Env, node: Node) -> None:
|
663
|
+
guard_term(node, 2, 2)
|
664
|
+
|
665
|
+
if type(node[1]) is not Sym:
|
666
|
+
raise MyError("class name must be an identifier")
|
667
|
+
|
668
|
+
module = node[1].val
|
669
|
+
try:
|
670
|
+
obj = __import__(f"auto_editor.lang.lib{module}", fromlist=["lang"])
|
671
|
+
except ImportError:
|
672
|
+
raise MyError(f"No module named `{module}`")
|
673
|
+
|
674
|
+
env.update(obj.all())
|
675
|
+
|
676
|
+
def syn_class(env: Env, node: Node) -> None:
|
677
|
+
if len(node) < 2:
|
678
|
+
raise MyError(f"{node[0]}: Expects at least 1 term")
|
679
|
+
|
680
|
+
if type(node[1]) is not Sym:
|
681
|
+
raise MyError("class name must be an identifier")
|
682
|
+
|
683
|
+
attr_len = len(node) - 2
|
684
|
+
attrs: Any = [None] * (attr_len * 2)
|
685
|
+
contracts = [None] * attr_len
|
686
|
+
|
687
|
+
for i, item in enumerate(node[2:]):
|
688
|
+
if type(item) is not tuple or len(item) != 2:
|
689
|
+
raise MyError(f"{node[0]}: Invalid syntax")
|
690
|
+
|
691
|
+
contracts[i] = my_eval(env, item[1])
|
692
|
+
attrs[i * 2] = item[0].val
|
693
|
+
attrs[i * 2 + 1] = contracts[i]
|
694
|
+
|
695
|
+
name = node[1].val
|
696
|
+
pred = name + "?"
|
697
|
+
attrs = tuple(attrs)
|
698
|
+
|
699
|
+
env[name] = Proc(
|
700
|
+
name,
|
701
|
+
lambda *args: PaletClass(name, attrs, list(args)),
|
702
|
+
(attr_len, attr_len),
|
703
|
+
*contracts,
|
704
|
+
)
|
705
|
+
env[pred] = Proc(
|
706
|
+
pred,
|
707
|
+
lambda v: type(v) is PaletClass and v.name == name and v.attrs == attrs,
|
708
|
+
(1, 1),
|
709
|
+
)
|
710
|
+
|
711
|
+
def attr(env: Env, node: Node) -> Any:
|
712
|
+
guard_term(node, 3, 3)
|
713
|
+
|
714
|
+
if type(node[2]) is not Sym:
|
715
|
+
raise MyError("@r: attribute must be an identifier")
|
716
|
+
|
717
|
+
base = my_eval(env, node[1])
|
718
|
+
if type(base) is PaletClass:
|
719
|
+
if type(name := node[2]) is not Sym:
|
720
|
+
raise MyError("@r: class attribute must be an identifier")
|
721
|
+
|
722
|
+
for i, item in enumerate(base.attrs[0::2]):
|
723
|
+
if name.val == item:
|
724
|
+
return base.values[i]
|
725
|
+
|
726
|
+
return my_eval(env, (node[2], node[1]))
|
727
|
+
|
728
|
+
# Procedure definitions
|
729
|
+
def minclip(oarr: BoolList, _min: int, /) -> BoolList:
|
730
|
+
arr = np.copy(oarr)
|
731
|
+
mut_remove_small(arr, _min, replace=1, with_=0)
|
732
|
+
return arr
|
733
|
+
|
734
|
+
def mincut(oarr: BoolList, _min: int, /) -> BoolList:
|
735
|
+
arr = np.copy(oarr)
|
736
|
+
mut_remove_small(arr, _min, replace=0, with_=1)
|
737
|
+
return arr
|
738
|
+
|
739
|
+
def maxclip(oarr: BoolList, _min: int, /) -> BoolList:
|
740
|
+
arr = np.copy(oarr)
|
741
|
+
mut_remove_large(arr, _min, replace=1, with_=0)
|
742
|
+
return arr
|
743
|
+
|
744
|
+
def maxcut(oarr: BoolList, _min: int, /) -> BoolList:
|
745
|
+
arr = np.copy(oarr)
|
746
|
+
mut_remove_large(arr, _min, replace=0, with_=1)
|
747
|
+
return arr
|
748
|
+
|
749
|
+
def margin(oarr: BoolList, start: int, end: int | None = None, /) -> BoolList:
|
750
|
+
arr = np.copy(oarr)
|
751
|
+
if end is None:
|
752
|
+
mut_margin(arr, start, start)
|
753
|
+
else:
|
754
|
+
mut_margin(arr, start, end)
|
755
|
+
return arr
|
756
|
+
|
757
|
+
def equal_num(*values: object) -> bool:
|
758
|
+
return all(values[0] == val for val in values[1:])
|
759
|
+
|
760
|
+
def minus(*vals: Number) -> Number:
|
761
|
+
if len(vals) == 1:
|
762
|
+
return -vals[0]
|
763
|
+
return reduce(lambda a, b: a - b, vals)
|
764
|
+
|
765
|
+
def num_div(z: Number, *w: Number) -> Number:
|
766
|
+
if len(w) == 0:
|
767
|
+
w = (z,)
|
768
|
+
z = 1
|
769
|
+
|
770
|
+
for num in w:
|
771
|
+
if num == 0:
|
772
|
+
raise MyError("/: division by zero")
|
773
|
+
|
774
|
+
z /= num
|
775
|
+
|
776
|
+
return z
|
777
|
+
|
778
|
+
def int_div(n: int, *m: int) -> int:
|
779
|
+
if 0 in m:
|
780
|
+
raise MyError("div: division by zero")
|
781
|
+
|
782
|
+
return reduce(lambda a, b: a // b, m, n)
|
783
|
+
|
784
|
+
def _sqrt(v: Number) -> Number:
|
785
|
+
r = complex_sqrt(v)
|
786
|
+
if r.imag == 0:
|
787
|
+
if int(r.real) == r.real:
|
788
|
+
return int(r.real)
|
789
|
+
return r.real
|
790
|
+
return r
|
791
|
+
|
792
|
+
def _xor(*vals: Any) -> bool | BoolList:
|
793
|
+
if is_boolarr(vals[0]):
|
794
|
+
check_args("xor", vals, (2, None), (is_boolarr,))
|
795
|
+
return reduce(lambda a, b: boolop(a, b, logical_xor), vals)
|
796
|
+
check_args("xor", vals, (2, None), (is_bool,))
|
797
|
+
return reduce(lambda a, b: a ^ b, vals)
|
798
|
+
|
799
|
+
def string_ref(s: str, ref: int) -> Char:
|
800
|
+
try:
|
801
|
+
return Char(s[ref])
|
802
|
+
except IndexError:
|
803
|
+
raise MyError(f"string index {ref} is out of range")
|
804
|
+
|
805
|
+
def number_to_string(val: Number) -> str:
|
806
|
+
if isinstance(val, complex):
|
807
|
+
join = "" if val.imag < 0 else "+"
|
808
|
+
return f"{val.real}{join}{val.imag}i"
|
809
|
+
return f"{val}"
|
810
|
+
|
811
|
+
def palet_join(v: Any, s: str) -> str:
|
812
|
+
try:
|
813
|
+
return s.join(v)
|
814
|
+
except Exception:
|
815
|
+
raise MyError("join: expected string?")
|
816
|
+
|
817
|
+
def _dtype_to_np(dtype: Sym) -> type[np.generic]:
|
818
|
+
np_dtype = dtype_map.get(dtype)
|
819
|
+
if np_dtype is None:
|
820
|
+
raise MyError(f"Invalid array dtype: {dtype}")
|
821
|
+
return np_dtype
|
822
|
+
|
823
|
+
def array_proc(dtype: Sym, *vals: Any) -> np.ndarray:
|
824
|
+
try:
|
825
|
+
return np.array(vals, dtype=_dtype_to_np(dtype))
|
826
|
+
except OverflowError:
|
827
|
+
raise MyError(f"number too large to be converted to {dtype}")
|
828
|
+
|
829
|
+
def make_array(dtype: Sym, size: int, v: int = 0) -> np.ndarray:
|
830
|
+
try:
|
831
|
+
return np.array([v] * size, dtype=_dtype_to_np(dtype))
|
832
|
+
except OverflowError:
|
833
|
+
raise MyError(f"number too large to be converted to {dtype}")
|
834
|
+
|
835
|
+
def vector_set(vec: list, pos: int, v: Any) -> None:
|
836
|
+
try:
|
837
|
+
vec[pos] = v
|
838
|
+
except IndexError:
|
839
|
+
raise MyError(f"vector-set: Invalid index {pos}")
|
840
|
+
|
841
|
+
def vector_extend(vec: list, *more_vecs: list) -> None:
|
842
|
+
for more in more_vecs:
|
843
|
+
vec.extend(more)
|
844
|
+
|
845
|
+
def list_append(*v: Quoted) -> Quoted:
|
846
|
+
result = Quoted(tuple())
|
847
|
+
for item in v:
|
848
|
+
result.val = result.val + item.val
|
849
|
+
return result
|
850
|
+
|
851
|
+
def palet_map(proc: Proc, seq: Any) -> Any:
|
852
|
+
if type(seq) is str:
|
853
|
+
return str(map(proc, seq))
|
854
|
+
if type(seq) is Quoted:
|
855
|
+
return Quoted(tuple(map(proc, seq.val)))
|
856
|
+
if isinstance(seq, list | range):
|
857
|
+
return list(map(proc, seq))
|
858
|
+
elif isinstance(seq, np.ndarray):
|
859
|
+
vectorized_proc = np.vectorize(proc)
|
860
|
+
return vectorized_proc(seq)
|
861
|
+
return proc(seq)
|
862
|
+
|
863
|
+
def palet_member(v: object, seq: Any) -> bool:
|
864
|
+
if type(v) is Char and type(seq) is str:
|
865
|
+
return v.val in seq
|
866
|
+
return v in seq
|
867
|
+
|
868
|
+
def splice(
|
869
|
+
arr: NDArray, v: int, start: int | None = None, end: int | None = None
|
870
|
+
) -> None:
|
871
|
+
arr[start:end] = v
|
872
|
+
|
873
|
+
def palet_hash(*args: Any) -> dict:
|
874
|
+
result = {}
|
875
|
+
if len(args) % 2 == 1:
|
876
|
+
raise MyError("hash: number of args must be even")
|
877
|
+
for key, item in zip(args[0::2], args[1::2]):
|
878
|
+
result[key] = item
|
879
|
+
return result
|
880
|
+
|
881
|
+
def hash_ref(h: dict, k: object) -> object:
|
882
|
+
try:
|
883
|
+
return h[k]
|
884
|
+
except Exception:
|
885
|
+
raise MyError("hash-ref: invalid key")
|
886
|
+
|
887
|
+
def hash_set(h: dict, k: object, v: object) -> None:
|
888
|
+
h[k] = v
|
889
|
+
|
890
|
+
def hash_remove(h: dict, v: object) -> None:
|
891
|
+
try:
|
892
|
+
del h[v]
|
893
|
+
except Exception:
|
894
|
+
pass
|
895
|
+
|
896
|
+
def palet_assert(expr: object, msg: str | bool = False) -> None:
|
897
|
+
if expr is not True:
|
898
|
+
raise MyError("assert-error" if msg is False else f"assert-error: {msg}")
|
899
|
+
|
900
|
+
def palet_system(cmd: str) -> bool:
|
901
|
+
try:
|
902
|
+
return run(cmd, shell=True).returncode == 0
|
903
|
+
except Exception:
|
904
|
+
return False
|
905
|
+
|
906
|
+
def palet_system_star(*cmd: str) -> bool:
|
907
|
+
try:
|
908
|
+
return run(cmd).returncode == 0
|
909
|
+
except Exception:
|
910
|
+
return False
|
911
|
+
|
912
|
+
def change_file_ext(a, ext) -> str:
|
913
|
+
import os.path
|
914
|
+
|
915
|
+
base_name = os.path.splitext(a)[0]
|
916
|
+
return f"{base_name}.{ext}" if ext else base_name
|
917
|
+
|
918
|
+
|
919
|
+
return {
|
920
|
+
# constants
|
921
|
+
"true": True,
|
922
|
+
"false": False,
|
923
|
+
"all": Sym("all"),
|
924
|
+
# syntax
|
925
|
+
"lambda": Syntax(syn_lambda),
|
926
|
+
"λ": Syntax(syn_lambda),
|
927
|
+
"define": Syntax(syn_define),
|
928
|
+
"define/c": Syntax(syn_definec),
|
929
|
+
"set!": Syntax(syn_set),
|
930
|
+
"incf": Syntax(syn_incf),
|
931
|
+
"decf": Syntax(syn_decf),
|
932
|
+
"&=": Syntax(syn_strappend),
|
933
|
+
"quote": Syntax(syn_quote),
|
934
|
+
"if": Syntax(syn_if),
|
935
|
+
"when": Syntax(syn_when),
|
936
|
+
"cond": Syntax(syn_cond),
|
937
|
+
"case": Syntax(syn_case),
|
938
|
+
"let": Syntax(syn_let),
|
939
|
+
"let*": Syntax(syn_let_star),
|
940
|
+
"import": Syntax(syn_import),
|
941
|
+
"class": Syntax(syn_class),
|
942
|
+
"@r": Syntax(attr),
|
943
|
+
# loops
|
944
|
+
"for": Syntax(syn_for),
|
945
|
+
"for-items": Syntax(syn_for_items),
|
946
|
+
# contracts
|
947
|
+
"number?": is_num,
|
948
|
+
"real?": is_real,
|
949
|
+
"int?": is_int,
|
950
|
+
"float?": is_float,
|
951
|
+
"frac?": is_frac,
|
952
|
+
"complex?": Contract("complex?", lambda v: type(v) is complex),
|
953
|
+
"nat?": is_nat,
|
954
|
+
"nat1?": is_nat1,
|
955
|
+
"threshold?": is_threshold,
|
956
|
+
"any": any_p,
|
957
|
+
"bool?": is_bool,
|
958
|
+
"void?": is_void,
|
959
|
+
"symbol?": (is_symbol := Contract("symbol?", lambda v: type(v) is Sym)),
|
960
|
+
"string?": is_str,
|
961
|
+
"char?": (is_char := Contract("char?", lambda v: type(v) is Char)),
|
962
|
+
"list?": (is_list := Contract("list?", lambda v: type(v) is Quoted or type(v) is tuple)),
|
963
|
+
"vector?": (is_vector := Contract("vector?", lambda v: type(v) is list)),
|
964
|
+
"array?": (is_array := Contract("array?", lambda v: isinstance(v, np.ndarray))),
|
965
|
+
"bool-array?": is_boolarr,
|
966
|
+
"range?": (is_range := Contract("range?", lambda v: type(v) is range)),
|
967
|
+
"iterable?": is_iterable,
|
968
|
+
"sequence?": is_sequence,
|
969
|
+
"procedure?": is_proc,
|
970
|
+
"contract?": is_cont,
|
971
|
+
"hash?": (is_hash := Contract("hash?", lambda v: isinstance(v, dict))),
|
972
|
+
"begin": Proc("begin", lambda *x: x[-1] if x else None, (0, None)),
|
973
|
+
"void": Proc("void", lambda *v: None, (0, 0)),
|
974
|
+
# control / b-arrays
|
975
|
+
"not": Proc("not", lambda v: not v if type(v) is bool else logical_not(v), (1, 1), bool_or_barr),
|
976
|
+
"and": Syntax(syn_and),
|
977
|
+
"or": Syntax(syn_or),
|
978
|
+
"xor": Proc("xor", _xor, (2, None), bool_or_barr),
|
979
|
+
# booleans
|
980
|
+
">": Proc(">", gt, (2, 2), is_real),
|
981
|
+
">=": Proc(">=", ge, (2, 2), is_real),
|
982
|
+
"<": Proc("<", lt, (2, 2), is_real),
|
983
|
+
"<=": Proc("<=", le, (2, 2), is_real),
|
984
|
+
"=": Proc("=", equal_num, (1, None), is_num),
|
985
|
+
"eq?": Proc("eq?", is_, (2, 2)),
|
986
|
+
"equal?": Proc("equal?", is_equal, (2, 2)),
|
987
|
+
"zero?": UserProc(env, "zero?", ["z"], (is_num,), ((Sym("="), Sym("z"), 0),)),
|
988
|
+
"positive?": UserProc(env, "positive?", ["x"], (is_real,), ((Sym(">"), Sym("x"), 0),)),
|
989
|
+
"negative?": UserProc(env, "negative?", ["x"], (is_real,), ((Sym("<"), Sym("x"), 0),)),
|
990
|
+
"even?": UserProc(
|
991
|
+
env, "even?", ["n"], (is_int,), ((Sym("zero?"), (Sym("mod"), Sym("n"), 2)),)),
|
992
|
+
"odd?": UserProc(
|
993
|
+
env, "odd?", ["n"], (is_int,), ((Sym("not"), (Sym("even?"), Sym("n"))),)),
|
994
|
+
">=/c": Proc(">=/c", gte_c, (1, 1), is_real),
|
995
|
+
">/c": Proc(">/c", gt_c, (1, 1), is_real),
|
996
|
+
"<=/c": Proc("<=/c", lte_c, (1, 1), is_real),
|
997
|
+
"</c": Proc("</c", lt_c, (1, 1), is_real),
|
998
|
+
"between/c": Proc("between/c", between_c, (2, 2), is_real),
|
999
|
+
# numbers
|
1000
|
+
"+": Proc("+", lambda *v: sum(v), (0, None), is_num),
|
1001
|
+
"-": Proc("-", minus, (1, None), is_num),
|
1002
|
+
"*": Proc("*", lambda *v: reduce(mul, v, 1), (0, None), is_num),
|
1003
|
+
"/": Proc("/", num_div, (1, None), is_num),
|
1004
|
+
"div": Proc("div", int_div, (2, None), is_int),
|
1005
|
+
"add1": Proc("add1", lambda z: z + 1, (1, 1), is_num),
|
1006
|
+
"sub1": Proc("sub1", lambda z: z - 1, (1, 1), is_num),
|
1007
|
+
"sqrt": Proc("sqrt", _sqrt, (1, 1), is_num),
|
1008
|
+
"real-part": Proc("real-part", lambda v: v.real, (1, 1), is_num),
|
1009
|
+
"imag-part": Proc("imag-part", lambda v: v.imag, (1, 1), is_num),
|
1010
|
+
# reals
|
1011
|
+
"pow": Proc("pow", pow, (2, 2), is_real),
|
1012
|
+
"abs": Proc("abs", abs, (1, 1), is_real),
|
1013
|
+
"round": Proc("round", round, (1, 1), is_real),
|
1014
|
+
"max": Proc("max", lambda *v: max(v), (1, None), is_real),
|
1015
|
+
"min": Proc("min", lambda *v: min(v), (1, None), is_real),
|
1016
|
+
"max-seq": Proc("max-seq", max, (1, 1), is_sequence),
|
1017
|
+
"min-seq": Proc("min-seq", min, (1, 1), is_sequence),
|
1018
|
+
"mod": Proc("mod", mod, (2, 2), is_int),
|
1019
|
+
"modulo": Proc("modulo", mod, (2, 2), is_int),
|
1020
|
+
# symbols
|
1021
|
+
"symbol->string": Proc("symbol->string", str, (1, 1), is_symbol),
|
1022
|
+
"string->symbol": Proc("string->symbol", Sym, (1, 1), is_str),
|
1023
|
+
# strings
|
1024
|
+
"string": Proc("string", lambda *v: reduce(add, v, ""), (0, None), is_char),
|
1025
|
+
"&": Proc("&", lambda *v: reduce(add, v, ""), (0, None), is_str),
|
1026
|
+
"split": Proc("split", str.split, (1, 2), is_str, is_str),
|
1027
|
+
"strip": Proc("strip", str.strip, (1, 1), is_str),
|
1028
|
+
"str-repeat": Proc("str-repeat", mul, (2, 2), is_str, is_int),
|
1029
|
+
"startswith": Proc("startswith", str.startswith, (2, 2), is_str),
|
1030
|
+
"endswith": Proc("endswith", str.endswith, (2, 2), is_str),
|
1031
|
+
"replace": Proc("replace", str.replace, (3, 4), is_str, is_str, is_str, is_int),
|
1032
|
+
"title": Proc("title", str.title, (1, 1), is_str),
|
1033
|
+
"lower": Proc("lower", str.lower, (1, 1), is_str),
|
1034
|
+
"upper": Proc("upper", str.upper, (1, 1), is_str),
|
1035
|
+
"join": Proc("join", palet_join, (2, 2), is_vector, is_str),
|
1036
|
+
"change-file-ext": Proc("change-file-ext", change_file_ext, (2, 2), is_str),
|
1037
|
+
# format
|
1038
|
+
"char->int": Proc("char->int", lambda c: ord(c.val), (1, 1), is_char),
|
1039
|
+
"int->char": Proc("int->char", Char, (1, 1), is_int),
|
1040
|
+
"~a": Proc("~a", lambda *v: "".join([display_str(a) for a in v]), (0, None)),
|
1041
|
+
"~s": Proc("~s", lambda *v: " ".join([display_str(a) for a in v]), (0, None)),
|
1042
|
+
"~v": Proc("~v", lambda *v: " ".join([print_str(a) for a in v]), (0, None)),
|
1043
|
+
# keyword
|
1044
|
+
"keyword?": is_keyw,
|
1045
|
+
"keyword->string": Proc("keyword->string", lambda v: v.val.val, (1, 1), is_keyw),
|
1046
|
+
"string->keyword": Proc("string->keyword", QuotedKeyword, (1, 1), is_str),
|
1047
|
+
# lists
|
1048
|
+
"list": Proc("list", lambda *a: Quoted(a), (0, None)),
|
1049
|
+
"append": Proc("append", list_append, (0, None), is_list),
|
1050
|
+
# vectors
|
1051
|
+
"vector": Proc("vector", lambda *a: list(a), (0, None)),
|
1052
|
+
"make-vector": Proc(
|
1053
|
+
"make-vector", lambda size, a=0: [a] * size, (1, 2), is_nat, any_p
|
1054
|
+
),
|
1055
|
+
"add!": Proc("add!", list.append, (2, 2), is_vector, any_p),
|
1056
|
+
"pop!": Proc("pop!", list.pop, (1, 1), is_vector),
|
1057
|
+
"vec-set!": Proc("vec-set!", vector_set, (3, 3), is_vector, is_int, any_p),
|
1058
|
+
"vec-append": Proc("vec-append", lambda *v: reduce(add, v, []), (0, None), is_vector),
|
1059
|
+
"vec-extend!": Proc("vec-extend!", vector_extend, (2, None), is_vector),
|
1060
|
+
"sort": Proc("sort", sorted, (1, 1), is_vector),
|
1061
|
+
"sort!": Proc("sort!", list.sort, (1, 1), is_vector),
|
1062
|
+
# arrays
|
1063
|
+
"array": Proc("array", array_proc, (2, None), is_symbol, is_real),
|
1064
|
+
"make-array": Proc("make-array", make_array, (2, 3), is_symbol, is_nat, is_real),
|
1065
|
+
"array-splice!": Proc(
|
1066
|
+
"array-splice!", splice, (2, 4), is_array, is_real, is_int, is_int
|
1067
|
+
),
|
1068
|
+
"array-copy": Proc("array-copy", np.copy, (1, 1), is_array),
|
1069
|
+
"count-nonzero": Proc("count-nonzero", np.count_nonzero, (1, 1), is_array),
|
1070
|
+
# bool arrays
|
1071
|
+
"bool-array": Proc(
|
1072
|
+
"bool-array", lambda *a: np.array(a, dtype=np.bool_), (1, None), is_nat
|
1073
|
+
),
|
1074
|
+
"margin": Proc("margin", margin, (2, 3), is_boolarr, is_int),
|
1075
|
+
"mincut": Proc("mincut", mincut, (2, 2), is_boolarr, is_nat),
|
1076
|
+
"minclip": Proc("minclip", minclip, (2, 2), is_boolarr, is_nat),
|
1077
|
+
"maxcut": Proc("maxcut", maxcut, (2, 2), is_boolarr, is_nat),
|
1078
|
+
"maxclip": Proc("maxclip", maxclip, (2, 2), is_boolarr, is_nat),
|
1079
|
+
# ranges
|
1080
|
+
"range": Proc("range", range, (1, 3), is_int, is_int, int_not_zero),
|
1081
|
+
# generic iterables
|
1082
|
+
"len": Proc("len", len, (1, 1), is_iterable),
|
1083
|
+
"member": Proc("member", palet_member, (2, 2), any_p, is_sequence),
|
1084
|
+
"reverse": Proc("reverse", lambda v: v[::-1], (1, 1), is_sequence),
|
1085
|
+
"ref": Proc("ref", ref, (2, 2), is_sequence, is_int),
|
1086
|
+
"slice": Proc("slice", p_slice, (2, 4), is_sequence, is_int),
|
1087
|
+
# procedures
|
1088
|
+
"map": Proc("map", palet_map, (2, 2), is_proc, is_sequence),
|
1089
|
+
"apply": Proc("apply", lambda p, s: p(*s), (2, 2), is_proc, is_sequence),
|
1090
|
+
"and/c": Proc("and/c", andc, (1, None), is_cont),
|
1091
|
+
"or/c": Proc("or/c", orc, (1, None), is_cont),
|
1092
|
+
"not/c": Proc("not/c", notc, (1, 1), is_cont),
|
1093
|
+
# hashs
|
1094
|
+
"hash": Proc("hash", palet_hash, (0, None)),
|
1095
|
+
"hash-ref": Proc("hash", hash_ref, (2, 2), is_hash, any_p),
|
1096
|
+
"hash-set!": Proc("hash-set!", hash_set, (3, 3), is_hash, any_p, any_p),
|
1097
|
+
"has-key?": Proc("has-key?", lambda h, k: k in h, (2, 2), is_hash, any_p),
|
1098
|
+
"hash-remove!": Proc("hash-remove!", hash_remove, (2, 2), is_hash, any_p),
|
1099
|
+
"hash-update!": UserProc(env, "hash-update!", ["h", "v", "up"], (is_hash, any_p),
|
1100
|
+
(
|
1101
|
+
(Sym("hash-set!"), Sym("h"), Sym("v"), (Sym("up"), (Sym("h"), Sym("v"))),),
|
1102
|
+
),
|
1103
|
+
),
|
1104
|
+
# i/o
|
1105
|
+
"open-output-file": Proc("open-output-file", initOutPort, (1, 1), is_str),
|
1106
|
+
"output-port?": (op := Contract("output-port?", lambda v: type(v) is OutputPort)),
|
1107
|
+
"close-port": Proc("close-port", OutputPort.close, (1, 1), op),
|
1108
|
+
"closed?": Proc("closed?", lambda o: o.closed, (1, 1), op),
|
1109
|
+
# printing
|
1110
|
+
"display": Proc("display",
|
1111
|
+
lambda v, f=None: print(display_str(v), end="", file=f), (1, 2), any_p, op),
|
1112
|
+
"displayln": Proc("displayln",
|
1113
|
+
lambda v, f=None: print(display_str(v), file=f), (1, 2), any_p, op),
|
1114
|
+
"print": Proc("print",
|
1115
|
+
lambda v, f=None: print(print_str(v), end="", file=f), (1, 2), any_p, op),
|
1116
|
+
"println": Proc("println",
|
1117
|
+
lambda v, f=None: print(print_str(v), file=f), (1, 2), any_p, op),
|
1118
|
+
# actions
|
1119
|
+
"assert": Proc("assert", palet_assert, (1, 2), any_p, orc(is_str, False)),
|
1120
|
+
"error": Proc("error", raise_, (1, 1), is_str),
|
1121
|
+
"sleep": Proc("sleep", sleep, (1, 1), is_int_or_float),
|
1122
|
+
"system": Proc("system", palet_system, (1, 1), is_str),
|
1123
|
+
"system*": Proc("system*", palet_system_star, (1, None), is_str),
|
1124
|
+
# conversions
|
1125
|
+
"number->string": Proc("number->string", number_to_string, (1, 1), is_num),
|
1126
|
+
"string->vector": Proc(
|
1127
|
+
"string->vector", lambda s: [Char(c) for c in s], (1, 1), is_str
|
1128
|
+
),
|
1129
|
+
"range->vector": Proc("range->vector", list, (1, 1), is_range),
|
1130
|
+
# reflexion
|
1131
|
+
"var-exists?": Proc("var-exists?", lambda sym: sym.val in env, (1, 1), is_symbol),
|
1132
|
+
"rename": Syntax(syn_rename),
|
1133
|
+
"delete": Syntax(syn_delete),
|
1134
|
+
"eval": Proc("eval",
|
1135
|
+
lambda tl: my_eval(env, tl.val) if type(tl) is Quoted
|
1136
|
+
else (my_eval(env, tl) if type(tl) is Sym else tl),
|
1137
|
+
(1, 1)
|
1138
|
+
),
|
1139
|
+
}
|