nedo-vision-worker-core 0.3.2__py3-none-any.whl → 0.3.4__py3-none-any.whl

nedo_vision_worker_core/util/PlatformDetector.py

@@ -0,0 +1,100 @@
+import os
+import platform
+import sys
+import torch
+
+
+class PlatformDetector:
+    """
+    Detects platform and multimedia stack capabilities.
+    """
+
+    @staticmethod
+    def is_linux() -> bool:
+        return sys.platform.startswith("linux")
+
+    @staticmethod
+    def is_windows() -> bool:
+        return sys.platform.startswith("win")
+
+    @staticmethod
+    def is_macos() -> bool:
+        return sys.platform == "darwin"
+
+    @staticmethod
+    def is_jetson() -> bool:
+        """
+        Determines if the platform is an NVIDIA Jetson device.
+        """
+        try:
+            # Device-tree model (most reliable)
+            if os.path.exists("/proc/device-tree/model"):
+                with open("/proc/device-tree/model", "r") as f:
+                    model = f.read().strip()
+                    if "NVIDIA Jetson" in model:
+                        return True
+
+            # Jetson-specific libs/paths
+            jetson_libraries = ["/usr/lib/aarch64-linux-gnu/tegra", "/etc/nv_tegra_release", "/etc/tegra-release"]
+            if any(os.path.exists(p) for p in jetson_libraries):
+                return True
+
+            # Arch alone is not definitive, but is a signal
+            if platform.machine() == "aarch64" and os.path.exists("/dev/nvhost-ctrl"):
+                return True
+
+        except Exception:
+            pass
+
+        return False
+
+    @staticmethod
+    def get_platform_type() -> str:
+        """
+        Returns a coarse platform type: 'jetson' | 'mac' | 'windows' | 'linux'
+        """
+        if PlatformDetector.is_jetson():
+            return "jetson"
+        if PlatformDetector.is_macos():
+            return "mac"
+        if PlatformDetector.is_windows():
+            return "windows"
+        return "linux"
+
+    @staticmethod
+    def has_gstreamer() -> bool:
+        """
+        Check if OpenCV was built with GStreamer support.
+        """
+        try:
+            import cv2
+            info = cv2.getBuildInformation()
+            return ("GStreamer:                   YES" in info) or ("GStreamer: YES" in info)
+        except Exception:
+            return False
+
+    @staticmethod
+    def has_nvidia_gpu() -> bool:
+        """
+        Heuristic for NVIDIA dGPU presence (desktop/server).
+        """
+        if PlatformDetector.is_windows():
+            return bool(os.environ.get("NVIDIA_VISIBLE_DEVICES", ""))  # WSL/Docker hint
+        if PlatformDetector.is_linux():
+            if os.path.exists("/proc/driver/nvidia/version"):
+                return True
+            if os.environ.get("NVIDIA_VISIBLE_DEVICES", "") not in ("", "none"):
+                return True
+        return False
+    
+    @staticmethod
+    def get_device() -> str:
+        """
+        Check for GPU availability and return the appropriate device.
+        """
+        if torch.cuda.is_available():
+            return "cuda"
+        # Add checks for other devices like MPS if needed
+        # elif torch.backends.mps.is_available():
+        #     return "mps"
+        return "cpu"

