canns 0.13.1__py3-none-any.whl → 0.14.0__py3-none-any.whl

canns/pipeline/asa_gui/main_window.py

@@ -0,0 +1,184 @@
+"""Main window for ASA GUI."""
+
+from __future__ import annotations
+
+import sys
+
+from PySide6.QtCore import QSettings, Qt, QUrl, Signal
+from PySide6.QtGui import QDesktopServices, QIcon
+from PySide6.QtWidgets import (
+    QApplication,
+    QComboBox,
+    QHBoxLayout,
+    QLabel,
+    QMainWindow,
+    QPushButton,
+    QStackedWidget,
+    QVBoxLayout,
+    QWidget,
+)
+
+from .controllers import AnalysisController, PreprocessController
+from .core import PipelineRunner, StateManager, WorkerManager
+from .resources import load_theme_qss
+from .views.pages.analysis_page import AnalysisPage
+from .views.pages.preprocess_page import PreprocessPage
+from .views.widgets.popup_combo import PopupComboBox
+
+
+class LogoLabel(QLabel):
+    clicked = Signal()
+
+    def mousePressEvent(self, event) -> None:  # pragma: no cover - UI callback
+        if event.button() == Qt.LeftButton:
+            self.clicked.emit()
+        super().mousePressEvent(event)
+
+
+class MainWindow(QMainWindow):
+    def __init__(self, parent=None) -> None:
+        super().__init__(parent)
+        self.setWindowTitle("ASA GUI")
+        self.resize(1200, 800)
+
+        self._settings = QSettings("canns", "asa_gui")
+
+        self._state_manager = StateManager()
+        self._runner = PipelineRunner()
+        self._workers = WorkerManager()
+
+        self._preprocess_controller = PreprocessController(self._state_manager, self._runner)
+        self._analysis_controller = AnalysisController(self._state_manager, self._runner)
+
+        self._build_ui()
+
+    def _build_ui(self) -> None:
+        root = QWidget()
+        layout = QVBoxLayout(root)
+
+        nav = QHBoxLayout()
+        self.logo_label = LogoLabel()
+        self.logo_label.setCursor(Qt.PointingHandCursor)
+        logo_height = 32 if sys.platform == "darwin" else 40
+        logo_pixmap = self._load_logo_pixmap(logo_height)
+        if logo_pixmap is not None:
+            self.logo_label.setPixmap(logo_pixmap)
+            self.logo_label.setFixedSize(logo_pixmap.size())
+            self.logo_label.setAlignment(Qt.AlignCenter)
+            nav.addWidget(self.logo_label)
+            self.logo_label.clicked.connect(self._open_homepage)
+        self.btn_preprocess = QPushButton("Preprocess")
+        self.btn_preprocess.setObjectName("navButton")
+        self.btn_preprocess.setCheckable(True)
+        self.btn_analysis = QPushButton("Analysis")
+        self.btn_analysis.setObjectName("navButton")
+        self.btn_analysis.setCheckable(True)
+        self.btn_preprocess.clicked.connect(lambda: self._stack.setCurrentIndex(0))
+        self.btn_analysis.clicked.connect(lambda: self._stack.setCurrentIndex(1))
+        nav.addWidget(self.btn_preprocess)
+        nav.addWidget(self.btn_analysis)
+        nav.addStretch(1)
+        self.theme_switch = PopupComboBox()
+        self.theme_switch.addItem("Light", userData="Light")
+        self.theme_switch.addItem("Dark", userData="Dark")
+        self.theme_switch.currentIndexChanged.connect(self._apply_theme)
+        self.theme_switch.setSizeAdjustPolicy(QComboBox.AdjustToContents)
+        self.theme_switch.setMinimumWidth(90)
+        nav.addWidget(self.theme_switch)
+        self.lang_switch = PopupComboBox()
+        self.lang_switch.addItem("EN", userData="en")
+        self.lang_switch.addItem("中文", userData="zh")
+        self.lang_switch.currentIndexChanged.connect(self._apply_language)
+        self.lang_switch.setSizeAdjustPolicy(QComboBox.AdjustToContents)
+        self.lang_switch.setMinimumWidth(90)
+        nav.addWidget(self.lang_switch)
+        layout.addLayout(nav)
+
+        self._stack = QStackedWidget()
+        self.preprocess_page = PreprocessPage(self._preprocess_controller, self._workers)
+        self.analysis_page = AnalysisPage(self._analysis_controller, self._workers)
+
+        self._stack.addWidget(self.preprocess_page)
+        self._stack.addWidget(self.analysis_page)
+        layout.addWidget(self._stack, 1)
+
+        self.preprocess_page.preprocess_completed.connect(self._on_preprocess_completed)
+        self._stack.currentChanged.connect(self._sync_nav)
+
+        self.setCentralWidget(root)
+        self._init_theme()
+        self._init_language()
+        self._init_icons(str(self.theme_switch.currentData() or "Light"))
+        self._sync_nav(self._stack.currentIndex())
+
+    def _go_analysis(self) -> None:
+        self._stack.setCurrentIndex(1)
+
+    def _on_preprocess_completed(self) -> None:
+        self._stack.setCurrentIndex(1)
+        self.analysis_page.load_state(self._state_manager.state)
+
+    def _sync_nav(self, index: int) -> None:
+        self.btn_preprocess.setChecked(index == 0)
+        self.btn_analysis.setChecked(index == 1)
+
+    def _init_theme(self) -> None:
+        theme = str(self._settings.value("theme", "Light"))
+        idx = 0 if theme.lower().startswith("light") else 1
+        self.theme_switch.setCurrentIndex(idx)
+        self._apply_theme()
+
+    def _init_language(self) -> None:
+        lang = str(self._settings.value("lang", "en")).lower()
+        idx = 0 if lang.startswith("en") else 1
+        self.lang_switch.setCurrentIndex(idx)
+        self._apply_language()
+
+    def _apply_theme(self) -> None:
+        try:
+            theme = str(self.theme_switch.currentData() or "Light")
+            qss = load_theme_qss(theme)
+            QApplication.instance().setStyleSheet(qss)
+            self._settings.setValue("theme", theme)
+            self._init_icons(theme)
+        except Exception:
+            pass
+
+    def _apply_language(self) -> None:
+        lang = self.lang_switch.currentData() or "en"
+        self._settings.setValue("lang", str(lang))
+        is_zh = str(lang).lower().startswith("zh")
+        self.btn_preprocess.setText("预处理" if is_zh else "Preprocess")
+        self.btn_analysis.setText("分析" if is_zh else "Analysis")
+        self.preprocess_page.apply_language(str(lang))
+        self.analysis_page.apply_language(str(lang))
+
+    def _init_icons(self, theme: str) -> None:
+        try:
+            import qtawesome as qta
+
+            color = "#34d399" if str(theme).lower().startswith("dark") else "#10b981"
+            self.btn_preprocess.setIcon(qta.icon("fa5s.sliders-h", color=color))
+            self.btn_analysis.setIcon(qta.icon("fa5s.chart-area", color=color))
+        except Exception:
+            pass
+
+    def _load_logo_pixmap(self, height: int):
+        from pathlib import Path
+
+        base = Path(__file__).resolve().parents[4] / "images"
+        logo_path = base / "logo.svg"
+        if not logo_path.exists():
+            return None
+        icon = QIcon(str(logo_path))
+        if icon.isNull():
+            return None
+        size = max(64, int(height) * 4)
+        pixmap = icon.pixmap(size, size)
+        return pixmap.scaledToHeight(int(height), Qt.SmoothTransformation)
+
+    def _open_homepage(self) -> None:
+        QDesktopServices.openUrl(QUrl("https://github.com/Routhleck/canns"))
+
+
+ASAGuiApp = MainWindow

