13namespace mixer_speaker {
15static const UBaseType_t MIXER_TASK_PRIORITY = 10;
17static const uint32_t TRANSFER_BUFFER_DURATION_MS = 50;
18static const uint32_t TASK_DELAY_MS = 25;
20static const size_t TASK_STACK_SIZE = 4096;
22static const int16_t MAX_AUDIO_SAMPLE_VALUE = INT16_MAX;
23static const int16_t MIN_AUDIO_SAMPLE_VALUE = INT16_MIN;
25static const char *
const TAG =
"speaker_mixer";
30static const std::vector<int16_t> DECIBEL_REDUCTION_TABLE = {
31 32767, 29201, 26022, 23189, 20665, 18415, 16410, 14624, 13032, 11613, 10349, 9222, 8218, 7324, 6527, 5816, 5183,
32 4619, 4116, 3668, 3269, 2913, 2596, 2313, 2061, 1837, 1637, 1459, 1300, 1158, 1032, 920, 820, 731,
33 651, 580, 517, 461, 411, 366, 326, 291, 259, 231, 206, 183, 163, 146, 130, 116, 103};
47 "Mixer Source Speaker\n"
48 " Buffer Duration: %" PRIu32
" ms",
53 ESP_LOGCONFIG(TAG,
" Timeout: never");
63 if (speakers_playback_frames > 0) {
72 esp_err_t err = this->
start_();
83 case ESP_ERR_NOT_SUPPORTED:
84 this->
status_set_error(LOG_STR(
"Failed to start mixer: unsupported bits per sample"));
86 case ESP_ERR_INVALID_ARG:
88 LOG_STR(
"Failed to start mixer: audio stream isn't compatible with the other audio stream."));
90 case ESP_ERR_INVALID_STATE:
91 this->
status_set_error(LOG_STR(
"Failed to start mixer: mixer task failed to start"));
105 this->stop_gracefully_) {
124 size_t bytes_written = 0;
126 std::shared_ptr<RingBuffer> temp_ring_buffer = this->
ring_buffer_.lock();
127 bytes_written = temp_ring_buffer->write_without_replacement(data,
length, ticks_to_wait);
128 if (bytes_written > 0) {
132 return bytes_written;
144 return ESP_ERR_NO_MEM;
146 std::shared_ptr<RingBuffer> temp_ring_buffer;
154 return ESP_ERR_NO_MEM;
177 return ((this->
transfer_buffer_.use_count() > 0) && this->transfer_buffer_->has_buffered_data());
202 size_t bytes_read = this->
transfer_buffer_->transfer_data_from_source(ticks_to_wait);
205 if (samples_to_duck > 0) {
206 int16_t *current_buffer =
reinterpret_cast<int16_t *
>(this->
transfer_buffer_->get_buffer_start() + current_length);
222 uint8_t total_ducking_steps = 0;
232 if ((duration > 0) && (total_ducking_steps > 0)) {
248 int8_t *current_ducking_db_reduction, uint32_t *ducking_transition_samples_remaining,
249 uint32_t samples_per_ducking_step, int8_t db_change_per_ducking_step) {
250 if (*ducking_transition_samples_remaining > 0) {
254 uint32_t ducking_steps_in_batch =
255 input_samples_to_duck / samples_per_ducking_step + (input_samples_to_duck % samples_per_ducking_step != 0);
257 for (uint32_t i = 0; i < ducking_steps_in_batch; ++i) {
258 uint32_t samples_left_in_step = *ducking_transition_samples_remaining % samples_per_ducking_step;
260 if (samples_left_in_step == 0) {
261 samples_left_in_step = samples_per_ducking_step;
264 uint32_t samples_to_duck = std::min(input_samples_to_duck, samples_left_in_step);
265 samples_to_duck = std::min(samples_to_duck, *ducking_transition_samples_remaining);
268 uint8_t safe_db_reduction_index =
269 clamp<uint8_t>(*current_ducking_db_reduction, 0, DECIBEL_REDUCTION_TABLE.size() - 1);
270 int16_t q15_scale_factor = DECIBEL_REDUCTION_TABLE[safe_db_reduction_index];
274 if (samples_left_in_step - samples_to_duck == 0) {
276 *current_ducking_db_reduction += db_change_per_ducking_step;
279 input_buffer += samples_to_duck;
280 *ducking_transition_samples_remaining -= samples_to_duck;
281 input_samples_to_duck -= samples_to_duck;
285 if ((*current_ducking_db_reduction > 0) && (input_samples_to_duck > 0)) {
288 uint8_t safe_db_reduction_index =
289 clamp<uint8_t>(*current_ducking_db_reduction, 0, DECIBEL_REDUCTION_TABLE.size() - 1);
290 int16_t q15_scale_factor = DECIBEL_REDUCTION_TABLE[safe_db_reduction_index];
299 " Number of output channels: %u",
307 ESP_LOGE(TAG,
"Failed to create event group");
314 uint32_t event_group_bits = xEventGroupGetBits(this->
event_group_);
317 ESP_LOGD(TAG,
"Starting speaker mixer");
318 xEventGroupClearBits(this->
event_group_, MixerEventGroupBits::STATE_STARTING);
321 this->
status_set_error(LOG_STR(
"Failed to allocate the mixer's internal buffer"));
322 xEventGroupClearBits(this->
event_group_, MixerEventGroupBits::ERR_ESP_NO_MEM);
325 ESP_LOGD(TAG,
"Started speaker mixer");
327 xEventGroupClearBits(this->
event_group_, MixerEventGroupBits::STATE_RUNNING);
330 ESP_LOGD(TAG,
"Stopping speaker mixer");
331 xEventGroupClearBits(this->
event_group_, MixerEventGroupBits::STATE_STOPPING);
340 bool all_stopped =
true;
343 all_stopped &= speaker->is_stopped();
356 return ESP_ERR_NOT_SUPPORTED;
365 return ESP_ERR_INVALID_ARG;
384 return ESP_ERR_NO_MEM;
393 return ESP_ERR_INVALID_STATE;
418 return ESP_ERR_INVALID_STATE;
425 uint32_t frames_to_transfer) {
426 uint8_t input_channels = input_stream_info.
get_channels();
427 uint8_t output_channels = output_stream_info.
get_channels();
428 const uint8_t max_input_channel_index = input_channels - 1;
430 if (input_channels == output_channels) {
431 size_t bytes_to_copy = input_stream_info.
frames_to_bytes(frames_to_transfer);
432 memcpy(output_buffer, input_buffer, bytes_to_copy);
437 for (uint32_t frame_index = 0; frame_index < frames_to_transfer; ++frame_index) {
438 for (uint8_t output_channel_index = 0; output_channel_index < output_channels; ++output_channel_index) {
439 uint8_t input_channel_index = std::min(output_channel_index, max_input_channel_index);
440 output_buffer[output_channels * frame_index + output_channel_index] =
441 input_buffer[input_channels * frame_index + input_channel_index];
449 uint32_t frames_to_mix) {
450 const uint8_t primary_channels = primary_stream_info.
get_channels();
451 const uint8_t secondary_channels = secondary_stream_info.
get_channels();
452 const uint8_t output_channels = output_stream_info.
get_channels();
454 const uint8_t max_primary_channel_index = primary_channels - 1;
455 const uint8_t max_secondary_channel_index = secondary_channels - 1;
457 for (uint32_t frames_index = 0; frames_index < frames_to_mix; ++frames_index) {
458 for (uint8_t output_channel_index = 0; output_channel_index < output_channels; ++output_channel_index) {
459 const uint32_t secondary_channel_index = std::min(output_channel_index, max_secondary_channel_index);
460 const int32_t secondary_sample = secondary_buffer[frames_index * secondary_channels + secondary_channel_index];
462 const uint32_t primary_channel_index = std::min(output_channel_index, max_primary_channel_index);
463 const int32_t primary_sample =
464 static_cast<int32_t
>(primary_buffer[frames_index * primary_channels + primary_channel_index]);
466 const int32_t added_sample = secondary_sample + primary_sample;
468 output_buffer[frames_index * output_channels + output_channel_index] =
469 static_cast<int16_t
>(clamp<int32_t>(added_sample, MIN_AUDIO_SAMPLE_VALUE, MAX_AUDIO_SAMPLE_VALUE));
484 if (output_transfer_buffer ==
nullptr) {
489 vTaskDelete(
nullptr);
496 bool sent_finished =
false;
499 uint32_t event_group_bits = xEventGroupGetBits(this_mixer->
event_group_);
505 output_transfer_buffer->transfer_data_to_sink(pdMS_TO_TICKS(TASK_DELAY_MS),
false);
507 const uint32_t output_frames_free =
508 this_mixer->
audio_stream_info_.value().bytes_to_frames(output_transfer_buffer->free());
510 std::vector<SourceSpeaker *> speakers_with_data;
511 std::vector<std::shared_ptr<audio::AudioSourceTransferBuffer>> transfer_buffers_with_data;
514 if (speaker->get_transfer_buffer().use_count() > 0) {
515 std::shared_ptr<audio::AudioSourceTransferBuffer> transfer_buffer = speaker->get_transfer_buffer().lock();
516 speaker->process_data_from_source(0);
518 if ((transfer_buffer->available() > 0) && !speaker->get_pause_state()) {
520 transfer_buffers_with_data.push_back(transfer_buffer);
521 speakers_with_data.push_back(speaker);
526 if (transfer_buffers_with_data.empty()) {
528 delay(TASK_DELAY_MS);
532 uint32_t frames_to_mix = output_frames_free;
534 if ((transfer_buffers_with_data.size() == 1) || this_mixer->
queue_mode_) {
543 const uint32_t frames_available_in_buffer =
544 active_stream_info.
bytes_to_frames(transfer_buffers_with_data[0]->available());
545 frames_to_mix = std::min(frames_to_mix, frames_available_in_buffer);
546 copy_frames(
reinterpret_cast<int16_t *
>(transfer_buffers_with_data[0]->get_buffer_start()), active_stream_info,
547 reinterpret_cast<int16_t *
>(output_transfer_buffer->get_buffer_end()),
551 transfer_buffers_with_data[0]->decrease_buffer_length(active_stream_info.
frames_to_bytes(frames_to_mix));
552 speakers_with_data[0]->pending_playback_frames_ += frames_to_mix;
555 output_transfer_buffer->increase_buffer_length(
560 if (!sent_finished) {
562 sent_finished =
true;
571 sent_finished =
false;
576 for (
size_t i = 0; i < transfer_buffers_with_data.size(); ++i) {
577 const uint32_t frames_available_in_buffer =
578 speakers_with_data[i]->get_audio_stream_info().bytes_to_frames(transfer_buffers_with_data[i]->available());
579 frames_to_mix = std::min(frames_to_mix, frames_available_in_buffer);
581 int16_t *primary_buffer =
reinterpret_cast<int16_t *
>(transfer_buffers_with_data[0]->get_buffer_start());
585 for (
size_t i = 1; i < transfer_buffers_with_data.size(); ++i) {
587 reinterpret_cast<int16_t *
>(transfer_buffers_with_data[i]->get_buffer_start()),
588 speakers_with_data[i]->get_audio_stream_info(),
589 reinterpret_cast<int16_t *
>(output_transfer_buffer->get_buffer_end()),
592 if (i != transfer_buffers_with_data.size() - 1) {
594 primary_buffer =
reinterpret_cast<int16_t *
>(output_transfer_buffer->get_buffer_end());
600 for (
size_t i = 0; i < transfer_buffers_with_data.size(); ++i) {
601 transfer_buffers_with_data[i]->decrease_buffer_length(
602 speakers_with_data[i]->get_audio_stream_info().frames_to_bytes(frames_to_mix));
603 speakers_with_data[i]->pending_playback_frames_ += frames_to_mix;
607 output_transfer_buffer->increase_buffer_length(
614 output_transfer_buffer.reset();
618 vTaskDelete(
nullptr);
virtual void mark_failed()
Mark this component as failed.
void status_clear_error()
An STL allocator that uses SPI or internal RAM.
void deallocate(T *p, size_t n)
static std::unique_ptr< RingBuffer > create(size_t len)
static std::unique_ptr< AudioSinkTransferBuffer > create(size_t buffer_size)
Creates a new sink transfer buffer.
static std::unique_ptr< AudioSourceTransferBuffer > create(size_t buffer_size)
Creates a new source transfer buffer.
size_t ms_to_bytes(uint32_t ms) const
Converts duration to bytes.
size_t frames_to_bytes(uint32_t frames) const
Converts frames to bytes.
uint8_t get_bits_per_sample() const
uint32_t ms_to_samples(uint32_t ms) const
Converts duration to samples.
uint32_t bytes_to_frames(size_t bytes) const
Convert bytes to frames.
uint8_t get_channels() const
uint32_t get_sample_rate() const
uint32_t bytes_to_samples(size_t bytes) const
Convert bytes to samples.
esp_err_t start_task_()
Starts the mixer task after allocating memory for the task stack.
esp_err_t delete_task_()
If the task is stopped, it sets the task handle to the nullptr and deallocates its stack.
void dump_config() override
esp_err_t start(audio::AudioStreamInfo &stream_info)
Starts the mixer task.
speaker::Speaker * get_output_speaker() const
std::vector< SourceSpeaker * > source_speakers_
StackType_t * task_stack_buffer_
static void mix_audio_samples(const int16_t *primary_buffer, audio::AudioStreamInfo primary_stream_info, const int16_t *secondary_buffer, audio::AudioStreamInfo secondary_stream_info, int16_t *output_buffer, audio::AudioStreamInfo output_stream_info, uint32_t frames_to_mix)
Mixes the primary and secondary streams taking into account the number of channels in each stream.
bool task_stack_in_psram_
TaskHandle_t task_handle_
static void copy_frames(const int16_t *input_buffer, audio::AudioStreamInfo input_stream_info, int16_t *output_buffer, audio::AudioStreamInfo output_stream_info, uint32_t frames_to_transfer)
Copies audio frames from the input buffer to the output buffer taking into account the number of chan...
static void audio_mixer_task(void *params)
EventGroupHandle_t event_group_
speaker::Speaker * output_speaker_
optional< audio::AudioStreamInfo > audio_stream_info_
uint32_t buffer_duration_ms_
std::shared_ptr< audio::AudioSourceTransferBuffer > transfer_buffer_
optional< uint32_t > timeout_ms_
float get_volume() override
void set_mute_state(bool mute_state) override
Mute state changes are passed to the parent's output speaker.
int8_t current_ducking_db_reduction_
static void duck_samples(int16_t *input_buffer, uint32_t input_samples_to_duck, int8_t *current_ducking_db_reduction, uint32_t *ducking_transition_samples_remaining, uint32_t samples_per_ducking_step, int8_t db_change_per_ducking_step)
Ducks audio samples by a specified amount.
size_t process_data_from_source(TickType_t ticks_to_wait)
Transfers audio from the ring buffer into the transfer buffer.
bool get_mute_state() override
uint32_t samples_per_ducking_step_
uint32_t pending_playback_frames_
void apply_ducking(uint8_t decibel_reduction, uint32_t duration)
Sets the ducking level for the source speaker.
std::weak_ptr< RingBuffer > ring_buffer_
int8_t target_ducking_db_reduction_
size_t play(const uint8_t *data, size_t length, TickType_t ticks_to_wait) override
uint32_t ducking_transition_samples_remaining_
void dump_config() override
int8_t db_change_per_ducking_step_
void set_volume(float volume) override
Volume state changes are passed to the parent's output speaker.
bool has_buffered_data() const override
uint32_t last_seen_data_ms_
value_type const & value() const
virtual void set_volume(float volume)
void add_audio_output_callback(std::function< void(uint32_t, int64_t)> &&callback)
Callback function for sending the duration of the audio written to the speaker since the last callbac...
virtual float get_volume()
CallbackManager< void(uint32_t, int64_t)> audio_output_callback_
void set_audio_stream_info(const audio::AudioStreamInfo &audio_stream_info)
audio::AudioStreamInfo & get_audio_stream_info()
virtual bool get_mute_state()
virtual void set_mute_state(bool mute_state)
audio::AudioStreamInfo audio_stream_info_
void scale_audio_samples(const int16_t *audio_samples, int16_t *output_buffer, int16_t scale_factor, size_t samples_to_scale)
Scales Q15 fixed point audio samples.
Providing packet encoding functions for exchanging data with a remote host.
void IRAM_ATTR HOT delay(uint32_t ms)
uint32_t IRAM_ATTR HOT millis()