diff --git a/environments/factory/assets/agent.png b/environments/factory/assets/agent.png new file mode 100644 index 0000000..bd2bfc8 Binary files /dev/null and b/environments/factory/assets/agent.png differ diff --git a/environments/factory/assets/dirt.jpg b/environments/factory/assets/dirt.jpg new file mode 100644 index 0000000..a78950e Binary files /dev/null and b/environments/factory/assets/dirt.jpg differ diff --git a/environments/factory/assets/dirt.png b/environments/factory/assets/dirt.png new file mode 100644 index 0000000..9e29d1b Binary files /dev/null and b/environments/factory/assets/dirt.png differ diff --git a/environments/factory/assets/wall.png b/environments/factory/assets/wall.png new file mode 100644 index 0000000..512c67c Binary files /dev/null and b/environments/factory/assets/wall.png differ diff --git a/environments/factory/renderer.py b/environments/factory/renderer.py new file mode 100644 index 0000000..5ff7582 --- /dev/null +++ b/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)]}) + diff --git a/requirements.txt b/requirements.txt index 68d27bc..c49ccfa 100644 --- a/requirements.txt +++ b/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