pandora_t_for_time/render_utils.py

import re
import os

import ox
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