mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-08 05:23:41 +08:00
Adeeb Shihadeh
115 lines
3.3 KiB
Python
115 lines
3.3 KiB
Python
from openpilot.starpilot.ui.raylib import rl
|
|
from openpilot.system.ui.lib.lru_cache import LRUCache
|
|
from openpilot.system.ui.lib.text_measure import measure_text_cached
|
|
from openpilot.system.ui.lib.application import font_fallback
|
|
from openpilot.system.ui.lib.multilang import tr_display
|
|
from openpilot.system.ui.lib.display_translation import DisplayText, native_text
|
|
|
|
|
|
def _break_long_word(font: rl.Font, word: str, font_size: int, max_width: int, spacing: float = 0) -> list[str]:
|
|
if not word:
|
|
return []
|
|
|
|
parts = []
|
|
remaining = word
|
|
|
|
while remaining:
|
|
if measure_text_cached(font, remaining, font_size, spacing).x <= max_width:
|
|
parts.append(remaining)
|
|
break
|
|
|
|
# Binary search for the longest substring that fits
|
|
left, right = 1, len(remaining)
|
|
best_fit = 1
|
|
|
|
while left <= right:
|
|
mid = (left + right) // 2
|
|
substring = remaining[:mid]
|
|
width = measure_text_cached(font, substring, font_size, spacing).x
|
|
|
|
if width <= max_width:
|
|
best_fit = mid
|
|
left = mid + 1
|
|
else:
|
|
right = mid - 1
|
|
|
|
# Add the part that fits
|
|
parts.append(remaining[:best_fit])
|
|
remaining = remaining[best_fit:]
|
|
|
|
return parts
|
|
|
|
|
|
_CACHE_MAXSIZE = 8192
|
|
_cache: LRUCache[int, list[str]] = LRUCache(_CACHE_MAXSIZE)
|
|
|
|
|
|
def wrap_text(font: rl.Font, text: str, font_size: int, max_width: int, spacing: float = 0) -> list[str]:
|
|
text = native_text(tr_display(text))
|
|
font = font_fallback(font)
|
|
spacing = round(spacing, 4)
|
|
key = hash((font.texture.id, text, font_size, max_width, spacing))
|
|
cached = _cache.get(key)
|
|
if cached is not None:
|
|
return cached
|
|
|
|
if not text or max_width <= 0:
|
|
return []
|
|
|
|
# Split text by newlines first to preserve explicit line breaks
|
|
paragraphs = text.split('\n')
|
|
all_lines: list[str] = []
|
|
|
|
for paragraph in paragraphs:
|
|
# Handle empty paragraphs (preserve empty lines)
|
|
if not paragraph.strip():
|
|
all_lines.append("")
|
|
continue
|
|
|
|
# Process each paragraph separately
|
|
words = paragraph.split()
|
|
if not words:
|
|
all_lines.append("")
|
|
continue
|
|
|
|
lines: list[str] = []
|
|
current_line: list[str] = []
|
|
|
|
for word in words:
|
|
word_width = measure_text_cached(font, word, font_size, spacing).x
|
|
|
|
# Check if word alone exceeds max width (need to break the word)
|
|
if word_width > max_width:
|
|
# Finish current line if it has content
|
|
if current_line:
|
|
lines.append(" ".join(current_line))
|
|
current_line = []
|
|
|
|
# Break the long word into parts
|
|
lines.extend(_break_long_word(font, word, font_size, max_width, spacing))
|
|
continue
|
|
|
|
# Measure the actual joined string to get accurate width (accounts for kerning, etc.)
|
|
test_line = " ".join(current_line + [word]) if current_line else word
|
|
test_width = measure_text_cached(font, test_line, font_size, spacing).x
|
|
|
|
# Check if word fits on current line
|
|
if test_width <= max_width:
|
|
current_line.append(word)
|
|
else:
|
|
# Start new line with this word
|
|
if current_line:
|
|
lines.append(" ".join(current_line))
|
|
current_line = [word]
|
|
|
|
# Add remaining words
|
|
if current_line:
|
|
lines.append(" ".join(current_line))
|
|
|
|
# Add all lines from this paragraph
|
|
all_lines.extend(lines)
|
|
|
|
all_lines = [DisplayText(line) for line in all_lines]
|
|
_cache[key] = all_lines
|
|
return all_lines
|