#!/usr/bin/ruby

EMPTY = 0
TEAM_1 = 1
TEAM_2 = 2
TEAM_1_MAN = ' w '
TEAM_1_KING = ' W '
TEAM_2_MAN = ' b '
TEAM_2_KING = ' B '
BOARD_EMPTY = ' _ '
BOARD_SIZE = 8

class Board
	attr :stats, true

	def initialize()
		@board = Array.new(BOARD_SIZE) {|i| Array.new(BOARD_SIZE, BOARD_EMPTY)}
		@stats = {TEAM_1 => {}, TEAM_2 => {}}
	end

	def setup()
		0.step(BOARD_SIZE) {|i|
		       	peice = (i < 3 ? TEAM_1_MAN : TEAM_2_MAN)	
			if [0,2,6].include? i
				start = 1
			elsif [1,5,7].include? i
				start = 0
			else
				next
			end
			start.step(BOARD_SIZE-1,2) {|j|
				@board[i][j] = peice
			}
		}
		@stats[TEAM_1]['count'] = 12
		@stats[TEAM_2]['count'] = 12
	end

	def set(x, y, thing)
		@board[y][x] = thing
	end

	def get(x, y)
		@board[y][x]
	end

	def to_s()
		str = '  '
		0.step(BOARD_SIZE-1) {|i|
			str += ' ' + i.to_s + ' '
		}
		str += "\n"
		i = 0;
		@board.each {|row|
			str += ('A'.ord + i).chr + ' '
			i += 1
			row.each {|cell| 
				str += cell
			} 
			str += "\n"
		}
		str += 'White: ' + @stats[TEAM_1]['count'].to_s + ' -- Black: ' + @stats[TEAM_2]['count'].to_s  + "\n" 
		return str
	end

	def piece_team(piece)
		if piece == TEAM_1_MAN or piece == TEAM_1_KING
			return TEAM_1
		elsif piece == TEAM_2_MAN or piece == TEAM_2_KING
			return TEAM_2
		else
			return EMPTY
		end
	end

	def empty?(x, y)
		return @board[y][x] == BOARD_EMPTY
	end

	def valid_coords(x, y)
		return (x >= 0 and x < BOARD_SIZE and y >= 0 and y < BOARD_SIZE)
	end

	def gen_moves(x, y, x_mod, y_mod)
		moves = []
		for xm in x_mod
			new_x = x + xm
			for ym in y_mod
				new_y = y + ym
				if valid_coords(new_x, new_y) and empty?(new_x, new_y)
					moves += [[new_x, new_y]]
				end
			end
		end
		return moves
	end

	# function to chart path of jumps
	# merging them together
	# only returns full paths
	#
	# base case: no jumps avail, return path
	# jumps avail: do all and merge and return
	def gen_jumps(path, x, y, team, x_mod, y_mod, taken=[])
		jumps = []
		for xm in x_mod
			new_x = x + xm
			for ym in y_mod
				new_y = y + ym
				if valid_coords(new_x, new_y) and piece?(new_x, new_y) and !taken.include?([new_x, new_y]) and team != team(new_x, new_y)
					taken.push([new_x, new_y])
					final_x = new_x + xm
					final_y = new_y + ym
					if (valid_coords(final_x, final_y) and empty?(final_x, final_y))
						jumps += gen_jumps(path + [[final_x, final_y]], final_x, final_y, team, x_mod, y_mod, taken)
					end
				end
			end
		end

		if jumps == []
			if path == []
				return []
			else 
				return [path]
			end
		else
			return jumps
		end
	end

	def piece?(x, y)
		return !empty?(x, y)
	end

	def team(x, y)
		if piece?(x, y)
			if @board[y][x] == TEAM_1_MAN or @board[y][x] == TEAM_1_KING
				return TEAM_1
			else
				return TEAM_2
			end
		end
		return EMPTY
	end

	def opposite_team(team)
		return (team == TEAM_1 ? TEAM_2 : TEAM_1)
	end

	def man?(x, y)
		return (@board[y][x] == TEAM_1_MAN or @board[y][x] == TEAM_2_MAN)
	end
	
	def king?(x, y)
		return (@board[y][x] == TEAM_1_KING or @board[y][x] == TEAM_2_KING)
	end

	def piece_valid_moves(x, y, team)
		moves = []
		if piece_team(@board[y][x]) == team
			direction = [1]
			if king?(x, y)
				direction = [-1, 1]
			elsif team == TEAM_2 
				direction = [-1]
			end

			moves = gen_moves(x, y, [-1,1], direction)
			moves += gen_jumps([], x, y, team, [-1, 1], direction)
		end
		return moves
	end

	def dup() 
		b = Board.new
		for x in 0..(BOARD_SIZE-1)
			for y in 0..(BOARD_SIZE-1)
				b.set(x, y, get(x, y))
			end
		end
		b.stats[TEAM_1] = @stats[TEAM_1].dup
		b.stats[TEAM_2] = @stats[TEAM_2].dup
		return b
	end

	def between(c1, c2)
		return [ c2[0] + (c1[0]-c2[0])/2, c2[1] + (c1[1]-c2[1])/2 ]
	end

	def move_piece(c1, c2)
		piece = get(c1[0], c1[1])
		set(c1[0], c1[1], BOARD_EMPTY)
		set(c2[0], c2[1], piece)
		if team(c2[0], c2[1]) == TEAM_1 and c2[1] == BOARD_SIZE-1
			set(c2[0], c2[1], TEAM_1_KING)
		elsif (team(c2[0], c2[1]) == TEAM_2 and c2[1] == 0)
			set(c2[0], c2[1], TEAM_2_KING)
		end
	end


	# doesnt validate
	def do_move(x, y, move, team)
		if move[0].is_a?(Array)
			# jump
			move.each { |jump|
				enemy = between([x,y], jump)
				set(enemy[0], enemy[1], BOARD_EMPTY)
				@stats[opposite_team(team)]['count'] -= 1
				move_piece([x,y], jump)
			}
		else
			# move
			move_piece([x,y], move) 
		end
	end	


	def gen_next_move_boards(team)
		boards = []
		for x in 0..(BOARD_SIZE-1)
			for y in 0..(BOARD_SIZE-1)
				if piece?(x, y) and team(x,y) == team
					moves = piece_valid_moves(x, y, team(x, y))
					moves.each { |move|
						board = dup()
						board.do_move(x, y, move, team(x, y))
						boards.push( { "board" => board, "move" => move})
					}
				end
			end
		end
		return boards
	end

	def score()
		scores = {}
		[TEAM_1, TEAM_2].each {|team|
			score = 0
			score += @stats[team]['count']
			score += 12 - @stats[opposite_team(team)]['count']
			if @stats[opposite_team(team)]['count'] == 0
				score += 12
			elsif @stats[team]['count'] == 0
				score -= 12
			end
			scores[team] = score
		}
		return scores	
	end

	def search(team)
		search_do(team, team, 0, 6, 100.0, 0.0, @stats.dup, [])
	end

	# the owner of the search: affects pruning bonuses
        # team - current player this turn
        # current depth - counting up to max_depth
        # max_depth - when we should stop our search
        # my_percent - the percent range this work covers
        # done_percent - the percent of work already done
	# stats - stats from 2 turns ago, ie before most recent opponent turn
	# maxs - depth indexed array of max scores
	def search_do(owner, team, current_depth, max_depth, my_percent, done_percent, stats, maxs)
		#puts current_depth.to_s + '/' + max_depth.to_s + ' ' + done_percent.to_s + "%"

		if current_depth < maxs.length and score()[team] < maxs[current_depth][team]
			return score()
		end
		if current_depth >= maxs.length or score()[team] > maxs[current_depth][team]
			maxs[current_depth] = score()
		end
		
		if current_depth >= max_depth
			return score()
		end

		if @stats[TEAM_1]['count'] == 0
			puts "TEAM_2 WON!"
			return score()
		elsif @stats[TEAM_2]['count'] == 0
			puts "TEAM_1 WON!"
			puts to_s
			return score()
		else
			moves = gen_next_move_boards(team)
			if moves == []
				puts "NO MOVES AVAILABLE?"
				return score()
			else
				jumps = moves.find_all { |item| item['move'][0].is_a?(Array)}
				if jumps.length > 0
					moves = jumps
				end
				done = 0.0
				item_percent = my_percent/moves.length
				max = nil
				for i in 0..moves.length-1 do
					depth_mod = 0
					move = moves[i]['board']
					# modify depth on piece loss
        				if  stats[team]['count'] > move.stats[team]['count']
						# search less is search owner lost peices
						if owner == team
							depth_mod = -4
						else
							depth_mod = 4
						end
					end
					sub_score = move.search_do(owner, opposite_team(team), current_depth + 1, max_depth + depth_mod, item_percent, done_percent+done,  @stats.dup, maxs)
       					done += item_percent
					if done >= 0.01
          					done_percent += done;
						puts "%.2f" % done_percent + "% depth: " + current_depth.to_s + "/" + max_depth.to_s + " max: " + maxs.to_s
						done = 0.0
        				end
					#puts "Score: " + sub_score.to_s
					if max == nil or sub_score[team] > max[team]
						max = sub_score
					end
				end
				return max
			end
		 end
	end

	def parse_coords(str)
		if str.length != 2
			return false
		end
		
		y = str.downcase[0]
		x = str[1]

		if x >= '0' and x <= '7'
			x = x.ord - '0'.ord
		else 
			return false
		end

		if y >= 'a' and y <= 'h'
			y = y.ord - 'a'.ord
		else 
			return false
		end
		return [x,y]
	end

	# todo: deal with loss when no move avail
	def play()
		setup()
		color = ''
		while color != 'w' and color != 'b'
			print "Choose color ([W]hite or [B]lack): "
			color = gets
			color = color.downcase[0]
		end

		turn = 1
		team = TEAM_1
		while true
			print "Turn " + turn.to_s + ": "
			if team == TEAM_1
				print "white "
				if color == 'w'
					print "(player)"
				else
					print "(ai)"
				end
			else
				print "black "
				if color == 'b'
					print '(player)'
				else
					print '(ai)'
				end
			end
			print "\n"
			puts to_s()

			if (team == TEAM_1 and color == 'w') or (team == TEAM_2 and color == 'b')
				valid = false
				while !valid
					print "Move piece: "
					from = gets().strip!
					print "to: "
					to = gets().strip!
					from = parse_coords(from)
					to = parse_coords(to)
					puts to.to_s + " or " + from.to_s
					if !to or !from
						next
					end
					if team(from[0], from[1]) != team
						next
					end
					if !piece_valid_moves(from[0], from[1], team).include?(to)
						next
					end
					valid = true
				end
				do_move(from[0], from[1], to, team)
			else
				puts "AI MOVE"
			end

			if @stats[TEAM_1]['count'] == 0
				puts "Black wins!"
			elsif @stats[TEAM_2]['count'] == 0
				puts "White wins!"
			end
			team = opposite_team(team)
			turn += 1
		end
	end

end