mediocre_chess 0.0.1 → 0.1.0

data/lib/mediocre_chess.rb

@@ -3,19 +3,77 @@ require 'java'
 require 'mediocre/mediocre_v0.34.jar'
 java_import 'mediocre.board.Board'
 java_import 'mediocre.board.Move'
+java_import 'mediocre.engine.Engine'
 
 module MediocreChess
-  # Input: FEN board rep., rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1
-  # Returns: Array of move strings, ['e2e4', 'd2d4', ...]
-  def self.moves_for_fen(fen_board)
-    board = Java::Mediocre.board.Board.new
-    board.inputFen(fen_board)
-    moves = Java::int[128].new
-    number_of_moves = board.gen_allLegalMoves(moves, 0)
-    results = []
-    number_of_moves.times do |i|
-      results << Java::mediocre.board.Move.inputNotation(moves[i])
-    end
-    results
+  class IllegalMoveException < Exception; end
+  class InvalidFenException < Exception; end
+  WHITE = 1; BLACK = -1
+
+  class Board
+    def initialize(fen_board = 'rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1')
+      @board = Java::Mediocre.board.Board.new
+      self.fen = fen_board
+    end
+
+    # FEN Methods
+    def fen; @board.getFen; end
+
+    def fen=(fen_board)
+        @board.inputFen(fen_board)
+      rescue
+        raise MediocreChess::InvalidFenException.new("'#{fen_board}' is an invalid FEN board. The board may be in a bad state.")
+    end
+
+    # Returns: Array of legal move strings, ['e2e4', 'd2d4', ...]
+    def moves
+      number_of_moves, moves = gen_legal_moves
+      results = []
+      number_of_moves.times do |i|
+        results << Java::mediocre.board.Move.inputNotation(moves[i])
+      end
+      results
+    end
+
+    # Input: move string, "e2e4"
+    # Raises: IllegalMoveException
+    def add!(move_string)
+      number_of_moves, moves = gen_legal_moves
+      number_of_moves.times do |i|
+        if Java::mediocre.board.Move.inputNotation(moves[i]) == move_string
+          @board.makeMove(moves[i])
+          return
+        end
+      end
+
+      raise MediocreChess::IllegalMoveException.new("'#{move_string}' is not a legal chess move.")
+    end
+
+    def stalemate?
+      number_of_moves, moves = gen_legal_moves
+      number_of_moves == 0 && !Java::Mediocre.engine.Engine.isInCheck(@board)
+    end
+
+    def checkmate?
+      number_of_moves, moves = gen_legal_moves
+      number_of_moves == 0 && Java::Mediocre.engine.Engine.isInCheck(@board)
+    end
+
+    # Returns: MediocreChess::WHITE if white won
+    #          MediocreChess::BLACK if black won
+    #          nil if the game isn't checkmated
+    def winner
+      if checkmate?
+        -1 * @board.toMove
+      end
+    end
+
+    private
+
+    def gen_legal_moves
+      moves = Java::int[128].new
+      number_of_moves = @board.gen_allLegalMoves(moves, 0)
+      [number_of_moves, moves]
+    end
   end
 end

data/spec/mediocre_chess_spec.rb

@@ -0,0 +1,64 @@
+$:.unshift(File.dirname(__FILE__))
+require 'spec_helper'
+
+describe MediocreChess::Board do
+  # The purpose of these tests are not to test Mediocre methods,
+  # but rather the JRuby wrapper.
+
+  before(:each) do
+    @board = MediocreChess::Board.new
+  end
+
+  context "fen boards" do
+    it "should be created with the default board" do
+      @board.fen.should == 'rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1'
+    end
+
+    it "should be properly received and generated" do
+      @board.fen = '8/3K4/2p5/p2b2r1/5k2/8/8/1q6 b - - 1 67'
+      @board.fen.should == '8/3K4/2p5/p2b2r1/5k2/8/8/1q6 b - - 1 67'
+    end
+  end
+
+  context "move generation" do
+    it "should generate moves of the correct format" do
+      moves = @board.moves
+      moves.count.should == 20 # verifying the moves work
+      moves.each { |move| move.should match /[a-h][1-8][a-h][1-8]/ }
+    end
+  end
+
+  context "move addition" do
+    it "should add legal moves" do
+      @board.add!('e2e4')
+      @board.fen.should == 'rnbqkbnr/pppppppp/8/8/4P3/8/PPPP1PPP/RNBQKBNR b KQkq e3 0 1'
+    end
+
+    it "should only add legal moves" do
+      lambda { @board.add!('e7e5') }.should raise_error(MediocreChess::IllegalMoveException)
+    end
+  end
+
+  context "game state" do
+    it "should be normal at the initial position" do
+      @board.checkmate?.should be false
+      @board.stalemate?.should be false
+      @board.winner.should be nil
+    end
+
+    it "should find checkmate" do
+      @board.fen = '2Q5/p4pkp/3p2p1/3P4/3b4/2P5/PP6/6qK w - - 0 37'
+      @board.checkmate?.should be true
+      @board.stalemate?.should be false
+      @board.winner.should be MediocreChess::BLACK
+    end
+
+    it "should find stalemate" do
+      @board.fen = '8/8/8/8/8/1K6/2Q5/k7 b - - 0 1'
+      @board.fen.should == '8/8/8/8/8/1K6/2Q5/k7 b - - 0 1'
+      @board.checkmate?.should be false
+      @board.stalemate?.should be true
+      @board.winner.should be nil 
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,3 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'rspec/core'
+require 'mediocre_chess'

metadata

@@ -3,10 +3,10 @@ name: mediocre_chess
 version: !ruby/object:Gem::Version 
   prerelease: false
   segments: 
-    - 0
     - 0
     - 1
-  version: 0.0.1
+    - 0
+  version: 0.1.0
 platform: ruby
 authors: 
   - Joey Robert
@@ -14,7 +14,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-11-16 00:00:00 +00:00
+date: 2010-11-17 00:00:00 +00:00
 default_executable: 
 dependencies: []
 
@@ -37,7 +37,7 @@ files:
   - lib/mediocre_chess.rb
   - README.md
 has_rdoc: true
-homepage: http://github.com/joeyrobert/mediocre_ruby
+homepage: https://github.com/joeyrobert/mediocre_chess
 licenses: []
 
 post_install_message: 
@@ -66,5 +66,6 @@ rubygems_version: 1.3.6
 signing_key: 
 specification_version: 3
 summary: Chess utility functions using the Mediocre chess engine
-test_files: []
-
+test_files: 
+  - spec/mediocre_chess_spec.rb
+  - spec/spec_helper.rb