midenc-codegen-masm 0.8.0

#![feature(debug_closure_helpers)]
#![feature(assert_matches)]
#![feature(iter_array_chunks)]
#![feature(iterator_try_collect)]
#![deny(warnings)]

extern crate alloc;

mod artifact;
mod data_segments;
mod emit;
mod emitter;
mod events;
pub mod intrinsics;
mod linker;
mod lower;
mod opt;
mod stack;

pub mod masm {
    pub use miden_assembly_syntax::{
        KernelLibrary, Library, Path as LibraryPathRef, PathBuf as LibraryPath,
        PathComponent as LibraryPathComponent,
        ast::*,
        debuginfo::{SourceSpan, Span, Spanned},
        parser::{IntValue, PushValue},
    };
}

use midenc_dialect_arith as arith;
use midenc_dialect_cf as cf;
use midenc_dialect_hir as hir;
use midenc_dialect_scf as scf;
use midenc_dialect_ub as ub;
use midenc_dialect_wasm as wasm;
use midenc_hir::{dialects::builtin, inventory};

pub(crate) use self::lower::HirLowering;
pub use self::{
    artifact::{MasmComponent, Rodata},
    events::{TRACE_FRAME_END, TRACE_FRAME_START, TraceEvent},
    lower::{NativePtr, ToMasmComponent},
    stack::{Constraint, Operand, OperandStack},
};

inventory::submit!(::midenc_hir::DialectRegistrationHookInfo::new::<builtin::BuiltinDialect>(
    lower_builtin_ops
));
inventory::submit!(::midenc_hir::DialectRegistrationHookInfo::new::<arith::ArithDialect>(
    lower_arith_ops
));
inventory::submit!(::midenc_hir::DialectRegistrationHookInfo::new::<cf::ControlFlowDialect>(
    lower_cf_ops
));
inventory::submit!(::midenc_hir::DialectRegistrationHookInfo::new::<scf::ScfDialect>(
    lower_scf_ops
));
inventory::submit!(::midenc_hir::DialectRegistrationHookInfo::new::<ub::UndefinedBehaviorDialect>(
    lower_ub_ops
));
inventory::submit!(::midenc_hir::DialectRegistrationHookInfo::new::<hir::HirDialect>(
    lower_hir_ops
));
inventory::submit!(::midenc_hir::DialectRegistrationHookInfo::new::<wasm::WasmDialect>(
    lower_wasm_ops
));

fn lower_builtin_ops(info: &mut midenc_hir::DialectInfo) {
    info.register_operation_trait::<builtin::Ret, dyn HirLowering>();
    info.register_operation_trait::<builtin::RetImm, dyn HirLowering>();
    info.register_operation_trait::<builtin::GlobalSymbol, dyn HirLowering>();
}

fn lower_arith_ops(info: &mut midenc_hir::DialectInfo) {
    info.register_operation_trait::<arith::Constant, dyn HirLowering>();
    info.register_operation_trait::<arith::Add, dyn HirLowering>();
    info.register_operation_trait::<arith::AddOverflowing, dyn HirLowering>();
    info.register_operation_trait::<arith::Sub, dyn HirLowering>();
    info.register_operation_trait::<arith::SubOverflowing, dyn HirLowering>();
    info.register_operation_trait::<arith::Mul, dyn HirLowering>();
    info.register_operation_trait::<arith::MulOverflowing, dyn HirLowering>();
    info.register_operation_trait::<arith::Exp, dyn HirLowering>();
    info.register_operation_trait::<arith::Div, dyn HirLowering>();
    info.register_operation_trait::<arith::Sdiv, dyn HirLowering>();
    info.register_operation_trait::<arith::Mod, dyn HirLowering>();
    info.register_operation_trait::<arith::Smod, dyn HirLowering>();
    info.register_operation_trait::<arith::Divmod, dyn HirLowering>();
    info.register_operation_trait::<arith::Sdivmod, dyn HirLowering>();
    info.register_operation_trait::<arith::And, dyn HirLowering>();
    info.register_operation_trait::<arith::Or, dyn HirLowering>();
    info.register_operation_trait::<arith::Xor, dyn HirLowering>();
    info.register_operation_trait::<arith::Band, dyn HirLowering>();
    info.register_operation_trait::<arith::Bor, dyn HirLowering>();
    info.register_operation_trait::<arith::Bxor, dyn HirLowering>();
    info.register_operation_trait::<arith::Shl, dyn HirLowering>();
    info.register_operation_trait::<arith::Shr, dyn HirLowering>();
    info.register_operation_trait::<arith::Ashr, dyn HirLowering>();
    info.register_operation_trait::<arith::Rotl, dyn HirLowering>();
    info.register_operation_trait::<arith::Rotr, dyn HirLowering>();
    info.register_operation_trait::<arith::Eq, dyn HirLowering>();
    info.register_operation_trait::<arith::Neq, dyn HirLowering>();
    info.register_operation_trait::<arith::Gt, dyn HirLowering>();
    info.register_operation_trait::<arith::Gte, dyn HirLowering>();
    info.register_operation_trait::<arith::Lt, dyn HirLowering>();
    info.register_operation_trait::<arith::Lte, dyn HirLowering>();
    info.register_operation_trait::<arith::Min, dyn HirLowering>();
    info.register_operation_trait::<arith::Max, dyn HirLowering>();
    info.register_operation_trait::<arith::Trunc, dyn HirLowering>();
    info.register_operation_trait::<arith::Zext, dyn HirLowering>();
    info.register_operation_trait::<arith::Sext, dyn HirLowering>();
    info.register_operation_trait::<arith::Incr, dyn HirLowering>();
    info.register_operation_trait::<arith::Neg, dyn HirLowering>();
    info.register_operation_trait::<arith::Inv, dyn HirLowering>();
    info.register_operation_trait::<arith::Ilog2, dyn HirLowering>();
    info.register_operation_trait::<arith::Pow2, dyn HirLowering>();
    info.register_operation_trait::<arith::Not, dyn HirLowering>();
    info.register_operation_trait::<arith::Bnot, dyn HirLowering>();
    info.register_operation_trait::<arith::IsOdd, dyn HirLowering>();
    info.register_operation_trait::<arith::Popcnt, dyn HirLowering>();
    info.register_operation_trait::<arith::Clz, dyn HirLowering>();
    info.register_operation_trait::<arith::Ctz, dyn HirLowering>();
    info.register_operation_trait::<arith::Clo, dyn HirLowering>();
    info.register_operation_trait::<arith::Cto, dyn HirLowering>();
    info.register_operation_trait::<arith::Join, dyn HirLowering>();
    info.register_operation_trait::<arith::Split, dyn HirLowering>();
}

