ComfyFactorio/maps/stone_maze/3x3_rooms.lua

--luacheck: ignore
local room = {}

room.empty = function(surface, cell_left_top, direction)
end

room.biters = function(surface, cell_left_top, direction)
    local amount = get_biter_amount() * 2
    local tile_positions = {}
    local left_top = {x = cell_left_top.x * grid_size, y = cell_left_top.y * grid_size}
    for x = 0.5, grid_size * 3 - 0.5, 1 do
        for y = 0.5, grid_size * 3 - 0.5, 1 do
            local pos = {left_top.x + x, left_top.y + y}
            tile_positions[#tile_positions + 1] = pos
        end
    end

    table.shuffle_table(tile_positions)

    for _, pos in pairs(tile_positions) do
        local enemy = get_biter()
        if surface.can_place_entity({name = enemy, position = pos}) then
            surface.create_entity({name = enemy, position = pos, force = 'enemy'})
            amount = amount - 1
        end
        if amount < 1 then
            break
        end
    end
end

room.spitters = function(surface, cell_left_top, direction)
    local amount = get_biter_amount() * 2
    local tile_positions = {}
    local left_top = {x = cell_left_top.x * grid_size, y = cell_left_top.y * grid_size}
    for x = 0.5, grid_size * 3 - 0.5, 1 do
        for y = 0.5, grid_size * 3 - 0.5, 1 do
            local pos = {left_top.x + x, left_top.y + y}
            tile_positions[#tile_positions + 1] = pos
        end
    end

    table.shuffle_table(tile_positions)

    for _, pos in pairs(tile_positions) do
        local enemy = get_spitter()
        if surface.can_place_entity({name = enemy, position = pos}) then
            surface.create_entity({name = enemy, position = pos, force = 'enemy'})
            amount = amount - 1
        end
        if amount < 1 then
            break
        end
    end
end

room.nests = function(surface, cell_left_top, direction)
    local amount = math.ceil(get_biter_amount() * 0.1)
    local tile_positions = {}
    local left_top = {x = cell_left_top.x * grid_size, y = cell_left_top.y * grid_size}
    for x = 0.5, grid_size * 3 - 0.5, 1 do
        for y = 0.5, grid_size * 3 - 0.5, 1 do
            local pos = {left_top.x + x, left_top.y + y}
            tile_positions[#tile_positions + 1] = pos
        end
    end

    table.shuffle_table(tile_positions)

    for _, pos in pairs(tile_positions) do
        if surface.can_place_entity({name = 'spitter-spawner', position = pos}) then
            if math.random(1, 4) == 1 then
                surface.create_entity({name = 'spitter-spawner', position = pos, force = 'enemy'})
            else
                surface.create_entity({name = 'biter-spawner', position = pos, force = 'enemy'})
            end
            amount = amount - 1
        end
        if amount < 1 then
            break
        end
    end
end

room.uranium_wasteland = function(surface, cell_left_top, direction)
    local left_top = {x = cell_left_top.x * grid_size, y = cell_left_top.y * grid_size}
    local center_pos = {x = left_top.x + grid_size * 1.5, y = left_top.y + grid_size * 1.5}

    map_functions.draw_noise_tile_circle(center_pos, 'water-green', surface, grid_size * 0.65)
    map_functions.draw_smoothed_out_ore_circle(center_pos, 'uranium-ore', surface, grid_size * 1.3, get_ore_amount())

    for x = math.floor(grid_size * 3 * 0.1), math.floor(grid_size * 3 * 0.9), 1 do
        for y = math.floor(grid_size * 3 * 0.1), math.floor(grid_size * 3 * 0.9), 1 do
            local pos = {x = left_top.x + x, y = left_top.y + y}
            local distance_to_center = math.sqrt((center_pos.x - pos.x) ^ 2 + (center_pos.y - pos.y) ^ 2)
            if math.random(1, 128) == 1 and distance_to_center < grid_size * 1.4 then
                spawn_enemy_gun_turret(surface, pos)
            end
            if math.random(1, 10) == 1 and distance_to_center < grid_size * 1.4 then
                if surface.can_place_entity({name = 'mineable-wreckage', position = pos, force = 'neutral'}) then
                    surface.create_entity({name = get_scrap(), position = pos, force = 'neutral'})
                end
            end
        end
    end

    room.biters(surface, cell_left_top, direction)
end

room.stone_block = function(surface, cell_left_top, direction)
    local left_top = {x = cell_left_top.x * grid_size, y = cell_left_top.y * grid_size}
    for x = 3.5, grid_size * 3 - 3.5, 1 do
        for y = 3.5, grid_size * 3 - 3.5, 1 do
            local pos = {left_top.x + x, left_top.y + y}
            if math.random(1, 5) ~= 1 then
                surface.create_entity({name = rock_raffle[math.random(1, #rock_raffle)], position = pos, force = 'neutral'})
            end
        end
    end

    for a = 1, math.random(1, 3), 1 do
        local chest =
            surface.create_entity(
            {
                name = 'steel-chest',
                position = {
                    left_top.x + math.random(math.floor(grid_size * 0.5), math.floor(grid_size * 2.5)),
                    left_top.y + math.random(math.floor(grid_size * 0.5), math.floor(grid_size * 2.5))
                },
                force = 'neutral'
            }
        )
        for a = 1, math.random(1, 4), 1 do
            chest.insert(get_loot_item_stack())
        end
    end
end

room.tree_square_nests = function(surface, cell_left_top, direction)
    local left_top = {x = cell_left_top.x * grid_size, y = cell_left_top.y * grid_size}

    local tree = tree_raffle[math.random(1, #tree_raffle)]

    for x = 0, grid_size * 3 - 1, 1 do
        for y = 0, grid_size * 3 - 1, 1 do
            local pos = {left_top.x + x, left_top.y + y}
            if x <= 1 or x >= grid_size * 3 - 2 or y <= 1 or y >= grid_size * 3 - 2 then
                surface.create_entity({name = tree, position = pos, force = 'neutral'})
            end
        end
    end
    room.nests(surface, cell_left_top, direction)
    room.spitters(surface, cell_left_top, direction)
    room.biters(surface, cell_left_top, direction)
end

local room_weights = {
    {func = room.uranium_wasteland, weight = 1},
    {func = room.stone_block, weight = 3},
    {func = room.tree_square_nests, weight = 3}
}

local room_shuffle = {}
for _, r in pairs(room_weights) do
    for c = 1, r.weight, 1 do
        room_shuffle[#room_shuffle + 1] = r.func
    end
end

return room_shuffle