mirror of
https://github.com/sbrl/Minetest-WorldEditAdditions.git
synced 2024-06-03 05:03:02 +00:00
67 lines
2.3 KiB
Lua
67 lines
2.3 KiB
Lua
local wea = worldeditadditions
|
|
local Vector3 = wea.Vector3
|
|
|
|
--- Applies the given brush with the given height and size to the given position.
|
|
-- @param pos1 Vector3 The position at which to apply the brush.
|
|
-- @param brush_name string The name of the brush to apply.
|
|
-- @param height number The height of the brush application.
|
|
-- @param brush_size Vector3 The size of the brush application. Values are interpreted on the X/Y coordinates, and NOT X/Z!
|
|
-- @returns bool, string|{ added: number, removed: number } A bool indicating whether the operation was successful or not, followed by either an error message as a string (if it was not successful) or a table of statistics (if it was successful).
|
|
local function apply(pos1, brush_name, height, brush_size)
|
|
-- 1: Get & validate brush
|
|
local success, brush, brush_size_actual = wea.sculpt.make_brush(brush_name, brush_size)
|
|
if not success then return success, brush end
|
|
|
|
local brush_size_terrain = Vector3.new(
|
|
brush_size_actual.x,
|
|
0,
|
|
brush_size_actual.y
|
|
)
|
|
local brush_size_radius = (brush_size_terrain / 2):floor()
|
|
|
|
local pos1_compute = pos1 - brush_size_radius
|
|
local pos2_compute = pos1 + brush_size_radius + Vector3.new(0, height, 0)
|
|
|
|
|
|
-- 2: Fetch the nodes in the specified area, extract heightmap
|
|
local manip, area = worldedit.manip_helpers.init(pos1_compute, pos2_compute)
|
|
local data = manip:get_data()
|
|
|
|
local heightmap, heightmap_size = wea.make_heightmap(
|
|
pos1_compute, pos2_compute,
|
|
manip, area,
|
|
data
|
|
)
|
|
local heightmap_orig = wea.table.shallowcopy(heightmap)
|
|
|
|
for z = pos2_compute.z, pos1_compute.z, -1 do
|
|
for x = pos2_compute.x, pos1_compute.x, -1 do
|
|
local next_index = 1 -- We use table.insert() in make_weighted
|
|
local placed_node = false
|
|
|
|
local hi = (z-pos1_compute.z)*heightmap_size.x + (x-pos1_compute.x)
|
|
|
|
local offset = brush[hi] * height
|
|
if height > 0 then offset = math.floor(offset)
|
|
else offset = math.ceil(offset) end
|
|
heightmap[hi] = heightmap[hi] + offset
|
|
end
|
|
end
|
|
|
|
-- 3: Save back to disk & return
|
|
local success2, stats = wea.apply_heightmap_changes(
|
|
pos1_compute, pos2_compute,
|
|
area, data,
|
|
heightmap_orig, heightmap,
|
|
heightmap_size
|
|
)
|
|
if not success2 then return success2, stats end
|
|
|
|
worldedit.manip_helpers.finish(manip, data)
|
|
|
|
return true, stats
|
|
end
|
|
|
|
|
|
return apply
|