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