Split program into multiple files

1 files changed, 81 insertions, 0 deletions

diff --git a/maze.rb b/maze.rb
new file mode 100644
index 0000000..d12a8f5
--- /dev/null
+++ b/maze.rb
@@ -0,0 +1,81 @@
+Tile = Struct.new(:pos, :up, :right, :down, :left)
+
+class Maze
+  attr_reader :width
+  attr_reader :height
+
+  def initialize(width, height)
+    @width = width
+    @height = height
+    @h_walls = Array.new(@width) { Array.new(@height + 1, true) }
+    @v_walls = Array.new(@width + 1) { Array.new(@height, true) }
+  end
+
+  def inBounds?(pos)
+    x = pos.x
+    y = pos.y
+
+    x >= 0 && y >= 0 && x < @width && y < @height
+  end
+
+  def get(pos)
+    raise IndexError unless inBounds?(pos)
+
+    x = pos.x
+    y = pos.y
+
+    Tile.new(pos, @h_walls[x][y], @v_walls[x + 1][y], @h_walls[x][y + 1], @v_walls[x][y])
+  end
+
+  def neighbors(pos)
+    neighbors = []
+
+    [:up, :right, :down, :left].each do |dir|
+      neighbors.push(pos.move(dir)) if inBounds?(pos.move(dir))
+    end
+
+    neighbors
+  end
+
+  def set(pos, dir, state)
+    raise IndexError if pos.x >= @width
+    raise IndexError if pos.y >= @height
+
+    case dir
+    when :up
+      @h_walls[pos.x][pos.y] = state
+    when :right
+      @v_walls[pos.x + 1][pos.y] = state
+    when :down
+      @h_walls[pos.x][pos.y + 1] = state
+    when :left
+      @v_walls[pos.x][pos.y] = state
+    end
+  end
+
+  def to_s
+    drawingField = Array.new(@height * 2 + 1) { " " * (@width * 2 + 1) }
+
+    @v_walls.each_index do |x|
+      @v_walls[x].each_index do |y|
+        next if @v_walls[x][y] == false
+
+        drawingField[y * 2][x * 2] = "█"
+        drawingField[y * 2 + 1][x * 2] = "█"
+        drawingField[y * 2 + 2][x * 2] = "█"
+      end
+    end
+
+    @h_walls.each_index do |x|
+      @h_walls[x].each_index do |y|
+        next if @h_walls[x][y] == false
+
+        drawingField[y * 2][x * 2] = "█"
+        drawingField[y * 2][x * 2 + 1] = "█"
+        drawingField[y * 2][x * 2 + 2] = "█"
+      end
+    end
+
+    drawingField.join("\n")
+  end
+end