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

Code Comments, Getting Dirty Env, Naming

Browse Source
This commit is contained in:
steffen-illium 2021-05-11 13:53:05 +02:00
parent 7d926c403d
commit 2460d8e5c2
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
environments/factory/simple_factory_getting_dirty.py
View File

@ -3,22 +3,25 @@ from environments.factory.base_factory import BaseFactory
from collections import namedtuple from collections import namedtuple
DirtProperties = namedtuple('DirtProperties', ['clean_amount', 'max_spawn_ratio', 'gain_amount']) DirtProperties = namedtuple('DirtProperties', ['clean_amount', 'max_spawn_ratio', 'gain_amount'],
defaults=[0.25, 0.1, 0.1])
class GettingDirty(BaseFactory): class GettingDirty(BaseFactory):
_dirt_indx = -1 _dirt_indx = -1
def __init__(self, *args, dirt_properties, **kwargs): def __init__(self, *args, dirt_properties:DirtProperties, **kwargs):
super(GettingDirty, self).__init__(*args, **kwargs) super(GettingDirty, self).__init__(*args, **kwargs)
self._dirt_properties = dirt_properties self._dirt_properties = dirt_properties
self.slice_strings.update({self.state.shape[0]-1: 'dirt'}) self.slice_strings.update({self.state.shape[0]-1: 'dirt'})
def spawn_dirt(self): def spawn_dirt(self):
free_for_dirt = self.free_cells free_for_dirt = self.free_cells
for x, y in free_for_dirt[:self._max_dirt_spawn_ratio * free_for_dirt.]: # randomly distribute dirt across the grid # randomly distribute dirt across the grid
self.state[self._dirt_indx, x, y] += 0.1 n_dirt_tiles = self._dirt_properties.max_spawn_ratio * len(free_for_dirt)
for x, y in free_for_dirt[:n_dirt_tiles]:
self.state[self._dirt_indx, x, y] += self._dirt_properties.gain_amount
def reset(self): def reset(self):
# ToDo: When self.reset returns the new states and stuff, use it here! # ToDo: When self.reset returns the new states and stuff, use it here!