Refactor LuaJIT codegen

codegen-luajit/src/backend/expression.rs

@@ -0,0 +1,130 @@
+use std::io::{Result, Write};
+
+use wasm_ast::node::{
+	BinOp, CmpOp, Expression, GetGlobal, GetLocal, LoadAt, MemoryGrow, MemorySize, Recall, Select,
+	UnOp, Value,
+};
+
+use super::manager::{
+	write_bin_call, write_cmp_op, write_condition, write_f32, write_f64, write_separated,
+	write_variable, Driver, Manager,
+};
+
+impl Driver for Recall {
+	fn write(&self, _: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write!(w, "reg_{} ", self.var)
+	}
+}
+
+impl Driver for Select {
+	fn write(&self, v: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write!(w, "(")?;
+		write_condition(&self.cond, v, w)?;
+		write!(w, "and ")?;
+		self.a.write(v, w)?;
+		write!(w, "or ")?;
+		self.b.write(v, w)?;
+		write!(w, ")")
+	}
+}
+
+impl Driver for GetLocal {
+	fn write(&self, v: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write_variable(self.var, v, w)
+	}
+}
+
+impl Driver for GetGlobal {
+	fn write(&self, _: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write!(w, "GLOBAL_LIST[{}].value ", self.var)
+	}
+}
+
+impl Driver for LoadAt {
+	fn write(&self, v: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write!(w, "load_{}(memory_at_0, ", self.what.as_name())?;
+		self.pointer.write(v, w)?;
+		write!(w, "+ {})", self.offset)
+	}
+}
+
+impl Driver for MemorySize {
+	fn write(&self, _: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write!(w, "memory_at_{}.min ", self.memory)
+	}
+}
+
+impl Driver for MemoryGrow {
+	fn write(&self, v: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write!(w, "rt.allocator.grow(memory_at_{}, ", self.memory)?;
+		self.value.write(v, w)?;
+		write!(w, ")")
+	}
+}
+
+impl Driver for Value {
+	fn write(&self, _: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		match self {
+			Self::I32(i) => write!(w, "{i} "),
+			Self::I64(i) => write!(w, "{i}LL "),
+			Self::F32(f) => write_f32(*f, w),
+			Self::F64(f) => write_f64(*f, w),
+		}
+	}
+}
+
+impl Driver for UnOp {
+	fn write(&self, v: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		let (a, b) = self.op.as_name();
+
+		write!(w, "{a}_{b}(")?;
+		self.rhs.write(v, w)?;
+		write!(w, ")")
+	}
+}
+
+impl Driver for BinOp {
+	fn write(&self, v: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		if let Some(op) = self.op.as_operator() {
+			write!(w, "(")?;
+			self.lhs.write(v, w)?;
+			write!(w, "{op} ")?;
+			self.rhs.write(v, w)?;
+			write!(w, ")")
+		} else {
+			write_bin_call(self.op.as_name(), &self.lhs, &self.rhs, v, w)
+		}
+	}
+}
+
+impl Driver for CmpOp {
+	fn write(&self, v: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write!(w, "(")?;
+		write_cmp_op(self, v, w)?;
+		write!(w, "and 1 or 0)")
+	}
+}
+
+impl Driver for Expression {
+	fn write(&self, v: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		match self {
+			Self::Recall(e) => e.write(v, w),
+			Self::Select(e) => e.write(v, w),
+			Self::GetLocal(e) => e.write(v, w),
+			Self::GetGlobal(e) => e.write(v, w),
+			Self::LoadAt(e) => e.write(v, w),
+			Self::MemorySize(e) => e.write(v, w),
+			Self::MemoryGrow(e) => e.write(v, w),
+			Self::Value(e) => e.write(v, w),
+			Self::UnOp(e) => e.write(v, w),
+			Self::BinOp(e) => e.write(v, w),
+			Self::CmpOp(e) => e.write(v, w),
+		}
+	}
+}
+
+impl Driver for &[Expression] {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write_separated(self.iter(), |e, w| e.write(mng, w), w)
+	}
+}

