viri-leds-dmx-sacn/lights/common.py

import logging
from colorsys import hsv_to_rgb

LOG = logging.getLogger('DMX')


class BaseDMXLight:
    def __init__(self, address):
        self.address = address
        self.intensity = 0
        self.red = 0
        self.green = 0
        self.blue = 0
        self.white = 0

    def __str__(self):
        return f'{self.name} ({self.address})'

    def _dump(self):
        raise NotImplementedError

    def dump(self):
        ret = self._dump()
        LOG.debug(f'{str(self)} -> {ret[1]}')
        return ret

    def rainbow(self, idx, angle, number_of_lights, intensity, brightness):
        my_degrees_dec = (
            (angle + (idx * (360 / number_of_lights))) % 360 / 360
        )
        r, g, b = hsv_to_rgb(
            my_degrees_dec,
            1,
            intensity / 100,
        )

        self.red = int(r * 255)
        self.green = int(g * 255)
        self.blue = int(b * 255)
        self.white = 0
        self.intensity = brightness

        return self.dump()