181 lines
5.9 KiB
Python
181 lines
5.9 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
'''
|
||
# @Time : 2026/1/7 10:41
|
||
# @Author : reenrr
|
||
# @File : 3D.py
|
||
# @Desc : 界面上的线条3D显示
|
||
'''
|
||
from re import T
|
||
import sys
|
||
|
||
from PySide6.QtWidgets import QApplication, QWidget, QVBoxLayout
|
||
from matplotlib.backends.backend_qtagg import FigureCanvasQTAgg as FigureCanvas
|
||
import matplotlib.pyplot as plt
|
||
from PySide6.QtCore import Qt
|
||
|
||
|
||
class Target3DWidget(QWidget):
|
||
"""
|
||
3D显示窗口，支持类型A和类型B
|
||
"""
|
||
|
||
VERTICES_A = [
|
||
# 底部截面（Z=0平面）
|
||
(0, 3, 0), (100, 3, 0),
|
||
(100, 3, 0), (100, 0, 0),
|
||
# 中间截面（Z=0.5平面）
|
||
(0, 0.5, 0.5), (0, 3, 0.5),
|
||
(0, 3, 0.5), (100, 3, 0.5),
|
||
(100, 3, 0.5), (100, 0.5, 0.5),
|
||
(100, 0.5, 0.5), (0, 0.5, 0.5),
|
||
# 顶部界面（Z=2平面）
|
||
(0, 0, 2), (0, 0.5, 2),
|
||
(0, 0.5, 2), (100, 0.5, 2),
|
||
(100, 0.5, 2), (100, 0, 2),
|
||
(100, 0, 2), (0, 0, 2),
|
||
# z方向的连线
|
||
(0, 3, 0.5), (0, 3, 0),
|
||
(100, 3, 0.5), (100, 3, 0),
|
||
(0, 0.5, 2), (0, 0.5, 0.5),
|
||
(100, 0.5, 2), (100, 0.5, 0.5),
|
||
(100, 0, 2), (100, 0, 0),
|
||
]
|
||
|
||
VERTICES_B = [
|
||
# 底部截面（Z=0平面）
|
||
(0, 3, 0), (100, 3, 0),
|
||
(100, 3, 0), (100, 0, 0),
|
||
# 中间截面（Z=0.5平面）
|
||
(0, 0.5, 0.5), (0, 2.5, 0.5),
|
||
(100, 2.5, 0.5), (100, 0.5, 0.5),
|
||
(100, 0.5, 0.5), (0, 0.5, 0.5),
|
||
# 中间截面（Z=1平面）
|
||
(0, 2.5, 1), (0, 3, 1),
|
||
(0, 3, 1), (100, 3, 1),
|
||
(100, 3, 1), (100, 2.5, 1),
|
||
(100, 2.5, 1), (0, 2.5, 1),
|
||
# 顶部界面（Z=2平面）
|
||
(0, 0, 2), (0, 0.5, 2),
|
||
(0, 0.5, 2), (100, 0.5, 2),
|
||
(100, 0.5, 2), (100, 0, 2),
|
||
(100, 0, 2), (0, 0, 2),
|
||
# z方向的连线
|
||
(0, 3, 1), (0, 3, 0),
|
||
(100, 3, 1), (100, 3, 0),
|
||
(0, 0.5, 2), (0, 0.5, 0.5),
|
||
(100, 0.5, 2), (100, 0.5, 0.5),
|
||
(100, 0, 2), (100, 0, 0),
|
||
(0, 2.5, 1), (0, 2.5, 0.5),
|
||
(100, 2.5, 1), (100, 2.5, 0.5)
|
||
]
|
||
|
||
ANNOTATIONS_A = [
|
||
("a", 100, 0.25, 2, 'red'),
|
||
("b", 100, 0.5, 1.25, 'magenta'),
|
||
("c", 100, 1, 0, 'green'),
|
||
("d", 100, 3, 0.25, 'blue'),
|
||
]
|
||
|
||
ANNOTATIONS_B = [
|
||
("a", 100, 0.25, 2, 'red'),
|
||
("b", 100, 0.5, 1.25, 'magenta'),
|
||
("c", 100, 1, 0, 'green'),
|
||
("d", 100, 3, 0.5, 'blue'),
|
||
("e", 100, 2.5, 0.6, 'orange'),
|
||
("f", 100, 2.75, 1, 'purple')
|
||
]
|
||
|
||
def __init__(self, anno: bool, index: int = 0, parent=None):
|
||
super().__init__(parent)
|
||
self._index = index
|
||
self._anno = anno
|
||
|
||
self.setWindowFlags(Qt.Window | Qt.FramelessWindowHint) # 无边框窗口
|
||
self.setAttribute(Qt.WA_TranslucentBackground) # 设置背景为透明
|
||
self.setStyleSheet("background: transparent;")
|
||
|
||
self.init_3d_plot()
|
||
self.setLayout(QVBoxLayout())
|
||
self.layout().setContentsMargins(0, 0, 0, 0) # 移除布局的内边距
|
||
self.layout().addWidget(self.canvas)
|
||
|
||
def init_3d_plot(self):
|
||
# 3D画布
|
||
self.fig = plt.figure()
|
||
self.fig.patch.set_facecolor('none') # 设置figure背景透明
|
||
self.canvas = FigureCanvas(self.fig)
|
||
self.ax = self.fig.add_subplot(111, projection='3d')
|
||
self.ax.set_facecolor('none') # 设置axes背景透明
|
||
|
||
# 绘制3D形状
|
||
self.draw_target_shape()
|
||
# 添加标注
|
||
if self._anno:
|
||
self.add_annotations()
|
||
|
||
# ========== 核心修改：X/Y/Z轴统一为1cm单位 ==========
|
||
self.ax.set_xlabel('X')
|
||
self.ax.set_ylabel('Y')
|
||
self.ax.set_zlabel('Z')
|
||
|
||
# 1. 清空刻度数值（隐藏刻度数字）
|
||
self.ax.set_xticklabels([])
|
||
self.ax.set_yticklabels([])
|
||
self.ax.set_zticklabels([])
|
||
|
||
# 2. 隐藏刻度线（针对3D轴的特殊设置）
|
||
self.ax.tick_params(axis='x', which='both', length=0) # X轴刻度线长度设为0
|
||
self.ax.tick_params(axis='y', which='both', length=0) # Y轴刻度线长度设为0
|
||
self.ax.tick_params(axis='z', which='both', length=0) # Z轴刻度线长度设为0
|
||
|
||
# 轴范围保留（保证3D图形显示范围正确）
|
||
self.ax.set_xlim(0, 120)
|
||
self.ax.set_ylim(0, 5)
|
||
self.ax.set_zlim(0, 5)
|
||
|
||
# 调整视角（适配X轴1cm刻度）
|
||
self.ax.view_init(elev=20, azim=60)
|
||
|
||
def draw_target_shape(self):
|
||
"""绘制3D结构"""
|
||
vertices = self.VERTICES_A if self._index == 0 else self.VERTICES_B
|
||
# 绘制所有棱边
|
||
for i in range(0, len(vertices), 2):
|
||
x = [vertices[i][0], vertices[i + 1][0]]
|
||
y = [vertices[i][1], vertices[i + 1][1]]
|
||
z = [vertices[i][2], vertices[i + 1][2]]
|
||
self.ax.plot(x, y, z, color='black', linewidth=2)
|
||
|
||
def add_annotations(self):
|
||
"""标注位置（适配X轴1cm单位）"""
|
||
annotations = self.ANNOTATIONS_A if self._index == 0 else self.ANNOTATIONS_B
|
||
for text, x, y, z, color in annotations:
|
||
self.ax.text(x, y, z, text, fontsize=14, color=color, weight='bold')
|
||
|
||
|
||
# ----------对外接口----------
|
||
def show(x: int, y: int, w: int, h: int, index: int):
|
||
"""
|
||
显示3D视图窗口
|
||
:param x: 窗口横坐标
|
||
:param y: 窗口纵坐标
|
||
:param w: 窗口宽度
|
||
:param h: 窗口高度
|
||
:param index: 窗口索引，0为A视图，1为B视图
|
||
"""
|
||
if index not in (0, 1):
|
||
raise ValueError("index must be 0 or 1")
|
||
|
||
app = QApplication(sys.argv)
|
||
window = Target3DWidget(anno=True, index=index)
|
||
window.setGeometry(x, y, w, h)
|
||
window.show()
|
||
sys.exit(app.exec())
|
||
|
||
|
||
# ----------测试接口----------
|
||
if __name__ == '__main__':
|
||
# show(100, 100, 700, 700, 0) # 显示A视图窗口
|
||
show(100, 100, 700, 700, 1) # 显示B视图窗口
|