just_shogi 0.1.0

data/lib/just_shogi/hand.rb

@@ -0,0 +1,48 @@
+require 'just_shogi/piece_factory'
+require 'just_shogi/promotion_factory'
+require 'just_shogi/pieces/piece'
+
+module JustShogi
+  class Hand
+    def initialize(player_number: , pieces: [])
+      @player_number = player_number
+      @pieces = if pieces.instance_of?(Array)
+                  if pieces.all? { |p| p.is_a?(Hash) }
+                    pieces.map { |p| JustShogi::PieceFactory.new(p).build }
+                  elsif pieces.all? { |p| p.is_a?(JustShogi::Piece) }
+                    pieces
+                  else
+                    raise ArgumentError, "all pieces must have the same class"
+                  end
+                else
+                  raise ArgumentError, "pieces must be an array"
+                end
+    end
+
+    attr_reader :player_number
+    attr_reader :pieces
+
+    def push_piece(piece)
+      factory = PromotionFactory.new(piece)
+      demoted_piece = factory.demotable? ? factory.demote : piece
+      demoted_piece.switch_player
+      @pieces.push(demoted_piece)
+    end
+
+    def pop_piece(id)
+      p = find_piece_by_id(id) 
+      @pieces.delete(p)
+    end
+
+    def find_piece_by_id(id)
+      pieces.find { |p| p.id == id } 
+    end
+
+    def as_json
+      {
+        player_number: player_number,
+        pieces: pieces.map(&:as_json) 
+      }
+    end
+  end
+end

data/lib/just_shogi/piece_factory.rb

@@ -0,0 +1,87 @@
+require 'just_shogi/pieces/fuhyou'
+require 'just_shogi/pieces/kakugyou'
+require 'just_shogi/pieces/hisha'
+require 'just_shogi/pieces/kyousha'
+require 'just_shogi/pieces/keima'
+require 'just_shogi/pieces/ginshou'
+require 'just_shogi/pieces/kinshou'
+require 'just_shogi/pieces/oushou'
+require 'just_shogi/pieces/gyokushou'
+
+require 'just_shogi/pieces/tokin'
+require 'just_shogi/pieces/narikyou'
+require 'just_shogi/pieces/narikei'
+require 'just_shogi/pieces/narigin'
+require 'just_shogi/pieces/ryuuma'
+require 'just_shogi/pieces/ryuuou'
+
+module JustShogi
+
+  # = Piece Factory
+  #
+  # Generates pieces from a hash of arguments
+  class PieceFactory
+
+    # A mapping of type descriptions to classes.
+    CLASSES = {
+      'fuhyou' => Fuhyou,
+      'kakugyou' => Kakugyou,
+      'hisha' => Hisha,
+      'kyousha' => Kyousha,
+      'keima' => Keima,
+      'ginshou' => Ginshou,
+      'kinshou' => Kinshou,
+      'oushou' => Oushou,
+      'gyokushou' => Gyokushou,
+      'tokin' => Tokin,
+      'narikyou' => Narikyou,
+      'narikei' => Narikei,
+      'narigin' => Narigin,
+      'ryuuma' => Ryuuma,
+      'ryuuou' => Ryuuou
+    }
+
+    # New objects can be instantiated by passing in a hash or the piece.
+    #
+    # @param [Hash,Piece] args
+    #   the initial attributes of the piece, hash requires type key
+    #
+    # ==== Example:
+    #   # Instantiates a new PieceFactory
+    #   JustShogi::PieceFactory.new({
+    #     type: 'fuhyou',
+    #     id: 1,
+    #     player_number: 2
+    #   })
+    def initialize(args)
+      @args = args
+    end
+
+    # Returns a piece based on the initial arguments.
+    #
+    # @return [Piece]
+    def build
+      case @args
+      when Hash
+        build_from_hash
+      when Piece
+        @args
+      when nil
+        nil
+      else
+        raise ArgumentError, "piece must be Hash or NilClass"
+      end
+    end
+
+    private
+
+    def build_from_hash
+      klass = CLASSES[@args[:type]]
+      if klass
+        klass.new(**@args)
+      else
+        raise ArgumentError, "invalid piece type: #{@args[:type].to_s}"
+      end
+    end
+  end
+end

data/lib/just_shogi/pieces/fuhyou.rb

