Merge pull request #2 from robjo82/develop

Develop

src/main.py

@@ -1,6 +1,171 @@
-def main():
-    print("Hello, world!")
+import copy
+import time
 
 
-if __name__ == "__main__":
-    main()
+# Initialisation du plateau de jeu
+def init_board():
+    board = [[' ' for _ in range(8)] for _ in range(8)]
+    board[3][3], board[4][4] = 'W', 'W'
+    board[3][4], board[4][3] = 'B', 'B'
+    return board
+
+
+# Affichage du plateau de jeu
+def display_board(board):
+    print("  0 1 2 3 4 5 6 7")
+    for i, row in enumerate(board):
+        print(i, end=' ')
+        for cell in row:
+            print(cell, end=' ')
+        print()
+
+
+# Vérification de la validité d'un mouvement
+def is_valid_move(board, x, y, player):
+    opponent = 'W' if player == 'B' else 'B'
+    if board[x][y] != ' ':
+        return False, []
+
+    directions = [(dx, dy) for dx in [-1, 0, 1] for dy in [-1, 0, 1] if dx != 0 or dy != 0]
+    flipped_cells = []
+
+    for dx, dy in directions:
+        temp_flips = []
+        nx, ny = x + dx, y + dy
+        while 0 <= nx < 8 and 0 <= ny < 8:
+            if board[nx][ny] == opponent:
+                temp_flips.append((nx, ny))
+            elif board[nx][ny] == player:
+                if temp_flips:
+                    flipped_cells.extend(temp_flips)
+                break
+            else:
+                break
+            nx += dx
+            ny += dy
+
+    return len(flipped_cells) > 0, flipped_cells
+
+
+# Appliquer un mouvement sur le plateau
+def make_move(board, x, y, player):
+    is_valid, flipped_cells = is_valid_move(board, x, y, player)
+    if is_valid:
+        board[x][y] = player
+        for fx, fy in flipped_cells:
+            board[fx][fy] = player
+    return is_valid
+
+
+# Fonction d'évaluation simple
+def evaluate_board(board, player):
+    player_score = sum(cell == player for row in board for cell in row)
+    opponent_score = sum(cell != ' ' and cell != player for row in board for cell in row)
+    return player_score - opponent_score
+
+
+# Mémoire des états déjà traités pour éviter les calculs redondants
+memo = {}
+
+
+# Algorithme Min-Max avec time-out et mémoire
+def minmax_with_memory(board, depth, maximizing, player, timeout):
+    if time.time() > timeout:
+        return None, None
+
+    board_str = ''.join(''.join(row) for row in board) + player
+    if board_str in memo:
+        return memo[board_str]
+
+    if depth == 0:
+        score = evaluate_board(board, player)
+        memo[board_str] = score, None
+        return score, None
+
+    opponent = 'W' if player == 'B' else 'B'
+    best_move = None
+
+    if maximizing:
+        max_eval = float('-inf')
+        for x in range(8):
+            for y in range(8):
+                new_board = copy.deepcopy(board)
+                if make_move(new_board, x, y, player):
+                    eval_value, _ = minmax_with_memory(new_board, depth - 1, False, player, timeout)
+                    if eval_value is None:  # Timeout occurred
+                        return None, None
+                    if eval_value > max_eval:
+                        max_eval = eval_value
+                        best_move = (x, y)
+        memo[board_str] = max_eval, best_move
+        return max_eval, best_move
+    else:
+        min_eval = float('inf')
+        for x in range(8):
+            for y in range(8):
+                new_board = copy.deepcopy(board)
+                if make_move(new_board, x, y, opponent):
+                    eval_value, _ = minmax_with_memory(new_board, depth - 1, True, player, timeout)
+                    if eval_value is None:  # Timeout occurred
+                        return None, None
+                    if eval_value < min_eval:
+                        min_eval = eval_value
+                        best_move = (x, y)
+        memo[board_str] = min_eval, best_move
+        return min_eval, best_move
+
+
+# Fonction pour récupérer le mouvement du joueur humain
+def get_human_move(board, player):
+    while True:
+        try:
+            x, y = map(int, input(f"Enter the coordinates where you want to place your '{player}' (row col): ").split())
+            is_valid, _ = is_valid_move(board, x, y, player)
+            if is_valid:
+                return x, y
+            else:
+                print("Invalid move. Try again.")
+        except ValueError:
+            print("Invalid input. Please enter two integers separated by a space.")
+
+
+# Fonction principale pour jouer au jeu
+def play_game():
+    board = init_board()
+    player_turn = 'W'
+
+    while True:
+        display_board(board)
+
+        if player_turn == 'W':
+            print("Human's turn:")
+            x, y = get_human_move(board, 'W')
+        else:
+            print("AI's turn:")
+            timeout = time.time() + 2  # 2 secondes de time-out pour l'IA
+            _, (x, y) = minmax_with_memory(board, 3, True, 'B', timeout)
+            if x is None and y is None:
+                print("AI timeout. Human wins!")
+                break
+
+        make_move(board, x, y, player_turn)
+
+        # Vérification de la fin du jeu
+        if all(cell != ' ' for row in board for cell in row):
+            w_score = sum(cell == 'W' for row in board for cell in row)
+            b_score = sum(cell == 'B' for row in board for cell in row)
+            print(f"Final scores - W: {w_score}, B: {b_score}")
+            if w_score > b_score:
+                print("Human wins!")
+            elif b_score > w_score:
+                print("AI wins!")
+            else:
+                print("It's a tie!")
+            break
+
+        player_turn = 'B' if player_turn == 'W' else 'W'
+
+
+# Jouer au jeu
+if __name__ == '__main__':
+    play_game()