summaryrefslogtreecommitdiff
path: root/kvmd/plugins/hid/ch9329/mouse.py
blob: da2259c499924356c691b28fbeebf98724834333 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
# ========================================================================== #
#                                                                            #
#    KVMD - The main PiKVM daemon.                                           #
#                                                                            #
#    Copyright (C) 2018-2022  Maxim Devaev <[email protected]>               #
#                                                                            #
#    This program 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.                                     #
#                                                                            #
#    This program 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 this program.  If not, see <https://www.gnu.org/licenses/>.  #
#                                                                            #
# ========================================================================== #

import math

from ....mouse import MouseRange


class Mouse:  # pylint: disable=too-many-instance-attributes
    def __init__(self) -> None:
        self.__active = "usb"
        self.__buttons = 0x00
        self.__to_x = [0, 0]
        self.__to_y = [0, 0]
        self.__wheel_y = 0
        self.__delta_x = 0
        self.__delta_y = 0

    def button(self, button: str, clicked: bool) -> list[int]:
        code = self.__button_code(button)
        if code and self.__buttons:
            self.__buttons &= ~code
        if clicked:
            self.__buttons |= code
        self.__wheel_y = 0
        if self.__active != "usb":
            self.__to_x = [0, 0]
            self.__to_y = [0, 0]
        return self.__absolute()

    def move(self, to_x: int, to_y: int) -> list[int]:
        assert MouseRange.MIN <= to_x <= MouseRange.MAX
        assert MouseRange.MIN <= to_y <= MouseRange.MAX
        self.__to_x = self.__to_fixed(to_x)
        self.__to_y = self.__to_fixed(to_y)
        self.__wheel_y = 0
        return self.__absolute()

    def wheel(self, delta_x: int, delta_y: int) -> list[int]:
        assert -127 <= delta_y <= 127
        _ = delta_x
        self.__wheel_y = 1 if delta_y > 0 else 255
        return self.__absolute()

    def relative(self, delta_x: int, delta_y: int) -> list[int]:
        assert -127 <= delta_x <= 127
        assert -127 <= delta_y <= 127
        delta_x = math.ceil(delta_x / 3)
        delta_y = math.ceil(delta_y / 3)
        self.__delta_x = delta_x if delta_x >= 0 else 255 + delta_x
        self.__delta_y = delta_y if delta_y >= 0 else 255 + delta_y
        return self.__relative()

    def active(self) -> str:
        return self.__active

    def set_active(self, name: str) -> None:
        self.__active = name

    def __absolute(self) -> list[int]:
        cmd = [
            0x00, 0x04, 0x07, 0x02,
            self.__buttons,
            self.__to_x[1], self.__to_x[0],
            self.__to_y[1], self.__to_y[0],
            0x00]
        if self.__wheel_y:
            cmd[9] = self.__wheel_y
        return cmd

    def __relative(self) -> list[int]:
        cmd = [
            0x00, 0x05, 0x05, 0x01,
            self.__buttons,
            self.__delta_x, self.__delta_y,
            0x00]
        return cmd

    def __button_code(self, name: str) -> int:
        code = 0x00
        match name:
            case "left":
                code = 0x01
            case "right":
                code = 0x02
            case "middle":
                code = 0x04
            case "up":
                code = 0x08
            case "down":
                code = 0x10
        return code

    def __to_fixed(self, num: int) -> list[int]:
        to_fixed = math.ceil(MouseRange.remap(num, 0, MouseRange.MAX) / 8)
        return [to_fixed >> 8, to_fixed & 0xFF]