界面优化

src/gui/panels/step9_panel.py

@@ -101,12 +101,9 @@ class Step9Panel(QWidget):
         mode_row = QHBoxLayout()
         self.mode_single_rb = QRadioButton("单个 CSV 文件")
         self.mode_folder_rb = QRadioButton("文件夹批量")
-        self.mode_single_rb.setChecked(True)
         self._mode_group = QButtonGroup(self)
         self._mode_group.addButton(self.mode_single_rb, 0)
         self._mode_group.addButton(self.mode_folder_rb, 1)
-        self.mode_single_rb.toggled.connect(self._on_step9_mode_changed)
-        self.mode_folder_rb.toggled.connect(self._on_step9_mode_changed)
         mode_row.addWidget(self.mode_single_rb)
         mode_row.addWidget(self.mode_folder_rb)
         mode_row.addStretch()
@@ -192,16 +189,36 @@ class Step9Panel(QWidget):
 
         layout.addStretch()
         self.setLayout(layout)
-        self._on_step9_mode_changed()
 
-    def _on_step9_mode_changed(self):
+        # 信号绑定与初始状态
+        self.mode_single_rb.toggled.connect(self._toggle_input_mode)
+        self.mode_folder_rb.toggled.connect(self._toggle_input_mode)
+        self.mode_single_rb.setChecked(True)  # 默认选中"单个 CSV"
+        self._toggle_input_mode()             # 根据默认值设置初始显示状态
+
+    def _toggle_input_mode(self):
+        """槽函数：根据单选框状态动态显示/隐藏对应的输入组件。"""
         folder_mode = self.mode_folder_rb.isChecked()
-        self.prediction_csv_file.setEnabled(not folder_mode)
-        self._folder_row_widget.setEnabled(folder_mode)
-        self.recursive_csv_cb.setEnabled(folder_mode)
+        # 单个 CSV 模式：显示单文件选择，隐藏文件夹选择
+        self.prediction_csv_file.setVisible(not folder_mode)
+        # 文件夹批量模式：显示文件夹选择 + 递归选项，隐藏单文件选择
+        self._folder_row_widget.setVisible(folder_mode)
+        self.recursive_csv_cb.setVisible(folder_mode)
+
+    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_prediction_csv_dir(self):
-        d = QFileDialog.getExistingDirectory(self, "选择预测结果 CSV 所在文件夹")
+        default = self._get_default_work_dir()
+        if default:
+            default = os.path.join(default, "11_12_13_predictions")
+        d = QFileDialog.getExistingDirectory(self, "选择预测结果 CSV 所在文件夹", default)
         if d:
             self.prediction_csv_dir_edit.setText(d)
 
@@ -281,9 +298,68 @@ class Step9Panel(QWidget):
             if p.parent and str(p.parent) != '.':
                 self.output_dir.set_path(str(p.parent))
 
+    def update_from_config(self, work_dir=None, pipeline=None):
+        """从全局配置自动填充预测结果目录
+
+        优先使用 Step8（机器学习预测）的输出目录作为待预测 CSV 目录；
+        其次回退到 Step8.5（回归预测）或 Step8.75（自定义回归预测）的输出目录。
+
+        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()
+        if not main_window:
+            return
+
+        # 1. 尝试从 Step8 界面读取机器学习预测输出目录（优先）
+        pred_dir = None
+        if hasattr(main_window, 'step8_panel'):
+            step8_output = main_window.step8_panel.output_file.get_path()
+            if step8_output:
+                pred_dir = str(Path(step8_output).parent)
+
+        # 2. 备选：从 Step8.5 界面读取非经验预测输出目录
+        if not pred_dir and hasattr(main_window, 'step8_5_panel'):
+            step8_5_output = main_window.step8_5_panel.output_file.get_path()
+            if step8_5_output:
+                pred_dir = str(Path(step8_5_output).parent)
+
+        # 3. 备选：从 Step8.75 界面读取自定义回归预测输出目录
+        if not pred_dir and hasattr(main_window, 'step8_75_panel'):
+            step8_75_output = main_window.step8_75_panel.output_dir_widget.get_path()
+            if step8_75_output:
+                pred_dir = step8_75_output
+
+        # 自动填入"预测CSV目录"（文件夹批量模式）
+        if pred_dir:
+            existing_dir = (self.prediction_csv_dir_edit.text() or "").strip()
+            if not existing_dir:
+                self.prediction_csv_dir_edit.setText(pred_dir)
+                # 切换到文件夹批量模式
+                self.mode_folder_rb.setChecked(True)
+
+        # 4. 自动填充输出目录（14_visualization）
+        if self.work_dir:
+            output_dir = os.path.join(self.work_dir, "14_visualization")
+            os.makedirs(output_dir, exist_ok=True)
+            existing_out = self.output_dir.get_path()
+            if not existing_out or not existing_out.strip():
+                self.output_dir.set_path(output_dir)
+
     def browse_output_dir(self):
         """浏览输出目录"""
-        dir_path = QFileDialog.getExistingDirectory(self, "选择输出模型目录", "")
+        default = self._get_default_work_dir()
+        if default:
+            default = os.path.join(default, "14_visualization")
+        dir_path = QFileDialog.getExistingDirectory(self, "选择输出分布图目录", default)
         if dir_path:
             self.output_dir.set_path(dir_path)