Merge pull request #1838 from kattni/rainbowio

Move to rainbowio

Author: kattni

3D_Printed_LED_Microphone_Flag/code.py

@@ -31,6 +31,7 @@
 
 import board
 import neopixel
+from rainbowio import colorwheel
 from analogio import AnalogIn
 
 n_pixels = 16  # Number of pixels you are using
@@ -53,21 +54,6 @@
 strip = neopixel.NeoPixel(led_pin, n_pixels, brightness=1, auto_write=False)
 
 
-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 (int(pos * 3), int(255 - (pos * 3)), 0)
-    elif pos < 170:
-        pos -= 85
-        return (int(255 - pos * 3), 0, int(pos * 3))
-    else:
-        pos -= 170
-        return (0, int(pos * 3), int(255 - pos * 3))
-
-
 def remapRange(value, leftMin, leftMax, rightMin, rightMax):
     # this remaps a value fromhere original (left) range to new (right) range
     # Figure out how 'wide' each range is
@@ -168,7 +154,7 @@ def drawLine(fromhere, to):
 
     # Fill the strip with rainbow gradient
     for i in range(0, len(strip)):
-        strip[i] = wheel(remapRange(i, 0, (n_pixels - 1), 30, 150))
+        strip[i] = colorwheel(remapRange(i, 0, (n_pixels - 1), 30, 150))
 
     # Scale the input logarithmically instead of linearly
     c = fscale(input_floor, input_ceiling, (n_pixels - 1), 0, peaktopeak, 2)
@@ -182,7 +168,7 @@ def drawLine(fromhere, to):
 
     # Set the peak dot to match the rainbow gradient
     y = n_pixels - peak
-    strip.fill = (y - 1, wheel(remapRange(y, 0, (n_pixels - 1), 30, 150)))
+    strip.fill = (y - 1, colorwheel(remapRange(y, 0, (n_pixels - 1), 30, 150)))
     strip.write()
 
     # Frame based peak dot animation

3D_Printed_Unicorn_Horn/code.py

@@ -2,6 +2,7 @@
 
 import board
 import neopixel
+from rainbowio import colorwheel
 from digitalio import DigitalInOut, Direction
 
 pixpin = board.D1
@@ -13,34 +14,19 @@
 strip = neopixel.NeoPixel(pixpin, numpix, brightness=1, auto_write=True)
 
 
-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 (int(pos * 3), int(255 - (pos*3)), 0)
-    elif pos < 170:
-        pos -= 85
-        return (int(255 - pos * 3), 0, int(pos * 3))
-    else:
-        pos -= 170
-        return (0, int(pos * 3), int(255 - pos * 3))
-
-
 def rainbow_cycle(wait):
     for j in range(255 * 5):
         for i in range(len(strip)):
             idx = int((i * 256 / len(strip)) + j)
-            strip[i] = wheel(idx & 255)
+            strip[i] = colorwheel(idx & 255)
         time.sleep(wait)
 
 
 def rainbow(wait):
     for j in range(255):
         for i in range(len(strip)):
             idx = int(i + j)
-            strip[i] = wheel(idx & 255)
+            strip[i] = colorwheel(idx & 255)
         time.sleep(wait)
 
 

Adafruit_FunHouse/dotstar_rainbow/code.py

@@ -2,7 +2,7 @@
 import time
 import board
 import adafruit_dotstar
-from _pixelbuf import colorwheel
+from rainbowio import colorwheel
 
 dots = adafruit_dotstar.DotStar(board.DOTSTAR_CLOCK, board.DOTSTAR_DATA, 5, auto_write=False)
 dots.brightness = 0.3

Adafruit_Neo_Trinkey/cap_touch_neopixel_brightness/code.py

@@ -3,7 +3,7 @@
 import board
 import touchio
 import neopixel
-from _pixelbuf import colorwheel
+from rainbowio import colorwheel
 
 touch1 = touchio.TouchIn(board.TOUCH1)
 touch2 = touchio.TouchIn(board.TOUCH2)

Adafruit_Prop_Maker_FeatherWing/Prop_Maker_3W_LED_Simpletest/code.py

