ComfyFactorio/maps/scrap_towny_ffa/terrain.lua

local Event = require 'utils.event'
local get_noise = require 'utils.get_noise'
local ScenarioTable = require 'maps.scrap_towny_ffa.table'
local Scrap = require 'maps.scrap_towny_ffa.scrap'
local get_perlin = require 'utils.get_perlin'

local table_insert = table.insert
local math_random = math.random
local math_floor = math.floor

local scrap_entities = {
    -- simple entity with owner
    {name = 'crash-site-spaceship-wreck-small-1'}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-small-1'}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-small-2'}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-small-3'}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-small-4'}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-small-5'}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-small-6'} -- these do not have mining animation
}

local scrap_entities_index = table.size(scrap_entities)

local scrap_containers = {
    -- containers
    {name = 'big-ship-wreck-1', size = 3}, -- these are not mineable normally
    {name = 'big-ship-wreck-1', size = 3}, -- these are not mineable normally
    {name = 'big-ship-wreck-1', size = 3}, -- these are not mineable normally
    {name = 'big-ship-wreck-2', size = 3}, -- these are not mineable normally
    {name = 'big-ship-wreck-2', size = 3}, -- these are not mineable normally
    {name = 'big-ship-wreck-2', size = 3}, -- these are not mineable normally
    {name = 'big-ship-wreck-3', size = 3}, -- these are not mineable normally
    {name = 'big-ship-wreck-3', size = 3}, -- these are not mineable normally
    {name = 'big-ship-wreck-3', size = 3}, -- these are not mineable normally
    {name = 'crash-site-chest-1', size = 8}, -- these do not have mining animation
    {name = 'crash-site-chest-2', size = 8}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-medium-1', size = 1}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-medium-1', size = 1}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-medium-1', size = 1}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-medium-1', size = 1}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-medium-2', size = 1}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-medium-2', size = 1}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-medium-2', size = 1}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-medium-2', size = 1}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-medium-3', size = 1}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-medium-3', size = 1}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-medium-3', size = 1}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-medium-3', size = 1}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-big-1', size = 2}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-big-1', size = 2}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-big-1', size = 2}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-big-2', size = 2}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-big-2', size = 2}, -- these do not have mining animation
    {name = 'crash-site-spaceship-wreck-big-2', size = 2} -- these do not have mining animation
}
local scrap_containers_index = table.size(scrap_containers)

-- loot chances and amounts for scrap containers

local container_loot_chance = {
    {name = 'advanced-circuit', chance = 15},
    --{name = "artillery-shell", chance = 1},
    {name = 'battery', chance = 15},
    {name = 'cannon-shell', chance = 4},
    --{name = "cluster-grenade", chance = 2},
    {name = 'construction-robot', chance = 1},
    {name = 'copper-cable', chance = 250},
    {name = 'copper-plate', chance = 250},
    {name = 'crude-oil-barrel', chance = 30},
    {name = 'defender-capsule', chance = 5},
    {name = 'destroyer-capsule', chance = 1},
    {name = 'distractor-capsule', chance = 2},
    {name = 'electric-engine-unit', chance = 2},
    {name = 'electronic-circuit', chance = 150},
    {name = 'empty-barrel', chance = 10},
    {name = 'engine-unit', chance = 5},
    {name = 'explosive-cannon-shell', chance = 2},
    --{name = "explosive-rocket", chance = 3},
    --{name = "explosive-uranium-cannon-shell", chance = 1},
    {name = 'explosives', chance = 5},
    {name = 'grenade', chance = 10},
    {name = 'heavy-oil-barrel', chance = 20},
    {name = 'iron-gear-wheel', chance = 500},
    {name = 'iron-plate', chance = 500},
    {name = 'iron-stick', chance = 50},
    {name = 'land-mine', chance = 3},
    {name = 'light-oil-barrel', chance = 20},
    {name = 'logistic-robot', chance = 1},
    {name = 'low-density-structure', chance = 1},
    {name = 'lubricant-barrel', chance = 20},
    {name = 'nuclear-fuel', chance = 1},
    {name = 'petroleum-gas-barrel', chance = 30},
    {name = 'pipe', chance = 100},
    {name = 'pipe-to-ground', chance = 10},
    {name = 'plastic-bar', chance = 5},
    {name = 'processing-unit', chance = 2},
    --{name = "rocket", chance = 3},
    --{name = "rocket-control-unit", chance = 1},
    --{name = "rocket-fuel", chance = 3},
    {name = 'solid-fuel', chance = 100},
    {name = 'steel-plate', chance = 150},
    {name = 'sulfuric-acid-barrel', chance = 15},
    --{name = "uranium-cannon-shell", chance = 1},
    {name = 'uranium-fuel-cell', chance = 1},
    --{name = "used-up-uranium-fuel-cell", chance = 1},
    {name = 'water-barrel', chance = 10}
}

