gstreamer/examples/gst/dvdplay.py

#!/usr/bin/env python
#
# gst-python
# Copyright (C) 2002 David I. Lehn
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Library General Public
# License as published by the Free Software Foundation; either
# version 2 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Library General Public License for more details.
#
# You should have received a copy of the GNU Library General Public
# License along with this library; if not, write to the
# Free Software Foundation, Inc., 59 Temple Place - Suite 330,
# Boston, MA 02111-1307, USA.
# 
# Author: David I. Lehn <dlehn@users.sourceforge.net>
#

import sys
#from gnome import *
from gstreamer import *
from gobject import GObject
import gtk
gtk.threads_init()

class DVDPlayer(object):
   def __init__(self):
      pass

   def idle(self, pipeline):
      #gtk.threads_enter()
      pipeline.iterate()
      #gtk.threads_leave()
      return 1

   def eof(self, sender):
      print 'EOS, quiting'
      sys.exit(0)
      
   def mpegparse_newpad(self, parser, pad, pipeline):
      #gtk.threads_enter()
      print '***** a new pad %s was created' % pad.get_name()
      if pad.get_name()[:6] == 'video_':
         pad.connect(self.v_queue.get_pad('sink'))
         self.pipeline.set_state(STATE_PAUSED)
         self.pipeline.add(self.v_thread)
         #self.v_thread.set_state(STATE_PLAYING)
         self.pipeline.set_state(STATE_PLAYING)
      elif pad.get_name() == 'private_stream_1.0':
         pad.connect(self.a_queue.get_pad('sink'))
         self.pipeline.set_state(STATE_PAUSED)
         self.pipeline.add(self.a_thread)
         #self.a_thread.set_state(STATE_PLAYING);
         self.pipeline.set_state(STATE_PLAYING)
      else:
         print 'unknown pad: %s' % pad.get_name()
      #gtk.threads_leave()

   def mpegparse_have_size(self, videosink, width, height):
      gtk.threads_enter()
      self.gtk_socket.set_usize(width,height)
      self.appwindow.show_all()
      gtk.threads_leave()

   def main(self):
      if len(sys.argv) < 5:
         print 'usage: %s dvdlocation title chapter angle' % sys.argv[0]
         return -1

      self.location = sys.argv[1]
      self.title = int(sys.argv[2])
      self.chapter = int(sys.argv[3])
      self.angle = int(sys.argv[4])

      #gst_init(&argc,&argv);
      #gnome_init('MPEG2 Video player','0.0.1',argc,argv);

      ret = self.build()
      if ret:
         return ret

      return self.run()

   def run(self):
      print 'setting to PLAYING state'

      gtk.threads_enter()

      self.pipeline.set_state(STATE_PLAYING)

      gtk.idle_add(self.idle,self.pipeline)

      gtk.main()

      self.pipeline.set_state(STATE_NULL)

      gtk.threads_leave()

      return 0

   def build_video_thread(self):
      # ***** pre-construct the video thread *****
      self.v_thread = Thread('v_thread')
      assert self.v_thread

      self.v_queue = gst_element_factory_make('queue','v_queue')
      assert self.v_queue

      self.v_decode = gst_element_factory_make('mpeg2dec','decode_video')
      assert self.v_decode

      self.color = gst_element_factory_make('colorspace','color')
      assert self.color

      self.efx = gst_element_factory_make('identity','identity')
      #self.efx = gst_element_factory_make('edgeTV','EdgeTV')
      #self.efx = gst_element_factory_make('agingTV','AgingTV')
      #effectv:  diceTV: DiceTV
      #effectv:  warpTV: WarpTV
      #effectv:  shagadelicTV: ShagadelicTV
      #effectv:  vertigoTV: VertigoTV
      #self.efx = gst_element_factory_make('revTV','RevTV')
      #self.efx = gst_element_factory_make('quarkTV','QuarkTV')
      assert self.efx

      self.color2 = gst_element_factory_make('colorspace','color2')
      assert self.color2

      self.show = gst_element_factory_make('xvideosink','show')
      #self.show = gst_element_factory_make('sdlvideosink','show')
      #self.show = gst_element_factory_make('fakesink','fakesinkv')
      assert self.show
      #self.show.set_property('silent', 0)
      #self.show.set_property('sync', 1)

      #self.deinterlace = gst_element_factory_make('deinterlace','deinterlace')
      self.deinterlace = gst_element_factory_make('identity','deinterlace')
      assert self.deinterlace

      last = None
      for e in (self.v_queue, self.v_decode, self.color, self.efx, self.color2,  self.deinterlace, self.show):
         self.v_thread.add(e)
         if last:
            last.connect(e)
         last = e

      #self.v_queue.connect(self.v_decode)
      #self.v_decode.connect(self.color)
      #self.color.connect(self.efx)
      #self.efx.connect(self.color2)
      #self.color2.connect(self.show)

   def build_audio_thread(self):
      # ***** pre-construct the audio thread *****
      self.a_thread = Thread('a_thread')
      assert self.a_thread

      self.a_queue = gst_element_factory_make('queue','a_queue')
      assert self.a_queue

      self.a_decode = gst_element_factory_make('a52dec','decode_audio')
      assert self.a_decode

      self.osssink = gst_element_factory_make('osssink','osssink')
      #self.osssink = gst_element_factory_make('fakesink','fakesinka')
      assert self.osssink
      #self.osssink.set_property('silent', 0)
      #self.osssink.set_property('sync', 0)

      for e in (self.a_queue, self.a_decode, self.osssink):
         self.a_thread.add(e)

      self.a_queue.connect(self.a_decode)
      self.a_decode.connect(self.osssink)

   def build(self):
      # ***** construct the main pipeline *****
      self.pipeline = Pipeline('pipeline')
      assert self.pipeline

      self.src = gst_element_factory_make('dvdreadsrc','src');
      assert self.src

      GObject.connect(self.src,'deep_notify',self.dnprint)
      self.src.set_property('location', self.location)
      self.src.set_property('title', self.title)
      self.src.set_property('chapter', self.chapter)
      self.src.set_property('angle', self.angle)

      self.parse = gst_element_factory_make('mpegdemux','parse')
      assert self.parse
      self.parse.set_property('sync', 0)

      self.pipeline.add(self.src)
      self.pipeline.add(self.parse)

      self.src.connect(self.parse)

      # pre-construct the audio/video threads
      self.build_video_thread()
      self.build_audio_thread()

      # ***** construct the GUI *****
      #self.appwindow = gnome_app_new('DVD Player','DVD Player')

      #self.gtk_socket = gtk_socket_new ()
      #gtk_socket.show()

      #gnome_app_set_contents(GNOME_APP(appwindow),
            #GTK_WIDGET(gtk_socket));

      #gtk_widget_realize (gtk_socket);
      #gtk_socket_steal (GTK_SOCKET (gtk_socket), 
            #gst_util_get_int_arg (GTK_OBJECT(show), 'xid'));

      GObject.connect(self.parse,'new_pad',self.mpegparse_newpad, self.pipeline)
      GObject.connect(self.src,'eos',self.eof)
      #GObject.connect(show,'have_size',self.mpegparse_have_size, self.pipeline)

      #GObject.connect(self.pipeline,'error',self.pipeline_error)
      #GObject.connect(self.pipeline,'deep_notify',self.dnprint)

      return 0

   def pipeline_error(self, sender, obj, error):
      print "(%s) ERROR: %s: %s" % (self, obj.name(), error)

   def dnprint(self, sender, obj, param):
      str = obj.get_property(param.name)
      print '%s: %s = %s' % (sender.get_name(), param.name, str)

if __name__ == '__main__':
   #gst_debug_set_categories(0xFFFFFFFFL)
   #gst_info_set_categories(0xFFFFFFFFL)
   player = DVDPlayer()
   ret = player.main()
   sys.exit(ret)