leds/light_manager.py

from time import sleep
import neopixel
import board
import logging

def defaultPattern(n, t):
    return (((n + t) * 5) % 255, (t * 42) % 255, (t * 50) % 255)

class LightStrip:
  def __init__(self, data_pin = board.D18, string_length = 300, brightness = 1, pixel_order = neopixel.GRB, max_changes = 5):
        self.data_pin = data_pin

        self.np = neopixel.NeoPixel(self.data_pin, string_length, brightness = brightness, auto_write=True, pixel_order = pixel_order)
        self.pattern = defaultPattern
        self.cur_tick = 0
        self.cur_index = 0
        self.max_changes = max_changes
        self.cur_pattern = []

  def set_light_level(self, level):
    self.np.brightness = level
  def pattern(self, pattern_callback):
    self.pattern = pattern_callback

  def get_next_pattern_tick(self):
    n = self.np.n
    t = self.cur_tick
    if not len(self.cur_pattern) >= n - 1:
      li = [(255, 255, 255)] * (n - len(self.cur_pattern))
      self.cur_pattern.extend(li)
    for i in range(n):
      self.cur_pattern[i] = self.pattern(i, t)
    self.cur_tick = t + 1
  
  def tick(self):
    np = self.np
    t = self.cur_tick
    n = np.n
    
    if not len(self.cur_pattern) >= n - 1:
      self.get_next_pattern_tick()
    changes = 0
    ind = self.cur_index
    for i in range(ind, n):
      col = self.cur_pattern[i]
      self.cur_index = i
      logging.debug("TEST VALUES: np {} col {} same {}".format(np[i], col, tuple(np[i]) != col))
      if tuple(np[i]) != col:
        changes = changes + 1
        np[i] = col
        logging.debug("CHANGE COLOR OF PIXEL {} TO {} ON TICK {}".format(i, col, t))
        if changes >= self.max_changes:
          break;
    if self.cur_index >= (n-1):
      self.get_next_pattern_tick()
      self.cur_index = 0