RedMew/map_gen/Diggy/Feature/DiggyCaveCollapse.lua

--[[-- info
    Provides the ability to collapse caves when digging.
]]

-- dependencies
require 'utils.list_utils'

local Event = require 'utils.event'
local Template = require 'map_gen.Diggy.Template'
local Mask = require 'map_gen.Diggy.Mask'
local PressureMap = require 'map_gen.Diggy.PressureMap'

-- this
local DiggyCaveCollapse = {}

--[[--
    @param surface LuaSurface
    @param position Position with x and y
    @param strength positive increases pressure, negative decreases pressure
]]
local function update_pressure_map(surface, position, strength)
    Mask.blur(position.x, position.y, strength, function (x, y, fraction)
        PressureMap.add(surface, {x = x, y = y}, fraction)
    end)

    PressureMap.process_maxed_values_buffer(surface, function (positions)
        local entities = {}
        local tiles = {}

        for _, position in pairs(positions) do
            table.insert(entities, {position = {x = position.x, y = position.y - 1}, name = 'sand-rock-big'})
            table.insert(entities, {position = {x = position.x + 1, y = position.y}, name = 'sand-rock-big'})
            table.insert(entities, {position = {x = position.x, y = position.y + 1}, name = 'sand-rock-big'})
            table.insert(entities, {position = {x = position.x - 1, y = position.y}, name = 'sand-rock-big'})
            table.insert(tiles, {position = {x = position.x, y = position.y}, name = 'out-of-map'})
        end

        for _, new_spawn in pairs({entities, tiles}) do
            for _, tile in pairs(new_spawn) do
                for _, entity in pairs(surface.find_entities_filtered({position = tile.position})) do
                    pcall(function() entity.die() end)
                    pcall(function() entity.destroy() end)
                end
            end
        end

        Template.insert(surface, tiles, entities, false)
    end)
end

--[[--
    Registers all event handlers.]

    @param config Table {@see Diggy.Config}.
]]
function DiggyCaveCollapse.register(config)
    local support_beam_entities = config.features.DiggyCaveCollapse.support_beam_entities;

    Event.add(defines.events.on_robot_built_entity, function(event)
        local strength = support_beam_entities[event.created_entity.name]

        if (not strength) then
            return
        end

        update_pressure_map(event.created_entity.surface, {
            x = event.created_entity.position.x,
            y = event.created_entity.position.y,
        }, -1 * strength)
    end)

    Event.add(defines.events.on_built_entity, function(event)
        local strength = support_beam_entities[event.created_entity.name]


        if (not strength) then
            return
        end

        update_pressure_map(event.created_entity.surface, {
            x = event.created_entity.position.x,
            y = event.created_entity.position.y,
        }, -1 * strength)
    end)

    Event.add(Template.events.on_placed_entity, function(event)
        local strength = support_beam_entities[event.entity.name]


        if (not strength) then
            return
        end

        update_pressure_map(event.entity.surface, {
            x = event.entity.position.x,
            y = event.entity.position.y,
        }, -1 * strength)
    end)

    Event.add(defines.events.on_entity_died, function(event)
        local strength = support_beam_entities[event.entity.name]

        if (not strength) then
            return
        end

        update_pressure_map(event.entity.surface, {
            x = event.entity.position.x,
            y = event.entity.position.y,
        }, strength)
    end)

    Event.add(defines.events.on_player_mined_entity, function(event)
        local strength = support_beam_entities[event.entity.name]

        if (not strength) then
            return
        end

        update_pressure_map(event.entity.surface, {
            x = event.entity.position.x,
            y = event.entity.position.y,
        }, strength)
    end)

    Event.add(Template.events.on_void_removed, function(event)
        local strength = support_beam_entities['out-of-map']

        update_pressure_map(event.surface, {
            x = event.old_tile.position.x,
            y = event.old_tile.position.y,
        }, strength)
    end)

    Event.add(Template.events.on_void_added, function(event)
        local strength = support_beam_entities['out-of-map']

        update_pressure_map(event.surface, {
            x = event.old_tile.position.x,
            y = event.old_tile.position.y,
        }, -1  * strength)
    end)
end

--[[--
    Initializes the Feature.

    @param config Table {@see Diggy.Config}.
]]
function DiggyCaveCollapse.initialize(config)

end

return DiggyCaveCollapse