blob: d0526a9037adb846e0c52bd28bd8ddb292533c46 (
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
|
require_relative 'pos'
require_relative 'maze'
class MazeGenerator
attr_reader :maze
def initialize(width, height, stack_limit)
@maze = Maze.new(width, height)
@stack_limit = stack_limit
@visitedTiles = Array.new(width) { Array.new(height) { false } }
end
def generate!()
start_pos = Pos.new(rand(@maze.width), rand(@maze.height))
@visitedTiles[start_pos.x][start_pos.y] = true
@stack = [start_pos]
if ENV["DEBUG"] == "visual"
print "\e[?1049h" # Save the state of the terminal
print "\e[2J" # Clear the screen
print "\e[0;0H" # Move the cursor to 0, 0
print "\n [ \e[1;35mGenerating maze...\e[0m ]\n\n" # Print some nice graphics
print "\e[s" # Save the cursor position
end
while !@stack.empty?
step()
if ENV["DEBUG"] == "visual"
print "\e[u" # Restore the cursor position
print @maze.to_s(" ")
print "\n\n"
puts " Stack size: #{@stack.length}"
if @stack.length < @stack_limit
puts " Current algorithm: Depth-first-search"
else
puts " Current algorithm: Breath-first-search"
end
end
end
if ENV["DEBUG"] == "visual"
print "\e[?1049l" # Restore the state of the terminal
end
end
def step()
current_tile = if @stack.length < @stack_limit
@stack.last
else
@stack.first
end
neighbors = @maze.neighbors(current_tile)
neighbors.select! do |neighbor|
@visitedTiles[neighbor.x][neighbor.y] == false
end
if neighbors.empty?
if @stack.length < @stack_limit
@stack.pop()
else
@stack.shift()
end
else
randomNeighbor = neighbors.sample
print "Removing wall between ", current_tile, " and ", randomNeighbor, "\n" if ENV["DEBUG"] == "log"
@maze.set(current_tile, randomNeighbor.dir_from(current_tile), false)
@stack.push(randomNeighbor)
@visitedTiles[randomNeighbor.x][randomNeighbor.y] = true
end
end
end
|
