mirror of
https://github.com/dekuNukem/USB4VC.git
synced 2025-10-31 11:26:46 -07:00
101 lines
3.6 KiB
Python
101 lines
3.6 KiB
Python
import os
|
|
import sys
|
|
import time
|
|
import json
|
|
import RPi.GPIO as GPIO
|
|
import usb4vc_usb_scan
|
|
import usb4vc_ui
|
|
import subprocess
|
|
|
|
# usb4vc_ui.reset_pboard()
|
|
|
|
def get_current_rpi_model():
|
|
current_model = 'Unknown'
|
|
try:
|
|
current_model = subprocess.getoutput("cat /proc/device-tree/model").replace('\n', '').replace('\r', '').strip()
|
|
except Exception as e:
|
|
print('get_current_rpi_model:', e)
|
|
return current_model
|
|
|
|
rpi_model_save_file = '/home/pi/rpi_model.txt'
|
|
|
|
def get_stored_rpi_model():
|
|
stored_model = 'Unknown'
|
|
try:
|
|
with open(rpi_model_save_file, 'r') as file:
|
|
stored_model = file.read().replace('\n', '').replace('\r', '').strip()
|
|
except Exception as e:
|
|
print('get_stored_rpi_model:', e)
|
|
return stored_model
|
|
|
|
def save_rpi_model(model_str):
|
|
try:
|
|
with open(rpi_model_save_file, 'w') as file:
|
|
file.write(model_str)
|
|
except Exception as e:
|
|
print('save_rpi_model:', e)
|
|
|
|
def check_rpi_model():
|
|
current_model = get_current_rpi_model()
|
|
stored_model = get_stored_rpi_model()
|
|
print('current_model', current_model)
|
|
print('stored_model', stored_model)
|
|
|
|
if "Raspberry Pi 5".lower() in current_model.lower():
|
|
usb4vc_ui.oled_print_model_changed()
|
|
os.system("cd /home/pi/xpadneo/; sudo ./uninstall.sh")
|
|
save_rpi_model(current_model)
|
|
return
|
|
elif current_model != stored_model:
|
|
usb4vc_ui.oled_print_model_changed()
|
|
for x in range(20):
|
|
print("!!!!!!!!!! DO NOT UNPLUG UNTIL I REBOOT !!!!!!!!!!")
|
|
print("!!!!!!!!!! DO NOT UNPLUG UNTIL I REBOOT !!!!!!!!!!")
|
|
print("!!!!!!!!!! DO NOT UNPLUG UNTIL I REBOOT !!!!!!!!!!")
|
|
time.sleep(0.1)
|
|
os.system("cd /home/pi/xpadneo/; sudo ./uninstall.sh")
|
|
usb4vc_ui.oled_print_oneline("Recompiling...")
|
|
os.system("cd /home/pi/xpadneo/; sudo ./install.sh")
|
|
usb4vc_ui.oled_print_reboot()
|
|
save_rpi_model(current_model)
|
|
time.sleep(2)
|
|
os.system('sudo reboot')
|
|
time.sleep(10)
|
|
|
|
check_rpi_model()
|
|
|
|
cmdline_path = "/boot/firmware/cmdline.txt"
|
|
try:
|
|
boot_config = os.popen(f'cat {cmdline_path}').read()
|
|
if "usbhid.mousepoll".lower() not in boot_config.lower():
|
|
print(f"@@@@@@@@@@@@ writing usbhid.mousepoll to {cmdline_path} @@@@@@@@@@@@")
|
|
new_config = boot_config.replace('\r', '').replace('\n', '').strip() + ' usbhid.mousepoll=4\n'
|
|
with open(cmdline_path, 'w') as ffff:
|
|
ffff.write(new_config)
|
|
elif "usbhid.mousepoll=0" in boot_config:
|
|
print(f"@@@@@@@@@@@@ writing usbhid.mousepoll to {cmdline_path} @@@@@@@@@@@@")
|
|
new_config = boot_config.replace('\r', '').replace('\n', '').strip().replace("usbhid.mousepoll=0", "usbhid.mousepoll=4")
|
|
with open(cmdline_path, 'w') as ffff:
|
|
ffff.write(new_config)
|
|
except Exception as e:
|
|
print('usbhid.mousepoll exception:', e)
|
|
|
|
usb4vc_ui.ui_init()
|
|
usb4vc_ui.ui_thread.start()
|
|
|
|
usb4vc_usb_scan.usb_device_scan_thread.start()
|
|
usb4vc_usb_scan.raw_input_event_parser_thread.start()
|
|
|
|
while 1:
|
|
time.sleep(2)
|
|
if os.path.exists("/sys/module/bluetooth/parameters/disable_ertm"):
|
|
try:
|
|
ertm_status = subprocess.getoutput("cat /sys/module/bluetooth/parameters/disable_ertm").replace('\n', '').replace('\r', '').strip()
|
|
if ertm_status != 'Y':
|
|
# print('ertm_status:', ertm_status)
|
|
# print("Disabling ERTM....")
|
|
subprocess.call('echo Y > /sys/module/bluetooth/parameters/disable_ertm')
|
|
# print("DONE")
|
|
except Exception:
|
|
continue
|