from PySide6.QtWidgets import ( QWidget, QVBoxLayout, QHBoxLayout, QLabel, QPushButton, QSlider, QCheckBox, QFileDialog, QSizePolicy, QFrame, QTabWidget, QGridLayout, QComboBox, QListWidget, QListWidgetItem, QGroupBox, QScrollArea ) from PySide6.QtCore import Signal, Qt from PySide6.QtGui import QPixmap, QIcon import json import os from pathlib import Path class SmartIntersectionOverlay(QFrame): """Advanced overlay for Smart Intersection analytics.""" def __init__(self, parent=None): super().__init__(parent) self.setStyleSheet(""" background: rgba(0,20,40,0.85); border: 2px solid #03DAC5; border-radius: 12px; color: #fff; font-family: 'Consolas', 'SF Mono', 'monospace'; font-size: 12px; """) self.setFixedHeight(140) self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed) self.setAttribute(Qt.WA_TransparentForMouseEvents) layout = QVBoxLayout(self) layout.setContentsMargins(16, 12, 16, 12) layout.setSpacing(4) # Title title = QLabel("🚦 Smart Intersection Analytics") title.setStyleSheet("color: #03DAC5; font-weight: bold; font-size: 14px;") layout.addWidget(title) # Scene data self.scene_label = QLabel("Scene: Multi-Camera Fusion") self.tracking_label = QLabel("Active Tracks: 0") self.roi_label = QLabel("ROI Events: 0") # Camera data self.camera_label = QLabel("Cameras: North(0) East(0) South(0) West(0)") # Analytics data self.analytics_label = QLabel("Analytics: Crosswalk(0) Lane(0) Safety(0)") for w in [self.scene_label, self.tracking_label, self.roi_label, self.camera_label, self.analytics_label]: w.setStyleSheet("color: #fff;") layout.addWidget(w) def update_smart_intersection(self, scene_data): """Update smart intersection specific data""" if not scene_data: return # Update tracking info active_tracks = scene_data.get('active_tracks', 0) self.tracking_label.setText(f"Active Tracks: {active_tracks}") # Update ROI events roi_events = scene_data.get('roi_events', 0) self.roi_label.setText(f"ROI Events: {roi_events}") # Update camera data cameras = scene_data.get('cameras', {}) north = cameras.get('north', 0) east = cameras.get('east', 0) south = cameras.get('south', 0) west = cameras.get('west', 0) self.camera_label.setText(f"Cameras: North({north}) East({east}) South({south}) West({west})") # Update analytics analytics = scene_data.get('analytics', {}) crosswalk = analytics.get('crosswalk_events', 0) lane = analytics.get('lane_events', 0) safety = analytics.get('safety_events', 0) self.analytics_label.setText(f"Analytics: Crosswalk({crosswalk}) Lane({lane}) Safety({safety})") class IntersectionROIWidget(QFrame): """Widget for defining and managing ROI regions for smart intersection""" roi_updated = Signal(dict) def __init__(self, parent=None): super().__init__(parent) self.setStyleSheet(""" QFrame { background: #1a1a1a; border: 1px solid #424242; border-radius: 8px; } """) self.setFixedWidth(300) layout = QVBoxLayout(self) layout.setContentsMargins(16, 16, 16, 16) # Title title = QLabel("🎯 Region of Interest (ROI)") title.setStyleSheet("color: #03DAC5; font-weight: bold; font-size: 14px;") layout.addWidget(title) # ROI Type selection type_layout = QHBoxLayout() type_layout.addWidget(QLabel("Type:")) self.roi_type = QComboBox() self.roi_type.addItems(["Crosswalk", "Traffic Lane", "Safety Zone", "Intersection Center"]) type_layout.addWidget(self.roi_type) layout.addLayout(type_layout) # ROI List self.roi_list = QListWidget() self.roi_list.setMaximumHeight(120) layout.addWidget(self.roi_list) # ROI Controls roi_controls = QHBoxLayout() self.add_roi_btn = QPushButton("Add ROI") self.delete_roi_btn = QPushButton("Delete") self.add_roi_btn.setStyleSheet("background: #27ae60; color: white; border-radius: 4px; padding: 6px;") self.delete_roi_btn.setStyleSheet("background: #e74c3c; color: white; border-radius: 4px; padding: 6px;") roi_controls.addWidget(self.add_roi_btn) roi_controls.addWidget(self.delete_roi_btn) layout.addLayout(roi_controls) # Analytics settings analytics_group = QGroupBox("Analytics Settings") analytics_layout = QVBoxLayout(analytics_group) self.enable_tracking = QCheckBox("Multi-Object Tracking") self.enable_speed = QCheckBox("Speed Estimation") self.enable_direction = QCheckBox("Direction Analysis") self.enable_safety = QCheckBox("Safety Monitoring") for cb in [self.enable_tracking, self.enable_speed, self.enable_direction, self.enable_safety]: cb.setChecked(True) cb.setStyleSheet("color: white;") analytics_layout.addWidget(cb) layout.addWidget(analytics_group) # Connect signals self.add_roi_btn.clicked.connect(self._add_roi) self.delete_roi_btn.clicked.connect(self._delete_roi) # Initialize with default ROIs self._init_default_rois() def _init_default_rois(self): """Initialize with default intersection ROIs""" default_rois = [ "North Crosswalk", "South Crosswalk", "East Crosswalk", "West Crosswalk", "Center Intersection", "North Lane", "South Lane", "East Lane", "West Lane" ] for roi in default_rois: item = QListWidgetItem(roi) item.setFlags(item.flags() | Qt.ItemIsUserCheckable) item.setCheckState(Qt.Checked) self.roi_list.addItem(item) def _add_roi(self): """Add new ROI""" roi_type = self.roi_type.currentText() roi_name = f"{roi_type}_{self.roi_list.count() + 1}" item = QListWidgetItem(roi_name) item.setFlags(item.flags() | Qt.ItemIsUserCheckable) item.setCheckState(Qt.Checked) self.roi_list.addItem(item) self._emit_roi_update() def _delete_roi(self): """Delete selected ROI""" current_row = self.roi_list.currentRow() if current_row >= 0: self.roi_list.takeItem(current_row) self._emit_roi_update() def _emit_roi_update(self): """Emit ROI configuration update""" roi_config = { 'rois': [], 'analytics': { 'tracking': self.enable_tracking.isChecked(), 'speed': self.enable_speed.isChecked(), 'direction': self.enable_direction.isChecked(), 'safety': self.enable_safety.isChecked() } } for i in range(self.roi_list.count()): item = self.roi_list.item(i) roi_config['rois'].append({ 'name': item.text(), 'enabled': item.checkState() == Qt.Checked }) self.roi_updated.emit(roi_config) class MultiCameraView(QFrame): """Multi-camera view for smart intersection""" def __init__(self, parent=None): super().__init__(parent) self.setStyleSheet(""" QFrame { background: #0a0a0a; border: 2px solid #424242; border-radius: 8px; } """) layout = QGridLayout(self) layout.setContentsMargins(8, 8, 8, 8) layout.setSpacing(4) # Create camera views self.camera_views = {} positions = [('North', 0, 1), ('West', 1, 0), ('East', 1, 2), ('South', 2, 1)] for pos_name, row, col in positions: view = self._create_camera_view(pos_name) self.camera_views[pos_name.lower()] = view layout.addWidget(view, row, col) # Center intersection view center_view = self._create_intersection_center() layout.addWidget(center_view, 1, 1) def _create_camera_view(self, position): """Create individual camera view""" view = QFrame() view.setStyleSheet(""" background: #1a1a1a; border: 1px solid #555; border-radius: 4px; """) view.setMinimumSize(160, 120) view.setMaximumSize(200, 150) layout = QVBoxLayout(view) layout.setContentsMargins(4, 4, 4, 4) # Title title = QLabel(f"📹 {position}") title.setStyleSheet("color: #03DAC5; font-weight: bold; font-size: 10px;") title.setAlignment(Qt.AlignCenter) layout.addWidget(title) # Video area video_area = QLabel("No feed") video_area.setStyleSheet("background: #000; color: #666; border: 1px dashed #333;") video_area.setAlignment(Qt.AlignCenter) video_area.setMinimumHeight(80) layout.addWidget(video_area) # Stats stats = QLabel("Objects: 0") stats.setStyleSheet("color: #aaa; font-size: 9px;") stats.setAlignment(Qt.AlignCenter) layout.addWidget(stats) return view def _create_intersection_center(self): """Create center intersection overview""" view = QFrame() view.setStyleSheet(""" background: #2a1a1a; border: 2px solid #03DAC5; border-radius: 8px; """) view.setMinimumSize(160, 120) view.setMaximumSize(200, 150) layout = QVBoxLayout(view) layout.setContentsMargins(8, 8, 8, 8) title = QLabel("🚦 Intersection") title.setStyleSheet("color: #03DAC5; font-weight: bold; font-size: 12px;") title.setAlignment(Qt.AlignCenter) layout.addWidget(title) # Intersection map map_area = QLabel("Scene Map") map_area.setStyleSheet("background: #000; color: #03DAC5; border: 1px solid #03DAC5;") map_area.setAlignment(Qt.AlignCenter) map_area.setMinimumHeight(80) layout.addWidget(map_area) # Total stats total_stats = QLabel("Total Objects: 0") total_stats.setStyleSheet("color: #03DAC5; font-size: 10px; font-weight: bold;") total_stats.setAlignment(Qt.AlignCenter) layout.addWidget(total_stats) return view def update_camera_feed(self, camera_position, pixmap, object_count=0): """Update specific camera feed""" if camera_position.lower() in self.camera_views: view = self.camera_views[camera_position.lower()] video_label = view.findChild(QLabel) if video_label and pixmap: scaled = pixmap.scaled(video_label.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation) video_label.setPixmap(scaled) # Update stats stats_labels = view.findChildren(QLabel) if len(stats_labels) >= 3: # title, video, stats stats_labels[2].setText(f"Objects: {object_count}") class EnhancedPerformanceOverlay(QFrame): """Enhanced performance metrics overlay with traffic light status.""" def __init__(self, parent=None): super().__init__(parent) self.setStyleSheet(""" QFrame { background: rgba(20, 30, 40, 0.95); border: 2px solid #03DAC5; border-radius: 12px; color: #fff; font-family: 'Segoe UI', 'Arial', sans-serif; } """) self.setFixedHeight(140) self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed) self.setAttribute(Qt.WA_TransparentForMouseEvents) layout = QVBoxLayout(self) layout.setContentsMargins(16, 12, 16, 12) layout.setSpacing(8) # Title row title_layout = QHBoxLayout() title = QLabel("📊 Real-time Performance Metrics") title.setStyleSheet(""" color: #03DAC5; font-weight: bold; font-size: 14px; margin-bottom: 4px; """) title_layout.addWidget(title) title_layout.addStretch() # Traffic light status self.traffic_light_status = QLabel("🚦 Traffic: Unknown") self.traffic_light_status.setStyleSheet(""" color: #FFD700; font-weight: bold; font-size: 13px; background: rgba(0,0,0,0.3); padding: 4px 8px; border-radius: 6px; """) title_layout.addWidget(self.traffic_light_status) layout.addLayout(title_layout) # Performance metrics row perf_layout = QHBoxLayout() perf_layout.setSpacing(16) # FPS and Inference in badges self.fps_label = QLabel("FPS: --") self.fps_label.setStyleSheet(""" background: #27AE60; color: white; font-weight: bold; font-size: 13px; padding: 6px 12px; border-radius: 8px; min-width: 70px; """) self.fps_label.setAlignment(Qt.AlignCenter) self.inference_label = QLabel("Inference: -- ms") self.inference_label.setStyleSheet(""" background: #3498DB; color: white; font-weight: bold; font-size: 13px; padding: 6px 12px; border-radius: 8px; min-width: 110px; """) self.inference_label.setAlignment(Qt.AlignCenter) perf_layout.addWidget(self.fps_label) perf_layout.addWidget(self.inference_label) perf_layout.addStretch() layout.addLayout(perf_layout) # System info row system_layout = QHBoxLayout() self.model_label = QLabel("Model: -") self.model_label.setStyleSheet(""" color: #E74C3C; font-weight: bold; font-size: 12px; background: rgba(231, 76, 60, 0.1); padding: 4px 8px; border-radius: 6px; """) self.device_label = QLabel("Device: -") self.device_label.setStyleSheet(""" color: #9B59B6; font-weight: bold; font-size: 12px; background: rgba(155, 89, 182, 0.1); padding: 4px 8px; border-radius: 6px; """) system_layout.addWidget(self.model_label) system_layout.addWidget(self.device_label) system_layout.addStretch() layout.addLayout(system_layout) # Vehicle counts row self.vehicle_stats_label = QLabel("🚗 Vehicles: 0 | 🚛 Trucks: 0 | 🚶 Pedestrians: 0 | 🏍️ Motorcycles: 0") self.vehicle_stats_label.setStyleSheet(""" color: #F39C12; font-weight: bold; font-size: 12px; background: rgba(243, 156, 18, 0.1); padding: 6px 10px; border-radius: 6px; """) layout.addWidget(self.vehicle_stats_label) def update_overlay(self, model, device, cars, trucks, peds, tlights, motorcycles): """Update performance metrics""" self.model_label.setText(f"Model: {model}") self.device_label.setText(f"Device: {device}") self.vehicle_stats_label.setText(f"🚗 Vehicles: {cars} | 🚛 Trucks: {trucks} | 🚶 Pedestrians: {peds} | 🏍️ Motorcycles: {motorcycles}") def update_performance_metrics(self, fps, inference_time): """Update FPS and inference time""" if fps is not None: self.fps_label.setText(f"FPS: {fps:.1f}") else: self.fps_label.setText("FPS: --") if inference_time is not None: self.inference_label.setText(f"Inference: {inference_time:.1f} ms") else: self.inference_label.setText("Inference: -- ms") def update_traffic_light_status(self, traffic_light_data): """Update traffic light status""" if traffic_light_data and isinstance(traffic_light_data, dict): color = traffic_light_data.get('color', 'unknown') confidence = traffic_light_data.get('confidence', 0) if color.lower() == 'red': icon = "🔴" text_color = "#E74C3C" elif color.lower() == 'yellow': icon = "🟡" text_color = "#F39C12" elif color.lower() == 'green': icon = "🟢" text_color = "#27AE60" else: icon = "⚫" text_color = "#95A5A6" self.traffic_light_status.setText(f"{icon} Traffic: {color.title()} ({confidence:.2f})") self.traffic_light_status.setStyleSheet(f""" color: {text_color}; font-weight: bold; font-size: 13px; background: rgba(0,0,0,0.3); padding: 4px 8px; border-radius: 6px; """) else: self.traffic_light_status.setText("🚦 Traffic: Unknown") self.traffic_light_status.setStyleSheet(""" color: #95A5A6; font-weight: bold; font-size: 13px; background: rgba(0,0,0,0.3); padding: 4px 8px; border-radius: 6px; """) class VideoDetectionTab(QWidget): file_selected = Signal(str) play_clicked = Signal() pause_clicked = Signal() stop_clicked = Signal() detection_toggled = Signal(bool) screenshot_clicked = Signal() seek_changed = Signal(int) auto_select_model_device = Signal() # Smart Intersection signals smart_intersection_enabled = Signal(bool) multi_camera_mode_enabled = Signal(bool) roi_configuration_changed = Signal(dict) scene_analytics_toggled = Signal(bool) def __init__(self): super().__init__() self.video_loaded = False self.smart_intersection_mode = False self.multi_camera_mode = False # Load smart intersection config self.load_smart_intersection_config() # Main layout main_layout = QHBoxLayout(self) main_layout.setContentsMargins(16, 16, 16, 16) main_layout.setSpacing(16) # Left panel - video and controls left_panel = self._create_left_panel() main_layout.addWidget(left_panel, 3) # 3/4 of the space # Right panel - smart intersection controls right_panel = self._create_right_panel() main_layout.addWidget(right_panel, 1) # 1/4 of the space def load_smart_intersection_config(self): """Load smart intersection configuration""" config_path = Path(__file__).parent.parent / "config" / "smart-intersection" / "desktop-config.json" try: if config_path.exists(): with open(config_path, 'r') as f: self.smart_config = json.load(f) else: self.smart_config = self._get_default_config() except Exception as e: print(f"Error loading smart intersection config: {e}") self.smart_config = self._get_default_config() def _get_default_config(self): """Get default smart intersection configuration""" return { "desktop_app_config": { "scene_analytics": { "enable_multi_camera": True, "enable_roi_analytics": True, "enable_vlm_integration": True }, "camera_settings": { "max_cameras": 4, "default_fps": 30 }, "analytics_settings": { "object_tracking": True, "speed_estimation": True, "direction_analysis": True, "safety_monitoring": True } } } def _create_left_panel(self): """Create main video panel""" panel = QWidget() layout = QVBoxLayout(panel) layout.setContentsMargins(0, 0, 0, 0) layout.setSpacing(16) # Smart Intersection Mode Toggle mode_bar = self._create_mode_bar() layout.addWidget(mode_bar) # File select bar file_bar = self._create_file_bar() layout.addWidget(file_bar) # Video display area (with tabs for different modes) self.video_tabs = QTabWidget() self.video_tabs.setStyleSheet(""" QTabWidget::pane { border: 1px solid #424242; background: #121212; border-radius: 8px; } QTabBar::tab { background: #232323; color: #fff; padding: 8px 16px; margin-right: 2px; border-top-left-radius: 8px; border-top-right-radius: 8px; } QTabBar::tab:selected { background: #03DAC5; color: #000; } """) # Single camera tab self.single_cam_widget = self._create_single_camera_view() self.video_tabs.addTab(self.single_cam_widget, "📹 Single Camera") # Multi-camera tab self.multi_cam_widget = MultiCameraView() self.video_tabs.addTab(self.multi_cam_widget, "🚦 Multi-Camera Intersection") layout.addWidget(self.video_tabs) # Analytics overlay self.analytics_overlay = self._create_analytics_overlay() layout.addWidget(self.analytics_overlay) # Control bar control_bar = self._create_control_bar() layout.addWidget(control_bar) return panel def _create_mode_bar(self): """Create smart intersection mode toggle bar""" bar = QFrame() bar.setStyleSheet(""" QFrame { background: #1a2332; border: 2px solid #03DAC5; border-radius: 12px; padding: 8px; } """) bar.setFixedHeight(60) layout = QHBoxLayout(bar) layout.setContentsMargins(16, 8, 16, 8) # Smart Intersection Toggle self.smart_intersection_toggle = QCheckBox("🚦 Smart Intersection Mode") self.smart_intersection_toggle.setStyleSheet(""" QCheckBox { color: #03DAC5; font-weight: bold; font-size: 14px; } QCheckBox::indicator { width: 20px; height: 20px; } QCheckBox::indicator:checked { background: #03DAC5; border: 2px solid #03DAC5; border-radius: 4px; } """) self.smart_intersection_toggle.toggled.connect(self._toggle_smart_intersection) layout.addWidget(self.smart_intersection_toggle) layout.addSpacing(32) # Multi-camera Toggle self.multi_camera_toggle = QCheckBox("📹 Multi-Camera Fusion") self.multi_camera_toggle.setStyleSheet(""" QCheckBox { color: #e67e22; font-weight: bold; font-size: 14px; } QCheckBox::indicator { width: 20px; height: 20px; } QCheckBox::indicator:checked { background: #e67e22; border: 2px solid #e67e22; border-radius: 4px; } """) self.multi_camera_toggle.toggled.connect(self._toggle_multi_camera) layout.addWidget(self.multi_camera_toggle) layout.addStretch() # Status indicator self.mode_status = QLabel("Standard Detection Mode") self.mode_status.setStyleSheet("color: #bbb; font-size: 12px;") layout.addWidget(self.mode_status) return bar def _create_file_bar(self): """Create file selection bar""" widget = QWidget() bar = QHBoxLayout(widget) self.file_btn = QPushButton() self.file_btn.setIcon(QIcon.fromTheme("folder-video")) self.file_btn.setText("Select Video") self.file_btn.setStyleSheet("padding: 8px 18px; border-radius: 8px; background: #232323; color: #fff;") self.file_label = QLabel("No file selected") self.file_label.setStyleSheet("color: #bbb; font-size: 13px;") self.file_btn.clicked.connect(self._select_file) bar.addWidget(self.file_btn) bar.addWidget(self.file_label) bar.addStretch() return widget def _create_single_camera_view(self): """Create single camera view widget""" widget = QWidget() layout = QVBoxLayout(widget) layout.setContentsMargins(0, 0, 0, 0) # Video frame video_frame = QFrame() video_frame.setStyleSheet(""" background: #121212; border: 1px solid #424242; border-radius: 8px; """) video_frame.setMinimumSize(640, 360) video_frame.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding) video_layout = QVBoxLayout(video_frame) video_layout.setContentsMargins(0, 0, 0, 0) video_layout.setAlignment(Qt.AlignCenter) self.video_label = QLabel() self.video_label.setAlignment(Qt.AlignCenter) self.video_label.setStyleSheet("background: transparent; color: #888; font-size: 18px;") self.video_label.setText("No video loaded. Please select a file.") self.video_label.setMinimumSize(640, 360) self.video_label.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding) video_layout.addWidget(self.video_label) layout.addWidget(video_frame) return widget def _create_analytics_overlay(self): """Create analytics overlay that switches based on mode""" container = QWidget() self.overlay_layout = QVBoxLayout(container) self.overlay_layout.setContentsMargins(0, 0, 0, 0) # Standard overlay self.standard_overlay = EnhancedPerformanceOverlay() self.standard_overlay.setStyleSheet(self.standard_overlay.styleSheet() + "border: 1px solid #03DAC5;") # Smart intersection overlay self.smart_overlay = SmartIntersectionOverlay() # Badge bar self.badge_bar = QHBoxLayout() self.badge_bar.setContentsMargins(0, 8, 0, 8) self.fps_badge = QLabel("FPS: --") self.fps_badge.setStyleSheet("background: #27ae60; color: #fff; border-radius: 12px; padding: 4px 24px; font-weight: bold; font-size: 15px;") self.fps_badge.setAlignment(Qt.AlignCenter) self.inference_badge = QLabel("Inference: -- ms") self.inference_badge.setStyleSheet("background: #2980b9; color: #fff; border-radius: 12px; padding: 4px 24px; font-weight: bold; font-size: 15px;") self.inference_badge.setAlignment(Qt.AlignCenter) self.badge_bar.addWidget(self.fps_badge) self.badge_bar.addSpacing(12) self.badge_bar.addWidget(self.inference_badge) self.badge_bar.addSpacing(18) # Add current overlay (start with standard) self.current_overlay = self.standard_overlay self.badge_bar.addWidget(self.current_overlay) self.badge_bar.addStretch() self.overlay_layout.addLayout(self.badge_bar) return container def _create_control_bar(self): """Create control bar""" widget = QWidget() control_bar = QHBoxLayout(widget) control_bar.setContentsMargins(0, 16, 0, 0) # Playback controls self.play_btn = QPushButton() self.play_btn.setIcon(QIcon.fromTheme("media-playback-start")) self.play_btn.setToolTip("Play") self.play_btn.setFixedSize(48, 48) self.play_btn.setEnabled(False) self.play_btn.setStyleSheet(self._button_style()) self.pause_btn = QPushButton() self.pause_btn.setIcon(QIcon.fromTheme("media-playback-pause")) self.pause_btn.setToolTip("Pause") self.pause_btn.setFixedSize(48, 48) self.pause_btn.setEnabled(False) self.pause_btn.setStyleSheet(self._button_style()) self.stop_btn = QPushButton() self.stop_btn.setIcon(QIcon.fromTheme("media-playback-stop")) self.stop_btn.setToolTip("Stop") self.stop_btn.setFixedSize(48, 48) self.stop_btn.setEnabled(False) self.stop_btn.setStyleSheet(self._button_style()) for btn, sig in zip([self.play_btn, self.pause_btn, self.stop_btn], [self.play_clicked.emit, self.pause_clicked.emit, self.stop_clicked.emit]): btn.clicked.connect(sig) control_bar.addWidget(self.play_btn) control_bar.addWidget(self.pause_btn) control_bar.addWidget(self.stop_btn) control_bar.addSpacing(16) # Progress bar self.progress = QSlider(Qt.Horizontal) self.progress.setStyleSheet("QSlider::groove:horizontal { height: 6px; background: #232323; border-radius: 3px; } QSlider::handle:horizontal { background: #03DAC5; border-radius: 8px; width: 18px; }") self.progress.setMinimumWidth(240) self.progress.setEnabled(False) self.progress.valueChanged.connect(self.seek_changed.emit) control_bar.addWidget(self.progress, 2) self.timestamp = QLabel("00:00 / 00:00") self.timestamp.setStyleSheet("color: #bbb; font-size: 13px;") control_bar.addWidget(self.timestamp) control_bar.addSpacing(16) # Detection toggle & screenshot self.detection_toggle = QCheckBox("Enable Detection") self.detection_toggle.setChecked(True) self.detection_toggle.setStyleSheet("color: #fff; font-size: 14px;") self.detection_toggle.setEnabled(False) self.detection_toggle.toggled.connect(self.detection_toggled.emit) control_bar.addWidget(self.detection_toggle) self.screenshot_btn = QPushButton() self.screenshot_btn.setIcon(QIcon.fromTheme("camera-photo")) self.screenshot_btn.setText("Screenshot") self.screenshot_btn.setToolTip("Save current frame as image") self.screenshot_btn.setEnabled(False) self.screenshot_btn.setStyleSheet(self._button_style()) self.screenshot_btn.clicked.connect(self.screenshot_clicked.emit) control_bar.addWidget(self.screenshot_btn) control_bar.addStretch() return widget def _create_right_panel(self): """Create right panel for smart intersection controls""" panel = QScrollArea() panel.setWidgetResizable(True) panel.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff) panel.setStyleSheet(""" QScrollArea { background: #1a1a1a; border: 1px solid #424242; border-radius: 8px; } """) content = QWidget() layout = QVBoxLayout(content) layout.setContentsMargins(16, 16, 16, 16) layout.setSpacing(16) # Smart Intersection Controls intersection_group = QGroupBox("🚦 Smart Intersection") intersection_group.setStyleSheet(""" QGroupBox { color: #03DAC5; font-weight: bold; font-size: 14px; border: 2px solid #03DAC5; border-radius: 8px; margin-top: 12px; padding-top: 8px; } QGroupBox::title { subcontrol-origin: margin; left: 16px; padding: 0 8px 0 8px; } """) intersection_layout = QVBoxLayout(intersection_group) # Scene Analytics Toggle self.scene_analytics_toggle = QCheckBox("Scene Analytics") self.scene_analytics_toggle.setChecked(True) self.scene_analytics_toggle.setStyleSheet("color: white; font-size: 12px;") self.scene_analytics_toggle.toggled.connect(self.scene_analytics_toggled.emit) intersection_layout.addWidget(self.scene_analytics_toggle) # Multi-object tracking self.multi_tracking_toggle = QCheckBox("Multi-Object Tracking") self.multi_tracking_toggle.setChecked(True) self.multi_tracking_toggle.setStyleSheet("color: white; font-size: 12px;") intersection_layout.addWidget(self.multi_tracking_toggle) # Speed estimation self.speed_estimation_toggle = QCheckBox("Speed Estimation") self.speed_estimation_toggle.setChecked(True) self.speed_estimation_toggle.setStyleSheet("color: white; font-size: 12px;") intersection_layout.addWidget(self.speed_estimation_toggle) layout.addWidget(intersection_group) # ROI Management self.roi_widget = IntersectionROIWidget() self.roi_widget.roi_updated.connect(self.roi_configuration_changed.emit) layout.addWidget(self.roi_widget) # Analytics Summary - Simplified analytics_group = QGroupBox("📊 Quick Stats") analytics_group.setStyleSheet(intersection_group.styleSheet().replace("#03DAC5", "#e67e22")) analytics_layout = QVBoxLayout(analytics_group) self.total_objects_label = QLabel("Total Objects: 0") self.active_vehicles_label = QLabel("Active Vehicles: 0") self.traffic_status_label = QLabel("Traffic Light: Unknown") for label in [self.total_objects_label, self.active_vehicles_label, self.traffic_status_label]: label.setStyleSheet("color: white; font-size: 12px; padding: 4px;") analytics_layout.addWidget(label) layout.addWidget(analytics_group) # Performance Monitoring perf_group = QGroupBox("⚡ Performance") perf_group.setStyleSheet(intersection_group.styleSheet().replace("#03DAC5", "#9b59b6")) perf_layout = QVBoxLayout(perf_group) self.gpu_usage_label = QLabel("GPU Usage: -%") self.memory_usage_label = QLabel("Memory: - MB") self.processing_time_label = QLabel("Processing: - ms") for label in [self.gpu_usage_label, self.memory_usage_label, self.processing_time_label]: label.setStyleSheet("color: white; font-size: 12px;") perf_layout.addWidget(label) layout.addWidget(perf_group) layout.addStretch() panel.setWidget(content) return panel def _toggle_smart_intersection(self, enabled): """Toggle smart intersection mode""" self.smart_intersection_mode = enabled self.smart_intersection_enabled.emit(enabled) # Switch overlay if enabled: self._switch_to_smart_overlay() self.mode_status.setText("🚦 Smart Intersection Active") self.mode_status.setStyleSheet("color: #03DAC5; font-weight: bold; font-size: 12px;") else: self._switch_to_standard_overlay() self.mode_status.setText("Standard Detection Mode") self.mode_status.setStyleSheet("color: #bbb; font-size: 12px;") # Enable/disable multi-camera toggle self.multi_camera_toggle.setEnabled(enabled) if not enabled: self.multi_camera_toggle.setChecked(False) def _toggle_multi_camera(self, enabled): """Toggle multi-camera mode""" self.multi_camera_mode = enabled self.multi_camera_mode_enabled.emit(enabled) if enabled: self.video_tabs.setCurrentIndex(1) # Switch to multi-camera tab self.mode_status.setText("🚦 Multi-Camera Intersection Active") else: self.video_tabs.setCurrentIndex(0) # Switch to single camera tab if self.smart_intersection_mode: self.mode_status.setText("🚦 Smart Intersection Active") def _switch_to_smart_overlay(self): """Switch to smart intersection overlay""" self.badge_bar.removeWidget(self.current_overlay) self.current_overlay.setParent(None) self.current_overlay = self.smart_overlay self.badge_bar.addWidget(self.current_overlay) def _switch_to_standard_overlay(self): """Switch to standard overlay""" self.badge_bar.removeWidget(self.current_overlay) self.current_overlay.setParent(None) self.current_overlay = self.standard_overlay self.badge_bar.addWidget(self.current_overlay) def _button_style(self): return """ QPushButton { background: #232323; border-radius: 24px; color: #fff; font-size: 15px; border: none; } QPushButton:hover { background: #03DAC5; color: #222; } QPushButton:pressed { background: #018786; } """ def _select_file(self): file_path, _ = QFileDialog.getOpenFileName(self, "Select Video File", "", "Video Files (*.mp4 *.avi *.mov *.mkv *.webm);;All Files (*)") if file_path: self.file_label.setText(file_path) self.file_selected.emit(file_path) self.video_loaded = True self._enable_controls(True) self.video_label.setText("") self.auto_select_model_device.emit() def _enable_controls(self, enabled): self.play_btn.setEnabled(enabled) self.pause_btn.setEnabled(enabled) self.stop_btn.setEnabled(enabled) self.progress.setEnabled(enabled) self.detection_toggle.setEnabled(enabled) self.screenshot_btn.setEnabled(enabled) if enabled: self.auto_select_model_device.emit() def update_display(self, pixmap): """Update display with new frame""" if pixmap: if self.multi_camera_mode: # In multi-camera mode, distribute to different camera views # For now, just update the single view scaled = pixmap.scaled(self.video_label.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation) self.video_label.setPixmap(scaled) else: scaled = pixmap.scaled(self.video_label.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation) self.video_label.setPixmap(scaled) self._set_controls_enabled(True) self.video_label.setStyleSheet("background: transparent; color: #888; font-size: 18px;") else: self.video_label.clear() self.video_label.setText("No video loaded. Please select a video file.") self._set_controls_enabled(False) self.video_label.setStyleSheet("background: transparent; color: #F44336; font-size: 18px;") def _set_controls_enabled(self, enabled): for btn in [self.play_btn, self.pause_btn, self.stop_btn, self.progress, self.detection_toggle, self.screenshot_btn]: btn.setEnabled(enabled) def update_stats(self, stats): """Update statistics display""" if self.smart_intersection_mode: # Update smart intersection overlay scene_data = { 'active_tracks': stats.get('total_objects', 0), 'roi_events': stats.get('roi_events', 0), 'cameras': { 'north': stats.get('north_objects', 0), 'east': stats.get('east_objects', 0), 'south': stats.get('south_objects', 0), 'west': stats.get('west_objects', 0) }, 'analytics': { 'crosswalk_events': stats.get('crosswalk_events', 0), 'lane_events': stats.get('lane_events', 0), 'safety_events': stats.get('safety_events', 0) } } self.smart_overlay.update_smart_intersection(scene_data) # Update right panel quick stats self.total_objects_label.setText(f"Total Objects: {stats.get('total_objects', 0)}") active_vehicles = stats.get('cars', 0) + stats.get('trucks', 0) + stats.get('motorcycles', 0) self.active_vehicles_label.setText(f"Active Vehicles: {active_vehicles}") traffic_light = stats.get('traffic_light', {}) if traffic_light and isinstance(traffic_light, dict): color = traffic_light.get('color', 'Unknown') self.traffic_status_label.setText(f"Traffic Light: {color.title()}") else: self.traffic_status_label.setText("Traffic Light: Unknown") else: # Update enhanced performance overlay cars = stats.get('cars', 0) trucks = stats.get('trucks', 0) peds = stats.get('peds', 0) tlights = stats.get('tlights', 0) motorcycles = stats.get('motorcycles', 0) model = stats.get('model', stats.get('model_name', '-')) device = stats.get('device', stats.get('device_name', '-')) # Update vehicle counts and system info self.standard_overlay.update_overlay(model, device, cars, trucks, peds, tlights, motorcycles) # Update performance metrics (FPS and inference time) fps = stats.get('fps', None) inference = stats.get('inference', stats.get('detection_time', stats.get('detection_time_ms', None))) self.standard_overlay.update_performance_metrics(fps, inference) # Update traffic light status traffic_light = stats.get('traffic_light', None) self.standard_overlay.update_traffic_light_status(traffic_light) # Update right panel quick stats for standard mode too total_objects = cars + trucks + peds + motorcycles + tlights self.total_objects_label.setText(f"Total Objects: {total_objects}") active_vehicles = cars + trucks + motorcycles self.active_vehicles_label.setText(f"Active Vehicles: {active_vehicles}") if traffic_light and isinstance(traffic_light, dict): color = traffic_light.get('color', 'Unknown') self.traffic_status_label.setText(f"Traffic Light: {color.title()}") else: self.traffic_status_label.setText("Traffic Light: Unknown") # Update performance badges (keeping the existing FPS and inference badges) fps = stats.get('fps', None) inference = stats.get('inference', stats.get('detection_time', stats.get('detection_time_ms', None))) if fps is not None: self.fps_badge.setText(f"FPS: {fps:.2f}") else: self.fps_badge.setText("FPS: --") if inference is not None: self.inference_badge.setText(f"Inference: {inference:.1f} ms") else: self.inference_badge.setText("Inference: -- ms") # Update performance panel (simplified to focus on essential metrics) self.gpu_usage_label.setText(f"GPU Usage: {stats.get('gpu_usage', 0):.1f}%") self.memory_usage_label.setText(f"Memory: {stats.get('memory_usage', 0):.1f} MB") self.processing_time_label.setText(f"Processing: {stats.get('processing_time', inference if inference else 0):.1f} ms") def update_progress(self, value, max_value, timestamp): self.progress.setMaximum(max_value) self.progress.setValue(value) if isinstance(timestamp, float) or isinstance(timestamp, int): timestamp_str = f"{timestamp:.2f}" else: timestamp_str = str(timestamp) self.timestamp.setText(timestamp_str) def update_multi_camera_feed(self, camera_position, pixmap, object_count=0): """Update specific camera feed in multi-camera mode""" if self.multi_camera_mode: self.multi_cam_widget.update_camera_feed(camera_position, pixmap, object_count) def get_smart_intersection_config(self): """Get current smart intersection configuration""" return { 'enabled': self.smart_intersection_mode, 'multi_camera': self.multi_camera_mode, 'scene_analytics': self.scene_analytics_toggle.isChecked(), 'multi_tracking': self.multi_tracking_toggle.isChecked(), 'speed_estimation': self.speed_estimation_toggle.isChecked() }