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