canns/pipeline/asa_gui/models/__init__.py

@@ -0,0 +1,7 @@
+"""Data models for ASA GUI."""
+
+from .config import AnalysisConfig
+from .job import JobResult, JobSpec
+from .presets import get_preset_params
+
+__all__ = ["AnalysisConfig", "JobSpec", "JobResult", "get_preset_params"]

canns/pipeline/asa_gui/models/config.py

@@ -0,0 +1,14 @@
+"""Lightweight config helpers for ASA GUI."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from typing import Any
+
+
+@dataclass
+class AnalysisConfig:
+    """Simple container for analysis parameters."""
+
+    mode: str = "tda"
+    params: dict[str, Any] = field(default_factory=dict)

canns/pipeline/asa_gui/models/job.py

@@ -0,0 +1,31 @@
+"""Job specification and results for ASA GUI."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from pathlib import Path
+from typing import Any
+
+
+@dataclass
+class JobSpec:
+    """Inputs and parameters for a single analysis run."""
+
+    input_mode: str  # "asa" | "batch" | "neuron_traj"
+    preset: str  # "grid" | "hd" | "none"
+    asa_file: Path | None = None
+    neuron_file: Path | None = None
+    traj_file: Path | None = None
+    out_dir: Path = Path("Results/asa_gui_job")
+    params: dict[str, Any] = field(default_factory=dict)
+
+
+@dataclass
+class JobResult:
+    """Result container for pipeline execution."""
+
+    ok: bool
+    out_dir: Path
+    artifacts: dict[str, Path] = field(default_factory=dict)
+    summary: dict[str, Any] = field(default_factory=dict)
+    errors: list[str] = field(default_factory=list)