fn lower_cf_ops(info: &mut midenc_hir::DialectInfo) {
    info.register_operation_trait::<cf::Select, dyn HirLowering>();
    info.register_operation_trait::<cf::CondBr, dyn HirLowering>();
}

fn lower_scf_ops(info: &mut midenc_hir::DialectInfo) {
    info.register_operation_trait::<scf::If, dyn HirLowering>();
    info.register_operation_trait::<scf::While, dyn HirLowering>();
    info.register_operation_trait::<scf::IndexSwitch, dyn HirLowering>();
    info.register_operation_trait::<scf::Condition, dyn HirLowering>();
    info.register_operation_trait::<scf::Yield, dyn HirLowering>();
}

fn lower_ub_ops(info: &mut midenc_hir::DialectInfo) {
    info.register_operation_trait::<ub::Unreachable, dyn HirLowering>();
    info.register_operation_trait::<ub::Poison, dyn HirLowering>();
}

fn lower_hir_ops(info: &mut midenc_hir::DialectInfo) {
    info.register_operation_trait::<hir::Assert, dyn HirLowering>();
    info.register_operation_trait::<hir::Assertz, dyn HirLowering>();
    info.register_operation_trait::<hir::AssertEq, dyn HirLowering>();
    info.register_operation_trait::<hir::PtrToInt, dyn HirLowering>();
    info.register_operation_trait::<hir::IntToPtr, dyn HirLowering>();
    info.register_operation_trait::<hir::Cast, dyn HirLowering>();
    info.register_operation_trait::<hir::Bitcast, dyn HirLowering>();
    //info.register_operation_trait::<hir::ConstantBytes, dyn HirLowering>();
    info.register_operation_trait::<hir::Exec, dyn HirLowering>();
    info.register_operation_trait::<hir::Call, dyn HirLowering>();
    info.register_operation_trait::<hir::Store, dyn HirLowering>();
    info.register_operation_trait::<hir::StoreLocal, dyn HirLowering>();
    info.register_operation_trait::<hir::Load, dyn HirLowering>();
    info.register_operation_trait::<hir::LoadLocal, dyn HirLowering>();
    info.register_operation_trait::<hir::MemGrow, dyn HirLowering>();
    info.register_operation_trait::<hir::MemSize, dyn HirLowering>();
    info.register_operation_trait::<hir::MemSet, dyn HirLowering>();
    info.register_operation_trait::<hir::MemCpy, dyn HirLowering>();
}

fn lower_wasm_ops(info: &mut midenc_hir::DialectInfo) {
    info.register_operation_trait::<wasm::SignExtend, dyn HirLowering>();
    info.register_operation_trait::<wasm::I32Load8S, dyn HirLowering>();
    info.register_operation_trait::<wasm::I32Load16S, dyn HirLowering>();
    info.register_operation_trait::<wasm::I64Load8S, dyn HirLowering>();
    info.register_operation_trait::<wasm::I64Load16S, dyn HirLowering>();
    info.register_operation_trait::<wasm::I64Load32S, dyn HirLowering>();
}