renatofilho@483: import pygst renatofilho@483: pygst.require("0.10") renatofilho@483: import gst renatofilho@483: import gobject renatofilho@483: renatofilho@483: class Media: renatofilho@483: class StreamData: renatofilho@483: stream_count = 0 renatofilho@483: renatofilho@483: def __init__ (self, pipe, abin, vbin): renatofilho@483: renatofilho@483: self.stream_count += 1 renatofilho@483: self.Id = self.stream_count renatofilho@483: self.Pipe = pipe renatofilho@483: self.Abin = abin renatofilho@483: self.Vbin = vbin renatofilho@483: self.Loop = gobject.MainLoop() renatofilho@483: self.ACaps = "" renatofilho@483: self.VCaps = "" renatofilho@483: self.Ready = False renatofilho@483: renatofilho@483: renatofilho@483: def __init__(self, config): renatofilho@483: # set gstreamer basic options renatofilho@483: self.config = config renatofilho@483: self.pipe = None renatofilho@483: self.streams = [] renatofilho@483: renatofilho@483: renatofilho@483: def setup(self, filename, mux, vcodec, vbitrate, renatofilho@483: fps, acodec, abitrate, width, height, port, options): renatofilho@483: renatofilho@483: ## Pipelines renatofilho@483: self.pipe = gst.Pipeline () renatofilho@483: uri = "file://" + filename renatofilho@483: print "Opening Uri:" + uri renatofilho@483: src = gst.element_make_from_uri (gst.URI_SRC, uri, "src") renatofilho@483: if (src is None): renatofilho@483: return None renatofilho@483: renatofilho@483: decode = gst.element_factory_make ("decodebin", "decode") renatofilho@483: if (decode is None): renatofilho@483: return None renatofilho@483: renatofilho@483: renatofilho@483: #video encode renatofilho@483: #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@483: vbin = gst.Bin () renatofilho@483: vqueue = gst.element_factory_make ("queue", "vqueue") renatofilho@483: vscale = gst.element_factory_make ("videoscale", "vscale") renatofilho@483: vrate = gst.element_factory_make ("videorate", "vrate") renatofilho@483: vencode = gst.element_factory_make ("ffenc_mpeg4", "vencode") renatofilho@483: vpay = gst.element_factory_make ("rtpmp4vpay", "vpay") renatofilho@483: vsink = gst.element_factory_make ("udpsink", "vsink") renatofilho@483: renatofilho@483: if (None in [vbin, vqueue, vscale, vrate, vencode, vpay, vsink]): renatofilho@483: print "Fail to create video encode elements." renatofilho@483: return None renatofilho@483: renatofilho@483: vscale_pad = vscale.get_pad("sink") renatofilho@483: if (vscale_pad is None): renatofilho@483: print "Fail to get vscale sink pad." renatofilho@483: return None renatofilho@483: renatofilho@483: vscale_caps = gst.caps_from_string ("video/x-raw-yuv, width=%s, height=%s" % (width, height)) renatofilho@483: if (vscale_caps is None): renatofilho@483: print "Fail to create video caps" renatofilho@483: return None renatofilho@483: renatofilho@483: if (not vscale_pad.set_caps (vscale_caps)): renatofilho@483: print "Fail to set video output caps" renatofilho@483: return None renatofilho@483: renatofilho@483: vencode.set_property ("bitrate", 256000) renatofilho@483: vencode.set_property ("me-method", 2) renatofilho@483: renatofilho@483: vsink.set_property ("host", "224.0.0.1") renatofilho@483: vsink.set_property ("port", 5000) renatofilho@483: renatofilho@483: vbin.add (vqueue, vscale, vrate, vencode, vpay, vsink) renatofilho@483: if (not gst.element_link_many (vqueue, vscale, vrate, vencode, vpay, vsink)): renatofilho@483: print "Fail to link video elements" renatofilho@483: return None renatofilho@483: renatofilho@483: vbin.add_pad (gst.GhostPad ("sink", vqueue.get_pad ("sink"))) renatofilho@483: renatofilho@483: #audio encode renatofilho@483: #audio/x-raw-int ! queue ! audioconvert ! faac ! rtpmp4gpay ! udpsink name=upd_audio host=224.0.0.1 port=5002 renatofilho@483: abin = gst.Bin () renatofilho@483: aqueue = gst.element_factory_make ("queue", "vqueue") renatofilho@483: aconvert = gst.element_factory_make ("audioconvert", "aconvert") renatofilho@483: aencode = gst.element_factory_make ("faac", "aencode") renatofilho@483: apay = gst.element_factory_make ("rtpmp4gpay", "apay") renatofilho@483: asink = gst.element_factory_make ("udpsink", "asink") renatofilho@483: renatofilho@483: if (None in [abin, aqueue, aconvert, aencode, apay, asink]): renatofilho@483: print "Fail to create video encode elements." renatofilho@483: return None renatofilho@483: renatofilho@483: asink.set_property ("host", "224.0.0.1") renatofilho@483: asink.set_property ("port", 5002) renatofilho@483: renatofilho@483: abin.add (aqueue, aconvert, aencode, apay, asink) renatofilho@483: if (not gst.element_link_many (aqueue, aconvert, aencode, apay, asink)): renatofilho@483: print "Fail to link video elements" renatofilho@483: return None renatofilho@483: renatofilho@483: abin.add_pad (gst.GhostPad ("sink", aqueue.get_pad ("sink"))) renatofilho@483: renatofilho@483: self.pipe.add (src, decode, abin, vbin) renatofilho@483: gst.element_link_many (src, decode) renatofilho@483: renatofilho@483: stream_data = self.StreamData (self.pipe, abin, vbin) renatofilho@483: renatofilho@483: bus = self.pipe.get_bus() renatofilho@483: bus.add_signal_watch() renatofilho@483: bus.connect("message", self.__on_bus_message, stream_data) renatofilho@483: renatofilho@483: decode.connect("new-decoded-pad", self.__on_decode_new_pad, stream_data) renatofilho@483: decode.connect("unknown-type", self.__on_decode_unknown_type, stream_data) renatofilho@483: renatofilho@483: renatofilho@483: self.pipe.set_state (gst.STATE_PAUSED) renatofilho@483: print "Running Pipe" renatofilho@483: stream_data.Loop.run () renatofilho@483: print "End run" renatofilho@483: renatofilho@483: a_caps = stream_data.ACaps renatofilho@483: v_caps = stream_data.VCaps renatofilho@483: stream_id = stream_data.Id renatofilho@483: renatofilho@483: self.streams.append (stream_data) renatofilho@483: renatofilho@483: def play(self): renatofilho@483: renatofilho@483: print "Trying to play pipeline: %s" % self.pipe renatofilho@483: try: renatofilho@483: if (self.pipe): renatofilho@483: self.pipe.set_state(gst.STATE_PLAYING) renatofilho@483: except gobject.GError, e: renatofilho@483: print "Error: " + str(e) renatofilho@483: renatofilho@483: renatofilho@483: def stop(self): renatofilho@483: renatofilho@483: print "Trying to stop pipeline: %s" % self.pipe renatofilho@483: try: renatofilho@483: if (self.pipeline): renatofilho@483: self.pipeline.set_state(gst.STATE_NULL) renatofilho@483: except gobject.GError, e: renatofilho@483: print "Error: " + str(e) renatofilho@483: renatofilho@483: def __on_bus_message (self, bus, message, stream_data): renatofilho@483: renatofilho@483: t = message.type renatofilho@483: if (t == gst.MESSAGE_STATE_CHANGED): renatofilho@483: oldstate = -1 renatofilho@483: newstate = -1 renatofilho@483: pending = -1 renatofilho@483: oldstate, newstate, pending = message.parse_state_changed () renatofilho@483: if ((oldstate == gst.STATE_READY) and \ renatofilho@483: (newstate == gst.STATE_PAUSED) and \ renatofilho@483: (pending == gst.STATE_VOID_PENDING) and \ renatofilho@483: (stream_data.Ready == False)): renatofilho@483: state_changed_status, current_state, pending_state = stream_data.Pipe.get_state () renatofilho@483: if ((current_state == gst.STATE_PAUSED) and \ renatofilho@483: (pending_state == gst.STATE_VOID_PENDING)): renatofilho@483: print "Pipe paused" renatofilho@483: self.__fill_sink_pads (stream_data) renatofilho@483: stream_data.Loop.quit () renatofilho@483: stream_data.Ready = True renatofilho@483: elif (t == gst.MESSAGE_ERROR): renatofilho@483: err, debug = message.parse_error() renatofilho@483: print "Error: %s" % err, debug renatofilho@483: stream_data.Loop.quit () renatofilho@483: stream_data.Ready = False renatofilho@483: renatofilho@483: return True renatofilho@483: renatofilho@483: renatofilho@483: def __fill_sink_pads (self, stream_data): renatofilho@483: renatofilho@483: asink = stream_data.Abin.get_by_name ("asink") renatofilho@483: vsink = stream_data.Vbin.get_by_name ("vsink") renatofilho@483: renatofilho@483: asink_pad = asink.get_pad ("sink") renatofilho@483: stream_data.ACaps = asink_pad.get_negotiated_caps().to_string() renatofilho@483: print "ACAPS " + stream_data.ACaps renatofilho@483: renatofilho@483: vsink_pad = vsink.get_pad ("sink") renatofilho@483: stream_data.VCaps = vsink_pad.get_negotiated_caps().to_string() renatofilho@483: print "ACAPS " + stream_data.VCaps renatofilho@483: renatofilho@483: renatofilho@483: renatofilho@483: def __on_decode_unknown_type (self, decode, pad, caps, stream_data): renatofilho@483: renatofilho@483: print "Unknown Type" renatofilho@483: return None renatofilho@483: renatofilho@483: def __on_decode_new_pad (self, decode, pad, arg1, stream_data): renatofilho@483: renatofilho@483: caps = pad.get_caps().to_string() renatofilho@483: print "New pad " + caps renatofilho@483: if (caps.rfind ("audio") != -1): renatofilho@483: apad = stream_data.Abin.get_pad ("sink") renatofilho@483: if (pad.link (apad) != gst.PAD_LINK_OK): renatofilho@483: print "Error on link audio pad" renatofilho@483: return None renatofilho@483: elif (caps.rfind ("video") != -1): renatofilho@483: vpad = stream_data.Vbin.get_pad ("sink") renatofilho@483: if (pad.link (vpad) != gst.PAD_LINK_OK): renatofilho@483: print "Error on link video pad" renatofilho@483: return None renatofilho@483: else: renatofilho@483: print "Invalid caps" renatofilho@483: renatofilho@483: