home *** CD-ROM | disk | FTP | other *** search
/ PC Welt 2006 November (DVD) / PCWELT_11_2006.ISO / casper / filesystem.squashfs / usr / share / doc / python-gst0.10 / examples / bps.py < prev    next >
Encoding:
Python Source  |  2006-07-31  |  3.0 KB  |  121 lines
  1. #!/usr/bin/env python
  2. # -*- Mode: Python -*-
  3. # vi:si:et:sw=4:sts=4:ts=4
  4.  
  5. # gst-python
  6. # Copyright (C) 2003 David I. Lehn
  7. #
  8. # This library is free software; you can redistribute it and/or
  9. # modify it under the terms of the GNU Library General Public
  10. # License as published by the Free Software Foundation; either
  11. # version 2 of the License, or (at your option) any later version.
  12. #
  13. # This library is distributed in the hope that it will be useful,
  14. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16. # Library General Public License for more details.
  17. #
  18. # You should have received a copy of the GNU Library General Public
  19. # License along with this library; if not, write to the
  20. # Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  21. # Boston, MA 02111-1307, USA.
  22. # 
  23. # Author: David I. Lehn <dlehn@users.sourceforge.net>
  24. #
  25.  
  26. import pygtk
  27. pygtk.require('2.0')
  28.  
  29. import sys
  30. import time
  31. import gobject
  32. import gtk
  33.  
  34. import pygst
  35. pygst.require('0.10')
  36.  
  37. import gst
  38.  
  39. class BPS(object):
  40.     def __init__(self):
  41.         self.buffers = 0
  42.         self.start = 0
  43.  
  44.     def done(self):
  45.         end = time.time()
  46.         dt = end - self.start
  47.         bps = self.buffers/dt
  48.         spb = dt/self.buffers
  49.         print '\t%d buffers / %fs\t= %f bps\t= %f spb' % (self.buffers, dt, bps, spb)
  50.  
  51.     def fakesrc(self, buffers):
  52.         src = gst.element_factory_make('fakesrc','src')
  53.         src.set_property('silent', 1)
  54.         src.set_property('num_buffers', buffers)
  55.         return src
  56.  
  57.     def fakesink(self):
  58.         sink = gst.element_factory_make('fakesink','sink')
  59.         sink.set_property('silent', 1)
  60.         return sink
  61.  
  62.     def build_pipeline(self, buffers):
  63.         pipeline = gst.Pipeline('pipeline')
  64.  
  65.         src = self.fakesrc(buffers)
  66.         pipeline.add(src)
  67.         sink = self.fakesink()
  68.         pipeline.add(sink)
  69.         src.link(sink)
  70.  
  71.         return pipeline
  72.  
  73.     def idle(self, pipeline):
  74.         return pipeline.iterate()
  75.  
  76.     def test(self):
  77.         self.bus = self.pipeline.get_bus()
  78.  
  79.         self.start = time.time()
  80.         
  81.         self.pipeline.set_state(gst.STATE_PLAYING)
  82.  
  83.         while 1:
  84.             msg = self.bus.poll(gst.MESSAGE_EOS | gst.MESSAGE_ERROR, gst.SECOND)
  85.             if msg:
  86.                 break
  87.  
  88.         self.pipeline.set_state(gst.STATE_NULL)
  89.         self.done()
  90.  
  91.     def run(self, buffers):
  92.         self.buffers = buffers
  93.         
  94.         print '# Testing buffer processing rate for "fakesrc ! fakesink"'
  95.         print '# bps = buffers per second'
  96.         print '# spb = seconds per buffer'
  97.         
  98.         self.pipeline = self.build_pipeline(buffers)
  99.         assert self.pipeline
  100.  
  101.         self.test()
  102.     
  103. def main(args):
  104.     "GStreamer Buffers-Per-Second tester"
  105.  
  106.     if len(args) < 2:
  107.         print 'usage: %s buffers' % args[0]
  108.         return 1
  109.     
  110.     bps = BPS()
  111.     
  112.     buffers = int(args[1])
  113.     if buffers < 1:
  114.     print 'buffers must be higher than 0'
  115.     return
  116.  
  117.     bps.run(buffers)
  118.  
  119. if __name__ == '__main__':
  120.     sys.exit(main(sys.argv))
  121.