cluster4npu/ui/dialogs/export_report_dialog.py

"""
ui/dialogs/export_report_dialog.py — 效能報告匯出對話框。

提供 ExportReportDialog(QDialog)，讓使用者選擇報告格式（PDF/CSV）與儲存路徑，
然後觸發 ReportExporter 執行匯出。

設計重點：
- _collect_report_data() 從各模組收集資料，每個來源都用 try/except 保護。
- 不在此模組執行實際 benchmark，只使用 history 的最新一筆作為 parallel_result。
- chart_image_bytes 留 None（截圖整合留未來）。
"""

from __future__ import annotations

from typing import TYPE_CHECKING, List, Optional

from PyQt5.QtWidgets import (
    QDialog,
    QFileDialog,
    QHBoxLayout,
    QLabel,
    QPushButton,
    QRadioButton,
    QVBoxLayout,
    QWidget,
    QLineEdit,
    QGroupBox,
    QProgressBar,
)
from PyQt5.QtCore import Qt

from core.performance.report_exporter import DeviceSummary, ReportData, ReportExporter

if TYPE_CHECKING:
    from core.performance.benchmarker import PerformanceBenchmarker
    from core.performance.history import PerformanceHistory


class ExportReportDialog(QDialog):
    """
    效能報告匯出對話框。

    使用者可選擇格式（PDF / CSV），指定儲存路徑後按匯出，
    對話框會呼叫 ReportExporter 產出檔案並顯示結果。
    """

    def __init__(
        self,
        parent: Optional[QWidget],
        benchmarker,                # PerformanceBenchmarker | None
        history,                    # PerformanceHistory | None
        device_manager,             # DeviceManager | None
        dashboard,                  # PerformanceDashboard | None
    ) -> None:
        super().__init__(parent)

        self._benchmarker = benchmarker
        self._history = history
        self._device_manager = device_manager
        self._dashboard = dashboard
        self._exporter = ReportExporter()

        # 預設格式為 PDF
        self._selected_format: str = "pdf"

        self._setup_ui()

    # ------------------------------------------------------------------
    # UI 建立
    # ------------------------------------------------------------------

    def _setup_ui(self) -> None:
        """建立對話框 UI。"""
        self.setWindowTitle("匯出效能報告")

        main_layout = QVBoxLayout()

        # 格式選擇
        format_group = QGroupBox("匯出格式")
        format_layout = QHBoxLayout()

        self._pdf_radio = QRadioButton("PDF")
        self._csv_radio = QRadioButton("CSV")
        self._pdf_radio.setChecked(True)
        self._pdf_radio.clicked.connect(lambda: self._set_format("pdf"))
        self._csv_radio.clicked.connect(lambda: self._set_format("csv"))

        format_layout.addWidget(self._pdf_radio)
        format_layout.addWidget(self._csv_radio)
        format_group.setLayout(format_layout)
        main_layout.addWidget(format_group)

        # 儲存路徑
        path_layout = QHBoxLayout()
        self._path_input = QLineEdit()
        self._path_input.setPlaceholderText("儲存路徑…")
        self._browse_btn = QPushButton("瀏覽")
        self._browse_btn.clicked.connect(self._on_browse)
        path_layout.addWidget(self._path_input)
        path_layout.addWidget(self._browse_btn)
        main_layout.addLayout(path_layout)

        # 進度條
        self._progress_bar = QProgressBar()
        self._progress_bar.setVisible(False)
        main_layout.addWidget(self._progress_bar)

        # 匯出按鈕
        self._export_btn = QPushButton("匯出")
        self._export_btn.clicked.connect(self._on_export)
        main_layout.addWidget(self._export_btn)

        # 狀態標籤
        self._status_label = QLabel("")
        main_layout.addWidget(self._status_label)

        self.setLayout(main_layout)

    # ------------------------------------------------------------------
    # 格式設定
    # ------------------------------------------------------------------

    def _set_format(self, fmt: str) -> None:
        """設定匯出格式（'pdf' 或 'csv'）。"""
        self._selected_format = fmt

    # ------------------------------------------------------------------
    # 事件處理
    # ------------------------------------------------------------------

    def _on_browse(self) -> None:
        """開啟 QFileDialog 讓使用者選擇儲存路徑。"""
        if self._selected_format == "pdf":
            file_filter = "PDF 檔案 (*.pdf)"
            default_suffix = ".pdf"
        else:
            file_filter = "CSV 檔案 (*.csv)"
            default_suffix = ".csv"

        path, _ = QFileDialog.getSaveFileName(
            self,
            "選擇儲存位置",
            f"performance_report{default_suffix}",
            file_filter,
        )
        if path:
            self._path_input.setText(path)

    def _on_export(self) -> None:
        """執行匯出：收集資料 -> 呼叫 ReportExporter。"""
        output_path = self._path_input.text().strip()
        if not output_path:
            self._status_label.setText("請先指定儲存路徑。")
            return

        data = self._collect_report_data()

        try:
            if self._selected_format == "pdf":
                result = self._exporter.export_pdf(data, output_path)
            else:
                result = self._exporter.export_csv(data, output_path)
            self._status_label.setText(f"匯出成功：{result}")
        except ImportError as e:
            self._status_label.setText(f"匯出失敗（缺少函式庫）：{e}")
        except ValueError as e:
            self._status_label.setText(f"匯出失敗（資料不足）：{e}")
        except Exception as e:
            self._status_label.setText(f"匯出失敗：{e}")

    # ------------------------------------------------------------------
    # 資料收集
    # ------------------------------------------------------------------

    def _collect_report_data(self) -> ReportData:
        """
        從各模組收集資料，組裝 ReportData。
        每個來源都用 try/except 保護，失敗時使用 None / 空值。
        不實際執行 benchmark，只使用 history 的最新一筆作為 parallel_result。
        """
        # 歷史記錄，同時從中取最近一筆 sequential / parallel 作為 result
        history_records: list = []
        seq_result = None
        par_result = None
        try:
            records = self._history.get_history(limit=20) if self._history else []
            history_records = list(records) if records else []
            seq_result = next((r for r in history_records if r.mode == "sequential"), None)
            par_result = next((r for r in history_records if r.mode == "parallel"), None)
        except Exception:
            history_records, seq_result, par_result = [], None, None

        # 從 benchmarker.history 取最新一筆作為 parallel_result（fallback，不執行新的 benchmark）
        if par_result is None:
            try:
                if self._benchmarker is not None:
                    hist = self._benchmarker.history
                    if hist:
                        par_result = hist[-1]
            except Exception:
                par_result = None

        # 裝置資訊
        devices: List[DeviceSummary] = []
        try:
            if self._device_manager is not None:
                raw_devices = self._device_manager.scan_devices() or []
                devices = self._convert_devices(raw_devices)
        except Exception:
            devices = []

        return ReportData(
            sequential_result=seq_result,
            parallel_result=par_result,
            speedup=None,
            history_records=history_records,
            devices=devices,
            chart_image_bytes=None,          # 截圖整合留未來
        )

    @staticmethod
    def _convert_devices(raw_devices: list) -> List[DeviceSummary]:
        """
        將 DeviceManager 回傳的裝置列表轉換為 DeviceSummary 列表。
        若轉換失敗，略過該裝置。
        """
        result: List[DeviceSummary] = []
        for dev in raw_devices:
            try:
                result.append(DeviceSummary(
                    device_id=str(getattr(dev, "device_id", getattr(dev, "id", "unknown"))),
                    product_name=str(getattr(dev, "product_name", getattr(dev, "model", "unknown"))),
                    firmware_version=str(getattr(dev, "firmware_version", "unknown")),
                    is_active=bool(getattr(dev, "is_active", True)),
                ))
            except Exception:
                continue
        return result