smithay/anvil/src/input_handler.rs

548 lines
22 KiB
Rust

use std::{process::Command, sync::atomic::Ordering};
use crate::AnvilState;
#[cfg(feature = "udev")]
use crate::udev::UdevData;
use smithay::{
backend::input::{
self, Event, InputBackend, InputEvent, KeyState, KeyboardKeyEvent, PointerAxisEvent,
PointerButtonEvent,
},
reexports::wayland_server::protocol::wl_pointer,
wayland::{
seat::{keysyms as xkb, AxisFrame, FilterResult, Keysym, ModifiersState},
SERIAL_COUNTER as SCOUNTER,
},
};
#[cfg(any(feature = "winit", feature = "x11"))]
use smithay::backend::input::PointerMotionAbsoluteEvent;
#[cfg(feature = "udev")]
use smithay::{
backend::{
input::{
Device, DeviceCapability, PointerMotionEvent, ProximityState, TabletToolButtonEvent,
TabletToolEvent, TabletToolProximityEvent, TabletToolTipEvent, TabletToolTipState,
},
session::Session,
},
utils::{Logical, Point},
wayland::tablet_manager::{TabletDescriptor, TabletSeatTrait},
};
impl<Backend> AnvilState<Backend> {
fn process_common_key_action(&mut self, action: KeyAction) {
match action {
KeyAction::None => (),
KeyAction::Quit => {
info!(self.log, "Quitting.");
self.running.store(false, Ordering::SeqCst);
}
KeyAction::Run(cmd) => {
info!(self.log, "Starting program"; "cmd" => cmd.clone());
if let Err(e) = Command::new(&cmd).spawn() {
error!(self.log,
"Failed to start program";
"cmd" => cmd,
"err" => format!("{:?}", e)
);
}
}
_ => unreachable!(
"Common key action handler encountered backend specific action {:?}",
action
),
}
}
fn keyboard_key_to_action<B: InputBackend>(&mut self, evt: B::KeyboardKeyEvent) -> KeyAction {
let keycode = evt.key_code();
let state = evt.state();
debug!(self.log, "key"; "keycode" => keycode, "state" => format!("{:?}", state));
let serial = SCOUNTER.next_serial();
let log = &self.log;
let time = Event::time(&evt);
let suppressed_keys = &mut self.suppressed_keys;
self.keyboard
.input(keycode, state, serial, time, |modifiers, handle| {
let keysym = handle.modified_sym();
debug!(log, "keysym";
"state" => format!("{:?}", state),
"mods" => format!("{:?}", modifiers),
"keysym" => ::xkbcommon::xkb::keysym_get_name(keysym)
);
// If the key is pressed and triggered a action
// we will not forward the key to the client.
// Additionally add the key to the suppressed keys
// so that we can decide on a release if the key
// should be forwarded to the client or not.
if let KeyState::Pressed = state {
let action = process_keyboard_shortcut(*modifiers, keysym);
if action.is_some() {
suppressed_keys.push(keysym);
}
action
.map(FilterResult::Intercept)
.unwrap_or(FilterResult::Forward)
} else {
let suppressed = suppressed_keys.contains(&keysym);
if suppressed {
suppressed_keys.retain(|k| *k != keysym);
FilterResult::Intercept(KeyAction::None)
} else {
FilterResult::Forward
}
}
})
.unwrap_or(KeyAction::None)
}
fn on_pointer_button<B: InputBackend>(&mut self, evt: B::PointerButtonEvent) {
let serial = SCOUNTER.next_serial();
let button = evt.button_code();
let state = match evt.state() {
input::ButtonState::Pressed => {
// change the keyboard focus unless the pointer is grabbed
if !self.pointer.is_grabbed() {
let under = self
.window_map
.borrow_mut()
.get_surface_and_bring_to_top(self.pointer_location);
self.keyboard
.set_focus(under.as_ref().map(|&(ref s, _)| s), serial);
}
wl_pointer::ButtonState::Pressed
}
input::ButtonState::Released => wl_pointer::ButtonState::Released,
};
self.pointer.button(button, state, serial, evt.time());
}
fn on_pointer_axis<B: InputBackend>(&mut self, evt: B::PointerAxisEvent) {
let source = match evt.source() {
input::AxisSource::Continuous => wl_pointer::AxisSource::Continuous,
input::AxisSource::Finger => wl_pointer::AxisSource::Finger,
input::AxisSource::Wheel | input::AxisSource::WheelTilt => wl_pointer::AxisSource::Wheel,
};
let horizontal_amount = evt
.amount(input::Axis::Horizontal)
.unwrap_or_else(|| evt.amount_discrete(input::Axis::Horizontal).unwrap() * 3.0);
let vertical_amount = evt
.amount(input::Axis::Vertical)
.unwrap_or_else(|| evt.amount_discrete(input::Axis::Vertical).unwrap() * 3.0);
let horizontal_amount_discrete = evt.amount_discrete(input::Axis::Horizontal);
let vertical_amount_discrete = evt.amount_discrete(input::Axis::Vertical);
{
let mut frame = AxisFrame::new(evt.time()).source(source);
if horizontal_amount != 0.0 {
frame = frame.value(wl_pointer::Axis::HorizontalScroll, horizontal_amount);
if let Some(discrete) = horizontal_amount_discrete {
frame = frame.discrete(wl_pointer::Axis::HorizontalScroll, discrete as i32);
}
} else if source == wl_pointer::AxisSource::Finger {
frame = frame.stop(wl_pointer::Axis::HorizontalScroll);
}
if vertical_amount != 0.0 {
frame = frame.value(wl_pointer::Axis::VerticalScroll, vertical_amount);
if let Some(discrete) = vertical_amount_discrete {
frame = frame.discrete(wl_pointer::Axis::VerticalScroll, discrete as i32);
}
} else if source == wl_pointer::AxisSource::Finger {
frame = frame.stop(wl_pointer::Axis::VerticalScroll);
}
self.pointer.axis(frame);
}
}
}
#[cfg(any(feature = "winit", feature = "x11"))]
impl<Backend> AnvilState<Backend> {
pub fn process_input_event_windowed<B: InputBackend>(&mut self, event: InputEvent<B>, output_name: &str) {
match event {
InputEvent::Keyboard { event } => match self.keyboard_key_to_action::<B>(event) {
KeyAction::ScaleUp => {
let current_scale = {
self.output_map
.borrow()
.find_by_name(output_name)
.map(|o| o.scale())
.unwrap_or(1.0)
};
self.output_map
.borrow_mut()
.update_scale_by_name(current_scale + 0.25f32, output_name);
}
KeyAction::ScaleDown => {
let current_scale = {
self.output_map
.borrow()
.find_by_name(output_name)
.map(|o| o.scale())
.unwrap_or(1.0)
};
self.output_map
.borrow_mut()
.update_scale_by_name(f32::max(1.0f32, current_scale + 0.25f32), output_name);
}
action => match action {
KeyAction::None | KeyAction::Quit | KeyAction::Run(_) => {
self.process_common_key_action(action)
}
_ => warn!(
self.log,
"Key action {:?} unsupported on on output {} backend.", action, output_name
),
},
},
InputEvent::PointerMotionAbsolute { event } => {
self.on_pointer_move_absolute_windowed::<B>(event, output_name)
}
InputEvent::PointerButton { event } => self.on_pointer_button::<B>(event),
InputEvent::PointerAxis { event } => self.on_pointer_axis::<B>(event),
_ => (), // other events are not handled in anvil (yet)
}
}
fn on_pointer_move_absolute_windowed<B: InputBackend>(
&mut self,
evt: B::PointerMotionAbsoluteEvent,
output_name: &str,
) {
let output_size = self
.output_map
.borrow()
.find_by_name(output_name)
.map(|o| o.size())
.unwrap();
let pos = evt.position_transformed(output_size);
self.pointer_location = pos;
let serial = SCOUNTER.next_serial();
let under = self.window_map.borrow().get_surface_under(pos);
self.pointer.motion(pos, under, serial, evt.time());
}
}
#[cfg(feature = "udev")]
impl AnvilState<UdevData> {
pub fn process_input_event<B: InputBackend>(&mut self, event: InputEvent<B>) {
match event {
InputEvent::Keyboard { event, .. } => match self.keyboard_key_to_action::<B>(event) {
#[cfg(feature = "udev")]
KeyAction::VtSwitch(vt) => {
info!(self.log, "Trying to switch to vt {}", vt);
if let Err(err) = self.backend_data.session.change_vt(vt) {
error!(self.log, "Error switching to vt {}: {}", vt, err);
}
}
KeyAction::Screen(num) => {
let geometry = self.output_map.borrow().find_by_index(num).map(|o| o.geometry());
if let Some(geometry) = geometry {
let x = geometry.loc.x as f64 + geometry.size.w as f64 / 2.0;
let y = geometry.size.h as f64 / 2.0;
self.pointer_location = (x, y).into()
}
}
KeyAction::ScaleUp => {
let mut output_map = self.output_map.borrow_mut();
let output = output_map
.find_by_position(self.pointer_location.to_i32_round())
.map(|o| (o.name().to_owned(), o.location(), o.scale()));
if let Some((name, output_location, scale)) = output {
let new_scale = scale + 0.25;
output_map.update_scale_by_name(new_scale, name);
let rescale = scale as f64 / new_scale as f64;
let output_location = output_location.to_f64();
let mut pointer_output_location = self.pointer_location - output_location;
pointer_output_location.x *= rescale;
pointer_output_location.y *= rescale;
self.pointer_location = output_location + pointer_output_location;
let under = self.window_map.borrow().get_surface_under(self.pointer_location);
self.pointer
.motion(self.pointer_location, under, SCOUNTER.next_serial(), 0);
}
}
KeyAction::ScaleDown => {
let mut output_map = self.output_map.borrow_mut();
let output = output_map
.find_by_position(self.pointer_location.to_i32_round())
.map(|o| (o.name().to_owned(), o.location(), o.scale()));
if let Some((name, output_location, scale)) = output {
let new_scale = f32::max(1.0, scale - 0.25);
output_map.update_scale_by_name(new_scale, name);
let rescale = scale as f64 / new_scale as f64;
let output_location = output_location.to_f64();
let mut pointer_output_location = self.pointer_location - output_location;
pointer_output_location.x *= rescale;
pointer_output_location.y *= rescale;
self.pointer_location = output_location + pointer_output_location;
let under = self.window_map.borrow().get_surface_under(self.pointer_location);
self.pointer
.motion(self.pointer_location, under, SCOUNTER.next_serial(), 0);
}
}
action => match action {
KeyAction::None | KeyAction::Quit | KeyAction::Run(_) => {
self.process_common_key_action(action)
}
_ => unreachable!(),
},
},
InputEvent::PointerMotion { event, .. } => self.on_pointer_move::<B>(event),
InputEvent::PointerButton { event, .. } => self.on_pointer_button::<B>(event),
InputEvent::PointerAxis { event, .. } => self.on_pointer_axis::<B>(event),
InputEvent::TabletToolAxis { event, .. } => self.on_tablet_tool_axis::<B>(event),
InputEvent::TabletToolProximity { event, .. } => self.on_tablet_tool_proximity::<B>(event),
InputEvent::TabletToolTip { event, .. } => self.on_tablet_tool_tip::<B>(event),
InputEvent::TabletToolButton { event, .. } => self.on_tablet_button::<B>(event),
InputEvent::DeviceAdded { device } => {
if device.has_capability(DeviceCapability::TabletTool) {
self.seat
.tablet_seat()
.add_tablet(&TabletDescriptor::from(&device));
}
}
InputEvent::DeviceRemoved { device } => {
if device.has_capability(DeviceCapability::TabletTool) {
let tablet_seat = self.seat.tablet_seat();
tablet_seat.remove_tablet(&TabletDescriptor::from(&device));
// If there are no tablets in seat we can remove all tools
if tablet_seat.count_tablets() == 0 {
tablet_seat.clear_tools();
}
}
}
_ => {
// other events are not handled in anvil (yet)
}
}
}
fn on_pointer_move<B: InputBackend>(&mut self, evt: B::PointerMotionEvent) {
let serial = SCOUNTER.next_serial();
self.pointer_location += evt.delta();
// clamp to screen limits
// this event is never generated by winit
self.pointer_location = self.clamp_coords(self.pointer_location);
let under = self.window_map.borrow().get_surface_under(self.pointer_location);
self.pointer
.motion(self.pointer_location, under, serial, evt.time());
}
fn on_tablet_tool_axis<B: InputBackend>(&mut self, evt: B::TabletToolAxisEvent) {
let output_map = self.output_map.borrow();
let pointer_location = &mut self.pointer_location;
let tablet_seat = self.seat.tablet_seat();
let window_map = self.window_map.borrow();
let output_geometry = output_map.with_primary().map(|o| o.geometry());
if let Some(rect) = output_geometry {
*pointer_location = evt.position_transformed(rect.size) + rect.loc.to_f64();
let under = window_map.get_surface_under(*pointer_location);
let tablet = tablet_seat.get_tablet(&TabletDescriptor::from(&evt.device()));
let tool = tablet_seat.get_tool(&evt.tool());
if let (Some(tablet), Some(tool)) = (tablet, tool) {
if evt.pressure_has_changed() {
tool.pressure(evt.pressure());
}
if evt.distance_has_changed() {
tool.distance(evt.distance());
}
if evt.tilt_has_changed() {
tool.tilt(evt.tilt());
}
if evt.slider_has_changed() {
tool.slider_position(evt.slider_position());
}
if evt.rotation_has_changed() {
tool.rotation(evt.rotation());
}
if evt.wheel_has_changed() {
tool.wheel(evt.wheel_delta(), evt.wheel_delta_discrete());
}
tool.motion(
*pointer_location,
under,
&tablet,
SCOUNTER.next_serial(),
evt.time(),
);
}
}
}
fn on_tablet_tool_proximity<B: InputBackend>(&mut self, evt: B::TabletToolProximityEvent) {
let output_map = self.output_map.borrow();
let pointer_location = &mut self.pointer_location;
let tablet_seat = self.seat.tablet_seat();
let window_map = self.window_map.borrow();
let output_geometry = output_map.with_primary().map(|o| o.geometry());
if let Some(rect) = output_geometry {
let tool = evt.tool();
tablet_seat.add_tool(&tool);
*pointer_location = evt.position_transformed(rect.size) + rect.loc.to_f64();
let under = window_map.get_surface_under(*pointer_location);
let tablet = tablet_seat.get_tablet(&TabletDescriptor::from(&evt.device()));
let tool = tablet_seat.get_tool(&tool);
if let (Some(under), Some(tablet), Some(tool)) = (under, tablet, tool) {
match evt.state() {
ProximityState::In => tool.proximity_in(
*pointer_location,
under,
&tablet,
SCOUNTER.next_serial(),
evt.time(),
),
ProximityState::Out => tool.proximity_out(evt.time()),
}
}
}
}
fn on_tablet_tool_tip<B: InputBackend>(&mut self, evt: B::TabletToolTipEvent) {
let tool = self.seat.tablet_seat().get_tool(&evt.tool());
if let Some(tool) = tool {
match evt.tip_state() {
TabletToolTipState::Down => {
tool.tip_down(SCOUNTER.next_serial(), evt.time());
// change the keyboard focus unless the pointer is grabbed
if !self.pointer.is_grabbed() {
let under = self
.window_map
.borrow_mut()
.get_surface_and_bring_to_top(self.pointer_location);
let serial = SCOUNTER.next_serial();
self.keyboard
.set_focus(under.as_ref().map(|&(ref s, _)| s), serial);
}
}
TabletToolTipState::Up => {
tool.tip_up(evt.time());
}
}
}
}
fn on_tablet_button<B: InputBackend>(&mut self, evt: B::TabletToolButtonEvent) {
let tool = self.seat.tablet_seat().get_tool(&evt.tool());
if let Some(tool) = tool {
tool.button(
evt.button(),
evt.button_state(),
SCOUNTER.next_serial(),
evt.time(),
);
}
}
fn clamp_coords(&self, pos: Point<f64, Logical>) -> Point<f64, Logical> {
if self.output_map.borrow().is_empty() {
return pos;
}
let (pos_x, pos_y) = pos.into();
let output_map = self.output_map.borrow();
let max_x = output_map.width();
let clamped_x = pos_x.max(0.0).min(max_x as f64);
let max_y = output_map.height(clamped_x as i32);
if let Some(max_y) = max_y {
let clamped_y = pos_y.max(0.0).min(max_y as f64);
(clamped_x, clamped_y).into()
} else {
(clamped_x, pos_y).into()
}
}
}
/// Possible results of a keyboard action
#[derive(Debug)]
enum KeyAction {
/// Quit the compositor
Quit,
/// Trigger a vt-switch
VtSwitch(i32),
/// run a command
Run(String),
/// Switch the current screen
Screen(usize),
ScaleUp,
ScaleDown,
/// Do nothing more
None,
}
fn process_keyboard_shortcut(modifiers: ModifiersState, keysym: Keysym) -> Option<KeyAction> {
if modifiers.ctrl && modifiers.alt && keysym == xkb::KEY_BackSpace
|| modifiers.logo && keysym == xkb::KEY_q
{
// ctrl+alt+backspace = quit
// logo + q = quit
Some(KeyAction::Quit)
} else if (xkb::KEY_XF86Switch_VT_1..=xkb::KEY_XF86Switch_VT_12).contains(&keysym) {
// VTSwicth
Some(KeyAction::VtSwitch(
(keysym - xkb::KEY_XF86Switch_VT_1 + 1) as i32,
))
} else if modifiers.logo && keysym == xkb::KEY_Return {
// run terminal
Some(KeyAction::Run("weston-terminal".into()))
} else if modifiers.logo && keysym >= xkb::KEY_1 && keysym <= xkb::KEY_9 {
Some(KeyAction::Screen((keysym - xkb::KEY_1) as usize))
} else if modifiers.logo && modifiers.shift && keysym == xkb::KEY_M {
Some(KeyAction::ScaleDown)
} else if modifiers.logo && modifiers.shift && keysym == xkb::KEY_P {
Some(KeyAction::ScaleUp)
} else {
None
}
}