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};
46 ESP_LOGCONFIG(TAG,
"Mixer Source Speaker");
51 ESP_LOGCONFIG(TAG,
" Timeout: never");
61 if (speakers_playback_frames > 0) {
70 esp_err_t err = this->
start_();
81 case ESP_ERR_NOT_SUPPORTED:
82 this->
status_set_error(
"Failed to start mixer: unsupported bits per sample");
84 case ESP_ERR_INVALID_ARG:
85 this->
status_set_error(
"Failed to start mixer: audio stream isn't compatible with the other audio stream.");
87 case ESP_ERR_INVALID_STATE:
102 this->stop_gracefully_) {
121 size_t bytes_written = 0;
123 std::shared_ptr<RingBuffer> temp_ring_buffer = this->
ring_buffer_.lock();
124 bytes_written = temp_ring_buffer->write_without_replacement(data,
length, ticks_to_wait);
125 if (bytes_written > 0) {
129 return bytes_written;
141 return ESP_ERR_NO_MEM;
143 std::shared_ptr<RingBuffer> temp_ring_buffer;
151 return ESP_ERR_NO_MEM;
174 return ((this->
transfer_buffer_.use_count() > 0) && this->transfer_buffer_->has_buffered_data());
199 size_t bytes_read = this->
transfer_buffer_->transfer_data_from_source(ticks_to_wait);
202 if (samples_to_duck > 0) {
203 int16_t *current_buffer =
reinterpret_cast<int16_t *
>(this->
transfer_buffer_->get_buffer_start() + current_length);
219 uint8_t total_ducking_steps = 0;
229 if ((duration > 0) && (total_ducking_steps > 0)) {
245 int8_t *current_ducking_db_reduction, uint32_t *ducking_transition_samples_remaining,
246 uint32_t samples_per_ducking_step, int8_t db_change_per_ducking_step) {
247 if (*ducking_transition_samples_remaining > 0) {
251 uint32_t ducking_steps_in_batch =
252 input_samples_to_duck / samples_per_ducking_step + (input_samples_to_duck % samples_per_ducking_step != 0);
254 for (uint32_t i = 0; i < ducking_steps_in_batch; ++i) {
255 uint32_t samples_left_in_step = *ducking_transition_samples_remaining % samples_per_ducking_step;
257 if (samples_left_in_step == 0) {
258 samples_left_in_step = samples_per_ducking_step;
261 uint32_t samples_to_duck = std::min(input_samples_to_duck, samples_left_in_step);
262 samples_to_duck = std::min(samples_to_duck, *ducking_transition_samples_remaining);
265 uint8_t safe_db_reduction_index =
266 clamp<uint8_t>(*current_ducking_db_reduction, 0, DECIBEL_REDUCTION_TABLE.size() - 1);
267 int16_t q15_scale_factor = DECIBEL_REDUCTION_TABLE[safe_db_reduction_index];
271 if (samples_left_in_step - samples_to_duck == 0) {
273 *current_ducking_db_reduction += db_change_per_ducking_step;
276 input_buffer += samples_to_duck;
277 *ducking_transition_samples_remaining -= samples_to_duck;
278 input_samples_to_duck -= samples_to_duck;
282 if ((*current_ducking_db_reduction > 0) && (input_samples_to_duck > 0)) {
285 uint8_t safe_db_reduction_index =
286 clamp<uint8_t>(*current_ducking_db_reduction, 0, DECIBEL_REDUCTION_TABLE.size() - 1);
287 int16_t q15_scale_factor = DECIBEL_REDUCTION_TABLE[safe_db_reduction_index];
294 ESP_LOGCONFIG(TAG,
"Speaker Mixer:");
295 ESP_LOGCONFIG(TAG,
" Number of output channels: %u", this->
output_channels_);
302 ESP_LOGE(TAG,
"Failed to create event group");
309 uint32_t event_group_bits = xEventGroupGetBits(this->
event_group_);
312 ESP_LOGD(TAG,
"Starting speaker mixer");
313 xEventGroupClearBits(this->
event_group_, MixerEventGroupBits::STATE_STARTING);
317 xEventGroupClearBits(this->
event_group_, MixerEventGroupBits::ERR_ESP_NO_MEM);
320 ESP_LOGD(TAG,
"Started speaker mixer");
322 xEventGroupClearBits(this->
event_group_, MixerEventGroupBits::STATE_RUNNING);
325 ESP_LOGD(TAG,
"Stopping speaker mixer");
326 xEventGroupClearBits(this->
event_group_, MixerEventGroupBits::STATE_STOPPING);
335 bool all_stopped =
true;
338 all_stopped &= speaker->is_stopped();
351 return ESP_ERR_NOT_SUPPORTED;
360 return ESP_ERR_INVALID_ARG;
379 return ESP_ERR_NO_MEM;
388 return ESP_ERR_INVALID_STATE;
413 return ESP_ERR_INVALID_STATE;
420 uint32_t frames_to_transfer) {
421 uint8_t input_channels = input_stream_info.
get_channels();
422 uint8_t output_channels = output_stream_info.
get_channels();
423 const uint8_t max_input_channel_index = input_channels - 1;
425 if (input_channels == output_channels) {
426 size_t bytes_to_copy = input_stream_info.
frames_to_bytes(frames_to_transfer);
427 memcpy(output_buffer, input_buffer, bytes_to_copy);
432 for (uint32_t frame_index = 0; frame_index < frames_to_transfer; ++frame_index) {
433 for (uint8_t output_channel_index = 0; output_channel_index < output_channels; ++output_channel_index) {
434 uint8_t input_channel_index = std::min(output_channel_index, max_input_channel_index);
435 output_buffer[output_channels * frame_index + output_channel_index] =
436 input_buffer[input_channels * frame_index + input_channel_index];
444 uint32_t frames_to_mix) {
445 const uint8_t primary_channels = primary_stream_info.
get_channels();
446 const uint8_t secondary_channels = secondary_stream_info.
get_channels();
447 const uint8_t output_channels = output_stream_info.
get_channels();
449 const uint8_t max_primary_channel_index = primary_channels - 1;
450 const uint8_t max_secondary_channel_index = secondary_channels - 1;
452 for (uint32_t frames_index = 0; frames_index < frames_to_mix; ++frames_index) {
453 for (uint8_t output_channel_index = 0; output_channel_index < output_channels; ++output_channel_index) {
454 const uint32_t secondary_channel_index = std::min(output_channel_index, max_secondary_channel_index);
455 const int32_t secondary_sample = secondary_buffer[frames_index * secondary_channels + secondary_channel_index];
457 const uint32_t primary_channel_index = std::min(output_channel_index, max_primary_channel_index);
458 const int32_t primary_sample =
459 static_cast<int32_t
>(primary_buffer[frames_index * primary_channels + primary_channel_index]);
461 const int32_t added_sample = secondary_sample + primary_sample;
463 output_buffer[frames_index * output_channels + output_channel_index] =
464 static_cast<int16_t
>(
clamp<int32_t>(added_sample, MIN_AUDIO_SAMPLE_VALUE, MAX_AUDIO_SAMPLE_VALUE));
479 if (output_transfer_buffer ==
nullptr) {
484 vTaskDelete(
nullptr);
491 bool sent_finished =
false;
494 uint32_t event_group_bits = xEventGroupGetBits(this_mixer->
event_group_);
500 output_transfer_buffer->transfer_data_to_sink(pdMS_TO_TICKS(TASK_DELAY_MS),
false);
502 const uint32_t output_frames_free =
503 this_mixer->
audio_stream_info_.value().bytes_to_frames(output_transfer_buffer->free());
505 std::vector<SourceSpeaker *> speakers_with_data;
506 std::vector<std::shared_ptr<audio::AudioSourceTransferBuffer>> transfer_buffers_with_data;
509 if (speaker->get_transfer_buffer().use_count() > 0) {
510 std::shared_ptr<audio::AudioSourceTransferBuffer> transfer_buffer = speaker->get_transfer_buffer().lock();
511 speaker->process_data_from_source(0);
513 if ((transfer_buffer->available() > 0) && !speaker->get_pause_state()) {
515 transfer_buffers_with_data.push_back(transfer_buffer);
516 speakers_with_data.push_back(speaker);
521 if (transfer_buffers_with_data.empty()) {
523 delay(TASK_DELAY_MS);
527 uint32_t frames_to_mix = output_frames_free;
529 if ((transfer_buffers_with_data.size() == 1) || this_mixer->
queue_mode_) {
538 const uint32_t frames_available_in_buffer =
539 active_stream_info.
bytes_to_frames(transfer_buffers_with_data[0]->available());
540 frames_to_mix = std::min(frames_to_mix, frames_available_in_buffer);
541 copy_frames(
reinterpret_cast<int16_t *
>(transfer_buffers_with_data[0]->get_buffer_start()), active_stream_info,
542 reinterpret_cast<int16_t *
>(output_transfer_buffer->get_buffer_end()),
546 transfer_buffers_with_data[0]->decrease_buffer_length(active_stream_info.
frames_to_bytes(frames_to_mix));
547 speakers_with_data[0]->pending_playback_frames_ += frames_to_mix;
550 output_transfer_buffer->increase_buffer_length(
555 if (!sent_finished) {
557 sent_finished =
true;
566 sent_finished =
false;
571 for (
int i = 0; i < transfer_buffers_with_data.size(); ++i) {
572 const uint32_t frames_available_in_buffer =
573 speakers_with_data[i]->get_audio_stream_info().bytes_to_frames(transfer_buffers_with_data[i]->available());
574 frames_to_mix = std::min(frames_to_mix, frames_available_in_buffer);
576 int16_t *primary_buffer =
reinterpret_cast<int16_t *
>(transfer_buffers_with_data[0]->get_buffer_start());
580 for (
int i = 1; i < transfer_buffers_with_data.size(); ++i) {
582 reinterpret_cast<int16_t *
>(transfer_buffers_with_data[i]->get_buffer_start()),
583 speakers_with_data[i]->get_audio_stream_info(),
584 reinterpret_cast<int16_t *
>(output_transfer_buffer->get_buffer_end()),
587 if (i != transfer_buffers_with_data.size() - 1) {
589 primary_buffer =
reinterpret_cast<int16_t *
>(output_transfer_buffer->get_buffer_end());
595 for (
int i = 0; i < transfer_buffers_with_data.size(); ++i) {
596 transfer_buffers_with_data[i]->decrease_buffer_length(
597 speakers_with_data[i]->get_audio_stream_info().frames_to_bytes(frames_to_mix));
598 speakers_with_data[i]->pending_playback_frames_ += frames_to_mix;
602 output_transfer_buffer->increase_buffer_length(
609 output_transfer_buffer.reset();
613 vTaskDelete(
nullptr);
virtual void mark_failed()
Mark this component as failed.
void status_clear_error()
void status_set_error(const char *message="unspecified")
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()
constexpr const T & clamp(const T &v, const T &lo, const T &hi, Compare comp)