Pasted

#!/usr/bin/env python3
"""
Inspired from here:
https://luator.de/linux/2022/03/05/xinput-rotate-mouse-axes.html
 
The “Coordinate Transformation Matrix” is a affine transformation matrix. To rotate by an angle α, you need the matrix
⎡cos(α)  -sin(α)  0⎤
⎢sin(α)   cos(α)  0⎥
⎣  0        0     1⎦
"""
import re
from argparse import ArgumentParser, Namespace
from math import acos, cos, degrees, radians, sin
from subprocess import run
 
LIST_REG = re.compile(
    r"(?P<name>\w[_\w\s-]+?)\s+id=(?P<id>\d+)\s+\[(master|slave)\s+pointer"
)
 
 
def rotate(degrees) -> list[float]:
    a = radians(degrees)
    return [
        cos(a),
        -sin(a),
        0.0,
        sin(a),
        cos(a),
        0.0,
        0.0,
        0.0,
        1.0,
    ]
 
 
def get_rotation(name):
    proc = run(["xinput", "list-props", name], capture_output=True, encoding="utf8")
    for line in proc.stdout.splitlines():
        if "Coordinate Transformation Matrix" in line:
            matrix = tuple(map(float, line.partition(":")[2].split(",")))
            deg = degrees(acos(matrix[0]))
            return deg
 
 
def get_args() -> Namespace:
    parser = ArgumentParser()
    sub = parser.add_subparsers(title="Commands", dest="command")
    ag_list = sub.add_parser("list", help="Command")
    ag_get = sub.add_parser("get")
    ag_get.add_argument("name", help="Name of the mouse pointer")
    ag_set = sub.add_parser("set")
    ag_reset = sub.add_parser("reset")
    ag_set.add_argument("name", help="Name of the mouse")
    ag_set.add_argument("angle", help="Angle in degrees", type=float)
    return parser.parse_args()
 
 
def list_dev(show=True):
    proc = run(["xinput", "list"], capture_output=True, encoding="utf8")
    devices = {}
 
    for line in proc.stdout.splitlines():
        if match := LIST_REG.search(line):
            name = match["name"]
            dev_id = int(match["id"])
            devices[dev_id] = name
 
    if not show:
        return devices
 
    for dev_id, name in sorted(devices.items()):
        rot = get_rotation(str(dev_id)) or 0.0
        print(f"[ {dev_id:3d} ] {name:<30} {rot:0.2f} °")
 
 
def set_prop(name: str | int, angle: float):
    name = str(name)
    transf_matrix = [str(v) for v in rotate(angle)]
    command = [
        "xinput",
        "set-prop",
        name,
        "Coordinate Transformation Matrix",
        *transf_matrix,
    ]
    run(command)
 
 
def main():
    args = get_args()
 
    match args.command:
        case "list":
            list_dev()
 
        case "get":
            deg = get_rotation(args.name)
            print(f"{deg:.2f} °")
 
        case "reset":
            for dev_id in list_dev(show=False):
                set_prop(dev_id, 0.0)
 
            list_dev()
 
        case "set":
            set_prop(args.name, args.angle)
 
 
if __name__ == "__main__":
    try:
        main()
    except Exception as e:
        print(e)