auto-editor 28.1.0__py3-none-any.whl → 29.0.0__py3-none-any.whl

auto_editor/exports/kdenlive.py

@@ -1,322 +0,0 @@
-import json
-import xml.etree.ElementTree as ET
-from os import getcwd
-from uuid import uuid4
-
-from auto_editor.timeline import Clip, v3
-from auto_editor.utils.func import aspect_ratio, to_timecode
-
-"""
-kdenlive uses the MLT timeline format
-
-See docs here:
-https://mltframework.org/docs/mltxml/
-
-kdenlive specifics:
-https://github.com/KDE/kdenlive/blob/master/dev-docs/fileformat.md
-"""
-
-
-def kdenlive_write(output: str, tl: v3) -> None:
-    mlt = ET.Element(
-        "mlt",
-        attrib={
-            "LC_NUMERIC": "C",
-            "version": "7.22.0",
-            "producer": "main_bin",
-            "root": f"{getcwd()}",
-        },
-    )
-
-    width, height = tl.res
-    num, den = aspect_ratio(width, height)
-    tb = tl.tb
-    seq_uuid = uuid4()
-
-    ET.SubElement(
-        mlt,
-        "profile",
-        attrib={
-            "description": "automatic",
-            "width": f"{width}",
-            "height": f"{height}",
-            "progressive": "1",
-            "sample_aspect_num": "1",
-            "sample_aspect_den": "1",
-            "display_aspect_num": f"{num}",
-            "display_aspect_den": f"{den}",
-            "frame_rate_num": f"{tb.numerator}",
-            "frame_rate_den": f"{tb.denominator}",
-            "colorspace": "709",
-        },
-    )
-
-    # Reserved producer0
-    global_out = to_timecode(len(tl) / tb, "standard")
-    producer = ET.SubElement(mlt, "producer", id="producer0")
-    ET.SubElement(producer, "property", name="length").text = global_out
-    ET.SubElement(producer, "property", name="eof").text = "continue"
-    ET.SubElement(producer, "property", name="resource").text = "black"
-    ET.SubElement(producer, "property", name="mlt_service").text = "color"
-    ET.SubElement(producer, "property", name="kdenlive:playlistid").text = "black_track"
-    ET.SubElement(producer, "property", name="mlt_image_format").text = "rgba"
-    ET.SubElement(producer, "property", name="aspect_ratio").text = "1"
-
-    # Get all clips
-    if tl.v:
-        clips = [clip for clip in tl.v[0] if isinstance(clip, Clip)]
-    elif tl.a:
-        clips = tl.a[0]
-    else:
-        clips = []
-
-    source_ids = {}
-    source_id = 4
-    clip_playlists = []
-    chains = 0
-    playlists = 0
-    producers = 1
-    a_channels = len(tl.a)
-    v_channels = len(tl.v)
-    warped_clips = [i for i, clip in enumerate(clips) if clip.speed != 1]
-
-    # create all producers for warped clips
-    for clip_idx in warped_clips:
-        for i in range(a_channels + v_channels):
-            clip = clips[clip_idx]
-            path = str(clip.src.path)
-
-            if path not in source_ids:
-                source_ids[path] = str(source_id)
-                source_id += 1
-
-            prod = ET.SubElement(
-                mlt,
-                "producer",
-                attrib={
-                    "id": f"producer{producers}",
-                    "in": "00:00:00.000",
-                    "out": global_out,
-                },
-            )
-            ET.SubElement(
-                prod, "property", name="resource"
-            ).text = f"{clip.speed}:{path}"
-            ET.SubElement(prod, "property", name="warp_speed").text = str(clip.speed)
-            ET.SubElement(prod, "property", name="warp_resource").text = path
-            ET.SubElement(prod, "property", name="warp_pitch").text = "0"
-            ET.SubElement(prod, "property", name="mlt_service").text = "timewarp"
-            ET.SubElement(prod, "property", name="kdenlive:id").text = source_ids[path]
-
-            if i < a_channels:
-                ET.SubElement(prod, "property", name="vstream").text = "0"
-                ET.SubElement(prod, "property", name="astream").text = str(
-                    a_channels - 1 - i
-                )
-                ET.SubElement(prod, "property", name="set.test_audio").text = "0"
-                ET.SubElement(prod, "property", name="set.test_video").text = "1"
-            else:
-                ET.SubElement(prod, "property", name="vstream").text = str(
-                    v_channels - 1 - (i - a_channels)
-                )
-                ET.SubElement(prod, "property", name="astream").text = "0"
-                ET.SubElement(prod, "property", name="set.test_audio").text = "1"
-                ET.SubElement(prod, "property", name="set.test_video").text = "0"
-
-            producers += 1
-
-    # create chains, playlists and tractors for audio channels
-    for i, audio in enumerate(tl.a):
-        path = str(audio[0].src.path)
-
-        if path not in source_ids:
-            source_ids[path] = str(source_id)
-            source_id += 1
-
-        chain = ET.SubElement(mlt, "chain", attrib={"id": f"chain{chains}"})
-        ET.SubElement(chain, "property", name="resource").text = path
-        ET.SubElement(
-            chain, "property", name="mlt_service"
-        ).text = "avformat-novalidate"
-        ET.SubElement(chain, "property", name="vstream").text = "0"
-        ET.SubElement(chain, "property", name="astream").text = str(a_channels - 1 - i)
-        ET.SubElement(chain, "property", name="set.test_audio").text = "0"
-        ET.SubElement(chain, "property", name="set.test_video").text = "1"
-        ET.SubElement(chain, "property", name="kdenlive:id").text = source_ids[path]
-
-        for _i in range(2):
-            playlist = ET.SubElement(mlt, "playlist", id=f"playlist{playlists}")
-            clip_playlists.append(playlist)
-            ET.SubElement(playlist, "property", name="kdenlive:audio_track").text = "1"
-            playlists += 1
-
-        tractor = ET.SubElement(
-            mlt,
-            "tractor",
-            attrib={"id": f"tractor{chains}", "in": "00:00:00.000", "out": global_out},
-        )
-        ET.SubElement(tractor, "property", name="kdenlive:audio_track").text = "1"
-        ET.SubElement(tractor, "property", name="kdenlive:timeline_active").text = "1"
-        ET.SubElement(tractor, "property", name="kdenlive:audio_rec")
-        ET.SubElement(
-            tractor,
-            "track",
-            attrib={"hide": "video", "producer": f"playlist{playlists - 2}"},
-        )
-        ET.SubElement(
-            tractor,
-            "track",
-            attrib={"hide": "video", "producer": f"playlist{playlists - 1}"},
-        )
-        chains += 1
-
-    # create chains, playlists and tractors for video channels
-    for i, video in enumerate(tl.v):
-        path = f"{video[0].src.path}"  # type: ignore
-
-        if path not in source_ids:
-            source_ids[path] = str(source_id)
-            source_id += 1
-
-        chain = ET.SubElement(mlt, "chain", attrib={"id": f"chain{chains}"})
-        ET.SubElement(chain, "property", name="resource").text = path
-        ET.SubElement(
-            chain, "property", name="mlt_service"
-        ).text = "avformat-novalidate"
-        ET.SubElement(chain, "property", name="vstream").text = str(v_channels - 1 - i)
-        ET.SubElement(chain, "property", name="astream").text = "0"
-        ET.SubElement(chain, "property", name="set.test_audio").text = "1"
-        ET.SubElement(chain, "property", name="set.test_video").text = "0"
-        ET.SubElement(chain, "property", name="kdenlive:id").text = source_ids[path]
-
-        for _i in range(2):
-            playlist = ET.SubElement(mlt, "playlist", id=f"playlist{playlists}")
-            clip_playlists.append(playlist)
-            playlists += 1
-
-        tractor = ET.SubElement(
-            mlt,
-            "tractor",
-            attrib={"id": f"tractor{chains}", "in": "00:00:00.000", "out": global_out},
-        )
-        ET.SubElement(tractor, "property", name="kdenlive:timeline_active").text = "1"
-        ET.SubElement(
-            tractor,
-            "track",
-            attrib={"hide": "audio", "producer": f"playlist{playlists - 2}"},
-        )
-        ET.SubElement(
-            tractor,
-            "track",
-            attrib={"hide": "audio", "producer": f"playlist{playlists - 1}"},
-        )
-        chains += 1
-
-    # final chain for the project bin
-    path = str(clips[0].src.path)
-    chain = ET.SubElement(mlt, "chain", attrib={"id": f"chain{chains}"})
-    ET.SubElement(chain, "property", name="resource").text = path
-    ET.SubElement(chain, "property", name="mlt_service").text = "avformat-novalidate"
-    ET.SubElement(chain, "property", name="audio_index").text = "1"
-    ET.SubElement(chain, "property", name="video_index").text = "0"
-    ET.SubElement(chain, "property", name="vstream").text = "0"
-    ET.SubElement(chain, "property", name="astream").text = "0"
-    ET.SubElement(chain, "property", name="kdenlive:id").text = source_ids[path]
-
-    groups = []
-    group_counter = 0
-    producers = 1
-
-    for clip in clips:
-        group_children: list[object] = []
-        _in = to_timecode(clip.offset / tb, "standard")
-        _out = to_timecode((clip.offset + clip.dur) / tb, "standard")
-        path = str(clip.src.path)
-
-        for i, playlist in enumerate(clip_playlists[::2]):
-            # adding 1 extra frame for each previous group to the start time works but feels hacky?
-            group_children.append(
-                {
-                    "data": f"{i}:{clip.start + group_counter}",
-                    "leaf": "clip",
-                    "type": "Leaf",
-                }
-            )
-            clip_prod = ""
-
-            if clip.speed == 1:
-                clip_prod = f"chain{i}"
-            else:
-                clip_prod = f"producer{producers}"
-                producers += 1
-
-            entry = ET.SubElement(
-                playlist,
-                "entry",
-                attrib={"producer": f"{clip_prod}", "in": _in, "out": _out},
-            )
-            ET.SubElement(entry, "property", name="kdenlive:id").text = source_ids[path]
-
-        groups.append({"children": group_children[:], "type": "Normal"})
-        group_counter += 1
-
-    # default sequence tractor
-    sequence = ET.SubElement(
-        mlt,
-        "tractor",
-        attrib={"id": f"{{{seq_uuid}}}", "in": "00:00:00.000", "out": "00:00:00.000"},
-    )
-    ET.SubElement(sequence, "property", name="kdenlive:uuid").text = f"{{{seq_uuid}}}"
-    ET.SubElement(sequence, "property", name="kdenlive:clipname").text = "Sequence 1"
-    ET.SubElement(
-        sequence, "property", name="kdenlive:sequenceproperties.groups"
-    ).text = json.dumps(groups, indent=4)
-    ET.SubElement(sequence, "track", producer="producer0")
-
-    for i in range(chains):
-        ET.SubElement(sequence, "track", producer=f"tractor{i}")
-
-    # main bin
-    playlist_bin = ET.SubElement(mlt, "playlist", id="main_bin")
-    ET.SubElement(
-        playlist_bin, "property", name="kdenlive:docproperties.uuid"
-    ).text = f"{{{seq_uuid}}}"
-    ET.SubElement(
-        playlist_bin, "property", name="kdenlive:docproperties.version"
-    ).text = "1.1"
-    ET.SubElement(playlist_bin, "property", name="xml_retain").text = "1"
-    ET.SubElement(
-        playlist_bin,
-        "entry",
-        attrib={
-            "producer": f"{{{seq_uuid}}}",
-            "in": "00:00:00.000",
-            "out": "00:00:00.000",
-        },
-    )
-    ET.SubElement(
-        playlist_bin,
-        "entry",
-        attrib={"producer": f"chain{chains}", "in": "00:00:00.000"},
-    )
-
-    # reserved last tractor for project
-    tractor = ET.SubElement(
-        mlt,
-        "tractor",
-        attrib={"id": f"tractor{chains}", "in": "00:00:00.000", "out": global_out},
-    )
-    ET.SubElement(tractor, "property", name="kdenlive:projectTractor").text = "1"
-    ET.SubElement(
-        tractor,
-        "track",
-        attrib={"producer": f"{{{seq_uuid}}}", "in": "00:00:00.000", "out": global_out},
-    )
-    tree = ET.ElementTree(mlt)
-
-    ET.indent(tree, space="\t", level=0)
-
-    if output == "-":
-        print(ET.tostring(mlt, encoding="unicode"))
-    else:
-        tree.write(output, xml_declaration=True, encoding="utf-8")

