43 lines
1.7 KiB
Python
43 lines
1.7 KiB
Python
|
# © Crown-owned copyright 2024, Defence Science and Technology Laboratory UK
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import random
|
||
|
|
from abc import abstractmethod
|
||
|
|
|
||
|
|
from primaite.game.agent.interface import AbstractScriptedAgent
|
||
|
|
|
||
|
|
|
||
|
|
class AbstractTAPAgent(AbstractScriptedAgent, identifier="Abstract_TAP"):
|
||
|
|
"""Base class for TAP agents to inherit from."""
|
||
|
|
|
||
|
|
config: "AbstractTAPAgent.ConfigSchema"
|
||
|
|
|
||
|
|
class ConfigSchema(AbstractScriptedAgent.ConfigSchema):
|
||
|
|
"""Configuration schema for Abstract TAP agents."""
|
||
|
|
|
||
|
|
agent_name: str = "Abstract_TAP"
|
||
|
|
starting_node_name: str
|
||
|
|
next_execution_timestep: int
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
def setup_agent(self) -> None:
|
||
|
|
"""Set up agent."""
|
||
|
|
pass
|
||
|
|
|
||
|
|
def _set_next_execution_timestep(self, timestep: int) -> None:
|
||
|
|
"""Set the next execution timestep with a configured random variance.
|
||
|
|
|
||
|
|
:param timestep: The timestep to add variance to.
|
||
|
|
"""
|
||
|
|
random_timestep_increment = random.randint(
|
||
|
|
-self.config.agent_settings.start_settings.variance, self.config.agent_settings.start_settings.variance
|
||
|
|
)
|
||
|
|
self.config.next_execution_timestep = timestep + random_timestep_increment
|
||
|
|
|
||
|
|
def _select_start_node(self) -> None:
|
||
|
|
"""Set the starting starting node of the agent to be a random node from this agent's action manager."""
|
||
|
|
# we are assuming that every node in the node manager has a data manipulation application at idx 0
|
||
|
|
num_nodes = len(self.config.action_manager.node_names)
|
||
|
|
starting_node_idx = random.randint(0, num_nodes - 1)
|
||
|
|
self.starting_node_name = self.config.action_manager.node_names[starting_node_idx]
|
||
|
|
self.config.logger.debug(f"Selected Starting node ID: {self.starting_node_name}")
|