33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
|
import time
|
||
|
|
import numpy as np
|
||
|
|
import pyaudio
|
||
|
|
import config
|
||
|
|
|
||
|
|
|
||
|
|
def start_stream(callback):
|
||
|
|
p = pyaudio.PyAudio()
|
||
|
|
frames_per_buffer = int(config.MIC_RATE / config.FPS)
|
||
|
|
stream = p.open(format=pyaudio.paInt16,
|
||
|
|
channels=1,
|
||
|
|
rate=config.MIC_RATE,
|
||
|
|
input=True,
|
||
|
|
frames_per_buffer=frames_per_buffer)
|
||
|
|
overflows = 0
|
||
|
|
prev_ovf_time = time.time()
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
y = np.fromstring(stream.read(frames_per_buffer), dtype=np.int16)
|
||
|
|
y = y.astype(np.float32)
|
||
|
|
callback(y)
|
||
|
|
except IOError:
|
||
|
|
overflows += 1
|
||
|
|
if time.time() > prev_ovf_time + 1:
|
||
|
|
prev_ovf_time = time.time()
|
||
|
|
if config.USE_GUI:
|
||
|
|
gui.label_error.setText('Audio buffer has overflowed {} times'.format(overflows))
|
||
|
|
else:
|
||
|
|
print('Audio buffer has overflowed {} times'.format(overflows))
|
||
|
|
stream.stop_stream()
|
||
|
|
stream.close()
|
||
|
|
p.terminate()
|