luaur-code-gen 0.1.0

Native (A64/X64) code generation for Luau (Rust).
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

use crate::enums::size_x_64::SizeX64;
use crate::functions::luau_reg_tag::luau_reg_tag;
use crate::functions::luau_reg_value::luau_reg_value;
use crate::records::assembly_builder_x_64::AssemblyBuilderX64;
use crate::records::ir_call_wrapper_x_64::IrCallWrapperX64;
use crate::records::ir_data::k_invalid_inst_idx;
use crate::records::ir_op::IrOp;
use crate::records::ir_reg_alloc_x_64::IrRegAllocX64;
use crate::records::native_context::NativeContext;
use crate::records::operand_x_64::OperandX64;
use crate::records::register_x_64::RegisterX64;
use luaur_vm::enums::lua_type::lua_Type;

const fn reg(index: u8, size: SizeX64) -> RegisterX64 {
    RegisterX64 {
        bits: (index << RegisterX64::INDEX_SHIFT) | size as u8,
    }
}

const R_NATIVE_CONTEXT: RegisterX64 = reg(13, SizeX64::qword);
const XMM0: RegisterX64 = reg(0, SizeX64::xmmword);
const XMM1: RegisterX64 = reg(1, SizeX64::xmmword);

fn s_temporary_slot() -> OperandX64 {
    OperandX64::mem(SizeX64::qword, RegisterX64::noreg, 1, RegisterX64::rsp, 0)
}

pub fn emit_builtin_math_modf(
    regs: &mut IrRegAllocX64,
    build: &mut AssemblyBuilderX64,
    ra: i32,
    arg: i32,
    nresults: i32,
) {
    let mut call_wrap = IrCallWrapperX64::ir_call_wrapper_x_64_ir_call_wrapper_x_64(
        regs,
        build,
        k_invalid_inst_idx,
    );
    call_wrap.add_argument_size_x_64_operand_x_64_ir_op(
        SizeX64::xmmword,
        luau_reg_value(arg),
        IrOp::ir_op(),
    );
    call_wrap.add_argument_size_x_64_operand_x_64_ir_op(
        SizeX64::qword,
        s_temporary_slot(),
        IrOp::ir_op(),
    );
    call_wrap.call(&OperandX64::mem(
        SizeX64::qword,
        RegisterX64::noreg,
        1,
        R_NATIVE_CONTEXT,
        core::mem::offset_of!(NativeContext, libm_modf) as i32,
    ));

    build.vmovsd_operand_x_64_operand_x_64(
        OperandX64::reg(XMM1),
        OperandX64::mem(SizeX64::qword, RegisterX64::noreg, 1, RegisterX64::rsp, 0),
    );
    build.vmovsd_operand_x_64_operand_x_64(luau_reg_value(ra), OperandX64::reg(XMM1));
    build.mov(
        luau_reg_tag(ra),
        OperandX64::imm(lua_Type::LUA_TNUMBER as i32),
    );

    if nresults > 1 {
        build.vmovsd_operand_x_64_operand_x_64(luau_reg_value(ra + 1), OperandX64::reg(XMM0));
        build.mov(
            luau_reg_tag(ra + 1),
            OperandX64::imm(lua_Type::LUA_TNUMBER as i32),
        );
    }
}