adding neopixel fidget spinner code

This commit is contained in:
BlitzCityDIY 2024-01-10 12:12:35 -05:00
parent ac9f36c19d
commit 3c63086bcf

View file

@ -0,0 +1,190 @@
# SPDX-FileCopyrightText: 2023 Liz Clark for Adafruit Industries
# SPDX-License-Identifier: MIT
import time
import random
import board
import neopixel
from adafruit_seesaw import seesaw, rotaryio, digitalio
from adafruit_debouncer import Button
from rainbowio import colorwheel
from adafruit_led_animation import color
# NeoPixel ring setup. Update num_pixels if using a different ring.
num_pixels = 24
pixels = neopixel.NeoPixel(board.D5, num_pixels, auto_write=False)
i2c = board.STEMMA_I2C()
seesaw = seesaw.Seesaw(i2c, addr=0x49)
buttons = []
for b in range(1, 6):
seesaw.pin_mode(b, seesaw.INPUT_PULLUP)
ss_pin = digitalio.DigitalIO(seesaw, b)
button = Button(ss_pin, long_duration_ms=1000)
buttons.append(button)
encoder = rotaryio.IncrementalEncoder(seesaw)
last_position = 0
button_names = ["Select", "Up", "Left", "Down", "Right"]
colors = [color.RED, color.YELLOW, color.ORANGE, color.GREEN,
color.TEAL, color.CYAN, color.BLUE, color.PURPLE, color.MAGENTA]
# rainbow cycle function
def rainbow_cycle(wait):
for j in range(255):
for i in range(num_pixels):
rc_index = (i * 256 // num_pixels) + j
pixels[i] = colorwheel(rc_index & 255)
pixels.show()
time.sleep(wait)
color_index = 0
game_mode = False
pixel = 0
num = 0
last_num = 0
now_color = 0
next_color = 1
speed = 0.1
level = 0.005
final_level = 0.001
new_target = True
while True:
if not game_mode:
for b in range(5):
buttons[b].update()
if buttons[b].released or buttons[b].pressed:
pixels.fill(color.BLACK)
position = encoder.position
if position != last_position:
pixels[last_position % num_pixels] = color.BLACK
pixels[position % num_pixels] = colors[color_index]
# print("Position: {}".format(position))
last_position = position
if buttons[0].pressed:
# print("Center button!")
pixels.fill(colors[color_index])
elif buttons[0].long_press:
# print("long press detected")
pixels.fill(color.BLACK)
new_target = True
game_mode = True
if buttons[1].pressed:
# print("Up button!")
color_index = (color_index + 1) % len(colors)
pixels[10] = colors[color_index]
if buttons[2].pressed:
# print("Left button!")
color_index = (color_index + 1) % len(colors)
pixels[4] = colors[color_index]
if buttons[3].pressed:
# print("Down button!")
color_index = (color_index - 1) % len(colors)
pixels[22] = colors[color_index]
if buttons[4].pressed:
# print("Right button!")
color_index = (color_index - 1) % len(colors)
pixels[16] = colors[color_index]
pixels.show()
if game_mode:
buttons[0].update()
if buttons[0].long_press:
# print("long press detected")
pixels.fill(color.BLACK)
pixels.show()
game_mode = False
pixels.fill(colors[color_index])
# if new level starting..
if new_target:
if buttons[0].released:
# randomize target location
y = random.randint(5, 22)
x = y - 1
z = y + 1
new_target = False
pixels[x] = color.WHITE
pixels[y] = colors[next_color]
pixels[z] = color.WHITE
else:
# delay without time.sleep()
if (pixel + speed) < time.monotonic():
# turn off pixel behind chaser
if num > 0:
last_num = num - 1
pixels[last_num] = color.BLACK
pixels.show()
# keep target pixels their colors when the chaser passes
if last_num in (x, y, z):
pixels[x] = color.WHITE
pixels[y] = colors[next_color]
pixels[z] = color.WHITE
# move chaser pixel by one
if num < num_pixels:
pixels[num] = colors[now_color]
pixels.show()
num += 1
# send chaser back to the beginning of the circle
if num == num_pixels:
last_num = num - 1
pixels[last_num] = color.BLACK
pixels.show()
num = 0
# if the chaser hits the target...
if last_num in [x, y, z] and not buttons[0].value:
# fills with the next color
pixels.fill(colors[next_color])
pixels.show()
# chaser resets
num = 0
time.sleep(0.5)
pixels.fill(color.BLACK)
pixels.show()
# speed increases for next level
speed = speed - level
# color updates
next_color = (next_color + 1) % 9
now_color = (now_color + 1) % 9
# setup for new target
new_target = True
# if the chaser misses the target...
if last_num not in [x, y, z] and not buttons[0].value:
# fills with current chaser color
pixels.fill(color.BLACK)
pixels.show()
# chaser is reset
num = 0
# speed is reset to default
speed = 0.1
# colors are reset
next_color = 1
now_color = 0
# setup for new target
new_target = True
# when you have beaten all the levels...
if speed < final_level:
# rainbows!
rainbow_cycle(0.01)
time.sleep(1)
# chaser is reset
num = 0
pixels.fill(color.BLACK)
pixels.show()
# speed is reset to default
speed = 0.1
# colors are reset
next_color = 1
now_color = 0
# setup for new target
new_target = True
# time.monotonic() is reset for the delay
pixel = time.monotonic()