RedMew/map_gen/maps/concrete_jungle.lua

local floor = math.floor
local RestrictEntities = require 'map_gen.shared.entity_placement_restriction'
local Event = require 'utils.event'
local b = require 'map_gen.shared.builders'

local ScenarioInfo = require 'features.gui.info'

ScenarioInfo.set_map_name('Concrete Jungle')
ScenarioInfo.set_map_description([[
Extensive underground mining have resulted in brittle soil.
New regulations require heavy objects being placed on top,
proper materials to support the soil!
]])

--- Items explicitly allowed on ores
RestrictEntities.add_allowed(
    {
        'transport-belt',
        'fast-transport-belt',
        'express-transport-belt',
        'underground-belt',
        'fast-underground-belt',
        'express-underground-belt',
        'small-electric-pole',
        'burner-mining-drill',
        'pumpjack',
        'car',
        'tank',
        'pipe',
        'pipe-to-ground',
        'offshore-pump'
    }
)

local tiles_tier_1 = {
    ['stone-path'] = true
}

local tiles_tier_2 = {
    ['concrete'] = true,
    ['hazard-concrete-right'] = true,
    ['hazard-concrete-left'] = true
}

local tiles_tier_3 = {
    ['refined-concrete'] = true,
    ['refined-hazard-concrete-right'] = true,
    ['refined-hazard-concrete-left'] = true
}

local tier_2 = {
    ['oil-refinery'] = true,
    ['chemical-plant'] = true,
    ['storage-tank'] = true,
    ['straight-rail'] = true,
    ['curved-rail'] = true,
    ['train-stop'] = true,
    ['solar-panel'] = true,
    ['flamethrower-turret'] = true,
    ['assembling-machine-2'] = true,
    ['steel-furnace'] = true,
    ['iron-chest'] = true,
    ['fast-inserter'] = true,
    ['filter-inserter'] = true,
    ['accumulator'] = true,
    ['big-electric-pole'] = true
}

local tier_3 = {
    ['rocket-silo'] = true,
    ['nuclear-reactor'] = true,
    ['centrifuge'] = true,
    ['heat-exchanger'] = true,
    ['heat-pipe'] = true,
    ['steam-turbine'] = true,
    ['artillery-turret'] = true,
    ['roboport'] = true,
    ['beacon'] = true,
    ['assembling-machine-3'] = true,
    ['electric-furnace'] = true,
    ['substation'] = true,
    ['laser-turret'] = true,
    ['steel-chest'] = true,
    ['stack-inserter'] = true,
    ['stack-filter-inserter'] = true,
    ['logistic-chest-active-provider'] = true,
    ['logistic-chest-passive-provider'] = true,
    ['logistic-chest-buffer'] = true,
    ['logistic-chest-storage'] = true,
    ['logistic-chest-requester'] = true
}

local tier_2_items = '[item=rail]'

for k, _ in pairs(tier_2) do
    if not (k == 'straight-rail' or k == 'curved-rail') then
        tier_2_items = tier_2_items .. ' [item=' .. k .. ']'
    end
end

local tier_3_items = ''

local tier_3_counter = 0
for k, _ in pairs(tier_3) do
    tier_3_items = tier_3_items .. ' [item=' .. k .. ']'
    tier_3_counter = tier_3_counter + 1
    if tier_3_counter > 14 then
        tier_3_counter = 0
        tier_3_items = tier_3_items .. '\n'
    end
end

ScenarioInfo.add_map_extra_info(
    [[
You may only build the factory on [item=stone-brick] [item=concrete] [item=refined-concrete].

Exceptions:
 [item=burner-mining-drill] [item=pumpjack] [item=small-electric-pole] [item=car] [item=tank] [item=pipe] [item=pipe-to-ground] [item=offshore-pump]
 [item=transport-belt] [item=fast-transport-belt] [item=express-transport-belt]  [item=underground-belt] [item=fast-underground-belt] [item=express-underground-belt]

Some buildings/entities require better ground support!


Ground support minimum concrete:
]] ..
        tier_2_items .. [[


Ground support minimum refined concrete:
]] .. tier_3_items
)

--- The logic for checking that there are resources under the entity's position
RestrictEntities.set_keep_alive_callback(
    function(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
        local name = entity.name

        if name == 'entity-ghost' then
            name = entity.ghost_name
        end

        local is_tier_2 = tier_2[name]
        local is_tier_3 = tier_3[name]

        for x = floor(left_top.x), floor(right_bottom.x) do
            for y = floor(left_top.y), floor(right_bottom.y) do
                local tile_name = get_tile(x, y).name
                if is_tier_2 then
                    if not (tiles_tier_2[tile_name] or tiles_tier_3[tile_name]) then
                        return false
                    end
                elseif is_tier_3 then
                    if not (tiles_tier_3[tile_name]) then
                        return false
                    end
                else
                    if not (tiles_tier_1[tile_name] or tiles_tier_2[tile_name] or tiles_tier_3[tile_name]) then
                        return false
                    end
                end
            end
        end
        return true
    end
)

--- Warning for players when their entities are destroyed (needs to be pre because of the stack)
local function on_destroy(event)
    local p = game.get_player(event.player_index)
    local name = event.stack.name
    if p and p.valid then
        if not (name == 'blueprint') then
            local tier = 'stone path'
            if (tier_2[name]) then
                tier = 'concrete'
            elseif (tier_3[name]) then
                tier = 'refined concrete'
            end
            p.print('[color=yellow]This [/color][color=red]' .. name .. '[/color][color=yellow] cannot be placed here, it needs ground support of at least [/color][color=red]' .. tier .. '[/color]')
        else
            p.print('[color=yellow]Some parts of this [/color][color=red]blueprint[/color][color=yellow] cannot be placed here, they need better ground support![/color]')
        end
    end
end

Event.add(RestrictEntities.events.on_pre_restricted_entity_destroyed, on_destroy)

local shape = b.circle(50)
local stone_path = b.tile('stone-path')

shape = b.invert(shape)
local map = b.if_else(shape, stone_path)

return map