renatofilho@484: import pygst
renatofilho@484: pygst.require("0.10")
renatofilho@484: import gst
renatofilho@484: import gobject
renatofilho@484: import socket
renatofilho@484: 
renatofilho@484: class Media:
renatofilho@484:     class StreamData:
renatofilho@484:         stream_count = 0
renatofilho@484: 		
renatofilho@484:         def __init__ (self, pipe, abin, vbin, sink):
renatofilho@484: 
renatofilho@484: 	    self.stream_count += 1
renatofilho@484: 	    self.Id = self.stream_count
renatofilho@484: 	    self.Pipe = pipe
renatofilho@484: 	    self.Abin = abin
renatofilho@484: 	    self.Vbin = vbin
renatofilho@484:             self.Sink = sink
renatofilho@484: 	    self.Loop = gobject.MainLoop()
renatofilho@484: 	    self.ACaps = ""
renatofilho@484: 	    self.VCaps = ""
renatofilho@484: 	    self.Ready = False
renatofilho@484: 
renatofilho@484: 
renatofilho@484:     def __init__(self, config):
renatofilho@484:         # set gstreamer basic options
renatofilho@484:         self.config = config
renatofilho@484:         self.pipe = None
renatofilho@484:         self.streams = []
renatofilho@484: 	self.socket = None
renatofilho@484: 	self.connection = None
renatofilho@484: 	self.addr = None
renatofilho@484: 
renatofilho@484: 
renatofilho@484:     def setup(self, filename, mux, vcodec, vbitrate,
renatofilho@484:               fps, acodec, abitrate, width, height, port, options):
renatofilho@484: 
renatofilho@484:         ## Pipelines
renatofilho@484:         self.pipe = gst.Pipeline ()
renatofilho@484:         uri = "file://" + filename
renatofilho@484:         print "Opening Uri:" + uri
renatofilho@484:         src = gst.element_make_from_uri (gst.URI_SRC, uri, "src")
renatofilho@484:         if (src is None):
renatofilho@484:             return None
renatofilho@484:         
renatofilho@484:         decode = gst.element_factory_make ("decodebin", "decode")
renatofilho@484:         if (decode is None):
renatofilho@484:             return None
renatofilho@484: 
renatofilho@484: 	mux = gst.element_factory_make ("avimux", "mux")
renatofilho@484:         if (mux is None):
renatofilho@484:             return None
renatofilho@484: 
renatofilho@484:         sink = gst.element_factory_make ("fdsink", "sink")
renatofilho@484:         if (sink is None):
renatofilho@484:             return None
renatofilho@484: 
renatofilho@484:         #Create socket
renatofilho@484:         self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
renatofilho@484: 	self.socket.bind(('', int (port)))
renatofilho@484:         
renatofilho@484:         #video encode 
renatofilho@484:         #queue ! videoscale ! video/x-raw-yuv,width=240,height=144 ! videorate ! ffenc_h263p bitrate=256000 me-method=2 ! rtph263ppay ! udpsink  host=224.0.0.1 port=5000
renatofilho@484:         vbin = gst.Bin ()
renatofilho@484:         vqueue = gst.element_factory_make ("queue", "vqueue")
renatofilho@484:         vscale = gst.element_factory_make ("videoscale", "vscale")
renatofilho@484:         vrate = gst.element_factory_make ("videorate", "vrate")
renatofilho@484:         vencode = gst.element_factory_make ("ffenc_h263p", "vencode")
renatofilho@484: 
renatofilho@484:         if (None in [vbin, vqueue, vscale, vrate, vencode]):
renatofilho@484:             print "Fail to create video encode elements."
renatofilho@484:             return None
renatofilho@484: 
renatofilho@484:         vscale_pad = vscale.get_pad("sink")
renatofilho@484:         if (vscale_pad is None):
renatofilho@484:             print "Fail to get vscale sink pad."
renatofilho@484:             return None
renatofilho@484: 
renatofilho@484:         vscale_caps = gst.caps_from_string ("video/x-raw-yuv, width=%s, height=%s" % (width, height))
renatofilho@484:         if (vscale_caps is None):
renatofilho@484:             print "Fail to create video caps"
renatofilho@484:             return None
renatofilho@484: 
renatofilho@484:         if (not vscale_pad.set_caps (vscale_caps)):
renatofilho@484:             print "Fail to set video output caps"
renatofilho@484:             return None
renatofilho@484:         
renatofilho@484:         vbin.add (vqueue, vscale, vrate, vencode)
renatofilho@484:         if (not gst.element_link_many (vqueue,  vscale, vrate, vencode)):
renatofilho@484:             print "Fail to link video elements"
renatofilho@484:             return None
renatofilho@484:         
renatofilho@484:         vbin.add_pad (gst.GhostPad ("sink", vqueue.get_pad ("sink")))
renatofilho@484: 	vbin.add_pad (gst.GhostPad ("src", vencode.get_pad ("src")))
renatofilho@484: 
renatofilho@484:         #audio encode
renatofilho@484:         #audio/x-raw-int ! queue ! audioconvert ! faac ! rtpmp4gpay !  udpsink name=upd_audio host=224.0.0.1 port=5002
renatofilho@484:         abin = gst.Bin ()
renatofilho@484:         aqueue = gst.element_factory_make ("queue", "vqueue")
renatofilho@484:         aconvert = gst.element_factory_make ("audioconvert", "aconvert")
renatofilho@484:         aencode = gst.element_factory_make ("ffenc_ac3", "aencode")
renatofilho@484: 
renatofilho@484:         if (None in [abin, aqueue, aconvert, aencode]):
renatofilho@484:             print "Fail to create video encode elements."
renatofilho@484:             return None
renatofilho@484: 
renatofilho@484:         abin.add (aqueue, aconvert, aencode)
renatofilho@484:         if (not gst.element_link_many (aqueue, aconvert, aencode)):
renatofilho@484:             print "Fail to link video elements"
renatofilho@484:             return None
renatofilho@484:         
renatofilho@484:         abin.add_pad (gst.GhostPad ("sink", aqueue.get_pad ("sink")))
renatofilho@484:         abin.add_pad (gst.GhostPad ("src", aencode.get_pad ("src")))
renatofilho@484: 
renatofilho@484:         #Finish Pipeline
renatofilho@484: 
renatofilho@484: 	self.pipe.add (src, decode, abin, vbin, mux, sink)
renatofilho@484: 	gst.element_link_many (src, decode)
renatofilho@484:         gst.element_link_many (mux, sink)
renatofilho@484: 
renatofilho@484:         #Linking decode with mux
renatofilho@484:         mux_audio = mux.get_pad ("audio_0")
renatofilho@484:         mux_video = mux.get_pad ("video_0")
renatofilho@484: 
renatofilho@484: 	audio_pad = abin.get_pad ("src")
renatofilho@484:         video_pad = vbin.get_pad ("src")
renatofilho@484: 
renatofilho@484:         if (audio_pad.link (mux_audio) != gst.PAD_LINK_OK):
renatofilho@484:             print "Fail to link audio with mux"
renatofilho@484:             return None
renatofilho@484:  
renatofilho@484:         if (video_pad.link (mux_video) != gst.PAD_LINK_OK):
renatofilho@484:             print "Fail to link audio with mux"
renatofilho@484:             return None
renatofilho@484: 
renatofilho@484: 	stream_data = self.StreamData (self.pipe, abin, vbin, sink)
renatofilho@484: 
renatofilho@484: 	bus = self.pipe.get_bus()
renatofilho@484: 	bus.add_signal_watch()
renatofilho@484: 	bus.connect("message", self.__on_bus_message, stream_data)
renatofilho@484: 	
renatofilho@484: 	decode.connect("new-decoded-pad", self.__on_decode_new_pad, stream_data)
renatofilho@484: 	decode.connect("unknown-type", self.__on_decode_unknown_type, stream_data)
renatofilho@484: 
renatofilho@484: 	
renatofilho@484: 	self.pipe.set_state (gst.STATE_PAUSED)
renatofilho@484:         print "Running Pipe"
renatofilho@484: 	stream_data.Loop.run ()
renatofilho@484:         print "End run"
renatofilho@484: 
renatofilho@484: 	a_caps = stream_data.ACaps
renatofilho@484: 	v_caps = stream_data.VCaps
renatofilho@484: 	stream_id = stream_data.Id
renatofilho@484: 
renatofilho@484:         self.streams.append (stream_data)
renatofilho@484: 
renatofilho@484:     def play(self):
renatofilho@484: 
renatofilho@484:         print "Trying to play pipeline: %s" % self.pipe
renatofilho@484:         try:
renatofilho@484:             if (self.pipe):
renatofilho@484: 		print "Waiting for connection"
renatofilho@484: 		self.socket.listen(1)
renatofilho@484: 		print "Connection Requested"
renatofilho@484: 	        #Create socket
renatofilho@484: 		self.connection, self.addr = self.socket.accept ()
renatofilho@484: 
renatofilho@484: 		stream_data = self.streams[0]
renatofilho@484: 		stream_data.Sink.set_property ("fd", self.connection.fileno());
renatofilho@484:                 print "Connected"
renatofilho@484: 
renatofilho@484:                 self.pipe.set_state(gst.STATE_PLAYING)
renatofilho@484:         except gobject.GError, e:
renatofilho@484:             print "Error: " + str(e)
renatofilho@484: 
renatofilho@484: 
renatofilho@484:     def stop(self):
renatofilho@484: 
renatofilho@484:         print "Trying to stop pipeline: %s" % self.pipe
renatofilho@484:         try:
renatofilho@484:             if (self.pipeline):
renatofilho@484:                 self.connection.close ()
renatofilho@484:                 self.pipeline.set_state(gst.STATE_NULL)
renatofilho@484:         except gobject.GError, e:
renatofilho@484:             print "Error: " + str(e)
renatofilho@484: 
renatofilho@484:     def __on_bus_message (self, bus, message, stream_data):
renatofilho@484: 
renatofilho@484:         t = message.type
renatofilho@484:         if (t == gst.MESSAGE_STATE_CHANGED):
renatofilho@484:             oldstate = -1
renatofilho@484:             newstate = -1
renatofilho@484:             pending = -1
renatofilho@484:             oldstate, newstate, pending = message.parse_state_changed ()
renatofilho@484:             if ((oldstate == gst.STATE_READY) and \
renatofilho@484:                 (newstate == gst.STATE_PAUSED) and \
renatofilho@484:                 (pending == gst.STATE_VOID_PENDING) and \
renatofilho@484:                 (stream_data.Ready == False)):
renatofilho@484:                 state_changed_status, current_state, pending_state = stream_data.Pipe.get_state () 
renatofilho@484: 		if ((current_state == gst.STATE_PAUSED) and \
renatofilho@484:                     (pending_state == gst.STATE_VOID_PENDING)):
renatofilho@484:                     print "Pipe paused"
renatofilho@484:                     stream_data.Loop.quit ()
renatofilho@484:                     stream_data.Ready = True
renatofilho@484:         elif (t == gst.MESSAGE_ERROR):
renatofilho@484:             err, debug = message.parse_error()
renatofilho@484: 	    print "Error: %s" % err, debug
renatofilho@484:             stream_data.Loop.quit ()
renatofilho@484:             stream_data.Ready = False
renatofilho@484: 
renatofilho@484:         return True
renatofilho@484:  
renatofilho@484:     def __on_decode_unknown_type (self, decode, pad, caps, stream_data):
renatofilho@484: 
renatofilho@484:         print "Unknown Type"
renatofilho@484:         return None
renatofilho@484: 
renatofilho@484:     def __on_decode_new_pad (self, decode, pad, arg1, stream_data):
renatofilho@484:         
renatofilho@484:         caps = pad.get_caps().to_string()
renatofilho@484:         print "New pad " + caps
renatofilho@484: 	if (caps.rfind ("audio") != -1):
renatofilho@484:             apad = stream_data.Abin.get_pad ("sink")
renatofilho@484:             if (pad.link (apad) != gst.PAD_LINK_OK):
renatofilho@484:                 print "Error on link audio pad"
renatofilho@484:                 return None
renatofilho@484:         elif (caps.rfind ("video") != -1):
renatofilho@484:             vpad = stream_data.Vbin.get_pad ("sink")
renatofilho@484:             if (pad.link (vpad) != gst.PAD_LINK_OK):
renatofilho@484:                 print "Error on link video pad"
renatofilho@484:                 return None
renatofilho@484:         else:
renatofilho@484:             print "Invalid caps"
renatofilho@484: 
renatofilho@484: