gui15qt.py

m=['MOUSE_BUTTON_LEFT', 'MOUSE_BUTTON_MIDDLE', 'MOUSE_BUTTON_RIGHT', 'MOUSE_SCROLL_UP', 'MOUSE_SCROLL_DOWN']
k=['\t', '\n', '\r', ' ', '!', '"', '#', '$', '%', '&', "'", '(',
')', '*', '+', ',', '-', '.', '/', '0', '1', '2', '3', '4', '5', '6', '7',
'8', '9', ':', ';', '<', '=', '>', '?', '@', '[', '\\', ']', '^', '_', '`',
'a', 'b', 'c', 'd', 'e','f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o',
'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', '{', '|', '}', '~',
'accept', 'add', 'alt', 'altleft', 'altright', 'apps', 'backspace',
'browserback', 'browserfavorites', 'browserforward', 'browserhome',
'browserrefresh', 'browsersearch', 'browserstop', 'capslock', 'clear',
'convert', 'ctrl', 'ctrlleft', 'ctrlright', 'decimal', 'del', 'delete',
'divide', 'down', 'end', 'enter', 'esc', 'escape', 'execute', 'f1', 'f10',
'f11', 'f12', 'f13', 'f14', 'f15', 'f16', 'f17', 'f18', 'f19', 'f2', 'f20',
'f21', 'f22', 'f23', 'f24', 'f3', 'f4', 'f5', 'f6', 'f7', 'f8', 'f9',
'final', 'fn', 'hanguel', 'hangul', 'hanja', 'help', 'home', 'insert', 'junja',
'kana', 'kanji', 'launchapp1', 'launchapp2', 'launchmail',
'launchmediaselect', 'left', 'modechange', 'multiply', 'nexttrack',
'nonconvert', 'num0', 'num1', 'num2', 'num3', 'num4', 'num5', 'num6',
'num7', 'num8', 'num9', 'numlock', 'pagedown', 'pageup', 'pause', 'pgdn',
'pgup', 'playpause', 'prevtrack', 'print', 'printscreen', 'prntscrn',
'prtsc', 'prtscr', 'return', 'right', 'scrolllock', 'select', 'separator',
'shift', 'shiftleft', 'shiftright', 'sleep', 'space', 'stop', 'subtract', 'tab',
'up', 'volumedown', 'volumemute', 'volumeup', 'win', 'winleft', 'winright', 'yen',
'command', 'option', 'optionleft', 'optionright']

import os
os.environ["QT_ENABLE_HIGHDPI_SCALING"]   = "0"
os.environ["QT_AUTO_SCREEN_SCALE_FACTOR"] = "0"
os.environ["QT_SCALE_FACTOR"]             = "1"
from PySide6.QtWidgets import QApplication, QMainWindow, QWidget, QPushButton, QLabel, QTabWidget, QComboBox, QVBoxLayout, QHBoxLayout, QScrollArea
from PySide6.QtCore import QSize, QPoint, Qt
from PySide6.QtGui import QPainter, QPen, QPixmap, QColor, QBrush, QFont

import queue
import threading
import winsound
import time
import asyncio
from bleak import BleakClient
from enum import Enum

import ctypes
ctypes.windll.user32.SetProcessDPIAware = lambda: None
ctypes.windll.shcore.SetProcessDpiAwareness = (lambda _: None)
import pyautogui
pyautogui.FAILSAFE = False

import win32gui # pip install pywin32
from win32.win32gui import FindWindow, GetWindowRect, GetForegroundWindow, GetWindowText

class Modes(Enum):
    No_Comm = 0x0000
    Normal_Comm = 0x0100
    VR_Comm = 0x0800
    
address = "2C:BA:BA:2F:46:93" # USE YOUR OWN DEVICE ADDRESS
MODEL_UUID = "00002a00-0000-1000-8000-00805f9b34fb"
INFO_UUID = "C8C51726-81BC-483B-A052-F7A14EA3D281"
COMMAND_UUID = "C8C51726-81BC-483B-A052-F7A14EA3D282"
battery_level_characteristic = '00002a19-0000-1000-8000-00805f9b34fb'
data54 = 0
data55 = 0
data56 = 0
data58 = 0
battery = 0

touched=0
count=0
dir=0
HOLD=0.03
MAX=315 # touch pad effective range
EDGE=50 # touch pad user defined effective region
detected_titles=[]
lastX=0
lastY=0
    
