Line data    Source code

       1              : # SPDX-FileCopyrightText: 2025 PairInteraction Developers
       2              : # SPDX-License-Identifier: LGPL-3.0-or-later
       3            1 : from __future__ import annotations
       4              : 
       5            1 : import logging
       6            1 : from typing import TYPE_CHECKING
       7              : 
       8            1 : from pairinteraction_gui.calculate.calculate_lifetimes import ParametersLifetimes, calculate_lifetimes
       9            1 : from pairinteraction_gui.config import KetConfigLifetimes
      10            1 : from pairinteraction_gui.page.base_page import CalculationPage
      11            1 : from pairinteraction_gui.plotwidget.plotwidget import PlotLifetimes
      12            1 : from pairinteraction_gui.qobjects import show_status_tip
      13              : 
      14              : if TYPE_CHECKING:
      15              :     from pairinteraction_gui.calculate.calculate_lifetimes import ResultsLifetimes
      16              : 
      17            1 : logger = logging.getLogger(__name__)
      18              : 
      19              : 
      20            1 : class LifetimesPage(CalculationPage):
      21              :     """Page for calculating lifetimes."""
      22              : 
      23            1 :     title = "Lifetimes"
      24            1 :     tooltip = "Calculate the lifetimes and transition rates for a specified ket"
      25              : 
      26            1 :     ket_config: KetConfigLifetimes
      27            1 :     plotwidget: PlotLifetimes  # type: ignore[assignment]
      28              : 
      29            1 :     def setupWidget(self) -> None:
      30            1 :         super().setupWidget()
      31            1 :         show_status_tip(self, "Ready", timeout=1)
      32              : 
      33              :         # all attributes of instance BaseConfig will be added to the toolbox in postSetupWidget
      34            1 :         self.ket_config = KetConfigLifetimes(self)
      35              : 
      36            1 :     def _create_plot_widget(self) -> PlotLifetimes:  # type: ignore[override]
      37            1 :         return PlotLifetimes(self)
      38              : 
      39            1 :     def calculate(self) -> tuple[ParametersLifetimes, ResultsLifetimes]:  # type: ignore[override]
      40            1 :         params = ParametersLifetimes.from_page(self)
      41            1 :         results = calculate_lifetimes(params)
      42            1 :         return params, results
      43              : 
      44            1 :     def update_plot(self, parameters: ParametersLifetimes, results: ResultsLifetimes) -> None:  # type: ignore[override]
      45            0 :         super().update_plot(parameters, results)  # type: ignore[arg-type]
      46            0 :         self.ket_config.set_lifetime(results.lifetime)
      47            0 :         show_status_tip(self, "Finished updating plot. Tip: Click on a bar to see transition details.", logger=logger)
      48              : 
      49            1 :     def _get_export_notebook_template_name(self) -> str:
      50            1 :         return "lifetimes.ipynb"
      51              : 
      52            1 :     def _get_export_replacements(self) -> dict[str, str]:
      53            1 :         parameters = ParametersLifetimes.from_page(self)
      54            1 :         return parameters.to_replacement_dict()