{nedo_vision_worker_core-0.3.2.dist-info → nedo_vision_worker_core-0.3.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker-core
-Version: 0.3.2
+Version: 0.3.4
 Summary: Nedo Vision Worker Core Library for AI Vision Processing
 Author-email: Willy Achmat Fauzi <willy.achmat@gmail.com>
 Maintainer-email: Willy Achmat Fauzi <willy.achmat@gmail.com>

{nedo_vision_worker_core-0.3.2.dist-info → nedo_vision_worker_core-0.3.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-nedo_vision_worker_core/__init__.py,sha256=gOVK9WXdcVvRyelxasgOBdnkqE2wbdbRZ3VnuSXgobc,1924
+nedo_vision_worker_core/__init__.py,sha256=0iPoN0eHddv9i7aZ8YJKl2Rn_P_BTsxTJMPggoVd_kc,1924
 nedo_vision_worker_core/cli.py,sha256=8YuKWsIgICUYXE_QtwyU3WzGhVjTWiAo5uzpFOmjNc8,5766
 nedo_vision_worker_core/core_service.py,sha256=dnHNjbslOeyeWqHDFnk_yKdfTICYzLyRIcuZNwF0Zf4,11323
 nedo_vision_worker_core/doctor.py,sha256=K_-hVV2-mdEefZ4Cfu5hMCiOxBiI1aXY8VtkkpK80Lc,10651
@@ -14,8 +14,7 @@ nedo_vision_worker_core/config/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrF
 nedo_vision_worker_core/database/DatabaseManager.py,sha256=o2e9jcqBbtf0Zd9TYuyQhXyKj9rjnNYbSShr7CxuAPk,9341
 nedo_vision_worker_core/database/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 nedo_vision_worker_core/detection/BaseDetector.py,sha256=bReQCTy4tEA1itvw3kkjBMQnx3Jn4MjnDkzdtPwmNPQ,757
-nedo_vision_worker_core/detection/DetectionManager.py,sha256=GSxdXbG6sQXgKEzxsFhi_CQCdy8JK3nytiJPYZUE-xo,3142
-nedo_vision_worker_core/detection/RFDETRDetector.py,sha256=FYZX0wh_y9R3wRYz5vratHcx-pCDnZUkeL2YI__UCPI,2908
+nedo_vision_worker_core/detection/RFDETRDetector.py,sha256=3T3zTFZW0pBv9E-pSpY4JP7wI0LOTM4hxzanvrEXMcE,3093
 nedo_vision_worker_core/detection/YOLODetector.py,sha256=oMCW4KanQCDbtz-ttMiCY5lP1rIgVH_LpfvZvi270j0,2290
 nedo_vision_worker_core/detection/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 nedo_vision_worker_core/detection/detection_processing/DetectionProcessor.py,sha256=wqf4hliR_CPkeoeRbBB3PEpQsmasC4mASJ4WyYjNyPE,948
@@ -58,17 +57,18 @@ nedo_vision_worker_core/models/worker_source_pipeline.py,sha256=xCD4i9pHr8Qy5B_h
 nedo_vision_worker_core/models/worker_source_pipeline_config.py,sha256=dGYTpcTFFu6pmGBufuWBHjv3Xs4RGAQwZn6jp6Ondvs,876
 nedo_vision_worker_core/models/worker_source_pipeline_debug.py,sha256=6S7TkN37FrAT4VwsEB38DWSad7QfvNhsOGtSEK8D1Qs,594
 nedo_vision_worker_core/models/worker_source_pipeline_detection.py,sha256=p6CJsiVCKprTYrNxJsiTB8njXdHkjZKVEyBceRVE6fY,560
+nedo_vision_worker_core/pipeline/ModelManager.py,sha256=K7lmVOo-KL7bnWtyafilZs23bzd6loCgfUz7xuAmlVw,6195
 nedo_vision_worker_core/pipeline/PipelineConfigManager.py,sha256=X55i9GyXcW9ylO6cj2UMAZFSxxPViacL4H4DZl60CAY,1157
-nedo_vision_worker_core/pipeline/PipelineManager.py,sha256=kJslTS1QD7sgmvZ4ZShxW2HI2u_xzmNM5yD0KlpPx_4,5485
+nedo_vision_worker_core/pipeline/PipelineManager.py,sha256=dAYLK5AXXUkfO7cyxLEGGa9SCQTP-9iAO_otffRgVsI,5482
 nedo_vision_worker_core/pipeline/PipelinePrepocessor.py,sha256=cCiVSHHqsKCtKYURdYoEjHJX2GnT6zd8kQ6ZukjQ3V0,1271
-nedo_vision_worker_core/pipeline/PipelineProcessor.py,sha256=Qw8gRk7cOuVBAvS2h12CF0VLkqLrCrIOIGqGQHnzrKk,26262
-nedo_vision_worker_core/pipeline/PipelineSyncThread.py,sha256=9eVGsm4rUfr5tpDHRYOshpiaZ5I484AjJdilhffHzls,11692
+nedo_vision_worker_core/pipeline/PipelineProcessor.py,sha256=92ef6fRpibY3VXNeydpD1kprRCRH3fye_cpztAEVrLQ,26350
+nedo_vision_worker_core/pipeline/PipelineSyncThread.py,sha256=LX_LXU9MTdD5rP-ElHcYR431_qTXNaKOWcCsE4E9Gd0,8613
 nedo_vision_worker_core/pipeline/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 nedo_vision_worker_core/preprocessing/ImageResizer.py,sha256=RvOazxe6dJQuiy0ZH4lIGbdFfiu0FLUVCHoMvxkDNT4,1324
 nedo_vision_worker_core/preprocessing/ImageRoi.py,sha256=iO7oQ-SdUSA_kTIVBuq_mdycXsiJNfiFD3J7-VTxiQ4,2141
 nedo_vision_worker_core/preprocessing/Preprocessor.py,sha256=uYIh0Ld4T1zEEHtKVLbUVBcF0kUwj5zCfPXn__bKwwU,477
 nedo_vision_worker_core/preprocessing/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
-nedo_vision_worker_core/repositories/AIModelRepository.py,sha256=8DaTcKQK89Nf_xTbAUtJUFuMJV9qiQ0VcInHdVFEjts,952
+nedo_vision_worker_core/repositories/AIModelRepository.py,sha256=SzBLwUrjDnc4QayZOA2zL0Jl8YQ-x6S1rde_zCBu6LQ,1591
 nedo_vision_worker_core/repositories/PPEDetectionRepository.py,sha256=C_0QL2sHiSlM9rPmhLmfs6hdZk9FDazy-aVLcznN5w0,6623
 nedo_vision_worker_core/repositories/RestrictedAreaRepository.py,sha256=umZ7IrgoEFqAa9ZZlH7KPUIbmp5yhBc0FSDUOY6UFag,4283
 nedo_vision_worker_core/repositories/WorkerSourcePipelineDebugRepository.py,sha256=_z-UyaYtg1Q5nqSbs_16ngYesyM3aji4VrP1ZBHm6Jk,2987
@@ -79,11 +79,11 @@ nedo_vision_worker_core/repositories/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8
 nedo_vision_worker_core/services/SharedVideoStreamServer.py,sha256=rhCineMKPG3GQbrMHlSHP4xhXaGZ6Rn1oqIajW5xpaY,9827
 nedo_vision_worker_core/services/VideoSharingDaemon.py,sha256=iY6afEKTOsphfHvmZTL0grezka2DS9DDq-1EIpVMy0Y,28524
 nedo_vision_worker_core/services/VideoSharingDaemonManager.py,sha256=sc8VZo5iwoOdR8uTiel5BKz6-eZ7wwLy3IwV_3tsAu0,10340
-nedo_vision_worker_core/streams/RTMPStreamer.py,sha256=GNooWE4V--GiqetqpWSj0TTmOns12DaIpLGy1qHFulc,10167
-nedo_vision_worker_core/streams/SharedVideoDeviceManager.py,sha256=N2cvlKfemD3OG4zEW5mnyup4JgqbzS_JvP2rbbHEBE0,16183
+nedo_vision_worker_core/streams/RTMPStreamer.py,sha256=Dblfutc1UVHj159KUHFYZ8xFEVhHVknZn_nAqKR6uCs,8695
+nedo_vision_worker_core/streams/SharedVideoDeviceManager.py,sha256=vSslwxbhKH6FPndR1HcSFIVWtF-iiOQMlSa4VvFa6M4,16265
 nedo_vision_worker_core/streams/StreamSyncThread.py,sha256=WmYAY9wFiFhLlxGdnvKGIjAqLwCBayNKdmAWzkbU0jM,3763
-nedo_vision_worker_core/streams/VideoStream.py,sha256=Mtj5FI4vEy-dhJN2hQZaD9D_OWBbsntVqWMSDfKn4wk,16024
-nedo_vision_worker_core/streams/VideoStreamManager.py,sha256=FpTNRS8F1LvLLYBMwwTeaiOPTz5yyX_HDYtcYJ97utU,12269
+nedo_vision_worker_core/streams/VideoStream.py,sha256=nGtJ4FAZ1Ek-8hVRopEt0bLWLpa10OtyUwdDEuXLObQ,13343
+nedo_vision_worker_core/streams/VideoStreamManager.py,sha256=i7uemC3tUdsrAhL8lvoxCYa0fwUh5oAVD6DDaxflXGE,12057
 nedo_vision_worker_core/streams/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 nedo_vision_worker_core/tracker/SFSORT.py,sha256=0kggw0l4yPZ55AKHdqVX6mu9ehHmJed7jcJ3JQoC4sk,14061
 nedo_vision_worker_core/tracker/TrackerManager.py,sha256=xtDMI657W2s7HM2lMGtwU0x5Hq74BZpLHd-5xk-278I,6152
@@ -93,10 +93,11 @@ nedo_vision_worker_core/util/DrawingUtils.py,sha256=sLptmzVaJakP_ZgbZsLL03RMH_9N
 nedo_vision_worker_core/util/ModelReadinessChecker.py,sha256=ywHvt_d7UlY3DyFEJrO4Iyl0zx3SaLKb-Qab5l5Q8n4,6548
 nedo_vision_worker_core/util/PersonAttributeMatcher.py,sha256=PhYTPYSF62Nfuc7dage03K6icw_bBBdpvXvnlzCbS30,2773
 nedo_vision_worker_core/util/PersonRestrictedAreaMatcher.py,sha256=iuzCU32BQKaZ3dIy0QHNg2yoWJA-XhTRwwYqCvFdDgg,1711
+nedo_vision_worker_core/util/PlatformDetector.py,sha256=GGL8UfeMQITR22EMYIRWnuOEnSqo7Dr5mb0PaFrl8AM,3006
 nedo_vision_worker_core/util/TablePrinter.py,sha256=wzLGgb1GFMeIbAP6HmKcZD33j4D-IlyqlyeR7C5yD7w,1137
 nedo_vision_worker_core/util/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
-nedo_vision_worker_core-0.3.2.dist-info/METADATA,sha256=ql_lATGf7e9bMoC80qbVUEqTq6IbqUAt7fk62kfcZEI,14370
-nedo_vision_worker_core-0.3.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nedo_vision_worker_core-0.3.2.dist-info/entry_points.txt,sha256=pIPafsvPnBw-fpBKBmc1NQCQ6PQY3ad8mZ6mn8_p5FI,70
-nedo_vision_worker_core-0.3.2.dist-info/top_level.txt,sha256=y8kusXjVYqtG8MSHYWTrk8bRrvjOrphKXYyzu943TTQ,24
-nedo_vision_worker_core-0.3.2.dist-info/RECORD,,
+nedo_vision_worker_core-0.3.4.dist-info/METADATA,sha256=3_UyMqyQc4m_h1qyECq1k7w3vWnN2Hr-ixnLK_BPEp0,14370
+nedo_vision_worker_core-0.3.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nedo_vision_worker_core-0.3.4.dist-info/entry_points.txt,sha256=pIPafsvPnBw-fpBKBmc1NQCQ6PQY3ad8mZ6mn8_p5FI,70
+nedo_vision_worker_core-0.3.4.dist-info/top_level.txt,sha256=y8kusXjVYqtG8MSHYWTrk8bRrvjOrphKXYyzu943TTQ,24
+nedo_vision_worker_core-0.3.4.dist-info/RECORD,,

nedo_vision_worker_core/detection/DetectionManager.py

@@ -1,83 +0,0 @@
-import logging
-try:
-    import torch
-    TORCH_AVAILABLE = True
-except ImportError:
-    TORCH_AVAILABLE = False
-from .BaseDetector import BaseDetector
-from .RFDETRDetector import RFDETRDetector
-from .YOLODetector import YOLODetector
-
-
-class DetectionManager:
-    def __init__(self, model=None):
-        self.detector: BaseDetector | None = None
-        self.model_metadata = None
-
-        if model:
-            self.load_model(model)
-
-    def load_model(self, model):
-        """
-        Loads a new model at runtime and replaces current detector if successful.
-        Checks download status before attempting to load the model.
-        """
-        if not model:
-            if self.detector:
-                logging.info("🧹 Model unloaded")
-            self.detector = None
-            self.model_metadata = None
-            return
-
-        # Check download status before loading
-        if not model.is_ready_for_use():
-            if model.is_downloading():
-                logging.warning(f"⏳ Model {model.name} is still downloading. Skipping load.")
-                self.detector = None
-                self.model_metadata = None
-                return
-            elif model.has_download_failed():
-                logging.error(f"❌ Model {model.name} download failed: {model.download_error}")
-                self.detector = None
-                self.model_metadata = None
-                return
-            else:
-                logging.warning(f"⚠️ Model {model.name} is not ready for use (status: {model.download_status})")
-                self.detector = None
-                self.model_metadata = None
-                return
-
-        detector_type = model.type.lower()
-
-        try:
-            if detector_type == "yolo":
-                detector = YOLODetector(model)
-            elif detector_type == "rf_detr":
-                detector = RFDETRDetector(model)
-            else:
-                raise ValueError(f"Unsupported model type: {detector_type}")
-
-            if detector.model is not None:
-                self.detector = detector
-                self.model_metadata = model
-                # Log device info
-                if TORCH_AVAILABLE:
-                    device = "GPU" if torch.cuda.is_available() else "CPU"
-                else:
-                    device = "CPU (torch not installed)"
-                logging.info(f"🚀 Model {model.name} loaded on {device}")
-                logging.info(f"📥 Model {model.name} with {detector_type} detector loaded")
-            else:
-                logging.error(f"❌ Error loading model: {model.name} with {detector_type} detector")
-                self.detector = None
-                self.model_metadata = None
-
-        except Exception as e:
-            logging.error(f"❌ Error loading model: {model.name} with {detector_type} detector: {e}")
-            self.detector = None
-            self.model_metadata = None
-
-    def detect_objects(self, frame, confidence_threshold=0.7, class_thresholds=None):
-        if not self.detector:
-            return []
-        return self.detector.detect_objects(frame, confidence_threshold, class_thresholds)