I am not able to restart or control the RTSP media if I am using nvh264enc instead of x264enc at the launch_str in the following code, because of this issue, we are not able to properly close the connections from clients. Please let me know if there is any solution to this.
#!/usr/bin/python
# --------------------------------------------------------------------------- #
# Supporting arguments
# --------------------------------------------------------------------------- #
# --------------------------------------------------------------------------- #
# Use gi to import GStreamer functionality
# --------------------------------------------------------------------------- #
import gi
gi.require_version('Gst', '1.0')
gi.require_version('GstRtspServer', '1.0')
gi.require_version('GstVideo', '1.0')
import argparse
import logging
import json
import time
import os.path
import subprocess
import signal
import sys
from ctypes import *
from gi.repository import GObject, Gst, Gio, GstVideo, GstRtspServer, GLib
from threading import Thread, Lock
import gi
gi.require_version('Gst', '1.0')
gi.require_version('GstRtspServer', '1.0')
gi.require_version('GstVideo', '1.0')
parser = argparse.ArgumentParser(description="gst-rtsp-launch-py V0.2")
parser.add_argument('-v', '--verbose', action='store_true',
help='Make script chatty')
args = parser.parse_args()
# --------------------------------------------------------------------------- #
# configure the service logging
# --------------------------------------------------------------------------- #
logging.basicConfig()
log = logging.getLogger()
# --------------------------------------------------------------------------- #
# import misc standard libraries
# --------------------------------------------------------------------------- #
if args.verbose:
log.setLevel(logging.DEBUG)
else:
log.setLevel(logging.INFO)
def on_debug(category, level, dfile, dfctn, dline, source, message, user_data):
if source:
print('Debug {} {}'.format(
Gst.DebugLevel.get_name(level), message.get()))
else:
print('Debug {}: {}'.format(
Gst.DebugLevel.get_name(level), message.get()))
if not Gst.debug_is_active():
Gst.debug_set_active(True)
level = Gst.debug_get_default_threshold()
Gst.debug_set_default_threshold(Gst.DebugLevel.INFO)
if level < Gst.DebugLevel.ERROR:
Gst.debug_set_default_threshold(Gst.DebugLevel.WARNING)
Gst.debug_add_log_function(on_debug, None)
Gst.debug_remove_log_function(Gst.debug_log_default)
cam_mutex = Lock()
# -------------------
class StreamServer:
def __init__(self):
signal.signal(signal.SIGTERM, self.exit_gracefully)
Gst.init(None)
self.mainloop = GLib.MainLoop()
self.server = GstRtspServer.RTSPServer()
self.mounts = self.server.get_mount_points()
self.factory = GstRtspServer.RTSPMediaFactory()
# Factory must be shared to allow multiple connections
self.factory.set_shared(True)
self.context_id = 0
self.running = False
self.stayAwake = True
GLib.threads_init()
log.info("StreamServer initialized")
def exit_gracefully(self, signum, frame):
self.stop()
self.stayAwake = False
def launch(self):
log.debug("StreamServer.launch")
if self.running:
log.debug("StreamServer.launch called on running instance.")
self.stop() # Need to stop any instances first
launch_str = 'videotestsrc ' \
' ! queue' \
' ! videoconvert' \
' ! video/x-raw,format=I420' \
' ! x264enc' \
' ! rtph264pay config-interval=1 pt=96 name=pay0' \
''
log.debug(launch_str)
cam_mutex.acquire()
try:
log.info("Starting service on port " +
str(8555)+" at url /video1")
self.factory.set_launch(launch_str)
self.server.set_service("8555")
self.mounts.add_factory("/video", self.factory)
self.context_id = self.server.attach(None)
# mainloop.run()
self.mainthread = Thread(target=self.mainloop.run)
self.mainthread.daemon = True
self.mainthread.start()
self.running = True
finally:
cam_mutex.release()
log.info("Running RTSP Server")
def start(self):
while True:
print("Sleeping for 10 seconds")
time.sleep(20)
self.restart()
def disconnect_all(self, a, b):
return GstRtspServer.RTSPFilterResult.REMOVE
def stop(self):
if self.running:
print("Suspending server")
log.debug("Suspending RTSP Server")
cam_mutex.acquire()
try:
print(self.context_id)
print(self.server.client_filter(self.disconnect_all))
time.sleep(0.3)
self.mainloop.quit()
self.mainthread.join()
self.mounts.remove_factory("/video")
GLib.Source.remove(self.context_id)
self.running = False
finally:
cam_mutex.release()
def restart(self):
# TODO: Manipulate the running pipe rather than destroying and recreating it.
self.stop()
self.launch()
if __name__ == '__main__':
streamServer = StreamServer()
streamServer.launch()
streamServer.start()
Please find the below details as well.
the GPU is NVIDIA GeForce RTX 3050 Ti Laptop GPU
The gstreamer version is 1.20.3
nvcc version is
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Tue_May__3_18:49:52_PDT_2022
Cuda compilation tools, release 11.7, V11.7.64
Build cuda_11.7.r11.7/compiler.31294372_0
using Video_Codec_SDK_9.0.20 to build the nvenc and nvdec
