ai_games-four_in_a_row 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4e1e9b6d64ac0459f8d5464252c00b21448d05c3
-  data.tar.gz: de6ba6b68141cbc0daab9dc3ee8a34a9f6198e18
+  metadata.gz: 78694aacaf1fee4e7f09ec115fdd059f692ee45f
+  data.tar.gz: 29e1558eb5acc4440b439b454d888636bde3eeaa
 SHA512:
-  metadata.gz: 2dc3298df00dcd690d3bf3a0d21b9dfa8be83d37a6e8540c579a861b6a5ba2ac734328b195a0304a677e257595a83ffcd28874095c2736d278efcd293d3f95a5
-  data.tar.gz: 5b8208f24fa26123f61fb1d19d2029e4b998f86d5d8008c56cfbc818e9463ff5796e4962fb8d4d997870af4d5f09f484405eee65fc07c6bdfe1d5f9b201a5985
+  metadata.gz: f6c6864f30229a00cb4d1e7c7368c8ce4943d440408b72140063210d043de00597ffadbe71797971bee416bd946d47a06cf15afa2fd779ccd68a3f6b1cf693d5
+  data.tar.gz: f95332b5907f6b97153bc5a82f7a8d5d97b9e15068da17e161f3ac0559b0af951b021ee3ac2b5439841a60b4e7928b011903576f5d1ea2ec1370780dc43d6be1

data/lib/ai_games/four_in_a_row/cell.rb

@@ -31,11 +31,20 @@ module AIGames
       # The bot who fills this cell
       attr_accessor :owner
 
+      # The neighbors of this cell
+      attr_accessor :neighbors
+
       # Initializes the cell with the given position, and optionally the owner.
       def initialize(row, column, owner = nil)
         @row = row
         @column = column
         @owner = owner
+        @neighbors = []
+      end
+
+      # Returns all neighbors having the same owner.
+      def neighbors_with_same_owner
+        neighbors.select { |x| x.owner == owner }
       end
     end
   end

data/lib/ai_games/four_in_a_row/playing_field.rb

@@ -47,6 +47,17 @@ module AIGames
         @fields[row][column]
       end
 
+      # Returns all cells in the given area. Pass in two ranges to limit area.
+      def get_cells(rows, columns)
+        cells = []
+        rows.each do |r|
+          columns.each do |c|
+            cells << @fields[r][c]
+          end
+        end
+        cells
+      end
+
       # Updates a cell. If the owner of the cell changes, all observers of the
       # playing field get notified.
       def set_cell(row, column, owner)
@@ -72,17 +83,43 @@ module AIGames
 
       private
 
+      # Fills the playing field with cells.
+      def create_cells
+        (0...rows).each do |r|
+          (0...columns).each do |c|
+            @fields[r][c] = Cell.new(r, c)
+          end
+        end
+      end
+
+      # Find all neighbors of a cell.
+      def find_neighbors(cell)
+        min_r = [0, cell.row - 1].max
+        max_r = [cell.row + 1, rows - 1].min
+        min_c = [0, cell.column - 1].max
+        max_c = [cell.column + 1, columns - 1].min
+
+        get_cells(min_r..max_r, min_c..max_c).select { |x| x unless x == cell }
+      end
+
+      # Sets up neighbor relationships for all cells.
+      def link_cells
+        (0...rows).each do |r|
+          (0...columns).each do |c|
+            cell = @fields[r][c]
+            cell.neighbors = find_neighbors(cell)
+          end
+        end
+      end
+
       # Resizes the playing field. Each time this method gets called, a new
       # array is created and filled with new cells.
       def resize_field(rows, columns)
         @fields = Array.new(rows) { Array.new(columns) }
         @columns = columns
         @rows = rows
-        (0...rows).each do |r|
-          (0...columns).each do |c|
-            @fields[r][c] = Cell.new(r, c)
-          end
-        end
+        create_cells
+        link_cells
       end
     end
   end

data/lib/ai_games/four_in_a_row/version.rb

@@ -1,5 +1,5 @@
 module AIGames
   module FourInARow
-    VERSION = '0.3.0'
+    VERSION = '0.4.0'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ai_games-four_in_a_row
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Jan David Nose
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2015-11-26 00:00:00.000000000 Z
+date: 2015-11-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ai_games-logger