|
楼主 |
发表于 2019-8-6 10:25:21
|
显示全部楼层
本帖最后由 勇Z的心 于 2019-8-6 11:04 编辑
import cv2
import gi
gi.require_version('Gst', '1.0')
gi.require_version('GstRtspServer', '1.0')
from gi.repository import Gst, GstRtspServer, GObject
class SensorFactory(GstRtspServer.RTSPMediaFactory):
def __init__(self, **properties):
super(SensorFactory, self).__init__(**properties)
self.cap = cv2.VideoCapture(0)
self.number_frames = 0
self.fps = 30
self.duration = 1 / self.fps * Gst.SECOND # duration of a frame in nanoseconds
self.launch_string = 'appsrc name=source is-live=true block=true format=GST_FORMAT_TIME '\
'caps=video/x-raw,format=BGR,width=640,height=480,framerate={}/1 '\
'! videoconvert ! video/x-raw,format=I420 '\
'! mpph264enc '\
'! rtph264pay config-interval=1 name=pay0 pt=96'.format(self.fps)
print('init end')
def on_need_data(self, src, lenght):
print('on_need_data start')
if self.cap.isOpened():
ret, frame = self.cap.read()
if ret:
data = frame.tostring()
buf = Gst.Buffer.new_allocate(None, len(data), None)
buf.fill(0, data)
buf.duration = self.duration
timestamp = self.number_frames * self.duration
buf.pts = buf.dts = int(timestamp)
buf.offset = timestamp
self.number_frames += 1
retval = src.emit('push-buffer', buf)
print('pushed buffer, frame {}, duration {} ns, durations {} s'.format(self.number_frames, self.duration, self.duration / Gst.SECOND))
if retval != Gst.FlowReturn.OK:
print(retval)
print('on_need_data end')
def do_create_element(self, url):
print('do_create_element')
return Gst.parse_launch(self.launch_string)
def do_configure(self, rtsp_media):
print('do_configure start')
self.number_frames = 0
appsrc = rtsp_media.get_element().get_child_by_name('source')
appsrc.connect('need-data', self.on_need_data)
print('do_configure end')
class GstServer(GstRtspServer.RTSPServer):
def __init__(self, **properties):
print('GstServer __init__ start')
super(GstServer, self).__init__(**properties)
self.factory = SensorFactory()
self.factory.set_shared(True)
self.get_mount_points().add_factory("/test", self.factory)
self.attach(None)
print('GstServer __init__ end')
GObject.threads_init()
Gst.init(None)
server = GstServer()
loop = GObject.MainLoop()
loop.run()
|
|