Minetest-WorldEditAdditions/worldeditadditions_commands/commands/replacemix.lua

local we_c = worldeditadditions_commands

-- ██████  ███████ ██████  ██       █████   ██████ ███████ ███    ███ ██ ██   ██
-- ██   ██ ██      ██   ██ ██      ██   ██ ██      ██      ████  ████ ██  ██ ██
-- ██████  █████   ██████  ██      ███████ ██      █████   ██ ████ ██ ██   ███
-- ██   ██ ██      ██      ██      ██   ██ ██      ██      ██  ██  ██ ██  ██ ██
-- ██   ██ ███████ ██      ███████ ██   ██  ██████ ███████ ██      ██ ██ ██   ██
worldedit.register_command("replacemix", {
	params = "<target_node> [<chance>] <replace_node_a> [<chance_a>] [<replace_node_b> [<chance_b>]] [<replace_node_N> [<chance_N>]] ...",
	description = "Replaces target_node with a mix of other nodes. Functions simmilarly to //mix. <chance> is optional and the chance to replace the target node at all. replace_node_a is the node to replace target_node with. If multiple nodes are specified in a space separated list, then when replacing an instance of target_node one is randomly chosen from the list. Just like with //mix, if a positive integer is present after a replace_node, that adds a weighting to that particular node making it more common.",
	privs = { worldedit = true },
	require_pos = 2,
	parse = function(params_text)
		if not params_text or params_text == "" then
			return false, "Error: No arguments specified"
		end
		
		local parts = worldeditadditions.split(params_text, "%s+", false)
		
		local target_node = nil
		local target_node_chance = 1
		local replace_nodes = {}
		
		local mode = "target_node"
		local last_node_name = nil
		for i, part in ipairs(parts) do
			if mode == "target_node" then
				target_node = worldedit.normalize_nodename(part)
				if not target_node then
					return false, "Error: Invalid target_node name"
				end
				mode = "target_chance"
			elseif mode == "target_chance" and tonumber(part) then
				target_node_chance = tonumber(part)
				mode = "replace_node"
			elseif (mode == "target_chance" and not tonumber(part)) or mode == "replace_node" then
				mode = "replace_node"
				if tonumber(part) then
					if not last_node_name then
						return false, "Error: No previous node name was found (this is a probably a bug)."
					end
					replace_nodes[last_node_name] = math.floor(tonumber(part))
				else
					if last_node_name and not replace_nodes[last_node_name] then
						replace_nodes[last_node_name] = 1
					end
					
					local node_name = worldedit.normalize_nodename(part)
					if not node_name then
						return false, "Error: Invalid replace node name '"..part.."'."
					end
					last_node_name = node_name
				end
			end
		end
		if not replace_nodes[last_node_name] then
			replace_nodes[last_node_name] = 1
		end
		
		-- We unconditionally math.floor here because when we tried to test for it directly it was unreliable
		return true, target_node, target_node_chance, replace_nodes
	end,
	nodes_needed = function(name) -- target_node, target_node_chance, replace_nodes
		return worldedit.volume(worldedit.pos1[name], worldedit.pos2[name])
	end,
	func = function(name, target_node, target_node_chance, replace_nodes)
		local start_time = worldeditadditions.get_ms_time()
		
		local success, changed, candidates, distribution = worldeditadditions.replacemix(
			worldedit.pos1[name], worldedit.pos2[name],
			target_node,
			target_node_chance,
			replace_nodes
		)
		if not success then
			return success, changed
		end
		
		local nodes_total = worldedit.volume(worldedit.pos1[name], worldedit.pos2[name])
		local percentage_replaced = worldeditadditions.round((changed / candidates)*100, 2)
		local distribution_table = worldeditadditions.format.node_distribution(
			distribution,
			changed,
			true -- Add a grand total to the bottom
		)
		
		local time_taken = worldeditadditions.get_ms_time() - start_time
		
		
		minetest.log("action", name .. " used //replacemix at "..worldeditadditions.vector.tostring(worldedit.pos1[name]).." - "..worldeditadditions.vector.tostring(worldedit.pos2[name])..", replacing " .. changed.." nodes (out of "..nodes_total.." nodes) in "..time_taken.."s")
		
		return true, distribution_table..changed.." out of "..candidates.." (~"..percentage_replaced.."%) candidates replaced in "..worldeditadditions.format.human_time(time_taken)
	end
})