audio-reactive-led-strip/python/lib/microphone.py

import time
import numpy as np
import pyaudio
import lib.config as config


def start_stream(callback):
    p = pyaudio.PyAudio()
    frames_per_buffer = int(config.settings["configuration"]["MIC_RATE"] / config.settings["configuration"]["FPS"])
    stream = p.open(format=pyaudio.paInt16,
                    channels=1,
                    rate=config.settings["configuration"]["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.settings["configuration"]["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()