# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. import os.path as osp from functools import partial from qtpy import QtCore, QtGui, QtWidgets from qtpy.QtGui import QIcon from qtpy.QtCore import QTimer from eiseg import pjpath, __APPNAME__, __VERSION__, logger from eiseg.widget.create import creat_dock, create_button, create_slider, create_text from widget import AnnotationScene, AnnotationView from widget.create import * from widget.table import TableWidget from widget.vtk import VTKWidget # log = logging.getLogger(__name__ + ".ui") class Ui_EISeg(object): def __init__(self): super(Ui_EISeg, self).__init__() self.tr = partial(QtCore.QCoreApplication.translate, "APP_EISeg") def setupUi(self, MainWindow): ## -- 主窗体设置 -- MainWindow.setObjectName("MainWindow") MainWindow.setMinimumSize(QtCore.QSize(1366, 768)) # 1366x768的屏幕显示不全 MainWindow.setWindowTitle(__APPNAME__ + " " + __VERSION__) MainWindow.setWindowIcon(QIcon()) # TODO: 默认图标需要换一个吗，貌似不能不显示图标 self.CentralWidget = QtWidgets.QWidget(MainWindow) self.CentralWidget.setObjectName("CentralWidget") MainWindow.setCentralWidget(self.CentralWidget) ## ----- ## -- 工具栏 -- toolBar = QtWidgets.QToolBar(self) sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Preferred, QtWidgets.QSizePolicy.Minimum) sizePolicy.setHorizontalStretch(0) sizePolicy.setVerticalStretch(0) sizePolicy.setHeightForWidth(toolBar.sizePolicy().hasHeightForWidth()) toolBar.setSizePolicy(sizePolicy) toolBar.setMinimumSize(QtCore.QSize(0, 33)) toolBar.setMovable(True) toolBar.setAllowedAreas(QtCore.Qt.BottomToolBarArea | QtCore.Qt.TopToolBarArea) toolBar.setObjectName("toolBar") self.toolBar = toolBar MainWindow.addToolBar(QtCore.Qt.TopToolBarArea, self.toolBar) ## ----- ## -- 状态栏 -- self.statusbar = QtWidgets.QStatusBar(MainWindow) self.statusbar.setObjectName("statusbar") self.statusbar.setStyleSheet("QStatusBar::item {border: none;}") MainWindow.setStatusBar(self.statusbar) self.statusbar.addPermanentWidget( self.show_logo(osp.join(pjpath, "resource/Paddle.png"))) ## ----- ## -- 图形区域 -- ImageRegion = QtWidgets.QHBoxLayout(self.CentralWidget) ImageRegion.setObjectName("ImageRegion") # 滑动区域 self.scrollArea = QtWidgets.QScrollArea(self.CentralWidget) self.scrollArea.setWidgetResizable(True) self.scrollArea.setObjectName("scrollArea") ImageRegion.addWidget(self.scrollArea) # 图形显示 self.scene = AnnotationScene() self.scene.addPixmap(QtGui.QPixmap()) self.canvas = AnnotationView(self.scene, self) sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Expanding) self.canvas.setSizePolicy(sizePolicy) self.canvas.setAlignment(QtCore.Qt.AlignCenter) self.canvas.setAutoFillBackground(False) self.canvas.setStyleSheet("background-color: White") self.canvas.setObjectName("canvas") self.scrollArea.setWidget(self.canvas) ## ----- ## -- 工作区 -- self.p_create_dock = partial(self.creat_dock, MainWindow) self.p_create_button = partial(self.create_button, self.CentralWidget) # 模型加载 widget = QtWidgets.QWidget() horizontalLayout = QtWidgets.QHBoxLayout(widget) self.ModelDock = self.p_create_dock("ModelDock", self.tr("模型选择"), widget) ModelRegion = QtWidgets.QVBoxLayout() ModelRegion.setObjectName("ModelRegion") # labShowSet = self.create_text(CentralWidget, "labShowSet", "模型选择") # ModelRegion.addWidget(labShowSet) # combo = QtWidgets.QComboBox(self) # # combo.addItems([self.tr(ModelsNick[m.__name__][0]) for m in MODELS]) # combo.addItems([self.tr(ModelsNick[m][0]) for m in ModelsNick.keys()]) # self.comboModelSelect = combo # ModelRegion.addWidget(self.comboModelSelect) # 网络参数 self.btnParamsSelect = self.p_create_button( "btnParamsLoad", self.tr("加载网络参数"), osp.join(pjpath, "resource/Model.png"), "Ctrl+D", ) ModelRegion.addWidget(self.btnParamsSelect) # 模型选择 self.cheWithMask = QtWidgets.QCheckBox(self) self.cheWithMask.setText(self.tr("使用掩膜")) self.cheWithMask.setChecked(True) ModelRegion.addWidget(self.cheWithMask) # with_mask ModelRegion.addSpacerItem( QtWidgets.QSpacerItem(0, 0, QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Expanding)) horizontalLayout.addLayout(ModelRegion) MainWindow.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.ModelDock) # 数据列表 # TODO: 数据列表加一个搜索功能 widget = QtWidgets.QWidget() horizontalLayout = QtWidgets.QHBoxLayout(widget) ListRegion = QtWidgets.QVBoxLayout() ListRegion.setObjectName("ListRegion") # labFiles = self.create_text(CentralWidget, "labFiles", "数据列表") # ListRegion.addWidget(labFiles) self.listFiles = QtWidgets.QListWidget(self.CentralWidget) self.listFiles.setObjectName("ListFiles") ListRegion.addWidget(self.listFiles) # ListRegion.addWidget(self.btnSave) horizontalLayout.addLayout(ListRegion) self.DataDock = self.p_create_dock("DataDock", self.tr("数据列表"), widget) MainWindow.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.DataDock) # 标签列表 widget = QtWidgets.QWidget() horizontalLayout = QtWidgets.QHBoxLayout(widget) LabelRegion = QtWidgets.QVBoxLayout() LabelRegion.setObjectName("LabelRegion") self.labelListTable = TableWidget( self.CentralWidget) # QtWidgets.QTableWidget(CentralWidget) self.labelListTable.horizontalHeader().hide() # 铺满 self.labelListTable.horizontalHeader().setSectionResizeMode( QtWidgets.QHeaderView.Stretch) self.labelListTable.verticalHeader().hide() self.labelListTable.setColumnWidth(0, 10) # self.labelListTable.setMinimumWidth() self.labelListTable.setObjectName("labelListTable") self.labelListTable.clearContents() self.labelListTable.setRowCount(0) self.labelListTable.setColumnCount(4) LabelRegion.addWidget(self.labelListTable) self.btnAddClass = self.p_create_button( "btnAddClass", self.tr("添加标签"), osp.join(pjpath, "resource/Label.png")) LabelRegion.addWidget(self.btnAddClass) horizontalLayout.addLayout(LabelRegion) self.LabelDock = self.p_create_dock("LabelDock", self.tr("标签列表"), widget) MainWindow.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.LabelDock) ## 滑块设置 # 分割阈值 self.p_create_slider = partial(self.create_slider, self.CentralWidget) widget = QtWidgets.QWidget() horizontalLayout = QtWidgets.QHBoxLayout(widget) ShowSetRegion = QtWidgets.QVBoxLayout() ShowSetRegion.setObjectName("ShowSetRegion") self.sldThresh, _, SegShowRegion = self.p_create_slider( "sldThresh", "labThresh", self.tr("分割阈值：")) ShowSetRegion.addLayout(SegShowRegion) ShowSetRegion.addWidget(self.sldThresh) # 透明度 self.sldOpacity, _, MaskShowRegion = self.p_create_slider( "sldOpacity", "labOpacity", self.tr("标签透明度："), 75) ShowSetRegion.addLayout(MaskShowRegion) ShowSetRegion.addWidget(self.sldOpacity) # 点大小 self.sldClickRadius, _, PointShowRegion = self.p_create_slider( "sldClickRadius", "labClickRadius", self.tr("点击可视化半径："), 3, 10, 0, 1) ShowSetRegion.addLayout(PointShowRegion) ShowSetRegion.addWidget(self.sldClickRadius) # 保存 self.btnSave = self.p_create_button( "btnSave", self.tr("保存"), osp.join(pjpath, "resource/Save.png"), "Ctrl+S", ) ShowSetRegion.addWidget(self.btnSave) ShowSetRegion.addSpacerItem( QtWidgets.QSpacerItem(0, 0, QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Expanding)) horizontalLayout.addLayout(ShowSetRegion) self.SegSettingDock = self.p_create_dock("SegSettingDock", self.tr("分割设置"), widget) MainWindow.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.SegSettingDock) ## 专业功能区工作区 ## 遥感影像设置 widget = QtWidgets.QWidget() horizontalLayout = QtWidgets.QHBoxLayout(widget) self.RSDock = self.p_create_dock("RSDock", self.tr("遥感设置"), widget) bandRegion = QtWidgets.QVBoxLayout() bandRegion.setObjectName("bandRegion") bandRegion.addWidget( create_text(self.CentralWidget, "bandSelection", self.tr("波段设置"))) text_list = ["R", "G", "B"] self.bandCombos = [] for txt in text_list: combo = QtWidgets.QComboBox() combo.addItems(["band_1"]) self.bandCombos.append(combo) hbandLayout = QtWidgets.QHBoxLayout() hbandLayout.setObjectName("hbandLayout") hbandLayout.addWidget( create_text(self.CentralWidget, "band" + txt, txt)) hbandLayout.addWidget(combo) hbandLayout.setStretch(1, 4) bandRegion.addLayout(hbandLayout) bandRegion.addWidget( create_text(self.CentralWidget, "resultSave", self.tr("保存设置"))) self.boundaryRegular = QtWidgets.QCheckBox(self.tr("建筑边界规范化")) self.boundaryRegular.setObjectName("boundaryRegular") bandRegion.addWidget(self.boundaryRegular) self.shpSave = QtWidgets.QCheckBox(self.tr("另存为shapefile")) self.shpSave.setObjectName("shpSave") bandRegion.addWidget(self.shpSave) horizontalLayout.addLayout(bandRegion) bandRegion.addWidget( create_text(self.CentralWidget, "showGeoInfo", self.tr("地理信息"))) self.edtGeoinfo = QtWidgets.QTextEdit(self.tr("无")) self.edtGeoinfo.setObjectName("edtGeoinfo") self.edtGeoinfo.setReadOnly(True) bandRegion.addWidget(self.edtGeoinfo) MainWindow.addDockWidget(QtCore.Qt.LeftDockWidgetArea, self.RSDock) ## 医学影像设置 widget = QtWidgets.QWidget() horizontalLayout = QtWidgets.QHBoxLayout(widget) MIRegion = QtWidgets.QVBoxLayout() MIRegion.setObjectName("MIRegion") # mi_text = create_text(CentralWidget, "sliceSelection", self.tr("切片选择")) # MIRegion.addWidget(mi_text) # self.sldMISlide, _, slideRegion = p_create_slider( # "sldMISlide", "labMISlide", self.tr("切片选择："), 1, 1, 1 # ) # self.sldMISlide.setMinimum(1) # MIRegion.addLayout(slideRegion) # MIRegion.addWidget(self.sldMISlide) self.sldWw, self.textWw, WwRegion = self.p_create_slider( "sldWw", "textWw", self.tr("窗宽："), 200, 2048, -2048, 1, True) MIRegion.addLayout(WwRegion) MIRegion.addWidget(self.sldWw) self.sldWc, self.textWc, WcRegion = self.p_create_slider( "sldWc", "textWc", self.tr("窗位："), 0, 2048, -2048, 1, True) MIRegion.addLayout(WcRegion) MIRegion.addWidget(self.sldWc) MIRegion.addSpacerItem( QtWidgets.QSpacerItem(0, 0, QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Expanding)) horizontalLayout.addLayout(MIRegion) self.MedDock = self.p_create_dock("MedDock", self.tr("医疗设置"), widget) MainWindow.addDockWidget(QtCore.Qt.LeftDockWidgetArea, self.MedDock) ## 宫格区域设置 widget = QtWidgets.QWidget() horizontalLayout = QtWidgets.QHBoxLayout(widget) self.GridDock = self.p_create_dock("GridDock", self.tr("宫格切换"), widget) GridRegion = QtWidgets.QVBoxLayout() GridRegion.setObjectName("GridRegion") # self.btnInitGrid = p_create_button( # "btnInitGrid", # self.tr("创建宫格"), # osp.join(pjpath, "resource/N2.png"), # "", # ) self.btnFinishedGrid = self.p_create_button( "btnFinishedGrid", self.tr("完成宫格"), osp.join(pjpath, "resource/Save.png"), "", ) hbandLayout = QtWidgets.QHBoxLayout() hbandLayout.setObjectName("hbandLayout") # hbandLayout.addWidget(self.btnInitGrid) hbandLayout.addWidget(self.btnFinishedGrid) GridRegion.addLayout(hbandLayout) # 创建宫格 self.cheSaveEvery = QtWidgets.QCheckBox(self) self.cheSaveEvery.setText(self.tr("保存每个宫格的标签")) self.cheSaveEvery.setChecked(False) GridRegion.addWidget(self.cheSaveEvery) self.gridTable = QtWidgets.QTableWidget(self.CentralWidget) self.gridTable.horizontalHeader().hide() self.gridTable.verticalHeader().hide() # 铺满 self.gridTable.horizontalHeader().setSectionResizeMode( QtWidgets.QHeaderView.Stretch) self.gridTable.verticalHeader().setSectionResizeMode( QtWidgets.QHeaderView.Stretch) self.gridTable.setObjectName("gridTable") self.gridTable.clearContents() self.gridTable.setColumnCount(1) self.gridTable.setRowCount(1) GridRegion.addWidget(self.gridTable) horizontalLayout.addLayout(GridRegion) MainWindow.addDockWidget(QtCore.Qt.LeftDockWidgetArea, self.GridDock) ## 视频设置 widget = QtWidgets.QWidget() horizontalLayout = QtWidgets.QHBoxLayout(widget) self.VSTDock = self.p_create_dock("VSTDock", self.tr("视频分割设置"), widget) VSTDock = QtWidgets.QVBoxLayout() VSTDock.setObjectName("VSTDock") self.btn3DParamsSelect = self.p_create_button( "btn3DParamsLoad", self.tr("加载传播网络参数"), osp.join(pjpath, "resource/Model.png"), "", ) VSTDock.addWidget(self.btn3DParamsSelect) self.btnPropagate = self.p_create_button( "btnPropagate", self.tr("传播"), osp.join(pjpath, "resource/Propagate.png"), "", ) self.proPropagete = QtWidgets.QProgressBar(self.CentralWidget) self.proPropagete.setAlignment(QtCore.Qt.AlignVCenter) self.proPropagete.setValue(0) VSTDock.addWidget(self.btnPropagate) proLayer = QtWidgets.QHBoxLayout() proLayer.addWidget( create_text(self.CentralWidget, None, self.tr("进度："))) proLayer.addWidget(self.proPropagete) VSTDock.addLayout(proLayer) VSTDock.addSpacerItem( QtWidgets.QSpacerItem(0, 0, QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Expanding)) horizontalLayout.addLayout(VSTDock) MainWindow.addDockWidget(QtCore.Qt.LeftDockWidgetArea, self.VSTDock) # 3d显示 widget = QtWidgets.QWidget() horizontalLayout = QtWidgets.QHBoxLayout(widget) self.TDDock = self.p_create_dock("TDDock", self.tr("3D显示"), widget) TDRegion = QtWidgets.QVBoxLayout() TDRegion.setObjectName("TDRegion") self.vtkWidget = VTKWidget(self.CentralWidget) TDRegion.addWidget(self.vtkWidget) horizontalLayout.addLayout(TDRegion) MainWindow.addDockWidget(QtCore.Qt.LeftDockWidgetArea, self.TDDock) # 视频播放功能区域 widget = QtWidgets.QWidget() verticalLayout = QtWidgets.QVBoxLayout(widget) self.VideoDock = self.p_create_dock("VideoDock", self.tr("时间轴"), widget) VideoRegion = QtWidgets.QHBoxLayout() VideoRegion.setObjectName("VideoRegion") self.videoPlay = self.p_create_button( "videoPlay", self.tr("播放"), osp.join(pjpath, "resource/Play.png"), "", ) height = self.videoPlay.height() min_size = QtCore.QSize(0, height) VideoRegion.addWidget(self.videoPlay) self.sldTime, self.textTime, _ = self.p_create_slider( "sldTime", "textTime", "", 0, 100, 0, 1, True) self.textTime.setMinimumSize(min_size) self.sldTime.setTickInterval(1) self.sldTime.setTickPosition(QtWidgets.QSlider.TicksBelow) VideoRegion.addWidget(self.sldTime) VideoRegion.addWidget( create_text(self.CentralWidget, None, self.tr("帧数："))) VideoRegion.addWidget(self.textTime) self.preFrameButton = self.p_create_button( "preFrameButton", self.tr("上一帧"), ) self.nextFrameButton = self.p_create_button( "nextFrameButton", self.tr("下一帧"), ) # self.preFrameButton = QtWidgets.QPushButton("上一帧") # self.nextFrameButton = QtWidgets.QPushButton("下一帧") VideoRegion.addWidget(self.preFrameButton) VideoRegion.addWidget(self.nextFrameButton) self.speedComboBox = QtWidgets.QComboBox() self.speedComboBox.setMinimumSize(min_size) v_txt = self.tr("倍速: ") self.speedComboBox.addItems([ v_txt + "0.25x", v_txt + "0.5x", v_txt + "1.0x", v_txt + "2.0x", v_txt + "3.0x", v_txt + "5.0x", v_txt + "10.0x", ]) self.speedComboBox.setCurrentIndex(2) VideoRegion.addWidget(self.speedComboBox) verticalLayout.addLayout(VideoRegion) self.VideoDock.setAllowedAreas(QtCore.Qt.BottomDockWidgetArea) MainWindow.addDockWidget(QtCore.Qt.BottomDockWidgetArea, self.VideoDock) # timer self.timer = QTimer() self.timer.setSingleShot(False) ## ----- QtCore.QMetaObject.connectSlotsByName(MainWindow) # log.debug("Set up UI finished") ## 创建文本 def create_text(self, parent, text_name=None, text_text=None): return create_text(parent, text_name, text_text) ## 创建按钮 def create_button(self, parent, btn_name, btn_text, ico_path=None, curt=None): return create_button(parent, btn_name, btn_text, ico_path, curt) ## 创建dock def creat_dock(self, parent, name, text, layout): return creat_dock(parent, name, text, layout) ## 显示Logo def show_logo(self, logo_path): labLogo = QtWidgets.QLabel() sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Maximum, QtWidgets.QSizePolicy.Maximum) labLogo.setSizePolicy(sizePolicy) labLogo.setMaximumSize(QtCore.QSize(100, 33)) labLogo.setPixmap(QtGui.QPixmap(logo_path)) labLogo.setScaledContents(True) labLogo.setObjectName("labLogo") return labLogo ## 创建滑块区域 def create_slider(self, parent, sld_name, text_name, text, default_value=50, max_value=100, min_value=0, text_rate=0.01, edit=False): return create_slider(parent, sld_name, text_name, text, default_value, max_value, min_value, text_rate, edit)