# To run this script
# open a Command Prompt or PowerShell in "C:\Program Files (x86)\gnubg>"
# and run ".\gnubg-cli.exe --python=D:\MuratMutantCubeGameStages.py"
# or run .\gnubg-cli.exe then type ">"+[ENTER] to get in Python interpreter
# and run "exec(open('D:\Python\MuratMutantCubeGameStages.py').read())"
# replace "D:\" with the path to the script file on your computer

import math
import sys

experiment = 'MuratMutantCubeGameStages' # used in filenames to save games and matches
# mutant doubles, takes and drops according to double/take points
# based on the 5 stages of a game

gamestages = [0, 7, 19, 37, 49]
cubepoints = [0.50, 0.55, 0.60, 0.65, 0.70]
takepoints = [0.0, 0.05, 0.10, 0.15, 0.20]

session = games = game = movecount = snum = gnum  = cycle = power = 0
opoints = xpoints = zpoints = wonroll = autocube = cubeyes = takeyes = 0
cubepoint = takepoint = 0

session = int(input('Enter beginning game: ')) # i.e. 0, 1000, 2000, etc.
games = int(input('Enter games to run: ')) # games in filenames will start with session+1
saveall = input('Save individual games? y/n: ')
matchlog = input('Save matches to logfile? y/n: ')
gameslog = input('Save games to logfile? y/n: ')

if matchlog == 'y': matchfile = open('D:\\' + experiment + 'Session.txt', 'a')
if gameslog == 'y': gamesfile = open('D:\\' + experiment + str(int(session)).zfill(6) + '.txt', 'a')

def getpos():
	global posinfo, dice, player, resigned, doubled, gamestate
	posinfo = gnubg.posinfo()
	dice = posinfo['dice']
	player = posinfo['turn']
	resigned = posinfo['resigned']
	doubled = posinfo['doubled']
	gamestate = posinfo['gamestate'] # 0=no game, 1=playing, 2=over, 3=resigned, 4=dropped

def getcube():
	global cubeinfo, cubeowner, cubevalue, beaver
	cubeinfo = gnubg.cubeinfo()
	cubeowner = cubeinfo['cubeowner'] # 0=bot, 1=mutant, -1=centerd
	cubevalue = cubeinfo['cube']
	beaver = cubeinfo['beavers']

def getwin():
	global cubeless, win
	cubeless = gnubg.evaluate()
	win = cubeless[0] # cubeless win%
#	win = cubeless[5] # cubeless equity

def makemove():
	global board, bestmove, move, movecount
	gnubg.command('roll')
	movecount = movecount + 1 # used in filenames and/or to determine game stages
	posinfo = gnubg.posinfo()
	dice = posinfo['dice'] # to fix gnubg bug, need this again to check if mutant danced
	if dice != (0, 0): # mutant can move, proceed
		bestmove = gnubg.findbestmove()
		board = gnubg.board()
		move = gnubg.movetupletostring(bestmove, board)
		gnubg.command(move)

def errorend(error):
	print(error)
	input('Enter to exit')
	sys.exit()

while game < games: # loop to run number of games entered above
	game = game + 1  
	gnubg.command('new game')
	getpos()
	wonroll = 0
	if dice != (0, 0): # mutant won the opening roll
		wonroll = 1
		movecount = -1 # adjust move count since bot auto-moves after mutant's turn
	else:
		movecount = 0
		wonroll = 0
	gamestate = 1
	cycle = cubevalue = points = winfactor = 0

	while gamestate == 1: # main game loop
		getpos()
		movecount = movecount + 1 # always increment as bot auto-rolls and auto-moves

		if gamestate > 1:
			break

		if resigned > 0:
			gnubg.command('agree')
			movecount = movecount - 1 # don't count them as moves
			break

		getcube()

		if cubevalue == 2048 and beaver == 1: # when bot beavers, cubeinfo is not current
			cubevalue = 4096
			print('beavering cubeowner ', cubeowner)
			input('contine \a') # just to know when this happens

		if cubevalue == 4096:
			cubevalue = 2
			gnubg.command('set cube value 2')
			cycle = cycle + 1
			input('cube cycled')

		if cubeowner == 0 and doubled ==0: # mutant has no cube access, just move
			if beaver > 0:
				input('Not doubled but beavered! \a') # this should never happen
			makemove()
			continue

		for i in reversed(range(5)):
			if movecount >= gamestages[i]:
				cubepoint = cubepoints[i]
				takepoint = takepoints[i]
				print('i', i, 'mcnt', movecount, 'gstg', gamestages[i], 'cpt', cubepoint, 'tpt', takepoint)
