luaur_code_gen/functions/
translate_inst_get_table.rs

1use crate::enums::ir_cmd::IrCmd;
2use crate::functions::is_userdata_bytecode_type::is_userdata_bytecode_type;
3use crate::records::bytecode_types::BytecodeTypes;
4use crate::records::fallback_stream_scope::FallbackStreamScope;
5use crate::records::ir_builder::IrBuilder;
6use crate::type_aliases::instruction_ir_translation::Instruction;
7use luaur_common::enums::luau_bytecode_type::LuauBytecodeType;
8use luaur_common::macros::luau_insn_a::LUAU_INSN_A;
9use luaur_common::macros::luau_insn_b::LUAU_INSN_B;
10use luaur_common::macros::luau_insn_c::LUAU_INSN_C;
11use luaur_vm::enums::lua_type::lua_Type;
12
13pub fn translate_inst_get_table(build: &mut IrBuilder, pc: *const Instruction, pcpos: i32) {
14    let insn = unsafe { *pc };
15    let ra = LUAU_INSN_A(insn) as u8;
16    let rb = LUAU_INSN_B(insn) as u8;
17    let rc = LUAU_INSN_C(insn) as u8;
18
19    let bc_types = build.function.get_bytecode_types_at(pcpos);
20
21    if is_userdata_bytecode_type(bc_types.a)
22        || bc_types.b == LuauBytecodeType::LBC_TYPE_STRING.0 as u8
23    {
24        let savedpc_arg = build.const_uint((pcpos + 1) as u32);
25        build.inst_ir_cmd_ir_op(IrCmd::SET_SAVEDPC, savedpc_arg);
26        let reg_ra = build.vm_reg(ra);
27        let reg_rb = build.vm_reg(rb);
28        let reg_rc = build.vm_reg(rc);
29        build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::GET_TABLE, reg_ra, reg_rb, reg_rc);
30        return;
31    }
32
33    let fallback = build.fallback_block(pcpos as u32);
34
35    let reg_rb = build.vm_reg(rb);
36    let tb = build.inst_ir_cmd_ir_op(IrCmd::LOAD_TAG, reg_rb);
37    let const_tag_table = build.const_tag(lua_Type::LUA_TTABLE as u8);
38    let exit_or_fallback = if bc_types.a == LuauBytecodeType::LBC_TYPE_TABLE.0 as u8 {
39        build.vm_exit(pcpos as u32)
40    } else {
41        fallback
42    };
43    build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::CHECK_TAG, tb, const_tag_table, exit_or_fallback);
44
45    let reg_rc = build.vm_reg(rc);
46    let tc = build.inst_ir_cmd_ir_op(IrCmd::LOAD_TAG, reg_rc);
47    let const_tag_number = build.const_tag(lua_Type::LUA_TNUMBER as u8);
48    let exit_or_fallback_c = if bc_types.b == LuauBytecodeType::LBC_TYPE_NUMBER.0 as u8 {
49        build.vm_exit(pcpos as u32)
50    } else {
51        fallback
52    };
53    build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::CHECK_TAG, tc, const_tag_number, exit_or_fallback_c);
54
55    let vb = build.inst_ir_cmd_ir_op(IrCmd::LOAD_POINTER, reg_rb);
56    let vc = build.inst_ir_cmd_ir_op(IrCmd::LOAD_DOUBLE, reg_rc);
57
58    let index = build.inst_ir_cmd_ir_op_ir_op(IrCmd::TRY_NUM_TO_INDEX, vc, fallback);
59
60    let const_int_one = build.const_int(1);
61    let index = build.inst_ir_cmd_ir_op_ir_op(IrCmd::SUB_INT, index, const_int_one);
62
63    build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::CHECK_ARRAY_SIZE, vb, index, fallback);
64    build.inst_ir_cmd_ir_op_ir_op(IrCmd::CHECK_NO_METATABLE, vb, fallback);
65
66    let arr_el = build.inst_ir_cmd_ir_op_ir_op(IrCmd::GET_ARR_ADDR, vb, index);
67
68    let arr_el_tval = build.inst_ir_cmd_ir_op(IrCmd::LOAD_TVALUE, arr_el);
69    let reg_ra = build.vm_reg(ra);
70    build.inst_ir_cmd_ir_op_ir_op(IrCmd::STORE_TVALUE, reg_ra, arr_el_tval);
71
72    let next = build.block_at_inst((pcpos + 1) as u32);
73
74    let mut scope = FallbackStreamScope::new(build, fallback, next);
75    let build = &mut *scope.build;
76
77    let savedpc_arg = build.const_uint((pcpos + 1) as u32);
78    build.inst_ir_cmd_ir_op(IrCmd::SET_SAVEDPC, savedpc_arg);
79    let reg_ra = build.vm_reg(ra);
80    let reg_rb = build.vm_reg(rb);
81    let reg_rc = build.vm_reg(rc);
82    build.inst_ir_cmd_ir_op_ir_op_ir_op(IrCmd::GET_TABLE, reg_ra, reg_rb, reg_rc);
83    build.inst_ir_cmd_ir_op(IrCmd::JUMP, next);
84}
luaur_code_gen/functions/translate_inst_get_table.rs

luaur_code_gen/functions/
translate_inst_get_table.rs
