vidi/Timeline.py: support setting the canvas size

[vidi-player.git] / vidi / Timeline.py

diff --git a/vidi/Timeline.py b/vidi/Timeline.py
index 48c6181..ef04c99 100755 (executable)
--- a/vidi/Timeline.py
+++ b/vidi/Timeline.py
@@ -29,31 +29,73 @@ GES.init()
 
 from .Player import Player
 
 
 from .Player import Player
 

+TITLE_BACKGROUND = 0xFF000000
+TITLE_COLOR = 0xFFFFFFFF
+TITLE_OUTLINE_COLOR = 0x00000000
+TITLE_FONT_DESC = "Georgia, 24"

 
 class Timeline(object):
 
 class Timeline(object):

-    def __init__(self):
+    def __init__(self, canvas_size=None):

         self.project = GES.Project(extractable_type=GES.Timeline)
         self.timeline = GES.Asset.extract(self.project)
 
         audio_track = GES.AudioTrack.new()
         video_track = GES.VideoTrack.new()
 
         self.project = GES.Project(extractable_type=GES.Timeline)
         self.timeline = GES.Asset.extract(self.project)
 
         audio_track = GES.AudioTrack.new()
         video_track = GES.VideoTrack.new()
 

+        if canvas_size:
+            width, height = canvas_size
+            caps = Gst.Caps.new_empty_simple("video/x-raw")
+            caps.set_value("width", width)
+            caps.set_value("height", height)
+            video_track.set_restriction_caps(caps)
+

         self.timeline.add_track(audio_track)
         self.timeline.add_track(video_track)
 
         self.layer = self.timeline.append_layer()
 
         self.timeline.add_track(audio_track)
         self.timeline.add_track(video_track)
 
         self.layer = self.timeline.append_layer()
 

-    def add_clip(self, clip_uri, start_time, duration):
+        ges_pipeline = GES.Pipeline()
+        ges_pipeline.set_timeline(self.timeline)
+        self.player = Player(ges_pipeline)
+
+    def add_title_clip(self, text, start_time, duration):
+        title_clip = GES.TitleClip()
+        title_clip.set_start(start_time * Gst.SECOND)
+        title_clip.set_duration(duration * Gst.SECOND)
+        self.layer.add_clip(title_clip)
+
+        # Now that the clip is inserted in the timeline, it has a source which
+        # can be used to set its properties. (comment taken from Pitivi)
+        title_source = title_clip.find_track_element(None, GES.TitleSource)
+        title_source.set_child_property("text", text)
+        title_source.set_child_property("background", TITLE_BACKGROUND)
+        title_source.set_child_property("color", TITLE_COLOR)
+        title_source.set_child_property("outline-color", TITLE_OUTLINE_COLOR)
+        title_source.set_child_property("font-desc", TITLE_FONT_DESC)
+        title_source.set_child_property("halignment", GES.TextVAlign.ABSOLUTE)
+        title_source.set_child_property("valignment", GES.TextHAlign.ABSOLUTE)
+
+    def add_clip(self, clip_path, start_time, duration):
+        clip_uri = Gst.filename_to_uri(clip_path)

         asset = GES.UriClipAsset.request_sync(clip_uri)
         self.layer.add_asset(asset, start_time * Gst.SECOND, 0,
                              duration * Gst.SECOND, GES.TrackType.UNKNOWN)
 
         asset = GES.UriClipAsset.request_sync(clip_uri)
         self.layer.add_asset(asset, start_time * Gst.SECOND, 0,
                              duration * Gst.SECOND, GES.TrackType.UNKNOWN)
 

+    def add_layer_clip(self, clip_path, start_time, duration):
+        """Add a clip on its own layer"""
+        clip_uri = Gst.filename_to_uri(clip_path)
+        asset = GES.UriClipAsset.request_sync(clip_uri)
+        new_layer = self.timeline.append_layer()
+        new_layer.add_asset(asset, start_time * Gst.SECOND, 0,
+                            duration * Gst.SECOND, GES.TrackType.UNKNOWN)
+

     def play(self):
         self.timeline.commit()
     def play(self):
         self.timeline.commit()

+        self.player.play()

 
 

-        ges_pipeline = GES.Pipeline()
-        ges_pipeline.set_timeline(self.timeline)
-        Player(ges_pipeline).play()
+    def stop(self):
+        self.player.stop()

 
 

-    def save(self, uri):
+    def save(self, path):
+        uri = Gst.filename_to_uri(path)

         self.project.save(self.timeline, uri, None, False)
         self.project.save(self.timeline, uri, None, False)