local container_loot_amounts = {
    ['advanced-circuit'] = 6,
    --["artillery-shell"] = 0.3,
    ['battery'] = 2,
    ['cannon-shell'] = 4,
    --["cluster-grenade"] = 0.3,
    ['construction-robot'] = 0.3,
    ['copper-cable'] = 24,
    ['copper-plate'] = 16,
    ['crude-oil-barrel'] = 3,
    ['defender-capsule'] = 2,
    ['destroyer-capsule'] = 0.3,
    ['distractor-capsule'] = 0.3,
    ['electric-engine-unit'] = 2,
    ['electronic-circuit'] = 8,
    ['empty-barrel'] = 3,
    ['engine-unit'] = 2,
    ['explosive-cannon-shell'] = 2,
    --["explosive-rocket"] = 2,
    --["explosive-uranium-cannon-shell"] = 2,
    ['explosives'] = 4,
    ['green-wire'] = 8,
    ['grenade'] = 6,
    ['heat-pipe'] = 1,
    ['heavy-oil-barrel'] = 3,
    ['iron-gear-wheel'] = 8,
    ['iron-plate'] = 16,
    ['iron-stick'] = 16,
    ['land-mine'] = 6,
    ['light-oil-barrel'] = 3,
    ['logistic-robot'] = 0.3,
    ['low-density-structure'] = 0.3,
    ['lubricant-barrel'] = 3,
    ['nuclear-fuel'] = 0.1,
    ['petroleum-gas-barrel'] = 3,
    ['pipe'] = 8,
    ['pipe-to-ground'] = 1,
    ['plastic-bar'] = 4,
    ['processing-unit'] = 2,
    ['red-wire'] = 8,
    --["rocket"] = 2,
    --["rocket-control-unit"] = 0.3,
    --["rocket-fuel"] = 0.3,
    ['solid-fuel'] = 4,
    ['steel-plate'] = 4,
    ['sulfuric-acid-barrel'] = 3,
    --["uranium-cannon-shell"] = 2,
    ['uranium-fuel-cell'] = 0.3,
    --["used-up-uranium-fuel-cell"] = 1,
    ['water-barrel'] = 3
}

local scrap_raffle = {}
for _, t in pairs(container_loot_chance) do
    for _ = 1, t.chance, 1 do
        table_insert(scrap_raffle, t.name)
    end
end

local size_of_scrap_raffle = #scrap_raffle

local function place_scrap(surface, position)
    local this = ScenarioTable.get_table()
    if this.spaceships == nil then
        this.spaceships = {}
    end
    -- place turrets
    if math_random(1, 700) == 1 then
        if position.x ^ 2 + position.x ^ 2 > 4096 then
            local e = surface.create_entity({name = 'gun-turret', position = position, force = 'enemy'})
            e.minable = false
            e.operable = false
            e.insert({name = 'piercing-rounds-magazine', count = 100})
            return
        end
    end

    -- place scrap containers with loot
    if math_random(1, 128) == 1 then
        local scrap = scrap_containers[math_random(1, scrap_containers_index)]
        local e = surface.create_entity({name = scrap.name, position = position, force = 'neutral'})
        e.minable = true
        local i = e.get_inventory(defines.inventory.chest)
        if i then
            local size = scrap.size
            for _ = 1, math_random(1, size), 1 do
                local loot = scrap_raffle[math_random(1, size_of_scrap_raffle)]
                local amount = container_loot_amounts[loot]
                local count = math_floor(amount * math_random(5, 35) * 0.1) + 1
                i.insert({name = loot, count = count})
            end
        end
        return
    end

    -- place scrap entities with loot
    local scrap = scrap_entities[math_random(1, scrap_entities_index)]
    local e = surface.create_entity({name = scrap.name, position = position, force = 'neutral'})
    e.minable = true
end

local function is_scrap_area(n)
    if n > 0.5 then
        return true
    end
    if n < -0.5 then
        return true
    end
end

local function move_away_biteys(surface, area)
    for _, e in pairs(surface.find_entities_filtered({type = {'unit-spawner', 'turret', 'unit'}, area = area})) do
        local position = surface.find_non_colliding_position(e.name, e.position, 96, 4)
        if position then
            surface.create_entity({name = e.name, position = position, force = 'enemy'})
            e.destroy()
        end
    end
end

local vectors = {{0, 0}, {1, 0}, {-1, 0}, {0, 1}, {0, -1}}

local function landfill_under(entity)
    -- landfill the area under the entity
    local surface = entity.surface
    for _, v in pairs(vectors) do
        local position = {entity.position.x + v[1], entity.position.y + v[2]}
        if not surface.get_tile(position).collides_with('resource-layer') then
            surface.set_tiles({{name = 'landfill', position = position}}, true)
        end
    end
