Coverage for orchestr_ant_ion / pipeline / __init__.py: 100%

1from __future__ import annotations 

2 

3import importlib 

4from typing import TYPE_CHECKING 

5 

6from orchestr_ant_ion.monitoring.gpu import PYNVML_AVAILABLE 

7from orchestr_ant_ion.pipeline.capture import CameraCapture, OpenCVCapture 

8from orchestr_ant_ion.pipeline.capture.gstreamer import ( 

9 GStreamerSubprocessCapture, 

10 find_gstreamer_launch, 

11 get_gstreamer_env, 

12) 

13from orchestr_ant_ion.pipeline.logging import ( 

14 attach_log_buffer, 

15 configure_logging, 

16 create_log_buffer, 

17) 

18from orchestr_ant_ion.pipeline.metrics.performance import PerformanceTracker 

19from orchestr_ant_ion.pipeline.monitoring.power import ( 

20 PowerMonitor, 

21 get_cpu_freq_ratio, 

22) 

23from orchestr_ant_ion.pipeline.monitoring.system import ( 

24 SystemMonitor, 

25) 

26from orchestr_ant_ion.pipeline.tracking.centroid import SimpleCentroidTracker 

27from orchestr_ant_ion.pipeline.types import ( 

28 CameraConfig, 

29 CaptureBackend, 

30 PerformanceMetrics, 

31 SystemStats, 

32 Track, 

33) 

34from orchestr_ant_ion.pipeline.ui.dearpygui import DearPyGuiViewer 

35 

36 

37if TYPE_CHECKING: 

38 from orchestr_ant_ion.pipeline.ui.wx import ( 

39 WxPythonViewer as WxPythonViewerType, 

40 ) 

41 

42WxPythonViewer: type[WxPythonViewerType] | None = None 

43try: 

44 _wx_mod = importlib.import_module("orchestr_ant_ion.pipeline.ui.wx") 

45 WxPythonViewer = getattr(_wx_mod, "WxPythonViewer", None) 

46except Exception: # pragma: no cover - optional dependency 

47 WxPythonViewer = None 

48 

49 

50__all__ = [ 

51 "PYNVML_AVAILABLE", 

52 "CameraCapture", 

53 "CameraConfig", 

54 "CaptureBackend", 

55 "DearPyGuiViewer", 

56 "GStreamerSubprocessCapture", 

57 "OpenCVCapture", 

58 "PerformanceMetrics", 

59 "PerformanceTracker", 

60 "PowerMonitor", 

61 "SimpleCentroidTracker", 

62 "SystemMonitor", 

63 "SystemStats", 

64 "Track", 

65 "WxPythonViewer", 

66 "attach_log_buffer", 

67 "configure_logging", 

68 "create_log_buffer", 

69 "find_gstreamer_launch", 

70 "get_cpu_freq_ratio", 

71 "get_gstreamer_env", 

72]