RubyShogi 0.1

data/lib/ruby_shogi/mgen_white.rb

@@ -0,0 +1,237 @@
+##
+# RubyShogi, a Shogi Engine
+# Author: Toni Helminen
+# License: GPLv3
+##
+
+module RubyShogi
+
+class MgenWhite < RubyShogi::Mgen
+	def initialize(board)
+		@board = board
+	end
+	
+	def push_move(me, to, promo) 
+		board2 = @board
+		copy = @board.copy_me
+		copy.from = @from_gen
+		copy.to = to
+		copy.promo = promo
+		copy.r50 += 1
+		copy.fullmoves += 1
+		copy.r50 = 0 if [1, 3, 5].include?(me)
+		copy.eat = copy.brd[to]
+		copy.white_pocket.push(eaten_piece(-copy.eat)) if copy.eat != 0
+		copy.wtm = !copy.wtm
+		copy.brd[@from_gen] = 0
+		copy.brd[to] = me
+		copy.wking = to if me == 14
+		#fail if copy.find_white_king != copy.wking
+		@board = copy
+		@moves.push << copy if !checks_b?
+		@board = board2
+	end
+	
+	def pawn_drop_checkmate?(to) 
+		@board.brd[to + 9] == -14 && !checks_b?(to + 9) ? true : false
+	end
+	
+	def add_new_drop_move(me, to)
+		return if me == 3 && to / 9 == 8
+		return if me == 5 && to / 9 >= 7
+		board2 = @board
+		copy = @board.copy_me
+		copy.from = @from_gen
+		copy.to = to
+		copy.drop = me
+		copy.eat = 0
+		copy.r50 += 1
+		copy.fullmoves += 1
+		copy.wtm = ! copy.wtm
+		copy.brd[@from_gen] = 0
+		copy.brd[to] = me
+		copy.white_pocket = remove_from_array(copy.white_pocket, me)
+		#fail if copy.find_white_king != copy.wking
+		@board = copy
+		if !checks_b? && !(me == -1 && pawn_drop_checkmate?(to))
+			@moves.push << copy
+		end
+		@board = board2
+	end
+	
+	def handle_capture(copy, eat) 
+		piece = case eat
+			when 1..2 then 1
+			when 3..4 then 3
+			when 5..6 then 5
+			when 7..8 then 7
+			when 9 then 9
+			when 10..11 then 10
+			when 12..13 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 handle_promotion?(me, to)
+		return true if must_promote?(me, to)
+		return false if to / 9 <= 5 && @from_gen / 9 <= 5
+		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
+		false
+	end
+	
+	def must_promote?(me, to)
+		if me == 5 && to / 9 >= 7
+			push_move(6, to, PROMO_YES)
+			return true
+		end
+		return false if to / 9 != 8
+		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 6..7
+			push_move(1, to, PROMO_STAY)
+			push_move(2, to, PROMO_YES)
+		when 8 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 < 81 && @board.walkable_w?(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_w?(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_w?(to)
+				break if !@board.empty?(to)
+			end
+		end
+	end
+	
+	def pawn_on_column?(c)
+		ret = false
+		9.times do |i|
+			to = 9 * i + c
+			if to != @from_gen && @board.brd[to] == 1
+				ret = true
+				break
+			end
+		end
+		ret
+	end
+	
+	def put_pawn_drops
+		(9*8).times do |i| 
+			@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.white_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
+			case @board.brd[i]
+			when 1 then	generate_pawn_moves
+			when 2 then generate_jump_moves(WHITE_GOLD_GENERAL_MOVES, 2)
+			when 3 then generate_slider_moves(WHITE_LANCE_MOVES, 3)
+			when 4 then generate_jump_moves(WHITE_GOLD_GENERAL_MOVES, 4)
+			when 5 then generate_jump_moves(WHITE_KNIGHT_MOVES, 5)
+			when 6 then generate_jump_moves(WHITE_GOLD_GENERAL_MOVES, 6)
+			when 7 then generate_jump_moves(WHITE_SILVER_GENERAL_MOVES, 7)
+			when 8 then generate_jump_moves(WHITE_GOLD_GENERAL_MOVES, 8)
+			when 9 then generate_jump_moves(WHITE_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 MgenWhite
+
+end # module RubyShogi

data/lib/ruby_shogi/perft.rb

@@ -0,0 +1,87 @@
+##
+# RubyShogi, a Shogi Engine
+# Author: Toni Helminen
+# License: GPLv3
+##
+
+module RubyShogi
+
+class Perft
+	attr_accessor :board
+	
+	NUMS = { # https://groups.google.com/forum/#!topic/shogi-l/U7hmtThbk1k
+		0 => 1,
+		1 => 30,
+		2 => 900,
+		3 => 25470,
+		4 => 719731,
+		5 => 19861490,
+		6 => 547581517
+	}
+	
+	def initialize(fen = nil)
+		@board = RubyShogi::Board.new(fen)
+	end
+	
+	def perft_number(depth)
+		return 1 if depth == 0
+		board = @board
+		mgen = @board.mgen_generator
+		n, moves = 0, mgen.generate_moves
+		return moves.length if depth <= 1
+		moves.each do |move|
+			@board = move
+			n += perft_number(depth - 1)
+		end
+		@board = board
+		n
+	end
+	
+	def randperft(depth)
+		@board = @board.randpos
+		perft(depth)
+	end
+	
+	def perft(depth)
+		puts "~~~ perft( #{depth} ) ~~~"
+		puts "[ fen: #{@board.pos2fen} ]"
+		total_time = 0
+		total_nodes = 0
+		copy = @board
+		(depth+1).times do |i|
+			start = Time.now
+			@board = copy
+			n = perft_number(i)
+			diff = Time.now - start
+			total_time += diff
+			total_nodes += n
+			nps = (diff == 0 or n == 1) ? n : (n / diff).to_i
+			puts "#{i}: #{n} | #{diff.round(3)}s | #{nps} nps"
+		end
+		total_time = 1 if total_time == 0
+		puts "= #{total_nodes} | #{total_time.round(3)}s | #{(total_nodes/total_time).to_i} nps"
+	end
+	
+	def suite(depth)
+		puts "~~~ suite( #{depth} ) ~~~"
+		total_time = 0
+		total_nodes = 0
+		copy = @board
+		(depth+1).times do |i|
+			start = Time.now
+			@board = copy
+			n = perft_number(i)
+			diff = Time.now - start
+			total_time += diff
+			total_nodes += n
+			nps = (diff == 0 or n == 1) ? n : (n / diff).to_i
+			error = ["ok", "error"][NUMS[i] - n == 0 ? 0 : 1]
+			break if i >= NUMS.length - 1
+			puts "#{i}: #{n} | #{diff.round(3)}s | #{nps} nps | #{error}"
+		end
+		total_time = 1 if total_time == 0
+		puts "= #{total_nodes} | #{total_time.round(3)}s | #{(total_nodes/total_time).to_i} nps"
+	end
+end # class Perft
+
+end # module ShurikenShogi

data/lib/ruby_shogi/ruby_shogi.rb

@@ -0,0 +1,66 @@
+##
+#
+# RubyShogi, a Shogi Engine
+# Copyright (C) 2019 Toni Helminen ( kalleankka1@gmail.com )
+#
+# RubyShogi is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# RubyShogi is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+##
+
+require_relative "./engine"
+require_relative "./zobrist"
+require_relative "./history"
+require_relative "./cmd"
+require_relative "./xboard"
+require_relative "./eval"
+require_relative "./utils"
+require_relative "./bench"
+require_relative "./tokens"
+require_relative "./board"
+require_relative "./mgen"
+require_relative "./mgen_white"
+require_relative "./mgen_black"
+require_relative "./perft"
+require_relative "./tactics"
+
+$stdout.sync = true
+$stderr.sync = true
+Thread.abort_on_exception = true
+$stderr.reopen("ruby_shogi-error.txt", "a+")
+
+module RubyShogi
+	NAME = "RubyShogi"
+	VERSION = "0.1"
+	AUTHOR = "Toni Helminen"
+
+	def RubyShogi.init
+		RubyShogi::Eval.init
+		RubyShogi::Zobrist.init
+	end
+	
+	# Start RubyShogi
+	#
+	# Example:
+	#   >> ruby_shogi -xboard
+	#   => enter xboard mode
+	def RubyShogi.go
+		cmd = RubyShogi::Cmd.new
+		cmd.args
+	end
+end # module RubyShogi
+
+RubyShogi.init # init just once
+
+if __FILE__ == $0
+	RubyShogi.go
+end

data/lib/ruby_shogi/tactics.rb

@@ -0,0 +1,33 @@
+##
+# RubyShogi, a Shogi Engine
+# Author: Toni Helminen
+# License: GPLv3
+##
+
+module RubyShogi
+
+module Tactics
+	TACTICS = [	
+		["3k5/2P1P4/3K5/9/1N7/9/9/9/9[-] w 0 1", "b5c7="], # mate in 1
+		["4k4/9/4K4/9/9/9/9/9/9[G] w 0 1", "G@e8"], # mate in 1
+		["9/9/9/9/9/9/4K4/R4R3/6k2[-] w 0 1", "a2a1"] # mate in 1
+	]
+	
+	def Tactics.run
+		puts "~~~ Tactics ~~~"
+		score, total = 0, 0
+		TACTICS.each do |tactic|
+			engine = RubyShogi::Engine.new
+			engine.printinfo = false
+			engine.board.fen(tactic[0])
+			engine.time = 50
+			result = engine.think
+			total += 1
+			score += 1 if tactic[1] == result
+			puts "#{total}. move #{result} | " + (tactic[1] == result ? "ok" : "error")
+		end
+		puts "= #{score} / #{total}"
+	end
+end # module Tactics
+
+end # module RubyShogi

data/lib/ruby_shogi/tokens.rb

@@ -0,0 +1,42 @@
+##
+# RubyShogi, a Shogi Engine
+# Author: Toni Helminen
+# License: GPLv3
+##
+
+module RubyShogi
+
+class Tokens
+	def initialize(tokens)
+		@tokens = tokens
+		@token_i = 0
+	end
+	
+	def peek(n)
+		return nil if @token_i + n < 0 || @token_i + n >= @tokens.length
+		@tokens[@token_i + n]
+	end
+	
+	def ok?
+		return @token_i < @tokens.length ? true : false
+	end
+	
+	def go_next
+		v = nil
+		if @token_i < @tokens.length
+			v = @tokens[@token_i]
+			@token_i += 1 
+		end
+		return v
+	end
+	
+	def forward
+		@token_i += 1 
+	end
+	
+	def cur
+		@tokens[@token_i]
+	end
+end # class Tokens
+
+end # module RubyShogi