viri-leds-dmx-sacn/dmx_queue.py

import logging
from colorsys import hsv_to_rgb
from queue import Empty
from threading import Thread
from time import sleep

from sacn import sACNsender

LOG = logging.getLogger('DMXQueue')

class DMXQueue:
    def __init__(self, args, queue, lights):
        self.args = args
        self.queue = queue
        self.lights = lights

        self.worker = Thread(target=self._worker)
        self.worker_should_be_running = False

        self.sacn = sACNsender()

    def start(self):
        self.sacn.start()
        self.sacn.activate_output(1)

        self.sacn[1].multicast = self.args.sacn_multicast
        if not self.args.sacn_multicast:
            self.sacn[1].destination = self.args.sacn_target

        self.dmx_data = 512*[0]
        for i in range(1, 513):
            self._dmx(i, 0)

        self.worker_should_be_running = True
        self.worker.start()

    def stop(self):
        LOG.info('Waiting for worker to terminate ...')
        self.worker_should_be_running = False
        self.worker.join()
        self.sacn.stop()

    def _dmx(self, addr, data):
        self.dmx_data[addr-1] = data
        self.sacn[1].dmx_data = tuple(self.dmx_data)

    def _bulk(self, start_addr, values):
        for idx, value in enumerate(values):
            self._dmx(start_addr+idx, value)

    def _update_all(self, intensity, red, green, blue, white=0):
        for light in self.lights:
            light.intensity = intensity
            light.red = red
            light.green = green
            light.blue = blue
            light.white = white
            self._bulk(*light.dump())

    def _worker(self):
        LOG.info('Worker startup')
        rotation = 0
        while self.worker_should_be_running:
            try:
                level, component, text = self.queue.get_nowait()

                forward = list(range(17))
                reverse = list(range(17))
                reverse.reverse()

                if level == 'error':
                    # three instances of two flashes each
                    for i in range(3):
                        for j in range(4):
                            self._update_all(255, 255, 0, 0, 50)
                            sleep(0.1)
                            self._update_all(0, 255, 0, 0)
                            sleep(0.1)
                        sleep(0.2)
                elif level == 'warn':
                    # warning: blink alternate, but slow
                    for i in range(6):
                        for idx, light in enumerate(self.lights):
                            light.red = 255
                            light.green = 150
                            light.blue = 0
                            light.white = 0

                            if (idx + i) % 2:
                                light.intensity = 255
                            else:
                                light.intensity = 0

                            self._bulk(*light.dump())
                        sleep(0.5)
                        self._update_all(0, 0, 0, 0)
                elif level == 'info':
                    for i in range(2):
                        for intensity_multiplier in forward + reverse:
                            self._update_all(intensity_multiplier*15, 0, 50, 255)
                            sleep(0.03)
                self.queue.task_done()
            except Empty:
                degrees_per_step = 360 / len(self.lights)

                for idx, light in enumerate(self.lights):
                    light_degrees_dec = (
                        (rotation + (idx * degrees_per_step)) % 360 / 360
                    )
                    r, g, b = hsv_to_rgb(
                        light_degrees_dec, 1, self.args.rainbow_intensity / 100
                    )

                    light.red = int(r * 255)
                    light.green = int(g * 255)
                    light.blue = int(b * 200)
                    light.intensity = self.args.rainbow_brightness
                    self._bulk(*light.dump())

                rotation = rotation + 1
                if rotation >= 360:
                    rotation = 0

                sleep(self.args.rainbow_speed / 1000)
        LOG.info('Worker shutdown')