dot/scripts/tablet
2024-04-08 11:36:09 +01:00

116 lines
3.2 KiB
Python
Executable file

#!/usr/bin/python
import sys
import numpy as np
import os
from Xlib import X, display
from Xlib.ext import randr
import subprocess
import re
# this is horriffic, gets the list of all screens
screens = [ (lambda s: [int(x) for x in (s.split("+")[1], s.split("+")[2], s.split("x")[0].split("/")[0], s.split("x")[1].split("/")[0])])(l.split(" ")[3]) for l in str(subprocess.check_output(["xrandr", "--listmonitors"])).split("\\n") if len(l.split(" ")) > 2]
#if len(sys.argv) < 3:
# print("Usage: tablet [dev number] [property number] [scale factor] [aspect ratio]")
#Arguments
scale = float(sys.argv[1]) if len(sys.argv) > 1 else 1
#tablet_ratio = (lambda s: float(s[0]) / float(s[1]))(sys.argv[4].split(":")) if len(sys.argv) > 4 else 16/9
tablet_ratio = 16/9
input_items = list(filter(lambda i: i[0], zip(subprocess.check_output(["xinput", "list", "--id-only"]).decode("ascii").split("\n"), subprocess.check_output(["xinput", "list", "--name-only"]).decode("ascii").split("\n"))))
input_items = sorted(input_items, key=(lambda i: int(i[0])))
print("Input devices found:")
for item in input_items:
print(f"{item[0]}: {item[1]}")
print()
dev = input("Select input device (id) to use: ")
prop = [ re.search(r"\((.+)\)", line).group(1) for line in subprocess.check_output(["xinput", "list-props", dev]).decode("ascii").split("\n") if re.search("Coordinate Transformation Matrix", line) ][0]
if len(screens) == 1:
target = screens[0]
else:
print("Multiple screens detected:\n")
for i, screen in enumerate(screens):
print(f"{i}: {screen}")
print()
targetno = input("Enter screen to map to (0): ")
if targetno:
target = screens[int(targetno)]
else:
target = screens[0]
#Physical offset of the selected screen
offsetx = target[0]
offsety = target[1]
#Size of the full screen areas
sh, sw = 0, 0
for screen in screens:
if screen[0]+screen[2] > sw:
sw = screen[0] + screen[2]
if screen[1]+screen[3] > sh:
sh = screen[1] + screen[3]
#Aspect ratio of the tablet, to avoid weird scaling problems
#Size of the tablet
#Prioritize the width when scaling between aspect ratios
tw = target[2]
th = tw / tablet_ratio
if "-r" in sys.argv or "-l" in sys.argv:
th = target[2]
tw = th / tablet_ratio
#make a variable that will try to make the scale the same over diferent resoultions
f = target[3] / target[3]
print ("f value =", f)
#calculate the new area size
aw = tw * scale * f
ah = th * scale * f
ox = (target[2] - aw) / 2 + offsetx
oy = (target[3] - ah) / 2 + offsety
#ox = offsetx
#oy = offsety
c0 = aw / sw
c2 = ah / sh
c1 = ox / sw
c3 = oy / sh
print(str(c0), "0", str(c1), "0", str(c2), str(c3), "0 0 1")
transform = np.matrix([[c0, 0, c1], [0, c2, c3], [0, 0, 1]])
leftrotate = np.matrix([[0, -1, 1], [1, 0, 0], [0, 0, 1]])
rightrotate = np.matrix([[0, 1, 0], [-1, 0, 1], [0, 0, 1]])
if "-r" in sys.argv:
transform *= rightrotate
if "-l" in sys.argv:
transform *= leftrotate
arr = np.squeeze(np.asarray(transform))
cmd = " ".join(["xinput set-prop", dev, prop, " ".join([str(x) for x in [arr[0, 0], arr[0, 1], arr[0, 2], arr[1, 0], arr[1, 1], arr[1, 2], arr[2, 0], arr[2, 1], arr[2, 2]]])])
print(cmd)
os.system(cmd)
#alias tablet2="xinput set-prop 18 156 2.45 0 -0.725 0 2.45 -0.725 0 0 1"