Main.py

import pygame
from Classes.Board import Board
from Classes.Button import Button
from Classes.Computer import Computer
import sys,getopt

def play(turn: str) -> None:
    """
    Function that runs the game.

    Parameters:
        turn (str): who will start the game.
    """

    board = Board(resolution,resolution,size)
    board.depth = depth
    board.startPos()
    computer = Computer(board)
    if turn == "first":
        board.turn = False
    else:
        board.turn = True      
    
    while True:
        screen.fill((0))
        mx, my = pygame.mouse.get_pos()

        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                sys.exit()
            elif event.type == pygame.MOUSEBUTTONDOWN:
                if event.button == 1:
                    board.click(mx,my)
            elif event.type == pygame.KEYDOWN:
                if event.key == pygame.K_ESCAPE:
                    main_menu()
                    break
        
        if board.turn == True:
            board.computerMove(computer)

        board.draw(screen)

        pygame.display.update()
        
        result =board.isWin()
        if result != False:
            winScreen(result)
            break

def winScreen(text) -> None:
    """
    Runs the Win Screen

    Parameters:
        text (str): stores who won.
    """
    
    
    while True:
        screen.fill((0))
        mouse_pos = pygame.mouse.get_pos()

        back_button = Button(image=None, pos=(centerY, (spacesX)*5), text_input="BACK", font=font, base_color="#d7fcd4", hovering_color="White")

        menu_text = font.render(f"{text} Wins", True, "#b68f40")
        menu_rect = menu_text.get_rect(center=(centerY, spacesX))
        screen.blit(menu_text, menu_rect)

        back_button.changeColor(mouse_pos)
        back_button.update(screen)
    

        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                sys.exit()

            if event.type == pygame.MOUSEBUTTONDOWN:
                if back_button.checkForInput(mouse_pos):
                    main_menu()
                    break
        pygame.display.update()

def about() -> None:
    """
    Function that runs about/how to play page
    """

    def text_seperator(text: str, x: int, y: int )-> list:
        """
        Seperates the text by lines and return a list

        Parameters:
                text (str): string to seperate

                x (int): x coordinate of the start possition of text

                y (int): y coordinate of the start possition of text

        Returns:
                lines (list): contains each line
        """

        lines = []
        text_font = pygame.font.SysFont("arialblack", fontSize//2)
        for line in text.splitlines():
            text = text_font.render(line, True, "#b68f40")
            rect = text.get_rect(center=(x, y))
            lines.append([text,rect])
            y+=fontSize
        return lines

    while True:
        screen.fill((0))
        mouse_pos = pygame.mouse.get_pos()

        text = """Each player moves one piece per turn.
        A piece can be moved one space forward or diagonally
        forward, if the target position is empty. A piece 
        may be also moved to a square occupied by an enemy 
        piece, if the target position is one space diagonally forward 
        The game ends if one player reaches opponent's home row.
        \n\n Made by Mehmet Sahin Uces"""

        back_button = Button(image=None, pos=(centerY, (spacesX)*5), text_input="BACK", font=font, base_color="#d7fcd4", hovering_color="White")

        for i in text_seperator(text,centerY//1.1,fontSize):
            screen.blit(i[0], i[1])

        back_button.changeColor(mouse_pos)
        back_button.update(screen)
    

        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                sys.exit()
            if event.type == pygame.MOUSEBUTTONDOWN:
                if back_button.checkForInput(mouse_pos):
                    main_menu()
                    break
        pygame.display.update()

def main_menu() -> None:
    """
    Function that runs Main Menu.
    """

    while True:
        screen.fill((0))

        mouse_pos = pygame.mouse.get_pos()

        menu_text = font.render("MAIN MENU", True, "#b68f40")
        menu_rect = menu_text.get_rect(center=(centerY, spacesX))

        first_button = Button(image=None, pos=(centerY, (spacesX)*2), text_input="FIRST TURN", font=font, base_color="#d7fcd4", hovering_color="White")
        second_button = Button(image=None, pos=(centerY, (spacesX)*3), text_input="SECOND TURN", font=font, base_color="#d7fcd4", hovering_color="White")
        about_button = Button(image=None, pos=(centerY, (spacesX)*4), text_input="HOW TO PLAY", font=font, base_color="#d7fcd4", hovering_color="White")
        quit_button = Button(image=None, pos=(centerY, (spacesX)*5), text_input="QUIT", font=font, base_color="#d7fcd4", hovering_color="White")

        screen.blit(menu_text, menu_rect)

        for button in [first_button,second_button,about_button, quit_button]:
            button.changeColor(mouse_pos)
            button.update(screen)
        
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                sys.exit()
            if event.type == pygame.MOUSEBUTTONDOWN:
                if first_button.checkForInput(mouse_pos):
                    play("first")
                if second_button.checkForInput(mouse_pos):
                    play("second")
                if about_button.checkForInput(mouse_pos):
                    about()
                if quit_button.checkForInput(mouse_pos):
                    pygame.quit()
                    sys.exit()

        pygame.display.update()


if __name__ == "__main__":
    pygame.init()
    resolution = 720
    size = 6
    depth = 3

    argv = sys.argv[1:]
    opts, args = getopt.getopt(argv, "r:s:d:", ["resolution =","size =","depth ="])
    for opt, arg in opts:
        if opt in ['-r','--resolution']:
            resolution = int(arg)
        if opt in ['-s','--size']:
            size = int(arg) if int(arg)>3 else 6
        if opt in ['-d','--depth']:
            depth = int(arg) if int(arg)>0 else 3

    centerY = resolution//2 #gets the center Y value
    spacesX = resolution//6 #gets the optimal spaces between buttons
    fontSize = resolution//18

    clock = pygame.time.Clock()
    clock.tick(30)
    screen = pygame.display.set_mode((resolution, resolution))
    font = pygame.font.SysFont("arialblack", fontSize)

    main_menu()