mirror of
https://github.com/sezanzeb/input-remapper
synced 2024-11-16 06:12:58 +00:00
213 lines
8.0 KiB
Python
213 lines
8.0 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/>.
|
|
|
|
|
|
import unittest
|
|
import asyncio
|
|
|
|
from evdev.ecodes import (
|
|
EV_REL,
|
|
REL_X,
|
|
REL_Y,
|
|
REL_WHEEL,
|
|
REL_HWHEEL,
|
|
EV_ABS,
|
|
ABS_X,
|
|
ABS_Y,
|
|
ABS_RX,
|
|
ABS_RY,
|
|
)
|
|
|
|
from keymapper.config import config
|
|
from keymapper.mapping import Mapping
|
|
from keymapper.injection.context import Context
|
|
from keymapper.injection.consumers.event_producer import EventProducer, MOUSE, WHEEL
|
|
|
|
from tests.test import (
|
|
InputDevice,
|
|
UInput,
|
|
MAX_ABS,
|
|
clear_write_history,
|
|
uinput_write_history,
|
|
quick_cleanup,
|
|
new_event,
|
|
MIN_ABS,
|
|
)
|
|
|
|
|
|
abs_state = [0, 0, 0, 0]
|
|
|
|
|
|
class TestEventProducer(unittest.IsolatedAsyncioTestCase):
|
|
def setUp(self):
|
|
loop = asyncio.new_event_loop()
|
|
asyncio.set_event_loop(loop)
|
|
|
|
self.mapping = Mapping()
|
|
self.context = Context(self.mapping)
|
|
|
|
uinput = UInput()
|
|
self.context.uinput = uinput
|
|
|
|
source = InputDevice("/dev/input/event30")
|
|
self.event_producer = EventProducer(self.context, source)
|
|
|
|
config.set("gamepad.joystick.x_scroll_speed", 1)
|
|
config.set("gamepad.joystick.y_scroll_speed", 1)
|
|
|
|
def tearDown(self):
|
|
quick_cleanup()
|
|
|
|
def assertClose(self, a, b, within):
|
|
"""a has to be within b - b * within, b + b * within."""
|
|
self.assertLess(a - abs(a) * within, b)
|
|
self.assertGreater(a + abs(a) * within, b)
|
|
|
|
async def test_assertClose(self):
|
|
self.assertClose(5, 5, 0.1)
|
|
self.assertClose(5, 5, 1)
|
|
self.assertClose(6, 5, 0.2)
|
|
self.assertClose(4, 5, 0.3)
|
|
self.assertRaises(AssertionError, lambda: self.assertClose(6, 5, 0.1))
|
|
self.assertRaises(AssertionError, lambda: self.assertClose(4, 5, 0.1))
|
|
|
|
self.assertClose(-5, -5, 0.1)
|
|
self.assertClose(-5, -5, 1)
|
|
self.assertClose(-6, -5, 0.2)
|
|
self.assertClose(-4, -5, 0.3)
|
|
self.assertRaises(AssertionError, lambda: self.assertClose(-6, -5, 0.1))
|
|
self.assertRaises(AssertionError, lambda: self.assertClose(-4, -5, 0.1))
|
|
|
|
async def do(self, a, b, c, d, expectation):
|
|
"""Present fake values to the loop and observe the outcome.
|
|
|
|
Depending on the configuration, the cursor or wheel should move.
|
|
"""
|
|
clear_write_history()
|
|
self.event_producer.context.update_purposes()
|
|
await self.event_producer.notify(new_event(EV_ABS, ABS_X, a))
|
|
await self.event_producer.notify(new_event(EV_ABS, ABS_Y, b))
|
|
await self.event_producer.notify(new_event(EV_ABS, ABS_RX, c))
|
|
await self.event_producer.notify(new_event(EV_ABS, ABS_RY, d))
|
|
|
|
# sleep long enough to test if multiple events are written
|
|
await asyncio.sleep(5 / 60)
|
|
|
|
history = [h.t for h in uinput_write_history]
|
|
self.assertGreater(len(history), 1)
|
|
self.assertIn(expectation, history)
|
|
|
|
for history_entry in history:
|
|
self.assertEqual(history_entry[:2], expectation[:2])
|
|
# if the injected cursor movement is 19 or 20 doesn't really matter
|
|
self.assertClose(history_entry[2], expectation[2], 0.1)
|
|
|
|
async def test_joystick_purpose_1(self):
|
|
asyncio.ensure_future(self.event_producer.run())
|
|
|
|
speed = 20
|
|
self.mapping.set("gamepad.joystick.non_linearity", 1)
|
|
self.mapping.set("gamepad.joystick.pointer_speed", speed)
|
|
self.mapping.set("gamepad.joystick.left_purpose", MOUSE)
|
|
self.mapping.set("gamepad.joystick.right_purpose", WHEEL)
|
|
|
|
min_abs = 0
|
|
# if `rest` is not exactly `max_abs / 2` decimal places might add up
|
|
# and cause higher or lower values to be written after a few events,
|
|
# which might be difficult to test.
|
|
max_abs = 256
|
|
rest = 128 # resting position of the cursor
|
|
self.event_producer.set_abs_range(min_abs, max_abs)
|
|
|
|
await self.do(max_abs, rest, rest, rest, (EV_REL, REL_X, speed))
|
|
await self.do(min_abs, rest, rest, rest, (EV_REL, REL_X, -speed))
|
|
await self.do(rest, max_abs, rest, rest, (EV_REL, REL_Y, speed))
|
|
await self.do(rest, min_abs, rest, rest, (EV_REL, REL_Y, -speed))
|
|
|
|
# vertical wheel event values are negative
|
|
await self.do(rest, rest, max_abs, rest, (EV_REL, REL_HWHEEL, 1))
|
|
await self.do(rest, rest, min_abs, rest, (EV_REL, REL_HWHEEL, -1))
|
|
await self.do(rest, rest, rest, max_abs, (EV_REL, REL_WHEEL, -1))
|
|
await self.do(rest, rest, rest, min_abs, (EV_REL, REL_WHEEL, 1))
|
|
|
|
async def test_joystick_purpose_2(self):
|
|
asyncio.ensure_future(self.event_producer.run())
|
|
|
|
speed = 30
|
|
config.set("gamepad.joystick.non_linearity", 1)
|
|
config.set("gamepad.joystick.pointer_speed", speed)
|
|
config.set("gamepad.joystick.left_purpose", WHEEL)
|
|
config.set("gamepad.joystick.right_purpose", MOUSE)
|
|
config.set("gamepad.joystick.x_scroll_speed", 1)
|
|
config.set("gamepad.joystick.y_scroll_speed", 2)
|
|
|
|
# vertical wheel event values are negative
|
|
await self.do(MAX_ABS, 0, 0, 0, (EV_REL, REL_HWHEEL, 1))
|
|
await self.do(MIN_ABS, 0, 0, 0, (EV_REL, REL_HWHEEL, -1))
|
|
await self.do(0, MAX_ABS, 0, 0, (EV_REL, REL_WHEEL, -2))
|
|
await self.do(0, MIN_ABS, 0, 0, (EV_REL, REL_WHEEL, 2))
|
|
|
|
await self.do(0, 0, MAX_ABS, 0, (EV_REL, REL_X, speed))
|
|
await self.do(0, 0, MIN_ABS, 0, (EV_REL, REL_X, -speed))
|
|
await self.do(0, 0, 0, MAX_ABS, (EV_REL, REL_Y, speed))
|
|
await self.do(0, 0, 0, MIN_ABS, (EV_REL, REL_Y, -speed))
|
|
|
|
async def test_joystick_purpose_3(self):
|
|
asyncio.ensure_future(self.event_producer.run())
|
|
|
|
speed = 40
|
|
self.mapping.set("gamepad.joystick.non_linearity", 1)
|
|
config.set("gamepad.joystick.pointer_speed", speed)
|
|
self.mapping.set("gamepad.joystick.left_purpose", MOUSE)
|
|
config.set("gamepad.joystick.right_purpose", MOUSE)
|
|
|
|
await self.do(MAX_ABS, 0, 0, 0, (EV_REL, REL_X, speed))
|
|
await self.do(MIN_ABS, 0, 0, 0, (EV_REL, REL_X, -speed))
|
|
await self.do(0, MAX_ABS, 0, 0, (EV_REL, REL_Y, speed))
|
|
await self.do(0, MIN_ABS, 0, 0, (EV_REL, REL_Y, -speed))
|
|
|
|
await self.do(0, 0, MAX_ABS, 0, (EV_REL, REL_X, speed))
|
|
await self.do(0, 0, MIN_ABS, 0, (EV_REL, REL_X, -speed))
|
|
await self.do(0, 0, 0, MAX_ABS, (EV_REL, REL_Y, speed))
|
|
await self.do(0, 0, 0, MIN_ABS, (EV_REL, REL_Y, -speed))
|
|
|
|
async def test_joystick_purpose_4(self):
|
|
asyncio.ensure_future(self.event_producer.run())
|
|
|
|
config.set("gamepad.joystick.left_purpose", WHEEL)
|
|
config.set("gamepad.joystick.right_purpose", WHEEL)
|
|
self.mapping.set("gamepad.joystick.x_scroll_speed", 2)
|
|
self.mapping.set("gamepad.joystick.y_scroll_speed", 3)
|
|
|
|
await self.do(MAX_ABS, 0, 0, 0, (EV_REL, REL_HWHEEL, 2))
|
|
await self.do(MIN_ABS, 0, 0, 0, (EV_REL, REL_HWHEEL, -2))
|
|
await self.do(0, MAX_ABS, 0, 0, (EV_REL, REL_WHEEL, -3))
|
|
await self.do(0, MIN_ABS, 0, 0, (EV_REL, REL_WHEEL, 3))
|
|
|
|
# vertical wheel event values are negative
|
|
await self.do(0, 0, MAX_ABS, 0, (EV_REL, REL_HWHEEL, 2))
|
|
await self.do(0, 0, MIN_ABS, 0, (EV_REL, REL_HWHEEL, -2))
|
|
await self.do(0, 0, 0, MAX_ABS, (EV_REL, REL_WHEEL, -3))
|
|
await self.do(0, 0, 0, MIN_ABS, (EV_REL, REL_WHEEL, 3))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|