mirror of
https://github.com/sezanzeb/input-remapper
synced 2024-11-13 19:10:50 +00:00
152 lines
5.0 KiB
Python
152 lines
5.0 KiB
Python
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
# key-mapper - GUI for device specific keyboard mappings
|
|
# Copyright (C) 2020 sezanzeb <proxima@hip70890b.de>
|
|
#
|
|
# This file is part of key-mapper.
|
|
#
|
|
# key-mapper is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# key-mapper is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with key-mapper. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
|
|
"""Keeps reading keycodes in the background for the UI to use."""
|
|
|
|
|
|
import evdev
|
|
import select
|
|
import multiprocessing
|
|
|
|
from keymapper.logger import logger
|
|
from keymapper.getdevices import get_devices, refresh_devices
|
|
from keymapper.state import KEYCODE_OFFSET
|
|
|
|
|
|
class _KeycodeReader:
|
|
"""Keeps reading keycodes in the background for the UI to use.
|
|
|
|
When a button was pressed, the newest keycode can be obtained from this
|
|
object. GTK has get_keycode for keyboard keys, but KeycodeReader also
|
|
has knowledge of buttons like the middle-mouse button.
|
|
"""
|
|
def __init__(self):
|
|
self.virtual_devices = []
|
|
self._pipe = None
|
|
self._process = None
|
|
|
|
def __del__(self):
|
|
self.stop_reading()
|
|
|
|
def stop_reading(self):
|
|
# TODO something like this for the injector?
|
|
if self._process is not None:
|
|
logger.debug('Terminating reader process')
|
|
self._process.terminate()
|
|
self._process = None
|
|
|
|
if self._pipe is not None:
|
|
logger.debug('Closing reader pipe')
|
|
self._pipe.close()
|
|
self._pipe = None
|
|
|
|
def clear(self):
|
|
"""Next time when reading don't return the previous keycode."""
|
|
# just call read to clear the pipe
|
|
self.read()
|
|
|
|
def start_reading(self, device_name):
|
|
"""Tell the evdev lib to start looking for keycodes.
|
|
|
|
If read is called without prior start_reading, no keycodes
|
|
will be available.
|
|
"""
|
|
self.stop_reading()
|
|
|
|
# make sure this sees up to date devices, including those created
|
|
# by key-mapper
|
|
refresh_devices()
|
|
|
|
self.virtual_devices = []
|
|
|
|
for name, group in get_devices(include_keymapper=True).items():
|
|
# also find stuff like "key-mapper {device}"
|
|
if device_name not in name:
|
|
continue
|
|
|
|
# Watch over each one of the potentially multiple devices per
|
|
# hardware
|
|
for path in group['paths']:
|
|
try:
|
|
device = evdev.InputDevice(path)
|
|
except FileNotFoundError:
|
|
continue
|
|
|
|
if evdev.ecodes.EV_KEY in device.capabilities():
|
|
self.virtual_devices.append(device)
|
|
|
|
logger.debug(
|
|
'Starting reading keycodes from "%s"',
|
|
'", "'.join([device.name for device in self.virtual_devices])
|
|
)
|
|
|
|
pipe = multiprocessing.Pipe()
|
|
self._process = multiprocessing.Process(
|
|
target=self._read_worker,
|
|
args=(pipe[1],)
|
|
)
|
|
self._process.start()
|
|
self._pipe = pipe[0]
|
|
|
|
def _consume_event(self, event, pipe):
|
|
"""Write the event code into the pipe if it is a key-down press."""
|
|
# value: 1 for down, 0 for up, 2 for hold.
|
|
if event.type == evdev.ecodes.EV_KEY and event.value == 1:
|
|
logger.spam(
|
|
'got code:%s value:%s',
|
|
event.code + KEYCODE_OFFSET, event.value
|
|
)
|
|
pipe.send(event.code + KEYCODE_OFFSET)
|
|
|
|
def _read_worker(self, pipe):
|
|
"""Process that reads keycodes and buffers them into a pipe."""
|
|
# using a process that blocks instead of read_one made it easier
|
|
# to debug via the logs, because the UI was not polling properly
|
|
# at some point which caused logs for events not to be written.
|
|
rlist = {device.fd: device for device in self.virtual_devices}
|
|
while True:
|
|
ready = select.select(rlist, [], [])[0]
|
|
for fd in ready:
|
|
try:
|
|
for event in rlist[fd].read():
|
|
self._consume_event(event, pipe)
|
|
except OSError:
|
|
logger.debug(
|
|
'Device "%s" disappeared from the reader',
|
|
rlist[fd].path
|
|
)
|
|
del rlist[fd]
|
|
|
|
def read(self):
|
|
"""Get the newest keycode or None if none was pressed."""
|
|
if self._pipe is None:
|
|
logger.debug('No pipe available to read from')
|
|
return None
|
|
|
|
newest_keycode = None
|
|
while self._pipe.poll():
|
|
newest_keycode = self._pipe.recv()
|
|
|
|
return newest_keycode
|
|
|
|
|
|
keycode_reader = _KeycodeReader()
|