393 lines
14 KiB
Python
393 lines
14 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
Subtitle Font Manager · 字幕字体管理器
|
|||
|
|
======================================
|
|||
|
|
固定项目字幕字体，不能依赖系统默认字体。
|
|||
|
|
|
|||
|
|
功能：
|
|||
|
|
1. 检查项目字体是否已安装
|
|||
|
|
2. 如果未安装，自动下载/复制到项目字体目录
|
|||
|
|
3. 生成字体配置文件（供 ASS 渲染器使用）
|
|||
|
|
4. 验证字体文件是否有效（支持中文）
|
|||
|
|
|
|||
|
|
依赖：
|
|||
|
|
pip install fonttools # 可选，用于验证字体
|
|||
|
|
# macOS: 系统字体通常在 /System/Library/Fonts/
|
|||
|
|
# Ubuntu: 系统字体通常在 /usr/share/fonts/
|
|||
|
|
|
|||
|
|
用法：
|
|||
|
|
# 检查并安装项目字体
|
|||
|
|
python subtitle-font-manager.py --check --install
|
|||
|
|
|
|||
|
|
# 列出可用的中文字体
|
|||
|
|
python subtitle-font-manager.py --list
|
|||
|
|
|
|||
|
|
# 设置项目字体
|
|||
|
|
python subtitle-font-manager.py --set-font "PingFang SC" --output project-font.json
|
|||
|
|
|
|||
|
|
# 验证字体文件是否有效
|
|||
|
|
python subtitle-font-manager.py --validate ./assets/fonts/PingFangSC-Regular.otf
|
|||
|
|
|
|||
|
|
# 生成 FFmpeg 字体路径配置
|
|||
|
|
python subtitle-font-manager.py --generate-ffmpeg-config --output ffmpeg-font.conf
|
|||
|
|
|
|||
|
|
输出：
|
|||
|
|
- project-font.json: 项目字体配置
|
|||
|
|
- ./assets/fonts/: 项目字体文件（如果需要下载）
|
|||
|
|
- ffmpeg-font.conf: FFmpeg 字体配置
|
|||
|
|
|
|||
|
|
路径：
|
|||
|
|
video-ai-system/engines/subtitle-pipeline/font-manager/subtitle-font-manager.py
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import argparse
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import shutil
|
|||
|
|
import sys
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
# 尝试导入 fonttools（可选）
|
|||
|
|
try:
|
|||
|
|
from fonttools import TTFont
|
|||
|
|
from fonttools.unicode import Unicode
|
|||
|
|
FONTTOOLS_AVAILABLE = True
|
|||
|
|
except ImportError:
|
|||
|
|
FONTTOOLS_AVAILABLE = False
|
|||
|
|
print("[WARN] 未安装 fonttools，将无法验证字体是否支持中文")
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 项目中文字体候选列表（按优先级排序）
|
|||
|
|
CHINESE_FONT_CANDIDATES = [
|
|||
|
|
{
|
|||
|
|
"name": "PingFang SC",
|
|||
|
|
"mac_path": "/System/Library/Fonts/PingFang.ttc",
|
|||
|
|
"windows_path": "C:/Windows/Fonts/pingfang.ttc",
|
|||
|
|
"linux_path": "/usr/share/fonts/truetype/pingfang/PingFangSC-Regular.otf",
|
|||
|
|
"download_url": None, # macOS 系统自带，无需下载
|
|||
|
|
"weight": "Regular"
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"name": "Hiragino Sans GB",
|
|||
|
|
"mac_path": "/System/Library/Fonts/Hiragino Sans GB.ttc",
|
|||
|
|
"windows_path": "C:/Windows/Fonts/hiragino.ttc",
|
|||
|
|
"linux_path": "/usr/share/fonts/truetype/hiragino/HiraginoSansGB-Regular.otf",
|
|||
|
|
"download_url": None,
|
|||
|
|
"weight": "Regular"
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"name": "Noto Sans CJK SC",
|
|||
|
|
"mac_path": "/System/Library/Fonts/Supplemental/Noto Sans CJK SC.ttc",
|
|||
|
|
"windows_path": "C:/Windows/Fonts/NotoSansCJK-Regular.ttc",
|
|||
|
|
"linux_path": "/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc",
|
|||
|
|
"download_url": "https://github.com/googlefonts/noto-cjk/raw/main/Sans/OTF/SimplifiedChinese/NotoSansCJKsc-Regular.otf",
|
|||
|
|
"weight": "Regular"
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"name": "Source Han Sans SC",
|
|||
|
|
"mac_path": "/System/Library/Fonts/Supplemental/Source Han Sans SC.ttc",
|
|||
|
|
"windows_path": "C:/Windows/Fonts/SourceHanSansSC-Regular.otf",
|
|||
|
|
"linux_path": "/usr/share/fonts/opentype/source-han-sans/SourceHanSansSC-Regular.otf",
|
|||
|
|
"download_url": "https://github.com/adobe-fonts/source-han-sans/raw/release/OTF/SimplifiedChinese/SourceHanSansSC-Regular.otf",
|
|||
|
|
"weight": "Regular"
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"name": "WenQuanYi Micro Hei",
|
|||
|
|
"mac_path": None,
|
|||
|
|
"windows_path": None,
|
|||
|
|
"linux_path": "/usr/share/fonts/truetype/wqy/wqy-microhei.ttc",
|
|||
|
|
"download_url": "https://github.com/anthonyfok/fonts-wqy-microhei/raw/master/wqy-microhei.ttc",
|
|||
|
|
"weight": "Regular"
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
|
|||
|
|
def find_font_file(font_info: dict) -> str:
|
|||
|
|
"""
|
|||
|
|
查找字体文件（跨平台）
|
|||
|
|
|
|||
|
|
:param font_info: 字体信息字典
|
|||
|
|
:return: 字体文件路径，如果未找到返回 None
|
|||
|
|
"""
|
|||
|
|
import platform
|
|||
|
|
system = platform.system()
|
|||
|
|
|
|||
|
|
if system == "Darwin": # macOS
|
|||
|
|
font_path = font_info.get("mac_path")
|
|||
|
|
elif system == "Windows":
|
|||
|
|
font_path = font_info.get("windows_path")
|
|||
|
|
elif system == "Linux":
|
|||
|
|
font_path = font_info.get("linux_path")
|
|||
|
|
else:
|
|||
|
|
font_path = None
|
|||
|
|
|
|||
|
|
if font_path and os.path.isfile(font_path):
|
|||
|
|
return font_path
|
|||
|
|
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def check_chinese_support(font_path: str) -> bool:
|
|||
|
|
"""
|
|||
|
|
检查字体是否支持中文（简体）
|
|||
|
|
|
|||
|
|
:param font_path: 字体文件路径
|
|||
|
|
:return: 是否支持中文
|
|||
|
|
"""
|
|||
|
|
if not FONTTOOLS_AVAILABLE:
|
|||
|
|
print("[WARN] 未安装 fonttools，跳过中文支持检查")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
font = TTFont(font_path)
|
|||
|
|
|
|||
|
|
# 检查是否包含常用简体中文字符（GB2312 范围）
|
|||
|
|
# GB2312 一级汉字：0x4E00-0x9FA5
|
|||
|
|
chinese_chars = [chr(code) for code in range(0x4E00, 0x9FA5)]
|
|||
|
|
|
|||
|
|
# 抽样检查（检查前100个常用汉字）
|
|||
|
|
sample_chars = chinese_chars[:100]
|
|||
|
|
|
|||
|
|
supported = 0
|
|||
|
|
for char in sample_chars:
|
|||
|
|
if char in font:
|
|||
|
|
supported += 1
|
|||
|
|
|
|||
|
|
support_ratio = supported / len(sample_chars)
|
|||
|
|
print(f"[INFO] 字体中文支持率：{support_ratio*100:.1f}%")
|
|||
|
|
|
|||
|
|
return support_ratio > 0.8 # 支持率 > 80% 认为支持中文
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[ERROR] 检查字体中文支持失败：{e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
|
|||
|
|
def install_font_to_project(font_info: dict, project_fonts_dir: str) -> str:
|
|||
|
|
"""
|
|||
|
|
安装字体到项目字体目录
|
|||
|
|
|
|||
|
|
:param font_info: 字体信息字典
|
|||
|
|
:param project_fonts_dir: 项目字体目录
|
|||
|
|
:return: 安装后的字体文件路径
|
|||
|
|
"""
|
|||
|
|
os.makedirs(project_fonts_dir, exist_ok=True)
|
|||
|
|
|
|||
|
|
# 查找系统字体文件
|
|||
|
|
system_font_path = find_font_file(font_info)
|
|||
|
|
|
|||
|
|
if not system_font_path:
|
|||
|
|
print(f"[ERROR] 未找到系统字体：{font_info['name']}")
|
|||
|
|
if font_info.get("download_url"):
|
|||
|
|
print(f"[INFO] 尝试从以下地址下载：{font_info['download_url']}")
|
|||
|
|
# TODO: 实现下载逻辑
|
|||
|
|
return None
|
|||
|
|
else:
|
|||
|
|
print(f"[ERROR] 该字体无下载地址，请手动安装")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
# 复制到项目字体目录
|
|||
|
|
font_filename = os.path.basename(system_font_path)
|
|||
|
|
project_font_path = os.path.join(project_fonts_dir, font_filename)
|
|||
|
|
|
|||
|
|
if not os.path.isfile(project_font_path):
|
|||
|
|
try:
|
|||
|
|
shutil.copy2(system_font_path, project_font_path)
|
|||
|
|
print(f"[OK] 字体已复制到项目目录：{project_font_path}")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[ERROR] 复制字体失败：{e}")
|
|||
|
|
return None
|
|||
|
|
else:
|
|||
|
|
print(f"[INFO] 字体已存在于项目目录：{project_font_path}")
|
|||
|
|
|
|||
|
|
return project_font_path
|
|||
|
|
|
|||
|
|
|
|||
|
|
def generate_project_font_config(font_path: str, output_path: str, font_name: str = None):
|
|||
|
|
"""
|
|||
|
|
生成项目字体配置文件
|
|||
|
|
|
|||
|
|
:param font_path: 字体文件路径
|
|||
|
|
:param output_path: 输出配置文件路径
|
|||
|
|
:param font_name: 字体名称（如果不指定，从文件读取）
|
|||
|
|
"""
|
|||
|
|
# 如果未指定字体名称，尝试从文件读取
|
|||
|
|
if not font_name and FONTTOOLS_AVAILABLE:
|
|||
|
|
try:
|
|||
|
|
font = TTFont(font_path)
|
|||
|
|
font_name = font.get("name").getDebugName(4) # 完整字体名称
|
|||
|
|
except Exception:
|
|||
|
|
font_name = os.path.splitext(os.path.basename(font_path))[0]
|
|||
|
|
|
|||
|
|
config = {
|
|||
|
|
"_comment": "项目字幕字体配置",
|
|||
|
|
"font_name": font_name or "Unknown",
|
|||
|
|
"font_path": os.path.abspath(font_path),
|
|||
|
|
"font_path_relative": os.path.relpath(font_path, os.path.dirname(output_path)),
|
|||
|
|
"install_date": __import__("datetime").datetime.now().isoformat(),
|
|||
|
|
"supports_chinese": check_chinese_support(font_path) if FONTTOOLS_AVAILABLE else None,
|
|||
|
|
"ass_fontname": font_name, # ASS 格式使用的字体名称
|
|||
|
|
"ffmpeg_font_path": font_path, # FFmpeg 使用的字体路径
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
with open(output_path, "w", encoding="utf-8") as f:
|
|||
|
|
json.dump(config, f, ensure_ascii=False, indent=2)
|
|||
|
|
|
|||
|
|
print(f"[OK] 项目字体配置已生成：{output_path}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def list_available_fonts():
|
|||
|
|
"""
|
|||
|
|
列出系统中可用的中文字体
|
|||
|
|
"""
|
|||
|
|
print("\n[INFO] 查找系统中可用的中文字体...")
|
|||
|
|
|
|||
|
|
available = []
|
|||
|
|
for font_info in CHINESE_FONT_CANDIDATES:
|
|||
|
|
font_path = find_font_file(font_info)
|
|||
|
|
if font_path:
|
|||
|
|
available.append({
|
|||
|
|
"name": font_info["name"],
|
|||
|
|
"path": font_path,
|
|||
|
|
"weight": font_info["weight"]
|
|||
|
|
})
|
|||
|
|
print(f" ✅ {font_info['name']} ({font_info['weight']})")
|
|||
|
|
print(f" 路径：{font_path}")
|
|||
|
|
else:
|
|||
|
|
print(f" ❌ {font_info['name']} (未找到)")
|
|||
|
|
|
|||
|
|
if not available:
|
|||
|
|
print("\n[WARN] 未找到任何中文字体，请手动安装中文字体")
|
|||
|
|
print("推荐字体：")
|
|||
|
|
print(" - macOS: PingFang SC（系统自带）")
|
|||
|
|
print(" - Ubuntu: fonts-noto-cjk（apt install fonts-noto-cjk）")
|
|||
|
|
print(" - Windows: 微软雅黑（系统自带）")
|
|||
|
|
|
|||
|
|
return available
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
parser = argparse.ArgumentParser(
|
|||
|
|
description="Subtitle Font Manager · 字幕字体管理器",
|
|||
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|||
|
|
epilog="""
|
|||
|
|
示例：
|
|||
|
|
python subtitle-font-manager.py --list
|
|||
|
|
python subtitle-font-manager.py --check --install
|
|||
|
|
python subtitle-font-manager.py --set-font "PingFang SC" --output project-font.json
|
|||
|
|
python subtitle-font-manager.py --validate ./assets/fonts/PingFangSC-Regular.otf
|
|||
|
|
"""
|
|||
|
|
)
|
|||
|
|
parser.add_argument("--list", action="store_true", help="列出可用的中文字体")
|
|||
|
|
parser.add_argument("--check", action="store_true", help="检查项目字体是否已安装")
|
|||
|
|
parser.add_argument("--install", action="store_true", help="如果未安装，自动安装字体到项目目录")
|
|||
|
|
parser.add_argument("--set-font", help="设置项目字体（字体名称）")
|
|||
|
|
parser.add_argument("--output", help="输出配置文件路径")
|
|||
|
|
parser.add_argument("--validate", help="验证字体文件是否有效")
|
|||
|
|
parser.add_argument("--generate-ffmpeg-config", action="store_true", help="生成 FFmpeg 字体配置")
|
|||
|
|
parser.add_argument("--project-fonts-dir", default="./assets/fonts/", help="项目字体目录")
|
|||
|
|
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
|
|||
|
|
if args.list:
|
|||
|
|
list_available_fonts()
|
|||
|
|
sys.exit(0)
|
|||
|
|
|
|||
|
|
if args.validate:
|
|||
|
|
if not os.path.isfile(args.validate):
|
|||
|
|
print(f"[ERROR] 字体文件不存在：{args.validate}")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
print(f"[INFO] 验证字体文件：{args.validate}")
|
|||
|
|
supports_chinese = check_chinese_support(args.validate)
|
|||
|
|
|
|||
|
|
if supports_chinese:
|
|||
|
|
print(f"[OK] 字体有效，支持中文")
|
|||
|
|
else:
|
|||
|
|
print(f"[WARN] 字体可能不支持中文，请手动检查")
|
|||
|
|
sys.exit(0)
|
|||
|
|
|
|||
|
|
if args.check or args.install:
|
|||
|
|
# 检查项目字体是否已安装
|
|||
|
|
project_font_config = os.path.join(args.project_fonts_dir, "../subtitle-styles/project-font.json")
|
|||
|
|
project_font_config = os.path.abspath(project_font_config)
|
|||
|
|
|
|||
|
|
if os.path.isfile(project_font_config):
|
|||
|
|
with open(project_font_config, "r", encoding="utf-8") as f:
|
|||
|
|
config = json.load(f)
|
|||
|
|
print(f"[INFO] 项目字体已配置：{config['font_name']}")
|
|||
|
|
print(f"[INFO] 字体路径：{config['font_path']}")
|
|||
|
|
|
|||
|
|
if not os.path.isfile(config["font_path"]):
|
|||
|
|
print(f"[WARN] 字体文件不存在，需要重新安装")
|
|||
|
|
args.install = True
|
|||
|
|
else:
|
|||
|
|
sys.exit(0)
|
|||
|
|
|
|||
|
|
if args.install:
|
|||
|
|
# 安装字体
|
|||
|
|
available = list_available_fonts()
|
|||
|
|
|
|||
|
|
if not available:
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
# 使用第一个可用的字体
|
|||
|
|
font_info = available[0]
|
|||
|
|
print(f"\n[INFO] 安装字体：{font_info['name']}")
|
|||
|
|
|
|||
|
|
# 查找字体信息
|
|||
|
|
font_info_full = next((f for f in CHINESE_FONT_CANDIDATES if f["name"] == font_info["name"]), None)
|
|||
|
|
|
|||
|
|
if font_info_full:
|
|||
|
|
project_font_path = install_font_to_project(font_info_full, args.project_fonts_dir)
|
|||
|
|
|
|||
|
|
if project_font_path and args.output:
|
|||
|
|
generate_project_font_config(project_font_path, args.output, font_info["name"])
|
|||
|
|
|
|||
|
|
if args.set_font:
|
|||
|
|
# 设置项目字体
|
|||
|
|
font_info = next((f for f in CHINESE_FONT_CANDIDATES if f["name"] == args.set_font), None)
|
|||
|
|
|
|||
|
|
if not font_info:
|
|||
|
|
print(f"[ERROR] 未找到字体：{args.set_font}")
|
|||
|
|
print(f"[INFO] 可用字体：{[f['name'] for f in CHINESE_FONT_CANDIDATES]}")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
project_font_path = install_font_to_project(font_info, args.project_fonts_dir)
|
|||
|
|
|
|||
|
|
if project_font_path and args.output:
|
|||
|
|
generate_project_font_config(project_font_path, args.output, args.set_font)
|
|||
|
|
|
|||
|
|
if args.generate_ffmpeg_config:
|
|||
|
|
# 生成 FFmpeg 字体配置
|
|||
|
|
# FFmpeg 可以使用 fontfile 滤镜指定字体文件
|
|||
|
|
output = args.output or "ffmpeg-font.conf"
|
|||
|
|
|
|||
|
|
# 读取项目字体配置
|
|||
|
|
project_font_config = os.path.join(args.project_fonts_dir, "../subtitle-styles/project-font.json")
|
|||
|
|
project_font_config = os.path.abspath(project_font_config)
|
|||
|
|
|
|||
|
|
if not os.path.isfile(project_font_config):
|
|||
|
|
print(f"[ERROR] 项目字体未配置，请先运行 --check --install")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
with open(project_font_config, "r", encoding="utf-8") as f:
|
|||
|
|
config = json.load(f)
|
|||
|
|
|
|||
|
|
# 生成 FFmpeg 滤镜字符串
|
|||
|
|
font_path = config["font_path"]
|
|||
|
|
ffmpeg_filter = f"subtitles=subtitles.ass:force_style='FontName={config['font_name']},FontFile={font_path}'"
|
|||
|
|
|
|||
|
|
with open(output, "w", encoding="utf-8") as f:
|
|||
|
|
f.write(f"# FFmpeg 字体配置\n")
|
|||
|
|
f.write(f"# 使用方法：ffmpeg -i input.mp4 -vf \"{ffmpeg_filter}\" output.mp4\n\n")
|
|||
|
|
f.write(f"font_path={font_path}\n")
|
|||
|
|
f.write(f"font_name={config['font_name']}\n")
|
|||
|
|
f.write(f"ffmpeg_filter={ffmpeg_filter}\n")
|
|||
|
|
|
|||
|
|
print(f"[OK] FFmpeg 字体配置已生成：{output}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|