RubyShogi 0.1

data/lib/ruby_shogi/eval.rb

@@ -0,0 +1,102 @@
+##
+# RubyShogi, a Shogi Engine
+# Author: Toni Helminen
+# License: GPLv3
+##
+
+module RubyShogi
+
+module Eval
+	# https://en.wikipedia.org/wiki/Shogi_strategy#Relative_piece_value
+	MATERIAL_SCORE = {
+		1 => 1,
+		2 => 4.2,
+		3 => 4.3,
+		4 => 6.3,
+		5 => 4.5,
+		6 => 6.4,
+		7 => 6.4,
+		8 => 6.7,
+		9 => 6.9,
+		10 => 8.9,
+		11 => 11.5,
+		12 => 10.4,
+		13 => 13,
+		14 => 0
+	}
+
+	MATERIAL_HAND_SCORE = {
+		1 => 1.15,
+		3 => 4.8,
+		5 => 5.1,
+		7 => 7.2,
+		9 => 7.8,
+		10 => 11.10,
+		12 => 10.4,
+		13 => 12.7
+	}
+	
+	CENTRAL_BONUS = [1,2,3,4,5,4,3,2,1].freeze
+	
+	CENTRAL_SCORE = {
+		1 => 0,
+		2 => 2,
+		3 => 1,
+		4 => 2,
+		5 => 1,
+		6 => 2,
+		7 => 2,
+		8 => 2,
+		9 => 2,
+		10 => 2,
+		11 => 2,
+		12 => 2,
+		13 => 2,
+		14 => 0
+	}
+	
+	EVAL_PST_MG = []
+
+	def Eval.init
+		return if EVAL_PST_MG.length > 0
+		14.times do |i|
+			arr = []
+			81.times do |j|
+				score = 0.1 * (MATERIAL_SCORE[i + 1] + 2 * CENTRAL_SCORE[i + 1] * (CENTRAL_BONUS[j % 9 ] + CENTRAL_BONUS[j / 9]))
+				arr.push(score)
+			end
+			EVAL_PST_MG.push(arr)
+		end
+		EVAL_PST_MG.freeze
+	end
+	
+	def Eval.eval(board)
+		score = 0
+		board.brd.each_with_index do |p, i|
+			score += case p
+				when 1..14 then EVAL_PST_MG[p - 1][i]
+				when -14..-1 then EVAL_PST_MG[-p - 1][i]
+				else 
+					0
+				end
+		end
+		0.01 * score
+	end
+	
+	def Eval.material(board)
+		score = 0
+		board.brd.each do |p|
+			score += case p
+				when 1..14 then MATERIAL_SCORE[p]
+				when -14..-1 then -MATERIAL_SCORE[-p]
+				else 
+					0
+				end
+		end
+		board.white_pocket.each { |p| score += MATERIAL_HAND_SCORE[p] }
+		board.black_pocket.each { |p| score -= MATERIAL_HAND_SCORE[-p] }
+		score
+	end
+end # module Eval
+
+end # module RubyShogi

data/lib/ruby_shogi/history.rb

@@ -0,0 +1,63 @@
+##
+# RubyShogi, a Shogi Engine
+# Author: Toni Helminen
+# License: GPLv3
+##
+
+module RubyShogi
+
+class History
+	def initialize
+		reset
+	end
+
+	def reset
+		@data = []
+		@pos = -1
+	end
+	
+	def debug
+		puts "@pos: #{@pos} .. @data: #{@data.length}"
+	end
+	
+	def remove
+		if @pos > 1
+			board = @data[@pos - 2]
+			@pos -= 2
+			return board
+		end
+		@data.last
+	end
+
+	def undo
+		if @pos > 0
+			board = @data[@pos - 1]
+			@pos -= 1
+			return board
+		end
+		@data.last
+	end
+	
+	def add(board)
+		@data.push(board)
+		@pos += 1
+	end
+	
+	def draw_too_long?
+		@data.length > 900 # I give up...
+	end
+	
+	def is_draw?(board, repsn = 4)
+		len, hash = @data.length, board.hash
+		i, n, reps = len - 1, 0, 0
+		while i > 0
+			break if n >= 100 
+			reps += 1 if hash == @data[i].hash
+			n, i = n + 1, i - 1
+			return true if reps >= repsn
+		end
+		false
+	end
+end # class History
+
+end # module RubyShogi

