界面优化

src/gui/panels/step8_75_panel.py

@@ -4,6 +4,8 @@
 Step8_75 面板 - 自定义回归预测
 """
 
+import os
+
 from PyQt5.QtWidgets import (
     QWidget, QVBoxLayout, QGroupBox,
     QPushButton, QCheckBox, QMessageBox, QFileDialog,
@@ -73,15 +75,70 @@ class Step8_75Panel(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.75 界面读取自定义回归模型目录
+        if main_window and hasattr(main_window, 'step6_75_panel'):
+            step6_75_models_dir = main_window.step6_75_panel.output_dir.text().strip()
+            if step6_75_models_dir:
+                existing_models = self.regression_models_dir.get_path()
+                if not existing_models or not existing_models.strip():
+                    self.regression_models_dir.set_path(step6_75_models_dir)
+
+        # 3. 自动填充输出目录（自定义回归预测目录）
+        if self.work_dir:
+            output_dir = os.path.join(self.work_dir, "11_12_13_predictions/Custom_Regression_Prediction")
+            os.makedirs(output_dir, exist_ok=True)
+            existing_out = self.output_dir_widget.get_path()
+            if not existing_out or not existing_out.strip():
+                self.output_dir_widget.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_regression_models_dir(self):
         """浏览回归模型目录"""
-        dir_path = QFileDialog.getExistingDirectory(self, "选择回归模型目录", "")
+        default = self._get_default_work_dir()
+        if default:
+            default = os.path.join(default, "9_Custom_Regression_Modeling")
+        dir_path = QFileDialog.getExistingDirectory(self, "选择回归模型目录", default)
         if dir_path:
             self.regression_models_dir.set_path(dir_path)
 
     def browse_output_dir(self):
         """浏览输出目录"""
-        dir_path = QFileDialog.getExistingDirectory(self, "选择输出目录", "")
+        default = self._get_default_work_dir()
+        if default:
+            default = os.path.join(default, "11_12_13_predictions/Custom_Regression_Prediction")
+        dir_path = QFileDialog.getExistingDirectory(self, "选择输出目录", default)
         if dir_path:
             self.output_dir_widget.set_path(dir_path)