min_max.py

"""AI pou le morpion !"""

import json
from copy import deepcopy
from morpion import Morpion


def move_hash(move):
    "Crée le hash correspondant"
    temp = "ABC"
    row, col = move
    return temp[row] + str(col + 1)


class MinMaxTree:
    "Class qui crée et update l'arbre de possibilités"

    def __init__(
        self, state, move: tuple, player_turn: int, father_hash: str = ""
    ) -> None:
        self.player = player_turn
        self.move = move
        self.state = state
        self.value = 0
        self.next = []
        if move != (None, None):
            self.hash = father_hash + move_hash(move)
        else:
            self.hash = ""
        simul_m = Morpion()
        simul_m.state = state
        simul_m.player_id = player_turn
        win, player = simul_m.winning_condition()
        if win:
            self.value = player
            self.best_move = (None, None)
        else:
            self.update_sons()
            self.update_value()
        del simul_m

    def update_sons(self):
        "Créé la liste des fils"
        for row, temp in enumerate(self.state):
            for col, value in enumerate(temp):
                if value == 0:
                    new_state = deepcopy(self.state)
                    new_state[row, col] = self.player
                    self.next.append(
                        MinMaxTree(new_state, (row, col), -self.player, self.hash)
                    )

    def update_value(self):
        "Update la valeur dans le noeud"
        self.best_move = (0, 0)
        func = max
        best_value = -2
        if self.player == 1:
            func = min
            best_value = 2
        for son in self.next:
            if best_value != func(best_value, son.value):
                self.best_move = son.move
                best_value = func(best_value, son.value)
        self.value = best_value

    def create_dict(self, base_dict=None):
        "Sauvegarde le calcul du meilleur coup"
        if base_dict is None:
            base_dict = {}
        base_dict[self.hash] = self.best_move
        for son in self.next:
            base_dict = son.create_dict(base_dict)
        return base_dict


tree = MinMaxTree(Morpion().state, (None, None), Morpion().player_id)
with open("min_max.json", "w", encoding="utf-8") as f:
    json.dump(tree.create_dict(), f, indent=4)