#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author:Hiuhung Wanimport turtle
from enum import Enumclass PotColor(Enum):Black = 1White = 2import time
class WuziBoard(object):def __init__(self, RowNum):turtle.speed(9)turtle.hideturtle()self.RowNum = RowNumself.halfDim = 500 /(RowNum-1)/2.0passdef drawBoard(self, ActionHis =None):turtle.screensize(400, 400, "white")turtle.title('五子棋')turtle.home()turtle.speed(0)time.sleep(5)for i in range( self.RowNum ):x = 0 - 250 + i * ( self.halfDim ) *2y = 0 -250turtle.penup()turtle.setpos(x, y)turtle.pendown()turtle.goto(x, y + 500)for i in range( self.RowNum ):x = 0 - 250y = 0 -250 + i * ( self.halfDim ) *2turtle.penup()turtle.setpos(x, y)turtle.pendown()turtle.setpos(x+500, y)if( ActionHis != None):self.drawNow( ActionHis )turtle.done()passdef action2potxy(self, action):x = 0 - 250 + action[0]*self.halfDim*2y = 0 - 250 + action[1]*self.halfDim*2return x,ydef drawNow(self, RunAction ):for potsite in RunAction:x,y = self.action2potxy( ( potsite[0], potsite[1] ) )turtle.penup()turtle.setpos(x, y)turtle.pendown()if( potsite[2] != PotColor.Black ) :turtle.dot(10,"Red")else:turtle.dot(10, "Black")if( potsite ==  RunAction[len(RunAction) - 1 ] ):if (potsite[2] != PotColor.Black):turtle.dot(20, "Red")else:turtle.dot(20, "Black")passdef drawAction(self):passpassdef main():ActionHis =[ (0,1,1), (1,1,2), (5,1,1),  ]wuziBoard = WuziBoard( 6 )wuziBoard.drawBoard( ActionHis )passif __name__ == "__main__":#test()main()

class GameFivePot(object):def __init__(self):self.potCount  =0;self.AllAction    =[]self.ActionHis = []for x in range(ROW_NUM):for y in range(ECO_NUM):self.AllAction     += [(x,y)]self.AvailAction = self.AllActionself.RunAction = [[0 for col in range(ROW_NUM)] for row in range(ECO_NUM)]def getActions(self):return self.AvailActiondef getRunAction(self):      return self.RunActiondef getActionHis(self):return  self.ActionHisdef is_over(self, action, potColor):x = action[0]y = action[1]dimCount =[1,1,1,1]#���� xiang qianfor x1 in range(x+1, x+5):if(x1 >= ROW_NUM ):breakif( self.RunAction[x1][y] == potColor ):dimCount[0] +=1 else:break#- xiang houfor x1 in range(x-1, x-5, -1 ):if(x1 < 0 ):breakif( self.RunAction[x1][y] == potColor ):dimCount[0] +=1 else:breakif( dimCount[0] >= 5 ):return True,True#���� ����for y1 in range(y+1, y+5):if(y1 >= ROW_NUM ):breakif( self.RunAction[x][y1] == potColor ):dimCount[1] +=1 else:break#- ����for y1 in range(y-1, y-5, -1 ):if(y1 < 0 ):breakif( self.RunAction[x][y1] == potColor ):dimCount[1] +=1 else:breakif( dimCount[1] >= 5 ):return True,True#-��б ����for offset in range(1 ,5):x1 = x+offsety1 = y+offsetif(y1 >= ROW_NUM or x1 >= ROW_NUM  ):breakif( self.RunAction[x1][y1] == potColor ):dimCount[2] +=1 else:break#- ����for offset in range(-1, -5, -1 ):x1 = x+offsety1 = y+offset            if(y1 < 0 or x1<0):breakif( self.RunAction[x1][y1] == potColor ):dimCount[2] +=1 else:breakif( dimCount[2] >= 5 ):return True,True#-��б ���� for offset in range(1 ,5):x1 = x+offsety1 = y-offsetif(y1 < 0 or x1 >= ROW_NUM  ):breakif( self.RunAction[x1][y1] == potColor ):dimCount[3] +=1 else:break#- ���� for offset in range(-1, -5, -1 ):x1 = x+offsety1 = y-offset            if(y1 >= ROW_NUM  or x1<0 ):breakif( self.RunAction[x1][y1] == potColor ):dimCount[3] +=1 else:breakif( dimCount[3] >= 5 ):return True,Trueif( len(self.AvailAction) == 0 ):return True,Falsereturn False,Falsepass   def action( self, action,potColor ):self.potCount +=1self.ActionHis += [  ( action[0], action[1], potColor )  ]self.AllAction.remove(  action  )self.RunAction[ action[0] ][  action[1] ] =potColorisOver, isWin = self.is_over(action, potColor)return self.RunAction,  isOver, isWindef __repr__(self):return "Game step count: {}, AvailAction len: {},  ".format( self.potCount,     len(self.AvailAction) )

class GamePlayer(object):def __init__(self, potColor ):self.actionHis = []self.color = potColordef getActionHis(self):return self.actionHisdef play(self, game):actions = game.getActions()action = self.choiceActions( actions )self.actionHis = self.actionHis +[action]gameInfo , isOver, isWin = game.action(action , self.color )return gameInfo , isOver, isWinpassdef choiceActions( self, actions ):action = random.choice( actions  )return actiondef __repr__(self):return "color: {}, actionHis: {},  ".format(    self.color, self.actionHis )