renderer.py

import lib.local_debug as local_debug
from configuration import configuration
from lib.safe_logging import safe_log
from renderers.debug import Renderer

if local_debug.is_debug():
    from renderers import debug
elif configuration.get_mode() == configuration.WS2801:
    from renderers import ws2801
elif configuration.get_mode() == configuration.WS281x:
    from renderers import ws281x
else:
    from renderers import debug


def get_renderer() -> Renderer:
    """
    Returns the renderer to use based on the type of
    LED lights given in the config.

    Returns:
        renderer -- Object that takes the colors and airport config and
        sets the LEDs.
    """

    if local_debug.is_debug():
        return Renderer(configuration.CONFIG[configuration.PIXEL_COUNT_KEY])

    if configuration.get_mode() == configuration.WS2801:
        pixel_count = configuration.CONFIG[configuration.PIXEL_COUNT_KEY]
        spi_port = configuration.CONFIG[configuration.SPI_PORT_KEY]
        spi_device = configuration.CONFIG[configuration.SPI_DEVICE_KEY]

        return ws2801.Ws2801Renderer(pixel_count, spi_port, spi_device)
    elif configuration.get_mode() == configuration.WS281x:
        pixel_count = configuration.CONFIG[configuration.PIXEL_COUNT_KEY]
        gpio_pin = configuration.CONFIG[configuration.GPIO_PIN_KEY]

        safe_log("Setting up WS281x on Pin{} for {} lights".format(
            gpio_pin,
            pixel_count))

        return ws281x.Ws281xRenderer(
            pixel_count,
            gpio_pin,
            configuration.get_pixel_order())

    return Renderer(configuration.CONFIG[configuration.PIXEL_COUNT_KEY])