Kevin AhrendtandCopilot
d759f1a567
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
60 lines
2.2 KiB
C++
60 lines
2.2 KiB
C++
#pragma once
|
|
|
|
#include "esphome/core/defines.h"
|
|
|
|
#ifdef USE_ESP32
|
|
|
|
#include "esphome/components/audio/audio.h"
|
|
#include "esphome/components/media_source/media_source.h"
|
|
#include "esphome/core/component.h"
|
|
|
|
#include <micro_decoder/decoder_source.h>
|
|
#include <micro_decoder/types.h>
|
|
|
|
#include <atomic>
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
namespace esphome::audio_http {
|
|
|
|
// Inherits from two unrelated listener-style interfaces:
|
|
// - media_source::MediaSource: this source reports state and writes audio *to* an orchestrator
|
|
// (the orchestrator calls set_listener() on us with a MediaSourceListener*).
|
|
// - micro_decoder::DecoderListener: the underlying decoder calls back *into* us with decoded
|
|
// audio and state changes (we call decoder_->set_listener(this) in setup()).
|
|
// The two set_listener() methods live on different base classes and serve opposite directions.
|
|
class AudioHTTPMediaSource : public Component, public media_source::MediaSource, public micro_decoder::DecoderListener {
|
|
public:
|
|
void setup() override;
|
|
void loop() override;
|
|
void dump_config() override;
|
|
|
|
void set_buffer_size(size_t buffer_size) { this->buffer_size_ = buffer_size; }
|
|
void set_task_stack_in_psram(bool task_stack_in_psram) { this->decoder_task_stack_in_psram_ = task_stack_in_psram; }
|
|
|
|
// MediaSource interface implementation
|
|
bool play_uri(const std::string &uri) override;
|
|
void handle_command(media_source::MediaSourceCommand command) override;
|
|
bool can_handle(const std::string &uri) const override;
|
|
|
|
// DecoderListener interface implementation
|
|
size_t on_audio_write(const uint8_t *data, size_t length, uint32_t timeout_ms) override;
|
|
void on_stream_info(const micro_decoder::AudioStreamInfo &info) override;
|
|
void on_state_change(micro_decoder::DecoderState state) override;
|
|
|
|
protected:
|
|
std::unique_ptr<micro_decoder::DecoderSource> decoder_;
|
|
audio::AudioStreamInfo stream_info_;
|
|
|
|
size_t buffer_size_{50000};
|
|
|
|
// Written from the main loop in handle_command(), read from the decoder task in
|
|
// on_audio_write(). Must be atomic to avoid a data race.
|
|
std::atomic<bool> pause_{false};
|
|
bool decoder_task_stack_in_psram_{false};
|
|
};
|
|
|
|
} // namespace esphome::audio_http
|
|
|
|
#endif // USE_ESP32
|