mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-05-23 07:16:44 +02:00
52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
from typing import Union, List, Tuple
|
|
|
|
from marl_factory_grid.environment.groups.collection import Collection
|
|
from marl_factory_grid.modules.batteries.entitites import Pod, Battery
|
|
|
|
|
|
class Batteries(Collection):
|
|
_entity = Battery
|
|
|
|
@property
|
|
def var_is_blocking_light(self):
|
|
return False
|
|
|
|
@property
|
|
def var_can_collide(self):
|
|
return False
|
|
|
|
@property
|
|
def var_can_move(self):
|
|
return False
|
|
|
|
@property
|
|
def var_has_position(self):
|
|
return False
|
|
|
|
@property
|
|
def obs_tag(self):
|
|
return self.__class__.__name__
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(Batteries, self).__init__(*args, **kwargs)
|
|
|
|
def spawn(self, agents, initial_charge_level):
|
|
batteries = [self._entity(initial_charge_level, agent) for _, agent in enumerate(agents)]
|
|
self.add_items(batteries)
|
|
|
|
# def spawn(self, coords_or_quantity: Union[int, List[Tuple[(int, int)]]], *entity_args): hat keine pos
|
|
# agents = entity_args[0]
|
|
# initial_charge_level = entity_args[1]
|
|
# batteries = [self._entity(initial_charge_level, agent) for _, agent in enumerate(agents)]
|
|
# self.add_items(batteries)
|
|
|
|
|
|
class ChargePods(Collection):
|
|
_entity = Pod
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(ChargePods, self).__init__(*args, **kwargs)
|
|
|
|
def __repr__(self):
|
|
return super(ChargePods, self).__repr__()
|