mirror of
https://github.com/erjiang/huion-keys.git
synced 2026-03-23 13:44:52 +03:00
partial threaded input implementation
removed the config reload signal because it's becoming harder to maintain thanks to threading
This commit is contained in:
parent
6d090cf00e
commit
6bc1bac531
251
huion_keys.py
251
huion_keys.py
@ -1,8 +1,8 @@
|
||||
#!/usr/bin/env python3
|
||||
import os
|
||||
import time
|
||||
import signal
|
||||
import argparse
|
||||
import threading
|
||||
import configparser
|
||||
|
||||
from _xdo_cffi import ffi, lib
|
||||
@ -17,10 +17,21 @@ TABLET_MODELS = {
|
||||
BUTTON_BINDINGS = {}
|
||||
BUTTON_BINDINGS_HOLD = {}
|
||||
CYCLE_BUTTON = None
|
||||
CYCLE_MODE = 1
|
||||
CYCLE_MODES = 1
|
||||
DIAL_MODES = {}
|
||||
|
||||
BUTTON_BITS = {
|
||||
0x01: 1,
|
||||
0x02: 2,
|
||||
0x04: 3,
|
||||
0x08: 4,
|
||||
0x10: 5,
|
||||
0x20: 6,
|
||||
0x40: 7,
|
||||
0x80: 8,
|
||||
}
|
||||
|
||||
|
||||
def main():
|
||||
#Commandline arguments processing
|
||||
parser = argparse.ArgumentParser(
|
||||
@ -41,8 +52,6 @@ def main():
|
||||
else:
|
||||
CONFIG_FILE_PATH = os.path.expanduser(args.config)
|
||||
|
||||
global CYCLE_MODES, CYCLE_MODE, CYCLE_BUTTON
|
||||
xdo = lib.xdo_new(ffi.NULL)
|
||||
if os.path.isfile(CONFIG_FILE_PATH):
|
||||
read_config(CONFIG_FILE_PATH)
|
||||
else:
|
||||
@ -50,73 +59,157 @@ def main():
|
||||
create_default_config(CONFIG_FILE_PATH)
|
||||
print("Created an example config file at " + CONFIG_FILE_PATH)
|
||||
return 1
|
||||
signal.signal(signal.SIGUSR1, handle_reload_signal) # Reload the config if recieved SIGUSR1
|
||||
prev_button = None
|
||||
|
||||
hidraw_paths = []
|
||||
while True:
|
||||
hidraw_path = None
|
||||
# search for a known tablet device
|
||||
for device_name, device_id in TABLET_MODELS.items():
|
||||
hidraw_path = get_tablet_hidraw(device_id)
|
||||
if hidraw_path is not None:
|
||||
print("Found %s at %s" % (device_name, hidraw_path))
|
||||
break
|
||||
if hidraw_path is None:
|
||||
print("Could not find tablet hidraw device")
|
||||
hidraw_paths = hidraw_paths + hidraw_path
|
||||
|
||||
if hidraw_paths:
|
||||
print("Found %s at %s" % (device_name, hidraw_paths))
|
||||
break
|
||||
elif not hidraw_paths:
|
||||
print("Could not find any tablet hidraw devices")
|
||||
time.sleep(2)
|
||||
continue
|
||||
try:
|
||||
hidraw = open(hidraw_path, 'rb')
|
||||
except PermissionError as e:
|
||||
print(e)
|
||||
print("Trying again in 5 seconds...")
|
||||
time.sleep(5)
|
||||
continue
|
||||
continue
|
||||
|
||||
threads = []
|
||||
for hidraw_path in hidraw_paths:
|
||||
thread = PollThread(hidraw_path)
|
||||
thread.daemon = True
|
||||
threads.append(thread)
|
||||
thread.start()
|
||||
|
||||
for thread in threads:
|
||||
thread.join()
|
||||
|
||||
class PollThread(threading.Thread):
|
||||
|
||||
cycle_mode = None
|
||||
hidraw_path = None
|
||||
xdo = None
|
||||
|
||||
SCROLL_STATE=None
|
||||
|
||||
def __init__(self, hidraw_path):
|
||||
super(PollThread, self).__init__()
|
||||
self.xdo = lib.xdo_new(ffi.NULL)
|
||||
self.hidraw_path = hidraw_path
|
||||
self.cycle_mode = 1
|
||||
|
||||
def run(self):
|
||||
global BUTTON_BINDINGS, BUTTON_BINDINGS_HOLD, CYCLE_MODES, CYCLE_BUTTON
|
||||
while True:
|
||||
try:
|
||||
btn = get_button_press(hidraw)
|
||||
hidraw = open(self.hidraw_path, 'rb')
|
||||
break
|
||||
except PermissionError as e:
|
||||
print(e)
|
||||
print("Trying again in 5 seconds...")
|
||||
time.sleep(5)
|
||||
continue
|
||||
|
||||
while True:
|
||||
try:
|
||||
btn = self.get_button_press(hidraw)
|
||||
except OSError as e:
|
||||
print("Lost connection with the tablet - searching for tablet...")
|
||||
time.sleep(3)
|
||||
break
|
||||
print("Got button %s" % (btn,))
|
||||
print(BUTTON_BINDINGS)
|
||||
if btn == CYCLE_BUTTON and CYCLE_BUTTON is not None:
|
||||
CYCLE_MODE = CYCLE_MODE + 1
|
||||
if CYCLE_MODE > CYCLE_MODES:
|
||||
CYCLE_MODE = 1
|
||||
print("Cycling to mode %s" % (CYCLE_MODE,))
|
||||
elif CYCLE_MODE in DIAL_MODES and btn in DIAL_MODES[CYCLE_MODE]:
|
||||
print("Sending %s from Mode %d" % (DIAL_MODES[CYCLE_MODE][btn], CYCLE_MODE),)
|
||||
lib.xdo_send_keysequence_window(
|
||||
xdo, lib.CURRENTWINDOW, DIAL_MODES[CYCLE_MODE][btn], 1000)
|
||||
elif btn in BUTTON_BINDINGS_HOLD:
|
||||
print("Pressing %s" % (BUTTON_BINDINGS_HOLD[btn],))
|
||||
lib.xdo_send_keysequence_window_down(xdo, lib.CURRENTWINDOW, BUTTON_BINDINGS_HOLD[btn], 12000)
|
||||
get_button_release(hidraw)
|
||||
print("Releasing %s" % (BUTTON_BINDINGS_HOLD[btn],))
|
||||
lib.xdo_send_keysequence_window_up(xdo, lib.CURRENTWINDOW, BUTTON_BINDINGS_HOLD[btn], 12000)
|
||||
elif btn in BUTTON_BINDINGS:
|
||||
print("Sending %s" % (BUTTON_BINDINGS[btn],))
|
||||
lib.xdo_send_keysequence_window(
|
||||
xdo, lib.CURRENTWINDOW, BUTTON_BINDINGS[btn], 1000)
|
||||
|
||||
self.cycle_mode = self.cycle_mode + 1
|
||||
if self.cycle_mode > CYCLE_MODES:
|
||||
self.cycle_mode = 1
|
||||
print("Cycling to mode %s" % (self.cycle_mode,))
|
||||
elif self.cycle_mode in DIAL_MODES and btn in DIAL_MODES[self.cycle_mode]:
|
||||
print("Sending %s from Mode %d" % (DIAL_MODES[self.cycle_mode][btn], self.cycle_mode),)
|
||||
lib.xdo_send_keysequence_window(
|
||||
self.xdo, lib.CURRENTWINDOW, DIAL_MODES[self.cycle_mode][btn], 1000)
|
||||
elif btn in BUTTON_BINDINGS_HOLD:
|
||||
print("Pressing %s" % (BUTTON_BINDINGS_HOLD[btn],))
|
||||
lib.xdo_send_keysequence_window_down(self.xdo, lib.CURRENTWINDOW, BUTTON_BINDINGS_HOLD[btn], 12000)
|
||||
self.get_button_release(hidraw)
|
||||
print("Releasing %s" % (BUTTON_BINDINGS_HOLD[btn],))
|
||||
lib.xdo_send_keysequence_window_up(xdo, lib.CURRENTWINDOW, BUTTON_BINDINGS_HOLD[btn], 12000)
|
||||
elif btn in BUTTON_BINDINGS:
|
||||
print("Sending %s" % (BUTTON_BINDINGS[btn],))
|
||||
lib.xdo_send_keysequence_window(
|
||||
self.xdo, lib.CURRENTWINDOW, BUTTON_BINDINGS[btn], 1000)
|
||||
|
||||
def get_button_press(self,hidraw):
|
||||
while True:
|
||||
sequence = hidraw.read(12)
|
||||
# 0xf7 is what my Kamvas Pro 22 reads
|
||||
# another model seems to send 0x08
|
||||
# Q620M reads as 0xf9
|
||||
if sequence[0] != 0xf7 and sequence[0] != 0x08 and sequence[0] != 0xf9:
|
||||
pass
|
||||
if sequence[1] == 0xe0: # buttons
|
||||
# doesn't seem like the tablet will let you push two buttons at once
|
||||
if sequence[4] > 0:
|
||||
return BUTTON_BITS[sequence[4]]
|
||||
elif sequence[5] > 0:
|
||||
# right-side buttons are 8-15, so add 8
|
||||
return BUTTON_BITS[sequence[5]] + 8
|
||||
else:
|
||||
# must be button release (all zeros)
|
||||
continue
|
||||
elif sequence[1] == 0xf0: # scroll strip
|
||||
scroll_pos = sequence[5]
|
||||
if scroll_pos == 0:
|
||||
# reset scroll state after lifting finger off scroll strip
|
||||
self.SCROLL_STATE = None
|
||||
elif self.SCROLL_STATE is not None:
|
||||
# scroll strip is numbered from top to bottom so a greater new
|
||||
# value means they scrolled down
|
||||
if scroll_pos > self.SCROLL_STATE:
|
||||
self.SCROLL_STATE = scroll_pos
|
||||
return 'scroll_down'
|
||||
elif scroll_pos < self.SCROLL_STATE:
|
||||
self.SCROLL_STATE = scroll_pos
|
||||
return 'scroll_up'
|
||||
else:
|
||||
self.SCROLL_STATE = scroll_pos
|
||||
continue
|
||||
elif sequence[1] == 0xf1: # dial on Q620M, practically 2 buttons
|
||||
if sequence[5] == 0x1:
|
||||
return 'dial_cw'
|
||||
elif sequence[5] == 0xff:
|
||||
return 'dial_ccw'
|
||||
else:
|
||||
continue
|
||||
|
||||
def get_button_release(self,hidraw):
|
||||
while True:
|
||||
sequence = hidraw.read(12)
|
||||
if sequence[1] == 0xe0 and sequence[4] == 0 and sequence[5] == 0:
|
||||
return True
|
||||
|
||||
def get_tablet_hidraw(device_id):
|
||||
"""Finds the /dev/hidrawX file that belongs to the given device ID (in xxxx:xxxx format)."""
|
||||
"""Finds the /dev/hidrawX file or files that belong to the given device ID (in xxxx:xxxx format)."""
|
||||
# TODO: is this too fragile?
|
||||
hidraws = os.listdir('/sys/class/hidraw')
|
||||
inputs = [];
|
||||
for h in hidraws:
|
||||
device_path = os.readlink(os.path.join('/sys/class/hidraw', h, 'device'))
|
||||
if device_id.upper() in device_path:
|
||||
# need to confirm that there's "input" because there are two hidraw
|
||||
# files listed for the tablet, but only one of them carries the
|
||||
# need to confirm that there's "input" because there are two or more hidraw
|
||||
# files listed for the tablet, but only few of them carry the
|
||||
# mouse/keyboard input
|
||||
if os.path.exists(os.path.join('/sys/class/hidraw', h, 'device/input')):
|
||||
return os.path.join('/dev', os.path.basename(h))
|
||||
inputs.append(os.path.join('/dev', os.path.basename(h)))
|
||||
if inputs:
|
||||
return inputs
|
||||
return None
|
||||
|
||||
|
||||
def read_config(config_file):
|
||||
global CYCLE_MODES, CYCLE_MODE, CYCLE_BUTTON
|
||||
global CYCLE_MODES, CYCLE_BUTTON, BUTTON_BINDINGS, BUTTON_BINDINGS_HOLD
|
||||
CONFIG = configparser.ConfigParser()
|
||||
CONFIG.read(config_file)
|
||||
# It is still better for performance to pre-encode these values
|
||||
@ -158,11 +251,6 @@ def read_config(config_file):
|
||||
for binding in CONFIG[key]:
|
||||
DIAL_MODES[mode][binding] = CONFIG[key][binding].encode('utf-8')
|
||||
|
||||
|
||||
def handle_reload_signal(signum, frame):
|
||||
print("SIGUSR1 recieved - reloading config..")
|
||||
read_config(CONFIG_FILE_PATH)
|
||||
|
||||
def make_rules():
|
||||
for device_name, device_id in TABLET_MODELS.items():
|
||||
print("# %s" % (device_name, ))
|
||||
@ -198,70 +286,5 @@ dial_cw=minus
|
||||
dial_ccw=equal
|
||||
""")
|
||||
|
||||
|
||||
BUTTON_BITS = {
|
||||
0x01: 1,
|
||||
0x02: 2,
|
||||
0x04: 3,
|
||||
0x08: 4,
|
||||
0x10: 5,
|
||||
0x20: 6,
|
||||
0x40: 7,
|
||||
0x80: 8,
|
||||
}
|
||||
|
||||
SCROLL_STATE=None
|
||||
|
||||
def get_button_press(hidraw):
|
||||
global SCROLL_STATE
|
||||
while True:
|
||||
sequence = hidraw.read(12)
|
||||
# 0xf7 is what my Kamvas Pro 22 reads
|
||||
# another model seems to send 0x08
|
||||
# Q620M reads as 0xf9
|
||||
if sequence[0] != 0xf7 and sequence[0] != 0x08 and sequence[0] != 0xf9:
|
||||
pass
|
||||
if sequence[1] == 0xe0: # buttons
|
||||
# doesn't seem like the tablet will let you push two buttons at once
|
||||
if sequence[4] > 0:
|
||||
return BUTTON_BITS[sequence[4]]
|
||||
elif sequence[5] > 0:
|
||||
# right-side buttons are 8-15, so add 8
|
||||
return BUTTON_BITS[sequence[5]] + 8
|
||||
else:
|
||||
# must be button release (all zeros)
|
||||
continue
|
||||
elif sequence[1] == 0xf0: # scroll strip
|
||||
scroll_pos = sequence[5]
|
||||
if scroll_pos == 0:
|
||||
# reset scroll state after lifting finger off scroll strip
|
||||
SCROLL_STATE = None
|
||||
elif SCROLL_STATE is not None:
|
||||
# scroll strip is numbered from top to bottom so a greater new
|
||||
# value means they scrolled down
|
||||
if scroll_pos > SCROLL_STATE:
|
||||
SCROLL_STATE = scroll_pos
|
||||
return 'scroll_down'
|
||||
elif scroll_pos < SCROLL_STATE:
|
||||
SCROLL_STATE = scroll_pos
|
||||
return 'scroll_up'
|
||||
else:
|
||||
SCROLL_STATE = scroll_pos
|
||||
continue
|
||||
elif sequence[1] == 0xf1: # dial on Q620M, practically 2 buttons
|
||||
if sequence[5] == 0x1:
|
||||
return 'dial_cw'
|
||||
elif sequence[5] == 0xff:
|
||||
return 'dial_ccw'
|
||||
else:
|
||||
continue
|
||||
|
||||
def get_button_release(hidraw):
|
||||
while True:
|
||||
sequence = hidraw.read(12)
|
||||
if sequence[1] == 0xe0 and sequence[4] == 0 and sequence[5] == 0:
|
||||
return True
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user