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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

use crate::enums::ir_block_kind::IrBlockKind;
use crate::enums::ir_op_kind::IrOpKind;
use crate::records::cfg_info::CfgInfo;
use crate::records::ir_block::IrBlock;
use crate::records::ir_function::IrFunction;
use crate::records::ir_inst::IrInst;
use crate::records::ir_op::IrOp;

pub fn compute_cfg_block_edges(function: &mut IrFunction) {
    let info: &mut CfgInfo = &mut function.cfg;

    // Clear existing data
    info.predecessors_offsets.clear();
    info.successors_offsets.clear();

    // Compute predecessors block edges
    info.predecessors_offsets.reserve(function.blocks.len());
    info.successors_offsets.reserve(function.blocks.len());

    let mut edge_count: u32 = 0;

    for block in &function.blocks {
        info.predecessors_offsets.push(edge_count);
        edge_count = edge_count.wrapping_add(block.use_count as u32);
    }

    info.predecessors.resize(edge_count as usize, 0);
    info.successors.resize(edge_count as usize, 0);

    edge_count = 0;

    for (block_idx, block) in function.blocks.iter().enumerate() {
        info.successors_offsets.push(edge_count);

        if block.kind == IrBlockKind::Dead {
            continue;
        }

        for inst_idx in block.start..=block.finish {
            let inst: &IrInst = &function.instructions[inst_idx as usize];

            let mut check_op = |op: &IrOp| {
                if op.kind() == IrOpKind::Block {
                    // Using predecessors list offset as write cursor position (adjusted back later)
                    let pred_pos = info.predecessors_offsets[op.index() as usize] as usize;
                    info.predecessors[pred_pos] = block_idx as u32;

                    info.predecessors_offsets[op.index() as usize] += 1;

                    info.successors[edge_count as usize] = op.index();
                    edge_count += 1;
                }
            };

            for op in &inst.ops {
                check_op(op);
            }
        }
    }

    // Offsets into the predecessor list were used as iterators in the previous loop
    // Adjust them back by subtracting block use count (predecessor count == block uses)
    for (block_idx, block) in function.blocks.iter().enumerate() {
        info.predecessors_offsets[block_idx] -= block.use_count as u32;
    }
}