vban-cmd-python/vbancmd/bus.py

from .errors import VMCMDErrors
from .channel import Channel
from . import kinds
from .meta import bus_mode_prop, channel_bool_prop, channel_label_prop


class OutputBus(Channel):
    """Base class for output buses."""

    @classmethod
    def make(cls, is_physical, remote, index, *args, **kwargs):
        """
        Factory function for output busses.
        Returns a physical/virtual bus of a kind.
        """
        BusModeMixin = _make_bus_mode_mixin(cls)
        OutputBus = PhysicalOutputBus if is_physical else VirtualOutputBus
        OB_cls = type(
            f"Bus{remote.kind.name}",
            (OutputBus,),
            {
                "levels": BusLevel(remote, index),
                "mode": BusModeMixin(remote, index),
                **{param: channel_bool_prop(param) for param in ["mute", "mono"]},
                "eq": channel_bool_prop("eq.On"),
                "eq_ab": channel_bool_prop("eq.ab"),
                "label": channel_label_prop(),
            },
        )
        return OB_cls(remote, index, *args, **kwargs)

    @property
    def identifier(self):
        return "bus"

    @property
    def gain(self) -> float:
        def fget():
            val = self.public_packet.busgain[self.index]
            if val < 10000:
                return -val
            elif val == ((1 << 16) - 1):
                return 0
            else:
                return ((1 << 16) - 1) - val

        val = self.getter("gain")
        if val is None:
            val = fget() * 0.01
        return round(val, 1)

    @gain.setter
    def gain(self, val: float):
        self.setter("gain", val)


class PhysicalOutputBus(OutputBus):
    @property
    def device(self) -> str:
        return

    @property
    def sr(self) -> int:
        return


class VirtualOutputBus(OutputBus):
    pass


class BusLevel(OutputBus):
    def __init__(self, remote, index):
        super().__init__(remote, index)
        self.level_map = _bus_maps[remote.kind.id]

    def getter_level(self, mode=None):
        def fget(i, data):
            val = data.outputlevels[i]
            return -val * 0.01

        range_ = self.level_map[self.index]
        data = self.public_packet
        levels = tuple(round(fget(i, data), 1) for i in range(*range_))
        return levels

    @property
    def all(self) -> tuple:
        return self.getter_level()


def _make_bus_level_map(kind):
    phys_out, virt_out = kind.outs
    return tuple((i, i + 8) for i in range(0, (phys_out + virt_out) * 8, 8))


_bus_maps = {kind.id: _make_bus_level_map(kind) for kind in kinds.all}


def _make_bus_mode_mixin(cls):
    """Creates a mixin of Bus Modes."""
    return type(
        "BusModeMixin",
        (cls,),
        {
            **{
                f"{mode.lower()}": bus_mode_prop(mode.lower())
                for mode in [
                    "normal",
                    "Amix",
                    "Bmix",
                    "Repeat",
                    "Composite",
                    "TVMix",
                    "UpMix21",
                    "UpMix41",
                    "UpMix61",
                    "CenterOnly",
                    "LFEOnly",
                    "RearOnly",
                ]
            },
        },
    )