renatofilho@484: import pygst renatofilho@484: pygst.require("0.10") renatofilho@484: import gst renatofilho@484: import gobject 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): 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.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: 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: 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_mpeg4", "vencode") renatofilho@484: vpay = gst.element_factory_make ("rtpmp4vpay", "vpay") renatofilho@484: vsink = gst.element_factory_make ("udpsink", "vsink") renatofilho@484: renatofilho@484: if (None in [vbin, vqueue, vscale, vrate, vencode, vpay, vsink]): 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: vencode.set_property ("bitrate", 256000) renatofilho@484: vencode.set_property ("me-method", 2) renatofilho@484: renatofilho@484: vsink.set_property ("host", "224.0.0.1") renatofilho@484: vsink.set_property ("port", 5000) renatofilho@484: renatofilho@484: vbin.add (vqueue, vscale, vrate, vencode, vpay, vsink) renatofilho@484: if (not gst.element_link_many (vqueue, vscale, vrate, vencode, vpay, vsink)): 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: 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 ("faac", "aencode") renatofilho@484: apay = gst.element_factory_make ("rtpmp4gpay", "apay") renatofilho@484: asink = gst.element_factory_make ("udpsink", "asink") renatofilho@484: renatofilho@484: if (None in [abin, aqueue, aconvert, aencode, apay, asink]): renatofilho@484: print "Fail to create video encode elements." renatofilho@484: return None renatofilho@484: renatofilho@484: asink.set_property ("host", "224.0.0.1") renatofilho@484: asink.set_property ("port", 5002) renatofilho@484: renatofilho@484: abin.add (aqueue, aconvert, aencode, apay, asink) renatofilho@484: if (not gst.element_link_many (aqueue, aconvert, aencode, apay, asink)): 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: renatofilho@484: self.pipe.add (src, decode, abin, vbin) renatofilho@484: gst.element_link_many (src, decode) renatofilho@484: renatofilho@484: stream_data = self.StreamData (self.pipe, abin, vbin) 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: 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.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: self.__fill_sink_pads (stream_data) 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: renatofilho@484: def __fill_sink_pads (self, stream_data): renatofilho@484: renatofilho@484: asink = stream_data.Abin.get_by_name ("asink") renatofilho@484: vsink = stream_data.Vbin.get_by_name ("vsink") renatofilho@484: renatofilho@484: asink_pad = asink.get_pad ("sink") renatofilho@484: stream_data.ACaps = asink_pad.get_negotiated_caps().to_string() renatofilho@484: print "ACAPS " + stream_data.ACaps renatofilho@484: renatofilho@484: vsink_pad = vsink.get_pad ("sink") renatofilho@484: stream_data.VCaps = vsink_pad.get_negotiated_caps().to_string() renatofilho@484: print "ACAPS " + stream_data.VCaps renatofilho@484: renatofilho@484: 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: