# HG changeset patch
# User renatofilho
# Date 1176313250 -3600
# Node ID 605cbc483e09d48470fd4d81154d00dc542255f4
# Parent 92acf23b87f0ca7e8a17edb1499b6ba29be7f48e
[svn r526] - bug fixes;
diff -r 92acf23b87f0 -r 605cbc483e09 gmyth-stream/server/plugins/media/gstreamer.py
--- a/gmyth-stream/server/plugins/media/gstreamer.py Wed Apr 11 18:39:39 2007 +0100
+++ b/gmyth-stream/server/plugins/media/gstreamer.py Wed Apr 11 18:40:50 2007 +0100
@@ -1,82 +1,105 @@
+#vim:ts=4:sw=4:et
import pygst
pygst.require("0.10")
import gst
import gobject
import socket
+from threading import Thread
class Media:
+ class StreamListener(Thread):
+ def __init__ (self, stream_data):
+ Thread.__init__(self)
+ self.stream = stream_data
+ print "Thread Created"
+
+ def run (self):
+ #Create socket
+ print "Waiting connection"
+ self.stream.Socket.listen(1)
+ self.stream.Connection, self.stream.Addr = self.stream.Socket.accept ()
+ print "Connection requested"
+ self.stream.Sink.set_property ("fd", self.stream.Connection.fileno())
+ self.stream.Pipe.set_state(gst.STATE_PLAYING)
+ print "PLAYING"
+
+
class StreamData:
stream_count = 0
-
- def __init__ (self, pipe, abin, vbin, sink):
- self.stream_count += 1
- self.Id = self.stream_count
- self.Pipe = pipe
- self.Abin = abin
- self.Vbin = vbin
- self.Sink = sink
- self.Loop = gobject.MainLoop()
- self.ACaps = ""
- self.VCaps = ""
- self.Ready = False
-
+ def __init__ (self, pipe, abin, vbin, sink):
+ self.stream_count += 1
+ self.Id = self.stream_count
+ self.Pipe = pipe
+ self.Abin = abin
+ self.Vbin = vbin
+ self.Sink = sink
+ self.Loop = gobject.MainLoop()
+ self.ACaps = ""
+ self.VCaps = ""
+ self.Ready = False
+ self.Socket = None
+ self.Connection = None
+ self.Addr = None
def __init__(self, config):
# set gstreamer basic options
self.config = config
- self.pipe = None
self.streams = []
- self.socket = None
- self.connection = None
- self.addr = None
+ self.socket = None
+ self.connection = None
+ self.addr = None
+ self.ready = False
- def setup(self, filename, mux, vcodec, vbitrate,
+ def setup(self, uri, mux, vcodec, vbitrate,
fps, acodec, abitrate, width, height, port, options):
## Pipelines
- self.pipe = gst.Pipeline ()
- uri = "file://" + filename
+ pipe = gst.Pipeline ()
print "Opening Uri:" + uri
src = gst.element_make_from_uri (gst.URI_SRC, uri, "src")
if (src is None):
+ print "Fail to create src element"
return None
-
+
decode = gst.element_factory_make ("decodebin", "decode")
if (decode is None):
+ print "Fail to create decodebin"
return None
- mux = gst.element_factory_make ("avimux", "mux")
+ mux = gst.element_factory_make ("avimux", "mux")
if (mux is None):
+ print "Fail to create mux"
return None
sink = gst.element_factory_make ("fdsink", "sink")
if (sink is None):
+ print "Fail to create fdsink"
return None
- #Create socket
- self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
- self.socket.bind(('', int (port)))
-
- #video encode
+
+ #video encode
#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
vbin = gst.Bin ()
vqueue = gst.element_factory_make ("queue", "vqueue")
vscale = gst.element_factory_make ("videoscale", "vscale")
+ colorspace = gst.element_factory_make ("ffmpegcolorspace", "")
vrate = gst.element_factory_make ("videorate", "vrate")
vencode = gst.element_factory_make ("ffenc_h263p", "vencode")
+
if (None in [vbin, vqueue, vscale, vrate, vencode]):
print "Fail to create video encode elements."
return None
+ '''
vscale_pad = vscale.get_pad("sink")
if (vscale_pad is None):
print "Fail to get vscale sink pad."
return None
- vscale_caps = gst.caps_from_string ("video/x-raw-yuv, width=%s, height=%s" % (width, height))
+ vscale_caps = gst.caps_from_string ("video/x-raw-yuv,width=%d,height=(int)%d" % (int(width), int(height)))
if (vscale_caps is None):
print "Fail to create video caps"
return None
@@ -84,21 +107,32 @@
if (not vscale_pad.set_caps (vscale_caps)):
print "Fail to set video output caps"
return None
-
- vbin.add (vqueue, vscale, vrate, vencode)
- if (not gst.element_link_many (vqueue, vscale, vrate, vencode)):
+ '''
+
+ vbin.add (vqueue, vscale, colorspace, vrate, vencode)
+ if (not vqueue.link (vscale)):
print "Fail to link video elements"
return None
-
+
+ if (not vscale.link (colorspace, \
+ gst.caps_from_string ("video/x-raw-yuv,width=%d,height=(int)%d" % (int(width), int(height))))):
+ print "Fail to link video elements"
+ return None
+
+ if (not gst.element_link_many (colorspace, vrate, vencode)):
+ print "Fail to link video elements"
+ return None
+
vbin.add_pad (gst.GhostPad ("sink", vqueue.get_pad ("sink")))
- vbin.add_pad (gst.GhostPad ("src", vencode.get_pad ("src")))
+ vbin.add_pad (gst.GhostPad ("src", vencode.get_pad ("src")))
#audio encode
#audio/x-raw-int ! queue ! audioconvert ! faac ! rtpmp4gpay ! udpsink name=upd_audio host=224.0.0.1 port=5002
abin = gst.Bin ()
aqueue = gst.element_factory_make ("queue", "vqueue")
aconvert = gst.element_factory_make ("audioconvert", "aconvert")
- aencode = gst.element_factory_make ("ffenc_ac3", "aencode")
+ #aencode = gst.element_factory_make ("ffenc_ac3", "aencode")
+ aencode = gst.element_factory_make ("lame", "aencode")
if (None in [abin, aqueue, aconvert, aencode]):
print "Fail to create video encode elements."
@@ -108,81 +142,66 @@
if (not gst.element_link_many (aqueue, aconvert, aencode)):
print "Fail to link video elements"
return None
-
+
abin.add_pad (gst.GhostPad ("sink", aqueue.get_pad ("sink")))
abin.add_pad (gst.GhostPad ("src", aencode.get_pad ("src")))
#Finish Pipeline
- self.pipe.add (src, decode, abin, vbin, mux, sink)
- gst.element_link_many (src, decode)
+ pipe.add (src, decode, abin, vbin, mux, sink)
+ gst.element_link_many (src, decode)
gst.element_link_many (mux, sink)
#Linking decode with mux
mux_audio = mux.get_pad ("audio_0")
mux_video = mux.get_pad ("video_0")
- audio_pad = abin.get_pad ("src")
+ audio_pad = abin.get_pad ("src")
video_pad = vbin.get_pad ("src")
if (audio_pad.link (mux_audio) != gst.PAD_LINK_OK):
print "Fail to link audio with mux"
return None
-
+
if (video_pad.link (mux_video) != gst.PAD_LINK_OK):
print "Fail to link audio with mux"
return None
- stream_data = self.StreamData (self.pipe, abin, vbin, sink)
+ stream_data = self.StreamData (pipe, abin, vbin, sink)
+ bus = pipe.get_bus()
+ bus.add_signal_watch()
+ bus.connect ("message", self.__on_bus_message, stream_data)
- bus = self.pipe.get_bus()
- bus.add_signal_watch()
- bus.connect("message", self.__on_bus_message, stream_data)
-
- decode.connect("new-decoded-pad", self.__on_decode_new_pad, stream_data)
- decode.connect("unknown-type", self.__on_decode_unknown_type, stream_data)
+ decode.connect("new-decoded-pad", self.__on_decode_new_pad, stream_data)
+ decode.connect("unknown-type", self.__on_decode_unknown_type, stream_data)
-
- self.pipe.set_state (gst.STATE_PAUSED)
+ pipe.set_state (gst.STATE_PAUSED)
print "Running Pipe"
- stream_data.Loop.run ()
+ stream_data.Loop.run ()
print "End run"
- a_caps = stream_data.ACaps
- v_caps = stream_data.VCaps
- stream_id = stream_data.Id
+
+ #Create socket
+ stream_data.Socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+ print "Bind on port %d" % port
+ stream_data.Socket.bind(('', int (port)))
self.streams.append (stream_data)
+ return True
- def play(self):
-
- print "Trying to play pipeline: %s" % self.pipe
- try:
- if (self.pipe):
- print "Waiting for connection"
- self.socket.listen(1)
- print "Connection Requested"
- #Create socket
- self.connection, self.addr = self.socket.accept ()
-
- stream_data = self.streams[0]
- stream_data.Sink.set_property ("fd", self.connection.fileno());
- print "Connected"
-
- self.pipe.set_state(gst.STATE_PLAYING)
- except gobject.GError, e:
- print "Error: " + str(e)
-
+ def play(self):
+ stream = self.streams[0]
+ current = self.StreamListener(stream)
+ current.start ()
+ print "Saindo"
+ return True
def stop(self):
+ stream = self.streams[0]
+ stream.Pipe.set_state(gst.STATE_NULL)
+ stream.Connection.close ()
+ return True
- print "Trying to stop pipeline: %s" % self.pipe
- try:
- if (self.pipeline):
- self.connection.close ()
- self.pipeline.set_state(gst.STATE_NULL)
- except gobject.GError, e:
- print "Error: " + str(e)
def __on_bus_message (self, bus, message, stream_data):
@@ -196,30 +215,30 @@
(newstate == gst.STATE_PAUSED) and \
(pending == gst.STATE_VOID_PENDING) and \
(stream_data.Ready == False)):
- state_changed_status, current_state, pending_state = stream_data.Pipe.get_state ()
- if ((current_state == gst.STATE_PAUSED) and \
+ state_changed_status, current_state, pending_state = stream_data.Pipe.get_state ()
+ if ((current_state == gst.STATE_PAUSED) and \
(pending_state == gst.STATE_VOID_PENDING)):
print "Pipe paused"
stream_data.Loop.quit ()
stream_data.Ready = True
elif (t == gst.MESSAGE_ERROR):
err, debug = message.parse_error()
- print "Error: %s" % err, debug
+ print "Error: %s" % err, debug
stream_data.Loop.quit ()
stream_data.Ready = False
return True
-
+
def __on_decode_unknown_type (self, decode, pad, caps, stream_data):
print "Unknown Type"
return None
def __on_decode_new_pad (self, decode, pad, arg1, stream_data):
-
+
caps = pad.get_caps().to_string()
print "New pad " + caps
- if (caps.rfind ("audio") != -1):
+ if (caps.rfind ("audio") != -1):
apad = stream_data.Abin.get_pad ("sink")
if (pad.link (apad) != gst.PAD_LINK_OK):
print "Error on link audio pad"
@@ -232,4 +251,3 @@
else:
print "Invalid caps"
-