added tooltips
This commit is contained in:
@@ -10,6 +10,8 @@ from vec import Vec2
|
||||
|
||||
|
||||
class SpeedMapDisplay(MapDisplay):
|
||||
APP_NAME = "Train Journey - Speed Map Display"
|
||||
|
||||
def __init__(self,
|
||||
surf: pygame.Surface,
|
||||
min_lon: float,
|
||||
@@ -61,14 +63,42 @@ class SpeedMapDisplay(MapDisplay):
|
||||
|
||||
def render(self) -> None:
|
||||
self.surf.fill((0, 0, 0))
|
||||
self._tooltip_surf.fill((0, 0, 0, 0))
|
||||
self.draw_cities()
|
||||
|
||||
if self._path is not None:
|
||||
self.draw_path(self._path)
|
||||
mpos = Vec2(*pygame.mouse.get_pos())
|
||||
self.tooltip_nearest(mpos)
|
||||
self.surf.blit(self._tooltip_surf, (0, 0))
|
||||
|
||||
def set_path(self, path: Path) -> None:
|
||||
self._path = path
|
||||
|
||||
def tooltip_nearest(self, mpos: Vec2) -> None:
|
||||
closest = None
|
||||
closest_i = 0
|
||||
min_dist = 0
|
||||
|
||||
for i, pt in enumerate(self._path.points):
|
||||
pt = Vec2(*self.real_to_screen(pt.x, pt.y))
|
||||
dist = (pt - mpos).mag
|
||||
if i == 0 or dist < min_dist:
|
||||
closest = pt
|
||||
closest_i = i
|
||||
min_dist = dist
|
||||
|
||||
pt = closest
|
||||
speed = self._path.extra_data[closest_i]
|
||||
col = (200, 150, 50)
|
||||
pygame.draw.circle(self.surf, col, (pt.x, pt.y), 2 * self.PATH_WIDTH, 2)
|
||||
pygame.draw.line(self.surf, col, (pt.x, pt.y), (mpos.x, mpos.y), 2)
|
||||
txt = self.font.render(f"{speed:.1f} km/h", True, (150, 200, 255))
|
||||
txt_size = Vec2(*txt.get_size())
|
||||
txt_pos = mpos - txt_size.scale(Vec2(0.5, 1))
|
||||
pygame.draw.rect(self._tooltip_surf, (0, 0, 0, 150), (txt_pos.x, txt_pos.y, txt_size.x, txt_size.y))
|
||||
self._tooltip_surf.blit(txt, (txt_pos.x, txt_pos.y))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
name = "data_28-03"
|
||||
|
||||
Reference in New Issue
Block a user