You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
input-remapper/keymapper/gui/row.py

410 lines
13 KiB
Python

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# key-mapper - GUI for device specific keyboard mappings
# Copyright (C) 2021 sezanzeb <proxima@sezanzeb.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/>.
"""A single, configurable key mapping."""
import evdev
from gi.repository import Gtk, GLib, Gdk
from keymapper.system_mapping import system_mapping
from keymapper.gui.custom_mapping import custom_mapping
from keymapper.logger import logger
from keymapper.key import Key
from keymapper.gui.reader import reader
CTX_KEYCODE = 2
store = Gtk.ListStore(str)
def populate_store():
"""Fill the dropdown for key suggestions with values."""
for name in system_mapping.list_names():
store.append([name])
extra = [
"mouse(up, 1)",
"mouse(down, 1)",
"mouse(left, 1)",
"mouse(right, 1)",
"wheel(up, 1)",
"wheel(down, 1)",
"wheel(left, 1)",
"wheel(right, 1)",
]
for key in extra:
# add some more keys to the dropdown list
store.append([key])
populate_store()
def to_string(key):
"""A nice to show description of the pressed key."""
if isinstance(key, Key):
return " + ".join([to_string(sub_key) for sub_key in key])
if isinstance(key[0], tuple):
raise Exception("deprecated stuff")
ev_type, code, value = key
if ev_type not in evdev.ecodes.bytype:
logger.error("Unknown key type for %s", key)
return "unknown"
if code not in evdev.ecodes.bytype[ev_type]:
logger.error("Unknown key code for %s", key)
return "unknown"
key_name = None
# first try to find the name in xmodmap to not display wrong
# names due to the keyboard layout
if ev_type == evdev.ecodes.EV_KEY:
key_name = system_mapping.get_name(code)
if key_name is None:
# if no result, look in the linux key constants. On a german
# keyboard for example z and y are switched, which will therefore
# cause the wrong letter to be displayed.
key_name = evdev.ecodes.bytype[ev_type][code]
if isinstance(key_name, list):
key_name = key_name[0]
if ev_type != evdev.ecodes.EV_KEY:
direction = {
# D-Pad
(evdev.ecodes.ABS_HAT0X, -1): "Left",
(evdev.ecodes.ABS_HAT0X, 1): "Right",
(evdev.ecodes.ABS_HAT0Y, -1): "Up",
(evdev.ecodes.ABS_HAT0Y, 1): "Down",
(evdev.ecodes.ABS_HAT1X, -1): "Left",
(evdev.ecodes.ABS_HAT1X, 1): "Right",
(evdev.ecodes.ABS_HAT1Y, -1): "Up",
(evdev.ecodes.ABS_HAT1Y, 1): "Down",
(evdev.ecodes.ABS_HAT2X, -1): "Left",
(evdev.ecodes.ABS_HAT2X, 1): "Right",
(evdev.ecodes.ABS_HAT2Y, -1): "Up",
(evdev.ecodes.ABS_HAT2Y, 1): "Down",
# joystick
(evdev.ecodes.ABS_X, 1): "Right",
(evdev.ecodes.ABS_X, -1): "Left",
(evdev.ecodes.ABS_Y, 1): "Down",
(evdev.ecodes.ABS_Y, -1): "Up",
(evdev.ecodes.ABS_RX, 1): "Right",
(evdev.ecodes.ABS_RX, -1): "Left",
(evdev.ecodes.ABS_RY, 1): "Down",
(evdev.ecodes.ABS_RY, -1): "Up",
# wheel
(evdev.ecodes.REL_WHEEL, -1): "Down",
(evdev.ecodes.REL_WHEEL, 1): "Up",
(evdev.ecodes.REL_HWHEEL, -1): "Left",
(evdev.ecodes.REL_HWHEEL, 1): "Right",
}.get((code, value))
if direction is not None:
key_name += f" {direction}"
key_name = key_name.replace("ABS_Z", "Trigger Left")
key_name = key_name.replace("ABS_RZ", "Trigger Right")
key_name = key_name.replace("ABS_HAT0X", "DPad")
key_name = key_name.replace("ABS_HAT0Y", "DPad")
key_name = key_name.replace("ABS_HAT1X", "DPad 2")
key_name = key_name.replace("ABS_HAT1Y", "DPad 2")
key_name = key_name.replace("ABS_HAT2X", "DPad 3")
key_name = key_name.replace("ABS_HAT2Y", "DPad 3")
key_name = key_name.replace("ABS_X", "Joystick")
key_name = key_name.replace("ABS_Y", "Joystick")
key_name = key_name.replace("ABS_RX", "Joystick 2")
key_name = key_name.replace("ABS_RY", "Joystick 2")
key_name = key_name.replace("BTN_", "Button ")
key_name = key_name.replace("KEY_", "")
key_name = key_name.replace("REL_", "")
key_name = key_name.replace("HWHEEL", "Wheel")
key_name = key_name.replace("WHEEL", "Wheel")
key_name = key_name.replace("_", " ")
key_name = key_name.replace(" ", " ")
return key_name
IDLE = 0
HOLDING = 1
class Row(Gtk.ListBoxRow):
"""A single, configurable key mapping."""
__gtype_name__ = "ListBoxRow"
def __init__(self, delete_callback, window, key=None, symbol=None):
"""Construct a row widget.
Parameters
----------
key : Key
"""
if key is not None and not isinstance(key, Key):
raise TypeError("Expected key to be a Key object")
super().__init__()
self.device = window.group
self.window = window
self.delete_callback = delete_callback
self.symbol_input = None
self.keycode_input = None
self.key = key
self.put_together(symbol)
self._state = IDLE
def refresh_state(self):
"""Refresh the state.
The state is needed to switch focus when no keys are held anymore,
but only if the row has been in the HOLDING state before.
"""
old_state = self._state
if not self.keycode_input.is_focus():
self._state = IDLE
return
unreleased_keys = reader.get_unreleased_keys()
if unreleased_keys is None and old_state == HOLDING and self.key:
# A key was pressed and then released.
# Switch to the symbol. idle_add this so that the
# keycode event won't write into the symbol input as well.
window = self.window.window
GLib.idle_add(lambda: window.set_focus(self.symbol_input))
if unreleased_keys is not None:
self._state = HOLDING
return
self._state = IDLE
def get_key(self):
"""Get the Key object from the left column.
Or None if no code is mapped on this row.
"""
return self.key
def get_symbol(self):
"""Get the assigned symbol from the middle column."""
symbol = self.symbol_input.get_text()
return symbol if symbol else None
def set_new_key(self, new_key):
"""Check if a keycode has been pressed and if so, display it.
Parameters
----------
new_key : Key
"""
if new_key is not None and not isinstance(new_key, Key):
raise TypeError("Expected new_key to be a Key object")
# the newest_keycode is populated since the ui regularly polls it
# in order to display it in the status bar.
previous_key = self.get_key()
# no input
if new_key is None:
return
# it might end up being a key combination
self._state = HOLDING
# keycode didn't change, do nothing
if new_key == previous_key:
return
# keycode is already set by some other row
existing = custom_mapping.get_symbol(new_key)
if existing is not None:
msg = f'"{to_string(new_key)}" already mapped to "{existing}"'
logger.info(msg)
self.window.show_status(CTX_KEYCODE, msg)
return
# it's legal to display the keycode
# always ask for get_child to set the label, otherwise line breaking
# has to be configured again.
self.set_keycode_input_label(to_string(new_key))
self.key = new_key
symbol = self.get_symbol()
# the symbol is empty and therefore the mapping is not complete
if symbol is None:
return
# else, the keycode has changed, the symbol is set, all good
custom_mapping.change(new_key=new_key, symbol=symbol, previous_key=previous_key)
def on_symbol_input_change(self, _):
"""When the output symbol for that keycode is typed in."""
key = self.get_key()
symbol = self.get_symbol()
if symbol is None:
return
if key is not None:
custom_mapping.change(new_key=key, symbol=symbol, previous_key=None)
def match(self, _, key, tree_iter):
"""Search the avilable names."""
value = store.get_value(tree_iter, 0)
return key in value.lower()
def show_click_here(self):
"""Show 'click here' on the keycode input button."""
if self.get_key() is not None:
return
self.set_keycode_input_label("click here")
self.keycode_input.set_opacity(0.3)
def show_press_key(self):
"""Show 'press key' on the keycode input button."""
if self.get_key() is not None:
return
self.set_keycode_input_label("press key")
self.keycode_input.set_opacity(1)
def on_keycode_input_focus(self, *_):
"""Refresh useful usage information."""
reader.clear()
self.show_press_key()
self.window.can_modify_mapping()
def on_keycode_input_unfocus(self, *_):
"""Refresh useful usage information and set some state stuff."""
self.show_click_here()
self.keycode_input.set_active(False)
self._state = IDLE
self.window.save_preset()
def set_keycode_input_label(self, label):
"""Set the label of the keycode input."""
self.keycode_input.set_label(label)
# make the child label widget break lines, important for
# long combinations
label = self.keycode_input.get_child()
label.set_line_wrap(True)
label.set_line_wrap_mode(2)
label.set_max_width_chars(13)
label.set_justify(Gtk.Justification.CENTER)
self.keycode_input.set_opacity(1)
def on_symbol_input_unfocus(self, symbol_input, _):
"""Save the preset and correct the input casing."""
symbol = symbol_input.get_text()
correct_case = system_mapping.correct_case(symbol)
if symbol != correct_case:
symbol_input.set_text(correct_case)
self.window.save_preset()
def put_together(self, symbol):
"""Create all child GTK widgets and connect their signals."""
delete_button = Gtk.EventBox()
delete_button.add(
Gtk.Image.new_from_icon_name("window-close", Gtk.IconSize.BUTTON)
)
delete_button.connect("button-press-event", self.on_delete_button_clicked)
delete_button.set_size_request(50, -1)
keycode_input = Gtk.ToggleButton()
self.keycode_input = keycode_input
keycode_input.set_size_request(140, -1)
if self.key is not None:
self.set_keycode_input_label(to_string(self.key))
else:
self.show_click_here()
# make the togglebutton go back to its normal state when doing
# something else in the UI
keycode_input.connect("focus-in-event", self.on_keycode_input_focus)
keycode_input.connect("focus-out-event", self.on_keycode_input_unfocus)
# don't leave the input when using arrow keys or tab. wait for the
# window to consume the keycode from the reader
keycode_input.connect("key-press-event", lambda *args: Gdk.EVENT_STOP)
symbol_input = Gtk.Entry()
self.symbol_input = symbol_input
symbol_input.set_alignment(0.5)
symbol_input.set_width_chars(4)
symbol_input.set_has_frame(False)
completion = Gtk.EntryCompletion()
completion.set_model(store)
completion.set_text_column(0)
completion.set_match_func(self.match)
symbol_input.set_completion(completion)
if symbol is not None:
symbol_input.set_text(symbol)
symbol_input.connect("changed", self.on_symbol_input_change)
symbol_input.connect("focus-out-event", self.on_symbol_input_unfocus)
box = Gtk.Box(orientation=Gtk.Orientation.HORIZONTAL)
box.set_homogeneous(False)
box.set_spacing(0)
box.pack_start(keycode_input, expand=False, fill=True, padding=0)
box.pack_start(symbol_input, expand=True, fill=True, padding=0)
box.pack_start(delete_button, expand=False, fill=True, padding=0)
box.show_all()
box.get_style_context().add_class("row-box")
self.add(box)
self.show_all()
def on_delete_button_clicked(self, *_):
"""Destroy the row and remove it from the config."""
key = self.get_key()
if key is not None:
custom_mapping.clear(key)
self.symbol_input.set_text("")
self.set_keycode_input_label("")
self.key = None
self.delete_callback(self)