nvda-voicemeeter/src/nvda_voicemeeter/window.py

import logging

import PySimpleGUI as psg

from .models import _make_cache
from .nvda import Nvda
from .parser import Parser

logger = logging.getLogger(__name__)


class Window(psg.Window):
    def __init__(self, title, vm):
        self.vm = vm
        self.kind = self.vm.kind
        super().__init__(title, self.make_layout(), finalize=True)
        self.logger = logger.getChild(type(self).__name__)
        self.cache = _make_cache(self.vm)
        self.nvda = Nvda()
        self.parser = Parser()
        self.register_events()

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_value, traceback):
        self.close()

    def make_layout(self) -> list:
        """Builds the window layout step by step"""

        def add_physical_device_opts(layout):
            devices = ["{type}: {name}".format(**self.vm.device.output(i)) for i in range(self.vm.device.outs)]
            layout.append(
                [
                    psg.Combo(
                        devices,
                        size=(22, 4),
                        expand_x=True,
                        enable_events=True,
                        key=f"HARDWARE OUT||A{i}",
                    )
                    for i in range(1, self.kind.phys_out + 1)
                ]
            )

        def add_asio_checkboxes(layout, i):
            data = list(range(99))
            layout.append(
                [psg.Spin(data, initial_value=0, size=2, enable_events=True, key=f"ASIO CHECKBOX||IN{i} 0")],
            )
            layout.append(
                [psg.Spin(data, initial_value=0, size=2, enable_events=True, key=f"ASIO CHECKBOX||IN{i} 1")],
            )

        hardware_out = list()
        [step(hardware_out) for step in (add_physical_device_opts,)]
        row0 = psg.Frame("Hardware Out", hardware_out)

        inner = list()
        asio_checkboxes_in1, asio_checkboxes_in2, asio_checkboxes_in3, asio_checkboxes_in4, asio_checkboxes_in5 = (
            [] for _ in range(5)
        )
        for i, checkbox_list in enumerate(
            (
                asio_checkboxes_in1,
                asio_checkboxes_in2,
                asio_checkboxes_in3,
                asio_checkboxes_in4,
                asio_checkboxes_in5,
            )
        ):
            [step(checkbox_list, i + 1) for step in (add_asio_checkboxes,)]
            inner.append(psg.Frame(f"In#{i + 1}", checkbox_list))

        asio_checkboxes = [inner]
        row1 = psg.Frame("PATCH ASIO Inputs to Strips", asio_checkboxes)

        return [[row0], [row1]]

    def register_events(self):
        for i in range(1, self.vm.kind.phys_out + 1):
            self[f"HARDWARE OUT||A{i}"].bind("<FocusIn>", "||FOCUS IN")
        for i in range(1, 6):
            self[f"ASIO CHECKBOX||IN{i} 0"].bind("<FocusIn>", "||FOCUS IN")
            self[f"ASIO CHECKBOX||IN{i} 1"].bind("<FocusIn>", "||FOCUS IN")

    def run(self):
        """
        Parses the event string and matches it to events

        Main thread will shutdown once a close or exit event occurs
        """
        while True:
            event, values = self.read()
            if event in (psg.WIN_CLOSED, "Exit"):
                break
            match self.parser.match.parseString(event):
                case [["HARDWARE", "OUT"], [key]]:
                    selection = values[f"HARDWARE OUT||{key}"]
                    driver, device_name = selection.split(":")
                    index = int(key[1]) - 1
                    setattr(self.vm.bus[index].device, driver, device_name.strip())
                    self.nvda.speak(f"{driver} {selection}")
                case [["HARDWARE", "OUT"], [key], ["FOCUS", "IN"]]:
                    self.nvda.speak(f"HARDWARE OUT {key} in focus")
                case [["ASIO", "CHECKBOX"], [in_num, side]]:
                    if int(side) == 0:
                        index = (2 * int(in_num[-1])) - 2
                    else:
                        index = 2 * int(in_num[-1]) - 1
                    val = values[f"ASIO CHECKBOX||{in_num} {side}"]
                    self.vm.patch.asio[index].set(val)
                    side = ("left", "right")[int(side)]
                    self.nvda.speak(f"Patch ASIO {in_num} {side} set to {val}")
                case [["ASIO", "CHECKBOX"], [in_num, side], ["FOCUS", "IN"]]:
                    side = ("left", "right")[int(side)]
                    num = int(in_num[-1])
                    self.nvda.speak(f"Patch ASIO inputs to strips IN#{num} {side} in focus")
                case _:
                    self.logger.error(f"Unknown event {event}")
            self.logger.debug(self.parser.match.parseString(event))


def request_window_object(title, vm):
    WINDOW_cls = Window
    return WINDOW_cls(title, vm)