python/gst-custom-player.py

   1 #!/usr/bin/env python
   2
   3 # Simple media player with GStreamer
   4 #
   5 # Copyright (C) 2013-2014  Antonio Ospite <ao2@ao2.it>
   6 #
   7 # This program is free software: you can redistribute it and/or modify
   8 # it under the terms of the GNU General Public License as published by
   9 # the Free Software Foundation, either version 3 of the License, or
  10 # (at your option) any later version.
  11 #
  12 # This program is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 # GNU General Public License for more details.
  16 #
  17 # You should have received a copy of the GNU General Public License
  18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19
  20 # References:
  21 # http://pygstdocs.berlios.de/pygst-reference
  22 # https://core.fluendo.com/gstreamer/svn/trunk/gst-fluendo-gdlsink/test/ismdplay.py
  23
  24 import sys
  25 import os
  26
  27 import gi
  28 gi.require_version('Gst', '1.0')
  29 from gi.repository import Gst
  30 Gst.init(None)
  31
  32 from gi.repository import GObject
  33 GObject.threads_init()
  34
  35 # The player window will have a fixed width and height.
  36 # This is just to demonstrate the use of capabilities.
  37 WIDTH = 640
  38 HEIGHT = 300
  39
  40
  41 class CustomVideoBin(Gst.Bin):
  42     def __init__(self):
  43         Gst.Bin.__init__(self, 'CustomVideoBin')
  44
  45         queue = Gst.ElementFactory.make('queue', 'vqueue')
  46         self.add(queue)
  47
  48         rescale = Gst.ElementFactory.make("videoscale", "rescale")
  49         self.add(rescale)
  50         queue.link(rescale)
  51
  52         caps = Gst.Caps("video/x-raw-yuv,format=(fourcc)AYUV,width=%d,height=%d,pixel-aspect-ratio=1/1" % (WIDTH, HEIGHT))
  53         capsfilter = Gst.ElementFactory.make("capsfilter", "filter")
  54         capsfilter.set_property("caps", caps)
  55         self.add(capsfilter)
  56         rescale.link(capsfilter)
  57
  58         videoconvert = Gst.ElementFactory.make("videoconvert", "videoconvert")
  59         self.add(videoconvert)
  60         capsfilter.link(videoconvert)
  61
  62         videosink = Gst.ElementFactory.make("autovideosink", "vidoesink")
  63         self.add(videosink)
  64         videoconvert.link(videosink)
  65
  66         sink = queue.get_static_pad('sink')
  67         self.add_pad(Gst.GhostPad('sink', sink))
  68
  69
  70 class CustomAudioBin(Gst.Bin):
  71     def __init__(self):
  72         Gst.Bin.__init__(self, 'CustomAudioBin')
  73
  74         queue = Gst.ElementFactory.make('queue', 'aqueue')
  75         self.add(queue)
  76
  77         audioconvert = Gst.ElementFactory.make("audioconvert", "audioconverter")
  78         self.add(audioconvert)
  79         queue.link(audioconvert)
  80
  81         audiosink = Gst.ElementFactory.make("autoaudiosink", "audiosink")
  82         self.add(audiosink)
  83         audioconvert.link(audiosink)
  84
  85         sink = queue.get_static_pad('sink')
  86         self.add_pad(Gst.GhostPad('sink', sink))
  87
  88
  89 class CustomPlayBin(Gst.Pipeline):
  90     __gproperties__ = {
  91         'source': (Gst.Element, "source", "Source element", GObject.PARAM_READABLE)
  92     }
  93
  94     def __init__(self, uri=None):
  95         Gst.Pipeline.__init__(self, 'CustomPlayBin')
  96
  97         self._uri = uri
  98
  99         self._playbin = Gst.ElementFactory.make("playbin", "playbin")
 100         self.add(self._playbin)
 101
 102         self._playbin.set_property("uri", self._uri)
 103         self._playbin.set_property("video-sink", CustomVideoBin())
 104         self._playbin.set_property("audio-sink", CustomAudioBin())
 105
 106     def set_uri(self, uri):
 107         self._uri = uri
 108         self._playbin.set_property("uri", self._uri)
 109
 110
 111 class GstPlayer:
 112     def __init__(self):
 113
 114         # The user can require some action at End Of Stream
 115         self.eos_cb = None
 116
 117         self.pipeline = CustomPlayBin()
 118
 119         bus = self.pipeline.get_bus()
 120         bus.add_signal_watch()
 121         bus.connect('message::eos', self.on_eos)
 122         bus.connect('message::tag', self.on_tag)
 123         bus.connect('message::error', self.on_error)
 124         bus.connect('message::state-changed', self.on_state_changed)
 125
 126     def on_eos(self, bus, msg):
 127         print 'on_eos'
 128         self.stop()
 129         if self.eos_cb:
 130             self.eos_cb()
 131
 132     def on_tag(self, bus, msg):
 133         print 'on_tag:'
 134         taglist = msg.parse_tag()
 135         print '\t', taglist.to_string()
 136
 137     def on_error(self, bus, msg):
 138         print 'on_error'
 139         error, debug = msg.parse_error()
 140         print "Error: %s" % error, debug
 141         self.stop()
 142
 143     def on_state_changed(self, bus, msg):
 144         print 'on_state_changed'
 145         if msg.src != self.pipeline:
 146             return
 147
 148         old_state, new_state, pending = msg.parse_state_changed()
 149
 150     def set_location(self, location):
 151         self.pipeline.set_uri(location)
 152
 153     def play(self):
 154         self.pipeline.set_state(Gst.State.PLAYING)
 155
 156     def pause(self):
 157         self.pipeline.set_state(Gst.State.PAUSED)
 158
 159     def stop(self):
 160         self.pipeline.set_state(Gst.State.NULL)
 161
 162
 163 class PlayerTUI():
 164     def __init__(self, location):
 165
 166         self.player = GstPlayer()
 167         self.player.eos_cb = self.quit
 168
 169         self.mainloop = GObject.MainLoop()
 170
 171         self.player.set_location(location)
 172         self.player.play()
 173
 174         GObject.io_add_watch(sys.stdin, GObject.IO_IN, self.on_stdin)
 175
 176         try:
 177             self.mainloop.run()
 178         except KeyboardInterrupt:
 179             self.quit()
 180
 181     def on_stdin(self, fd, event):
 182         # The user has to send a newline fo this to go on
 183         c = os.read(fd.fileno(), 1)
 184
 185         if c == "q":
 186             self.quit()
 187         elif c == "s":
 188             self.player.pause()
 189         elif c == "r":
 190             self.player.play()
 191
 192         return True
 193
 194     def quit(self):
 195         self.player.stop()
 196         self.player = None
 197         self.mainloop.quit()
 198
 199
 200 def main(args):
 201     def usage():
 202         sys.stdout.write("usage: %s <URI-OF-MEDIA-FILE>\n" % args[0])
 203
 204     if len(args) != 2:
 205         usage()
 206         sys.exit(1)
 207
 208     uri = Gst.filename_to_uri(args[1])
 209
 210     tui = PlayerTUI(uri)
 211
 212 if __name__ == '__main__':
 213     sys.exit(main(sys.argv))