steffen/FIKS_Entwicklungsumgebung
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added vsibility radius around agent

Browse Source
This commit is contained in:
romue 2021-05-18 12:03:12 +02:00
parent 556e8923a3
commit 8c8a8594fc
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
environments/factory/renderer.py
View File

@ -16,7 +16,6 @@ class Entity:
class Renderer:
BG_COLOR = (99, 110, 114)
WHITE = (200, 200, 200)
PINK = (0.5, 255, 118, 117)
def __init__(self, grid_w=16, grid_h=16, cell_size=40, fps=4, grid_lines=True, view_radius=2):
self.grid_h = grid_h
@ -76,7 +75,8 @@ class Renderer:
if 'agent' in asset and self.view_radius > 0:
visibility_rect = bp['dest'].inflate((self.view_radius*2)*self.cell_size, (self.view_radius*2)*self.cell_size)
shape_surf = pygame.Surface(visibility_rect.size, pygame.SRCALPHA)
pygame.draw.rect(shape_surf, self.PINK, shape_surf.get_rect())
pygame.draw.rect(shape_surf, self.WHITE, shape_surf.get_rect())
shape_surf.set_alpha(96)
self.screen.blit(shape_surf, visibility_rect)
self.screen.blit(**bp)
pygame.display.flip()

2
environments/factory/simple_factory_getting_dirty.py
View File

@ -37,7 +37,7 @@ class GettingDirty(BaseFactory):
def render(self):
if not self.renderer: # lazy init
height, width = self.state.shape[1:]
self.renderer = Renderer(width, height, view_radius=0)
self.renderer = Renderer(width, height, view_radius=2)
dirt = [Entity('dirt', [x, y], min(self.state[DIRT_INDEX, x, y], 1), 'scale')
for x, y in np.argwhere(self.state[DIRT_INDEX] > h.IS_FREE_CELL)]