canns/pipeline/asa_gui/models/presets.py

@@ -0,0 +1,21 @@
+"""Preset parameter hints for ASA GUI."""
+
+from __future__ import annotations
+
+from typing import Any
+
+
+def get_preset_params(preset: str) -> dict[str, Any]:
+    """Return default parameter overrides for a preset."""
+    preset = (preset or "none").lower()
+    if preset == "grid":
+        return {
+            "tda": {"maxdim": 2},
+            "decode": {"num_circ": 2},
+        }
+    if preset == "hd":
+        return {
+            "tda": {"maxdim": 1},
+            "decode": {"num_circ": 1},
+        }
+    return {}

canns/pipeline/asa_gui/resources/__init__.py

@@ -0,0 +1,16 @@
+"""Static resources for ASA GUI."""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+
+def load_theme_qss(theme: str) -> str:
+    """Load the QSS for a given theme name."""
+    name = (theme or "Light").strip().lower()
+    if name in {"dark", "dark mode", "darkmode"}:
+        fname = "dark.qss"
+    else:
+        fname = "light.qss"
+    path = Path(__file__).parent / fname
+    return path.read_text(encoding="utf-8")

canns/pipeline/asa_gui/resources/dark.qss

@@ -0,0 +1,167 @@
+/* Dark theme */
+QWidget {
+    font-family: "SF Pro Text", "Helvetica Neue", "Segoe UI", "Arial", sans-serif;
+    font-size: 11pt;
+    color: #e5e7eb;
+    background-color: #111827;
+}
+
+QLabel {
+    background: transparent;
+}
+
+QLabel#muted {
+    color: #9ca3af;
+}
+
+QFrame#card, QGroupBox#card {
+    background-color: #1f2937;
+    border: 1px solid #374151;
+    border-radius: 12px;
+}
+
+QGroupBox {
+    border: 1px solid #374151;
+    border-radius: 10px;
+    margin-top: 18px;
+    background-color: #1f2937;
+}
+
+QGroupBox::title {
+    subcontrol-origin: margin;
+    left: 10px;
+    padding: 0 6px;
+    color: #34d399;
+    font-weight: 600;
+}
+
+QLineEdit, QSpinBox, QDoubleSpinBox, QComboBox {
+    background-color: #374151;
+    color: #e5e7eb;
+    border: 1px solid #4b5563;
+    border-radius: 6px;
+    padding: 4px 8px;
+    min-width: 120px;
+    selection-background-color: #34d399;
+}
+
+QLineEdit:focus, QSpinBox:focus, QDoubleSpinBox:focus, QComboBox:focus {
+    border: 1px solid #34d399;
+}
+
+
+QPushButton {
+    background-color: #374151;
+    color: #e5e7eb;
+    border: 1px solid #4b5563;
+    border-radius: 6px;
+    padding: 6px 12px;
+    font-weight: 600;
+}
+
+QPushButton:hover {
+    background-color: #4b5563;
+    border-color: #6b7280;
+}
+
+QPushButton#btn_run {
+    background-color: #059669;
+    color: #ffffff;
+    border: none;
+}
+
+QPushButton#btn_run:hover {
+    background-color: #10b981;
+}
+
+QPushButton#btn_stop {
+    background-color: #991b1b;
+    color: #ffffff;
+    border: none;
+}
+
+QPushButton#btn_stop:hover {
+    background-color: #b91c1c;
+}
+
+QPushButton#navButton {
+    background-color: transparent;
+    border: none;
+    color: #9ca3af;
+    font-weight: 600;
+    padding: 6px 10px;
+}
+
+QPushButton#navButton:checked {
+    color: #34d399;
+    border-bottom: 2px solid #34d399;
+}
+
+QTabWidget::pane {
+    border: 1px solid #374151;
+    background: #1f2937;
+    border-radius: 8px;
+}
+
+QTabBar::tab {
+    background: transparent;
+    border: none;
+    padding: 8px 14px;
+    margin-right: 4px;
+    color: #9ca3af;
+    font-weight: 600;
+}
+
+QTabBar::tab:selected {
+    color: #34d399;
+    border-bottom: 2px solid #34d399;
+}
+
+QProgressBar {
+    border: none;
+    background-color: #374151;
+    border-radius: 4px;
+    text-align: center;
+    color: #e5e7eb;
+}
+
+QProgressBar::chunk {
+    background-color: #059669;
+    border-radius: 4px;
+}
+
+QTextEdit, QListWidget {
+    background-color: #0b0f14;
+    color: #a7f3d0;
+    border: 1px solid #374151;
+    border-radius: 8px;
+    font-family: "Consolas", "Menlo", "Monaco", monospace;
+}
+
+QSplitter::handle {
+    background: #374151;
+}
+
+QScrollArea {
+    background: transparent;
+    border: none;
+}
+
+QGraphicsView {
+    background: #0f172a;
+    border: 1px solid #374151;
+    border-radius: 8px;
+}
+
+QFrame#DropZone {
+    border: 2px dashed #6b7280;
+    border-radius: 12px;
+    background-color: #111827;
+    color: #9ca3af;
+}
+
+QFrame#DropZone[drag="true"] {
+    background-color: #0f172a;
+    border-color: #34d399;
+    color: #34d399;
+}

