123
5 days ago
parent
3cc936ba14
commit
cb52c09b01
@ -1,8 +1,8 @@
|
|||||||
# Default ignored files
|
# Default ignored files
|
||||||
/shelf/
|
/shelf/
|
||||||
/workspace.xml
|
/workspace.xml
|
||||||
# 基于编辑器的 HTTP 客户端请求
|
# 基于编辑器的 HTTP 客户端请求
|
||||||
/httpRequests/
|
/httpRequests/
|
||||||
# Datasource local storage ignored files
|
# Datasource local storage ignored files
|
||||||
/dataSources/
|
/dataSources/
|
||||||
/dataSources.local.xml
|
/dataSources.local.xml
|
||||||
|
|||||||
@ -1,8 +1,8 @@
|
|||||||
<?xml version="1.0" encoding="UTF-8"?>
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
<project version="4">
|
<project version="4">
|
||||||
<component name="ProjectModuleManager">
|
<component name="ProjectModuleManager">
|
||||||
<modules>
|
<modules>
|
||||||
<module fileurl="file://$PROJECT_DIR$/.idea/pythonProject2.iml" filepath="$PROJECT_DIR$/.idea/pythonProject2.iml" />
|
<module fileurl="file://$PROJECT_DIR$/.idea/pythonProject2.iml" filepath="$PROJECT_DIR$/.idea/pythonProject2.iml" />
|
||||||
</modules>
|
</modules>
|
||||||
</component>
|
</component>
|
||||||
</project>
|
</project>
|
||||||
@ -1,8 +1,8 @@
|
|||||||
<?xml version="1.0" encoding="UTF-8"?>
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
<module type="PYTHON_MODULE" version="4">
|
<module type="PYTHON_MODULE" version="4">
|
||||||
<component name="NewModuleRootManager">
|
<component name="NewModuleRootManager">
|
||||||
<content url="file://$MODULE_DIR$" />
|
<content url="file://$MODULE_DIR$" />
|
||||||
<orderEntry type="inheritedJdk" />
|
<orderEntry type="inheritedJdk" />
|
||||||
<orderEntry type="sourceFolder" forTests="false" />
|
<orderEntry type="sourceFolder" forTests="false" />
|
||||||
</component>
|
</component>
|
||||||
</module>
|
</module>
|
||||||
Binary file not shown.
File diff suppressed because it is too large
Load Diff
@ -0,0 +1 @@
|
|||||||
|
|
||||||
@ -1,261 +1,261 @@
|
|||||||
import cv2
|
import cv2
|
||||||
import time
|
import time
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from src import PersonDetector, DistanceCalculator, MapManager, config
|
from src import PersonDetector, DistanceCalculator, MapManager, config
|
||||||
|
|
||||||
class RealTimePersonDistanceDetector:
|
class RealTimePersonDistanceDetector:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.detector = PersonDetector()
|
self.detector = PersonDetector()
|
||||||
self.distance_calculator = DistanceCalculator()
|
self.distance_calculator = DistanceCalculator()
|
||||||
self.cap = None
|
self.cap = None
|
||||||
self.fps_counter = 0
|
self.fps_counter = 0
|
||||||
self.fps_time = time.time()
|
self.fps_time = time.time()
|
||||||
self.current_fps = 0
|
self.current_fps = 0
|
||||||
|
|
||||||
# 初始化地图管理器
|
# 初始化地图管理器
|
||||||
if config.ENABLE_MAP_DISPLAY:
|
if config.ENABLE_MAP_DISPLAY:
|
||||||
self.map_manager = MapManager(
|
self.map_manager = MapManager(
|
||||||
api_key=config.GAODE_API_KEY,
|
api_key=config.GAODE_API_KEY,
|
||||||
camera_lat=config.CAMERA_LATITUDE,
|
camera_lat=config.CAMERA_LATITUDE,
|
||||||
camera_lng=config.CAMERA_LONGITUDE
|
camera_lng=config.CAMERA_LONGITUDE
|
||||||
)
|
)
|
||||||
self.map_manager.set_camera_position(
|
self.map_manager.set_camera_position(
|
||||||
config.CAMERA_LATITUDE,
|
config.CAMERA_LATITUDE,
|
||||||
config.CAMERA_LONGITUDE,
|
config.CAMERA_LONGITUDE,
|
||||||
config.CAMERA_HEADING
|
config.CAMERA_HEADING
|
||||||
)
|
)
|
||||||
print("🗺️ 地图管理器已初始化")
|
print("🗺️ 地图管理器已初始化")
|
||||||
else:
|
else:
|
||||||
self.map_manager = None
|
self.map_manager = None
|
||||||
|
|
||||||
def initialize_camera(self):
|
def initialize_camera(self):
|
||||||
"""初始化摄像头"""
|
"""初始化摄像头"""
|
||||||
self.cap = cv2.VideoCapture(config.CAMERA_INDEX)
|
self.cap = cv2.VideoCapture(config.CAMERA_INDEX)
|
||||||
if not self.cap.isOpened():
|
if not self.cap.isOpened():
|
||||||
raise Exception(f"无法开启摄像头 {config.CAMERA_INDEX}")
|
raise Exception(f"无法开启摄像头 {config.CAMERA_INDEX}")
|
||||||
|
|
||||||
# 设置摄像头参数
|
# 设置摄像头参数
|
||||||
self.cap.set(cv2.CAP_PROP_FRAME_WIDTH, config.FRAME_WIDTH)
|
self.cap.set(cv2.CAP_PROP_FRAME_WIDTH, config.FRAME_WIDTH)
|
||||||
self.cap.set(cv2.CAP_PROP_FRAME_HEIGHT, config.FRAME_HEIGHT)
|
self.cap.set(cv2.CAP_PROP_FRAME_HEIGHT, config.FRAME_HEIGHT)
|
||||||
self.cap.set(cv2.CAP_PROP_FPS, config.FPS)
|
self.cap.set(cv2.CAP_PROP_FPS, config.FPS)
|
||||||
|
|
||||||
# 获取实际设置的参数
|
# 获取实际设置的参数
|
||||||
actual_width = int(self.cap.get(cv2.CAP_PROP_FRAME_WIDTH))
|
actual_width = int(self.cap.get(cv2.CAP_PROP_FRAME_WIDTH))
|
||||||
actual_height = int(self.cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
|
actual_height = int(self.cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
|
||||||
actual_fps = int(self.cap.get(cv2.CAP_PROP_FPS))
|
actual_fps = int(self.cap.get(cv2.CAP_PROP_FPS))
|
||||||
|
|
||||||
print(f"摄像头初始化成功:")
|
print(f"摄像头初始化成功:")
|
||||||
print(f" 分辨率: {actual_width}x{actual_height}")
|
print(f" 分辨率: {actual_width}x{actual_height}")
|
||||||
print(f" 帧率: {actual_fps} FPS")
|
print(f" 帧率: {actual_fps} FPS")
|
||||||
|
|
||||||
def calculate_fps(self):
|
def calculate_fps(self):
|
||||||
"""计算实际帧率"""
|
"""计算实际帧率"""
|
||||||
self.fps_counter += 1
|
self.fps_counter += 1
|
||||||
current_time = time.time()
|
current_time = time.time()
|
||||||
if current_time - self.fps_time >= 1.0:
|
if current_time - self.fps_time >= 1.0:
|
||||||
self.current_fps = self.fps_counter
|
self.current_fps = self.fps_counter
|
||||||
self.fps_counter = 0
|
self.fps_counter = 0
|
||||||
self.fps_time = current_time
|
self.fps_time = current_time
|
||||||
|
|
||||||
def draw_info_panel(self, frame, person_count=0):
|
def draw_info_panel(self, frame, person_count=0):
|
||||||
"""绘制信息面板"""
|
"""绘制信息面板"""
|
||||||
height, width = frame.shape[:2]
|
height, width = frame.shape[:2]
|
||||||
|
|
||||||
# 绘制顶部信息栏
|
# 绘制顶部信息栏
|
||||||
info_height = 60
|
info_height = 60
|
||||||
cv2.rectangle(frame, (0, 0), (width, info_height), (0, 0, 0), -1)
|
cv2.rectangle(frame, (0, 0), (width, info_height), (0, 0, 0), -1)
|
||||||
|
|
||||||
# 显示FPS
|
# 显示FPS
|
||||||
fps_text = f"FPS: {self.current_fps}"
|
fps_text = f"FPS: {self.current_fps}"
|
||||||
cv2.putText(frame, fps_text, (10, 25), config.FONT, 0.6, (0, 255, 0), 2)
|
cv2.putText(frame, fps_text, (10, 25), config.FONT, 0.6, (0, 255, 0), 2)
|
||||||
|
|
||||||
# 显示人员计数
|
# 显示人员计数
|
||||||
person_text = f"Persons: {person_count}"
|
person_text = f"Persons: {person_count}"
|
||||||
cv2.putText(frame, person_text, (150, 25), config.FONT, 0.6, (0, 255, 255), 2)
|
cv2.putText(frame, person_text, (150, 25), config.FONT, 0.6, (0, 255, 255), 2)
|
||||||
|
|
||||||
# 显示模型信息
|
# 显示模型信息
|
||||||
model_text = self.detector.get_model_info()
|
model_text = self.detector.get_model_info()
|
||||||
cv2.putText(frame, model_text, (10, 45), config.FONT, 0.5, (255, 255, 255), 1)
|
cv2.putText(frame, model_text, (10, 45), config.FONT, 0.5, (255, 255, 255), 1)
|
||||||
|
|
||||||
# 显示操作提示
|
# 显示操作提示
|
||||||
help_text = "Press 'q' to quit | 'c' to calibrate | 'r' to reset | 'm' to open map"
|
help_text = "Press 'q' to quit | 'c' to calibrate | 'r' to reset | 'm' to open map"
|
||||||
text_size = cv2.getTextSize(help_text, config.FONT, 0.5, 1)[0]
|
text_size = cv2.getTextSize(help_text, config.FONT, 0.5, 1)[0]
|
||||||
cv2.putText(frame, help_text, (width - text_size[0] - 10, 25),
|
cv2.putText(frame, help_text, (width - text_size[0] - 10, 25),
|
||||||
config.FONT, 0.5, (255, 255, 0), 1)
|
config.FONT, 0.5, (255, 255, 0), 1)
|
||||||
|
|
||||||
# 显示地图状态
|
# 显示地图状态
|
||||||
if self.map_manager:
|
if self.map_manager:
|
||||||
map_status = "Map: ON"
|
map_status = "Map: ON"
|
||||||
cv2.putText(frame, map_status, (10, height - 10),
|
cv2.putText(frame, map_status, (10, height - 10),
|
||||||
config.FONT, 0.5, (0, 255, 255), 1)
|
config.FONT, 0.5, (0, 255, 255), 1)
|
||||||
|
|
||||||
return frame
|
return frame
|
||||||
|
|
||||||
def calibrate_distance(self, detections):
|
def calibrate_distance(self, detections):
|
||||||
"""距离校准模式"""
|
"""距离校准模式"""
|
||||||
if len(detections) == 0:
|
if len(detections) == 0:
|
||||||
print("未检测到人体，无法校准")
|
print("未检测到人体，无法校准")
|
||||||
return
|
return
|
||||||
|
|
||||||
print("\n=== 距离校准模式 ===")
|
print("\n=== 距离校准模式 ===")
|
||||||
print("请确保画面中有一个人，并输入该人距离摄像头的真实距离")
|
print("请确保画面中有一个人，并输入该人距离摄像头的真实距离")
|
||||||
|
|
||||||
try:
|
try:
|
||||||
real_distance = float(input("请输入真实距离（厘米）: "))
|
real_distance = float(input("请输入真实距离（厘米）: "))
|
||||||
|
|
||||||
# 使用第一个检测到的人进行校准
|
# 使用第一个检测到的人进行校准
|
||||||
detection = detections[0]
|
detection = detections[0]
|
||||||
x1, y1, x2, y2, conf = detection
|
x1, y1, x2, y2, conf = detection
|
||||||
bbox_height = y2 - y1
|
bbox_height = y2 - y1
|
||||||
|
|
||||||
# 更新参考参数
|
# 更新参考参数
|
||||||
config.REFERENCE_DISTANCE = real_distance
|
config.REFERENCE_DISTANCE = real_distance
|
||||||
config.REFERENCE_HEIGHT_PIXELS = bbox_height
|
config.REFERENCE_HEIGHT_PIXELS = bbox_height
|
||||||
|
|
||||||
# 重新初始化距离计算器
|
# 重新初始化距离计算器
|
||||||
self.distance_calculator = DistanceCalculator()
|
self.distance_calculator = DistanceCalculator()
|
||||||
|
|
||||||
print(f"校准完成！")
|
print(f"校准完成！")
|
||||||
print(f"参考距离: {real_distance}cm")
|
print(f"参考距离: {real_distance}cm")
|
||||||
print(f"参考像素高度: {bbox_height}px")
|
print(f"参考像素高度: {bbox_height}px")
|
||||||
|
|
||||||
except ValueError:
|
except ValueError:
|
||||||
print("输入无效，校准取消")
|
print("输入无效，校准取消")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"校准失败: {e}")
|
print(f"校准失败: {e}")
|
||||||
|
|
||||||
def process_frame(self, frame):
|
def process_frame(self, frame):
|
||||||
"""处理单帧图像"""
|
"""处理单帧图像"""
|
||||||
# 检测人体
|
# 检测人体
|
||||||
detections = self.detector.detect_persons(frame)
|
detections = self.detector.detect_persons(frame)
|
||||||
|
|
||||||
# 计算距离并更新地图位置
|
# 计算距离并更新地图位置
|
||||||
distances = []
|
distances = []
|
||||||
if self.map_manager:
|
if self.map_manager:
|
||||||
self.map_manager.clear_persons()
|
self.map_manager.clear_persons()
|
||||||
|
|
||||||
for i, detection in enumerate(detections):
|
for i, detection in enumerate(detections):
|
||||||
bbox = detection[:4] # [x1, y1, x2, y2]
|
bbox = detection[:4] # [x1, y1, x2, y2]
|
||||||
x1, y1, x2, y2 = bbox
|
x1, y1, x2, y2 = bbox
|
||||||
distance = self.distance_calculator.get_distance(bbox)
|
distance = self.distance_calculator.get_distance(bbox)
|
||||||
distance_str = self.distance_calculator.format_distance(distance)
|
distance_str = self.distance_calculator.format_distance(distance)
|
||||||
distances.append(distance_str)
|
distances.append(distance_str)
|
||||||
|
|
||||||
# 更新地图上的人员位置
|
# 更新地图上的人员位置
|
||||||
if self.map_manager:
|
if self.map_manager:
|
||||||
# 计算人体中心点
|
# 计算人体中心点
|
||||||
center_x = (x1 + x2) / 2
|
center_x = (x1 + x2) / 2
|
||||||
center_y = (y1 + y2) / 2
|
center_y = (y1 + y2) / 2
|
||||||
|
|
||||||
# 将距离从厘米转换为米
|
# 将距离从厘米转换为米
|
||||||
distance_meters = distance / 100.0
|
distance_meters = distance / 100.0
|
||||||
|
|
||||||
# 添加到地图
|
# 添加到地图
|
||||||
self.map_manager.add_person_position(
|
self.map_manager.add_person_position(
|
||||||
center_x, center_y, distance_meters,
|
center_x, center_y, distance_meters,
|
||||||
frame.shape[1], frame.shape[0], # width, height
|
frame.shape[1], frame.shape[0], # width, height
|
||||||
f"P{i+1}"
|
f"P{i+1}"
|
||||||
)
|
)
|
||||||
|
|
||||||
# 绘制检测结果
|
# 绘制检测结果
|
||||||
frame = self.detector.draw_detections(frame, detections, distances)
|
frame = self.detector.draw_detections(frame, detections, distances)
|
||||||
|
|
||||||
# 绘制信息面板
|
# 绘制信息面板
|
||||||
frame = self.draw_info_panel(frame, len(detections))
|
frame = self.draw_info_panel(frame, len(detections))
|
||||||
|
|
||||||
# 计算FPS
|
# 计算FPS
|
||||||
self.calculate_fps()
|
self.calculate_fps()
|
||||||
|
|
||||||
return frame, detections
|
return frame, detections
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
"""运行主程序"""
|
"""运行主程序"""
|
||||||
try:
|
try:
|
||||||
print("正在初始化...")
|
print("正在初始化...")
|
||||||
self.initialize_camera()
|
self.initialize_camera()
|
||||||
|
|
||||||
print("系统启动成功!")
|
print("系统启动成功!")
|
||||||
print("操作说明:")
|
print("操作说明:")
|
||||||
print(" - 按 'q' 键退出程序")
|
print(" - 按 'q' 键退出程序")
|
||||||
print(" - 按 'c' 键进入距离校准模式")
|
print(" - 按 'c' 键进入距离校准模式")
|
||||||
print(" - 按 'r' 键重置为默认参数")
|
print(" - 按 'r' 键重置为默认参数")
|
||||||
print(" - 按 's' 键保存当前帧")
|
print(" - 按 's' 键保存当前帧")
|
||||||
if self.map_manager:
|
if self.map_manager:
|
||||||
print(" - 按 'm' 键打开地图显示")
|
print(" - 按 'm' 键打开地图显示")
|
||||||
print(" - 按 'h' 键设置摄像头朝向")
|
print(" - 按 'h' 键设置摄像头朝向")
|
||||||
print("\n开始实时检测...")
|
print("\n开始实时检测...")
|
||||||
|
|
||||||
frame_count = 0
|
frame_count = 0
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
ret, frame = self.cap.read()
|
ret, frame = self.cap.read()
|
||||||
if not ret:
|
if not ret:
|
||||||
print("无法读取摄像头画面")
|
print("无法读取摄像头画面")
|
||||||
break
|
break
|
||||||
|
|
||||||
# 处理帧
|
# 处理帧
|
||||||
processed_frame, detections = self.process_frame(frame)
|
processed_frame, detections = self.process_frame(frame)
|
||||||
|
|
||||||
# 显示结果
|
# 显示结果
|
||||||
cv2.imshow('Real-time Person Distance Detection', processed_frame)
|
cv2.imshow('Real-time Person Distance Detection', processed_frame)
|
||||||
|
|
||||||
# 处理按键
|
# 处理按键
|
||||||
key = cv2.waitKey(1) & 0xFF
|
key = cv2.waitKey(1) & 0xFF
|
||||||
|
|
||||||
if key == ord('q'):
|
if key == ord('q'):
|
||||||
print("用户退出程序")
|
print("用户退出程序")
|
||||||
break
|
break
|
||||||
elif key == ord('c'):
|
elif key == ord('c'):
|
||||||
# 校准模式
|
# 校准模式
|
||||||
self.calibrate_distance(detections)
|
self.calibrate_distance(detections)
|
||||||
elif key == ord('r'):
|
elif key == ord('r'):
|
||||||
# 重置参数
|
# 重置参数
|
||||||
print("重置为默认参数")
|
print("重置为默认参数")
|
||||||
self.distance_calculator = DistanceCalculator()
|
self.distance_calculator = DistanceCalculator()
|
||||||
elif key == ord('s'):
|
elif key == ord('s'):
|
||||||
# 保存当前帧
|
# 保存当前帧
|
||||||
filename = f"capture_{int(time.time())}.jpg"
|
filename = f"capture_{int(time.time())}.jpg"
|
||||||
cv2.imwrite(filename, processed_frame)
|
cv2.imwrite(filename, processed_frame)
|
||||||
print(f"已保存截图: {filename}")
|
print(f"已保存截图: {filename}")
|
||||||
elif key == ord('m') and self.map_manager:
|
elif key == ord('m') and self.map_manager:
|
||||||
# 打开地图显示
|
# 打开地图显示
|
||||||
print("正在打开地图...")
|
print("正在打开地图...")
|
||||||
self.map_manager.open_map()
|
self.map_manager.open_map()
|
||||||
elif key == ord('h') and self.map_manager:
|
elif key == ord('h') and self.map_manager:
|
||||||
# 设置摄像头朝向
|
# 设置摄像头朝向
|
||||||
try:
|
try:
|
||||||
heading = float(input("请输入摄像头朝向角度 (0-360°, 0为正北): "))
|
heading = float(input("请输入摄像头朝向角度 (0-360°, 0为正北): "))
|
||||||
if 0 <= heading <= 360:
|
if 0 <= heading <= 360:
|
||||||
self.map_manager.update_camera_heading(heading)
|
self.map_manager.update_camera_heading(heading)
|
||||||
else:
|
else:
|
||||||
print("角度必须在0-360度之间")
|
print("角度必须在0-360度之间")
|
||||||
except ValueError:
|
except ValueError:
|
||||||
print("输入无效")
|
print("输入无效")
|
||||||
|
|
||||||
frame_count += 1
|
frame_count += 1
|
||||||
|
|
||||||
except KeyboardInterrupt:
|
except KeyboardInterrupt:
|
||||||
print("\n程序被用户中断")
|
print("\n程序被用户中断")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"程序运行出错: {e}")
|
print(f"程序运行出错: {e}")
|
||||||
finally:
|
finally:
|
||||||
self.cleanup()
|
self.cleanup()
|
||||||
|
|
||||||
def cleanup(self):
|
def cleanup(self):
|
||||||
"""清理资源"""
|
"""清理资源"""
|
||||||
if self.cap:
|
if self.cap:
|
||||||
self.cap.release()
|
self.cap.release()
|
||||||
cv2.destroyAllWindows()
|
cv2.destroyAllWindows()
|
||||||
print("资源已清理，程序结束")
|
print("资源已清理，程序结束")
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
"""主函数"""
|
"""主函数"""
|
||||||
print("=" * 50)
|
print("=" * 50)
|
||||||
print("实时人体距离检测系统")
|
print("实时人体距离检测系统")
|
||||||
print("=" * 50)
|
print("=" * 50)
|
||||||
|
|
||||||
detector = RealTimePersonDistanceDetector()
|
detector = RealTimePersonDistanceDetector()
|
||||||
detector.run()
|
detector.run()
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@ -1,97 +1,97 @@
|
|||||||
1#!/usr/bin/env python3
|
1#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
"""
|
"""
|
||||||
无人机战场态势感知系统 - 启动脚本
|
无人机战场态势感知系统 - 启动脚本
|
||||||
让用户选择运行模式
|
让用户选择运行模式
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import sys
|
import sys
|
||||||
import os
|
import os
|
||||||
|
|
||||||
def show_menu():
|
def show_menu():
|
||||||
"""显示菜单"""
|
"""显示菜单"""
|
||||||
print("=" * 60)
|
print("=" * 60)
|
||||||
print("🚁 无人机战场态势感知系统")
|
print("🚁 无人机战场态势感知系统")
|
||||||
print("=" * 60)
|
print("=" * 60)
|
||||||
print()
|
print()
|
||||||
print("请选择运行模式:")
|
print("请选择运行模式:")
|
||||||
print()
|
print()
|
||||||
print("1. 🌐 Web模式 (推荐)")
|
print("1. 🌐 Web模式 (推荐)")
|
||||||
print(" • 地图作为主界面")
|
print(" • 地图作为主界面")
|
||||||
print(" • 通过浏览器操作")
|
print(" • 通过浏览器操作")
|
||||||
print(" • 可视化程度更高")
|
print(" • 可视化程度更高")
|
||||||
print(" • 支持远程访问")
|
print(" • 支持远程访问")
|
||||||
print()
|
print()
|
||||||
print("2. 🖥️ 传统模式")
|
print("2. 🖥️ 传统模式")
|
||||||
print(" • 直接显示摄像头画面")
|
print(" • 直接显示摄像头画面")
|
||||||
print(" • 键盘快捷键操作")
|
print(" • 键盘快捷键操作")
|
||||||
print(" • 性能更好")
|
print(" • 性能更好")
|
||||||
print(" • 适合本地使用")
|
print(" • 适合本地使用")
|
||||||
print()
|
print()
|
||||||
print("3. ⚙️ 配置摄像头位置")
|
print("3. ⚙️ 配置摄像头位置")
|
||||||
print(" • 设置GPS坐标")
|
print(" • 设置GPS坐标")
|
||||||
print(" • 配置朝向角度")
|
print(" • 配置朝向角度")
|
||||||
print(" • 设置API Key")
|
print(" • 设置API Key")
|
||||||
print()
|
print()
|
||||||
print("4. 🧪 运行系统测试")
|
print("4. 🧪 运行系统测试")
|
||||||
print(" • 检查各模块状态")
|
print(" • 检查各模块状态")
|
||||||
print(" • 验证系统功能")
|
print(" • 验证系统功能")
|
||||||
print()
|
print()
|
||||||
print("0. ❌ 退出")
|
print("0. ❌ 退出")
|
||||||
print()
|
print()
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
"""主函数"""
|
"""主函数"""
|
||||||
while True:
|
while True:
|
||||||
show_menu()
|
show_menu()
|
||||||
try:
|
try:
|
||||||
choice = input("请输入选择 (0-4): ").strip()
|
choice = input("请输入选择 (0-4): ").strip()
|
||||||
|
|
||||||
if choice == "1":
|
if choice == "1":
|
||||||
print("\n🌐 启动Web模式...")
|
print("\n🌐 启动Web模式...")
|
||||||
import main_web
|
import main_web
|
||||||
main_web.main()
|
main_web.main()
|
||||||
break
|
break
|
||||||
|
|
||||||
elif choice == "2":
|
elif choice == "2":
|
||||||
print("\n🖥️ 启动传统模式...")
|
print("\n🖥️ 启动传统模式...")
|
||||||
import main
|
import main
|
||||||
main.main()
|
main.main()
|
||||||
break
|
break
|
||||||
|
|
||||||
elif choice == "3":
|
elif choice == "3":
|
||||||
print("\n⚙️ 配置摄像头位置...")
|
print("\n⚙️ 配置摄像头位置...")
|
||||||
import sys
|
import sys
|
||||||
sys.path.append('tools')
|
sys.path.append('tools')
|
||||||
import setup_camera_location
|
import setup_camera_location
|
||||||
setup_camera_location.main()
|
setup_camera_location.main()
|
||||||
print("\n配置完成，请重新选择运行模式")
|
print("\n配置完成，请重新选择运行模式")
|
||||||
input("按回车键继续...")
|
input("按回车键继续...")
|
||||||
|
|
||||||
elif choice == "4":
|
elif choice == "4":
|
||||||
print("\n🧪 运行系统测试...")
|
print("\n🧪 运行系统测试...")
|
||||||
import sys
|
import sys
|
||||||
sys.path.append('tests')
|
sys.path.append('tests')
|
||||||
import test_system
|
import test_system
|
||||||
test_system.main()
|
test_system.main()
|
||||||
print("\n测试完成")
|
print("\n测试完成")
|
||||||
input("按回车键继续...")
|
input("按回车键继续...")
|
||||||
|
|
||||||
elif choice == "0":
|
elif choice == "0":
|
||||||
print("\n👋 再见!")
|
print("\n👋 再见!")
|
||||||
sys.exit(0)
|
sys.exit(0)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
print("\n❌ 无效选择，请重新输入")
|
print("\n❌ 无效选择，请重新输入")
|
||||||
input("按回车键继续...")
|
input("按回车键继续...")
|
||||||
|
|
||||||
except KeyboardInterrupt:
|
except KeyboardInterrupt:
|
||||||
print("\n\n👋 再见!")
|
print("\n\n👋 再见!")
|
||||||
sys.exit(0)
|
sys.exit(0)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"\n❌ 运行出错: {e}")
|
print(f"\n❌ 运行出错: {e}")
|
||||||
input("按回车键继续...")
|
input("按回车键继续...")
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
@ -1,51 +1,51 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
"""
|
"""
|
||||||
实时人体距离检测系统 - 核心模块包
|
实时人体距离检测系统 - 核心模块包
|
||||||
|
|
||||||
包含以下模块:
|
包含以下模块:
|
||||||
- config: 配置文件
|
- config: 配置文件
|
||||||
- person_detector: 人体检测模块
|
- person_detector: 人体检测模块
|
||||||
- distance_calculator: 距离计算模块
|
- distance_calculator: 距离计算模块
|
||||||
"""
|
"""
|
||||||
|
|
||||||
__version__ = "1.0.0"
|
__version__ = "1.0.0"
|
||||||
__author__ = "Distance Detection System"
|
__author__ = "Distance Detection System"
|
||||||
|
|
||||||
# 导入核心模块
|
# 导入核心模块
|
||||||
from .config import *
|
from .config import *
|
||||||
from .person_detector import PersonDetector
|
from .person_detector import PersonDetector
|
||||||
from .distance_calculator import DistanceCalculator
|
from .distance_calculator import DistanceCalculator
|
||||||
from .map_manager import MapManager
|
from .map_manager import MapManager
|
||||||
from .web_server import WebServer
|
from .web_server import WebServer
|
||||||
from .mobile_connector import MobileConnector, MobileDevice
|
from .mobile_connector import MobileConnector, MobileDevice
|
||||||
from .orientation_detector import OrientationDetector
|
from .orientation_detector import OrientationDetector
|
||||||
from .web_orientation_detector import WebOrientationDetector
|
from .web_orientation_detector import WebOrientationDetector
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
'PersonDetector',
|
'PersonDetector',
|
||||||
'DistanceCalculator',
|
'DistanceCalculator',
|
||||||
'MapManager',
|
'MapManager',
|
||||||
'WebServer',
|
'WebServer',
|
||||||
'MobileConnector',
|
'MobileConnector',
|
||||||
'MobileDevice',
|
'MobileDevice',
|
||||||
'CAMERA_INDEX',
|
'CAMERA_INDEX',
|
||||||
'FRAME_WIDTH',
|
'FRAME_WIDTH',
|
||||||
'FRAME_HEIGHT',
|
'FRAME_HEIGHT',
|
||||||
'FPS',
|
'FPS',
|
||||||
'MODEL_PATH',
|
'MODEL_PATH',
|
||||||
'CONFIDENCE_THRESHOLD',
|
'CONFIDENCE_THRESHOLD',
|
||||||
'IOU_THRESHOLD',
|
'IOU_THRESHOLD',
|
||||||
'KNOWN_PERSON_HEIGHT',
|
'KNOWN_PERSON_HEIGHT',
|
||||||
'FOCAL_LENGTH',
|
'FOCAL_LENGTH',
|
||||||
'REFERENCE_DISTANCE',
|
'REFERENCE_DISTANCE',
|
||||||
'REFERENCE_HEIGHT_PIXELS',
|
'REFERENCE_HEIGHT_PIXELS',
|
||||||
'FONT',
|
'FONT',
|
||||||
'FONT_SCALE',
|
'FONT_SCALE',
|
||||||
'FONT_THICKNESS',
|
'FONT_THICKNESS',
|
||||||
'BOX_COLOR',
|
'BOX_COLOR',
|
||||||
'TEXT_COLOR',
|
'TEXT_COLOR',
|
||||||
'TEXT_BG_COLOR',
|
'TEXT_BG_COLOR',
|
||||||
'PERSON_CLASS_ID'
|
'PERSON_CLASS_ID'
|
||||||
]
|
]
|
||||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@ -0,0 +1,131 @@
|
|||||||
|
# Air模块配置文件
|
||||||
|
# RoboMaster TT (Tello TLW004) 无人机配置
|
||||||
|
|
||||||
|
# 无人机基本配置
|
||||||
|
drone:
|
||||||
|
type: "tello" # 无人机类型
|
||||||
|
model: "TLW004" # 型号
|
||||||
|
name: "RoboMaster TT" # 显示名称
|
||||||
|
|
||||||
|
# 网络连接配置
|
||||||
|
connection:
|
||||||
|
ip: "192.168.10.1" # 无人机IP地址
|
||||||
|
cmd_port: 8889 # 命令端口
|
||||||
|
state_port: 8890 # 状态端口
|
||||||
|
video_port: 11111 # 视频端口
|
||||||
|
timeout: 5 # 连接超时时间(秒)
|
||||||
|
|
||||||
|
# 视频流配置
|
||||||
|
video:
|
||||||
|
# 支持的视频流格式
|
||||||
|
formats:
|
||||||
|
udp: "udp://{ip}:{port}"
|
||||||
|
rtsp: "rtsp://{ip}:554/live"
|
||||||
|
http: "http://{ip}:8080/video"
|
||||||
|
|
||||||
|
# 默认视频流URL
|
||||||
|
default_stream: "udp://192.168.10.1:11111"
|
||||||
|
|
||||||
|
# 视频参数
|
||||||
|
resolution:
|
||||||
|
width: 960
|
||||||
|
height: 720
|
||||||
|
fps: 30
|
||||||
|
|
||||||
|
# 缓冲设置
|
||||||
|
buffer_size: 10
|
||||||
|
timeout: 10
|
||||||
|
|
||||||
|
# 录制设置
|
||||||
|
recording:
|
||||||
|
enabled: false
|
||||||
|
format: "mp4"
|
||||||
|
quality: "high"
|
||||||
|
|
||||||
|
# 图像分析配置
|
||||||
|
analysis:
|
||||||
|
# 检测阈值
|
||||||
|
confidence_threshold: 0.25
|
||||||
|
part_confidence_threshold: 0.3
|
||||||
|
|
||||||
|
# 模型路径（相对于项目根目录）
|
||||||
|
models:
|
||||||
|
ship_detector: "models/best.pt"
|
||||||
|
part_detector: "models/part_detectors/best.pt"
|
||||||
|
classifier: "models/custom/best.pt"
|
||||||
|
|
||||||
|
# 检测类别
|
||||||
|
ship_classes:
|
||||||
|
- "航空母舰"
|
||||||
|
- "驱逐舰"
|
||||||
|
- "护卫舰"
|
||||||
|
- "潜艇"
|
||||||
|
- "商船"
|
||||||
|
- "油轮"
|
||||||
|
|
||||||
|
# 部件类别
|
||||||
|
part_classes:
|
||||||
|
- "舰桥"
|
||||||
|
- "雷达"
|
||||||
|
- "舰炮"
|
||||||
|
- "导弹发射器"
|
||||||
|
- "直升机甲板"
|
||||||
|
- "烟囱"
|
||||||
|
|
||||||
|
# Web界面配置
|
||||||
|
web:
|
||||||
|
host: "0.0.0.0"
|
||||||
|
port: 5000
|
||||||
|
debug: true
|
||||||
|
|
||||||
|
# 静态文件路径
|
||||||
|
static_folder: "web/static"
|
||||||
|
template_folder: "web/templates"
|
||||||
|
|
||||||
|
# 上传设置
|
||||||
|
upload:
|
||||||
|
max_file_size: "10MB"
|
||||||
|
allowed_extensions: [".jpg", ".jpeg", ".png", ".mp4", ".avi"]
|
||||||
|
save_path: "uploads/drone_captures"
|
||||||
|
|
||||||
|
# 安全设置
|
||||||
|
safety:
|
||||||
|
max_height: 100 # 最大飞行高度(米)
|
||||||
|
max_distance: 500 # 最大飞行距离(米)
|
||||||
|
min_battery: 15 # 最低电量百分比
|
||||||
|
return_home_battery: 30 # 自动返航电量
|
||||||
|
|
||||||
|
# 飞行限制区域
|
||||||
|
no_fly_zones: []
|
||||||
|
|
||||||
|
# 日志配置
|
||||||
|
logging:
|
||||||
|
level: "INFO"
|
||||||
|
format: "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
||||||
|
|
||||||
|
# 日志文件
|
||||||
|
files:
|
||||||
|
main: "logs/air_main.log"
|
||||||
|
drone: "logs/drone.log"
|
||||||
|
video: "logs/video.log"
|
||||||
|
analysis: "logs/analysis.log"
|
||||||
|
|
||||||
|
# 性能配置
|
||||||
|
performance:
|
||||||
|
# GPU使用
|
||||||
|
use_gpu: true
|
||||||
|
gpu_memory_fraction: 0.7
|
||||||
|
|
||||||
|
# 多线程设置
|
||||||
|
max_workers: 4
|
||||||
|
|
||||||
|
# 内存限制
|
||||||
|
max_memory_usage: "2GB"
|
||||||
|
|
||||||
|
# 开发调试配置
|
||||||
|
debug:
|
||||||
|
enabled: false
|
||||||
|
save_frames: false
|
||||||
|
frame_save_path: "debug/frames"
|
||||||
|
log_commands: true
|
||||||
|
mock_drone: false # 是否使用模拟无人机
|
||||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,13 @@
|
|||||||
|
"""
|
||||||
|
无人机接口子系统(DroneInterface)
|
||||||
|
-------------
|
||||||
|
与无人机建立通信连接
|
||||||
|
接收无人机传回的视频流
|
||||||
|
将视频流转发给图像分析子系统
|
||||||
|
向无人机发送控制命令(如需要)
|
||||||
|
"""
|
||||||
|
|
||||||
|
from .drone_manager import DroneManager
|
||||||
|
from .video_receiver import VideoReceiver
|
||||||
|
|
||||||
|
__all__ = ['DroneManager', 'VideoReceiver']
|
||||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@ -0,0 +1,12 @@
|
|||||||
|
"""
|
||||||
|
图像分析子系统(ImageAnalyzer)
|
||||||
|
-------------
|
||||||
|
调用模型进行舰船检测、分类和部件识别
|
||||||
|
处理图像预处理和后处理
|
||||||
|
生成分析结果报告
|
||||||
|
提供API接口供Web应用调用
|
||||||
|
"""
|
||||||
|
|
||||||
|
from .analyzer import ImageAnalyzer
|
||||||
|
|
||||||
|
__all__ = ['ImageAnalyzer']
|
||||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,138 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="zh-CN">
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>{% block title %}舰船识别系统{% endblock %} - ShipAI</title>
|
||||||
|
<!-- Bootstrap CSS -->
|
||||||
|
<link href="https://cdn.jsdelivr.net/npm/bootstrap@5.2.3/dist/css/bootstrap.min.css" rel="stylesheet">
|
||||||
|
<!-- Font Awesome 图标 -->
|
||||||
|
<link href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.2/css/all.min.css" rel="stylesheet">
|
||||||
|
<!-- 自定义CSS -->
|
||||||
|
<link rel="stylesheet" href="{{ url_for('static', filename='css/style.css') }}">
|
||||||
|
{% block extra_css %}{% endblock %}
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<!-- 导航栏 -->
|
||||||
|
<nav class="navbar navbar-expand-lg navbar-dark bg-dark">
|
||||||
|
<div class="container">
|
||||||
|
<a class="navbar-brand" href="{{ url_for('index') }}">ShipAI - 智能舰船识别系统</a>
|
||||||
|
<button class="navbar-toggler" type="button" data-bs-toggle="collapse" data-bs-target="#navbarNav">
|
||||||
|
<span class="navbar-toggler-icon"></span>
|
||||||
|
</button>
|
||||||
|
<div class="collapse navbar-collapse" id="navbarNav">
|
||||||
|
<ul class="navbar-nav me-auto">
|
||||||
|
<li class="nav-item">
|
||||||
|
<a class="nav-link" href="{{ url_for('index') }}">首页</a>
|
||||||
|
</li>
|
||||||
|
<li class="nav-item">
|
||||||
|
<a class="nav-link" href="{{ url_for('gallery') }}">样本图库</a>
|
||||||
|
</li>
|
||||||
|
<li class="nav-item">
|
||||||
|
<a class="nav-link" href="{{ url_for('drone_control') }}">无人机控制</a>
|
||||||
|
</li>
|
||||||
|
<li class="nav-item dropdown">
|
||||||
|
<a class="nav-link dropdown-toggle" href="#" id="analysisDropdown" role="button" data-bs-toggle="dropdown">
|
||||||
|
分析工具
|
||||||
|
</a>
|
||||||
|
<div class="dropdown-menu">
|
||||||
|
<a class="dropdown-item" href="{{ url_for('image_analysis') }}">图像分析</a>
|
||||||
|
<a class="dropdown-item" href="{{ url_for('analytics') }}">分析报告</a>
|
||||||
|
<a class="dropdown-item" href="{{ url_for('data_storage') }}">数据存储</a>
|
||||||
|
</div>
|
||||||
|
</li>
|
||||||
|
<li class="nav-item dropdown">
|
||||||
|
<a class="nav-link dropdown-toggle" href="#" id="modelDropdown" role="button" data-bs-toggle="dropdown">
|
||||||
|
模型管理
|
||||||
|
</a>
|
||||||
|
<div class="dropdown-menu">
|
||||||
|
<a class="dropdown-item" href="{{ url_for('model_settings') }}">模型设置</a>
|
||||||
|
<a class="dropdown-item" href="{{ url_for('annotation_tool') }}">图像标注</a>
|
||||||
|
<a class="dropdown-item" href="{{ url_for('train_model') }}">模型训练</a>
|
||||||
|
</div>
|
||||||
|
</li>
|
||||||
|
<li class="nav-item">
|
||||||
|
<a class="nav-link {% if request.path == '/ship-database' %}active{% endif %}" href="/ship-database">舰船数据库</a>
|
||||||
|
</li>
|
||||||
|
<li class="nav-item dropdown">
|
||||||
|
<a class="nav-link dropdown-toggle" href="#" id="modelsDropdown" role="button" data-bs-toggle="dropdown" aria-expanded="false">
|
||||||
|
部件检测
|
||||||
|
</a>
|
||||||
|
<ul class="dropdown-menu" aria-labelledby="modelsDropdown">
|
||||||
|
<li><a class="dropdown-item" href="{{ url_for('part_detection') }}">部件库管理</a></li>
|
||||||
|
<li><a class="dropdown-item" href="{{ url_for('annotation_tool', type='part') }}">部件标注工具</a></li>
|
||||||
|
<li><a class="dropdown-item" href="{{ url_for('train_part_model') }}">部件模型训练</a></li>
|
||||||
|
</ul>
|
||||||
|
</li>
|
||||||
|
<li class="nav-item" id="nav-history">
|
||||||
|
<a class="nav-link" href="{{ url_for('detection_history') }}">检测历史</a>
|
||||||
|
</li>
|
||||||
|
<li class="nav-item">
|
||||||
|
<a class="nav-link" href="{{ url_for('about') }}">关于我们</a>
|
||||||
|
</li>
|
||||||
|
</ul>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</nav>
|
||||||
|
|
||||||
|
<!-- 消息提示 -->
|
||||||
|
<div class="container mt-3">
|
||||||
|
{% with messages = get_flashed_messages(with_categories=true) %}
|
||||||
|
{% if messages %}
|
||||||
|
{% for category, message in messages %}
|
||||||
|
<div class="alert alert-{{ category if category != 'message' else 'info' }} alert-dismissible fade show">
|
||||||
|
{{ message }}
|
||||||
|
<button type="button" class="btn-close" data-bs-dismiss="alert"></button>
|
||||||
|
</div>
|
||||||
|
{% endfor %}
|
||||||
|
{% endif %}
|
||||||
|
{% endwith %}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- 主要内容 -->
|
||||||
|
<main class="py-4">
|
||||||
|
{% block content %}{% endblock %}
|
||||||
|
</main>
|
||||||
|
|
||||||
|
<!-- 页脚 -->
|
||||||
|
<footer class="bg-dark text-white py-4 mt-5">
|
||||||
|
<div class="container">
|
||||||
|
<div class="row">
|
||||||
|
<div class="col-md-6">
|
||||||
|
<h5>ShipAI - 智能舰船识别系统</h5>
|
||||||
|
<p>基于深度学习的海上舰船自动识别与分析平台</p>
|
||||||
|
</div>
|
||||||
|
<div class="col-md-6 text-md-end">
|
||||||
|
<p>© {{ current_year }} ShipAI 团队</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</footer>
|
||||||
|
|
||||||
|
<!-- JavaScript -->
|
||||||
|
<script src="https://cdn.jsdelivr.net/npm/bootstrap@5.2.3/dist/js/bootstrap.bundle.min.js"></script>
|
||||||
|
<script src="https://code.jquery.com/jquery-3.6.0.min.js"></script>
|
||||||
|
<script src="{{ url_for('static', filename='js/modal_fix.js') }}"></script>
|
||||||
|
<script>
|
||||||
|
// 初始化所有模态框
|
||||||
|
document.addEventListener('DOMContentLoaded', function() {
|
||||||
|
// 使所有具有data-bs-toggle="modal"属性的元素正确工作
|
||||||
|
var modalTriggers = document.querySelectorAll('[data-bs-toggle="modal"]');
|
||||||
|
modalTriggers.forEach(function(trigger) {
|
||||||
|
trigger.addEventListener('click', function() {
|
||||||
|
var targetId = this.getAttribute('data-bs-target');
|
||||||
|
if (targetId) {
|
||||||
|
var modalElement = document.querySelector(targetId);
|
||||||
|
if (modalElement) {
|
||||||
|
var modal = new bootstrap.Modal(modalElement);
|
||||||
|
modal.show();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
{% block scripts %}{% endblock %}
|
||||||
|
{% block extra_js %}{% endblock %}
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
File diff suppressed because it is too large
Load Diff
@ -1,303 +1,303 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
"""
|
"""
|
||||||
手机连接器模块
|
手机连接器模块
|
||||||
用于接收手机传送的摄像头图像、GPS位置和设备信息
|
用于接收手机传送的摄像头图像、GPS位置和设备信息
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import cv2
|
import cv2
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import json
|
import json
|
||||||
import time
|
import time
|
||||||
import threading
|
import threading
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
import base64
|
import base64
|
||||||
import socket
|
import socket
|
||||||
import struct
|
import struct
|
||||||
from typing import Dict, List, Optional, Tuple, Callable
|
from typing import Dict, List, Optional, Tuple, Callable
|
||||||
from . import config
|
from . import config
|
||||||
|
|
||||||
class MobileDevice:
|
class MobileDevice:
|
||||||
"""移动设备信息类"""
|
"""移动设备信息类"""
|
||||||
def __init__(self, device_id: str, device_name: str):
|
def __init__(self, device_id: str, device_name: str):
|
||||||
self.device_id = device_id
|
self.device_id = device_id
|
||||||
self.device_name = device_name
|
self.device_name = device_name
|
||||||
self.last_seen = time.time()
|
self.last_seen = time.time()
|
||||||
self.is_online = True
|
self.is_online = True
|
||||||
self.current_location = None # (lat, lng, accuracy)
|
self.current_location = None # (lat, lng, accuracy)
|
||||||
self.battery_level = 100
|
self.battery_level = 100
|
||||||
self.signal_strength = 100
|
self.signal_strength = 100
|
||||||
self.camera_info = {}
|
self.camera_info = {}
|
||||||
self.connection_info = {}
|
self.connection_info = {}
|
||||||
|
|
||||||
def update_status(self, data: dict):
|
def update_status(self, data: dict):
|
||||||
"""更新设备状态"""
|
"""更新设备状态"""
|
||||||
self.last_seen = time.time()
|
self.last_seen = time.time()
|
||||||
self.is_online = True
|
self.is_online = True
|
||||||
|
|
||||||
if 'gps' in data:
|
if 'gps' in data:
|
||||||
self.current_location = (
|
self.current_location = (
|
||||||
data['gps'].get('latitude'),
|
data['gps'].get('latitude'),
|
||||||
data['gps'].get('longitude'),
|
data['gps'].get('longitude'),
|
||||||
data['gps'].get('accuracy', 0)
|
data['gps'].get('accuracy', 0)
|
||||||
)
|
)
|
||||||
|
|
||||||
if 'battery' in data:
|
if 'battery' in data:
|
||||||
self.battery_level = data['battery']
|
self.battery_level = data['battery']
|
||||||
|
|
||||||
if 'signal' in data:
|
if 'signal' in data:
|
||||||
self.signal_strength = data['signal']
|
self.signal_strength = data['signal']
|
||||||
|
|
||||||
if 'camera_info' in data:
|
if 'camera_info' in data:
|
||||||
self.camera_info = data['camera_info']
|
self.camera_info = data['camera_info']
|
||||||
|
|
||||||
def is_location_valid(self) -> bool:
|
def is_location_valid(self) -> bool:
|
||||||
"""检查GPS位置是否有效"""
|
"""检查GPS位置是否有效"""
|
||||||
if not self.current_location:
|
if not self.current_location:
|
||||||
return False
|
return False
|
||||||
lat, lng, _ = self.current_location
|
lat, lng, _ = self.current_location
|
||||||
return lat is not None and lng is not None and -90 <= lat <= 90 and -180 <= lng <= 180
|
return lat is not None and lng is not None and -90 <= lat <= 90 and -180 <= lng <= 180
|
||||||
|
|
||||||
class MobileConnector:
|
class MobileConnector:
|
||||||
"""手机连接器主类"""
|
"""手机连接器主类"""
|
||||||
|
|
||||||
def __init__(self, port: int = 8080):
|
def __init__(self, port: int = 8080):
|
||||||
self.port = port
|
self.port = port
|
||||||
self.server_socket = None
|
self.server_socket = None
|
||||||
self.is_running = False
|
self.is_running = False
|
||||||
self.devices = {} # device_id -> MobileDevice
|
self.devices = {} # device_id -> MobileDevice
|
||||||
self.frame_callbacks = [] # 帧数据回调函数列表
|
self.frame_callbacks = [] # 帧数据回调函数列表
|
||||||
self.location_callbacks = [] # 位置数据回调函数列表
|
self.location_callbacks = [] # 位置数据回调函数列表
|
||||||
self.device_callbacks = [] # 设备状态回调函数列表
|
self.device_callbacks = [] # 设备状态回调函数列表
|
||||||
self.client_threads = []
|
self.client_threads = []
|
||||||
|
|
||||||
# 统计信息
|
# 统计信息
|
||||||
self.total_frames_received = 0
|
self.total_frames_received = 0
|
||||||
self.total_data_received = 0
|
self.total_data_received = 0
|
||||||
self.start_time = time.time()
|
self.start_time = time.time()
|
||||||
|
|
||||||
def add_frame_callback(self, callback: Callable):
|
def add_frame_callback(self, callback: Callable):
|
||||||
"""添加帧数据回调函数"""
|
"""添加帧数据回调函数"""
|
||||||
self.frame_callbacks.append(callback)
|
self.frame_callbacks.append(callback)
|
||||||
|
|
||||||
def add_location_callback(self, callback: Callable):
|
def add_location_callback(self, callback: Callable):
|
||||||
"""添加位置数据回调函数"""
|
"""添加位置数据回调函数"""
|
||||||
self.location_callbacks.append(callback)
|
self.location_callbacks.append(callback)
|
||||||
|
|
||||||
def add_device_callback(self, callback: Callable):
|
def add_device_callback(self, callback: Callable):
|
||||||
"""添加设备状态回调函数"""
|
"""添加设备状态回调函数"""
|
||||||
self.device_callbacks.append(callback)
|
self.device_callbacks.append(callback)
|
||||||
|
|
||||||
def start_server(self):
|
def start_server(self):
|
||||||
"""启动服务器"""
|
"""启动服务器"""
|
||||||
try:
|
try:
|
||||||
self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||||
self.server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
self.server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||||
self.server_socket.bind(('0.0.0.0', self.port))
|
self.server_socket.bind(('0.0.0.0', self.port))
|
||||||
self.server_socket.listen(5)
|
self.server_socket.listen(5)
|
||||||
self.is_running = True
|
self.is_running = True
|
||||||
|
|
||||||
print(f"📱 手机连接服务器启动成功，端口: {self.port}")
|
print(f"📱 手机连接服务器启动成功，端口: {self.port}")
|
||||||
print(f"🌐 等待手机客户端连接...")
|
print(f"🌐 等待手机客户端连接...")
|
||||||
|
|
||||||
# 启动服务器监听线程
|
# 启动服务器监听线程
|
||||||
server_thread = threading.Thread(target=self._server_loop, daemon=True)
|
server_thread = threading.Thread(target=self._server_loop, daemon=True)
|
||||||
server_thread.start()
|
server_thread.start()
|
||||||
|
|
||||||
# 启动设备状态监控线程
|
# 启动设备状态监控线程
|
||||||
monitor_thread = threading.Thread(target=self._device_monitor, daemon=True)
|
monitor_thread = threading.Thread(target=self._device_monitor, daemon=True)
|
||||||
monitor_thread.start()
|
monitor_thread.start()
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"❌ 启动服务器失败: {e}")
|
print(f"❌ 启动服务器失败: {e}")
|
||||||
return False
|
return False
|
||||||
|
|
||||||
def stop_server(self):
|
def stop_server(self):
|
||||||
"""停止服务器"""
|
"""停止服务器"""
|
||||||
self.is_running = False
|
self.is_running = False
|
||||||
if self.server_socket:
|
if self.server_socket:
|
||||||
self.server_socket.close()
|
self.server_socket.close()
|
||||||
|
|
||||||
# 清理客户端连接
|
# 清理客户端连接
|
||||||
for thread in self.client_threads:
|
for thread in self.client_threads:
|
||||||
if thread.is_alive():
|
if thread.is_alive():
|
||||||
thread.join(timeout=1.0)
|
thread.join(timeout=1.0)
|
||||||
|
|
||||||
print("📱 手机连接服务器已停止")
|
print("📱 手机连接服务器已停止")
|
||||||
|
|
||||||
def _server_loop(self):
|
def _server_loop(self):
|
||||||
"""服务器主循环"""
|
"""服务器主循环"""
|
||||||
while self.is_running:
|
while self.is_running:
|
||||||
try:
|
try:
|
||||||
client_socket, address = self.server_socket.accept()
|
client_socket, address = self.server_socket.accept()
|
||||||
print(f"📱 新的手机客户端连接: {address}")
|
print(f"📱 新的手机客户端连接: {address}")
|
||||||
|
|
||||||
# 为每个客户端创建处理线程
|
# 为每个客户端创建处理线程
|
||||||
client_thread = threading.Thread(
|
client_thread = threading.Thread(
|
||||||
target=self._handle_client,
|
target=self._handle_client,
|
||||||
args=(client_socket, address),
|
args=(client_socket, address),
|
||||||
daemon=True
|
daemon=True
|
||||||
)
|
)
|
||||||
client_thread.start()
|
client_thread.start()
|
||||||
self.client_threads.append(client_thread)
|
self.client_threads.append(client_thread)
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
if self.is_running:
|
if self.is_running:
|
||||||
print(f"⚠️ 服务器接受连接时出错: {e}")
|
print(f"⚠️ 服务器接受连接时出错: {e}")
|
||||||
break
|
break
|
||||||
|
|
||||||
def _handle_client(self, client_socket, address):
|
def _handle_client(self, client_socket, address):
|
||||||
"""处理客户端连接"""
|
"""处理客户端连接"""
|
||||||
device_id = None
|
device_id = None
|
||||||
try:
|
try:
|
||||||
while self.is_running:
|
while self.is_running:
|
||||||
# 接收数据长度
|
# 接收数据长度
|
||||||
length_data = self._recv_all(client_socket, 4)
|
length_data = self._recv_all(client_socket, 4)
|
||||||
if not length_data:
|
if not length_data:
|
||||||
break
|
break
|
||||||
|
|
||||||
data_length = struct.unpack('!I', length_data)[0]
|
data_length = struct.unpack('!I', length_data)[0]
|
||||||
|
|
||||||
# 接收JSON数据
|
# 接收JSON数据
|
||||||
json_data = self._recv_all(client_socket, data_length)
|
json_data = self._recv_all(client_socket, data_length)
|
||||||
if not json_data:
|
if not json_data:
|
||||||
break
|
break
|
||||||
|
|
||||||
try:
|
try:
|
||||||
data = json.loads(json_data.decode('utf-8'))
|
data = json.loads(json_data.decode('utf-8'))
|
||||||
device_id = data.get('device_id')
|
device_id = data.get('device_id')
|
||||||
|
|
||||||
if device_id:
|
if device_id:
|
||||||
self._process_mobile_data(device_id, data, address)
|
self._process_mobile_data(device_id, data, address)
|
||||||
self.total_data_received += len(json_data)
|
self.total_data_received += len(json_data)
|
||||||
|
|
||||||
except json.JSONDecodeError as e:
|
except json.JSONDecodeError as e:
|
||||||
print(f"⚠️ JSON解析错误: {e}")
|
print(f"⚠️ JSON解析错误: {e}")
|
||||||
continue
|
continue
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"⚠️ 处理客户端 {address} 时出错: {e}")
|
print(f"⚠️ 处理客户端 {address} 时出错: {e}")
|
||||||
finally:
|
finally:
|
||||||
client_socket.close()
|
client_socket.close()
|
||||||
if device_id and device_id in self.devices:
|
if device_id and device_id in self.devices:
|
||||||
self.devices[device_id].is_online = False
|
self.devices[device_id].is_online = False
|
||||||
print(f"📱 设备 {device_id} 已断开连接")
|
print(f"📱 设备 {device_id} 已断开连接")
|
||||||
|
|
||||||
def _recv_all(self, socket, length):
|
def _recv_all(self, socket, length):
|
||||||
"""接收指定长度的数据"""
|
"""接收指定长度的数据"""
|
||||||
data = b''
|
data = b''
|
||||||
while len(data) < length:
|
while len(data) < length:
|
||||||
packet = socket.recv(length - len(data))
|
packet = socket.recv(length - len(data))
|
||||||
if not packet:
|
if not packet:
|
||||||
return None
|
return None
|
||||||
data += packet
|
data += packet
|
||||||
return data
|
return data
|
||||||
|
|
||||||
def _process_mobile_data(self, device_id: str, data: dict, address):
|
def _process_mobile_data(self, device_id: str, data: dict, address):
|
||||||
"""处理手机发送的数据"""
|
"""处理手机发送的数据"""
|
||||||
# 更新或创建设备信息
|
# 更新或创建设备信息
|
||||||
if device_id not in self.devices:
|
if device_id not in self.devices:
|
||||||
device_name = data.get('device_name', f'Mobile-{device_id[:8]}')
|
device_name = data.get('device_name', f'Mobile-{device_id[:8]}')
|
||||||
self.devices[device_id] = MobileDevice(device_id, device_name)
|
self.devices[device_id] = MobileDevice(device_id, device_name)
|
||||||
print(f"📱 新设备注册: {device_name} ({device_id[:8]})")
|
print(f"📱 新设备注册: {device_name} ({device_id[:8]})")
|
||||||
|
|
||||||
# 触发设备状态回调
|
# 触发设备状态回调
|
||||||
for callback in self.device_callbacks:
|
for callback in self.device_callbacks:
|
||||||
try:
|
try:
|
||||||
callback('device_connected', self.devices[device_id])
|
callback('device_connected', self.devices[device_id])
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"⚠️ 设备回调错误: {e}")
|
print(f"⚠️ 设备回调错误: {e}")
|
||||||
|
|
||||||
device = self.devices[device_id]
|
device = self.devices[device_id]
|
||||||
device.update_status(data)
|
device.update_status(data)
|
||||||
device.connection_info = {'address': address}
|
device.connection_info = {'address': address}
|
||||||
|
|
||||||
# 处理图像数据
|
# 处理图像数据
|
||||||
if 'frame' in data:
|
if 'frame' in data:
|
||||||
try:
|
try:
|
||||||
frame_data = base64.b64decode(data['frame'])
|
frame_data = base64.b64decode(data['frame'])
|
||||||
frame = cv2.imdecode(
|
frame = cv2.imdecode(
|
||||||
np.frombuffer(frame_data, np.uint8),
|
np.frombuffer(frame_data, np.uint8),
|
||||||
cv2.IMREAD_COLOR
|
cv2.IMREAD_COLOR
|
||||||
)
|
)
|
||||||
|
|
||||||
if frame is not None:
|
if frame is not None:
|
||||||
self.total_frames_received += 1
|
self.total_frames_received += 1
|
||||||
|
|
||||||
# 触发帧数据回调
|
# 触发帧数据回调
|
||||||
for callback in self.frame_callbacks:
|
for callback in self.frame_callbacks:
|
||||||
try:
|
try:
|
||||||
callback(device_id, frame, device)
|
callback(device_id, frame, device)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"⚠️ 帧回调错误: {e}")
|
print(f"⚠️ 帧回调错误: {e}")
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"⚠️ 图像数据处理错误: {e}")
|
print(f"⚠️ 图像数据处理错误: {e}")
|
||||||
|
|
||||||
# 处理GPS位置数据
|
# 处理GPS位置数据
|
||||||
if 'gps' in data and device.is_location_valid():
|
if 'gps' in data and device.is_location_valid():
|
||||||
for callback in self.location_callbacks:
|
for callback in self.location_callbacks:
|
||||||
try:
|
try:
|
||||||
callback(device_id, device.current_location, device)
|
callback(device_id, device.current_location, device)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"⚠️ 位置回调错误: {e}")
|
print(f"⚠️ 位置回调错误: {e}")
|
||||||
|
|
||||||
def _device_monitor(self):
|
def _device_monitor(self):
|
||||||
"""设备状态监控"""
|
"""设备状态监控"""
|
||||||
while self.is_running:
|
while self.is_running:
|
||||||
try:
|
try:
|
||||||
current_time = time.time()
|
current_time = time.time()
|
||||||
offline_devices = []
|
offline_devices = []
|
||||||
|
|
||||||
for device_id, device in self.devices.items():
|
for device_id, device in self.devices.items():
|
||||||
# 超过30秒没有数据认为离线
|
# 超过30秒没有数据认为离线
|
||||||
if current_time - device.last_seen > 30:
|
if current_time - device.last_seen > 30:
|
||||||
if device.is_online:
|
if device.is_online:
|
||||||
device.is_online = False
|
device.is_online = False
|
||||||
offline_devices.append(device_id)
|
offline_devices.append(device_id)
|
||||||
|
|
||||||
# 通知离线设备
|
# 通知离线设备
|
||||||
for device_id in offline_devices:
|
for device_id in offline_devices:
|
||||||
print(f"📱 设备 {device_id[:8]} 已离线")
|
print(f"📱 设备 {device_id[:8]} 已离线")
|
||||||
for callback in self.device_callbacks:
|
for callback in self.device_callbacks:
|
||||||
try:
|
try:
|
||||||
callback('device_disconnected', self.devices[device_id])
|
callback('device_disconnected', self.devices[device_id])
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"⚠️ 设备回调错误: {e}")
|
print(f"⚠️ 设备回调错误: {e}")
|
||||||
|
|
||||||
time.sleep(5) # 每5秒检查一次
|
time.sleep(5) # 每5秒检查一次
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"⚠️ 设备监控错误: {e}")
|
print(f"⚠️ 设备监控错误: {e}")
|
||||||
time.sleep(5)
|
time.sleep(5)
|
||||||
|
|
||||||
def get_online_devices(self) -> List[MobileDevice]:
|
def get_online_devices(self) -> List[MobileDevice]:
|
||||||
"""获取在线设备列表"""
|
"""获取在线设备列表"""
|
||||||
return [device for device in self.devices.values() if device.is_online]
|
return [device for device in self.devices.values() if device.is_online]
|
||||||
|
|
||||||
def get_device_by_id(self, device_id: str) -> Optional[MobileDevice]:
|
def get_device_by_id(self, device_id: str) -> Optional[MobileDevice]:
|
||||||
"""根据ID获取设备"""
|
"""根据ID获取设备"""
|
||||||
return self.devices.get(device_id)
|
return self.devices.get(device_id)
|
||||||
|
|
||||||
def get_statistics(self) -> dict:
|
def get_statistics(self) -> dict:
|
||||||
"""获取连接统计信息"""
|
"""获取连接统计信息"""
|
||||||
online_count = len(self.get_online_devices())
|
online_count = len(self.get_online_devices())
|
||||||
total_count = len(self.devices)
|
total_count = len(self.devices)
|
||||||
uptime = time.time() - self.start_time
|
uptime = time.time() - self.start_time
|
||||||
|
|
||||||
return {
|
return {
|
||||||
'online_devices': online_count,
|
'online_devices': online_count,
|
||||||
'total_devices': total_count,
|
'total_devices': total_count,
|
||||||
'frames_received': self.total_frames_received,
|
'frames_received': self.total_frames_received,
|
||||||
'data_received_mb': self.total_data_received / (1024 * 1024),
|
'data_received_mb': self.total_data_received / (1024 * 1024),
|
||||||
'uptime_seconds': uptime,
|
'uptime_seconds': uptime,
|
||||||
'avg_frames_per_second': self.total_frames_received / uptime if uptime > 0 else 0
|
'avg_frames_per_second': self.total_frames_received / uptime if uptime > 0 else 0
|
||||||
}
|
}
|
||||||
|
|
||||||
def send_command_to_device(self, device_id: str, command: dict):
|
def send_command_to_device(self, device_id: str, command: dict):
|
||||||
"""向指定设备发送命令（预留接口）"""
|
"""向指定设备发送命令（预留接口）"""
|
||||||
# TODO: 实现向手机发送控制命令的功能
|
# TODO: 实现向手机发送控制命令的功能
|
||||||
pass
|
pass
|
||||||
File diff suppressed because it is too large
Load Diff
@ -1,392 +1,392 @@
|
|||||||
<!DOCTYPE html>
|
<!DOCTYPE html>
|
||||||
<html>
|
<html>
|
||||||
|
|
||||||
<head>
|
<head>
|
||||||
<meta charset="UTF-8">
|
<meta charset="UTF-8">
|
||||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
<title>设备选择器测试</title>
|
<title>设备选择器测试</title>
|
||||||
<style>
|
<style>
|
||||||
body {
|
body {
|
||||||
background: linear-gradient(135deg, #1e3c72, #2a5298);
|
background: linear-gradient(135deg, #1e3c72, #2a5298);
|
||||||
color: white;
|
color: white;
|
||||||
font-family: 'Microsoft YaHei', sans-serif;
|
font-family: 'Microsoft YaHei', sans-serif;
|
||||||
margin: 0;
|
margin: 0;
|
||||||
padding: 20px;
|
padding: 20px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.container {
|
.container {
|
||||||
max-width: 500px;
|
max-width: 500px;
|
||||||
margin: 0 auto;
|
margin: 0 auto;
|
||||||
padding: 20px;
|
padding: 20px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.video-container {
|
.video-container {
|
||||||
background: rgba(0, 0, 0, 0.4);
|
background: rgba(0, 0, 0, 0.4);
|
||||||
border-radius: 15px;
|
border-radius: 15px;
|
||||||
margin: 20px 0;
|
margin: 20px 0;
|
||||||
overflow: hidden;
|
overflow: hidden;
|
||||||
}
|
}
|
||||||
|
|
||||||
.video-header {
|
.video-header {
|
||||||
display: flex;
|
display: flex;
|
||||||
justify-content: space-between;
|
justify-content: space-between;
|
||||||
align-items: center;
|
align-items: center;
|
||||||
padding: 10px;
|
padding: 10px;
|
||||||
background: rgba(0, 0, 0, 0.3);
|
background: rgba(0, 0, 0, 0.3);
|
||||||
border-radius: 8px 8px 0 0;
|
border-radius: 8px 8px 0 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
.device-select-btn {
|
.device-select-btn {
|
||||||
background: #2196F3;
|
background: #2196F3;
|
||||||
color: white;
|
color: white;
|
||||||
border: none;
|
border: none;
|
||||||
padding: 8px 12px;
|
padding: 8px 12px;
|
||||||
border-radius: 4px;
|
border-radius: 4px;
|
||||||
font-size: 12px;
|
font-size: 12px;
|
||||||
cursor: pointer;
|
cursor: pointer;
|
||||||
}
|
}
|
||||||
|
|
||||||
.device-selector {
|
.device-selector {
|
||||||
position: fixed;
|
position: fixed;
|
||||||
top: 0;
|
top: 0;
|
||||||
left: 0;
|
left: 0;
|
||||||
width: 100%;
|
width: 100%;
|
||||||
height: 100%;
|
height: 100%;
|
||||||
background: rgba(0, 0, 0, 0.8);
|
background: rgba(0, 0, 0, 0.8);
|
||||||
z-index: 1000;
|
z-index: 1000;
|
||||||
display: flex;
|
display: flex;
|
||||||
align-items: center;
|
align-items: center;
|
||||||
justify-content: center;
|
justify-content: center;
|
||||||
}
|
}
|
||||||
|
|
||||||
.device-selector-content {
|
.device-selector-content {
|
||||||
background: rgba(0, 20, 40, 0.95);
|
background: rgba(0, 20, 40, 0.95);
|
||||||
border: 2px solid #00aaff;
|
border: 2px solid #00aaff;
|
||||||
border-radius: 15px;
|
border-radius: 15px;
|
||||||
padding: 20px;
|
padding: 20px;
|
||||||
max-width: 90%;
|
max-width: 90%;
|
||||||
max-height: 80%;
|
max-height: 80%;
|
||||||
overflow-y: auto;
|
overflow-y: auto;
|
||||||
backdrop-filter: blur(10px);
|
backdrop-filter: blur(10px);
|
||||||
}
|
}
|
||||||
|
|
||||||
.device-selector h3 {
|
.device-selector h3 {
|
||||||
margin: 0 0 15px 0;
|
margin: 0 0 15px 0;
|
||||||
color: #00aaff;
|
color: #00aaff;
|
||||||
text-align: center;
|
text-align: center;
|
||||||
}
|
}
|
||||||
|
|
||||||
.device-list {
|
.device-list {
|
||||||
margin: 15px 0;
|
margin: 15px 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
.device-item {
|
.device-item {
|
||||||
background: rgba(255, 255, 255, 0.1);
|
background: rgba(255, 255, 255, 0.1);
|
||||||
border-radius: 8px;
|
border-radius: 8px;
|
||||||
padding: 15px;
|
padding: 15px;
|
||||||
margin: 10px 0;
|
margin: 10px 0;
|
||||||
cursor: pointer;
|
cursor: pointer;
|
||||||
transition: all 0.3s ease;
|
transition: all 0.3s ease;
|
||||||
border: 2px solid transparent;
|
border: 2px solid transparent;
|
||||||
}
|
}
|
||||||
|
|
||||||
.device-item:hover {
|
.device-item:hover {
|
||||||
background: rgba(255, 255, 255, 0.2);
|
background: rgba(255, 255, 255, 0.2);
|
||||||
border-color: #00aaff;
|
border-color: #00aaff;
|
||||||
}
|
}
|
||||||
|
|
||||||
.device-item.selected {
|
.device-item.selected {
|
||||||
background: rgba(0, 170, 255, 0.3);
|
background: rgba(0, 170, 255, 0.3);
|
||||||
border-color: #00aaff;
|
border-color: #00aaff;
|
||||||
}
|
}
|
||||||
|
|
||||||
.device-name {
|
.device-name {
|
||||||
font-weight: bold;
|
font-weight: bold;
|
||||||
color: #00aaff;
|
color: #00aaff;
|
||||||
margin-bottom: 5px;
|
margin-bottom: 5px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.device-id {
|
.device-id {
|
||||||
font-size: 12px;
|
font-size: 12px;
|
||||||
color: #ccc;
|
color: #ccc;
|
||||||
font-family: monospace;
|
font-family: monospace;
|
||||||
}
|
}
|
||||||
|
|
||||||
.device-kind {
|
.device-kind {
|
||||||
display: inline-block;
|
display: inline-block;
|
||||||
background: #4CAF50;
|
background: #4CAF50;
|
||||||
color: white;
|
color: white;
|
||||||
padding: 2px 6px;
|
padding: 2px 6px;
|
||||||
border-radius: 3px;
|
border-radius: 3px;
|
||||||
font-size: 10px;
|
font-size: 10px;
|
||||||
margin-top: 5px;
|
margin-top: 5px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.device-selector-buttons {
|
.device-selector-buttons {
|
||||||
display: flex;
|
display: flex;
|
||||||
justify-content: space-between;
|
justify-content: space-between;
|
||||||
margin-top: 20px;
|
margin-top: 20px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.btn {
|
.btn {
|
||||||
padding: 10px 20px;
|
padding: 10px 20px;
|
||||||
border: none;
|
border: none;
|
||||||
border-radius: 8px;
|
border-radius: 8px;
|
||||||
cursor: pointer;
|
cursor: pointer;
|
||||||
font-weight: bold;
|
font-weight: bold;
|
||||||
margin: 0 5px;
|
margin: 0 5px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.btn-primary {
|
.btn-primary {
|
||||||
background: #007bff;
|
background: #007bff;
|
||||||
color: white;
|
color: white;
|
||||||
}
|
}
|
||||||
|
|
||||||
.btn-secondary {
|
.btn-secondary {
|
||||||
background: #6c757d;
|
background: #6c757d;
|
||||||
color: white;
|
color: white;
|
||||||
}
|
}
|
||||||
|
|
||||||
.btn:disabled {
|
.btn:disabled {
|
||||||
background: #666;
|
background: #666;
|
||||||
cursor: not-allowed;
|
cursor: not-allowed;
|
||||||
}
|
}
|
||||||
|
|
||||||
.loading {
|
.loading {
|
||||||
text-align: center;
|
text-align: center;
|
||||||
color: #ccc;
|
color: #ccc;
|
||||||
padding: 20px;
|
padding: 20px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.log {
|
.log {
|
||||||
background: rgba(0, 0, 0, 0.3);
|
background: rgba(0, 0, 0, 0.3);
|
||||||
border-radius: 8px;
|
border-radius: 8px;
|
||||||
padding: 10px;
|
padding: 10px;
|
||||||
margin: 20px 0;
|
margin: 20px 0;
|
||||||
max-height: 200px;
|
max-height: 200px;
|
||||||
overflow-y: auto;
|
overflow-y: auto;
|
||||||
font-family: monospace;
|
font-family: monospace;
|
||||||
font-size: 12px;
|
font-size: 12px;
|
||||||
}
|
}
|
||||||
</style>
|
</style>
|
||||||
</head>
|
</head>
|
||||||
|
|
||||||
<body>
|
<body>
|
||||||
<div class="container">
|
<div class="container">
|
||||||
<h1>🧪 设备选择器测试</h1>
|
<h1>🧪 设备选择器测试</h1>
|
||||||
|
|
||||||
<div class="video-container">
|
<div class="video-container">
|
||||||
<div class="video-header">
|
<div class="video-header">
|
||||||
<span>📹 视频设备</span>
|
<span>📹 视频设备</span>
|
||||||
<button class="device-select-btn" onclick="showDeviceSelector()">📷 选择设备</button>
|
<button class="device-select-btn" onclick="showDeviceSelector()">📷 选择设备</button>
|
||||||
</div>
|
</div>
|
||||||
<div id="videoPlaceholder" style="text-align: center; padding: 40px; color: #ccc;">
|
<div id="videoPlaceholder" style="text-align: center; padding: 40px; color: #ccc;">
|
||||||
点击"选择设备"开始使用摄像头
|
点击"选择设备"开始使用摄像头
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<!-- 设备选择弹窗 -->
|
<!-- 设备选择弹窗 -->
|
||||||
<div class="device-selector" id="deviceSelector" style="display: none;">
|
<div class="device-selector" id="deviceSelector" style="display: none;">
|
||||||
<div class="device-selector-content">
|
<div class="device-selector-content">
|
||||||
<h3>📷 选择视频设备</h3>
|
<h3>📷 选择视频设备</h3>
|
||||||
|
|
||||||
<!-- 本地设备列表 -->
|
<!-- 本地设备列表 -->
|
||||||
<div>
|
<div>
|
||||||
<h4 style="color: #4CAF50; margin: 15px 0 10px 0;">📱 本地设备</h4>
|
<h4 style="color: #4CAF50; margin: 15px 0 10px 0;">📱 本地设备</h4>
|
||||||
<div class="device-list" id="localDeviceList">
|
<div class="device-list" id="localDeviceList">
|
||||||
<div class="loading">正在扫描本地设备...</div>
|
<div class="loading">正在扫描本地设备...</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div class="device-selector-buttons">
|
<div class="device-selector-buttons">
|
||||||
<button class="btn btn-secondary" onclick="hideDeviceSelector()">❌ 取消</button>
|
<button class="btn btn-secondary" onclick="hideDeviceSelector()">❌ 取消</button>
|
||||||
<button class="btn btn-primary" onclick="refreshDevices()">🔄 刷新设备</button>
|
<button class="btn btn-primary" onclick="refreshDevices()">🔄 刷新设备</button>
|
||||||
<button class="btn" onclick="useSelectedDevice()" id="useDeviceBtn" disabled>✅ 使用选择的设备</button>
|
<button class="btn" onclick="useSelectedDevice()" id="useDeviceBtn" disabled>✅ 使用选择的设备</button>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div class="log" id="logPanel">
|
<div class="log" id="logPanel">
|
||||||
<div>系统初始化中...</div>
|
<div>系统初始化中...</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<script>
|
<script>
|
||||||
let availableDevices = [];
|
let availableDevices = [];
|
||||||
let selectedDeviceId = null;
|
let selectedDeviceId = null;
|
||||||
let selectedDeviceInfo = null;
|
let selectedDeviceInfo = null;
|
||||||
|
|
||||||
// 日志函数
|
// 日志函数
|
||||||
function log(message, type = 'info') {
|
function log(message, type = 'info') {
|
||||||
const logPanel = document.getElementById('logPanel');
|
const logPanel = document.getElementById('logPanel');
|
||||||
const timestamp = new Date().toLocaleTimeString();
|
const timestamp = new Date().toLocaleTimeString();
|
||||||
const entry = document.createElement('div');
|
const entry = document.createElement('div');
|
||||||
entry.style.color = type === 'error' ? '#ff6b6b' : type === 'success' ? '#51cf66' : '#74c0fc';
|
entry.style.color = type === 'error' ? '#ff6b6b' : type === 'success' ? '#51cf66' : '#74c0fc';
|
||||||
entry.textContent = `${timestamp} - ${message}`;
|
entry.textContent = `${timestamp} - ${message}`;
|
||||||
logPanel.appendChild(entry);
|
logPanel.appendChild(entry);
|
||||||
logPanel.scrollTop = logPanel.scrollHeight;
|
logPanel.scrollTop = logPanel.scrollHeight;
|
||||||
}
|
}
|
||||||
|
|
||||||
// 扫描设备
|
// 扫描设备
|
||||||
async function scanDevices() {
|
async function scanDevices() {
|
||||||
log('正在扫描可用视频设备...', 'info');
|
log('正在扫描可用视频设备...', 'info');
|
||||||
try {
|
try {
|
||||||
if (!navigator.mediaDevices || !navigator.mediaDevices.enumerateDevices) {
|
if (!navigator.mediaDevices || !navigator.mediaDevices.enumerateDevices) {
|
||||||
throw new Error('浏览器不支持设备枚举功能');
|
throw new Error('浏览器不支持设备枚举功能');
|
||||||
}
|
}
|
||||||
|
|
||||||
const devices = await navigator.mediaDevices.enumerateDevices();
|
const devices = await navigator.mediaDevices.enumerateDevices();
|
||||||
availableDevices = devices.filter(device => device.kind === 'videoinput');
|
availableDevices = devices.filter(device => device.kind === 'videoinput');
|
||||||
|
|
||||||
log(`发现 ${availableDevices.length} 个视频设备`, 'success');
|
log(`发现 ${availableDevices.length} 个视频设备`, 'success');
|
||||||
|
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
log(`设备扫描失败: ${error.message}`, 'error');
|
log(`设备扫描失败: ${error.message}`, 'error');
|
||||||
availableDevices = [];
|
availableDevices = [];
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// 显示设备选择器
|
// 显示设备选择器
|
||||||
async function showDeviceSelector() {
|
async function showDeviceSelector() {
|
||||||
log('打开设备选择器', 'info');
|
log('打开设备选择器', 'info');
|
||||||
const selector = document.getElementById('deviceSelector');
|
const selector = document.getElementById('deviceSelector');
|
||||||
selector.style.display = 'flex';
|
selector.style.display = 'flex';
|
||||||
|
|
||||||
await scanDevices();
|
await scanDevices();
|
||||||
updateDeviceList();
|
updateDeviceList();
|
||||||
}
|
}
|
||||||
|
|
||||||
// 隐藏设备选择器
|
// 隐藏设备选择器
|
||||||
function hideDeviceSelector() {
|
function hideDeviceSelector() {
|
||||||
document.getElementById('deviceSelector').style.display = 'none';
|
document.getElementById('deviceSelector').style.display = 'none';
|
||||||
clearDeviceSelection();
|
clearDeviceSelection();
|
||||||
}
|
}
|
||||||
|
|
||||||
// 刷新设备
|
// 刷新设备
|
||||||
async function refreshDevices() {
|
async function refreshDevices() {
|
||||||
document.getElementById('localDeviceList').innerHTML = '<div class="loading">正在扫描设备...</div>';
|
document.getElementById('localDeviceList').innerHTML = '<div class="loading">正在扫描设备...</div>';
|
||||||
|
|
||||||
await scanDevices();
|
await scanDevices();
|
||||||
updateDeviceList();
|
updateDeviceList();
|
||||||
}
|
}
|
||||||
|
|
||||||
// 更新设备列表
|
// 更新设备列表
|
||||||
function updateDeviceList() {
|
function updateDeviceList() {
|
||||||
const localList = document.getElementById('localDeviceList');
|
const localList = document.getElementById('localDeviceList');
|
||||||
|
|
||||||
if (availableDevices.length === 0) {
|
if (availableDevices.length === 0) {
|
||||||
localList.innerHTML = '<div style="color: #ff6b6b; text-align: center; padding: 20px;">未发现本地摄像头设备<br><small>请确保已连接摄像头并允许浏览器访问</small></div>';
|
localList.innerHTML = '<div style="color: #ff6b6b; text-align: center; padding: 20px;">未发现本地摄像头设备<br><small>请确保已连接摄像头并允许浏览器访问</small></div>';
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
localList.innerHTML = '';
|
localList.innerHTML = '';
|
||||||
availableDevices.forEach((device, index) => {
|
availableDevices.forEach((device, index) => {
|
||||||
const deviceItem = document.createElement('div');
|
const deviceItem = document.createElement('div');
|
||||||
deviceItem.className = 'device-item';
|
deviceItem.className = 'device-item';
|
||||||
deviceItem.onclick = () => selectDevice(device.deviceId, {
|
deviceItem.onclick = () => selectDevice(device.deviceId, {
|
||||||
label: device.label || `摄像头 ${index + 1}`,
|
label: device.label || `摄像头 ${index + 1}`,
|
||||||
kind: device.kind,
|
kind: device.kind,
|
||||||
isRemote: false
|
isRemote: false
|
||||||
});
|
});
|
||||||
|
|
||||||
const deviceName = device.label || `摄像头 ${index + 1}`;
|
const deviceName = device.label || `摄像头 ${index + 1}`;
|
||||||
const isFrontCamera = deviceName.toLowerCase().includes('front') || deviceName.toLowerCase().includes('前');
|
const isFrontCamera = deviceName.toLowerCase().includes('front') || deviceName.toLowerCase().includes('前');
|
||||||
const isBackCamera = deviceName.toLowerCase().includes('back') || deviceName.toLowerCase().includes('后');
|
const isBackCamera = deviceName.toLowerCase().includes('back') || deviceName.toLowerCase().includes('后');
|
||||||
|
|
||||||
let cameraIcon = '📷';
|
let cameraIcon = '📷';
|
||||||
if (isFrontCamera) cameraIcon = '🤳';
|
if (isFrontCamera) cameraIcon = '🤳';
|
||||||
else if (isBackCamera) cameraIcon = '📹';
|
else if (isBackCamera) cameraIcon = '📹';
|
||||||
|
|
||||||
deviceItem.innerHTML = `
|
deviceItem.innerHTML = `
|
||||||
<div class="device-name">${cameraIcon} ${deviceName}</div>
|
<div class="device-name">${cameraIcon} ${deviceName}</div>
|
||||||
<div class="device-id">${device.deviceId}</div>
|
<div class="device-id">${device.deviceId}</div>
|
||||||
<div class="device-kind">本地设备</div>
|
<div class="device-kind">本地设备</div>
|
||||||
`;
|
`;
|
||||||
localList.appendChild(deviceItem);
|
localList.appendChild(deviceItem);
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
// 选择设备
|
// 选择设备
|
||||||
function selectDevice(deviceId, deviceInfo) {
|
function selectDevice(deviceId, deviceInfo) {
|
||||||
// 清除之前的选择
|
// 清除之前的选择
|
||||||
document.querySelectorAll('.device-item').forEach(item => {
|
document.querySelectorAll('.device-item').forEach(item => {
|
||||||
item.classList.remove('selected');
|
item.classList.remove('selected');
|
||||||
});
|
});
|
||||||
|
|
||||||
// 选择当前设备
|
// 选择当前设备
|
||||||
event.currentTarget.classList.add('selected');
|
event.currentTarget.classList.add('selected');
|
||||||
selectedDeviceId = deviceId;
|
selectedDeviceId = deviceId;
|
||||||
selectedDeviceInfo = deviceInfo;
|
selectedDeviceInfo = deviceInfo;
|
||||||
|
|
||||||
// 启用使用按钮
|
// 启用使用按钮
|
||||||
document.getElementById('useDeviceBtn').disabled = false;
|
document.getElementById('useDeviceBtn').disabled = false;
|
||||||
|
|
||||||
log(`已选择设备: ${deviceInfo.label}`, 'info');
|
log(`已选择设备: ${deviceInfo.label}`, 'info');
|
||||||
}
|
}
|
||||||
|
|
||||||
// 清除设备选择
|
// 清除设备选择
|
||||||
function clearDeviceSelection() {
|
function clearDeviceSelection() {
|
||||||
selectedDeviceId = null;
|
selectedDeviceId = null;
|
||||||
selectedDeviceInfo = null;
|
selectedDeviceInfo = null;
|
||||||
document.getElementById('useDeviceBtn').disabled = true;
|
document.getElementById('useDeviceBtn').disabled = true;
|
||||||
document.querySelectorAll('.device-item').forEach(item => {
|
document.querySelectorAll('.device-item').forEach(item => {
|
||||||
item.classList.remove('selected');
|
item.classList.remove('selected');
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
// 使用选择的设备
|
// 使用选择的设备
|
||||||
async function useSelectedDevice() {
|
async function useSelectedDevice() {
|
||||||
if (!selectedDeviceId || !selectedDeviceInfo) {
|
if (!selectedDeviceId || !selectedDeviceInfo) {
|
||||||
log('请先选择一个设备', 'error');
|
log('请先选择一个设备', 'error');
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
log(`正在启动设备: ${selectedDeviceInfo.label}`, 'info');
|
log(`正在启动设备: ${selectedDeviceInfo.label}`, 'info');
|
||||||
|
|
||||||
const constraints = {
|
const constraints = {
|
||||||
video: {
|
video: {
|
||||||
deviceId: { exact: selectedDeviceId },
|
deviceId: { exact: selectedDeviceId },
|
||||||
width: { ideal: 640 },
|
width: { ideal: 640 },
|
||||||
height: { ideal: 480 }
|
height: { ideal: 480 }
|
||||||
},
|
},
|
||||||
audio: false
|
audio: false
|
||||||
};
|
};
|
||||||
|
|
||||||
const stream = await navigator.mediaDevices.getUserMedia(constraints);
|
const stream = await navigator.mediaDevices.getUserMedia(constraints);
|
||||||
|
|
||||||
// 创建视频元素显示
|
// 创建视频元素显示
|
||||||
const placeholder = document.getElementById('videoPlaceholder');
|
const placeholder = document.getElementById('videoPlaceholder');
|
||||||
placeholder.innerHTML = `
|
placeholder.innerHTML = `
|
||||||
<video autoplay muted playsinline style="width: 100%; height: auto;"></video>
|
<video autoplay muted playsinline style="width: 100%; height: auto;"></video>
|
||||||
<div style="font-size: 12px; color: #ccc; margin-top: 10px;">
|
<div style="font-size: 12px; color: #ccc; margin-top: 10px;">
|
||||||
正在使用: ${selectedDeviceInfo.label}
|
正在使用: ${selectedDeviceInfo.label}
|
||||||
</div>
|
</div>
|
||||||
`;
|
`;
|
||||||
|
|
||||||
const videoElement = placeholder.querySelector('video');
|
const videoElement = placeholder.querySelector('video');
|
||||||
videoElement.srcObject = stream;
|
videoElement.srcObject = stream;
|
||||||
|
|
||||||
hideDeviceSelector();
|
hideDeviceSelector();
|
||||||
log(`设备启动成功: ${selectedDeviceInfo.label}`, 'success');
|
log(`设备启动成功: ${selectedDeviceInfo.label}`, 'success');
|
||||||
|
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
let errorMsg = error.message;
|
let errorMsg = error.message;
|
||||||
if (error.name === 'NotAllowedError') {
|
if (error.name === 'NotAllowedError') {
|
||||||
errorMsg = '设备权限被拒绝，请允许访问摄像头';
|
errorMsg = '设备权限被拒绝，请允许访问摄像头';
|
||||||
} else if (error.name === 'NotFoundError') {
|
} else if (error.name === 'NotFoundError') {
|
||||||
errorMsg = '设备未找到或已被占用';
|
errorMsg = '设备未找到或已被占用';
|
||||||
}
|
}
|
||||||
|
|
||||||
log(`设备启动失败: ${errorMsg}`, 'error');
|
log(`设备启动失败: ${errorMsg}`, 'error');
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// 初始化
|
// 初始化
|
||||||
window.addEventListener('load', () => {
|
window.addEventListener('load', () => {
|
||||||
log('设备选择器测试页面已加载', 'success');
|
log('设备选择器测试页面已加载', 'success');
|
||||||
});
|
});
|
||||||
</script>
|
</script>
|
||||||
</body>
|
</body>
|
||||||
|
|
||||||
</html>
|
</html>
|
||||||
@ -0,0 +1,470 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="zh-CN">
|
||||||
|
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>🛤️ 无人机轨迹模拟测试</title>
|
||||||
|
<style>
|
||||||
|
* {
|
||||||
|
margin: 0;
|
||||||
|
padding: 0;
|
||||||
|
box-sizing: border-box;
|
||||||
|
}
|
||||||
|
|
||||||
|
body {
|
||||||
|
font-family: 'Microsoft YaHei', sans-serif;
|
||||||
|
background: linear-gradient(135deg, #1e3c72, #2a5298);
|
||||||
|
color: white;
|
||||||
|
min-height: 100vh;
|
||||||
|
padding: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.container {
|
||||||
|
max-width: 800px;
|
||||||
|
margin: 0 auto;
|
||||||
|
background: rgba(255, 255, 255, 0.1);
|
||||||
|
border-radius: 20px;
|
||||||
|
padding: 30px;
|
||||||
|
backdrop-filter: blur(10px);
|
||||||
|
box-shadow: 0 20px 40px rgba(0, 0, 0, 0.3);
|
||||||
|
}
|
||||||
|
|
||||||
|
.header {
|
||||||
|
text-align: center;
|
||||||
|
margin-bottom: 30px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.header h1 {
|
||||||
|
font-size: 2.5em;
|
||||||
|
margin-bottom: 10px;
|
||||||
|
text-shadow: 2px 2px 4px rgba(0, 0, 0, 0.5);
|
||||||
|
}
|
||||||
|
|
||||||
|
.header p {
|
||||||
|
font-size: 1.1em;
|
||||||
|
opacity: 0.8;
|
||||||
|
}
|
||||||
|
|
||||||
|
.control-panel {
|
||||||
|
background: rgba(255, 255, 255, 0.1);
|
||||||
|
border-radius: 15px;
|
||||||
|
padding: 25px;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.control-group {
|
||||||
|
margin-bottom: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.control-group h3 {
|
||||||
|
color: #00aaff;
|
||||||
|
margin-bottom: 15px;
|
||||||
|
font-size: 1.2em;
|
||||||
|
border-bottom: 2px solid rgba(0, 170, 255, 0.3);
|
||||||
|
padding-bottom: 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.button-grid {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(auto-fit, minmax(120px, 1fr));
|
||||||
|
gap: 15px;
|
||||||
|
margin-bottom: 15px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn {
|
||||||
|
padding: 12px 20px;
|
||||||
|
border: none;
|
||||||
|
border-radius: 8px;
|
||||||
|
cursor: pointer;
|
||||||
|
font-size: 14px;
|
||||||
|
font-weight: bold;
|
||||||
|
color: white;
|
||||||
|
transition: all 0.3s ease;
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn:hover {
|
||||||
|
transform: translateY(-2px);
|
||||||
|
box-shadow: 0 5px 15px rgba(0, 0, 0, 0.3);
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-circle {
|
||||||
|
background: linear-gradient(135deg, #9C27B0, #7B1FA2);
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-line {
|
||||||
|
background: linear-gradient(135deg, #FF9800, #F57C00);
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-random {
|
||||||
|
background: linear-gradient(135deg, #607D8B, #455A64);
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-stop {
|
||||||
|
background: linear-gradient(135deg, #F44336, #D32F2F);
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-clear {
|
||||||
|
background: linear-gradient(135deg, #FF5722, #D32F2F);
|
||||||
|
}
|
||||||
|
|
||||||
|
.input-group {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 10px;
|
||||||
|
margin-bottom: 10px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.input-group label {
|
||||||
|
min-width: 100px;
|
||||||
|
font-weight: bold;
|
||||||
|
}
|
||||||
|
|
||||||
|
.input-group input,
|
||||||
|
.input-group select {
|
||||||
|
flex: 1;
|
||||||
|
padding: 8px 12px;
|
||||||
|
border: none;
|
||||||
|
border-radius: 6px;
|
||||||
|
background: rgba(255, 255, 255, 0.2);
|
||||||
|
color: white;
|
||||||
|
font-size: 14px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.input-group input::placeholder {
|
||||||
|
color: rgba(255, 255, 255, 0.6);
|
||||||
|
}
|
||||||
|
|
||||||
|
.status-panel {
|
||||||
|
background: rgba(0, 0, 0, 0.3);
|
||||||
|
border-radius: 10px;
|
||||||
|
padding: 20px;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.status-item {
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
align-items: center;
|
||||||
|
padding: 8px 0;
|
||||||
|
border-bottom: 1px solid rgba(255, 255, 255, 0.1);
|
||||||
|
}
|
||||||
|
|
||||||
|
.status-item:last-child {
|
||||||
|
border-bottom: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.status-value {
|
||||||
|
font-weight: bold;
|
||||||
|
color: #4CAF50;
|
||||||
|
}
|
||||||
|
|
||||||
|
.alert {
|
||||||
|
padding: 12px 15px;
|
||||||
|
border-radius: 8px;
|
||||||
|
margin-bottom: 15px;
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.alert.success {
|
||||||
|
background: rgba(76, 175, 80, 0.2);
|
||||||
|
border-left: 4px solid #4CAF50;
|
||||||
|
}
|
||||||
|
|
||||||
|
.alert.error {
|
||||||
|
background: rgba(244, 67, 54, 0.2);
|
||||||
|
border-left: 4px solid #F44336;
|
||||||
|
}
|
||||||
|
|
||||||
|
.alert.info {
|
||||||
|
background: rgba(33, 150, 243, 0.2);
|
||||||
|
border-left: 4px solid #2196F3;
|
||||||
|
}
|
||||||
|
|
||||||
|
.drone-list {
|
||||||
|
max-height: 200px;
|
||||||
|
overflow-y: auto;
|
||||||
|
background: rgba(0, 0, 0, 0.2);
|
||||||
|
border-radius: 8px;
|
||||||
|
padding: 10px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.drone-item {
|
||||||
|
padding: 8px 10px;
|
||||||
|
margin-bottom: 5px;
|
||||||
|
background: rgba(255, 255, 255, 0.1);
|
||||||
|
border-radius: 6px;
|
||||||
|
font-size: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.link-section {
|
||||||
|
text-align: center;
|
||||||
|
margin-top: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.link-section a {
|
||||||
|
color: #00aaff;
|
||||||
|
text-decoration: none;
|
||||||
|
font-weight: bold;
|
||||||
|
padding: 10px 20px;
|
||||||
|
background: rgba(0, 170, 255, 0.2);
|
||||||
|
border-radius: 8px;
|
||||||
|
transition: all 0.3s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.link-section a:hover {
|
||||||
|
background: rgba(0, 170, 255, 0.4);
|
||||||
|
transform: translateY(-2px);
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
|
||||||
|
<body>
|
||||||
|
<div class="container">
|
||||||
|
<div class="header">
|
||||||
|
<h1>🛤️ 无人机轨迹模拟测试</h1>
|
||||||
|
<p>测试无人机轨迹记录和可视化功能</p>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div id="alertContainer"></div>
|
||||||
|
|
||||||
|
<!-- 快速启动面板 -->
|
||||||
|
<div class="control-panel">
|
||||||
|
<div class="control-group">
|
||||||
|
<h3>🚀 快速启动模拟</h3>
|
||||||
|
<div class="button-grid">
|
||||||
|
<button class="btn btn-circle" onclick="startQuickSimulation('circle')">
|
||||||
|
🔄 圆形轨迹
|
||||||
|
</button>
|
||||||
|
<button class="btn btn-line" onclick="startQuickSimulation('line')">
|
||||||
|
↔️ 直线轨迹
|
||||||
|
</button>
|
||||||
|
<button class="btn btn-random" onclick="startQuickSimulation('random')">
|
||||||
|
🎲 随机轨迹
|
||||||
|
</button>
|
||||||
|
<button class="btn btn-stop" onclick="stopSimulation()">
|
||||||
|
⏹️ 停止模拟
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- 高级配置面板 -->
|
||||||
|
<div class="control-panel">
|
||||||
|
<div class="control-group">
|
||||||
|
<h3>⚙️ 高级配置</h3>
|
||||||
|
<div class="input-group">
|
||||||
|
<label>轨迹类型:</label>
|
||||||
|
<select id="simulationType">
|
||||||
|
<option value="circle">圆形轨迹</option>
|
||||||
|
<option value="line">直线往返</option>
|
||||||
|
<option value="random">随机游走</option>
|
||||||
|
</select>
|
||||||
|
</div>
|
||||||
|
<div class="input-group">
|
||||||
|
<label>无人机数量:</label>
|
||||||
|
<input type="number" id="droneCount" value="2" min="1" max="5" placeholder="1-5架">
|
||||||
|
</div>
|
||||||
|
<div class="input-group">
|
||||||
|
<label>更新间隔:</label>
|
||||||
|
<input type="number" id="updateSpeed" value="3" min="1" max="10" placeholder="秒">
|
||||||
|
</div>
|
||||||
|
<div class="button-grid">
|
||||||
|
<button class="btn btn-circle" onclick="startCustomSimulation()">
|
||||||
|
🛠️ 启动自定义模拟
|
||||||
|
</button>
|
||||||
|
<button class="btn btn-clear" onclick="clearAllTrajectories()">
|
||||||
|
🗑️ 清除所有轨迹
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- 状态显示面板 -->
|
||||||
|
<div class="status-panel">
|
||||||
|
<h3 style="color: #00aaff; margin-bottom: 15px;">📊 模拟状态</h3>
|
||||||
|
<div class="status-item">
|
||||||
|
<span>模拟状态:</span>
|
||||||
|
<span class="status-value" id="simulationStatus">未启动</span>
|
||||||
|
</div>
|
||||||
|
<div class="status-item">
|
||||||
|
<span>轨迹类型:</span>
|
||||||
|
<span class="status-value" id="simulationType">无</span>
|
||||||
|
</div>
|
||||||
|
<div class="status-item">
|
||||||
|
<span>无人机数量:</span>
|
||||||
|
<span class="status-value" id="droneCountStatus">0</span>
|
||||||
|
</div>
|
||||||
|
<div class="status-item">
|
||||||
|
<span>轨迹记录:</span>
|
||||||
|
<span class="status-value" id="trajectoryRecordingStatus">开启</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- 无人机列表 -->
|
||||||
|
<div class="control-panel">
|
||||||
|
<div class="control-group">
|
||||||
|
<h3>🚁 模拟无人机列表</h3>
|
||||||
|
<div class="drone-list" id="droneList">
|
||||||
|
<div style="text-align: center; color: #999;">暂无模拟无人机</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- 链接到主系统 -->
|
||||||
|
<div class="link-section">
|
||||||
|
<a href="/" target="_blank">🎯 打开主系统查看轨迹</a>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
// 显示提示信息
|
||||||
|
function showAlert(message, type = 'info') {
|
||||||
|
const alertContainer = document.getElementById('alertContainer');
|
||||||
|
const alert = document.createElement('div');
|
||||||
|
alert.className = `alert ${type}`;
|
||||||
|
alert.textContent = message;
|
||||||
|
|
||||||
|
alertContainer.appendChild(alert);
|
||||||
|
alert.style.display = 'block';
|
||||||
|
|
||||||
|
setTimeout(() => {
|
||||||
|
alert.style.display = 'none';
|
||||||
|
alertContainer.removeChild(alert);
|
||||||
|
}, 5000);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 快速启动模拟
|
||||||
|
async function startQuickSimulation(type) {
|
||||||
|
await startSimulation(type, 2, 3);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 自定义启动模拟
|
||||||
|
async function startCustomSimulation() {
|
||||||
|
const type = document.getElementById('simulationType').value;
|
||||||
|
const droneCount = parseInt(document.getElementById('droneCount').value) || 2;
|
||||||
|
const speed = parseInt(document.getElementById('updateSpeed').value) || 3;
|
||||||
|
|
||||||
|
await startSimulation(type, droneCount, speed);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 启动轨迹模拟
|
||||||
|
async function startSimulation(type, droneCount = 2, speed = 3) {
|
||||||
|
try {
|
||||||
|
showAlert(`🚀 正在启动${droneCount}架无人机的${type}轨迹模拟...`, 'info');
|
||||||
|
|
||||||
|
const response = await fetch('/api/test/start_simulation', {
|
||||||
|
method: 'POST',
|
||||||
|
headers: { 'Content-Type': 'application/json' },
|
||||||
|
body: JSON.stringify({
|
||||||
|
type: type,
|
||||||
|
drone_count: droneCount,
|
||||||
|
speed: speed
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
const result = await response.json();
|
||||||
|
if (result.status === 'success') {
|
||||||
|
showAlert(`✅ ${result.message}`, 'success');
|
||||||
|
updateStatus();
|
||||||
|
} else {
|
||||||
|
showAlert(`❌ 启动模拟失败: ${result.message}`, 'error');
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
showAlert(`❌ 启动模拟失败: ${error.message}`, 'error');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 停止轨迹模拟
|
||||||
|
async function stopSimulation() {
|
||||||
|
try {
|
||||||
|
showAlert('🛑 正在停止轨迹模拟...', 'info');
|
||||||
|
|
||||||
|
const response = await fetch('/api/test/stop_simulation', {
|
||||||
|
method: 'POST'
|
||||||
|
});
|
||||||
|
|
||||||
|
const result = await response.json();
|
||||||
|
if (result.status === 'success') {
|
||||||
|
showAlert('✅ 轨迹模拟已停止', 'success');
|
||||||
|
updateStatus();
|
||||||
|
} else {
|
||||||
|
showAlert(`❌ 停止模拟失败: ${result.message}`, 'error');
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
showAlert(`❌ 停止模拟失败: ${error.message}`, 'error');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 清除所有轨迹
|
||||||
|
async function clearAllTrajectories() {
|
||||||
|
try {
|
||||||
|
// 这个功能需要在主页面中执行
|
||||||
|
showAlert('🗑️ 请在主系统页面中点击"清除"按钮来清除轨迹', 'info');
|
||||||
|
} catch (error) {
|
||||||
|
showAlert(`❌ 清除轨迹失败: ${error.message}`, 'error');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新状态显示
|
||||||
|
async function updateStatus() {
|
||||||
|
try {
|
||||||
|
const response = await fetch('/api/test/simulation_status');
|
||||||
|
const result = await response.json();
|
||||||
|
|
||||||
|
if (result.status === 'success') {
|
||||||
|
// 更新状态显示
|
||||||
|
document.getElementById('simulationStatus').textContent =
|
||||||
|
result.simulation_active ? '运行中' : '未启动';
|
||||||
|
document.getElementById('simulationStatus').style.color =
|
||||||
|
result.simulation_active ? '#4CAF50' : '#F44336';
|
||||||
|
|
||||||
|
document.getElementById('simulationType').textContent =
|
||||||
|
result.simulation_type || '无';
|
||||||
|
document.getElementById('droneCountStatus').textContent =
|
||||||
|
result.simulated_drones.length || '0';
|
||||||
|
|
||||||
|
// 更新无人机列表
|
||||||
|
updateDroneList(result.simulated_drones);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('获取状态失败:', error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新无人机列表
|
||||||
|
function updateDroneList(drones) {
|
||||||
|
const droneList = document.getElementById('droneList');
|
||||||
|
|
||||||
|
if (!drones || drones.length === 0) {
|
||||||
|
droneList.innerHTML = '<div style="text-align: center; color: #999;">暂无模拟无人机</div>';
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const html = drones.map((drone, index) => `
|
||||||
|
<div class="drone-item">
|
||||||
|
🚁 ${drone.name || `无人机-${index + 1}`}
|
||||||
|
<br>
|
||||||
|
<span style="color: #999;">
|
||||||
|
ID: ${drone.device_id.substring(0, 12)}...
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
`).join('');
|
||||||
|
|
||||||
|
droneList.innerHTML = html;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 页面加载完成后的初始化
|
||||||
|
window.addEventListener('load', function () {
|
||||||
|
updateStatus();
|
||||||
|
|
||||||
|
// 定期更新状态
|
||||||
|
setInterval(updateStatus, 3000);
|
||||||
|
|
||||||
|
showAlert('🎯 轨迹模拟测试页面已加载', 'success');
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
|
||||||
|
</html>
|
||||||
Loading…
Reference in new issue