Skip to main content

luaur_vm/functions/
vector_normalize.rs

1use crate::functions::lua_l_checkvector::lua_l_checkvector;
2use crate::functions::lua_pushvector_lapi::lua_pushvector_lua_state_f32_f32_f32_f32;
3use crate::functions::lua_pushvector_lapi_alt_b::lua_pushvector_lua_state_f32_f32_f32;
4use crate::macros::lua_vector_size::LUA_VECTOR_SIZE;
5use crate::type_aliases::lua_state::lua_State;
6
7#[allow(non_snake_case)]
8pub unsafe fn vector_normalize(L: *mut lua_State) -> core::ffi::c_int {
9    let v = lua_l_checkvector(L, 1);
10
11    if LUA_VECTOR_SIZE == 4 {
12        let v0 = *v.offset(0);
13        let v1 = *v.offset(1);
14        let v2 = *v.offset(2);
15        let v3 = *v.offset(3);
16
17        let inv_sqrt = 1.0f32 / (v0 * v0 + v1 * v1 + v2 * v2 + v3 * v3).sqrt();
18        lua_pushvector_lua_state_f32_f32_f32_f32(
19            L,
20            v0 * inv_sqrt,
21            v1 * inv_sqrt,
22            v2 * inv_sqrt,
23            v3 * inv_sqrt,
24        );
25    } else {
26        let v0 = *v.offset(0);
27        let v1 = *v.offset(1);
28        let v2 = *v.offset(2);
29
30        let inv_sqrt = 1.0f32 / (v0 * v0 + v1 * v1 + v2 * v2).sqrt();
31        lua_pushvector_lua_state_f32_f32_f32(L, v0 * inv_sqrt, v1 * inv_sqrt, v2 * inv_sqrt);
32    }
33
34    1
35}