399 lines
12 KiB
C++
399 lines
12 KiB
C++
#include "lua_commons.h"
|
|
#include "api_lua.h"
|
|
|
|
#include <iostream>
|
|
#include "../scripting.h"
|
|
#include "lua_util.h"
|
|
#include "LuaState.h"
|
|
|
|
#include "../../../engine.h"
|
|
#include "../../../assets/Assets.h"
|
|
#include "../../../graphics/ui/gui_util.hpp"
|
|
#include "../../../graphics/ui/elements/UINode.hpp"
|
|
#include "../../../graphics/ui/elements/Button.hpp"
|
|
#include "../../../graphics/ui/elements/CheckBox.hpp"
|
|
#include "../../../graphics/ui/elements/TextBox.hpp"
|
|
#include "../../../graphics/ui/elements/TrackBar.hpp"
|
|
#include "../../../graphics/ui/elements/Panel.hpp"
|
|
#include "../../../graphics/ui/elements/Menu.hpp"
|
|
#include "../../../frontend/UiDocument.h"
|
|
#include "../../../frontend/locale/langs.h"
|
|
#include "../../../util/stringutil.h"
|
|
|
|
struct DocumentNode {
|
|
UiDocument* document;
|
|
gui::UINode* node;
|
|
};
|
|
|
|
static DocumentNode getDocumentNode(lua_State* L, const std::string& name, const std::string& nodeName) {
|
|
auto doc = scripting::engine->getAssets()->getLayout(name);
|
|
if (doc == nullptr) {
|
|
luaL_error(L, "document '%s' not found", name.c_str());
|
|
}
|
|
auto node = doc->get(nodeName);
|
|
if (node == nullptr) {
|
|
luaL_error(L, "document '%s' has no element with id '%s'", name.c_str(), nodeName.c_str());
|
|
}
|
|
return {doc, node.get()};
|
|
}
|
|
|
|
static bool getattr(lua_State* L, gui::TrackBar* bar, const std::string& attr) {
|
|
if (bar == nullptr)
|
|
return false;
|
|
if (attr == "value") {
|
|
lua_pushnumber(L, bar->getValue()); return true;
|
|
} else if (attr == "min") {
|
|
lua_pushnumber(L, bar->getMin()); return true;
|
|
} else if (attr == "max") {
|
|
lua_pushnumber(L, bar->getMax());
|
|
return true;
|
|
} else if (attr == "step") {
|
|
lua_pushnumber(L, bar->getStep());
|
|
return true;
|
|
} else if (attr == "trackWidth") {
|
|
lua_pushnumber(L, bar->getTrackWidth());
|
|
return true;
|
|
} else if (attr == "trackColor") {
|
|
return lua::pushcolor_arr(L, bar->getTrackColor());
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static bool setattr(lua_State* L, gui::TrackBar* bar, const std::string& attr) {
|
|
if (bar == nullptr)
|
|
return false;
|
|
if (attr == "value") {
|
|
bar->setValue(lua_tonumber(L, 4));
|
|
return true;
|
|
} else if (attr == "min") {
|
|
bar->setMin(lua_tonumber(L, 4));
|
|
return true;
|
|
} else if (attr == "max") {
|
|
bar->setMax(lua_tonumber(L, 4));
|
|
return true;
|
|
} else if (attr == "step") {
|
|
bar->setStep(lua_tonumber(L, 4));
|
|
return true;
|
|
} else if (attr == "trackWidth") {
|
|
bar->setTrackWidth(lua_tonumber(L, 4));
|
|
return true;
|
|
} else if (attr == "trackColor") {
|
|
bar->setTrackColor(lua::tocolor(L, 4));
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static bool getattr(lua_State* L, gui::Button* button, const std::string& attr) {
|
|
if (button == nullptr)
|
|
return false;
|
|
if (attr == "text") {
|
|
lua_pushstring(L, util::wstr2str_utf8(button->getText()).c_str());
|
|
return true;
|
|
} else if (attr == "pressedColor") {
|
|
return lua::pushcolor_arr(L, button->getPressedColor());
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static bool getattr(lua_State* L, gui::Label* label, const std::string& attr) {
|
|
if (label == nullptr)
|
|
return false;
|
|
if (attr == "text") {
|
|
lua_pushstring(L, util::wstr2str_utf8(label->getText()).c_str());
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static bool getattr(lua_State* L, gui::FullCheckBox* box, const std::string& attr) {
|
|
if (box == nullptr)
|
|
return false;
|
|
if (attr == "checked") {
|
|
lua_pushboolean(L, box->isChecked());
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static bool getattr(lua_State* L, gui::TextBox* box, const std::string& attr) {
|
|
if (box == nullptr)
|
|
return false;
|
|
if (attr == "text") {
|
|
lua_pushstring(L, util::wstr2str_utf8(box->getText()).c_str());
|
|
return true;
|
|
} else if (attr == "placeholder") {
|
|
lua_pushstring(L, util::wstr2str_utf8(box->getPlaceholder()).c_str());
|
|
return true;
|
|
} else if (attr == "valid") {
|
|
lua_pushboolean(L, box->validate());
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static DocumentNode getDocumentNode(lua_State* L) {
|
|
lua_getfield(L, 1, "docname");
|
|
lua_getfield(L, 1, "name");
|
|
auto docname = lua_tostring(L, -2);
|
|
auto name = lua_tostring(L, -1);
|
|
auto node = getDocumentNode(L, docname, name);
|
|
lua_pop(L, 2);
|
|
return node;
|
|
}
|
|
|
|
static int menu_back(lua_State* L) {
|
|
auto node = getDocumentNode(L);
|
|
auto menu = dynamic_cast<gui::Menu*>(node.node);
|
|
menu->back();
|
|
return 0;
|
|
}
|
|
|
|
static int menu_reset(lua_State* L) {
|
|
auto node = getDocumentNode(L);
|
|
auto menu = dynamic_cast<gui::Menu*>(node.node);
|
|
menu->reset();
|
|
return 0;
|
|
}
|
|
|
|
static bool getattr(lua_State* L, gui::Menu* menu, const std::string& attr) {
|
|
if (menu == nullptr)
|
|
return false;
|
|
if (attr == "page") {
|
|
lua_pushstring(L, menu->getCurrent().name.c_str());
|
|
return true;
|
|
} else if (attr == "back") {
|
|
lua_pushcfunction(L, menu_back);
|
|
return true;
|
|
} else if (attr == "reset") {
|
|
lua_pushcfunction(L, menu_reset);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static bool setattr(lua_State* L, gui::FullCheckBox* box, const std::string& attr) {
|
|
if (box == nullptr)
|
|
return false;
|
|
if (attr == "checked") {
|
|
box->setChecked(lua_toboolean(L, 4));
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static bool setattr(lua_State* L, gui::Button* button, const std::string& attr) {
|
|
if (button == nullptr)
|
|
return false;
|
|
if (attr == "text") {
|
|
button->setText(util::str2wstr_utf8(lua_tostring(L, 4)));
|
|
return true;
|
|
} else if (attr == "pressedColor") {
|
|
button->setPressedColor(lua::tocolor(L, 4));
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static bool setattr(lua_State* L, gui::TextBox* box, const std::string& attr) {
|
|
if (box == nullptr)
|
|
return false;
|
|
if (attr == "text") {
|
|
box->setText(util::str2wstr_utf8(lua_tostring(L, 4)));
|
|
return true;
|
|
} else if (attr == "placeholder") {
|
|
box->setPlaceholder(util::str2wstr_utf8(lua_tostring(L, 4)));
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static bool setattr(lua_State* L, gui::Label* label, const std::string& attr) {
|
|
if (label == nullptr)
|
|
return false;
|
|
if (attr == "text") {
|
|
label->setText(util::str2wstr_utf8(lua_tostring(L, 4)));
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static bool setattr(lua_State* L, gui::Menu* menu, const std::string& attr) {
|
|
if (menu == nullptr)
|
|
return false;
|
|
if (attr == "page") {
|
|
auto page = lua_tostring(L, 4);
|
|
menu->setPage(page);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static int container_add(lua_State* L) {
|
|
auto docnode = getDocumentNode(L);
|
|
auto node = dynamic_cast<gui::Container*>(docnode.node);
|
|
auto xmlsrc = lua_tostring(L, 2);
|
|
try {
|
|
auto subnode = guiutil::create(xmlsrc, docnode.document->getEnvironment());
|
|
node->add(subnode);
|
|
gui::UINode::getIndices(subnode, docnode.document->getMapWriteable());
|
|
} catch (const std::exception& err) {
|
|
luaL_error(L, err.what());
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static bool getattr(lua_State* L, gui::Container* container, const std::string& attr) {
|
|
if (container == nullptr)
|
|
return false;
|
|
|
|
if (attr == "add") {
|
|
lua_pushcfunction(L, container_add);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static int l_gui_getattr(lua_State* L) {
|
|
auto docname = lua_tostring(L, 1);
|
|
auto element = lua_tostring(L, 2);
|
|
const std::string attr = lua_tostring(L, 3);
|
|
auto docnode = getDocumentNode(L, docname, element);
|
|
auto node = docnode.node;
|
|
|
|
if (attr == "color") {
|
|
return lua::pushcolor_arr(L, node->getColor());
|
|
} else if (attr == "pos") {
|
|
return lua::pushvec2_arr(L, node->getPos());
|
|
} else if (attr == "size") {
|
|
return lua::pushvec2_arr(L, node->getSize());
|
|
} else if (attr == "hoverColor") {
|
|
return lua::pushcolor_arr(L, node->getHoverColor());
|
|
} else if (attr == "interactive") {
|
|
lua_pushboolean(L, node->isInteractive());
|
|
return 1;
|
|
} else if (attr == "visible") {
|
|
lua_pushboolean(L, node->isVisible());
|
|
return 1;
|
|
}
|
|
|
|
if (getattr(L, dynamic_cast<gui::Container*>(node), attr))
|
|
return 1;
|
|
if (getattr(L, dynamic_cast<gui::Button*>(node), attr))
|
|
return 1;
|
|
if (getattr(L, dynamic_cast<gui::Label*>(node), attr))
|
|
return 1;
|
|
if (getattr(L, dynamic_cast<gui::TextBox*>(node), attr))
|
|
return 1;
|
|
if (getattr(L, dynamic_cast<gui::TrackBar*>(node), attr))
|
|
return 1;
|
|
if (getattr(L, dynamic_cast<gui::FullCheckBox*>(node), attr))
|
|
return 1;
|
|
if (getattr(L, dynamic_cast<gui::Menu*>(node), attr))
|
|
return 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int l_gui_getviewport(lua_State* L) {
|
|
lua::pushvec2_arr(L, scripting::engine->getGUI()->getContainer()->getSize());
|
|
return 1;
|
|
}
|
|
|
|
static int l_gui_setattr(lua_State* L) {
|
|
auto docname = lua_tostring(L, 1);
|
|
auto element = lua_tostring(L, 2);
|
|
const std::string attr = lua_tostring(L, 3);
|
|
|
|
auto docnode = getDocumentNode(L, docname, element);
|
|
auto node = docnode.node;
|
|
if (attr == "pos") {
|
|
node->setPos(lua::tovec2(L, 4));
|
|
} else if (attr == "size") {
|
|
node->setSize(lua::tovec2(L, 4));
|
|
} else if (attr == "color") {
|
|
node->setColor(lua::tocolor(L, 4));
|
|
} else if (attr == "hoverColor") {
|
|
node->setHoverColor(lua::tocolor(L, 4));
|
|
} else if (attr == "interactive") {
|
|
node->setInteractive(lua_toboolean(L, 4));
|
|
} else if (attr == "visible") {
|
|
node->setVisible(lua_toboolean(L, 4));
|
|
} else {
|
|
if (setattr(L, dynamic_cast<gui::Button*>(node), attr))
|
|
return 0;
|
|
if (setattr(L, dynamic_cast<gui::Label*>(node), attr))
|
|
return 0;
|
|
if (setattr(L, dynamic_cast<gui::TextBox*>(node), attr))
|
|
return 0;
|
|
if (setattr(L, dynamic_cast<gui::TrackBar*>(node), attr))
|
|
return 0;
|
|
if (setattr(L, dynamic_cast<gui::FullCheckBox*>(node), attr))
|
|
return 0;
|
|
if (setattr(L, dynamic_cast<gui::Menu*>(node), attr))
|
|
return 0;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int l_gui_get_env(lua_State* L) {
|
|
auto name = lua_tostring(L, 1);
|
|
auto doc = scripting::engine->getAssets()->getLayout(name);
|
|
if (doc == nullptr) {
|
|
luaL_error(L, "document '%s' not found", name);
|
|
}
|
|
lua_getglobal(L, lua::LuaState::envName(*doc->getEnvironment()).c_str());
|
|
return 1;
|
|
}
|
|
|
|
static int l_gui_str(lua_State* L) {
|
|
auto text = util::str2wstr_utf8(lua_tostring(L, 1));
|
|
if (!lua_isnoneornil(L, 2)) {
|
|
auto context = util::str2wstr_utf8(lua_tostring(L, 2));
|
|
lua_pushstring(L, util::wstr2str_utf8(langs::get(text, context)).c_str());
|
|
} else {
|
|
lua_pushstring(L, util::wstr2str_utf8(langs::get(text)).c_str());
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
static int l_gui_reindex(lua_State* L) {
|
|
auto name = lua_tostring(L, 1);
|
|
auto doc = scripting::engine->getAssets()->getLayout(name);
|
|
if (doc == nullptr) {
|
|
luaL_error(L, "document '%s' not found", name);
|
|
}
|
|
doc->rebuildIndices();
|
|
return 0;
|
|
}
|
|
|
|
/// @brief gui.get_locales_info() -> table of tables
|
|
static int l_gui_get_locales_info(lua_State* L) {
|
|
auto& locales = langs::locales_info;
|
|
lua_createtable(L, 0, locales.size());
|
|
for (auto& entry : locales) {
|
|
lua_createtable(L, 0, 1);
|
|
lua_pushstring(L, entry.second.name.c_str());
|
|
lua_setfield(L, -2, "name");
|
|
lua_setfield(L, -2, entry.first.c_str());
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
/// @brief gui.get_locale() -> string
|
|
static int l_gui_get_locale(lua_State* L) {
|
|
lua_pushstring(L, langs::current->getId().c_str());
|
|
return 1;
|
|
}
|
|
|
|
const luaL_Reg guilib [] = {
|
|
{"get_viewport", lua_wrap_errors<l_gui_getviewport>},
|
|
{"getattr", lua_wrap_errors<l_gui_getattr>},
|
|
{"setattr", lua_wrap_errors<l_gui_setattr>},
|
|
{"get_env", lua_wrap_errors<l_gui_get_env>},
|
|
{"str", lua_wrap_errors<l_gui_str>},
|
|
{"reindex", lua_wrap_errors<l_gui_reindex>},
|
|
{"get_locale", lua_wrap_errors<l_gui_get_locale>},
|
|
{"get_locales_info", lua_wrap_errors<l_gui_get_locales_info>},
|
|
{NULL, NULL}
|
|
};
|