154 lines
4.3 KiB
Python
154 lines
4.3 KiB
Python
import re
|
|
import os
|
|
|
|
import lxml.etree
|
|
|
|
from .render_kdenlive import melt_xml
|
|
|
|
def parse_lang(lang):
|
|
if lang and "," in lang:
|
|
lang = lang.split(',')
|
|
if isinstance(lang, list):
|
|
tlang = lang[1:]
|
|
lang = lang[0]
|
|
else:
|
|
tlang = None
|
|
if lang == "en":
|
|
lang = None
|
|
return lang, tlang
|
|
|
|
def random_int(seq, length):
|
|
n = n_ = length - 1
|
|
#print('len', n)
|
|
if n == 0:
|
|
return n
|
|
r = seq() / 9 * 10
|
|
base = 10
|
|
while n > 10:
|
|
n /= 10
|
|
r += seq() / 9 * 10
|
|
base += 10
|
|
r = int(round(n_ * r / base))
|
|
return r
|
|
|
|
def random_choice(seq, items, pop=False):
|
|
n = random_int(seq, len(items))
|
|
if pop:
|
|
return items.pop(n)
|
|
return items[n]
|
|
|
|
def chance(seq, chance):
|
|
return (seq() / 10) < chance
|
|
|
|
def get_clip_by_seqid(clips, seqid):
|
|
selected = None
|
|
for i, clip in enumerate(clips):
|
|
if clip['seqid'] == seqid:
|
|
selected = i
|
|
break
|
|
if selected is not None:
|
|
return clips.pop(i)
|
|
return None
|
|
|
|
def get_scene_duration(scene):
|
|
if isinstance(scene, str):
|
|
with open(scene) as fd:
|
|
scene = json.load(fd)
|
|
duration = 0
|
|
for key, value in scene.items():
|
|
for name, clips in value.items():
|
|
for clip in clips:
|
|
duration += int(clip['duration'] * 24)
|
|
return duration / 24
|
|
|
|
def get_offset_duration(prefix):
|
|
duration = 0
|
|
for root, folders, files in os.walk(prefix):
|
|
for f in files:
|
|
if f == 'scene.json':
|
|
duration += get_scene_duration(scene)
|
|
return duration
|
|
|
|
def get_track_duration(scene, k, n):
|
|
duration = 0
|
|
for key, value in scene.items():
|
|
if key == k:
|
|
for name, clips in value.items():
|
|
if name == n:
|
|
for clip in clips:
|
|
duration += int(clip['duration'] * 24)
|
|
return duration / 24
|
|
|
|
def get_project_duration(file):
|
|
out = melt_xml(file)
|
|
chain = lxml.etree.fromstring(out).xpath('producer')[0]
|
|
duration = int(chain.attrib['out']) + 1
|
|
return duration
|
|
|
|
def fix_overlaps(data):
|
|
previous = None
|
|
for sub in data:
|
|
if previous is None:
|
|
previous = sub
|
|
else:
|
|
if sub['in'] < previous['out']:
|
|
previous['out'] = sub['in'] - 0.001
|
|
previous = sub
|
|
return data
|
|
|
|
def shift_clips(data, offset):
|
|
for clip in data:
|
|
clip['in'] += offset
|
|
clip['out'] += offset
|
|
return data
|
|
|
|
def ass_encode(subs, options):
|
|
if "lang" in options:
|
|
langs = options["lang"].split(',')
|
|
else:
|
|
langs = list(subs[0]["values"])
|
|
#print('ass_encode', langs, options)
|
|
#print(subs)
|
|
|
|
header = '''[Script Info]
|
|
ScriptType: v4.00+
|
|
PlayResX: 1920
|
|
PlayResY: 1080
|
|
ScaledBorderAndShadow: yes
|
|
YCbCr Matrix: None
|
|
|
|
[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
|
|
'''
|
|
ass = header
|
|
offset = options.get("sub_margin", 10)
|
|
spacing = options.get("sub_spacing", 20)
|
|
height = 42
|
|
styles = []
|
|
for lang in reversed(langs):
|
|
if isinstance(options.get("font"), list) and lang in options["font"]:
|
|
font = options["font"][lang]
|
|
else:
|
|
font = 'SimHei' if lang in ('zh', 'jp') else 'Menlo'
|
|
if isinstance(options.get("font_size"), list) and lang in options["font_size"]:
|
|
size = options["font_size"][lang]
|
|
else:
|
|
size = 46 if font == 'SimHei' else 42
|
|
|
|
styles.append(
|
|
f'Style: {lang},{font},{size},&Hffffff,&Hffffff,&H0,&H0,0,0,0,0,100,100,0,0,1,1,0,2,10,10,{offset},1'
|
|
)
|
|
offset += size + spacing
|
|
ass += '\n'.join(reversed(styles)) + '\n'
|
|
events = [
|
|
'Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text'
|
|
]
|
|
for sub in subs:
|
|
start = ox.format_timecode(sub['in']).rstrip('0')
|
|
stop = ox.format_timecode(sub['out']).rstrip('0')
|
|
for lang in reversed(langs):
|
|
value = sub['values'][lang]
|
|
event = f'Dialogue: 0,{start},{stop},{lang},,0,0,0,,{value}'
|
|
events.append(event)
|
|
ass += '\n\n[Events]\n' + '\n'.join(events) + '\n'
|
|
return ass
|