Move python experiments to a python/ subdir
[experiments/gstreamer.git] / python / gst-trick-mode-looping-1.py
diff --git a/python/gst-trick-mode-looping-1.py b/python/gst-trick-mode-looping-1.py
new file mode 100755 (executable)
index 0000000..fe00556
--- /dev/null
@@ -0,0 +1,96 @@
+#!/usr/bin/env python
+
+# Get a test sample with:
+# youtube-dl -t http://www.youtube.com/watch?v=yWa-YXiSk2Y
+
+import sys
+
+import gobject
+gobject.threads_init()
+
+import gst
+
+
+class Player:
+    def __init__(self, filename, rate):
+        self._filename = filename
+        self._rate = rate
+
+        self._player = gst.element_factory_make("playbin2", "player")
+        self._player.set_property("uri", filename)
+
+        bus = self._player.get_bus()
+        bus.add_signal_watch()
+        bus.connect('message::eos', self.on_eos)
+        bus.connect('message::state-changed', self.on_state_changed)
+
+    def run(self):
+        self._player.set_state(gst.STATE_PLAYING)
+        loop = gobject.MainLoop()
+        loop.run()
+
+    def on_eos(self, bus, msg):
+        sys.stderr.write(".")
+        self._player.set_state(gst.STATE_PLAYING)
+        seek_event = gst.event_new_seek(self._rate,
+                gst.FORMAT_TIME,
+                gst.SEEK_FLAG_FLUSH | gst.SEEK_FLAG_ACCURATE,
+                gst.SEEK_TYPE_SET, 0,
+                gst.SEEK_TYPE_NONE, 0)
+        self._player.send_event(seek_event)
+
+    def on_state_changed(self, bus, msg):
+        if msg.src != self._player:
+            return
+
+        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:
+            self.set_rate(self._rate)
+
+    def set_rate(self, rate):
+        self._rate = rate
+        try:
+            position, format = self._player.query_position(gst.FORMAT_TIME)
+        except:
+            position = 0
+
+        # Create the seek event
+        if rate > 0:
+            seek_event = gst.event_new_seek(rate,
+                    gst.FORMAT_TIME,
+                    gst.SEEK_FLAG_FLUSH | gst.SEEK_FLAG_ACCURATE,
+                    gst.SEEK_TYPE_SET, position,
+                    gst.SEEK_TYPE_NONE, 0)
+        else:
+            seek_event = gst.event_new_seek(rate,
+                    gst.FORMAT_TIME,
+                    gst.SEEK_FLAG_FLUSH | gst.SEEK_FLAG_ACCURATE,
+                    gst.SEEK_TYPE_SET, 0,
+                    gst.SEEK_TYPE_SET, position)
+
+        if seek_event:
+            self._player.send_event(seek_event)
+            gst.info("rate set to %s" % rate)
+        else:
+            gst.warining("change rate failed")
+
+
+def main(args):
+    def usage():
+        sys.stdout.write("usage: %s <URI-OF-MEDIA-FILE>\n" % args[0])
+
+    if len(args) != 2:
+        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)
+
+    player = Player(args[1], 3.0)
+    player.run()
+
+if __name__ == '__main__':
+    sys.exit(main(sys.argv))