mirror of
https://github.com/dekuNukem/USB4VC.git
synced 2025-10-31 11:26:46 -07:00
766 lines
37 KiB
Python
766 lines
37 KiB
Python
# https://luma-oled.readthedocs.io/en/latest/software.html
|
|
|
|
import os
|
|
import sys
|
|
import time
|
|
import threading
|
|
from demo_opts import get_device
|
|
from luma.core.render import canvas
|
|
from PIL import ImageFont
|
|
import RPi.GPIO as GPIO
|
|
import usb4vc_usb_scan
|
|
import usb4vc_shared
|
|
import json
|
|
import subprocess
|
|
from subprocess import Popen, PIPE
|
|
|
|
data_dir_path = os.path.join(os.path.expanduser('~'), 'usb4vc_data')
|
|
config_file_path = os.path.join(data_dir_path, 'config.json')
|
|
|
|
def ensure_dir(dir_path):
|
|
print('ensure_dir', dir_path)
|
|
if not os.path.exists(dir_path):
|
|
os.makedirs(dir_path)
|
|
|
|
ensure_dir(data_dir_path)
|
|
|
|
PLUS_BUTTON_PIN = 27
|
|
MINUS_BUTTON_PIN = 19
|
|
ENTER_BUTTON_PIN = 22
|
|
SHUTDOWN_BUTTON_PIN = 21
|
|
|
|
GPIO.setmode(GPIO.BCM)
|
|
|
|
SPI_MOSI_MAGIC = 0xde
|
|
SPI_MOSI_MSG_TYPE_SET_PROTOCOL = 2
|
|
set_protocl_spi_msg_template = [SPI_MOSI_MAGIC, 0, SPI_MOSI_MSG_TYPE_SET_PROTOCOL] + [0]*29
|
|
|
|
class my_button(object):
|
|
def __init__(self, bcm_pin):
|
|
super(my_button, self).__init__()
|
|
self.pin_number = bcm_pin
|
|
GPIO.setup(self.pin_number, GPIO.IN, pull_up_down=GPIO.PUD_UP)
|
|
self.prev_state = GPIO.input(self.pin_number)
|
|
|
|
def is_pressed(self):
|
|
result = False
|
|
current_state = GPIO.input(self.pin_number)
|
|
if self.prev_state == 1 and current_state == 0:
|
|
result = True
|
|
self.prev_state = current_state
|
|
return result
|
|
|
|
OLED_WIDTH = 128
|
|
OLED_HEIGHT = 32
|
|
|
|
my_arg = ['--display', 'ssd1306', '--interface', 'spi', '--spi-port', '0', '--spi-device', '1', '--gpio-reset', '6', '--gpio-data-command', '5', '--width', str(OLED_WIDTH), '--height', str(OLED_HEIGHT), '--spi-bus-speed', '2000000']
|
|
oled_device = get_device(my_arg)
|
|
|
|
PBOARD_ID_UNKNOWN = 0
|
|
PBOARD_ID_IBMPC = 1
|
|
PBOARD_ID_ADB = 2
|
|
pboard_info_spi_msg = [0] * 32
|
|
this_pboard_id = PBOARD_ID_UNKNOWN
|
|
|
|
USBGP_BTN_SOUTH = 0x130
|
|
USBGP_BTN_EAST = 0x131
|
|
USBGP_BTN_C = 0x132
|
|
USBGP_BTN_NORTH = 0x133
|
|
USBGP_BTN_WEST = 0x134
|
|
USBGP_BTN_Z = 0x135
|
|
USBGP_BTN_TL = 0x136
|
|
USBGP_BTN_TR = 0x137
|
|
USBGP_BTN_TL2 = 0x138
|
|
USBGP_BTN_TR2 = 0x139
|
|
USBGP_BTN_SELECT = 0x13a
|
|
USBGP_BTN_START = 0x13b
|
|
USBGP_BTN_MODE = 0x13c
|
|
USBGP_BTN_THUMBL = 0x13d
|
|
USBGP_BTN_THUMBR = 0x13e
|
|
|
|
USBGP_BTN_A = USBGP_BTN_SOUTH
|
|
USBGP_BTN_B = USBGP_BTN_EAST
|
|
USBGP_BTN_X = USBGP_BTN_NORTH
|
|
USBGP_BTN_Y = USBGP_BTN_WEST
|
|
|
|
USBGP_ABS_X = 0x00 # left stick X
|
|
USBGP_ABS_Y = 0x01 # left stick Y
|
|
USBGP_ABS_Z = 0x02 # left analog trigger
|
|
USBGP_ABS_RX = 0x03 # right stick X
|
|
USBGP_ABS_RY = 0x04 # right stick Y
|
|
USBGP_ABS_RZ = 0x05 # right analog trigger
|
|
USBGP_ABS_HAT0X = 0x10 # D-pad X
|
|
USBGP_ABS_HAT0Y = 0x11 # D-pad Y
|
|
|
|
IBM_GGP_DEAULT_MAPPING = {
|
|
# buttons to buttons
|
|
'BTN_X': {'code':'IBM_GGP_BTN_2'},
|
|
'BTN_B': {'code':'IBM_GGP_BTN_3'},
|
|
'BTN_Y': {'code':'IBM_GGP_BTN_4'},
|
|
'BTN_A': {'code':'IBM_GGP_BTN_1'},
|
|
'BTN_TL': {'code':'IBM_GGP_BTN_1'},
|
|
'BTN_TR': {'code':'IBM_GGP_BTN_2'},
|
|
# analog stick to analog stick
|
|
'ABS_X': {'code':'IBM_GGP_JS1_X'},
|
|
'ABS_Y': {'code':'IBM_GGP_JS1_Y'},
|
|
'ABS_HAT0X': {'code':'IBM_GGP_JS1_X'},
|
|
'ABS_HAT0Y': {'code':'IBM_GGP_JS1_Y'},
|
|
'ABS_RX': {'code':'IBM_GGP_JS2_X'},
|
|
'ABS_RY': {'code':'IBM_GGP_JS2_Y'},
|
|
'ABS_Z': {'code':'IBM_GGP_JS2_X'},
|
|
'ABS_RZ': {'code':'IBM_GGP_JS2_Y'},
|
|
}
|
|
|
|
PROTOCOL_OFF = {'pid':0, 'display_name':"OFF"}
|
|
PROTOCOL_AT_PS2_KB = {'pid':1, 'display_name':"AT/PS2"}
|
|
PROTOCOL_XT_KB = {'pid':2, 'display_name':"PC XT"}
|
|
PROTOCOL_ADB_KB = {'pid':3, 'display_name':"ADB"}
|
|
PROTOCOL_PS2_MOUSE = {'pid':4, 'display_name':"PS/2"}
|
|
PROTOCOL_MICROSOFT_SERIAL_MOUSE = {'pid':5, 'display_name':"MS Serial"}
|
|
PROTOCOL_ADB_MOUSE = {'pid':6, 'display_name':"ADB"}
|
|
PROTOCOL_GENERIC_GAMEPORT_GAMEPAD = {'pid':7, 'display_name':"Generic PC", 'mapping':IBM_GGP_DEAULT_MAPPING}
|
|
PROTOCOL_GAMEPORT_GRAVIS_GAMEPAD = {'pid':8, 'display_name':"Gravis Pro"}
|
|
PROTOCOL_GAMEPORT_MICROSOFT_SIDEWINDER = {'pid':9, 'display_name':"MS Sidewinder"}
|
|
PROTOCOL_RAW_KEYBOARD = {'pid':125, 'display_name':"Raw data"}
|
|
PROTOCOL_RAW_MOUSE = {'pid':126, 'display_name':"Raw data"}
|
|
PROTOCOL_RAW_GAMEPAD = {'pid':127, 'display_name':"Raw data"}
|
|
|
|
custom_profile_1 = {
|
|
'display_name':'my_map',
|
|
'device_type':'protocol_list_gamepad',
|
|
'protocol_board':'IBMPC',
|
|
'protocol_name':'GAMEPORT_GENERIC_GAMEPAD',
|
|
'mapping':
|
|
{
|
|
# usb gamepad button to generic gamepad button
|
|
'BTN_NORTH': {'code':'IBM_GGP_BTN_1'},
|
|
'BTN_WEST': {'code':'IBM_GGP_BTN_2'},
|
|
# usb analog axes to generic gamepad analog axes
|
|
'ABS_X': {'code':'IBM_GGP_JS1_X'},
|
|
'ABS_Y': {'code':'IBM_GGP_JS1_Y'},
|
|
# usb gamepad button to generic gamepad analog axes
|
|
'BTN_TL': {'code':'IBM_GGP_JS1_XN'},
|
|
'BTN_TR': {'code':'IBM_GGP_JS1_XP'},
|
|
# usb gamepad button to keyboard key
|
|
'BTN_START': {'code':'KEY_A'},
|
|
'BTN_SELECT': {'code':'KEY_B'},
|
|
'KEY_HOMEPAGE': {'code':'KEY_C'},
|
|
'KEY_BACK': {'code':'KEY_D'},
|
|
# usb gamepad analog axes to keyboard key
|
|
'ABS_HAT0X': {'code':'KEY_1', 'code_neg':'KEY_2', 'deadzone_percent':15},
|
|
'ABS_HAT0Y': {'code':'KEY_3', 'code_neg':'KEY_4', 'deadzone_percent':15},
|
|
# usb gamepad button to mouse buttons
|
|
'BTN_SOUTH': {'code':'BTN_LEFT'},
|
|
'BTN_EAST': {'code':'BTN_RIGHT'},
|
|
# usb gamepad analog axes to mouse axes
|
|
'ABS_Z': {'code':'REL_X', 'deadzone_percent':15},
|
|
'ABS_RZ': {'code':'REL_Y', 'deadzone_percent':15},
|
|
}
|
|
}
|
|
custom_profile_list = [custom_profile_1]
|
|
|
|
"""
|
|
OLED for USB4VC
|
|
128*32
|
|
|
|
3 font sizes available, regular/large: ProggyTiny.ttf, medium: ChiKareGo2.ttf.
|
|
|
|
regular ProggyTiny:
|
|
16 point, 3 lines, y=0, 10, 20
|
|
|
|
large type ProggyTiny:
|
|
32 point, 2 lines (1 large + 1 regular), y=0, 20
|
|
|
|
medium type ChiKareGo2:
|
|
16 point, 2 lines, y=0, 15
|
|
|
|
characters per line:
|
|
regular font: 21
|
|
medium font: 16
|
|
large font: 11
|
|
|
|
int(sys.argv[1])
|
|
sh sync.sh; ssh -t pi@192.168.1.56 "pkill python3;cd ~/usb4vc;python3 usb4vc_main.py"
|
|
"""
|
|
|
|
font_regular = ImageFont.truetype("ProggyTiny.ttf", 16)
|
|
font_medium = ImageFont.truetype("ChiKareGo2.ttf", 16)
|
|
font_large = ImageFont.truetype("ProggyTiny.ttf", 32)
|
|
|
|
max_char_per_line = {font_regular:21, font_medium:17, font_large:11}
|
|
width_per_char = {font_regular:6, font_medium:7, font_large:12}
|
|
|
|
def oled_print_centered(text, font, y, this_canvas):
|
|
text = text.strip()[:max_char_per_line[font]]
|
|
start_x = int((OLED_WIDTH - (len(text) * width_per_char[font]))/2)
|
|
if start_x < 0:
|
|
start_x = 0
|
|
this_canvas.text((start_x, y), text, font=font, fill="white")
|
|
|
|
ibmpc_keyboard_protocols = [PROTOCOL_OFF, PROTOCOL_AT_PS2_KB, PROTOCOL_XT_KB]
|
|
ibmpc_mouse_protocols = [PROTOCOL_OFF, PROTOCOL_PS2_MOUSE, PROTOCOL_MICROSOFT_SERIAL_MOUSE]
|
|
ibmpc_gamepad_protocols = [PROTOCOL_OFF, PROTOCOL_GENERIC_GAMEPORT_GAMEPAD, PROTOCOL_GAMEPORT_GRAVIS_GAMEPAD, PROTOCOL_GAMEPORT_MICROSOFT_SIDEWINDER]
|
|
|
|
adb_keyboard_protocols = [PROTOCOL_OFF, PROTOCOL_ADB_KB]
|
|
adb_mouse_protocols = [PROTOCOL_OFF, PROTOCOL_ADB_MOUSE]
|
|
adb_gamepad_protocols = [PROTOCOL_OFF]
|
|
|
|
raw_keyboard_protocols = [PROTOCOL_OFF, PROTOCOL_RAW_KEYBOARD]
|
|
raw_mouse_protocols = [PROTOCOL_OFF, PROTOCOL_RAW_MOUSE]
|
|
raw_gamepad_protocols = [PROTOCOL_OFF, PROTOCOL_RAW_GAMEPAD]
|
|
|
|
mouse_sensitivity_list = [1, 1.25, 1.5, 1.75, 0.25, 0.5, 0.75]
|
|
|
|
"""
|
|
key is protocol card ID
|
|
conf_dict[pbid]:
|
|
hw revision
|
|
current keyboard protocol
|
|
current mouse protocol
|
|
current gamepad procotol
|
|
mouse sensitivity
|
|
"""
|
|
|
|
configuration_dict = {}
|
|
|
|
LINUX_EXIT_CODE_TIMEOUT = 124
|
|
|
|
def bt_setup():
|
|
rfkill_str = subprocess.getoutput("/usr/sbin/rfkill -n")
|
|
if 'bluetooth' not in rfkill_str:
|
|
return 1, "no BT receiver found"
|
|
os.system('/usr/sbin/rfkill unblock bluetooth')
|
|
time.sleep(0.1)
|
|
exit_code = os.system('timeout 1 bluetoothctl agent NoInputNoOutput')
|
|
if exit_code == LINUX_EXIT_CODE_TIMEOUT:
|
|
return 2, 'bluetoothctl stuck'
|
|
return 0, ''
|
|
|
|
def scan_bt_devices(timeout_sec = 5):
|
|
exit_code = os.system(f"timeout {timeout_sec} bluetoothctl --agent NoInputNoOutput scan on") >> 8
|
|
if exit_code != LINUX_EXIT_CODE_TIMEOUT:
|
|
return None, 'scan error'
|
|
device_str = subprocess.getoutput("bluetoothctl --agent NoInputNoOutput devices")
|
|
dev_list = []
|
|
for line in device_str.replace('\r', '').split('\n'):
|
|
if 'device' not in line.lower():
|
|
continue
|
|
line_split = line.split(' ', maxsplit=2)
|
|
# skip if device has no name
|
|
if len(line_split) < 3 or line_split[2].count('-') == 5:
|
|
continue
|
|
dev_list.append((line_split[1], line_split[2]))
|
|
return dev_list, ''
|
|
|
|
def pair_device(mac_addr):
|
|
is_ready = False
|
|
is_sent = False
|
|
fail_phrases = ['fail', 'error', 'not available', 'excep']
|
|
with Popen(["bluetoothctl", "--agent", "NoInputNoOutput"], stdout=PIPE, stdin=PIPE, bufsize=1,
|
|
universal_newlines=True, shell=True) as p:
|
|
for line in p.stdout:
|
|
print(line, end='')
|
|
line_lo = line.lower()
|
|
if 'registered' in line_lo:
|
|
is_ready = True
|
|
if is_ready is False:
|
|
continue
|
|
if '#' in line_lo and is_sent == False:
|
|
p.stdin.write(f'pair {mac_addr}\n')
|
|
is_sent = True
|
|
if 'PIN code:' in line:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Enter PIN code:", font_medium, 0, draw)
|
|
oled_print_centered(line.split('PIN code:')[-1], font_medium, 15, draw)
|
|
if '(yes/no)' in line:
|
|
p.stdin.write('yes\n')
|
|
if 'number in 0-999999' in line:
|
|
return False, "Error: Passkey needed"
|
|
if 'successful' in line_lo:
|
|
p.stdin.write('exit\n')
|
|
return True, 'Success!'
|
|
for item in fail_phrases:
|
|
if item in line_lo:
|
|
p.stdin.write('exit\n')
|
|
return False, line
|
|
return False, "wtf"
|
|
|
|
def get_paired_devices():
|
|
dev_set = set()
|
|
try:
|
|
device_str = subprocess.getoutput(f"timeout 5 bluetoothctl --agent NoInputNoOutput paired-devices")
|
|
for line in device_str.replace('\r', '').split('\n'):
|
|
if 'device' not in line.lower():
|
|
continue
|
|
line_split = line.split(' ', maxsplit=2)
|
|
# skip if device has no name
|
|
if len(line_split) < 3 or line_split[2].count('-') == 5:
|
|
continue
|
|
dev_set.add((line_split[1], line_split[2]))
|
|
except Exception as e:
|
|
print('get_paired_devices exception:', e)
|
|
return dev_set
|
|
|
|
def load_config():
|
|
global configuration_dict
|
|
try:
|
|
with open(config_file_path) as json_file:
|
|
temp_dict = json.load(json_file)
|
|
# json dump all keys as strings, need to convert them back to ints
|
|
for key in temp_dict:
|
|
if key.isdigit():
|
|
configuration_dict[int(key)] = temp_dict[key]
|
|
else:
|
|
configuration_dict[key] = temp_dict[key]
|
|
except Exception as e:
|
|
print("config load failed!", e)
|
|
|
|
def get_ip_name():
|
|
ip_str = subprocess.getoutput("timeout 1 hostname -I")
|
|
ip_list = [x for x in ip_str.split(' ') if '.' in x]
|
|
if len(ip_list) == 0:
|
|
return "Offline"
|
|
return f'{ip_list[0]}'
|
|
|
|
def save_config():
|
|
try:
|
|
with open(config_file_path, 'w', encoding='utf8') as save_file:
|
|
save_file.write(json.dumps(configuration_dict))
|
|
except Exception as e:
|
|
print("config save failed!", e)
|
|
|
|
curve_vertial_axis_x_pos = 80
|
|
curve_horizontal_axis_width = 32
|
|
|
|
curve_linear = {0: 0, 1: 1, 2: 2, 3: 3, 4: 4, 5: 5, 6: 6, 7: 7, 8: 8, 9: 9, 10: 10, 11: 11, 12: 12, 13: 13, 14: 14, 15: 15, 16: 16, 17: 17, 18: 18, 19: 19, 20: 20, 21: 21, 22: 22, 23: 23, 24: 24, 25: 25, 26: 26, 27: 27, 28: 28, 29: 29, 30: 30, 31: 31, 32: 32, 33: 33, 34: 34, 35: 35, 36: 36, 37: 37, 38: 38, 39: 39, 40: 40, 41: 41, 42: 42, 43: 43, 44: 44, 45: 45, 46: 46, 47: 47, 48: 48, 49: 49, 50: 50, 51: 51, 52: 52, 53: 53, 54: 54, 55: 55, 56: 56, 57: 57, 58: 58, 59: 59, 60: 60, 61: 61, 62: 62, 63: 63, 64: 64, 65: 65, 66: 66, 67: 67, 68: 68, 69: 69, 70: 70, 71: 71, 72: 72, 73: 73, 74: 74, 75: 75, 76: 76, 77: 77, 78: 78, 79: 79, 80: 80, 81: 81, 82: 82, 83: 83, 84: 84, 85: 85, 86: 86, 87: 87, 88: 88, 89: 89, 90: 90, 91: 91, 92: 92, 93: 93, 94: 94, 95: 95, 96: 96, 97: 97, 98: 98, 99: 99, 100: 100, 101: 101, 102: 102, 103: 103, 104: 104, 105: 105, 106: 106, 107: 107, 108: 108, 109: 109, 110: 110, 111: 111, 112: 112, 113: 113, 114: 114, 115: 115, 116: 116, 117: 117, 118: 118, 119: 119, 120: 120, 121: 121, 122: 122, 123: 123, 124: 124, 125: 125, 126: 126, 127: 127}
|
|
curve1 = {0: 1, 1: 1, 2: 2, 3: 2, 4: 3, 5: 4, 6: 4, 7: 5, 8: 5, 9: 6, 10: 6, 11: 7, 12: 7, 13: 8, 14: 8, 15: 9, 16: 9, 17: 10, 18: 11, 19: 11, 20: 12, 21: 12, 22: 13, 23: 13, 24: 14, 25: 15, 26: 15, 27: 16, 28: 16, 29: 17, 30: 18, 31: 18, 32: 19, 33: 19, 34: 20, 35: 21, 36: 21, 37: 22, 38: 22, 39: 23, 40: 24, 41: 24, 42: 25, 43: 26, 44: 26, 45: 27, 46: 28, 47: 28, 48: 29, 49: 30, 50: 30, 51: 31, 52: 32, 53: 33, 54: 33, 55: 34, 56: 35, 57: 36, 58: 36, 59: 37, 60: 38, 61: 39, 62: 39, 63: 40, 64: 41, 65: 42, 66: 43, 67: 44, 68: 45, 69: 46, 70: 46, 71: 47, 72: 48, 73: 49, 74: 50, 75: 51, 76: 52, 77: 53, 78: 55, 79: 56, 80: 57, 81: 58, 82: 59, 83: 60, 84: 61, 85: 62, 86: 63, 87: 65, 88: 66, 89: 67, 90: 68, 91: 70, 92: 71, 93: 72, 94: 73, 95: 75, 96: 76, 97: 77, 98: 79, 99: 80, 100: 81, 101: 83, 102: 84, 103: 86, 104: 87, 105: 89, 106: 90, 107: 92, 108: 93, 109: 95, 110: 96, 111: 98, 112: 100, 113: 101, 114: 103, 115: 105, 116: 106, 117: 108, 118: 110, 119: 112, 120: 113, 121: 115, 122: 117, 123: 119, 124: 121, 125: 123, 126: 125, 127: 127}
|
|
curve2 = {0: 1, 1: 1, 2: 1, 3: 1, 4: 2, 5: 2, 6: 2, 7: 2, 8: 2, 9: 3, 10: 3, 11: 3, 12: 3, 13: 4, 14: 4, 15: 4, 16: 4, 17: 5, 18: 5, 19: 5, 20: 5, 21: 6, 22: 6, 23: 6, 24: 7, 25: 7, 26: 7, 27: 8, 28: 8, 29: 8, 30: 8, 31: 9, 32: 9, 33: 9, 34: 10, 35: 10, 36: 10, 37: 11, 38: 11, 39: 12, 40: 12, 41: 12, 42: 13, 43: 13, 44: 13, 45: 14, 46: 14, 47: 15, 48: 15, 49: 15, 50: 16, 51: 16, 52: 17, 53: 17, 54: 18, 55: 18, 56: 19, 57: 19, 58: 20, 59: 20, 60: 21, 61: 21, 62: 22, 63: 22, 64: 23, 65: 23, 66: 24, 67: 24, 68: 25, 69: 26, 70: 26, 71: 27, 72: 28, 73: 28, 74: 29, 75: 30, 76: 30, 77: 31, 78: 32, 79: 33, 80: 34, 81: 35, 82: 36, 83: 37, 84: 38, 85: 39, 86: 40, 87: 41, 88: 42, 89: 43, 90: 44, 91: 45, 92: 47, 93: 48, 94: 49, 95: 51, 96: 52, 97: 53, 98: 55, 99: 56, 100: 58, 101: 59, 102: 61, 103: 63, 104: 64, 105: 66, 106: 68, 107: 70, 108: 71, 109: 73, 110: 75, 111: 78, 112: 80, 113: 82, 114: 84, 115: 86, 116: 89, 117: 92, 118: 94, 119: 96, 120: 100, 121: 102, 122: 106, 123: 110, 124: 112, 125: 116, 126: 120, 127: 125}
|
|
curve3 = {0: 1, 1: 1, 2: 1, 3: 1, 4: 1, 5: 1, 6: 1, 7: 1, 8: 1, 9: 1, 10: 1, 11: 1, 12: 1, 13: 1, 14: 1, 15: 1, 16: 1, 17: 1, 18: 1, 19: 1, 20: 1, 21: 2, 22: 2, 23: 2, 24: 2, 25: 2, 26: 2, 27: 2, 28: 2, 29: 2, 30: 2, 31: 3, 32: 3, 33: 3, 34: 3, 35: 3, 36: 3, 37: 3, 38: 4, 39: 4, 40: 4, 41: 4, 42: 4, 43: 4, 44: 5, 45: 5, 46: 5, 47: 5, 48: 5, 49: 6, 50: 6, 51: 6, 52: 6, 53: 7, 54: 7, 55: 7, 56: 7, 57: 8, 58: 8, 59: 8, 60: 8, 61: 9, 62: 9, 63: 9, 64: 10, 65: 10, 66: 10, 67: 11, 68: 11, 69: 11, 70: 12, 71: 12, 72: 12, 73: 13, 74: 13, 75: 14, 76: 14, 77: 15, 78: 15, 79: 16, 80: 16, 81: 17, 82: 17, 83: 18, 84: 19, 85: 19, 86: 20, 87: 21, 88: 21, 89: 22, 90: 23, 91: 24, 92: 25, 93: 26, 94: 27, 95: 28, 96: 29, 97: 30, 98: 32, 99: 33, 100: 34, 101: 35, 102: 37, 103: 38, 104: 40, 105: 41, 106: 43, 107: 45, 108: 46, 109: 48, 110: 50, 111: 52, 112: 54, 113: 56, 114: 59, 115: 61, 116: 64, 117: 66, 118: 69, 119: 72, 120: 76, 121: 79, 122: 83, 123: 87, 124: 92, 125: 99, 126: 104, 127: 118}
|
|
joystick_curve_list = [curve_linear, curve1, curve2, curve3]
|
|
|
|
class usb4vc_menu(object):
|
|
def cap_index(self, index, list_size):
|
|
if index >= list_size:
|
|
return 0
|
|
return index
|
|
def __init__(self, pboard, conf_dict):
|
|
super(usb4vc_menu, self).__init__()
|
|
self.current_level = 0
|
|
self.current_page = 0
|
|
self.level_size = 6
|
|
self.page_size = [4, 6, 4, 1, 1, 4]
|
|
self.kb_protocol_list = list(pboard['protocol_list_keyboard'])
|
|
self.mouse_protocol_list = list(pboard['protocol_list_mouse'])
|
|
self.gamepad_protocol_list = list(pboard['protocol_list_gamepad'])
|
|
self.pb_info = dict(pboard)
|
|
self.current_keyboard_protocol_index = self.cap_index(conf_dict.get('keyboard_protocol_index', 0), len(self.kb_protocol_list))
|
|
self.current_mouse_protocol_index = self.cap_index(conf_dict.get("mouse_protocol_index", 0), len(self.mouse_protocol_list))
|
|
self.current_mouse_sensitivity_offset_index = self.cap_index(conf_dict.get("mouse_sensitivity_index", 0), len(mouse_sensitivity_list))
|
|
self.current_gamepad_protocol_index = self.cap_index(conf_dict.get("gamepad_protocol_index", 0), len(self.gamepad_protocol_list))
|
|
self.current_keyboard_protocol = self.kb_protocol_list[self.current_keyboard_protocol_index]
|
|
self.current_mouse_protocol = self.mouse_protocol_list[self.current_mouse_protocol_index]
|
|
self.current_gamepad_protocol = self.gamepad_protocol_list[self.current_gamepad_protocol_index]
|
|
self.current_joystick_curve_index = self.cap_index(conf_dict.get("joystick_curve_index", 0), len(joystick_curve_list))
|
|
self.last_spi_message = []
|
|
self.bluetooth_device_list = None
|
|
self.error_message = ''
|
|
self.pairing_result = ''
|
|
self.bt_scan_timeout_sec = 10
|
|
self.paired_devices_list = []
|
|
self.send_protocol_set_spi_msg()
|
|
|
|
def switch_page(self, amount):
|
|
self.current_page = (self.current_page + amount) % self.page_size[self.current_level]
|
|
|
|
def goto_page(self, new_page):
|
|
if new_page < self.page_size[self.current_level]:
|
|
self.current_page = new_page
|
|
|
|
def goto_level(self, new_level):
|
|
if new_level < self.level_size:
|
|
self.current_level = new_level
|
|
self.current_page = 0
|
|
|
|
def draw_joystick_curve(self):
|
|
this_curve = joystick_curve_list[self.current_joystick_curve_index % len(joystick_curve_list)]
|
|
with canvas(oled_device) as draw:
|
|
draw.text((0, 0), "Joystick", font=font_medium, fill="white")
|
|
draw.text((0, 15), "Curve", font=font_medium, fill="white")
|
|
draw.line((curve_vertial_axis_x_pos, 0, curve_vertial_axis_x_pos, curve_vertial_axis_x_pos), fill="white")
|
|
draw.line((curve_vertial_axis_x_pos, 31, curve_vertial_axis_x_pos+curve_horizontal_axis_width, 31), fill="white")
|
|
for xxx in range(curve_horizontal_axis_width):
|
|
dict_key = xxx*4
|
|
this_point_x = xxx + curve_vertial_axis_x_pos
|
|
this_point_y = OLED_HEIGHT - this_curve[dict_key]//4 - 1
|
|
draw.line((this_point_x,this_point_y,this_point_x,this_point_y), fill="white")
|
|
|
|
def display_page(self, level, page):
|
|
if level == 0:
|
|
if page == 0:
|
|
with canvas(oled_device) as draw:
|
|
mouse_count, kb_count, gp_count = usb4vc_usb_scan.get_device_count()
|
|
draw.text((0, 0), f"KBD {kb_count} {self.current_keyboard_protocol['display_name']}", font=font_regular, fill="white")
|
|
draw.text((0, 10), f"MOS {mouse_count} {self.current_mouse_protocol['display_name']}", font=font_regular, fill="white")
|
|
draw.text((0, 20), f"GPD {gp_count} {self.current_gamepad_protocol['display_name']}", font=font_regular, fill="white")
|
|
if page == 1:
|
|
with canvas(oled_device) as draw:
|
|
if 'Unknown' in self.pb_info['full_name']:
|
|
draw.text((0, 0), f"{self.pb_info['full_name']} PID {this_pboard_id}", font=font_regular, fill="white")
|
|
else:
|
|
draw.text((0, 0), f"{self.pb_info['full_name']}", font=font_regular, fill="white")
|
|
draw.text((0, 10), f"PB {self.pb_info['fw_ver'][0]}.{self.pb_info['fw_ver'][1]}.{self.pb_info['fw_ver'][2]} RPi {usb4vc_shared.RPI_APP_VERSION_TUPLE[0]}.{usb4vc_shared.RPI_APP_VERSION_TUPLE[1]}.{usb4vc_shared.RPI_APP_VERSION_TUPLE[2]}", font=font_regular, fill="white")
|
|
draw.text((0, 20), f"IP: {get_ip_name()}", font=font_regular, fill="white")
|
|
|
|
if page == 2:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Remove BT Device", font_medium, 0, draw)
|
|
oled_print_centered("(experimental)", font_regular, 20, draw)
|
|
if page == 3:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Pair Bluetooth", font_medium, 0, draw)
|
|
oled_print_centered("(experimental)", font_regular, 20, draw)
|
|
|
|
if level == 1:
|
|
if page == 0:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Keyboard Protocol", font_medium, 0, draw)
|
|
oled_print_centered(self.kb_protocol_list[self.current_keyboard_protocol_index]['display_name'], font_medium, 15, draw)
|
|
if page == 1:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Mouse Protocol", font_medium, 0, draw)
|
|
oled_print_centered(self.mouse_protocol_list[self.current_mouse_protocol_index]['display_name'], font_medium, 15, draw)
|
|
if page == 2:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Gamepad Protocol", font_medium, 0, draw)
|
|
oled_print_centered(self.gamepad_protocol_list[self.current_gamepad_protocol_index]['display_name'], font_medium, 15, draw)
|
|
if page == 3:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Mouse Sensitivity", font_medium, 0, draw)
|
|
oled_print_centered(f"{mouse_sensitivity_list[self.current_mouse_sensitivity_offset_index]}", font_medium, 15, draw)
|
|
if page == 4:
|
|
self.draw_joystick_curve()
|
|
if page == 5:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Save & Quit", font_medium, 10, draw)
|
|
if level == 2:
|
|
if page == 0:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Put your device in", font_regular, 0, draw)
|
|
oled_print_centered("pairing mode now.", font_regular, 10, draw)
|
|
oled_print_centered("Press enter to start", font_regular, 20, draw)
|
|
if page == 1:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Scanning...", font_medium, 0, draw)
|
|
oled_print_centered("Please wait", font_medium, 15, draw)
|
|
result, self.error_message = bt_setup()
|
|
if result != 0:
|
|
self.goto_page(3)
|
|
self.display_curent_page()
|
|
return
|
|
paired_devices_set = get_paired_devices()
|
|
self.bluetooth_device_list, self.error_message = scan_bt_devices(self.bt_scan_timeout_sec)
|
|
self.bluetooth_device_list = list(set(self.bluetooth_device_list) - paired_devices_set)
|
|
if len(self.bluetooth_device_list) == 0:
|
|
self.error_message = "Nothing was found"
|
|
self.goto_page(3)
|
|
self.display_curent_page()
|
|
return
|
|
print("BT LIST:", self.bluetooth_device_list)
|
|
# set up level 3 menu structure
|
|
self.page_size[3] = len(self.bluetooth_device_list) + 1
|
|
self.goto_level(3)
|
|
self.display_curent_page()
|
|
if page == 2:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Pairing result:", font_medium, 0, draw)
|
|
oled_print_centered(self.pairing_result, font_regular, 20, draw)
|
|
if page == 3:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Bluetooth Error!", font_medium, 0, draw)
|
|
oled_print_centered(self.error_message, font_regular, 20, draw)
|
|
if level == 3:
|
|
if page == self.page_size[3] - 1:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Exit", font_medium, 10, draw)
|
|
else:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered(f"Found {len(self.bluetooth_device_list)}. Pair this?", font_regular, 0, draw)
|
|
oled_print_centered(f"{self.bluetooth_device_list[page][1]}", font_regular, 10, draw)
|
|
oled_print_centered(f"{self.bluetooth_device_list[page][0]}", font_regular, 20, draw)
|
|
if level == 4:
|
|
if page == self.page_size[4] - 1:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Exit", font_medium, 10, draw)
|
|
else:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered(f"Remove this?", font_regular, 0, draw)
|
|
oled_print_centered(f"{self.paired_devices_list[page][1]}", font_regular, 10, draw)
|
|
oled_print_centered(f"{self.paired_devices_list[page][0]}", font_regular, 20, draw)
|
|
if level == 5:
|
|
if page == 0:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Power Down", font_medium, 10, draw)
|
|
if page == 1:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Relaunch", font_medium, 10, draw)
|
|
if page == 2:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Reboot", font_medium, 10, draw)
|
|
if page == 3:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Cancel", font_medium, 10, draw)
|
|
|
|
def send_protocol_set_spi_msg(self):
|
|
status_dict = {}
|
|
for index, item in enumerate(self.kb_protocol_list):
|
|
if item['pid'] & 0x7f in status_dict and status_dict[item['pid'] & 0x7f] == 1:
|
|
continue
|
|
status_dict[item['pid'] & 0x7f] = 0
|
|
if index == self.current_keyboard_protocol_index:
|
|
status_dict[item['pid'] & 0x7f] = 1
|
|
|
|
for index, item in enumerate(self.mouse_protocol_list):
|
|
if item['pid'] & 0x7f in status_dict and status_dict[item['pid'] & 0x7f] == 1:
|
|
continue
|
|
status_dict[item['pid'] & 0x7f] = 0
|
|
if index == self.current_mouse_protocol_index:
|
|
status_dict[item['pid'] & 0x7f] = 1
|
|
|
|
for index, item in enumerate(self.gamepad_protocol_list):
|
|
if item['pid'] & 0x7f in status_dict and status_dict[item['pid'] & 0x7f] == 1:
|
|
continue
|
|
status_dict[item['pid'] & 0x7f] = 0
|
|
if index == self.current_gamepad_protocol_index:
|
|
status_dict[item['pid'] & 0x7f] = 1
|
|
|
|
protocol_bytes = []
|
|
for key in status_dict:
|
|
if key == PROTOCOL_OFF['pid']:
|
|
continue
|
|
if status_dict[key]:
|
|
protocol_bytes.append(key | 0x80)
|
|
else:
|
|
protocol_bytes.append(key)
|
|
|
|
this_msg = list(set_protocl_spi_msg_template)
|
|
this_msg[3:3+len(protocol_bytes)] = protocol_bytes
|
|
|
|
self.current_keyboard_protocol = self.kb_protocol_list[self.current_keyboard_protocol_index]
|
|
self.current_mouse_protocol = self.mouse_protocol_list[self.current_mouse_protocol_index]
|
|
self.current_gamepad_protocol = self.gamepad_protocol_list[self.current_gamepad_protocol_index]
|
|
|
|
# print('this', this_msg)
|
|
# print('last', self.last_spi_message)
|
|
if this_msg == self.last_spi_message:
|
|
print("SPI: no need to send")
|
|
return
|
|
print("set_protocol:", [hex(x) for x in this_msg])
|
|
usb4vc_usb_scan.set_protocol(this_msg)
|
|
print('new status:', [hex(x) for x in usb4vc_usb_scan.get_pboard_info()])
|
|
self.last_spi_message = list(this_msg)
|
|
|
|
def action(self, level, page):
|
|
if level == 0:
|
|
if page == 2:
|
|
self.paired_devices_list = list(get_paired_devices())
|
|
self.page_size[4] = len(self.paired_devices_list) + 1
|
|
self.goto_level(4)
|
|
elif page == 3:
|
|
self.goto_level(2)
|
|
else:
|
|
self.goto_level(1)
|
|
if level == 1:
|
|
if page == 0:
|
|
self.current_keyboard_protocol_index = (self.current_keyboard_protocol_index + 1) % len(self.kb_protocol_list)
|
|
if page == 1:
|
|
self.current_mouse_protocol_index = (self.current_mouse_protocol_index + 1) % len(self.mouse_protocol_list)
|
|
if page == 2:
|
|
self.current_gamepad_protocol_index = (self.current_gamepad_protocol_index + 1) % len(self.gamepad_protocol_list)
|
|
if page == 3:
|
|
self.current_mouse_sensitivity_offset_index = (self.current_mouse_sensitivity_offset_index + 1) % len(mouse_sensitivity_list)
|
|
if page == 4:
|
|
self.current_joystick_curve_index = (self.current_joystick_curve_index + 1) % len(joystick_curve_list)
|
|
self.draw_joystick_curve()
|
|
if page == 5:
|
|
configuration_dict[this_pboard_id]["keyboard_protocol_index"] = self.current_keyboard_protocol_index
|
|
configuration_dict[this_pboard_id]["mouse_protocol_index"] = self.current_mouse_protocol_index
|
|
configuration_dict[this_pboard_id]["mouse_sensitivity_index"] = self.current_mouse_sensitivity_offset_index
|
|
configuration_dict[this_pboard_id]["gamepad_protocol_index"] = self.current_gamepad_protocol_index
|
|
configuration_dict[this_pboard_id]["joystick_curve_index"] = self.current_joystick_curve_index
|
|
print(configuration_dict)
|
|
save_config()
|
|
self.send_protocol_set_spi_msg()
|
|
self.goto_level(0)
|
|
if level == 2:
|
|
if page == 0:
|
|
self.switch_page(1)
|
|
if page == 2:
|
|
self.goto_level(0)
|
|
if page == 3:
|
|
self.goto_level(0)
|
|
if level == 3:
|
|
if page == self.page_size[3] - 1:
|
|
self.goto_level(0)
|
|
else:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Pairing...", font_medium, 0, draw)
|
|
oled_print_centered("Please wait", font_medium, 15, draw)
|
|
print("pairing", self.bluetooth_device_list[page])
|
|
bt_mac_addr = self.bluetooth_device_list[page][0]
|
|
is_successful, result_message = pair_device(bt_mac_addr)
|
|
self.pairing_result = result_message.split('.')[-1].strip()[-22:]
|
|
if is_successful:
|
|
os.system(f'timeout {self.bt_scan_timeout_sec} bluetoothctl --agent NoInputNoOutput trust {bt_mac_addr}')
|
|
os.system(f'timeout {self.bt_scan_timeout_sec} bluetoothctl --agent NoInputNoOutput connect {bt_mac_addr}')
|
|
self.goto_level(2)
|
|
self.goto_page(2)
|
|
if level == 4:
|
|
if page == self.page_size[4] - 1:
|
|
self.goto_level(0)
|
|
else:
|
|
os.system(f'timeout 5 bluetoothctl --agent NoInputNoOutput untrust {self.paired_devices_list[page][0]}')
|
|
os.system(f'timeout 5 bluetoothctl --agent NoInputNoOutput remove {self.paired_devices_list[page][0]}')
|
|
self.goto_level(0)
|
|
if level == 5:
|
|
if page == 0:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Shutting down...", font_medium, 10, draw)
|
|
os.system("sudo halt")
|
|
while 1:
|
|
time.sleep(1)
|
|
if page == 1:
|
|
oled_device.clear()
|
|
os._exit(0)
|
|
if page == 2:
|
|
with canvas(oled_device) as draw:
|
|
oled_print_centered("Rebooting...", font_medium, 10, draw)
|
|
os.system("sudo reboot")
|
|
while 1:
|
|
time.sleep(1)
|
|
if page == 3:
|
|
self.goto_level(0)
|
|
self.display_curent_page()
|
|
|
|
def action_current_page(self):
|
|
self.action(self.current_level, self.current_page);
|
|
|
|
def display_curent_page(self):
|
|
self.display_page(self.current_level, self.current_page)
|
|
|
|
def update_usb_status(self):
|
|
if self.current_level == 0 and self.current_page == 0:
|
|
self.display_page(0, 0)
|
|
|
|
pboard_database = {
|
|
PBOARD_ID_UNKNOWN:{'author':'Unknown', 'fw_ver':(0,0,0), 'full_name':'Unknown', 'hw_rev':0, 'protocol_list_keyboard':raw_keyboard_protocols, 'protocol_list_mouse':raw_mouse_protocols, 'protocol_list_gamepad':raw_gamepad_protocols},
|
|
PBOARD_ID_IBMPC:{'author':'dekuNukem', 'fw_ver':(0,0,0), 'full_name':'IBM PC Compatible', 'hw_rev':0, 'protocol_list_keyboard':ibmpc_keyboard_protocols, 'protocol_list_mouse':ibmpc_mouse_protocols, 'protocol_list_gamepad':ibmpc_gamepad_protocols},
|
|
PBOARD_ID_ADB:{'author':'dekuNukem', 'fw_ver':(0,0,0), 'full_name':'Apple Desktop Bus', 'hw_rev':0, 'protocol_list_keyboard':adb_keyboard_protocols, 'protocol_list_mouse':adb_mouse_protocols, 'protocol_list_gamepad':adb_gamepad_protocols},
|
|
}
|
|
|
|
def get_pboard_dict(pid):
|
|
if pid not in pboard_database:
|
|
pid = 0
|
|
return pboard_database[pid]
|
|
|
|
def get_mouse_sensitivity():
|
|
return mouse_sensitivity_list[configuration_dict[this_pboard_id]["mouse_sensitivity_index"]]
|
|
|
|
def ui_init():
|
|
global pboard_info_spi_msg
|
|
global this_pboard_id
|
|
load_config()
|
|
pboard_info_spi_msg = usb4vc_usb_scan.get_pboard_info()
|
|
this_pboard_id = pboard_info_spi_msg[3]
|
|
if this_pboard_id in pboard_database:
|
|
# load custom profile mapping into protocol list
|
|
for item in custom_profile_list:
|
|
this_mapping_bid = usb4vc_shared.board_id_lookup.get(item['protocol_board'], 0)
|
|
if this_mapping_bid == this_pboard_id and item['device_type'] in pboard_database[this_pboard_id]:
|
|
this_mapping_pid = usb4vc_shared.protocol_id_lookup.get(item['protocol_name'])
|
|
this_profile = {'pid':this_mapping_pid, 'display_name':item['display_name'], 'mapping':item['mapping']}
|
|
pboard_database[this_pboard_id][item['device_type']].append(this_profile)
|
|
pboard_database[this_pboard_id]['hw_rev'] = pboard_info_spi_msg[4]
|
|
pboard_database[this_pboard_id]['fw_ver'] = (pboard_info_spi_msg[5], pboard_info_spi_msg[6], pboard_info_spi_msg[7])
|
|
if 'rpi_app_ver' not in configuration_dict:
|
|
configuration_dict['rpi_app_ver'] = usb4vc_shared.RPI_APP_VERSION_TUPLE
|
|
if this_pboard_id not in configuration_dict:
|
|
configuration_dict[this_pboard_id] = {"keyboard_protocol_index":1, "mouse_protocol_index":1, "mouse_sensitivity_index":0, "gamepad_protocol_index":1}
|
|
|
|
plus_button = my_button(PLUS_BUTTON_PIN)
|
|
minus_button = my_button(MINUS_BUTTON_PIN)
|
|
enter_button = my_button(ENTER_BUTTON_PIN)
|
|
shutdown_button = my_button(SHUTDOWN_BUTTON_PIN)
|
|
|
|
class oled_sleep_control(object):
|
|
def __init__(self):
|
|
super(oled_sleep_control, self).__init__()
|
|
self.is_sleeping = False
|
|
self.last_input_event = time.time()
|
|
def sleep(self):
|
|
if self.is_sleeping is False:
|
|
print("sleeping!")
|
|
oled_device.clear()
|
|
self.is_sleeping = True
|
|
def wakeup(self):
|
|
if self.is_sleeping:
|
|
print("waking up!")
|
|
my_menu.display_curent_page()
|
|
self.last_input_event = time.time()
|
|
self.is_sleeping = False
|
|
def check_sleep(self):
|
|
if time.time() - self.last_input_event > 180:
|
|
self.sleep()
|
|
else:
|
|
self.wakeup()
|
|
def kick(self):
|
|
self.last_input_event = time.time()
|
|
|
|
my_oled = oled_sleep_control()
|
|
my_menu = None
|
|
def ui_worker():
|
|
global my_menu
|
|
print(configuration_dict)
|
|
print("ui_worker started")
|
|
my_menu = usb4vc_menu(get_pboard_dict(this_pboard_id), configuration_dict[this_pboard_id])
|
|
my_menu.display_page(0, 0)
|
|
while 1:
|
|
time.sleep(0.1)
|
|
if my_oled.is_sleeping is False:
|
|
my_menu.update_usb_status();
|
|
|
|
if plus_button.is_pressed():
|
|
my_oled.kick()
|
|
if my_oled.is_sleeping:
|
|
my_oled.wakeup()
|
|
elif my_menu.current_level != 2:
|
|
my_menu.switch_page(1)
|
|
my_menu.display_curent_page()
|
|
|
|
if minus_button.is_pressed():
|
|
my_oled.kick()
|
|
if my_oled.is_sleeping:
|
|
my_oled.wakeup()
|
|
elif my_menu.current_level != 2:
|
|
my_menu.switch_page(-1)
|
|
my_menu.display_curent_page()
|
|
|
|
if enter_button.is_pressed():
|
|
my_oled.kick()
|
|
if my_oled.is_sleeping:
|
|
my_oled.wakeup()
|
|
else:
|
|
my_menu.action_current_page()
|
|
|
|
if shutdown_button.is_pressed():
|
|
my_oled.kick()
|
|
if my_oled.is_sleeping:
|
|
my_oled.wakeup()
|
|
my_menu.goto_level(5)
|
|
my_menu.display_curent_page()
|
|
my_oled.check_sleep()
|
|
|
|
def get_gamepad_protocol():
|
|
return my_menu.current_gamepad_protocol
|
|
|
|
ui_thread = threading.Thread(target=ui_worker, daemon=True)
|