Lua: Add Vector:set(x, y, z)

praydog · Oct 18, 2024 · 9bc4513 · 9bc4513
1 parent 18108c9
commit 9bc4513
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/lua-api/lib/src/datatypes/Quaternion.cpp b/lua-api/lib/src/datatypes/Quaternion.cpp
@@ -4,6 +4,7 @@ namespace lua::datatypes {
     void bind_quaternions(sol::state_view& lua) {
         #define BIND_QUATERNION_LIKE(name, datatype) \
             lua.new_usertype<name>(#name, \
+                "set", [](name& v, datatype x, datatype y, datatype z, datatype w) { v.x = x; v.y = y; v.z = z; v.w = w; }, \
                 "x", &name::x, \
                 "y", &name::y, \
                 "z", &name::z, \

diff --git a/lua-api/lib/src/datatypes/Vector.cpp b/lua-api/lib/src/datatypes/Vector.cpp
@@ -4,6 +4,11 @@ namespace lua::datatypes {
     void bind_vectors(sol::state_view& lua) {
         #define BIND_VECTOR3_LIKE(name, datatype) \
             lua.new_usertype<name>(#name, \
+                "set", [](sol::object o, datatype x, datatype y, datatype z, datatype w) -> sol::object { \
+                    name& v = o.as<name>(); \
+                    v.x = x; v.y = y; v.z = z; \
+                    return o; \
+                }, \
                 "clone", [](name& v) -> name { return v; }, \
                 "x", &name::x, \
                 "y", &name::y, \
@@ -38,7 +43,12 @@ namespace lua::datatypes {
         BIND_VECTOR3_LIKE_END();
 
         #define BIND_VECTOR4_LIKE(name, datatype) \
-        lua.new_usertype<name>(#name, \
+            lua.new_usertype<name>(#name, \
+                "set", [](sol::object o, datatype x, datatype y, datatype z, datatype w) -> sol::object { \
+                    name& v = o.as<name>(); \
+                    v.x = x; v.y = y; v.z = z; v.w = w; \
+                    return o; \
+                }, \
                 "clone", [](name& v) -> name { return v; }, \
                 "x", &name::x, \
                 "y", &name::y, \