1
0
mirror of https://github.com/Refactorio/RedMew.git synced 2024-12-16 10:19:27 +02:00
RedMew/map_gen/Diggy/Feature/ScatteredResources.lua
2018-11-13 10:39:41 -05:00

138 lines
4.7 KiB
Lua

--[[-- info
Provides the ability to spawn random ores all over the place.
]]
-- dependencies
local Event = require 'utils.event'
local Debug = require 'map_gen.Diggy.Debug'
local Template = require 'map_gen.Diggy.Template'
local Perlin = require 'map_gen.shared.perlin_noise'
local random = math.random
local sqrt = math.sqrt
local ceil = math.ceil
local floor = math.floor
-- this
local ScatteredResources = {}
local function get_name_by_weight(collection)
local pre_calculated = random()
local current = 0
local sum = 0
for _, weight in pairs(collection) do
sum = sum + weight
end
local target = pre_calculated * sum
for name, weight in pairs(collection) do
current = current + weight
if (current >= target) then
return name
end
end
Debug.print('Current \'' .. current .. '\' should be higher or equal to random \'' .. pre_calculated .. '\'')
end
--[[--
Registers all event handlers.
]]
function ScatteredResources.register(config)
local noise_resource_threshold = config.noise_resource_threshold
local noise_variance = config.noise_variance
local cluster_mode = config.cluster_mode
local distance_probability_modifier = config.distance_probability_modifier
local resource_probability = config.resource_probability
local max_resource_probability = config.max_resource_probability
local resource_weights = config.resource_weights
local resource_richness_weights = config.resource_richness_weights
local distance_richness_modifier = config.distance_richness_modifier
local liquid_value_modifiers = config.liquid_value_modifiers
local resource_richness_values = config.resource_richness_values
local minimum_resource_distance = config.minimum_resource_distance
local cluster_yield_multiplier = config.cluster_yield_multiplier
local function spawn_resource(surface, x, y, distance)
local resource_name = get_name_by_weight(resource_weights)
if (minimum_resource_distance[resource_name] > distance) then
return
end
local min_max = resource_richness_values[get_name_by_weight(resource_richness_weights)]
local amount = ceil(random(min_max[1], min_max[2]) * (1 + ((distance / distance_richness_modifier) * 0.01)))
if liquid_value_modifiers[resource_name] then
amount = amount * liquid_value_modifiers[resource_name]
end
if (cluster_mode) then
amount = amount * cluster_yield_multiplier
end
local position = {x = x, y = y}
Template.resources(surface, {{name = resource_name, position = position, amount = amount}})
end
local seed
local function get_noise(surface, x, y)
seed = seed or surface.map_gen_settings.seed + surface.index + 200
return Perlin.noise(x * noise_variance, y * noise_variance, seed)
end
Event.add(Template.events.on_void_removed, function (event)
local position = event.position
local x = position.x
local y = position.y
local surface = event.surface
local distance = floor(sqrt(x * x + y * y))
if (cluster_mode and get_noise(surface, x, y) > noise_resource_threshold) then
spawn_resource(surface, x, y, distance)
return
end
local calculated_probability = resource_probability + ((distance / distance_probability_modifier) * 0.01)
local probability = max_resource_probability
if (calculated_probability < probability) then
probability = calculated_probability
end
-- cluster mode reduces the max probability to reduce max spread
if (cluster_mode) then
probability = probability * 0.5
end
if (probability > random()) then
spawn_resource(surface, x, y, distance)
end
end)
if (config.display_resource_fields) then
Event.add(defines.events.on_chunk_generated, function (event)
local surface = event.surface
local area = event.area
for x = area.left_top.x, area.left_top.x + 31 do
for y = area.left_top.y, area.left_top.y + 31 do
if get_noise(surface, x, y) >= noise_resource_threshold then
Debug.print_grid_value('ore', surface, {x = x, y = y}, nil, nil, true)
end
end
end
end)
end
end
function ScatteredResources.get_extra_map_info(config)
return [[Scattered Resources, resources are everywhere!
Scans of the mine have shown greater amounts of resources to be deeper in the mine]]
end
return ScatteredResources