159 lines
5.6 KiB
Python
159 lines
5.6 KiB
Python
from adafruit_ble import BLERadio
|
|
from adafruit_ble.advertising.standard import ProvideServicesAdvertisement
|
|
from adafruit_ble.services.nordic import UARTService
|
|
from kmk.extensions import Extension
|
|
from kmk.hid import HIDModes
|
|
from kmk.kmktime import ticks_diff, ticks_ms
|
|
from kmk.matrix import intify_coordinate
|
|
|
|
|
|
class BLE_Split(Extension):
|
|
def __init__(
|
|
self, split_flip=True, split_side=None, split_offsets=[], hid_type=HIDModes.BLE
|
|
):
|
|
self._is_target = True
|
|
self._uart_buffer = []
|
|
self.hid_type = hid_type
|
|
self.split_flip = split_flip
|
|
self.split_side = split_side
|
|
self.split_offsets = split_offsets
|
|
self._ble = BLERadio()
|
|
self._ble_last_scan = ticks_ms() - 5000
|
|
self._is_target = True
|
|
self._connection_count = 0
|
|
self._uart = None
|
|
self._advertisment = None
|
|
self._advertising = False
|
|
|
|
def __repr__(self):
|
|
return f'BLE_SPLIT({self._to_dict()})'
|
|
|
|
def _to_dict(self):
|
|
return f'BLE_Split( _ble={self._ble} _ble_last_scan={self._ble_last_scan} _is_target={self._is_target} _uart_buffer={self._uart_buffer} _split_flip={self.split_flip} _split_side={self.split_side} _split_offsets={self.split_offsets} )'
|
|
|
|
def during_bootup(self, keyboard):
|
|
if self.split_side == 'Left':
|
|
self._is_target = True
|
|
else:
|
|
self._is_target = False
|
|
|
|
if self.split_flip and not self._is_target:
|
|
keyboard.col_pins = list(reversed(keyboard.col_pins))
|
|
if self.split_side == 'Left':
|
|
self._is_target = True
|
|
self.target_advertise()
|
|
elif self.split_side == 'Right':
|
|
self._is_target = False
|
|
self.initiator_scan()
|
|
|
|
# Attempt to sanely guess a coord_mapping if one is not provided.
|
|
if not keyboard.coord_mapping:
|
|
keyboard.coord_mapping = []
|
|
|
|
rows_to_calc = len(keyboard.row_pins)
|
|
cols_to_calc = len(keyboard.col_pins)
|
|
|
|
if self.split_offsets:
|
|
rows_to_calc *= 2
|
|
cols_to_calc *= 2
|
|
|
|
for ridx in range(rows_to_calc):
|
|
for cidx in range(cols_to_calc):
|
|
keyboard.coord_mapping.append(intify_coordinate(ridx, cidx))
|
|
|
|
def before_matrix_scan(self, keyboard_state):
|
|
self.check_all_connections()
|
|
if self._is_target:
|
|
return self._receive_from_initiator()
|
|
|
|
def after_matrix_scan(self, keyboard_state, matrix_update):
|
|
if matrix_update is not None and not self._is_target:
|
|
self._send_to_target(matrix_update)
|
|
|
|
def check_all_connections(self):
|
|
self._connection_count = len(self._ble.connections)
|
|
if self._is_target and self._connection_count < 2:
|
|
self.target_advertise()
|
|
elif not self._is_target and self._connection_count < 1:
|
|
self.initiator_scan()
|
|
|
|
def initiator_scan(self):
|
|
self._uart = None
|
|
# See if any existing connections are providing UARTService.
|
|
self._connection_count = len(self._ble.connections)
|
|
if self._connection_count > 0 and not self._uart:
|
|
for connection in self._ble.connections:
|
|
if UARTService in connection:
|
|
self._uart = connection
|
|
break
|
|
|
|
if not self._uart:
|
|
print('Scanning...')
|
|
for adv in self._ble.start_scan(ProvideServicesAdvertisement, timeout=20):
|
|
print('Scanning...')
|
|
if UARTService in adv.services:
|
|
self._uart = self._ble.connect(adv)
|
|
self._ble.stop_scan()
|
|
print('Scan complete')
|
|
break
|
|
self._ble.stop_scan()
|
|
return
|
|
|
|
def send(self, data):
|
|
if self._uart and self._uart.connected:
|
|
try:
|
|
self._uart[UARTService].write(data)
|
|
except OSError:
|
|
try:
|
|
self._uart.disconnect()
|
|
except: # noqa: E722
|
|
print('UART disconnect failed')
|
|
print('Connection error')
|
|
self._uart = None
|
|
return
|
|
|
|
def target_advertise(self):
|
|
self._ble.stop_advertising()
|
|
print('Advertising')
|
|
# Uart must not change on this connection if reconnecting
|
|
if not self._uart:
|
|
self._uart = UARTService()
|
|
advertisement = ProvideServicesAdvertisement(self._uart)
|
|
|
|
try:
|
|
self._ble.start_advertising(advertisement)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
self.ble_time_reset()
|
|
while not self.ble_rescan_timer():
|
|
self._connection_count = len(self._ble.connections)
|
|
if self._connection_count > 1:
|
|
self.ble_time_reset()
|
|
print('Advertising complete')
|
|
break
|
|
self._ble.stop_advertising()
|
|
|
|
def ble_rescan_timer(self):
|
|
return bool(ticks_diff(ticks_ms(), self._ble_last_scan) > 5000)
|
|
|
|
def ble_time_reset(self):
|
|
self._ble_last_scan = ticks_ms()
|
|
return self
|
|
|
|
def _send_to_target(self, update):
|
|
update[1] += self.split_offsets[update[0]]
|
|
if self._uart is not None and self._uart.connected:
|
|
self.send(update)
|
|
|
|
def _receive_from_initiator(self):
|
|
if self._uart is not None and self._uart.in_waiting > 0 or self._uart_buffer:
|
|
while self._uart.in_waiting >= 3:
|
|
self._uart_buffer.append(self._uart.read(3))
|
|
if self._uart_buffer:
|
|
update = bytearray(self._uart_buffer.pop(0))
|
|
|
|
return update
|
|
|
|
return None
|