Factor out operator analysis and simplify output

codegen-luau/src/analyzer/localize.rs

@@ -5,6 +5,8 @@ use wasm_ast::{
 	visit::{Driver, Visitor},
 };
 
+use super::operator::{bin_symbol_of, cmp_symbol_of};
+
 struct Visit {
 	result: BTreeSet<(&'static str, &'static str)>,
 }
@@ -29,7 +31,7 @@ impl Visitor for Visit {
 	}
 
 	fn visit_bin_op(&mut self, v: &BinOp) {
-		if v.op.as_operator().is_some() {
+		if bin_symbol_of(v.op).is_some() {
 			return;
 		}
 
@@ -39,7 +41,7 @@ impl Visitor for Visit {
 	}
 
 	fn visit_cmp_op(&mut self, v: &CmpOp) {
-		if v.op.as_operator().is_some() {
+		if cmp_symbol_of(v.op).is_some() {
 			return;
 		}
 

codegen-luau/src/analyzer/mod.rs

@@ -1,2 +1,3 @@
 pub mod localize;
 pub mod memory;
+pub mod operator;

codegen-luau/src/analyzer/operator.rs

@@ -0,0 +1,30 @@
+use wasm_ast::node::{BinOpType, CmpOpType};
+
+pub fn bin_symbol_of(op: BinOpType) -> Option<&'static str> {
+	let result = match op {
+		BinOpType::Add_I32 | BinOpType::Add_I64 | BinOpType::Add_FN => "+",
+		BinOpType::Sub_I32 | BinOpType::Sub_I64 | BinOpType::Sub_FN => "-",
+		BinOpType::Mul_I32 | BinOpType::Mul_I64 | BinOpType::Mul_FN => "*",
+		BinOpType::DivS_I32 | BinOpType::DivS_I64 | BinOpType::Div_FN => "/",
+		BinOpType::RemS_I32 | BinOpType::RemU_I32 | BinOpType::RemS_I64 | BinOpType::RemU_I64 => {
+			"%"
+		}
+		_ => return None,
+	};
+
+	Some(result)
+}
+
+pub fn cmp_symbol_of(op: CmpOpType) -> Option<&'static str> {
+	let result = match op {
+		CmpOpType::Eq_I32 | CmpOpType::Eq_I64 | CmpOpType::Eq_FN => "==",
+		CmpOpType::Ne_I32 | CmpOpType::Ne_I64 | CmpOpType::Ne_FN => "~=",
+		CmpOpType::LtS_I32 | CmpOpType::LtS_I64 | CmpOpType::Lt_FN => "<",
+		CmpOpType::GtS_I32 | CmpOpType::GtS_I64 | CmpOpType::Gt_FN => ">",
+		CmpOpType::LeS_I32 | CmpOpType::LeS_I64 | CmpOpType::Le_FN => "<=",
+		CmpOpType::GeS_I32 | CmpOpType::GeS_I64 | CmpOpType::Ge_FN => ">=",
+		_ => return None,
+	};
+
+	Some(result)
+}

codegen-luau/src/backend/expression.rs

@@ -5,6 +5,8 @@ use wasm_ast::node::{
 	UnOp, Value,
 };
 
+use crate::analyzer::operator::bin_symbol_of;
+
 use super::manager::{write_f32, write_f64, write_separated, write_variable, Driver, Manager};
 
 impl Driver for Recall {
@@ -80,32 +82,22 @@ impl Driver for UnOp {
 	}
 }
 
-fn write_bin_op(bin_op: &BinOp, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
-	let op = bin_op.op.as_operator().unwrap();
-
-	write!(w, "(")?;
-	bin_op.lhs.write(mng, w)?;
-	write!(w, "{op} ")?;
-	bin_op.rhs.write(mng, w)?;
-	write!(w, ")")
-}
-
-fn write_bin_op_call(bin_op: &BinOp, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
-	let (a, b) = bin_op.op.as_name();
-
-	write!(w, "{a}_{b}(")?;
-	bin_op.lhs.write(mng, w)?;
-	write!(w, ", ")?;
-	bin_op.rhs.write(mng, w)?;
-	write!(w, ")")
-}
-
 impl Driver for BinOp {
 	fn write(&self, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
-		if self.op.as_operator().is_some() {
-			write_bin_op(self, mng, w)
+		if let Some(symbol) = bin_symbol_of(self.op) {
+			write!(w, "(")?;
+			self.lhs.write(mng, w)?;
+			write!(w, "{symbol} ")?;
+			self.rhs.write(mng, w)?;
+			write!(w, ")")
 		} else {
-			write_bin_op_call(self, mng, w)
+			let (head, tail) = self.op.as_name();
+
+			write!(w, "{head}_{tail}(")?;
+			self.lhs.write(mng, w)?;
+			write!(w, ", ")?;
+			self.rhs.write(mng, w)?;
+			write!(w, ")")
 		}
 	}
 }

codegen-luau/src/backend/statement.rs

@@ -259,14 +259,14 @@ fn write_variable_list(ir: &Intermediate, w: &mut dyn Write) -> Result<()> {
 
 	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)?;
+		write_separated(range, |_, w| w.write_all(b"0"), w)?;
+		write!(w, " ")?;
 	}
 
 	if ir.num_stack != 0 {

codegen-luau/src/translator.rs

@@ -308,11 +308,6 @@ pub fn translate(wasm: &Module, type_info: &TypeInfo, w: &mut dyn Write) -> Resu
 
 	write_localize_used(&func_list, w)?;
 
-	write!(w, "local ZERO_i32 = 0 ")?;
-	write!(w, "local ZERO_i64 = 0 ")?;
-	write!(w, "local ZERO_f32 = 0.0 ")?;
-	write!(w, "local ZERO_f64 = 0.0 ")?;
-
 	write_named_array("FUNC_LIST", wasm.functions_space(), w)?;
 	write_named_array("TABLE_LIST", wasm.table_space(), w)?;
 	write_named_array("MEMORY_LIST", wasm.memory_space(), w)?;