Albireo: Emit WAT

Author: minoki

albireo/Makefile

@@ -16,6 +16,7 @@ sources = \
   tokenize.lex.sml \
   parse.grm.sml \
   parse.sml \
+  emit.sml \
   main.sml
 
 albireo: albireo.mlb $(sources)

albireo/albireo.mlb

@@ -9,6 +9,7 @@ in
     tokenize.lex.sml
     parse.grm.sml
     parse.sml
+    emit.sml
     main.sml
   end
 end

albireo/emit.sml

@@ -0,0 +1,43 @@
+(*
+ * Copyright (c) 2025 ARATA Mizuki
+ * This file is part of LunarML.
+ *)
+structure AlbireoEmit :> sig
+val emit : AlbireoSyntax.dec list -> string
+end = struct
+structure S = AlbireoSyntax
+fun emitHeapTy S.HT_ANY = "any"
+  | emitHeapTy S.HT_I31 = "i31"
+  | emitHeapTy S.HT_EXTERN = "extern"
+fun emitTy S.I32 = "i32"
+  | emitTy S.I64 = "i64"
+  | emitTy S.F32 = "f32"
+  | emitTy S.F64 = "f64"
+  | emitTy S.V128 = "v128"
+  | emitTy (S.REF { nullable, to }) =
+      if nullable then
+        "(ref null " ^ emitHeapTy to ^ ")"
+      else
+        "(ref " ^ emitHeapTy to ^ ")"
+fun compileExp (S.IntConstExp (i, SOME S.IT_64)) = ["i64.const " ^ CharVector.map (fn #"~" => #"-" | c => c) (Int.toString i)]
+  | compileExp (S.IntConstExp (i, _)) = ["i32.const " ^ CharVector.map (fn #"~" => #"-" | c => c) (Int.toString i)]
+  | compileExp (S.VarExp id) = ["local.get $" ^ id]
+  | compileExp (S.BinExp (p, a, b)) =
+      let val insn = case p of
+                       S.PlusOp => "i32.add"
+                     | S.MinusOp => "i32.sub"
+                     | S.TimesOp => "i32.mul"
+                     | S.DivOp => "i32.div_s"
+      in compileExp a @ compileExp b @ [insn]
+      end
+fun emitDec (S.FuncDec { name, params, results, body }) =
+  let val name' = "(export \"" ^ name ^ "\")"
+      val params' = List.map (fn (id, ty) => "(param $" ^ id ^ " " ^ emitTy ty ^ ")") params
+      val results' = List.map (fn ty => "(result " ^ emitTy ty ^ ")") results
+      val body' = case body of
+          [S.ReturnStat exps] => List.concat (List.map compileExp exps) @ ["return"]
+        | _ => raise Fail "unsupported function body"
+  in "(func " ^ name' ^ "\n" ^ String.concatWith "\n" (params' @ results' @ body') ^ ")"
+  end
+fun emit decs = "(module\n" ^ String.concatWith "\n" (List.map emitDec decs) ^ ")"
+end;

albireo/main.sml

@@ -1,13 +1,17 @@
-fun makeInputFromString s =
-  let val r = ref s
-  in fn _ => let val x = !r in r := ""; x end
+fun compile filename =
+  let
+    val strm = TextIO.openIn filename
+    fun getInput n = TextIO.inputN (strm, n)
+    val lexer = AlbireoParser.makeLexer getInput filename;
+    fun onError (message, p1, p2) = raise Fail message;
+  in
+    let val (program, _) = AlbireoParser.parse ((* lookahead *) 0, lexer, onError, filename)
+    in print (AlbireoEmit.emit program ^ "\n")
+    end
+    handle AlbireoParser.ParseError => ()
+         | Fail s => print (s ^ "\n");
+    TextIO.closeIn strm
   end;
-val name = "<test input>";
-val lexer = AlbireoParser.makeLexer (makeInputFromString "func foo(x: i32) -> i32 { return x * x; }") name;
-fun onError (message, p1, p2) = raise Fail message;
-val () =
-  let val (program, _) = AlbireoParser.parse ((* lookahead *) 0, lexer, onError, name)
-  in print "OK.\n"
-  end
-  handle AlbireoParser.ParseError => ()
-       | Fail s => print (s ^ "\n");
+fun main [filename] = compile filename
+  | main _ = print "albireo <filename>";
+val () = main (CommandLine.arguments ());

albireo/runtest.mjs

@@ -0,0 +1,7 @@
+import { readFileSync } from "node:fs";
+const wasmBuffer = readFileSync("test.wasm");
+const wasmModule = await WebAssembly.instantiate(wasmBuffer);
+const { foo } = wasmModule.instance.exports;
+console.log(foo(1));
+console.log(foo(2));
+console.log(foo(3));

albireo/test.albireo

@@ -0,0 +1,4 @@
+func foo(x: i32) -> i32
+{
+    return x * x + 2 * x;
+}