sciveo 0.1.74__tar.gz → 0.1.75__tar.gz

{sciveo-0.1.74 → sciveo-0.1.75}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sciveo
-Version: 0.1.74
+Version: 0.1.75
 Description-Content-Type: text/markdown
 Provides-Extra: mon
 Provides-Extra: net

{sciveo-0.1.74 → sciveo-0.1.75}/sciveo/cli.py

@@ -27,8 +27,9 @@ def main():
   parser.add_argument(
     'command',
     choices=[
-      'init', 'monitor', 'scan', 'nvr', 'predictors-server',
-      'media-server', 'media-run',
+      'init', 'monitor', 'scan',
+      'nvr', 'rtsp',
+      'predictors-server', 'media-server', 'media-run',
       'watchdog',
     ],
     help='Command to execute')
@@ -46,9 +47,12 @@ def main():
   parser.add_argument('--output-path', type=str, default=None, help='Output Path')
   parser.add_argument('--width', type=str, default=None, help='width')
   parser.add_argument('--height', type=str, default=None, help='height')
+  parser.add_argument('--fps', type=int, help='FPS')
+  parser.add_argument('--stream', type=str, default=None, help='Stream')
   parser.add_argument('--rate', type=int, help='Rate number')
   parser.add_argument('--processor', type=str, help='Processor name')
   parser.add_argument('--src', type=str, default=None, help='Source')
+  parser.add_argument('--src-id', type=int, default=None, help='Source Id')
   parser.add_argument('--dst', type=str, default=None, help='Destination')
   parser.add_argument('--value', type=float, help='Value')
   parser.add_argument('--threshold', type=float, default=None, help='Threshold')
@@ -94,6 +98,22 @@ def main():
   elif args.command == 'nvr':
     from sciveo.media.capture.nvr import VideoRecorder
     VideoRecorder(args.input_path).start()
+  elif args.command == 'rtsp':
+    if args.src is None and args.src_id is None:
+      from sciveo.media.capture.nvr import RTSPVideoPlayer
+      player = RTSPVideoPlayer(f"rtsp://{args.host}:{args.port}/{args.stream}")
+      player.run()
+    else:
+      from sciveo.media.capture.gst_server import CamFactory, ColorBarFactory
+      if args.src_id is not None:
+        server = CamFactory(args.src_id, args.width, args.height, args.fps)
+      elif args.src is not None:
+        if "color" in args.src:
+          server = ColorBarFactory(args.width, args.height, args.fps)
+        else:
+          # warning("RTSP src", args.src, "not recognised")
+          server = CamFactory(args.src, args.width, args.height, args.fps)
+      server.serve(host=args.host, port=args.port, stream=args.stream)
   elif args.command == 'media-server':
     from sciveo.media.pipelines.server import __START_SCIVEO_MEDIA_SERVER__
     __START_SCIVEO_MEDIA_SERVER__()

sciveo-0.1.75/sciveo/media/capture/gst_server.py

