mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-06-23 12:01:36 +02:00
23 lines
612 B
Python
23 lines
612 B
Python
from marl_factory_grid.environment import constants as c
|
|
from marl_factory_grid.environment.entity.wall import Wall
|
|
from marl_factory_grid.environment.groups.collection import Collection
|
|
|
|
|
|
class Walls(Collection):
|
|
_entity = Wall
|
|
symbol = c.SYMBOL_WALL
|
|
|
|
@property
|
|
def var_has_position(self):
|
|
return True
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(Walls, self).__init__(*args, **kwargs)
|
|
self._value = c.VALUE_OCCUPIED_CELL
|
|
|
|
def by_pos(self, pos: (int, int)):
|
|
try:
|
|
return super().by_pos(pos)[0]
|
|
except IndexError:
|
|
return None
|