|
|
import pygame as pg
|
|
|
|
|
|
#设置棋盘上的棋子
|
|
|
def set_chess(x, y, color):
|
|
|
if board[x][y] != ' ':
|
|
|
print('该位置已有棋子') #有了
|
|
|
return False
|
|
|
else:#没有棋子
|
|
|
board[x][y] = color
|
|
|
return True
|
|
|
|
|
|
#检查是否获胜
|
|
|
def check_win(board):
|
|
|
for list_str in board: #遍历棋盘的每一行
|
|
|
if ''.join(list_str).find('O'*5) != -1: #如果找到连续的五个'O'，表示白棋获胜，并返回一个0表示白棋获胜
|
|
|
print('白棋获胜')
|
|
|
return 0
|
|
|
elif ''.join(list_str).find('X'*5) != -1:
|
|
|
print('黑棋获胜')
|
|
|
return 1
|
|
|
else:
|
|
|
return -1 #都没有获胜
|
|
|
|
|
|
#检查所有方向的获胜情况
|
|
|
def check_win_all(board):
|
|
|
board_c = [[] for line in range(29)] #水平方向的棋盘副本
|
|
|
for x in range(15):
|
|
|
for y in range(15):
|
|
|
board_c[x-y].append(board[x][y]) #根据斜线填充棋盘副本
|
|
|
board_d = [[] for line in range(29)] #反对角线方向的棋盘副本
|
|
|
for x in range(15):
|
|
|
for y in range(15):
|
|
|
board_d[x+y].append(board[x][y]) #根据反对角线填充棋盘副本
|
|
|
#返回四个方向的检查结果
|
|
|
return [check_win(board), check_win([list(l) for l in zip(*board)]), check_win(board_c), check_win(board_d)]
|
|
|
|
|
|
|
|
|
def main():
|
|
|
pg.init()
|
|
|
clock = pg.time.Clock()
|
|
|
screen = pg.display.set_mode((WIDTH, HEIGHT)) #设置显示模式
|
|
|
|
|
|
black = pg.image.load("data/storn_black.png").convert_alpha()
|
|
|
white = pg.image.load("data/storn_white.png").convert_alpha()
|
|
|
background = pg.image.load("data/bg.png").convert_alpha()
|
|
|
|
|
|
objects = [] #棋子列表
|
|
|
pg.display.set_caption("五子棋") #标题
|
|
|
flag = 0 #当前下棋的标记
|
|
|
going = True #游戏是否继续的标志
|
|
|
chess_list = [black, white] #黑棋和白棋的列表
|
|
|
letter_list = ['X', 'O'] #黑棋和白棋的标记列表
|
|
|
while going:
|
|
|
screen.blit(background, (0, 0))
|
|
|
for event in pg.event.get():
|
|
|
if event.type == pg.QUIT: #点击关闭窗口
|
|
|
going = False
|
|
|
elif event.type == pg.KEYDOWN and event.key == pg.K_ESCAPE: #如果按下Esc键
|
|
|
going = False
|
|
|
elif event.type == pg.MOUSEBUTTONDOWN: #鼠标点击
|
|
|
pos = pg.mouse.get_pos() #获取位置
|
|
|
a, b = round((pos[0] - 27)/40), round((pos[1] - 27)/40) #计算棋盘坐标
|
|
|
x, y = max(0, a) if a < 0 else min(a, 14), max(0, b) if b < 0 else min(b, 14) #限制棋子棋盘内
|
|
|
if set_chess(x, y, letter_list[flag]): #成功放置棋子
|
|
|
objects.append([chess_list[flag], (9+x*40, 9+y*40)]) #将棋子添加到棋子列表
|
|
|
flag = [1, 0][flag] #转换下棋方
|
|
|
if 0 in check_win_all(board) or 1 in check_win_all(board): #检查是否有人获胜
|
|
|
going = False #有人获胜，结束游戏
|
|
|
for o in objects:
|
|
|
screen.blit(o[0], o[1])
|
|
|
clock.tick(60)
|
|
|
pg.display.update()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
WIDTH = 615
|
|
|
HEIGHT = 615
|
|
|
board = [[' ']*15 for line in range(15)]
|
|
|
main()
|
|
|
pg.quit() |