@@ -0,0 +1,137 @@
+#
+# Pavlin Georgiev, Softel Labs
+#
+# This is a proprietary file and may not be copied,
+# distributed, or modified without express permission
+# from the owner. For licensing inquiries, please
+# contact pavlin@softel.bg.
+#
+# 2025
+#
+
+import numpy as np
+
+import gi
+gi.require_version('Gst', '1.0')
+gi.require_version('GstRtspServer', '1.0')
+from gi.repository import Gst, GObject, GstRtspServer, GLib
+Gst.init(None)
+
+from sciveo.tools.logger import *
+from sciveo.media.capture.cam import CameraDaemon
+
+
+class RTSPFactoryBase(GstRtspServer.RTSPMediaFactory):
+  def __init__(self, width, height, fps):
+    super().__init__()
+    self.width = width
+    self.height = height
+    self.fps = fps
+    self.set_shared(True)
+
+  def do_create_element(self, url):
+    pipeline_str = (
+      f"appsrc name=mysrc is-live=true block=true format=time "
+      f"caps=video/x-raw,format=BGR,width={self.width},height={self.height},framerate={self.fps}/1 "
+      "! videoconvert "
+      "! x264enc tune=zerolatency speed-preset=ultrafast "
+      "! rtph264pay config-interval=1 name=pay0 pt=96"
+    )
+    return Gst.parse_launch(pipeline_str)
+
+  def do_configure(self, rtsp_media):
+    self.appsrc = rtsp_media.get_element().get_child_by_name("mysrc")
+    self.appsrc.connect("need-data", self.on_need_data)
+    self.timestamp = 0
+
+  def on_need_data(self, src, length):
+    frame = self.next_frame()
+
+    data = frame.tobytes()
+    buf = Gst.Buffer.new_allocate(None, len(data), None)
+    buf.fill(0, data)
+    buf.duration = Gst.SECOND // self.fps
+    buf.pts = buf.dts = self.timestamp
+    self.timestamp += buf.duration
+    src.emit("push-buffer", buf)
+
+  def next_frame(self):
+    return np.zeros((self.height, self.width, 3), dtype=np.uint8)
+
+  def serve(self, host="0.0.0.0", port=8554, stream="test"):
+    rtsp_url = f"rtsp://{host}:{port}/{stream}"
+    server = GstRtspServer.RTSPServer()
+    server.set_address(host)
+    server.set_service(str(port))
+
+    mount_points = server.get_mount_points()
+    mount_points.add_factory(f"/{stream}", self)
+
+    server.attach(None)
+    info(f"RTSP server {type(self).__name__} running at {rtsp_url}")
+    GLib.MainLoop().run()
+
+
+"""
+  RTSP TV-like Color Bars Test Generator
+"""
+class ColorBarsGenerator:
+  """
+  Generates standard vertical color bars (SMPTE/EBU-like) for testing.
+  """
+  def __init__(self, width=640, height=480):
+    self.width = width
+    self.height = height
+    self.colors = [
+      (255, 255, 255),  # White
+      (0, 255, 255),    # Yellow
+      (255, 255, 0),    # Cyan
+      (0, 255, 0),      # Green
+      (255, 0, 255),    # Magenta
+      (0, 0, 255),      # Red
+      (255, 0, 0),      # Blue
+      (0, 0, 0)         # Black
+    ]
+    self.n_colors = len(self.colors)
+
+  def next_frame(self):
+    frame = np.zeros((self.height, self.width, 3), dtype=np.uint8)
+    bar_width = self.width // self.n_colors
+
+    for i, color in enumerate(self.colors):
+      start_x = i * bar_width
+      end_x = start_x + bar_width if i < self.n_colors - 1 else self.width
+      frame[:, start_x:end_x] = color
+
+    return frame
+
+
+class ColorBarFactory(RTSPFactoryBase):
+  def __init__(self, width, height, fps):
+    super().__init__(width, height, fps)
+    self.gen = ColorBarsGenerator(width=width, height=height)
+
+  def next_frame(self):
+    return self.gen.next_frame()
+
+
+class CamFactory(ColorBarFactory):
+  def __init__(self, cam_id, width, height, fps):
+    super().__init__(width, height, fps)
+    self.cam = CameraDaemon(cam_id=cam_id)
+    self.cam.start()
+
+  def next_frame(self):
+    frame = self.cam.read()
+    if frame is None:
+      frame = super().next_frame()
+    return frame
+
+
+if __name__ == "__main__":
+
+    # server = ColorBarFactory(640, 480, 10)
+    # server.serve(host="0.0.0.0", port=8554, stream="test")
+
+    server = CamFactory(0, 640, 480, 30)
+    server.serve(host="0.0.0.0", port=8554, stream="camera")

{sciveo-0.1.74 → sciveo-0.1.75}/sciveo/media/capture/nvr.py

@@ -197,5 +197,63 @@ class VideoRecorder:
       exception(e, cmd)
 
 
