Skip to main content

luaur_vm/functions/
luau_f_integerarshift.rs

1use crate::enums::lua_type::lua_Type;
2use crate::macros::lvalue::lvalue;
3use crate::macros::setlvalue::setlvalue;
4use crate::macros::ttisinteger::ttisinteger;
5use crate::type_aliases::lua_state::lua_State;
6use crate::type_aliases::stk_id::StkId;
7use crate::type_aliases::t_value::TValue;
8
9#[allow(non_snake_case)]
10pub unsafe fn luau_f_integerarshift(
11    _L: *mut lua_State,
12    res: StkId,
13    arg0: *mut TValue,
14    nresults: core::ffi::c_int,
15    args: StkId,
16    nparams: core::ffi::c_int,
17) -> core::ffi::c_int {
18    if nparams >= 2 && nresults <= 1 && ttisinteger!(arg0) && ttisinteger!(args) {
19        let n: i64 = lvalue!(arg0);
20        let i: i64 = lvalue!(args);
21
22        if i >= -63 && i <= 63 {
23            setlvalue!(
24                res,
25                if i < 0 {
26                    ((n as u64) << (-i)) as i64
27                } else {
28                    n >> i
29                }
30            );
31        } else if i < -63 {
32            setlvalue!(res, 0);
33        } else {
34            setlvalue!(res, if n < 0 { -1 } else { 0 });
35        }
36
37        1
38    } else {
39        -1
40    }
41}