106 lines
2.2 KiB
Rust
106 lines
2.2 KiB
Rust
use std::{
|
|
collections::HashMap,
|
|
io::{Result, Write},
|
|
ops::Range,
|
|
};
|
|
|
|
use wasm_ast::node::{BrTable, CmpOp, Expression};
|
|
|
|
use crate::analyzer::operator::cmp_symbol_of;
|
|
|
|
#[derive(Default)]
|
|
pub struct Manager {
|
|
table_map: HashMap<usize, usize>,
|
|
label_list: Vec<usize>,
|
|
num_label: usize,
|
|
num_param: usize,
|
|
}
|
|
|
|
impl Manager {
|
|
pub fn get_table_index(&self, table: &BrTable) -> usize {
|
|
let id = table as *const _ as usize;
|
|
|
|
self.table_map[&id]
|
|
}
|
|
|
|
pub fn set_table_map(&mut self, map: HashMap<usize, usize>) {
|
|
self.table_map = map;
|
|
}
|
|
|
|
pub fn set_num_param(&mut self, num: usize) {
|
|
self.num_param = num;
|
|
}
|
|
|
|
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_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_cmp_op(cmp: &CmpOp, mng: &mut Manager, w: &mut dyn Write) -> Result<()> {
|
|
if let Some(symbol) = cmp_symbol_of(cmp.op_type()) {
|
|
cmp.lhs().write(mng, w)?;
|
|
write!(w, "{symbol} ")?;
|
|
cmp.rhs().write(mng, w)
|
|
} else {
|
|
let (head, tail) = cmp.op_type().as_name();
|
|
|
|
write!(w, "{head}_{tail}(")?;
|
|
cmp.lhs().write(mng, w)?;
|
|
write!(w, ", ")?;
|
|
cmp.rhs().write(mng, w)?;
|
|
write!(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 ")
|
|
}
|
|
}
|