401 lines
14 KiB
Python
401 lines
14 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
ASS Subtitle Renderer · ASS字幕渲染器
|
|||
|
|
=====================================
|
|||
|
|
用 ASS/libass 正式渲染字幕，支持字体、粗体、描边、边距、对齐。
|
|||
|
|
不再用 Pillow PNG 凑合。
|
|||
|
|
|
|||
|
|
依赖：
|
|||
|
|
pip install pysrt
|
|||
|
|
# FFmpeg 需要编译时启用 libass 支持（通常默认启用）
|
|||
|
|
|
|||
|
|
用法：
|
|||
|
|
# 生成 ASS 字幕文件
|
|||
|
|
python ass-subtitle-renderer.py --srt input.srt --style reference-drama --output subtitles.ass
|
|||
|
|
|
|||
|
|
# 用 FFmpeg + libass 烧字幕到视频
|
|||
|
|
python ass-subtitle-renderer.py --srt input.srt --video input.mp4 --output output.mp4 --style reference-drama
|
|||
|
|
|
|||
|
|
# 指定自定义样式配置
|
|||
|
|
python ass-subtitle-renderer.py --srt input.srt --config custom-style.json --output subtitles.ass
|
|||
|
|
|
|||
|
|
# 批量处理多集
|
|||
|
|
python ass-subtitle-renderer.py --batch ./srt/ --style reference-drama --output-dir ./ass/
|
|||
|
|
|
|||
|
|
输出：
|
|||
|
|
- ASS 字幕文件（可直接用 FFmpeg 烧录）
|
|||
|
|
- 烧录字幕后的视频（可选）
|
|||
|
|
|
|||
|
|
路径：
|
|||
|
|
video-ai-system/engines/subtitle-pipeline/reference-analysis/ass-subtitle-renderer.py
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import argparse
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import subprocess
|
|||
|
|
import sys
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
import pysrt
|
|||
|
|
except ImportError:
|
|||
|
|
print("[ERROR] 缺少依赖：pysrt")
|
|||
|
|
print("请先安装：pip install pysrt")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 默认样式配置（参考短剧风格）
|
|||
|
|
DEFAULT_STYLE = {
|
|||
|
|
"video_width": 1080,
|
|||
|
|
"video_height": 1920,
|
|||
|
|
"font_family": "PingFang SC",
|
|||
|
|
"font_size": 38,
|
|||
|
|
"font_color": "&HFFFFFF&", # 白色（ASS格式：&HBBGGRR&）
|
|||
|
|
"font_color_hex": "#FFFFFF",
|
|||
|
|
"bold": True,
|
|||
|
|
"italic": False,
|
|||
|
|
"underline": False,
|
|||
|
|
"strikeout": False,
|
|||
|
|
"stroke_enabled": True,
|
|||
|
|
"stroke_width": 2,
|
|||
|
|
"stroke_color": "&H000000&", # 黑色描边
|
|||
|
|
"stroke_opacity": 0.92, # &H40&（0-255，0=透明，255=不透明）
|
|||
|
|
"shadow_enabled": False,
|
|||
|
|
"shadow_depth": 0,
|
|||
|
|
"alignment": 2, # 2=中下
|
|||
|
|
"margin_left": 100,
|
|||
|
|
"margin_right": 100,
|
|||
|
|
"margin_vertical": 50,
|
|||
|
|
"line_spacing": 1.45,
|
|||
|
|
"letter_spacing": 0,
|
|||
|
|
"background_enabled": False,
|
|||
|
|
"fade_in_ms": 150,
|
|||
|
|
"fade_out_ms": 150,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def load_style_config(config_path: str = None, style_name: str = "reference-drama") -> dict:
|
|||
|
|
"""
|
|||
|
|
加载样式配置
|
|||
|
|
|
|||
|
|
:param config_path: 自定义配置文件路径
|
|||
|
|
:param style_name: 预设样式名称
|
|||
|
|
:return: 样式字典
|
|||
|
|
"""
|
|||
|
|
if config_path and os.path.isfile(config_path):
|
|||
|
|
with open(config_path, "r", encoding="utf-8") as f:
|
|||
|
|
config = json.load(f)
|
|||
|
|
print(f"[INFO] 已加载自定义样式配置：{config_path}")
|
|||
|
|
return config
|
|||
|
|
|
|||
|
|
# 尝试从 assets/subtitle-styles/ 加载预设样式
|
|||
|
|
style_file = os.path.join(
|
|||
|
|
os.path.dirname(__file__),
|
|||
|
|
"..", "..", "..", "assets", "subtitle-styles",
|
|||
|
|
f"subtitle-style.{style_name}.json"
|
|||
|
|
)
|
|||
|
|
style_file = os.path.abspath(style_file)
|
|||
|
|
|
|||
|
|
if os.path.isfile(style_file):
|
|||
|
|
with open(style_file, "r", encoding="utf-8") as f:
|
|||
|
|
config = json.load(f)
|
|||
|
|
print(f"[INFO] 已加载预设样式：{style_name}")
|
|||
|
|
return config
|
|||
|
|
|
|||
|
|
print(f"[WARN] 未找到样式配置，使用默认样式")
|
|||
|
|
return DEFAULT_STYLE
|
|||
|
|
|
|||
|
|
|
|||
|
|
def srt_time_to_ass(srt_time) -> str:
|
|||
|
|
"""
|
|||
|
|
将 pysrt 时间对象转换为 ASS 时间戳格式
|
|||
|
|
|
|||
|
|
:param srt_time: pysrt 时间对象
|
|||
|
|
:return: ASS 时间戳字符串（H:MM:SS.cc）
|
|||
|
|
"""
|
|||
|
|
hours = srt_time.hours
|
|||
|
|
minutes = srt_time.minutes
|
|||
|
|
seconds = srt_time.seconds
|
|||
|
|
milliseconds = srt_time.milliseconds
|
|||
|
|
|
|||
|
|
# ASS 时间戳格式：H:MM:SS.cc（cc=厘秒，1/100秒）
|
|||
|
|
centiseconds = milliseconds // 10
|
|||
|
|
return f"{hours}:{minutes:02d}:{seconds:02d}.{centiseconds:02d}"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def generate_ass_file(srt_path: str, output_path: str, style: dict) -> bool:
|
|||
|
|
"""
|
|||
|
|
将 SRT 字幕转换为 ASS 格式
|
|||
|
|
|
|||
|
|
:param srt_path: SRT 文件路径
|
|||
|
|
:param output_path: 输出 ASS 文件路径
|
|||
|
|
:param style: 样式字典
|
|||
|
|
:return: 是否成功
|
|||
|
|
"""
|
|||
|
|
if not os.path.isfile(srt_path):
|
|||
|
|
print(f"[ERROR] SRT 文件不存在：{srt_path}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# 加载 SRT
|
|||
|
|
try:
|
|||
|
|
subs = pysrt.open(srt_path, encoding="utf-8")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[ERROR] 加载 SRT 失败：{e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
print(f"[INFO] 找到 {len(subs)} 条字幕，开始生成 ASS 文件...")
|
|||
|
|
|
|||
|
|
# 构建 ASS 文件内容
|
|||
|
|
ass_lines = []
|
|||
|
|
|
|||
|
|
# 1. [Script Info] 节
|
|||
|
|
ass_lines.append("[Script Info]")
|
|||
|
|
ass_lines.append("; Script generated by ASS Subtitle Renderer")
|
|||
|
|
ass_lines.append(f"PlayResX: {style.get('video_width', 1080)}")
|
|||
|
|
ass_lines.append(f"PlayResY: {style.get('video_height', 1920)}")
|
|||
|
|
ass_lines.append("Aspect Ratio: 9:16")
|
|||
|
|
ass_lines.append("")
|
|||
|
|
|
|||
|
|
# 2. [V4+ Styles] 节
|
|||
|
|
ass_lines.append("[V4+ Styles]")
|
|||
|
|
ass_lines.append(
|
|||
|
|
"Format: Name, Fontname, Fontsize, PrimaryColour, SecondaryColour, OutlineColour, BackColour, "
|
|||
|
|
"Bold, Italic, Underline, StrikeOut, ScaleX, ScaleY, Spacing, Angle, "
|
|||
|
|
"BorderStyle, Outline, Shadow, Alignment, MarginL, MarginR, MarginV, Encoding"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 构建样式行
|
|||
|
|
font_name = style.get("font_family", "PingFang SC")
|
|||
|
|
font_size = style.get("font_size", 38)
|
|||
|
|
primary_colour = style.get("font_color", "&HFFFFFF&")
|
|||
|
|
outline_colour = style.get("stroke_color", "&H000000&")
|
|||
|
|
|
|||
|
|
# 背景色（如果启用背景）
|
|||
|
|
if style.get("background_enabled", False):
|
|||
|
|
bg_opacity = int(style.get("background_opacity", 0.6) * 255)
|
|||
|
|
back_colour = f"&H{bg_opacity:02X}000000&" # &HAAKKBBDD&
|
|||
|
|
else:
|
|||
|
|
back_colour = "&H00000000&" # 透明
|
|||
|
|
|
|||
|
|
bold = -1 if style.get("bold", True) else 0
|
|||
|
|
italic = -1 if style.get("italic", False) else 0
|
|||
|
|
underline = -1 if style.get("underline", False) else 0
|
|||
|
|
strikeout = -1 if style.get("strikeout", False) else 0
|
|||
|
|
|
|||
|
|
outline_width = style.get("stroke_width", 2) if style.get("stroke_enabled", True) else 0
|
|||
|
|
shadow_depth = style.get("shadow_depth", 0) if style.get("shadow_enabled", False) else 0
|
|||
|
|
|
|||
|
|
alignment = style.get("alignment", 2)
|
|||
|
|
margin_l = style.get("margin_left", 100)
|
|||
|
|
margin_r = style.get("margin_right", 100)
|
|||
|
|
margin_v = style.get("margin_vertical", 50)
|
|||
|
|
|
|||
|
|
style_line = (
|
|||
|
|
f"Style: Default,{font_name},{font_size},{primary_colour},&H000000FF&,{outline_colour},{back_colour},"
|
|||
|
|
f"{bold},{italic},{underline},{strikeout},100,100,{style.get('letter_spacing', 0)},0,"
|
|||
|
|
f"1,{outline_width},{shadow_depth},{alignment},{margin_l},{margin_r},{margin_v},1"
|
|||
|
|
)
|
|||
|
|
ass_lines.append(style_line)
|
|||
|
|
ass_lines.append("")
|
|||
|
|
|
|||
|
|
# 3. [Events] 节
|
|||
|
|
ass_lines.append("[Events]")
|
|||
|
|
ass_lines.append("Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text")
|
|||
|
|
|
|||
|
|
for sub in subs:
|
|||
|
|
start_time = srt_time_to_ass(sub.start)
|
|||
|
|
end_time = srt_time_to_ass(sub.end)
|
|||
|
|
text = sub.text.strip()
|
|||
|
|
|
|||
|
|
# 添加淡入淡出效果（如果启用）
|
|||
|
|
fade_in = style.get("fade_in_ms", 150)
|
|||
|
|
fade_out = style.get("fade_out_ms", 150)
|
|||
|
|
|
|||
|
|
if fade_in > 0 and fade_out > 0:
|
|||
|
|
# ASS 淡入淡出标签：{\fad(fade_in,fade_out)}
|
|||
|
|
effect_tags = f"{{\\fad({fade_in},{fade_out})}}"
|
|||
|
|
else:
|
|||
|
|
effect_tags = ""
|
|||
|
|
|
|||
|
|
# 转义特殊字符（ASS 格式）
|
|||
|
|
text = text.replace("\\", "\\\\")
|
|||
|
|
text = text.replace("{", "\\{")
|
|||
|
|
text = text.replace("}", "\\}")
|
|||
|
|
text = text.replace("\n", "\\N") # ASS 换行符
|
|||
|
|
|
|||
|
|
# 构建对话行
|
|||
|
|
dialogue_line = (
|
|||
|
|
f"Dialogue: 0,{start_time},{end_time},Default,,0,0,0,,{effect_tags}{text}"
|
|||
|
|
)
|
|||
|
|
ass_lines.append(dialogue_line)
|
|||
|
|
|
|||
|
|
# 写入 ASS 文件
|
|||
|
|
try:
|
|||
|
|
os.makedirs(os.path.dirname(os.path.abspath(output_path)), exist_ok=True)
|
|||
|
|
with open(output_path, "w", encoding="utf-8-sig") as f: # UTF-8 BOM for ASS
|
|||
|
|
f.write("\n".join(ass_lines))
|
|||
|
|
|
|||
|
|
print(f"[OK] ASS 字幕文件已生成：{output_path}")
|
|||
|
|
print(f"[INFO] 共 {len(subs)} 条字幕")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[ERROR] 写入 ASS 文件失败：{e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
|
|||
|
|
def burn_subtitles_with_ffmpeg(video_path: str, ass_path: str, output_path: str) -> bool:
|
|||
|
|
"""
|
|||
|
|
用 FFmpeg + libass 烧录字幕到视频
|
|||
|
|
|
|||
|
|
:param video_path: 输入视频路径
|
|||
|
|
:param ass_path: ASS 字幕文件路径
|
|||
|
|
:param output_path: 输出视频路径
|
|||
|
|
:return: 是否成功
|
|||
|
|
"""
|
|||
|
|
if not os.path.isfile(video_path):
|
|||
|
|
print(f"[ERROR] 视频文件不存在：{video_path}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
if not os.path.isfile(ass_path):
|
|||
|
|
print(f"[ERROR] ASS 字幕文件不存在：{ass_path}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# 使用 subtitles 滤镜（更通用，支持 SRT/ASS 等多种格式）
|
|||
|
|
# 正确语法：subtitles=filename='path' （需要 filename= 前缀）
|
|||
|
|
ass_path_escaped = ass_path.replace(":", "\\:").replace("'", "'\\''")
|
|||
|
|
|
|||
|
|
cmd = [
|
|||
|
|
"ffmpeg", "-y",
|
|||
|
|
"-i", video_path,
|
|||
|
|
"-vf", f"subtitles=filename='{ass_path_escaped}'",
|
|||
|
|
"-c:v", "libx264",
|
|||
|
|
"-pix_fmt", "yuv420p",
|
|||
|
|
"-c:a", "copy",
|
|||
|
|
"-shortest",
|
|||
|
|
output_path
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
print(f"[INFO] 开始用 FFmpeg + libass 烧录字幕...")
|
|||
|
|
print(f"[INFO] 命令：{' '.join(cmd[:10])}...") # 只打印前10个参数
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
result = subprocess.run(cmd, check=False, capture_output=True, text=True)
|
|||
|
|
except FileNotFoundError:
|
|||
|
|
print("[ERROR] 未找到 ffmpeg，请先安装 FFmpeg")
|
|||
|
|
print("安装方法：")
|
|||
|
|
print(" macOS: brew install ffmpeg")
|
|||
|
|
print(" Ubuntu: sudo apt install ffmpeg")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
if result.returncode != 0:
|
|||
|
|
print("[ERROR] FFmpeg 烧录字幕失败")
|
|||
|
|
print(result.stderr[-2000:]) # 打印最后2000字符的错误信息
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
if os.path.isfile(output_path) and os.path.getsize(output_path) > 0:
|
|||
|
|
print(f"[OK] 字幕已烧录到视频：{output_path}")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
print(f"[ERROR] 输出视频未生成：{output_path}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
|
|||
|
|
def batch_process(srt_dir: str, output_dir: str, style: dict, burn_video: bool = False, video_dir: str = None):
|
|||
|
|
"""
|
|||
|
|
批量处理 SRT 文件
|
|||
|
|
|
|||
|
|
:param srt_dir: SRT 文件目录
|
|||
|
|
:param output_dir: 输出目录
|
|||
|
|
:param style: 样式字典
|
|||
|
|
:param burn_video: 是否烧录到视频
|
|||
|
|
:param video_dir: 视频文件目录（如果 burn_video=True）
|
|||
|
|
"""
|
|||
|
|
os.makedirs(output_dir, exist_ok=True)
|
|||
|
|
|
|||
|
|
for file in os.listdir(srt_dir):
|
|||
|
|
if file.lower().endswith(".srt"):
|
|||
|
|
srt_path = os.path.join(srt_dir, file)
|
|||
|
|
ass_output = os.path.join(output_dir, file.replace(".srt", ".ass"))
|
|||
|
|
|
|||
|
|
print(f"\n[INFO] 处理文件：{file}")
|
|||
|
|
ok = generate_ass_file(srt_path, ass_output, style)
|
|||
|
|
|
|||
|
|
if ok and burn_video and video_dir:
|
|||
|
|
# 查找对应的视频文件
|
|||
|
|
video_file = file.replace(".srt", ".mp4")
|
|||
|
|
video_path = os.path.join(video_dir, video_file)
|
|||
|
|
|
|||
|
|
if os.path.isfile(video_path):
|
|||
|
|
video_output = os.path.join(output_dir, video_file)
|
|||
|
|
burn_subtitles_with_ffmpeg(video_path, ass_output, video_output)
|
|||
|
|
else:
|
|||
|
|
print(f"[WARN] 未找到对应的视频文件：{video_file}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
parser = argparse.ArgumentParser(
|
|||
|
|
description="ASS Subtitle Renderer · ASS字幕渲染器",
|
|||
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|||
|
|
epilog="""
|
|||
|
|
示例：
|
|||
|
|
python ass-subtitle-renderer.py --srt input.srt --style reference-drama --output subtitles.ass
|
|||
|
|
python ass-subtitle-renderer.py --srt input.srt --video input.mp4 --output output.mp4 --style reference-drama
|
|||
|
|
python ass-subtitle-renderer.py --batch ./srt/ --style reference-drama --output-dir ./ass/
|
|||
|
|
"""
|
|||
|
|
)
|
|||
|
|
parser.add_argument("--srt", help="输入 SRT 文件路径")
|
|||
|
|
parser.add_argument("--video", help="输入视频路径（可选，用于直接烧录字幕）")
|
|||
|
|
parser.add_argument("--output", help="输出文件路径（ASS 或 MP4）")
|
|||
|
|
parser.add_argument("--style", default="reference-drama", help="样式名称或配置文件路径")
|
|||
|
|
parser.add_argument("--config", help="自定义样式配置文件路径（覆盖 --style）")
|
|||
|
|
parser.add_argument("--batch", help="批量处理 SRT 文件目录")
|
|||
|
|
parser.add_argument("--output-dir", help="批量处理输出目录")
|
|||
|
|
parser.add_argument("--burn-video", action="store_true", help="是否烧录字幕到视频")
|
|||
|
|
parser.add_argument("--video-dir", help="视频文件目录（配合 --burn-video 使用）")
|
|||
|
|
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
|
|||
|
|
# 加载样式配置
|
|||
|
|
if args.config:
|
|||
|
|
style = load_style_config(config_path=args.config)
|
|||
|
|
else:
|
|||
|
|
style = load_style_config(style_name=args.style)
|
|||
|
|
|
|||
|
|
if args.srt:
|
|||
|
|
# 单文件处理
|
|||
|
|
if not args.output:
|
|||
|
|
print("[ERROR] 请指定 --output")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
# 生成 ASS 文件
|
|||
|
|
ok = generate_ass_file(args.srt, args.output, style)
|
|||
|
|
if not ok:
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
# 如果指定了视频，则烧录字幕
|
|||
|
|
if args.video:
|
|||
|
|
video_output = args.output.replace(".ass", ".mp4")
|
|||
|
|
burn_ok = burn_subtitles_with_ffmpeg(args.video, args.output, video_output)
|
|||
|
|
if not burn_ok:
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
elif args.batch:
|
|||
|
|
# 批量处理
|
|||
|
|
if not args.output_dir:
|
|||
|
|
print("[ERROR] 批量处理需要指定 --output-dir")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
batch_process(args.batch, args.output_dir, style, args.burn_video, args.video_dir)
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
print("[ERROR] 请指定 --srt 或 --batch")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
print("\n[OK] 处理完成")
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|