dot/scripts/tablet

117 lines
3.2 KiB
Text
Raw Normal View History

2021-10-09 21:20:41 +00:00
#!/usr/bin/python
import sys
2023-03-10 17:06:03 +00:00
import numpy as np
import os
2023-11-06 10:51:50 +00:00
from Xlib import X, display
from Xlib.ext import randr
2021-10-09 21:20:41 +00:00
2023-11-06 10:51:50 +00:00
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]")
2022-07-31 21:09:31 +00:00
2021-10-09 21:20:41 +00:00
#Arguments
2023-11-06 10:51:50 +00:00
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:")
2021-10-09 21:20:41 +00:00
2023-11-06 10:51:50 +00:00
for item in input_items:
print(f"{item[0]}: {item[1]}")
2022-07-31 21:09:31 +00:00
2023-11-06 10:51:50 +00:00
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[1]
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]
2021-10-09 21:20:41 +00:00
#Physical offset of the selected screen
offsetx = target[0]
offsety = target[1]
#Size of the full screen areas
2023-11-06 10:51:50 +00:00
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]
2021-10-09 21:20:41 +00:00
#Aspect ratio of the tablet, to avoid weird scaling problems
#Size of the tablet
2023-11-06 10:51:50 +00:00
#Prioritize the width when scaling between aspect ratios
2023-11-20 23:33:59 +00:00
tw = target[2]
2021-10-09 21:20:41 +00:00
th = tw / tablet_ratio
2023-03-10 17:06:03 +00:00
if "-r" in sys.argv or "-l" in sys.argv:
2023-11-20 23:33:59 +00:00
th = target[2]
2023-03-10 17:06:03 +00:00
tw = th / tablet_ratio
2021-10-09 21:20:41 +00:00
#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")
2023-03-10 17:06:03 +00:00
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))
2023-11-06 10:51:50 +00:00
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)
2021-10-09 21:20:41 +00:00
#alias tablet2="xinput set-prop 18 156 2.45 0 -0.725 0 2.45 -0.725 0 0 1"