diff options
author | kwolekr <kwolekr@minetest.net> | 2014-12-10 01:49:57 -0500 |
---|---|---|
committer | kwolekr <kwolekr@minetest.net> | 2014-12-10 01:49:57 -0500 |
commit | dcc48976ce92dc0385037e8bf16075396bcb6943 (patch) | |
tree | 909b831ec2d2d59d88efabca0b9d459a82a1800e /src/script/lua_api | |
parent | 4cf8042b7c8c4dd56cb0a749fb03d9653aa48983 (diff) | |
download | minetest-dcc48976ce92dc0385037e8bf16075396bcb6943.tar.gz minetest-dcc48976ce92dc0385037e8bf16075396bcb6943.tar.bz2 minetest-dcc48976ce92dc0385037e8bf16075396bcb6943.zip |
Move get_schematic and read_schematic to l_mapgen.cpp
Diffstat (limited to 'src/script/lua_api')
-rw-r--r-- | src/script/lua_api/l_mapgen.cpp | 109 | ||||
-rw-r--r-- | src/script/lua_api/l_mapgen.h | 2 |
2 files changed, 108 insertions, 3 deletions
diff --git a/src/script/lua_api/l_mapgen.cpp b/src/script/lua_api/l_mapgen.cpp index 03a2ee0d5..3fe6fb991 100644 --- a/src/script/lua_api/l_mapgen.cpp +++ b/src/script/lua_api/l_mapgen.cpp @@ -85,7 +85,114 @@ struct EnumString ModApiMapgen::es_Rotation[] = }; -static void read_schematic_replacements(lua_State *L, +/////////////////////////////////////////////////////////////////////////////// + + +bool read_schematic(lua_State *L, int index, Schematic *schem, + INodeDefManager *ndef, std::map<std::string, std::string> &replace_names) +{ + //// Get schematic size + lua_getfield(L, index, "size"); + v3s16 size = read_v3s16(L, -1); + lua_pop(L, 1); + + //// Get schematic data + lua_getfield(L, index, "data"); + luaL_checktype(L, -1, LUA_TTABLE); + + int numnodes = size.X * size.Y * size.Z; + MapNode *schemdata = new MapNode[numnodes]; + int i = 0; + + lua_pushnil(L); + while (lua_next(L, -2)) { + if (i >= numnodes) { + i++; + lua_pop(L, 1); + continue; + } + + // same as readnode, except param1 default is MTSCHEM_PROB_CONST + lua_getfield(L, -1, "name"); + std::string name = luaL_checkstring(L, -1); + lua_pop(L, 1); + + u8 param1; + lua_getfield(L, -1, "param1"); + param1 = !lua_isnil(L, -1) ? lua_tonumber(L, -1) : MTSCHEM_PROB_ALWAYS; + lua_pop(L, 1); + + u8 param2; + lua_getfield(L, -1, "param2"); + param2 = !lua_isnil(L, -1) ? lua_tonumber(L, -1) : 0; + lua_pop(L, 1); + + std::map<std::string, std::string>::iterator it; + it = replace_names.find(name); + if (it != replace_names.end()) + name = it->second; + + schemdata[i] = MapNode(ndef, name, param1, param2); + + i++; + lua_pop(L, 1); + } + + if (i != numnodes) { + errorstream << "read_schematic: incorrect number of " + "nodes provided in raw schematic data (got " << i << + ", expected " << numnodes << ")." << std::endl; + delete schemdata; + return false; + } + + //// Get Y-slice probability values (if present) + u8 *slice_probs = new u8[size.Y]; + for (i = 0; i != size.Y; i++) + slice_probs[i] = MTSCHEM_PROB_ALWAYS; + + lua_getfield(L, index, "yslice_prob"); + if (lua_istable(L, -1)) { + lua_pushnil(L); + while (lua_next(L, -2)) { + if (getintfield(L, -1, "ypos", i) && i >= 0 && i < size.Y) { + slice_probs[i] = getintfield_default(L, -1, + "prob", MTSCHEM_PROB_ALWAYS); + } + lua_pop(L, 1); + } + } + + // Here, we read the nodes directly from the INodeDefManager - there is no + // need for pending node resolutions so we'll mark this schematic as updated + schem->flags = SCHEM_CIDS_UPDATED; + + schem->size = size; + schem->schemdata = schemdata; + schem->slice_probs = slice_probs; + return true; +} + + +bool get_schematic(lua_State *L, int index, Schematic *schem, + INodeDefManager *ndef, std::map<std::string, std::string> &replace_names) +{ + if (index < 0) + index = lua_gettop(L) + 1 + index; + + if (lua_istable(L, index)) { + return read_schematic(L, index, schem, ndef, replace_names); + } else if (lua_isstring(L, index)) { + NodeResolver *resolver = ndef->getResolver(); + const char *filename = lua_tostring(L, index); + return schem->loadSchematicFromFile(filename, resolver, replace_names); + } else { + return false; + } +} + + +void read_schematic_replacements(lua_State *L, std::map<std::string, std::string> &replace_names, int index) { lua_pushnil(L); diff --git a/src/script/lua_api/l_mapgen.h b/src/script/lua_api/l_mapgen.h index 72bf1f59b..76f60a2d7 100644 --- a/src/script/lua_api/l_mapgen.h +++ b/src/script/lua_api/l_mapgen.h @@ -76,6 +76,4 @@ public: static void Initialize(lua_State *L, int top); }; - - #endif /* L_MAPGEN_H_ */ |