Wasynth/src/backend/ast/operation.rs
2021-11-27 00:43:25 -05:00

504 lines
14 KiB
Rust

use std::convert::TryFrom;
use parity_wasm::elements::Instruction;
#[allow(non_camel_case_types)]
#[derive(Clone, Copy)]
pub enum Load {
I32,
I64,
F32,
F64,
I32_I8,
I32_U8,
I32_I16,
I32_U16,
I64_I8,
I64_U8,
I64_I16,
I64_U16,
I64_I32,
I64_U32,
}
impl Load {
pub fn as_name(self) -> &'static str {
match self {
Self::I32 => "i32",
Self::I64 => "i64",
Self::F32 => "f32",
Self::F64 => "f64",
Self::I32_I8 => "i32_i8",
Self::I32_U8 => "i32_u8",
Self::I32_I16 => "i32_i16",
Self::I32_U16 => "i32_u16",
Self::I64_I8 => "i64_i8",
Self::I64_U8 => "i64_u8",
Self::I64_I16 => "i64_i16",
Self::I64_U16 => "i64_u16",
Self::I64_I32 => "i64_i32",
Self::I64_U32 => "i64_u32",
}
}
}
impl TryFrom<&Instruction> for Load {
type Error = ();
fn try_from(inst: &Instruction) -> Result<Self, Self::Error> {
let result = match inst {
Instruction::I32Load(_, _) => Self::I32,
Instruction::I64Load(_, _) => Self::I64,
Instruction::F32Load(_, _) => Self::F32,
Instruction::F64Load(_, _) => Self::F64,
Instruction::I32Load8S(_, _) => Self::I32_I8,
Instruction::I32Load8U(_, _) => Self::I32_U8,
Instruction::I32Load16S(_, _) => Self::I32_I16,
Instruction::I32Load16U(_, _) => Self::I32_U16,
Instruction::I64Load8S(_, _) => Self::I64_I8,
Instruction::I64Load8U(_, _) => Self::I64_U8,
Instruction::I64Load16S(_, _) => Self::I64_I16,
Instruction::I64Load16U(_, _) => Self::I64_U16,
Instruction::I64Load32S(_, _) => Self::I64_I32,
Instruction::I64Load32U(_, _) => Self::I64_U32,
_ => return Err(()),
};
Ok(result)
}
}
#[allow(non_camel_case_types)]
#[derive(Clone, Copy)]
pub enum Store {
I32,
I64,
F32,
F64,
I32_N8,
I32_N16,
I64_N8,
I64_N16,
I64_N32,
}
impl Store {
pub fn as_name(self) -> &'static str {
match self {
Self::I32 => "i32",
Self::I64 => "i64",
Self::F32 => "f32",
Self::F64 => "f64",
Self::I32_N8 => "i32_n8",
Self::I32_N16 => "i32_n16",
Self::I64_N8 => "i64_n8",
Self::I64_N16 => "i64_n16",
Self::I64_N32 => "i64_n32",
}
}
}
impl TryFrom<&Instruction> for Store {
type Error = ();
fn try_from(inst: &Instruction) -> Result<Self, Self::Error> {
let result = match inst {
Instruction::I32Store(_, _) => Self::I32,
Instruction::I64Store(_, _) => Self::I64,
Instruction::F32Store(_, _) => Self::F32,
Instruction::F64Store(_, _) => Self::F64,
Instruction::I32Store8(_, _) => Self::I32_N8,
Instruction::I32Store16(_, _) => Self::I32_N16,
Instruction::I64Store8(_, _) => Self::I64_N8,
Instruction::I64Store16(_, _) => Self::I64_N16,
Instruction::I64Store32(_, _) => Self::I64_N32,
_ => return Err(()),
};
Ok(result)
}
}
// Order of mnemonics is:
// operation_result_parameter
#[allow(non_camel_case_types)]
#[derive(Clone, Copy)]
pub enum UnOp {
Eqz_I32,
Eqz_I64,
Clz_I32,
Ctz_I32,
Popcnt_I32,
Clz_I64,
Ctz_I64,
Popcnt_I64,
Abs_FN,
Neg_FN,
Ceil_FN,
Floor_FN,
Trunc_FN,
Nearest_FN,
Sqrt_FN,
Wrap_I32_I64,
Trunc_I32_F32,
Trunc_U32_F32,
Trunc_I32_F64,
Trunc_U32_F64,
Extend_I64_I32,
Extend_U64_I32,
Trunc_I64_F32,
Trunc_U64_F32,
Trunc_I64_F64,
Trunc_U64_F64,
Convert_F32_I32,
Convert_F32_U32,
Convert_F32_I64,
Convert_F32_U64,
Demote_F32_F64,
Convert_F64_I32,
Convert_F64_U32,
Convert_F64_I64,
Convert_F64_U64,
Promote_F64_F32,
Reinterpret_I32_F32,
Reinterpret_I64_F64,
Reinterpret_F32_I32,
Reinterpret_F64_I64,
}
impl UnOp {
pub fn as_operator(self) -> Option<&'static str> {
let op = match self {
Self::Neg_FN => "-",
_ => return None,
};
Some(op)
}
pub fn as_name(self) -> (&'static str, &'static str) {
match self {
Self::Eqz_I32 => ("eqz", "i32"),
Self::Eqz_I64 => ("eqz", "i64"),
Self::Clz_I32 => ("clz", "i32"),
Self::Ctz_I32 => ("ctz", "i32"),
Self::Popcnt_I32 => ("popcnt", "i32"),
Self::Clz_I64 => ("clz", "i64"),
Self::Ctz_I64 => ("ctz", "i64"),
Self::Popcnt_I64 => ("popcnt", "i64"),
Self::Abs_FN => ("math", "abs"),
Self::Neg_FN => ("neg", "num"),
Self::Ceil_FN => ("math", "ceil"),
Self::Floor_FN => ("math", "floor"),
Self::Trunc_FN => ("trunc", "num"),
Self::Nearest_FN => ("nearest", "num"),
Self::Sqrt_FN => ("math", "sqrt"),
Self::Wrap_I32_I64 => ("wrap", "i32_i64"),
Self::Trunc_I32_F32 => ("trunc", "i32_f32"),
Self::Trunc_U32_F32 => ("trunc", "u32_f32"),
Self::Trunc_I32_F64 => ("trunc", "i32_f64"),
Self::Trunc_U32_F64 => ("trunc", "u32_f64"),
Self::Extend_I64_I32 => ("extend", "i64_i32"),
Self::Extend_U64_I32 => ("extend", "u64_i32"),
Self::Trunc_I64_F32 => ("trunc", "i64_f32"),
Self::Trunc_U64_F32 => ("trunc", "u64_f32"),
Self::Trunc_I64_F64 => ("trunc", "i64_f64"),
Self::Trunc_U64_F64 => ("trunc", "u64_f64"),
Self::Convert_F32_I32 => ("convert", "f32_i32"),
Self::Convert_F32_U32 => ("convert", "f32_u32"),
Self::Convert_F32_I64 => ("convert", "f32_i64"),
Self::Convert_F32_U64 => ("convert", "f32_u64"),
Self::Demote_F32_F64 => ("demote", "f32_f64"),
Self::Convert_F64_I32 => ("convert", "f64_i32"),
Self::Convert_F64_U32 => ("convert", "f64_u32"),
Self::Convert_F64_I64 => ("convert", "f64_i64"),
Self::Convert_F64_U64 => ("convert", "f64_u64"),
Self::Promote_F64_F32 => ("promote", "f64_f32"),
Self::Reinterpret_I32_F32 => ("reinterpret", "i32_f32"),
Self::Reinterpret_I64_F64 => ("reinterpret", "i64_f64"),
Self::Reinterpret_F32_I32 => ("reinterpret", "f32_i32"),
Self::Reinterpret_F64_I64 => ("reinterpret", "f64_i64"),
}
}
}
impl TryFrom<&Instruction> for UnOp {
type Error = ();
fn try_from(inst: &Instruction) -> Result<Self, Self::Error> {
let result = match inst {
Instruction::I32Eqz => Self::Eqz_I32,
Instruction::I64Eqz => Self::Eqz_I64,
Instruction::I32Clz => Self::Clz_I32,
Instruction::I32Ctz => Self::Ctz_I32,
Instruction::I32Popcnt => Self::Popcnt_I32,
Instruction::I64Clz => Self::Clz_I64,
Instruction::I64Ctz => Self::Ctz_I64,
Instruction::I64Popcnt => Self::Popcnt_I64,
Instruction::F32Abs | Instruction::F64Abs => Self::Abs_FN,
Instruction::F32Neg | Instruction::F64Neg => Self::Neg_FN,
Instruction::F32Ceil | Instruction::F64Ceil => Self::Ceil_FN,
Instruction::F32Floor | Instruction::F64Floor => Self::Floor_FN,
Instruction::F32Trunc | Instruction::F64Trunc => Self::Trunc_FN,
Instruction::F32Nearest | Instruction::F64Nearest => Self::Nearest_FN,
Instruction::F32Sqrt | Instruction::F64Sqrt => Self::Sqrt_FN,
Instruction::I32WrapI64 => Self::Wrap_I32_I64,
Instruction::I32TruncSF32 => Self::Trunc_I32_F32,
Instruction::I32TruncUF32 => Self::Trunc_U32_F32,
Instruction::I32TruncSF64 => Self::Trunc_I32_F64,
Instruction::I32TruncUF64 => Self::Trunc_U32_F64,
Instruction::I64ExtendSI32 => Self::Extend_I64_I32,
Instruction::I64ExtendUI32 => Self::Extend_U64_I32,
Instruction::I64TruncSF32 => Self::Trunc_I64_F32,
Instruction::I64TruncUF32 => Self::Trunc_U64_F32,
Instruction::I64TruncSF64 => Self::Trunc_I64_F64,
Instruction::I64TruncUF64 => Self::Trunc_U64_F64,
Instruction::F32ConvertSI32 => Self::Convert_F32_I32,
Instruction::F32ConvertUI32 => Self::Convert_F32_U32,
Instruction::F32ConvertSI64 => Self::Convert_F32_I64,
Instruction::F32ConvertUI64 => Self::Convert_F32_U64,
Instruction::F32DemoteF64 => Self::Demote_F32_F64,
Instruction::F64ConvertSI32 => Self::Convert_F64_I32,
Instruction::F64ConvertUI32 => Self::Convert_F64_U32,
Instruction::F64ConvertSI64 => Self::Convert_F64_I64,
Instruction::F64ConvertUI64 => Self::Convert_F64_U64,
Instruction::F64PromoteF32 => Self::Promote_F64_F32,
Instruction::I32ReinterpretF32 => Self::Reinterpret_I32_F32,
Instruction::I64ReinterpretF64 => Self::Reinterpret_I64_F64,
Instruction::F32ReinterpretI32 => Self::Reinterpret_F32_I32,
Instruction::F64ReinterpretI64 => Self::Reinterpret_F64_I64,
_ => return Err(()),
};
Ok(result)
}
}
#[allow(non_camel_case_types)]
#[derive(Clone, Copy)]
pub enum BinOp {
Eq_I32,
Ne_I32,
LtS_I32,
LtU_I32,
GtS_I32,
GtU_I32,
LeS_I32,
LeU_I32,
GeS_I32,
GeU_I32,
Eq_I64,
Ne_I64,
LtS_I64,
LtU_I64,
GtS_I64,
GtU_I64,
LeS_I64,
LeU_I64,
GeS_I64,
GeU_I64,
Add_I32,
Sub_I32,
Mul_I32,
DivS_I32,
DivU_I32,
RemS_I32,
RemU_I32,
And_I32,
Or_I32,
Xor_I32,
Shl_I32,
ShrS_I32,
ShrU_I32,
Rotl_I32,
Rotr_I32,
Add_I64,
Sub_I64,
Mul_I64,
DivS_I64,
DivU_I64,
RemS_I64,
RemU_I64,
And_I64,
Or_I64,
Xor_I64,
Shl_I64,
ShrS_I64,
ShrU_I64,
Rotl_I64,
Rotr_I64,
Eq_FN,
Ne_FN,
Lt_FN,
Gt_FN,
Le_FN,
Ge_FN,
Add_FN,
Sub_FN,
Mul_FN,
Div_FN,
Min_FN,
Max_FN,
Copysign_FN,
}
impl BinOp {
pub fn as_operator(self) -> Option<&'static str> {
let op = match self {
Self::Add_FN => "+",
Self::Sub_FN => "-",
Self::Mul_FN => "*",
Self::Div_FN => "/",
Self::RemS_I32 | Self::RemU_I32 | Self::RemS_I64 | Self::RemU_I64 => "%",
_ => return None,
};
Some(op)
}
pub fn as_name(self) -> (&'static str, &'static str) {
match self {
Self::Eq_I32 => ("eq", "i32"),
Self::Ne_I32 => ("ne", "i32"),
Self::LtS_I32 => ("lt", "i32"),
Self::LtU_I32 => ("lt", "u32"),
Self::GtS_I32 => ("gt", "i32"),
Self::GtU_I32 => ("gt", "u32"),
Self::LeS_I32 => ("le", "i32"),
Self::LeU_I32 => ("le", "u32"),
Self::GeS_I32 => ("ge", "i32"),
Self::GeU_I32 => ("ge", "u32"),
Self::Eq_I64 => ("eq", "i64"),
Self::Ne_I64 => ("ne", "i64"),
Self::LtS_I64 => ("lt", "i64"),
Self::LtU_I64 => ("lt", "u64"),
Self::GtS_I64 => ("gt", "i64"),
Self::GtU_I64 => ("gt", "u64"),
Self::LeS_I64 => ("le", "i64"),
Self::LeU_I64 => ("le", "u64"),
Self::GeS_I64 => ("ge", "i64"),
Self::GeU_I64 => ("ge", "u64"),
Self::Add_I32 => ("add", "i32"),
Self::Sub_I32 => ("sub", "i32"),
Self::Mul_I32 => ("mul", "i32"),
Self::DivS_I32 => ("div", "i32"),
Self::DivU_I32 => ("div", "u32"),
Self::RemS_I32 => ("rem", "i32"),
Self::RemU_I32 => ("rem", "u32"),
Self::And_I32 => ("band", "i32"),
Self::Or_I32 => ("bor", "i32"),
Self::Xor_I32 => ("bxor", "i32"),
Self::Shl_I32 => ("shl", "i32"),
Self::ShrS_I32 => ("shr", "i32"),
Self::ShrU_I32 => ("shr", "u32"),
Self::Rotl_I32 => ("rotl", "i32"),
Self::Rotr_I32 => ("rotr", "i32"),
Self::Add_I64 => ("add", "i64"),
Self::Sub_I64 => ("sub", "i64"),
Self::Mul_I64 => ("mul", "i64"),
Self::DivS_I64 => ("div", "i64"),
Self::DivU_I64 => ("div", "u64"),
Self::RemS_I64 => ("rem", "i64"),
Self::RemU_I64 => ("rem", "u64"),
Self::And_I64 => ("band", "i64"),
Self::Or_I64 => ("bor", "i64"),
Self::Xor_I64 => ("bxor", "i64"),
Self::Shl_I64 => ("shl", "i64"),
Self::ShrS_I64 => ("shr", "i64"),
Self::ShrU_I64 => ("shr", "u64"),
Self::Rotl_I64 => ("rotl", "i64"),
Self::Rotr_I64 => ("rotr", "i64"),
Self::Eq_FN => ("eq", "num"),
Self::Ne_FN => ("ne", "num"),
Self::Lt_FN => ("lt", "num"),
Self::Gt_FN => ("gt", "num"),
Self::Le_FN => ("le", "num"),
Self::Ge_FN => ("ge", "num"),
Self::Add_FN => ("add", "num"),
Self::Sub_FN => ("sub", "num"),
Self::Mul_FN => ("mul", "num"),
Self::Div_FN => ("div", "num"),
Self::Min_FN => ("math", "min"),
Self::Max_FN => ("math", "max"),
Self::Copysign_FN => ("copysign", "num"),
}
}
}
impl TryFrom<&Instruction> for BinOp {
type Error = ();
fn try_from(inst: &Instruction) -> Result<Self, Self::Error> {
let result = match inst {
Instruction::I32Eq => Self::Eq_I32,
Instruction::I32Ne => Self::Ne_I32,
Instruction::I32LtS => Self::LtS_I32,
Instruction::I32LtU => Self::LtU_I32,
Instruction::I32GtS => Self::GtS_I32,
Instruction::I32GtU => Self::GtU_I32,
Instruction::I32LeS => Self::LeS_I32,
Instruction::I32LeU => Self::LeU_I32,
Instruction::I32GeS => Self::GeS_I32,
Instruction::I32GeU => Self::GeU_I32,
Instruction::I64Eq => Self::Eq_I64,
Instruction::I64Ne => Self::Ne_I64,
Instruction::I64LtS => Self::LtS_I64,
Instruction::I64LtU => Self::LtU_I64,
Instruction::I64GtS => Self::GtS_I64,
Instruction::I64GtU => Self::GtU_I64,
Instruction::I64LeS => Self::LeS_I64,
Instruction::I64LeU => Self::LeU_I64,
Instruction::I64GeS => Self::GeS_I64,
Instruction::I64GeU => Self::GeU_I64,
Instruction::I32Add => Self::Add_I32,
Instruction::I32Sub => Self::Sub_I32,
Instruction::I32Mul => Self::Mul_I32,
Instruction::I32DivS => Self::DivS_I32,
Instruction::I32DivU => Self::DivU_I32,
Instruction::I32RemS => Self::RemS_I32,
Instruction::I32RemU => Self::RemU_I32,
Instruction::I32And => Self::And_I32,
Instruction::I32Or => Self::Or_I32,
Instruction::I32Xor => Self::Xor_I32,
Instruction::I32Shl => Self::Shl_I32,
Instruction::I32ShrS => Self::ShrS_I32,
Instruction::I32ShrU => Self::ShrU_I32,
Instruction::I32Rotl => Self::Rotl_I32,
Instruction::I32Rotr => Self::Rotr_I32,
Instruction::I64Add => Self::Add_I64,
Instruction::I64Sub => Self::Sub_I64,
Instruction::I64Mul => Self::Mul_I64,
Instruction::I64DivS => Self::DivS_I64,
Instruction::I64DivU => Self::DivU_I64,
Instruction::I64RemS => Self::RemS_I64,
Instruction::I64RemU => Self::RemU_I64,
Instruction::I64And => Self::And_I64,
Instruction::I64Or => Self::Or_I64,
Instruction::I64Xor => Self::Xor_I64,
Instruction::I64Shl => Self::Shl_I64,
Instruction::I64ShrS => Self::ShrS_I64,
Instruction::I64ShrU => Self::ShrU_I64,
Instruction::I64Rotl => Self::Rotl_I64,
Instruction::I64Rotr => Self::Rotr_I64,
Instruction::F32Eq | Instruction::F64Eq => Self::Eq_FN,
Instruction::F32Ne | Instruction::F64Ne => Self::Ne_FN,
Instruction::F32Lt | Instruction::F64Lt => Self::Lt_FN,
Instruction::F32Gt | Instruction::F64Gt => Self::Gt_FN,
Instruction::F32Le | Instruction::F64Le => Self::Le_FN,
Instruction::F32Ge | Instruction::F64Ge => Self::Ge_FN,
Instruction::F32Add | Instruction::F64Add => Self::Add_FN,
Instruction::F32Sub | Instruction::F64Sub => Self::Sub_FN,
Instruction::F32Mul | Instruction::F64Mul => Self::Mul_FN,
Instruction::F32Div | Instruction::F64Div => Self::Div_FN,
Instruction::F32Min | Instruction::F64Min => Self::Min_FN,
Instruction::F32Max | Instruction::F64Max => Self::Max_FN,
Instruction::F32Copysign | Instruction::F64Copysign => Self::Copysign_FN,
_ => {
return Err(());
}
};
Ok(result)
}
}