耿直版黑白棋pygame

JAY饭

        本来想写一写比较复杂的黑白棋AI，但是坑太大，暂时不去想了，就用pygame做了一个极度耿直和简陋版的人机黑白棋。
先凑合看，素材的几张图片都在下面，大家可以拿去试着玩。棋力极低，而且因为字体的缘故就没写得分和输赢。
抛砖引玉一下。另求黑白棋的AI设计策略。

1. import random
   
2. import pygame
   
3. from pygame.locals import *
   
4. from sys import exit
   
5. import time
   
6. 
   
7. 
   
8. class Point():
   
9. 
   
10.     def __init__(self,lst,a,b):
    
11.         self.ite = lst
    
12.         self.x ,self.y = a,b
    
13.         self.step = [(1,0),(0,1),(-1,0),(0,-1),(1,1),(-1,1),(1,-1),(-1,-1)]
    
14.         self.dic,self.dic_true = {1:[],2:[]},{1:[],2:[]}
    
15.         self.white,self.black = 0,0
    
16.         
    
17.     def point(self):
    
18.         self.dic,self.dic_true = {1:[],2:[]},{1:[],2:[]}
    
19.         for i,j in self.step:
    
20.             a,b = self.x + i,self.y+j
    
21.             if 0 <= a < 8 and 0 <= b <8:
    
22.                  if self.ite[a][b] != 0:
    
23.                     sign = self.ite[a][b]
    
24.                     self.dic[sign].append([a,b])
    
25.                     self.ext(sign,a,b,i,j)
    
26.                     
    
27.     def ext(self,sign,a,b,i,j):
    
28.         while True:
    
29.             a += i
    
30.             b += j
    
31.             if 0<=a<8 and 0<=b<8 and self.ite[a][b]!=0:
    
32.                 if self.ite[a][b] == sign:
    
33.                     self.dic[sign].append([a,b])
    
34.                 else:
    
35.                     if sign == 1: self.dic_true[2].append(tuple(self.dic[1]))
    
36.                     else:
    
37.                         self.dic_true[1].append(tuple(self.dic[2]))
    
38.             else: break
    
39.         self.dic[sign] = []
    
40. 
    
41.     def count(self):
    
42.         if len(self.dic_true[1]):
    
43.             self.white = 1
    
44.         if len(self.dic_true[2]):
    
45.             self.black = 1
    
46.         return self.white,self.black
    
47. 
    
48. class Reversi():
    
49. 
    
50.     def __init__(self):
    
51.         self.lst = [[0]*8 for i in range(8)]
    
52.         self.lst[3][3],self.lst[3][4] = 1,2
    
53.         self.lst[4][3],self.lst[4][4] = 2,1
    
54.         self.Man,self.AI = 0,0
    
55. 
    
56.     def possiblecoo(self):
    
57.         self.Man,self.AI = 0,0
    
58.         for i in range(8):
    
59.             for j in range(8):
    
60.                 if self.lst[i][j] == 0:
    
61.                     if self.cohesion(i,j):
    
62.                         P = Point(self.lst,i,j)
    
63.                         P.point()
    
64.                         P.count()
    
65.                         if P.white:
    
66.                             self.Man += 1
    
67.                         if P.black:
    
68.                             self.AI += 1
    
69.         return self.AI-self.Man
    
70.                            
    
71.     def cohesion(self,poi_x,poi_y):
    
72.         around =[(1,0),(0,1),(-1,0),(0,-1),(1,1),(-1,1),(1,-1),(-1,-1)]
    
73.         for a,b in around:
    
74.             a1,b1 = poi_x+a,poi_y+b
    
75.             if 0 <= a1 < 8 and 0 <= b1 <8 and self.lst[a1][b1] != 0:
    
76.                 return True
    
77. 
    
78.     def man_play(self,a,b):
    
79.         P = Point(self.lst,a,b)
    
80.         P.point()
    
81.         if len(P.dic_true[1]):
    
82.             self.lst[a][b] = 1
    
83.             for i in P.dic_true[1]:
    
84.                 for [j,k] in i:
    
85.                     self.lst[j][k] = 1                    
    
86.             return True
    
87.         else:
    
88.             return None
    
89.         
    
90.     def check(self):
    
91.         pos = []
    
92.         for i in range(8):
    
93.             for j in range(8):
    
94.                 if self.lst[i][j] == 0:
    
95.                     if self.cohesion(i,j):
    
96.                         P = Point(self.lst,i,j)
    
97.                         P.point()
    
98.                         if len(P.dic_true[2]):
    
99.                              pos.append([i,j,P.dic_true[2]])
    
100.         return pos
     
101.         
     
102.     def AI_play_check(self):
     
