luaur-compiler 0.1.2

Luau source-to-bytecode compiler (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

//! Node: `cxx:Function:Luau.Compiler:Compiler/src/ValueTracking.cpp:104:trackValues`
//!
//! `trackValues` — run a `ValueVisitor` over the AST root to record, for every
//! local, its initializer and whether it is ever reassigned (and which globals
//! are written). The C++ visitor holds references to `globals`/`variables`; the
//! Rust `ValueVisitor` owns them, so the constructor moves the maps in and we
//! move the populated results back out after the walk.

use crate::enums::global::Global;
use crate::records::value_visitor::ValueVisitor;
use crate::records::variable::Variable;
use luaur_ast::records::ast_local::AstLocal;
use luaur_ast::records::ast_name::AstName;
use luaur_ast::records::ast_node::AstNode;
use luaur_common::records::dense_hash_map::DenseHashMap;

pub fn track_values(
    globals: &mut DenseHashMap<AstName, Global>,
    variables: &mut DenseHashMap<*mut AstLocal, Variable>,
    root: *mut AstNode,
) {
    let mut visitor = ValueVisitor::value_visitor(globals, variables);

    unsafe {
        luaur_ast::visit::dispatch_node(root, &mut visitor);
    }

    *globals = visitor.globals;
    *variables = visitor.variables;
}