duxingchen/WQ_GUI
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

步骤六页面修改

Browse Source
This commit is contained in:
DXC
2026-06-26 13:56:57 +08:00
parent 5fb2db4a07
commit abbe218608
1 changed files with 112 additions and 102 deletions
Download Patch File Download Diff File Expand all files Collapse all files

214
src/gui/panels/step6_feature_panel.py
View File

@ -1,22 +1,22 @@
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
""" """
Step6 面板 - 光谱特征提取 Step6 面板 - 光谱特征提取 (完美对齐无跳动重构版)
""" """
import os import os
import sys import sys
from pathlib import Path from pathlib import Path
# 路径归一化 helper（与 pipeline.get_step_output_dir 互为表里） # 路径归一化 helper
_HERE = os.path.dirname(os.path.abspath(__file__)) _HERE = os.path.dirname(os.path.abspath(__file__))
if _HERE not in sys.path: if _HERE not in sys.path:
sys.path.insert(0, _HERE) sys.path.insert(0, _HERE)
from src.gui.panels._step_path_resolver import resolve_subdir from src.gui.panels._step_path_resolver import resolve_subdir
from PyQt5.QtWidgets import ( from PyQt5.QtWidgets import (
QWidget, QVBoxLayout, QGroupBox, QFormLayout, QLabel, QWidget, QVBoxLayout, QHBoxLayout, QGroupBox, QFormLayout,
QSpinBox, QPushButton, QCheckBox, QMessageBox, QLabel, QSpinBox, QPushButton, QMessageBox, QSizePolicy
) )
from PyQt5.QtGui import QFont from PyQt5.QtGui import QFont
from PyQt5.QtCore import Qt from PyQt5.QtCore import Qt
@ -27,91 +27,148 @@ from src.gui.styles import ModernStylesheet
class Step6FeaturePanel(QWidget): class Step6FeaturePanel(QWidget):
"""步骤6：光谱特征提取""" """步骤6：光谱特征提取"""
def __init__(self, parent=None): def __init__(self, parent=None):
super().__init__(parent) super().__init__(parent)
self.init_ui() self.init_ui()
def init_ui(self): def init_ui(self):
layout = QVBoxLayout() # 1. 注入全局样式系统
self.setStyleSheet(ModernStylesheet.get_main_stylesheet())
# 标题 # 主布局：增加四周留白(24px)和模块间的呼吸间距(20px)
title = QLabel("步骤6：光谱特征提取") main_layout = QVBoxLayout()
title.setFont(QFont("Arial", 12, QFont.Bold)) main_layout.setContentsMargins(24, 24, 24, 24)
layout.addWidget(title) main_layout.setSpacing(20)
# ==========================================
# 卡片 1：输入数据
# ==========================================
input_group = QGroupBox("📁 输入数据")
input_layout = QVBoxLayout()
input_layout.setSpacing(16)
input_layout.setContentsMargins(20, 24, 20, 20)
step6_glint_hint = QLabel(
"💡 提示：独立运行本步骤时必须选择耀斑掩膜（通常为步骤2输出的 severe_glint_area.dat），用于在采样时自动避开耀斑像元。"
)
step6_glint_hint.setWordWrap(True)
step6_glint_hint.setStyleSheet(f"""
QLabel {{
color: {ModernStylesheet.COLORS['primary']};
background-color: {ModernStylesheet.COLORS['selected']};
border: 1px solid {ModernStylesheet.COLORS['border_light']};
border-radius: 6px;
padding: 10px 14px;
margin-bottom: 4px;
}}
""")
input_layout.addWidget(step6_glint_hint)
# 去耀斑影像文件（用于独立运行）
self.deglint_img_file = FileSelectWidget( self.deglint_img_file = FileSelectWidget(
"去耀斑影像:", "去耀斑影像:",
"Image Files (*.bsq *.dat *.tif);;All Files (*.*)" "Image Files (*.bsq *.dat *.tif);;All Files (*.*)"
) )
layout.addWidget(self.deglint_img_file) self.deglint_img_file.label.setMinimumWidth(100)
# 处理后的CSV文件（用于独立运行）
self.csv_file = FileSelectWidget( self.csv_file = FileSelectWidget(
"处理后CSV:", "处理后CSV:",
"CSV Files (*.csv);;All Files (*.*)" "CSV Files (*.csv);;All Files (*.*)"
) )
layout.addWidget(self.csv_file) self.csv_file.label.setMinimumWidth(100)
# 水体掩膜文件（可选，用于独立运行）
self.water_mask_file = FileSelectWidget( self.water_mask_file = FileSelectWidget(
"水体掩膜:", "水体掩膜:",
"Mask Files (*.dat *.tif);;All Files (*.*)" "Mask Files (*.dat *.tif);;All Files (*.*)"
) )
self.water_mask_file.label.setMinimumWidth(100)
self.water_mask_file.line_edit.setPlaceholderText("可选，如不选择则自动生成") self.water_mask_file.line_edit.setPlaceholderText("可选，如不选择则自动生成")
layout.addWidget(self.water_mask_file)
self.glint_mask_file = FileSelectWidget( self.glint_mask_file = FileSelectWidget(
"耀斑掩膜:", "耀斑掩膜:",
"Mask Files (*.dat *.tif);;All Files (*.*)" "Mask Files (*.dat *.tif);;All Files (*.*)"
) )
layout.addWidget(self.glint_mask_file) self.glint_mask_file.label.setMinimumWidth(100)
step6_glint_hint = QLabel(
"提示：独立运行本步骤时必须选择耀斑掩膜（通常为步骤2输出的 severe_glint_area.dat），用于在采样时避开耀斑像元。"
)
step6_glint_hint.setWordWrap(True)
step6_glint_hint.setStyleSheet("color: #666; font-size: 10px;")
layout.addWidget(step6_glint_hint)
# 参数设置 input_layout.addWidget(self.deglint_img_file)
params_group = QGroupBox("提取参数") input_layout.addWidget(self.csv_file)
input_layout.addWidget(self.water_mask_file)
input_layout.addWidget(self.glint_mask_file)
input_group.setLayout(input_layout)
main_layout.addWidget(input_group)
# ==========================================
# 卡片 2：提取参数
# ==========================================
params_group = QGroupBox("⚙️ 提取参数")
params_layout = QFormLayout() params_layout = QFormLayout()
params_layout.setSpacing(16)
params_layout.setContentsMargins(20, 24, 20, 20)
self.radius = QSpinBox() self.radius = QSpinBox()
self.radius.setRange(1, 50) self.radius.setRange(1, 50)
self.radius.setValue(5) self.radius.setValue(5)
params_layout.addRow("采样半径(像素):", self.radius) self.radius.setSuffix(" px")
self.radius.setButtonSymbols(QSpinBox.NoButtons)
self._add_row_with_fixed_label(params_layout, "采样半径:", self.radius)
self.source_epsg = QSpinBox() self.source_epsg = QSpinBox()
self.source_epsg.setRange(1000, 99999) self.source_epsg.setRange(1000, 99999)
self.source_epsg.setValue(4326) self.source_epsg.setValue(4326)
params_layout.addRow("源坐标系EPSG:", self.source_epsg) self.source_epsg.setButtonSymbols(QSpinBox.NoButtons)
self._add_row_with_fixed_label(params_layout, "源坐标系 EPSG:", self.source_epsg)
params_group.setLayout(params_layout) params_group.setLayout(params_layout)
layout.addWidget(params_group) main_layout.addWidget(params_group)
# ==========================================
# 卡片 3：输出与执行
# ==========================================
output_group = QGroupBox("🚀 输出与执行")
output_layout = QVBoxLayout()
output_layout.setSpacing(16)
output_layout.setContentsMargins(20, 24, 20, 20)
# 输出文件路径
self.output_file = FileSelectWidget( self.output_file = FileSelectWidget(
"输出训练数据:", "结果保存至:",
"CSV Files (*.csv);;All Files (*.*)" "CSV Files (*.csv);;All Files (*.*)",
mode="save"
) )
self.output_file.label.setMinimumWidth(100)
self.output_file.line_edit.setPlaceholderText("training_spectra.csv") self.output_file.line_edit.setPlaceholderText("training_spectra.csv")
layout.addWidget(self.output_file) output_layout.addWidget(self.output_file)
# 启用步骤 action_layout = QHBoxLayout()
self.enable_checkbox = QCheckBox("启用此步骤") action_layout.addStretch()
self.enable_checkbox.setChecked(True)
layout.addWidget(self.enable_checkbox)
# 独立运行按钮 self.run_btn = QPushButton("独立运行步骤")
self.run_btn = QPushButton("独立运行此步骤") self.run_btn.setStyleSheet(ModernStylesheet.get_button_stylesheet('primary'))
self.run_btn.setStyleSheet(ModernStylesheet.get_button_stylesheet('success')) self.run_btn.setMinimumWidth(140)
self.run_btn.clicked.connect(self._on_run_single_clicked) self.run_btn.clicked.connect(self._on_run_single_clicked)
layout.addWidget(self.run_btn) action_layout.addWidget(self.run_btn)
layout.addStretch() output_layout.addLayout(action_layout)
self.setLayout(layout) output_group.setLayout(output_layout)
# 信号连接：影像文件路径变化时动态更新波段范围 main_layout.addWidget(output_group)
main_layout.addStretch()
self.setLayout(main_layout)
def _add_row_with_fixed_label(self, form_layout, label_text, widget):
"""辅助方法：创建绝对锁死宽度的对齐行"""
lbl = QLabel(label_text)
lbl.setMinimumWidth(100)
row_layout = QHBoxLayout()
row_layout.setContentsMargins(0, 0, 0, 0)
row_layout.addWidget(lbl)
if hasattr(widget, 'setSizePolicy'):
widget.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
row_layout.addWidget(widget)
form_layout.addRow(row_layout)
def get_config(self): def get_config(self):
"""获取配置""" """获取配置"""
@ -119,21 +176,22 @@ class Step6FeaturePanel(QWidget):
'radius': self.radius.value(), 'radius': self.radius.value(),
'source_epsg': self.source_epsg.value(), 'source_epsg': self.source_epsg.value(),
} }
# 添加独立运行所需的文件路径
deglint_img_path = self.deglint_img_file.get_path() deglint_img_path = self.deglint_img_file.get_path()
if deglint_img_path: if deglint_img_path:
config['deglint_img_path'] = deglint_img_path config['deglint_img_path'] = deglint_img_path
csv_path = self.csv_file.get_path() csv_path = self.csv_file.get_path()
if csv_path: if csv_path:
config['csv_path'] = csv_path config['csv_path'] = csv_path
water_mask_path = self.water_mask_file.get_path() water_mask_path = self.water_mask_file.get_path()
if water_mask_path: if water_mask_path:
config['boundary_path'] = water_mask_path config['boundary_path'] = water_mask_path
glint_mask_path = self.glint_mask_file.get_path() glint_mask_path = self.glint_mask_file.get_path()
if glint_mask_path: if glint_mask_path:
config['glint_mask_path'] = glint_mask_path config['glint_mask_path'] = glint_mask_path
# 注意：step6_extract_spectra 不接受 output_path / training_csv_path
# 参数，输出路径由 pipeline 内部根据 training_spectra_dir 自动生成。
return config return config
def set_config(self, config): def set_config(self, config):
@ -152,13 +210,7 @@ class Step6FeaturePanel(QWidget):
self.glint_mask_file.set_path(config['glint_mask_path']) self.glint_mask_file.set_path(config['glint_mask_path'])
def update_from_config(self, work_dir=None, pipeline=None): def update_from_config(self, work_dir=None, pipeline=None):
"""从全局配置/Pipeline 或 Step1Panel 自动填充路径，实现上下游数据流转 """从全局配置/Pipeline 自动填充路径，实现上下游数据流转"""
Args:
work_dir: 工作目录路径
pipeline: Pipeline 实例，用于获取步骤1生成的水域掩膜路径
"""
# 保存工作目录引用
if work_dir: if work_dir:
self.work_dir = work_dir self.work_dir = work_dir
elif hasattr(self, 'work_dir') and self.work_dir: elif hasattr(self, 'work_dir') and self.work_dir:
@ -166,55 +218,46 @@ class Step6FeaturePanel(QWidget):
else: else:
self.work_dir = None self.work_dir = None
# 1. 尝试从 Pipeline 获取水体掩膜路径
mask_path = None mask_path = None
if pipeline and hasattr(pipeline, 'water_mask_path') and pipeline.water_mask_path: if pipeline and hasattr(pipeline, 'water_mask_path') and pipeline.water_mask_path:
mask_path = pipeline.water_mask_path mask_path = pipeline.water_mask_path
# 2. 如果 Pipeline 中没有，则尝试直接从 Step1 界面读取
main_window = self.window() main_window = self.window()
if not mask_path and hasattr(main_window, 'step1_panel'): if not mask_path and hasattr(main_window, 'step1_panel'):
if main_window.step1_panel.use_ndwi_radio.isChecked(): if main_window.step1_panel.use_ndwi_radio.isChecked():
mask_path = main_window.step1_panel.output_file.get_path() mask_path = main_window.step1_panel.output_file.get_path()
else: else:
mask_path = main_window.step1_panel.mask_file.get_path() mask_path = main_window.step1_panel.mask_file.get_path()
# 若为相对路径，使用 work_dir 合成为绝对路径
if mask_path and not os.path.isabs(mask_path): if mask_path and not os.path.isabs(mask_path):
mask_path = os.path.join(self.work_dir or '', mask_path).replace('\\', '/') mask_path = os.path.join(self.work_dir or '', mask_path).replace('\\', '/')
# 填充水体掩膜路径
if mask_path: if mask_path:
# 若为相对路径，使用 work_dir 合成为绝对路径
if not os.path.isabs(mask_path): if not os.path.isabs(mask_path):
mask_path = os.path.join(self.work_dir or '', mask_path).replace('\\', '/') mask_path = os.path.join(self.work_dir or '', mask_path).replace('\\', '/')
self.water_mask_file.set_path(mask_path) self.water_mask_file.set_path(mask_path)
# 3. 尝试从 Step2 界面读取耀斑掩膜路径
main_window = self.window()
if hasattr(main_window, 'step2_panel'): if hasattr(main_window, 'step2_panel'):
glint_path = main_window.step2_panel.output_file.get_path() glint_path = main_window.step2_panel.output_file.get_path()
if glint_path: if glint_path:
# 若为相对路径，使用 work_dir 合成为绝对路径
if not os.path.isabs(glint_path): if not os.path.isabs(glint_path):
glint_path = os.path.join(self.work_dir or '', glint_path).replace('\\', '/') glint_path = os.path.join(self.work_dir or '', glint_path).replace('\\', '/')
self.glint_mask_file.set_path(glint_path) self.glint_mask_file.set_path(glint_path)
# 4. 自动填充去耀斑影像路径（上游 Step 3 的输出，修复张冠李戴）
deglint_path = None deglint_path = None
if pipeline and hasattr(pipeline, 'step_outputs'): if pipeline and hasattr(pipeline, 'step_outputs'):
step3_outputs = getattr(pipeline, 'step_outputs', {}).get('step3', {}) step3_outputs = getattr(pipeline, 'step_outputs', {}).get('step3', {})
deglint_path = ( deglint_path = (
step3_outputs.get('deglint_image') step3_outputs.get('deglint_image')
or step3_outputs.get('output_path') or step3_outputs.get('output_path')
or step3_outputs.get('output_file') or step3_outputs.get('output_file')
or step3_outputs.get('deglint_img_path') or step3_outputs.get('deglint_img_path')
) )
# 回退：从 step3 面板 widget 直接读取（可能是相对路径）
if not deglint_path and hasattr(main_window, 'step3_panel'): if not deglint_path and hasattr(main_window, 'step3_panel'):
step3_widget = getattr(main_window.step3_panel, 'output_file', None) step3_widget = getattr(main_window.step3_panel, 'output_file', None)
if step3_widget is not None and hasattr(step3_widget, 'get_path'): if step3_widget is not None and hasattr(step3_widget, 'get_path'):
deglint_path = step3_widget.get_path() or "" deglint_path = step3_widget.get_path() or ""
# 兜底：扫描 3_deglint 目录下的 .bsq（优先 goodman，兼容 kutser/插值）
if not deglint_path and self.work_dir: if not deglint_path and self.work_dir:
deglint_dir = resolve_subdir(self.work_dir, 'deglint') deglint_dir = resolve_subdir(self.work_dir, 'deglint')
if os.path.isdir(deglint_dir): if os.path.isdir(deglint_dir):
@ -222,20 +265,17 @@ class Step6FeaturePanel(QWidget):
f for f in os.listdir(deglint_dir) f for f in os.listdir(deglint_dir)
if f.lower().endswith('.bsq') if f.lower().endswith('.bsq')
] ]
# 优先匹配 goodman（默认算法），其次按文件名排序保证稳定
bsq_files.sort(key=lambda n: (0 if 'goodman' in n.lower() else 1, n)) bsq_files.sort(key=lambda n: (0 if 'goodman' in n.lower() else 1, n))
if bsq_files: if bsq_files:
deglint_path = os.path.join(deglint_dir, bsq_files[0]).replace('\\', '/') deglint_path = os.path.join(deglint_dir, bsq_files[0]).replace('\\', '/')
if deglint_path: if deglint_path:
# 若为相对路径，使用 work_dir 合成为绝对路径
if not os.path.isabs(deglint_path): if not os.path.isabs(deglint_path):
deglint_path = os.path.join(self.work_dir or '', deglint_path).replace('\\', '/') deglint_path = os.path.join(self.work_dir or '', deglint_path).replace('\\', '/')
existing_deglint = self.deglint_img_file.get_path() existing_deglint = self.deglint_img_file.get_path()
if not existing_deglint or not existing_deglint.strip(): if not existing_deglint or not existing_deglint.strip():
self.deglint_img_file.set_path(deglint_path) self.deglint_img_file.set_path(deglint_path)
# 5. 自动填充输出路径（基于工作目录）
if self.work_dir: if self.work_dir:
output_dir = resolve_subdir(self.work_dir, 'spectral_feature') output_dir = resolve_subdir(self.work_dir, 'spectral_feature')
os.makedirs(output_dir, exist_ok=True) os.makedirs(output_dir, exist_ok=True)
@ -244,12 +284,9 @@ class Step6FeaturePanel(QWidget):
else: else:
self.output_file.set_path("") self.output_file.set_path("")
# 6. 尝试从 Step5 Clean 界面读取已处理的清洗后 CSV 路径，自动填入本面板
main_window = self.window()
if main_window and hasattr(main_window, 'step5_clean_panel'): if main_window and hasattr(main_window, 'step5_clean_panel'):
step5_clean_output_path = main_window.step5_clean_panel.output_file.get_path() step5_clean_output_path = main_window.step5_clean_panel.output_file.get_path()
if step5_clean_output_path: if step5_clean_output_path:
# 若为相对路径，使用 work_dir 合成为绝对路径
if not os.path.isabs(step5_clean_output_path): if not os.path.isabs(step5_clean_output_path):
step5_clean_output_path = os.path.join( step5_clean_output_path = os.path.join(
self.work_dir or '', step5_clean_output_path self.work_dir or '', step5_clean_output_path
@ -259,7 +296,6 @@ class Step6FeaturePanel(QWidget):
self.csv_file.set_path(step5_clean_output_path) self.csv_file.set_path(step5_clean_output_path)
def _on_run_single_clicked(self): def _on_run_single_clicked(self):
"""通过 EventBus 发布单步执行请求（解耦面板与 PipelineExecutor）。"""
from src.gui.core.event_bus import global_event_bus from src.gui.core.event_bus import global_event_bus
deglint_img_path = self.deglint_img_file.get_path() deglint_img_path = self.deglint_img_file.get_path()
@ -268,7 +304,7 @@ class Step6FeaturePanel(QWidget):
QMessageBox.warning(self, "输入错误", "请选择去耀斑影像文件！") QMessageBox.warning(self, "输入错误", "请选择去耀斑影像文件！")
return return
if not csv_path: if not csv_path:
QMessageBox.warning(self, "输入错误", "请选择处理后的CSV文件！") QMessageBox.warning(self, "输入错误", "请选择处理后的 CSV 文件！")
return return
if not self.glint_mask_file.get_path(): if not self.glint_mask_file.get_path():
QMessageBox.warning( QMessageBox.warning(
@ -283,30 +319,4 @@ class Step6FeaturePanel(QWidget):
global_event_bus.publish('RequestRunSingleStep', { global_event_bus.publish('RequestRunSingleStep', {
'step_name': 'step6_feature', 'step_name': 'step6_feature',
'config': config, 'config': config,
}) })
def run_step(self):
"""独立运行步骤6（旧版 parent 链上溯方式，保留兼容）。"""
# 验证输入
deglint_img_path = self.deglint_img_file.get_path()
csv_path = self.csv_file.get_path()
if not deglint_img_path:
QMessageBox.warning(self, "输入错误", "请选择去耀斑影像文件！")
return
if not csv_path:
QMessageBox.warning(self, "输入错误", "请选择处理后的CSV文件！")
return
if not self.glint_mask_file.get_path():
QMessageBox.warning(
self,
"输入错误",
"独立运行光谱特征提取时，必须选择耀斑掩膜文件。\n\n"
"请提供与去耀斑影像对应的耀斑二值掩膜（一般为步骤2输出的 severe_glint_area.dat）。",
)
return
# 获取主窗口并运行步骤
main_window = self.window()
if hasattr(main_window, 'run_single_step'):
config = {'step6_feature': self.get_config()}
main_window.run_single_step('step6_feature', config)