luaur_code_gen/functions/
translate_builtin_vector.rs

1use crate::enums::builtin_impl_type::BuiltinImplType;
2use crate::enums::ir_cmd::IrCmd;
3use crate::functions::builtin_check_double::builtin_check_double;
4use crate::functions::builtin_load_double::builtin_load_double;
5use crate::macros::codegen_assert::CODEGEN_ASSERT;
6use crate::records::builtin_impl_result::BuiltinImplResult;
7use crate::records::ir_builder::IrBuilder;
8use crate::records::ir_op::IrOp;
9use luaur_vm::macros::lua_vector_size::LUA_VECTOR_SIZE;
10
11pub fn translate_builtin_vector(
12    build: &mut IrBuilder,
13    nparams: i32,
14    ra: i32,
15    arg: i32,
16    args: IrOp,
17    arg3: IrOp,
18    nresults: i32,
19    pcpos: i32,
20) -> BuiltinImplResult {
21    if nparams < 2 || nresults > 1 {
22        return BuiltinImplResult {
23            r#type: BuiltinImplType::None,
24            actual_result_count: -1,
25        };
26    }
27
28    CODEGEN_ASSERT!(LUA_VECTOR_SIZE == 3);
29
30    let ra_op = build.vm_reg(ra as u8);
31
32    if nparams == 2 {
33        let arg_op = build.vm_reg(arg as u8);
34        builtin_check_double(build, arg_op, pcpos);
35        builtin_check_double(build, args, pcpos);
36
37        let x = builtin_load_double(build, arg_op);
38        let y = builtin_load_double(build, args);
39
40        let xf = build.inst_ir_cmd_ir_op(IrCmd::NUM_TO_FLOAT, x);
41        let yf = build.inst_ir_cmd_ir_op(IrCmd::NUM_TO_FLOAT, y);
42
43        let zero = build.const_double(0.0);
44        build.inst_ir_cmd_ir_op_ir_op_ir_op_ir_op(IrCmd::STORE_VECTOR, ra_op, xf, yf, zero);
45    } else {
46        let arg_op = build.vm_reg(arg as u8);
47        builtin_check_double(build, arg_op, pcpos);
48        builtin_check_double(build, args, pcpos);
49        builtin_check_double(build, arg3, pcpos);
50
51        let x = builtin_load_double(build, arg_op);
52        let y = builtin_load_double(build, args);
53        let z = builtin_load_double(build, arg3);
54
55        let xf = build.inst_ir_cmd_ir_op(IrCmd::NUM_TO_FLOAT, x);
56        let yf = build.inst_ir_cmd_ir_op(IrCmd::NUM_TO_FLOAT, y);
57        let zf = build.inst_ir_cmd_ir_op(IrCmd::NUM_TO_FLOAT, z);
58
59        build.inst_ir_cmd_ir_op_ir_op_ir_op_ir_op(IrCmd::STORE_VECTOR, ra_op, xf, yf, zf);
60    }
61
62    let tag = build.const_tag(luaur_vm::enums::lua_type::lua_Type::LUA_TVECTOR as u8);
63    build.inst_ir_cmd_ir_op_ir_op(IrCmd::STORE_TAG, ra_op, tag);
64
65    BuiltinImplResult {
66        r#type: BuiltinImplType::Full,
67        actual_result_count: 1,
68    }
69}
luaur_code_gen/functions/translate_builtin_vector.rs

luaur_code_gen/functions/
translate_builtin_vector.rs