def enumerate_window(): # Retrieves a list of all visible top-level window handles.    
    windows = []
    handles = []
    def enum_windows_callback(hwnd, lParam): 
        if win32gui.IsWindowVisible(hwnd): handles.append(hwnd)
        return True  # Continue enumeration
    win32gui.EnumWindows(enum_windows_callback, None)

    for handle in handles:
        try:
            title = win32gui.GetWindowText(handle)            
            if title:  # Only print handles with a visible title            
                windows.append({"id":handle, "title":title})
        except Exception as e:
            print(f"Could not get title for handle {handle}: {e}")

    #for i in range(len(windows)): print(f"[{windows[i]['id']:<8}] {windows[i]['title']}")
    #print()
    return windows
    
class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        self.setFixedSize(QSize(624*2+40,624))        
        self.setWindowTitle("Gear VR Controller Mapper")
               
        self.label = QLabel(self)
        self.pixmap = QPixmap('Gear-vr.png')        
        self.label.setPixmap(self.pixmap)
        # self.setCentralWidget(self.label)
        
        """
        layout0 = QHBoxLayout()
        layout0.setContentsMargins(0,0,0,0)
        layout0.setSpacing(10)
        layout0.addWidget(self.label)        
        layout0v = QVBoxLayout()
        layout0v.addWidget(QLabel("Target window title: (Editable and partial matching)"))
        layout0v.addWidget(QLabel("Target window title: (Editable and partial matching)"))
        layout0v.addWidget(QLabel("Target window title: (Editable and partial matching)"))
        #layout0.addLayout(layout0v)
        widget0 = QWidget()
        widget0.setLayout(layout0)
        """
        
        layout1 = QVBoxLayout()
        layout1.addWidget(QLabel("Target window title: (Editable and partial matching)"))
        c11 = QComboBox()        
        c11.setEditable(True)
        windows = enumerate_window()
        for i in range(len(windows)): c11.addItem(windows[i]['title'])        
        layout1.addWidget(c11)
        
        layout1.addWidget(QLabel("[Touchpad] button:"))
        c12 = QComboBox()                
        for i in m: c12.addItem(i)        
        layout1.addWidget(c12)
        c13 = QComboBox()                
        for i in k: c13.addItem(i)        
        layout1.addWidget(c13)
        
        #layout1.addWidget(QLabel("[Touchpad] touch: Upper region"))
        #layout1.addWidget(QLabel("[Touchpad] touch: Lower region"))
        #layout1.addWidget(QLabel("[Touchpad] touch: Left region"))
        #layout1.addWidget(QLabel("[Touchpad] touch: RIght region"))
        layout1.addWidget(QLabel("[Trigger] button:"))
        c14 = QComboBox()                
        for i in m: c14.addItem(i)        
        layout1.addWidget(c14)
        c15 = QComboBox()                
        for i in k: c15.addItem(i)        
        layout1.addWidget(c15)
        
        layout1.addWidget(QLabel("[HOME] button:"))
        layout1.addWidget(QLabel("[BACK] button:"))
        layout1.addWidget(QLabel("[+] button:"))
        layout1.addWidget(QLabel("[-] button:"))
        
        widget1 = QWidget()
        widget1.setLayout(layout1)
        scroll1 = QScrollArea()        
        scroll1.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
        scroll1.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
        scroll1.setWidgetResizable(True)
        scroll1.setWidget(widget1)        
        
        self.tabs = QTabWidget()
        self.tabs.setTabPosition(QTabWidget.East)
        #self.tabs.addTab(widget0,"LIVE")
        self.tabs.addTab(self.label, "LIVE")
        self.tabs.addTab(scroll1, "1")
        
        scroll2 = QScrollArea()        
        scroll3 = QScrollArea()                
        scroll4 = QScrollArea()        
        self.tabs.addTab(scroll2, "2")
        self.tabs.addTab(scroll3, "3")
        self.tabs.addTab(scroll4, "4")
        
        self.setCentralWidget(self.tabs)
        
        self.draw()

    def draw(self):
        canvas = self.pixmap.copy() #self.label.pixmap() # work from clean base image
        painter = QPainter(canvas)
        pen = QPen()
        pen.setWidth(3)
        
        font = QFont()
        font.setFamily('Times')
        # font.setBold(True)
        font.setPointSize(10)
        painter.setFont(font)
        
        brush = QBrush() # for buttons
        brush.setColor(QColor(0xFF, 0, 0, 0x80))
        brush.setStyle(Qt.SolidPattern)
        painter.setBrush(brush)
                
        # Draw text, battery
        pen.setColor(QColor(0xFF, 0xFF, 0xFF, 0xA0))
        painter.setPen(pen)        
        painter.drawText(20, 590, f'Battery {battery}%')
                
        c=100
        # Draw Touchpad crosshair
        x = (((data54 & 0xF) << 6) +((data55 & 0xFC) >> 2))
        y = (((data55 & 0x3) << 8) +((data56 & 0xFF) >> 0))
        if x!=0 and y!=0:
            x=(x-316//2)//2
            y=(y-316//2)//2
        if ((data58 &  8) ==  8):  color = pen.setColor(QColor(0xFF, 0, 0, 0x80))
        else: pen.setColor(QColor(0xFF, 0xFF, 0xFF, 0x80))            
        painter.setPen(pen)
        painter.drawLine(312+x, 180+y-c, 312+x, 180+y+c)
        painter.drawLine(312+x-c, 180+y, 312+x+c, 180+y)
                
        r=24
        # Draw button status
        pen.setColor(QColor(0xFF, 0, 0, 0x80))
        painter.setPen(pen)        
        if ((data58 &  1) ==  1): # "Trigger"
            painter.drawEllipse(QPoint(312, 100), 16, 16) 
            painter.drawEllipse(QPoint(312+30, 100), 16, 16)
            painter.drawEllipse(QPoint(312-30, 100), 16, 16)
        if ((data58 &  2) ==  2): painter.drawEllipse(QPoint(338, 280), r, r) # "Home"
        if ((data58 &  4) ==  4): painter.drawEllipse(QPoint(284, 280), r, r) # "Back"        
        if ((data58 & 16) == 16): painter.drawEllipse(QPoint(312, 325), r, r) # "+"
        if ((data58 & 32) == 32): painter.drawEllipse(QPoint(312, 380), r, r) # "-"

        painter.end()
        self.label.setPixmap(canvas)

running = True
async def cantusenotify(client, INFO_UUID):
    while running:
        global data54, data55, data56, data58
        data = await client.read_gatt_char(INFO_UUID) # VR mode, 30 ms report rate
        data54 = data[54]
        data55 = data[55]
        data56 = data[56]
        data58 = data[58]
        if window.tabs.currentIndex() == 0:
            if not window.isMinimized(): 
                window.draw()
                
async def run(address):
    global battery
    async with BleakClient(address) as client:
        await client.write_gatt_char(COMMAND_UUID, bytearray.fromhex("0800"), response = True) # VR mode, 30 ms report rate, NEED BOTH MODE ON
        await client.write_gatt_char(COMMAND_UUID, bytearray.fromhex("0100"), response = True) # Sensor mode, slow 200-2000 ms, NEED BOTH MODE ON
        #model_number = await client.read_gatt_char(MODEL_UUID)
        #current_mode = await client.read_gatt_char(COMMAND_UUID)
        batt = await client.read_gatt_char(battery_level_characteristic)
        battery = int.from_bytes(batt, 'big')
        #print("Model: {0} ".format("".join(map(chr,model_number))))
        #print("Mode: {0} ({1})".format("".join(current_mode.hex()), Modes(int("0x" + current_mode.hex(), base=16)).name))
        #print("Battery: ", int.from_bytes(battery, 'big'),"\n")
        winsound.Beep(1440, 500)        
        await cantusenotify(client, INFO_UUID)

app = QApplication([])
font = QFont()
font.setPointSize(6)
app.setFont(font)

window = MainWindow()
#window.setWindowFlags(window.windowFlags() | Qt.CustomizeWindowHint)
#window.setWindowFlags(window.windowFlags() & ~Qt.WindowMaximizeButtonHint) # not working
#window.setWindowFlags(window.windowFlags() & ~Qt.WindowMinMaxButtonsHint) # need Alt-F4 to close
window.show()

def bt_worker(): asyncio.run(run(address))
bt_thread = threading.Thread(target=bt_worker, args=())
bt_thread.start()

app.exec()
running = False # stop worker thread