vban-cmd-python/vban_cmd/bus.py

import time
from abc import abstractmethod
from enum import IntEnum
from typing import Union

from .iremote import IRemote
from .meta import bus_mode_prop, channel_bool_prop, channel_label_prop

BusModes = IntEnum(
    "BusModes",
    "normal amix bmix repeat composite tvmix upmix21 upmix41 upmix61 centeronly lfeonly rearonly",
    start=0,
)


class Bus(IRemote):
    """
    Implements the common interface

    Defines concrete implementation for bus
    """

    @abstractmethod
    def __str__(self):
        pass

    @property
    def identifier(self) -> str:
        return f"bus[{self.index}]"

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

        val = self.getter("gain")
        return round(val if val else fget(), 1)

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

    def fadeto(self, target: float, time_: int):
        self.setter("FadeTo", f"({target}, {time_})")
        time.sleep(self._remote.DELAY)

    def fadeby(self, change: float, time_: int):
        self.setter("FadeBy", f"({change}, {time_})")
        time.sleep(self._remote.DELAY)


class BusEQ(IRemote):
    @classmethod
    def make(cls, remote, index):
        BUSEQ_cls = type(
            f"BusEQ{remote.kind}",
            (cls,),
            {
                **{param: channel_bool_prop(param) for param in ["on", "ab"]},
            },
        )
        return BUSEQ_cls(remote, index)

    @property
    def identifier(self) -> str:
        return f"bus[{self.index}].eq"


class PhysicalBus(Bus):
    def __str__(self):
        return f"{type(self).__name__}{self.index}"

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

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


class VirtualBus(Bus):
    def __str__(self):
        return f"{type(self).__name__}{self.index}"


class BusLevel(IRemote):
    def __init__(self, remote, index):
        super().__init__(remote, index)
        self.level_map = tuple(
            (i, i + 8)
            for i in range(0, (remote.kind.phys_out + remote.kind.virt_out) * 8, 8)
        )
        self.range = self.level_map[self.index]

    def getter(self):
        """Returns a tuple of level values for the channel."""

        def fget(i):
            return round((((1 << 16) - 1) - i) * -0.01, 1)

        if not self._remote.stopped() and self._remote.event.ldirty:
            return tuple(
                fget(i)
                for i in self._remote.cache["bus_level"][self.range[0] : self.range[-1]]
            )
        return tuple(
            fget(i)
            for i in self._remote._get_levels(self.public_packet)[1][
                self.range[0] : self.range[-1]
            ]
        )

    @property
    def identifier(self) -> str:
        return f"bus[{self.index}]"

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

    @property
    def isdirty(self) -> bool:
        """
        Returns dirty status for this specific channel.

        Expected to be used in a callback only.
        """
        return any(self._remote._bus_comp[self.range[0] : self.range[-1]])

    is_updated = isdirty


def _make_bus_mode_mixin():
    """Creates a mixin of Bus Modes."""

    modestates = {
        "normal": [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
        "amix": [1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1],
        "repeat": [0, 2, 2, 0, 0, 2, 2, 0, 0, 2, 2],
        "bmix": [1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3],
        "composite": [0, 0, 0, 4, 4, 4, 4, 0, 0, 0, 0],
        "tvmix": [1, 0, 1, 4, 5, 4, 5, 0, 1, 0, 1],
        "upmix21": [0, 2, 2, 4, 4, 6, 6, 0, 0, 2, 2],
        "upmix41": [1, 2, 3, 4, 5, 6, 7, 0, 1, 2, 3],
        "upmix61": [0, 0, 0, 0, 0, 0, 0, 8, 8, 8, 8],
        "centeronly": [1, 0, 1, 0, 1, 0, 1, 8, 9, 8, 9],
        "lfeonly": [0, 2, 2, 0, 0, 2, 2, 8, 8, 10, 10],
        "rearonly": [1, 2, 3, 0, 1, 2, 3, 8, 9, 10, 11],
    }

    def identifier(self) -> str:
        return f"bus[{self.index}].mode"

    def get(self):
        states = [
            (int.from_bytes(self.public_packet.busstate[self.index], "little") & val)
            >> 4
            for val in self._modes.modevals
        ]
        for k, v in modestates.items():
            if states == v:
                return k

    return type(
        "BusModeMixin",
        (IRemote,),
        {
            "identifier": property(identifier),
            "modestates": modestates,
            **{mode.name: bus_mode_prop(mode.name) for mode in BusModes},
            "get": get,
        },
    )


def bus_factory(phys_bus, remote, i) -> Union[PhysicalBus, VirtualBus]:
    """
    Factory method for buses

    Returns a physical or virtual bus subclass
    """
    BUS_cls = PhysicalBus if phys_bus else VirtualBus
    BUSMODEMIXIN_cls = _make_bus_mode_mixin()
    return type(
        f"{BUS_cls.__name__}{remote.kind}",
        (BUS_cls,),
        {
            "eq": BusEQ.make(remote, i),
            "levels": BusLevel(remote, i),
            "mode": BUSMODEMIXIN_cls(remote, i),
            **{param: channel_bool_prop(param) for param in ["mute", "mono"]},
            "label": channel_label_prop(),
        },
    )(remote, i)


def request_bus_obj(phys_bus, remote, i) -> Bus:
    """
    Bus entry point. Wraps factory method.

    Returns a reference to a bus subclass of a kind
    """
    return bus_factory(phys_bus, remote, i)