31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
"""
|
|
Copyright © IQ.Lvbs, apart of Project Teal Lvbs, All Rights Reserved, licensed under https://konn3kt.com/tos
|
|
"""
|
|
import pyray as rl
|
|
from iqpilot.selfdrive.ui.ui_state import ui_state, UIStatus
|
|
|
|
|
|
ACTIVE_CONFIDENCE_TOP = rl.Color(0x22, 0xB8, 0xB9, 0xFF)
|
|
ACTIVE_CONFIDENCE_BOTTOM = rl.Color(0x0C, 0x94, 0x96, 0xFF)
|
|
MEDIUM_CONFIDENCE_TOP = rl.Color(255, 200, 0, 255)
|
|
MEDIUM_CONFIDENCE_BOTTOM = rl.Color(255, 115, 0, 255)
|
|
LOW_CONFIDENCE_TOP = rl.Color(255, 0, 21, 255)
|
|
LOW_CONFIDENCE_BOTTOM = rl.Color(255, 0, 89, 255)
|
|
|
|
|
|
class IQConfidenceBall:
|
|
@staticmethod
|
|
def get_animate_status_probs():
|
|
if ui_state.status == UIStatus.LAT_ONLY:
|
|
return ui_state.sm['modelV2'].meta.disengagePredictions.steerOverrideProbs
|
|
|
|
return ui_state.sm['modelV2'].meta.disengagePredictions.brakeDisengageProbs
|
|
|
|
@staticmethod
|
|
def get_lat_long_dot_colors(confidence: float) -> tuple[rl.Color, rl.Color]:
|
|
if confidence > 0.5:
|
|
return ACTIVE_CONFIDENCE_TOP, ACTIVE_CONFIDENCE_BOTTOM
|
|
if confidence > 0.2:
|
|
return MEDIUM_CONFIDENCE_TOP, MEDIUM_CONFIDENCE_BOTTOM
|
|
return LOW_CONFIDENCE_TOP, LOW_CONFIDENCE_BOTTOM
|