Move MazeGenerator to seperate file

2 files changed, 57 insertions, 56 deletions

diff --git a/main.rb b/main.rb
index 07b2fa9..19f8376 100644
--- a/main.rb
+++ b/main.rb
@@ -1,59 +1,4 @@
-require_relative './pos.rb'
-require_relative './maze.rb'
-
-class MazeGenerator
-  attr_reader :maze
-
-  def initialize(width, height)
-    @maze = Maze.new(width, height)
-    @visitedTiles = Array.new(width) { Array.new(height) { false } }
-  end
-
-  def generate()
-    @visitedCells = [@currentPos]
-    @stack = [Pos.new(rand(@maze.width), rand(@maze.height))]
-
-    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
-
-    print "\e[s" if ENV["DEBUG"] == "visual"
-
-    while !@stack.empty?
-      step()
-      if ENV["DEBUG"] == "visual"
-        print "\e[u" # Restore the cursor position
-        print @maze.to_s("   ")
-      end
-    end
-
-    if ENV["DEBUG"] == "visual"
-      print "\e[?1049l" # Restore the state of the terminal
-    end
-  end
-
-  def step()
-    neighbors = @maze.neighbors(@stack.last)
-    neighbors.select! do |neighbor|
-      @visitedTiles[neighbor.x][neighbor.y] == false
-    end
-
-    if neighbors.empty?
-      @stack.pop()
-    else
-      randomNeighbor = neighbors.sample
-      print "Removing wall between ", @stack.last, " and ", randomNeighbor, "\n" if ENV["DEBUG"] == "log"
-      @maze.set(@stack.last, randomNeighbor.dir_from(@stack.last), false)
-
-      @stack.push(randomNeighbor)
-      @visitedTiles[randomNeighbor.x][randomNeighbor.y] = true
-    end
-  end
-end
+require_relative 'maze_generator'
 
 mazeGenerator = MazeGenerator.new(ARGV[0].to_i, ARGV[1].to_i)
 mazeGenerator.generate
diff --git a/maze_generator.rb b/maze_generator.rb
new file mode 100644
index 0000000..9797624
--- /dev/null
+++ b/maze_generator.rb
@@ -0,0 +1,56 @@
+require_relative 'pos'
+require_relative 'maze'
+
+class MazeGenerator
+  attr_reader :maze
+
+  def initialize(width, height)
+    @maze = Maze.new(width, height)
+    @visitedTiles = Array.new(width) { Array.new(height) { false } }
+  end
+
+  def generate()
+    @visitedCells = [@currentPos]
+    @stack = [Pos.new(rand(@maze.width), rand(@maze.height))]
+
+    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
+
+    print "\e[s" if ENV["DEBUG"] == "visual"
+
+    while !@stack.empty?
+      step()
+      if ENV["DEBUG"] == "visual"
+        print "\e[u" # Restore the cursor position
+        print @maze.to_s("   ")
+      end
+    end
+
+    if ENV["DEBUG"] == "visual"
+      print "\e[?1049l" # Restore the state of the terminal
+    end
+  end
+
+  def step()
+    neighbors = @maze.neighbors(@stack.last)
+    neighbors.select! do |neighbor|
+      @visitedTiles[neighbor.x][neighbor.y] == false
+    end
+
+    if neighbors.empty?
+      @stack.pop()
+    else
+      randomNeighbor = neighbors.sample
+      print "Removing wall between ", @stack.last, " and ", randomNeighbor, "\n" if ENV["DEBUG"] == "log"
+      @maze.set(@stack.last, randomNeighbor.dir_from(@stack.last), false)
+
+      @stack.push(randomNeighbor)
+      @visitedTiles[randomNeighbor.x][randomNeighbor.y] = true
+    end
+  end
+end