mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-10-31 12:37:27 +01:00
128 lines
3.3 KiB
Python
128 lines
3.3 KiB
Python
from abc import ABC, abstractmethod
|
|
from collections import defaultdict
|
|
from typing import Union
|
|
|
|
from environment import constants as c
|
|
|
|
|
|
class Object:
|
|
|
|
"""Generell Objects for Organisation and Maintanance such as Actions etc..."""
|
|
|
|
_u_idx = defaultdict(lambda: 0)
|
|
|
|
def __bool__(self):
|
|
return True
|
|
|
|
@property
|
|
def observers(self):
|
|
return self._observers
|
|
|
|
@property
|
|
def name(self):
|
|
if self._str_ident is not None:
|
|
return f'{self.__class__.__name__}[{self._str_ident}]'
|
|
return f'{self.__class__.__name__}#{self.identifier_int}'
|
|
|
|
@property
|
|
def identifier(self):
|
|
if self._str_ident is not None:
|
|
return self._str_ident
|
|
else:
|
|
return self.name
|
|
|
|
def __init__(self, str_ident: Union[str, None] = None, **kwargs):
|
|
self._observers = []
|
|
self._str_ident = str_ident
|
|
self.identifier_int = self._identify_and_count_up()
|
|
self._collection = None
|
|
|
|
if kwargs:
|
|
print(f'Following kwargs were passed, but ignored: {kwargs}')
|
|
|
|
def __repr__(self):
|
|
return f'{self.name}'
|
|
|
|
def __eq__(self, other) -> bool:
|
|
return other == self.identifier
|
|
|
|
def __hash__(self):
|
|
return hash(self.identifier)
|
|
|
|
def _identify_and_count_up(self):
|
|
idx = Object._u_idx[self.__class__.__name__]
|
|
Object._u_idx[self.__class__.__name__] += 1
|
|
return idx
|
|
|
|
def set_collection(self, collection):
|
|
self._collection = collection
|
|
|
|
def add_observer(self, observer):
|
|
self.observers.append(observer)
|
|
observer.notify_change_pos(self)
|
|
|
|
def del_observer(self, observer):
|
|
self.observers.remove(observer)
|
|
|
|
|
|
class EnvObject(Object):
|
|
|
|
"""Objects that hold Information that are observable, but have no position on the env grid. Inventories etc..."""
|
|
|
|
_u_idx = defaultdict(lambda: 0)
|
|
|
|
@property
|
|
def obs_tag(self):
|
|
try:
|
|
return self._collection.name or self.name
|
|
except AttributeError:
|
|
return self.name
|
|
|
|
@property
|
|
def is_blocking_light(self):
|
|
try:
|
|
return self._collection.is_blocking_light or False
|
|
except AttributeError:
|
|
return False
|
|
|
|
@property
|
|
def can_move(self):
|
|
try:
|
|
return self._collection.can_move or False
|
|
except AttributeError:
|
|
return False
|
|
|
|
@property
|
|
def is_blocking_pos(self):
|
|
try:
|
|
return self._collection.is_blocking_pos or False
|
|
except AttributeError:
|
|
return False
|
|
|
|
@property
|
|
def has_position(self):
|
|
try:
|
|
return self._collection.has_position or False
|
|
except AttributeError:
|
|
return False
|
|
|
|
@property
|
|
def can_collide(self):
|
|
try:
|
|
return self._collection.can_collide or False
|
|
except AttributeError:
|
|
return False
|
|
|
|
@property
|
|
def encoding(self):
|
|
return c.VALUE_OCCUPIED_CELL
|
|
|
|
def __init__(self, **kwargs):
|
|
super(EnvObject, self).__init__(**kwargs)
|
|
|
|
def change_parent_collection(self, other_collection):
|
|
other_collection.add_item(self)
|
|
self._collection.delete_env_object(self)
|
|
self._collection = other_collection
|
|
return self._collection == other_collection
|