mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-05-23 15:26:43 +02:00
136 lines
4.6 KiB
Python
136 lines
4.6 KiB
Python
import abc
|
|
from random import shuffle
|
|
from typing import List
|
|
|
|
from marl_factory_grid.environment.entity.agent import Agent
|
|
from marl_factory_grid.utils import helpers as h
|
|
from marl_factory_grid.utils.results import TickResult, DoneResult
|
|
from marl_factory_grid.environment import rewards as r, constants as c
|
|
|
|
|
|
class Rule(abc.ABC):
|
|
|
|
@property
|
|
def name(self):
|
|
return self.__class__.__name__
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
def __repr__(self):
|
|
return f'{self.name}'
|
|
|
|
def on_init(self, state, lvl_map):
|
|
return []
|
|
|
|
def on_reset(self):
|
|
return []
|
|
|
|
def tick_pre_step(self, state) -> List[TickResult]:
|
|
return []
|
|
|
|
def tick_step(self, state) -> List[TickResult]:
|
|
return []
|
|
|
|
def tick_post_step(self, state) -> List[TickResult]:
|
|
return []
|
|
|
|
def on_check_done(self, state) -> List[DoneResult]:
|
|
return []
|
|
|
|
|
|
class SpawnAgents(Rule):
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
pass
|
|
|
|
def on_init(self, state, lvl_map):
|
|
agent_conf = state.agents_conf
|
|
# agents = Agents(lvl_map.size)
|
|
agents = state[c.AGENT]
|
|
empty_positions = state.entities.empty_positions()[:len(agent_conf)]
|
|
for agent_name in agent_conf:
|
|
actions = agent_conf[agent_name]['actions'].copy()
|
|
observations = agent_conf[agent_name]['observations'].copy()
|
|
positions = agent_conf[agent_name]['positions'].copy()
|
|
if positions:
|
|
shuffle(positions)
|
|
while True:
|
|
try:
|
|
pos = positions.pop()
|
|
except IndexError:
|
|
raise ValueError(f'It was not possible to spawn an Agent on the available position: '
|
|
f'\n{agent_name[agent_name]["positions"].copy()}')
|
|
if agents.by_pos(pos) and state.check_pos_validity(pos):
|
|
continue
|
|
else:
|
|
agents.add_item(Agent(actions, observations, pos, str_ident=agent_name))
|
|
break
|
|
else:
|
|
agents.add_item(Agent(actions, observations, empty_positions.pop(), str_ident=agent_name))
|
|
pass
|
|
|
|
|
|
class MaxStepsReached(Rule):
|
|
|
|
def __init__(self, max_steps: int = 500):
|
|
super().__init__()
|
|
self.max_steps = max_steps
|
|
|
|
def on_init(self, state, lvl_map):
|
|
pass
|
|
|
|
def on_check_done(self, state):
|
|
if self.max_steps <= state.curr_step:
|
|
return [DoneResult(validity=c.VALID, identifier=self.name, reward=0)]
|
|
return [DoneResult(validity=c.NOT_VALID, identifier=self.name, reward=0)]
|
|
|
|
|
|
class AssignGlobalPositions(Rule):
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def on_init(self, state, lvl_map):
|
|
from marl_factory_grid.environment.entity.util import GlobalPosition
|
|
for agent in state[c.AGENT]:
|
|
gp = GlobalPosition(lvl_map.level_shape)
|
|
gp.bind_to(agent)
|
|
state[c.GLOBALPOSITIONS].add_item(gp)
|
|
return []
|
|
|
|
|
|
class Collision(Rule):
|
|
|
|
def __init__(self, done_at_collisions: bool = False):
|
|
super().__init__()
|
|
self.done_at_collisions = done_at_collisions
|
|
self.curr_done = False
|
|
|
|
def tick_post_step(self, state) -> List[TickResult]:
|
|
self.curr_done = False
|
|
pos_with_collisions = state.get_all_pos_with_collisions()
|
|
results = list()
|
|
for pos in pos_with_collisions:
|
|
guests = [x for x in state.entities.pos_dict[pos] if x.var_can_collide]
|
|
if len(guests) >= 2:
|
|
for i, guest in enumerate(guests):
|
|
try:
|
|
guest.set_state(TickResult(identifier=c.COLLISION, reward=r.COLLISION,
|
|
validity=c.NOT_VALID, entity=self))
|
|
except AttributeError:
|
|
pass
|
|
results.append(TickResult(entity=guest, identifier=c.COLLISION,
|
|
reward=r.COLLISION, validity=c.VALID))
|
|
self.curr_done = True if self.done_at_collisions else False
|
|
return results
|
|
|
|
def on_check_done(self, state) -> List[DoneResult]:
|
|
if self.done_at_collisions:
|
|
inter_entity_collision_detected = self.curr_done
|
|
move_failed = any(h.is_move(x.state.identifier) and not x.state.validity for x in state[c.AGENT])
|
|
if inter_entity_collision_detected or move_failed:
|
|
return [DoneResult(validity=c.VALID, identifier=c.COLLISION, reward=r.COLLISION)]
|
|
return [DoneResult(validity=c.NOT_VALID, identifier=self.name, reward=0)]
|