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

extern crate alloc;

use crate::functions::append::append;
use crate::functions::append_blocks::append_blocks;
use crate::functions::successors::successors;
use crate::records::ir_function::IrFunction;
use crate::records::ir_to_string_context::ir_to_string_context;
use alloc::string::String;
use core::ffi::c_void;

pub fn to_dot_cfg(function: &IrFunction) -> String {
    let mut result = String::new();

    {
        let mut ctx = ir_to_string_context {
            result: &mut result,
            blocks: &function.blocks,
            constants: &function.constants,
            cfg: &function.cfg,
            vm_exit_info: &function.vm_exit_info,
            proto: function.proto as *mut c_void,
        };

        ctx.result.push_str("digraph CFG {\n");
        ctx.result.push_str("node[shape=record]\n");

        append_blocks(
            &mut ctx, function, /* include_inst */ false, /* include_in */ false,
            /* include_out */ false, /* include_def */ true,
        );

        let cfg = ctx.cfg;

        let mut i = 0usize;
        while i < function.blocks.len() && i < cfg.successors_offsets.len() {
            let succ = successors(cfg, i as u32);

            for target in succ {
                append(
                    &mut ctx.result,
                    format_args!("b{} -> b{};\n", i as u32, target),
                );
            }

            i += 1;
        }

        ctx.result.push_str("}\n");
    }

    result
}