KaraokeTestudaiBackend/subtitle_generator.py

68 lines
3.2 KiB
Python
Raw Normal View History

2024-12-17 21:14:21 +10:30
# Substation Alpha (ASS) generation
# Colour values are &HAABBGGRR, &HBBGGRR, or &HAA.
# Alpha is actually inverted, i.e. transparency - FF is transparent, 00 is opaque.
ass_preamble = '''[Script Info]
ScriptType: v4.00+
WrapStyle: 0
ScaledBorderAndShadow: yes
YCbCr Matrix: TV.709
PlayResX: {PlayResX}
PlayResY: {PlayResY}
[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
Style: Default,Arial,72,&H002A0A00,&H000019FF,&H00FFFFFF,&H00000000,0,0,0,0,100,100,0,0,1,2.5,0,8,30,30,25,1
Style: Kanji,Migu 1P,{KanjiSize},&H{KaraokeColourPast},&H{KaraokeColourFuture},&H00FFFFFF,&H00000000,0,0,0,0,100,100,0,0,1,4.0,0,2,30,30,25,1
Style: Furigana,Migu 1P,{FuriSize},&H{KaraokeColourPast},&H{KaraokeColourFuture},&H00FFFFFF,&H00000000,0,0,0,0,100,100,0,0,1,2.5,0,2,30,30,0,1
Style: Romaji,Migu 1P,{RomajiSize},&H{KaraokeColourPast},&H{KaraokeColourFuture},&H00FFFFFF,&H00000000,0,0,0,0,100,100,0,0,1,2.5,0,8,30,30,25,1
Style: Translation,Migu 1P,{TranslationSize},&H00FFFFFF,&H000019FF,&H00000000,&H00000000,0,1,0,0,100,100,0,0,1,1.0,3,8,30,30,25,1
[Events]
Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text
'''
format_defaults = {
'PlayResX': 1280,
'PlayResY': 720,
'TranslationSize': 48,
'RomajiSize': 60,
'KanjiSize': 72,
'FuriSize': 36,
'KaraokeColourFuture': '000019FF',
'KaraokeColourPast': 'E02A0A00',
}
from format import LyricTrack
def generate_ass(filename: str, lyric_track: LyricTrack, format_overloads: dict = None):
format_dict = format_defaults.copy()
if format_overloads:
format_dict.update(format_overloads)
preamble = ass_preamble.format(format_dict)
# Kanji Furigana layout stuff
size_kanji = format_dict['KanjiSize']
size_furi = format_dict['FuriSize']
with open(filename, 'w') as file:
file.write(preamble)
# for line in lines:
# #
# for syllable in line:
# t, kanji, furi, romaji = syllable
example_layout = '''
Dialogue: 0,0:01:08.00,0:01:26.00,Kanji,,,,,,{\k0}{\K100}{\K100}{\K100}{\K100}{\K100}{\K100}から{\K100}{\K100}って{\K100}{\K100}{\K100}{\K100} {\K100}{\K100}{\K100}{\K100}{\K100}{\K100}
Dialogue: 0,0:01:08.00,0:01:26.00,Furigana,, 0,1130,,,{\k0}{\K100}あめ
Dialogue: 0,0:01:08.00,0:01:26.00,Furigana,, 0, 900,,,{\k200}{\K100}ゆき
Dialogue: 0,0:01:08.00,0:01:26.00,Furigana,, 0, 700,,,{\k400}{\K100}てん
Dialogue: 0,0:01:08.00,0:01:26.00,Furigana,, 0, 370,,,{\k600}{\K100}
Dialogue: 0,0:01:08.00,0:01:26.00,Furigana,, 0, 0,,,{\k800}{\K100}
Dialogue: 0,0:01:08.00,0:01:26.00,Furigana,, 260, 0,,,{\k1000}{\K100}うるお
Dialogue: 0,0:01:08.00,0:01:26.00,Furigana,, 570, 0,,,{\k1200}{\K100}
Dialogue: 0,0:01:08.00,0:01:26.00,Furigana,, 800, 0,,,{\k1400}{\K100}
Dialogue: 0,0:01:08.00,0:01:26.00,Translation,,,,,,Rain and snow fall from the heavens. Moisten the earth and make it sprout
Dialogue: 0,0:01:08.00,0:01:26.00,Romaji,,,,,,{\K0}{\K100}ame {\K100}ya {\K100}yuki {\K100}ga {\K100}ten {\K100}kara {\K100}fu{\K100}tte {\K100}chi {\K100}wo {\K100}uruo{\K100}shi {\K100}me {\K100}wo {\K100}da{\K100}sa {\K100}se{\K100}ru
'''