mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-07-04 05:35:53 +00:00
f51f2f70de
Implement lazy loading asset cache so gesdemux use the formatters assets while GES hasn't been initialized. And set extensions to temporary files as some formatters require the information (otio)
108 lines
3.7 KiB
Python
108 lines
3.7 KiB
Python
#!/usr/bin/env python
|
|
# -*- Mode: Python -*-
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
|
#
|
|
# Copyright (C) 2019 Igalia S.L
|
|
# Authors:
|
|
# Thibault Saunier <tsaunier@igalia.com>
|
|
#
|
|
|
|
import sys
|
|
|
|
import gi
|
|
import tempfile
|
|
gi.require_version("GES", "1.0")
|
|
gi.require_version("Gst", "1.0")
|
|
|
|
from gi.repository import GObject
|
|
from gi.repository import Gst
|
|
Gst.init(None)
|
|
from gi.repository import GES
|
|
from gi.repository import GLib
|
|
from collections import OrderedDict
|
|
|
|
try:
|
|
import opentimelineio as otio
|
|
otio.adapters.from_name('xges')
|
|
except Exception as e:
|
|
Gst.info("Could not load OpenTimelineIO: %s" % e)
|
|
otio = None
|
|
|
|
class GESOtioFormatter(GES.Formatter):
|
|
def do_save_to_uri(self, timeline, uri, overwrite):
|
|
if not Gst.uri_is_valid(uri) or Gst.uri_get_protocol(uri) != "file":
|
|
Gst.error("Protocol not supported for file: %s" % uri)
|
|
return False
|
|
|
|
with tempfile.NamedTemporaryFile(suffix=".xges") as tmpxges:
|
|
timeline.get_asset().save(timeline, "file://" + tmpxges.name, None, overwrite)
|
|
|
|
linker = otio.media_linker.MediaLinkingPolicy.ForceDefaultLinker
|
|
otio_timeline = otio.adapters.read_from_file(tmpxges.name, "xges", media_linker_name=linker)
|
|
location = Gst.uri_get_location(uri)
|
|
out_adapter = otio.adapters.from_filepath(location)
|
|
otio.adapters.write_to_file(otio_timeline, Gst.uri_get_location(uri), out_adapter.name)
|
|
|
|
return True
|
|
|
|
def do_can_load_uri(self, uri):
|
|
try:
|
|
if not Gst.uri_is_valid(uri) or Gst.uri_get_protocol(uri) != "file":
|
|
return False
|
|
except GLib.Error as e:
|
|
Gst.error(str(e))
|
|
return False
|
|
|
|
if uri.endswith(".xges"):
|
|
return False
|
|
|
|
try:
|
|
return otio.adapters.from_filepath(Gst.uri_get_location(uri)) is not None
|
|
except Exception as e:
|
|
Gst.info("Could not load %s -> %s" % (uri, e))
|
|
return False
|
|
|
|
|
|
def do_load_from_uri(self, timeline, uri):
|
|
location = Gst.uri_get_location(uri)
|
|
in_adapter = otio.adapters.from_filepath(location)
|
|
assert(in_adapter) # can_load_uri should have ensured it is loadable
|
|
|
|
linker = otio.media_linker.MediaLinkingPolicy.ForceDefaultLinker
|
|
otio_timeline = otio.adapters.read_from_file(
|
|
location,
|
|
in_adapter.name,
|
|
media_linker_name=linker
|
|
)
|
|
|
|
with tempfile.NamedTemporaryFile(suffix=".xges") as tmpxges:
|
|
otio.adapters.write_to_file(otio_timeline, tmpxges.name, "xges")
|
|
formatter = GES.Formatter.get_default().extract()
|
|
timeline.get_asset().add_formatter(formatter)
|
|
return formatter.load_from_uri(timeline, "file://" + tmpxges.name)
|
|
|
|
if otio is not None:
|
|
GObject.type_register(GESOtioFormatter)
|
|
known_extensions_mimetype_map = [
|
|
("otio", "xml", "fcpxml"),
|
|
("application/vnd.pixar.opentimelineio+json", "application/vnd.apple-xmeml+xml", "application/vnd.apple-fcp+xml")
|
|
]
|
|
|
|
extensions = []
|
|
for adapter in otio.plugins.ActiveManifest().adapters:
|
|
if adapter.name != 'xges':
|
|
extensions.extend(adapter.suffixes)
|
|
|
|
extensions_mimetype_map = [[], []]
|
|
for i, ext in enumerate(known_extensions_mimetype_map[0]):
|
|
if ext in extensions:
|
|
extensions_mimetype_map[0].append(ext)
|
|
extensions_mimetype_map[1].append(known_extensions_mimetype_map[1][i])
|
|
extensions.remove(ext)
|
|
extensions_mimetype_map[0].extend(extensions)
|
|
|
|
GES.FormatterClass.register_metas(GESOtioFormatter, "otioformatter",
|
|
"GES Formatter using OpenTimelineIO",
|
|
','.join(extensions_mimetype_map[0]),
|
|
';'.join(extensions_mimetype_map[1]), 0.1, Gst.Rank.SECONDARY)
|