steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-06-26 05:01:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

initial n steps

Browse Source
This commit is contained in:
Steffen Illium
2023-11-23 17:32:32 +01:00
parent 2f29ef703c
commit 0ec260f6a2
8 changed files with 92 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
marl_factory_grid/environment/groups/global_entities.py
View File

@ -4,15 +4,17 @@ from random import shuffle
from typing import Dict
from marl_factory_grid.environment.groups.objects import Objects
from marl_factory_grid.utils.helpers import POS_MASK
from marl_factory_grid.utils.helpers import POS_MASK_8, POS_MASK_4
class Entities(Objects):
_entity = Objects
@staticmethod
def neighboring_positions(pos):
return [tuple(x) for x in (POS_MASK + pos).reshape(-1, 2)]
def neighboring_positions(self, pos):
return [tuple(x) for x in (POS_MASK_8 + pos).reshape(-1, 2) if tuple(x) in self._floor_positions]
def neighboring_4_positions(self, pos):
return [tuple(x) for x in (POS_MASK_4 + pos) if tuple(x) in self._floor_positions]
def get_entities_near_pos(self, pos):
return [y for x in itemgetter(*self.neighboring_positions(pos))(self.pos_dict) for y in x]