From 9d505f208a29f2ba1b4ed7e0bbdddbd964aa8b95 Mon Sep 17 00:00:00 2001 From: ZodiusInfuser Date: Wed, 25 Aug 2021 15:06:05 +0100 Subject: [PATCH] Added MP accelerometer example --- micropython/examples/plasma2040/level.py | 206 +++++++++++++++++++++++ 1 file changed, 206 insertions(+) create mode 100644 micropython/examples/plasma2040/level.py diff --git a/micropython/examples/plasma2040/level.py b/micropython/examples/plasma2040/level.py new file mode 100644 index 00000000..1ea5c6d0 --- /dev/null +++ b/micropython/examples/plasma2040/level.py @@ -0,0 +1,206 @@ +import plasma +from plasma import plasma2040 +import time +import math +import random + +# Import helpers for RGB LEDs and Buttons +from pimoroni import RGBLED, Button + +# Import msa301 and I2C helper +from breakout_msa301 import BreakoutMSA301 +from pimoroni_i2c import PimoroniI2C + +# A simple balancing game, where you use the MSA301 accelerometer to line up a band with a goal on the strip. +# This can either be done using: +# - Angle mode: Where position on the strip directly matches the accelerometer's angle +# - Velocity mode: Where tilting the accelerometer changes the speed the band moves at +# When the goal position is reached, a new position is randomly selected + +# Press "A" to change the game mode. +# Press "B" to start or stop the game mode. +# Press "Boot" to invert the direction of the accelerometer tilt + +# Set how many LEDs you have +NUM_LEDS = 30 + +# How many times the LEDs will be updated per second +UPDATES = 60 + +# The sensitivity of the accelerometer input +ANGLE_SENSITIVITY = 0.05 +VELOCITY_SENSITIVITY = 0.2 / UPDATES + +# The band colour hues to show in Angle mode +ANGLE_MODE_GOAL_HUE = 0.333 +ANGLE_MODE_EDGE_HUE = 0.0 + +# The band colour hues to show in Velocity mode +VELOCITY_MODE_GOAL_HUE = 0.667 +VELOCITY_MODE_EDGE_HUE = 1.0 + +# The width and colour settings for the band +BAND_PIXEL_WIDTH = 2.0 +BAND_SATURATION = 1.0 +BAND_IN_GOAL_SATURATION = 0.5 +BAND_BRIGHTNESS = 1.0 + +# The width and colour settings for the goal +# Goal should be wider than the band by a small amount +GOAL_PIXEL_WIDTH = BAND_PIXEL_WIDTH + 2.0 +GOAL_BRIGHTNESS = 0.1 + +# The percentage of the new angle (between 0.0 and 1.0) to apply to the last angle +# Has the effect of smoothing out the reading, at the cost of making it slower to react +SMOOTHING_FACTOR = 0.1 + + +# Pick *one* LED type by uncommenting the relevant line below: + +# APA102 / DotStar™ LEDs +led_strip = plasma.APA102(NUM_LEDS, 0, 0, plasma2040.DAT, plasma2040.CLK) + +# WS2812 / NeoPixel™ LEDs +# led_strip = plasma.WS2812(NUM_LEDS, 0, 0, plasma2040.DAT) + +user_sw = Button(plasma2040.USER_SW, repeat_time=0) +button_a = Button(plasma2040.BUTTON_A, repeat_time=0) +button_b = Button(plasma2040.BUTTON_B, repeat_time=0) +led = RGBLED(plasma2040.LED_R, plasma2040.LED_G, plasma2040.LED_B) + +PINS_PLASMA2040 = {"sda": plasma2040.SDA, "scl": plasma2040.SCL} + +i2c = PimoroniI2C(**PINS_PLASMA2040) +msa = BreakoutMSA301(i2c) + + +ANGLE, VELOCITY = range(2) + +# Maps a value from one range to another +def map(x, in_min, in_max, out_min, out_max): + return (x - in_min) * (out_max - out_min) / (in_max - in_min) + out_min + +# Shows a band and goal with the given widths at the positions on the strip +def colour_band(centre_position, width, goal_position, goal_width, hue): + if centre_position >= 0.0 and width > 0.0 and goal_width > 0.0: + band_pixels_start = centre_position - (width / 2) + band_pixels_end = centre_position + (width / 2) + + goal_pixels_start = goal_position - (goal_width / 2) + goal_pixels_end = goal_position + (goal_width / 2) + + # Go through each led in the strip + for i in range(NUM_LEDS): + + # Set saturation and brightness values for if the led is inside or outside of the goal + saturation = BAND_SATURATION + brightness = 0.0 + if i >= goal_pixels_start and i < goal_pixels_end: + saturation = BAND_IN_GOAL_SATURATION + brightness = GOAL_BRIGHTNESS + + i2 = i + 1 + if i2 <= band_pixels_end: + if i2 <= band_pixels_start: + # Outside of the band + led_strip.set_hsv(i, hue, 0.0, brightness) + elif i <= band_pixels_start: + # Transition into the band + val = map(band_pixels_start, float(i), float(i2), BAND_BRIGHTNESS, brightness) + sat = map(band_pixels_start, float(i), float(i2), BAND_SATURATION, saturation) + led_strip.set_hsv(i, hue, sat, val) + else: + # Inside the band + led_strip.set_hsv(i, hue, 1.0, 1.0) + + elif i <= band_pixels_end: + # Transition out of the band + val = map(band_pixels_end, float(i), float(i2), brightness, BAND_BRIGHTNESS) + sat = map(band_pixels_end, float(i), float(i2), saturation, BAND_SATURATION) + led_strip.set_hsv(i, hue, sat, val) + else: + # Outside of the band + led_strip.set_hsv(i, hue, 0.0, brightness) + + +mode = ANGLE + +goal_position = 0.0 +measured_angle = 0.0 +invert = False +game_mode = False + +# Start updating the LED strip +led_strip.start() + +while True: + # Read the x and y axes of the accelerometer + x = msa.get_x_axis() + y = msa.get_y_axis() + + # Convert those values to an angle in degrees, and invert if selected + new_measured_angle = (math.atan2(x, -y) * 180.0) / math.pi + if invert: + new_measured_angle = -new_measured_angle + print("Angle:", new_measured_angle, "deg") + + # Smooth out the measured angle + measured_angle = ((new_measured_angle - measured_angle) * SMOOTHING_FACTOR) + measured_angle + + if mode == ANGLE: + # Apply the measured angle directly to the used angle, clamping it between -1 and +1 + band_position = measured_angle * ANGLE_SENSITIVITY + band_position = min(1.0, max(-1.0, band_position)) + + # Convert the difference between the band and goal positions into a colour hue + position_diff = min(abs(band_position - goal_position), 1.0) + hue = map(position_diff, 0.0, 1.0, ANGLE_MODE_GOAL_HUE, ANGLE_MODE_EDGE_HUE) + + elif mode == VELOCITY: + band_position += measured_angle * VELOCITY_SENSITIVITY + band_position = min(1.0, max(-1.0, band_position)) + + # Convert the difference between the band and goal positions into a colour hue + position_diff = min(abs(band_position - goal_position), 1.0) + hue = map(position_diff, 0.0, 1.0, VELOCITY_MODE_GOAL_HUE, VELOCITY_MODE_EDGE_HUE) + + # Convert the band and goal positions to positions on the LED strip + strip_band_position = map(band_position, -1.0, 1.0, 0.0, float(NUM_LEDS)) + strip_goal_position = map(goal_position, -1.0, 1.0, 0.0, float(NUM_LEDS)) + + # Draw the band and goal + colour_band(strip_band_position, BAND_PIXEL_WIDTH, strip_goal_position, GOAL_PIXEL_WIDTH, hue) + + sw_pressed = user_sw.read() + a_pressed = button_a.read() + b_pressed = button_b.read() + + if b_pressed: + game_mode = not game_mode + + if sw_pressed: + invert = not invert + + if mode == ANGLE: + if game_mode: + led.set_rgb(255, 255, 0) + else: + led.set_rgb(0, 255, 0) + if a_pressed: + mode = VELOCITY + + elif mode == VELOCITY: + if game_mode: + led.set_rgb(255, 0, 255) + else: + led.set_rgb(0, 0, 255) + if a_pressed: + mode = ANGLE + + if game_mode: + # Check if the band is within the goal, and if so, set a new goal + if(strip_band_position >= strip_goal_position - (GOAL_PIXEL_WIDTH - BAND_PIXEL_WIDTH) / 2 and + strip_band_position <= strip_goal_position + (GOAL_PIXEL_WIDTH - BAND_PIXEL_WIDTH) / 2): + goal_position = random.uniform(-1.0, 1.0) + + time.sleep(1.0 / UPDATES)