mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-01 21:53:42 +08:00
FrogAi
3be51ac310
The openpilot theme changes according to the current/upcoming holiday. Minor holidays last a day, while major holidays (Easter, Christmas, Halloween, etc.) last a week.
73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
import datetime
|
|
|
|
from openpilot.common.params import Params
|
|
|
|
from openpilot.selfdrive.frogpilot.controls.lib.frogpilot_functions import update_frogpilot_toggles
|
|
|
|
class ThemeManager:
|
|
def __init__(self):
|
|
self.params_memory = Params("/dev/shm/params")
|
|
|
|
self.previous_theme_id = 0
|
|
|
|
@staticmethod
|
|
def calculate_easter(year):
|
|
a = year % 19
|
|
b = year // 100
|
|
c = year % 100
|
|
d = b // 4
|
|
e = b % 4
|
|
f = (b + 8) // 25
|
|
g = (b - f + 1) // 3
|
|
h = (19 * a + b - d - g + 15) % 30
|
|
i = c // 4
|
|
k = c % 4
|
|
l = (32 + 2 * e + 2 * i - h - k) % 7
|
|
m = (a + 11 * h + 22 * l) // 451
|
|
month = (h + l - 7 * m + 114) // 31
|
|
day = ((h + l - 7 * m + 114) % 31) + 1
|
|
return datetime.datetime(year, month, day)
|
|
|
|
@staticmethod
|
|
def calculate_thanksgiving(year):
|
|
november_first = datetime.datetime(year, 11, 1)
|
|
day_of_week = november_first.weekday()
|
|
return november_first + datetime.timedelta(days=(3 - day_of_week + 21) % 7 + 21)
|
|
|
|
@staticmethod
|
|
def is_within_week_of(target_date, current_date):
|
|
start_of_week = target_date - datetime.timedelta(days=target_date.weekday())
|
|
end_of_week = start_of_week + datetime.timedelta(days=6)
|
|
return start_of_week <= current_date <= end_of_week
|
|
|
|
def update_holiday(self):
|
|
current_date = datetime.datetime.now()
|
|
year = current_date.year
|
|
|
|
holidays = {
|
|
"april_fools": (datetime.datetime(year, 4, 1), 1),
|
|
"christmas_week": (datetime.datetime(year, 12, 25), 2),
|
|
"cinco_de_mayo": (datetime.datetime(year, 5, 5), 3),
|
|
"easter_week": (self.calculate_easter(year), 4),
|
|
"fourth_of_july": (datetime.datetime(year, 7, 4), 5),
|
|
"halloween_week": (datetime.datetime(year, 10, 31), 6),
|
|
"new_years": (datetime.datetime(year, 1, 1), 7),
|
|
"st_patricks": (datetime.datetime(year, 3, 17), 8),
|
|
"thanksgiving_week": (self.calculate_thanksgiving(year), 9),
|
|
"valentines": (datetime.datetime(year, 2, 14), 10),
|
|
"world_frog_day": (datetime.datetime(year, 3, 20), 11)
|
|
}
|
|
|
|
for holiday, (date, theme_id) in holidays.items():
|
|
if holiday.endswith("_week") and self.is_within_week_of(date, current_date) or current_date.date() == date.date():
|
|
if theme_id != self.previous_theme_id:
|
|
self.params_memory.put_int("CurrentHolidayTheme", theme_id)
|
|
update_frogpilot_toggles()
|
|
self.previous_theme_id = theme_id
|
|
return
|
|
|
|
if self.previous_theme_id != 0:
|
|
self.params_memory.put_int("CurrentHolidayTheme", "0")
|
|
update_frogpilot_toggles()
|
|
self.previous_theme_id = 0
|