change led type from RGBW to RGB Rainbow is working

[rubkec]

i changed the line ORDER = neopixel_spi.RGBW to ORDER = neopixel_spi.RGB

now my rainbow effect is working correct, but no flash or countdown anymore. do i need to change something else?

[peteoheat]

Yeah unfortunately I wrote the code to work on LED string which is RGBW and this first version only supports that. It will be a relatively easy change to have it support others, and I'll get that done for the next version when I have time.

[lxne]

i changed the line ORDER = neopixel_spi.RGBW to ORDER = neopixel_spi.RGB

now my rainbow effect is working correct, but no flash or countdown anymore. do i need to change something else?

I had the same issue. After changing the part after 'return (r, g, b) if ORDER in (neopixel_spi.RGB, neopixel_spi.GRB) else (r, g, b, 0)' it works for me.

Here my hole file with the changes: You might have to change num_pixels, ORDER and, for correct countdown timing, the multiplier under 'def countdown(time_seconds, pixels)'

"""Plugin to manage the RGB lights via GPIO."""

import pibooth
import board
import neopixel_spi
import time
import threading
from pibooth.utils import LOGGER

__version__ = "0.0.1"
num_pixels = 32
spi = board.SPI()
ORDER = neopixel_spi.GRB

def wheel(pos):
    # Input a value 0 to 255 to get a color value.
    # The colours are a transition r - g - b - back to r.
    if pos < 0 or pos > 255:
        r = g = b = 0
    elif pos < 85:
        r = int(pos * 3)
        g = int(255 - pos * 3)
        b = 0
    elif pos < 170:
        pos -= 85
        r = int(255 - pos * 3)
        g = 0
        b = int(pos * 3)
    else:
        pos -= 170
        r = 0
        g = int(pos * 3)
        b = int(255 - pos * 3)
    return (r, g, b) if ORDER in (neopixel_spi.RGB, neopixel_spi.GRB) else (r, g, b, 0)

def thread_cycle(pixels):
    t = threading.currentThread()
    LOGGER.info("Got to thread cycle")
    LOGGER.info(pixels)
    while getattr(t, "do_run", True):
        LOGGER.info("starting rainbow cycle")
        rainbow_cycle(0.001, pixels, t)
    print("Stopping rainbow cycle")
    pixels.fill((0,0,0))
    pixels.show()

def countdown(time_seconds, pixels):
    raw_time_between_pixels = time_seconds / num_pixels
    # I needed the multiplier below in order to synchronise the on-screen countdown with my neopixel countdown
    # I arrived at this number through trial and error. Perhaps it's due to the SPI clock speed?
    # It needs to be adjusted to the LED count and preview time.
#    time_between_pixels = raw_time_between_pixels * 1.75
    time_between_pixels = raw_time_between_pixels * 1.7 # 32 LEDs, 5 sec. preview

    pixels.fill((255, 0, 0))
    pixels.show()
    for i in range(0, num_pixels):
        pixels[num_pixels - i - 1] = (255, 255, 255)
        pixels.show()
        time.sleep(time_between_pixels)
    pixels.fill((0, 0, 0))
    pixels.show()
    time.sleep(0.2)
    pixels.fill((255, 255, 255))
    pixels.show()

def rainbow_cycle(wait, pixels, t):
    LOGGER.info("running rainbow cycle")
    for j in range(255):
        for i in range(num_pixels):
            pixel_index = (i * 256 // num_pixels) + j
            pixels[i] = wheel(pixel_index & 255)
        if getattr(t, "do_run", False):
            LOGGER.info("stopping rainbow cycle")
            return
        else:
            pixels.show()
            time.sleep(wait)

@pibooth.hookimpl
def pibooth_startup(app):
    LOGGER.info("Starting up")
    pixels = neopixel_spi.NeoPixel_SPI(
        spi, num_pixels, bpp=4, brightness=0.2, auto_write=False, pixel_order=ORDER, bit0=0b10000000
    )
    app.pixels = pixels

@pibooth.hookimpl
def state_wait_enter(app):
    LOGGER.info("Starting proc")
    proc = threading.Thread(target=thread_cycle, args=[app.pixels])
    proc.daemon = True
    proc.start()
    app.neopixels_proc = proc;

def state_wait_do(app):
    LOGGER.info("wait do")

@pibooth.hookimpl
def state_wait_exit(app):
    LOGGER.info("Stopping rainbow process")
    app.neopixels_proc.do_run = False
    LOGGER.info("Stopped rainbow process")

@pibooth.hookimpl
def state_choose_enter(app):
    LOGGER.info("choosing enter")
    app.pixels.fill((255,0,0))
    app.pixels.show()

@pibooth.hookimpl
def state_preview_enter(app):
    app.pixels.fill((0,255,0))
    app.pixels.show()

    proc = threading.Thread(target=countdown, args=[3, app.pixels])
    proc.daemon = True
    proc.start()
    app.neopixels_proc = proc;
    LOGGER.info("In preview enter")

@pibooth.hookimpl
def state_preview_exit(app):
    app.neopixels_proc.do_run = False
    app.pixels.fill((255,255,255))
    app.pixels.show()
    LOGGER.info("In preview exit")

@pibooth.hookimpl
def state_capture_exit(app):
    app.pixels.fill((0,0,0))
    app.pixels.show()
    LOGGER.info("In capture exit")