Public feed
#604385
#!/usr/bin/env pythonimport pygstpygst.require('0.10')import gstimport sysimport gobjectgobject.threads_init()class PyTee(gst.Element): __gstdetails__ = ('Tee Plugin', "tee", "Simple Tee Plugin", "Corey O.") _srcpadtemplate = gst.PadTemplate ("src%d", gst.PAD_SRC, gst.PAD_REQUEST, gst.caps_new_any()) _sinkpadtemplate = gst.PadTemplate ("sink", gst.PAD_SINK, gst.PAD_ALWAYS, gst.caps_new_any()) __gsttemplates__ = (_srcpadtemplate, _sinkpadtemplate) def __init__(self): gst.Element.__init__(self) self.src_pad_count = 0 self.src_pads = [] self.sinkpad = gst.Pad(self._sinkpadtemplate) self.add_pad(self.sinkpad) self.sinkpad.set_chain_function(self.chainfunc) self.sinkpad.set_event_function(self.eventfunc) def do_request_new_pad(self, template, name=None): name = "src%d" % self.src_pad_count self.src_pad_count += 1 pad = gst.Pad(template, name) self.src_pads.append(pad) self.add_pad(pad) return pad def do_release_pad(self, pad): self.src_pads.pop(self.src_pads.index(pad)) self.remove_pad(pad) def chainfunc(self, pad, buffer): if len(self.src_pads) == 0: return gst.FLOW_NOT_LINKED else: for pad in self.src_pads: status = pad.push(buffer) if status != gst.FLOW_OK: return status return gst.FLOW_OK def eventfunc(self, pad, event): for srcpad in self.src_pads: status = srcpad.push_event(event) return statusgobject.type_register(PyTee)gst.element_register(PyTee, 'pytee')def on_message(bus, message): type = message.type if type == gst.MESSAGE_ERROR: err, debug = message.parse_error() print 'MESSAGE_ERROR =', err print 'MESSAGE_DEBUG =', debugdef main(args): # NOTE: the following two pipelines work properly #pipeline = gst.parse_launch ("fakesrc ! pytee name=tee .src0 ! fakesink") #pipeline = gst.parse_launch ("fakesrc ! pytee name=tee .src0 ! fakesink tee.src1 ! fakesink") # this pipeline causes a segfault pipeline = gst.parse_launch ("fakesrc ! pytee name=tee ! fakesink") bus = pipeline.get_bus() bus.add_signal_watch() bus.connect('message', on_message) pipeline.set_state(gst.STATE_PLAYING) try: gobject.MainLoop().run() except KeyboardInterrupt: pipeline.set_state(gst.STATE_NULL) return 0 return -1if __name__ == '__main__': sys.exit(main(sys.argv))