You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
2.4 KiB
75 lines
2.4 KiB
4 months ago
|
import pyray as rl
|
||
|
|
from openpilot.system.ui.lib.widget import Widget
|
||
|
|
from openpilot.system.ui.lib.scroll_panel import GuiScrollPanel
|
||
|
|
|
||
|
|
ITEM_SPACING = 40
|
||
|
|
LINE_COLOR = rl.GRAY
|
||
|
|
LINE_PADDING = 40
|
||
|
|
|
||
|
|
|
||
|
|
class LineSeparator(Widget):
|
||
|
|
def __init__(self, height: int = 1):
|
||
|
|
super().__init__()
|
||
|
|
self._rect = rl.Rectangle(0, 0, 0, height)
|
||
|
|
|
||
|
|
def set_parent_rect(self, parent_rect: rl.Rectangle) -> None:
|
||
|
|
super().set_parent_rect(parent_rect)
|
||
|
|
self._rect.width = parent_rect.width
|
||
|
|
|
||
|
|
def _render(self, _):
|
||
|
|
rl.draw_line(int(self._rect.x) + LINE_PADDING, int(self._rect.y),
|
||
|
|
int(self._rect.x + self._rect.width) - LINE_PADDING * 2, int(self._rect.y),
|
||
|
|
LINE_COLOR)
|
||
|
|
|
||
|
|
|
||
|
|
class Scroller(Widget):
|
||
|
|
def __init__(self, items: list[Widget], spacing: int = ITEM_SPACING, line_separator: bool = False, pad_end: bool = True):
|
||
|
|
super().__init__()
|
||
|
|
self._items: list[Widget] = []
|
||
|
|
self._spacing = spacing
|
||
|
|
self._line_separator = line_separator
|
||
|
|
self._pad_end = pad_end
|
||
|
|
|
||
|
|
self.scroll_panel = GuiScrollPanel()
|
||
|
|
|
||
|
|
for item in items:
|
||
|
|
self.add_widget(item)
|
||
|
|
|
||
|
|
def add_widget(self, item: Widget) -> None:
|
||
|
|
if self._line_separator and len(self._items) > 0:
|
||
|
|
self._items.append(LineSeparator())
|
||
|
|
self._items.append(item)
|
||
|
|
item.set_touch_valid_callback(self.scroll_panel.is_touch_valid)
|
||
|
|
|
||
|
|
def _render(self, _):
|
||
|
|
# TODO: don't draw items that are not in the viewport
|
||
|
|
visible_items = [item for item in self._items if item.is_visible]
|
||
|
|
content_height = sum(item.rect.height for item in visible_items) + self._spacing * (len(visible_items))
|
||
|
|
if not self._pad_end:
|
||
|
|
content_height -= self._spacing
|
||
|
|
scroll = self.scroll_panel.handle_scroll(self._rect, rl.Rectangle(0, 0, self._rect.width, content_height))
|
||
|
|
|
||
|
|
rl.begin_scissor_mode(int(self._rect.x), int(self._rect.y),
|
||
|
|
int(self._rect.width), int(self._rect.height))
|
||
|
|
|
||
|
|
cur_height = 0
|
||
|
|
for idx, item in enumerate(visible_items):
|
||
|
|
if not item.is_visible:
|
||
|
|
continue
|
||
|
|
|
||
|
|
# Nicely lay out items vertically
|
||
|
|
x = self._rect.x
|
||
|
|
y = self._rect.y + cur_height + self._spacing * (idx != 0)
|
||
|
|
cur_height += item.rect.height + self._spacing * (idx != 0)
|
||
|
|
|
||
|
|
# Consider scroll
|
||
|
|
x += scroll.x
|
||
|
|
y += scroll.y
|
||
|
|
|
||
|
|
# Update item state
|
||
|
|
item.set_position(x, y)
|
||
|
|
item.set_parent_rect(self._rect)
|
||
|
|
item.render()
|
||
|
|
|
||
|
|
rl.end_scissor_mode()
|