diff --git a/examples/cheerlights_bulb.py b/examples/cheerlights_bulb.py new file mode 100644 index 0000000..9853876 --- /dev/null +++ b/examples/cheerlights_bulb.py @@ -0,0 +1,276 @@ +import time + +import machine +import network +import plasma +import requests +from ft6236 import FT6236 +from picographics import DISPLAY_PRESTO, PicoGraphics +from picovector import ANTIALIAS_BEST, PicoVector, Polygon, Transform +from presto import Presto + +BULB_OUTLINE = [(130.44, 0.0), + (150.36, 1.51), + (165.02, 4.64), + (183.83, 11.41), + (201.36, 20.97), + (213.45, 29.85), + (231.09, 47.51), + (239.96, 59.6), + (249.51, 77.14), + (256.26, 95.95), + (260.51, 120.55), + (260.54, 140.53), + (257.63, 160.3), + (249.3, 183.81), + (239.13, 201.01), + (226.92, 216.84), + (210.84, 235.98), + (199.26, 252.27), + (194.56, 261.09), + (189.52, 275.19), + (186.53, 299.95), + (184.32, 304.38), + (175.48, 308.37), + (85.48, 308.37), + (76.62, 304.46), + (74.37, 300.04), + (71.38, 275.29), + (66.36, 261.17), + (61.67, 252.35), + (50.1, 236.05), + (34.02, 216.92), + (21.8, 201.09), + (11.63, 183.89), + (3.27, 160.39), + (0.35, 140.63), + (0.36, 120.64), + (3.37, 100.89), + (11.33, 77.23), + (23.67, 55.53), + (32.98, 43.78), + (51.27, 26.8), + (68.01, 15.9), + (90.97, 6.09), + (110.42, 1.53), + (130.35, 0.0)] + +BULB_INNER = [(130.44, 6.81), + (150.35, 8.47), + (164.95, 11.87), + (183.54, 19.17), + (200.7, 29.4), + (216.02, 42.22), + (226.08, 53.33), + (237.34, 69.85), + (245.83, 87.92), + (251.31, 107.14), + (253.57, 126.99), + (252.61, 146.94), + (248.29, 166.44), + (244.81, 175.81), + (235.5, 193.48), + (223.9, 209.77), + (208.12, 229.16), + (196.2, 245.21), + (186.84, 262.85), + (183.75, 272.35), + (180.29, 297.0), + (172.24, 302.02), + (92.24, 302.05), + (82.74, 299.85), + (80.07, 295.69), + (76.77, 270.99), + (69.14, 252.56), + (58.07, 235.92), + (36.12, 208.67), + (24.64, 192.3), + (15.53, 174.52), + (12.17, 165.1), + (7.62, 140.58), + (7.67, 120.59), + (8.92, 110.68), + (13.83, 91.31), + (21.8, 72.99), + (32.58, 56.17), + (45.86, 41.25), + (61.32, 28.59), + (78.6, 18.56), + (97.28, 11.47), + (106.99, 9.12), + (126.84, 6.86)] + +# How long we'll wait between updates +INTERVAL = 60 + +machine.freq(264000000) + +# Check and import the SSID and Password from secrets.py +try: + from secrets import WIFI_PASSWORD, WIFI_SSID + if WIFI_SSID == "": + raise ValueError("WIFI_SSID in 'secrets.py' is empty!") + if WIFI_PASSWORD == "": + raise ValueError("WIFI_PASSWORD in 'secrets.py' is empty!") +except ImportError: + raise ImportError("'secrets.py' is missing from your Plasma 2350 W!") +except ValueError as e: + print(e) + +# Setup for the Presto display +presto = Presto() +display = PicoGraphics(DISPLAY_PRESTO, buffer=memoryview(presto)) +WIDTH, HEIGHT = display.get_bounds() + +# We'll need this for the touch element of the screen +touch = FT6236() + +# Pico Vector +vector = PicoVector(display) +vector.set_antialiasing(ANTIALIAS_BEST) + +# Colours for use later +WHITE = display.create_pen(255, 255, 255) +BLACK = display.create_pen(0, 0, 0) +GRAY = display.create_pen(75, 75, 75) + +wlan = network.WLAN(network.STA_IF) + + +def connect(): + # Connect to the network specified in secrets.py + wlan.active(True) + wlan.connect(WIFI_SSID, WIFI_PASSWORD) + while wlan.isconnected() is False: + print("Attempting connection to {}".format(WIFI_SSID)) + time.sleep(1) + + +# Start connection to the network +connect() + +# WS2812 / NeoPixel™ LEDs used for the backlight +bl = plasma.WS2812(7, 0, 0, 33) +bl.start() + +# Centre points +CX = WIDTH // 2 +CY = HEIGHT // 2 + +# Shape constants +BAR_W = 60 +BAR_H = 10 +HALF_BAR_W = BAR_W // 2 +HALF_BAR_H = BAR_H // 2 +BAR_Y_START = 160 + +# Define our vector shapes +bars = Polygon() +bars.rectangle(CX - HALF_BAR_W, BAR_Y_START + 10, 60, 10, (5, 5, 5, 5)) +bars.rectangle(CX - HALF_BAR_W, BAR_Y_START + 25, 60, 10, (5, 5, 5, 5)) +bars.rectangle(CX - HALF_BAR_W, BAR_Y_START + 40, 60, 10, (5, 5, 5, 5)) +end = Polygon() +end.arc(CX - 14, BAR_Y_START + 55, 14, 90, 270) + +bulb = Polygon() +bulb.path(*BULB_INNER) +bulb_outline = Polygon() +bulb_outline.path(*BULB_OUTLINE) + +transform = Transform() + + +def draw_bulb(colour): + + display.set_pen(GRAY) + + transform.reset() + vector.set_transform(transform) + transform.rotate(180, (CX - 7, BAR_Y_START + 55)) + vector.draw(end) + + transform.reset() + + vector.draw(bars) + + cl = display.create_pen(*colour) + + transform.reset() + + transform.translate(54, 11) + transform.scale(0.5, 0.5) + + display.set_pen(BLACK) + vector.draw(bulb_outline) + + display.set_pen(cl) + vector.draw(bulb) + + +def get_cheerlight(): + try: + print("Getting new colour...") + req = requests.get("http://api.thingspeak.com/channels/1417/field/2/last.json", timeout=None) + json = req.json() + req.close() + print("Success!") + + colour = tuple(int(json['field2'][i:i + 2], 16) for i in (1, 3, 5)) + + return colour + + except OSError: + print("Error: Failed to get new colour") + return (255, 255, 255) + + +bulb_on = True +last_updated = time.time() + +# Get the first colour from cheerlights +colour = get_cheerlight() + +while True: + + # Poll the touch so we can see if anything changed since the last time + touch.poll() + + if wlan.isconnected(): + # If the user is touching the screen we'll do the following + if touch.state: + bulb_on = not bulb_on + print(f"Bulb is {bulb_on}") + # Wait for the user to stop touching the screen + while touch.state: + touch.poll() + + if bulb_on: + if time.time() - last_updated > INTERVAL: + colour = get_cheerlight() + last_updated = time.time() + + display.set_pen(WHITE) + display.clear() + + draw_bulb(colour) + + for i in range(7): + bl.set_rgb(i, *colour) + + time.sleep(0.02) + + else: + display.set_pen(BLACK) + display.clear() + + for i in range(7): + bl.set_rgb(i, 0, 0, 0) + + time.sleep(0.02) + + draw_bulb((50, 50, 50)) + + else: + print("Lost connection to network {}".format(WIFI_SSID)) + + presto.update(display) diff --git a/examples/secrets.py b/examples/secrets.py new file mode 100644 index 0000000..88c656c --- /dev/null +++ b/examples/secrets.py @@ -0,0 +1,2 @@ +WIFI_SSID = "" +WIFI_PASSWORD = "" \ No newline at end of file