43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
import board
|
|
import adafruit_ws2801
|
|
|
|
class WS2801:
|
|
def __init__(self, numleds, brightness, odata=board.MOSI, oclock=board.SCLK):
|
|
self.leds = adafruit_ws2801.WS2801(oclock, odata, numleds, brightness=brightness, auto_write=False)
|
|
self.values = []
|
|
self.brightness = brightness
|
|
self.numleds = numleds
|
|
|
|
def set_leds(self, vals):
|
|
if len(vals) == 1:
|
|
val = vals[0]
|
|
vals = []
|
|
for i in range(0, self.numleds):
|
|
vals.append(val)
|
|
|
|
if len(vals) == self.numleds:
|
|
for i in range(self.numleds):
|
|
self.leds[i] = (
|
|
int(vals[i][0] * self.brightness),
|
|
int(vals[i][1] * self.brightness),
|
|
int(vals[i][2] * self.brightness)
|
|
)
|
|
self.values = vals
|
|
else:
|
|
print("Incorrect vals length")
|
|
return
|
|
self.leds.show()
|
|
|
|
def set_brightness(self, brightness):
|
|
if brightness > 1.0:
|
|
brightness = 1.0
|
|
elif brightness < 0.0:
|
|
brightness = 0.0
|
|
|
|
self.brightness = brightness
|
|
|
|
self.set_leds(self.values)
|
|
|
|
|
|
|