Init

agent/base_agent.py

@@ -0,0 +1,81 @@
+import logging
+
+from agent.agent import Agent
+from world.robot import T1, Robot
+from behaviors.behavior_manager import BehaviorManager
+from world.world import World
+from communication.server import Server
+from communication.world_parser import WorldParser
+
+logger = logging.getLogger(__file__)
+
+
+class Base_Agent:
+    def __init__(
+            self,
+            team_name: str = "Default",
+            number: int = 1,
+            host: str = "localhost",
+            port: int = 60000,
+            field: str = 'fifa'
+    ):
+        """
+        Initializes the agent and all its main components.
+
+        Args:
+            team_name (str): The name of the team the agent belongs to.
+            number (int): The player number assigned to this agent.
+            host (str): The host address of the simulator server.
+            port (int): The port number of the simulator server.
+            field (str): The name of the field configuration to use.
+        """
+
+        self.world: World = World(agent=self, team_name=team_name, number=number, field_name=field)
+        self.world_parser: WorldParser = WorldParser(agent=self)
+        self.server: Server = Server(
+            host=host, port=port, world_parser=self.world_parser
+        )
+        self.robot: Robot = T1(agent=self)
+        self.skills_manager: BehaviorManager = BehaviorManager(agent=self)
+        self.decision_maker: Agent = Agent(agent=self)
+
+    def run(self):
+        """
+        Starts the agent’s main control loop.
+
+        This method:
+          1. Connects to the simulator server.
+          2. Sends the initial configuration (init message).
+          3. Enters the main loop, where it:
+             - Receives and parses world updates.
+             - Updates internal world representation.
+             - Executes the decision-making process.
+             - Sends the next set of commands to the server.
+        """
+        self.server.connect()
+
+        self.server.send_immediate(
+            f"(init {self.robot.name} {self.world.team_name} {self.world.number})"
+        )
+
+        while True:
+            try:
+                self.server.receive()
+
+                self.world.update()
+
+                self.decision_maker.update_current_behavior()
+
+                self.server.send()
+            except Exception:
+                self.shutdown()
+                raise
+
+    def shutdown(self):
+        """
+        Safely shuts down the agent.
+
+        Logs a shutdown message and closes the server connection.
+        """
+        logger.info("Shutting down.")
+        self.server.shutdown()