forked from xiaozhi/xiaozhi-esp32
Switch to 2.0 branch (#1152)
* Adapt boards to v2 partition tables * fix esp log error * fix display style * reset emotion after download assets * fix compiling * update assets default url * Add user only tools * Add image cache * smaller cache and buffer, more heap * use MAIN_EVENT_CLOCK_TICK to avoid audio glitches * bump to 2.0.0 * fix compiling errors --------- Co-authored-by: Xiaoxia <terrence.huang@tenclass.com>
This commit is contained in:
148
scripts/spiffs_assets/build_all.py
Normal file
148
scripts/spiffs_assets/build_all.py
Normal file
@@ -0,0 +1,148 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Build multiple spiffs assets partitions with different parameter combinations
|
||||
|
||||
This script calls build.py with different combinations of:
|
||||
- wakenet_models
|
||||
- text_fonts
|
||||
- emoji_collections
|
||||
|
||||
And generates assets.bin files with names like:
|
||||
wn9_nihaoxiaozhi_tts-font_puhui_common_20_4-emojis_32.bin
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
import shutil
|
||||
import subprocess
|
||||
import argparse
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
def ensure_dir(directory):
|
||||
"""Ensure directory exists, create if not"""
|
||||
os.makedirs(directory, exist_ok=True)
|
||||
|
||||
|
||||
def get_file_path(base_dir, filename):
|
||||
"""Get full path for a file, handling 'none' case"""
|
||||
if filename == "none":
|
||||
return None
|
||||
return os.path.join(base_dir, f"{filename}.bin" if not filename.startswith("emojis_") else filename)
|
||||
|
||||
|
||||
def build_assets(wakenet_model, text_font, emoji_collection, build_dir, final_dir):
|
||||
"""Build assets.bin using build.py with given parameters"""
|
||||
|
||||
# Prepare arguments for build.py
|
||||
cmd = [sys.executable, "build.py"]
|
||||
|
||||
if wakenet_model != "none":
|
||||
wakenet_path = os.path.join("../../managed_components/espressif__esp-sr/model/wakenet_model", wakenet_model)
|
||||
cmd.extend(["--wakenet_model", wakenet_path])
|
||||
|
||||
if text_font != "none":
|
||||
text_font_path = os.path.join("../../components/xiaozhi-fonts/build", f"{text_font}.bin")
|
||||
cmd.extend(["--text_font", text_font_path])
|
||||
|
||||
if emoji_collection != "none":
|
||||
emoji_path = os.path.join("../../components/xiaozhi-fonts/build", emoji_collection)
|
||||
cmd.extend(["--emoji_collection", emoji_path])
|
||||
|
||||
print(f"\n正在构建: {wakenet_model}-{text_font}-{emoji_collection}")
|
||||
print(f"执行命令: {' '.join(cmd)}")
|
||||
|
||||
try:
|
||||
# Run build.py
|
||||
result = subprocess.run(cmd, check=True, cwd=os.path.dirname(__file__))
|
||||
|
||||
# Generate output filename
|
||||
output_name = f"{wakenet_model}-{text_font}-{emoji_collection}.bin"
|
||||
|
||||
# Copy generated assets.bin to final directory with new name
|
||||
src_path = os.path.join(build_dir, "assets.bin")
|
||||
dst_path = os.path.join(final_dir, output_name)
|
||||
|
||||
if os.path.exists(src_path):
|
||||
shutil.copy2(src_path, dst_path)
|
||||
print(f"✓ 成功生成: {output_name}")
|
||||
return True
|
||||
else:
|
||||
print(f"✗ 错误: 未找到生成的 assets.bin 文件")
|
||||
return False
|
||||
|
||||
except subprocess.CalledProcessError as e:
|
||||
print(f"✗ 构建失败: {e}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"✗ 未知错误: {e}")
|
||||
return False
|
||||
|
||||
|
||||
def main():
|
||||
# Configuration
|
||||
wakenet_models = [
|
||||
"none",
|
||||
"wn9_nihaoxiaozhi_tts",
|
||||
"wn9s_nihaoxiaozhi"
|
||||
]
|
||||
|
||||
text_fonts = [
|
||||
"none",
|
||||
"font_puhui_common_14_1",
|
||||
"font_puhui_common_16_4",
|
||||
"font_puhui_common_20_4",
|
||||
"font_puhui_common_30_4",
|
||||
]
|
||||
|
||||
emoji_collections = [
|
||||
"none",
|
||||
"emojis_32",
|
||||
"emojis_64",
|
||||
]
|
||||
|
||||
# Get script directory
|
||||
script_dir = os.path.dirname(os.path.abspath(__file__))
|
||||
|
||||
# Set directory paths
|
||||
build_dir = os.path.join(script_dir, "build")
|
||||
final_dir = os.path.join(build_dir, "final")
|
||||
|
||||
# Ensure directories exist
|
||||
ensure_dir(build_dir)
|
||||
ensure_dir(final_dir)
|
||||
|
||||
print("开始构建多个 SPIFFS assets 分区...")
|
||||
print(f"输出目录: {final_dir}")
|
||||
|
||||
# Track successful builds
|
||||
successful_builds = 0
|
||||
total_combinations = len(wakenet_models) * len(text_fonts) * len(emoji_collections)
|
||||
|
||||
# Build all combinations
|
||||
for wakenet_model in wakenet_models:
|
||||
for text_font in text_fonts:
|
||||
for emoji_collection in emoji_collections:
|
||||
if build_assets(wakenet_model, text_font, emoji_collection, build_dir, final_dir):
|
||||
successful_builds += 1
|
||||
|
||||
print(f"\n构建完成!")
|
||||
print(f"成功构建: {successful_builds}/{total_combinations}")
|
||||
print(f"输出文件位置: {final_dir}")
|
||||
|
||||
# List generated files
|
||||
if os.path.exists(final_dir):
|
||||
files = [f for f in os.listdir(final_dir) if f.endswith('.bin')]
|
||||
if files:
|
||||
print("\n生成的文件:")
|
||||
for file in sorted(files):
|
||||
file_size = os.path.getsize(os.path.join(final_dir, file))
|
||||
print(f" {file} ({file_size:,} bytes)")
|
||||
else:
|
||||
print("\n未找到生成的 .bin 文件")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user