mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-20 06:25:56 +08:00
James Vecellio-Grant
3e29a0ccfe
* raylib: progress bar * freaking test dir * easier to see * smoother updating * final --------- Co-authored-by: Jason Wen <haibin.wen3@gmail.com>
58 lines
2.3 KiB
Python
58 lines
2.3 KiB
Python
"""
|
|
Copyright (c) 2021-, Haibin Wen, sunnypilot, and a number of other contributors.
|
|
|
|
This file is part of sunnypilot and is licensed under the MIT License.
|
|
See the LICENSE.md file in the root directory for more details.
|
|
"""
|
|
import pyray as rl
|
|
from openpilot.system.ui.lib.application import gui_app, FontWeight
|
|
from openpilot.system.ui.lib.text_measure import measure_text_cached
|
|
from openpilot.system.ui.widgets.list_view import ListItem, ItemAction
|
|
|
|
|
|
class ProgressBarAction(ItemAction):
|
|
def __init__(self, width=600):
|
|
super().__init__(width=width)
|
|
self.progress = 0.0
|
|
self.text = ""
|
|
self.show_progress = False
|
|
self.text_color = rl.GRAY
|
|
self._font = gui_app.font(FontWeight.NORMAL)
|
|
|
|
def update(self, progress, text, show_progress=False, text_color=rl.GRAY):
|
|
self.progress = progress
|
|
self.text = text
|
|
self.show_progress = show_progress
|
|
self.text_color = text_color
|
|
|
|
def _render(self, rect: rl.Rectangle):
|
|
font_size = 40
|
|
text_size = measure_text_cached(self._font, self.text, font_size)
|
|
padding = 30
|
|
bar_width = text_size.x + 2 * padding
|
|
text_x = (bar_width - text_size.x) / 2
|
|
|
|
if self.show_progress and len(parts := self.text.split(' - ', 1)) == 2:
|
|
prefix = parts[0]
|
|
max_prefix_w = measure_text_cached(self._font, "100%", font_size).x
|
|
current_prefix_w = measure_text_cached(self._font, prefix, font_size).x
|
|
|
|
bar_width = (text_size.x - current_prefix_w + max_prefix_w) + 2 * padding
|
|
text_x = padding + (max_prefix_w - current_prefix_w)
|
|
|
|
bar_height = 60
|
|
bar_rect = rl.Rectangle(rect.x + rect.width - bar_width, rect.y + (rect.height - bar_height) / 2, bar_width, bar_height)
|
|
|
|
if self.show_progress:
|
|
inner_rect = rl.Rectangle(bar_rect.x + 4, bar_rect.y + 4, bar_rect.width - 8, bar_rect.height - 8)
|
|
if inner_rect.width > 0:
|
|
fill_width = max(0, min(inner_rect.width, inner_rect.width * (self.progress / 100.0)))
|
|
rl.draw_rectangle_rounded(rl.Rectangle(inner_rect.x, inner_rect.y, fill_width, inner_rect.height), 0.2, 10, rl.Color(30, 121, 232, 255))
|
|
|
|
rl.draw_text_ex(self._font, self.text, rl.Vector2(bar_rect.x + text_x, bar_rect.y + (bar_height - text_size.y) / 2), font_size, 0, self.text_color)
|
|
|
|
|
|
def progress_item(title):
|
|
action = ProgressBarAction()
|
|
return ListItem(title=title, action_item=action)
|