luaur-code-gen 0.1.0

use crate::enums::ir_cmd::IrCmd;
use crate::enums::ir_condition::IrCondition;
use crate::records::ir_builder::IrBuilder;
use crate::records::ir_op::IrOp;
use crate::type_aliases::instruction_ir_builder::Instruction;

use luaur_common::macros::luau_insn_a::LUAU_INSN_A;
use luaur_common::macros::luau_insn_aux_kb::LUAU_INSN_AUX_KB;
use luaur_common::macros::luau_insn_aux_not::LUAU_INSN_AUX_NOT;
use luaur_vm::enums::lua_type::lua_Type;

pub fn translate_inst_jumpx_eq_b_shortcut(
    build: &mut IrBuilder,
    pc: *const Instruction,
    pcpos: i32,
) {
    let rr = LUAU_INSN_A(unsafe { *(pc.add(2) as *const u32) });

    let ra = LUAU_INSN_A(unsafe { *(pc as *const u32) });
    let aux = unsafe { *(pc.add(1) as *const u32) };
    let not_ = LUAU_INSN_AUX_NOT(aux) != 0;

    let next = build.block_at_inst((pcpos + 4) as u32);

    let vm_reg_ra = build.vm_reg(ra as u8);
    let ta = build.inst_ir_cmd_ir_op(IrCmd::LOAD_TAG, vm_reg_ra);
    let va = build.inst_ir_cmd_ir_op(IrCmd::LOAD_INT, vm_reg_ra);
    let vb = build.const_int(LUAU_INSN_AUX_KB(aux) as i32);

    let const_tag_boolean = build.const_tag(lua_Type::LUA_TBOOLEAN as u8);
    let cond = if not_ {
        IrCondition::NotEqual
    } else {
        IrCondition::Equal
    };
    let cond_op = build.cond(cond);

    let result = build.inst_ir_cmd_ir_op_ir_op_ir_op_ir_op_ir_op(
        IrCmd::CMP_SPLIT_TVALUE,
        ta,
        const_tag_boolean,
        va,
        vb,
        cond_op,
    );

    let vm_reg_rr = build.vm_reg(rr as u8);
    let const_tag_boolean_2 = build.const_tag(lua_Type::LUA_TBOOLEAN as u8);
    build.inst_ir_cmd_ir_op_ir_op(IrCmd::STORE_TAG, vm_reg_rr, const_tag_boolean_2);
    build.inst_ir_cmd_ir_op_ir_op(IrCmd::STORE_INT, vm_reg_rr, result);
    build.inst_ir_cmd_ir_op(IrCmd::JUMP, next);

    if build.is_internal_block(next) {
        build.begin_block(next);
    }
}