reversi.py

   1 '''
   2 Created on 2015/08/25
   3
   4 @author: fuke
   5 '''
   6 import pygame,os,sys
   7 from pygame.locals import *
   8 from network import Comp
   9
  10
  11 pygame.font.init()
  12 comp = Comp()
  13 screen = pygame.display.set_mode((400,400))
  14
  15 none, black, white = 0,1,2
  16
  17 def Path():
  18     for name in sys.argv:
  19         if os.path.exists(name) == True:
  20             return os.path.dirname(name)
  21
  22 img = pygame.image.load(os.path.join(Path(),'png','tips.png')).convert()
  23 bkblack = pygame.image.load(os.path.join(Path(),'png','min_black.png')).convert()
  24 bkwhite = pygame.image.load(os.path.join(Path(),'png','min_white.png')).convert()
  25
  26 index_x = 6
  27 index_y = 5
  28 size = 50
  29
  30 class Player():
  31     auto = False
  32     stone = none
  33
  34 class Grid():
  35     def __init__(self):
  36         self.grid = [[none for y in range(8)] for x in range(8)]
  37         self.stone = none
  38
  39     def Assign(self,item):
  40         for x in range(8):
  41             for y in range(8):
  42                 self.grid[x][y] = item.grid[x][y]
  43
  44 class StoneGrid():
  45     def __init__(self):
  46         self.item = Grid()
  47         self.buffer = Grid()
  48         self.arr = Grid()
  49         self.turn_number = 0
  50         self.turn_index = 0
  51         self.active = True
  52         self.gameover = False
  53         self.text = pygame.font.SysFont(pygame.font.get_fonts()[0], 25, True).render('reversi',False,(0,0,255)).convert()
  54
  55     def Clear(self):
  56         for x in range(8):
  57             for y in range(8):
  58                 self.item.grid[x][y] = none
  59         self.item.grid[3][3] = black
  60         self.item.grid[4][4] = black
  61         self.item.grid[4][3] = white
  62         self.item.grid[3][4] = white
  63         self.turn_index = 0
  64         self.turn_number = 0
  65
  66     def CalScore(self,stone,x,y):
  67         self.buffer.Assign(self.item)
  68         if self.CanSetStone(stone, x, y, True) == True:
  69             self.score = 0
  70             for i in range(8):
  71                 for j in range(8):
  72                     if self.CanSetStone(stone, i, j, False) == True:
  73                         self.score += 1
  74             self.item.Assign(self.buffer)
  75         else:
  76             self.score = -1
  77
  78     def CanSetStone(self,stone,x,y,reverse):
  79         p = [True,False]
  80         def Method(m,n):
  81             if p[0] == False:
  82                 return
  83             i = 1
  84             while True:
  85                 d = x+m*i
  86                 e = y+n*i
  87                 if not ((0 <= d)and(d < 8)and(0 <= e)and(e < 8)):
  88                     break
  89                 s = self.item.grid[d][e]
  90                 if s == none:
  91                     break
  92                 elif s == stone:
  93                     if i > 1:
  94                         if reverse == True:
  95                             if p[1] == False:
  96                                 self.item.grid[x][y] = stone
  97                                 p[1] = True
  98                             j = 1
  99                             while j <= i-1:
 100                                 self.item.grid[x+m*j][y+n*j] = stone
 101                                 j += 1
 102                         else:
 103                             p[0],p[1] = False,True
 104                     break
 105                 else:
 106                     i += 1
 107
 108         if self.item.grid[x][y] == none:
 109             Method(-1,-1)
 110             Method(-1, 0)
 111             Method(-1, 1)
 112             Method(0, -1)
 113             Method(0, 1)
 114             Method(1, -1)
 115             Method(1, 0)
 116             Method(1, 1)
 117         return p[1]
 118
 119     def NextStone(self,stone,pos):
 120         result = False
 121         n = 0
 122         self.score = 0
 123         for i in range(8):
 124             for j in range(8):
 125                 self.CalScore(stone, i, j)
 126                 if self.score != -1:
 127                     if result == False:
 128                         result = True
 129                     self.arr.grid[i][j] = self.score
 130                     if self.score > n:
 131                         n = self.score
 132                     pos[0],pos[1] = i,j
 133                 else:
 134                     self.arr.grid[i][j] = -1
 135         if n > 0:
 136             for i in range(8):
 137                 for j in range(8):
 138                     k = self.arr.grid[i][j]
 139                     if k != -1:
 140                         self.arr.grid[i][j] = k / n
 141         return result
 142
 143     def Start(self):
 144         global index
 145         index = player1
 146         self.Clear()
 147         self.active = True
 148         self.gameover = False
 149
 150     def ReStart(self):
 151         self.active = True
 152         self.gameover = False
 153         self.turn_index = self.turn_number
 154
 155     def Pause(self):
 156         self.active = False
 157
 158     def Paint(self):
 159         if self.effect_stone == black:
 160             s = bkwhite
 161         else:
 162             s = bkblack
 163         i = 0
 164         while i < len(self.list):
 165             p = self.list[i]
 166             screen.blit(s,(p.Left*50,p.Top*50),(p.X*50,p.Y*50,50,50))
 167             i += 1
 168
 169 def Paint():
 170     for x in range(8):
 171         for y in range(8):
 172             s = stone_grid.item.grid[x][y]
 173             r = pygame.Rect(x*size,y*size,(x+1)*size,(y+1)*size)
 174             if s == white:
 175                 screen.blit(img,r,pygame.Rect(2*size,0,3*size,size))
 176             elif s == black:
 177                 screen.blit(img,r,pygame.Rect(size,0,2*size,size))
 178             else:
 179                 screen.blit(img,r,pygame.Rect(0,0,size,size))
 180     if stone_grid.gameover == True:
 181         screen.blit(stone_grid.text,pygame.Rect(0,0,800,50))
 182     pygame.display.update()
 183
 184 def ChangePlayer():
 185     def Main():
 186         global index
 187         Paint()
 188         if index == player1:
 189             index = player2
 190             return 'white'
 191         else:
 192             index = player1
 193             return 'black'
 194
 195     def Execute():
 196         for x in range(8):
 197             for y in range(8):
 198                 if stone_grid.CanSetStone(index.stone, x, y, False) == True:
 199                     return True
 200         return False
 201
 202     s = Main()+str(stone_grid.turn_index+1)
 203     if Execute() == False:
 204         s = Main()+str(stone_grid.turn_index+1)
 205         if Execute() == False:
 206             s = 'game over'
 207             i = 0
 208             j = 0
 209             for x in range(8):
 210                 for y in range(8):
 211                     if stone_grid.item.grid[x][y] == black:
 212                         i += 1
 213                     elif stone_grid.item.grid[x][y] == white:
 214                         j += 1
 215             pygame.display.set_caption(s+str(stone_grid.turn_index+1))
 216             if i > j:
 217                 s = 'Player1 Win!'
 218             elif i < j:
 219                 s = 'Player2 Win!'
 220             else:
 221                 s = 'Draw'
 222             stone_grid.text = pygame.font.SysFont(pygame.font.get_fonts()[0],25,True).render(s+'(Player1){0}(Player2){1}'.format(i,j),False,(0,0,255)).convert()
 223             stone_grid.gameover = True
 224             Paint()
 225         else:
 226             pygame.display.set_caption(s)
 227             stone_grid.turn_index += 1
 228     else:
 229         pygame.display.set_caption(s)
 230         stone_grid.turn_index += 1
 231
 232 def CompStone():
 233     pos = [0,0]
 234     stone_grid.active = False
 235     if stone_grid.NextStone(index.stone, pos) == True:
 236         if index.stone == black:
 237             pre = comp.sente_stone(stone_grid.item.grid,stone_grid.arr.grid)
 238         elif index.stone == white:
 239             pre = comp.gote_stone(stone_grid.item.grid,stone_grid.arr.grid)
 240         stone_grid.CanSetStone(index.stone, pre[0], pre[1], True)
 241         print(pre)
 242     stone_grid.active = True
 243
 244 player1 = Player()
 245 player2 = Player()
 246 index = player1
 247 player1.stone = black
 248 player1.auto = False
 249 player2.auto = True
 250 player2.stone = white
 251 stone_grid = StoneGrid()
 252 stone_grid.Start()
 253 pygame.event.get()
 254 temp = pygame.time.get_ticks()
 255 Paint()
 256 tt = (player1.auto == True)and(player2.auto == True)
 257 if tt == True:
 258     k = 10
 259 else:
 260     k = 150
 261 while True:
 262     if pygame.time.get_ticks()-temp > 2*k:
 263         if (stone_grid.active == True)and(index.auto == True):
 264             CompStone()
 265             ChangePlayer()
 266         temp = pygame.time.get_ticks()
 267     pygame.time.wait(k)
 268     for x in pygame.event.get():
 269         if x.type == QUIT:
 270             sys.exit()
 271     t = pygame.mouse.get_pressed()[0]
 272     if (stone_grid.gameover == True)and((t == True)or(tt == True)):
 273         stone_grid.Start()
 274     if (index.auto == False)and(stone_grid.active == True)and(t == True):
 275         stone_grid.active = False
 276         s = pygame.mouse.get_pos()
 277         if stone_grid.CanSetStone(index.stone,s[0]//size,s[1]//size,True) == True:
 278             stone_grid.turn_index += 1
 279             ChangePlayer()
 280         stone_grid.active = True
 281