mirror of
https://github.com/Refactorio/RedMew.git
synced 2024-12-12 10:04:40 +02:00
74 lines
1.8 KiB
Lua
74 lines
1.8 KiB
Lua
local Global = require 'utils.global'
|
|
local RS = require 'map_gen.shared.redmew_surface'
|
|
local table = require 'utils.table'
|
|
|
|
local seed = nil -- Set to number to force seed.
|
|
|
|
local resource_rate = 768 -- Number of tiles per resouce on average.
|
|
|
|
local generator
|
|
|
|
Global.register_init(
|
|
{},
|
|
function(tbl)
|
|
tbl.generator = game.create_random_generator()
|
|
tbl.seed = seed or RS.get_surface().map_gen_settings.seed
|
|
end,
|
|
function(tbl)
|
|
generator = tbl.generator
|
|
seed = tbl.seed
|
|
end
|
|
)
|
|
|
|
local function value(base, mult, pow)
|
|
return function(x, y)
|
|
local d_sq = x * x + y * y
|
|
return base + mult * d_sq ^ (pow / 2)
|
|
end
|
|
end
|
|
|
|
return function(tiles_per_resource)
|
|
resource_rate = tiles_per_resource or 768
|
|
|
|
local ores = {
|
|
{'iron-ore', value(40000, 20, 1.5), 100},
|
|
{'copper-ore', value(40000, 20, 1.5), 60},
|
|
{'coal', value(40000, 20, 1.25), 25},
|
|
{'stone', value(20000, 20, 1.25), 15},
|
|
{'uranium-ore', value(5000, 5, 1.35), 5},
|
|
{'crude-oil', value(600000, 200, 1.25), 4}
|
|
}
|
|
|
|
local total_weights = {}
|
|
local t = 0
|
|
for _, v in ipairs(ores) do
|
|
t = t + v[3]
|
|
table.insert(total_weights, t)
|
|
end
|
|
|
|
t = resource_rate * t
|
|
table.insert(ores, {false})
|
|
table.insert(total_weights, t)
|
|
|
|
return function(x, y, world)
|
|
local r_seed = bit32.band(x * 374761393 + y * 668265263 + seed, 4294967295)
|
|
generator.re_seed(r_seed)
|
|
local i = generator(t)
|
|
|
|
local index = table.binary_search(total_weights, i)
|
|
if (index < 0) then
|
|
index = bit32.bnot(index)
|
|
end
|
|
|
|
local ore_data = ores[index]
|
|
local ore = ore_data[1]
|
|
if not ore then
|
|
return
|
|
end
|
|
|
|
local amount = ore_data[2](world.x, world.y)
|
|
|
|
return {name = ore, amount = amount}
|
|
end
|
|
end
|