dialects: (wasm) Add binary encoding interface

tests/dialects/wasm/test_wasm_encoding.py

@@ -0,0 +1,7 @@
+from xdsl.dialects import wasm
+
+
+def test_empty_module():
+    module = wasm.WasmModule()
+
+    assert module.wasm() == b"\x00asm\x01\x00\x00\x00"

xdsl/dialects/wasm/encoding.py

@@ -0,0 +1,22 @@
+"""
+Helpers for encoding modules in the `wasm` dialect to the WebAssembly binary format.
+"""
+
+import abc
+from typing import BinaryIO
+
+
+class WasmBinaryEncodingContext:
+    """
+    A class to store the state of encoding.
+    """
+
+
+class EncodingException(Exception): ...
+
+
+class WasmBinaryEncodable(abc.ABC):
+
+    @abc.abstractmethod
+    def encode(self, ctx: WasmBinaryEncodingContext, io: BinaryIO) -> None:
+        raise NotImplementedError()

xdsl/dialects/wasm/ops.py

@@ -9,6 +9,9 @@
 at the bottom of this file.
 """
 
+from io import BytesIO
+from typing import BinaryIO
+
 from typing_extensions import Self
 
 from xdsl.irdl import (
@@ -18,13 +21,15 @@
 from xdsl.parser import Parser
 from xdsl.printer import Printer
 
+from .encoding import WasmBinaryEncodable, WasmBinaryEncodingContext
+
 ##==------------------------------------------------------------------------==##
 # WebAssembly module
 ##==------------------------------------------------------------------------==##
 
 
 @irdl_op_definition
-class WasmModule(IRDLOperation):
+class WasmModule(IRDLOperation, WasmBinaryEncodable):
     """
     wasm> WebAssembly programs are organized into modules, which are the unit of
     deployment, loading, and compilation. A module collects definitions for
@@ -60,6 +65,20 @@ def print(self, printer: Printer):
             printer.print_string(" attributes ")
             printer.print_attr_dict(attr_dict)
 
+    def encode(self, ctx: WasmBinaryEncodingContext, io: BinaryIO) -> None:
+        # https://webassembly.github.io/spec/core/binary/modules.html#binary-module
+        magic = b"\x00asm"
+        version = b"\x01\x00\x00\x00"
+        io.write(magic)
+        io.write(version)
+
+    def wasm(self) -> bytes:
+        ctx = WasmBinaryEncodingContext()
+        io = BytesIO()
+        self.encode(ctx, io)
+        res = io.getvalue()
+        return res
+
 
 """
 --- Licensing Terms for the WebAssembly Specification