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(
"Failed to start mixer: unsupported bits per sample");
86 case ESP_ERR_INVALID_ARG:
87 this->
status_set_error(
"Failed to start mixer: audio stream isn't compatible with the other audio stream.");
89 case ESP_ERR_INVALID_STATE:
104 this->stop_gracefully_) {
123 size_t bytes_written = 0;
125 std::shared_ptr<RingBuffer> temp_ring_buffer = this->
ring_buffer_.lock();
126 bytes_written = temp_ring_buffer->write_without_replacement(data,
length, ticks_to_wait);
127 if (bytes_written > 0) {
131 return bytes_written;
143 return ESP_ERR_NO_MEM;
145 std::shared_ptr<RingBuffer> temp_ring_buffer;
153 return ESP_ERR_NO_MEM;
176 return ((this->
transfer_buffer_.use_count() > 0) && this->transfer_buffer_->has_buffered_data());
201 size_t bytes_read = this->
transfer_buffer_->transfer_data_from_source(ticks_to_wait);
204 if (samples_to_duck > 0) {
205 int16_t *current_buffer =
reinterpret_cast<int16_t *
>(this->
transfer_buffer_->get_buffer_start() + current_length);
221 uint8_t total_ducking_steps = 0;
231 if ((duration > 0) && (total_ducking_steps > 0)) {
247 int8_t *current_ducking_db_reduction, uint32_t *ducking_transition_samples_remaining,
248 uint32_t samples_per_ducking_step, int8_t db_change_per_ducking_step) {
249 if (*ducking_transition_samples_remaining > 0) {
253 uint32_t ducking_steps_in_batch =
254 input_samples_to_duck / samples_per_ducking_step + (input_samples_to_duck % samples_per_ducking_step != 0);
256 for (uint32_t i = 0; i < ducking_steps_in_batch; ++i) {
257 uint32_t samples_left_in_step = *ducking_transition_samples_remaining % samples_per_ducking_step;
259 if (samples_left_in_step == 0) {
260 samples_left_in_step = samples_per_ducking_step;
263 uint32_t samples_to_duck = std::min(input_samples_to_duck, samples_left_in_step);
264 samples_to_duck = std::min(samples_to_duck, *ducking_transition_samples_remaining);
267 uint8_t safe_db_reduction_index =
268 clamp<uint8_t>(*current_ducking_db_reduction, 0, DECIBEL_REDUCTION_TABLE.size() - 1);
269 int16_t q15_scale_factor = DECIBEL_REDUCTION_TABLE[safe_db_reduction_index];
273 if (samples_left_in_step - samples_to_duck == 0) {
275 *current_ducking_db_reduction += db_change_per_ducking_step;
278 input_buffer += samples_to_duck;
279 *ducking_transition_samples_remaining -= samples_to_duck;
280 input_samples_to_duck -= samples_to_duck;
284 if ((*current_ducking_db_reduction > 0) && (input_samples_to_duck > 0)) {
287 uint8_t safe_db_reduction_index =
288 clamp<uint8_t>(*current_ducking_db_reduction, 0, DECIBEL_REDUCTION_TABLE.size() - 1);
289 int16_t q15_scale_factor = DECIBEL_REDUCTION_TABLE[safe_db_reduction_index];
298 " Number of output channels: %u",
306 ESP_LOGE(TAG,
"Failed to create event group");
313 uint32_t event_group_bits = xEventGroupGetBits(this->
event_group_);
316 ESP_LOGD(TAG,
"Starting speaker mixer");
317 xEventGroupClearBits(this->
event_group_, MixerEventGroupBits::STATE_STARTING);
321 xEventGroupClearBits(this->
event_group_, MixerEventGroupBits::ERR_ESP_NO_MEM);
324 ESP_LOGD(TAG,
"Started speaker mixer");
326 xEventGroupClearBits(this->
event_group_, MixerEventGroupBits::STATE_RUNNING);
329 ESP_LOGD(TAG,
"Stopping speaker mixer");
330 xEventGroupClearBits(this->
event_group_, MixerEventGroupBits::STATE_STOPPING);
339 bool all_stopped =
true;
342 all_stopped &= speaker->is_stopped();
355 return ESP_ERR_NOT_SUPPORTED;
364 return ESP_ERR_INVALID_ARG;
383 return ESP_ERR_NO_MEM;
392 return ESP_ERR_INVALID_STATE;
417 return ESP_ERR_INVALID_STATE;
424 uint32_t frames_to_transfer) {
425 uint8_t input_channels = input_stream_info.
get_channels();
426 uint8_t output_channels = output_stream_info.
get_channels();
427 const uint8_t max_input_channel_index = input_channels - 1;
429 if (input_channels == output_channels) {
430 size_t bytes_to_copy = input_stream_info.
frames_to_bytes(frames_to_transfer);
431 memcpy(output_buffer, input_buffer, bytes_to_copy);
436 for (uint32_t frame_index = 0; frame_index < frames_to_transfer; ++frame_index) {
437 for (uint8_t output_channel_index = 0; output_channel_index < output_channels; ++output_channel_index) {
438 uint8_t input_channel_index = std::min(output_channel_index, max_input_channel_index);
439 output_buffer[output_channels * frame_index + output_channel_index] =
440 input_buffer[input_channels * frame_index + input_channel_index];
448 uint32_t frames_to_mix) {
449 const uint8_t primary_channels = primary_stream_info.
get_channels();
450 const uint8_t secondary_channels = secondary_stream_info.
get_channels();
451 const uint8_t output_channels = output_stream_info.
get_channels();
453 const uint8_t max_primary_channel_index = primary_channels - 1;
454 const uint8_t max_secondary_channel_index = secondary_channels - 1;
456 for (uint32_t frames_index = 0; frames_index < frames_to_mix; ++frames_index) {
457 for (uint8_t output_channel_index = 0; output_channel_index < output_channels; ++output_channel_index) {
458 const uint32_t secondary_channel_index = std::min(output_channel_index, max_secondary_channel_index);
459 const int32_t secondary_sample = secondary_buffer[frames_index * secondary_channels + secondary_channel_index];
461 const uint32_t primary_channel_index = std::min(output_channel_index, max_primary_channel_index);
462 const int32_t primary_sample =
463 static_cast<int32_t
>(primary_buffer[frames_index * primary_channels + primary_channel_index]);
465 const int32_t added_sample = secondary_sample + primary_sample;
467 output_buffer[frames_index * output_channels + output_channel_index] =
468 static_cast<int16_t
>(clamp<int32_t>(added_sample, MIN_AUDIO_SAMPLE_VALUE, MAX_AUDIO_SAMPLE_VALUE));
483 if (output_transfer_buffer ==
nullptr) {
488 vTaskDelete(
nullptr);
495 bool sent_finished =
false;
498 uint32_t event_group_bits = xEventGroupGetBits(this_mixer->
event_group_);
504 output_transfer_buffer->transfer_data_to_sink(pdMS_TO_TICKS(TASK_DELAY_MS),
false);
506 const uint32_t output_frames_free =
507 this_mixer->
audio_stream_info_.value().bytes_to_frames(output_transfer_buffer->free());
509 std::vector<SourceSpeaker *> speakers_with_data;
510 std::vector<std::shared_ptr<audio::AudioSourceTransferBuffer>> transfer_buffers_with_data;
513 if (speaker->get_transfer_buffer().use_count() > 0) {
514 std::shared_ptr<audio::AudioSourceTransferBuffer> transfer_buffer = speaker->get_transfer_buffer().lock();
515 speaker->process_data_from_source(0);
517 if ((transfer_buffer->available() > 0) && !speaker->get_pause_state()) {
519 transfer_buffers_with_data.push_back(transfer_buffer);
520 speakers_with_data.push_back(speaker);
525 if (transfer_buffers_with_data.empty()) {
527 delay(TASK_DELAY_MS);
531 uint32_t frames_to_mix = output_frames_free;
533 if ((transfer_buffers_with_data.size() == 1) || this_mixer->
queue_mode_) {
542 const uint32_t frames_available_in_buffer =
543 active_stream_info.
bytes_to_frames(transfer_buffers_with_data[0]->available());
544 frames_to_mix = std::min(frames_to_mix, frames_available_in_buffer);
545 copy_frames(
reinterpret_cast<int16_t *
>(transfer_buffers_with_data[0]->get_buffer_start()), active_stream_info,
546 reinterpret_cast<int16_t *
>(output_transfer_buffer->get_buffer_end()),
550 transfer_buffers_with_data[0]->decrease_buffer_length(active_stream_info.
frames_to_bytes(frames_to_mix));
551 speakers_with_data[0]->pending_playback_frames_ += frames_to_mix;
554 output_transfer_buffer->increase_buffer_length(
559 if (!sent_finished) {
561 sent_finished =
true;
570 sent_finished =
false;
575 for (
int i = 0; i < transfer_buffers_with_data.size(); ++i) {
576 const uint32_t frames_available_in_buffer =
577 speakers_with_data[i]->get_audio_stream_info().bytes_to_frames(transfer_buffers_with_data[i]->available());
578 frames_to_mix = std::min(frames_to_mix, frames_available_in_buffer);
580 int16_t *primary_buffer =
reinterpret_cast<int16_t *
>(transfer_buffers_with_data[0]->get_buffer_start());
584 for (
int i = 1; i < transfer_buffers_with_data.size(); ++i) {
586 reinterpret_cast<int16_t *
>(transfer_buffers_with_data[i]->get_buffer_start()),
587 speakers_with_data[i]->get_audio_stream_info(),
588 reinterpret_cast<int16_t *
>(output_transfer_buffer->get_buffer_end()),
591 if (i != transfer_buffers_with_data.size() - 1) {
593 primary_buffer =
reinterpret_cast<int16_t *
>(output_transfer_buffer->get_buffer_end());
599 for (
int i = 0; i < transfer_buffers_with_data.size(); ++i) {
600 transfer_buffers_with_data[i]->decrease_buffer_length(
601 speakers_with_data[i]->get_audio_stream_info().frames_to_bytes(frames_to_mix));
602 speakers_with_data[i]->pending_playback_frames_ += frames_to_mix;
606 output_transfer_buffer->increase_buffer_length(
613 output_transfer_buffer.reset();
617 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()