81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
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() |