luaur_code_gen/functions/
translate_builtin_int_64_binary.rs

1use crate::enums::builtin_impl_type::BuiltinImplType;
2use crate::enums::int_64_binary::Int64Binary;
3use crate::enums::ir_cmd::IrCmd;
4use crate::enums::ir_condition::IrCondition;
5use crate::functions::builtin_check_int_64::builtin_check_int_64;
6use crate::functions::builtin_load_int_64::builtin_load_int_64;
7use crate::macros::codegen_assert::CODEGEN_ASSERT;
8use crate::records::builtin_impl_result::BuiltinImplResult;
9use crate::records::ir_builder::IrBuilder;
10use crate::records::ir_op::IrOp;
11use luaur_vm::enums::lua_type::lua_Type;
12
13pub fn translate_builtin_int_64_binary(
14    build: &mut IrBuilder,
15    nparams: i32,
16    ra: i32,
17    arg: i32,
18    args: IrOp,
19    nresults: i32,
20    pcpos: i32,
21    op: Int64Binary,
22) -> BuiltinImplResult {
23    if nparams < 2 || nresults > 1 {
24        return BuiltinImplResult {
25            r#type: BuiltinImplType::None,
26            actual_result_count: -1,
27        };
28    }
29
30    let vm_reg_arg = build.vm_reg(arg as u8);
31    builtin_check_int_64(build, vm_reg_arg, pcpos);
32    builtin_check_int_64(build, args, pcpos);
33
34    let va = builtin_load_int_64(build, vm_reg_arg);
35    let vb = builtin_load_int_64(build, args);
36
37    let bin_op = match op {
38        Int64Binary::Add => build.inst_ir_cmd_ir_op_ir_op(IrCmd::ADD_INT64, va, vb),
39        Int64Binary::Sub => build.inst_ir_cmd_ir_op_ir_op(IrCmd::SUB_INT64, va, vb),
40        Int64Binary::Mul => build.inst_ir_cmd_ir_op_ir_op(IrCmd::MUL_INT64, va, vb),
41        Int64Binary::Div => {
42            let exit = build.vm_exit(pcpos as u32);
43            build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::CHECK_DIV_INT64, va, vb, exit);
44            build.inst_ir_cmd_ir_op_ir_op(IrCmd::DIV_INT64, va, vb)
45        }
46        Int64Binary::Idiv => {
47            let exit = build.vm_exit(pcpos as u32);
48            build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::CHECK_DIV_INT64, va, vb, exit);
49            build.inst_ir_cmd_ir_op_ir_op(IrCmd::IDIV_INT64, va, vb)
50        }
51        Int64Binary::Udiv => {
52            let zero = build.const_int_64(0);
53            let cond = build.cond(IrCondition::NotEqual);
54            let exit = build.vm_exit(pcpos as u32);
55            build.inst_ir_cmd_ir_op_ir_op_ir_op_ir_op(IrCmd::CHECK_CMP_INT64, vb, zero, cond, exit);
56            build.inst_ir_cmd_ir_op_ir_op(IrCmd::UDIV_INT64, va, vb)
57        }
58        Int64Binary::Rem => {
59            let zero = build.const_int_64(0);
60            let cond = build.cond(IrCondition::NotEqual);
61            let exit = build.vm_exit(pcpos as u32);
62            build.inst_ir_cmd_ir_op_ir_op_ir_op_ir_op(IrCmd::CHECK_CMP_INT64, vb, zero, cond, exit);
63            build.inst_ir_cmd_ir_op_ir_op(IrCmd::REM_INT64, va, vb)
64        }
65        Int64Binary::Urem => {
66            let zero = build.const_int_64(0);
67            let cond = build.cond(IrCondition::NotEqual);
68            let exit = build.vm_exit(pcpos as u32);
69            build.inst_ir_cmd_ir_op_ir_op_ir_op_ir_op(IrCmd::CHECK_CMP_INT64, vb, zero, cond, exit);
70            build.inst_ir_cmd_ir_op_ir_op(IrCmd::UREM_INT64, va, vb)
71        }
72        Int64Binary::Mod => {
73            let zero = build.const_int_64(0);
74            let cond = build.cond(IrCondition::NotEqual);
75            let exit = build.vm_exit(pcpos as u32);
76            build.inst_ir_cmd_ir_op_ir_op_ir_op_ir_op(IrCmd::CHECK_CMP_INT64, vb, zero, cond, exit);
77            build.inst_ir_cmd_ir_op_ir_op(IrCmd::MOD_INT64, va, vb)
78        }
79    };
80
81    let vm_reg_ra = build.vm_reg(ra as u8);
82    build.inst_ir_cmd_ir_op_ir_op(IrCmd::STORE_INT64, vm_reg_ra, bin_op);
83    let tag = build.const_tag(lua_Type::LUA_TINTEGER as u8);
84    build.inst_ir_cmd_ir_op_ir_op(IrCmd::STORE_TAG, vm_reg_ra, tag);
85
86    BuiltinImplResult {
87        r#type: BuiltinImplType::Full,
88        actual_result_count: 1,
89    }
90}
luaur_code_gen/functions/translate_builtin_int_64_binary.rs

luaur_code_gen/functions/
translate_builtin_int_64_binary.rs
