99 lines
2.7 KiB
Python
99 lines
2.7 KiB
Python
from random import randint
|
|
|
|
from kmk.keys import make_argumented_key
|
|
from kmk.modules import Module
|
|
|
|
|
|
class RapidFireMeta:
|
|
def __init__(
|
|
self,
|
|
kc,
|
|
repeat=100,
|
|
wait=200,
|
|
randomize_repeat=False,
|
|
randomize_magnitude=15,
|
|
toggle=False,
|
|
):
|
|
self.kc = kc
|
|
self.repeat = repeat
|
|
self.wait = wait
|
|
self.randomize_repeat = randomize_repeat
|
|
self.randomize_magnitude = randomize_magnitude
|
|
self.toggle = toggle
|
|
|
|
|
|
class RapidFire(Module):
|
|
_active_keys = {}
|
|
_toggled_keys = []
|
|
_waiting_keys = []
|
|
|
|
def __init__(self):
|
|
make_argumented_key(
|
|
validator=RapidFireMeta,
|
|
names=('RF',),
|
|
on_press=self._rf_pressed,
|
|
on_release=self._rf_released,
|
|
)
|
|
|
|
def _get_repeat(self, key):
|
|
if key.meta.randomize_repeat:
|
|
return key.meta.repeat + randint(
|
|
-key.meta.randomize_magnitude, key.meta.randomize_magnitude
|
|
)
|
|
return key.meta.repeat
|
|
|
|
def _on_timer_timeout(self, key, keyboard):
|
|
keyboard.tap_key(key.meta.kc)
|
|
if key in self._waiting_keys:
|
|
self._waiting_keys.remove(key)
|
|
if key.meta.toggle and key not in self._toggled_keys:
|
|
self._toggled_keys.append(key)
|
|
self._active_keys[key] = keyboard.set_timeout(
|
|
self._get_repeat(key), lambda: self._on_timer_timeout(key, keyboard)
|
|
)
|
|
|
|
def _rf_pressed(self, key, keyboard, *args, **kwargs):
|
|
if key in self._toggled_keys:
|
|
self._toggled_keys.remove(key)
|
|
self._deactivate_key(key, keyboard)
|
|
return
|
|
keyboard.tap_key(key.meta.kc)
|
|
self._waiting_keys.append(key)
|
|
self._active_keys[key] = keyboard.set_timeout(
|
|
key.meta.wait, lambda: self._on_timer_timeout(key, keyboard)
|
|
)
|
|
|
|
def _rf_released(self, key, keyboard, *args, **kwargs):
|
|
if key not in self._active_keys:
|
|
return
|
|
if key in self._toggled_keys:
|
|
if key not in self._waiting_keys:
|
|
return
|
|
self._toggled_keys.remove(key)
|
|
self._deactivate_key(key, keyboard)
|
|
|
|
def _deactivate_key(self, key, keyboard):
|
|
keyboard.cancel_timeout(self._active_keys[key])
|
|
self._active_keys.pop(key)
|
|
|
|
def during_bootup(self, keyboard):
|
|
return
|
|
|
|
def before_matrix_scan(self, keyboard):
|
|
return
|
|
|
|
def before_hid_send(self, keyboard):
|
|
return
|
|
|
|
def after_hid_send(self, keyboard):
|
|
return
|
|
|
|
def on_powersave_enable(self, keyboard):
|
|
return
|
|
|
|
def on_powersave_disable(self, keyboard):
|
|
return
|
|
|
|
def after_matrix_scan(self, keyboard):
|
|
return
|