Minetest-WorldEditAdditions/worldeditadditions/lib/ellipsoid.lua

82 lines
3 KiB
Lua
Raw Normal View History

2022-09-18 23:18:03 +00:00
local wea_c = worldeditadditions_core
local Vector3 = wea_c.Vector3
---
-- @module worldeditadditions
-- ███████ ██ ██ ██ ██████ ███████ ███████
-- ██ ██ ██ ██ ██ ██ ██ ██
-- █████ ██ ██ ██ ██████ ███████ █████
-- ██ ██ ██ ██ ██ ██ ██
-- ███████ ███████ ███████ ██ ██ ███████ ███████
2021-10-13 21:50:37 +00:00
--- Fills an ellipsoidal area around the given position with the target node.
-- The resulting ellipsoid may optionally be hollow (in which case
-- nodes inside the ellipsoid are left untouched).
-- @param position Vector3 The centre position of the ellipsoid.
-- @param radius Vector3 The radius of the ellipsoid in all 3 dimensions.
-- @param target_node string The name of the node to use to fill the ellipsoid.
-- @param hollow bool Whether the ellipsoid should be hollow or not.
-- @returns number The number of nodes filled to create the (optionally hollow) ellipsoid. This number will be lower with hollow ellipsoids, since the internals of an ellipsoid aren't altered.
2020-05-10 22:29:49 +00:00
function worldeditadditions.ellipsoid(position, radius, target_node, hollow)
2022-09-18 23:18:03 +00:00
radius = Vector3.clone(radius)
-- position = { x, y, z }
2022-09-18 23:18:03 +00:00
local hollow_inner_radius = radius - 1
-- Fetch the nodes in the specified area
-- OPTIMIZE: We should be able to calculate a more efficient box-area here
2018-06-09 12:10:42 +00:00
local manip, area = worldedit.manip_helpers.init_radius(position, math.max(radius.x, radius.y, radius.z))
local data = manip:get_data()
local node_id = minetest.get_content_id(target_node)
local stride_z, stride_y = area.zstride, area.ystride
2018-06-09 12:10:42 +00:00
local count = 0 -- The number of nodes replaced
2018-06-09 18:38:23 +00:00
local idx_z_base = area:index(position.x - radius.x, position.y - radius.y, position.z - radius.z) -- initial z offset
for z = -radius.z, radius.z do
local idx_y_base = idx_z_base
for y = -radius.y, radius.y do
local i = idx_y_base
for x = -radius.x, radius.x do
2022-09-18 23:18:03 +00:00
local here = Vector3.new(x, y, z)
-- If we're inside the ellipse, then fill it in
2022-09-18 23:18:03 +00:00
local comp = here / radius
local ellipsoid_dist = Vector3.dot(comp, comp)
2018-06-09 19:02:30 +00:00
if ellipsoid_dist <= 1 then
local place_ok = not hollow;
2018-06-09 18:35:32 +00:00
2018-06-09 19:02:30 +00:00
if not place_ok then
-- It must be hollow! Do some additional calculations.
2022-09-18 23:18:03 +00:00
local h_comp = here / hollow_inner_radius
2018-06-09 19:02:30 +00:00
-- It's only ok to place it if it's outside our inner ellipse
2022-09-18 23:18:03 +00:00
place_ok = Vector3.dot(h_comp, h_comp) >= 1
2018-06-09 19:02:30 +00:00
end
2018-06-09 18:35:32 +00:00
2018-06-09 19:02:30 +00:00
if place_ok then
data[i] = node_id
count = count + 1
end
end
i = i + 1
end
idx_y_base = idx_y_base + stride_y
end
idx_z_base = idx_z_base + stride_z
end
-- Save the modified nodes back to disk & return
worldedit.manip_helpers.finish(manip, data)
2018-06-09 12:10:42 +00:00
return count
end