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)