mirror of
https://github.com/Refactorio/RedMew.git
synced 2024-12-16 10:19:27 +02:00
117 lines
3.5 KiB
Lua
117 lines
3.5 KiB
Lua
--[[-- info
|
|
Provides the ability to spawn random ores all over the place.
|
|
]]
|
|
|
|
-- dependencies
|
|
local Event = require 'utils.event'
|
|
local Template = require 'map_gen.Diggy.Template'
|
|
|
|
-- this
|
|
local ScatteredResources = {}
|
|
|
|
global.ScatteredResources = {
|
|
can_spawn_resources = false
|
|
}
|
|
|
|
local function get_name_by_random(collection)
|
|
local random = math.random()
|
|
local current = 0
|
|
|
|
for name, probability in pairs(collection) do
|
|
current = current + probability
|
|
if (current >= random) then
|
|
return name
|
|
end
|
|
end
|
|
|
|
error('Current \'' .. current .. '\' should be higher or equal to random \'' .. random .. '\'')
|
|
end
|
|
|
|
local function spawn_resource(config, surface, x, y, distance)
|
|
local resource_name = get_name_by_random(config.resource_chances)
|
|
|
|
if (config.minimum_resource_distance[resource_name] > distance) then
|
|
return
|
|
end
|
|
|
|
local amount
|
|
|
|
if ('tree' == resource_name) then
|
|
local trees = {
|
|
'dead-dry-hairy-tree',
|
|
'dead-grey-trunk',
|
|
'dead-tree-desert',
|
|
}
|
|
|
|
resource_name = trees[math.random(1, 3)]
|
|
amount = 1
|
|
else
|
|
local min_max = config.resource_richness_values[get_name_by_random(config.resource_richness_probability)]
|
|
amount = math.ceil(math.random(min_max[1], min_max[2]) * (1 + ((distance / config.distance_richness_modifier) / 100)))
|
|
end
|
|
|
|
if ('crude-oil' == resource_name) then
|
|
amount = amount * config.oil_value_modifier
|
|
end
|
|
|
|
Template.insert(surface, {}, {{name = resource_name, position = {x = x, y = y}, amount = amount}})
|
|
end
|
|
|
|
--[[--
|
|
Registers all event handlers.
|
|
]]
|
|
function ScatteredResources.register(config)
|
|
function sum(t)
|
|
local sum = 0
|
|
for _, v in pairs(t) do
|
|
sum = sum + v
|
|
end
|
|
|
|
return sum
|
|
end
|
|
|
|
local resource_sum = sum(config.features.ScatteredResources.resource_chances)
|
|
if (1 ~= resource_sum) then
|
|
error('Expected a sum of 1.00, got \'' .. resource_sum .. '\' for config.feature.ScatteredResources.resource_chances.')
|
|
end
|
|
|
|
local richness_sum = sum(config.features.ScatteredResources.resource_richness_probability)
|
|
if (1 ~= richness_sum) then
|
|
error('Expected a sum of 1.00, got \'' .. richness_sum .. '\' for config.feature.ScatteredResources.resource_richness_probability.')
|
|
end
|
|
|
|
Event.add(Template.events.on_void_removed, function(event)
|
|
if (not global.ScatteredResources.can_spawn_resources) then
|
|
return
|
|
end
|
|
|
|
local x = event.old_tile.position.x
|
|
local y = event.old_tile.position.y
|
|
|
|
local feature_config = config.features.ScatteredResources;
|
|
|
|
local distance = math.floor(math.sqrt(x^2 + y^2))
|
|
local calculated_probability = feature_config.resource_probability + ((distance / feature_config.distance_probability_modifier) / 100)
|
|
local probability = 0.7
|
|
|
|
if (calculated_probability < probability) then
|
|
probability = calculated_probability
|
|
end
|
|
|
|
if (probability > math.random()) then
|
|
spawn_resource(feature_config, event.surface, x, y, distance)
|
|
end
|
|
end)
|
|
end
|
|
|
|
--[[--
|
|
Initializes the Feature.
|
|
|
|
@param config Table {@see Diggy.Config}.
|
|
]]
|
|
function ScatteredResources.initialize(config)
|
|
global.ScatteredResources.can_spawn_resources = true
|
|
end
|
|
|
|
return ScatteredResources
|