mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-26 11:41:09 +00:00
378 lines
13 KiB
Python
378 lines
13 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright (c) 2015, Thibault Saunier
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU Lesser General Public
|
|
# License as published by the Free Software Foundation; either
|
|
# version 2.1 of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
# Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public
|
|
# License along with this program; if not, write to the
|
|
# Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
|
|
# Boston, MA 02110-1301, USA.
|
|
|
|
from . import overrides_hack
|
|
|
|
import os
|
|
import tempfile
|
|
|
|
import gi
|
|
gi.require_version("Gst", "1.0")
|
|
gi.require_version("GES", "1.0")
|
|
|
|
from gi.repository import Gst # noqa
|
|
Gst.init(None) # noqa
|
|
from gi.repository import GES # noqa
|
|
GES.init()
|
|
|
|
from . import common # noqa
|
|
|
|
import unittest # noqa
|
|
|
|
|
|
class TestCopyPaste(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.timeline = GES.Timeline.new_audio_video()
|
|
self.assertEqual(len(self.timeline.get_tracks()), 2)
|
|
self.layer = self.timeline.append_layer()
|
|
|
|
def testCopyClipRemoveAndPaste(self):
|
|
clip1 = GES.TestClip.new()
|
|
clip1.props.duration = 10
|
|
|
|
self.layer.add_clip(clip1)
|
|
|
|
self.assertEqual(len(clip1.get_children(False)), 2)
|
|
|
|
copy = clip1.copy(True)
|
|
self.assertEqual(len(self.layer.get_clips()), 1)
|
|
|
|
self.layer.remove_clip(clip1)
|
|
|
|
copy.paste(10)
|
|
self.assertEqual(len(self.layer.get_clips()), 1)
|
|
|
|
def testCopyPasteTitleClip(self):
|
|
clip1 = GES.TitleClip.new()
|
|
clip1.props.duration = 10
|
|
|
|
self.layer.add_clip(clip1)
|
|
self.assertEqual(len(clip1.get_children(False)), 1)
|
|
|
|
copy = clip1.copy(True)
|
|
self.assertEqual(len(self.layer.get_clips()), 1)
|
|
|
|
copy.paste(10)
|
|
self.assertEqual(len(self.layer.get_clips()), 2)
|
|
|
|
|
|
class TestTransitionClip(unittest.TestCase):
|
|
|
|
def test_serialize_invert(self):
|
|
timeline = GES.Timeline.new()
|
|
timeline.add_track(GES.VideoTrack.new())
|
|
layer = timeline.append_layer()
|
|
|
|
clip1 = GES.TransitionClip.new_for_nick("crossfade")
|
|
clip1.props.duration = Gst.SECOND
|
|
self.assertTrue(layer.add_clip(clip1))
|
|
|
|
vtransition, = clip1.children
|
|
vtransition.set_inverted(True)
|
|
self.assertEqual(vtransition.props.invert, True)
|
|
|
|
with tempfile.NamedTemporaryFile() as tmpxges:
|
|
uri = Gst.filename_to_uri(tmpxges.name)
|
|
timeline.save_to_uri(uri, None, True)
|
|
|
|
timeline = GES.Timeline.new_from_uri(uri)
|
|
project = timeline.get_asset()
|
|
mainloop = common.create_main_loop()
|
|
project.connect("loaded", lambda _, __: mainloop.quit())
|
|
mainloop.run()
|
|
self.assertIsNotNone(timeline)
|
|
layer, = timeline.get_layers()
|
|
clip, = layer.get_clips()
|
|
vtransition, = clip.children
|
|
self.assertEqual(vtransition.props.invert, True)
|
|
|
|
class TestTitleClip(unittest.TestCase):
|
|
|
|
def testSetColor(self):
|
|
timeline = GES.Timeline.new_audio_video()
|
|
clip = GES.TitleClip.new()
|
|
timeline.append_layer().add_clip(clip )
|
|
self.assertTrue(clip.set_child_property('color', 1))
|
|
self.assertTrue(clip.set_child_property('color', 4294967295))
|
|
|
|
def testGetPropertyNotInTrack(self):
|
|
title_clip = GES.TitleClip.new()
|
|
self.assertEqual(title_clip.props.text, "")
|
|
self.assertEqual(title_clip.props.font_desc, "Serif 36")
|
|
|
|
def test_split_effect(self):
|
|
timeline = GES.Timeline.new()
|
|
timeline.add_track(GES.VideoTrack.new())
|
|
layer = timeline.append_layer()
|
|
|
|
clip1 = GES.TitleClip.new()
|
|
clip1.props.duration = Gst.SECOND
|
|
self.assertTrue(layer.add_clip(clip1))
|
|
|
|
effect = GES.Effect.new("agingtv")
|
|
self.assertTrue(clip1.add(effect))
|
|
|
|
children1 = clip1.get_children(True)
|
|
self.assertNotEqual(children1[0].props.priority,
|
|
children1[1].props.priority)
|
|
|
|
clip2 = clip1.split(Gst.SECOND / 2)
|
|
|
|
children1 = clip1.get_children(True)
|
|
self.assertNotEqual(children1[0].props.priority,
|
|
children1[1].props.priority)
|
|
|
|
children2 = clip2.get_children(True)
|
|
self.assertNotEqual(children2[0].props.priority,
|
|
children2[1].props.priority)
|
|
|
|
|
|
class TestUriClip(common.GESSimpleTimelineTest):
|
|
def test_max_duration_on_extract(self):
|
|
asset = GES.UriClipAsset.request_sync(common.get_asset_uri("audio_video.ogg"))
|
|
clip = asset.extract()
|
|
|
|
self.assertEqual(clip.props.max_duration, Gst.SECOND)
|
|
|
|
|
|
class TestTrackElements(common.GESSimpleTimelineTest):
|
|
|
|
def test_add_to_layer_with_effect_remove_add(self):
|
|
timeline = GES.Timeline.new_audio_video()
|
|
video_track, audio_track = timeline.get_tracks()
|
|
layer = timeline.append_layer()
|
|
|
|
test_clip = GES.TestClip()
|
|
self.assertEqual(test_clip.get_children(True), [])
|
|
self.assertTrue(layer.add_clip(test_clip))
|
|
audio_source = test_clip.find_track_element(None, GES.AudioSource)
|
|
video_source = test_clip.find_track_element(None, GES.VideoSource)
|
|
|
|
self.assertTrue(test_clip.set_child_property("volume", 0.0))
|
|
self.assertEqual(audio_source.get_child_property("volume")[1], 0.0)
|
|
|
|
effect = GES.Effect.new("agingtv")
|
|
test_clip.add(effect)
|
|
self.assertEqual(audio_source.props.track, audio_track)
|
|
self.assertEqual(video_source.props.track, video_track)
|
|
self.assertEqual(effect.props.track, video_track)
|
|
|
|
children = test_clip.get_children(True)
|
|
layer.remove_clip(test_clip)
|
|
self.assertEqual(test_clip.get_children(True), children)
|
|
self.assertEqual(audio_source.props.track, None)
|
|
self.assertEqual(video_source.props.track, None)
|
|
self.assertEqual(effect.props.track, None)
|
|
|
|
self.assertTrue(layer.add_clip(test_clip))
|
|
self.assertEqual(test_clip.get_children(True), children)
|
|
self.assertEqual(audio_source.props.track, audio_track)
|
|
self.assertEqual(video_source.props.track, video_track)
|
|
self.assertEqual(effect.props.track, video_track)
|
|
|
|
audio_source = test_clip.find_track_element(None, GES.AudioSource)
|
|
self.assertFalse(audio_source is None)
|
|
self.assertEqual(audio_source.get_child_property("volume")[1], 0.0)
|
|
self.assertEqual(audio_source.props.track, audio_track)
|
|
self.assertEqual(video_source.props.track, video_track)
|
|
self.assertEqual(effect.props.track, video_track)
|
|
|
|
def test_effects_priority(self):
|
|
timeline = GES.Timeline.new_audio_video()
|
|
layer = timeline.append_layer()
|
|
|
|
test_clip = GES.TestClip.new()
|
|
layer.add_clip(test_clip)
|
|
self.assert_effects(test_clip)
|
|
|
|
effect1 = GES.Effect.new("agingtv")
|
|
test_clip.add(effect1)
|
|
self.assert_effects(test_clip, effect1)
|
|
|
|
test_clip.set_top_effect_index(effect1, 1)
|
|
self.assert_effects(test_clip, effect1)
|
|
test_clip.set_top_effect_index(effect1, 10)
|
|
self.assert_effects(test_clip, effect1)
|
|
|
|
effect2 = GES.Effect.new("dicetv")
|
|
test_clip.add(effect2)
|
|
self.assert_effects(test_clip, effect1, effect2)
|
|
|
|
test_clip.remove(effect1)
|
|
self.assert_effects(test_clip, effect2)
|
|
|
|
def test_effects_index(self):
|
|
timeline = GES.Timeline.new_audio_video()
|
|
layer = timeline.append_layer()
|
|
|
|
test_clip = GES.TestClip.new()
|
|
layer.add_clip(test_clip)
|
|
self.assert_effects(test_clip)
|
|
|
|
ref_effects_list = []
|
|
|
|
def add_effect(effect):
|
|
test_clip.add(effect)
|
|
ref_effects_list.append(effect)
|
|
|
|
self.assert_effects(test_clip, *ref_effects_list)
|
|
|
|
def move_effect(old_index, new_index):
|
|
effect = ref_effects_list[old_index]
|
|
self.assertTrue(test_clip.set_top_effect_index(effect, new_index))
|
|
|
|
ref_effects_list.insert(new_index, ref_effects_list.pop(old_index))
|
|
|
|
self.assert_effects(test_clip, *ref_effects_list)
|
|
|
|
effects = ["agingtv", "dicetv", "burn", "gamma", "edgetv", "alpha",
|
|
"exclusion", "chromahold", "coloreffects", "videobalance"]
|
|
|
|
for effect in effects:
|
|
add_effect(GES.Effect.new(effect))
|
|
|
|
move_effect(3, 8)
|
|
move_effect(5, 6)
|
|
move_effect(0, 9)
|
|
|
|
self.assert_effects(test_clip, *ref_effects_list)
|
|
|
|
def test_signal_order_when_removing_effect(self):
|
|
timeline = GES.Timeline.new_audio_video()
|
|
layer = timeline.append_layer()
|
|
|
|
test_clip = GES.TestClip.new()
|
|
layer.add_clip(test_clip)
|
|
self.assert_effects(test_clip)
|
|
|
|
effect1 = GES.Effect.new("agingtv")
|
|
test_clip.add(effect1)
|
|
effect2 = GES.Effect.new("dicetv")
|
|
test_clip.add(effect2)
|
|
self.assert_effects(test_clip, effect1, effect2)
|
|
|
|
mainloop = common.create_main_loop()
|
|
|
|
signals = []
|
|
|
|
def handler_cb(*args):
|
|
signals.append(args[-1])
|
|
|
|
test_clip.connect("child-removed", handler_cb, "child-removed")
|
|
effect2.connect("notify::priority", handler_cb, "notify::priority")
|
|
test_clip.remove(effect1)
|
|
test_clip.disconnect_by_func(handler_cb)
|
|
effect2.disconnect_by_func(handler_cb)
|
|
self.assert_effects(test_clip, effect2)
|
|
|
|
mainloop.run(until_empty=True)
|
|
|
|
self.assertEqual(signals, ["child-removed", "notify::priority"])
|
|
|
|
def test_moving_core_track_elements(self):
|
|
clip = self.append_clip()
|
|
clip1 = self.append_clip()
|
|
title_clip = self.append_clip(asset_type=GES.TitleClip)
|
|
|
|
track_element = clip.find_track_element(None, GES.VideoSource)
|
|
self.assertTrue(clip.remove(track_element))
|
|
|
|
track_element1 = clip1.find_track_element(None, GES.VideoSource)
|
|
self.assertTrue(clip1.remove(track_element1))
|
|
|
|
self.assertTrue(clip1.add(track_element))
|
|
self.assertIsNotNone(track_element.get_track())
|
|
# We can add another TestSource to the clip as it has the same parent
|
|
# asset
|
|
self.assertTrue(clip1.add(track_element1))
|
|
# We already have a core TrackElement for the video track, not adding
|
|
# a second one.
|
|
self.assertIsNone(track_element1.get_track())
|
|
|
|
clip1.remove(track_element)
|
|
clip1.remove(track_element1)
|
|
title = title_clip.find_track_element(None, GES.VideoSource)
|
|
self.assertTrue(title_clip.remove(title))
|
|
# But we can't add an element that has been created by a TitleClip
|
|
self.assertFalse(clip.add(title))
|
|
self.assertFalse(title_clip.add(track_element))
|
|
self.assertTrue(clip.add(track_element))
|
|
self.assertTrue(clip1.add(track_element1))
|
|
|
|
def test_ungroup_regroup(self):
|
|
clip = self.append_clip()
|
|
children = clip.get_children(True)
|
|
|
|
clip1, clip2 = GES.Container.ungroup(clip, True)
|
|
|
|
self.assertEqual(clip, clip1)
|
|
clip1_child, = clip1.get_children(True)
|
|
clip2_child, = clip2.get_children(True)
|
|
self.assertCountEqual (children, [clip1_child, clip2_child])
|
|
|
|
# can freely move children between the ungrouped clips
|
|
self.assertTrue(clip1.remove(clip1_child))
|
|
self.assertTrue(clip2.add(clip1_child))
|
|
|
|
self.assertTrue(clip2.remove(clip2_child))
|
|
self.assertTrue(clip1.add(clip2_child))
|
|
|
|
grouped = GES.Container.group([clip1, clip2])
|
|
self.assertEqual(grouped, clip1)
|
|
|
|
self.assertCountEqual(clip1.get_children(True),
|
|
[clip1_child, clip2_child])
|
|
self.assertEqual(clip2.get_children(True), [])
|
|
|
|
# can freely move children between the grouped clips
|
|
self.assertTrue(clip1.remove(clip2_child))
|
|
self.assertTrue(clip2.add(clip2_child))
|
|
|
|
self.assertTrue(clip1.remove(clip1_child))
|
|
self.assertTrue(clip2.add(clip1_child))
|
|
|
|
self.assertTrue(clip2.remove(clip1_child))
|
|
self.assertTrue(clip1.add(clip1_child))
|
|
|
|
self.assertTrue(clip2.remove(clip2_child))
|
|
self.assertTrue(clip1.add(clip2_child))
|
|
|
|
# clip2 no longer part of the timeline
|
|
self.assertIsNone(clip2.props.layer)
|
|
self.assertEqual(clip1.props.layer, self.layer)
|
|
self.assertIsNone(clip2.props.timeline)
|
|
self.assertEqual(clip1.props.timeline, self.timeline)
|
|
|
|
def test_image_source_asset(self):
|
|
asset = GES.UriClipAsset.request_sync(common.get_asset_uri("png.png"))
|
|
clip = self.layer.add_asset(asset, 0, 0, Gst.SECOND, GES.TrackType.UNKNOWN)
|
|
|
|
image_src, = clip.get_children(True)
|
|
|
|
self.assertTrue(image_src.get_asset().is_image())
|
|
self.assertTrue(isinstance(image_src, GES.VideoUriSource))
|
|
imagefreeze, = [e for e in image_src.get_nleobject().iterate_recurse()
|
|
if e.get_factory().get_name() == "imagefreeze"]
|
|
|
|
asset = GES.UriClipAsset.request_sync(common.get_asset_uri("audio_video.ogg"))
|
|
clip = self.layer.add_asset(asset, Gst.SECOND, 0, Gst.SECOND, GES.TrackType.VIDEO)
|
|
video_src, = clip.get_children(True)
|
|
self.assertEqual([e for e in video_src.get_nleobject().iterate_recurse()
|
|
if e.get_factory().get_name() == "imagefreeze"], [])
|