mirror of
https://github.com/DarkFlippers/unleashed-firmware
synced 2024-11-30 08:20:21 +00:00
107 lines
3.6 KiB
Python
107 lines
3.6 KiB
Python
import json
|
|
import os
|
|
import sys
|
|
from functools import reduce
|
|
|
|
from SCons.Errors import StopError
|
|
|
|
|
|
def _load_sdk_data(sdk_root):
|
|
split_vars = {
|
|
"cc_args",
|
|
"cpp_args",
|
|
"linker_args",
|
|
"linker_libs",
|
|
}
|
|
subst_vars = split_vars | {
|
|
"sdk_symbols",
|
|
}
|
|
sdk_data = {}
|
|
with open(os.path.join(sdk_root, "sdk.opts")) as f:
|
|
sdk_json_data = json.load(f)
|
|
replacements = {
|
|
sdk_json_data["app_ep_subst"]: "${APP_ENTRY}",
|
|
sdk_json_data["sdk_path_subst"]: sdk_root.replace("\\", "/"),
|
|
sdk_json_data["map_file_subst"]: "${TARGET}",
|
|
}
|
|
|
|
def do_value_substs(src_value):
|
|
if isinstance(src_value, str):
|
|
return reduce(
|
|
lambda acc, kv: acc.replace(*kv), replacements.items(), src_value
|
|
)
|
|
elif isinstance(src_value, list):
|
|
return [do_value_substs(v) for v in src_value]
|
|
else:
|
|
return src_value
|
|
|
|
for key, value in sdk_json_data.items():
|
|
if key in split_vars:
|
|
value = value.split()
|
|
if key in subst_vars:
|
|
value = do_value_substs(value)
|
|
sdk_data[key] = value
|
|
|
|
return sdk_data
|
|
|
|
|
|
def _load_state_file(state_dir_node, filename: str) -> dict:
|
|
state_path = os.path.join(state_dir_node.abspath, filename)
|
|
if not os.path.exists(state_path):
|
|
raise StopError(f"State file {state_path} not found")
|
|
|
|
with open(state_path, "r") as f:
|
|
return json.load(f)
|
|
|
|
|
|
def generate(env, **kw):
|
|
sdk_current_sdk_dir_node = env["UFBT_CURRENT_SDK_DIR"]
|
|
|
|
sdk_components_filename = kw.get("SDK_COMPONENTS", "components.json")
|
|
ufbt_state_filename = kw.get("UFBT_STATE", "ufbt_state.json")
|
|
|
|
sdk_state = _load_state_file(sdk_current_sdk_dir_node, sdk_components_filename)
|
|
ufbt_state = _load_state_file(sdk_current_sdk_dir_node, ufbt_state_filename)
|
|
|
|
if not (sdk_components := sdk_state.get("components", {})):
|
|
raise StopError("SDK state file doesn't contain components data")
|
|
|
|
sdk_data = _load_sdk_data(
|
|
sdk_current_sdk_dir_node.Dir(sdk_components["sdk_headers.dir"]).abspath
|
|
)
|
|
|
|
if not sdk_state["meta"]["hw_target"].endswith(sdk_data["hardware"]):
|
|
raise StopError("SDK state file doesn't match hardware target")
|
|
|
|
scripts_dir = sdk_current_sdk_dir_node.Dir(sdk_components["scripts.dir"])
|
|
env.SetDefault(
|
|
# Paths
|
|
SDK_DEFINITION=env.File(sdk_data["sdk_symbols"]),
|
|
FBT_SCRIPT_DIR=scripts_dir,
|
|
FBT_DEBUG_DIR=scripts_dir.Dir("debug"),
|
|
LIBPATH=sdk_current_sdk_dir_node.Dir(sdk_components["lib.dir"]),
|
|
FW_ELF=sdk_current_sdk_dir_node.File(sdk_components["firmware.elf"]),
|
|
FW_BIN=sdk_current_sdk_dir_node.File(sdk_components["full.bin"]),
|
|
UPDATE_BUNDLE_DIR=sdk_current_sdk_dir_node.Dir(sdk_components["update.dir"]),
|
|
SVD_FILE="${FBT_DEBUG_DIR}/STM32WB55_CM4.svd",
|
|
# Build variables
|
|
ROOT_DIR=env.Dir("#"),
|
|
FIRMWARE_BUILD_CFG="firmware",
|
|
TARGET_HW=int(sdk_data["hardware"]),
|
|
CFLAGS_APP=sdk_data["cc_args"],
|
|
CXXFLAGS_APP=sdk_data["cpp_args"],
|
|
LINKFLAGS_APP=sdk_data["linker_args"],
|
|
LIBS=sdk_data["linker_libs"],
|
|
# ufbt state
|
|
# UFBT_STATE_DIR=ufbt_state_dir_node,
|
|
# UFBT_CURRENT_SDK_DIR=sdk_current_sdk_dir_node,
|
|
UFBT_STATE=ufbt_state,
|
|
UFBT_BOOTSTRAP_SCRIPT="${UFBT_SCRIPT_DIR}/bootstrap.py",
|
|
UFBT_SCRIPT_ROOT=scripts_dir.Dir("ufbt"),
|
|
)
|
|
|
|
sys.path.insert(0, env["FBT_SCRIPT_DIR"].abspath)
|
|
|
|
|
|
def exists(env):
|
|
return True
|