1
0
mirror of https://github.com/Refactorio/RedMew.git synced 2024-12-14 10:13:13 +02:00
RedMew/map_gen/maps/diggy/feature/starting_zone.lua
2019-01-31 20:15:41 -05:00

104 lines
3.4 KiB
Lua

--[[-- info
Provides the ability to create a pre-configured starting zone.
]]
-- dependencies
local Event = require 'utils.event'
local Token = require 'utils.token'
local Template = require 'map_gen.maps.diggy.template'
local Retailer = require 'features.retailer'
local DiggyCaveCollapse = require 'map_gen.maps.diggy.feature.diggy_cave_collapse'
local RS = require 'map_gen.shared.redmew_surface'
local insert = table.insert
local random = math.random
local sqrt = math.sqrt
local floor = math.floor
local pairs = pairs
local raise_event = script.raise_event
-- this
local StartingZone = {}
--[[--
Registers all event handlers.
]]
function StartingZone.register(config)
local callback_token
local starting_zone_size = config.starting_size
local function on_chunk_generated(event)
if event.surface ~= RS.get_surface() then
return
end
local start_point_area = {{-0.9, -0.9}, {0.9, 0.9}}
local start_point_cleanup = {{-0.9, -0.9}, {1.9, 1.9}}
local surface = event.surface
-- hack to figure out whether the important chunks are generated via diggy.feature.refresh_map.
if (4 ~= surface.count_tiles_filtered({start_point_area, name = 'lab-dark-1'})) then
return
end
-- ensure a clean starting point
for _, entity in pairs(surface.find_entities_filtered({area = start_point_cleanup, type = 'resource'})) do
entity.destroy()
end
local tiles = {}
local rocks = {}
local dirt_range = floor(starting_zone_size * 0.5)
local rock_range = starting_zone_size - 2
local stress_hack = floor(starting_zone_size * 0.1)
for x = -starting_zone_size, starting_zone_size do
for y = -starting_zone_size, starting_zone_size do
local distance = floor(sqrt(x * x + y * y))
if (distance < starting_zone_size) then
if (distance > dirt_range) then
insert(tiles, {name = 'dirt-' .. random(1, 7), position = {x = x, y = y}})
else
insert(tiles, {name = 'stone-path', position = {x = x, y = y}})
end
if (distance > rock_range) then
insert(rocks, {name = 'rock-big', position = {x = x, y = y}})
end
-- hack to avoid starting area from collapsing
if (distance > stress_hack) then
DiggyCaveCollapse.stress_map_add(surface, {x = x, y = y}, -0.5)
end
end
end
end
Template.insert(surface, tiles, rocks)
local position = config.market_spawn_position;
local player_force = game.forces.player;
local market = surface.create_entity({name = 'market', position = position})
market.destructible = false
Retailer.set_market_group_label('player', 'Diggy Market')
Retailer.add_market('player', market)
player_force.add_chart_tag(surface, {
text = 'Market',
position = position,
})
raise_event(Template.events.on_placed_entity, {entity = market})
Event.remove_removable(defines.events.on_chunk_generated, callback_token)
end
callback_token = Token.register(on_chunk_generated)
Event.add_removable(defines.events.on_chunk_generated, callback_token)
end
return StartingZone