move vm_path and dll loading into cdll.py

define binds explicitly in Binds.

bump to version 0.3
This commit is contained in:
onyx-and-iris 2023-09-24 17:25:43 +01:00
parent 1b2608801f
commit f24ef8442e
5 changed files with 100 additions and 52 deletions

View File

@ -1,51 +1,41 @@
import ctypes as ct import ctypes as ct
import winreg from ctypes.wintypes import CHAR, FLOAT, LONG
from pathlib import Path
from .error import VMError from .cdll import libc
from .error import VMCAPIError
class Binds: class Binds:
VM_KEY = "VB:Voicemeeter {17359A74-1236-5467}" bind_login = libc.VBVMR_Login
BITS = 64 if ct.sizeof(ct.c_voidp) == 8 else 32 bind_login.restype = LONG
bind_login.argtypes = None
def __init__(self): bind_logout = libc.VBVMR_Logout
dll_path = Path(self.__vmpath()).parent.joinpath( bind_logout.restype = LONG
f'VoicemeeterRemote{"64" if self.BITS == 64 else ""}.dll' bind_logout.argtypes = None
)
if self.BITS == 64:
self.libc = ct.CDLL(str(dll_path))
else:
self.libc = ct.WinDLL(str(dll_path))
def __vmpath(self): bind_run_voicemeeter = libc.VBVMR_RunVoicemeeter
with winreg.OpenKey( bind_run_voicemeeter.restype = LONG
winreg.HKEY_LOCAL_MACHINE, bind_run_voicemeeter.argtypes = [LONG]
r"{}".format(
"\\".join( bind_get_voicemeeter_type = libc.VBVMR_GetVoicemeeterType
( bind_get_voicemeeter_type.restype = LONG
"\\".join( bind_get_voicemeeter_type.argtypes = [ct.POINTER(LONG)]
filter(
None, bind_is_parameters_dirty = libc.VBVMR_IsParametersDirty
( bind_is_parameters_dirty.restype = LONG
"SOFTWARE", bind_is_parameters_dirty.argtypes = None
"WOW6432Node" if self.BITS == 64 else "",
"Microsoft", bind_get_parameter_float = libc.VBVMR_GetParameterFloat
"Windows", bind_get_parameter_float.restype = LONG
"CurrentVersion", bind_get_parameter_float.argtypes = [ct.POINTER(CHAR), ct.POINTER(FLOAT)]
"Uninstall",
), bind_set_parameter_float = libc.VBVMR_SetParameterFloat
) bind_set_parameter_float.restype = LONG
), bind_set_parameter_float.argtypes = [ct.POINTER(CHAR), FLOAT]
self.VM_KEY,
)
)
),
) as vm_key:
return winreg.QueryValueEx(vm_key, r"UninstallString")[0]
def call(self, fn, *args, ok=(0,)): def call(self, fn, *args, ok=(0,)):
retval = getattr(self.libc, fn)(*args) retval = fn(*args)
if retval not in ok: if retval not in ok:
raise VMError(f"{fn} returned {retval}") raise VMCAPIError(fn.bind_namebind_, retval)
return retval return retval

View File

@ -0,0 +1,49 @@
import ctypes as ct
import platform
import winreg
from pathlib import Path
from .error import VMError
BITS = 64 if ct.sizeof(ct.c_voidp) == 8 else 32
if platform.system() != "Windows":
raise VMError("Only Windows OS supported")
VM_KEY = "VB:Voicemeeter {17359A74-1236-5467}"
REG_KEY = "\\".join(
filter(
None,
(
"SOFTWARE",
"WOW6432Node" if BITS == 64 else "",
"Microsoft",
"Windows",
"CurrentVersion",
"Uninstall",
),
)
)
def get_vmpath():
with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, r"{}".format("\\".join((REG_KEY, VM_KEY)))) as vm_key:
return winreg.QueryValueEx(vm_key, r"UninstallString")[0]
try:
vm_parent = Path(get_vmpath()).parent
except FileNotFoundError as e:
raise VMError(f"Unable to fetch DLL path from the registry") from e
DLL_NAME = f'VoicemeeterRemote{"64" if BITS == 64 else ""}.dll'
dll_path = vm_parent.joinpath(DLL_NAME)
if not dll_path.is_file():
raise VMError(f"Could not find {dll_path}")
if BITS == 64:
libc = ct.CDLL(str(dll_path))
else:
libc = ct.WinDLL(str(dll_path))

