+ end # enf of while
+ end # def run
+end # class
+
+class Piece
+ PROMOTE = {"FU" => "TO", "KY" => "NY", "KE" => "NK",
+ "GI" => "NG", "KA" => "UM", "HI" => "RY"}
+ def initialize(board, x, y, sente, promoted=false)
+ @board = board
+ @x = x
+ @y = y
+ @sente = sente
+ @promoted = promoted
+
+ if ((x == 0) || (y == 0))
+ if (sente)
+ hands = board.sente_hands
+ else
+ hands = board.gote_hands
+ end
+ hands.push(self)
+ hands.sort! {|a, b|
+ a.name <=> b.name
+ }
+ else
+ @board.array[x][y] = self
+ end
+ end
+ attr_accessor :promoted, :sente, :x, :y, :board
+
+ def room_of_head?(x, y, name)
+ true
+ end
+
+ def movable_grids
+ return adjacent_movable_grids + far_movable_grids
+ end
+
+ def far_movable_grids
+ return []
+ end
+
+ def jump_to?(x, y)
+ if ((1 <= x) && (x <= 9) && (1 <= y) && (y <= 9))
+ if ((@board.array[x][y] == nil) || # dst is empty
+ (@board.array[x][y].sente != @sente)) # dst is enemy
+ return true
+ end
+ end
+ return false
+ end
+
+ def put_to?(x, y)
+ if ((1 <= x) && (x <= 9) && (1 <= y) && (y <= 9))
+ if (@board.array[x][y] == nil) # dst is empty?
+ return true
+ end
+ end
+ return false
+ end
+
+ def adjacent_movable_grids
+ grids = Array::new
+ if (@promoted)
+ moves = @promoted_moves
+ else
+ moves = @normal_moves
+ end
+ moves.each do |(dx, dy)|
+ if (@sente)
+ cand_y = @y - dy
+ else
+ cand_y = @y + dy
+ end
+ cand_x = @x + dx
+ if (jump_to?(cand_x, cand_y))
+ grids.push([cand_x, cand_y])
+ end
+ end
+ return grids
+ end
+
+ def move_to?(x, y, name)
+ return false if (! room_of_head?(x, y, name))
+ return false if ((name != @name) && (name != @promoted_name))
+ return false if (@promoted && (name != @promoted_name)) # can't un-promote
+
+ if (! @promoted)
+ return false if (((@x == 0) || (@y == 0)) && (name != @name)) # can't put promoted piece
+ if (@sente)
+ return false if ((4 <= @y) && (4 <= y) && (name != @name)) # can't promote
+ else
+ return false if ((6 >= @y) && (6 >= y) && (name != @name))
+ end
+ end
+
+ if ((@x == 0) || (@y == 0))
+ return jump_to?(x, y)
+ else
+ return movable_grids.include?([x, y])
+ end
+ end
+
+ def move_to(x, y)
+ if ((@x == 0) || (@y == 0))
+ if (@sente)
+ @board.sente_hands.delete(self)
+ else
+ @board.gote_hands.delete(self)
+ end
+ @board.array[x][y] = self
+ elsif ((x == 0) || (y == 0))
+ @promoted = false # clear promoted flag before moving to hands
+ if (@sente)
+ @board.sente_hands.push(self)
+ else
+ @board.gote_hands.push(self)
+ end
+ @board.array[@x][@y] = nil
+ else
+ @board.array[@x][@y] = nil
+ @board.array[x][y] = self
+ end
+ @x = x
+ @y = y
+ end
+
+ def point
+ @point
+ end
+
+ def name
+ @name
+ end
+
+ def promoted_name
+ @promoted_name
+ end
+
+ def to_s
+ if (@sente)
+ sg = "+"
+ else
+ sg = "-"
+ end
+ if (@promoted)
+ n = @promoted_name
+ else
+ n = @name
+ end
+ return sg + n
+ end
+end
+
+class PieceFU < Piece
+ def initialize(*arg)
+ @point = 1
+ @normal_moves = [[0, +1]]
+ @promoted_moves = [[0, +1], [+1, +1], [-1, +1], [+1, +0], [-1, +0], [0, -1]]
+ @name = "FU"
+ @promoted_name = "TO"
+ super
+ end
+ def room_of_head?(x, y, name)
+ if (name == "FU")
+ if (@sente)
+ return false if (y == 1)
+ else
+ return false if (y == 9)
+ end
+ ## 2fu check
+ c = 0
+ iy = 1
+ while (iy <= 9)
+ if ((iy != @y) && # not source position
+ @board.array[x][iy] &&
+ (@board.array[x][iy].sente == @sente) && # mine
+ (@board.array[x][iy].name == "FU") &&
+ (@board.array[x][iy].promoted == false))
+ return false
+ end
+ iy = iy + 1
+ end
+ end
+ return true
+ end
+end
+
+class PieceKY < Piece
+ def initialize(*arg)
+ @point = 1
+ @normal_moves = []
+ @promoted_moves = [[0, +1], [+1, +1], [-1, +1], [+1, +0], [-1, +0], [0, -1]]
+ @name = "KY"
+ @promoted_name = "NY"
+ super
+ end
+ def room_of_head?(x, y, name)
+ if (name == "KY")
+ if (@sente)
+ return false if (y == 1)
+ else
+ return false if (y == 9)
+ end
+ end
+ return true
+ end
+ def far_movable_grids
+ grids = Array::new
+ if (@promoted)
+ return []
+ else
+ if (@sente) # up
+ cand_x = @x
+ cand_y = @y - 1
+ while (jump_to?(cand_x, cand_y))
+ grids.push([cand_x, cand_y])
+ break if (! put_to?(cand_x, cand_y))
+ cand_y = cand_y - 1
+ end
+ else # down
+ cand_x = @x
+ cand_y = @y + 1
+ while (jump_to?(cand_x, cand_y))
+ grids.push([cand_x, cand_y])
+ break if (! put_to?(cand_x, cand_y))
+ cand_y = cand_y + 1
+ end
+ end
+ return grids
+ end
+ end
+end
+class PieceKE < Piece
+ def initialize(*arg)
+ @point = 1
+ @normal_moves = [[+1, +2], [-1, +2]]
+ @promoted_moves = [[0, +1], [+1, +1], [-1, +1], [+1, +0], [-1, +0], [0, -1]]
+ @name = "KE"
+ @promoted_name = "NK"
+ super
+ end
+ def room_of_head?(x, y, name)
+ if (name == "KE")
+ if (@sente)
+ return false if ((y == 1) || (y == 2))
+ else
+ return false if ((y == 9) || (y == 8))
+ end
+ end
+ return true
+ end
+end
+class PieceGI < Piece
+ def initialize(*arg)
+ @point = 1
+ @normal_moves = [[0, +1], [+1, +1], [-1, +1], [+1, -1], [-1, -1]]
+ @promoted_moves = [[0, +1], [+1, +1], [-1, +1], [+1, +0], [-1, +0], [0, -1]]
+ @name = "GI"
+ @promoted_name = "NG"
+ super
+ end
+end
+class PieceKI < Piece
+ def initialize(*arg)
+ @point = 1
+ @normal_moves = [[0, +1], [+1, +1], [-1, +1], [+1, +0], [-1, +0], [0, -1]]
+ @promoted_moves = []
+ @name = "KI"
+ @promoted_name = nil
+ super
+ end
+end
+class PieceKA < Piece
+ def initialize(*arg)
+ @point = 5
+ @normal_moves = []
+ @promoted_moves = [[0, +1], [+1, 0], [-1, 0], [0, -1]]
+ @name = "KA"
+ @promoted_name = "UM"
+ super
+ end
+ def far_movable_grids
+ grids = Array::new
+ ## up right
+ cand_x = @x - 1
+ cand_y = @y - 1
+ while (jump_to?(cand_x, cand_y))
+ grids.push([cand_x, cand_y])
+ break if (! put_to?(cand_x, cand_y))
+ cand_x = cand_x - 1
+ cand_y = cand_y - 1
+ end
+ ## down right
+ cand_x = @x - 1
+ cand_y = @y + 1
+ while (jump_to?(cand_x, cand_y))
+ grids.push([cand_x, cand_y])
+ break if (! put_to?(cand_x, cand_y))
+ cand_x = cand_x - 1
+ cand_y = cand_y + 1
+ end
+ ## up left
+ cand_x = @x + 1
+ cand_y = @y - 1
+ while (jump_to?(cand_x, cand_y))
+ grids.push([cand_x, cand_y])
+ break if (! put_to?(cand_x, cand_y))
+ cand_x = cand_x + 1
+ cand_y = cand_y - 1
+ end
+ ## down left
+ cand_x = @x + 1
+ cand_y = @y + 1
+ while (jump_to?(cand_x, cand_y))
+ grids.push([cand_x, cand_y])
+ break if (! put_to?(cand_x, cand_y))
+ cand_x = cand_x + 1
+ cand_y = cand_y + 1
+ end
+ return grids
+ end
+end
+class PieceHI < Piece
+ def initialize(*arg)
+ @point = 5
+ @normal_moves = []
+ @promoted_moves = [[+1, +1], [-1, +1], [+1, -1], [-1, -1]]
+ @name = "HI"
+ @promoted_name = "RY"
+ super
+ end
+ def far_movable_grids
+ grids = Array::new
+ ## up
+ cand_x = @x
+ cand_y = @y - 1
+ while (jump_to?(cand_x, cand_y))
+ grids.push([cand_x, cand_y])
+ break if (! put_to?(cand_x, cand_y))
+ cand_y = cand_y - 1
+ end
+ ## down
+ cand_x = @x
+ cand_y = @y + 1
+ while (jump_to?(cand_x, cand_y))
+ grids.push([cand_x, cand_y])
+ break if (! put_to?(cand_x, cand_y))
+ cand_y = cand_y + 1
+ end
+ ## right
+ cand_x = @x - 1
+ cand_y = @y
+ while (jump_to?(cand_x, cand_y))
+ grids.push([cand_x, cand_y])
+ break if (! put_to?(cand_x, cand_y))
+ cand_x = cand_x - 1
+ end
+ ## down
+ cand_x = @x + 1
+ cand_y = @y
+ while (jump_to?(cand_x, cand_y))
+ grids.push([cand_x, cand_y])
+ break if (! put_to?(cand_x, cand_y))
+ cand_x = cand_x + 1
+ end
+ return grids
+ end
+end
+class PieceOU < Piece
+ def initialize(*arg)
+ @point = 0
+ @normal_moves = [[0, +1], [+1, +1], [-1, +1], [+1, +0], [-1, +0], [0, -1], [+1, -1], [-1, -1]]
+ @promoted_moves = []
+ @name = "OU"
+ @promoted_name = nil
+ super
+ end
+end
+
+class Board
+ def initialize
+ @sente_hands = Array::new
+ @gote_hands = Array::new
+ @history = Hash::new(0)
+ @sente_history = Hash::new(0)
+ @gote_history = Hash::new(0)
+ @array = [[], [], [], [], [], [], [], [], [], []]
+ @move_count = 0
+ @teban = nil # black => true, white => false
+ end
+ attr_accessor :array, :sente_hands, :gote_hands, :history, :sente_history, :gote_history
+ attr_reader :move_count
+
+ def initial
+ PieceKY::new(self, 1, 1, false)
+ PieceKE::new(self, 2, 1, false)
+ PieceGI::new(self, 3, 1, false)
+ PieceKI::new(self, 4, 1, false)
+ PieceOU::new(self, 5, 1, false)
+ PieceKI::new(self, 6, 1, false)
+ PieceGI::new(self, 7, 1, false)
+ PieceKE::new(self, 8, 1, false)
+ PieceKY::new(self, 9, 1, false)
+ PieceKA::new(self, 2, 2, false)
+ PieceHI::new(self, 8, 2, false)
+ (1..9).each do |i|
+ PieceFU::new(self, i, 3, false)
+ end
+
+ PieceKY::new(self, 1, 9, true)
+ PieceKE::new(self, 2, 9, true)
+ PieceGI::new(self, 3, 9, true)
+ PieceKI::new(self, 4, 9, true)
+ PieceOU::new(self, 5, 9, true)
+ PieceKI::new(self, 6, 9, true)
+ PieceGI::new(self, 7, 9, true)
+ PieceKE::new(self, 8, 9, true)
+ PieceKY::new(self, 9, 9, true)
+ PieceKA::new(self, 8, 8, true)
+ PieceHI::new(self, 2, 8, true)
+ (1..9).each do |i|
+ PieceFU::new(self, i, 7, true)
+ end
+ @teban = true
+ end
+
+ def have_piece?(hands, name)
+ piece = hands.find { |i|
+ i.name == name
+ }
+ return piece
+ end
+
+ def move_to(x0, y0, x1, y1, name, sente)
+ if (sente)
+ hands = @sente_hands
+ else
+ hands = @gote_hands
+ end
+
+ if ((x0 == 0) || (y0 == 0))
+ piece = have_piece?(hands, name)
+ return :illegal if (! piece.move_to?(x1, y1, name)) # TODO null check for the piece?
+ piece.move_to(x1, y1)
+ else
+ return :illegal if (! @array[x0][y0].move_to?(x1, y1, name)) # TODO null check?
+ if (@array[x0][y0].name != name) # promoted ?
+ @array[x0][y0].promoted = true
+ end
+ if (@array[x1][y1]) # capture
+ if (@array[x1][y1].name == "OU")
+ return :outori # return board update
+ end
+ @array[x1][y1].sente = @array[x0][y0].sente
+ @array[x1][y1].move_to(0, 0)
+ hands.sort! {|a, b| # TODO refactor. Move to Piece class
+ a.name <=> b.name
+ }
+ end
+ @array[x0][y0].move_to(x1, y1)
+ end
+ @move_count += 1
+ @teban = @teban ? false : true
+ return true
+ end
+
+ def look_for_ou(sente)
+ x = 1
+ while (x <= 9)
+ y = 1
+ while (y <= 9)
+ if (@array[x][y] &&
+ (@array[x][y].name == "OU") &&
+ (@array[x][y].sente == sente))
+ return @array[x][y]
+ end
+ y = y + 1
+ end
+ x = x + 1
+ end
+ raise "can't find ou"
+ end
+
+ # note checkmate, but check. sente is checked.
+ def checkmated?(sente) # sente is loosing
+ ou = look_for_ou(sente)
+ x = 1
+ while (x <= 9)
+ y = 1
+ while (y <= 9)
+ if (@array[x][y] &&
+ (@array[x][y].sente != sente))
+ if (@array[x][y].movable_grids.include?([ou.x, ou.y]))
+ return true
+ end
+ end
+ y = y + 1
+ end
+ x = x + 1
+ end
+ return false
+ end
+
+ def uchifuzume?(sente)
+ rival_ou = look_for_ou(! sente) # rival's ou
+ if (sente) # rival is gote
+ if ((rival_ou.y != 9) &&
+ (@array[rival_ou.x][rival_ou.y + 1]) &&
+ (@array[rival_ou.x][rival_ou.y + 1].name == "FU") &&
+ (@array[rival_ou.x][rival_ou.y + 1].sente == sente)) # uchifu true
+ fu_x = rival_ou.x
+ fu_y = rival_ou.y + 1
+ else
+ return false
+ end
+ else # gote
+ if ((rival_ou.y != 1) &&
+ (@array[rival_ou.x][rival_ou.y - 1]) &&
+ (@array[rival_ou.x][rival_ou.y - 1].name == "FU") &&
+ (@array[rival_ou.x][rival_ou.y - 1].sente == sente)) # uchifu true
+ fu_x = rival_ou.x
+ fu_y = rival_ou.y - 1
+ else
+ return false
+ end
+ end
OSDN Git Service
