BAPSicle/helpers/device_manager.py

from typing import Any, Dict, List, Optional, Tuple
import sounddevice as sd
from helpers.os_environment import isLinux, isMacOS, isWindows
import glob

# TODO: https://wiki.libsdl.org/FAQUsingSDL maybe try setting some of these env variables for choosing different host APIs?
WINDOWS_APIS = ["Windows DirectSound"]


class DeviceManager:
    @classmethod
    def _isOutput(cls, device: Dict[str, Any]) -> bool:
        return device["max_output_channels"] > 0

    @classmethod
    def _isHostAPI(cls, host_api) -> bool:
        return host_api

    @classmethod
    def _getAudioDevices(cls) -> sd.DeviceList:
        # To update the list of devices
        # Sadly this doesn't work on MacOS.
        if not isMacOS():
            sd._terminate()
            sd._initialize()
        devices: sd.DeviceList = sd.query_devices()
        return devices

    @classmethod
    def getAudioOutputs(cls) -> Tuple[List[Dict]]:
        host_apis = sd.query_hostapis()
        devices: sd.DeviceList = cls._getAudioDevices()

        valid_host_apis = []
        for host_api_id in range(len(host_apis)):
            if isWindows() and host_apis[host_api_id]["name"] not in WINDOWS_APIS:
                continue

            host_api_devices = (device for device in devices if device["hostapi"] == host_api_id)

            outputs: List[Dict] = list(filter(cls._isOutput, host_api_devices))
            outputs = sorted(outputs, key=lambda k: k["name"])

            valid_host_apis.append(host_apis[host_api_id])
            valid_host_apis[-1]["output_devices"] = outputs

        return host_apis

    @classmethod
    def getSerialPorts(cls) -> List[Optional[str]]:
        """Lists serial port names

        :raises EnvironmentError:
            On unsupported or unknown platforms
        :returns:
            A list of the serial ports available on the system
        """
        # TODO: Get list of COM ports properly. (Can't use )
        if isWindows():
            ports = ["COM%s" % (i + 1) for i in range(8)]
        elif isLinux():
            # this excludes your current terminal "/dev/tty"
            ports = glob.glob("/dev/tty[A-Za-z]*")
        elif isMacOS():
            ports = glob.glob("/dev/tty.*")
        else:
            raise EnvironmentError("Unsupported platform")

        valid: List[str] = ports

        result: List[Optional[str]] = []

        if len(valid) > 0:
            valid.sort()

        result.append(None)  # Add the None option
        result.extend(valid)

        return result