@@ -0,0 +1,23 @@
+require 'just_shogi/pieces/piece'
+
+module JustShogi
+
+  # = Fuhyou
+  #
+  # The piece that moves forward one space.
+  class Fuhyou < Piece
+
+    # All the squares that the piece can move to and/or capture.
+    #
+    # @param [Square] square
+    #   the origin square.
+    #
+    # @param [GameState] game_state
+    #   the current game state.
+    #
+    # @return [SquareSet]
+    def destinations(square, game_state)
+      game_state.squares.in_range(square, 1).in_direction(square, forwards_direction).orthogonal(square).unoccupied_or_occupied_by_opponent(player_number).unblocked(square, game_state.squares)
+    end
+  end
+end

data/lib/just_shogi/pieces/ginshou.rb

@@ -0,0 +1,23 @@
+require 'just_shogi/pieces/piece'
+
+module JustShogi
+
+  # = Ginshou
+  #
+  # The piece that can move 1 space diagonally or forwards orthogonally
+  class Ginshou < Piece
+
+    # All the squares that the piece can move to and/or capture.
+    #
+    # @param [Square] square
+    #   the origin square.
+    #
+    # @param [GameState] game_state
+    #   the current game state.
+    #
+    # @return [SquareSet]
+    def destinations(square, game_state)
+      (game_state.squares.diagonal(square) | game_state.squares.in_direction(square, forwards_direction)).in_range(square, 1).unoccupied_or_occupied_by_opponent(player_number).unblocked(square, game_state.squares)
+    end
+  end
+end

data/lib/just_shogi/pieces/gyokushou.rb

@@ -0,0 +1,9 @@
+require 'just_shogi/pieces/ou_base'
+
+module JustShogi
+
+  # = Gyokushou
+  #
+  # The piece that moves 1 space away.
+  class Gyokushou < OuBase; end
+end

data/lib/just_shogi/pieces/hisha.rb

@@ -0,0 +1,23 @@
+require 'just_shogi/pieces/piece'
+
+module JustShogi
+
+  # = Hisha
+  #
+  # The piece that moves any number of squares orthogonally.
+  class Hisha < Piece
+    
+    # All the squares that the piece can move to and/or capture.
+    #
+    # @param [Square] square
+    #   the origin square.
+    #
+    # @param [GameState] game_state
+    #   the current game state.
+    #
+    # @return [SquareSet]
+    def destinations(square, game_state)
+      game_state.squares.orthogonal(square).unoccupied_or_occupied_by_opponent(player_number).unblocked(square, game_state.squares)
+    end
+  end
+end

data/lib/just_shogi/pieces/kakugyou.rb

@@ -0,0 +1,23 @@
+require 'just_shogi/pieces/piece'
+
+module JustShogi
+
+  # = Kakugyou
+  #
+  # The piece that moves diagonally any number of squares
+  class Kakugyou < Piece
+
+    # All the squares that the piece can move to and/or capture.
+    #
+    # @param [Square] square
+    #   the origin square.
+    #
+    # @param [GameState] game_state
+    #   the current game state.
+    #
+    # @return [SquareSet]
+    def destinations(square, game_state)
+      game_state.squares.diagonal(square).unoccupied_or_occupied_by_opponent(player_number).unblocked(square, game_state.squares)
+    end
+  end
+end

data/lib/just_shogi/pieces/keima.rb

@@ -0,0 +1,23 @@
+require 'just_shogi/pieces/piece'
+
+module JustShogi
+
+  # = Keima
+  #
+  # The piece that jumps over pieces 2v1h forwards.
+  class Keima < Piece
+  
+    # All the suares that the piece can move to and/or capture.
+    #
+    # @param [Square] square
+    #   the origin square.
+    #
+    # @param [GameState] game_state
+    #   the current game state.
+    #
+    # @return [SquareSet]
+    def destinations(square, game_state)
+      game_state.squares.in_direction(square, forwards_direction).ranks_away(square, 2).files_away(square, 1).unoccupied_or_occupied_by_opponent(player_number)
+    end
+  end
+end

data/lib/just_shogi/pieces/kin_base.rb

@@ -0,0 +1,23 @@
+require 'just_shogi/pieces/piece'
+
+module JustShogi
+  
+  # = KinBase
+  #
+  # A piece that moves 1 space orthogonally or forwards diagonally
+  class KinBase < Piece
+    
+    # All the squares that the piece can move to and/or capture.
+    #
+    # @param [Square] square
+    #   the origin square.
+    #
+    # @param [GameState] game_state
+    #   the current game state.
+    #
+    # @return [SquareSet]
+    def destinations(square, game_state)
+      (game_state.squares.orthogonal(square) | game_state.squares.in_direction(square, forwards_direction)).in_range(square, 1).unoccupied_or_occupied_by_opponent(player_number).unblocked(square, game_state.squares)
+    end
+  end
+end

