import logging import pickle from pathlib import Path import PySimpleGUI as psg from .builder import Builder from .models import _make_bus_mode_cache, _make_output_cache from .nvda import Nvda from .parser import Parser from .util import ( _patch_insert_channels, get_asio_checkbox_index, get_insert_checkbox_index, get_patch_composite_list, open_context_menu_for_buttonmenu, ) logger = logging.getLogger(__name__) psg.theme("Dark Blue 3") class NVDAVMWindow(psg.Window): """Represents the main window of the Voicemeeter NVDA application""" DEFAULT_BIN = "default.bin" def __init__(self, title, vm): self.vm = vm self.kind = self.vm.kind self.logger = logger.getChild(type(self).__name__) self.cache = {"outputs": _make_output_cache(self.vm), "busmode": _make_bus_mode_cache(self.vm)} self.nvda = Nvda() self.parser = Parser() self.builder = Builder(self) layout = self.builder.run() super().__init__(title, layout, return_keyboard_events=True, finalize=True) buttonmenu_opts = {"takefocus": 1, "highlightthickness": 1} for i in range(self.kind.phys_out): self[f"HARDWARE OUT||A{i + 1}"].Widget.config(**buttonmenu_opts) if self.kind.name != "basic": [self[f"PATCH COMPOSITE||PC{i + 1}"].Widget.config(**buttonmenu_opts) for i in range(self.kind.phys_out)] self["ASIO BUFFER"].Widget.config(**buttonmenu_opts) self.register_events() def __enter__(self): default_config = Path.cwd() / self.DEFAULT_BIN if default_config.exists(): try: with open(default_config, "rb") as f: if config := pickle.load(f): self.vm.set("command.load", config) self.logger.debug(f"config {config} loaded") self.TKroot.after( 200, self.nvda.speak, f"config {Path(config).stem} has been loaded", ) except EOFError: self.logger.debug("no data in default bin. silently continuing...") return self def __exit__(self, exc_type, exc_value, traceback): self.close() def register_events(self): """Registers events for widgets""" # TABS self["tabs"].bind("", "||FOCUS IN") self.bind("", "CTRL-TAB") self.bind("", "CTRL-SHIFT-TAB") # Hardware Out for i in range(self.vm.kind.phys_out): self[f"HARDWARE OUT||A{i + 1}"].bind("", "||FOCUS IN") self[f"HARDWARE OUT||A{i + 1}"].bind("", "||KEY SPACE", propagate=False) self[f"HARDWARE OUT||A{i + 1}"].bind("", "||KEY ENTER", propagate=False) # Patch ASIO if self.kind.name != "basic": for i in range(self.kind.phys_out): self[f"ASIO CHECKBOX||IN{i + 1} 0"].bind("", "||FOCUS IN") self[f"ASIO CHECKBOX||IN{i + 1} 1"].bind("", "||FOCUS IN") # Patch Composite if self.kind.name != "basic": for i in range(self.vm.kind.phys_out): self[f"PATCH COMPOSITE||PC{i + 1}"].bind("", "||FOCUS IN") self[f"PATCH COMPOSITE||PC{i + 1}"].bind("", "||KEY SPACE", propagate=False) self[f"PATCH COMPOSITE||PC{i + 1}"].bind("", "||KEY ENTER", propagate=False) # Patch Insert if self.kind.name != "basic": for i in range(self.kind.num_strip): if i < self.kind.phys_in: self[f"INSERT CHECKBOX||IN{i + 1} 0"].bind("", "||FOCUS IN") self[f"INSERT CHECKBOX||IN{i + 1} 1"].bind("", "||FOCUS IN") else: [self[f"INSERT CHECKBOX||IN{i + 1} {j}"].bind("", "||FOCUS IN") for j in range(8)] # Strip Outputs for i in range(self.kind.num_strip): for j in range(self.kind.phys_out): self[f"STRIP {i}||A{j + 1}"].bind("", "||FOCUS IN") self[f"STRIP {i}||A{j + 1}"].bind("", "||KEY ENTER") for j in range(self.kind.virt_out): self[f"STRIP {i}||B{j + 1}"].bind("", "||FOCUS IN") self[f"STRIP {i}||B{j + 1}"].bind("", "||KEY ENTER") # Bus Modes for i in range(self.kind.num_bus): self[f"BUS {i}||MODE"].bind("", "||FOCUS IN") # ASIO Buffer if self.kind.name != "basic": self["ASIO BUFFER"].bind("", "||FOCUS IN") self["ASIO BUFFER"].bind("", "||KEY SPACE", propagate=False) self["ASIO BUFFER"].bind("", "||KEY ENTER", propagate=False) 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() self.logger.debug(f"event::{event}") self.logger.debug(f"values::{values}") if event in (psg.WIN_CLOSED, "Exit"): break elif event == "tabs": self.nvda.speak(f"tab {values['tabs']}") match parsed_cmd := self.parser.match.parseString(event): # Focus tabgroup case ["CTRL-TAB"] | ["CTRL-SHIFT-TAB"]: self["tabs"].set_focus() # Menus case [["Restart", "Audio", "Engine"], ["MENU"]]: self.perform_long_operation(self.vm.command.restart, "ENGINE RESTART||END") case [["ENGINE", "RESTART"], ["END"]]: self.TKroot.after( 200, self.nvda.speak, "Audio Engine restarted", ) case [["Save", "Settings"], ["MENU"]]: def popup_get_text(message, title=None): layout = [ [psg.Text(message)], [psg.Input(key="Text Input")], [psg.Button("Ok"), psg.Button("Cancel")], ] window = psg.Window(title, layout, finalize=True) window["Text Input"].bind("", "||FOCUS IN") window["Ok"].bind("", "||FOCUS IN") window["Cancel"].bind("", "||FOCUS IN") filename = None while True: event, values = window.read() if event in (psg.WIN_CLOSED, "Cancel"): break if event.endswith("||FOCUS IN"): label = event.split("||")[0] self.TKroot.after( 200 if label == "Text Input" else 1, self.nvda.speak, label, ) elif event == "Ok": filename = values["Text Input"] break window.close() return filename filename = popup_get_text("Filename", title="Save As") if filename: configpath = Path.home() / "Documents" / "Voicemeeter" / f"{filename.removesuffix('xml')}.xml" self.vm.set("command.save", str(configpath)) self.logger.debug(f"saving config file to {configpath}") self.TKroot.after( 200, self.nvda.speak, f"config file {filename} has been saved", ) case [["Load", "Settings"], ["MENU"]]: def popup_get_file(message, title=None): layout = [ [psg.Text(message)], [psg.Input(key="Text Input"), psg.FilesBrowse("Browse")], [psg.Button("Ok"), psg.Button("Cancel")], ] window = psg.Window(title, layout, finalize=True) window["Text Input"].bind("", "||FOCUS IN") window["Browse"].bind("", "||FOCUS IN") window["Ok"].bind("", "||FOCUS IN") window["Cancel"].bind("", "||FOCUS IN") filename = None while True: event, values = window.read() if event in (psg.WIN_CLOSED, "Cancel"): break if event.endswith("||FOCUS IN"): label = event.split("||")[0] self.TKroot.after( 200 if label == "Text Input" else 1, self.nvda.speak, label, ) elif event == "Ok": filename = values["Text Input"] break window.close() return filename configpath = Path.home() / "Documents" / "Voicemeeter" filename = popup_get_file("Filename", title="Load Settings") if filename: configpath = configpath / filename self.vm.set("command.load", str(configpath)) self.logger.debug(f"loading config file from {configpath}") self.TKroot.after( 200, self.nvda.speak, f"config file {Path(filename).stem} has been loaded", ) case [["Load", "Settings", "on", "Startup"], ["MENU"]]: def popup_get_text(message, title=None): layout = [ [psg.Text(message)], [psg.Input(key="Text Input")], [psg.Button("Ok"), psg.Button("Cancel")], ] window = psg.Window(title, layout, finalize=True) window["Text Input"].bind("", "||FOCUS IN") window["Ok"].bind("", "||FOCUS IN") window["Cancel"].bind("", "||FOCUS IN") filename = None while True: event, values = window.read() if event in (psg.WIN_CLOSED, "Cancel"): break if event.endswith("||FOCUS IN"): label = event.split("||")[0] self.TKroot.after( 200 if label == "Text Input" else 1, self.nvda.speak, label, ) elif event == "Ok": filename = values["Text Input"] break window.close() return filename filename = popup_get_text("Filename", title="Load on startup") if filename: configpath = Path.home() / "Documents" / "Voicemeeter" / f"{filename.removesuffix('xml')}.xml" with open(self.DEFAULT_BIN, "wb") as f: pickle.dump(str(configpath), f) self.TKroot.after( 200, self.nvda.speak, f"config {filename} set as default on startup", ) else: with open(self.DEFAULT_BIN, "wb") as f: f.truncate() self.logger.debug("default bin was truncated") # Tabs case [["tabs"], ["FOCUS", "IN"]]: self.nvda.speak(f"tab {values['tabs']}") # Hardware out case [["HARDWARE", "OUT"], [key]]: selection = values[f"HARDWARE OUT||{key}"] index = int(key[1]) - 1 match selection.split(":"): case [device_name]: setattr(self.vm.bus[index].device, "wdm", "") self.TKroot.after(200, self.nvda.speak, f"HARDWARE OUT {key} device selection removed") case [driver, device_name]: setattr(self.vm.bus[index].device, driver, device_name.strip()) phonetic = {"mme": "em em e"} self.TKroot.after( 200, self.nvda.speak, f"HARDWARE OUT {key} set {phonetic.get(driver, driver)} {device_name}", ) case [["HARDWARE", "OUT"], [key], ["FOCUS", "IN"]]: self.nvda.speak(f"HARDWARE OUT {key} {self.vm.bus[int(key[-1]) - 1].device.name}") case [["HARDWARE", "OUT"], [key], ["KEY", "SPACE" | "ENTER"]]: open_context_menu_for_buttonmenu(self, f"HARDWARE OUT||{key}") # Patch ASIO case [["ASIO", "CHECKBOX"], [in_num, channel]]: index = get_asio_checkbox_index(int(channel), int(in_num[-1])) val = values[f"ASIO CHECKBOX||{in_num} {channel}"] self.vm.patch.asio[index].set(val) channel = ("left", "right")[int(channel)] self.nvda.speak(f"Patch ASIO {in_num} {channel} set to {val}") case [["ASIO", "CHECKBOX"], [in_num, channel], ["FOCUS", "IN"]]: val = values[f"ASIO CHECKBOX||{in_num} {channel}"] channel = ("left", "right")[int(channel)] num = int(in_num[-1]) self.nvda.speak(f"Patch ASIO inputs to strips IN#{num} {channel} {val}") # Patch COMPOSITE case [["PATCH", "COMPOSITE"], [key]]: val = values[f"PATCH COMPOSITE||{key}"] index = int(key[-1]) - 1 self.vm.patch.composite[index].set(get_patch_composite_list(self.kind).index(val) + 1) self.TKroot.after(200, self.nvda.speak, f"PATCH COMPOSITE {key[-1]} set {val}") case [["PATCH", "COMPOSITE"], [key], ["FOCUS", "IN"]]: if values[f"PATCH COMPOSITE||{key}"]: val = values[f"PATCH COMPOSITE||{key}"] else: index = int(key[-1]) - 1 val = get_patch_composite_list(self.kind)[self.vm.patch.composite[index].get() - 1] self.nvda.speak(f"Patch COMPOSITE {key[-1]} {val}") case [["PATCH", "COMPOSITE"], [key], ["KEY", "SPACE" | "ENTER"]]: open_context_menu_for_buttonmenu(self, f"PATCH COMPOSITE||{key}") # Patch INSERT case [["INSERT", "CHECKBOX"], [in_num, channel]]: index = get_insert_checkbox_index( self.kind, int(channel), int(in_num[-1]), ) val = values[f"INSERT CHECKBOX||{in_num} {channel}"] self.vm.patch.insert[index].on = val self.nvda.speak( f"PATCH INSERT {in_num} {_patch_insert_channels[int(channel)]} set to {'on' if val else 'off'}" ) case [["INSERT", "CHECKBOX"], [in_num, channel], ["FOCUS", "IN"]]: index = get_insert_checkbox_index( self.kind, int(channel), int(in_num[-1]), ) val = values[f"INSERT CHECKBOX||{in_num} {channel}"] channel = _patch_insert_channels[int(channel)] num = int(in_num[-1]) self.nvda.speak(f"Patch INSERT IN#{num} {channel} {'on' if val else 'off'}") # ASIO Buffer case ["ASIO BUFFER"]: if values[event] == "Default": val = 0 else: val = values[event] self.vm.option.buffer("asio", val) self.TKroot.after(200, self.nvda.speak, f"ASIO BUFFER {val if val else 'default'}") case [["ASIO", "BUFFER"], ["FOCUS", "IN"]]: val = int(self.vm.get("option.buffer.asio")) self.nvda.speak(f"ASIO BUFFER {val if val else 'default'}") case [["ASIO", "BUFFER"], ["KEY", "SPACE" | "ENTER"]]: open_context_menu_for_buttonmenu(self, "ASIO BUFFER") # Strip outputs case [["STRIP", index], [output]]: val = not self.cache["outputs"][f"STRIP {index}||{output}"] setattr(self.vm.strip[int(index)], output, val) self.cache["outputs"][f"STRIP {index}||{output}"] = val self.nvda.speak(f"STRIP {index} {output} {label if label else ''} {'on' if val else 'off'}") case [["STRIP", index], [output], ["FOCUS", "IN"]]: val = self.cache["outputs"][f"STRIP {index}||{output}"] label = self.vm.strip[int(index)].label self.nvda.speak(f"STRIP {index} {output} {label if label else ''} {'on' if val else 'off'}") case [["STRIP", index], [output], ["KEY", "ENTER"]]: self.find_element_with_focus().click() # Bus modes case [["BUS", index], ["MODE"]]: val = self.cache["busmode"][event] if val != "normal": self.vm.bus[int(index)].mode.normal = True self.cache["busmode"][event] = "normal" else: self.vm.bus[int(index)].mode.composite = True self.cache["busmode"][event] = "composite" label = self.vm.bus[int(index)].label self.TKroot.after( 200, self.nvda.speak, f"BUS {index} {label if label else ''} bus mode {self.cache['busmode'][event]}", ) case [["BUS", index], ["MODE"], ["FOCUS", "IN"]]: label = self.vm.bus[int(index)].label self.nvda.speak( f"BUS {index} {label if label else ''} bus mode {self.cache['busmode'][f'BUS {index}||MODE']}" ) # Unknown case _: self.logger.error(f"Unknown event {event}") self.logger.debug(f"parsed::{parsed_cmd}") def request_window_object(kind_id, vm): NVDAVMWindow_cls = NVDAVMWindow return NVDAVMWindow_cls(f"Voicemeeter {kind_id.capitalize()} NVDA", vm)