auto_editor/exports/shotcut.py

@@ -1,147 +0,0 @@
-import xml.etree.ElementTree as ET
-
-from auto_editor.timeline import Clip, v3
-from auto_editor.utils.func import aspect_ratio, to_timecode
-
-"""
-Shotcut uses the MLT timeline format
-
-See docs here:
-https://mltframework.org/docs/mltxml/
-
-"""
-
-
-def shotcut_write_mlt(output: str, tl: v3) -> None:
-    mlt = ET.Element(
-        "mlt",
-        attrib={
-            "LC_NUMERIC": "C",
-            "version": "7.9.0",
-            "title": "Shotcut version 22.09.23",
-            "producer": "main_bin",
-        },
-    )
-
-    width, height = tl.res
-    num, den = aspect_ratio(width, height)
-    tb = tl.tb
-
-    ET.SubElement(
-        mlt,
-        "profile",
-        attrib={
-            "description": "automatic",
-            "width": f"{width}",
-            "height": f"{height}",
-            "progressive": "1",
-            "sample_aspect_num": "1",
-            "sample_aspect_den": "1",
-            "display_aspect_num": f"{num}",
-            "display_aspect_den": f"{den}",
-            "frame_rate_num": f"{tb.numerator}",
-            "frame_rate_den": f"{tb.denominator}",
-            "colorspace": "709",
-        },
-    )
-
-    playlist_bin = ET.SubElement(mlt, "playlist", id="main_bin")
-    ET.SubElement(playlist_bin, "property", name="xml_retain").text = "1"
-
-    global_out = to_timecode(len(tl) / tb, "standard")
-
-    producer = ET.SubElement(mlt, "producer", id="bg")
-
-    ET.SubElement(producer, "property", name="length").text = global_out
-    ET.SubElement(producer, "property", name="eof").text = "pause"
-    ET.SubElement(producer, "property", name="resource").text = "#000"  # background
-    ET.SubElement(producer, "property", name="mlt_service").text = "color"
-    ET.SubElement(producer, "property", name="mlt_image_format").text = "rgba"
-    ET.SubElement(producer, "property", name="aspect_ratio").text = "1"
-
-    playlist = ET.SubElement(mlt, "playlist", id="background")
-    ET.SubElement(
-        playlist,
-        "entry",
-        attrib={"producer": "bg", "in": "00:00:00.000", "out": global_out},
-    ).text = "1"
-
-    chains = 0
-    producers = 0
-
-    if tl.v:
-        clips = [clip for clip in tl.v[0] if isinstance(clip, Clip)]
-    elif tl.a:
-        clips = tl.a[0]
-    else:
-        clips = []
-
-    for clip in clips:
-        src = clip.src
-        length = to_timecode((clip.offset + clip.dur) / tb, "standard")
-
-        if clip.speed == 1:
-            resource = f"{src.path}"
-            caption = f"{src.path.stem}"
-            chain = ET.SubElement(
-                mlt, "chain", attrib={"id": f"chain{chains}", "out": length}
-            )
-        else:
-            chain = ET.SubElement(
-                mlt, "producer", attrib={"id": f"producer{producers}", "out": length}
-            )
-            resource = f"{clip.speed}:{src.path}"
-            caption = f"{src.path.stem} ({clip.speed}x)"
-
-            producers += 1
-
-        ET.SubElement(chain, "property", name="length").text = length
-        ET.SubElement(chain, "property", name="resource").text = resource
-
-        if clip.speed != 1:
-            ET.SubElement(chain, "property", name="warp_speed").text = f"{clip.speed}"
-            ET.SubElement(chain, "property", name="warp_pitch").text = "1"
-            ET.SubElement(chain, "property", name="mlt_service").text = "timewarp"
-
-        ET.SubElement(chain, "property", name="caption").text = caption
-
-        chains += 1
-
-    main_playlist = ET.SubElement(mlt, "playlist", id="playlist0")
-    ET.SubElement(main_playlist, "property", name="shotcut:video").text = "1"
-    ET.SubElement(main_playlist, "property", name="shotcut:name").text = "V1"
-
-    producers = 0
-    for i, clip in enumerate(clips):
-        _in = to_timecode(clip.offset / tb, "standard")
-        _out = to_timecode((clip.offset + clip.dur) / tb, "standard")
-
-        tag_name = f"chain{i}"
-        if clip.speed != 1:
-            tag_name = f"producer{producers}"
-            producers += 1
-
-        ET.SubElement(
-            main_playlist,
-            "entry",
-            attrib={"producer": tag_name, "in": _in, "out": _out},
-        )
-
-    tractor = ET.SubElement(
-        mlt,
-        "tractor",
-        attrib={"id": "tractor0", "in": "00:00:00.000", "out": global_out},
-    )
-    ET.SubElement(tractor, "property", name="shotcut").text = "1"
-    ET.SubElement(tractor, "property", name="shotcut:projectAudioChannels").text = "2"
-    ET.SubElement(tractor, "track", producer="background")
-    ET.SubElement(tractor, "track", producer="playlist0")
-
-    tree = ET.ElementTree(mlt)
-
-    ET.indent(tree, space="\t", level=0)
-
-    if output == "-":
-        print(ET.tostring(mlt, encoding="unicode"))
-    else:
-        tree.write(output, xml_declaration=True, encoding="utf-8")