canns/pipeline/asa_gui/resources/light.qss

@@ -0,0 +1,163 @@
+/* Light theme */
+QWidget {
+    font-family: "SF Pro Text", "Helvetica Neue", "Segoe UI", "Arial", sans-serif;
+    font-size: 11pt;
+    color: #374151;
+    background-color: #f3f4f6;
+}
+
+QLabel {
+    background: transparent;
+}
+
+QLabel#muted {
+    color: #6b7280;
+}
+
+QFrame#card, QGroupBox#card {
+    background-color: #ffffff;
+    border: 1px solid #e5e7eb;
+    border-radius: 12px;
+}
+
+QGroupBox {
+    border: 1px solid #e5e7eb;
+    border-radius: 10px;
+    margin-top: 18px;
+    background-color: #ffffff;
+}
+
+QGroupBox::title {
+    subcontrol-origin: margin;
+    left: 10px;
+    padding: 0 6px;
+    color: #10b981;
+    font-weight: 600;
+}
+
+QLineEdit, QSpinBox, QDoubleSpinBox, QComboBox {
+    background-color: #ffffff;
+    border: 1px solid #d1d5db;
+    border-radius: 6px;
+    padding: 4px 8px;
+    min-width: 120px;
+    selection-background-color: #10b981;
+}
+
+QLineEdit:focus, QSpinBox:focus, QDoubleSpinBox:focus, QComboBox:focus {
+    border: 1px solid #10b981;
+    background-color: #f0fdf4;
+}
+
+
+QPushButton {
+    background-color: #ffffff;
+    border: 1px solid #d1d5db;
+    border-radius: 6px;
+    padding: 6px 12px;
+    font-weight: 600;
+}
+
+QPushButton:hover {
+    background-color: #f9fafb;
+    border-color: #9ca3af;
+}
+
+QPushButton#btn_run {
+    background-color: #10b981;
+    color: #ffffff;
+    border: none;
+}
+
+QPushButton#btn_run:hover {
+    background-color: #059669;
+}
+
+QPushButton#btn_stop {
+    background-color: #ef4444;
+    color: #ffffff;
+    border: none;
+}
+
+QPushButton#btn_stop:hover {
+    background-color: #dc2626;
+}
+
+QPushButton#navButton {
+    background-color: transparent;
+    border: none;
+    color: #6b7280;
+    font-weight: 600;
+    padding: 6px 10px;
+}
+
+QPushButton#navButton:checked {
+    color: #10b981;
+    border-bottom: 2px solid #10b981;
+}
+
+QTabWidget::pane {
+    border: 1px solid #e5e7eb;
+    background: #ffffff;
+    border-radius: 8px;
+}
+
+QTabBar::tab {
+    background: transparent;
+    border: none;
+    padding: 8px 14px;
+    margin-right: 4px;
+    color: #6b7280;
+    font-weight: 600;
+}
+
+QTabBar::tab:selected {
+    color: #10b981;
+    border-bottom: 2px solid #10b981;
+}
+
+QProgressBar {
+    border: none;
+    background-color: #e5e7eb;
+    border-radius: 4px;
+    text-align: center;
+}
+
+QProgressBar::chunk {
+    background-color: #10b981;
+    border-radius: 4px;
+}
+
+QTextEdit, QListWidget {
+    background-color: #ffffff;
+    border: 1px solid #e5e7eb;
+    border-radius: 8px;
+}
+
+QSplitter::handle {
+    background: #e5e7eb;
+}
+
+QScrollArea {
+    background: transparent;
+    border: none;
+}
+
+QGraphicsView {
+    background: #ffffff;
+    border: 1px solid #e5e7eb;
+    border-radius: 8px;
+}
+
+QFrame#DropZone {
+    border: 2px dashed #9ca3af;
+    border-radius: 12px;
+    background-color: #f9fafb;
+    color: #6b7280;
+}
+
+QFrame#DropZone[drag="true"] {
+    background-color: #f0fdf4;
+    border-color: #10b981;
+    color: #10b981;
+}