#!/usr/bin/env python # -*- coding: utf-8 -*- """ Step3 面板 - 耀斑去除 """ import os from PyQt5.QtWidgets import ( QWidget, QVBoxLayout, QGroupBox, QFormLayout, QDoubleSpinBox, QSpinBox, QComboBox, QCheckBox, QPushButton, QLabel, QLineEdit, QMessageBox, ) from PyQt5.QtCore import Qt # 从公共组件库导入 from src.gui.components.custom_widgets import FileSelectWidget from src.gui.styles import ModernStylesheet class Step3Panel(QWidget): """步骤3：耀斑去除""" def __init__(self, parent=None): super().__init__(parent) self.init_ui() def init_ui(self): layout = QVBoxLayout() # 标题 # 影像文件 self.img_file = FileSelectWidget( "影像文件:", "Image Files (*.bsq *.dat *.tif);;All Files (*.*)" ) layout.addWidget(self.img_file) # 水域掩膜/边界：完整流程可由步骤1自动生成；独立单步运行时须手动指定 self.water_mask_file = FileSelectWidget( "水域掩膜/边界:", "Mask/Boundary (*.dat *.tif *.shp);;All Files (*.*)" ) layout.addWidget(self.water_mask_file) step3_mask_hint = QLabel( "提示：独立运行本步骤时必须选择水域掩膜或边界（与影像同区域的 .dat/.tif 掩膜，或 .shp 矢量）。" ) step3_mask_hint.setWordWrap(True) step3_mask_hint.setStyleSheet("color: #666; font-size: 10px;") layout.addWidget(step3_mask_hint) # 方法选择 method_group = QGroupBox("去耀斑方法") method_layout = QVBoxLayout() self.method = QComboBox() for text, data in [('Goodman方法', 'goodman'), ('Kutser方法', 'kutser'), ('Hedley方法', 'hedley'), ('SUGAR算法', 'sugar')]: self.method.addItem(text, data) self.method.currentIndexChanged.connect(self._on_method_changed) method_layout.addWidget(self.method) method_group.setLayout(method_layout) layout.addWidget(method_group) # Goodman参数组 self.goodman_group = QGroupBox("Goodman方法参数") goodman_layout = QFormLayout() self.nir_lower = QSpinBox() self.nir_lower.setRange(0, 200) self.nir_lower.setValue(65) goodman_layout.addRow("NIR下波段索引:", self.nir_lower) self.nir_upper = QSpinBox() self.nir_upper.setRange(0, 200) self.nir_upper.setValue(91) goodman_layout.addRow("NIR上波段索引:", self.nir_upper) self.goodman_a = QDoubleSpinBox() self.goodman_a.setDecimals(6) self.goodman_a.setRange(0, 1) self.goodman_a.setValue(0.000019) goodman_layout.addRow("参数A:", self.goodman_a) self.goodman_b = QDoubleSpinBox() self.goodman_b.setDecimals(2) self.goodman_b.setRange(0, 1) self.goodman_b.setValue(0.1) goodman_layout.addRow("参数B:", self.goodman_b) self.goodman_group.setLayout(goodman_layout) layout.addWidget(self.goodman_group) # Kutser参数组 self.kutser_group = QGroupBox("Kutser方法参数") kutser_layout = QFormLayout() self.oxy_band = QSpinBox() self.oxy_band.setRange(0, 200) self.oxy_band.setValue(8) kutser_layout.addRow("氧吸收波段索引:", self.oxy_band) self.lower_oxy = QDoubleSpinBox() self.lower_oxy.setDecimals(2) self.lower_oxy.setRange(0, 1000) self.lower_oxy.setValue(756.54) kutser_layout.addRow("下氧吸收波长(nm):", self.lower_oxy) self.upper_oxy = QDoubleSpinBox() self.upper_oxy.setDecimals(2) self.upper_oxy.setRange(0, 1000) self.upper_oxy.setValue(766.54) kutser_layout.addRow("上氧吸收波长(nm):", self.upper_oxy) self.nir_band = QSpinBox() self.nir_band.setRange(0, 200) self.nir_band.setValue(65) kutser_layout.addRow("NIR波段索引:", self.nir_band) self.kutser_group.setLayout(kutser_layout) self.kutser_group.setVisible(False) layout.addWidget(self.kutser_group) # Hedley参数组 self.hedley_group = QGroupBox("Hedley方法参数") hedley_layout = QFormLayout() self.hedley_nir_band = QSpinBox() self.hedley_nir_band.setRange(0, 200) self.hedley_nir_band.setValue(47) hedley_layout.addRow("NIR波段索引:", self.hedley_nir_band) self.hedley_group.setLayout(hedley_layout) self.hedley_group.setVisible(False) layout.addWidget(self.hedley_group) # SUGAR参数组 self.sugar_group = QGroupBox("SUGAR方法参数") sugar_layout = QFormLayout() self.sugar_iter = QSpinBox() self.sugar_iter.setRange(1, 20) self.sugar_iter.setValue(3) self.sugar_iter.setSpecialValueText("自动") sugar_layout.addRow("迭代次数:", self.sugar_iter) self.sugar_sigma = QDoubleSpinBox() self.sugar_sigma.setDecimals(2) self.sugar_sigma.setRange(0.1, 10) self.sugar_sigma.setValue(1.0) sugar_layout.addRow("LoG平滑σ:", self.sugar_sigma) self.sugar_estimate_background = QCheckBox() self.sugar_estimate_background.setChecked(True) sugar_layout.addRow("估计背景光谱:", self.sugar_estimate_background) self.sugar_glint_mask_method = QComboBox() self.sugar_glint_mask_method.addItems(['cdf', 'otsu']) self.sugar_glint_mask_method.setCurrentText('cdf') sugar_layout.addRow("耀斑掩膜方法:", self.sugar_glint_mask_method) self.sugar_termination_thresh = QDoubleSpinBox() self.sugar_termination_thresh.setDecimals(2) self.sugar_termination_thresh.setRange(1, 100) self.sugar_termination_thresh.setValue(20.0) sugar_layout.addRow("终止阈值:", self.sugar_termination_thresh) self.sugar_bounds = QLineEdit() self.sugar_bounds.setText("[(1, 2)]") sugar_layout.addRow("优化边界:", self.sugar_bounds) self.sugar_group.setLayout(sugar_layout) self.sugar_group.setVisible(False) layout.addWidget(self.sugar_group) # 插值选项 interp_group = QGroupBox("0值像素插值") interp_layout = QFormLayout() self.interpolate_zeros = QCheckBox("启用插值") interp_layout.addRow("", self.interpolate_zeros) self.interp_method = QComboBox() for text, data in [('最近邻插值', 'nearest'), ('双线性插值', 'bilinear'), ('样条插值', 'spline'), ('克里金插值', 'kriging')]: self.interp_method.addItem(text, data) self.interp_method.setCurrentIndex(1) # 默认双线性插值 interp_layout.addRow("插值方法:", self.interp_method) interp_group.setLayout(interp_layout) layout.addWidget(interp_group) # # 实测经纬度参考点 # self.ref_csv_file = FileSelectWidget( # "实测经纬度CSV:", # "CSV Files (*.csv);;All Files (*.*)" # ) # self.ref_csv_file.line_edit.setPlaceholderText("可选：包含 Lon/Lat 列的 CSV 文件") # layout.addWidget(self.ref_csv_file) # 交互式预览按钮 # self.preview_btn = QPushButton("👁️ 打开交互式影像预览") # self.preview_btn.setStyleSheet(ModernStylesheet.get_button_stylesheet('info')) # self.preview_btn.clicked.connect(self.open_interactive_viewer) # layout.addWidget(self.preview_btn) # 输出文件路径 self.output_file = FileSelectWidget( "输出影像:", "Image Files (*.bsq *.dat *.tif);;All Files (*.*)" ) self.output_file.line_edit.setPlaceholderText("deglint_image.dat") layout.addWidget(self.output_file) # 启用步骤 self.enable_checkbox = QCheckBox("启用此步骤") self.enable_checkbox.setChecked(True) layout.addWidget(self.enable_checkbox) # 独立运行按钮 self.run_btn = QPushButton("独立运行此步骤") self.run_btn.setStyleSheet(ModernStylesheet.get_button_stylesheet('success')) self.run_btn.clicked.connect(self.run_step) layout.addWidget(self.run_btn) layout.addStretch() self.setLayout(layout) # 信号连接：影像文件路径变化时动态更新波段范围 self.img_file.line_edit.textChanged.connect(self._update_band_ranges) def open_interactive_viewer(self): """打开交互式影像预览""" from src.gui.water_quality_gui import InteractiveViewerDialog img_path = self.img_file.get_path() if not img_path or not os.path.isfile(img_path): QMessageBox.warning(self, "警告", "请先选择影像文件！") return water_mask = self.water_mask_file.get_path() dialog = InteractiveViewerDialog(img_path, self) if water_mask and os.path.isfile(water_mask): dialog.load_water_mask(water_mask) dialog.exec_() def _update_band_ranges(self, file_path): """根据选择的影像动态限制波段索引的输入范围""" from osgeo import gdal if not file_path or not os.path.isfile(file_path): return try: dataset = gdal.Open(file_path) if dataset is None: return raster_count = dataset.RasterCount max_band = max(0, raster_count - 1) self.nir_lower.setMaximum(max_band) self.nir_upper.setMaximum(max_band) self.oxy_band.setMaximum(max_band) self.nir_band.setMaximum(max_band) self.hedley_nir_band.setMaximum(max_band) dataset = None except Exception: pass def update_from_config(self, work_dir=None, pipeline=None): """ 从 Step1Panel 自动填充水域掩膜路径，实现上下游数据流转 Args: work_dir: 工作目录路径 pipeline: Pipeline 实例（未使用，保留接口兼容性） """ # 保存工作目录引用 if work_dir: self.work_dir = work_dir elif hasattr(self, 'work_dir') and self.work_dir: pass # 保持现有工作目录 else: self.work_dir = None # 从 Step1 界面读取水域掩膜路径 main_window = self.window() if hasattr(main_window, 'step1_panel'): if main_window.step1_panel.use_ndwi_radio.isChecked(): # NDWI模式，读取输出框的路径 mask_path = main_window.step1_panel.output_file.get_path() else: # 导入现有模式，读取输入框的路径 mask_path = main_window.step1_panel.mask_file.get_path() if mask_path: # 若为相对路径，使用 work_dir 合成为绝对路径 if not os.path.isabs(mask_path): mask_path = os.path.join(self.work_dir or '', mask_path).replace('\\', '/') self.water_mask_file.set_path(mask_path) # 自动填充输出路径（基于工作目录） if self.work_dir: output_dir = os.path.join(self.work_dir, "3_deglint") os.makedirs(output_dir, exist_ok=True) default_output_path = os.path.join(output_dir, "deglint_image.dat").replace('\\', '/') self.output_file.set_path(default_output_path) else: self.output_file.set_path("") def _on_method_changed(self, index): """方法改变时更新参数显示""" method_id = self.method.currentData() self.goodman_group.setVisible(method_id == 'goodman') self.kutser_group.setVisible(method_id == 'kutser') self.hedley_group.setVisible(method_id == 'hedley') self.sugar_group.setVisible(method_id == 'sugar') def get_config(self): """获取配置""" config = { 'img_path': self.img_file.get_path(), 'method': self.method.currentData(), # 使用 currentData() 获取英文ID 'enabled': self.enable_checkbox.isChecked(), 'interpolate_zeros': self.interpolate_zeros.isChecked(), 'interpolation_method': self.interp_method.currentData(), # 使用 currentData() } water_mask_path = self.water_mask_file.get_path() if water_mask_path: config['water_mask'] = water_mask_path output_path = self.output_file.get_path() if output_path: config['output_path'] = output_path method = self.method.currentData() # 使用 currentData() if method == 'goodman': config['nir_lower'] = self.nir_lower.value() config['nir_upper'] = self.nir_upper.value() config['goodman_A'] = self.goodman_a.value() config['goodman_B'] = self.goodman_b.value() elif method == 'kutser': config['oxy_band'] = self.oxy_band.value() config['lower_oxy'] = self.lower_oxy.value() config['upper_oxy'] = self.upper_oxy.value() config['nir_band'] = self.nir_band.value() elif method == 'hedley': config['hedley_nir_band'] = self.hedley_nir_band.value() elif method == 'sugar': config['sugar_iter'] = self.sugar_iter.value() if self.sugar_iter.value() > 0 else None config['sugar_sigma'] = self.sugar_sigma.value() config['sugar_estimate_background'] = self.sugar_estimate_background.isChecked() config['sugar_glint_mask_method'] = self.sugar_glint_mask_method.currentData() config['sugar_termination_thresh'] = self.sugar_termination_thresh.value() # 解析bounds字符串 try: import ast config['sugar_bounds'] = ast.literal_eval(self.sugar_bounds.text()) except: config['sugar_bounds'] = [(1, 2)] # 默认值 return config def set_config(self, config): """设置配置""" if 'img_path' in config: self.img_file.set_path(config['img_path']) if 'water_mask' in config: self.water_mask_file.set_path(config['water_mask']) if 'output_path' in config: self.output_file.set_path(config['output_path']) if 'reference_csv' in config: self.ref_csv_file.set_path(config['reference_csv']) if 'method' in config: idx = self.method.findData(config['method']) # 使用 findData() if idx >= 0: self.method.setCurrentIndex(idx) if 'enabled' in config: self.enable_checkbox.setChecked(config['enabled']) if 'interpolate_zeros' in config: self.interpolate_zeros.setChecked(config['interpolate_zeros']) if 'interpolation_method' in config: idx = self.interp_method.findData(config['interpolation_method']) # 使用 findData() if idx >= 0: self.interp_method.setCurrentIndex(idx) # Goodman参数 if 'nir_lower' in config: self.nir_lower.setValue(config['nir_lower']) if 'nir_upper' in config: self.nir_upper.setValue(config['nir_upper']) if 'goodman_A' in config: self.goodman_a.setValue(config['goodman_A']) if 'goodman_B' in config: self.goodman_b.setValue(config['goodman_B']) # Kutser参数 if 'oxy_band' in config: self.oxy_band.setValue(config['oxy_band']) if 'lower_oxy' in config: self.lower_oxy.setValue(config['lower_oxy']) if 'upper_oxy' in config: self.upper_oxy.setValue(config['upper_oxy']) if 'nir_band' in config: self.nir_band.setValue(config['nir_band']) # Hedley参数 if 'hedley_nir_band' in config: self.hedley_nir_band.setValue(config['hedley_nir_band']) # SUGAR参数 if 'sugar_iter' in config: self.sugar_iter.setValue(config['sugar_iter'] if config['sugar_iter'] is not None else 0) if 'sugar_sigma' in config: self.sugar_sigma.setValue(config['sugar_sigma']) if 'sugar_estimate_background' in config: self.sugar_estimate_background.setChecked(config['sugar_estimate_background']) if 'sugar_glint_mask_method' in config: idx = self.sugar_glint_mask_method.findData(config['sugar_glint_mask_method']) # 使用 findData() if idx >= 0: self.sugar_glint_mask_method.setCurrentIndex(idx) if 'sugar_termination_thresh' in config: self.sugar_termination_thresh.setValue(config['sugar_termination_thresh']) if 'sugar_bounds' in config: self.sugar_bounds.setText(str(config['sugar_bounds'])) def run_step(self): """独立运行步骤3""" # 验证输入 img_path = self.img_file.get_path() if not img_path: QMessageBox.warning(self, "输入错误", "请选择影像文件！") return if self.enable_checkbox.isChecked(): water_mask_path = self.water_mask_file.get_path() if not water_mask_path: QMessageBox.warning( self, "输入错误", "独立运行耀斑去除时，必须选择水域掩膜或边界文件。\n\n" "请提供与当前影像空间一致的水域栅格掩膜（.dat/.tif），或水域矢量边界（.shp）。\n" "若刚跑过完整流程，可使用步骤1生成的水域掩膜文件。", ) return # 获取主窗口并运行步骤 main_window = self.window() if hasattr(main_window, 'run_single_step'): config = {'step3': self.get_config()} main_window.run_single_step('step3', config)