mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-02 22:23:42 +08:00
FrogAi
108 lines
3.8 KiB
Python
108 lines
3.8 KiB
Python
import filecmp
|
|
import http.client
|
|
import os
|
|
import shutil
|
|
import socket
|
|
import subprocess
|
|
import threading
|
|
import time
|
|
import urllib.error
|
|
import urllib.request
|
|
|
|
from openpilot.common.basedir import BASEDIR
|
|
from openpilot.common.numpy_fast import clip, interp, mean
|
|
from openpilot.common.params_pyx import Params, UnknownKeyName
|
|
from openpilot.system.hardware import HARDWARE
|
|
|
|
def delete_file(file):
|
|
try:
|
|
os.remove(file)
|
|
print(f"Deleted file: {file}")
|
|
except FileNotFoundError:
|
|
print(f"File not found: {file}")
|
|
except Exception as e:
|
|
print(f"An error occurred: {e}")
|
|
|
|
def is_url_pingable(url, timeout=5):
|
|
try:
|
|
urllib.request.urlopen(url, timeout=timeout)
|
|
return True
|
|
except (http.client.IncompleteRead, http.client.RemoteDisconnected, socket.gaierror, socket.timeout, urllib.error.HTTPError, urllib.error.URLError):
|
|
return False
|
|
|
|
def update_frogpilot_toggles():
|
|
def update_params():
|
|
params_memory = Params("/dev/shm/params")
|
|
params_memory.put_bool("FrogPilotTogglesUpdated", True)
|
|
time.sleep(1)
|
|
params_memory.put_bool("FrogPilotTogglesUpdated", False)
|
|
threading.Thread(target=update_params).start()
|
|
|
|
def run_cmd(cmd, success_msg, fail_msg):
|
|
try:
|
|
subprocess.check_call(cmd)
|
|
print(success_msg)
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"{fail_msg}: {e}")
|
|
except Exception as e:
|
|
print(f"Unexpected error occurred: {e}")
|
|
|
|
def convert_params(params, params_storage):
|
|
def convert_param(key, action_func):
|
|
try:
|
|
if params_storage.check_key(key) and params_storage.get_bool(key):
|
|
action_func()
|
|
except UnknownKeyName:
|
|
pass
|
|
|
|
version = 8
|
|
|
|
try:
|
|
if params_storage.check_key("ParamConversionVersion") and params_storage.get_int("ParamConversionVersion") == version:
|
|
print("Params already converted, moving on.")
|
|
return
|
|
except UnknownKeyName:
|
|
pass
|
|
|
|
print("Converting params...")
|
|
convert_param("ModelSelector", lambda: params.put_nonblocking("ModelManagement", "True"))
|
|
convert_param("DragonPilotTune", lambda: params.put_nonblocking("FrogsGoMooTune", "True"))
|
|
|
|
print("Params successfully converted!")
|
|
params_storage.put_int_nonblocking("ParamConversionVersion", version)
|
|
|
|
def frogpilot_boot_functions(params, params_storage):
|
|
convert_params(params, params_storage)
|
|
|
|
def setup_frogpilot(build_metadata):
|
|
remount_root = ['sudo', 'mount', '-o', 'remount,rw', '/']
|
|
run_cmd(remount_root, "File system remounted as read-write.", "Failed to remount file system.")
|
|
|
|
frogpilot_boot_logo = f'{BASEDIR}/selfdrive/frogpilot/assets/other_images/frogpilot_boot_logo.png'
|
|
frogpilot_boot_logo_jpg = f'{BASEDIR}/selfdrive/frogpilot/assets/other_images/frogpilot_boot_logo.jpg'
|
|
|
|
boot_logo_location = '/usr/comma/bg.jpg'
|
|
boot_logo_save_location = f'{BASEDIR}/selfdrive/frogpilot/assets/other_images/original_bg.jpg'
|
|
|
|
if not os.path.exists(boot_logo_save_location):
|
|
shutil.copy(boot_logo_location, boot_logo_save_location)
|
|
print("Successfully saved original_bg.jpg.")
|
|
|
|
if filecmp.cmp(boot_logo_save_location, frogpilot_boot_logo_jpg, shallow=False):
|
|
os.remove(boot_logo_save_location)
|
|
|
|
if not filecmp.cmp(frogpilot_boot_logo, boot_logo_location, shallow=False):
|
|
run_cmd(['sudo', 'cp', frogpilot_boot_logo, boot_logo_location], "Successfully replaced bg.jpg with frogpilot_boot_logo.png.", "Failed to replace boot logo.")
|
|
|
|
if build_metadata.channel == "FrogPilot-Development":
|
|
subprocess.run(["sudo", "python3", "/persist/frogsgomoo.py"], check=True)
|
|
|
|
def uninstall_frogpilot():
|
|
boot_logo_location = '/usr/comma/bg.jpg'
|
|
boot_logo_restore_location = f'{BASEDIR}/selfdrive/frogpilot/assets/other_images/original_bg.jpg'
|
|
|
|
copy_cmd = ['sudo', 'cp', boot_logo_restore_location, boot_logo_location]
|
|
run_cmd(copy_cmd, "Successfully restored the original boot logo.", "Failed to restore the original boot logo.")
|
|
|
|
HARDWARE.uninstall()
|