diff --git a/src/logic/scripting/lua/api_lua.hpp b/src/logic/scripting/lua/api_lua.hpp index 83130ec2..592340c3 100644 --- a/src/logic/scripting/lua/api_lua.hpp +++ b/src/logic/scripting/lua/api_lua.hpp @@ -22,6 +22,9 @@ extern const luaL_Reg packlib []; extern const luaL_Reg playerlib []; extern const luaL_Reg timelib []; extern const luaL_Reg tomllib []; +extern const luaL_Reg vec2lib []; +extern const luaL_Reg vec3lib []; +extern const luaL_Reg vec4lib []; extern const luaL_Reg worldlib []; // Lua Overrides diff --git a/src/logic/scripting/lua/libmat4.cpp b/src/logic/scripting/lua/libmat4.cpp index 16cc2098..17ab752b 100644 --- a/src/logic/scripting/lua/libmat4.cpp +++ b/src/logic/scripting/lua/libmat4.cpp @@ -60,9 +60,9 @@ static int l_mul(lua::State* L) { } case 3: { if (len2 == 4) { - return lua::setvec4(L, 3, matrix1 * lua::tovec4(L, 2)); + return lua::setvec(L, 3, matrix1 * lua::tovec4(L, 2)); } else if (len2 == 3) { - return lua::setvec3(L, 3, matrix1 * glm::vec4(lua::tovec3(L, 2), 1.0f)); + return lua::setvec(L, 3, matrix1 * glm::vec4(lua::tovec3(L, 2), 1.0f)); } return lua::setmat4(L, 3, matrix1 * lua::tomat4(L, 2)); } diff --git a/src/logic/scripting/lua/libvecn.cpp b/src/logic/scripting/lua/libvecn.cpp new file mode 100644 index 00000000..62f4b163 --- /dev/null +++ b/src/logic/scripting/lua/libvecn.cpp @@ -0,0 +1,43 @@ +#include "api_lua.hpp" + +#include + +template< + uint n, + glm::vec(*tofunc)(lua::State*, int), + int(*setfunc)(lua::State*, int, glm::vec) +> +static int l_add(lua::State* L) { + uint argc = lua::gettop(L); + auto a = tofunc(L, 1); + auto b = tofunc(L, 2); + switch (argc) { + case 2: + lua::createtable(L, n, 0); + for (uint i = 0; i < n; i++) { + lua::pushnumber(L, a[i]+b[i]); + lua::rawseti(L, i+1); + } + return 1; + case 3: + return setfunc(L, 3, a + b); + default: { + throw std::runtime_error("invalid arguments number (2 or 3 expected)"); + } + } +} + +const luaL_Reg vec2lib [] = { + {"add", lua::wrap>>}, + {NULL, NULL} +}; + +const luaL_Reg vec3lib [] = { + {"add", lua::wrap>>}, + {NULL, NULL} +}; + +const luaL_Reg vec4lib [] = { + {"add", lua::wrap>>}, + {NULL, NULL} +}; diff --git a/src/logic/scripting/lua/lua_engine.cpp b/src/logic/scripting/lua/lua_engine.cpp index a64216a6..92e59285 100644 --- a/src/logic/scripting/lua/lua_engine.cpp +++ b/src/logic/scripting/lua/lua_engine.cpp @@ -42,6 +42,9 @@ static void create_libs(lua::State* L) { openlib(L, "player", playerlib); openlib(L, "time", timelib); openlib(L, "toml", tomllib); + openlib(L, "vec2", vec2lib); + openlib(L, "vec3", vec3lib); + openlib(L, "vec4", vec4lib); openlib(L, "world", worldlib); addfunc(L, "print", lua::wrap); diff --git a/src/logic/scripting/lua/lua_util.hpp b/src/logic/scripting/lua/lua_util.hpp index a774d283..308ef83e 100644 --- a/src/logic/scripting/lua/lua_util.hpp +++ b/src/logic/scripting/lua/lua_util.hpp @@ -221,19 +221,10 @@ namespace lua { } return 1; } - /// @brief pushes vector table to the stack and updates it with glm vec4 - inline int setvec4(lua::State* L, int idx, glm::vec4 vec) { + template + inline int setvec(lua::State* L, int idx, glm::vec vec) { pushvalue(L, idx); - for (uint i = 0; i < 4; i++) { - pushnumber(L, vec[i]); - rawseti(L, i+1); - } - return 1; - } - /// @brief pushes vector table to the stack and updates it with glm vec3 - inline int setvec3(lua::State* L, int idx, glm::vec3 vec) { - pushvalue(L, idx); - for (uint i = 0; i < 3; i++) { + for (int i = 0; i < n; i++) { pushnumber(L, vec[i]); rawseti(L, i+1); }