+class RTSPVideoPlayer:
+  def __init__(self, rtsp_url, name=None):
+    self.rtsp_url = rtsp_url
+    if name is None:
+      name = rtsp_url
+    self.name = name
+    self.cap = None
+    self.is_running = False
+
+  def open(self):
+    info("Opening RTSP", self.rtsp_url)
+    self.cap = cv2.VideoCapture(self.rtsp_url)
+    if not self.cap.isOpened():
+      error("Failed to open RTSP URL", self.rtsp_url)
+      self.cap = None
+
+  def close(self):
+    if self.cap:
+      info("Closing RTSP", self.rtsp_url)
+      try:
+        self.cap.release()
+      except:
+        pass
+      self.cap = None
+    try:
+      cv2.destroyWindow(self.window_name)
+    except:
+      pass
+
+  def run(self):
+    self.is_running = True
+    while(self.is_running):
+      self.loop()
+
+  def loop(self):
+    if self.cap is None:
+      self.open()
+      if self.cap is None:
+        time.sleep(1)
+        return
+
+    ret, frame = self.cap.read()
+    if not ret or frame is None:
+      error("Failed to read frame, reconnecting...")
+      self.close()
+      time.sleep(1)
+      return
+
+    cv2.imshow(self.name, frame)
+    if cv2.waitKey(1) & 0xFF == ord('q'):
+      info("Quit key pressed, stopping player")
+      self.close()
+      self.is_running = False
+
+  def finalise(self):
+    self.close()
+
+
 if __name__ == '__main__':
   VideoRecorder("./cams.json").start()

sciveo-0.1.75/sciveo/media/capture/rtsp.py

@@ -0,0 +1,80 @@
+#
+# Pavlin Georgiev, Softel Labs
+#
+# This is a proprietary file and may not be copied,
+# distributed, or modified without express permission
+# from the owner. For licensing inquiries, please
+# contact pavlin@softel.bg.
+#
+# 2025
+#
+
+import time
+import cv2
+
+from sciveo.tools.logger import *
+
+
+class RTSPVideoPlayer:
+  def __init__(self, rtsp_url, name=None):
+    self.rtsp_url = rtsp_url
+    if name is None:
+      name = rtsp_url
+    self.name = name
+    self.cap = None
+    self.is_running = False
+
+  def open(self):
+    info("Opening RTSP", self.rtsp_url)
+    self.cap = cv2.VideoCapture(self.rtsp_url)
+    if not self.cap.isOpened():
+      error("Failed to open RTSP URL", self.rtsp_url)
+      self.cap = None
+
+  def close(self):
+    if self.cap:
+      info("Closing RTSP", self.rtsp_url)
+      try:
+        self.cap.release()
+      except:
+        pass
+      self.cap = None
+    try:
+      cv2.destroyWindow(self.window_name)
+    except:
+      pass
+
+  def run(self):
+    self.is_running = True
+    while(self.is_running):
+      self.loop()
+
+  def loop(self):
+    if self.cap is None:
+      self.open()
+      if self.cap is None:
+        time.sleep(1)
+        return
+
+    ret, frame = self.cap.read()
+    if not ret or frame is None:
+      error("Failed to read frame, reconnecting...")
+      self.close()
+      time.sleep(1)
+      return
+
+    cv2.imshow(self.name, frame)
+    if cv2.waitKey(1) & 0xFF == ord('q'):
+      info("Quit key pressed, stopping player")
+      self.close()
+      self.is_running = False
+
+  def finalise(self):
+    self.close()
+
+
+if __name__ == '__main__':
+
+  player = RTSPVideoPlayer("rtsp://192.168.10.118:8554/test")
+  # player = RTSPVideoPlayer("rtsp://192.168.10.118:8554/camera")
+  player.run()

sciveo-0.1.75/sciveo/version.py

@@ -0,0 +1,2 @@
+
+__version__ = '0.1.75'

{sciveo-0.1.74 → sciveo-0.1.75}/sciveo.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sciveo
-Version: 0.1.74
+Version: 0.1.75
 Description-Content-Type: text/markdown
 Provides-Extra: mon
 Provides-Extra: net

{sciveo-0.1.74 → sciveo-0.1.75}/sciveo.egg-info/SOURCES.txt

@@ -29,9 +29,11 @@ sciveo/db/read_table.py
 sciveo/media/__init__.py
 sciveo/media/capture/__init__.py
 sciveo/media/capture/cam.py
+sciveo/media/capture/gst_server.py
 sciveo/media/capture/motion_detection.py
 sciveo/media/capture/nvr.py
 sciveo/media/capture/readers.py
+sciveo/media/capture/rtsp.py
 sciveo/media/ml/__init__.py
 sciveo/media/ml/base.py
 sciveo/media/ml/encoders/__init__.py

sciveo-0.1.74/sciveo/version.py

@@ -1,2 +0,0 @@
-
-__version__ = '0.1.74'