Content """ Code for the MCH2022 badge to control the 5 neopixels in the kite. The buttons A and B can be used to cycle through the default colors provided by matplotlib. Getting the Color List ---------------------- >>> import matplotlib >>> print(matplotlib.colors.TABLEAU_COLORS) >>> print(matplotlib.colors.BASE_COLORS) >>> print(tuple( ... (lambda s: tuple(int(f"0x{s[2 * i:2 * i + 2]}", 16) ... for i in range(3)))(c[1:]) ... for c in [matplotlib.colors.to_hex(f"C{j}") ... for j in range(10)]) See Also -------- [[https://www.badge.team/docs/badges/mch2022/software-development/micropython/neopixels/]]: The code is taken from there """ # imports from machine import Pin from neopixel import NeoPixel import mch22 import buttons import display # Pin 19 controls the power supply to SD card and neopixels powerPin = Pin(19, Pin.OUT) # Pin 5 is the LED's data line dataPin = Pin(5, Pin.OUT) # create a neopixel object for 5 pixels np = NeoPixel(dataPin, 5) # turn on power to the LEDs powerPin.on() def reboot(pressed): """ reboot on button press """ if pressed: mch22.exit_python() def convert_hexstring_to_int_tuple(color): """ Convert color Example ------- >>> convert_hexstring_to_int_tuple("#1f77b4") (31, 119, 180) """ s = color[-6:] return tuple(int(f"0x{s[2 * i:2 * i + 2]}", 16) for i in range(3)) BASE_COLORS = [ ('b', (0, 0, 255)), ('g', (0, 127, 0)), ('r', (255, 0, 0)), ('c', (0, 192, 192)), ('m', (192, 0, 192)), ('y', (192, 192, 0)), ('k', (0, 0, 0)), ('w', (255, 255, 255))] TABLEAU_COLORS = [ (c_str, convert_hexstring_to_int_tuple(c)) for (c_str, c) in [ ('tab:blue', '#1f77b4'), ('tab:orange', '#ff7f0e'), ('tab:green', '#2ca02c'), ('tab:red', '#d62728'), ('tab:purple', '#9467bd'), ('tab:brown', '#8c564b'), ('tab:pink', '#e377c2'), ('tab:gray', '#7f7f7f'), ('tab:olive', '#bcbd22'), ('tab:cyan', '#17becf')]] i = 0 color_list = TABLEAU_COLORS def cycle_color(pressed): """ Cycle through a list of colors """ if not pressed: # if we don't add this, the function is also called on release return global i, color_list color_str, color = color_list[(i) % len(color_list)] color_hex = int("0x" + "".join([ f"{n:02x}" for n in color]), 16) for pxl in [0, 1, 3]: # set some colors for the pixels (RGB) # (This is the center row of the kite) np[pxl] = color # pixel to the right np[4] = color_list[(i - 1) % len(color_list)][-1] # pixel to the left np[2] = color_list[(i + 1) % len(color_list)][-1] # send colors out to LEDs np.write() display.drawFill(color_hex) # Add what color we are displaying to the display text_color = 0xFFFFFF if color_str != 'w' else 0x000000 font = "roboto_regular12" display.drawText( 10, 10, f"Color: {color_str} - {color} - {hex(color_hex)}", text_color, font) display.drawText( 10, 60, "Press <A> and <B> to cycle colors. \n<SELECT> to change color list.", text_color, font) display.drawText( 10, 120, "Press <MENU> to quit", text_color, font) display.flush() i += 1 def cycle_backwards(pressed): """ cycle the other way around """ global i if pressed: i -= 2 cycle_color(pressed) def toggle(pressed): """ switch between the two color lists """ if not pressed: return global i, color_list if BASE_COLORS is color_list: color_list = TABLEAU_COLORS else: color_list = BASE_COLORS i -= 1 cycle_color(pressed) buttons.attach(buttons.BTN_MENU, reboot) buttons.attach(buttons.BTN_A, cycle_color) buttons.attach(buttons.BTN_B, cycle_backwards) buttons.attach(buttons.BTN_SELECT, toggle) # call function once for initialization cycle_color(True)