mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-05-22 14:56:43 +02:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
from typing import List
|
|
|
|
import marl_factory_grid.modules.maintenance.constants
|
|
from marl_factory_grid.environment.rules import Rule
|
|
from marl_factory_grid.utils.results import TickResult, DoneResult
|
|
from marl_factory_grid.environment import constants as c
|
|
from . import constants as M
|
|
|
|
|
|
class MoveMaintainers(Rule):
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def tick_step(self, state) -> List[TickResult]:
|
|
for maintainer in state[M.MAINTAINERS]:
|
|
maintainer.tick(state)
|
|
# Todo: Return a Result Object.
|
|
return []
|
|
|
|
|
|
class DoneAtMaintainerCollision(Rule):
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def on_check_done(self, state) -> List[DoneResult]:
|
|
agents = list(state[c.AGENT].values())
|
|
m_pos = state[M.MAINTAINERS].positions
|
|
done_results = []
|
|
for agent in agents:
|
|
if agent.pos in m_pos:
|
|
done_results.append(DoneResult(entity=agent, validity=c.VALID, identifier=self.name,
|
|
reward=marl_factory_grid.modules.maintenance.constants.MAINTAINER_COLLISION_REWARD))
|
|
return done_results
|