codegen-luajit/src/backend/manager.rs

@@ -0,0 +1,119 @@
+use std::{
+	io::{Result, Write},
+	ops::Range,
+};
+
+use wasm_ast::node::{CmpOp, Expression};
+
+#[derive(Default)]
+pub struct Manager {
+	label_list: Vec<usize>,
+	num_label: usize,
+	pub num_param: usize,
+}
+
+impl Manager {
+	pub fn label_list(&self) -> &[usize] {
+		&self.label_list
+	}
+
+	pub fn push_label(&mut self) -> usize {
+		self.label_list.push(self.num_label);
+		self.num_label += 1;
+
+		self.num_label - 1
+	}
+
+	pub fn pop_label(&mut self) {
+		self.label_list.pop().unwrap();
+	}
+}
+
+pub trait Driver {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()>;
+}
+
+pub fn write_separated<I, T, M>(mut iter: I, mut func: M, w: &mut dyn Write) -> Result<()>
+where
+	M: FnMut(T, &mut dyn Write) -> Result<()>,
+	I: Iterator<Item = T>,
+{
+	match iter.next() {
+		Some(first) => func(first, w)?,
+		None => return Ok(()),
+	}
+
+	iter.try_for_each(|v| {
+		write!(w, ", ")?;
+		func(v, w)
+	})
+}
+
+pub fn write_ascending(prefix: &str, range: Range<usize>, w: &mut dyn Write) -> Result<()> {
+	write_separated(range, |i, w| write!(w, "{prefix}_{i}"), w)
+}
+
+pub fn write_f32(number: f32, w: &mut dyn Write) -> Result<()> {
+	let sign = if number.is_sign_negative() { "-" } else { "" };
+
+	if number.is_infinite() {
+		write!(w, "{sign}math.huge ")
+	} else if number.is_nan() {
+		write!(w, "{sign}0/0 ")
+	} else {
+		write!(w, "{number:e} ")
+	}
+}
+
+pub fn write_f64(number: f64, w: &mut dyn Write) -> Result<()> {
+	let sign = if number.is_sign_negative() { "-" } else { "" };
+
+	if number.is_infinite() {
+		write!(w, "{sign}math.huge ")
+	} else if number.is_nan() {
+		write!(w, "{sign}0/0 ")
+	} else {
+		write!(w, "{number:e} ")
+	}
+}
+
+pub fn write_variable(var: usize, mng: &Manager, w: &mut dyn Write) -> Result<()> {
+	if let Some(rem) = var.checked_sub(mng.num_param) {
+		write!(w, "loc_{rem} ")
+	} else {
+		write!(w, "param_{var} ")
+	}
+}
+
+pub fn write_bin_call(
+	name: (&str, &str),
+	lhs: &Expression,
+	rhs: &Expression,
+	mng: &mut Manager,
+	w: &mut dyn Write,
+) -> Result<()> {
+	write!(w, "{}_{}(", name.0, name.1)?;
+	lhs.write(mng, w)?;
+	write!(w, ", ")?;
+	rhs.write(mng, w)?;
+	write!(w, ")")
+}
+
+pub fn write_cmp_op(cmp: &CmpOp, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+	if let Some(op) = cmp.op.as_operator() {
+		cmp.lhs.write(mng, w)?;
+		write!(w, "{op} ")?;
+		cmp.rhs.write(mng, w)
+	} else {
+		write_bin_call(cmp.op.as_name(), &cmp.lhs, &cmp.rhs, mng, w)
+	}
+}
+
+pub fn write_condition(data: &Expression, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+	if let Expression::CmpOp(node) = data {
+		write_cmp_op(node, mng, w)
+	} else {
+		data.write(mng, w)?;
+		write!(w, "~= 0 ")
+	}
+}

codegen-luajit/src/backend/mod.rs

@@ -0,0 +1,4 @@
+pub mod manager;
+
+mod expression;
+mod statement;

