Merge pull request #5 from ericgfx/master

Create kitt.py

Author: sonnynomnom

README.md

@@ -51,6 +51,7 @@ https://www.codecademy.com/learn/learn-circuitpython
 
 - [x] [`code.py`](circuit-playground-part-1/bike-light/code.py)
 - [x] [`bedazzle_bike_light.py`](circuit-playground-part-1/bike-light/bedazzle_bike_light.py) (by [cephir909](https://www.codecademy.com/cephir909))
+- [x] [`KITT.py`](circuit-playground-part-1/bike-light/KITT.py) (by [ericgfx](https://www.codecademy.com/EricGrossDesign))
 
 ## 2. Circuit Playground Express: Part 2 ##
 

circuit-playground-part-1/bike-light/kitt.py

@@ -0,0 +1,89 @@
+# K.I.T.T. BikeLight
+# Use Button A to change color
+# Use Button B to change style
+# Ride safe and don't Hassle the Hoff
+# 
+# Created by Eric Gross. Sublime Text on MacBookPro on 5/27/19.
+# CodeCademy 2019: Hardware Programming with CircuitPython: BikeLight
+
+
+
+from adafruit_circuitplayground.express import cpx
+import time
+
+colorIndex = 0
+functionIndex = 0
+rainbow =((255, 0, 0),(255, 85, 0),(255, 255, 0),(0, 255, 0),(34, 139, 34),(0, 255, 255),(0, 0, 255),(0, 0, 139),(255, 0, 255),(75, 0, 130),(0,0,0))
+
+
+###### Function to Change Color ######
+def checkPress(colorIndex):
+  if cpx.button_a:
+    colorIndex += 1
+    if colorIndex > 10:
+      return 0
+  return colorIndex
+
+###### Function to Change Function ######
+def functionPress(functionIndex):
+  if cpx.button_b:
+    functionIndex += 1
+    if functionIndex > 2:
+      return 0
+  return functionIndex
+
+
+###### Color and Blinking #####
+def blinkLed(colorIndex, d = 0.5, ledIndex = 10, turnOff = True):
+  if ledIndex == 10:
+    cpx.pixels.fill((rainbow[colorIndex][0],rainbow[colorIndex][1],rainbow[colorIndex][2]))
+    time.sleep(d)
+  else:
+    cpx.pixels[ledIndex] = (rainbow[colorIndex][0],rainbow[colorIndex][1],rainbow[colorIndex][2])
+    time.sleep(d)
+  if turnOff == True:
+  	cpx.pixels.fill((0,0,0))
+  	time.sleep(d)
+  return
+
+
+def kitt(d = 0.1):
+  global colorIndex
+  colorIndex = checkPress(colorIndex)
+  for x in range(0,10):
+    blinkLed(10, d, x, False)
+    blinkLed(colorIndex, 0, x, False)
+  time.sleep(d)
+  colorIndex = checkPress(colorIndex)
+  for x in range(9,-1,-1):
+    blinkLed(10, d, x, False)
+    blinkLed(colorIndex, 0, x, False)
+  time.sleep(d)
+  return
+
+def rainbowSpin(d = 0.1):
+  global colorIndex
+  for x in range(0,10):
+    blinkLed(colorIndex, d, x, False)
+  colorIndex += 1
+  if colorIndex > 10:
+    colorIndex = 0
+  return
+
+
+##########################
+###### Main Program ######
+##########################
+while True:
+  if cpx.switch:
+    functionIndex = functionPress(functionIndex)
+    colorIndex = checkPress(colorIndex)
+    if functionIndex == 0:
+      blinkLed(colorIndex)
+    elif functionIndex == 1:
+      kitt()
+    else:
+      rainbowSpin()
+  else:
+    # shut it
+    cpx.pixels.fill((0, 0, 0))