ids aligned to hero vehicle

This commit is contained in:
Manish 2019-01-29 16:07:54 +01:00
parent 13c0c7d6b6
commit 6c784bca85
1 changed files with 26 additions and 23 deletions

View File

@ -12,6 +12,8 @@
Welcome to CARLA No Rendering Mode Visualizer
I : Toggle HUD
H : Hero Mode
Mouse Wheel : Zoom In / Zoom Out
Mouse Drag : Move Map in No Hero Mode
ESC : quit
"""
@ -407,16 +409,19 @@ class ModuleHUD (object):
def add_info(self, module_name, info):
self._info_text[module_name] = info
def renderActorId(self, vehicle_id_surface, list_actors, transform_helper, translation_offset, angle):
if self._show_info:
def render_actors_ids(self, vehicle_id_surface, list_actors, transform_helper, translation_offset, hero_actor):
vehicle_id_surface.fill(COLOR_BLACK)
if self._show_info:
vehicle_id_surface.set_alpha(150)
v_offset = 4
for actor in list_actors:
location = actor.get_location()
x, y = transform_helper.convert_world_to_screen_point((location.x, location.y - v_offset))
angle = 0
if hero_actor is not None:
angle = -hero_actor.get_transform().rotation.yaw - 90
color_surface = pygame.Surface((len(str(actor.id)) * 8, 14))
color_surface.fill(COLOR_BLACK)
color_surface.set_alpha(200)
@ -823,6 +828,16 @@ class ModuleWorld(object):
img2.blit(img, (w-pos[0], h-pos[1]))
return pygame.transform.rotate(img2, angle)
def clip_surfaces(self, clipping_rect):
self.map_surface.set_clip(clipping_rect)
self.vehicles_surface.set_clip(clipping_rect)
self.traffic_light_surface.set_clip(clipping_rect)
self.speed_limits_surface.set_clip(clipping_rect)
self.walkers_surface.set_clip(clipping_rect)
self.vehicle_id_surface.set_clip(clipping_rect)
self.hero_actor_surface.set_clip(clipping_rect)
self.result_surface.set_clip(clipping_rect)
def render(self, display):
if not self.map_rendered:
@ -929,24 +944,19 @@ class ModuleWorld(object):
(self.vehicle_id_surface, (0, 0)),
(self.hero_actor_surface, (0, 0))
)
self.hud_module.renderActorId(self.vehicle_id_surface, vehicles, self.transform_helper, (0, 0), angle)
self.hud_module.render_actors_ids(self.vehicle_id_surface, vehicles,
self.transform_helper, (0, 0), self.hero_actor)
rotated_result_surface = self.result_surface
if self.hero_actor is not None:
hero_surface = pygame.Surface((self.surface_size, self.surface_size), pygame.SRCALPHA)
# Apply clipping rect
clipping_rect = pygame.Rect(-translation_offset[0] - hero_surface.get_width() / 2,
-translation_offset[1] - hero_surface.get_height() / 2, self.hud_module.dim[0], self.hud_module.dim[1])
# Apply clipping rect
self.map_surface.set_clip(clipping_rect)
self.vehicles_surface.set_clip(clipping_rect)
self.traffic_light_surface.set_clip(clipping_rect)
self.speed_limits_surface.set_clip(clipping_rect)
self.walkers_surface.set_clip(clipping_rect)
self.vehicle_id_surface.set_clip(clipping_rect)
self.hero_actor_surface.set_clip(clipping_rect)
self.result_surface.set_clip(clipping_rect)
self.clip_surfaces(clipping_rect)
self.result_surface.blits(surfaces)
hero_surface.fill(COLOR_BROWN)
hero_surface.blit(self.result_surface, (translation_offset[0] + hero_surface.get_width()/2,
translation_offset[1] + hero_surface.get_height()/2))
@ -958,19 +968,12 @@ class ModuleWorld(object):
final_offset = rotated_result_surface.get_rect(center=center_offset)
display.blit(rotated_result_surface, final_offset)
else:
# Apply clipping rect
clipping_rect = pygame.Rect(-translation_offset[0] - center_offset[0], -translation_offset[1],
self.hud_module.dim[0], self.hud_module.dim[1])
self.map_surface.set_clip(clipping_rect)
self.vehicles_surface.set_clip(clipping_rect)
self.traffic_light_surface.set_clip(clipping_rect)
self.speed_limits_surface.set_clip(clipping_rect)
self.walkers_surface.set_clip(clipping_rect)
self.vehicle_id_surface.set_clip(clipping_rect)
self.hero_actor_surface.set_clip(clipping_rect)
self.result_surface.set_clip(clipping_rect)
self.clip_surfaces(clipping_rect)
self.result_surface.blits(surfaces)
display.blit(rotated_result_surface, (translation_offset[0] + center_offset[0],
translation_offset[1]))