def __init__(self, dev, DEVICE_CONFIG, led_id, freq, filename):
if not validate_params(DEVICE_CONFIG, freq, led_id):
raise ValueError("Invalid music LED stream parameters")
success, (ready, free) = fetch_free_steps_buffer_info(
dev, led_id, DEVICE_CONFIG['stream_led_cnt'])
if not success:
raise Exception("Device communication problem occurred")
if not ready:
raise Exception("LEDs are not ready")
try:
self.file = wave.open(filename, 'rb')
except (wave.Error, FileNotFoundError) as e:
raise e
self.dev_params = {
'max_free': free,
'dev': dev,
'stream_led_cnt': DEVICE_CONFIG['stream_led_cnt'],
'led_id': led_id,
}
self.send_error_event = threading.Event()
self.queue = queue.Queue()
self.p = pyaudio.PyAudio()
self.stream = self.p.open(
format=self.p.get_format_from_width(self.file.getsampwidth()),
channels=self.file.getnchannels(),
rate=self.file.getframerate(),
output=True,
input=False,
stream_callback=self.music_callback,
frames_per_buffer=self.file.getframerate() // freq,
start=False)
self.led_effects = {
'BASE_DURATION': 1 / freq,
'SUBSTEP_CNT': 10,
'sent_cnt': 0,
'reminder': 0,
'duration_increment': 0,
'start_time': None,
'out_latency': self.stream.get_output_latency(),
}
def send_led_effects(self):
while True:
data = self.queue.get()
if data is None:
# LED stream ended
return
data = np.frombuffer(data, dtype=np.int16)
duration = self.led_effects['BASE_DURATION'] + \
self.led_effects['reminder'] + \
self.led_effects['duration_increment']
r, g, b = MusicLedStream.gen_led_color(data)
step = Step(r=int(r * 255),
g=int(g * 255),
b=int(b * 255),
substep_count=self.led_effects['SUBSTEP_CNT'],
substep_time=int(duration * MS_PER_SEC /
self.led_effects['SUBSTEP_CNT']))
if step.substep_time < 0:
step.substep_time = 0
self.led_effects['reminder'] = duration - \
(step.substep_count * step.substep_time / MS_PER_SEC)
success = led_send_single_step(self.dev_params['dev'], step,
self.dev_params['led_id'])
if not success:
print("Device communication problem")
self.send_error_event.set()
return
self.led_effects['sent_cnt'] += 1
success, (ready, free) = fetch_free_steps_buffer_info(
self.dev_params['dev'], self.dev_params['led_id'],
self.dev_params['stream_led_cnt'])
if not success or not ready:
self.send_error_event.set()
if not ready:
print("LEDs are not ready")
else:
print("Device communication problem")
return
cur_step_speaker = (time.time() - self.led_effects['start_time'] - \
self.led_effects['out_latency']) \
// self.led_effects['BASE_DURATION']
if cur_step_speaker < 0:
# Speaker not ready yet
continue
in_buf_speaker = self.led_effects['sent_cnt'] - cur_step_speaker - 1
in_buf_dev = self.dev_params['max_free'] - free
# LED effect on device is synchronized with music played by speaker.
# Ensure that device will always have buffered LED effect.
if in_buf_speaker < 1:
in_buf_speaker = 1
DURATION_INCREMENT_CONST_FACTOR = 0.01
DURATION_INCREMENT_VARIABLE_FACTOR = 0.1
if in_buf_speaker != 0:
duration_inc_variable = (in_buf_speaker - in_buf_dev) / \
abs(in_buf_speaker) * DURATION_INCREMENT_VARIABLE_FACTOR
else:
duration_inc_variable = 0
self.led_effects['duration_increment'] = duration_inc_variable
if in_buf_speaker > in_buf_dev:
self.led_effects[
'duration_increment'] += DURATION_INCREMENT_CONST_FACTOR
if in_buf_speaker < in_buf_dev:
self.led_effects[
'duration_increment'] -= DURATION_INCREMENT_CONST_FACTOR
self.led_effects['duration_increment'] *= self.led_effects[
'BASE_DURATION']