sciveo 0.1.73__tar.gz → 0.1.75__tar.gz

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

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

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

@@ -27,9 +27,10 @@ def main():
   parser.add_argument(
     'command',
     choices=[
-      'init', 'monitor', 'scan', 'nvr', 'predictors-server',
-      'media-server', 'media-run',
-      'watchdog'
+      'init', 'monitor', 'scan',
+      'nvr', 'rtsp',
+      'predictors-server', 'media-server', 'media-run',
+      'watchdog',
     ],
     help='Command to execute')
 
@@ -46,20 +47,31 @@ 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')
   parser.add_argument('--execute', type=str, default=None, help='Execute command')
   parser.add_argument('--pid', type=int, default=None, help='Process PID')
+  parser.add_argument('--serial', type=str, default=None, help='Serial Name')
 
   args = parser.parse_args()
 
   if args.command == 'monitor':
-    from sciveo.monitoring.start import MonitorStart
-    MonitorStart(period=args.period, block=args.block, output_path=args.output_path)()
+    if args.src is None:
+      from sciveo.monitoring.start import MonitorStart
+      MonitorStart(period=args.period, block=args.block, output_path=args.output_path)()
+    elif args.src.startswith("power"):
+      from sciveo.monitoring.power.ems300 import PowerEMS300
+      mon = PowerEMS300(serial=args.serial, host=args.host, port=args.port, period=args.period, output_path=args.output_path)
+      mon.start()
+      while(True):
+        time.sleep(3600)
   elif args.command == 'scan':
     from sciveo.network.tools import NetworkTools
     host=args.host
@@ -86,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.73 → 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/monitoring/power/ems300.py

@@ -0,0 +1,107 @@
+#
+# Stanislav 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 s.georgiev@softel.bg.
+#
+# 2025
+#
+
+import os
+import json
+import datetime
+import time
+from pymodbus.client import ModbusTcpClient
+from sciveo.tools.logger import *
+from sciveo.tools.daemon import DaemonBase
+# from sciveo.api.base import APIRemoteClient
+from sciveo.monitoring.power.tools import *
+
+
+class PowerEMS300(DaemonBase):
+  def __init__(self, serial, host, port=502, device_id=247, delay=0.01, period=30, output_path=None):
+    super().__init__(period=period)
+    self.serial = serial
+    self.host = host
+    self.port = port
+    self.device_id = device_id
+    self.delay = delay
+    self.client = None
+
+    if output_path is not None and os.path.isdir(output_path):
+      self.output_path = os.path.join(output_path, f"{self.serial}.json")
+    else:
+      self.output_path = output_path
+
+    # self.api = APIRemoteClient()
+    self.client = ModbusTcpClient(host=self.host, port=self.port)
+
+  def connect(self):
+    if not self.client.connected:
+      if self.client.connect():
+        info(type(self.client).__name__, "connected", (self.host, self.port))
+      else:
+        error("Connect FAIL", (self.host, self.port))
+    return self.client.connected
+
+  def send(self, data):
+    api_result = self.api.POST_SCI("monitor", {"data": data})
+    debug(data, "api_result", api_result)
+
+  def output_write(self, data):
+    if os.path.isfile(self.output_path):
+      with open(self.output_path, "rb+") as fp:
+        fp.seek(-2, 2)
+        fp.truncate()
+      with open(self.output_path, "a") as fp:
+        fp.write(f", {json.dumps(data)}]\n")
+    else:
+      with open(self.output_path, "w") as fp:
+        fp.write(json.dumps([data]) + "\n")
+
+  def save(self, data):
+    try:
+      if self.output_path is not None:
+        info("save", self.output_path, data)
+        self.output_write(data)
+    except Exception as e:
+      error(e, "writing failed for", self.output_path)
+
+  def loop(self):
+    if not self.connect():
+      warning("Not connected", (self.host, self.port))
+      return
+
+    try:
+      data = {
+        "local_time": datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+        "serial": self.serial,
+      }
+
+      current_info = [8026, 8044, 8060, 8776, 8778, 10101, 10103, 10105, 10713, 10715, 10716, 10945]
+      for addr in current_info:
+        result = read_input_registers(self.client, EMS300_REG_MAP, addr, self.device_id)
+        if result is not None:
+          value, name, dtype = result
+          debug(f"{name}: {value}")
+          data[name] = value
+        time.sleep(self.delay)
+
+      # self.send(data)
+      self.save(data)
+    except Exception as e:
+      exception(e)
+
+  def close(self):
+    self.client.close()
+
+
+if __name__ == "__main__":
+
+  mon = PowerEMS300(serial="EMS300-1", host="192.168.86.184", port=502, period=30, output_path="ems300.json")
+  mon.start()
+
+  while(True):
+    time.sleep(30)