界面优化

src/gui/panels/step8_panel.py

@@ -4,6 +4,9 @@
 Step8 面板 - 机器学习预测
 """
 
+import os
+from pathlib import Path
+
 from PyQt5.QtWidgets import (
     QWidget, QVBoxLayout, QGroupBox, QFormLayout,
     QPushButton, QCheckBox, QComboBox, QLineEdit, QMessageBox,
@@ -76,9 +79,61 @@ class Step8Panel(QWidget):
         layout.addStretch()
         self.setLayout(layout)
 
+    def update_from_config(self, work_dir=None, pipeline=None):
+        """从全局配置自动填充采样光谱和模型目录
+
+        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
+
+        main_window = self.window()
+
+        # 1. 尝试从 Step7 界面读取全湖采样点 CSV 路径
+        if main_window and hasattr(main_window, 'step7_panel'):
+            step7_output_path = main_window.step7_panel.output_file.get_path()
+            if step7_output_path:
+                existing = self.sampling_csv_file.get_path()
+                if not existing or not existing.strip():
+                    self.sampling_csv_file.set_path(step7_output_path)
+
+        # 2. 尝试从 Step6 界面读取监督模型目录
+        if main_window and hasattr(main_window, 'step6_panel'):
+            step6_models_dir = main_window.step6_panel.output_dir.get_path()
+            if step6_models_dir:
+                existing_models = self.models_dir_file.get_path()
+                if not existing_models or not existing_models.strip():
+                    self.models_dir_file.set_path(step6_models_dir)
+
+        # 3. 自动填充输出路径（机器学习预测目录）
+        if self.work_dir:
+            output_dir = os.path.join(self.work_dir, "11_12_13_predictions/Machine_Learning_Prediction")
+            os.makedirs(output_dir, exist_ok=True)
+            existing_out = self.output_file.get_path()
+            if not existing_out or not existing_out.strip():
+                self.output_file.set_path(output_dir)
+
+    def _get_default_work_dir(self):
+        """获取 work_dir，优先用 panel 自身缓存的，否则尝试从主窗口取"""
+        if hasattr(self, 'work_dir') and self.work_dir:
+            return str(self.work_dir)
+        mw = self.window()
+        if mw and hasattr(mw, 'work_dir') and mw.work_dir:
+            return str(mw.work_dir)
+        return ""
+
     def browse_models_dir(self):
         """浏览模型目录"""
-        dir_path = QFileDialog.getExistingDirectory(self, "选择模型目录", "")
+        default = self._get_default_work_dir()
+        if default:
+            default = os.path.join(default, "7_Supervised_Model_Training")
+        dir_path = QFileDialog.getExistingDirectory(self, "选择模型目录", default)
         if dir_path:
             self.models_dir_file.set_path(dir_path)