128 lines
3.8 KiB
Python
128 lines
3.8 KiB
Python
import time
|
|
import random
|
|
import digitalio
|
|
import audioio
|
|
import busio
|
|
import board
|
|
import adafruit_rgbled
|
|
import adafruit_lis3dh
|
|
|
|
# CUSTOMISE COLORS HERE:
|
|
MAIN_COLOR = (255, 0, 0) # Default is red
|
|
HIT_COLOR = (255, 255, 255) # Default is white
|
|
|
|
# CUSTOMISE SENSITIVITY HERE: smaller numbers = more sensitive to motion
|
|
HIT_THRESHOLD = 650
|
|
SWING_THRESHOLD = 125
|
|
|
|
# Set to the length in seconds of the "on.wav" file
|
|
POWER_ON_SOUND_DURATION = 1.7
|
|
|
|
POWER_PIN = board.D10
|
|
|
|
enable = digitalio.DigitalInOut(POWER_PIN)
|
|
enable.direction = digitalio.Direction.OUTPUT
|
|
enable.value = False
|
|
|
|
# Pin the Red LED is connected to
|
|
RED_LED = board.D11
|
|
|
|
# Pin the Green LED is connected to
|
|
GREEN_LED = board.D12
|
|
|
|
# Pin the Blue LED is connected to
|
|
BLUE_LED = board.D13
|
|
|
|
# Create the RGB LED object
|
|
led = adafruit_rgbled.RGBLED(RED_LED, GREEN_LED, BLUE_LED)
|
|
|
|
audio = audioio.AudioOut(board.A0) # Speaker
|
|
|
|
# Set up accelerometer on I2C bus, 4G range:
|
|
i2c = busio.I2C(board.SCL, board.SDA)
|
|
accel = adafruit_lis3dh.LIS3DH_I2C(i2c)
|
|
accel.range = adafruit_lis3dh.RANGE_4_G
|
|
|
|
COLOR_HIT = HIT_COLOR # "hit" color is HIT_COLOR set above
|
|
COLOR_SWING = MAIN_COLOR # "swing" color is MAIN_COLOR set above
|
|
|
|
|
|
def play_wav(name, loop=False):
|
|
"""
|
|
Play a WAV file in the 'sounds' directory.
|
|
:param name: partial file name string, complete name will be built around
|
|
this, e.g. passing 'foo' will play file 'sounds/foo.wav'.
|
|
:param loop: if True, sound will repeat indefinitely (until interrupted
|
|
by another sound).
|
|
"""
|
|
print("playing", name)
|
|
try:
|
|
wave_file = open('sounds/' + name + '.wav', 'rb')
|
|
wave = audioio.WaveFile(wave_file)
|
|
audio.play(wave, loop=loop)
|
|
except: # pylint: disable=bare-except
|
|
return
|
|
|
|
|
|
# List of swing wav files without the .wav in the name for use with play_wav()
|
|
swing_sounds = [
|
|
'swing1',
|
|
'swing2',
|
|
'swing3',
|
|
'swing4',
|
|
'swing5',
|
|
'swing6',
|
|
'swing7',
|
|
'swing8',
|
|
]
|
|
|
|
# List of hit wav files without the .wav in the name for use with play_wav()
|
|
hit_sounds = [
|
|
'hit1',
|
|
'hit2',
|
|
'hit3',
|
|
'hit4',
|
|
'hit5',
|
|
'hit6',
|
|
'hit7',
|
|
'hit8',
|
|
]
|
|
|
|
mode = 0 # Initial mode = OFF
|
|
|
|
# Main loop
|
|
while True:
|
|
if mode == 0: # If currently off...
|
|
enable.value = True
|
|
play_wav('on') # Power up!
|
|
led.color = MAIN_COLOR
|
|
time.sleep(POWER_ON_SOUND_DURATION)
|
|
play_wav('idle', loop=True) # Play idle sound now
|
|
mode = 1 # Idle mode
|
|
|
|
elif mode >= 1: # If not OFF mode...
|
|
x, y, z = accel.acceleration # Read accelerometer
|
|
accel_total = x * x + z * z
|
|
# (Y axis isn't needed, due to the orientation that the Prop-Maker
|
|
# Wing is mounted. Also, square root isn't needed, since we're
|
|
# comparing thresholds...use squared values instead.)
|
|
if accel_total > HIT_THRESHOLD: # Large acceleration = HIT
|
|
play_wav(random.choice(hit_sounds)) # Start playing 'hit' sound
|
|
COLOR_ACTIVE = COLOR_HIT # Set color to fade from
|
|
mode = 3 # HIT mode
|
|
elif mode == 1 and accel_total > SWING_THRESHOLD: # Mild = SWING
|
|
play_wav(random.choice(swing_sounds)) # Randomly choose from available swing sounds
|
|
led.color = MAIN_COLOR # Set color to main color
|
|
mode = 2 # SWING mode
|
|
elif mode == 1:
|
|
# Idle color
|
|
led.color = MAIN_COLOR
|
|
elif mode > 1: # If in SWING or HIT mode...
|
|
if audio.playing: # And sound currently playing...
|
|
if mode == 2: # If SWING,
|
|
led.color = MAIN_COLOR
|
|
else:
|
|
led.color = HIT_COLOR # Set color to hit color
|
|
else: # No sound now, but still SWING or HIT modes
|
|
play_wav('idle', loop=True) # Resume idle sound
|
|
mode = 1 # Return to idle mode
|