From: Perttu Ahola <celeron55@gmail.com>
Date: Sun, 8 Apr 2012 11:37:12 +0000 (+0300)
Subject: EnvRef:find_nodes_in_area(minp, maxp, nodenames)
X-Git-Url: http://81.2.79.47:8989/gitweb/?a=commitdiff_plain;h=e29a5b11e4d6356c41806f3cc43c62d810e8cdb3;p=zefram%2Fminetest%2Fminetest_engine.git

EnvRef:find_nodes_in_area(minp, maxp, nodenames)
---

diff --git a/doc/lua_api.txt b/doc/lua_api.txt
index e4f357f2..9c70037a 100644
--- 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 7ce10c41..726f43b9 100644
--- 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}
 };