emulate.py

from board import GameBoard
from solution import AIPlayer
from player import HumanPlayer
import aux_funcs

def render_board(state: GameBoard):
    """Display the hexagonal board in terminal format."""
    print(f"\n{'-'*75}")
    print("HEX Board State (Even-R)")
    print(f"{'-'*75}")
    
    print("Player 1 (L ← → R): ●")
    print("Player 2 (T ↑ ↓ B): ○")
    print("Empty: □\n")
    
    print("    ", end="")
    for c in range(state.size):
        print(f"{c:3}", end="")
    print()
    
    for r in range(state.size):
        offset = "  " if r % 2 == 0 else ""
        print(f"{r:2} {offset}", end="")
        
        for c in range(state.size):
            val = state.board[r][c]
            if val == 0:
                print(" □ ", end="")
            elif val == 1:
                print(" ● ", end="")
            else:
                print(" ○ ", end="")
        print()
    
    print(f"{'-'*75}\n")


def select_player_mode(num: int) -> str:
    """Ask user to choose player type (AI or human)."""
    while True:
        print(f"\nPlayer {num} configuration:")
        print("1. AI Player")
        print("2. Human Player")
        opt = input("Choose (1 or 2): ").strip()
        
        if opt == "1":
            return "ai"
        elif opt == "2":
            return "human"
        else:
            print("Invalid option.")


def run_game(size: int = 4):
    """Start and manage HEX game session."""
    print("\n" + "="*70)
    print("PLAYER SETUP")
    print("="*70)
    
    p1_mode = select_player_mode(1)
    p2_mode = select_player_mode(2)
    
    if p1_mode == "ai":
        p1 = AIPlayer(1)
        print("\n✓ Player 1: AI")
    else:
        p1 = HumanPlayer(1)
        print("\n✓ Player 1: Human")
    
    if p2_mode == "ai":
        p2 = AIPlayer(2)
        print("✓ Player 2: AI")
    else:
        p2 = HumanPlayer(2)
        print("✓ Player 2: Human")
    
    board = GameBoard(size)
    
    print("\n" + "="*70)
    print(f"START HEX {size}x{size}")
    print("="*70)
    print("Player 1 (●): Connect left ← → right")
    print("Player 2 (○): Connect top ↑ ↓ bottom")
    print("="*70 + "\n")
    
    step = 0
    finished = False
    
    while not finished:
        if aux_funcs.is_board_full(board):
            print("\nDraw - board is full.")
            finished = True
            break
        
        print(f"\nTurn {step + 1}: Player 1 (●)")
        render_board(board)
        
        try:
            p1.move(board)
            if board.has_won(1):
                print("\nPlayer 1 (●) WINS! Connected left ← → right")
                render_board(board)
                finished = True
                break
        except ValueError as e:
            print(f"Error: {e}")
            finished = True
            break
        
        if aux_funcs.is_board_full(board):
            print("\nDraw - board is full.")
            finished = True
            break
        
        print(f"\nTurn {step + 1}: Player 2 (○)")
        render_board(board)
        
        try:
            p2.move(board)
            if board.has_won(2):
                print("\nPlayer 2 (○) WINS! Connected top ↑ ↓ bottom")
                render_board(board)
                finished = True
                break
        except ValueError as e:
            print(f"Error: {e}")
            finished = True
            break
        
        step += 1


if __name__ == "__main__":
    run_game(size=10)