diff --git a/src/logic/scripting/lua/libs/libgui.cpp b/src/logic/scripting/lua/libs/libgui.cpp index 89c778c3..8a1f12db 100644 --- a/src/logic/scripting/lua/libs/libgui.cpp +++ b/src/logic/scripting/lua/libs/libgui.cpp @@ -14,6 +14,7 @@ #include "graphics/ui/elements/TextBox.hpp" #include "graphics/ui/elements/TrackBar.hpp" #include "graphics/ui/elements/InlineFrame.hpp" +#include "graphics/ui/elements/ModelViewer.hpp" #include "graphics/ui/gui_util.hpp" #include "graphics/ui/markdown.hpp" #include "graphics/core/Font.hpp" @@ -351,6 +352,8 @@ static int p_get_src(UINode* node, lua::State* L) { return lua::pushstring(L, image->getTexture()); } else if (auto iframe = dynamic_cast(node)) { return lua::pushstring(L, iframe->getSrc()); + } else if (auto modelviewer = dynamic_cast(node)) { + return lua::pushstring(L, modelviewer->getModel()); } return 0; } @@ -693,6 +696,8 @@ static void p_set_src(UINode* node, lua::State* L, int idx) { image->setTexture(lua::require_string(L, idx)); } else if (auto iframe = dynamic_cast(node)) { iframe->setSrc(lua::require_string(L, idx)); + } else if (auto modelviewer = dynamic_cast(node)) { + modelviewer->setModel(lua::require_string(L, idx)); } } static void p_set_region(UINode* node, lua::State* L, int idx) {