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)
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
