937 lines
35 KiB
Python
937 lines
35 KiB
Python
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 SmartIntersectionTab(QWidget):
|
|
"""Smart intersection tab with advanced analytics and multi-camera support"""
|
|
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 = True # Always enabled for this tab
|
|
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 (always enabled for this tab)
|
|
self.smart_intersection_toggle = QCheckBox("🚦 Smart Intersection Mode")
|
|
self.smart_intersection_toggle.setChecked(True)
|
|
self.smart_intersection_toggle.setEnabled(False) # Always on for this tab
|
|
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;
|
|
}
|
|
""")
|
|
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("🚦 Smart Intersection Active")
|
|
self.mode_status.setStyleSheet("color: #03DAC5; font-weight: bold; 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 for smart intersection"""
|
|
container = QWidget()
|
|
self.overlay_layout = QVBoxLayout(container)
|
|
self.overlay_layout.setContentsMargins(0, 0, 0, 0)
|
|
|
|
# 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)
|
|
self.badge_bar.addWidget(self.smart_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
|
|
analytics_group = QGroupBox("📊 Analytics Summary")
|
|
analytics_group.setStyleSheet(intersection_group.styleSheet().replace("#03DAC5", "#e67e22"))
|
|
analytics_layout = QVBoxLayout(analytics_group)
|
|
|
|
self.total_objects_label = QLabel("Total Objects: 0")
|
|
self.crosswalk_events_label = QLabel("Crosswalk Events: 0")
|
|
self.lane_events_label = QLabel("Lane Violations: 0")
|
|
self.safety_alerts_label = QLabel("Safety Alerts: 0")
|
|
|
|
for label in [self.total_objects_label, self.crosswalk_events_label,
|
|
self.lane_events_label, self.safety_alerts_label]:
|
|
label.setStyleSheet("color: white; font-size: 12px;")
|
|
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_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
|
|
self.mode_status.setText("🚦 Smart Intersection Active")
|
|
|
|
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 for smart intersection"""
|
|
# 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 analytics
|
|
self.total_objects_label.setText(f"Total Objects: {stats.get('total_objects', 0)}")
|
|
self.crosswalk_events_label.setText(f"Crosswalk Events: {stats.get('crosswalk_events', 0)}")
|
|
self.lane_events_label.setText(f"Lane Violations: {stats.get('lane_events', 0)}")
|
|
self.safety_alerts_label.setText(f"Safety Alerts: {stats.get('safety_events', 0)}")
|
|
|
|
# Update performance 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
|
|
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', 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()
|
|
}
|