data/lib/just_shogi/pieces/kinshou.rb

@@ -0,0 +1,9 @@
+require 'just_shogi/pieces/kin_base'
+
+module JustShogi
+  
+  # = Kinshou
+  #
+  # The piece that can move 1 space orthogonally or forwards diagonally
+  class Kinshou < KinBase; end
+end

data/lib/just_shogi/pieces/kyousha.rb

@@ -0,0 +1,23 @@
+require 'just_shogi/pieces/piece'
+
+module JustShogi
+
+  # = Kyousha
+  #
+  # The piece that moves forward any number of spaces.
+  class Kyousha < Piece
+
+    # All the squares that the piece can move to and/or capture.
+    #
+    # @param [Square] square
+    #   the origin square.
+    #
+    # @param [GameState] game_state
+    #   the current game state.
+    #
+    # @return [SquareSet]
+    def destinations(square, game_state)
+      game_state.squares.orthogonal(square).in_direction(square, forwards_direction).unoccupied_or_occupied_by_opponent(player_number).unblocked(square, game_state.squares)
+    end
+  end
+end

data/lib/just_shogi/pieces/narigin.rb

@@ -0,0 +1,9 @@
+require 'just_shogi/pieces/kin_base'
+
+module JustShogi
+
+  # = Narigin
+  #
+  # The piece that can move 1 space orthogonally or forwards diagonally
+  class Narigin < KinBase; end
+end

data/lib/just_shogi/pieces/narikei.rb

@@ -0,0 +1,9 @@
+require 'just_shogi/pieces/kin_base'
+
+module JustShogi
+
+  # = Narikei
+  #
+  # The piece that can move 1 space orthogonally or forwards diagonally
+  class Narikei < KinBase; end
+end

data/lib/just_shogi/pieces/narikyou.rb

@@ -0,0 +1,9 @@
+require 'just_shogi/pieces/kin_base'
+
+module JustShogi
+  
+  # = Narikyou 
+  #
+  # The piece that can move 1 space orthogonally or forwards diagonally
+  class Narikyou < KinBase; end
+end

data/lib/just_shogi/pieces/ou_base.rb

@@ -0,0 +1,70 @@
+require 'just_shogi/pieces/piece'
+
+module JustShogi
+
+  # = Ou Base 
+  #
+  # The piece that moves 1 space away.
+  class OuBase < Piece
+
+    # All the squares that the piece can move to and/or capture.
+    #
+    # @param [Square] square
+    #   the origin square.
+    #
+    # @param [GameState] game_state
+    #   the current game state.
+    #
+    # @return [SquareSet]
+    def destinations(square, game_state)
+      base_destinations(square, game_state) # - checked_squares(square, game_state) - shared_king_squares(game_state)
+    end
+
+    # All the squares that the king could move to normally.
+    #
+    # @param [Square] square
+    #   the origin square.
+    #
+    # @param [GameState] game_state
+    #   the current game state.
+    #
+    # @return [SquareSet]
+    def base_destinations(square, game_state)
+      game_state.squares.at_range(square, 1).unoccupied_or_occupied_by_opponent(player_number)
+    end
+
+    # All the squares that the king could not move to because of check.
+    #
+    # @param [Square] square
+    #   the origin square.
+    #
+    # @param [GameState] game_state
+    #   the current game state.
+    #
+    # @return [SquareSet]
+    def checked_squares(square, game_state)
+      dup = game_state.clone
+      # set the piece to nil to handle case where a piece threatens squares behind this piece.
+      dup.squares.find_ou_for_player(player_number).piece = nil
+      dup.squares.threatened_by(opponent, dup)
+    end
+
+    # All the squares that the king could not move to because antoher king is nearby.
+    #
+    # @param [GameState] game_state
+    #   the current game state.
+    #
+    # @return [SquareSet]
+    def shared_king_squares(game_state)
+      all = game_state.squares.occupied_by_piece(JustShogi::OuBase).map { |s| s.piece.base_destinations(s, game_state) }
+
+      all.reduce(nil) do |memo, set|
+        if memo
+          memo & set
+        else
+          set
+        end
+      end
+    end
+  end
+end

data/lib/just_shogi/pieces/oushou.rb

@@ -0,0 +1,9 @@
+require 'just_shogi/pieces/ou_base'
+
+module JustShogi
+
+  # = Oushou
+  #
+  # The piece that moves 1 space away.
+  class Oushou < OuBase; end
+end