forked from M-Labs/artiq
9fbd6de30c
Some changes are due to deprecations in Qt6 which were outright removed in PyQt, for instance QRegExp or the x()/y() QMouseEvent properties. Most of the diff is due to enumeration values now no longer being available directly in the parent namespace. This commit is purposefully restricted to the mechanical changes, no reformatting/… is done to keep the diff easy to validate.
85 lines
2.7 KiB
Python
85 lines
2.7 KiB
Python
import re
|
|
from math import inf, copysign
|
|
from PyQt6 import QtCore, QtGui, QtWidgets
|
|
|
|
|
|
_float_acceptable = re.compile(
|
|
r"([-+]?\d*(?:\d|\.\d|\d\.)\d*)(?:[eE]([-+]?\d+))?",
|
|
)
|
|
_float_intermediate = re.compile(
|
|
r"[-+]?\d*\.?\d*(?:(?:(?<=\d)|(?<=\d\.))[eE][-+]?\d*)?",
|
|
)
|
|
_exp_shorten = re.compile(r"e\+?0*")
|
|
|
|
|
|
class ScientificSpinBox(QtWidgets.QDoubleSpinBox):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.setGroupSeparatorShown(False)
|
|
self.setInputMethodHints(QtCore.Qt.InputMethodHint.ImhNone)
|
|
self.setCorrectionMode(self.CorrectionMode.CorrectToPreviousValue)
|
|
# singleStep: resolution for step, buttons, accelerators
|
|
# decimals: absolute rounding granularity
|
|
# sigFigs: number of significant digits shown
|
|
self.setSigFigs()
|
|
self.setRelativeStep()
|
|
self.setRange(-inf, inf)
|
|
self.setValue(0)
|
|
# self.setKeyboardTracking(False)
|
|
|
|
def setSigFigs(self, d=None):
|
|
if d is None:
|
|
d = self.decimals() + 3
|
|
self._sig_figs = max(1, int(d))
|
|
self._fmt = "{{:.{}g}}".format(self._sig_figs)
|
|
|
|
def sigFigs(self):
|
|
return self._sig_figs
|
|
|
|
def setRelativeStep(self, s=None):
|
|
if s is None:
|
|
s = 1 + self.singleStep()
|
|
self._relative_step = max(1 + 10**-self.decimals(), float(s))
|
|
|
|
def relativeStep(self):
|
|
return self._relative_step
|
|
|
|
def setGroupSeparatorShown(self, s):
|
|
if s:
|
|
raise NotImplementedError
|
|
|
|
def textFromValue(self, v):
|
|
t = self._fmt.format(v)
|
|
t = re.sub(_exp_shorten, "e", t, 1)
|
|
return t
|
|
|
|
def valueFromText(self, text):
|
|
clean = text
|
|
if self.prefix():
|
|
clean = clean.split(self.prefix(), 1)[-1]
|
|
if self.suffix():
|
|
clean = clean.rsplit(self.suffix(), 1)[0]
|
|
return round(float(clean), self.decimals())
|
|
|
|
def validate(self, text, pos):
|
|
clean = text
|
|
if self.prefix():
|
|
clean = clean.split(self.prefix(), 1)[-1]
|
|
if self.suffix():
|
|
clean = clean.rsplit(self.suffix(), 1)[0]
|
|
try:
|
|
float(clean) # faster than matching
|
|
return QtGui.QValidator.State.Acceptable, text, pos
|
|
except ValueError:
|
|
if re.fullmatch(_float_intermediate, clean):
|
|
return QtGui.QValidator.State.Intermediate, text, pos
|
|
return QtGui.QValidator.State.Invalid, text, pos
|
|
|
|
def stepBy(self, s):
|
|
if abs(s) < 10: # unaccelerated buttons, keys, wheel/trackpad
|
|
super().stepBy(s)
|
|
else: # accelerated PageUp/Down or CTRL-wheel
|
|
v = self.value()
|
|
v *= self._relative_step**(s/copysign(10., v))
|
|
self.setValue(v)
|