class mainWindow(QMainWindow):
def __init__(self):
super(mainWindow, self).__init__()
self.__image = None # SimpleITK.Image
self.__grade = None # Numpy array with shape [num_slice, num_class]
self.child_stat = statAnalyzeWidget()
self.cls_SVC = clsABWidget(ifSVC=True)
self.cls_Logi = clsABWidget(ifSVC=False)
self.statusBar().showMessage('Initializing UI...')
self.initUI()
self.statusBar().showMessage('Initializing Core...')
self.main = mainLogic()
self.statusBar().showMessage('Initializing Signal...')
self.initSignals()
self.statusBar().showMessage('Ready')
def initUI(self):
self.setWindowTitle('Radiomics System')
screenGeometry = QApplication.desktop().screenGeometry()
aspectRatio = 4 / 3.0
blockLen = min(screenGeometry.width()/aspectRatio, \
screenGeometry.height())
self.resize(blockLen * aspectRatio, blockLen)
#qr = self.frameGeometry()
#cp = QDesktopWidget().availableGeometry().center()
#qr.moveCenter(cp)
#self.move(qr.topLeft())
self.initUI_Menubar()
self.initUI_Toolbar()
self.initUI_ContextMenu()
self.initUI_Layout()
self.show()
def initUI_Menubar(self):
menubar = self.menuBar()
self.menuLoad = menubar.addMenu('Load')
# self.menutools = QMenu(menubar)
# self.menutools.setObjectName("menutools")
self.menuImage = self.menuLoad.addMenu('Image')
self.menuROI = self.menuLoad.addMenu('ROI')
# self.menuload.setObjectName("menuload")
self.actionimagedir = QAction('DICOM DIR', self)
self.actionimagedir.triggered.connect(self.actLoadImgDir)
self.actionimagefiles = QAction('NIfTI File', self)
self.actionimagefiles.triggered.connect(self.actLoadStudy)
self.actionROIdir = QAction('DICOM DIR', self)
self.actionROIdir.triggered.connect(self.actLoadROIDir)
self.actionROIfiles = QAction('NIfTI File', self)
self.actionROIfiles.triggered.connect(self.actLoadROI)
self.menuImage.addAction(self.actionimagedir)
self.menuImage.addAction(self.actionimagefiles)
self.menuROI.addAction(self.actionROIdir)
self.menuROI.addAction(self.actionROIfiles)
self.menuSave = menubar.addMenu('Save')
self.actionSaveImg = QAction('Image', self)
self.actionSaveImg.triggered.connect(self.actSaveImg)
self.actionSaveSelROI = QAction('Selected ROI', self)
self.actionSaveSelROI.triggered.connect(self.actSaveSelROI)
self.actionSaveAllROI = QAction('All ROI', self)
self.actionSaveAllROI.triggered.connect(self.actSaveAllROI)
self.menuSave.addAction(self.actionSaveImg)
self.menuSave.addAction(self.actionSaveSelROI)
self.menuSave.addAction(self.actionSaveAllROI)
self.menuRadiomics = menubar.addMenu('Radiomics')
self.menuExt = self.menuRadiomics.addMenu('Feature Extraction')
self.actionPyExt = QAction('PyRadiomics', self)
self.actionPyExt.triggered.connect(self.actFeatureExt)
self.actionCuExt = QAction('cuRadiomics(2D only)', self)
self.actionCuExt.triggered.connect(self.actCuFeatureExt)
self.menuExt.addAction(self.actionPyExt)
self.menuExt.addAction(self.actionCuExt)
self.actionFeatSel = QAction('Feature Selection', self)
self.actionFeatSel.triggered.connect(self.child_stat.show)
self.menuRadiomics.addAction(self.actionFeatSel)
LogiRegAct = QAction('Logit Regression', self)
LogiRegAct.triggered.connect(self.cls_Logi.show)
SVCAct = QAction('Support Vector Classification', self)
SVCAct.triggered.connect(self.cls_SVC.show)
menubar = self.menuBar()
fileMenu = menubar.addMenu('Classification')
fileMenu.addAction(LogiRegAct)
fileMenu.addAction(SVCAct)
DenoiseAct = QAction('Wavelet Denoise', self)
DenoiseAct.triggered.connect(self.actDenoise)
ThredAct = QAction('Threshold Seg', self)
ThredAct.triggered.connect(self.actThreshold)
GasSmoothAct = QAction('Guassian Smooth', self)
GasSmoothAct.triggered.connect(self.actGasSmooth)
MeanSmoothAct = QAction('Mean Smooth', self)
MeanSmoothAct.triggered.connect(self.actMeanSmooth)
MedSmoothAct = QAction('Median Smooth', self)
MedSmoothAct.triggered.connect(self.actMedSmooth)
ResampleAct = QAction('Resample', self)
ResampleAct.triggered.connect(self.actResample)
menubar = self.menuBar()
fileMenu = menubar.addMenu('Pre-processing')
fileMenu.addAction(DenoiseAct)
fileMenu.addAction(ThredAct)
fileMenu.addAction(GasSmoothAct)
fileMenu.addAction(MeanSmoothAct)
fileMenu.addAction(MedSmoothAct)
fileMenu.addAction(ResampleAct)
SRAct = QAction('Super Resolution', self)
SRAct.triggered.connect(self.actGetSR)
DPSegAct = QAction('Segmentation', self)
DPSegAct.triggered.connect(self.actDPSeg)
segbar = self.menuBar()
segMenu = segbar.addMenu('Deep Learning')
segMenu.addAction(SRAct)
segMenu.addAction(DPSegAct)
#loadStudyAct = QAction('Load Dicom Study', self)
#fileMenu.addAction(loadStudyAct)
#exitAct = QAction('Quit', self)
#exitAct.triggered.connect(qApp.quit)
#fileMenu.addAction(exitAct)
def initUI_Toolbar(self):
pass
def initUI_ContextMenu(self):
pass
def initUI_Layout(self):
self.volumeViewer = volumeViewerWidget(self)
self.FeatureDisp = featureDispWidget(self)
self.dicomInfo = dicomInfoWidget(self)
# self.controlPanel = controlPannelWidget(self)
self.annotationPanel = annotationPannelWidget(self)
#self.FeatureSel = featureSelWidget(self)
#self.gradeDisp = gradeDispWidget(self)
# self.refViewer = curveWidget(self)
self.SRViewer = volumeViewerWidget(self)
colormap = (plt.cm.bwr(np.array( \
list([x for x in range(256)]), dtype=np.uint8 \
))*255).astype(np.uint8).reshape(256,1,4) #RGBA
colormap[:, 0, 3] = list(range(255, 0, -2)) + list(range(1, 256, 2))
# dock and central widget
# volumeViewer
self.setCentralWidget(self.volumeViewer)
# dicomInfo
self.dockDicomInfo = QDockWidget("DICOM Info", self)
self.dockDicomInfo.setObjectName("dockDicomInfo")
self.addDockWidget(Qt.RightDockWidgetArea, self.dockDicomInfo)
self.dockDicomInfo.setWidget(self.dicomInfo)
# FeatureExtract
self.dockFeatureDisp = QDockWidget("Extracted Feature", self)
self.dockFeatureDisp.setObjectName("dockFeatureDisp")
self.addDockWidget(Qt.RightDockWidgetArea, self.dockFeatureDisp)
self.dockFeatureDisp.setWidget(self.FeatureDisp)
self.tabifyDockWidget(self.dockDicomInfo, self.dockFeatureDisp)
# Annoation
self.dockAnnotation = QDockWidget("Annotation Tools", self)
self.dockAnnotation.setObjectName("dockAnnotation")
self.addDockWidget(Qt.RightDockWidgetArea, self.dockAnnotation)
self.dockAnnotation.setWidget(self.annotationPanel)
self.dockAnnotation.showMinimized()
self.resizeDocks([self.dockAnnotation, self.dockFeatureDisp], \
[1,10], Qt.Vertical)
# self.tabifyDockWidget(self.dockAnnotation, self.dockFeatureDisp)
# controlPanel
# self.dockControlPanel = QDockWidget("Control Panel", self)
# self.dockControlPanel.setObjectName("dockControlPanel")
# self.addDockWidget(Qt.RightDockWidgetArea, self.dockControlPanel)
# self.dockControlPanel.setWidget(self.controlPanel)
# self.dockControlPanel.showMinimized()
# self.resizeDocks([self.dockControlPanel, self.dockFeatureDisp], \
# [1,10], Qt.Vertical)
# FeatureSelection
# self.dockFeatureSel = QDockWidget("Selected Feature", self)
# self.dockFeatureSel.setObjectName("dockFeatureSel")
# self.addDockWidget(Qt.RightDockWidgetArea, self.dockFeatureSel)
# self.dockFeatureSel.setWidget(self.FeatureSel)
# self.tabifyDockWidget(self.dockFeatureDisp, self.dockFeatureSel)
# # gradeDisp
# self.dockGradeDisp = QDockWidget("Grade (Prediction)", self)
# self.dockGradeDisp.setObjectName("dockGradeDisp")
# self.addDockWidget(Qt.RightDockWidgetArea, self.dockGradeDisp)
# self.dockGradeDisp.setWidget(self.gradeDisp)
# self.tabifyDockWidget(self.dockDicomInfo, self.dockGradeDisp)
# reference image and SR image
# self.dockRefViewer = QDockWidget("Reference", self)
# self.dockRefViewer.setObjectName("dockRefViewer")
# self.addDockWidget(Qt.LeftDockWidgetArea, self.dockRefViewer)
# self.dockRefViewer.setWidget(self.refViewer)
# self.dockRefViewer.setFloating(True)
# self.dockRefViewer.setVisible(False)
self.dockSRViewer = QDockWidget("Super-Resolution", self)
self.dockSRViewer.setObjectName("dockSRViewer")
self.addDockWidget(Qt.LeftDockWidgetArea, self.dockSRViewer)
self.dockSRViewer.setWidget(self.SRViewer)
self.dockSRViewer.setFloating(True)
self.dockSRViewer.setVisible(False)
def initSignals(self):
# self.controlPanel.btnLoad.clicked.connect(self.actLoadStudy)
# self.controlPanel.btnROI.clicked.connect(self.actLoadROI)
# self.controlPanel.btnExt.clicked.connect(self.actFeatureExt)
# self.controlPanel.btnSel.clicked.connect(self.child_stat.show)
self.annotationPanel.btnDoAnn.clicked.connect(self.actDoAnn)
self.annotationPanel.btnDoAnn.setIcon(
QIcon("./qdarkstyle/polygon.png"))
self.annotationPanel.btnPoly.clicked.connect(self.actSetPoly)
self.annotationPanel.btnPoly.setIcon(
(QIcon("./qdarkstyle/polygon.png")))
# self.annotationPanel.btnPoly.setStyleSheet("QPushButton{border-image: url(./qdarkstyle/polygon.png)}")
self.annotationPanel.btnCircle.clicked.connect(self.actSetCircle)
self.annotationPanel.btnCircle.setIcon(
(QIcon("./qdarkstyle/circle.png")))
# self.annotationPanel.btnTri.clicked.connect(self.actSetTri)
# self.annotationPanel.btnTri.setIcon((QIcon("./qdarkstyle/tri.png")))
self.annotationPanel.btnJux.clicked.connect(self.actSetJux)
self.annotationPanel.btnJux.setIcon((QIcon("./qdarkstyle/juxing.png")))
# childBoxGeo.addWidget(self.btnPoly)
# childBoxGeo.addWidget(self.btnCircle)
# childBoxGeo.addWidget(self.btnTri)
# childBoxGeo.addWidget(self.btnJux)
self.annotationPanel.btnAccROI.clicked.connect(self.actAccROI)
self.annotationPanel.btnClrSelROI.clicked.connect(self.actClrSelROI)
self.annotationPanel.btnClrAllROI.clicked.connect(self.actClrAllROI)
# self.annotationPanel.btnSaveSelROI.clicked.connect(self.actSaveSelROI)
# self.annotationPanel.btnSaveAllROI.clicked.connect(self.actSaveAllROI)
# self.controlPanel.btnCla.clicked.connect(self.actGetGrade)
# self.controlPanel.btnSeg.clicked.connect(self.actGetSeg)
# self.controlPanel.btnRef.clicked.connect(self.actGetReference)
# self.controlPanel.btnSR.clicked.connect(self.actGetSR)
#self.volumeViewer.sliderIndex.valueChanged.connect( \
# lambda x: self.refViewer.setIndex(x-1))
self.FeatureDisp.cellPressed.connect( \
lambda row, col: self.volumeViewer.sliderIndex.setValue(row+1))
def actSetPoly(self):
self.__type = 'Poly'
self.actDoAnn()
return
def actSetCircle(self):
self.__type = 'Circle'
self.actDoAnn()
return
# def actSetTri(self):
# self.__type ='Tri'
# self.actDoAnn()
# return
def actSetJux(self):
self.__type = 'Jux'
self.actDoAnn()
return
def actDoAnn(self):
if self.annotationPanel.rbS.isChecked() == True:
self.volumeViewer.setROI(ax='S', type=self.__type)
if self.annotationPanel.rbA.isChecked() == True:
self.volumeViewer.setROI(ax='A', type=self.__type)
if self.annotationPanel.rbC.isChecked() == True:
self.volumeViewer.setROI(ax='C', type=self.__type)
return
def actAccROI(self):
# labelArr=self.volumeViewer.accROI()
# print(np.unique(labelArr))
# print(labelArr.dtype)
self.main.setROI(self.volumeViewer.accROI())
# self.volumeViewer.setLabel(ROI)
return
def actSaveSelROI(self):
self.volumeViewer.saveSelROI()
return
def actSaveAllROI(self):
self.volumeViewer.saveAllROI()
return
def actClrSelROI(self):
self.volumeViewer.clrSelROI()
return
def actClrAllROI(self):
self.volumeViewer.clrAllROI()
return
def actDenoise(self):
self.volumeViewer.denoise()
return
def actGasSmooth(self):
self.volumeViewer.setPreprocessMethod(method='Gaussian')
return
def actMeanSmooth(self):
self.volumeViewer.setPreprocessMethod(method='Mean')
return
def actMedSmooth(self):
self.volumeViewer.setPreprocessMethod(method='Median')
return
def actResample(self):
self.volumeViewer.setPreprocessMethod(method='Resample')
return
def actThreshold(self):
self.volumeViewer.setSegMethod(method='Threshold')
return
def actDPSeg(self):
pass
@pyqtSlot()
def actLoadImgDir(self, directory=None):
start = time.time()
self.statusBar().showMessage('Loading Study...')
if directory is None:
dialog = QFileDialog(self)
dialog.setFileMode(QFileDialog.DirectoryOnly)
dialog.setViewMode(QFileDialog.List)
dialog.setOption(QFileDialog.ShowDirsOnly, True)
if dialog.exec_():
directory = str(dialog.selectedFiles()[0])
else:
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
try:
images = ReadSagittalPDs(directory)
except Exception as err:
msgBox = QMessageBox(self)
msgBox.setText(str(type(err)) + str(err))
msgBox.exec()
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
selectDialog = dicomSelectDialog(images)
if selectDialog.exec_():
image = images[selectDialog.selectedIndex]
else:
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
# get image data
# image_out = sitk.GetImageFromArray(sitk.GetArrayFromImage(image))
#
# # setup other image characteristics
# image_out.SetOrigin(image.GetOrigin())
# image_out.SetSpacing(image.GetSpacing())
# # set to RAI
# image_out.SetDirection(tuple(-0.0, 0.0, -1.0, 1.0, -0.0, 0.0, 0.0, -1.0, 0.0))
# # sitk.WriteImage(image_out, 'test.mha')
# image = image_out
self.main.setImage(image)
self.volumeViewer.setImage(image)
self.dicomInfo.setImage(image)
self.dockDicomInfo.setVisible(True)
# self.dockRefViewer.setVisible(False)
self.dockSRViewer.setVisible(False)
# self.gradeDisp.setGrade(None)
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
def actLoadStudy(self, directory=None):
start = time.time()
self.statusBar().showMessage('Loading Study...')
directory = QFileDialog.getOpenFileName(self,
"Select one file to open",
"./", "Files (*.nii *.dcm)")
print(directory)
directory = str(directory[0])
#if directory is None:
#dialog = QFileDialog(self)
# dialog.setFileMode(QFileDialog.DirectoryOnly)
# dialog.setViewMode(QFileDialog.List)
# dialog.setOption(QFileDialog.ShowDirsOnly, True)
# if dialog.exec_():
# directory = str(dialog.selectedFiles()[0])
# else:
# self.statusBar().showMessage( \
# 'Ready ({:.2f}s)'.format(time.time() - start))
# return
try:
#images = ReadSagittalPDs(directory)
image = ReadImage(directory)
except Exception as err:
msgBox = QMessageBox(self)
msgBox.setText(str(type(err)) + str(err))
msgBox.exec()
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
# selectDialog = dicomSelectDialog(images)
# if selectDialog.exec_():
# image = images[selectDialog.selectedIndex]
# else:
# self.statusBar().showMessage( \
# 'Ready ({:.2f}s)'.format(time.time() - start))
# return
self.main.setImage(image)
self.volumeViewer.setImage(image)
self.dicomInfo.setImage(image)
self.dockDicomInfo.setVisible(True)
# self.dockRefViewer.setVisible(False)
self.dockSRViewer.setVisible(False)
# self.gradeDisp.setGrade(None)
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
def actLoadROIDir(self, directory=None):
start = time.time()
self.statusBar().showMessage('Loading Study...')
if directory is None:
dialog = QFileDialog(self)
dialog.setFileMode(QFileDialog.DirectoryOnly)
dialog.setViewMode(QFileDialog.List)
dialog.setOption(QFileDialog.ShowDirsOnly, True)
if dialog.exec_():
directory = str(dialog.selectedFiles()[0])
else:
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
try:
ROIs = ReadSagittalPDs(directory)
except Exception as err:
msgBox = QMessageBox(self)
msgBox.setText(str(type(err)) + str(err))
msgBox.exec()
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
selectDialog = dicomSelectDialog(images)
if selectDialog.exec_():
ROI = ROIs[selectDialog.selectedIndex]
else:
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
self.main.setROI(ROI)
self.volumeViewer.setLabel(ROI)
# self.dockRefViewer.setVisible(False)
self.dockSRViewer.setVisible(False)
# self.gradeDisp.setGrade(None)
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
def actLoadROI(self, directory=None):
start = time.time()
self.statusBar().showMessage('Loading ROI...')
directory = QFileDialog.getOpenFileName(self,
"Select one file to open",
"./", "Files (*.nii *.dcm)")
print(directory)
directory = str(directory[0])
try:
ROI = ReadROI(directory)
except Exception as err:
msgBox = QMessageBox(self)
msgBox.setText(str(type(err)) + str(err))
msgBox.exec()
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
self.main.setROI(ROI)
self.volumeViewer.setLabel(ROI)
self.dockSRViewer.setVisible(False)
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
def actFeatureExt(self):
start = time.time()
self.statusBar().showMessage('Extracting Feature...')
try:
feature_extracted = self.main.getFeature()
except Exception as err:
msgBox = QMessageBox(self)
msgBox.setText(str(type(err)) + str(err))
msgBox.exec()
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
self.FeatureDisp.setFeature(feature_extracted)
self.dockFeatureDisp.setVisible(True)
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
def actCuFeatureExt(self):
start = time.time()
self.statusBar().showMessage('Extracting Feature...')
try:
feature_extracted = self.main.getCuFeature()
except Exception as err:
msgBox = QMessageBox(self)
msgBox.setText(str(type(err)) + str(err))
msgBox.exec()
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
self.FeatureDisp.setFeature(feature_extracted)
self.dockFeatureDisp.setVisible(True)
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
def actSaveImg(self):
pass
# def actFeatureSel(self):
# start = time.time()
# self.statusBar().showMessage('Statistical Analyzing...')
# directory = QFileDialog.getOpenFileName(self,
# "Select the feature folder",
# "./",
# "Files (*.nii *.dcm)")
# print(directory)
# directory = str(directory[0])
# try:
# feature_selected = self.main.featureSel()
# except Exception as err:
# msgBox = QMessageBox(self)
# msgBox.setText(str(type(err)) + str(err))
# msgBox.exec()
# self.statusBar().showMessage( \
# 'Ready ({:.2f}s)'.format(time.time() - start))
# return
# msgBox = QMessageBox(self)
# if feature_selected is True:
# msgBox.setText('Done! Please check the folder.')
# return
@pyqtSlot()
def actGetGrade(self):
start = time.time()
self.statusBar().showMessage('Predicting Grade...')
try:
grade = self.main.getGrade()
except Exception as err:
msgBox = QMessageBox(self)
msgBox.setText(str(type(err)) + str(err))
msgBox.exec()
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
self.gradeDisp.setGrade(grade)
self.dockGradeDisp.setVisible(True)
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
msgBox = QMessageBox(self)
msgBox.setText('得出诊断意见：'+
'我们从四千多个膝关节测共振图层（健康人和患者）'+\
'与医生的诊断结果中，利用深度学习的方法，'+\
'提取图像的特征及其与诊断结果的关系。'+\
'利用训练出的模型，'+\
'计算机可以自动从图像中预测诊断意见。'+'\n'+\
'提示水肿区域：'+\
'我们从三百多层带有水肿区域的磁共振图像'+\
'与医生手工勾勒出的水肿区域中，'+\
'利用深度学习的方法提取特征。利用训练出的模型，'+\
'计算机可以自动从图像中预测图像是否有水肿，'+\
'以及水肿区域的位置。')
#msgBox.exec()
return
@pyqtSlot()
def actGetSeg(self):
start = time.time()
self.statusBar().showMessage('Predicting Segmentation...')
try:
seg = self.main.getSeg()
except Exception as err:
msgBox = QMessageBox(self)
msgBox.setText(str(type(err)) + str(err))
msgBox.exec()
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
self.volumeViewer.setLabel(seg)
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
msgBox = QMessageBox(self)
msgBox.setText('生成分割结果：'+\
'我们从100多例患者膝关节磁共振的手工标记结果中，'+\
'利用深度学习的方法提取特征。利用训练出的模型，'+\
'计算机可以自动从图像中得到膝关节结构的分割结果。')
#msgBox.exec()
@pyqtSlot()
def actGetSR(self):
start = time.time()
self.statusBar().showMessage('Enhancing Resolution...')
try:
SR = self.main.getSuperResolution()
except Exception as err:
msgBox = QMessageBox(self)
msgBox.setText(str(type(err)) + str(err))
msgBox.exec()
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
self.SRViewer.setImage(SR)
if not self.dockSRViewer.isVisible():
self.dockSRViewer.setVisible(True)
if self.dockSRViewer.isFloating():
self.dockSRViewer.resize( \
self.SRViewer.viewerSlice.viewportSizeHint()*1.4)
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
msgBox = QMessageBox(self)
msgBox.setText('分辨率增强（层间距）')
#msgBox.exec()
@pyqtSlot()
def actGetReference(self):
start = time.time()
self.statusBar().showMessage('Predicting Reference...')
try:
#'PatientAge': '0010|1010', \
img = self.main.getImage()
self.refViewer.openCSV(os.path.join( \
os.path.dirname(__file__), 'plot.csv'))
age = img.GetMetaData('0010|1010')
age = int(age[0:3])
thickness = self.main.getThickness()
except Exception as err:
msgBox = QMessageBox(self)
msgBox.setText(str(type(err)) + str(err))
msgBox.exec()
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
return
#print(age, *thickness)
self.refViewer.plotCurve(age, *thickness)
if not self.dockRefViewer.isVisible():
self.dockRefViewer.setVisible(True)
if self.dockRefViewer.isFloating():
self.dockRefViewer.resize( \
self.refViewer.sizeHint()*1.1)
self.statusBar().showMessage( \
'Ready ({:.2f}s)'.format(time.time() - start))
msgBox = QMessageBox(self)
msgBox.setText('生成参考')
class MainWindow(QMainWindow):
def __init__(self):
super().__init__()
self.initUI()
def initUI(self):
maxWidth = QDesktopWidget().availableGeometry().width()
maxHeight = QDesktopWidget().availableGeometry().height()
workSpace = QWidget(self)
#### MainWindow Configuration
self.setWindowTitle("Application")
self.setWindowIcon(QIcon("icon.png"))
self.tray_icon = QSystemTrayIcon(QIcon("icon.png"))
self.tray_icon.show()
#### Tray MENU #################
trayIconMenu = QMenu(self)
trayIconMenu.setTitle("Main menu")
trayIconMenu.addAction("Show Main Window", self.showMainWindow)
trayIconMenu.addAction("Switch chat", self.toggleDockWidget)
trayIconMenu.addSeparator()
trayIconMenu.addAction("Quit", self.close)
self.tray_icon.setContextMenu(trayIconMenu)
######################################
#### MENUBAR description: #################
mainmenu = self.menuBar()
file = mainmenu.addMenu("File")
file.addAction("isFloating", self.floatingInfo)
file.addAction("Toggle Chat", self.toggleDockWidget)
file.addAction("Close", self.close)
######################
self.dockWidget = QDockWidget("Messenger", self)
self.dockWidget.setFloating(True)
self.dockWidget.setVisible(True)
self.dockWidget.setMinimumSize(300, 460)
self.dockWidget.setMaximumSize(maxWidth / 2, maxHeight)
self.dockWidget.setBaseSize(maxWidth / 2, maxHeight)
self.dockWidget.resize(maxWidth / 4, maxHeight / 1.5)
self.dockWidget.setFeatures(QDockWidget.NoDockWidgetFeatures)
self.dockWidget.setVisible(False)
#### TODO:" set dockWidget widgets" ################
messengerWidget = QWidget()
masterVertical = QVBoxLayout()
messengerWidget.setLayout(masterVertical)
slaveHorizont = QHBoxLayout()
subSlaveVertical = QVBoxLayout()
msgSpace = QListView()
textEdit = QTextEdit()
sendBtn = QPushButton("Send")
masterVertical.addWidget(msgSpace)
masterVertical.addLayout(slaveHorizont)
slaveHorizont.addWidget(textEdit)
slaveHorizont.addLayout(subSlaveVertical)
subSlaveVertical.addWidget(sendBtn)
# self.dockWidget.setLayout(QVBoxLayout())
# print(self.dockWidget.setLayout(QVBoxLayout()))
############################################################
# self.statusBar()
toggleDockWidgetBtn = QPushButton("Show geomerty", workSpace)
toggleDockWidgetBtn.move(10, 40)
toggleDockWidgetBtn.clicked.connect(self.showGeometry)
#### TEST BUTONS ####
hideBtn = QPushButton("hide main", workSpace)
hideBtn.move(10, 70)
hideBtn.clicked.connect(self.hideMainWindow)
#######################################################
self.setGeometry(300, 300, 300, 500)
self.dockWidget.resize(333, 1000)
self.setCentralWidget(workSpace)
self.addDockWidget(Qt.RightDockWidgetArea, self.dockWidget)
def toggleDockWidget(self):
flag = False if self.dockWidget.isVisible() else True
text = "Messenger status was changed to {}".format(
"visible" if flag else "hide")
self.dockWidget.setVisible(flag)
# self.dockWidget.setGeometry(650, 300, 200, 500)
#### SET DOCKWINDOW GEAMETRY AND POSITION
qtRect = self.dockWidget.frameGeometry()
rightPoint = QDesktopWidget().availableGeometry().bottomRight()
#### FOR LINUX POSITION ###############################
# qtRect.moveBottomRight(rightPoint)
# self.dockWidget.move(qtRect.bottomRight())
#### FOR WINDOWS POSITION #############################
qtRect.moveBottomRight(rightPoint)
self.dockWidget.move(qtRect.topLeft())
##############################################################
self.statusBar().showMessage(text, 500)
#### test functionality
def showGeometry(self):
availGeometry = QDesktopWidget().availableGeometry()
self.statusBar().showMessage("Available geometry.")
print("Available geometry: ", availGeometry)
print("Right point: ",
QDesktopWidget().availableGeometry().bottomRight())
print("DockWidget frame bottomRight: ",
self.dockWidget.frameGeometry().bottomRight())
print("DockWidget frame topLeft: ",
self.dockWidget.frameGeometry().topLeft())
print("\n" + "|---|" * 15 + "\n")
def hideMainWindow(self):
self.hide()
def showMainWindow(self):
self.show()
def floatingInfo(self):
flag = False if self.dockWidget.isFloating() else True
print("dockWidget status is: ", flag)
self.dockWidget.setFloating(flag)
