vrsub/scripts/unify_ass_style.py

"""统一历史 VR 双目字幕（*.CN_dual_eye.ass）的样式与位置。

背景
----
`nodes/ass.py` 生成的字幕样式在迭代中多次变化，媒体库（如 /mnt/fnOS/123）
里由旧版本批量生成的 *.CN_dual_eye.ass 因此混有多种历史样式：

- 最早期：底部对齐（an2）+ 实心白字（&H00FFFFFF），MarginV≈1200；
- 过渡期：底部对齐 + 半透明白（&H80FFFFFF），MarginV≈1020；
- 2026-09 新默认：顶部对齐（an8）+ 70% 透明（&HB3FFFFFF）+ MarginV=120；
- 用户实测合适值：顶部对齐 + 70% 透明 + MarginV=700。

本脚本把库内所有目标字幕**原地改写**为当前 nodes/ass.py 的统一样式
（an8 顶部对齐 + 70% 透明 + DEFAULT_MARGIN_TOP=700，单点定义见
nodes/ass.py 顶部常量），保证"之后生成的字幕"与"历史字幕"外观一致。

实现要点
--------
- 不对手写字符串做手术，而是解析旧文件的 [Script Info] 分辨率与全部
  Dialogue 事件后，用 nodes/ass.py 的 ass_header()/dialogue_line()
  重新拼装——与代码新生成产物共用同一出口，两边永不漂移；以后要调样式
  只需改 nodes/ass.py 一处常量。
- 事件文本原样保留（含原有 \\N 多行分隔与事件内其它标签），只规整前导
  \\an 对齐标签：历史 an2（底部）/an8（顶部）一律统一为 an8。
- 仅改写样式声明与事件对齐标签，不触碰时间轴、事件文本与左右眼对称性。
- 文件最终以 UTF-8 写回、行尾 \\n、末尾一个换行——与 write_ass() 一致。

用法（在 vrsub 根目录执行）
----------------------------
    # 预览（默认 dry-run，不改盘）
    uv run python scripts/unify_ass_style.py /mnt/fnOS/123

    # 真正原地改写
    uv run python scripts/unify_ass_style.py /mnt/fnOS/123 --apply

    # 指定递归匹配模式 / 只处理某旧 MarginV
    uv run python scripts/unify_ass_style.py /mnt/fnOS/123 --pattern "*.ass"
    uv run python scripts/unify_ass_style.py /mnt/fnOS/123 --only-margin 1200

安全说明
--------
- 默认 --dry-run 只打印不写入；--apply 才真正原地改写。
- 改写前脚本先打印库内旧样式分布，便于核对范围。
- 不产生 .bak 副本（用户已确认直接原地改写）；如需回滚请先自行备份。
"""

from __future__ import annotations

import argparse
import re
import sys
from collections import Counter
from pathlib import Path

# 复用节点单一事实来源：统一脚本必须与"新生成字幕"完全一致。
from nodes.ass import (  # noqa: E402
    DEFAULT_MARGIN_TOP,
    ass_header,
    dialogue_line,
)

_DIALOGUE_RE = re.compile(r"^Dialogue: ")
_AN_TAG_RE = re.compile(r"\{\\an[0-9]\}")


def parse_resolution(text: str) -> tuple[int, int] | None:
    """从 [Script Info] 解析 PlayResX/PlayResY；解析失败返回 None。"""
    width = height = None
    for line in text.splitlines():
        m = re.match(r"^\s*PlayResX:\s*(\d+)", line)
        if m:
            width = int(m.group(1))
            continue
        m = re.match(r"^\s*PlayResY:\s*(\d+)", line)
        if m:
            height = int(m.group(1))
    if width is None or height is None:
        return None
    return width, height


def inspect_style(text: str) -> tuple[str, str] | None:
    """提取旧样式 (PrimaryColour, MarginV) 用于统计/过滤。

    只有存在 "Style: LeftEye" 行才视为 VR 双眼字幕产物；普通 ASS 字幕
    （无左右眼样式）跳过，避免误伤。"""
    for line in text.splitlines():
        if line.startswith("Style: LeftEye"):
            fields = line.split(",")
            # 字段：Name0,Fontname1,Fontsize2,Primary3,...,Alignment18,MarginL19,MarginR20,MarginV21
            primary = fields[3].strip() if len(fields) > 3 else "?"
            margin_v = fields[21].strip() if len(fields) > 21 else "?"
            return primary, margin_v
    return None


