Files
dog-trainer/dog-trainer.py
2023-12-02 03:15:11 +01:00

140 lines
3.7 KiB
Python
Executable File

import RPi.GPIO as GPIO
import time
from datetime import datetime
import subprocess
import sys
PLAY_DEVICE = "my_usb_sound_device_with_converter" # as defined in '~/.asoundrc'
REC_DEVICE = "my_usb_sound_device" # as defined in '~/.asoundrc'
MAX_RECORDING_TIME = 30 # limit recordings to only e.g. 30 seconds long
TRIM_RECORDING_END = 0.8 # remove the clicking noise from the end of recordings
RECORD_BUTTON_PRESS_TIME = 2 # pressing and holding a button longer then e.g. 2 seconds triggers recording otherwise playing
PINS = [17, 23] # listen to changes on this GPIOs e.g. [4, 17, 23, 0, 5, 6]
GPIO.setmode(GPIO.BCM) # use BCM pin layout
GPIO.setwarnings(False)
def setup_pins():
for pin in PINS:
GPIO.setup(pin, GPIO.IN, pull_up_down=GPIO.PUD_OFF)
def terminate_recording(proc, pin):
print("Terminating recording...")
proc.terminate()
proc.wait(timeout=5)
print(f"Cutting last {TRIM_RECORDING_END} second (click noise) from the recording")
audio_duration = subprocess.run(
[
"ffprobe",
"-v",
"error",
"-select_streams",
"a:0",
"-show_entries",
"stream=duration",
"-of",
"default=noprint_wrappers=1:nokey=1",
f"recording_temp_{pin}.wav",
],
stdout=subprocess.PIPE,
).stdout.decode("utf-8")
subprocess.call(
[
"ffmpeg",
"-hide_banner",
"-y",
"-i",
f"recording_temp_{pin}.wav",
"-t",
f"{float(audio_duration) - TRIM_RECORDING_END}",
"-acodec",
"copy",
f"recording_{pin}.wav",
]
)
DATE_TIME = datetime.now().strftime("%Y-%m-%d_%H:%M:%S")
print(f"Archiving old sound to archive/recording_{pin}_{DATE_TIME}.wav")
subprocess.call(
["mv", f"recording_temp_{pin}.wav", f"archive/recording_{pin}__{DATE_TIME}.wav"]
)
play_sound(pin)
def play(file):
subprocess.call(["aplay", "-D", PLAY_DEVICE, file])
def play_sound(pin):
play(f"recording_{pin}.wav")
def play_beep():
play("beep.wav")
def play_intro():
play("Windows_95.wav")
def record_sound(pin):
print(f"GPIO {pin} LOW - Starting recording_temp_{pin}.wav")
proc = subprocess.Popen(
[
"arecord",
"-D",
REC_DEVICE,
"-f",
"S16_LE",
"-r",
"16000",
"-c",
"1",
"-d",
f"{MAX_RECORDING_TIME}",
f"recording_temp_{pin}.wav",
]
)
print(f"Recording in progress (max {MAX_RECORDING_TIME} seconds)...")
wait_for_button_release_or_timeout(pin, MAX_RECORDING_TIME)
print("GPIO went HIGH - terminating recording")
play_beep()
terminate_recording(proc, pin)
def wait_for_button_release_or_timeout(pin, timeout):
timeout_start = time.time()
while time.time() < timeout_start + timeout:
if GPIO.input(pin) == GPIO.HIGH:
break
time.sleep(0.05)
def handle_pin(pin):
print(f"GPIO {pin} LOW - event triggered")
timeout_start = time.time()
wait_for_button_release_or_timeout(pin, RECORD_BUTTON_PRESS_TIME)
if time.time() - timeout_start >= RECORD_BUTTON_PRESS_TIME:
# Pin was LOW for longer time; recording
play_beep()
record_sound(pin)
else:
# Pin was LOW for shorter time; playing
play_sound(pin)
try:
setup_pins()
subprocess.call(["mkdir", "-p", "archive"])
play_intro()
for pin in PINS:
GPIO.add_event_detect(pin, GPIO.FALLING, callback=handle_pin, bouncetime=1000)
while True:
time.sleep(0.01)
except Exception:
print("Cought exception...")
sys.exit()
finally:
GPIO.cleanup()