feat: 实现 2 pass AST 到字节码编译器

2026-06-05 02:47:03 +08:00
parent 60c40d9134
commit 6de8ed1297
3 changed files with 357 additions and 0 deletions
@@ -0,0 +1,47 @@
+use std::fmt;
+
+use crate::compiler::Constant;
+use crate::opcode::{self, OpCode};
+
+pub struct CompiledModule {
+    pub bytecode: Vec<opcode::Instr>,
+    pub constants: Vec<Constant>,
+    pub n_registers: u8,
+}
+
+impl fmt::Display for CompiledModule {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        writeln!(f, "; n_registers = {}", self.n_registers)?;
+        writeln!(f, "; constants:")?;
+        for (i, c) in self.constants.iter().enumerate() {
+            writeln!(f, ";   [{i}] = {:?}", c)?;
+        }
+        writeln!(f)?;
+        for (offset, &instr) in self.bytecode.iter().enumerate() {
+            let op = opcode::opcode(instr);
+            let rd = opcode::rd(instr);
+            let a = opcode::a(instr);
+            let b = opcode::b(instr);
+            write!(f, "  {offset:04}  {op}")?;
+            match op {
+                OpCode::LOAD_CONST => {
+                    write!(f, " r{rd}, const[{}]", opcode::imm16(instr))?;
+                }
+                OpCode::JMP | OpCode::JMP_IF_FALSE | OpCode::JMP_IF_TRUE => {
+                    write!(f, " r{rd}, {offset:+}", offset = opcode::offset16(instr))?;
+                }
+                OpCode::RETURN | OpCode::HALT | OpCode::NOP => {
+                    write!(f, " r{rd}")?;
+                }
+                OpCode::NEG => {
+                    write!(f, " r{rd}, r{a}")?;
+                }
+                _ => {
+                    write!(f, " r{rd}, r{a}, r{b}")?;
+                }
+            }
+            writeln!(f)?;
+        }
+        Ok(())
+    }
+}