mirror of https://github.com/esphome/esphome.git
128 lines
3.6 KiB
Python
128 lines
3.6 KiB
Python
import os
|
|
|
|
import esphome.config_validation as cv
|
|
import esphome.codegen as cg
|
|
import esphome.final_validate as fv
|
|
|
|
from esphome.components import esp32
|
|
from esphome.components.i2c import i2c_ns, I2CBus
|
|
|
|
|
|
from esphome.const import CONF_ID, CONF_BOARD
|
|
|
|
CODEOWNERS = ["@jesserockz"]
|
|
DEPENDENCIES = ["esp32"]
|
|
|
|
CONF_ESP_ADF_ID = "esp_adf_id"
|
|
CONF_ESP_ADF = "esp_adf"
|
|
|
|
esp_adf_ns = cg.esphome_ns.namespace("esp_adf")
|
|
ESPADF = esp_adf_ns.class_("ESPADF", cg.Component)
|
|
ESPADFPipeline = esp_adf_ns.class_("ESPADFPipeline", cg.Parented.template(ESPADF))
|
|
|
|
SUPPORTED_BOARDS = {
|
|
"esp32s3box": "CONFIG_ESP32_S3_BOX_BOARD",
|
|
"esp32s3boxlite": "CONFIG_ESP32_S3_BOX_LITE_BOARD",
|
|
"esp32s3box3": "CONFIG_ESP32_S3_BOX_3_BOARD",
|
|
"esp32s3korvo1": "CONFIG_ESP32_S3_KORVO1_BOARD",
|
|
}
|
|
|
|
|
|
def _default_board(config):
|
|
config = config.copy()
|
|
if board := config.get(CONF_BOARD) is None:
|
|
board = esp32.get_board()
|
|
if board in SUPPORTED_BOARDS:
|
|
config[CONF_BOARD] = board
|
|
return config
|
|
|
|
|
|
def final_validate_usable_board(platform: str):
|
|
def _validate(adf_config):
|
|
board = adf_config.get(CONF_BOARD)
|
|
if board not in SUPPORTED_BOARDS:
|
|
raise cv.Invalid(f"Board {board} is not supported by esp-adf {platform}")
|
|
return adf_config
|
|
|
|
return cv.Schema(
|
|
{cv.Required(CONF_ESP_ADF_ID): fv.id_declaration_match_schema(_validate)},
|
|
extra=cv.ALLOW_EXTRA,
|
|
)
|
|
|
|
|
|
CONFIG_SCHEMA = cv.All(
|
|
cv.Schema(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(ESPADF),
|
|
cv.Optional(CONF_BOARD): cv.string_strict,
|
|
}
|
|
),
|
|
_default_board,
|
|
cv.only_with_esp_idf,
|
|
)
|
|
|
|
|
|
async def to_code(config):
|
|
var = cg.new_Pvariable(config[CONF_ID])
|
|
await cg.register_component(var, config)
|
|
|
|
cg.add_define("USE_ESP_ADF")
|
|
|
|
cg.add_platformio_option("build_unflags", "-Wl,--end-group")
|
|
|
|
esp32.add_idf_component(
|
|
name="esp-adf",
|
|
repo="https://github.com/X-Ryl669/esp-adf",
|
|
path="components",
|
|
ref="with-i2c-cb",
|
|
components=["*"],
|
|
submodules=["components/esp-sr", "components/esp-adf-libs"],
|
|
)
|
|
|
|
cg.add_platformio_option(
|
|
"board_build.embed_txtfiles", "components/dueros_service/duer_profile"
|
|
)
|
|
# This is required for esp-dsp that have files with the same basename and different extension (.S,.c)
|
|
cg.add_platformio_option(
|
|
"board_build.esp-idf.preserve_source_file_extension", "yes"
|
|
)
|
|
|
|
if board := config.get(CONF_BOARD):
|
|
cg.add_define("USE_ESP_ADF_BOARD")
|
|
|
|
esp32.add_idf_sdkconfig_option(SUPPORTED_BOARDS[board], True)
|
|
# esp32.add_idf_sdkconfig_option("CONFIG_SPIRAM_BOOT_INIT", True)
|
|
# esp32.add_idf_sdkconfig_option("CONFIG_SPIRAM_ALLOW_STACK_EXTERNAL_MEMORY", True)
|
|
|
|
esp32.add_extra_script(
|
|
"pre",
|
|
"apply_adf_patches.py",
|
|
os.path.join(os.path.dirname(__file__), "apply_adf_patches.py.script"),
|
|
)
|
|
esp32.add_extra_build_file(
|
|
"esp_adf_patches/idf_v4.4_freertos.patch",
|
|
"https://github.com/X-Ryl669/esp-adf/raw/with-i2c-cb/idf_patches/idf_v4.4_freertos.patch",
|
|
)
|
|
|
|
|
|
# I2C Bus below
|
|
ADFI2CBus = i2c_ns.class_("ADFI2CBus", I2CBus, cg.Component)
|
|
|
|
|
|
# Patch the I2C config schema to use the right I2C bus
|
|
def _patch_idfi2cbus(config):
|
|
from esphome.cpp_generator import MockObjClass
|
|
|
|
if "i2c" in fv.full_config.get():
|
|
for i2c_inst in fv.full_config.get()["i2c"]:
|
|
i2c_inst["id"].type = MockObjClass(
|
|
"i2c::ADFI2CBus",
|
|
# pylint: disable=protected-access
|
|
parents=i2c_inst["id"].type._parents,
|
|
)
|
|
|
|
return config
|
|
|
|
|
|
FINAL_VALIDATE_SCHEMA = _patch_idfi2cbus
|