Minetest-WorldEditAdditions/worldeditadditions/lib/forest.lua
2021-07-04 14:35:02 +01:00

104 lines
3.3 KiB
Lua

--- Overlap command. Places a specified node on top of each column.
-- @module worldeditadditions.layers
function worldeditadditions.forest(pos1, pos2, density_multiplier, sapling_weights)
pos1, pos2 = worldedit.sort_pos(pos1, pos2)
local weight_total = 0
for _name,weight in pairs(sapling_weights) do
weight_total = weight_total + weight
end
sapling_weights["air"] = math.ceil(weight_total * 100 * 1/density_multiplier)
-- This command requires the bonemeal mod to be installed
-- We check here too because other mods might call this function directly and bypass the chat command system
if not minetest.get_modpath("bonemeal") then
return false, "Bonemeal mod not loaded"
end
local overlay_stats = worldeditadditions.overlay(pos1, pos2, sapling_weights)
-- Fetch the nodes in the specified area
local manip, area = worldedit.manip_helpers.init(pos1, pos2)
local data = manip:get_data()
local id_air = minetest.get_content_id("air")
local group_cache = {}
local stats = { attempts = {}, failures = 0, placed = {} }
for z = pos2.z, pos1.z, -1 do
for x = pos2.x, pos1.x, -1 do
for y = pos2.y, pos1.y, -1 do
local i = area:index(x, y, z)
local node_id = data[i]
if not group_cache[node_id] then
group_cache[node_id] = worldeditadditions.is_sapling(node_id)
end
if group_cache[node_id] then
local did_grow = false
local new_id_at_pos
local new_name_at_pos
for i=1,100 do
bonemeal:on_use(
{ x = x, y = y, z = z },
4,
nil
)
new_name_at_pos = minetest.get_node({ z = z, y = y, x = x }).name
new_id_at_pos = minetest.get_content_id(new_name_at_pos)
if not group_cache[new_id_at_pos] then
group_cache[new_id_at_pos] = worldeditadditions.is_sapling(new_id_at_pos)
end
if not group_cache[new_id_at_pos] then
did_grow = true
-- Log the number of attempts it took to grow
table.insert(stats.attempts, i)
-- Update the running total of saplings that grew
if not stats.placed[node_id] then
stats.placed[node_id] = 0
end
stats.placed[node_id] = stats.placed[node_id] + 1
-- print("incrementing id", node_id, "to", stats.placed[node_id])
break
end
end
if not did_grow then
-- print("[//forest] Failed to grow sapling, detected node id", new_id_at_pos, "name", new_name_at_pos, "was originally", minetest.get_name_from_content_id(node_id))
-- We can't set it to air here because then when we save back we would overwrite all the newly grown trees
stats.failures = stats.failures + 1
end
end
end
end
end
-- Re-fetch the Voxel Manipulator to accountn for the new trees
manip, area = worldedit.manip_helpers.init(pos1, pos2)
data = manip:get_data()
for z = pos2.z, pos1.z, -1 do
for x = pos2.x, pos1.x, -1 do
for y = pos2.y, pos1.y, -1 do
local i = area:index(x, y, z)
if not group_cache[data[i]] then
group_cache[data[i]] = worldeditadditions.is_sapling(data[i])
end
if group_cache[data[i]] then
data[i] = id_air
end
end
end
end
stats.successes = #stats.attempts
stats.attempts_avg = worldeditadditions.average(stats.attempts)
-- Save the modified nodes back to disk & return
worldedit.manip_helpers.finish(manip, data)
return true, stats
end