Minetest-WorldEditAdditions/worldeditadditions/lib/noise/apply_2d.lua

63 lines
2.2 KiB
Lua
Raw Normal View History

local wea = worldeditadditions
2022-09-19 02:16:22 +02:00
local wea_c = worldeditadditions_core
--- Applies the given noise field to the given heightmap.
-- Mutates the given heightmap.
2022-09-19 02:16:22 +02:00
-- @param heightmap number[] A table of ZERO indexed numbers representing the heghtmap - see wea_c.terrain.make_heightmap().
-- @param noise number[] An table identical in structure to the heightmap containing the noise values to apply.
-- @param heightmap_size {x:number,z:number} A 2d vector representing the size of the heightmap.
-- @param region_height number The height of the defined region.
-- @param apply_mode string The apply mode to use to apply the noise to the heightmap.
-- @returns bool[,string] A boolean value representing whether the application was successful or not. If false, then an error message as a string is also returned describing the error that occurred.
function wea.noise.apply_2d(heightmap, noise, heightmap_size, pos1, pos2, apply_mode)
if type(apply_mode) ~= "string" and type(apply_mode) ~= "number" then
return false, "Error: Expected value of type string or number for apply_mode, but received value of type "..type(apply_mode)
end
local region_height = pos2.y - pos1.y
-- print("NOISE APPLY_2D\n")
2022-09-19 02:16:22 +02:00
wea_c.format.array_2d(noise, heightmap_size.x)
local height = tonumber(apply_mode)
-- print("DEBUG apply_mode", apply_mode, "as height", height)
for z = heightmap_size.z - 1, 0, -1 do
for x = heightmap_size.x - 1, 0, -1 do
local i = (z * heightmap_size.x) + x
if apply_mode == "add" then
2022-09-19 02:16:22 +02:00
heightmap[i] = wea_c.round(heightmap[i] + noise[i])
elseif apply_mode == "multiply" then
2022-09-19 02:16:22 +02:00
heightmap[i] = wea_c.round(heightmap[i] * noise[i])
elseif height then
-- Rescale from 0 - 1 to -1 - +1
local rescaled = (noise[i] * 2) - 1
-- Rescale to match the height specified
rescaled = rescaled * height
2022-09-19 02:16:22 +02:00
rescaled = math.floor(wea_c.clamp(
heightmap[i] + rescaled,
0, region_height
))
heightmap[i] = rescaled
else
return false, "Error: Unknown apply mode '"..apply_mode.."'"
end
end
end
-- for z = heightmap_size.z - 1, 0, -1 do
-- x = 0
-- heightmap[(z * heightmap_size.x) + x] = z
-- end
-- print("HEIGHTMAP\n")
2022-09-19 02:16:22 +02:00
-- wea_c.format.array_2d(heightmap, heightmap_size.x)
return true
end