forked from chiyu1468/AirTrapMine
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
117 lines
4.0 KiB
Python
117 lines
4.0 KiB
Python
1 month ago
|
#!/usr/bin/env python3
|
||
|
|
from PyQt6.QtWidgets import QTableWidget, QTableWidgetItem, QHeaderView, QLabel
|
||
|
|
from PyQt6.QtCore import Qt
|
||
|
|
|
||
|
|
class OverviewTable(QTableWidget):
|
||
|
|
"""總覽表格，顯示所有無人機的狀態資訊"""
|
||
|
|
|
||
|
|
# 默認的資訊類型和映射
|
||
|
|
DEFAULT_INFO_TYPES = ["模式", "ARM", "電壓", "經度", "緯度", "高度", "位置", "速度", "地速", "航向",
|
||
|
|
"空速", "油門", "HUD ALT", "爬升率", "Roll", "Pitch", "Yaw", "丟包", "延遲"]
|
||
|
|
|
||
|
|
DEFAULT_INFO_TYPE_MAP = {
|
||
|
|
"mode": 0,
|
||
|
|
"armed": 1,
|
||
|
|
"battery": 2,
|
||
|
|
"longitude": 3,
|
||
|
|
"latitude": 4,
|
||
|
|
"altitude": 5,
|
||
|
|
"local": 6,
|
||
|
|
"velocity": 7,
|
||
|
|
"groundspeed": 8,
|
||
|
|
"heading": 9,
|
||
|
|
"airspeed": 10,
|
||
|
|
"throttle": 11,
|
||
|
|
"hud_alt": 12,
|
||
|
|
"climb": 13,
|
||
|
|
"roll": 14,
|
||
|
|
"pitch": 15,
|
||
|
|
"yaw": 16,
|
||
|
|
"loss_rate": 17,
|
||
|
|
"ping": 18
|
||
|
|
}
|
||
|
|
|
||
|
|
def __init__(self, info_types=None, info_type_map=None, parent=None):
|
||
|
|
super().__init__(parent)
|
||
|
|
|
||
|
|
# 使用提供的或默認的資訊類型
|
||
|
|
self.info_types = info_types if info_types is not None else self.DEFAULT_INFO_TYPES
|
||
|
|
self.info_type_map = info_type_map if info_type_map is not None else self.DEFAULT_INFO_TYPE_MAP
|
||
|
|
self.drones = {} # 存儲無人機面板的引用
|
||
|
|
|
||
|
|
# 初始化表格
|
||
|
|
self.setColumnCount(1)
|
||
|
|
self.setRowCount(len(self.info_types))
|
||
|
|
self.setHorizontalHeaderLabels(["資訊"])
|
||
|
|
header = self.horizontalHeader()
|
||
|
|
header.setSectionResizeMode(0, QHeaderView.ResizeMode.ResizeToContents)
|
||
|
|
self.verticalHeader().setVisible(False)
|
||
|
|
|
||
|
|
# 設置第一列的資訊類型
|
||
|
|
for i, txt in enumerate(self.info_types):
|
||
|
|
item = QTableWidgetItem(txt)
|
||
|
|
item.setFlags(Qt.ItemFlag.ItemIsEnabled)
|
||
|
|
item.setTextAlignment(Qt.AlignmentFlag.AlignCenter)
|
||
|
|
self.setItem(i, 0, item)
|
||
|
|
|
||
|
|
def set_drones(self, drones):
|
||
|
|
"""設置無人機面板字典的引用"""
|
||
|
|
self.drones = drones
|
||
|
|
|
||
|
|
def update_table(self, drone_id=None, field=None, value=None):
|
||
|
|
"""更新總覽表格
|
||
|
|
|
||
|
|
Args:
|
||
|
|
drone_id: 無人機 ID
|
||
|
|
field: 欄位名稱 (如 'mode', 'altitude' 等)
|
||
|
|
value: 要更新的值
|
||
|
|
"""
|
||
|
|
# 更新特定儲存格
|
||
|
|
if drone_id and field and value:
|
||
|
|
if drone_id not in self.drones:
|
||
|
|
return
|
||
|
|
|
||
|
|
col = 1 + list(self.drones.keys()).index(drone_id)
|
||
|
|
row = self.info_type_map.get(field, -1)
|
||
|
|
|
||
|
|
if row == -1:
|
||
|
|
return # 無效的欄位
|
||
|
|
|
||
|
|
item = self.item(row, col)
|
||
|
|
if not item:
|
||
|
|
item = QTableWidgetItem()
|
||
|
|
self.setItem(row, col, item)
|
||
|
|
|
||
|
|
item.setText(value)
|
||
|
|
item.setTextAlignment(Qt.AlignmentFlag.AlignCenter)
|
||
|
|
|
||
|
|
# 如果沒有指定更新，刷新整個表格
|
||
|
|
if drone_id is None:
|
||
|
|
self.refresh_all()
|
||
|
|
|
||
|
|
def refresh_all(self):
|
||
|
|
"""刷新整個表格"""
|
||
|
|
cols = 1 + len(self.drones)
|
||
|
|
self.setColumnCount(cols)
|
||
|
|
headers = ["資訊"] + list(self.drones.keys())
|
||
|
|
self.setHorizontalHeaderLabels(headers)
|
||
|
|
|
||
|
|
for col, did in enumerate(self.drones, start=1):
|
||
|
|
panel = self.drones[did]
|
||
|
|
for field, row in self.info_type_map.items():
|
||
|
|
lbl = panel.findChild(QLabel, f"{did}_{field}")
|
||
|
|
val = lbl.text() if lbl else "--"
|
||
|
|
val_item = QTableWidgetItem(val)
|
||
|
|
val_item.setTextAlignment(Qt.AlignmentFlag.AlignCenter)
|
||
|
|
self.setItem(row, col, val_item)
|
||
|
|
|
||
|
|
def add_drone_column(self, drone_id):
|
||
|
|
"""當新增無人機時，添加一列"""
|
||
|
|
if drone_id in self.drones:
|
||
|
|
self.refresh_all()
|
||
|
|
|
||
|
|
def remove_drone_column(self, drone_id):
|
||
|
|
"""當移除無人機時，刷新表格"""
|
||
|
|
if drone_id in self.drones:
|
||
|
|
self.refresh_all()
|