59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
OFF = (0, 0, 0)
|
|
RED = (255, 0, 0)
|
|
YELLOW = (255, 150, 0)
|
|
GREEN = (0, 255, 0)
|
|
CYAN = (0, 255, 255)
|
|
BLUE = (0, 0, 255, 0)
|
|
PURPLE = (180, 0, 255)
|
|
|
|
|
|
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:
|
|
return (0, 0, 0)
|
|
if pos < 85:
|
|
return (255 - pos * 3, pos * 3, 0)
|
|
if pos < 170:
|
|
pos -= 85
|
|
return (0, 255 - pos * 3, pos * 3)
|
|
pos -= 170
|
|
return (pos * 3, 0, 255 - pos * 3)
|
|
|
|
|
|
def color_chase(pixels, num_pixels, color, color2=OFF, speed=100, animation_state=0):
|
|
if animation_state not in range(num_pixels):
|
|
color = color2
|
|
pixels[int(animation_state - num_pixels)] = color
|
|
|
|
else:
|
|
pixels[animation_state] = color
|
|
|
|
pixels.show()
|
|
animation_state += 1
|
|
|
|
if animation_state >= num_pixels * 2:
|
|
animation_state = 0
|
|
|
|
return animation_state, 0
|
|
|
|
|
|
def rainbow_cycle(pixels, num_pixels, speed=100, animation_state=0, color_state=0):
|
|
color_state += 1
|
|
if color_state in range(255):
|
|
print(animation_state)
|
|
animation_state +=1
|
|
if animation_state in range(num_pixels):
|
|
rc_index = (animation_state * 256 // num_pixels) + animation_state
|
|
print(pixels[animation_state])
|
|
print(wheel(rc_index & 255))
|
|
pixels[animation_state] = wheel(rc_index & 255)
|
|
else:
|
|
pixels.show()
|
|
return 0, color_state
|
|
else:
|
|
return animation_state, 0
|
|
|
|
|
|
|