File size: 3,259 Bytes
3e3d7b1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 | from __future__ import annotations
import html
from dataclasses import dataclass
from pathlib import Path
from renderer.templates import get_template
@dataclass
class SubtitleEvent:
start: float
end: float
text: str
class SubtitleGenerator:
def from_scenes(self, scenes: list, total_duration: float | None = None) -> list[SubtitleEvent]:
events: list[SubtitleEvent] = []
for scene in scenes:
if scene.caption:
events.append(SubtitleEvent(scene.start, scene.start + scene.duration, scene.caption))
if not events and total_duration:
events.append(SubtitleEvent(0, total_duration, ""))
return events
def write_srt(self, events: list[SubtitleEvent], output: Path) -> Path:
lines: list[str] = []
for idx, event in enumerate(events, start=1):
lines.extend([str(idx), f"{_srt_time(event.start)} --> {_srt_time(event.end)}", event.text, ""])
output.write_text("\n".join(lines), encoding="utf-8")
return output
def write_ass(self, events: list[SubtitleEvent], output: Path, template_key: str) -> Path:
template = get_template(template_key)
body = [
"[Script Info]",
"ScriptType: v4.00+",
"PlayResX: 1080",
"PlayResY: 1920",
"",
"[V4+ Styles]",
"Format: Name,Fontname,Fontsize,PrimaryColour,SecondaryColour,OutlineColour,BackColour,"
"Bold,Italic,Underline,StrikeOut,ScaleX,ScaleY,Spacing,Angle,BorderStyle,Outline,Shadow,"
"Alignment,MarginL,MarginR,MarginV,Encoding",
template.ass_style(),
"",
"[Events]",
"Format: Layer,Start,End,Style,Name,MarginL,MarginR,MarginV,Effect,Text",
]
for event in events:
text = _ass_escape(event.text)
if template.effect == "karaoke":
text = _karaoke_text(text, event.end - event.start)
elif template.effect == "zoom":
text = r"{\t(0,180,\fscx115\fscy115)\t(180,360,\fscx100\fscy100)}" + text
elif template.effect == "bounce":
text = r"{\t(0,120,\frz-2)\t(120,240,\frz2)\t(240,360,\frz0)}" + text
body.append(f"Dialogue: 0,{_ass_time(event.start)},{_ass_time(event.end)},Default,,0,0,0,,{text}")
output.write_text("\n".join(body), encoding="utf-8")
return output
def _karaoke_text(text: str, duration: float) -> str:
words = text.split()
if not words:
return text
centiseconds = max(1, int(duration * 100 / len(words)))
return "".join(f"{{\\k{centiseconds}}}{word} " for word in words).strip()
def _srt_time(seconds: float) -> str:
ms = int(round(seconds * 1000))
h, rem = divmod(ms, 3600000)
m, rem = divmod(rem, 60000)
s, ms = divmod(rem, 1000)
return f"{h:02d}:{m:02d}:{s:02d},{ms:03d}"
def _ass_time(seconds: float) -> str:
cs = int(round(seconds * 100))
h, rem = divmod(cs, 360000)
m, rem = divmod(rem, 6000)
s, cs = divmod(rem, 100)
return f"{h}:{m:02d}:{s:02d}.{cs:02d}"
def _ass_escape(text: str) -> str:
return html.escape(text).replace("\n", r"\N").replace("{", r"\{").replace("}", r"\}")
|