end

local function on_player_mined_entity(event)
    local entity = event.entity
    if not entity.valid then
        return
    end
    if Scrap.is_scrap(entity) then
        landfill_under(entity)
    end
end

local function on_entity_died(event)
    local entity = event.entity
    if not entity.valid then
        return
    end
    if Scrap.is_scrap(entity) then
        landfill_under(entity)
    end
end

--local function on_init(event)
--
--end

local function on_chunk_generated(event)
    --log("scrap_towny_ffa::on_chunk_generated")
    local surface = event.surface
    local this = ScenarioTable.get_table()
    local map_surface = game.get_surface(this.active_surface_index)
    if not map_surface or not map_surface.valid then
        return
    end
    if (surface.name ~= map_surface.name) then
        return
    end
    local seed = surface.map_gen_settings.seed
    local left_top_x = event.area.left_top.x
    local left_top_y = event.area.left_top.y

    local position
    local noise

    if this.surface_terrain == 'forest' then
        for x = 0, 31, 1 do
            for y = 0, 31, 1 do
                position = {x = left_top_x + x, y = left_top_y + y}
                local cave_ponds = get_perlin('cave_ponds', position, seed)
                local bridges = get_perlin('cave_rivers', position, seed)

                if bridges > 0.15 and bridges < 0.25 then
                    surface.set_tiles({{name = 'water', position = position}}, true)
                    if cave_ponds > 0.1 and cave_ponds < 0.25 then
                        surface.set_tiles({{name = 'dirt-1', position = position}}, true)
                    end
                    if math_random(1, 48) == 1 then
                        surface.create_entity({name = 'fish', position = position, force = 'neutral'})
                    end
                end
                if math_random(1, 3) > 1 then
                    if not surface.get_tile(position).collides_with('resource-layer') then
                        noise = get_noise('scrap_towny_ffa', position, seed)
                        if is_scrap_area(noise) then
                            place_scrap(surface, position)
                        end
                    end
                end
            end
        end
    else
        for x = 0, 31, 1 do
            for y = 0, 31, 1 do
                position = {x = left_top_x + x, y = left_top_y + y}
                if math_random(1, 3) > 1 then
                    if not surface.get_tile(position).collides_with('resource-layer') then
                        noise = get_noise('scrap_towny_ffa', position, seed)
                        if is_scrap_area(noise) then
                            place_scrap(surface, position)
                        end
                    end
                end
            end
        end
    end

    local chunk_position = event.position
    --log('chunk_position = {' .. chunk_position.x .. ',' .. chunk_position.y .. '}')
    if chunk_position.x >= -33 and chunk_position.x <= 32 and chunk_position.y >= -33 and chunk_position.y <= 32 then
        if chunk_position.x == -33 or chunk_position.x == 32 or chunk_position.y == -33 or chunk_position.y == 32 then
            local area = {{x = left_top_x, y = left_top_y}, {x = left_top_x + 31, y = left_top_y + 31}}
            local entities = surface.find_entities(area)
            for _, e in pairs(entities) do
                e.destroy()
            end
            for x = 0, 31, 1 do
                for y = 0, 31, 1 do
                    position = {x = left_top_x + x, y = left_top_y + y}
                    surface.set_tiles({{name = 'water-shallow', position = position}}, true)
                end
            end
            return
        end
    end

    if chunk_position.x < -33 or chunk_position.x > 32 or chunk_position.y < -33 or chunk_position.y > 32 then
        local area = {{x = left_top_x, y = left_top_y}, {x = left_top_x + 31, y = left_top_y + 31}}
        local entities = surface.find_entities(area)
        for _, e in pairs(entities) do
            e.destroy()
        end
        for x = 0, 31, 1 do
            for y = 0, 31, 1 do
                position = {x = left_top_x + x, y = left_top_y + y}
                surface.set_tiles({{name = 'deepwater', position = position}}, true)
            end
        end
        return
    end

    move_away_biteys(surface, event.area)
    --this.chunk_generated[key] = true
end

local function on_chunk_charted(event)
    local force = event.force
    local this = ScenarioTable.get_table()
    local surface = game.get_surface(this.active_surface_index)
    if not surface or not surface.valid then
        return
    end
    if force.valid then
        if force == game.forces['player'] or force == game.forces['rogue'] then
            force.clear_chart(surface)
        end
    end
end

-- local on_init = function ()
-- 	local this = ScenarioTable.get_table()
--  this.chunk_generated = {}
-- end

-- Event.on_init(on_init)
Event.add(defines.events.on_chunk_generated, on_chunk_generated)
Event.add(defines.events.on_chunk_charted, on_chunk_charted)
Event.add(defines.events.on_player_mined_entity, on_player_mined_entity)
Event.add(defines.events.on_entity_died, on_entity_died)