@@ -1,6 +1,7 @@
 """Simple rainbow swirl example for 3W LED"""
 import pwmio
 import board
+from rainbowio import colorwheel
 import digitalio
 
 enable = digitalio.DigitalInOut(board.D10)
@@ -11,24 +12,9 @@
 green = pwmio.PWMOut(board.D12, duty_cycle=0, frequency=20000)
 blue = pwmio.PWMOut(board.D13, duty_cycle=0, frequency=20000)
 
-
-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)
-
-
 while True:
     for i in range(255):
-        r, g, b = wheel(i)
+        r, g, b = colorwheel(i)
         red.duty_cycle = int(r * 65536 / 256)
         green.duty_cycle = int(g * 65536 / 256)
         blue.duty_cycle = int(b * 65536 / 256)

Adafruit_Prop_Maker_FeatherWing/Prop_Maker_NeoPixel_Simpletest/code.py

@@ -1,6 +1,7 @@
 """Simple rainbow example for 30-pixel NeoPixel strip"""
 import digitalio
 import board
+from rainbowio import colorwheel
 import neopixel
 
 NUM_PIXELS = 30  # NeoPixel strip length (in pixels)
@@ -11,21 +12,6 @@
 
 strip = neopixel.NeoPixel(board.D5, NUM_PIXELS, brightness=1)
 
-
-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)
-
-
 while True:
     for i in range(255):
-        strip.fill((wheel(i)))
+        strip.fill((colorwheel(i)))

Animated_NeoPixel_Glow_Fur_Scarf/code.py

@@ -63,21 +63,6 @@
                0xFFFF33, 0xFFFF66, 0xFFFF99, 0xFFFFCC]
 
 
-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 (int(pos * 3), int(255 - (pos * 3)), 0)
-    elif pos < 170:
-        pos -= 85
-        return (int(255 - pos * 3), 0, int(pos * 3))
-    else:
-        pos -= 170
-        return (0, int(pos * 3), int(255 - pos * 3))
-
-
 def remapRange(value, leftMin, leftMax, rightMin, rightMax):
     # this remaps a value fromhere original (left) range to new (right) range
     # Figure out how 'wide' each range is

Black_Lives_Matter_Kit/Capacitive_Touch_and_NeoPixels/code.py

@@ -3,7 +3,7 @@
 import touchio
 import digitalio
 import neopixel
-from adafruit_pypixelbuf import colorwheel
+from rainbowio import colorwheel
 
 pixels = neopixel.NeoPixel(board.NEOPIXEL, 6, auto_write=False)
 red_led = digitalio.DigitalInOut(board.D13)

Black_Lives_Matter_Kit/Sound_Reactive_NeoPixels/code.py

@@ -3,7 +3,7 @@
 import board
 import audiobusio
 import neopixel
-from adafruit_pypixelbuf import colorwheel
+from rainbowio import colorwheel
 
 # Increase this number to use this example in louder environments. As you increase the number, it
 # increases the level of sound needed to change the color of the LEDs. 5 is good for quiet up to

CP101_StateMachines/brute-force/code.py

@@ -23,6 +23,7 @@
 import audioio
 import audiocore
 import pwmio
+from rainbowio import colorwheel
 from adafruit_motor import servo
 import neopixel
 from adafruit_debouncer import Debouncer
@@ -131,30 +132,18 @@ def random_color():
     blue = random_color_byte()
     return (red, green, blue)
 
-# Color cycling.
-
-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 int(255 - pos*3), int(pos*3), 0
-    if pos < 170:
-        pos -= 85
-        return 0, int(255 - pos*3), int(pos*3)
-    pos -= 170
-    return int(pos * 3), 0, int(255 - (pos*3))
 
+# Color cycling.
 def cycle_sequence(seq):
     while True:
         for elem in seq:
             yield elem
 
+
 def rainbow_lamp(seq):
     g = cycle_sequence(seq)
     while True:
-        strip.fill(wheel(next(g)))
+        strip.fill(colorwheel(next(g)))
         strip.show()
         yield
 

CP101_StateMachines/classes/code.py

@@ -23,6 +23,7 @@
 import audioio
 import audiocore
 import pwmio
