From 98ccb7698da36d828e50e057f57891bc1388d541 Mon Sep 17 00:00:00 2001
From: ZodiusInfuser <christopher.parrott2@gmail.com>
Date: Mon, 21 Mar 2022 15:30:43 +0000
Subject: [PATCH] Added sensor feedback example

---
 .../examples/servo2040/current_meter.py       |  2 +-
 .../examples/servo2040/sensor_feedback.py     | 49 +++++++++++++++++++
 2 files changed, 50 insertions(+), 1 deletion(-)
 create mode 100644 micropython/examples/servo2040/sensor_feedback.py

diff --git a/micropython/examples/servo2040/current_meter.py b/micropython/examples/servo2040/current_meter.py
index 1ab4d554..6f659a0e 100644
--- a/micropython/examples/servo2040/current_meter.py
+++ b/micropython/examples/servo2040/current_meter.py
@@ -65,7 +65,7 @@ while user_sw.raw() is not True:
 
     # Update all the LEDs
     for i in range(servo2040.NUM_LEDS):
-        # Calculate the LED's hue, with Red for high currents and green for low
+        # Calculate the LED's hue, with Red for high currents and Green for low
         hue = (1.0 - i / (servo2040.NUM_LEDS - 1)) * 0.333
 
         # Calculate the current level the LED represents
diff --git a/micropython/examples/servo2040/sensor_feedback.py b/micropython/examples/servo2040/sensor_feedback.py
new file mode 100644
index 00000000..8ac0c2fc
--- /dev/null
+++ b/micropython/examples/servo2040/sensor_feedback.py
@@ -0,0 +1,49 @@
+import time
+from machine import Pin
+from pimoroni import Analog, AnalogMux, Button
+from plasma import WS2812
+from servo import servo2040
+
+# Press "Boot" to exit the program.
+
+BRIGHTNESS = 0.4      # The brightness of the LEDs
+UPDATES = 50          # How many times to update LEDs and Servos per second
+
+# Create the user button
+user_sw = Button(servo2040.USER_SW)
+
+# Set up the shared analog inputs
+sen_adc = Analog(servo2040.SHARED_ADC)
+
+# Set up the analog multiplexer, including the pin for controlling pull-up/pull-down
+mux = AnalogMux(servo2040.ADC_ADDR_0, servo2040.ADC_ADDR_1, servo2040.ADC_ADDR_2,
+                muxed_pin=Pin(servo2040.SHARED_ADC))
+
+# Set up the sensor addresses and have them pulled down by default
+sensor_addrs = list(range(servo2040.SENSOR_1_ADDR, servo2040.SENSOR_6_ADDR + 1))
+for addr in sensor_addrs:
+    mux.configure_pull(addr, Pin.PULL_DOWN)
+
+# Create the LED bar, using PIO 1 and State Machine 0
+led_bar = WS2812(servo2040.NUM_LEDS, 1, 0, servo2040.LED_DAT)
+
+# Start updating the LED bar
+led_bar.start()
+
+
+# Read sensors until the user button is pressed
+while user_sw.raw() is not True:
+
+    # Read each sensor in turn and print its voltage
+    for i in range(len(sensor_addrs)):
+        mux.select(sensor_addrs[i])
+        sensor_voltage = sen_adc.read_voltage()
+        
+        # Calculate the LED's hue, with Green for high voltages and Blue for low
+        hue = (2.0 - (sensor_voltage / 3.3)) * 0.333
+        led_bar.set_hsv(i, hue, 1.0, BRIGHTNESS)            
+            
+        print("S", i + 1, " = ", round(sensor_voltage, 3), sep="", end=", ")
+    print()
+
+    time.sleep(1.0 / UPDATES)
\ No newline at end of file