def parse_dialogues(text: str) -> list[tuple[str, str, str, str]]:
    """抽取全部 Dialogue 事件为 (开始, 结束, 样式, 文本)。

    历史文件每条可能带前导 {\an2}/{\an8} 对齐标签；这里剥掉（统一由
    dialogue_line 补 an8），文本其余内容原样保留。"""
    entries: list[tuple[str, str, str, str]] = []
    for line in text.splitlines():
        if not _DIALOGUE_RE.match(line):
            continue
        # Dialogue: 0,Start,End,Style,Name,MarginL,MarginR,MarginV,Effect,Text
        parts = line.split(",", 9)
        if len(parts) < 10:
            continue
        text_field = _AN_TAG_RE.sub("", parts[9], count=1)
        entries.append((parts[1], parts[2], parts[3], text_field))
    return entries


def rewrite_content(original: str) -> str | None:
    """把整份 ASS 文本重建为统一样式；已符合目标样式时返回 None。

    返回的新文本 = ass_header()（新样式行，MarginV=DEFAULT_MARGIN_TOP）
    + dialogue_line() 重建的全部事件，即与 write_ass() 产物逐字节一致
    （除事件文本本身）。"""
    res = parse_resolution(original)
    if res is None:
        return None  # 无 PlayResX/Y，非本工具产物
    width, height = res
    if inspect_style(original) is None:
        return None  # 无 LeftEye 样式行，非 VR 双目字幕

    lines = [ass_header(width, height, margin_top=DEFAULT_MARGIN_TOP).rstrip("\n")]
    for start, end, style, text in parse_dialogues(original):
        lines.append(dialogue_line(style, start, end, text))
    rewritten = "\n".join(lines) + "\n"
    return None if rewritten == original else rewritten


def main() -> int:
    parser = argparse.ArgumentParser(description="统一历史 VR 双目字幕样式与位置")
    parser.add_argument("root", help="要扫描的媒体库根目录（递归）")
    parser.add_argument(
        "--pattern",
        default="*.CN_dual_eye.ass",
        help="递归匹配的文件名模式（默认 *.CN_dual_eye.ass）",
    )
    parser.add_argument(
        "--apply",
        action="store_true",
        help="真正原地改写文件；不带此参数等价于 dry-run 预览",
    )
    parser.add_argument(
        "--only-margin",
        type=int,
        default=None,
        help="只处理旧样式 MarginV 等于该值的文件（如 1200），其余跳过",
    )
    args = parser.parse_args()

    root = Path(args.root)
    if not root.is_dir():
        print(f"错误：目录不存在或不可读: {root}", file=sys.stderr)
        return 2

    files = sorted(root.rglob(args.pattern))
    if not files:
        print(f"未找到匹配 {args.pattern} 的文件")
        return 0

    # 第一遍统计：库内旧样式分布 + 圈定目标文件。
    distribution: Counter[tuple[str, str]] = Counter()
    unreadable = 0
    target_files: list[Path] = []
    for p in files:
        try:
            text = p.read_text(encoding="utf-8")
        except (UnicodeDecodeError, OSError):
            unreadable += 1
            continue
        styled = inspect_style(text)
        if styled is None:
            unreadable += 1
            continue
        primary, margin = styled
        try:
            margin_i = int(margin)
        except ValueError:
            margin_i = -1
        if args.only_margin is not None and margin_i != args.only_margin:
            continue
        distribution[(primary, margin)] += 1
        target_files.append(p)

    print(f"扫描 {root}：共 {len(files)} 个匹配文件，可识别 VR 字幕 {len(target_files)}，跳过 {unreadable}")
    print("旧样式分布（PrimaryColour, MarginV）：")
    for (primary, margin), count in distribution.most_common():
        print(f"  {count:4d}  {primary}  MarginV={margin}")
    if not target_files:
        print("没有需要统一的文件。")
        return 0

    print(f"\n目标样式：an8 顶部对齐 + 70%透明(&HB3FFFFFF) + MarginV={DEFAULT_MARGIN_TOP}（与代码新生成一致）")

    changed = failed = 0
    for p in target_files:
        try:
            original = p.read_text(encoding="utf-8")
            rewritten = rewrite_content(original)
        except Exception as exc:  # 单文件失败不中断整体
            print(f"  [错误] {p}: {exc}")
            failed += 1
            continue
        if rewritten is None:
            continue  # 已是目标样式，跳过
        changed += 1
        marker = "改写" if args.apply else "将改写"
        if args.apply:
            p.write_text(rewritten, encoding="utf-8")
        # 摘要：旧样式 -> 新样式。
        styled = inspect_style(original)
        old_desc = f"{styled[0]}@{styled[1]}" if styled else "?"
        print(f"  [{marker}] {p.relative_to(root)}  ({old_desc} -> &HB3FFFFFF@{DEFAULT_MARGIN_TOP})")

    print(f"\n完成：{changed} 个文件{'已' if args.apply else '待'}统一，{failed} 个失败跳过，其余已是最新样式。")
    if not args.apply:
        print("（以上为预览，未改动任何文件；确认无误后加 --apply 执行真正改写）")
    return 0


if __name__ == "__main__":
    raise SystemExit(main())