+from rainbowio import colorwheel
 from adafruit_motor import servo
 import neopixel
 from adafruit_debouncer import Debouncer
@@ -113,19 +114,6 @@ def random_color():
 
 # Color cycling.
 
-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 int(255 - pos*3), int(pos*3), 0
-    if pos < 170:
-        pos -= 85
-        return 0, int(255 - pos*3), int(pos*3)
-    pos -= 170
-    return int(pos * 3), 0, int(255 - (pos*3))
-
 def cycle_sequence(seq):
     while True:
         for elem in seq:
@@ -134,7 +122,7 @@ def cycle_sequence(seq):
 def rainbow_lamp(seq):
     g = cycle_sequence(seq)
     while True:
-        strip.fill(wheel(next(g)))
+        strip.fill(colorwheel(next(g)))
         strip.show()
         yield
 

CircuitPlayground_Christmas_Tree/code.py

@@ -23,9 +23,10 @@
                            pixel_order=ORDER)
 
 
-def wheel(pos):
+def colorwheel(pos):
     # Input a value 0 to 255 to get a color value.
     # The colours are a transition r - g - b - back to r.
+    # Works with RGB or RGBW LEDs.
     if pos < 0 or pos > 255:
         r = g = b = 0
     elif pos < 85:
@@ -49,7 +50,7 @@ def rainbow_swirl(wait):
     for j in range(255):
         for i in range(num_pixels):
             pixel_index = (i * 256 // num_pixels) + j
-            pixels[i] = wheel(pixel_index & 255)
+            pixels[i] = colorwheel(pixel_index & 255)
         pixels.show()
         time.sleep(wait)
 
@@ -58,7 +59,7 @@ def rainbow_fill(wait):
     for j in range(255):
         for i in range(num_pixels):
             pixel_index = int(i + j)
-            pixels[i] = wheel(pixel_index & 255)
+            pixels[i] = colorwheel(pixel_index & 255)
         pixels.show()
         time.sleep(wait)
 

CircuitPython_AT_Hand_Raiser/code.py

@@ -8,6 +8,7 @@
 
 from time import sleep
 import board
+from rainbowio import colorwheel
 import adafruit_dotstar
 import supervisor
 
@@ -66,19 +67,6 @@ def hex2rgb(hex_code):
 
 mode='wheel'
 
-# standard function to rotate around the colorwheel
-def wheel(cpos):
-    # Input a value 0 to 255 to get a color value.
-    # The colours are a transition r - g - b - back to r.
-    if cpos < 85:
-        return (int(cpos * 3), int(255 - (cpos * 3)), 0)
-    elif cpos < 170:
-        cpos -= 85
-        return (int(255 - (cpos * 3)), 0, int(cpos * 3))
-    else:
-        cpos -= 170
-        return (0, int(cpos * 3), int(255 - cpos * 3))
-
 # We start by turning off pixels
 pixels.fill(black)
 pixels.show()
@@ -148,7 +136,7 @@ def wheel(cpos):
         elif mode == 'wheel':
             sleep(.05)
             pos = (pos + 1) % 255
-            pixels.fill(wheel(pos))
+            pixels.fill(colorwheel(pos))
             pixels.show()
         elif mode == 'solid':
             pixels.fill(targetColor)

CircuitPython_Display_Text/colormask_example/code.py

@@ -8,24 +8,10 @@
 
 import board
 import displayio
+from rainbowio import colorwheel
 from adafruit_bitmap_font import bitmap_font
 from adafruit_display_text import bitmap_label as label
 
-
-def wheel(pos):
-    # input a value 0 to 255 to get a color value
-    # the colors are a transition r-g-b-back to r.
-    if pos < 1 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)
-
-
 # Make the display context. Change size if you want
 display = board.DISPLAY
 
@@ -58,7 +44,7 @@ def wheel(pos):
 rainbow_palette = displayio.Palette(255)
 
 for i in range(0, 255):
-    rainbow_palette[i] = int("".join("%02x" % i for i in wheel(i)), 16)
+    rainbow_palette[i] = int("".join("%02x" % i for i in colorwheel(i)), 16)
 
 for y in range(rainbow_bitmap.height):
     for x in range(rainbow_bitmap.width):