from functools import partial from threading import Thread from rscder.plugins.misc.main import AlgFrontend from rscder.gui.actions import ActionManager from rscder.plugins.basic import BasicPlugin from PyQt5.QtWidgets import QAction, QToolBar, QMenu, QDialog, QHBoxLayout, QVBoxLayout, QPushButton, QWidget, QLabel, QLineEdit, QPushButton, QComboBox, QDialogButtonBox from rscder.gui.layercombox import PairLayerCombox from rscder.utils.icons import IconInstance from rscder.plugins.filter_collection import FILTER from .scripts import VEG_CD from rscder.plugins.thres import THRES from rscder.plugins.misc import table_layer, AlgSelectWidget from rscder.plugins.follow import FOLLOW import os class VegtationCDMethod(QDialog): def __init__(self, parent=None, alg: AlgFrontend = None): super(VegtationCDMethod, self).__init__(parent) self.alg = alg self.setWindowTitle('植被变化检测:{}'.format(alg.get_name())) self.setWindowIcon(IconInstance().LOGO) self.initUI() self.setMinimumWidth(500) def initUI(self): # 图层 self.layer_combox = PairLayerCombox(self) layerbox = QHBoxLayout() layerbox.addWidget(self.layer_combox) self.filter_select = AlgSelectWidget(self, FILTER) self.param_widget = self.alg.get_widget(self) self.unsupervised_menu = self.param_widget self.thres_select = AlgSelectWidget(self, THRES) self.ok_button = QPushButton('确定', self) self.ok_button.setIcon(IconInstance().OK) self.ok_button.clicked.connect(self.accept) self.ok_button.setDefault(True) self.cancel_button = QPushButton('取消', self) self.cancel_button.setIcon(IconInstance().CANCEL) self.cancel_button.clicked.connect(self.reject) self.cancel_button.setDefault(False) buttonbox = QDialogButtonBox(self) buttonbox.addButton(self.ok_button, QDialogButtonBox.NoRole) buttonbox.addButton(self.cancel_button, QDialogButtonBox.NoRole) buttonbox.setCenterButtons(True) totalvlayout = QVBoxLayout() totalvlayout.addLayout(layerbox) totalvlayout.addWidget(self.filter_select) if self.param_widget is not None: totalvlayout.addWidget(self.param_widget) totalvlayout.addWidget(self.thres_select) totalvlayout.addStretch(1) hbox = QHBoxLayout() hbox.addStretch(1) hbox.addWidget(buttonbox) totalvlayout.addLayout(hbox) # totalvlayout.addStretch() self.setLayout(totalvlayout) @FOLLOW.register class VegetationCDFollow(AlgFrontend): @staticmethod def get_name(): return '植被变化检测' @staticmethod def get_icon(): return IconInstance().VEGETATION @staticmethod def get_widget(parent=None): widget = QWidget(parent) layer_combox = PairLayerCombox(widget) layer_combox.setObjectName('layer_combox') filter_select = AlgSelectWidget(widget, FILTER) filter_select.setObjectName('filter_select') unsupervised_select = AlgSelectWidget(widget, VEG_CD) unsupervised_select.setObjectName('unsupervised_select') thres_select = AlgSelectWidget(widget, THRES) thres_select.setObjectName('thres_select') totalvlayout = QVBoxLayout() totalvlayout.addWidget(layer_combox) totalvlayout.addWidget(filter_select) totalvlayout.addWidget(unsupervised_select) totalvlayout.addWidget(thres_select) totalvlayout.addStretch() widget.setLayout(totalvlayout) return widget @staticmethod def get_params(widget: QWidget = None): if widget is None: return dict() layer_combox = widget.findChild(PairLayerCombox, 'layer_combox') filter_select = widget.findChild(AlgSelectWidget, 'filter_select') unsupervised_select = widget.findChild( AlgSelectWidget, 'unsupervised_select') thres_select = widget.findChild(AlgSelectWidget, 'thres_select') layer1 = layer_combox.layer1 pth1 = layer_combox.layer1.path pth2 = layer_combox.layer2.path falg, fparams = filter_select.get_alg_and_params() cdalg, cdparams = unsupervised_select.get_alg_and_params() thalg, thparams = thres_select.get_alg_and_params() if cdalg is None or thalg is None: return dict() return dict( layer1=layer1, pth1=pth1, pth2=pth2, falg=falg, fparams=fparams, cdalg=cdalg, cdparams=cdparams, thalg=thalg, thparams=thparams, ) @staticmethod def run_alg(layer1=None, pth1=None, pth2=None, falg=None, fparams=None, cdalg=None, cdparams=None, thalg=None, thparams=None, send_message=None): if cdalg is None or thalg is None: return name = layer1.name method_info = dict() if falg is not None: pth1 = falg.run_alg( pth1, name=name, send_message=send_message, **fparams) pth2 = falg.run_alg( pth2, name=name, send_message=send_message, **fparams) method_info['滤波算法'] = falg.get_name() if falg is not None: try: os.remove(pth1) os.remove(pth2) # send_message.emit('删除临时文件') except: # send_message.emit('删除临时文件失败！') pass cdpth = cdalg.run_alg(pth1, pth2, layer1.layer_parent, send_message=send_message, **cdparams) if falg is not None: try: os.remove(pth1) os.remove(pth2) # send_message.emit('删除临时文件') except: # send_message.emit('删除临时文件失败！') pass if cdpth is None: return thpth, th = thalg.run_alg( cdpth, name=name, send_message=send_message, **thparams) method_info['变化检测算法'] = cdalg.get_name() method_info['二值化算法'] = thalg.get_name() # table_layer(thpth, layer1, name, send_message) table_layer(thpth, layer1, name, cdpath=cdpth, th=th, method_info=method_info, send_message=send_message) class VegtationPlugin(BasicPlugin): @staticmethod def info(): return { 'name': 'VegtationPlugin', 'description': 'VegtationPlugin', 'author': 'RSCDER', 'version': '1.0.0', } def set_action(self): veg_menu = ActionManager().veg_menu veg_toolbar = ActionManager().add_toolbar('Veg Change') # veg_menu.setIcon(IconInstance().UNSUPERVISED) # ActionManager().veg_menu.addMenu(veg_menu) for key in VEG_CD.keys(): alg: AlgFrontend = VEG_CD[key] if alg.get_name() is None: name = key else: name = alg.get_name() action = QAction(name, veg_menu) func = partial(self.run_cd, alg) action.triggered.connect(func) veg_toolbar.addAction(action) veg_menu.addAction(action) def run_cd(self, alg): dialog = VegtationCDMethod(self.mainwindow, alg) dialog.show() if dialog.exec_() == QDialog.Accepted: t = Thread(target=self.run_cd_alg, args=(dialog,)) t.start() def run_cd_alg(self, w: VegtationCDMethod): layer1 = w.layer_combox.layer1 pth1 = w.layer_combox.layer1.path pth2 = w.layer_combox.layer2.path name = layer1.layer_parent.name falg, fparams = w.filter_select.get_alg_and_params() cdalg = w.alg cdparams = w.alg.get_params(w.param_widget) thalg, thparams = w.thres_select.get_alg_and_params() if cdalg is None or thalg is None: return method_info = dict() if falg is not None: pth1 = falg.run_alg( pth1, name=name, send_message=self.send_message, **fparams) pth2 = falg.run_alg( pth2, name=name, send_message=self.send_message, **fparams) method_info['滤波算法'] = falg.get_name() cdpth = cdalg.run_alg(pth1, pth2, layer1.layer_parent, send_message=self.send_message, **cdparams) if cdpth is None: return if falg is not None: try: os.remove(pth1) os.remove(pth2) # send_message.emit('删除临时文件') except: # send_message.emit('删除临时文件失败！') pass thpth, th = thalg.run_alg( cdpth, name=name, send_message=self.send_message, **thparams) method_info['变化检测算法'] = cdalg.get_name() method_info['二值化算法'] = thalg.get_name() # table_layer(thpth, layer1, name, self.send_message) table_layer(thpth, layer1, name, cdpath=cdpth, th=th, method_info=method_info, send_message=self.send_message)