-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathDisplay_widget.py
115 lines (104 loc) · 4.87 KB
/
Display_widget.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
from PyQt5 import QtWidgets
from PyQt5.QtCore import QSize, Qt
from PyQt5.QtGui import QIcon, QPalette, QFont, QCursor
from PyQt5.QtWidgets import QMenu, QTreeWidgetItem, QLabel, QSizePolicy, QTreeWidget, QFileDialog,QWidget
from PyQt5 import QtCore
import sys
from Slices_Viewer_Widget import Slice_Viewer_Widget
from Signal_Central_Process_Unit import SCPU
from VTK_Viewer_widget import VTK_Viewer_widget
import numpy as np
from tools import *
class Display_widget(QWidget):
pbar_signal = QtCore.pyqtSignal()
def __init__(self):
super().__init__()
self.init_ui()
self.init_data()
self.init_SCPU_signal_connection()
def init_ui(self):
#self.setWindowTitle("FT-ITK")
self.main_widget = QtWidgets.QWidget()
self.main_layout = QtWidgets.QGridLayout()
#self.main_widget.setObjectName('main_widget')
#self.main_widget.setLayout(self.main_layout)
#self.setCentralWidget(self.main_widget) # 设置窗口主部件
self.setLayout(self.main_layout)
self.Axial_Viewer = Slice_Viewer_Widget(type="axial") # axial view,横断位
self.Sagittal_Viewer = Slice_Viewer_Widget(type="sagittal") # sagittal view，矢状位
self.Coronal_Viewer = Slice_Viewer_Widget(type="coronal") # coronal view，冠状位
# self.Others_Viewer = Slice_Viewer_Widget()
self.VTK_Viewer=VTK_Viewer_widget()
self.main_layout.addWidget(self.Axial_Viewer,0,0)
self.main_layout.addWidget(self.Sagittal_Viewer,0,1)
self.main_layout.addWidget(self.Coronal_Viewer,1,0)
self.main_layout.addWidget(self.VTK_Viewer,1,1)
# self.data=np.load("medical_files/CT/0001.npy")
# self.label_data=np.load("medical_files/Label/0001.npy")
# self.VTK_Viewer.load_nii()
# self.main_layout.addWidget(self.Others_Viewer,1,1)
# self.Axial_Viewer.load_data_from_father(self.data)
# self.Sagittal_Viewer.load_data_from_father(self.data)
# self.Coronal_Viewer.load_data_from_father(self.data)
# self.Axial_Viewer.load_label_data_from_father(self.label_data)
# self.Sagittal_Viewer.load_label_data_from_father(self.label_data)
# self.Coronal_Viewer.load_label_data_from_father(self.label_data)
self.main_layout.setSpacing(2)
def init_data(self):
self.SCPU=SCPU() #Signal_Central_Process_Unit(SCPU)
self.data=None
self.label_data=None
self.current_file_path=None
def load_data(self,file_path):
self.current_file_path=file_path
if ".npy" in file_path: # numpy_file
data = np.load(file_path)
self.data=data
elif ".nii" in file_path: #nii file
data,*_ =get_medical_image(file_path)
data = np.array(data)
data = data.transpose(1, 2, 0)
self.data=data
self.Axial_Viewer.load_data_from_father(self.data)
self.Sagittal_Viewer.load_data_from_father(self.data)
self.Coronal_Viewer.load_data_from_father(self.data)
self.VTK_Viewer.load_nii()
def load_label_data(self,label_file_path=None):
if self.current_file_path is not None:
self.pbar_signal.emit()
if label_file_path is None:
label_file_path1=self.current_file_path.replace("CT","Label")
if ".npy" in label_file_path1: # numpy_file
data = np.load(label_file_path1)
self.label_data = data
elif ".nii" in label_file_path: # nii file
data, *_ = get_medical_image(label_file_path1)
data = np.array(data)
data = data.transpose(1, 2, 0)
self.label_data = data
self.Axial_Viewer.load_label_data_from_father(self.label_data)
self.Sagittal_Viewer.load_label_data_from_father(self.label_data)
self.Coronal_Viewer.load_label_data_from_father(self.label_data)
self.VTK_Viewer.load_nii()
else:
pass
def init_SCPU_signal_connection(self):
self.Axial_Viewer.output_signal.connect(self.SCPU.Process_Core)
self.Sagittal_Viewer.output_signal.connect(self.SCPU.Process_Core)
self.Coronal_Viewer.output_signal.connect(self.SCPU.Process_Core)
self.SCPU.command_to_axial.connect(self.Axial_Viewer.handle_SCPU_command)
self.SCPU.command_to_sagittal.connect(self.Sagittal_Viewer.handle_SCPU_command)
self.SCPU.command_to_coronal.connect(self.Coronal_Viewer.handle_SCPU_command)
def flash(self):
self.Axial_Viewer.flash()
self.Sagittal_Viewer.flash()
self.Coronal_Viewer.flash()
def clear_label(self):
self.Axial_Viewer.clear_label()
self.Sagittal_Viewer.clear_label()
self.Coronal_Viewer.clear_label()
if __name__ == '__main__':
app = QtWidgets.QApplication(sys.argv)
gui = Display_widget()
gui.show()
sys.exit(app.exec_())