chess_rb 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8387c3c7b4b44a16284cf21d3598c9e6f1377fc9
-  data.tar.gz: 9e9d8036aeb5e539135a2ea3f2fa7f6781dd5734
+  metadata.gz: ec725359b2743656a3d31112e5d9a3735644f5ec
+  data.tar.gz: 06be1b022d68f92885c13e0478eeec92462f23be
 SHA512:
-  metadata.gz: 8e22e5ccee3e43694dcaafe311c3c76a84389b4235af5c5ea76857829cc1fad4d47c536abdcde4b0ebf112db7a7a6f5b0d426334b0a1deab9ac581443377a6e7
-  data.tar.gz: b55e649dcf6243982b3493ce22ae99950b0770fe91fa628b98572c2d2d0c94866e13f7505beb491819730c450167b96c4f8393b2604a77ff68cde4bfe5022e25
+  metadata.gz: 763cee97ef7d9da45c039e820d132af8669a72449c906dbab794e85ac9c47734bebb1545103d0103cd7ee5b85c61fac225b1ebfef1dfeb242994f81812e05be8
+  data.tar.gz: d89f6eb86d04ec9507a4fa3deb539df42beddfe45c9c3633e11e954b6a7a8207fa5ba41c1bb5a7cf8ab5a1b6c8b453638e378780a92ac5315df28d62dc43b3e1

data/lib/chess_rb/notation.rb

@@ -35,12 +35,14 @@ class ChessRB::Notation
 
     san += "=#{move.promotion}" if move.promotion
 
-    undo_info = board.piece_on(move.to)
-    board.make_move(move)
-    if (board.check?)
-      san += board.mate? ? "#" : "+"
+    if !board.squares_with(['WK', 'BK']).empty?
+      undo_info = board.piece_on(move.to)
+      board.make_move(move)
+      if (board.check?)
+        san += board.mate? ? "#" : "+"
+      end
+      board.undo_move(move, undo_info)
     end
-    board.undo_move(move, undo_info)
 
     return san
   end

data/lib/chess_rb/position.rb

@@ -38,13 +38,16 @@ class ChessRB::Position
     end
   end
 
-  def squares_with(piece)
+  def squares_with(pieces)
     squares = []
-    code = ChessRB::Piece.const_get(piece)
+    codes = []
+    pieces.each do |piece|
+      codes << ChessRB::Piece.const_get(piece)
+    end
 
     @board.each_with_index do |r , i|
       r.each_with_index do |p, j|
-        if p == code
+        if codes.include?(p)
           squares << [i, j]
         end
       end
@@ -59,7 +62,7 @@ class ChessRB::Position
 
     t = turn().upcase
     not_t = t == 'W' ? 'B' : 'W'
-    king_vector = Vector[*(squares_with(t + 'K')[0])]
+    king_vector = Vector[*(squares_with([t + 'K'])[0])]
 
     # check pawn squares
     pawn_vectors = t == 'W' ? [[-1, -1], [1, -1]] : [[-1, 1], [1, 1]]
@@ -124,7 +127,7 @@ class ChessRB::Position
     return false if !check?
 
     t = turn().upcase
-    king_vector = Vector[*(squares_with(t + 'K')[0])]
+    king_vector = Vector[*(squares_with([t + 'K'])[0])]
 
     around_vectors = [[1, 0], [0, 1], [-1, 0], [0, -1], [1, 1], [-1, 1],
       [1, -1], [-1, -1]]
@@ -227,7 +230,6 @@ class ChessRB::Position
       fen += count.to_s if count != 0
     end
 
-    puts fen.to_s
     return fen
   end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: chess_rb
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Sabar Dasgupta
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-06-22 00:00:00.000000000 Z
+date: 2014-06-23 00:00:00.000000000 Z
 dependencies: []
 description: Chess functions written in Ruby
 email: sabar.dasgupta@gmail.com