#				input('continue')
				break

		getwin()
		cubeyes = takeyes = cubelow = 0
		print('win%', win, 'cpt', cubepoint, 'tpt', takepoint)
#		input('continue')

		if doubled == 1: # reverse GWC
			win = 1 - win

		if cubeowner == 1 and doubled == 1: # can mutants own cube and be doubled?
#			if beaver > 0:
			input('Mutants owns and doubled! \a') # this should never happen

		if win > cubepoint: # mutant's GWC > cubepoint
			cubeyes = 1
		elif win > takepoint: # mutant's GWC > takepoint
			takeyes = 1

		if cubeowner == 1 and beaver == 2: # respond to bot's raccoon
			movecount = movecount -1 # don't count cube actions as moves
			if win == 1: input('win 100\a') # this should never happen
			if win == 0: input('win 0\a') # this should never happen
			if cubeyes: # in case bot beavers with GWC <= mutant's cubepoint
				input('bot racconed low \a') # can this happen?
				gnubg.command('take') # bot racconed low, so take
				continue
			elif takeyes:
				gnubg.command('take') # bot racconed < mutant's takepoint, so take
				continue
			else:
				input('dropping1\a') # just to see if this ever happens
				gnubg.command('drop') # mutant drops with GWC <= takepoint
				break

		if cubeowner == 0 and beaver == 1: # respond to bot's beaver
			movecount = movecount -1 # don't count cube actions as moves
			if cubeyes: input('bot beavered low \a') # can this happen?
			if cubevalue == 4096: input('bot beavered, cube 4096 \a' ) # this should not happen
			if cubevalue == 2048: input('bot beavered, cube 2048 \a') # this should not happen
			gnubg.command('beaver') # mutant always raccoons since it doubled with GWC > cubepoint
			continue

		if cubeowner < 1 and doubled == 1: # respond to bot's double
			movecount = movecount -1 # don't count cube actions as moves
			if cubeyes: # in case bot doubles with GWC <= mutant's cubepoint
				input('bot doubled low \a') # can this happen?
				gnubg.command('beaver') # mutant beavers with GWC > cubepoint
				continue
			elif takeyes:
				gnubg.command('take') # mutant takes with GWC > takepoint
				continue
			else:
				gnubg.command('drop') # mutant drops with GWC <= takepoint
				break

		if cubeyes: # at this point mutant can initiate a cube action
			if movecount > 0: # can double after opening roll
				movecount = movecount -1 # don't count cube actions as moves
				gnubg.command('double') # mutant doubles with GWC > cubepoint
				continue

		makemove()

