forked from M-Labs/thermostat
ctrl_panel: Remove need for "mA" hack
Remove all instances of mA scaling scattered all around the code and specify it in the parameter tree with a single source of truth. Done by adding the option "pinSiPrefix" for all Parameters of type `int` or `float`, and using it for current Parameters with unit "mA".
This commit is contained in:
parent
d96d36fb0c
commit
c59e3e7ac4
|
@ -4,6 +4,7 @@ from pyqtgraph.parametertree import (
|
||||||
Parameter,
|
Parameter,
|
||||||
registerParameterType,
|
registerParameterType,
|
||||||
)
|
)
|
||||||
|
import pytec.gui.view.pin_si_prefix
|
||||||
|
|
||||||
|
|
||||||
class MutexParameter(pTypes.ListParameter):
|
class MutexParameter(pTypes.ListParameter):
|
||||||
|
@ -136,10 +137,10 @@ class CtrlPanel(QObject):
|
||||||
)
|
)
|
||||||
self.params[channel].child(
|
self.params[channel].child(
|
||||||
"pid", "pid_output_clamping", "output_min"
|
"pid", "pid_output_clamping", "output_min"
|
||||||
).setValue(settings["parameters"]["output_min"] * 1000)
|
).setValue(settings["parameters"]["output_min"])
|
||||||
self.params[channel].child(
|
self.params[channel].child(
|
||||||
"pid", "pid_output_clamping", "output_max"
|
"pid", "pid_output_clamping", "output_max"
|
||||||
).setValue(settings["parameters"]["output_max"] * 1000)
|
).setValue(settings["parameters"]["output_max"])
|
||||||
self.params[channel].child(
|
self.params[channel].child(
|
||||||
"output", "control_method", "target"
|
"output", "control_method", "target"
|
||||||
).setValue(settings["target"])
|
).setValue(settings["target"])
|
||||||
|
@ -154,7 +155,7 @@ class CtrlPanel(QObject):
|
||||||
)
|
)
|
||||||
self.params[channel].child(
|
self.params[channel].child(
|
||||||
"output", "control_method", "i_set"
|
"output", "control_method", "i_set"
|
||||||
).setValue(settings["i_set"] * 1000)
|
).setValue(settings["i_set"])
|
||||||
if settings["temperature"] is not None:
|
if settings["temperature"] is not None:
|
||||||
self.params[channel].child("temperature").setValue(
|
self.params[channel].child("temperature").setValue(
|
||||||
settings["temperature"]
|
settings["temperature"]
|
||||||
|
@ -188,10 +189,10 @@ class CtrlPanel(QObject):
|
||||||
pwm_params["max_v"]["value"]
|
pwm_params["max_v"]["value"]
|
||||||
)
|
)
|
||||||
self.params[channel].child("output", "limits", "max_i_pos").setValue(
|
self.params[channel].child("output", "limits", "max_i_pos").setValue(
|
||||||
pwm_params["max_i_pos"]["value"] * 1000
|
pwm_params["max_i_pos"]["value"]
|
||||||
)
|
)
|
||||||
self.params[channel].child("output", "limits", "max_i_neg").setValue(
|
self.params[channel].child("output", "limits", "max_i_neg").setValue(
|
||||||
pwm_params["max_i_neg"]["value"] * 1000
|
pwm_params["max_i_neg"]["value"]
|
||||||
)
|
)
|
||||||
|
|
||||||
for limit in "max_i_pos", "max_i_neg", "max_v":
|
for limit in "max_i_pos", "max_i_neg", "max_v":
|
||||||
|
|
|
@ -48,14 +48,16 @@
|
||||||
"title": "Set Current",
|
"title": "Set Current",
|
||||||
"type": "float",
|
"type": "float",
|
||||||
"value": 0,
|
"value": 0,
|
||||||
"step": 100,
|
"step": 0.1,
|
||||||
"limits": [
|
"limits": [
|
||||||
-2000,
|
-2,
|
||||||
2000
|
2
|
||||||
],
|
],
|
||||||
"triggerOnShow": true,
|
"triggerOnShow": true,
|
||||||
"decimals": 6,
|
"decimals": 6,
|
||||||
"suffix": "mA",
|
"pinSiPrefix": "m",
|
||||||
|
"suffix": "A",
|
||||||
|
"siPrefix": true,
|
||||||
"param": [
|
"param": [
|
||||||
"pwm",
|
"pwm",
|
||||||
"ch",
|
"ch",
|
||||||
|
@ -97,13 +99,15 @@
|
||||||
"title": "Max Cooling Current",
|
"title": "Max Cooling Current",
|
||||||
"type": "float",
|
"type": "float",
|
||||||
"value": 0,
|
"value": 0,
|
||||||
"step": 100,
|
"step": 0.1,
|
||||||
"decimals": 6,
|
"decimals": 6,
|
||||||
"limits": [
|
"limits": [
|
||||||
0,
|
0,
|
||||||
2000
|
2
|
||||||
],
|
],
|
||||||
"suffix": "mA",
|
"siPrefix": true,
|
||||||
|
"pinSiPrefix": "m",
|
||||||
|
"suffix": "A",
|
||||||
"param": [
|
"param": [
|
||||||
"pwm",
|
"pwm",
|
||||||
"ch",
|
"ch",
|
||||||
|
@ -117,13 +121,15 @@
|
||||||
"title": "Max Heating Current",
|
"title": "Max Heating Current",
|
||||||
"type": "float",
|
"type": "float",
|
||||||
"value": 0,
|
"value": 0,
|
||||||
"step": 100,
|
"step": 0.1,
|
||||||
"decimals": 6,
|
"decimals": 6,
|
||||||
|
"siPrefix": true,
|
||||||
|
"pinSiPrefix": "m",
|
||||||
|
"suffix": "A",
|
||||||
"limits": [
|
"limits": [
|
||||||
0,
|
0,
|
||||||
2000
|
2
|
||||||
],
|
],
|
||||||
"suffix": "mA",
|
|
||||||
"param": [
|
"param": [
|
||||||
"pwm",
|
"pwm",
|
||||||
"ch",
|
"ch",
|
||||||
|
@ -296,13 +302,15 @@
|
||||||
"name": "output_min",
|
"name": "output_min",
|
||||||
"title": "Minimum",
|
"title": "Minimum",
|
||||||
"type": "float",
|
"type": "float",
|
||||||
"step": 100,
|
"step": 0.1,
|
||||||
"limits": [
|
"limits": [
|
||||||
-2000,
|
-2,
|
||||||
2000
|
2
|
||||||
],
|
],
|
||||||
"decimals": 6,
|
"decimals": 6,
|
||||||
"suffix": "mA",
|
"siPrefix": true,
|
||||||
|
"pinSiPrefix": "m",
|
||||||
|
"suffix": "A",
|
||||||
"param": [
|
"param": [
|
||||||
"pid",
|
"pid",
|
||||||
"ch",
|
"ch",
|
||||||
|
@ -315,13 +323,15 @@
|
||||||
"name": "output_max",
|
"name": "output_max",
|
||||||
"title": "Maximum",
|
"title": "Maximum",
|
||||||
"type": "float",
|
"type": "float",
|
||||||
"step": 100,
|
"step": 0.1,
|
||||||
"limits": [
|
"limits": [
|
||||||
-2000,
|
-2,
|
||||||
2000
|
2
|
||||||
],
|
],
|
||||||
"decimals": 6,
|
"decimals": 6,
|
||||||
"suffix": "mA",
|
"siPrefix": true,
|
||||||
|
"pinSiPrefix": "m",
|
||||||
|
"suffix": "A",
|
||||||
"param": [
|
"param": [
|
||||||
"pid",
|
"pid",
|
||||||
"ch",
|
"ch",
|
||||||
|
@ -358,12 +368,14 @@
|
||||||
"type": "float",
|
"type": "float",
|
||||||
"value": 0,
|
"value": 0,
|
||||||
"decimals": 6,
|
"decimals": 6,
|
||||||
"step": 100,
|
"step": 0.1,
|
||||||
"limits": [
|
"limits": [
|
||||||
-2000,
|
-2,
|
||||||
2000
|
2
|
||||||
],
|
],
|
||||||
"suffix": "mA",
|
"siPrefix": true,
|
||||||
|
"pinSiPrefix": "m",
|
||||||
|
"suffix": "A",
|
||||||
"pid_autotune": [
|
"pid_autotune": [
|
||||||
"test_current",
|
"test_current",
|
||||||
"ch"
|
"ch"
|
||||||
|
|
|
@ -0,0 +1,95 @@
|
||||||
|
from pyqtgraph import SpinBox
|
||||||
|
import pyqtgraph.functions as fn
|
||||||
|
from pyqtgraph.parametertree import registerParameterItemType
|
||||||
|
from pyqtgraph.parametertree.parameterTypes import SimpleParameter, NumericParameterItem
|
||||||
|
|
||||||
|
|
||||||
|
class PinSIPrefixSpinBox(SpinBox):
|
||||||
|
"""
|
||||||
|
Extension of PyQtGraph's SpinBox widget.
|
||||||
|
Adds:
|
||||||
|
|
||||||
|
* The "pinSiPrefix" option, where the siPrefix could be fixed to a
|
||||||
|
particular scale instead of as determined by its value.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def setOpts(self, **opts):
|
||||||
|
if "pinSiPrefix" in opts:
|
||||||
|
self.opts["pinSiPrefix"] = opts.pop("pinSiPrefix")
|
||||||
|
|
||||||
|
super().setOpts(**opts)
|
||||||
|
|
||||||
|
def formatText(self, prev=None):
|
||||||
|
"""
|
||||||
|
In addition to pyqtgraph.SpinBox's formatting, incorporate the
|
||||||
|
'pinSiPrefix' mechanism, where SI prefixes could be fixed.
|
||||||
|
"""
|
||||||
|
# Code modified from the PyQtGraph source
|
||||||
|
|
||||||
|
# get the number of decimal places to print
|
||||||
|
decimals = self.opts['decimals']
|
||||||
|
suffix = self.opts['suffix']
|
||||||
|
prefix = self.opts['prefix']
|
||||||
|
pin_si_prefix = self.opts.get("pinSiPrefix")
|
||||||
|
|
||||||
|
# format the string
|
||||||
|
val = self.value()
|
||||||
|
if self.opts['siPrefix'] is True:
|
||||||
|
# SI prefix was requested, so scale the value accordingly
|
||||||
|
if pin_si_prefix is not None and pin_si_prefix in fn.SI_PREFIX_EXPONENTS:
|
||||||
|
# fixed scale
|
||||||
|
s = 10**-fn.SI_PREFIX_EXPONENTS[pin_si_prefix]
|
||||||
|
p = pin_si_prefix
|
||||||
|
elif self.val == 0 and prev is not None:
|
||||||
|
# special case: if it's zero use the previous prefix
|
||||||
|
(s, p) = fn.siScale(prev)
|
||||||
|
else:
|
||||||
|
(s, p) = fn.siScale(val)
|
||||||
|
parts = {'value': val, 'suffix': suffix, 'decimals': decimals, 'siPrefix': p, 'scaledValue': s*val, 'prefix':prefix}
|
||||||
|
|
||||||
|
else:
|
||||||
|
# no SI prefix /suffix requested; scale is 1
|
||||||
|
parts = {'value': val, 'suffix': suffix, 'decimals': decimals, 'siPrefix': '', 'scaledValue': val, 'prefix':prefix}
|
||||||
|
|
||||||
|
parts['prefixGap'] = '' if parts['prefix'] == '' else ' '
|
||||||
|
parts['suffixGap'] = '' if (parts['suffix'] == '' and parts['siPrefix'] == '') else ' '
|
||||||
|
|
||||||
|
return self.opts['format'].format(**parts)
|
||||||
|
|
||||||
|
|
||||||
|
class PinSIPrefixNumericParameterItem(NumericParameterItem):
|
||||||
|
"""
|
||||||
|
Subclasses PyQtGraph's `NumericParameterItem` and uses
|
||||||
|
PinSIPrefixSpinBox for editing.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def makeWidget(self):
|
||||||
|
opts = self.param.opts
|
||||||
|
t = opts['type']
|
||||||
|
defs = {
|
||||||
|
'value': 0, 'min': None, 'max': None,
|
||||||
|
'step': 1.0, 'dec': False,
|
||||||
|
'siPrefix': False, 'suffix': '', 'decimals': 3,
|
||||||
|
'pinSiPrefix': None,
|
||||||
|
}
|
||||||
|
if t == 'int':
|
||||||
|
defs['int'] = True
|
||||||
|
defs['minStep'] = 1.0
|
||||||
|
for k in defs:
|
||||||
|
if k in opts:
|
||||||
|
defs[k] = opts[k]
|
||||||
|
if 'limits' in opts:
|
||||||
|
defs['min'], defs['max'] = opts['limits']
|
||||||
|
w = PinSIPrefixSpinBox()
|
||||||
|
w.setOpts(**defs)
|
||||||
|
w.sigChanged = w.sigValueChanged
|
||||||
|
w.sigChanging = w.sigValueChanging
|
||||||
|
return w
|
||||||
|
|
||||||
|
|
||||||
|
registerParameterItemType(
|
||||||
|
"float", PinSIPrefixNumericParameterItem, SimpleParameter, override=True
|
||||||
|
)
|
||||||
|
registerParameterItemType(
|
||||||
|
"int", PinSIPrefixNumericParameterItem, SimpleParameter, override=True
|
||||||
|
)
|
|
@ -267,9 +267,6 @@ class MainWindow(QtWidgets.QMainWindow):
|
||||||
for inner_param, change, data in changes:
|
for inner_param, change, data in changes:
|
||||||
if change == "value":
|
if change == "value":
|
||||||
if inner_param.opts.get("param", None) is not None:
|
if inner_param.opts.get("param", None) is not None:
|
||||||
if inner_param.opts.get("suffix", None) == "mA":
|
|
||||||
data /= 1000 # Given in mA
|
|
||||||
|
|
||||||
thermostat_param = inner_param.opts["param"]
|
thermostat_param = inner_param.opts["param"]
|
||||||
if thermostat_param[1] == "ch":
|
if thermostat_param[1] == "ch":
|
||||||
thermostat_param[1] = ch
|
thermostat_param[1] = ch
|
||||||
|
|
Loading…
Reference in New Issue