View File

@ -3,45 +3,45 @@ import ctypes as ct
from logHandler import log from logHandler import log
from .binds import Binds from .binds import Binds
from .cdll import BITS
from .context import Context, StripStrategy from .context import Context, StripStrategy
from .kinds import KindId from .kinds import KindId
class Controller: class Controller(Binds):
def __init__(self): def __init__(self):
self.binds = Binds()
self.ctx = Context(StripStrategy(self, 0)) self.ctx = Context(StripStrategy(self, 0))
def login(self): def login(self):
retval = self.binds.call("VBVMR_Login", ok=(0, 1)) retval = self.call(self.bind_login, ok=(0, 1))
log.info("INFO - logged into Voicemeeter Remote API") log.info("INFO - logged into Voicemeeter Remote API")
return retval return retval
def logout(self): def logout(self):
self.binds.call("VBVMR_Logout") self.call(self.bind_logout)
log.info("NFO - logged out of Voicemeeter Remote API") log.info("NFO - logged out of Voicemeeter Remote API")
@property @property
def kind_id(self): def kind_id(self):
c_type = ct.c_long() c_type = ct.c_long()
self.binds.call("VBVMR_GetVoicemeeterType", ct.byref(c_type)) self.call(self.bind_get_voicemeeter_type, ct.byref(c_type))
return KindId(c_type.value).name.lower() return KindId(c_type.value).name.lower()
def run_voicemeeter(self, kind_id): def run_voicemeeter(self, kind_id):
val = kind_id.value val = kind_id.value
if val == 3 and Binds.BITS == 64: if val == 3 and BITS == 64:
val = 6 val = 6
self.binds.call("VBVMR_RunVoicemeeter", val) self.call(self.bind_run_voicemeeter, val)
def __clear(self): def __clear(self):
while self.binds.call("VBVMR_IsParametersDirty", ok=(0, 1)) == 1: while self.call(self.bind_is_parameters_dirty, ok=(0, 1)) == 1:
pass pass
def _get(self, param): def _get(self, param):
self.__clear() self.__clear()
buf = ct.c_float() buf = ct.c_float()
self.binds.call("VBVMR_GetParameterFloat", param.encode(), ct.byref(buf)) self.call(self.bind_get_parameter_float, param.encode(), ct.byref(buf))
return buf.value return buf.value
def _set(self, param, val): def _set(self, param, val):
self.binds.call("VBVMR_SetParameterFloat", param.encode(), ct.c_float(float(val))) self.call(self.bind_set_parameter_float, param.encode(), ct.c_float(float(val)))

View File

@ -1,2 +1,11 @@
class VMError(Exception): class VMError(Exception):
"""Base VMError class""" """Base voicemeeterlib exception class."""
class VMCAPIError(VMError):
"""Exception raised when the C-API returns an error code"""
def __init__(self, fn_name, code):
self.fn_name = fn_name
self.code = code
super().__init__(f"{self.fn_name} returned {self.code}")

View File

@ -28,7 +28,7 @@ addon_info = {
The add-on requires Voicemeeter to be installed.""" The add-on requires Voicemeeter to be installed."""
), ),
# version # version
"addon_version": "0.2", "addon_version": "0.3",
# Author(s) # Author(s)
"addon_author": "onyx-and-iris <code@onyxandiris.online>", "addon_author": "onyx-and-iris <code@onyxandiris.online>",
# URL for the add-on documentation support # URL for the add-on documentation support