界面优化

src/gui/panels/step4_panel.py

@@ -0,0 +1,182 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+Step4 面板 - 数据预处理
+"""
+
+import os
+
+import pandas as pd
+from PyQt5.QtWidgets import (
+    QWidget, QVBoxLayout, QGroupBox, QHBoxLayout, QLabel,
+    QSpinBox, QPushButton, QCheckBox, QTableView,
+    QAbstractItemView, QHeaderView, QMessageBox,
+)
+from PyQt5.QtCore import Qt
+
+from src.gui.components.custom_widgets import FileSelectWidget
+from src.gui.styles import ModernStylesheet
+
+
+class Step4Panel(QWidget):
+    """步骤4：数据预处理"""
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.init_ui()
+
+    def init_ui(self):
+        layout = QVBoxLayout()
+
+        # 标题
+
+        # CSV文件
+        self.csv_file = FileSelectWidget(
+            "水质参数文件:",
+            "CSV Files (*.csv);;All Files (*.*)"
+        )
+        layout.addWidget(self.csv_file)
+
+        hint = QLabel("提示: 处理CSV文件，筛选剔除异常值")
+        hint.setStyleSheet("color: #666; font-size: 10px;")
+        layout.addWidget(hint)
+
+        preview_group = QGroupBox("CSV数据预览")
+        preview_layout = QVBoxLayout()
+
+        controls_layout = QHBoxLayout()
+        controls_layout.addWidget(QLabel("预览行数:"))
+        self.preview_rows_spin = QSpinBox()
+        self.preview_rows_spin.setRange(1, 200)
+        self.preview_rows_spin.setValue(10)
+        controls_layout.addWidget(self.preview_rows_spin)
+        self.preview_btn = QPushButton("刷新预览")
+        self.preview_btn.clicked.connect(self.load_csv_preview)
+        controls_layout.addWidget(self.preview_btn)
+        controls_layout.addStretch()
+
+        self.preview_table = QTableView()
+        self.preview_table.setEditTriggers(QAbstractItemView.NoEditTriggers)
+        self.preview_table.setSelectionBehavior(QAbstractItemView.SelectRows)
+        self.preview_table.setSelectionMode(QAbstractItemView.SingleSelection)
+        self.preview_table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
+        self.preview_table.verticalHeader().setVisible(False)
+        self.preview_table.setMinimumHeight(200)
+
+        self.preview_status_label = QLabel("请选择CSV文件并点击刷新预览")
+        self.preview_status_label.setStyleSheet("color: #666; font-size: 11px;")
+
+        preview_layout.addLayout(controls_layout)
+        preview_layout.addWidget(self.preview_table)
+        preview_layout.addWidget(self.preview_status_label)
+        preview_group.setLayout(preview_layout)
+        layout.addWidget(preview_group)
+
+        # 输出文件路径
+        self.output_file = FileSelectWidget(
+            "输出处理后CSV:",
+            "CSV Files (*.csv);;All Files (*.*)"
+        )
+        self.output_file.line_edit.setPlaceholderText("processed_data.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)
+        self.reset_preview()
+
+    def get_config(self):
+        """获取配置"""
+        config = {
+            'csv_path': self.csv_file.get_path(),
+        }
+        output_path = self.output_file.get_path()
+        if output_path:
+            config['output_path'] = output_path
+        return config
+
+    def set_config(self, config):
+        """设置配置"""
+        if 'csv_path' in config:
+            self.csv_file.set_path(config['csv_path'])
+            self.load_csv_preview()
+        if 'output_path' in config:
+            self.output_file.set_path(config['output_path'])
+
+    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
+
+        if self.work_dir:
+            output_dir = os.path.join(self.work_dir, "4_processed_data")
+            os.makedirs(output_dir, exist_ok=True)
+            default_output_path = os.path.join(output_dir, "processed_data.csv").replace('\\', '/')
+            self.output_file.set_path(default_output_path)
+        else:
+            self.output_file.set_path("")
+
+    def run_step(self):
+        """独立运行步骤4"""
+        # 验证输入
+        csv_path = self.csv_file.get_path()
+        if not csv_path:
+            QMessageBox.warning(self, "输入错误", "请选择水质参数文件！")
+            return
+
+        # 获取主窗口并运行步骤
+        main_window = self.window()
+        if hasattr(main_window, 'run_single_step'):
+            config = {'step4': self.get_config()}
+            main_window.run_single_step('step4', config)
+
+    def reset_preview(self, message="请选择CSV文件并点击刷新预览"):
+        """重置预览表格"""
+        from src.gui.water_quality_gui import PandasTableModel
+        empty_model = PandasTableModel(pd.DataFrame())
+        self.preview_table.setModel(empty_model)
+        self.preview_status_label.setText(message)
+
+    def load_csv_preview(self):
+        """加载CSV预览数据"""
+        from src.gui.water_quality_gui import PandasTableModel
+        csv_path = self.csv_file.get_path()
+        if not csv_path:
+            self.reset_preview("请先选择CSV文件")
+            return
+        if not os.path.exists(csv_path):
+            self.reset_preview("文件不存在，请检查路径")
+            return
+
+        try:
+            rows_to_preview = max(1, self.preview_rows_spin.value())
+            df = pd.read_csv(csv_path, nrows=rows_to_preview)
+            if df.empty:
+                self.reset_preview("CSV文件为空")
+                return
+
+            model = PandasTableModel(df)
+            self.preview_table.setModel(model)
+            self.preview_status_label.setText(
+                f"预览 {len(df)} 行，{len(df.columns)} 列（总行数可能更多）"
+            )
+        except Exception as exc:
+            self.reset_preview(f"加载失败: {exc}")