mirror of
https://github.com/Refactorio/RedMew.git
synced 2024-12-12 10:04:40 +02:00
597 lines
19 KiB
Lua
597 lines
19 KiB
Lua
require 'map_gen.presets.crash_site.blueprint_extractor'
|
|
require 'map_gen.presets.crash_site.entity_died_events'
|
|
require 'map_gen.presets.crash_site.weapon_balance'
|
|
|
|
local b = require 'map_gen.shared.builders'
|
|
local Global = require('utils.global')
|
|
local Random = require 'map_gen.shared.random'
|
|
local OutpostBuilder = require 'map_gen.presets.crash_site.outpost_builder'
|
|
local math = require "utils.math"
|
|
local degrees = math.degrees
|
|
|
|
-- leave seeds nil to have them filled in based on teh map seed.
|
|
local outpost_seed = nil --91000
|
|
local ore_seed = nil --92000
|
|
|
|
local small_iron_plate_factory = require 'map_gen.presets.crash_site.outpost_data.small_iron_plate_factory'
|
|
local medium_iron_plate_factory = require 'map_gen.presets.crash_site.outpost_data.medium_iron_plate_factory'
|
|
local big_iron_plate_factory = require 'map_gen.presets.crash_site.outpost_data.big_iron_plate_factory'
|
|
|
|
local small_copper_plate_factory = require 'map_gen.presets.crash_site.outpost_data.small_copper_plate_factory'
|
|
local medium_copper_plate_factory = require 'map_gen.presets.crash_site.outpost_data.medium_copper_plate_factory'
|
|
local big_copper_plate_factory = require 'map_gen.presets.crash_site.outpost_data.big_copper_plate_factory'
|
|
|
|
local small_stone_factory = require 'map_gen.presets.crash_site.outpost_data.small_stone_factory'
|
|
local medium_stone_factory = require 'map_gen.presets.crash_site.outpost_data.medium_stone_factory'
|
|
local big_stone_factory = require 'map_gen.presets.crash_site.outpost_data.big_stone_factory'
|
|
|
|
local small_gear_factory = require 'map_gen.presets.crash_site.outpost_data.small_gear_factory'
|
|
local medium_gear_factory = require 'map_gen.presets.crash_site.outpost_data.medium_gear_factory'
|
|
local big_gear_factory = require 'map_gen.presets.crash_site.outpost_data.big_gear_factory'
|
|
|
|
local small_circuit_factory = require 'map_gen.presets.crash_site.outpost_data.small_circuit_factory'
|
|
local medium_circuit_factory = require 'map_gen.presets.crash_site.outpost_data.medium_circuit_factory'
|
|
local big_circuit_factory = require 'map_gen.presets.crash_site.outpost_data.big_circuit_factory'
|
|
|
|
local small_ammo_factory = require 'map_gen.presets.crash_site.outpost_data.small_ammo_factory'
|
|
local medium_ammo_factory = require 'map_gen.presets.crash_site.outpost_data.medium_ammo_factory'
|
|
local big_ammo_factory = require 'map_gen.presets.crash_site.outpost_data.big_ammo_factory'
|
|
|
|
local small_weapon_factory = require 'map_gen.presets.crash_site.outpost_data.small_weapon_factory'
|
|
local medium_weapon_factory = require 'map_gen.presets.crash_site.outpost_data.medium_weapon_factory'
|
|
local big_weapon_factory = require 'map_gen.presets.crash_site.outpost_data.big_weapon_factory'
|
|
|
|
local small_science_factory = require 'map_gen.presets.crash_site.outpost_data.small_science_factory'
|
|
local medium_science_factory = require 'map_gen.presets.crash_site.outpost_data.medium_science_factory'
|
|
local big_science_factory = require 'map_gen.presets.crash_site.outpost_data.big_science_factory'
|
|
|
|
local small_oil_refinery = require 'map_gen.presets.crash_site.outpost_data.small_oil_refinery'
|
|
local medium_oil_refinery = require 'map_gen.presets.crash_site.outpost_data.medium_oil_refinery'
|
|
local big_oil_refinery = require 'map_gen.presets.crash_site.outpost_data.big_oil_refinery'
|
|
|
|
local small_chemical_factory = require 'map_gen.presets.crash_site.outpost_data.small_chemical_factory'
|
|
local medium_chemical_factory = require 'map_gen.presets.crash_site.outpost_data.medium_chemical_factory'
|
|
local big_chemical_factory = require 'map_gen.presets.crash_site.outpost_data.big_chemical_factory'
|
|
|
|
local small_power_factory = require 'map_gen.presets.crash_site.outpost_data.small_power_factory'
|
|
local medium_power_factory = require 'map_gen.presets.crash_site.outpost_data.medium_power_factory'
|
|
local big_power_factory = require 'map_gen.presets.crash_site.outpost_data.big_power_factory'
|
|
|
|
local thin_walls = require 'map_gen.presets.crash_site.outpost_data.thin_walls'
|
|
|
|
local function init()
|
|
local outpost_random = Random.new(outpost_seed, outpost_seed * 2)
|
|
|
|
local outpost_builder = OutpostBuilder.new(outpost_random)
|
|
|
|
local stage1a = {
|
|
small_iron_plate_factory,
|
|
small_gear_factory,
|
|
small_copper_plate_factory,
|
|
small_stone_factory
|
|
}
|
|
|
|
local stage1a_pos = {
|
|
{4, 5},
|
|
{5, 4},
|
|
{5, 6},
|
|
{6, 5}
|
|
}
|
|
|
|
local stage1b = {
|
|
small_circuit_factory,
|
|
small_science_factory,
|
|
small_oil_refinery,
|
|
small_chemical_factory,
|
|
small_power_factory
|
|
}
|
|
|
|
local stage1b_pos = {
|
|
{4, 4},
|
|
{4, 6},
|
|
{6, 4},
|
|
{6, 6},
|
|
{3, 5},
|
|
{5, 3},
|
|
{5, 7},
|
|
{7, 5}
|
|
}
|
|
|
|
local stage2 = {
|
|
medium_iron_plate_factory,
|
|
medium_copper_plate_factory,
|
|
medium_stone_factory,
|
|
medium_gear_factory,
|
|
medium_circuit_factory,
|
|
small_ammo_factory,
|
|
small_ammo_factory,
|
|
small_weapon_factory,
|
|
small_science_factory,
|
|
medium_science_factory,
|
|
medium_oil_refinery,
|
|
medium_chemical_factory,
|
|
medium_power_factory
|
|
}
|
|
|
|
local stage2_pos = {
|
|
{2, 5},
|
|
{5, 2},
|
|
{5, 8},
|
|
{8, 5},
|
|
{3, 3},
|
|
{3, 4},
|
|
{3, 6},
|
|
{3, 7},
|
|
{4, 3},
|
|
{4, 7},
|
|
{6, 3},
|
|
{6, 7},
|
|
{7, 3},
|
|
{7, 4},
|
|
{7, 6},
|
|
{7, 7}
|
|
}
|
|
|
|
local stage3 = {
|
|
big_iron_plate_factory,
|
|
big_copper_plate_factory,
|
|
big_stone_factory,
|
|
big_gear_factory,
|
|
big_circuit_factory,
|
|
medium_ammo_factory,
|
|
medium_ammo_factory,
|
|
medium_weapon_factory,
|
|
medium_science_factory,
|
|
big_science_factory,
|
|
big_oil_refinery,
|
|
big_chemical_factory,
|
|
big_power_factory
|
|
}
|
|
|
|
local stage4 = {
|
|
big_iron_plate_factory,
|
|
big_copper_plate_factory,
|
|
big_gear_factory,
|
|
big_circuit_factory,
|
|
big_ammo_factory,
|
|
big_ammo_factory,
|
|
big_ammo_factory,
|
|
big_weapon_factory,
|
|
big_weapon_factory,
|
|
big_weapon_factory,
|
|
big_science_factory,
|
|
big_oil_refinery,
|
|
big_chemical_factory
|
|
}
|
|
|
|
local function fast_remove(tbl, index)
|
|
local count = #tbl
|
|
if index > count then
|
|
return
|
|
elseif index < count then
|
|
tbl[index] = tbl[count]
|
|
end
|
|
|
|
tbl[count] = nil
|
|
end
|
|
|
|
local function itertor_builder(arr, random)
|
|
local copy = {}
|
|
|
|
return function()
|
|
if #copy == 0 then
|
|
for i = 1, #arr do
|
|
copy[i] = arr[i]
|
|
end
|
|
end
|
|
|
|
local i = random:next_int(1, #copy)
|
|
local res = copy[i]
|
|
|
|
fast_remove(copy, i)
|
|
|
|
return res
|
|
end
|
|
end
|
|
|
|
local stage1a_iter = itertor_builder(stage1a, outpost_random)
|
|
local stage1b_iter = itertor_builder(stage1b, outpost_random)
|
|
|
|
local stage2_iter = itertor_builder(stage2, outpost_random)
|
|
local stage3_iter = itertor_builder(stage3, outpost_random)
|
|
local stage4_iter = itertor_builder(stage4, outpost_random)
|
|
|
|
local start_outpost = outpost_builder:do_outpost(thin_walls)
|
|
|
|
local function remove_water(shape)
|
|
return function(x, y, world)
|
|
local tile = shape(x, y, world)
|
|
|
|
if type(tile) == 'table' then
|
|
local gen_tile = world.surface.get_tile(world.x, world.y)
|
|
if gen_tile.collides_with('water-tile') then
|
|
tile.tile = 'grass-1'
|
|
end
|
|
else
|
|
local gen_tile = world.surface.get_tile(world.x, world.y)
|
|
if gen_tile.collides_with('water-tile') then
|
|
tile = 'grass-1'
|
|
end
|
|
end
|
|
|
|
return tile
|
|
end
|
|
end
|
|
|
|
start_outpost = remove_water(start_outpost)
|
|
|
|
local start_patch = b.circle(10)
|
|
local start_iron_patch =
|
|
b.resource(
|
|
b.translate(start_patch, -32, -32),
|
|
'iron-ore',
|
|
function()
|
|
return 1500
|
|
end
|
|
)
|
|
local start_copper_patch =
|
|
b.resource(
|
|
b.translate(start_patch, 32, -32),
|
|
'copper-ore',
|
|
function()
|
|
return 1200
|
|
end
|
|
)
|
|
local start_stone_patch =
|
|
b.resource(
|
|
b.translate(start_patch, 32, 32),
|
|
'stone',
|
|
function()
|
|
return 900
|
|
end
|
|
)
|
|
local start_coal_patch =
|
|
b.resource(
|
|
b.translate(start_patch, -32, 32),
|
|
'coal',
|
|
function()
|
|
return 1350
|
|
end
|
|
)
|
|
|
|
local start_resources = b.any({start_iron_patch, start_copper_patch, start_stone_patch, start_coal_patch})
|
|
start_outpost = b.apply_entity(start_outpost, start_resources)
|
|
|
|
local water_corner =
|
|
b.any {
|
|
b.translate(b.rectangle(6, 16), -6, 0),
|
|
b.translate(b.rectangle(16, 6), 0, -6)
|
|
}
|
|
water_corner = b.change_tile(water_corner, true, 'water')
|
|
water_corner = b.translate(water_corner, -60, -60)
|
|
|
|
start_outpost =
|
|
b.any {
|
|
water_corner,
|
|
b.rotate(water_corner, degrees(90)),
|
|
b.rotate(water_corner, degrees(180)),
|
|
b.rotate(water_corner, degrees(270)),
|
|
start_outpost
|
|
}
|
|
|
|
local pattern = {}
|
|
|
|
for r = 1, 10 do
|
|
local row = {}
|
|
pattern[r] = row
|
|
end
|
|
|
|
pattern[5][5] = start_outpost
|
|
|
|
local outpost_offset = 59
|
|
local grid_block_size = 190
|
|
local grid_number_of_blocks = 9
|
|
|
|
local half_total_size = grid_block_size * 0.5 * 8
|
|
|
|
for _, pos in ipairs(stage1a_pos) do
|
|
local r, c = pos[1], pos[2]
|
|
|
|
local row = pattern[r]
|
|
|
|
local template = stage1a_iter()
|
|
local shape = outpost_builder:do_outpost(template)
|
|
|
|
local x = outpost_random:next_int(-outpost_offset, outpost_offset)
|
|
local y = outpost_random:next_int(-outpost_offset, outpost_offset)
|
|
shape = b.translate(shape, x, y)
|
|
|
|
row[c] = shape
|
|
end
|
|
|
|
for _, pos in ipairs(stage1b_pos) do
|
|
local r, c = pos[1], pos[2]
|
|
|
|
local row = pattern[r]
|
|
|
|
local template = stage1b_iter()
|
|
local shape = outpost_builder:do_outpost(template)
|
|
|
|
local x = outpost_random:next_int(-outpost_offset, outpost_offset)
|
|
local y = outpost_random:next_int(-outpost_offset, outpost_offset)
|
|
shape = b.translate(shape, x, y)
|
|
|
|
row[c] = shape
|
|
end
|
|
|
|
for _, pos in ipairs(stage2_pos) do
|
|
local r, c = pos[1], pos[2]
|
|
|
|
local row = pattern[r]
|
|
|
|
local template = stage2_iter()
|
|
local shape = outpost_builder:do_outpost(template)
|
|
|
|
local x = outpost_random:next_int(-outpost_offset, outpost_offset)
|
|
local y = outpost_random:next_int(-outpost_offset, outpost_offset)
|
|
shape = b.translate(shape, x, y)
|
|
|
|
row[c] = shape
|
|
end
|
|
|
|
for r = 2, 8 do
|
|
local row = pattern[r]
|
|
for c = 2, 8 do
|
|
if not row[c] then
|
|
local template = stage3_iter()
|
|
local shape = outpost_builder:do_outpost(template)
|
|
|
|
local x = outpost_random:next_int(-outpost_offset, outpost_offset)
|
|
local y = outpost_random:next_int(-outpost_offset, outpost_offset)
|
|
shape = b.translate(shape, x, y)
|
|
|
|
row[c] = shape
|
|
end
|
|
end
|
|
end
|
|
|
|
for r = 1, grid_number_of_blocks do
|
|
local row = pattern[r]
|
|
for c = 1, grid_number_of_blocks do
|
|
if not row[c] then
|
|
local template = stage4_iter()
|
|
local shape = outpost_builder:do_outpost(template)
|
|
|
|
local x = outpost_random:next_int(-outpost_offset, outpost_offset)
|
|
local y = outpost_random:next_int(-outpost_offset, outpost_offset)
|
|
shape = b.translate(shape, x, y)
|
|
|
|
row[c] = shape
|
|
end
|
|
end
|
|
end
|
|
|
|
local outposts = b.grid_pattern(pattern, 10, 10, grid_block_size, grid_block_size)
|
|
--outposts = b.if_else(outposts, b.full_shape)
|
|
|
|
local spawners = {
|
|
'biter-spawner',
|
|
'spitter-spawner'
|
|
}
|
|
|
|
local worms = {
|
|
'small-worm-turret',
|
|
'medium-worm-turret',
|
|
'big-worm-turret'
|
|
}
|
|
|
|
local max_spawner_chance = 1 / 256
|
|
local spawner_chance_factor = 1 / (256 * 512)
|
|
local max_worm_chance = 1 / 32
|
|
local worm_chance_factor = 1 / (32 * 512)
|
|
|
|
local scale_factor = 1 / 32
|
|
|
|
local function enemy(x, y, world)
|
|
local wx, wy = world.x, world.y
|
|
local d = math.sqrt(wx * wx + wy * wy)
|
|
|
|
--[[ if Perlin.noise(x * scale_factor, y * scale_factor, enemy_seed) < 0 then
|
|
return nil
|
|
end ]]
|
|
local spawner_chance = d - 128
|
|
|
|
if spawner_chance > 0 then
|
|
spawner_chance = spawner_chance * spawner_chance_factor
|
|
spawner_chance = math.min(spawner_chance, max_spawner_chance)
|
|
|
|
if math.random() < spawner_chance then
|
|
return {name = spawners[math.random(2)]}
|
|
end
|
|
end
|
|
|
|
local worm_chance = d - 128
|
|
|
|
if worm_chance > 0 then
|
|
worm_chance = worm_chance * worm_chance_factor
|
|
worm_chance = math.min(worm_chance, max_worm_chance)
|
|
|
|
if math.random() < worm_chance then
|
|
if d < 256 then
|
|
return {name = 'small-worm-turret'}
|
|
else
|
|
local max_lvl
|
|
local min_lvl
|
|
if d < 512 then
|
|
max_lvl = 2
|
|
min_lvl = 1
|
|
else
|
|
max_lvl = 3
|
|
min_lvl = 2
|
|
end
|
|
local lvl = math.random() ^ (384 / d) * max_lvl
|
|
lvl = math.ceil(lvl)
|
|
--local lvl = math.floor(d / 256) + 1
|
|
lvl = math.clamp(lvl, min_lvl, 3)
|
|
return {name = worms[lvl]}
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
local enemy_shape = b.apply_entity(b.full_shape, enemy)
|
|
|
|
local ores_patch = b.circle(16)
|
|
local function value(base, mult, pow)
|
|
return function(x, y)
|
|
local d_sq = x * x + y * y
|
|
return base + mult * d_sq ^ (pow / 2) -- d^pow
|
|
end
|
|
end
|
|
|
|
local function non_transform(shape)
|
|
return shape
|
|
end
|
|
|
|
local function uranium_transform(shape)
|
|
return b.scale(shape, 0.5)
|
|
end
|
|
|
|
local function oil_transform(shape)
|
|
shape = b.scale(shape, 0.5)
|
|
shape = b.throttle_world_xy(shape, 1, 5, 1, 5)
|
|
return shape
|
|
end
|
|
|
|
local ores = {
|
|
{weight = 300},
|
|
{transform = non_transform, resource = 'iron-ore', value = value(500, 0.75, 1.1), weight = 16},
|
|
{transform = non_transform, resource = 'copper-ore', value = value(400, 0.75, 1.1), weight = 10},
|
|
{transform = non_transform, resource = 'stone', value = value(250, 0.3, 1.05), weight = 5},
|
|
{transform = non_transform, resource = 'coal', value = value(400, 0.8, 1.075), weight = 8},
|
|
{transform = uranium_transform, resource = 'uranium-ore', value = value(200, 0.3, 1.025), weight = 3},
|
|
{transform = oil_transform, resource = 'crude-oil', value = value(180000, 50, 1.025), weight = 6}
|
|
}
|
|
|
|
local total_ore_weights = {}
|
|
local ore_t = 0
|
|
for _, v in ipairs(ores) do
|
|
ore_t = ore_t + v.weight
|
|
table.insert(total_ore_weights, ore_t)
|
|
end
|
|
|
|
local random_ore = Random.new(ore_seed, ore_seed * 2)
|
|
local ore_pattern = {}
|
|
|
|
for r = 1, 50 do
|
|
local row = {}
|
|
ore_pattern[r] = row
|
|
for c = 1, 50 do
|
|
local i = random_ore:next_int(1, ore_t)
|
|
local index = table.binary_search(total_ore_weights, i)
|
|
if (index < 0) then
|
|
index = bit32.bnot(index)
|
|
end
|
|
local ore_data = ores[index]
|
|
|
|
local transform = ore_data.transform
|
|
if not transform then
|
|
row[c] = b.no_entity
|
|
else
|
|
local ore_shape = transform(ores_patch)
|
|
|
|
local x = random_ore:next_int(-24, 24)
|
|
local y = random_ore:next_int(-24, 24)
|
|
ore_shape = b.translate(ore_shape, x, y)
|
|
|
|
local ore = b.resource(ore_shape, ore_data.resource, ore_data.value)
|
|
row[c] = ore
|
|
end
|
|
end
|
|
end
|
|
|
|
local ore_grid = b.grid_pattern_full_overlap(ore_pattern, 50, 50, 64, 64)
|
|
ore_grid = b.choose(b.rectangle(138), b.no_entity, ore_grid)
|
|
|
|
local map = b.if_else(outposts, enemy_shape)
|
|
--map = b.change_tile(map, true, 'grass-1')
|
|
map = b.if_else(map, b.full_shape)
|
|
|
|
map = b.translate(map, -half_total_size, -half_total_size)
|
|
|
|
map = b.apply_entity(map, ore_grid)
|
|
|
|
--map = b.apply_entity(map, enemy)
|
|
|
|
local market = {
|
|
callback = outpost_builder.market_set_items_callback,
|
|
data = {
|
|
{name = 'raw-wood', price = 1},
|
|
{name = 'iron-plate', price = 2},
|
|
{name = 'stone', price = 2},
|
|
{name = 'coal', price = 1.25},
|
|
{name = 'raw-fish', price = 4},
|
|
{name = 'firearm-magazine', price = 5},
|
|
{name = 'science-pack-1', price = 20},
|
|
{name = 'science-pack-2', price = 40},
|
|
{name = 'military-science-pack', price = 80},
|
|
{name = 'science-pack-3', price = 120},
|
|
{name = 'production-science-pack', price = 240},
|
|
{name = 'high-tech-science-pack', price = 360},
|
|
{name = 'small-plane', price = 100}
|
|
}
|
|
}
|
|
|
|
local factory = {
|
|
callback = outpost_builder.magic_item_crafting_callback,
|
|
data = {
|
|
output = {min_rate = 0.5 / 60, distance_factor = 0, item = 'coin'}
|
|
}
|
|
}
|
|
|
|
local spawn = {
|
|
size = 2,
|
|
[1] = {
|
|
market = market,
|
|
[15] = {entity = {name = 'market', callback = 'market'}}
|
|
},
|
|
[2] = {
|
|
force = 'player',
|
|
factory = factory,
|
|
[15] = {entity = {name = 'electric-furnace', callback = 'factory'}}
|
|
}
|
|
}
|
|
|
|
local spawn_shape = outpost_builder.to_shape(spawn)
|
|
spawn_shape = b.change_tile(spawn_shape, false, 'stone-path')
|
|
|
|
map = b.choose(b.rectangle(16, 16), spawn_shape, map)
|
|
|
|
local bounds = b.rectangle(grid_block_size * grid_number_of_blocks)
|
|
map = b.choose(bounds, map, b.empty_shape)
|
|
|
|
return map
|
|
end
|
|
|
|
local map
|
|
|
|
Global.register_init(
|
|
{},
|
|
function(tbl)
|
|
local seed = game.surfaces[1].map_gen_settings.seed
|
|
tbl.outpost_seed = outpost_seed or seed
|
|
tbl.ore_seed = ore_seed or seed
|
|
|
|
global.scenario.config.fish_market.enable = false
|
|
end,
|
|
function(tbl)
|
|
outpost_seed = tbl.outpost_seed
|
|
ore_seed = tbl.ore_seed
|
|
map = init()
|
|
end
|
|
)
|
|
|
|
return function(x, y, world)
|
|
return map(x, y, world)
|
|
end
|