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
31
32
33
34
35
36

use crate::records::compile_error::CompileError;
use crate::records::compiler::Compiler;
use crate::records::reg_scope::RegScope;
use luaur_ast::records::ast_expr_constant_integer::AstExprConstantInteger;
use luaur_ast::records::ast_expr_unary::AstExprUnary;
use luaur_ast::rtti;

impl Compiler {
    pub fn compile_expr_unary(&mut self, expr: *mut AstExprUnary, target: u8) {
        unsafe {
            let expr_ref = &*expr;
            let mut rs = self.reg_scope_compiler();

            if luaur_common::FFlag::LuauIntegerType2.get()
                && expr_ref.op == luaur_ast::records::ast_expr_unary::AstExprUnaryOp::Minus
            {
                let cint = rtti::ast_node_as::<AstExprConstantInteger>(expr_ref.expr as *mut _);
                if !cint.is_null() {
                    let cid =
                        (*self.bytecode).add_constant_integer((!((*cint).value as u64) + 1) as i64);
                    if cid < 0 {
                        CompileError::raise(
                            &expr_ref.base.base.location,
                            format_args!("Exceeded constant limit; simplify the code to compile"),
                        );
                    }
                    self.emit_load_k(target, cid);
                    return;
                }
            }

            let re = self.compile_expr_auto(expr_ref.expr, &mut rs);
            (*self.bytecode).emit_abc(self.get_unary_op(expr_ref.op), target, re, 0);
        }
    }
}