55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
from typing import Any
|
|
|
|
import esphome.codegen as cg
|
|
from esphome.components import audio, esp32, media_source, psram
|
|
import esphome.config_validation as cv
|
|
from esphome.const import CONF_ID, CONF_TASK_STACK_IN_PSRAM
|
|
from esphome.types import ConfigType
|
|
|
|
CODEOWNERS = ["@kahrendt"]
|
|
AUTO_LOAD = ["audio"]
|
|
DEPENDENCIES = ["audio_file"]
|
|
|
|
audio_file_ns = cg.esphome_ns.namespace("audio_file")
|
|
AudioFileMediaSource = audio_file_ns.class_(
|
|
"AudioFileMediaSource", cg.Component, media_source.MediaSource
|
|
)
|
|
|
|
|
|
def _request_micro_decoder(config: ConfigType) -> ConfigType:
|
|
audio.request_micro_decoder_support()
|
|
return config
|
|
|
|
|
|
def _validate_task_stack_in_psram(value: Any) -> bool:
|
|
if value := cv.boolean(value):
|
|
return cv.requires_component(psram.DOMAIN)(value)
|
|
return value
|
|
|
|
|
|
CONFIG_SCHEMA = cv.All(
|
|
media_source.media_source_schema(
|
|
AudioFileMediaSource,
|
|
)
|
|
.extend(
|
|
{
|
|
cv.Optional(CONF_TASK_STACK_IN_PSRAM): _validate_task_stack_in_psram,
|
|
}
|
|
)
|
|
.extend(cv.COMPONENT_SCHEMA),
|
|
cv.only_on_esp32,
|
|
_request_micro_decoder,
|
|
)
|
|
|
|
|
|
async def to_code(config: ConfigType) -> None:
|
|
var = cg.new_Pvariable(config[CONF_ID])
|
|
await cg.register_component(var, config)
|
|
await media_source.register_media_source(var, config)
|
|
|
|
if config.get(CONF_TASK_STACK_IN_PSRAM):
|
|
cg.add(var.set_task_stack_in_psram(True))
|
|
esp32.add_idf_sdkconfig_option(
|
|
"CONFIG_SPIRAM_ALLOW_STACK_EXTERNAL_MEMORY", True
|
|
)
|