pacai.core.isolation.isolator

 1import abc
 2import random
 3
 4import pacai.core.action
 5import pacai.core.agentaction
 6import pacai.core.agentinfo
 7import pacai.core.gamestate
 8
 9class AgentIsolator(abc.ABC):
10    """
11    An isolator isolates an agent instance from the game being played.
12    This "isolation" allows the game to hide or protect state from a agent.
13    For example, without isolation an agent can just directly modify the state
14    to get all the points and end the game.
15
16    All communication between the game engine and agent should be done via the isolator.
17    """
18
19    @abc.abstractmethod
20    def init_agents(self, agent_infos: dict[int, pacai.core.agentinfo.AgentInfo]) -> None:
21        """
22        Initialize the agents this isolator will be responsible for.
23        """
24
25    @abc.abstractmethod
26    def game_start(self,
27            rng: random.Random,
28            initial_state: pacai.core.gamestate.GameState,
29            timeout: float,
30            ) -> dict[int, pacai.core.agentaction.AgentActionRecord]:
31        """
32        Pass along the initial game state to each agent and all them the allotted time to start.
33        """
34
35    @abc.abstractmethod
36    def game_complete(self,
37            final_state: pacai.core.gamestate.GameState,
38            timeout: float,
39            ) -> dict[int, pacai.core.agentaction.AgentActionRecord]:
40        """
41        Notify all agents that the game is over.
42        """
43
44    @abc.abstractmethod
45    def get_action(self,
46            state: pacai.core.gamestate.GameState,
47            user_inputs: list[pacai.core.action.Action],
48            timeout: float,
49            ) -> pacai.core.agentaction.AgentActionRecord:
50        """
51        Get the current agent's next action.
52        User inputs may be provided by the UI if available.
53        """
54
55    @abc.abstractmethod
56    def close(self) -> None:
57        """
58        Close the isolator and release all owned resources.
59        This call should not communicate with agents and forcibly release any agent resources.
60        """