1
0
mirror of https://github.com/Refactorio/RedMew.git synced 2024-12-16 10:19:27 +02:00
RedMew/map_gen/maps/rail_grid/rail_grid_restrictions.lua
2019-10-08 10:51:07 +02:00

84 lines
2.2 KiB
Lua

local Event = require 'utils.event'
local Global = require 'utils.global'
local RestrictEntities = require 'map_gen.shared.entity_placement_restriction'
local Popup = require 'features.gui.popup'
local Token = require 'utils.token'
local floor = math.floor
local players_popuped = {}
Global.register(
players_popuped,
function(tbl)
players_popuped = tbl
end
)
local rail_entities = {
['straight-rail'] = true,
['curved-rail'] = true
}
local function all_on_landfill(entity)
local get_tile = entity.surface.get_tile
local area = entity.bounding_box
local left_top = area.left_top
local right_bottom = area.right_bottom
for x = floor(left_top.x), floor(right_bottom.x) do
for y = floor(left_top.y), floor(right_bottom.y) do
if get_tile(x, y).name ~= 'landfill' then
return false
end
end
end
return true
end
RestrictEntities.set_keep_alive_callback(
Token.register(
function(entity)
local name = entity.name
if name == 'entity-ghost' then
name = entity.ghost_name
end
if not rail_entities[name] then
return true
end
return all_on_landfill(entity)
end
)
)
-- On first time player places rail entity on invalid tile, show popup explaining the rail mechanic.
local function restricted_entity_destroyed(event)
local p = event.player
if not p or not p.valid then
return
end
if players_popuped[p.index] then
return
end
Popup.player(p, 'Rails can only be built on green tiles.', nil, nil, 'rail_grid')
players_popuped[p.index] = true
end
-- On player join print a notice explaining the rail mechanic
local function player_joined_game(event)
local player_index = event.player_index
local player = game.get_player(player_index)
if not player or not player.valid then
return
end
player.print("Welcome to RedMew's Rail Grids Map. Rails can only be built on green tiles.", {r = 0, g = 1, b = 0, a = 1})
end
Event.add(RestrictEntities.events.on_restricted_entity_destroyed, restricted_entity_destroyed)
Event.add(defines.events.on_player_joined_game, player_joined_game)