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 : from typing import TYPE_CHECKING
       6             : 
       7           1 : from PySide6.QtWidgets import (
       8             :     QDoubleSpinBox,
       9             :     QSpinBox,
      10             : )
      11             : 
      12             : if TYPE_CHECKING:
      13             :     from PySide6.QtWidgets import (
      14             :         QWidget,
      15             :     )
      16             : 
      17             : 
      18           1 : class IntSpinBox(QSpinBox):
      19             :     """Custom spin box for integer values."""
      20             : 
      21           1 :     def __init__(
      22             :         self,
      23             :         parent: QWidget | None = None,
      24             :         vmin: int = 0,
      25             :         vmax: int = 999,
      26             :         vdefault: int = 0,
      27             :         vstep: int | None = None,
      28             :         suffix: str | None = None,
      29             :         tooltip: str | None = None,
      30             :     ) -> None:
      31             :         """Initialize the integer spin box."""
      32           1 :         super().__init__(parent)
      33             : 
      34           1 :         self.setRange(int(vmin), int(vmax))
      35           1 :         self.setValue(int(vdefault))
      36           1 :         self.setSingleStep(vstep if vstep is not None else 1)
      37             : 
      38           1 :         if suffix:
      39           0 :             self.setSuffix(suffix)
      40           1 :         if tooltip:
      41           1 :             self.setToolTip(tooltip)
      42             : 
      43             : 
      44           1 : class HalfIntSpinBox(QDoubleSpinBox):
      45             :     """Custom spin box for half int values."""
      46             : 
      47           1 :     def __init__(
      48             :         self,
      49             :         parent: QWidget | None = None,
      50             :         vmin: float = 0.5,
      51             :         vmax: float = 999.5,
      52             :         vdefault: float = 0.5,
      53             :         vstep: int | None = None,
      54             :         suffix: str | None = None,
      55             :         tooltip: str | None = None,
      56             :     ) -> None:
      57             :         """Initialize the double spin box."""
      58           1 :         super().__init__(parent)
      59             : 
      60           1 :         vstep = vstep if vstep is not None else 1
      61           1 :         assert vdefault % 1 == 0.5, "Default value must be a half integer."  # NOSONAR
      62           1 :         assert vstep % 1 == 0, "Step value must be an integer."
      63             : 
      64           1 :         self.setRange(vmin, vmax)
      65           1 :         self.setValue(vdefault)
      66           1 :         self.setSingleStep(vstep)
      67           1 :         self.setDecimals(1)
      68             : 
      69           1 :         if suffix:
      70           0 :             self.setSuffix(suffix)
      71           1 :         if tooltip:
      72           1 :             self.setToolTip(tooltip)
      73             : 
      74           1 :     def valueFromText(self, text: str) -> float:
      75             :         """Convert text to value, ensuring it's a half integer."""
      76           0 :         value = super().valueFromText(text)
      77           0 :         return round(value - 0.49) + 0.5
      78             : 
      79             : 
      80           1 : class DoubleSpinBox(QDoubleSpinBox):
      81             :     """Custom spin box for double values."""
      82             : 
      83           1 :     def __init__(
      84             :         self,
      85             :         parent: QWidget | None = None,
      86             :         vmin: float = 0,
      87             :         vmax: float = 999.9,
      88             :         vdefault: float = 0,
      89             :         vstep: float | None = None,
      90             :         suffix: str | None = None,
      91             :         decimals: int = 1,
      92             :         tooltip: str | None = None,
      93             :     ) -> None:
      94             :         """Initialize the double spin box."""
      95           1 :         super().__init__(parent)
      96             : 
      97           1 :         self.setDecimals(decimals)
      98           1 :         self.setSingleStep(vstep if vstep is not None else 10**-decimals)
      99           1 :         self.setRange(vmin, vmax)
     100           1 :         self.setValue(vdefault)
     101             : 
     102           1 :         if suffix:
     103           0 :             self.setSuffix(suffix)
     104           1 :         if tooltip:
     105           1 :             self.setToolTip(tooltip)