Export the type tree with enum keys to lua.
Alexander Gavrilov
parent
296d3a0af3
commit
dbbd9acfad
@ -0,0 +1,218 @@
|
|||||||
|
/*
|
||||||
|
https://github.com/peterix/dfhack
|
||||||
|
Copyright (c) 2009-2011 Petr Mrázek (peterix@gmail.com)
|
||||||
|
|
||||||
|
This software is provided 'as-is', without any express or implied
|
||||||
|
warranty. In no event will the authors be held liable for any
|
||||||
|
damages arising from the use of this software.
|
||||||
|
|
||||||
|
Permission is granted to anyone to use this software for any
|
||||||
|
purpose, including commercial applications, and to alter it and
|
||||||
|
redistribute it freely, subject to the following restrictions:
|
||||||
|
|
||||||
|
1. The origin of this software must not be misrepresented; you must
|
||||||
|
not claim that you wrote the original software. If you use this
|
||||||
|
software in a product, an acknowledgment in the product documentation
|
||||||
|
would be appreciated but is not required.
|
||||||
|
|
||||||
|
2. Altered source versions must be plainly marked as such, and
|
||||||
|
must not be misrepresented as being the original software.
|
||||||
|
|
||||||
|
3. This notice may not be removed or altered from any source
|
||||||
|
distribution.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "Internal.h"
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
#include <map>
|
||||||
|
|
||||||
|
#include "MemAccess.h"
|
||||||
|
#include "Core.h"
|
||||||
|
#include "VersionInfo.h"
|
||||||
|
#include "tinythread.h"
|
||||||
|
// must be last due to MS stupidity
|
||||||
|
#include "DataDefs.h"
|
||||||
|
#include "DataIdentity.h"
|
||||||
|
|
||||||
|
#include "MiscUtils.h"
|
||||||
|
|
||||||
|
#include <lua.h>
|
||||||
|
#include <lauxlib.h>
|
||||||
|
|
||||||
|
using namespace DFHack;
|
||||||
|
|
||||||
|
static luaL_Reg no_functions[] = { { NULL, NULL } };
|
||||||
|
|
||||||
|
inline void lua_dup(lua_State *state) { lua_pushvalue(state, -1); }
|
||||||
|
inline void lua_swap(lua_State *state) { lua_insert(state, -2); }
|
||||||
|
|
||||||
|
static int change_error(lua_State *state)
|
||||||
|
{
|
||||||
|
luaL_error(state, "Attempt to change a read-only table.\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
static void freeze_table(lua_State *state, bool leave_metatable = false, const char *name = NULL)
|
||||||
|
{
|
||||||
|
// rv = {}; setmetatable(rv, { __index = in, __newindex = change_error, __metatable = name })
|
||||||
|
int base = lua_gettop(state);
|
||||||
|
lua_newtable(state);
|
||||||
|
lua_swap(state);
|
||||||
|
lua_setfield(state, base, "__index");
|
||||||
|
lua_getfield(state, LUA_REGISTRYINDEX, "DFHack.ChangeError");
|
||||||
|
lua_setfield(state, base, "__newindex");
|
||||||
|
lua_newtable(state);
|
||||||
|
lua_swap(state);
|
||||||
|
lua_dup(state);
|
||||||
|
lua_setmetatable(state, base);
|
||||||
|
if (name)
|
||||||
|
{
|
||||||
|
lua_pushstring(state, name);
|
||||||
|
lua_setfield(state, -2, "__metatable");
|
||||||
|
}
|
||||||
|
// result: [frozen table] [metatable]
|
||||||
|
if (!leave_metatable)
|
||||||
|
lua_pop(state, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void SaveTypeInfo(lua_State *state, void *node)
|
||||||
|
{
|
||||||
|
lua_getfield(state, LUA_REGISTRYINDEX, "DFHack.DFTypes");
|
||||||
|
lua_pushlightuserdata(state, node);
|
||||||
|
lua_pushvalue(state, -3);
|
||||||
|
lua_settable(state, -3);
|
||||||
|
lua_pop(state, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
static bool RegisterTypeInfo(lua_State *state, void *node)
|
||||||
|
{
|
||||||
|
lua_getfield(state, LUA_REGISTRYINDEX, "DFHack.DFTypes");
|
||||||
|
int base = lua_gettop(state);
|
||||||
|
|
||||||
|
lua_pushlightuserdata(state, node);
|
||||||
|
lua_rawget(state, base);
|
||||||
|
|
||||||
|
bool added = false;
|
||||||
|
|
||||||
|
if (lua_isnil(state, -1))
|
||||||
|
{
|
||||||
|
lua_pop(state, 1);
|
||||||
|
|
||||||
|
lua_newtable(state);
|
||||||
|
lua_pushlightuserdata(state, node);
|
||||||
|
lua_pushvalue(state, -2);
|
||||||
|
lua_rawset(state, base);
|
||||||
|
|
||||||
|
added = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
lua_remove(state, -2);
|
||||||
|
return added;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void RenderTypeChildren(lua_State *state, const std::vector<compound_identity*> &children);
|
||||||
|
|
||||||
|
static void RenderType(lua_State *state, compound_identity *node)
|
||||||
|
{
|
||||||
|
assert(node->getName());
|
||||||
|
|
||||||
|
lua_newtable(state);
|
||||||
|
if (!lua_checkstack(state, 20))
|
||||||
|
return;
|
||||||
|
|
||||||
|
int base = lua_gettop(state);
|
||||||
|
|
||||||
|
lua_pushlightuserdata(state, node);
|
||||||
|
lua_setfield(state, base, "_identity");
|
||||||
|
|
||||||
|
switch (node->type())
|
||||||
|
{
|
||||||
|
case IDTYPE_ENUM:
|
||||||
|
{
|
||||||
|
enum_identity *eid = (enum_identity*)node;
|
||||||
|
const char *const *keys = eid->getKeys();
|
||||||
|
|
||||||
|
// For enums, set mapping between keys and values
|
||||||
|
for (int64_t i = eid->getFirstItem(), j = 0; i <= eid->getLastItem(); i++, j++)
|
||||||
|
{
|
||||||
|
if (!keys[j])
|
||||||
|
continue;
|
||||||
|
|
||||||
|
lua_pushinteger(state, i);
|
||||||
|
lua_pushstring(state, keys[j]);
|
||||||
|
lua_dup(state);
|
||||||
|
lua_pushinteger(state, i);
|
||||||
|
|
||||||
|
lua_settable(state, base);
|
||||||
|
lua_settable(state, base);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (eid->getFirstItem() <= eid->getLastItem())
|
||||||
|
{
|
||||||
|
lua_pushstring(state, "_first_item");
|
||||||
|
lua_pushinteger(state, eid->getFirstItem());
|
||||||
|
lua_settable(state, base);
|
||||||
|
|
||||||
|
lua_pushstring(state, "_last_item");
|
||||||
|
lua_pushinteger(state, eid->getLastItem());
|
||||||
|
lua_settable(state, base);
|
||||||
|
}
|
||||||
|
|
||||||
|
SaveTypeInfo(state, node);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
default:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
RenderTypeChildren(state, node->getScopeChildren());
|
||||||
|
|
||||||
|
assert(base == lua_gettop(state));
|
||||||
|
|
||||||
|
freeze_table(state, false, node->getName());
|
||||||
|
}
|
||||||
|
|
||||||
|
static void RenderTypeChildren(lua_State *state, const std::vector<compound_identity*> &children)
|
||||||
|
{
|
||||||
|
for (size_t i = 0; i < children.size(); i++)
|
||||||
|
{
|
||||||
|
RenderType(state, children[i]);
|
||||||
|
lua_pushstring(state, children[i]->getName());
|
||||||
|
lua_swap(state);
|
||||||
|
lua_settable(state, -3);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void DoAttach(lua_State *state)
|
||||||
|
{
|
||||||
|
int base = lua_gettop(state);
|
||||||
|
|
||||||
|
lua_pushcfunction(state, change_error);
|
||||||
|
lua_setfield(state, LUA_REGISTRYINDEX, "DFHack.ChangeError");
|
||||||
|
|
||||||
|
luaL_register(state, "df", no_functions);
|
||||||
|
|
||||||
|
{
|
||||||
|
// Assign df a metatable with read-only contents
|
||||||
|
lua_newtable(state);
|
||||||
|
|
||||||
|
// Render the type structure
|
||||||
|
RenderTypeChildren(state, compound_identity::getTopScope());
|
||||||
|
|
||||||
|
freeze_table(state, true, "df");
|
||||||
|
lua_remove(state, -2);
|
||||||
|
lua_setmetatable(state, -2);
|
||||||
|
}
|
||||||
|
|
||||||
|
lua_pop(state, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
void DFHack::AttachDFGlobals(lua_State *state)
|
||||||
|
{
|
||||||
|
if (luaL_newmetatable(state, "DFHack.DFTypes"))
|
||||||
|
DoAttach(state);
|
||||||
|
|
||||||
|
lua_pop(state, 1);
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue