updated base factory

environments/factory/renderer.py

@@ -0,0 +1,60 @@
+import numpy as np
+import pygame
+from pathlib import Path
+
+
+class Renderer:
+    BG_COLOR = (99, 110, 114)
+    WHITE = (200, 200, 200)
+
+    def __init__(self, grid_w=16, grid_h=16, cell_size=25, fps=4, grid_lines=True, assets=['wall', 'agent']):
+        self.grid_h = grid_h
+        self.grid_w = grid_w
+        self.cell_size = cell_size
+        self.fps = fps#
+        self.grid_lines = grid_lines
+        pygame.init()
+        self.screen_size = (grid_h*cell_size, grid_w*cell_size)
+        self.screen = pygame.display.set_mode(self.screen_size)
+        self.clock = pygame.time.Clock()
+        self.assets = {name: self.load_asset(name, 0.97) for name in assets}
+        self.fill_bg()
+
+    def fill_bg(self):
+        self.screen.fill(Renderer.BG_COLOR)
+        if self.grid_lines:
+            h, w = self.screen_size
+            for x in range(0, w, self.cell_size):
+                for y in range(0, h, self.cell_size):
+                    rect = pygame.Rect(x, y, self.cell_size, self.cell_size)
+                    pygame.draw.rect(self.screen, Renderer.WHITE, rect, 1)
+
+    def render_asset(self, r, c, name):
+        img = self.assets[name]
+        o = self.cell_size//2
+        r_, c_ = r*self.cell_size + o, c*self.cell_size + o
+        rect = img.get_rect()
+        rect.centerx, rect.centery = c_, r_
+        self.screen.blit(img, rect)
+        return c_, r_
+
+    def load_asset(self, name, factor=1.0):
+        s = int(factor*self.cell_size)
+        wall_img = pygame.image.load(str(Path(__file__).parent / 'assets' / f'{name}.png')).convert_alpha()
+        wall_img = pygame.transform.scale(wall_img, (s, s))
+        return wall_img
+
+    def render(self, pos_dict):
+        self.fill_bg()
+        for asset, positions in pos_dict.items():
+            for x, y in positions:
+                self.render_asset(x, y, asset)
+        pygame.display.flip()
+        self.clock.tick(self.fps)
+
+
+if __name__ == '__main__':
+    renderer = Renderer(fps=2, cell_size=40, assets=['wall', 'agent', 'dirt'])
+    for i in range(15):
+        renderer.render({'agent': [(5, 5)], 'wall': [(0, i), (i, 0)], 'dirt': [(3,3), (3,4)]})
+

requirements.txt

@@ -7,6 +7,7 @@ kiwisolver==1.3.1
 matplotlib==3.4.1
 numpy==1.20.2
 pandas==1.2.4
+pygame==2.0.0
 pep517==0.10.0
 Pillow==8.2.0
 pyparsing==2.4.7