luaur-code-gen 0.1.3

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

use crate::enums::abix_64::ABIX64;
use crate::enums::category_x_64::CategoryX64;
use crate::enums::size_x_64::SizeX64;
use crate::macros::codegen_assert::CODEGEN_ASSERT;
use crate::methods::ir_call_wrapper_x_64_get_next_argument_target::{
    systemv_gpr_order, windows_gpr_order, xmm_order,
};
use crate::records::assembly_builder_x_64::AssemblyBuilderX64;
use crate::records::ir_call_wrapper_x_64::IrCallWrapperX64;
use crate::records::register_x_64::RegisterX64;

impl IrCallWrapperX64 {
    // C++: `static RegisterX64 suggestArgumentRegister(SizeX64 size, AssemblyBuilderX64& build);`
    // (static template; instantiated for N = 0..3). No instance state is touched, so this is an
    // associated function in Rust as well -- callers invoke it as
    // `IrCallWrapperX64::suggest_argument_register::<N>(size, build)`.
    pub fn suggest_argument_register<const N: usize>(
        size: SizeX64,
        build: &mut AssemblyBuilderX64,
    ) -> RegisterX64 {
        // static_assert(N <= 3, "Argument index must be 0-3 (Windows passes args 4+ on the stack)");
        const { assert!(N <= 3) };

        if size == SizeX64::xmmword {
            return xmm_order()[N].base;
        }

        let gpr_order = if build.abi == ABIX64::Windows {
            windows_gpr_order()
        } else {
            systemv_gpr_order()
        };

        let mut target = gpr_order[N];
        CODEGEN_ASSERT!(target.cat == CategoryX64::reg);

        target.base = RegisterX64 {
            bits: (target.base.bits & RegisterX64::INDEX_MASK) | size as u8,
        };
        target.base
    }
}