mirror of
https://github.com/MaSzyna-EU07/maszyna.git
synced 2026-07-18 00:49:19 +02:00
3a600334a7
We could have done away with it just because the memcell values are always the last parameter. However, this might not always be the case in the future. Also, it's better for consistency reasons.
373 lines
9.0 KiB
C++
373 lines
9.0 KiB
C++
#include "stdafx.h"
|
|
#include "scripting/lua.h"
|
|
#include "world/Event.h"
|
|
#include "utilities/Logs.h"
|
|
#include "world/MemCell.h"
|
|
#include "vehicle/Driver.h"
|
|
#include "simulation/simulation.h"
|
|
|
|
lua::lua()
|
|
{
|
|
state = luaL_newstate();
|
|
if (!state)
|
|
throw std::runtime_error("cannot create lua state");
|
|
lua_atpanic(state, atpanic);
|
|
luaL_openlibs(state);
|
|
|
|
static constexpr luaL_Reg api[] =
|
|
{
|
|
{"event_create", scriptapi_event_create},
|
|
{"event_find", scriptapi_event_find},
|
|
{"event_exists", scriptapi_event_exists},
|
|
{"event_getname", scriptapi_event_getname},
|
|
{"event_dispatch", scriptapi_event_dispatch},
|
|
{"event_dispatch_n", scriptapi_event_dispatch_n},
|
|
{"track_find", scriptapi_track_find},
|
|
{"track_isoccupied", scriptapi_track_isoccupied},
|
|
{"track_isoccupied_n", scriptapi_track_isoccupied_n},
|
|
{"isolated_find", scriptapi_isolated_find},
|
|
{"isolated_isoccupied", scriptapi_isolated_isoccupied},
|
|
{"isolated_isoccupied_n", scriptapi_isolated_isoccupied_n},
|
|
{"train_getname", scriptapi_train_getname},
|
|
{"dynobj_putvalues", scriptapi_dynobj_putvalues},
|
|
{"memcell_find", scriptapi_memcell_find},
|
|
{"memcell_read", scriptapi_memcell_read},
|
|
{"memcell_read_n", scriptapi_memcell_read_n},
|
|
{"memcell_update", scriptapi_memcell_update},
|
|
{"memcell_update_n", scriptapi_memcell_update_n},
|
|
{"random", scriptapi_random},
|
|
{"writelog", scriptapi_writelog},
|
|
{"writeerrorlog", scriptapi_writeerrorlog},
|
|
{nullptr, nullptr}
|
|
};
|
|
luaL_register(state, "eu07.events", api);
|
|
}
|
|
|
|
lua::~lua()
|
|
{
|
|
lua_close(state);
|
|
state = nullptr;
|
|
}
|
|
|
|
std::string lua::get_error() const
|
|
{
|
|
return lua_tostring(state, -1);
|
|
}
|
|
|
|
void lua::interpret(const std::string& file) const
|
|
{
|
|
if (luaL_dofile(state, file.c_str())) {
|
|
const char *str = lua_tostring(state, -1);
|
|
ErrorLog(std::string(str), logtype::lua);
|
|
}
|
|
}
|
|
|
|
int lua::atpanic(lua_State *s)
|
|
{
|
|
std::string err = lua_tostring(s, 1);
|
|
ErrorLog(err, logtype::lua);
|
|
return 0;
|
|
}
|
|
|
|
/// Dereferences the function handler from the Lua registry.
|
|
void lua::unref(lua_State *L, const int ref)
|
|
{
|
|
luaL_unref(L, LUA_REGISTRYINDEX, ref);
|
|
}
|
|
|
|
/// Executes the callback function of an event created with `scriptapi_event_create`.
|
|
void lua::dispatch_event(lua_State *L, const int handler, basic_event *event, const TDynamicObject *activator)
|
|
{
|
|
lua_rawgeti(L, LUA_REGISTRYINDEX, handler);
|
|
lua_pushlightuserdata(L, event);
|
|
lua_pushlightuserdata(L, const_cast<TDynamicObject *>(activator));
|
|
lua_call(L, 2, 0);
|
|
}
|
|
|
|
void lua::push_memcell_values(lua_State *L, const TMemCell *mc)
|
|
{
|
|
lua_createtable(L, 0, 3);
|
|
lua_pushstring(L, "str");
|
|
lua_pushstring(L, mc ? mc->Text().c_str() : nullptr);
|
|
lua_settable(L, -3);
|
|
lua_pushstring(L, "num1");
|
|
lua_pushnumber(L, mc ? mc->Value1() : 0.0);
|
|
lua_settable(L, -3);
|
|
lua_pushstring(L, "num2");
|
|
lua_pushnumber(L, mc ? mc->Value2() : 0.0);
|
|
lua_settable(L, -3);
|
|
}
|
|
|
|
memcell_values lua::get_memcell_values(lua_State *L, int idx)
|
|
{
|
|
// For negative indices, we need to account for the extra item that is the key name/value extracted from the table.
|
|
if (idx < 0)
|
|
idx--;
|
|
memcell_values mc{nullptr, 0.0, 0.0};
|
|
lua_pushstring(L, "str");
|
|
lua_gettable(L, idx);
|
|
if (lua_isstring(L, -1))
|
|
mc.str = lua_tostring(L, -1);
|
|
lua_pop(L, 1);
|
|
lua_pushstring(L, "num1");
|
|
lua_gettable(L, idx);
|
|
if (lua_isnumber(L, -1))
|
|
mc.num1 = lua_tonumber(L, -1);
|
|
lua_pop(L, 1);
|
|
lua_pushstring(L, "num2");
|
|
lua_gettable(L, idx);
|
|
if (lua_isnumber(L, -1))
|
|
mc.num2 = lua_tonumber(L, -1);
|
|
lua_pop(L, 1);
|
|
return mc;
|
|
}
|
|
|
|
int lua::scriptapi_event_create(lua_State *L)
|
|
{
|
|
std::string name = lua_tostring(L, 1);
|
|
double delay = lua_tonumber(L, 2);
|
|
double randomdelay = lua_tonumber(L, 3);
|
|
lua_pushvalue(L, 4);
|
|
int funcRef = luaL_ref(L, LUA_REGISTRYINDEX);
|
|
|
|
basic_event *event = new lua_event(L, funcRef);
|
|
event->m_name = name;
|
|
event->m_delay = delay;
|
|
event->m_delayrandom = randomdelay;
|
|
if (simulation::Events.insert(event))
|
|
{
|
|
lua_pushlightuserdata(L, event);
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_event_find(lua_State *L)
|
|
{
|
|
std::string name = lua_tostring(L, 1);
|
|
basic_event *event = simulation::Events.FindEvent(name);
|
|
if (event)
|
|
{
|
|
lua_pushlightuserdata(L, event);
|
|
return 1;
|
|
}
|
|
ErrorLog("lua: missing event: " + name);
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_event_exists(lua_State *L)
|
|
{
|
|
std::string name = lua_tostring(L, 1);
|
|
basic_event *event = simulation::Events.FindEvent(name);
|
|
lua_pushboolean(L, event != nullptr);
|
|
return 1;
|
|
}
|
|
|
|
int lua::scriptapi_event_getname(lua_State *L)
|
|
{
|
|
auto *event = static_cast<basic_event *>(lua_touserdata(L, 1));
|
|
if (event)
|
|
{
|
|
lua_pushstring(L, event->m_name.c_str());
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_event_dispatch(lua_State *L)
|
|
{
|
|
auto *event = static_cast<basic_event *>(lua_touserdata(L, 1));
|
|
auto *activator = static_cast<TDynamicObject *>(lua_touserdata(L, 2));
|
|
double delay = lua_tonumber(L, 3);
|
|
if (event)
|
|
simulation::Events.AddToQuery(event, activator, delay);
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_event_dispatch_n(lua_State *L)
|
|
{
|
|
std::string name = lua_tostring(L, 1);
|
|
auto *activator = static_cast<TDynamicObject *>(lua_touserdata(L, 2));
|
|
double delay = lua_tonumber(L, 3);
|
|
basic_event *event = simulation::Events.FindEvent(name);
|
|
if (event)
|
|
simulation::Events.AddToQuery(event, activator, delay);
|
|
else
|
|
ErrorLog("lua: missing event: " + name);
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_track_find(lua_State *L)
|
|
{
|
|
std::string name = lua_tostring(L, 1);
|
|
TTrack *track = simulation::Paths.find(name);
|
|
if (track)
|
|
{
|
|
lua_pushlightuserdata(L, track);
|
|
return 1;
|
|
}
|
|
ErrorLog("lua: missing track: " + name);
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_track_isoccupied(lua_State *L)
|
|
{
|
|
auto *track = static_cast<TTrack *>(lua_touserdata(L, 1));
|
|
if (track)
|
|
{
|
|
lua_pushboolean(L, !track->IsEmpty());
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_track_isoccupied_n(lua_State *L)
|
|
{
|
|
std::string name = lua_tostring(L, 1);
|
|
TTrack *track = simulation::Paths.find(name);
|
|
if (track)
|
|
{
|
|
lua_pushboolean(L, !track->IsEmpty());
|
|
return 1;
|
|
}
|
|
ErrorLog("lua: missing track: " + name);
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_isolated_find(lua_State *L)
|
|
{
|
|
std::string name = lua_tostring(L, 1);
|
|
TIsolated *isolated = TIsolated::Find(name);
|
|
if (isolated)
|
|
{
|
|
lua_pushlightuserdata(L, isolated);
|
|
return 1;
|
|
}
|
|
ErrorLog("lua: missing isolated: " + name);
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_isolated_isoccupied(lua_State *L)
|
|
{
|
|
auto *isolated = static_cast<TIsolated *>(lua_touserdata(L, 1));
|
|
if (isolated)
|
|
{
|
|
lua_pushboolean(L, isolated->Busy());
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_isolated_isoccupied_n(lua_State *L)
|
|
{
|
|
std::string name = lua_tostring(L, 1);
|
|
TIsolated *isolated = TIsolated::Find(name);
|
|
if (isolated)
|
|
{
|
|
lua_pushboolean(L, isolated->Busy());
|
|
return 1;
|
|
}
|
|
ErrorLog("lua: missing isolated: " + name);
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_train_getname(lua_State *L)
|
|
{
|
|
auto *dyn = static_cast<TDynamicObject *>(lua_touserdata(L, 1));
|
|
if (dyn && dyn->Mechanik)
|
|
{
|
|
lua_pushstring(L, dyn->Mechanik->TrainName().c_str());
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_dynobj_putvalues(lua_State *L)
|
|
{
|
|
auto *dyn = static_cast<TDynamicObject *>(lua_touserdata(L, 1));
|
|
auto [str, num1, num2] = get_memcell_values(L, 2);
|
|
if (!dyn)
|
|
return 0;
|
|
TLocation loc{};
|
|
if (dyn->Mechanik)
|
|
dyn->Mechanik->PutCommand(str, num1, num2, loc);
|
|
else
|
|
dyn->MoverParameters->PutCommand(str, num1, num2, loc);
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_memcell_find(lua_State *L)
|
|
{
|
|
std::string str = lua_tostring(L, 1);
|
|
TMemCell *mc = simulation::Memory.find(str);
|
|
if (mc)
|
|
{
|
|
lua_pushlightuserdata(L, mc);
|
|
return 1;
|
|
}
|
|
ErrorLog("lua: missing memcell: " + str);
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_memcell_read(lua_State *L)
|
|
{
|
|
auto *mc = static_cast<TMemCell *>(lua_touserdata(L, 1));
|
|
push_memcell_values(L, mc);
|
|
return 1;
|
|
}
|
|
|
|
int lua::scriptapi_memcell_read_n(lua_State *L)
|
|
{
|
|
std::string str = lua_tostring(L, 1);
|
|
TMemCell *mc = simulation::Memory.find(str);
|
|
push_memcell_values(L, mc);
|
|
if (!mc)
|
|
ErrorLog("lua: missing memcell: " + str);
|
|
return 1;
|
|
}
|
|
|
|
int lua::scriptapi_memcell_update(lua_State *L)
|
|
{
|
|
auto *mc = static_cast<TMemCell *>(lua_touserdata(L, 1));
|
|
auto [str, num1, num2] = get_memcell_values(L, 2);
|
|
if (mc)
|
|
mc->UpdateValues(str, num1, num2,
|
|
basic_event::flags::text | basic_event::flags::value1 | basic_event::flags::value2);
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_memcell_update_n(lua_State *L)
|
|
{
|
|
std::string mstr = lua_tostring(L, 1);
|
|
auto [str, num1, num2] = get_memcell_values(L, 2);
|
|
TMemCell *mc = simulation::Memory.find(mstr);
|
|
if (mc)
|
|
mc->UpdateValues(str, num1, num2,
|
|
basic_event::flags::text | basic_event::flags::value1 | basic_event::flags::value2);
|
|
else
|
|
ErrorLog("lua: missing memcell: " + mstr);
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_random(lua_State *L)
|
|
{
|
|
double a = lua_tonumber(L, 1);
|
|
double b = lua_tonumber(L, 2);
|
|
lua_pushnumber(L, Random(a, b));
|
|
return 1;
|
|
}
|
|
|
|
int lua::scriptapi_writelog(lua_State *L)
|
|
{
|
|
std::string txt = lua_tostring(L, 1);
|
|
WriteLog("lua: log: " + txt, logtype::lua);
|
|
return 0;
|
|
}
|
|
|
|
int lua::scriptapi_writeerrorlog(lua_State *L)
|
|
{
|
|
std::string txt = lua_tostring(L, 1);
|
|
ErrorLog("lua: log: " + txt, logtype::lua);
|
|
return 0;
|
|
}
|