# game ended, build game filename and save

	getcube()

	gameinfo = gnubg.match(analysis=0, boards=0, statistics=1, verbose=0)['games'][-1]

	gnum = '_G' + str(session + game).zfill(5) # game number

	winner = "_" + gameinfo['info']['winner'] #	O or X won

	points = gameinfo['info']['points-won'] # points won integer

	winfactor = points / cubevalue

	if winfactor == 3:
		wintype = '_W3' # won backgammon
	elif winfactor == 2:
		wintype = '_W2' # won gammon
	elif winfactor == 1:
		wintype = '_W1' # won normal
	else: errorend('invalid multiplier') # this should never happen

	if cycle > 0 : # calculate effective cube value
		power = math.log(cubevalue, 2) + (cycle * 11) # 2^12 = 4096
		cubevalue = int(math.pow(2, power))

	cval = '_C' + str(cubevalue).zfill(16) # cube value formatted

	points = int(cubevalue * winfactor) # effective points won

	if winner == '_O': # to fix gnubg bug, players reversed
		winner = '_X'
		xpoints = xpoints + points
	elif winner == '_X':
		winner = '_O'
		opoints = opoints + points
	else: errorend('invalid winner') # this should never happen

	points = str(points).zfill(16) # points won formatted
	autocube = autocube + wonroll

	if gameinfo['game'][-1]['action'] == 'resign':
		endtype = '_R' # resigned
	elif gameinfo['game'][-1]['action'] == 'drop':
		endtype = '_D' # dropped
	elif gameinfo['game'][-1]['action'] == 'move':
		endtype = '_F' # finished
	else: errorend('invalid how game ended') # this should never happen

	lastacted = gameinfo['game'][-1]['player'] # player who made last action

	if lastacted == 'O': # fix gnubg bug, players reversed
		lastacted = 'X'
	elif lastacted == 'X':
		lastacted = 'O'
	else: errorend('invalid last player') # this should never happen

	if lastacted == 'X':
		movecount = movecount - 1 # it wasn't subtracted for mutant in game loop, do it now

	mcnt = '_M' + str(movecount).zfill(3) # number of moves in game for both players
	if wonroll == 1:
		mcnt = mcnt + '_W' # flag games when mutant won opening roll

	cubeskill = round(gameinfo['stats']['O']['cube']['error-skill'], 4) # cube error for mutant
	cubeskill = '_E' + str(int(cubeskill * 1000)).zfill(6) # formatted without decimal

	filename = experiment + gnum + winner + points + cubeskill + cval + wintype + endtype + lastacted + mcnt
	if gameslog == 'y': gamesfile.write(filename + '\n')
	filename = 'save game D:\\' + filename + '.sgf'
	if saveall == 'y': gnubg.command(filename)

# session ended, build match filename and save

snum = '_S' + str(session).zfill(5) # session begin number + games in session
opoints = '_O' + str(opoints).zfill(16) # O's total points won
xpoints = '_X' + str(xpoints).zfill(16) # X's total points won
autocube = '_A' + str(autocube).zfill(3) # count for mutant's opening roll wins (no cube)

gameinfo = gnubg.match(analysis=0, boards=0, statistics=1, verbose=0)['stats']
cubecount = gameinfo['O']['cube']['total-cube'] # total cube actions for mutant
cubecount = '_CC' + str(cubecount).zfill(6) # formatted without decimal
cubeskill = round(gameinfo['O']['cube']['error-skill'], 4) # total cube error for mutant
cubeskill = '_ES' + str(int(cubeskill * 10000)).zfill(8) # formatted without decimal
cubecost = round(gameinfo['O']['cube']['error-cost'], 2) # total cube error for mutant
cubecost = '_EC' + str(int(cubecost * 100)).zfill(8) # formatted without decimal

filename =  experiment + snum + gnum + opoints + xpoints + cubecount + cubeskill + cubecost + autocube
if matchlog == 'y': matchfile.write(filename + '\n')
filename = 'save match D:\\' + filename + '.sgf'
gnubg.command(filename)

gamesfile.close()
matchfile.close()
print('\a') # sounds bell to alert that it's done

# all prefixes starting with an "_" are unique single characters
# to use in wild-card filename searches within a directory
# in order to select certain games, i.e. *_W3* to select games ended in a backgammon
# prefixes used: _G, _O, _X, _C, _W, _R, _D, _F, _M, _E, _S, _CC, _ES, _EC _A, _W