86 lines
2.3 KiB
Python
86 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
|
"""Record 5s from the USB Speaker Phone with a live RMS meter, save out.wav.
|
|
|
|
Pass criteria for the record stage:
|
|
- Level meter visibly responds to speech and to silence.
|
|
- out.wav is written and non-empty.
|
|
"""
|
|
import os
|
|
os.environ.setdefault("PA_ALSA_PLUGHW", "1")
|
|
|
|
import sys
|
|
import time
|
|
|
|
import numpy as np
|
|
import sounddevice as sd
|
|
import soundfile as sf
|
|
|
|
SAMPLE_RATE = 16_000
|
|
DURATION_S = 5
|
|
CHANNELS = 1
|
|
DTYPE = "int16"
|
|
OUT_WAV = "out.wav"
|
|
|
|
|
|
def find_usb_device() -> int:
|
|
for idx, dev in enumerate(sd.query_devices()):
|
|
name = dev["name"].lower()
|
|
if "usb" in name and dev["max_input_channels"] >= 1:
|
|
return idx
|
|
print("USB audio device not found. Devices:", file=sys.stderr)
|
|
print(sd.query_devices(), file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
|
|
def rms(int16_buf: np.ndarray) -> float:
|
|
if int16_buf.size == 0:
|
|
return 0.0
|
|
x = int16_buf.astype(np.float32) / 32768.0
|
|
return float(np.sqrt((x * x).mean() + 1e-12))
|
|
|
|
|
|
def record(device: int) -> np.ndarray:
|
|
print(f"Recording {DURATION_S}s from device {device} ({sd.query_devices(device)['name']!r})")
|
|
total_frames = SAMPLE_RATE * DURATION_S
|
|
buf = np.zeros((total_frames, CHANNELS), dtype=DTYPE)
|
|
frames_written = 0
|
|
start = time.monotonic()
|
|
|
|
def callback(indata, frames, time_info, status):
|
|
nonlocal frames_written
|
|
if status:
|
|
print(f"\n[status] {status}", file=sys.stderr)
|
|
end = min(frames_written + frames, total_frames)
|
|
buf[frames_written:end] = indata[: end - frames_written]
|
|
frames_written = end
|
|
|
|
with sd.InputStream(
|
|
samplerate=SAMPLE_RATE,
|
|
channels=CHANNELS,
|
|
dtype=DTYPE,
|
|
device=device,
|
|
blocksize=1024,
|
|
callback=callback,
|
|
):
|
|
while frames_written < total_frames:
|
|
elapsed = time.monotonic() - start
|
|
tail = buf[max(0, frames_written - 1600) : frames_written]
|
|
level = rms(tail)
|
|
bar = "#" * int(level * 60)
|
|
print(f"\r{elapsed:4.1f}s |{bar:<60}|", end="", flush=True)
|
|
time.sleep(0.05)
|
|
print()
|
|
return buf
|
|
|
|
|
|
def main() -> int:
|
|
device = find_usb_device()
|
|
audio = record(device)
|
|
sf.write(OUT_WAV, audio, SAMPLE_RATE)
|
|
print(f"Wrote {OUT_WAV} ({audio.shape[0]} frames @ {SAMPLE_RATE} Hz)")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|