[wasm][ast] Support ref.cast instruction (#1234)

SamChou19815 · Dec 16, 2024 · 38b2cf5 · 38b2cf5
1 parent 6c826d0
commit 38b2cf5
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/crates/samlang-ast/src/wasm.rs b/crates/samlang-ast/src/wasm.rs
@@ -10,6 +10,10 @@ pub enum InlineInstruction {
     pointer_type: String,
     value: Box<InlineInstruction>,
   },
+  Cast {
+    pointer_type: String,
+    value: Box<InlineInstruction>,
+  },
   Binary(Box<InlineInstruction>, hir::BinaryOperator, Box<InlineInstruction>),
   Load {
     index: usize,
@@ -60,6 +64,13 @@ impl InlineInstruction {
         value.pretty_print(collector, heap, table);
         collector.push(')');
       }
+      InlineInstruction::Cast { pointer_type, value } => {
+        collector.push_str("(ref.cast $");
+        collector.push_str(pointer_type);
+        collector.push(' ');
+        value.pretty_print(collector, heap, table);
+        collector.push(')');
+      }
       InlineInstruction::Binary(v1, op, v2) => {
         let op_s = match op {
           hir::BinaryOperator::MUL => "mul",
@@ -377,6 +388,10 @@ mod tests {
                 pointer_type: "Foo".to_string(),
                 value: Box::new(InlineInstruction::Const(0)),
               }),
+              Instruction::Inline(InlineInstruction::Cast {
+                pointer_type: "Foo".to_string(),
+                value: Box::new(InlineInstruction::Const(0)),
+              }),
             ],
             s2: vec![
               Instruction::Inline(InlineInstruction::Binary(
@@ -520,6 +535,7 @@ mod tests {
     (local.get $a)
     (local.set $b (i32.const 0))
     (ref.test $Foo (i32.const 0))
+    (ref.cast $Foo (i32.const 0))
   ) (else
     (i32.add (i32.const 0) (i32.const 0))
     (i32.sub (i32.const 0) (i32.const 0))