# 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
'''