103.         right = {}
     
104.         pos = self.check()
     
105.         lst_x = [tuple(i) for i in self.lst]
     
106.         for [x,y,p] in pos:
     
107.             self.lst[x][y] = 2
     
108.             for i in p:
     
109.                 for [a,b] in i:
     
110.                     self.lst[a][b] = 2
     
111.             temp_comp = self.possiblecoo()
     
112.             right[temp_comp] = [x,y,p]
     
113.             self.lst = [list(i) for i in lst_x]
     
114.         temp = right[max(right)]
     
115.         self.lst[temp[0]][temp[1]] = 2
     
116.         for i in temp[2]:
     
117.             for [a,b] in i:
     
118.                 self.lst[a][b] = 2
     
119. 
     
120. pygame.init()
     
121. pygame.font.init()
     
122. bg = 255,255,255
     
123. screen = pygame.display.set_mode((800,600),RESIZABLE)
     
124. pygame.display.set_caption('黑白棋')
     
125. 
     
126. imgbackground = pygame.image.load('渔舟唱晚.jpg')
     
127. imgchessboard = pygame.image.load('棋盘.jpg').convert()
     
128. imgblack = pygame.image.load('黑.png').convert_alpha()
     
129. imgwhite = pygame.image.load('白.png').convert_alpha()
     
130.    
     
131. chessboard_pos = imgchessboard.get_rect()
     
132. chessboard_pos[0],chessboard_pos[1] = 260,160
     
133. whiteamount = []
     
134. blackamount = []
     
135. 
     
136. R = Reversi()
     
137. 
     
138. def sure_point(x,y):
     
139.     unit = 35
     
140.     i = int((x-260)/unit)
     
141.     j = int((y-160)/unit)
     
142.     pos_x,pos_y = 260+int(i*unit),160+int(j*unit)
     
143.     return pos_x,pos_y
     
144. 
     
145. def sure_lst(i,j):
     
146.     unit = 35
     
147.     return [260+int(i*unit),160+int(j*unit)]
     
148. 
     
149. def lst_point(a,b):
     
150.     unit = 35
     
151.     i,j = int((a-260)/unit),int((b-160)/unit)
     
152.     return i,j
     
153. 
     
154. def get_chess():
     
155.     for event in pygame.event.get():
     
156.         if event.type == QUIT:
     
157.             pygame.quit()
     
158.             exit()
     
159.         if event.type == MOUSEBUTTONDOWN:
     
160.             if event.button == 1:
     
161.                 chess_x,chess_y = pygame.mouse.get_pos()
     
162.                 if 260<=chess_x<=540 and 160<=chess_y<=440 and [chess_x,chess_y] not in whiteamount:
     
163.                     a,b = sure_point(chess_x,chess_y)
     
164.                     i,j = lst_point(a,b)
     
165.                     if R.lst[i][j] == 0:
     
166.                         return a,b,i,j
     
167. 
     
168.         
     
169. def show():
     
170.     global whiteamount,blackamount
     
171.     T = False
     
172.     while True:
     
173.         temp = get_chess()
     
174.         if temp:
     
175.             i,j = temp[2],temp[3]
     
176.             if R.man_play(i,j):
     
177.                 T = True   
     
178.         whiteamount,blackamount = [],[]
     
179.         for i in range(8):
     
180.             for j in range(8):
     
181.                 if R.lst[i][j] == 1:
     
182.                     whiteamount.append(sure_lst(i,j))        
     
183.                 elif R.lst[i][j] == 2:
     
184.                     blackamount.append(sure_lst(i,j))
     
185.         
     
186.         screen.fill(bg)
     
187.         screen.blit(imgbackground,(0,0))
     
188.         screen.blit(imgchessboard,chessboard_pos)
     
189.         for i in whiteamount:
     
190.             img_w = imgwhite
     
191.             img_w_pos = img_w.get_rect()
     
192.             img_w_pos[0],img_w_pos[1] = i[0],i[1]
     
193.             screen.blit(img_w,img_w_pos)
     
194.         for i in blackamount:
     
195.             img_b = imgblack
     
196.             img_b_pos = img_b.get_rect()
     
197.             img_b_pos[0],img_b_pos[1] = i[0],i[1]
     
198.             screen.blit(img_b,img_b_pos)
     
199.         pygame.display.flip()
     
200.         if T:
     
201.             time.sleep(0.5)
     
202.             R.AI_play_check()
     
203.             T = False
     
204. show()
     
205. 
     

复制代码

JAY饭

然后大家自己写进上面代码的图片名位置里面
效果图：  

素材图~~~~~~~~~~~~~~~~~~~~

JAY饭

明天起来写注释