Cosmetic fixes, updated rainbow_cycle

examples/neopixel_simpletest.py

@@ -1,61 +1,74 @@
-# CircuitPython demo - NeoPixel
-
 import time
 import board
 import neopixel
 
 
-# On CircuitPlayground Express -> Board.NEOPIXEL
-# Otherwise choose an open pin connected to the Data In of the NeoPixel strip,
-# such as board.D1
-# pylint: disable=no-member
-pixpin = board.NEOPIXEL
+# On CircuitPlayground Express, and boards with built in status NeoPixel -> board.NEOPIXEL
+# Otherwise choose an open pin connected to the Data In of the NeoPixel strip, i.e. board.D1
+pixel_pin = board.NEOPIXEL
 
-# The number of pixels in the strip
-numpix = 10
+# The number of NeoPixels
+num_pixels = 10
 
-# number of colors in each pixel, =3 for RGB, =4 for RGB plus white
-BPP = 3
+# The order of the pixel colors - RGB or GRB. Some NeoPixels have red and green reversed!
+# For RGBW NeoPixels, simply change the ORDER to RGBW or GRBW.
+ORDER = neopixel.GRB
 
-strip = neopixel.NeoPixel(pixpin, numpix, bpp=BPP, brightness=0.3, auto_write=False)
+pixels = neopixel.NeoPixel(pixel_pin, num_pixels, brightness=0.2, auto_write=False,
+ pixel_order=ORDER)
 
-def format_tuple(r, g, b):
- if BPP == 3:
- return (r, g, b)
- return (r, g, b, 0)
 
 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 format_tuple(0, 0, 0)
- if pos < 85:
- return format_tuple(int(pos * 3), int(255 - (pos*3)), 0)
- if pos < 170:
+ 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
- return format_tuple(int(255 - pos*3), 0, int(pos*3))
- pos -= 170
- return format_tuple(0, int(pos*3), int(255 - pos*3))
+ 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 == neopixel.RGB or ORDER == neopixel.GRB else (r, g, b, 0)
+
 
 def rainbow_cycle(wait):
  for j in range(255):
- for i in range(strip.n):
- idx = int((i * 256 / len(strip)) + j)
- strip[i] = wheel(idx & 255)
- strip.show()
+ for i in range(num_pixels):
+ pixel_index = (i * 256 // num_pixels) + j
+ pixels[i] = wheel(pixel_index & 255)
+ pixels.show()
  time.sleep(wait)
 
+
 while True:
- strip.fill(format_tuple(255, 0, 0))
- strip.show()
+ # Comment this line out if you have RGBW/GRBW NeoPixels
+ pixels.fill((255, 0, 0))
+ # Uncomment this line if you have RGBW/GRBW NeoPixels
+ # pixels.fill((255, 0, 0, 0))
+ pixels.show()
  time.sleep(1)
 
- strip.fill(format_tuple(0, 255, 0))
- strip.show()
+ # Comment this line out if you have RGBW/GRBW NeoPixels
+ pixels.fill((0, 255, 0))
+ # Uncomment this line if you have RGBW/GRBW NeoPixels
+ # pixels.fill((0, 255, 0, 0))
+ pixels.show()
  time.sleep(1)
 
- strip.fill(format_tuple(0, 0, 255))
- strip.show()
+ # Comment this line out if you have RGBW/GRBW NeoPixels
+ pixels.fill((0, 0, 255))
+ # Uncomment this line if you have RGBW/GRBW NeoPixels
+ # pixels.fill((0, 0, 255, 0))
+ pixels.show()
  time.sleep(1)
 
- rainbow_cycle(0.001) # rainbowcycle with 1ms delay per step
+ rainbow_cycle(0.001) # rainbow cycle with 1ms delay per step