Merge pull request #2556 from adafruit/adabot_prop-maker

Adding code and audio for adabot

Author: TheKitty

Adabot_RP2040_Prop-Maker_Feather/WAVs/adabot.wav

@@ -0,0 +1,147 @@
+# SPDX-FileCopyrightText: 2023 Liz Clark for Adafruit Industries
+#
+# SPDX-License-Identifier: MIT
+
+import os
+import random
+import board
+import audiocore
+import audiobusio
+import audiomixer
+import pwmio
+import neopixel
+import adafruit_lis3dh
+from adafruit_ticks import ticks_ms, ticks_add, ticks_diff
+from digitalio import DigitalInOut, Direction, Pull
+from adafruit_motor import servo
+from adafruit_led_animation.animation.comet import Comet
+from adafruit_led_animation.animation.pulse import Pulse
+from adafruit_led_animation.color import RED, BLUE, BLACK
+
+# enable external power pin
+# provides power to the external components
+external_power = DigitalInOut(board.EXTERNAL_POWER)
+external_power.direction = Direction.OUTPUT
+external_power.value = True
+
+i2c = board.I2C()
+int1 = DigitalInOut(board.ACCELEROMETER_INTERRUPT)
+lis3dh = adafruit_lis3dh.LIS3DH_I2C(i2c, int1=int1)
+lis3dh.range = adafruit_lis3dh.RANGE_2_G
+
+switch = DigitalInOut(board.EXTERNAL_BUTTON)
+switch.direction = Direction.INPUT
+switch.pull = Pull.UP
+switch_state = False
+
+wavs = []
+for filename in os.listdir('/WAVs'):
+    if filename.lower().endswith('.wav') and not filename.startswith('.'):
+        wavs.append("/WAVs/"+filename)
+
+audio = audiobusio.I2SOut(board.I2S_BIT_CLOCK, board.I2S_WORD_SELECT, board.I2S_DATA)
+mixer = audiomixer.Mixer(voice_count=1, sample_rate=22050, channel_count=1,
+                         bits_per_sample=16, samples_signed=True, buffer_size=32768)
+
+mixer.voice[0].level = 1
+track_number = 0
+wav_filename = wavs[track_number]
+wav_file = open(wav_filename, "rb")
+wave = audiocore.WaveFile(wav_file)
+audio.play(mixer)
+mixer.voice[0].play(wave)
+
+def open_audio(num):
+    n = wavs[num]
+    f = open(n, "rb")
+    w = audiocore.WaveFile(f)
+    return w
+
+PIXEL_PIN = board.EXTERNAL_NEOPIXELS
+SERVO_PIN = board.EXTERNAL_SERVO
+NUM_PIXELS = 8
+ORDER = neopixel.GRB
+BRIGHTNESS = 0.6
+
+PWM = pwmio.PWMOut(SERVO_PIN, duty_cycle=2 ** 15, frequency=50)
+SERVO = servo.Servo(PWM)
+
+pixel = neopixel.NeoPixel(board.NEOPIXEL, 1)
+pixel.brightness = 1
+
+PIXELS = neopixel.NeoPixel(PIXEL_PIN, NUM_PIXELS, auto_write=False,
+                           pixel_order=ORDER)
+LARSON = Comet(PIXELS, bounce=True, speed=0.6/NUM_PIXELS,
+               tail_length=NUM_PIXELS//2,
+               color=(RED[0] * BRIGHTNESS,
+                      RED[1] * BRIGHTNESS,
+                      RED[2] * BRIGHTNESS))
+pulse = Pulse(PIXELS, speed=0.1, color=BLUE, period=3)
+
+SERVO.angle = POSITION = NEXT_POSITION = 90
+MOVING = False
+START_TIME = ticks_ms()
+DURATION = 1000
+
+adabot_talk = False
+
+clock = ticks_ms()
+prop_time = 1000
+adabot_nap = False
+
+mixer.voice[0].play(wave)
+while mixer.playing:
+    LARSON.animate()
+
+while True:
+    if ticks_diff(ticks_ms(), clock) >= prop_time:
+        x, y, z = [
+        value / adafruit_lis3dh.STANDARD_GRAVITY for value in lis3dh.acceleration
+        ]
+        if z > 0.9:
+            adabot_nap = True
+            SERVO.angle = POSITION = NEXT_POSITION = 90
+            LARSON.color=(BLUE[0] * BRIGHTNESS,
+                          BLUE[1] * BRIGHTNESS,
+                          BLUE[2] * BRIGHTNESS)
+        else:
+            adabot_nap = False
+            LARSON.color=(RED[0] * BRIGHTNESS,
+                          RED[1] * BRIGHTNESS,
+                          RED[2] * BRIGHTNESS)
+        if not adabot_nap:
+            MOVING = not MOVING
+            if MOVING:
+                POSITION = NEXT_POSITION
+                while abs(POSITION - NEXT_POSITION) < 10:
+                    NEXT_POSITION = random.uniform(0, 180)
+                DURATION = 0.2 + 0.6 * abs(POSITION - NEXT_POSITION) / 180
+            else:
+                SERVO.angle = NEXT_POSITION
+                DURATION = random.uniform(0.5, 2.5)
+        clock = ticks_add(clock, prop_time)
+    if MOVING:
+        FRACTION = 0.0 / DURATION
+        FRACTION = (3 * FRACTION ** 2) - (2 * FRACTION ** 3)
+        SERVO.angle = POSITION + (NEXT_POSITION - POSITION) * FRACTION
+    if adabot_talk:
+        wave = open_audio(random.randint(1, 7))
+        mixer.voice[0].play(wave)
+        while mixer.playing:
+            LARSON.animate()
+        if not mixer.playing:
+            adabot_talk = False
+            PIXELS.fill(BLACK)
+            PIXELS.show()
+    elif adabot_nap:
+        LARSON.animate()
+    else:
+        pulse.animate()
+
+    if not switch.value and switch_state is False:
+        PIXELS.fill(BLACK)
+        PIXELS.show()
+        adabot_talk = True
+        switch_state = True
+    if switch.value and switch_state is True:
+        switch_state = False