python--pygame實現五子棋

2018.03.15
author：wills

今天學習瞭如何使用pygame來製作小遊戲，下面是五子棋的代碼，我的理解都寫在註釋裏了

import pygame
# 導入pygame模塊
print(pygame.ver)
# 檢查pygame的版本，檢查pygame有沒有導入成功

EMPTY = 0
BLACK = 1
WHITE = 2
# 定義三個常量函數，用來表示白棋，黑棋，以及 空

black_color = [0, 0, 0]
# 定義黑色（黑棋用，畫棋盤）
white_color = [255, 255, 255]
# 定義白色（白棋用）

# 定義棋盤這個類
class RenjuBoard(object):

    def __init__(self):
        # self._board = board = [[EMPTY] * 15 for _ in range(15)]
        # 將棋盤每一個交叉點都看作列表的一個元素位，一共有15*15共225個元素
        self._board = [[]] * 15
        self.reset()
    #重置棋盤
    def reset(self):
        for row in range(len(self._board)):
            self._board[row] = [EMPTY] * 15
    #定義棋盤上的下棋函數，row表示行，col表示列，is_black表示判斷當前點位該下黑棋，還是白棋
    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
    # 給棋盤定義一個函數將自己在screen上面畫出來，使用pygame.draw()函數。並且順便將下了的棋子也畫出來
    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,
        # 給棋盤加一個外框，使美觀
        pygame.draw.rect(screen, black_color, [36, 36, 568, 568], 3)
        # 在棋盤上標出，天元以及另外4個特殊點位
        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循環取得棋盤上所有交叉點的座標
        for row in range(len(self._board)):
            for col in range(len(self._board[row])):
                # 將下在棋盤上的棋子畫出來
                if self._board[row][col] != EMPTY:
                    ccolor = black_color \
                        if self._board[row][col] == BLACK else white_color
                    # 取得這個交叉點下的棋子的顏色，並將棋子畫出來
                    pos = [40 * (col + 1), 40 * (row + 1)]
                    # 畫出棋子
                    pygame.draw.circle(screen, ccolor, pos, 18, 0)

# 定義函數，傳入當前棋盤上的棋子列表，輸出結果，不管黑棋白棋勝，都是傳回False，未出結果則爲True
def is_win(board):
    for n in range(15):
        # 判斷垂直方向勝利
        flag = 0
        # flag是一個標籤，表示是否有連續以上五個相同顏色的棋子
        for b in board._board:
            if b[n] == 1:
                flag += 1
                if flag == 5:
                    print('黑棋勝')
                    return False
            else:
            # else表示此時沒有連續相同的棋子，標籤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():
    # 創建棋盤對象
    board = RenjuBoard()
    # 用於判斷是下黑棋還是白棋
    is_black = True
    # pygame初始化函數，固定寫法
    pygame.init()
    pygame.display.set_caption('五子棋') # 改標題
    # pygame.display.set_mode()表示建立個窗口，左上角爲座標原點，往右爲x正向，往下爲y軸正向
    screen = pygame.display.set_mode((640,640))
    # 給窗口填充顏色，顏色用三原色數字列表表示
    screen.fill([125,95,24])
    board.draw(screen)  # 給棋盤類發命令，調用draw()函數將棋盤畫出來
    pygame.display.flip()  # 刷新窗口顯示

    running = True
    # while 主循環的標籤，以便跳出循環
    while running:
        # 遍歷建立窗口後發生的所有事件，固定寫法
        for event in pygame.event.get():
            # 根據事件的類型，進行判斷
            if event.type == pygame.QUIT:
                running = False

            elif event.type == pygame.KEYUP:
                pass
            # pygame.MOUSEBUTTONDOWN表示鼠標的鍵被按下
            elif event.type == pygame.MOUSEBUTTONDOWN and \
                    event.button == 1:# button表示鼠標左鍵
                x, y = event.pos  # 拿到鼠標當前在窗口上的位置座標
                # 將鼠標的(x, y)窗口座標，轉化換爲棋盤上的座標
                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()
                    # 調用判斷勝負函數
                    if not is_win(board):
                        #break
                    running = False
                    # 這裏我有個bug沒找到解決辦法，就是判斷出勝負後，使用break跳出事件遍歷的for循環，但是老是不能跳出來，導致勝負分出來了還可以繼續下，這裏我採用判斷勝負後就將running標籤賦值爲False，跳出主循環，但是這樣棋盤的窗口也沒了。明天再找找bug在哪


    pygame.quit()



if __name__ == '__main__':
    main()