qt_widgets

class DoubleSpinBox(QDoubleSpinBox):
    """A double spin box that ignores wheel events."""

    def wheelEvent(self: Self, event: Any) -> None:
        """Ignore wheel events.

        Parameters
        ----------
        event : Any
            The wheel event.
        """
        event.ignore()

class PowerOfTwoSpinBox(QSpinBox):
    """A spin box that only accepts power of two values.

    Parameters
    ----------
    min_val : int
        Minimum value.
    max_val : int
        Maximum value.
    default : int
        Default value.
    *args : Any
        Additional arguments.
    **kwargs : Any
        Additional keyword arguments.
    """

    def __init__(
        self: Self, min_val: int, max_val: int, default: int, *args: Any, **kwargs: Any
    ) -> None:
        """Initialize the widget.

        Parameters
        ----------
        min_val : int
            Minimum value.
        max_val : int
            Maximum value.
        default : int
            Default value.
        *args : Any
            Additional arguments.
        **kwargs : Any
            Additional keyword arguments.

        Raises
        ------
        ValueError
            If min or max are not power of 2.
        """
        super().__init__(*args, **kwargs)

        # min or max are not power of 2
        if min_val & (min_val - 1) != 0:
            raise ValueError(f"Minimum value must be a power of 2, got {min_val}.")

        if max_val & (max_val - 1) != 0:
            raise ValueError(f"Maximum value must be a power of 2, got {max_val}.")

        self.setRange(min_val, max_val)
        self.setSingleStep(1)
        self.setValue(default)

    def stepBy(self: Self, steps: int) -> None:
        """Step the value by a given number of steps.

        Parameters
        ----------
        steps : int
            Number of steps to step the value by.
        """
        current_value = self.value()
        current_power = self._get_base_2_log(current_value)

        # Step up or down by adjusting the power of two
        new_power = current_power + steps
        new_value = 2**new_power
        self.setValue(new_value)

    def _get_base_2_log(self: Self, value: int) -> int:
        """Get base-2 logarithm.

        Parameters
        ----------
        value : int
            The value to get the power of two for.

        Returns
        -------
        int
            The power of two of the given value.
        """
        return int(math.log2(value))

    def textFromValue(self: Self, value: int) -> str:
        """Get the text representation of the value.

        Parameters
        ----------
        value : int
            The value.

        Returns
        -------
        str
            The text representation of the value.
        """
        return str(value)

    def valueFromText(self: Self, text: str) -> int:
        """Get the value from the text representation.

        Parameters
        ----------
        text : str
            The text representation.

        Returns
        -------
        int
            The value.
        """
        return int(text)

class SpinBox(QSpinBox):
    """A spin box that ignores wheel events."""

    def wheelEvent(self: Self, event: Any) -> None:
        """Ignore wheel events.

        Parameters
        ----------
        event : Any
            The wheel event.
        """
        event.ignore()