116 lines
3.9 KiB
Python
116 lines
3.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright: Ankitects Pty Ltd and contributors
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
from aqt import colors
|
|
from aqt.qt import *
|
|
from aqt.theme import theme_manager
|
|
|
|
|
|
class Switch(QAbstractButton):
|
|
"""A horizontal slider to toggle between two states which can be denoted by short strings.
|
|
The left state is the default and corresponds to isChecked=True.
|
|
"""
|
|
|
|
_margin: int = 2
|
|
|
|
def __init__(
|
|
self,
|
|
radius: int = 10,
|
|
left_label: str = "",
|
|
right_label: str = "",
|
|
parent: QWidget = None,
|
|
) -> None:
|
|
super().__init__(parent=parent)
|
|
self.setCheckable(True)
|
|
super().setChecked(True)
|
|
self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
|
|
self._left_label = left_label
|
|
self._right_label = right_label
|
|
self._path_radius = radius
|
|
self._knob_radius = radius - self._margin
|
|
self._left_position = self._position = self._path_radius + self._margin
|
|
self._right_position = 3 * self._path_radius + self._margin
|
|
|
|
@pyqtProperty(int) # type: ignore
|
|
def position(self) -> int:
|
|
return self._position
|
|
|
|
@position.setter # type: ignore
|
|
def position(self, position: int) -> None:
|
|
self._position = position
|
|
self.update()
|
|
|
|
@property
|
|
def start_position(self) -> int:
|
|
return self._right_position if self.isChecked() else self._left_position
|
|
|
|
@property
|
|
def end_position(self) -> int:
|
|
return self._left_position if self.isChecked() else self._right_position
|
|
|
|
@property
|
|
def label(self) -> str:
|
|
return self._left_label if self.isChecked() else self._right_label
|
|
|
|
def sizeHint(self) -> QSize:
|
|
return QSize(
|
|
4 * self._path_radius + 2 * self._margin,
|
|
2 * self._path_radius + 2 * self._margin,
|
|
)
|
|
|
|
def setChecked(self, checked: bool) -> None:
|
|
super().setChecked(checked)
|
|
self._position = self.end_position
|
|
self.update()
|
|
|
|
def paintEvent(self, _event: QPaintEvent) -> None:
|
|
painter = QPainter(self)
|
|
painter.setRenderHint(QPainter.Antialiasing, True)
|
|
painter.setPen(Qt.NoPen)
|
|
self._paint_path(painter)
|
|
self._paint_knob(painter)
|
|
self._paint_label(painter)
|
|
|
|
def _paint_path(self, painter: QPainter) -> None:
|
|
painter.setBrush(QBrush(theme_manager.qcolor(colors.FRAME_BG)))
|
|
rectangle = QRectF(
|
|
self._margin,
|
|
self._margin,
|
|
self.width() - 2 * self._margin,
|
|
self.height() - 2 * self._margin,
|
|
)
|
|
painter.drawRoundedRect(rectangle, self._path_radius, self._path_radius)
|
|
|
|
def _current_knob_rectangle(self) -> QRectF:
|
|
return QRectF(
|
|
self.position - self._knob_radius, # type: ignore
|
|
2 * self._margin,
|
|
2 * self._knob_radius,
|
|
2 * self._knob_radius,
|
|
)
|
|
|
|
def _paint_knob(self, painter: QPainter) -> None:
|
|
painter.setBrush(QBrush(theme_manager.qcolor(colors.HIGHLIGHT_BG)))
|
|
painter.drawEllipse(self._current_knob_rectangle())
|
|
|
|
def _paint_label(self, painter: QPainter) -> None:
|
|
painter.setPen(QColor("white"))
|
|
font = painter.font()
|
|
font.setPixelSize(int(1.5 * self._knob_radius))
|
|
painter.setFont(font)
|
|
painter.drawText(self._current_knob_rectangle(), Qt.AlignCenter, self.label)
|
|
|
|
def mouseReleaseEvent(self, event: QMouseEvent) -> None:
|
|
super().mouseReleaseEvent(event)
|
|
if event.button() == Qt.LeftButton:
|
|
animation = QPropertyAnimation(self, b"position", self)
|
|
animation.setDuration(100)
|
|
animation.setStartValue(self.start_position)
|
|
animation.setEndValue(self.end_position)
|
|
animation.start()
|
|
|
|
def enterEvent(self, event: QEvent) -> None:
|
|
self.setCursor(Qt.PointingHandCursor)
|
|
super().enterEvent(event)
|