{ :emit } Where Define (emit w program) Define (write s) (IO.write_all w s) In Let emit (emitter (string_encoder program.constants) write) In Begin { (write "#include <support.h>\n") (LIST.for_each (LIST.reverse program.functions) emit) (write "void P(void){") (emit program.init) (write ";}\n") (write "const U2 record_layouts[]={") (LIST.for_each program.record_indexes Func index Begin { (LIST.for_each index Func label Begin { (write (Z.show label)) (write ",") }) (write "Z,") }) (write "};\n") (write "_Alignas(4) const uint8_t constant_bytes[]=\"") (LIST.for_each program.constants Func c (write_string_literal write c)) (write "\";\n") } Where Define (emitter encode_string write) Let compile Define (write s) 'write.s Define (sequence terms) 'sequence.terms Define (compile term) 'compile.term In Let write_comma (write ",") In Define (comma_sep terms) (sequence (LIST.join write_comma terms)) Define (comma_pre terms) (sequence (LIST.fold terms 'nil Func {term terms} [write_comma & term & terms])) In Func term Match term { | 'stmt.stmt Match stmt { | 'expr.expr (sequence (Reduce & (write "({X t=s52(),x=") (compile 'expr.expr) (write ";s15(t);x;});") 'nil)) | 'assign.{var expr} (sequence (Reduce & (compile 'expr.var) (write "=") (compile 'expr.expr) (write ";") 'nil)) | 'env_access.{closure_var pairs} Define (compile_pair {xm j}) (sequence (Reduce & (compile 'expr.xm) (write "=env[") (write (Z.show j)) (write "]") 'nil)) In (sequence (Reduce & (write "const X*env=s62(") (compile 'expr.closure_var) (write ")") (comma_pre (LIST.map pairs compile_pair)) (write ";") 'nil)) | 'decls.pairs Define (compile_pair {var maybe_init}) (sequence (Reduce & (compile 'expr.var) Match maybe_init { | 'nothing 'nil | 'just.expr (Reduce & (sequence (Reduce & (write "=") (compile 'expr.expr) 'nil)) 'nil) })) In Match pairs { | 'cons._ (sequence (Reduce & (write "X ") (comma_sep (LIST.map pairs compile_pair)) (write ";") 'nil)) } } | 'expr.expr Match expr { | 'true (write "15") | 'false (write "271") | 'num.i (write (Z.show i)) | 'empty_tuple (write "31") | 'labeled_empty_tuple.label (write (Z.show (encode_empty_variant label))) | 'alloc_labeled_value.{label expr} (sequence (Reduce & (write "s27(") (write (Z.show label)) (write ",") (compile 'expr.expr) (write ")") 'nil)) | 'remove_label.expr (sequence (Reduce & (write "s06(") (compile 'expr.expr) (write ")") 'nil)) | 'alloc_tuple.exprs Match exprs { | 'nil (compile 'expr.'empty_tuple) | 'cons._ (sequence (Reduce & (write "s78(") (write (Z.show (LIST.length exprs))) (write ",(X[]){") (comma_sep (LIST.map exprs Func expr (compile 'expr.expr))) (write "})") 'nil)) } | 'alloc_record.{layout exprs} Match exprs { | 'cons._ (sequence (Reduce & (write "s30(") (write (Z.show (LIST.length exprs))) (write ",(X[]){") (comma_sep (LIST.map exprs Func expr (compile 'expr.expr))) (write "},") (write (Z.show layout)) (write ")") 'nil)) } | 'tuple_fetch.{code i} (sequence (Reduce & (write "s68(") (compile 'expr.code) (write ",") (write (Z.show i)) (write ")") 'nil)) | 'record_fetch.{code i} (sequence (Reduce & (write "s31(") (compile 'expr.code) (write ",") (write (Z.show i)) (write ")") 'nil)) | 'iterate.{inits expr} (sequence (Reduce & (write "({__label__ l;") Match inits { | 'nil (write "U4 t=s52();") | 'cons._ (compile 'stmt.'decls.inits) } (write "l:") (compile 'expr.expr) (write ";})") 'nil)) | 'continue.stmts (sequence (Reduce & (write "({") Match stmts { | 'nil (write "s15(t);") | 'cons._ (sequence (LIST.map stmts Func stmt (compile 'stmt.stmt))) } (write "goto l;31;})") 'nil)) | 'stmt_expr.{stmts expr} (sequence (Reduce & (write "({") (sequence (LIST.map stmts Func stmt (compile 'stmt.stmt))) (compile 'expr.expr) (write ";})") 'nil)) | 'func.{i args stmts expr} Define (compile_arg arg) (sequence [(write "X ") & (compile 'expr.arg) & 'nil]) In (sequence (Reduce & (write "static X f") (write (Z.show i)) (write "(") (comma_sep (LIST.map args compile_arg)) (write "){return ") Block Let expr Match stmts { | 'nil expr | 'cons._ 'stmt_expr.{stmts expr} } In (compile 'expr.expr) (write ";}\n") 'nil)) | 'prim_app.{name args} (sequence (Reduce & (write (RUNTIME.prim_short_name name)) (write "(") (comma_sep (LIST.map args Func expr (compile 'expr.expr))) (write ")") 'nil)) | 'closure_app_known.{i func args} (sequence (Reduce & (write "f") (write (Z.show i)) (write "(") (comma_sep (LIST.map [func & args] Func expr (compile 'expr.expr))) (write ")") 'nil)) | 'closure_app.{func args} (sequence (Reduce & (write "({X c=") (compile 'expr.func) (write ";((X(*)(X") (sequence (LIST.map args Func _ (write ",X"))) (write "))s35(c,") (write (Z.show (LIST.length args))) (write "))(c") (comma_pre (LIST.map args Func arg (compile 'expr.arg))) (write ");})") 'nil)) | 'alloc_closure.{i free num_params} (sequence (Reduce & (write "s75(f") (write (Z.show i)) (write ",") (write (Z.show num_params)) (write ",") (write (Z.show (LIST.length free))) Match free { | 'nil (write ",0") | 'cons._ (sequence (Reduce & (write ",(X[]){") (comma_sep (LIST.map free Func expr (compile 'expr.expr))) (write "}") 'nil)) } (write ")") 'nil)) | 'labeled.{label expr} (sequence (Reduce & (write "({__label__ ") (compile 'expr.label) (write ";") (compile 'expr.label) (write ":") (compile 'expr.expr) (write ";})") 'nil)) | 'stuck_cond (write "s89()") | 'halt (write "s87()") | 'match.{expr_var pat_var expr clauses} Define (compile_clause clause) Match clause { | 'default.body (sequence (Reduce & (write "default:") Match body { | 'expr.expr (sequence (Reduce & (write "r=") (compile 'expr.expr) (write ";break;") 'nil)) | 'stuck (write "s53();") } 'nil)) | 'labeled.{label expr} (sequence (Reduce & (write "case ") (write (Z.show label)) (write ":r=") (compile 'expr.expr) (write ";break;") 'nil)) } In (sequence (Reduce & (write "({X r,") (compile 'expr.expr_var) (write "=") (compile 'expr.expr) (write ",") (compile 'expr.pat_var) (write "=") (compile 'expr.'remove_label.expr_var) (write ";switch(s09(") (compile 'expr.expr_var) (write ")){") (sequence (LIST.map clauses compile_clause)) (write "}r;})") 'nil)) | 'if.{test then else} (sequence (Reduce & (write "(") (compile 'expr.test) (write "==15?") (compile 'expr.then) (write ":") (compile 'expr.else) (write ")") 'nil)) | 'x.i (sequence [(write "x") & (write (Z.show i)) & 'nil]) | 'c.i (write (Z.show (encode_string i))) | 'p.i (sequence [(write "p") & (write (Z.show i)) & 'nil]) } } In Func expr Unfold term From (compile 'expr.expr) Begin Match term { | 'write.s (write s) | 'sequence.terms Iterate terms Begin Match terms { | 'nil | 'cons.{term terms} (Fold term) (Continue terms) } | 'compile.term (Fold (compile term)) } Where Define (encode_empty_variant label) [[label * 256] + TAG_IMMEDIATE_VARIANT] Define (write_string_literal write c) Begin { Let m (string_literal_length c) When [m > 255] { (OS.die "String constant is too big.") } (write "\\") (write (octal [m % 256])) (write "\\0\\0\\000") (write (STRING.clip c 1 [(STRING.length c) - 1])) Let padding [-m % 4] Iterate i From 0 When [i < padding] { (write "\\0") (Continue [i + 1]) } } Define (string_encoder constants) Let STRING_MAP (SEARCH.MAP Z.compare [Func {i _} i]) In Let {string_map _ _} (LIST.reduce constants {STRING_MAP.empty 0 0} Func {{map i offset} c} Let map Let value [[offset * 16] + TAG_STATIC_CHUNK_RO] In (STRING_MAP.insert map {i value}) Let m (string_literal_length c) In Let offset [offset + [[m + 7] / 4]] In {map [i + 1] offset}) In Func i Match (STRING_MAP.search string_map i) { | 'just.{_ value} value } Where Define (octal n) Iterate {m digits n} From {0 'nil n} Cond { | [n > 0] (Continue [m + 1] [[n % 8] & digits] [n / 8]) | [m = 0] "0" | True (CHUNK.new_ro m Func d (LIST.for_each_numbered digits Func {i digit} (CHUNK.store_byte d i [`0` + digit]))) } Define (string_literal_length c) Let k [(STRING.length c) - 1] In Iterate {m i} From {0 1} Cond { | [i = k] m | [(STRING.fetch c i) = `\\`] (Continue [m + 1] [i + 2]) | True (Continue [m + 1] [i + 1]) } Let TAG_IMMEDIATE_VARIANT 47 Let TAG_STATIC_CHUNK_RO 9 Where Open Z { :Prefix - :Infix > :Infix < :Infix = :Infix + :Infix - :Infix * :Infix / :Infix % } Open LIST {:Infix &} Where Let CHUNK Package "chunk" Let FUNC Package "func" Let IO Package "io" Let LIST Package "list" Let OS Package "os" Let QUEUE Package "queue" Let RUNTIME Package "runtime" Let SEARCH Package "search" Let STDIO Package "stdio" Let STRING Package "string" Let Z Package "z"