|
@@ -0,0 +1,295 @@
|
|
|
+import copy
|
|
|
+import os
|
|
|
+import select
|
|
|
+import time
|
|
|
+
|
|
|
+import psutil
|
|
|
+import Xlib.display
|
|
|
+from Xlib import XK, X, Xatom
|
|
|
+
|
|
|
+from rescriptoon.actions import *
|
|
|
+
|
|
|
+EXTENDED_CONTROLS_DEFAULT_TOGGLE_KEYSYM_NAME = "grave"
|
|
|
+EXTENDED_CONTROLS_PID_XPROPERTY_NAME = "_TOONCHER_EXTENDED_CONTROLS_PID"
|
|
|
+TOONTOWN_WINDOW_NAME = "Toontown Rewritten"
|
|
|
+
|
|
|
+EXTENDED_CONTROLS_DEFAULT_KEYSYM_MAPPINGS = {
|
|
|
+ XK.XK_w: RewriteKeyEventAction(keysym=XK.XK_Up, target_engine_index=0),
|
|
|
+ XK.XK_a: RewriteKeyEventAction(keysym=XK.XK_Left, target_engine_index=0),
|
|
|
+ XK.XK_s: RewriteKeyEventAction(keysym=XK.XK_Down, target_engine_index=0),
|
|
|
+ XK.XK_d: RewriteKeyEventAction(keysym=XK.XK_Right, target_engine_index=0),
|
|
|
+ XK.XK_Control_L: RewriteKeyEventAction(
|
|
|
+ keysym=XK.XK_Control_L, target_engine_index=0
|
|
|
+ ),
|
|
|
+ XK.XK_v: RewriteKeyEventAction(keysym=XK.XK_Delete, target_engine_index=0),
|
|
|
+ XK.XK_i: RewriteKeyEventAction(keysym=XK.XK_Up, target_engine_index=1),
|
|
|
+ XK.XK_j: RewriteKeyEventAction(keysym=XK.XK_Left, target_engine_index=1),
|
|
|
+ XK.XK_k: RewriteKeyEventAction(keysym=XK.XK_Down, target_engine_index=1),
|
|
|
+ XK.XK_l: RewriteKeyEventAction(keysym=XK.XK_Right, target_engine_index=1),
|
|
|
+ XK.XK_slash: RewriteKeyEventAction(keysym=XK.XK_Control_L, target_engine_index=1),
|
|
|
+ XK.XK_n: RewriteKeyEventAction(keysym=XK.XK_Delete, target_engine_index=1),
|
|
|
+ XK.XK_space: RewriteKeyEventAction(
|
|
|
+ keysym=XK.XK_Control_L, target_engine_index=TargetEngine.All
|
|
|
+ ),
|
|
|
+ XK.XK_e: SelectGagAction(
|
|
|
+ target_engine_index=0, column_index=4, factor_y=-0.047
|
|
|
+ ),
|
|
|
+ XK.XK_o: SelectGagAction(
|
|
|
+ target_engine_index=1, column_index=4, factor_y=-0.047
|
|
|
+ ),
|
|
|
+ XK.XK_f: SelectGagAction(
|
|
|
+ target_engine_index=0, column_index=5, factor_y=-0.047
|
|
|
+ ),
|
|
|
+ XK.XK_semicolon: SelectGagAction(
|
|
|
+ target_engine_index=1, column_index=5, factor_y=-0.047
|
|
|
+ ),
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+def x_find_window(parent_window, filter_callback):
|
|
|
+ matching = []
|
|
|
+ for child_window in parent_window.query_tree().children:
|
|
|
+ if filter_callback(child_window):
|
|
|
+ matching.append(child_window)
|
|
|
+ matching += x_find_window(child_window, filter_callback)
|
|
|
+ return matching
|
|
|
+
|
|
|
+
|
|
|
+def x_find_window_by_pid(display, pid):
|
|
|
+ pid_prop = display.intern_atom("_NET_WM_PID")
|
|
|
+
|
|
|
+ def filter_callback(window):
|
|
|
+ prop = window.get_full_property(pid_prop, X.AnyPropertyType)
|
|
|
+ return prop and prop.value.tolist() == [pid]
|
|
|
+
|
|
|
+ return x_find_window(display.screen().root, filter_callback)
|
|
|
+
|
|
|
+
|
|
|
+def x_wait_for_event(xdisplay, timeout_seconds):
|
|
|
+ """ Wait up to `timeout_seconds` seconds for a xevent.
|
|
|
+ Return True, if a xevent is available.
|
|
|
+ Return False, if the timeout was reached. """
|
|
|
+ rlist = select.select(
|
|
|
+ [xdisplay.display.socket],
|
|
|
+ [],
|
|
|
+ [],
|
|
|
+ timeout_seconds,
|
|
|
+ )[0]
|
|
|
+ return len(rlist) > 0
|
|
|
+
|
|
|
+
|
|
|
+class Rescriptoon:
|
|
|
+ def __init__(
|
|
|
+ self,
|
|
|
+ primary_engine_pid,
|
|
|
+ primary_engine_window_name=None,
|
|
|
+ toggle_keysym_name=EXTENDED_CONTROLS_DEFAULT_TOGGLE_KEYSYM_NAME,
|
|
|
+ ):
|
|
|
+ self._primary_engine_pid = primary_engine_pid
|
|
|
+ self._primary_engine_window_name = primary_engine_window_name
|
|
|
+ self._xdisplay = Xlib.display.Display()
|
|
|
+ self._toggle_keysym = XK.string_to_keysym(toggle_keysym_name)
|
|
|
+ if self._toggle_keysym == X.NoSymbol:
|
|
|
+ raise Exception(
|
|
|
+ "Extended keyboard controls toggle:"
|
|
|
+ + " Unknown keysym name '{}'".format(toggle_keysym_name)
|
|
|
+ )
|
|
|
+ self._keysym_mappings = copy.deepcopy(
|
|
|
+ EXTENDED_CONTROLS_DEFAULT_KEYSYM_MAPPINGS,
|
|
|
+ )
|
|
|
+ if self._toggle_keysym in self._keysym_mappings:
|
|
|
+ print(
|
|
|
+ "INFO Extended Controls:"
|
|
|
+ + " Ignoring mapping for toggle key '{}'".format(toggle_keysym_name)
|
|
|
+ )
|
|
|
+ self._keysym_mappings[self._toggle_keysym] = ToggleExtendedControlsAction()
|
|
|
+ self._default_action = ForwardKeyEventAction()
|
|
|
+ self._primary_engine_window = None
|
|
|
+ self._engine_windows_by_target_index = None
|
|
|
+ self._active_key_registry = {}
|
|
|
+ self._enabled = False
|
|
|
+
|
|
|
+ @property
|
|
|
+ def engine_running(self):
|
|
|
+ return psutil.pid_exists(self._primary_engine_pid)
|
|
|
+
|
|
|
+ @property
|
|
|
+ def xdisplay(self):
|
|
|
+ return self._xdisplay
|
|
|
+
|
|
|
+ @property
|
|
|
+ def primary_engine_window(self):
|
|
|
+ return self._primary_engine_window
|
|
|
+
|
|
|
+ def _wait_for_engine_window(self, timeout_seconds=20, search_interval_seconds=2):
|
|
|
+ start_epoch = time.time()
|
|
|
+ while self.engine_running and (time.time() - start_epoch) <= timeout_seconds:
|
|
|
+ windows = x_find_window_by_pid(self._xdisplay, self._primary_engine_pid,)
|
|
|
+ assert len(windows) <= 1
|
|
|
+ if len(windows) == 1:
|
|
|
+ return windows[0]
|
|
|
+ time.sleep(search_interval_seconds)
|
|
|
+ return None
|
|
|
+
|
|
|
+ def run(self):
|
|
|
+ self._primary_engine_window = self._wait_for_engine_window()
|
|
|
+ if not self._primary_engine_window:
|
|
|
+ raise Exception("Could not find the game's window.")
|
|
|
+ self._grab_key(self._xdisplay.keysym_to_keycode(self._toggle_keysym),)
|
|
|
+ self._primary_engine_window.change_property(
|
|
|
+ self.xdisplay.intern_atom(EXTENDED_CONTROLS_PID_XPROPERTY_NAME),
|
|
|
+ Xatom.CARDINAL,
|
|
|
+ format=32,
|
|
|
+ data=[os.getpid()],
|
|
|
+ mode=X.PropModeReplace,
|
|
|
+ )
|
|
|
+ if self._primary_engine_window_name:
|
|
|
+ self._primary_engine_window.set_wm_name(self._primary_engine_window_name,)
|
|
|
+ print(
|
|
|
+ "INFO Changed engine's window name to {!r}".format(
|
|
|
+ self._primary_engine_window_name,
|
|
|
+ )
|
|
|
+ )
|
|
|
+ if not self.enabled:
|
|
|
+ keysym_name = XK.keysym_to_string(self._toggle_keysym)
|
|
|
+ print(
|
|
|
+ "INFO Extended Controls are currently disabled."
|
|
|
+ + " Press key '{}' to enable.".format(keysym_name)
|
|
|
+ )
|
|
|
+ while self.engine_running:
|
|
|
+ while self.xdisplay.pending_events():
|
|
|
+ self._handle_xevent(self._xdisplay.next_event())
|
|
|
+ self._check_active_key_registry()
|
|
|
+
|
|
|
+
|
|
|
+ x_wait_for_event(self.xdisplay, timeout_seconds=0.05)
|
|
|
+
|
|
|
+ def _handle_xevent(self, xevent):
|
|
|
+ if isinstance(xevent, Xlib.protocol.event.KeyPress) or isinstance(
|
|
|
+ xevent, Xlib.protocol.event.KeyRelease
|
|
|
+ ):
|
|
|
+ self._handle_xkeyevent(xevent)
|
|
|
+
|
|
|
+ def _handle_xkeyevent(self, xkeyevent):
|
|
|
+ self._update_active_key_registry(xkeyevent)
|
|
|
+ keysym_in = self._xdisplay.keycode_to_keysym(xkeyevent.detail, index=0,)
|
|
|
+ if keysym_in in self._keysym_mappings:
|
|
|
+ action = self._keysym_mappings[keysym_in]
|
|
|
+ else:
|
|
|
+ action = self._default_action
|
|
|
+ action.execute(self, xkeyevent)
|
|
|
+
|
|
|
+ def enable(self):
|
|
|
+ for keysym in self._keysym_mappings.keys():
|
|
|
+ if keysym != self._toggle_keysym:
|
|
|
+ self._grab_key(self._xdisplay.keysym_to_keycode(keysym),)
|
|
|
+ self._enabled = True
|
|
|
+
|
|
|
+ self._engine_windows_by_target_index = None
|
|
|
+ print("INFO Enabled Extended Controls")
|
|
|
+
|
|
|
+ def disable(self):
|
|
|
+ for keysym in self._keysym_mappings.keys():
|
|
|
+ if keysym != self._toggle_keysym:
|
|
|
+ self._ungrab_key(self._xdisplay.keysym_to_keycode(keysym),)
|
|
|
+ self._enabled = False
|
|
|
+ print("INFO Disabled Extended Controls")
|
|
|
+
|
|
|
+ @property
|
|
|
+ def enabled(self):
|
|
|
+ return self._enabled
|
|
|
+
|
|
|
+ def toggle(self):
|
|
|
+ if self.enabled:
|
|
|
+ self.disable()
|
|
|
+ else:
|
|
|
+ self.enable()
|
|
|
+
|
|
|
+ def _grab_key(self, keycode):
|
|
|
+ self.primary_engine_window.grab_key(
|
|
|
+ keycode,
|
|
|
+ X.AnyModifier,
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ True,
|
|
|
+ X.GrabModeAsync,
|
|
|
+ X.GrabModeAsync,
|
|
|
+ )
|
|
|
+
|
|
|
+ def _ungrab_key(self, keycode):
|
|
|
+ self.primary_engine_window.ungrab_key(keycode, X.AnyModifier)
|
|
|
+
|
|
|
+ def find_engine_windows(self):
|
|
|
+ controls_xprop = self.xdisplay.intern_atom(
|
|
|
+ EXTENDED_CONTROLS_PID_XPROPERTY_NAME,
|
|
|
+ )
|
|
|
+ return x_find_window(
|
|
|
+ self.xdisplay.screen().root,
|
|
|
+ lambda w: w.get_wm_name() == TOONTOWN_WINDOW_NAME
|
|
|
+ or w.get_full_property(controls_xprop, X.AnyPropertyType),
|
|
|
+ )
|
|
|
+
|
|
|
+ @property
|
|
|
+ def engine_windows_by_target_index(self):
|
|
|
+ if not self._engine_windows_by_target_index:
|
|
|
+ win_by_index = {}
|
|
|
+ for target_index, win in enumerate(self.find_engine_windows()):
|
|
|
+ print(
|
|
|
+ "INFO Engine window {} has no target index, assuming {}".format(
|
|
|
+ win.id, target_index,
|
|
|
+ )
|
|
|
+ )
|
|
|
+ if not target_index in win_by_index:
|
|
|
+ win_by_index[target_index] = []
|
|
|
+ win_by_index[target_index].append(win)
|
|
|
+ self._engine_windows_by_target_index = win_by_index
|
|
|
+ return self._engine_windows_by_target_index
|
|
|
+
|
|
|
+ @property
|
|
|
+ def engine_windows(self):
|
|
|
+ return [w for g in self.engine_windows_by_target_index.values() for w in g]
|
|
|
+
|
|
|
+ def _update_active_key_registry(self, xkeyevent):
|
|
|
+
|
|
|
+ keycode = xkeyevent.detail
|
|
|
+ if isinstance(xkeyevent, Xlib.protocol.event.KeyPress):
|
|
|
+ self._active_key_registry[keycode] = xkeyevent
|
|
|
+ elif keycode in self._active_key_registry:
|
|
|
+ del self._active_key_registry[keycode]
|
|
|
+
|
|
|
+ def _check_active_key_registry(self):
|
|
|
+ """
|
|
|
+ WORKAROUND
|
|
|
+ For an unknown reason some key release events don't get queued
|
|
|
+ when multiple keys are being released simultaneously.
|
|
|
+ So we keep a hashmap of supposedly currently pressed keys
|
|
|
+ and periodically compare it with xdispaly.query_keymap().
|
|
|
+
|
|
|
+ ref: https://stackoverflow.com/q/18160792/5894777
|
|
|
+ """
|
|
|
+
|
|
|
+ keymap = self.xdisplay.query_keymap()
|
|
|
+ missed_releases = []
|
|
|
+ for keycode, press_event in self._active_key_registry.items():
|
|
|
+ byte_index = keycode >> 3
|
|
|
+ bit_index = keycode & ((1 << 3) - 1)
|
|
|
+ if not keymap[byte_index] & (1 << bit_index):
|
|
|
+ print("DEBUG missed release event of key {}".format(keycode))
|
|
|
+ missed_releases.append(
|
|
|
+ Xlib.protocol.event.KeyRelease(
|
|
|
+ window=press_event.window,
|
|
|
+ detail=press_event.detail,
|
|
|
+ state=press_event.state,
|
|
|
+ root_x=press_event.root_x,
|
|
|
+ root_y=press_event.root_y,
|
|
|
+ event_x=press_event.event_x,
|
|
|
+ event_y=press_event.event_y,
|
|
|
+ child=press_event.child,
|
|
|
+ root=press_event.root,
|
|
|
+ time=X.CurrentTime,
|
|
|
+ same_screen=press_event.same_screen,
|
|
|
+ )
|
|
|
+ )
|
|
|
+ for release_event in missed_releases:
|
|
|
+ self._handle_xkeyevent(release_event)
|