界面优化

src/gui/panels/step5_panel.py

@@ -0,0 +1,218 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+Step5 面板 - 光谱提取
+"""
+
+import os
+
+from PyQt5.QtWidgets import (
+    QWidget, QVBoxLayout, QGroupBox, QFormLayout, QLabel,
+    QSpinBox, QPushButton, QCheckBox, QMessageBox,
+)
+from PyQt5.QtGui import QFont
+from PyQt5.QtCore import Qt
+
+from src.gui.components.custom_widgets import FileSelectWidget
+from src.gui.styles import ModernStylesheet
+
+
+class Step5Panel(QWidget):
+    """步骤5：光谱提取"""
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.init_ui()
+
+    def init_ui(self):
+        layout = QVBoxLayout()
+
+        # 标题
+        title = QLabel("步骤5：训练样本光谱提取")
+        title.setFont(QFont("Arial", 12, QFont.Bold))
+        layout.addWidget(title)
+
+        # 去耀斑影像文件（用于独立运行）
+        self.deglint_img_file = FileSelectWidget(
+            "去耀斑影像:",
+            "Image Files (*.bsq *.dat *.tif);;All Files (*.*)"
+        )
+        layout.addWidget(self.deglint_img_file)
+
+        # 处理后的CSV文件（用于独立运行）
+        self.csv_file = FileSelectWidget(
+            "处理后CSV:",
+            "CSV Files (*.csv);;All Files (*.*)"
+        )
+        layout.addWidget(self.csv_file)
+
+        # 水体掩膜文件（可选，用于独立运行）
+        self.water_mask_file = FileSelectWidget(
+            "水体掩膜:",
+            "Mask Files (*.dat *.tif);;All Files (*.*)"
+        )
+        self.water_mask_file.line_edit.setPlaceholderText("可选，如不选择则自动生成")
+        layout.addWidget(self.water_mask_file)
+
+        self.glint_mask_file = FileSelectWidget(
+            "耀斑掩膜:",
+            "Mask Files (*.dat *.tif);;All Files (*.*)"
+        )
+        layout.addWidget(self.glint_mask_file)
+        step5_glint_hint = QLabel(
+            "提示：独立运行本步骤时必须选择耀斑掩膜（通常为步骤2输出的 severe_glint_area.dat），用于在采样时避开耀斑像元。"
+        )
+        step5_glint_hint.setWordWrap(True)
+        step5_glint_hint.setStyleSheet("color: #666; font-size: 10px;")
+        layout.addWidget(step5_glint_hint)
+
+        # 参数设置
+        params_group = QGroupBox("提取参数")
+        params_layout = QFormLayout()
+
+        self.radius = QSpinBox()
+        self.radius.setRange(1, 50)
+        self.radius.setValue(5)
+        params_layout.addRow("采样半径(像素):", self.radius)
+
+        self.source_epsg = QSpinBox()
+        self.source_epsg.setRange(1000, 99999)
+        self.source_epsg.setValue(4326)
+        params_layout.addRow("源坐标系EPSG:", self.source_epsg)
+
+        params_group.setLayout(params_layout)
+        layout.addWidget(params_group)
+
+        # 输出文件路径
+        self.output_file = FileSelectWidget(
+            "输出训练数据:",
+            "CSV Files (*.csv);;All Files (*.*)"
+        )
+        self.output_file.line_edit.setPlaceholderText("training_spectra.csv")
+        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)
+        # 信号连接：影像文件路径变化时动态更新波段范围
+
+    def get_config(self):
+        """获取配置"""
+        config = {
+            'radius': self.radius.value(),
+            'source_epsg': self.source_epsg.value(),
+        }
+        # 添加独立运行所需的文件路径
+        deglint_img_path = self.deglint_img_file.get_path()
+        if deglint_img_path:
+            config['deglint_img_path'] = deglint_img_path
+        csv_path = self.csv_file.get_path()
+        if csv_path:
+            config['csv_path'] = csv_path
+        water_mask_path = self.water_mask_file.get_path()
+        if water_mask_path:
+            config['boundary_path'] = water_mask_path
+        glint_mask_path = self.glint_mask_file.get_path()
+        if glint_mask_path:
+            config['glint_mask_path'] = glint_mask_path
+        # 注意：step5_extract_training_spectra 不接受 output_path / training_spectra_path
+        # 参数，输出路径由 pipeline 内部根据 training_spectra_dir 自动生成。
+        return config
+
+    def set_config(self, config):
+        """设置配置"""
+        if 'radius' in config:
+            self.radius.setValue(config['radius'])
+        if 'source_epsg' in config:
+            self.source_epsg.setValue(config['source_epsg'])
+        if 'deglint_img_path' in config:
+            self.deglint_img_file.set_path(config['deglint_img_path'])
+        if 'csv_path' in config:
+            self.csv_file.set_path(config['csv_path'])
+        if 'boundary_path' in config:
+            self.water_mask_file.set_path(config['boundary_path'])
+        if 'glint_mask_path' in config:
+            self.glint_mask_file.set_path(config['glint_mask_path'])
+
+    def update_from_config(self, work_dir=None, pipeline=None):
+        """从全局配置/Pipeline 或 Step1Panel 自动填充路径，实现上下游数据流转
+
+        Args:
+            work_dir: 工作目录路径
+            pipeline: Pipeline 实例，用于获取步骤1生成的水域掩膜路径
+        """
+        # 保存工作目录引用
+        if work_dir:
+            self.work_dir = work_dir
+        elif hasattr(self, 'work_dir') and self.work_dir:
+            pass
+        else:
+            self.work_dir = None
+
+        # 1. 尝试从 Pipeline 获取水体掩膜路径
+        mask_path = None
+        if pipeline and hasattr(pipeline, 'water_mask_path') and pipeline.water_mask_path:
+            mask_path = pipeline.water_mask_path
+
+        # 2. 如果 Pipeline 中没有，则尝试直接从 Step1 界面读取
+        main_window = self.window()
+        if not mask_path and hasattr(main_window, 'step1_panel'):
+            if main_window.step1_panel.use_ndwi_radio.isChecked():
+                mask_path = main_window.step1_panel.output_file.get_path()
+            else:
+                mask_path = main_window.step1_panel.mask_file.get_path()
+
+        # 填充水体掩膜路径
+        if mask_path:
+            self.water_mask_file.set_path(mask_path)
+
+        # 3. 尝试从 Step2 界面读取耀斑掩膜路径
+        main_window = self.window()
+        if hasattr(main_window, 'step2_panel'):
+            glint_path = main_window.step2_panel.output_file.get_path()
+            if glint_path:
+                self.glint_mask_file.set_path(glint_path)
+
+        # 4. 自动填充输出路径（基于工作目录）
+        if self.work_dir:
+            output_dir = os.path.join(self.work_dir, "5_training_spectra")
+            os.makedirs(output_dir, exist_ok=True)
+            default_output_path = os.path.join(output_dir, "training_spectra.csv").replace('\\', '/')
+            self.output_file.set_path(default_output_path)
+        else:
+            self.output_file.set_path("")
+
+    def run_step(self):
+        """独立运行步骤5"""
+        # 验证输入
+        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 = {'step5': self.get_config()}
+            main_window.run_single_step('step5', config)