EnvRef:find_nodes_in_area(minp, maxp, nodenames)

diff --git a/doc/lua_api.txt b/doc/lua_api.txt
index e4f357f2dfe9d9d835164d3ed255341cc24135e4..9c70037a55d84e1722f35df1c748428a9aac8092 100644 (file)
--- a/doc/lua_api.txt
+++ b/doc/lua_api.txt
@@ -668,6 +668,8 @@ methods:
 - get_timeofday()
 - find_node_near(pos, radius, nodenames) -> pos or nil
   ^ nodenames: eg. {"ignore", "group:tree"} or "default:dirt"
+- find_nodes_in_area(minp, maxp, nodenames) -> list of positions
+  ^ nodenames: eg. {"ignore", "group:tree"} or "default:dirt"
 - get_perlin(seeddiff, octaves, persistence, scale)
   ^ Return world-specific perlin noise (int(worldseed)+seeddiff)
 Deprecated:

diff --git a/src/scriptapi.cpp b/src/scriptapi.cpp
index 7ce10c415f32f64aa9d823eecb3bc59d676cae1d..726f43b9c7705ddff077e3969c4a047fa5447c04 100644 (file)
--- a/src/scriptapi.cpp
+++ b/src/scriptapi.cpp
@@ -3346,6 +3346,55 @@ private:
                return 0;
        }
 
+       // EnvRef:find_nodes_in_area(minp, maxp, nodenames) -> list of positions
+       // nodenames: eg. {"ignore", "group:tree"} or "default:dirt"
+       static int l_find_nodes_in_area(lua_State *L)
+       {
+               EnvRef *o = checkobject(L, 1);
+               ServerEnvironment *env = o->m_env;
+               if(env == NULL) return 0;
+               INodeDefManager *ndef = get_server(L)->ndef();
+               v3s16 minp = read_v3s16(L, 2);
+               v3s16 maxp = read_v3s16(L, 3);
+               std::set<content_t> filter;
+               if(lua_istable(L, 4)){
+                       int table = 4;
+                       lua_pushnil(L);
+                       while(lua_next(L, table) != 0){
+                               // key at index -2 and value at index -1
+                               luaL_checktype(L, -1, LUA_TSTRING);
+                               ndef->getIds(lua_tostring(L, -1), filter);
+                               // removes value, keeps key for next iteration
+                               lua_pop(L, 1);
+                       }
+               } else if(lua_isstring(L, 4)){
+                       ndef->getIds(lua_tostring(L, 4), filter);
+               }
+
+               // Get the table insert function
+               lua_getglobal(L, "table");
+               lua_getfield(L, -1, "insert");
+               int table_insert = lua_gettop(L);
+               
+               lua_newtable(L);
+               int table = lua_gettop(L);
+               for(s16 x=minp.X; x<=maxp.X; x++)
+               for(s16 y=minp.Y; y<=maxp.Y; y++)
+               for(s16 z=minp.Z; z<=maxp.Z; z++)
+               {
+                       v3s16 p(x,y,z);
+                       content_t c = env->getMap().getNodeNoEx(p).getContent();
+                       if(filter.count(c) != 0){
+                               lua_pushvalue(L, table_insert);
+                               lua_pushvalue(L, table);
+                               push_v3s16(L, p);
+                               if(lua_pcall(L, 2, 0, 0))
+                                       script_error(L, "error: %s", lua_tostring(L, -1));
+                       }
+               }
+               return 1;
+       }
+
        //      EnvRef:get_perlin(seeddiff, octaves, persistence, scale)
        //  returns world-specific PerlinNoise
        static int l_get_perlin(lua_State *L)
@@ -3441,6 +3490,7 @@ const luaL_reg EnvRef::methods[] = {
        method(EnvRef, set_timeofday),
        method(EnvRef, get_timeofday),
        method(EnvRef, find_node_near),
+       method(EnvRef, find_nodes_in_area),
        method(EnvRef, get_perlin),
        {0,0}
 };