data/lib/ruby_shogi/mgen.rb

@@ -0,0 +1,169 @@
+##
+# RubyShogi, a Shogi Engine
+# Author: Toni Helminen
+# License: GPLv3
+##
+
+module RubyShogi
+
+class Mgen
+	# both
+	ROOK_MOVES = [[1, 0], [0, 1], [-1, 0], [0, -1]].freeze
+	BISHOP_MOVES = [[1, 1], [-1, 1], [1, -1], [-1, -1]].freeze
+	KING_MOVES = (ROOK_MOVES + BISHOP_MOVES).freeze
+	
+	PROMOTED_BISHOP_MOVES = ROOK_MOVES
+	PROMOTED_ROOK_MOVES = BISHOP_MOVES
+	
+	# white
+	WHITE_GOLD_GENERAL_MOVES = [[1, 0], [-1, 0], [0, -1], [0, 1], [1, 1], [-1, 1]].freeze
+	WHITE_SILVER_GENERAL_MOVES = [[-1, -1], [-1, 1], [0, 1], [1, 1], [1, -1]].freeze
+	WHITE_KNIGHT_MOVES = [[-1, 2], [1, 2]].freeze
+	WHITE_LANCE_MOVES = [[0, 1]].freeze
+	
+	# black
+	BLACK_GOLD_GENERAL_MOVES = [[1, 0], [-1, 0], [0, 1], [0, -1], [1, -1], [-1, -1]].freeze
+	BLACK_SILVER_GENERAL_MOVES = [[-1, 1], [-1, -1], [0, -1], [1, -1], [1, 1]].freeze
+	BLACK_KNIGHT_MOVES = [[-1, -2], [1, -2]].freeze
+	BLACK_LANCE_MOVES = [[0, -1]].freeze
+
+	# promotions
+	PROMO_NO   = 0
+	PROMO_STAY = 1
+	PROMO_YES  = 2
+	
+	attr_accessor :pseudo_moves, :only_captures
+	
+	def initialize(board)
+		@board, @moves = board, []
+		@x_gen, @y_gen, @from_gen = 0, 0, 0 # move generation
+		@x_checks, @y_checks = 0, 0 # checks
+		@pseudo_moves = false # 3x speed up
+	end
+
+	##
+	# Utils
+	##
+	
+	def print_move_list
+		@moves.each_with_index { |board, i| puts "#{i}: #{board.move_str}" }
+	end
+	
+	def is_on_board?(x, y)	
+		x >= 0 && x <= 8 && y >= 0 && y <= 8
+	end
+	
+	def remove_from_array(array, x)
+		found, a = false, []
+		array.each do |q|
+			if q == x
+				a.push(q) if found
+				found = true 
+			else
+				a.push(q)
+			end
+		end
+		fail if !found
+		a
+	end
+	
+	def good_coord?(i)
+		i >= 0 && i <= 80
+	end
+	
+	def eaten_piece(eat) 
+		case eat
+		when 2 then 1
+		when 4 then 3
+		when 6 then 5
+		when 8 then 7
+		when 11 then 10
+		when 13 then 12
+		else eat
+		end
+	end
+
+	##
+	# Checks
+	##
+	
+	def pawn_checks_w?(here)
+		@x_checks + 9 * (@y_checks + 1) == here
+	end
+	
+	def pawn_checks_b?(here)
+		@x_checks + 9 * (@y_checks - 1) == here
+	end
+
+	def slider_checks_to?(slider, here)
+		slider.each do |jmp|
+			px, py = @x_checks, @y_checks
+			loop do
+				px, py = px + jmp[0], py + jmp[1]
+				break if !is_on_board?(px, py)
+				to = px + py * 9
+				return true if to == here
+				break if !@board.empty?(to)
+			end
+		end
+		false
+	end
+	
+	def jump_checks_to?(jumps, here)
+		jumps.each do |jmp|
+			px, py = @x_checks + jmp[0], @y_checks + jmp[1]
+			return true if px + py * 9 == here
+		end
+		false
+	end
+	
+	def checks_w?#(here)
+		here = @board.bking
+		81.times do |i|
+			@x_checks, @y_checks = i % 9, i / 9 
+			case @board.brd[i]
+			when 1 then return true if pawn_checks_w?(here)
+			when 2 then return true if jump_checks_to?(WHITE_GOLD_GENERAL_MOVES, here)
+			when 3 then return true if slider_checks_to?(WHITE_LANCE_MOVES, here)
+			when 4 then return true if jump_checks_to?(WHITE_GOLD_GENERAL_MOVES, here)
+			when 5 then return true if jump_checks_to?(WHITE_KNIGHT_MOVES, here)
+			when 6 then return true if jump_checks_to?(WHITE_GOLD_GENERAL_MOVES, here)
+			when 7 then return true if jump_checks_to?(WHITE_SILVER_GENERAL_MOVES, here)
+			when 8 then return true if jump_checks_to?(WHITE_GOLD_GENERAL_MOVES, here)
+			when 9 then return true if jump_checks_to?(WHITE_GOLD_GENERAL_MOVES, here)
+			when 10 then return true if slider_checks_to?(BISHOP_MOVES, here)
+			when 11 then return true if slider_checks_to?(BISHOP_MOVES, here) || jump_checks_to?(PROMOTED_BISHOP_MOVES, here)
+			when 12 then return true if slider_checks_to?(ROOK_MOVES, here)
+			when 13 then return true if slider_checks_to?(ROOK_MOVES, here) || jump_checks_to?(PROMOTED_ROOK_MOVES, here)
+			when 14 then return true if jump_checks_to?(KING_MOVES, here)
+			end
+		end
+		false
+	end
+	
+	def checks_b?#(here)
+		here = @board.wking
+		81.times do |i|
+			@x_checks, @y_checks = i % 9, i / 9 
+			case @board.brd[i]
+			when -1 then return true if pawn_checks_b?(here)
+			when -2 then return true if jump_checks_to?(BLACK_GOLD_GENERAL_MOVES, here)
+			when -3 then return true if slider_checks_to?(BLACK_LANCE_MOVES, here)
+			when -4 then return true if jump_checks_to?(BLACK_GOLD_GENERAL_MOVES, here)
+			when -5 then return true if jump_checks_to?(BLACK_KNIGHT_MOVES, here)
+			when -6 then return true if jump_checks_to?(BLACK_GOLD_GENERAL_MOVES, here)
+			when -7 then return true if jump_checks_to?(BLACK_SILVER_GENERAL_MOVES, here)
+			when -8 then return true if jump_checks_to?(BLACK_GOLD_GENERAL_MOVES, here)
+			when -9 then return true if jump_checks_to?(BLACK_GOLD_GENERAL_MOVES, here)
+			when -10 then return true if slider_checks_to?(BISHOP_MOVES, here)
+			when -11 then return true if slider_checks_to?(BISHOP_MOVES, here) || jump_checks_to?(PROMOTED_BISHOP_MOVES, here)
+			when -12 then return true if slider_checks_to?(ROOK_MOVES, here)
+			when -13 then return true if slider_checks_to?(ROOK_MOVES, here) || jump_checks_to?(PROMOTED_ROOK_MOVES, here)
+			when -14 then return true if jump_checks_to?(KING_MOVES, here)
+			end
+		end
+		false
+	end
+end # class Mgen
+
+end # module RubyShogi

