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

import math
import sys

experiment = 'MuratMutantChecker1stWorst' # used in filenames to save games and matches
# mutant always makes the worst move

session = games = game = movecount = snum = gnum  = cycle = power = seed = 0
opoints = xpoints = zpoints = wonroll = autocube = cubeyes = takeyes = beaveryes = 0
cubepoint = takepoint = beaverpoint = 0
cubeaction = ['double', 'double', '', '', '', 'double', '', 'redouble', 'redouble', '', '', '', '', '', '', '', '', '', '', '', '']
cubereply = ['take', 'pass', 'take', 'take', 'pass', 'beaver', 'beaver', 'take', 'pass', 'take', 'take', 'pass', 'beaver', 'take', 'take', 'beaver', 'take', 'take', 'pass', 'pass', '']

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: ')
#seed = int(input('Enter RNG seed: '))
seed = 340000 + session
if seed > 0 : gnubg.command('set seed ' + str(seed))

games = 1000
matchlog = gameslog = 'y'

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 geteval() :
	global evalinfo, cubeint
	evalinfo = gnubg.cfevaluate()
	cubeint = evalinfo[4]

def makemove():
	global board, allmoves, pickmove, convmove, 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
		allmoves = gnubg.hint()
		pickmove = allmoves['hint'][-1]['move']
		board = gnubg.board()
		convmove = gnubg.parsemove(pickmove)
		move = gnubg.movetupletostring(convmove, 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
	gnubg.command('new game')
	game = game + 1  
	getpos()
	if dice != (0, 0): # mutant won the opening roll
		movecount = -1 # adjust move count since bot auto-moves after mutant's turn
	else: movecount = 0
	gamestate = 1
	cycle = cubevalue = points = winfactor = 0

	makemove()
	gnubg.command('end game')

# 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 :
		power = math.log(cubevalue, 2) + (cycle * 11)
		cubevalue = int(math.pow(2, power))

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

	points = int(cubevalue * winfactor)

	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 within the game loop

	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 = '_ES' + str(int(cubeskill * 1000)).zfill(6) # formatted without decimal
	cubecost = round(gameinfo['stats']['O']['cube']['error-cost'], 4) # cube error for mutant
	cubecost = '_EC' + str(int(cubecost * 1000)).zfill(6) # formatted without decimal

	filename = experiment + gnum + winner + points + cubeskill + cubecost + 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(8) # O's total points won
xpoints = '_X' + str(xpoints).zfill(8) # 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'], 4) # total cube error for mutant
cubecost = '_EC' + str(int(cubecost * 10000)).zfill(8) # formatted without decimal
diceseed = '_DS' + str(int(seed)).zfill(8)

filename =  experiment + snum + gnum + opoints + xpoints + cubecount + cubeskill + cubecost + diceseed + 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, _DS, _F, _M, _S, _CC, _ES, _EC _A, _W