712 lines
25 KiB
Python
712 lines
25 KiB
Python
#!/usr/bin/env python3
|
||
from PyQt6.QtWidgets import (QWidget, QVBoxLayout, QHBoxLayout, QLabel, QCheckBox, QApplication)
|
||
from PyQt6.QtCore import pyqtSignal
|
||
from PyQt6.QtGui import QPainter, QPen, QColor, QFont, QPolygonF
|
||
from PyQt6.QtCore import QPointF, Qt
|
||
import math
|
||
|
||
|
||
def _get_font_scale():
|
||
app = QApplication.instance()
|
||
if app is None:
|
||
return 1.0
|
||
scale = app.property("font_scale")
|
||
try:
|
||
return float(scale) if scale is not None else 1.0
|
||
except (TypeError, ValueError):
|
||
return 1.0
|
||
|
||
|
||
def _scale_stylesheet_font_sizes(stylesheet, scale):
|
||
if not stylesheet or 'font-size' not in stylesheet:
|
||
return stylesheet
|
||
|
||
import re
|
||
|
||
def repl(match):
|
||
size = float(match.group(1))
|
||
unit = match.group(2)
|
||
scaled = max(1.0, size * scale)
|
||
text = f"{scaled:.2f}".rstrip('0').rstrip('.')
|
||
return f"font-size: {text}{unit}"
|
||
|
||
return re.sub(r'font-size\s*:\s*([0-9]+(?:\.[0-9]+)?)\s*(px|pt)', repl, stylesheet)
|
||
|
||
|
||
def _set_scaled_stylesheet(widget, stylesheet):
|
||
widget._base_stylesheet = stylesheet
|
||
scaled = _scale_stylesheet_font_sizes(stylesheet, _get_font_scale())
|
||
widget._applied_stylesheet = scaled
|
||
widget.setStyleSheet(scaled)
|
||
|
||
|
||
def _reapply_scaled_stylesheet(widget):
|
||
current_stylesheet = widget.styleSheet()
|
||
base_stylesheet = getattr(widget, '_base_stylesheet', None)
|
||
applied_stylesheet = getattr(widget, '_applied_stylesheet', None)
|
||
|
||
if current_stylesheet != applied_stylesheet:
|
||
base_stylesheet = current_stylesheet
|
||
widget._base_stylesheet = base_stylesheet
|
||
|
||
if base_stylesheet is not None:
|
||
scaled = _scale_stylesheet_font_sizes(base_stylesheet, _get_font_scale())
|
||
widget._applied_stylesheet = scaled
|
||
if current_stylesheet != scaled:
|
||
widget.setStyleSheet(scaled)
|
||
|
||
|
||
def _apply_scaled_font(widget):
|
||
base_font = getattr(widget, '_base_font_for_scale', None)
|
||
if base_font is None:
|
||
app = QApplication.instance()
|
||
app_base_font = app.property("base_app_font") if app else None
|
||
base_font = QFont(app_base_font) if app_base_font is not None else QFont(widget.font())
|
||
widget._base_font_for_scale = QFont(base_font)
|
||
|
||
scaled_font = QFont(base_font)
|
||
scale = _get_font_scale()
|
||
if base_font.pointSizeF() > 0:
|
||
scaled_font.setPointSizeF(max(1.0, base_font.pointSizeF() * scale))
|
||
elif base_font.pointSize() > 0:
|
||
scaled_font.setPointSize(max(1, int(round(base_font.pointSize() * scale))))
|
||
widget.setFont(scaled_font)
|
||
|
||
class DronePanel(QWidget):
|
||
"""單個無人機面板類別"""
|
||
|
||
# 定義信號
|
||
mode_change_requested = pyqtSignal(str) # drone_id
|
||
arm_requested = pyqtSignal(str) # drone_id
|
||
takeoff_requested = pyqtSignal(str) # drone_id
|
||
setpoint_requested = pyqtSignal(str) # drone_id
|
||
selection_changed = pyqtSignal(str, int) # drone_id, state
|
||
|
||
def __init__(self, drone_id, parent=None):
|
||
super().__init__(parent)
|
||
self.drone_id = drone_id
|
||
|
||
# 提取資訊 (格式: s{socket_seq}_{system_id}, 如 s0_11, s1_12)
|
||
parts = drone_id.split('_')
|
||
if len(parts) >= 2:
|
||
self.socket_seq = parts[0][1:] # socket 序號 (移除 's' 前綴)
|
||
self.system_id = parts[1] # system ID
|
||
self.display_id = f"ID:{self.system_id}" # 顯示為 ID:11, ID:12
|
||
else:
|
||
self.socket_seq = "?"
|
||
self.system_id = "?"
|
||
self.display_id = drone_id
|
||
|
||
self.attitude_indicator = None
|
||
self._init_ui()
|
||
self.apply_font_scale()
|
||
|
||
def _init_ui(self):
|
||
"""初始化UI"""
|
||
self.setObjectName(f"panel_{self.drone_id}")
|
||
self.setFixedHeight(160)
|
||
_set_scaled_stylesheet(self, """
|
||
background-color: #2A2A2A;
|
||
border-radius: 8px;
|
||
""")
|
||
|
||
# 主佈局
|
||
main_layout = QHBoxLayout(self)
|
||
main_layout.setContentsMargins(8, 8, 8, 8)
|
||
main_layout.setSpacing(0)
|
||
|
||
# 創建內容容器（包含 info 和 control）
|
||
content_widget = QWidget()
|
||
_set_scaled_stylesheet(content_widget, "background-color: #333; border-radius: 6px;")
|
||
content_layout = QHBoxLayout(content_widget)
|
||
content_layout.setContentsMargins(8, 8, 8, 8)
|
||
content_layout.setSpacing(8)
|
||
|
||
# 左側資訊區域
|
||
info_widget = self._create_info_section()
|
||
|
||
# 右側態度指示器
|
||
attitude_widget = self._create_attitude_indicator()
|
||
|
||
# 將 info 和 attitude 加入內容容器
|
||
content_layout.addWidget(info_widget, 1)
|
||
content_layout.addWidget(attitude_widget, 0)
|
||
|
||
# 將內容容器加入主佈局
|
||
main_layout.addWidget(content_widget)
|
||
|
||
def _create_info_section(self):
|
||
"""創建資訊顯示區域"""
|
||
info_widget = QWidget()
|
||
info_layout = QVBoxLayout(info_widget)
|
||
info_layout.setContentsMargins(0, 0, 0, 0)
|
||
info_layout.setSpacing(4)
|
||
|
||
# 頂部標題欄
|
||
header = QWidget()
|
||
header_layout = QHBoxLayout(header)
|
||
header_layout.setContentsMargins(0, 0, 0, 0)
|
||
|
||
# 勾選框
|
||
self.checkbox = QCheckBox()
|
||
self.checkbox.setObjectName(f"{self.drone_id}_checkbox")
|
||
_set_scaled_stylesheet(self.checkbox, """
|
||
QCheckBox {
|
||
color: #DDD;
|
||
}
|
||
QCheckBox::indicator {
|
||
width: 16px;
|
||
height: 16px;
|
||
border: 2px solid #888888;
|
||
border-radius: 3px;
|
||
background: transparent;
|
||
}
|
||
QCheckBox::indicator:checked {
|
||
background-color: #7FFFD4;
|
||
border: 2px solid #888888;
|
||
}
|
||
""")
|
||
self.checkbox.stateChanged.connect(
|
||
lambda state: self.selection_changed.emit(self.drone_id, state)
|
||
)
|
||
|
||
# ID 顯示
|
||
self.id_label = QLabel(self.display_id)
|
||
_set_scaled_stylesheet(self.id_label, """
|
||
font-weight: bold;
|
||
font-size: 14px;
|
||
color: #7FFFD4;
|
||
min-width: 80px;
|
||
""")
|
||
|
||
header_layout.addWidget(self.checkbox)
|
||
header_layout.addWidget(self.id_label)
|
||
header_layout.addStretch()
|
||
|
||
info_layout.addWidget(header)
|
||
|
||
# 第一行：狀態 (模式 + ARM狀態)
|
||
status_row = self._create_status_row()
|
||
info_layout.addWidget(status_row)
|
||
|
||
# 第二行：電池（拆成百分比與電壓兩欄）
|
||
battery_row = self._create_battery_row()
|
||
info_layout.addWidget(battery_row)
|
||
|
||
# 第三行：高度
|
||
altitude_row = self._create_altitude_row()
|
||
info_layout.addWidget(altitude_row)
|
||
|
||
# 第四行：GNSS 定位狀態
|
||
gnss_row = self._create_gnss_row()
|
||
info_layout.addWidget(gnss_row)
|
||
|
||
# 第五行：航向 + 速度
|
||
nav_row = self._create_nav_row()
|
||
info_layout.addWidget(nav_row)
|
||
|
||
return info_widget
|
||
|
||
def _create_attitude_indicator(self):
|
||
"""創建態度指示器（ADI 人工地平儀）"""
|
||
self.attitude_indicator = AttitudeIndicator(self.drone_id)
|
||
self.attitude_indicator.setFixedSize(90, 100)
|
||
return self.attitude_indicator
|
||
|
||
def _create_status_row(self):
|
||
"""創建狀態行"""
|
||
status_row = QWidget()
|
||
status_layout = QHBoxLayout(status_row)
|
||
status_layout.setContentsMargins(0, 0, 0, 0)
|
||
|
||
status_title = QLabel("狀態：")
|
||
_set_scaled_stylesheet(status_title, "color: #888; min-width: 50px;")
|
||
|
||
self.mode_label = QLabel("--")
|
||
self.mode_label.setObjectName(f"{self.drone_id}_mode")
|
||
_set_scaled_stylesheet(self.mode_label, "color: #DDD;")
|
||
|
||
self.armed_label = QLabel("--")
|
||
self.armed_label.setObjectName(f"{self.drone_id}_armed")
|
||
_set_scaled_stylesheet(self.armed_label, "color: #DDD;")
|
||
|
||
status_layout.addWidget(status_title)
|
||
status_layout.addWidget(self.mode_label)
|
||
status_layout.addWidget(self.armed_label)
|
||
status_layout.addStretch()
|
||
|
||
return status_row
|
||
|
||
def _create_connection_row(self):
|
||
"""創建連接資訊行 (Socket Seq + 連接方式)"""
|
||
connection_row = QWidget()
|
||
connection_layout = QHBoxLayout(connection_row)
|
||
connection_layout.setContentsMargins(0, 0, 0, 0)
|
||
|
||
connection_title = QLabel("Socket：")
|
||
_set_scaled_stylesheet(connection_title, "color: #888; min-width: 50px;")
|
||
|
||
# 根據解析的 drone_id 資訊設定初始值
|
||
self.socket_seq_label = QLabel(self.socket_seq)
|
||
self.socket_seq_label.setObjectName(f"{self.drone_id}_socket_seq")
|
||
_set_scaled_stylesheet(self.socket_seq_label, "color: #DDD;")
|
||
|
||
connection_sep = QLabel(" - ")
|
||
_set_scaled_stylesheet(connection_sep, "color: #DDD;")
|
||
|
||
# 設定連接方式顯示
|
||
connection_type_map = {
|
||
'r': 'ROS2',
|
||
'u': 'UDP',
|
||
's': 'Serial',
|
||
'w': 'WS'
|
||
}
|
||
connection_type = connection_type_map.get(self.type_prefix, 'Unknown')
|
||
|
||
self.connection_type_label = QLabel(connection_type)
|
||
self.connection_type_label.setObjectName(f"{self.drone_id}_connection_type")
|
||
_set_scaled_stylesheet(self.connection_type_label, "color: #DDD;")
|
||
|
||
connection_layout.addWidget(connection_title)
|
||
connection_layout.addWidget(self.socket_seq_label)
|
||
connection_layout.addWidget(connection_sep)
|
||
connection_layout.addWidget(self.connection_type_label)
|
||
connection_layout.addStretch()
|
||
|
||
return connection_row
|
||
|
||
def _create_battery_row(self):
|
||
"""創建電池行"""
|
||
battery_row = QWidget()
|
||
battery_layout = QHBoxLayout(battery_row)
|
||
battery_layout.setContentsMargins(0, 0, 0, 0)
|
||
# 顯示百分比
|
||
battery_title = QLabel("電池：")
|
||
_set_scaled_stylesheet(battery_title, "color: #888; min-width: 50px;")
|
||
|
||
self.battery_pct_label = QLabel("--")
|
||
self.battery_pct_label.setObjectName(f"{self.drone_id}_battery_pct")
|
||
_set_scaled_stylesheet(self.battery_pct_label, "color: #DDD;")
|
||
|
||
# 分隔符
|
||
separator1 = QLabel(" - ")
|
||
separator1.setObjectName(f"{self.drone_id}_battery_sep1")
|
||
_set_scaled_stylesheet(separator1, "color: #DDD;")
|
||
|
||
# 顯示電壓
|
||
self.battery_vol_label = QLabel("--")
|
||
self.battery_vol_label.setObjectName(f"{self.drone_id}_battery_vol")
|
||
_set_scaled_stylesheet(self.battery_vol_label, "color: #DDD;")
|
||
|
||
# 分隔符
|
||
separator2 = QLabel(" - ")
|
||
separator2.setObjectName(f"{self.drone_id}_battery_sep2")
|
||
_set_scaled_stylesheet(separator2, "color: #DDD;")
|
||
|
||
# 顯示電池節數 (S count)
|
||
self.battery_cells_label = QLabel("--")
|
||
self.battery_cells_label.setObjectName(f"{self.drone_id}_battery_cells")
|
||
_set_scaled_stylesheet(self.battery_cells_label, "color: #DDD;")
|
||
|
||
battery_layout.addWidget(battery_title)
|
||
battery_layout.addWidget(self.battery_pct_label)
|
||
battery_layout.addWidget(separator1)
|
||
battery_layout.addWidget(self.battery_vol_label)
|
||
battery_layout.addWidget(separator2)
|
||
battery_layout.addWidget(self.battery_cells_label)
|
||
battery_layout.addStretch()
|
||
|
||
return battery_row
|
||
|
||
def _create_altitude_row(self):
|
||
"""創建高度和速度行"""
|
||
altitude_row = QWidget()
|
||
altitude_layout = QHBoxLayout(altitude_row)
|
||
altitude_layout.setContentsMargins(0, 0, 0, 0)
|
||
|
||
altitude_title = QLabel("高度：")
|
||
_set_scaled_stylesheet(altitude_title, "color: #888; min-width: 50px;")
|
||
|
||
self.altitude_label = QLabel("--")
|
||
self.altitude_label.setObjectName(f"{self.drone_id}_altitude")
|
||
_set_scaled_stylesheet(self.altitude_label, "color: #DDD;")
|
||
|
||
speed_title = QLabel("速度：")
|
||
_set_scaled_stylesheet(speed_title, "color: #888; margin-left: 10px;")
|
||
|
||
self.speed_label = QLabel("--")
|
||
self.speed_label.setObjectName(f"{self.drone_id}_speed")
|
||
_set_scaled_stylesheet(self.speed_label, "color: #DDD;")
|
||
|
||
altitude_layout.addWidget(altitude_title)
|
||
altitude_layout.addWidget(self.altitude_label)
|
||
altitude_layout.addWidget(speed_title)
|
||
altitude_layout.addWidget(self.speed_label)
|
||
altitude_layout.addStretch()
|
||
|
||
return altitude_row
|
||
|
||
def _create_gnss_row(self):
|
||
"""創建 GNSS 定位狀態行。"""
|
||
gnss_row = QWidget()
|
||
gnss_layout = QHBoxLayout(gnss_row)
|
||
gnss_layout.setContentsMargins(0, 0, 0, 0)
|
||
|
||
fix_title = QLabel("定位：")
|
||
_set_scaled_stylesheet(fix_title, "color: #888; min-width: 50px;")
|
||
|
||
self.fix_type_label = QLabel("--")
|
||
self.fix_type_label.setObjectName(f"{self.drone_id}_fix_type_label")
|
||
_set_scaled_stylesheet(self.fix_type_label, "color: #DDD;")
|
||
|
||
error_title = QLabel("誤差(H/V)：")
|
||
_set_scaled_stylesheet(error_title, "color: #888; margin-left: 8px;")
|
||
|
||
self.gnss_error_label = QLabel("--")
|
||
self.gnss_error_label.setObjectName(f"{self.drone_id}_gnss_error")
|
||
_set_scaled_stylesheet(self.gnss_error_label, "color: #DDD;")
|
||
|
||
gnss_layout.addWidget(fix_title)
|
||
gnss_layout.addWidget(self.fix_type_label)
|
||
gnss_layout.addWidget(error_title)
|
||
gnss_layout.addWidget(self.gnss_error_label)
|
||
gnss_layout.addStretch()
|
||
|
||
return gnss_row
|
||
|
||
def _create_position_row(self):
|
||
"""位置行已移除（位置座標不再顯示於面板）。"""
|
||
return QWidget()
|
||
|
||
def _create_nav_row(self):
|
||
"""創建導航行（已移除，不再顯示）"""
|
||
return QWidget()
|
||
|
||
def update_field(self, field, text, color=None):
|
||
"""更新指定欄位的值"""
|
||
label = self.findChild(QLabel, f"{self.drone_id}_{field}")
|
||
if label and label.text() != text:
|
||
label.setText(text)
|
||
if color:
|
||
label.setStyleSheet(f"color: {color};")
|
||
|
||
def set_connection_info(self, socket_seq, connection_type):
|
||
"""設定連接資訊（Socket Seq 和連接方式）
|
||
connection_type: 'UDP' | 'Serial' | 'WS'
|
||
"""
|
||
self.socket_seq_label.setText(str(socket_seq))
|
||
|
||
# 顯示友善的連接方式
|
||
type_display = {
|
||
'UDP': 'UDP',
|
||
'Serial': 'Serial',
|
||
'WS': 'WS'
|
||
}.get(connection_type, connection_type)
|
||
|
||
self.connection_type_label.setText(type_display)
|
||
|
||
|
||
def update_attitude(self, heading, roll, pitch):
|
||
"""更新態度指示器"""
|
||
if self.attitude_indicator:
|
||
self.attitude_indicator.update_attitude(heading, roll, pitch)
|
||
|
||
def get_checkbox(self):
|
||
"""獲取勾選框"""
|
||
return self.checkbox
|
||
|
||
def set_checked(self, checked):
|
||
"""設置勾選狀態"""
|
||
self.checkbox.setChecked(checked)
|
||
|
||
def is_checked(self):
|
||
"""獲取勾選狀態"""
|
||
return self.checkbox.isChecked()
|
||
|
||
def apply_font_scale(self):
|
||
"""重新套用目前字體倍率到 panel 內所有元件。"""
|
||
_apply_scaled_font(self)
|
||
_reapply_scaled_stylesheet(self)
|
||
for child in self.findChildren(QWidget):
|
||
_apply_scaled_font(child)
|
||
_reapply_scaled_stylesheet(child)
|
||
if self.attitude_indicator:
|
||
self.attitude_indicator.update()
|
||
|
||
class SocketGroupPanel(QWidget):
|
||
# 定義信號
|
||
group_selection_changed = pyqtSignal(str, int) # socket_id, state
|
||
|
||
def __init__(self, socket_id, color='#AAAAAA', socket_type=None, parent=None):
|
||
super().__init__(parent)
|
||
self.socket_id = socket_id
|
||
self.color = color
|
||
self.socket_type = socket_type
|
||
self._init_ui()
|
||
self.apply_font_scale()
|
||
|
||
def _init_ui(self):
|
||
"""初始化UI"""
|
||
self.setObjectName(f"socket_group_{self.socket_id}")
|
||
_set_scaled_stylesheet(self, """
|
||
background-color: #1E1E1E;
|
||
border-radius: 12px;
|
||
""")
|
||
|
||
layout = QVBoxLayout(self)
|
||
layout.setContentsMargins(10, 10, 10, 10)
|
||
layout.setSpacing(6)
|
||
|
||
# Socket 分組標題行 - 包含勾選框
|
||
title_row = QWidget()
|
||
title_layout = QHBoxLayout(title_row)
|
||
title_layout.setContentsMargins(0, 0, 0, 0)
|
||
|
||
# 分組勾選框
|
||
self.group_checkbox = QCheckBox()
|
||
self.group_checkbox.setObjectName(f"socket_{self.socket_id}_checkbox")
|
||
_set_scaled_stylesheet(self.group_checkbox, f"""
|
||
QCheckBox {{ color: #DDD; }}
|
||
QCheckBox::indicator {{
|
||
width: 14px;
|
||
height: 14px;
|
||
border: 2px solid #888888;
|
||
border-radius: 3px;
|
||
background: transparent;
|
||
}}
|
||
QCheckBox::indicator:checked {{
|
||
background-color: {self.color};
|
||
border: 2px solid #888888;
|
||
}}
|
||
QCheckBox::indicator:indeterminate {{
|
||
background-color: #666;
|
||
border: 2px solid #888888;
|
||
}}
|
||
""")
|
||
self.group_checkbox.stateChanged.connect(
|
||
lambda state: self.group_selection_changed.emit(self.socket_id, state)
|
||
)
|
||
|
||
# Socket 分組標題
|
||
if self.socket_type:
|
||
title_text = f"{self.socket_type} {self.socket_id}"
|
||
else:
|
||
title_text = f"Socket {self.socket_id}"
|
||
self.title_label = QLabel(title_text)
|
||
_set_scaled_stylesheet(self.title_label, f"""
|
||
font-weight: bold;
|
||
font-size: 16px;
|
||
color: {self.color};
|
||
margin-bottom: 8px;
|
||
padding: 4px 8px;
|
||
border-radius: 6px;
|
||
""")
|
||
|
||
title_layout.addWidget(self.group_checkbox)
|
||
title_layout.addWidget(self.title_label)
|
||
title_layout.addStretch()
|
||
|
||
layout.addWidget(title_row)
|
||
|
||
# 創建子容器用於放置該 socket 下的所有無人機面板
|
||
self.drones_container = QWidget()
|
||
self.drones_layout = QVBoxLayout(self.drones_container)
|
||
self.drones_layout.setContentsMargins(0, 0, 0, 0)
|
||
self.drones_layout.setSpacing(4)
|
||
|
||
layout.addWidget(self.drones_container)
|
||
|
||
def add_drone_panel(self, panel):
|
||
"""添加無人機面板到分組"""
|
||
self.drones_layout.addWidget(panel)
|
||
|
||
def clear_drones(self):
|
||
"""清空所有無人機面板"""
|
||
while self.drones_layout.count():
|
||
item = self.drones_layout.takeAt(0)
|
||
if item.widget():
|
||
item.widget().setParent(None)
|
||
|
||
def get_checkbox(self):
|
||
"""獲取分組勾選框"""
|
||
return self.group_checkbox
|
||
|
||
def set_checked(self, checked):
|
||
"""設置分組勾選狀態"""
|
||
self.group_checkbox.setChecked(checked)
|
||
|
||
def set_socket_type(self, conn_type):
|
||
"""設置 socket 類型並更新標題"""
|
||
self.title_label.setText(f"{conn_type} {self.socket_id}")
|
||
|
||
def set_check_state(self, state):
|
||
"""設置分組勾選狀態（支持半選）"""
|
||
self.group_checkbox.setCheckState(state)
|
||
|
||
def apply_font_scale(self):
|
||
"""重新套用目前字體倍率到 socket 分組面板。"""
|
||
_apply_scaled_font(self)
|
||
_reapply_scaled_stylesheet(self)
|
||
for child in self.findChildren(QWidget):
|
||
_apply_scaled_font(child)
|
||
_reapply_scaled_stylesheet(child)
|
||
|
||
|
||
class AttitudeIndicator(QWidget):
|
||
"""
|
||
人工地平儀 (ADI) — 仿 Mission Planner 風格
|
||
上半部顯示 roll/pitch 人工地平儀，下方細條顯示航向
|
||
"""
|
||
|
||
def __init__(self, drone_id, parent=None):
|
||
super().__init__(parent)
|
||
self.drone_id = drone_id
|
||
self.heading = 0.0 # 航向 yaw (0–360)
|
||
self.roll = 0.0 # 滾轉 (deg, left- negative)
|
||
self.pitch = 0.0 # 俯仰 (deg, nose-up positive)
|
||
self.setStyleSheet("background-color: transparent;")
|
||
|
||
def update_attitude(self, heading, roll, pitch):
|
||
self.heading = heading % 360
|
||
self.roll = roll
|
||
self.pitch = pitch
|
||
self.update()
|
||
|
||
# ------------------------------------------------------------------ helpers
|
||
def _adi_rect(self):
|
||
"""Returns the square rect used for the ADI ball."""
|
||
w, h = self.width(), self.height()
|
||
side = min(w, h - 14) # leave 14 px at bottom for heading strip
|
||
x = (w - side) / 2
|
||
y = 0
|
||
return x, y, side, side
|
||
|
||
# ------------------------------------------------------------------ paint
|
||
def paintEvent(self, event):
|
||
p = QPainter(self)
|
||
p.setRenderHint(QPainter.RenderHint.Antialiasing)
|
||
self._draw_adi(p)
|
||
self._draw_heading_strip(p)
|
||
|
||
# ---- artificial horizon ------------------------------------------------
|
||
def _draw_adi(self, p):
|
||
from PyQt6.QtGui import QPainterPath
|
||
x0, y0, side, _ = self._adi_rect()
|
||
cx = x0 + side / 2
|
||
cy = y0 + side / 2
|
||
r = side / 2 - 1
|
||
|
||
# clip to circle
|
||
clip_path = QPainterPath()
|
||
clip_path.addEllipse(QPointF(cx, cy), r, r)
|
||
p.setClipPath(clip_path)
|
||
|
||
# pixels-per-degree for pitch (10 deg ≈ side/5)
|
||
ppd = side / 50.0
|
||
|
||
# ---- rotate + translate canvas for roll & pitch
|
||
p.save()
|
||
p.translate(cx, cy)
|
||
p.rotate(-self.roll) # ✅ 負值：NED 坐標系轉換
|
||
pitch_offset = self.pitch * ppd
|
||
|
||
# sky (above horizon)
|
||
sky_color = QColor(30, 100, 180)
|
||
p.fillRect(int(-r*2), int(-r*2 + pitch_offset), int(r*4), int(r*4), sky_color)
|
||
|
||
# ground (below horizon)
|
||
ground_color = QColor(140, 90, 40)
|
||
p.fillRect(int(-r*2), int(pitch_offset), int(r*4), int(r*4), ground_color)
|
||
|
||
# horizon line
|
||
p.setPen(QPen(QColor("#FFFFFF"), 2))
|
||
p.drawLine(int(-r), int(pitch_offset), int(r), int(pitch_offset))
|
||
|
||
# pitch ladder (every 10°, ±30°)
|
||
p.setPen(QPen(QColor(255, 255, 255, 180), 1))
|
||
ladder_font = QFont("Arial")
|
||
ladder_font.setPointSizeF(max(1.0, 6 * _get_font_scale()))
|
||
p.setFont(ladder_font)
|
||
for deg in range(-30, 31, 10):
|
||
if deg == 0:
|
||
continue
|
||
yy = int(pitch_offset - deg * ppd)
|
||
half = int(r * (0.35 if deg % 20 == 0 else 0.22))
|
||
p.drawLine(-half, yy, half, yy)
|
||
|
||
p.restore()
|
||
p.setClipping(False)
|
||
|
||
# ---- roll arc & tick marks (outside clip, fixed frame) ----
|
||
p.save()
|
||
p.translate(cx, cy)
|
||
|
||
arc_r = r - 2
|
||
p.setPen(QPen(QColor("#FFFFFF"), 1))
|
||
# draw arc from -60° to +60° (Qt arc: 0=3o'clock, CCW, 16ths of deg)
|
||
p.drawArc(int(-arc_r), int(-arc_r), int(2*arc_r), int(2*arc_r),
|
||
(90 - 60) * 16, 120 * 16)
|
||
|
||
# tick marks at 0, ±10, ±20, ±30, ±45, ±60
|
||
for deg in [0, 10, 20, 30, 45, 60, -10, -20, -30, -45, -60]:
|
||
rad = math.radians(deg - 90)
|
||
tick = 6 if deg % 30 == 0 else 4
|
||
x1 = arc_r * math.cos(rad)
|
||
y1 = arc_r * math.sin(rad)
|
||
x2 = (arc_r - tick) * math.cos(rad)
|
||
y2 = (arc_r - tick) * math.sin(rad)
|
||
p.drawLine(QPointF(x1, y1), QPointF(x2, y2))
|
||
|
||
# roll pointer triangle (rotates with roll)
|
||
p.rotate(-self.roll) # ✅ 負值：NED 坐標系轉換
|
||
ptr_r = arc_r - 1
|
||
tri = QPolygonF([
|
||
QPointF(0, -ptr_r),
|
||
QPointF(-5, -ptr_r + 9),
|
||
QPointF(5, -ptr_r + 9),
|
||
])
|
||
p.setBrush(QColor("#FFFFFF"))
|
||
p.setPen(Qt.PenStyle.NoPen)
|
||
p.drawPolygon(tri)
|
||
p.restore()
|
||
|
||
# ---- fixed aircraft symbol ----
|
||
p.save()
|
||
p.translate(cx, cy)
|
||
p.setPen(QPen(QColor("#FFD700"), 2))
|
||
# left wing
|
||
p.drawLine(int(-r*0.5), 0, int(-r*0.15), 0)
|
||
p.drawLine(int(-r*0.15), 0, int(-r*0.15), int(r*0.12))
|
||
# right wing
|
||
p.drawLine(int(r*0.15), 0, int(r*0.5), 0)
|
||
p.drawLine(int(r*0.15), 0, int(r*0.15), int(r*0.12))
|
||
# centre dot
|
||
p.setBrush(QColor("#FFD700"))
|
||
p.drawEllipse(QPointF(0, 0), 2.5, 2.5)
|
||
p.restore()
|
||
|
||
# ---- outer ring ----
|
||
p.setPen(QPen(QColor("#888888"), 1))
|
||
p.setBrush(Qt.BrushStyle.NoBrush)
|
||
p.drawEllipse(QPointF(cx, cy), r, r)
|
||
|
||
# ---- heading strip at bottom ------------------------------------------
|
||
def _draw_heading_strip(self, p):
|
||
w = self.width()
|
||
x0, y0, side, _ = self._adi_rect()
|
||
strip_y = y0 + side
|
||
strip_h = self.height() - strip_y
|
||
if strip_h < 4:
|
||
return
|
||
|
||
# background
|
||
p.fillRect(0, int(strip_y), w, strip_h, QColor(30, 30, 30))
|
||
|
||
# heading text centred (bigger)
|
||
p.setPen(QPen(QColor("#FFFFFF")))
|
||
heading_font = QFont("Arial", weight=QFont.Weight.Bold)
|
||
heading_font.setPointSizeF(max(1.0, 10 * _get_font_scale()))
|
||
p.setFont(heading_font)
|
||
hdg_str = f"{int(self.heading)}°"
|
||
p.drawText(0, int(strip_y), w, strip_h, Qt.AlignmentFlag.AlignCenter, hdg_str)
|