summaryrefslogtreecommitdiff
path: root/logic/MapGenerator.lua
blob: f95ffedb482719169d5a0158b1854fb902d682e4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
local Room = require "objects.room"

local mapGenerator = {}

function mapGenerator.generate(roomCount, initial)
    local firstRoom = Room:new(initial and "initialRoom" or nil)
    local map = { firstRoom }
    local available = utils.copy(map)

    for i = 2, (roomCount or config.map.initialRoomCount) do
        -- connect the room with a new room
        local newRoom = Room:new()
        local prevRoom = utils.any(available)
        prevRoom:connect(newRoom)

        if prevRoom:isDoorFull() then
            table.remove(available, utils.indexof(available, prevRoom))
        end

        table.insert(map, newRoom)
        table.insert(available, newRoom)
    end

    return map
end

function mapGenerator.continue(roomCount, from)
    -- continue game by extending the map
    local newMap = mapGenerator.generate(roomCount or config.map.extendedRoomCount, true)
    local room = from or G.currentRoom
    
    -- attempt to find an available room
    local connected = false
    local attempts = 0
    local maxAttempts = 10
    
    while not connected and attempts < maxAttempts do
        -- if doorfull
        if room:isDoorFull() then
            -- find an available room
            for _, existingRoom in ipairs(G.allRooms) do
                if not existingRoom:isDoorFull() then
                    room = existingRoom
                    break
                end
            end
        end
        
        -- try to connect
        local entranceDoor, exitDoor = room:connect(newMap[1])
        if entranceDoor then
            connected = true
            if entranceDoor then
                G.BodyLifeCycleManager:create(entranceDoor, { entranceDoor.w, entranceDoor.h }, "static")
            end
            if exitDoor then
                newMap[1]:removeDoor(exitDoor.location)
            end
        else
            -- if failed, generate a new map
            newMap = mapGenerator.generate(roomCount or config.map.extendedRoomCount, true)
        end
        
        attempts = attempts + 1
    end
    
    if not connected then
        print("Warning: Failed to connect new rooms after " .. maxAttempts .. " attempts")
        return false
    end

    for _, newRoom in ipairs(newMap) do
        table.insert(G.allRooms, newRoom)
    end

    G.mapExpanded = true
    return true
end

return mapGenerator