luaur_code_gen/functions/
translate_builtin_vector_dot.rs

1use crate::enums::builtin_impl_type::BuiltinImplType;
2use crate::enums::ir_cmd::IrCmd;
3use crate::records::builtin_impl_result::BuiltinImplResult;
4use crate::records::ir_builder::IrBuilder;
5use crate::records::ir_op::IrOp;
6use luaur_vm::enums::lua_type::lua_Type;
7
8pub fn translate_builtin_vector_dot(
9    build: &mut IrBuilder,
10    nparams: i32,
11    ra: i32,
12    arg: i32,
13    args: IrOp,
14    arg3: IrOp,
15    nresults: i32,
16    pcpos: i32,
17) -> BuiltinImplResult {
18    let arg1 = build.vm_reg(arg as u8);
19
20    if nparams != 2
21        || nresults > 1
22        || arg1.kind() == crate::enums::ir_op_kind::IrOpKind::Constant
23        || args.kind() == crate::enums::ir_op_kind::IrOpKind::Constant
24    {
25        return BuiltinImplResult {
26            r#type: BuiltinImplType::None,
27            actual_result_count: -1,
28        };
29    }
30
31    let fallback = build.vm_exit(pcpos as u32);
32    build.load_and_check_tag(arg1, lua_Type::LUA_TVECTOR as u8, fallback);
33    build.load_and_check_tag(args, lua_Type::LUA_TVECTOR as u8, fallback);
34
35    let a = build.inst_ir_cmd_ir_op(IrCmd::LOAD_TVALUE, arg1);
36    let b = build.inst_ir_cmd_ir_op(IrCmd::LOAD_TVALUE, args);
37
38    let sum = build.inst_ir_cmd_ir_op_ir_op(IrCmd::DOT_VEC, a, b);
39    let sum = build.inst_ir_cmd_ir_op(IrCmd::FLOAT_TO_NUM, sum);
40
41    let ra_reg = build.vm_reg(ra as u8);
42    build.inst_ir_cmd_ir_op_ir_op(IrCmd::STORE_DOUBLE, ra_reg, sum);
43    let tag = build.const_tag(lua_Type::LUA_TNUMBER as u8);
44    build.inst_ir_cmd_ir_op_ir_op(IrCmd::STORE_TAG, ra_reg, tag);
45
46    let _ = arg3;
47
48    BuiltinImplResult {
49        r#type: BuiltinImplType::Full,
50        actual_result_count: 1,
51    }
52}
luaur_code_gen/functions/translate_builtin_vector_dot.rs

luaur_code_gen/functions/
translate_builtin_vector_dot.rs
