Skip to main content

luaur_vm/functions/
setnodevector.rs

1use crate::enums::lua_status::lua_Status;
2use crate::enums::lua_type::lua_Type;
3use crate::functions::lua_d_throw_ldo::luaD_throw;
4use crate::functions::lua_g_pusherror::lua_g_pusherror;
5use crate::macros::ceillog_2::ceillog2;
6use crate::macros::dummynode::dummynode;
7use crate::macros::lua_m_newarray::luaM_newarray;
8use crate::macros::setnilvalue::setnilvalue;
9use crate::records::lua_node::LuaNode;
10use crate::records::lua_table::LuaTable;
11use crate::type_aliases::lua_state::lua_State;
12use core::ffi::c_int;
13
14const MAXBITS: c_int = 26;
15
16unsafe fn runerror(l: *mut lua_State, msg: *const core::ffi::c_char) -> ! {
17    lua_g_pusherror(l, msg);
18    luaD_throw(l, lua_Status::LUA_ERRRUN as c_int);
19}
20
21#[allow(non_snake_case)]
22pub unsafe fn setnodevector(l: *mut lua_State, t: *mut LuaTable, mut size: c_int) {
23    let lsize: c_int;
24
25    if size == 0 {
26        (*t).node = dummynode as *mut LuaNode;
27        lsize = 0;
28    } else {
29        lsize = ceillog2(size as u32);
30        if lsize > MAXBITS {
31            runerror(l, c"table overflow".as_ptr());
32        }
33
34        size = 1 << lsize;
35        (*t).node = luaM_newarray!(l, size as usize, LuaNode, (*t).memcat);
36
37        let mut i = 0;
38        while i < size {
39            let n = (*t).node.add(i as usize);
40            (*n).key.set_next(0);
41            (*n).key.value = Default::default();
42            (*n).key.extra = [0];
43            (*n).key.set_tt(lua_Type::LUA_TNIL as i32);
44            setnilvalue!(core::ptr::addr_of_mut!((*n).val));
45            i += 1;
46        }
47    }
48
49    (*t).lsizenode = lsize as u8;
50    (*t).nodemask8 = ((1 << lsize) - 1) as u8;
51    (*t).union.lastfree = size;
52}