"""
Copyright © IQ.Lvbs, apart of Project Teal Lvbs, All Rights Reserved, licensed under https://konn3kt.com/tos
"""
import pyray as rl
from iqpilot.cereal import log

from iqpilot.selfdrive.ui import UI_BORDER_SIZE
from iqpilot.selfdrive.ui.ui_state import ui_state
from iqpilot.selfdrive.ui.onroad.driver_state import BTN_SIZE
from iqpilot.system.ui.lib.application import gui_app

EventName = log.OnroadEvent.EventName
SOFT_WARNING_EVENTS = {
  EventName.commIssue,
  EventName.commIssueAvgFreq,
  EventName.selfdrivedLagging,
}
ICON_SIZE = 96
_SPEED_BOX_X_OFFSET = 60
_SPEED_BOX_Y_OFFSET = 45
_SPEED_BOX_WIDTH = 180
_SPEED_BOX_HEIGHT = 228

_DM_OFFSET = UI_BORDER_SIZE + BTN_SIZE // 2

class SoftWarningRenderer:
  def __init__(self):
    self._icon = gui_app.texture("icons_mici/offroad_alerts/orange_warning.png", ICON_SIZE, ICON_SIZE)
    self._active = False

  def update(self) -> None:
    sm = ui_state.sm
    self._active = any(e.name in SOFT_WARNING_EVENTS for e in sm['onroadEvents'])

  def render(self, rect: rl.Rectangle) -> None:
    if not self._active:
      return
    speed_cx = rect.x + _SPEED_BOX_X_OFFSET + _SPEED_BOX_WIDTH / 2
    speed_cy = rect.y + _SPEED_BOX_Y_OFFSET + _SPEED_BOX_HEIGHT / 2

    dm_cx = rect.x + _DM_OFFSET
    dm_cy = rect.y + rect.height - _DM_OFFSET

    mid_x = (speed_cx + dm_cx) / 2
    mid_y = (speed_cy + dm_cy) / 2

    draw_x = int(mid_x - ICON_SIZE / 2)
    draw_y = int(mid_y - ICON_SIZE / 2)

    rl.draw_texture(self._icon, draw_x, draw_y, rl.WHITE)