gstreamer/tests/check/python/test_clip.py

# -*- 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 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 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_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()

        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.assertFalse(clip1.add(track_element))
        self.assertFalse(clip.add(track_element1))

        self.assertTrue(clip.add(track_element))
        self.assertTrue(clip1.add(track_element1))

    def test_ungroup_regroup(self):
        clip = self.append_clip()
        clip1, clip2 = GES.Container.ungroup(clip, True)

        self.assertEqual(clip, clip1)
        clip2_child, = clip2.get_children(True)

        self.assertTrue(clip2.remove(clip2_child))
        self.assertTrue(self.layer.remove_clip(clip2))

        self.assertTrue(clip1.add(clip2_child))