auto-editor 23.38.1__py3-none-any.whl → 23.40.1__py3-none-any.whl

auto_editor/lang/palet.py

@@ -27,7 +27,7 @@ from auto_editor.utils.func import boolop, mut_margin
 
 if TYPE_CHECKING:
     from collections.abc import Callable
-    from typing import Any, NoReturn
+    from typing import Any, Literal, NoReturn
 
     from numpy.typing import NDArray
 
@@ -49,7 +49,7 @@ class ClosingError(MyError):
 LPAREN, RPAREN, LBRAC, RBRAC, LCUR, RCUR, EOF = "(", ")", "[", "]", "{", "}", "EOF"
 VAL, QUOTE, SEC, DB, DOT, VLIT = "VAL", "QUOTE", "SEC", "DB", "DOT", "VLIT"
 SEC_UNITS = ("s", "sec", "secs", "second", "seconds")
-METHODS = ("audio:", "motion:", "pixeldiff:", "subtitle:", "none:", "all/e:")
+METHODS = ("audio:", "motion:", "pixeldiff:", "subtitle:")
 brac_pairs = {LPAREN: RPAREN, LBRAC: RBRAC, LCUR: RCUR}
 
 str_escape = {
@@ -458,43 +458,15 @@ class Parser:
 ###############################################################################
 
 
-def check_args(
-    o: str,
-    values: list | tuple,
-    arity: tuple[int, int | None],
-    cont: list[Contract] | None,
-) -> None:
-    lower, upper = arity
-    amount = len(values)
-
-    assert not (upper is not None and lower > upper)
-    base = f"`{o}` has an arity mismatch. Expected "
-
-    if lower == upper and len(values) != lower:
-        raise MyError(f"{base}{lower}, got {amount}")
-    if upper is None and amount < lower:
-        raise MyError(f"{base}at least {lower}, got {amount}")
-    if upper is not None and (amount > upper or amount < lower):
-        raise MyError(f"{base}between {lower} and {upper}, got {amount}")
-
-    if cont is None:
-        return
-
-    for i, val in enumerate(values):
-        check = cont[-1] if i >= len(cont) else cont[i]
-        if not check_contract(check, val):
-            exp = f"{check}" if callable(check) else print_str(check)
-            raise MyError(f"`{o}` expected a {exp}, got {print_str(val)}")
-
-
 is_cont = Contract("contract?", is_contract)
 is_iterable = Contract(
     "iterable?",
-    lambda v: type(v) in (str, range) or isinstance(v, (list, dict, np.ndarray)),
+    lambda v: type(v) in (str, range, Quoted)
+    or isinstance(v, (list, dict, np.ndarray)),
 )
 is_sequence = Contract(
     "sequence?",
-    lambda v: type(v) in (str, range) or isinstance(v, (list, np.ndarray)),
+    lambda v: type(v) in (str, range, Quoted) or isinstance(v, (list, np.ndarray)),
 )
 is_boolarr = Contract(
     "bool-array?",
@@ -504,9 +476,32 @@ bool_or_barr = Contract(
     "(or/c bool? bool-array?)",
     lambda v: type(v) is bool or is_boolarr(v),
 )
-is_keyw = Contract(
-    "keyword?", lambda v: type(v) is list and len(v) == 2 and type(v[1]) is Keyword
-)
+is_keyw = Contract("keyword?", lambda v: type(v) is QuotedKeyword)
+
+
+@dataclass(slots=True)
+class OutputPort:
+    name: str
+    port: Any
+    write: Any
+    closed: bool
+
+    def close(self) -> None:
+        if not self.closed:
+            self.port.close()
+
+    def __str__(self) -> str:
+        return f"#<output-port:{self.name}>"
+
+    __repr__ = __str__
+
+
+def initOutPort(name: str) -> OutputPort | Literal[False]:
+    try:
+        port = open(name, "w", encoding="utf-8")
+    except Exception:
+        return False
+    return OutputPort(name, port, port.write, False)
 
 
 def raise_(msg: str) -> None:
@@ -565,9 +560,9 @@ def _sqrt(v: Number) -> Number:
 
 def _xor(*vals: Any) -> bool | BoolList:
     if is_boolarr(vals[0]):
-        check_args("xor", vals, (2, None), [is_boolarr])
+        check_args("xor", vals, (2, None), (is_boolarr,))
         return reduce(lambda a, b: boolop(a, b, logical_xor), vals)
-    check_args("xor", vals, (2, None), [is_bool])
+    check_args("xor", vals, (2, None), (is_bool,))
     return reduce(lambda a, b: a ^ b, vals)
 
 
@@ -593,6 +588,13 @@ def number_to_string(val: Number) -> str:
     return f"{val}"
 
 
+def palet_join(v: Any, s: str) -> str:
+    try:
+        return s.join(v)
+    except Exception:
+        raise MyError("join: expected string?")
+
+
 dtype_map = {
     Sym("bool"): np.bool_,
     Sym("int8"): np.int8,
@@ -655,11 +657,11 @@ def maxcut(oarr: BoolList, _min: int) -> BoolList:
 
 def margin(a: int, b: Any, c: Any = None) -> BoolList:
     if c is None:
-        check_args("margin", [a, b], (2, 2), [is_int, is_boolarr])
+        check_args("margin", [a, b], (2, 2), (is_int, is_boolarr))
         oarr = b
         start, end = a, a
     else:
-        check_args("margin", [a, b, c], (3, 3), [is_int, is_int, is_boolarr])
+        check_args("margin", [a, b, c], (3, 3), (is_int, is_int, is_boolarr))
         oarr = c
         start, end = a, b
 
@@ -680,21 +682,14 @@ def vector_extend(vec: list, *more_vecs: list) -> None:
         vec.extend(more)
 
 
-def palet_map(proc: Proc, seq: str | list | range | NDArray) -> Any:
+def palet_map(proc: Proc, seq: Any) -> Any:
     if type(seq) is str:
-        return str(map(proc.proc, seq))
+        return str(map(proc, seq))
+    if type(seq) is Quoted:
+        return Quoted(list(map(proc, seq.val)))
     if isinstance(seq, (list, range)):
-        return list(map(proc.proc, seq))
-
-    if isinstance(seq, np.ndarray):
-        if proc.arity[0] != 0:
-            raise MyError("map: procedure must take at least one arg")
-        check_args(proc.name, [0], (1, 1), None)
-        return proc.proc(seq)
-
-
-def apply(proc: Proc, seq: str | list | range) -> Any:
-    return reduce(proc.proc, seq)
+        return list(map(proc, seq))
+    return proc(seq)
 
 
 def ref(seq: Any, ref: int) -> Any:
@@ -784,8 +779,8 @@ class UserProc(Proc):
         env: Env,
         name: str,
         parms: list[str],
+        contracts: tuple[Any, ...],
         body: list,
-        contracts: list[Any] | None = None,
     ):
         self.env = env
         self.name = name
@@ -800,6 +795,8 @@ class UserProc(Proc):
             self.arity = len(parms), len(parms)
 
     def __call__(self, *args: Any) -> Any:
+        check_args(self.name, args, self.arity, self.contracts)
+
         if self.arity[1] is None:
             args = tuple(
                 list(args[: len(self.parms) - 1]) + [list(args[len(self.parms) - 1 :])]
@@ -923,7 +920,7 @@ def syn_lambda(env: Env, node: list) -> UserProc:
 
         parms.append(f"{item}")
 
-    return UserProc(env, "", parms, node[2:])
+    return UserProc(env, "", parms, (), node[2:])
 
 
 def syn_define(env: Env, node: list) -> None:
@@ -947,7 +944,7 @@ def syn_define(env: Env, node: list) -> None:
                 if type(item) is Sym:
                     raise MyError(f"{node[0]}: {item} must be a keyword")
                 if type(item) is not Keyword:
-                    raise MyError(f"{node[0]}: must be an identifier or keyword")
+                    raise MyError(f"{node[0]}: must be a keyword")
                 kparms.append(item.val)
             else:
                 if type(item) is Keyword:
@@ -961,7 +958,7 @@ def syn_define(env: Env, node: list) -> None:
         if kw_only:
             env[n] = KeywordProc(env, n, parms, kparms, body, (len(parms), None))
         else:
-            env[n] = UserProc(env, n, parms, body)
+            env[n] = UserProc(env, n, parms, (), body)
         return None
 
     elif type(node[1]) is not Sym:
@@ -988,7 +985,7 @@ def syn_define(env: Env, node: list) -> None:
 
             parms.append(f"{item}")
 
-        env[n] = UserProc(env, n, parms, body)
+        env[n] = UserProc(env, n, parms, (), body)
 
     else:
         for item in node[2:-1]:
@@ -1008,7 +1005,7 @@ def syn_definec(env: Env, node: list) -> None:
 
     n = node[1][0].val
 
-    contracts: list[Proc | Contract] = []
+    contracts: list[Any] = []
     parms: list[str] = []
     for item in node[1][1:]:
         if item == Sym("->"):
@@ -1025,7 +1022,7 @@ def syn_definec(env: Env, node: list) -> None:
         parms.append(f"{item[0]}")
         contracts.append(con)
 
-    env[n] = UserProc(env, n, parms, node[2:], contracts)
+    env[n] = UserProc(env, n, parms, tuple(contracts), node[2:])
     return None
 
 
@@ -1126,10 +1123,12 @@ def syn_for(env: Env, node: list) -> None:
                 my_eval(env, c)
 
 
-def syn_quote(env: Env, node: list) -> list:
+def syn_quote(env: Env, node: list) -> Any:
     guard_term(node, 2, 2)
-    if type(node[1]) is list or type(node[1]) is Keyword:
-        return [list, node[1]]
+    if type(node[1]) is Keyword:
+        return QuotedKeyword(node[1])
+    if type(node[1]) is list:
+        return Quoted(node[1])
     return node[1]
 
 
@@ -1180,7 +1179,7 @@ def syn_and(env: Env, node: list) -> Any:
 
     if is_boolarr(first):
         vals = [first] + [my_eval(env, n) for n in node[2:]]
-        check_args(node[0], vals, (2, None), [is_boolarr])
+        check_args(node[0], vals, (2, None), (is_boolarr,))
         return reduce(lambda a, b: boolop(a, b, logical_and), vals)
 
     raise MyError(f"{node[0]} expects (or/c bool? bool-array?)")
@@ -1204,7 +1203,7 @@ def syn_or(env: Env, node: list) -> Any:
 
     if is_boolarr(first):
         vals = [first] + [my_eval(env, n) for n in node[2:]]
-        check_args(node[0], vals, (2, None), [is_boolarr])
+        check_args(node[0], vals, (2, None), (is_boolarr,))
         return reduce(lambda a, b: boolop(a, b, logical_or), vals)
 
     raise MyError(f"{node[0]} expects (or/c bool? bool-array?)")
@@ -1329,12 +1328,26 @@ def syn_class(env: Env, node: list) -> Any:
 def attr(env: Env, node: list) -> Any:
     guard_term(node, 3, 3)
 
-    if not isinstance(node[2], Sym):
+    if type(node[2]) is not Sym:
         raise MyError("@r: attribute must be an identifier")
 
     return my_eval(env, [node[2], node[1]])
 
 
+def edit_none() -> np.ndarray:
+    if "@levels" not in env:
+        raise MyError("Can't use `none` if there's no input media")
+
+    return env["@levels"].none()
+
+
+def edit_all() -> np.ndarray:
+    if "@levels" not in env:
+        raise MyError("Can't use `all/e` if there's no input media")
+
+    return env["@levels"].all()
+
+
 def my_eval(env: Env, node: object) -> Any:
     if type(node) is Sym:
         val = env.get(node.val)
@@ -1353,10 +1366,9 @@ def my_eval(env: Env, node: object) -> Any:
             raise MyError("Can't use edit methods if there's no input files")
         return edit_method(node.val, env["@filesetup"], env)
 
-    if isinstance(node, list):
+    if type(node) is list:
         if not node:
             raise MyError("Illegal () expression")
-
         if node[0] is list:  # Handle vector literal
             return [my_eval(env, item) for item in node[1]]
 
@@ -1382,13 +1394,7 @@ def my_eval(env: Env, node: object) -> Any:
         if type(oper) is Syntax:
             return oper(env, node)
 
-        values = [my_eval(env, c) for c in node[1:]]
-        if type(oper) is Contract:
-            check_args(oper.name, values, (1, 1), None)
-        else:
-            check_args(oper.name, values, oper.arity, oper.contracts)
-
-        return oper(*values)
+        return oper(*(my_eval(env, c) for c in node[1:]))
 
     return node
 
@@ -1400,6 +1406,9 @@ env.update({
     "true": True,
     "false": False,
     "all": Sym("all"),
+    # edit procedures
+    "none": Proc("none", edit_none, (0, 0)),
+    "all/e": Proc("all/e", edit_all, (0, 0)),
     # syntax
     "lambda": Syntax(syn_lambda),
     "λ": Syntax(syn_lambda),
@@ -1424,10 +1433,11 @@ env.update({
     "number?": is_num,
     "real?": is_real,
     "int?": is_int,
-    "uint?": is_uint,
-    "nat?": is_nat,
     "float?": is_float,
     "frac?": is_frac,
+    "complex?": Contract("complex?", lambda v: type(v) is complex),
+    "nat?": is_nat,
+    "nat1?": is_nat1,
     "threshold?": is_threshold,
     "any": any_p,
     "bool?": is_bool,
@@ -1435,7 +1445,7 @@ env.update({
     "symbol?": (is_symbol := Contract("symbol?", lambda v: type(v) is Sym)),
     "string?": is_str,
     "char?": (is_char := Contract("char?", lambda v: type(v) is Char)),
-    "vector?": (is_vector := Contract("vector?", lambda v: isinstance(v, list))),
+    "vector?": (is_vector := Contract("vector?", lambda v: type(v) is list)),
     "array?": (is_array := Contract("array?", lambda v: isinstance(v, np.ndarray))),
     "bool-array?": is_boolarr,
     "range?": (is_range := Contract("range?", lambda v: type(v) is range)),
@@ -1447,156 +1457,162 @@ env.update({
     "begin": Proc("begin", lambda *x: x[-1] if x else None, (0, None)),
     "void": Proc("void", lambda *v: None, (0, 0)),
     # control / b-arrays
-    "not": Proc("not", lambda v: not v if type(v) is bool else logical_not(v), (1, 1), [bool_or_barr]),
+    "not": Proc("not", lambda v: not v if type(v) is bool else logical_not(v), (1, 1), bool_or_barr),
     "and": Syntax(syn_and),
     "or": Syntax(syn_or),
-    "xor": Proc("xor", _xor, (2, None), [bool_or_barr]),
+    "xor": Proc("xor", _xor, (2, None), bool_or_barr),
     # booleans
-    ">": Proc(">", lambda a, b: a > b, (2, 2), [is_real, is_real]),
-    ">=": Proc(">=", lambda a, b: a >= b, (2, 2), [is_real, is_real]),
-    "<": Proc("<", lambda a, b: a < b, (2, 2), [is_real, is_real]),
-    "<=": Proc("<=", lambda a, b: a <= b, (2, 2), [is_real, is_real]),
-    "=": Proc("=", equal_num, (1, None), [is_num]),
+    ">": Proc(">", lambda a, b: a > b, (2, 2), is_real),
+    ">=": Proc(">=", lambda a, b: a >= b, (2, 2), is_real),
+    "<": Proc("<", lambda a, b: a < b, (2, 2), is_real),
+    "<=": Proc("<=", lambda a, b: a <= b, (2, 2), is_real),
+    "=": Proc("=", equal_num, (1, None), is_num),
     "eq?": Proc("eq?", lambda a, b: a is b, (2, 2)),
     "equal?": Proc("equal?", is_equal, (2, 2)),
-    "zero?": UserProc(env, "zero?", ["z"], [[Sym("="), Sym("z"), 0]], [is_num]),
-    "positive?": UserProc(
-        env, "positive?", ["x"], [[Sym(">"), Sym("x"), 0]], [is_real]
-    ),
-    "negative?": UserProc(
-        env, "negative?", ["x"], [[Sym("<"), Sym("x"), 0]], [is_real]
-    ),
+    "zero?": UserProc(env, "zero?", ["z"], (is_num,), [[Sym("="), Sym("z"), 0]]),
+    "positive?": UserProc(env, "positive?", ["x"], (is_real,), [[Sym(">"), Sym("x"), 0]]),
+    "negative?": UserProc(env, "negative?", ["x"], (is_real,), [[Sym("<"), Sym("x"), 0]]),
     "even?": UserProc(
-        env, "even?", ["n"], [[Sym("zero?"), [Sym("mod"), Sym("n"), 2]]], [is_int]
+        env, "even?", ["n"], (is_int,), [[Sym("zero?"), [Sym("mod"), Sym("n"), 2]]]
     ),
     "odd?": UserProc(
-        env, "odd?", ["n"], [[Sym("not"), [Sym("even?"), Sym("n")]]], [is_int]
+        env, "odd?", ["n"], (is_int,), [[Sym("not"), [Sym("even?"), Sym("n")]]]
     ),
-    ">=/c": Proc(">=/c", gte_c, (1, 1), [is_real]),
-    ">/c": Proc(">/c", gt_c, (1, 1), [is_real]),
-    "<=/c": Proc("<=/c", lte_c, (1, 1), [is_real]),
-    "</c": Proc("</c", lt_c, (1, 1), [is_real]),
-    "between/c": Proc("between/c", between_c, (2, 2), [is_real, is_real]),
+    ">=/c": Proc(">=/c", gte_c, (1, 1), is_real),
+    ">/c": Proc(">/c", gt_c, (1, 1), is_real),
+    "<=/c": Proc("<=/c", lte_c, (1, 1), is_real),
+    "</c": Proc("</c", lt_c, (1, 1), is_real),
+    "between/c": Proc("between/c", between_c, (2, 2), is_real),
     # numbers
-    "+": Proc("+", lambda *v: sum(v), (0, None), [is_num]),
-    "-": Proc("-", minus, (1, None), [is_num]),
-    "*": Proc("*", mul, (0, None), [is_num]),
-    "/": Proc("/", num_div, (1, None), [is_num]),
-    "div": Proc("div", int_div, (2, None), [is_int]),
-    "add1": Proc("add1", lambda z: z + 1, (1, 1), [is_num]),
-    "sub1": Proc("sub1", lambda z: z - 1, (1, 1), [is_num]),
-    "sqrt": Proc("sqrt", _sqrt, (1, 1), [is_num]),
-    "real-part": Proc("real-part", lambda v: v.real, (1, 1), [is_num]),
-    "imag-part": Proc("imag-part", lambda v: v.imag, (1, 1), [is_num]),
+    "+": Proc("+", lambda *v: sum(v), (0, None), is_num),
+    "-": Proc("-", minus, (1, None), is_num),
+    "*": Proc("*", mul, (0, None), is_num),
+    "/": Proc("/", num_div, (1, None), is_num),
+    "div": Proc("div", int_div, (2, None), is_int),
+    "add1": Proc("add1", lambda z: z + 1, (1, 1), is_num),
+    "sub1": Proc("sub1", lambda z: z - 1, (1, 1), is_num),
+    "sqrt": Proc("sqrt", _sqrt, (1, 1), is_num),
+    "real-part": Proc("real-part", lambda v: v.real, (1, 1), is_num),
+    "imag-part": Proc("imag-part", lambda v: v.imag, (1, 1), is_num),
     # reals
-    "pow": Proc("pow", pow, (2, 2), [is_real]),
-    "exp": Proc("exp", math.exp, (1, 1), [is_real]),
-    "abs": Proc("abs", abs, (1, 1), [is_real]),
-    "ceil": Proc("ceil", math.ceil, (1, 1), [is_real]),
-    "floor": Proc("floor", math.floor, (1, 1), [is_real]),
-    "round": Proc("round", round, (1, 1), [is_real]),
-    "max": Proc("max", lambda *v: max(v), (1, None), [is_real]),
-    "min": Proc("min", lambda *v: min(v), (1, None), [is_real]),
-    "sin": Proc("sin", math.sin, (1, 1), [is_real]),
-    "cos": Proc("cos", math.cos, (1, 1), [is_real]),
-    "log": Proc("log", math.log, (1, 2), [is_real, is_real]),
-    "tan": Proc("tan", math.tan, (1, 1), [is_real]),
-    "mod": Proc("mod", lambda a, b: a % b, (2, 2), [is_int]),
-    "modulo": Proc("modulo", lambda a, b: a % b, (2, 2), [is_int]),
+    "pow": Proc("pow", pow, (2, 2), is_real),
+    "exp": Proc("exp", math.exp, (1, 1), is_real),
+    "abs": Proc("abs", abs, (1, 1), is_real),
+    "ceil": Proc("ceil", math.ceil, (1, 1), is_real),
+    "floor": Proc("floor", math.floor, (1, 1), is_real),
+    "round": Proc("round", round, (1, 1), is_real),
+    "max": Proc("max", lambda *v: max(v), (1, None), is_real),
+    "min": Proc("min", lambda *v: min(v), (1, None), is_real),
+    "sin": Proc("sin", math.sin, (1, 1), is_real),
+    "cos": Proc("cos", math.cos, (1, 1), is_real),
+    "log": Proc("log", math.log, (1, 2), is_real),
+    "tan": Proc("tan", math.tan, (1, 1), is_real),
+    "mod": Proc("mod", lambda a, b: a % b, (2, 2), is_int),
+    "modulo": Proc("modulo", lambda a, b: a % b, (2, 2), is_int),
     # symbols
-    "symbol->string": Proc("symbol->string", str, (1, 1), [is_symbol]),
-    "string->symbol": Proc("string->symbol", Sym, (1, 1), [is_str]),
+    "symbol->string": Proc("symbol->string", str, (1, 1), is_symbol),
+    "string->symbol": Proc("string->symbol", Sym, (1, 1), is_str),
     # strings
-    "string": Proc("string", string_append, (0, None), [is_char]),
-    "&": Proc("&", string_append, (0, None), [is_str]),
-    "split": Proc("split", str.split, (1, 2), [is_str, is_str]),
-    "strip": Proc("strip", str.strip, (1, 1), [is_str]),
-    "str-repeat": Proc("str-repeat", lambda s, a: s * a, (2, 2), [is_str, is_int]),
-    "startswith": Proc("startswith", str.startswith, (2, 2), [is_str, is_str]),
-    "endswith": Proc("endswith", str.endswith, (2, 2), [is_str, is_str]),
-    "replace": Proc("replace", str.replace, (3, 4), [is_str, is_str, is_str, is_int]),
-    "title": Proc("title", str.title, (1, 1), [is_str]),
-    "lower": Proc("lower", str.lower, (1, 1), [is_str]),
-    "upper": Proc("upper", str.upper, (1, 1), [is_str]),
+    "string": Proc("string", string_append, (0, None), is_char),
+    "&": Proc("&", string_append, (0, None), is_str),
+    "split": Proc("split", str.split, (1, 2), is_str, is_str),
+    "strip": Proc("strip", str.strip, (1, 1), is_str),
+    "str-repeat": Proc("str-repeat", lambda s, a: s * a, (2, 2), is_str, is_int),
+    "startswith": Proc("startswith", str.startswith, (2, 2), is_str),
+    "endswith": Proc("endswith", str.endswith, (2, 2), is_str),
+    "replace": Proc("replace", str.replace, (3, 4), is_str, is_str, is_str, is_int),
+    "title": Proc("title", str.title, (1, 1), is_str),
+    "lower": Proc("lower", str.lower, (1, 1), is_str),
+    "upper": Proc("upper", str.upper, (1, 1), is_str),
+    "join": Proc("join", palet_join, (2, 2), is_vector, is_str),
     # format
-    "char->int": Proc("char->int", lambda c: ord(c.val), (1, 1), [is_char]),
-    "int->char": Proc("int->char", Char, (1, 1), [is_int]),
+    "char->int": Proc("char->int", lambda c: ord(c.val), (1, 1), is_char),
+    "int->char": Proc("int->char", Char, (1, 1), is_int),
     "~a": Proc("~a", lambda *v: "".join([display_str(a) for a in v]), (0, None)),
     "~s": Proc("~s", lambda *v: " ".join([display_str(a) for a in v]), (0, None)),
     "~v": Proc("~v", lambda *v: " ".join([print_str(a) for a in v]), (0, None)),
     # keyword
     "keyword?": is_keyw,
-    "keyword->string": Proc("keyword->string", lambda k: k[1].val, (1, 1), [is_keyw]),
-    "string->keyword": Proc("string->keyword", lambda s: [list, Keyword(s)], (1, 1), [is_str]),
+    "keyword->string": Proc("keyword->string", lambda v: v.val.val, (1, 1), is_keyw),
+    "string->keyword": Proc("string->keyword", QuotedKeyword, (1, 1), is_str),
     # vectors
     "vector": Proc("vector", lambda *a: list(a), (0, None)),
     "make-vector": Proc(
-        "make-vector", lambda size, a=0: [a] * size, (1, 2), [is_uint, any_p]
+        "make-vector", lambda size, a=0: [a] * size, (1, 2), is_nat, any_p
     ),
-    "vector-append": Proc("vector-append", vector_append, (0, None), [is_vector]),
-    "vector-pop!": Proc("vector-pop!", list.pop, (1, 1), [is_vector]),
-    "vector-add!": Proc("vector-add!", list.append, (2, 2), [is_vector, any_p]),
-    "vector-set!": Proc("vector-set!", vector_set, (3, 3), [is_vector, is_int, any_p]),
-    "vector-extend!": Proc("vector-extend!", vector_extend, (2, None), [is_vector]),
-    "sort": Proc("sort", sorted, (1, 1), [is_vector]),
-    "sort!": Proc("sort!", list.sort, (1, 1), [is_vector]),
+    "add!": Proc("add!", list.append, (2, 2), is_vector, any_p),
+    "pop!": Proc("pop!", list.pop, (1, 1), is_vector),
+    "vec-set!": Proc("vec-set!", vector_set, (3, 3), is_vector, is_int, any_p),
+    "vec-append": Proc("vec-append", vector_append, (0, None), is_vector),
+    "vec-extend!": Proc("vec-extend!", vector_extend, (2, None), is_vector),
+    "sort": Proc("sort", sorted, (1, 1), is_vector),
+    "sort!": Proc("sort!", list.sort, (1, 1), is_vector),
     # arrays
-    "array": Proc("array", array_proc, (2, None), [is_symbol, is_real]),
-    "make-array": Proc("make-array", make_array, (2, 3), [is_symbol, is_uint, is_real]),
+    "array": Proc("array", array_proc, (2, None), is_symbol, is_real),
+    "make-array": Proc("make-array", make_array, (2, 3), is_symbol, is_nat, is_real),
     "array-splice!": Proc(
-        "array-splice!", splice, (2, 4), [is_array, is_real, is_int, is_int]
+        "array-splice!", splice, (2, 4), is_array, is_real, is_int, is_int
     ),
-    "array-copy": Proc("array-copy", np.copy, (1, 1), [is_array]),
-    "count-nonzero": Proc("count-nonzero", np.count_nonzero, (1, 1), [is_array]),
+    "array-copy": Proc("array-copy", np.copy, (1, 1), is_array),
+    "count-nonzero": Proc("count-nonzero", np.count_nonzero, (1, 1), is_array),
     # bool arrays
     "bool-array": Proc(
-        "bool-array", lambda *a: np.array(a, dtype=np.bool_), (1, None), [is_uint]
+        "bool-array", lambda *a: np.array(a, dtype=np.bool_), (1, None), is_nat
     ),
-    "margin": Proc("margin", margin, (2, 3), None),
-    "mincut": Proc("mincut", mincut, (2, 2), [is_boolarr, is_uint]),
-    "minclip": Proc("minclip", minclip, (2, 2), [is_boolarr, is_uint]),
-    "maxcut": Proc("maxcut", maxcut, (2, 2), [is_boolarr, is_uint]),
-    "maxclip": Proc("maxclip", maxclip, (2, 2), [is_boolarr, is_uint]),
+    "margin": Proc("margin", margin, (2, 3)),
+    "mincut": Proc("mincut", mincut, (2, 2), is_boolarr, is_nat),
+    "minclip": Proc("minclip", minclip, (2, 2), is_boolarr, is_nat),
+    "maxcut": Proc("maxcut", maxcut, (2, 2), is_boolarr, is_nat),
+    "maxclip": Proc("maxclip", maxclip, (2, 2), is_boolarr, is_nat),
     # ranges
-    "range": Proc("range", range, (1, 3), [is_int, is_int, int_not_zero]),
+    "range": Proc("range", range, (1, 3), is_int, is_int, int_not_zero),
     # generic iterables
-    "len": Proc("len", len, (1, 1), [is_iterable]),
-    "reverse": Proc("reverse", lambda v: v[::-1], (1, 1), [is_sequence]),
-    "ref": Proc("ref", ref, (2, 2), [is_sequence, is_int]),
-    "slice": Proc("slice", p_slice, (2, 4), [is_sequence, is_int]),
+    "len": Proc("len", len, (1, 1), is_iterable),
+    "reverse": Proc("reverse", lambda v: v[::-1], (1, 1), is_sequence),
+    "ref": Proc("ref", ref, (2, 2), is_sequence, is_int),
+    "slice": Proc("slice", p_slice, (2, 4), is_sequence, is_int),
     # procedures
-    "map": Proc("map", palet_map, (2, 2), [is_proc, is_sequence]),
-    "apply": Proc("apply", apply, (2, 2), [is_proc, is_sequence]),
-    "and/c": Proc("and/c", andc, (1, None), [is_cont]),
-    "or/c": Proc("or/c", orc, (1, None), [is_cont]),
-    "not/c": Proc("not/c", notc, (1, 1), [is_cont]),
+    "map": Proc("map", palet_map, (2, 2), is_proc, is_sequence),
+    "apply": Proc("apply", lambda p, s: p(*s), (2, 2), is_proc, is_sequence),
+    "and/c": Proc("and/c", andc, (1, None), is_cont),
+    "or/c": Proc("or/c", orc, (1, None), is_cont),
+    "not/c": Proc("not/c", notc, (1, 1), is_cont),
     # hashs
     "hash": Proc("hash", palet_hash, (0, None)),
-    "hash-ref": Proc("hash", hash_ref, (2, 2), [is_hash, any_p]),
-    "hash-set!": Proc("hash-set!", hash_set, (3, 3), [is_hash, any_p, any_p]),
-    "has-key?": Proc("has-key?", lambda h, k: k in h, (2, 2), [is_hash, any_p]),
-    "hash-remove!": Proc("hash-remove!", hash_remove, (2, 2), [is_hash, any_p]),
-    "hash-update!": UserProc(env, "hash-update!", ["h", "v", "up"],
+    "hash-ref": Proc("hash", hash_ref, (2, 2), is_hash, any_p),
+    "hash-set!": Proc("hash-set!", hash_set, (3, 3), is_hash, any_p, any_p),
+    "has-key?": Proc("has-key?", lambda h, k: k in h, (2, 2), is_hash, any_p),
+    "hash-remove!": Proc("hash-remove!", hash_remove, (2, 2), is_hash, any_p),
+    "hash-update!": UserProc(env, "hash-update!", ["h", "v", "up"], (is_hash, any_p),
         [[Sym("hash-set!"), Sym("h"), Sym("v"), [Sym("up"), [Sym("hash-ref"), Sym("h"), Sym("v")]]]],
-        [is_hash, any_p, any_p],
     ),
+    # i/o
+    "open-output-file": Proc("open-output-file", initOutPort, (1, 1), is_str),
+    "output-port?": (op := Contract("output-port?", lambda v: type(v) is OutputPort)),
+    "close-port": Proc("close-port", OutputPort.close, (1, 1), op),
+    "closed?": Proc("closed?", lambda o: o.closed, (1, 1), op),
+    # printing
+    "display": Proc("display",
+        lambda v, f=None: print(display_str(v), end="", file=f), (1, 2), any_p, op),
+    "displayln": Proc("displayln",
+        lambda v, f=None: print(display_str(v), file=f), (1, 2), any_p, op),
+    "print": Proc("print",
+        lambda v, f=None: print(print_str(v), end="", file=f), (1, 2), any_p, op),
+    "println": Proc("println",
+        lambda v, f=None: print(print_str(v), file=f), (1, 2), any_p, op),
     # actions
-    "assert": Proc("assert", palet_assert, (1, 2), [any_p, orc(is_str, False)]),
-    "display": Proc("display", lambda v: print(display_str(v), end=""), (1, 1)),
-    "displayln": Proc("displayln", lambda v: print(display_str(v)), (1, 1)),
-    "error": Proc("error", raise_, (1, 1), [is_str]),
-    "sleep": Proc("sleep", sleep, (1, 1), [is_int_or_float]),
-    "print": Proc("print", lambda v: print(print_str(v), end=""), (1, 1)),
-    "println": Proc("println", lambda v: print(print_str(v)), (1, 1)),
-    "system": Proc("system", palet_system, (1, 1), [is_str]),
+    "assert": Proc("assert", palet_assert, (1, 2), any_p, orc(is_str, False)),
+    "error": Proc("error", raise_, (1, 1), is_str),
+    "sleep": Proc("sleep", sleep, (1, 1), is_int_or_float),
+    "system": Proc("system", palet_system, (1, 1), is_str),
     # conversions
-    "number->string": Proc("number->string", number_to_string, (1, 1), [is_num]),
+    "number->string": Proc("number->string", number_to_string, (1, 1), is_num),
     "string->vector": Proc(
-        "string->vector", lambda s: [Char(c) for c in s], (1, 1), [is_str]
+        "string->vector", lambda s: [Char(c) for c in s], (1, 1), is_str
     ),
-    "range->vector": Proc("range->vector", list, (1, 1), [is_range]),
+    "range->vector": Proc("range->vector", list, (1, 1), is_range),
     # reflexion
-    "var-exists?": Proc("var-exists?", lambda sym: sym.val in env, (1, 1), [is_symbol]),
+    "var-exists?": Proc("var-exists?", lambda sym: sym.val in env, (1, 1), is_symbol),
     "rename": Syntax(syn_rename),
     "delete": Syntax(syn_delete),
 })