#!/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"