main.py

import pygame

from gomoku import Gomoku
from gomoku.palette import COLOR_BLACK, COLOR_RED, COLOR_GRAY, \
    COLOR_GREEN, COLOR_WHITE


if __name__ == "__main__":
    pygame.init()
    pygame.font.init()

    stone = {}
    stone["white"], stone["black"] = [], []
    player1_score, player2_score = 0, 0
    game = Gomoku()
    game.draw_main()
    game.draw_score(player1_score, player2_score)

    play_order = None

    while True:
        event = pygame.event.poll()
        if event.type == pygame.MOUSEBUTTONDOWN:

            x_stone, y_stone = game.play_get_pos()

            # New game.
            if (125 + 45 * 16) > x_stone > 45 * 16 and 90 > y_stone > 45:
                stone["white"], stone["black"] = [], []
                player1_score, player2_score = 0, 0
                game = Gomoku()
                game.draw_main()
                game.draw_score(player1_score, player2_score)
                game.text_draw("GAME START", game.w_h//2, 30, COLOR_GREEN, 35)
                play_order = True

            # Next game.
            if (125 + 45 * 16) > x_stone > 45 * 16 and 160 > y_stone > 115:
                stone["white"], stone["black"] = [], []
                game = Gomoku()
                game.draw_main()
                game.draw_score(player1_score, player2_score)
                game.text_draw("NEXT GAME START", game.w_h//2, 30, COLOR_GREEN, 35)
                play_order = True

            # Draw a white stone (Player 1).
            if play_order is None:
                pass
            elif not play_order:
                game.text_draw("PLAYER 1", 45 * 16 + 65, game.w_h // 2 - 90,
                               COLOR_RED, 20)
                game.text_draw("PLAYER 2", 45 * 16 + 65, game.w_h // 2 + 20,
                               COLOR_BLACK, 20)
                if 45 <= x_stone <= game.w_h and 45 <= y_stone <= game.w_h:
                    x_stone, y_stone = game.play_draw_stone_pos()
                    stone, play_order = game.play_draw_stone(
                        stone, play_order, "white", COLOR_WHITE,
                        x_stone, y_stone)
                    game.text_draw("PLAYER 1", 45 * 16 + 65, game.w_h // 2 - 90,
                                   COLOR_GRAY, 20)
                    game.text_draw("PLAYER 2", 45 * 16 + 65, game.w_h // 2 + 20,
                                   COLOR_RED, 20)
                    player1_score, play_order = game.score(
                        stone, "white", player1_score, play_order)
                    if len(stone["white"]) + len(stone["black"]) == 225:
                        game.text_draw("DRAW", 45 * 16 + 65, game.w_h // 2 + 120,
                                       (200, 0, 0), 45)
                        play_order = None

            # Draw a black stone (Player 2).
            elif play_order:
                game.text_draw("PLAYER 1", 45 * 16 + 65, game.w_h // 2 - 90,
                               COLOR_GRAY, 20)
                game.text_draw("PLAYER 2", 45 * 16 + 65, game.w_h // 2 + 20,
                               COLOR_RED, 20)
                if 45 <= x_stone <= game.w_h and 45 <= y_stone <= game.w_h:
                    x_stone, y_stone = game.play_draw_stone_pos()
                    stone, play_order = game.play_draw_stone(
                        stone, play_order, "black", COLOR_BLACK, x_stone, y_stone)
                    game.text_draw("PLAYER 1", 45 * 16 + 65, game.w_h // 2 - 90,
                                   COLOR_RED, 20)
                    game.text_draw("PLAYER 2", 45 * 16 + 65, game.w_h // 2 + 20,
                                   COLOR_BLACK, 20)
                    player2_score, play_order = game.score(
                        stone, "black", player2_score, play_order)
                    if len(stone["white"]) + len(stone["black"]) == 225:
                        game.text_draw("DRAW", 45 * 16 + 65, game.w_h//2 + 120,
                                       (200, 0, 0), 45)
                        play_order = None
        game.interactive_button()
        pygame.display.update()