def _update_display(self): for event in pygame.event.get(): if event.type == pygame.QUIT: return pygame.QUIT display = pygame.display.get_surface() display.fill(white) plot_map(self.map, display) for images in self._agent_images: for surf, rectangle in images: display.blit(surf, rectangle) display.blit(self._info_surface, (0, 0), None, pygame.BLEND_RGB_SUB) self._info_surface.fill(black) # clear notifications from previous round pygame.display.update()
def _update_display(self): for event in pygame.event.get(): if event.type == pygame.QUIT: return pygame.QUIT display = pygame.display.get_surface() display.fill(white) plot_map(self.map, self.obs, display) for images in self._agent_images: for surf, rectangle in images: display.blit(surf, rectangle) display.blit(self._info_surface, (0, 0), None, pygame.BLEND_RGB_SUB) self._info_surface.fill(black) # clear notifications from previous round pygame.display.update()
def _prepare_visualization(self): red = (254, 0, 0) pygame.init() screen = pygame.display.set_mode(self.size) screen.fill(white) scale = plot_map(self.map, screen) for state in self.agent_states.values(): s = pygame.Surface((25, 15)) s.set_colorkey(white) s.fill(white) pygame.draw.rect(s, red, pygame.Rect(0, 0, 15, 15)) pygame.draw.polygon(s, red, [(15, 0), (25, 8), (15, 15)], 0) self._agent_surfaces.append(s) self._agent_images.append([self._get_agent_image(s, state, scale)]) self._map_surface = screen return scale
def _prepare_visualization(self): red = (254, 0, 0) #my_col = (0, 0, 120) pygame.init() screen = pygame.display.set_mode(self.size) screen.fill(white) #screen.fill(my_col) scale = plot_map(self.map, screen) for state in self.agent_states.values(): s = pygame.Surface((25, 15)) # рисуем прозрачное тело машинки s.set_colorkey(white) s.fill(white) pygame.draw.rect(s, red, pygame.Rect(0, 0, 15, 15)) # рисуем красный корпус машины pygame.draw.polygon(s, red, [(15, 0), (25, 8), (15, 15)], 0) # рисуем красный "наконечник" машины self._agent_surfaces.append(s) self._agent_images.append([self._get_agent_image(s, state, scale)]) self._map_surface = screen return scale