codegen-luajit/src/backend/statement.rs

@@ -0,0 +1,285 @@
+use std::{
+	io::{Result, Write},
+	ops::Range,
+};
+
+use wasm_ast::node::{
+	Backward, Br, BrIf, BrTable, Call, CallIndirect, Else, Forward, If, Intermediate, Memorize,
+	Return, SetGlobal, SetLocal, Statement, StoreAt,
+};
+
+use crate::analyzer::memory;
+
+use super::manager::{
+	write_ascending, write_condition, write_separated, write_variable, Driver, Manager,
+};
+
+impl Driver for Memorize {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write!(w, "reg_{} = ", self.var)?;
+		self.value.write(mng, w)
+	}
+}
+
+impl Driver for Forward {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		let label = mng.push_label();
+
+		self.body.iter().try_for_each(|s| s.write(mng, w))?;
+
+		write!(w, "::continue_at_{label}::")?;
+
+		mng.pop_label();
+
+		Ok(())
+	}
+}
+
+impl Driver for Backward {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		let label = mng.push_label();
+
+		write!(w, "::continue_at_{label}::")?;
+		write!(w, "while true do ")?;
+
+		self.body.iter().try_for_each(|s| s.write(mng, w))?;
+
+		write!(w, "break ")?;
+		write!(w, "end ")?;
+
+		mng.pop_label();
+
+		Ok(())
+	}
+}
+
+impl Driver for Else {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write!(w, "else ")?;
+
+		self.body.iter().try_for_each(|s| s.write(mng, w))
+	}
+}
+
+impl Driver for If {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		let label = mng.push_label();
+
+		write!(w, "if ")?;
+		write_condition(&self.cond, mng, w)?;
+		write!(w, "then ")?;
+
+		self.truthy.iter().try_for_each(|s| s.write(mng, w))?;
+
+		if let Some(s) = &self.falsey {
+			s.write(mng, w)?;
+		}
+
+		write!(w, "::continue_at_{label}::")?;
+		write!(w, "end ")?;
+
+		mng.pop_label();
+
+		Ok(())
+	}
+}
+
+fn write_br_at(up: usize, mng: &Manager, w: &mut dyn Write) -> Result<()> {
+	let level = mng.label_list().iter().nth_back(up).unwrap();
+
+	write!(w, "goto continue_at_{level} ")
+}
+
+impl Driver for Br {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write_br_at(self.target, mng, w)
+	}
+}
+
+impl Driver for BrIf {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write!(w, "if ")?;
+		write_condition(&self.cond, mng, w)?;
+		write!(w, "then ")?;
+		write_br_at(self.target, mng, w)?;
+		write!(w, "end ")
+	}
+}
+
+fn condense_jump_table(list: &[u32]) -> Vec<(usize, usize, u32)> {
+	let mut result = Vec::new();
+	let mut index = 0;
+
+	while index < list.len() {
+		let start = index;
+
+		loop {
+			index += 1;
+
+			// if end of list or next value is not equal, break
+			if index == list.len() || list[index - 1] != list[index] {
+				break;
+			}
+		}
+
+		result.push((start, index - 1, list[start]));
+	}
+
+	result
+}
+
+impl Driver for BrTable {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write!(w, "temp = ")?;
+		self.cond.write(mng, w)?;
+		write!(w, " ")?;
+
+		for (start, end, dest) in condense_jump_table(&self.data.table) {
+			if start == end {
+				write!(w, "if temp == {start} then ")?;
+			} else {
+				write!(w, "if temp >= {start} and temp <= {end} then ")?;
+			}
+
+			write_br_at(dest.try_into().unwrap(), mng, w)?;
+			write!(w, "else")?;
+		}
+
+		write!(w, " ")?;
+		write_br_at(self.data.default.try_into().unwrap(), mng, w)?;
+		write!(w, "end ")
+	}
+}
+
+impl Driver for Return {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write!(w, "do return ")?;
+		self.list.as_slice().write(mng, w)?;
+		write!(w, "end ")
+	}
+}
+
+fn write_call_store(result: Range<usize>, w: &mut dyn Write) -> Result<()> {
+	if result.is_empty() {
+		return Ok(());
+	}
+
+	write_ascending("reg", result, w)?;
+	write!(w, " = ")
+}
+
+impl Driver for Call {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write_call_store(self.result.clone(), w)?;
+
+		write!(w, "FUNC_LIST[{}](", self.func)?;
+		self.param_list.as_slice().write(mng, w)?;
+		write!(w, ")")
+	}
+}
+
+impl Driver for CallIndirect {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write_call_store(self.result.clone(), w)?;
+
+		write!(w, "TABLE_LIST[{}].data[", self.table)?;
+		self.index.write(mng, w)?;
+		write!(w, "](")?;
+		self.param_list.as_slice().write(mng, w)?;
+		write!(w, ")")
+	}
+}
+
+impl Driver for SetLocal {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write_variable(self.var, mng, w)?;
+		write!(w, "= ")?;
+		self.value.write(mng, w)
+	}
+}
+
+impl Driver for SetGlobal {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write!(w, "GLOBAL_LIST[{}].value = ", self.var)?;
+		self.value.write(mng, w)
+	}
+}
+
+impl Driver for StoreAt {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write!(w, "store_{}(memory_at_0, ", self.what.as_name())?;
+		self.pointer.write(mng, w)?;
+		write!(w, "+ {}, ", self.offset)?;
+		self.value.write(mng, w)?;
+		write!(w, ")")
+	}
+}
+
+impl Driver for Statement {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		match self {
+			Self::Unreachable => write!(w, "error(\"out of code bounds\")"),
+			Self::Memorize(s) => s.write(mng, w),
+			Self::Forward(s) => s.write(mng, w),
+			Self::Backward(s) => s.write(mng, w),
+			Self::If(s) => s.write(mng, w),
+			Self::Br(s) => s.write(mng, w),
+			Self::BrIf(s) => s.write(mng, w),
+			Self::BrTable(s) => s.write(mng, w),
+			Self::Return(s) => s.write(mng, w),
+			Self::Call(s) => s.write(mng, w),
+			Self::CallIndirect(s) => s.write(mng, w),
+			Self::SetLocal(s) => s.write(mng, w),
+			Self::SetGlobal(s) => s.write(mng, w),
+			Self::StoreAt(s) => s.write(mng, w),
+		}
+	}
+}
+
+fn write_parameter_list(ir: &Intermediate, w: &mut dyn Write) -> Result<()> {
+	write!(w, "function(")?;
+	write_ascending("param", 0..ir.num_param, w)?;
+	write!(w, ")")
+}
+
+fn write_variable_list(ir: &Intermediate, w: &mut dyn Write) -> Result<()> {
+	let mut total = 0;
+
+	for data in &ir.local_data {
+		let range = total..total + usize::try_from(data.count()).unwrap();
+		let typed = data.value_type();
+
+		total = range.end;
+
+		write!(w, "local ")?;
+		write_ascending("loc", range.clone(), w)?;
+		write!(w, " = ")?;
+		write_separated(range, |_, w| write!(w, "ZERO_{typed} "), w)?;
+	}
+
+	if ir.num_stack != 0 {
+		write!(w, "local ")?;
+		write_ascending("reg", 0..ir.num_stack, w)?;
+		write!(w, " ")?;
+	}
+
+	Ok(())
+}
+
+impl Driver for Intermediate {
+	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
+		write_parameter_list(self, w)?;
+
+		for v in memory::visit(self) {
+			write!(w, "local memory_at_{v} = MEMORY_LIST[{v}]")?;
+		}
+
+		write_variable_list(self, w)?;
+		write!(w, "local temp ")?;
+
+		mng.num_param = self.num_param;
+		self.code.write(mng, w)?;
+
+		write!(w, "end ")
+	}
+}