github-actions[bot]
de77c89cf6
version: sunnypilot v2026.002.000 (staging) date: 2026-07-21T07:44:02 master commit: 2c334ede443d7391d27575af8c854a095ba702a8
27 lines
883 B
Python
27 lines
883 B
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.
|
|
"""
|
|
from openpilot.selfdrive.ui.onroad.augmented_road_view import BORDER_COLORS
|
|
from openpilot.selfdrive.ui.ui_state import ui_state, UIStatus
|
|
|
|
|
|
class ConfidenceBallSP:
|
|
@staticmethod
|
|
def get_animate_status_probs():
|
|
if ui_state.status == UIStatus.LAT_ONLY:
|
|
return ui_state.sm['modelV2'].meta.disengagePredictions.steerOverrideProbs
|
|
|
|
# UIStatus.LONG_ONLY
|
|
return ui_state.sm['modelV2'].meta.disengagePredictions.brakeDisengageProbs
|
|
|
|
@staticmethod
|
|
def get_lat_long_dot_color():
|
|
if ui_state.status == UIStatus.LAT_ONLY:
|
|
return BORDER_COLORS[UIStatus.LAT_ONLY]
|
|
|
|
# UIStatus.LONG_ONLY
|
|
return BORDER_COLORS[UIStatus.LONG_ONLY]
|