mirror of
https://github.com/sbrl/Minetest-WorldEditAdditions.git
synced 2024-11-25 16:43:05 +00:00
85 lines
3.4 KiB
Lua
85 lines
3.4 KiB
Lua
--- Flood-fill command for complex lakes etc.
|
|
-- @module worldeditadditions.floodfill
|
|
|
|
local wea = worldeditadditions
|
|
|
|
function worldeditadditions.floodfill(start_pos, radius, replace_node)
|
|
-- Calculate the area we want to modify
|
|
local pos1 = vector.add(start_pos, { x = radius, y = 0, z = radius })
|
|
local pos2 = vector.subtract(start_pos, { x = radius, y = radius, z = radius })
|
|
pos1, pos2 = worldedit.sort_pos(pos1, pos2)
|
|
-- pos2 will always have the highest co-ordinates now
|
|
|
|
|
|
-- Fetch the nodes in the specified area
|
|
local manip, area = worldedit.manip_helpers.init(pos1, pos2)
|
|
local data = manip:get_data()
|
|
|
|
-- Setup for the floodfill operation itself
|
|
local start_pos_index = area:index(start_pos.x, start_pos.y, start_pos.z)
|
|
|
|
local search_id = data[start_pos_index]
|
|
local replace_id = minetest.get_content_id(replace_node)
|
|
local radius_sq = radius*radius
|
|
|
|
if search_id == replace_id then
|
|
return false
|
|
end
|
|
|
|
local count = 0
|
|
local remaining_nodes = wea.Queue.new() remaining_nodes:enqueue(start_pos_index)
|
|
|
|
-- Do the floodfill
|
|
while remaining_nodes:is_empty() == false do
|
|
local cur = remaining_nodes:dequeue()
|
|
|
|
-- Replace this node
|
|
data[cur] = replace_id
|
|
count = count + 1
|
|
|
|
-- Check all the nearby nodes
|
|
-- We don't need to go upwards here, since we're filling in lake-style
|
|
local xplus = cur + 1 -- +X
|
|
if data[xplus] == search_id and
|
|
worldeditadditions.vector.lengthsquared(vector.subtract(area:position(xplus), start_pos)) < radius_sq and
|
|
not remaining_nodes:contains(xplus) then
|
|
-- minetest.log("action", "[floodfill] [+X] index " .. xplus .. " is a " .. data[xplus] .. ", searching for a " .. search_id)
|
|
remaining_nodes:enqueue(xplus)
|
|
end
|
|
local xminus = cur - 1 -- -X
|
|
if data[xminus] == search_id and
|
|
worldeditadditions.vector.lengthsquared(vector.subtract(area:position(xminus), start_pos)) < radius_sq and
|
|
not remaining_nodes:contains(xminus) then
|
|
-- minetest.log("action", "[floodfill] [-X] index " .. xminus .. " is a " .. data[xminus] .. ", searching for a " .. search_id)
|
|
remaining_nodes:enqueue(xminus)
|
|
end
|
|
local zplus = cur + area.zstride -- +Z
|
|
if data[zplus] == search_id and
|
|
worldeditadditions.vector.lengthsquared(vector.subtract(area:position(zplus), start_pos)) < radius_sq and
|
|
not remaining_nodes:contains(zplus) then
|
|
-- minetest.log("action", "[floodfill] [+Z] index " .. zplus .. " is a " .. data[zplus] .. ", searching for a " .. search_id)
|
|
remaining_nodes:enqueue(zplus)
|
|
end
|
|
local zminus = cur - area.zstride -- -Z
|
|
if data[zminus] == search_id and
|
|
worldeditadditions.vector.lengthsquared(vector.subtract(area:position(zminus), start_pos)) < radius_sq and
|
|
not remaining_nodes:contains(zminus) then
|
|
-- minetest.log("action", "[floodfill] [-Z] index " .. zminus .. " is a " .. data[zminus] .. ", searching for a " .. search_id)
|
|
remaining_nodes:enqueue(zminus)
|
|
end
|
|
local yminus = cur - area.ystride -- -Y
|
|
if data[yminus] == search_id and
|
|
worldeditadditions.vector.lengthsquared(vector.subtract(area:position(yminus), start_pos)) < radius_sq and
|
|
not remaining_nodes:contains(yminus) then
|
|
-- minetest.log("action", "[floodfill] [-Y] index " .. yminus .. " is a " .. data[yminus] .. ", searching for a " .. search_id)
|
|
remaining_nodes:enqueue(yminus)
|
|
end
|
|
|
|
count = count + 1
|
|
end
|
|
|
|
-- Save the modified nodes back to disk & return
|
|
worldedit.manip_helpers.finish(manip, data)
|
|
|
|
return count
|
|
end
|