From d9627dfd1f0687017e4b80c60b4c9f323d10a8b5 Mon Sep 17 00:00:00 2001
From: Mikayla Dobson <mikaylaherself@gmail.com>
Date: Sun, 6 Aug 2023 15:36:32 -0500
Subject: [PATCH] some new functionality

---
 main.py | 50 ++++++++++++++++++++++++++++++++++++++++----------
 1 file changed, 40 insertions(+), 10 deletions(-)

diff --git a/main.py b/main.py
index d6f7b6e..d5c64a9 100644
--- a/main.py
+++ b/main.py
@@ -1,4 +1,4 @@
-from machine import Pin, PWM, freq
+from machine import Pin, PWM
 from time import sleep
 
 U16_MAX = 65535
@@ -10,14 +10,27 @@ class Color:
     def duty_u16(self, duty):
         self.pin.duty_u16(duty)
 
-    def breathe(self):
+    def breathe(self, interval=0.0001):
         for duty in range(0, U16_MAX, 20):
             self.duty_u16(duty)
-            sleep(0.0001)
+            sleep(interval)
 
         for duty in range(U16_MAX, 0, -20):
             self.duty_u16(duty)
-            sleep(0.0001)
+            sleep(interval)
+
+class RGB:
+    """
+    Convert a traditional (8-bit) RGB color to a 16-bit color
+    """
+    def __init__(self, r, g, b):
+        self.r = r * 256
+        self.g = g * 256
+        self.b = b * 256
+
+    def to_u16(self):
+        return self.r, self.g, self.b
+
 
 
 class StripLights:
@@ -36,10 +49,10 @@ class StripLights:
         self.G.duty_u16(0)
         self.B.duty_u16(0)
 
-    def breathe(self):
-        self.R.breathe()
-        self.G.breathe()
-        self.B.breathe()
+    def breathe(self, sleep_interval=0.0001):
+        self.R.breathe(sleep_interval)
+        self.G.breathe(sleep_interval)
+        self.B.breathe(sleep_interval)
 
     def blink(self, sleep_interval=1):
         self.on()
@@ -49,7 +62,9 @@ class StripLights:
         print('off')
         sleep(sleep_interval)
 
-    def set_color(self, r, g, b):
+    def set_color(self, rgb):
+        r, g, b = rgb
+
         if (r < 0 or r > U16_MAX):
             raise "Red value out of accepted range"
 
@@ -63,10 +78,25 @@ class StripLights:
         self.G.duty_u16(g)
         self.B.duty_u16(b)
 
+    def color_cycle(self, color_list, sleep_interval=1):
+        for color in color_list:
+            self.set_color(color)
+            sleep(sleep_interval)
+
+COLORS = {
+    'PURPLE': RGB(255, 0, 255),
+    'RED': RGB(255, 0, 0),
+    'GREEN': RGB(0, 255, 0),
+    'BLUE': RGB(0, 0, 255)
+}
+
 def main():
     lights = StripLights()
 
     while True:
-        lights.blink()
+        lights.color_cycle([color.to_u16() for color in COLORS.values()])
+
+    # while True:
+    #     lights.breathe(0.01)
 
 main()