323 lines
10 KiB
C
323 lines
10 KiB
C
#define _POSIX_C_SOURCE 200809L
|
|
#include <assert.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <time.h>
|
|
#include <wayland-server-core.h>
|
|
#include <wayland-server-protocol.h>
|
|
#include <xkbcommon/xkbcommon.h>
|
|
#include "types/wlr_keyboard.h"
|
|
#include "util/signal.h"
|
|
#include "wlr/interfaces/wlr_keyboard.h"
|
|
#include "wlr/types/wlr_keyboard.h"
|
|
#include "wlr/types/wlr_keyboard_group.h"
|
|
#include "wlr/util/log.h"
|
|
|
|
struct keyboard_group_device {
|
|
struct wlr_keyboard *keyboard;
|
|
struct wl_listener key;
|
|
struct wl_listener modifiers;
|
|
struct wl_listener keymap;
|
|
struct wl_listener repeat_info;
|
|
struct wl_listener destroy;
|
|
struct wl_list link; // wlr_keyboard_group::devices
|
|
};
|
|
|
|
struct keyboard_group_key {
|
|
uint32_t keycode;
|
|
size_t count;
|
|
struct wl_list link; // wlr_keyboard_group::keys
|
|
};
|
|
|
|
static void keyboard_set_leds(struct wlr_keyboard *kb, uint32_t leds) {
|
|
struct wlr_keyboard_group *group = wlr_keyboard_group_from_wlr_keyboard(kb);
|
|
struct keyboard_group_device *device;
|
|
wl_list_for_each(device, &group->devices, link) {
|
|
wlr_keyboard_led_update(device->keyboard, leds);
|
|
}
|
|
}
|
|
|
|
static const struct wlr_keyboard_impl impl = {
|
|
.name = "keyboard-group",
|
|
.led_update = keyboard_set_leds
|
|
};
|
|
|
|
struct wlr_keyboard_group *wlr_keyboard_group_create(void) {
|
|
struct wlr_keyboard_group *group =
|
|
calloc(1, sizeof(struct wlr_keyboard_group));
|
|
if (!group) {
|
|
wlr_log(WLR_ERROR, "Failed to allocate wlr_keyboard_group");
|
|
return NULL;
|
|
}
|
|
|
|
wlr_keyboard_init(&group->keyboard, &impl, "wlr_keyboard_group");
|
|
wl_list_init(&group->devices);
|
|
wl_list_init(&group->keys);
|
|
|
|
wl_signal_init(&group->events.enter);
|
|
wl_signal_init(&group->events.leave);
|
|
|
|
return group;
|
|
}
|
|
|
|
struct wlr_keyboard_group *wlr_keyboard_group_from_wlr_keyboard(
|
|
struct wlr_keyboard *keyboard) {
|
|
if (keyboard->impl != &impl) {
|
|
return NULL;
|
|
}
|
|
return (struct wlr_keyboard_group *)keyboard;
|
|
}
|
|
|
|
static bool process_key(struct keyboard_group_device *group_device,
|
|
struct wlr_event_keyboard_key *event) {
|
|
struct wlr_keyboard_group *group = group_device->keyboard->group;
|
|
|
|
struct keyboard_group_key *key, *tmp;
|
|
wl_list_for_each_safe(key, tmp, &group->keys, link) {
|
|
if (key->keycode != event->keycode) {
|
|
continue;
|
|
}
|
|
if (event->state == WL_KEYBOARD_KEY_STATE_PRESSED) {
|
|
key->count++;
|
|
return false;
|
|
}
|
|
if (event->state == WL_KEYBOARD_KEY_STATE_RELEASED) {
|
|
key->count--;
|
|
if (key->count > 0) {
|
|
return false;
|
|
}
|
|
wl_list_remove(&key->link);
|
|
free(key);
|
|
}
|
|
break;
|
|
}
|
|
|
|
if (event->state == WL_KEYBOARD_KEY_STATE_PRESSED) {
|
|
struct keyboard_group_key *key =
|
|
calloc(1, sizeof(struct keyboard_group_key));
|
|
if (!key) {
|
|
wlr_log(WLR_ERROR, "Failed to allocate keyboard_group_key");
|
|
return false;
|
|
}
|
|
key->keycode = event->keycode;
|
|
key->count = 1;
|
|
wl_list_insert(&group->keys, &key->link);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
static void handle_keyboard_key(struct wl_listener *listener, void *data) {
|
|
struct keyboard_group_device *group_device =
|
|
wl_container_of(listener, group_device, key);
|
|
if (process_key(group_device, data)) {
|
|
wlr_keyboard_notify_key(&group_device->keyboard->group->keyboard, data);
|
|
}
|
|
}
|
|
|
|
static void handle_keyboard_modifiers(struct wl_listener *listener,
|
|
void *data) {
|
|
// Sync the effective layout (group modifier) to all keyboards. The rest of
|
|
// the modifiers will be derived from the wlr_keyboard_group's key state
|
|
struct keyboard_group_device *group_device =
|
|
wl_container_of(listener, group_device, modifiers);
|
|
struct wlr_keyboard_modifiers mods = group_device->keyboard->modifiers;
|
|
|
|
struct keyboard_group_device *device;
|
|
wl_list_for_each(device, &group_device->keyboard->group->devices, link) {
|
|
if (mods.depressed != device->keyboard->modifiers.depressed ||
|
|
mods.latched != device->keyboard->modifiers.latched ||
|
|
mods.locked != device->keyboard->modifiers.locked ||
|
|
mods.group != device->keyboard->modifiers.group) {
|
|
wlr_keyboard_notify_modifiers(device->keyboard,
|
|
mods.depressed, mods.latched, mods.locked, mods.group);
|
|
return;
|
|
}
|
|
}
|
|
|
|
wlr_keyboard_notify_modifiers(&group_device->keyboard->group->keyboard,
|
|
mods.depressed, mods.latched, mods.locked, mods.group);
|
|
}
|
|
|
|
static void handle_keyboard_keymap(struct wl_listener *listener, void *data) {
|
|
struct keyboard_group_device *group_device =
|
|
wl_container_of(listener, group_device, keymap);
|
|
struct wlr_keyboard *keyboard = group_device->keyboard;
|
|
|
|
if (!wlr_keyboard_keymaps_match(keyboard->group->keyboard.keymap,
|
|
keyboard->keymap)) {
|
|
struct keyboard_group_device *device;
|
|
wl_list_for_each(device, &keyboard->group->devices, link) {
|
|
if (!wlr_keyboard_keymaps_match(keyboard->keymap,
|
|
device->keyboard->keymap)) {
|
|
wlr_keyboard_set_keymap(device->keyboard, keyboard->keymap);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
wlr_keyboard_set_keymap(&keyboard->group->keyboard, keyboard->keymap);
|
|
}
|
|
|
|
static void handle_keyboard_repeat_info(struct wl_listener *listener,
|
|
void *data) {
|
|
struct keyboard_group_device *group_device =
|
|
wl_container_of(listener, group_device, repeat_info);
|
|
struct wlr_keyboard *keyboard = group_device->keyboard;
|
|
|
|
struct keyboard_group_device *device;
|
|
wl_list_for_each(device, &keyboard->group->devices, link) {
|
|
struct wlr_keyboard *devkb = device->keyboard;
|
|
if (devkb->repeat_info.rate != keyboard->repeat_info.rate ||
|
|
devkb->repeat_info.delay != keyboard->repeat_info.delay) {
|
|
wlr_keyboard_set_repeat_info(devkb, keyboard->repeat_info.rate,
|
|
keyboard->repeat_info.delay);
|
|
return;
|
|
}
|
|
}
|
|
|
|
wlr_keyboard_set_repeat_info(&keyboard->group->keyboard,
|
|
keyboard->repeat_info.rate, keyboard->repeat_info.delay);
|
|
}
|
|
|
|
static void refresh_state(struct keyboard_group_device *device,
|
|
enum wl_keyboard_key_state state) {
|
|
struct wl_array keys;
|
|
wl_array_init(&keys);
|
|
|
|
for (size_t i = 0; i < device->keyboard->num_keycodes; i++) {
|
|
struct timespec now;
|
|
clock_gettime(CLOCK_MONOTONIC, &now);
|
|
struct wlr_event_keyboard_key event = {
|
|
.time_msec = (int64_t)now.tv_sec * 1000 + now.tv_nsec / 1000000,
|
|
.keycode = device->keyboard->keycodes[i],
|
|
.update_state = true,
|
|
.state = state
|
|
};
|
|
|
|
// Update the group's key state and determine whether this is a unique
|
|
// key that needs to be passed on to the compositor
|
|
if (process_key(device, &event)) {
|
|
// Update state for wlr_keyboard_group's keyboard
|
|
keyboard_key_update(&device->keyboard->group->keyboard, &event);
|
|
keyboard_modifier_update(&device->keyboard->group->keyboard);
|
|
keyboard_led_update(&device->keyboard->group->keyboard);
|
|
|
|
// Add the key to the array
|
|
uint32_t *key = wl_array_add(&keys, sizeof(uint32_t));
|
|
*key = event.keycode;
|
|
}
|
|
}
|
|
|
|
// If there are any unique keys, emit the enter/leave event
|
|
if (keys.size > 0) {
|
|
if (state == WL_KEYBOARD_KEY_STATE_PRESSED) {
|
|
wlr_signal_emit_safe(&device->keyboard->group->events.enter, &keys);
|
|
} else {
|
|
wlr_signal_emit_safe(&device->keyboard->group->events.leave, &keys);
|
|
}
|
|
}
|
|
|
|
wl_array_release(&keys);
|
|
}
|
|
|
|
static void remove_keyboard_group_device(struct keyboard_group_device *device) {
|
|
refresh_state(device, WL_KEYBOARD_KEY_STATE_RELEASED);
|
|
device->keyboard->group = NULL;
|
|
wl_list_remove(&device->link);
|
|
wl_list_remove(&device->key.link);
|
|
wl_list_remove(&device->modifiers.link);
|
|
wl_list_remove(&device->keymap.link);
|
|
wl_list_remove(&device->repeat_info.link);
|
|
wl_list_remove(&device->destroy.link);
|
|
free(device);
|
|
}
|
|
|
|
static void handle_keyboard_destroy(struct wl_listener *listener, void *data) {
|
|
struct keyboard_group_device *device =
|
|
wl_container_of(listener, device, destroy);
|
|
remove_keyboard_group_device(device);
|
|
}
|
|
|
|
bool wlr_keyboard_group_add_keyboard(struct wlr_keyboard_group *group,
|
|
struct wlr_keyboard *keyboard) {
|
|
if (keyboard->group) {
|
|
wlr_log(WLR_ERROR, "A wlr_keyboard can only belong to one group");
|
|
return false;
|
|
}
|
|
|
|
if (keyboard->impl == &impl) {
|
|
wlr_log(WLR_ERROR, "Cannot add a group's keyboard to a group");
|
|
return false;
|
|
}
|
|
|
|
if (!wlr_keyboard_keymaps_match(group->keyboard.keymap, keyboard->keymap)) {
|
|
wlr_log(WLR_ERROR, "Device keymap does not match keyboard group's");
|
|
return false;
|
|
}
|
|
|
|
struct keyboard_group_device *device =
|
|
calloc(1, sizeof(struct keyboard_group_device));
|
|
if (!device) {
|
|
wlr_log(WLR_ERROR, "Failed to allocate keyboard_group_device");
|
|
return false;
|
|
}
|
|
|
|
device->keyboard = keyboard;
|
|
keyboard->group = group;
|
|
wl_list_insert(&group->devices, &device->link);
|
|
|
|
wl_signal_add(&keyboard->events.key, &device->key);
|
|
device->key.notify = handle_keyboard_key;
|
|
|
|
wl_signal_add(&keyboard->events.modifiers, &device->modifiers);
|
|
device->modifiers.notify = handle_keyboard_modifiers;
|
|
|
|
wl_signal_add(&keyboard->events.keymap, &device->keymap);
|
|
device->keymap.notify = handle_keyboard_keymap;
|
|
|
|
wl_signal_add(&keyboard->events.repeat_info, &device->repeat_info);
|
|
device->repeat_info.notify = handle_keyboard_repeat_info;
|
|
|
|
wl_signal_add(&keyboard->events.destroy, &device->destroy);
|
|
device->destroy.notify = handle_keyboard_destroy;
|
|
|
|
struct wlr_keyboard *group_kb = &group->keyboard;
|
|
if (keyboard->modifiers.group != group_kb->modifiers.group) {
|
|
wlr_keyboard_notify_modifiers(keyboard, keyboard->modifiers.depressed,
|
|
keyboard->modifiers.latched, keyboard->modifiers.locked,
|
|
group_kb->modifiers.group);
|
|
}
|
|
if (keyboard->repeat_info.rate != group_kb->repeat_info.rate ||
|
|
keyboard->repeat_info.delay != group_kb->repeat_info.delay) {
|
|
wlr_keyboard_set_repeat_info(keyboard, group_kb->repeat_info.rate,
|
|
group_kb->repeat_info.delay);
|
|
}
|
|
|
|
refresh_state(device, WL_KEYBOARD_KEY_STATE_PRESSED);
|
|
return true;
|
|
}
|
|
|
|
void wlr_keyboard_group_remove_keyboard(struct wlr_keyboard_group *group,
|
|
struct wlr_keyboard *keyboard) {
|
|
struct keyboard_group_device *device, *tmp;
|
|
wl_list_for_each_safe(device, tmp, &group->devices, link) {
|
|
if (device->keyboard == keyboard) {
|
|
remove_keyboard_group_device(device);
|
|
return;
|
|
}
|
|
}
|
|
wlr_log(WLR_ERROR, "keyboard not found in group");
|
|
}
|
|
|
|
void wlr_keyboard_group_destroy(struct wlr_keyboard_group *group) {
|
|
struct keyboard_group_device *device, *tmp;
|
|
wl_list_for_each_safe(device, tmp, &group->devices, link) {
|
|
wlr_keyboard_group_remove_keyboard(group, device->keyboard);
|
|
}
|
|
wlr_keyboard_finish(&group->keyboard);
|
|
wl_list_remove(&group->events.enter.listener_list);
|
|
wl_list_remove(&group->events.leave.listener_list);
|
|
free(group);
|
|
}
|