gst-raiplay-download.sh: rename tsdemux from "ts" to "demux"

[experiments/gstreamer.git] / python / gst-trick-mode-looping-3.py

diff --git a/python/gst-trick-mode-looping-3.py b/python/gst-trick-mode-looping-3.py
index 94b99af..1fef9d3 100755 (executable)
--- a/python/gst-trick-mode-looping-3.py
+++ b/python/gst-trick-mode-looping-3.py
@@ -1,39 +1,54 @@
 #!/usr/bin/env python
 #!/usr/bin/env python

-
+#
+# A simple "trick-mode" looping player.
+# Version 3, based on segment seeking.
+#
+# TODO: the trickmodes work way better but the looping functionality
+# needs to be fixed.
+#

 # Get a test sample with:
 # youtube-dl -t http://www.youtube.com/watch?v=yWa-YXiSk2Y
 
 import sys
 
 # Get a test sample with:
 # youtube-dl -t http://www.youtube.com/watch?v=yWa-YXiSk2Y
 
 import sys
 

-import gobject
-gobject.threads_init()
+import gi
+gi.require_version('Gst', '1.0')
+from gi.repository import Gst
+Gst.init(None)

 
 

-import gst
+from gi.repository import GObject
+GObject.threads_init()

 
 
 class Player:
 
 
 class Player:

-    def __init__(self, filename, rate):
-        self._filename = filename
+    def __init__(self, uri, rate):

         self._rate = rate
 
         self._rate = rate
 

-        self._player = gst.element_factory_make("playbin2", "player")
-        self._player.set_property("uri", filename)
+        self._player = Gst.ElementFactory.make("playbin", "player")
+        self._player.set_property("uri", uri)

 
         bus = self._player.get_bus()
         bus.add_signal_watch()
 
         bus = self._player.get_bus()
         bus.add_signal_watch()

-        bus.connect('message::state-changed', self.on_state_changed)
+        bus.connect('message::error', self.on_error)

         bus.connect('message::segment-done', self.on_segment_done)
         bus.connect('message::segment-done', self.on_segment_done)

+        bus.connect('message::state-changed', self.on_state_changed)

 
     def run(self):
 
     def run(self):

-        self._player.set_state(gst.STATE_PLAYING)
-        loop = gobject.MainLoop()
-        loop.run()
+        self._player.set_state(Gst.State.PLAYING)
+        self.loop = GObject.MainLoop()
+        self.loop.run()
+
+    def quit(self):
+        self._player.set_state(Gst.State.NULL)
+        self.loop.quit()

 
     def on_segment_done(self, bus, msg):
 
     def on_segment_done(self, bus, msg):

-        self._player.seek(self._rate, gst.FORMAT_TIME,
-                gst.SEEK_FLAG_SEGMENT | gst.SEEK_FLAG_SKIP | gst.SEEK_FLAG_ACCURATE,
-                gst.SEEK_TYPE_SET, 0,
-                gst.SEEK_TYPE_END, 0)
+        self.set_seek(0)
+
+    def on_error(self, bus, msg):
+        (err, debug) = msg.parse_error()
+        print "Error: %s" % err, debug
+        self.quit()

 
     def on_state_changed(self, bus, msg):
         if msg.src != self._player:
 
     def on_state_changed(self, bus, msg):
         if msg.src != self._player:


@@ -42,35 +57,51 @@ class Player:
         print 'on_state_changed'
         old_state, new_state, pending = msg.parse_state_changed()
         print "%s -> %s" % (old_state, new_state)
         print 'on_state_changed'
         old_state, new_state, pending = msg.parse_state_changed()
         print "%s -> %s" % (old_state, new_state)

-        if old_state == gst.STATE_READY and new_state == gst.STATE_PAUSED:
+        if old_state == Gst.State.READY and new_state == Gst.State.PAUSED:

             self.set_rate(self._rate)
 
     def set_rate(self, rate):
         self._rate = rate
             self.set_rate(self._rate)
 
     def set_rate(self, rate):
         self._rate = rate

+        self.set_seek(0, True)

 
 

-        if rate == 0:
-            self._player.set_state(gst.STATE_PAUSED)
+    def set_seek(self, position, flush=False):
+        flags = Gst.SeekFlags.SEGMENT | Gst.SeekFlags.SKIP | Gst.SeekFlags.ACCURATE
+
+        if flush:
+            flags |= Gst.SeekFlags.FLUSH
+
+        if self._rate >= 0:
+            seek_event = Gst.Event.new_seek(self._rate,
+                                            Gst.Format.TIME,
+                                            flags,
+                                            Gst.SeekType.SET, position,
+                                            Gst.SeekType.NONE, 0)

         else:
         else:

-            self._player.set_state(gst.STATE_PLAYING)
-            self._player.seek(self._rate, gst.FORMAT_TIME,
-                    gst.SEEK_FLAG_SEGMENT | gst.SEEK_FLAG_SKIP | gst.SEEK_FLAG_FLUSH | gst.SEEK_FLAG_ACCURATE,
-                    gst.SEEK_TYPE_SET, 0,
-                    gst.SEEK_TYPE_END, 0)
+            seek_event = Gst.Event.new_seek(self._rate,
+                                            Gst.Format.TIME,
+                                            flags,
+                                            Gst.SeekType.NONE, 0,
+                                            Gst.SeekType.END, position)
+
+        if seek_event:
+            self._player.send_event(seek_event)
+            Gst.info("rate set to %s" % self._rate)
+        else:
+            Gst.warining("change rate failed")

 
 
 def main(args):
     def usage():
 
 
 def main(args):
     def usage():

-        sys.stdout.write("usage: %s <URI-OF-MEDIA-FILE>\n" % args[0])
+        sys.stdout.write("usage: %s <filename> <speedrate>\n" % args[0])

 
 

-    if len(args) != 2:
+    if len(args) != 3:

         usage()
         sys.exit(1)
 
         usage()
         sys.exit(1)
 

-    if not gst.uri_is_valid(args[1]):
-        sys.stderr.write("Error: Invalid URI: %s\n" % args[1])
-        sys.exit(1)
+    uri = Gst.filename_to_uri(args[1])
+    rate = float(args[2])

 
 

-    player = Player(args[1], 3.0)
+    player = Player(uri, rate)

     player.run()
 
 if __name__ == '__main__':
     player.run()
 
 if __name__ == '__main__':