mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-05-22 14:56:43 +02:00
32 lines
1.2 KiB
Python
32 lines
1.2 KiB
Python
from typing import Union
|
|
|
|
from marl_factory_grid.environment.actions import Action
|
|
from marl_factory_grid.modules.doors.entitites import Door
|
|
from marl_factory_grid.modules.doors import constants as d, rewards as r
|
|
from marl_factory_grid.environment import constants as c
|
|
from marl_factory_grid.utils.results import ActionResult
|
|
|
|
|
|
class DoorUse(Action):
|
|
|
|
def __init__(self, **kwargs):
|
|
super().__init__(d.ACTION_DOOR_USE, r.USE_DOOR_VALID, r.USE_DOOR_FAIL, **kwargs)
|
|
|
|
def do(self, entity, state) -> Union[None, ActionResult]:
|
|
# Check if agent really is standing on a door:
|
|
entities_close = state.entities.get_entities_near_pos(entity.pos)
|
|
|
|
valid = False
|
|
for door in [e for e in entities_close if isinstance(e, Door)]:
|
|
try:
|
|
# Will always be true, when there is at least a single door.
|
|
valid = door.use()
|
|
state.print(f'{entity.name} just used a {door.name} at {door.pos}')
|
|
|
|
except AttributeError:
|
|
pass
|
|
if not valid:
|
|
# When he doesn't stand necxxt to a door tell me.
|
|
state.print(f'{entity.name} just tried to use a door at {entity.pos}, but there is none.')
|
|
return self.get_result(valid, entity)
|