mirror of
https://github.com/esphome/esphome.git
synced 2026-08-22 22:26:21 +00:00
281 lines
9.2 KiB
Python
281 lines
9.2 KiB
Python
from dataclasses import dataclass
|
|
|
|
from esphome import automation
|
|
import esphome.codegen as cg
|
|
from esphome.components import esp32, network, psram, socket, wifi
|
|
import esphome.config_validation as cv
|
|
from esphome.const import (
|
|
CONF_BUFFER_SIZE,
|
|
CONF_ID,
|
|
CONF_SAMPLE_RATE,
|
|
CONF_TASK_STACK_IN_PSRAM,
|
|
)
|
|
from esphome.core import CORE, ID
|
|
from esphome.cpp_generator import TemplateArgsType
|
|
from esphome.types import ConfigType
|
|
|
|
# mdns for autodiscovery
|
|
AUTO_LOAD = ["mdns"]
|
|
CODEOWNERS = ["@kahrendt"]
|
|
DEPENDENCIES = ["network"]
|
|
DOMAIN = "sendspin"
|
|
|
|
CONF_SENDSPIN_ID = "sendspin_id"
|
|
|
|
CONF_INITIAL_STATIC_DELAY = "initial_static_delay"
|
|
CONF_FIXED_DELAY = "fixed_delay"
|
|
CONF_DECODE_MEMORY = "decode_memory"
|
|
|
|
# sendspin-cpp library lives in the global `sendspin` namespace.
|
|
sendspin_library_ns = cg.global_ns.namespace("sendspin")
|
|
|
|
# Library Enums
|
|
SendspinCodecFormat = sendspin_library_ns.enum("SendspinCodecFormat", is_class=True)
|
|
CODEC_FORMAT_FLAC = SendspinCodecFormat.enum("FLAC")
|
|
CODEC_FORMAT_OPUS = SendspinCodecFormat.enum("OPUS")
|
|
CODEC_FORMAT_PCM = SendspinCodecFormat.enum("PCM")
|
|
CODEC_FORMAT_UNSUPPORTED = SendspinCodecFormat.enum("UNSUPPORTED")
|
|
|
|
# Library Structs
|
|
AudioSupportedFormatObject = sendspin_library_ns.struct("AudioSupportedFormatObject")
|
|
PlayerRoleConfig = sendspin_library_ns.struct("PlayerRoleConfig")
|
|
|
|
# MemoryLocation enum (from sendspin/types.h) controls SPIRAM-vs-internal-RAM placement
|
|
# preference for the player role's transfer buffers.
|
|
SendspinMemoryLocation = sendspin_library_ns.enum("MemoryLocation", is_class=True)
|
|
|
|
MEMORY_PSRAM = "psram"
|
|
MEMORY_INTERNAL = "internal"
|
|
MEMORY_LOCATIONS = [MEMORY_PSRAM, MEMORY_INTERNAL]
|
|
MEMORY_LOCATION_ENUM = {
|
|
MEMORY_PSRAM: SendspinMemoryLocation.PREFER_EXTERNAL,
|
|
MEMORY_INTERNAL: SendspinMemoryLocation.PREFER_INTERNAL,
|
|
}
|
|
|
|
# Trailing underscore avoids clashing with sendspin-cpp's global `sendspin` namespace.
|
|
# Analysis tools strip the trailing underscore (same pattern as `template_`).
|
|
sendspin_ns = cg.esphome_ns.namespace("sendspin_")
|
|
SendspinHub = sendspin_ns.class_(
|
|
"SendspinHub",
|
|
cg.Component,
|
|
)
|
|
|
|
|
|
SendspinSwitchCommandAction = sendspin_ns.class_(
|
|
"SendspinSwitchCommandAction",
|
|
automation.Action,
|
|
cg.Parented.template(SendspinHub),
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class SendspinConfiguration:
|
|
artwork_support: bool = False
|
|
controller_support: bool = False
|
|
metadata_support: bool = False
|
|
player_support: bool = False
|
|
visualizer_support: bool = False
|
|
|
|
player_config: ConfigType | None = None
|
|
|
|
|
|
def _get_data() -> SendspinConfiguration:
|
|
if DOMAIN not in CORE.data:
|
|
CORE.data[DOMAIN] = SendspinConfiguration()
|
|
return CORE.data[DOMAIN]
|
|
|
|
|
|
def request_artwork_support() -> None:
|
|
"""Request artwork role support for Sendspin."""
|
|
_get_data().artwork_support = True
|
|
|
|
|
|
def request_controller_support() -> None:
|
|
"""Request controller role support for Sendspin."""
|
|
_get_data().controller_support = True
|
|
|
|
|
|
def request_metadata_support() -> None:
|
|
"""Request metadata role support for Sendspin."""
|
|
_get_data().metadata_support = True
|
|
|
|
|
|
def request_player_support() -> None:
|
|
"""Request player role support for Sendspin."""
|
|
_get_data().player_support = True
|
|
|
|
|
|
def request_visualizer_support() -> None:
|
|
"""Request visualizer role support for Sendspin."""
|
|
_get_data().visualizer_support = True
|
|
|
|
|
|
def register_player_config(config: ConfigType) -> None:
|
|
"""Register the player role config from the media source subcomponent."""
|
|
data = _get_data()
|
|
request_player_support()
|
|
if data.player_config is not None:
|
|
raise cv.Invalid(
|
|
"Only one sendspin media_source player configuration is supported"
|
|
)
|
|
data.player_config = config
|
|
|
|
|
|
def _request_high_performance_networking(config: ConfigType) -> ConfigType:
|
|
"""Request high performance networking for Sendspin streaming.
|
|
|
|
Also enables wake_loop_threadsafe support for fast defer() callbacks
|
|
from background threads (WebSocket handler, image decoder).
|
|
"""
|
|
network.require_high_performance_networking()
|
|
# Socket consumption varies by mode:
|
|
# - Server mode: 1 listening socket + 4 client connections (established connection, unproven connections, and a spare)
|
|
# - Client mode: 1 outbound connection
|
|
socket.consume_sockets(
|
|
1, "sendspin_websocket_server", socket.SocketType.TCP_LISTEN
|
|
)(config)
|
|
socket.consume_sockets(4, "sendspin_websocket_server")(config)
|
|
socket.consume_sockets(1, "sendspin_websocket_client")(config)
|
|
|
|
wifi.enable_runtime_power_save_control()
|
|
wifi.enable_runtime_roaming_suppression()
|
|
return config
|
|
|
|
|
|
CONFIG_SCHEMA = cv.All(
|
|
cv.Schema(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(SendspinHub),
|
|
cv.Optional(CONF_TASK_STACK_IN_PSRAM): psram.validate_task_stack_in_psram,
|
|
}
|
|
),
|
|
cv.only_on_esp32,
|
|
_request_high_performance_networking,
|
|
)
|
|
|
|
|
|
def _request_controller_role(config: ConfigType) -> ConfigType:
|
|
"""Request the controller role for the sendspin.switch action."""
|
|
request_controller_support()
|
|
return config
|
|
|
|
|
|
SENDSPIN_SIMPLE_ACTION_SCHEMA = cv.All(
|
|
automation.maybe_simple_id(
|
|
cv.Schema(
|
|
{
|
|
cv.GenerateID(): cv.use_id(SendspinHub),
|
|
}
|
|
)
|
|
),
|
|
_request_controller_role,
|
|
)
|
|
|
|
|
|
@automation.register_action(
|
|
"sendspin.switch",
|
|
SendspinSwitchCommandAction,
|
|
SENDSPIN_SIMPLE_ACTION_SCHEMA,
|
|
synchronous=True,
|
|
)
|
|
async def sendspin_switch_to_code(
|
|
config: ConfigType,
|
|
action_id: ID,
|
|
template_arg: cg.TemplateArguments,
|
|
args: TemplateArgsType,
|
|
):
|
|
var = cg.new_Pvariable(action_id, template_arg)
|
|
await cg.register_parented(var, config[CONF_ID])
|
|
return var
|
|
|
|
|
|
async def to_code(config: ConfigType) -> None:
|
|
var = cg.new_Pvariable(config[CONF_ID])
|
|
await cg.register_component(var, config)
|
|
|
|
if config.get(CONF_TASK_STACK_IN_PSRAM):
|
|
cg.add(var.set_task_stack_in_psram(True))
|
|
psram.request_external_task_stack()
|
|
|
|
# sendspin-cpp library
|
|
esp32.add_idf_component(name="sendspin/sendspin-cpp", ref="0.7.0")
|
|
|
|
cg.add_define("USE_SENDSPIN", True) # for MDNS
|
|
|
|
data = _get_data()
|
|
|
|
# The color role is not yet wired up in ESPHome; disable it in the library for now.
|
|
esp32.add_idf_sdkconfig_option("CONFIG_SENDSPIN_ENABLE_COLOR", False)
|
|
|
|
# Configure Sendspin roles based on requested features (ESPHome internally via USE_SENDSPIN_*)
|
|
# and disable building unused code paths in the sendspin-cpp library (IDF SDKConfig via CONFIG_SENDSPIN_ENABLE_*).
|
|
if data.artwork_support:
|
|
cg.add_define("USE_SENDSPIN_ARTWORK", True)
|
|
else:
|
|
esp32.add_idf_sdkconfig_option("CONFIG_SENDSPIN_ENABLE_ARTWORK", False)
|
|
|
|
if data.controller_support:
|
|
cg.add_define("USE_SENDSPIN_CONTROLLER", True)
|
|
else:
|
|
esp32.add_idf_sdkconfig_option("CONFIG_SENDSPIN_ENABLE_CONTROLLER", False)
|
|
|
|
if data.metadata_support:
|
|
cg.add_define("USE_SENDSPIN_METADATA", True)
|
|
else:
|
|
esp32.add_idf_sdkconfig_option("CONFIG_SENDSPIN_ENABLE_METADATA", False)
|
|
|
|
if data.player_support:
|
|
cg.add_define("USE_SENDSPIN_PLAYER", True)
|
|
|
|
# Configures the player role. We always assume support for 16 bits per sample mono and stereo FLAC, Opus, and PCM at the configured sample rate
|
|
# (with Opus only supported at 48 kHz since that's the only sample rate it supports). Users can configure the specific formats via the Sendspin server
|
|
player_cfg = data.player_config
|
|
sample_rate = player_cfg[CONF_SAMPLE_RATE]
|
|
|
|
# OPUS only supports 48 kHz audio
|
|
codecs = [CODEC_FORMAT_FLAC]
|
|
if sample_rate == 48000:
|
|
codecs.append(CODEC_FORMAT_OPUS)
|
|
codecs.append(CODEC_FORMAT_PCM)
|
|
|
|
def _audio_format(codec, channels):
|
|
return cg.StructInitializer(
|
|
AudioSupportedFormatObject,
|
|
("codec", codec),
|
|
("channels", channels),
|
|
("sample_rate", sample_rate),
|
|
("bit_depth", 16),
|
|
)
|
|
|
|
audio_format_structs = [
|
|
_audio_format(codec, channels) for codec in codecs for channels in (2, 1)
|
|
]
|
|
|
|
psram_stack = player_cfg.get(CONF_TASK_STACK_IN_PSRAM, False)
|
|
if psram_stack:
|
|
psram.request_external_task_stack()
|
|
|
|
player_struct_fields = [
|
|
("audio_formats", audio_format_structs),
|
|
("audio_buffer_capacity", player_cfg[CONF_BUFFER_SIZE]),
|
|
("fixed_delay_us", player_cfg[CONF_FIXED_DELAY]),
|
|
("initial_static_delay_ms", player_cfg[CONF_INITIAL_STATIC_DELAY]),
|
|
("psram_stack", psram_stack),
|
|
]
|
|
if (decode_memory := player_cfg.get(CONF_DECODE_MEMORY)) is not None:
|
|
player_struct_fields.append(
|
|
("decode_buffer_location", MEMORY_LOCATION_ENUM[decode_memory])
|
|
)
|
|
player_config_struct = cg.StructInitializer(
|
|
PlayerRoleConfig,
|
|
*player_struct_fields,
|
|
)
|
|
cg.add(var.set_player_config(player_config_struct))
|
|
else:
|
|
esp32.add_idf_sdkconfig_option("CONFIG_SENDSPIN_ENABLE_PLAYER", False)
|
|
|
|
if data.visualizer_support:
|
|
cg.add_define("USE_SENDSPIN_VISUALIZER", True)
|
|
else:
|
|
esp32.add_idf_sdkconfig_option("CONFIG_SENDSPIN_ENABLE_VISUALIZER", False)
|