auto_editor/ffwrapper.py

@@ -1,187 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass
-from fractions import Fraction
-from pathlib import Path
-
-import av
-
-from auto_editor.utils.log import Log
-
-
-def mux(input: Path, output: Path, stream: int) -> None:
-    input_container = av.open(input, "r")
-    output_container = av.open(output, "w")
-
-    input_audio_stream = input_container.streams.audio[stream]
-    output_audio_stream = output_container.add_stream("pcm_s16le")
-
-    for frame in input_container.decode(input_audio_stream):
-        output_container.mux(output_audio_stream.encode(frame))
-
-    output_container.mux(output_audio_stream.encode(None))
-
-    output_container.close()
-    input_container.close()
-
-
-@dataclass(slots=True, frozen=True)
-class VideoStream:
-    width: int
-    height: int
-    codec: str
-    fps: Fraction
-    duration: float
-    sar: Fraction
-    time_base: Fraction | None
-    pix_fmt: str | None
-    color_range: int
-    color_space: int
-    color_primaries: int
-    color_transfer: int
-    bitrate: int
-    lang: str | None
-
-
-@dataclass(slots=True, frozen=True)
-class AudioStream:
-    codec: str
-    samplerate: int
-    layout: str
-    channels: int
-    duration: float
-    bitrate: int
-    lang: str | None
-
-
-@dataclass(slots=True, frozen=True)
-class SubtitleStream:
-    codec: str
-    ext: str
-    lang: str | None
-
-
-@dataclass(slots=True, frozen=True)
-class FileInfo:
-    path: Path
-    bitrate: int
-    duration: float
-    timecode: str  # in SMPTE
-    videos: tuple[VideoStream, ...]
-    audios: tuple[AudioStream, ...]
-    subtitles: tuple[SubtitleStream, ...]
-
-    def get_res(self) -> tuple[int, int]:
-        if self.videos:
-            return self.videos[0].width, self.videos[0].height
-        return 1920, 1080
-
-    def get_fps(self) -> Fraction:
-        if self.videos:
-            return self.videos[0].fps
-        return Fraction(30)
-
-    def get_sr(self) -> int:
-        if self.audios:
-            return self.audios[0].samplerate
-        return 48000
-
-    @classmethod
-    def init(self, path: str, log: Log) -> FileInfo:
-        try:
-            cont = av.open(path, "r")
-        except av.error.FileNotFoundError:
-            log.error(f"Input file doesn't exist: {path}")
-        except av.error.IsADirectoryError:
-            log.error(f"Expected a media file, but got a directory: {path}")
-        except av.error.InvalidDataError:
-            log.error(f"Invalid data when processing: {path}")
-
-        videos: tuple[VideoStream, ...] = ()
-        audios: tuple[AudioStream, ...] = ()
-        subtitles: tuple[SubtitleStream, ...] = ()
-
-        for v in cont.streams.video:
-            if v.duration is not None and v.time_base is not None:
-                vdur = float(v.duration * v.time_base)
-            else:
-                vdur = 0.0
-
-            fps = v.average_rate
-            if (fps is None or fps < 1) and v.name in {"png", "mjpeg", "webp"}:
-                fps = Fraction(25)
-            if fps is None or fps == 0:
-                fps = Fraction(30)
-
-            if v.sample_aspect_ratio is None:
-                sar = Fraction(1)
-            else:
-                sar = v.sample_aspect_ratio
-
-            cc = v.codec_context
-
-            if v.name is None:
-                log.error(f"Can't detect codec for video stream {v}")
-
-            videos += (
-                VideoStream(
-                    v.width,
-                    v.height,
-                    v.codec.canonical_name,
-                    fps,
-                    vdur,
-                    sar,
-                    v.time_base,
-                    getattr(v.format, "name", None),
-                    cc.color_range,
-                    cc.colorspace,
-                    cc.color_primaries,
-                    cc.color_trc,
-                    0 if v.bit_rate is None else v.bit_rate,
-                    v.language,
-                ),
-            )
-
-        for a in cont.streams.audio:
-            adur = 0.0
-            if a.duration is not None and a.time_base is not None:
-                adur = float(a.duration * a.time_base)
-
-            a_cc = a.codec_context
-            audios += (
-                AudioStream(
-                    a_cc.codec.canonical_name,
-                    0 if a_cc.sample_rate is None else a_cc.sample_rate,
-                    a.layout.name,
-                    a_cc.channels,
-                    adur,
-                    0 if a_cc.bit_rate is None else a_cc.bit_rate,
-                    a.language,
-                ),
-            )
-
-        for s in cont.streams.subtitles:
-            codec = s.codec_context.name
-            sub_exts = {"mov_text": "srt", "ass": "ass", "webvtt": "vtt"}
-            ext = sub_exts.get(codec, "vtt")
-            subtitles += (SubtitleStream(codec, ext, s.language),)
-
-        def get_timecode() -> str:
-            for d in cont.streams.data:
-                if (result := d.metadata.get("timecode")) is not None:
-                    return result
-            for v in cont.streams.video:
-                if (result := v.metadata.get("timecode")) is not None:
-                    return result
-            return "00:00:00:00"
-
-        timecode = get_timecode()
-        bitrate = 0 if cont.bit_rate is None else cont.bit_rate
-        dur = 0 if cont.duration is None else cont.duration / av.time_base
-
-        cont.close()
-
-        return FileInfo(Path(path), bitrate, dur, timecode, videos, audios, subtitles)
-
-    def __repr__(self) -> str:
-        return f"@{self.path.name}"