data/lib/ruby_shogi/mgen_black.rb

@@ -0,0 +1,238 @@
+##
+# RubyShogi, a Shogi Engine
+# Author: Toni Helminen
+# License: GPLv3
+##
+
+module RubyShogi
+
+class MgenBlack < RubyShogi::Mgen
+	def initialize(board)
+		@board = board
+	end
+	
+	def handle_capture(copy, eat) 
+		piece = case eat
+			when -2..-1 then -1
+			when -4..-3 then -3
+			when -6..-5 then -5
+			when -8..-7 then -7
+			when -9 then -9
+			when -11..-10 then -10
+			when -13..-12 then -12
+			end
+		copy.white_pocket.push(piece)
+	end
+	
+	def add_new_move(me, to, promo = 0)
+		return if handle_promotion?(me, to)
+		push_move(me, to, promo)
+	end
+	
+	def push_move(me, to, promo) 
+		board2 = @board
+		copy = @board.copy_me
+		copy.from = @from_gen
+		copy.promo = promo
+		copy.r50 += 1
+		copy.fullmoves += 1
+		copy.r50 = 0 if [-1, -3, -5].include?(me)
+		copy.to = to
+		copy.eat = copy.brd[to]
+		copy.black_pocket.push(-1 * eaten_piece(copy.eat)) if copy.eat != 0
+		copy.wtm = !copy.wtm
+		copy.brd[@from_gen] = 0
+		copy.brd[to] = me
+		copy.bking = to if me == -14
+		#fail if copy.find_black_king != copy.bking
+		@board = copy
+		@moves.push << copy if !checks_w?
+		@board = board2
+	end
+	
+	def pawn_drop_checkmate?(to) 
+		@board.brd[to - 9] == 14 && !checks_w?(to - 9) ? true : false
+	end
+	
+	def add_new_drop_move(me, to) 
+		return if me == -3 && to / 9 == 0
+		return if me == -5 && to / 9 <= 1
+		board2 = @board
+		copy = @board.copy_me
+		copy.from = @from_gen
+		copy.to = to
+		copy.drop = me
+		copy.r50 += 1
+		copy.fullmoves += 1
+		copy.eat = 0
+		copy.wtm = ! copy.wtm
+		copy.brd[@from_gen] = 0
+		copy.brd[to] = me
+		copy.black_pocket = remove_from_array(copy.black_pocket, me)
+		#fail if copy.find_black_king != copy.bking
+		@board = copy
+		if !checks_w? && !(me == 1 && pawn_drop_checkmate?(to))
+			@moves.push << copy
+		end
+		@board = board2
+	end
+	
+	def handle_promotion?(me, to)
+		return true if must_promote?(me, to)
+		return false if to / 9 > 2 && @from_gen / 9 > 2
+		case me
+		when -1 
+			push_move(-1, to, PROMO_STAY)
+			push_move(-2, to, PROMO_YES)
+			return true
+		when -3 
+			push_move(-3, to, PROMO_STAY)
+			push_move(-4, to, PROMO_YES)
+			return true
+		when -5 
+			push_move(-5, to, PROMO_STAY)
+			push_move(-6, to, PROMO_YES)
+			return true
+		when -7 
+			push_move(-7, to, PROMO_STAY)
+			push_move(-8, to, PROMO_YES)
+			return true
+		when -10 
+			push_move(-10, to, PROMO_STAY)
+			push_move(-11, to, PROMO_YES)
+			return true
+		when -12 
+			push_move(-12, to, PROMO_STAY)
+			push_move(-13, to, PROMO_YES)
+			return true
+		end
+	end
+	
+	def must_promote?(me, to)
+		if me == -5 && to / 9 <= 1
+			push_move(-6, to, PROMO_YES)
+			return true
+		end
+		return false if to / 9 != 0
+		case me
+		when -1 
+			push_move(-2, to, PROMO_YES)
+			return true
+		when -3
+			push_move(-4, to, PROMO_YES)
+			return true
+		end
+		false
+	end
+
+	def add_new_pawn_move(to)
+		case to / 9
+		when 1..2
+			push_move(-1, to, PROMO_STAY)
+			push_move(-2, to, PROMO_YES)
+		when 0 then push_move(-2, to, PROMO_YES)
+		else
+			push_move(-1, to, PROMO_NO)
+		end
+	end
+	
+	def generate_pawn_moves
+		to = @x_gen + (@y_gen - 1) * 9
+		add_new_pawn_move(to) if to >= 0 && @board.walkable_b?(to)
+	end
+		
+	def generate_jump_moves(jumps, me)
+		jumps.each do |jmp|
+			px, py = @x_gen + jmp[0], @y_gen + jmp[1]
+			to = px + 9 * py
+			add_new_move(me, to) if is_on_board?(px, py) && @board.walkable_b?(to)
+		end
+	end
+	
+	def generate_slider_moves(slider, me)
+		slider.each do | jmp |
+			px, py = @x_gen, @y_gen
+			loop do
+				px, py = px + jmp[0], py + jmp[1]
+				break if !is_on_board?(px, py)
+				to = px + 9 * py
+				add_new_move(me, to) if @board.walkable_b?(to)
+				break if !@board.empty?(to)
+			end
+		end
+	end
+	
+	def pawn_on_column?(c)
+		ret = false
+		9.times do |i|
+			to = -9 * i + 8 * 9 + c
+			if to != @from_gen && @board.brd[to] == -1
+				ret = true
+				break
+			end
+		end
+		ret
+	end
+	
+	def put_pawn_drops
+		(9*8).times do |i2|
+			i = i2 + 9
+			@x_gen, @y_gen, @from_gen = i % 9, i / 9, i
+			add_new_drop_move(-1, i) if !pawn_on_column?(i % 9 ) && @board.brd[i] == 0
+		end
+	end
+	
+	def put_drops(piece)
+		81.times do |i|
+			@x_gen, @y_gen, @from_gen = i % 9, i / 9, i
+			add_new_drop_move(piece, i) if @board.brd[i] == 0
+		end
+	end
+	
+	def generate_drops
+		@board.black_pocket.each do |piece|
+			case piece
+			when -1 then put_pawn_drops
+			when -3 then put_drops(-3)
+			when -5 then put_drops(-5)
+			when -7 then put_drops(-7)
+			when -9 then put_drops(-9)
+			when -10 then put_drops(-10)
+			when -11 then put_drops(-11)
+			when -12 then put_drops(-12)
+			end
+		end
+		@moves
+	end
+	
+	def generate_moves
+		@moves = []
+		81.times do |i|
+			@x_gen, @y_gen, @from_gen = i % 9, i / 9, i
+			#puts @board.brd[i]
+			case @board.brd[i]
+			when -1 then generate_pawn_moves
+			when -2 then generate_jump_moves(BLACK_GOLD_GENERAL_MOVES, -2)
+			when -3 then generate_slider_moves(BLACK_LANCE_MOVES, -3)
+			when -4 then generate_jump_moves(BLACK_GOLD_GENERAL_MOVES, -4)
+			when -5 then generate_jump_moves(BLACK_KNIGHT_MOVES, -5)
+			when -6 then generate_jump_moves(BLACK_GOLD_GENERAL_MOVES, -6)
+			when -7 then generate_jump_moves(BLACK_SILVER_GENERAL_MOVES, -7)
+			when -8 then generate_jump_moves(BLACK_GOLD_GENERAL_MOVES, -8)
+			when -9 then generate_jump_moves(BLACK_GOLD_GENERAL_MOVES, -9)
+			when -10 then generate_slider_moves(BISHOP_MOVES, -10)
+			when -11 
+				generate_slider_moves(BISHOP_MOVES, -11)
+				generate_jump_moves(PROMOTED_BISHOP_MOVES, -11)
+			when -12 then generate_slider_moves(ROOK_MOVES, -12)
+			when -13 
+				generate_slider_moves(ROOK_MOVES, -13)
+				generate_jump_moves(PROMOTED_ROOK_MOVES, -13)
+			when -14 then generate_jump_moves(KING_MOVES, -14)
+			end
+		end
+		generate_drops
+	end
+end # class MgenBlack
+
+end # module RubyShogi