同樣需要安裝pygame
# 導(dǎo)入pygame模塊
import pygame
# 定義三個(gè)常量函數(shù),用來(lái)表示白棋,黑棋盾沫,以及 空
EMPTY = 0
BLACK = 1
WHITE = 2
# 定義黑色(黑棋用虹蒋,畫(huà)棋盤(pán))
black_color = [0, 0, 0]
# 定義白色(白棋用)
white_color = [255, 255, 255]
# 定義棋盤(pán)這個(gè)類(lèi)
class RenjuBoard(object):
def __init__(self):
# self._board = board = [[EMPTY] * 15 for _ in range(15)]
# 將棋盤(pán)每一個(gè)交叉點(diǎn)都看作列表的一個(gè)元素位爽冕,一共有15*15共225個(gè)元素
self._board = [[]] * 15
self.reset()
# 重置棋盤(pán)
def reset(self):
for row in range(len(self._board)):
self._board[row] = [EMPTY] * 15
# 定義棋盤(pán)上的下棋函數(shù)谐岁,row表示行骡湖,col表示列充石,
# is_black表示判斷當(dāng)前點(diǎn)位該下黑棋莫换,還是白棋
def move(self, row, col, is_black):
if self._board[row][col] == EMPTY:
self._board[row][col] = BLACK if is_black else WHITE
return True
return False
# 給棋盤(pán)定義一個(gè)函數(shù)將自己在screen上面畫(huà)出來(lái),
# 使用pygame.draw()函數(shù)赫冬。并且順便將下了的棋子也畫(huà)出來(lái)
def draw(self, screen):
for h in range(1, 16):
pygame.draw.line(screen, black_color,
[40, h * 40], [600, h * 40], 1)
pygame.draw.line(screen, black_color, [40, h * 40], [600, h * 40], 1)
# 給棋盤(pán)加一個(gè)外框浓镜,使美觀
pygame.draw.rect(screen, black_color, [36, 36, 568, 568], 3)
# 在棋盤(pán)上標(biāo)出,天元以及另外4個(gè)特殊點(diǎn)位
pygame.draw.circle(screen, black_color, [320, 320], 5, 0)
pygame.draw.circle(screen, black_color, [160, 160], 3, 0)
pygame.draw.circle(screen, black_color, [160, 480], 3, 0)
pygame.draw.circle(screen, black_color, [480, 160], 3, 0)
pygame.draw.circle(screen, black_color, [480, 480], 3, 0)
# 做2次for循環(huán)取得棋盤(pán)上所有交叉點(diǎn)的坐標(biāo)
for row in range(len(self._board)):
for col in range(len(self._board[row])):
# 將下在棋盤(pán)上的棋子畫(huà)出來(lái)
if self._board[row][col] != EMPTY:
ccolor = black_color \
if self._board[row][col] == BLACK else white_color
# 取得這個(gè)交叉點(diǎn)下的棋子的顏色劲厌,并將棋子畫(huà)出來(lái)
pos = [40 * (col + 1), 40 * (row + 1)]
# 畫(huà)出棋子
pygame.draw.circle(screen, ccolor, pos, 18, 0)
# 定義函數(shù)膛薛,傳入當(dāng)前棋盤(pán)上的棋子列表,
# 輸出結(jié)果补鼻,不管黑棋白棋勝哄啄,都是傳回False,未出結(jié)果則為T(mén)rue
def is_win(board):
for n in range(15):
# 判斷垂直方向勝利
flag = 0
# flag是一個(gè)標(biāo)簽风范,表示是否有連續(xù)五個(gè)以上相同顏色的棋子
for b in board._board:
if b[n] == 1:
flag += 1
if flag == 5:
print('黑棋勝')
return False
else:
# else表示此時(shí)沒(méi)有連續(xù)相同的棋子咨跌,標(biāo)簽flag重置為0
flag = 0
flag = 0
for b in board._board:
if b[n] == 2:
flag += 1
if flag == 5:
print('白棋勝')
return False
else:
flag = 0
# 判斷水平方向勝利
flag = 0
for b in board._board[n]:
if b == 1:
flag += 1
if flag == 5:
print('黑棋勝')
return False
else:
flag = 0
flag = 0
for b in board._board[n]:
if b == 2:
flag += 1
if flag == 5:
print('白棋勝')
return False
else:
flag = 0
# 判斷正斜方向勝利
for x in range(4, 25):
flag = 0
for i, b in enumerate(board._board):
if 14 >= x - i >= 0 and b[x - i] == 1:
flag += 1
if flag == 5:
print('黑棋勝')
return False
else:
flag = 0
for x in range(4, 25):
flag = 0
for i, b in enumerate(board._board):
if 14 >= x - i >= 0 and b[x - i] == 2:
flag += 1
if flag == 5:
print('白棋勝')
return False
else:
flag = 0
# 判斷反斜方向勝利
for x in range(11, -11, -1):
flag = 0
for i, b in enumerate(board._board):
if 0 <= x + i <= 14 and b[x + i] == 1:
flag += 1
if flag == 5:
print('黑棋勝')
return False
else:
flag = 0
for x in range(11, -11, -1):
flag = 0
for i, b in enumerate(board._board):
if 0 <= x + i <= 14 and b[x + i] == 2:
flag += 1
if flag == 5:
print('白棋勝')
return False
else:
flag = 0
return True
def main():
# 創(chuàng)建棋盤(pán)對(duì)象
board = RenjuBoard()
# 用于判斷是下黑棋還是白棋
is_black = True
# pygame初始化函數(shù),固定寫(xiě)法
pygame.init()
pygame.display.set_caption('五子棋') # 改標(biāo)題
# pygame.display.set_mode()表示建立個(gè)窗口硼婿,
# 左上角為坐標(biāo)原點(diǎn)锌半,往右為x正向,往下為y軸正向
screen = pygame.display.set_mode((640, 640))
# 給窗口填充顏色寇漫,顏色用三原色數(shù)字列表表示
screen.fill([125, 95, 24])
board.draw(screen) # 給棋盤(pán)類(lèi)發(fā)命令刊殉,
# 調(diào)用draw()函數(shù)將棋盤(pán)畫(huà)出來(lái)
pygame.display.flip() # 刷新窗口顯示
running = True
# while 主循環(huán)的標(biāo)簽殉摔,以便跳出循環(huán)
while running:
# 遍歷建立窗口后發(fā)生的所有事件,固定寫(xiě)法
for event in pygame.event.get():
# 根據(jù)事件的類(lèi)型记焊,進(jìn)行判斷
if event.type == pygame.QUIT:
running = False
elif event.type == pygame.KEYUP:
pass
# pygame.MOUSEBUTTONDOWN表示鼠標(biāo)的鍵被按下
elif event.type == pygame.MOUSEBUTTONDOWN and \
event.button == 1: # button表示鼠標(biāo)左鍵
x, y = event.pos # 拿到鼠標(biāo)當(dāng)前在窗口上的位置坐標(biāo)
# 將鼠標(biāo)的(x, y)窗口坐標(biāo)逸月,轉(zhuǎn)換為棋盤(pán)上的坐標(biāo)
row = round((y - 40) / 40)
col = round((x - 40) / 40)
if board.move(row, col, is_black):
is_black = not is_black
screen.fill([125, 95, 24])
board.draw(screen)
pygame.display.flip()
# 調(diào)用判斷勝負(fù)函數(shù)
if not is_win(board):
# break
running = False
pygame.quit()
if __name__ == '__main__':
main()