gui: new ScientificSpinBox (from 09f9293) closes #460

This commit is contained in:
Robert Jördens 2016-08-11 16:51:56 +02:00
parent 1c8202e207
commit 6fe23b8899
1 changed files with 60 additions and 57 deletions

View File

@ -1,71 +1,74 @@
import re import re
from PyQt5 import QtGui, QtWidgets from math import inf, copysign
from PyQt5 import QtCore, QtGui, QtWidgets
# after
# http://jdreaver.com/posts/2014-07-28-scientific-notation-spin-box-pyside.html
_inf = float("inf") _float_acceptable = re.compile(
# Regular expression to find floats. Match groups are the whole string, the r"([-+]?\d*(?:\d|\.\d|\d\.)\d*)(?:[eE]([-+]?\d+))?",
# whole coefficient, the decimal part of the coefficient, and the exponent )
# part. _float_intermediate = re.compile(
_float_re = re.compile(r"(([+-]?\d+(\.\d*)?|\.\d+)([eE][+-]?\d+)?)") r"[-+]?\d*\.?\d*(?:(?:(?<=\d)|(?<=\d\.))[eE][-+]?\d*)?",
)
_exp_shorten = re.compile(r"e\+?0*")
def valid_float_string(string):
match = _float_re.search(string)
if match:
return match.groups()[0] == string
return False
class FloatValidator(QtGui.QValidator):
def validate(self, string, position):
if valid_float_string(string):
return self.Acceptable, string, position
if string == "" or string[position-1] in "eE.-+":
return self.Intermediate, string, position
return self.Invalid, string, position
def fixup(self, text):
match = _float_re.search(text)
if match:
return match.groups()[0]
return ""
class ScientificSpinBox(QtWidgets.QDoubleSpinBox): class ScientificSpinBox(QtWidgets.QDoubleSpinBox):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self.setMinimum(-_inf) self.setGroupSeparatorShown(False)
self.setMaximum(_inf) self.setInputMethodHints(QtCore.Qt.ImhNone)
self.validator = FloatValidator() self.setCorrectionMode(self.CorrectToPreviousValue)
self.setDecimals(20) # singleStep: resolution for step, buttons, accelerators
# decimals: absolute rounding granularity
# precision: number of significant digits shown, %g precision
self.setPrecision()
self.setRelativeStep()
self.setRange(-inf, inf)
self.setValue(0)
# self.setKeyboardTracking(False)
def validate(self, text, position): def setPrecision(self, d=None):
return self.validator.validate(text, position) if d is None:
d = self.decimals() + 3
self._precision = max(1, int(d))
self._fmt = "{{:.{}g}}".format(self._precision)
def fixup(self, text): def precision(self):
return self.validator.fixup(text) return self._precision
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): def valueFromText(self, text):
return float(text) return round(float(text), self.decimals())
def textFromValue(self, value): def validate(self, text, pos):
return format_float(value) try:
float(text) # faster than matching
return QtGui.QValidator.Acceptable, text, pos
except ValueError:
if re.fullmatch(_float_intermediate, text):
return QtGui.QValidator.Intermediate, text, pos
return QtGui.QValidator.Invalid, text, pos
def stepBy(self, steps): def stepBy(self, s):
text = self.cleanText() if abs(s) < 10: # unaccelerated buttons, keys, wheel/trackpad
groups = _float_re.search(text).groups() super().stepBy(s)
decimal = float(groups[1]) else: # accelerated PageUp/Down or CTRL-wheel
decimal += steps v = self.value()
new_string = "{:g}".format(decimal) + (groups[3] if groups[3] else "") v *= self._relative_step**(s/copysign(10., v))
self.lineEdit().setText(new_string) self.setValue(v)
def format_float(value):
"""Modified form of the 'g' format specifier."""
string = "{:g}".format(value)
string = string.replace("e+", "e")
string = re.sub("e(-?)0*(\d+)", r"e\1\2", string)
return string