mirror of https://github.com/m-labs/artiq
471 lines
16 KiB
Python
471 lines
16 KiB
Python
import logging
|
|
from collections import OrderedDict
|
|
|
|
from PyQt5 import QtCore, QtGui, QtWidgets
|
|
|
|
from artiq.gui.tools import LayoutWidget, disable_scroll_wheel
|
|
from artiq.gui.scanwidget import ScanWidget
|
|
from artiq.gui.scientific_spinbox import ScientificSpinBox
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class StringEntry(QtWidgets.QLineEdit):
|
|
def __init__(self, argument):
|
|
QtWidgets.QLineEdit.__init__(self)
|
|
self.setText(argument["state"])
|
|
def update(text):
|
|
argument["state"] = text
|
|
self.textEdited.connect(update)
|
|
|
|
@staticmethod
|
|
def state_to_value(state):
|
|
return state
|
|
|
|
@staticmethod
|
|
def default_state(procdesc):
|
|
return procdesc.get("default", "")
|
|
|
|
|
|
class BooleanEntry(QtWidgets.QCheckBox):
|
|
def __init__(self, argument):
|
|
QtWidgets.QCheckBox.__init__(self)
|
|
self.setChecked(argument["state"])
|
|
def update(checked):
|
|
argument["state"] = bool(checked)
|
|
self.stateChanged.connect(update)
|
|
|
|
@staticmethod
|
|
def state_to_value(state):
|
|
return state
|
|
|
|
@staticmethod
|
|
def default_state(procdesc):
|
|
return procdesc.get("default", False)
|
|
|
|
|
|
class EnumerationEntry(QtWidgets.QComboBox):
|
|
def __init__(self, argument):
|
|
QtWidgets.QComboBox.__init__(self)
|
|
disable_scroll_wheel(self)
|
|
choices = argument["desc"]["choices"]
|
|
self.addItems(choices)
|
|
idx = choices.index(argument["state"])
|
|
self.setCurrentIndex(idx)
|
|
def update(index):
|
|
argument["state"] = choices[index]
|
|
self.currentIndexChanged.connect(update)
|
|
|
|
@staticmethod
|
|
def state_to_value(state):
|
|
return state
|
|
|
|
@staticmethod
|
|
def default_state(procdesc):
|
|
if "default" in procdesc:
|
|
return procdesc["default"]
|
|
else:
|
|
return procdesc["choices"][0]
|
|
|
|
|
|
class NumberEntryInt(QtWidgets.QSpinBox):
|
|
def __init__(self, argument):
|
|
QtWidgets.QSpinBox.__init__(self)
|
|
disable_scroll_wheel(self)
|
|
procdesc = argument["desc"]
|
|
self.setSingleStep(procdesc["step"])
|
|
if procdesc["min"] is not None:
|
|
self.setMinimum(procdesc["min"])
|
|
else:
|
|
self.setMinimum(-((1 << 31) - 1))
|
|
if procdesc["max"] is not None:
|
|
self.setMaximum(procdesc["max"])
|
|
else:
|
|
self.setMaximum((1 << 31) - 1)
|
|
if procdesc["unit"]:
|
|
self.setSuffix(" " + procdesc["unit"])
|
|
|
|
self.setValue(argument["state"])
|
|
def update(value):
|
|
argument["state"] = value
|
|
self.valueChanged.connect(update)
|
|
|
|
@staticmethod
|
|
def state_to_value(state):
|
|
return state
|
|
|
|
@staticmethod
|
|
def default_state(procdesc):
|
|
if "default" in procdesc:
|
|
return procdesc["default"]
|
|
else:
|
|
return 0
|
|
|
|
|
|
class NumberEntryFloat(ScientificSpinBox):
|
|
def __init__(self, argument):
|
|
ScientificSpinBox.__init__(self)
|
|
disable_scroll_wheel(self)
|
|
procdesc = argument["desc"]
|
|
scale = procdesc["scale"]
|
|
self.setDecimals(procdesc["ndecimals"])
|
|
self.setPrecision()
|
|
self.setSingleStep(procdesc["step"]/scale)
|
|
self.setRelativeStep()
|
|
if procdesc["min"] is not None:
|
|
self.setMinimum(procdesc["min"]/scale)
|
|
else:
|
|
self.setMinimum(float("-inf"))
|
|
if procdesc["max"] is not None:
|
|
self.setMaximum(procdesc["max"]/scale)
|
|
else:
|
|
self.setMaximum(float("inf"))
|
|
if procdesc["unit"]:
|
|
self.setSuffix(" " + procdesc["unit"])
|
|
|
|
self.setValue(argument["state"]/scale)
|
|
def update(value):
|
|
argument["state"] = value*scale
|
|
self.valueChanged.connect(update)
|
|
|
|
@staticmethod
|
|
def state_to_value(state):
|
|
return state
|
|
|
|
@staticmethod
|
|
def default_state(procdesc):
|
|
if "default" in procdesc:
|
|
return procdesc["default"]
|
|
else:
|
|
return 0.0
|
|
|
|
|
|
class _NoScan(LayoutWidget):
|
|
def __init__(self, procdesc, state):
|
|
LayoutWidget.__init__(self)
|
|
|
|
scale = procdesc["scale"]
|
|
self.value = ScientificSpinBox()
|
|
disable_scroll_wheel(self.value)
|
|
self.value.setDecimals(procdesc["ndecimals"])
|
|
self.value.setPrecision()
|
|
if procdesc["global_min"] is not None:
|
|
self.value.setMinimum(procdesc["global_min"]/scale)
|
|
else:
|
|
self.value.setMinimum(float("-inf"))
|
|
if procdesc["global_max"] is not None:
|
|
self.value.setMaximum(procdesc["global_max"]/scale)
|
|
else:
|
|
self.value.setMaximum(float("inf"))
|
|
self.value.setSingleStep(procdesc["global_step"]/scale)
|
|
self.value.setRelativeStep()
|
|
if procdesc["unit"]:
|
|
self.value.setSuffix(" " + procdesc["unit"])
|
|
self.addWidget(QtWidgets.QLabel("Value:"), 0, 0)
|
|
self.addWidget(self.value, 0, 1)
|
|
|
|
self.value.setValue(state["value"]/scale)
|
|
def update(value):
|
|
state["value"] = value*scale
|
|
self.value.valueChanged.connect(update)
|
|
|
|
self.repetitions = QtWidgets.QSpinBox()
|
|
self.repetitions.setMinimum(1)
|
|
self.repetitions.setMaximum((1 << 31) - 1)
|
|
disable_scroll_wheel(self.repetitions)
|
|
self.addWidget(QtWidgets.QLabel("Repetitions:"), 1, 0)
|
|
self.addWidget(self.repetitions, 1, 1)
|
|
|
|
self.repetitions.setValue(state["repetitions"])
|
|
|
|
def update_repetitions(value):
|
|
state["repetitions"] = value
|
|
self.repetitions.valueChanged.connect(update_repetitions)
|
|
|
|
|
|
class _RangeScan(LayoutWidget):
|
|
def __init__(self, procdesc, state):
|
|
LayoutWidget.__init__(self)
|
|
|
|
scale = procdesc["scale"]
|
|
|
|
def apply_properties(widget):
|
|
widget.setDecimals(procdesc["ndecimals"])
|
|
if procdesc["global_min"] is not None:
|
|
widget.setMinimum(procdesc["global_min"]/scale)
|
|
else:
|
|
widget.setMinimum(float("-inf"))
|
|
if procdesc["global_max"] is not None:
|
|
widget.setMaximum(procdesc["global_max"]/scale)
|
|
else:
|
|
widget.setMaximum(float("inf"))
|
|
if procdesc["global_step"] is not None:
|
|
widget.setSingleStep(procdesc["global_step"]/scale)
|
|
if procdesc["unit"]:
|
|
widget.setSuffix(" " + procdesc["unit"])
|
|
|
|
scanner = ScanWidget()
|
|
disable_scroll_wheel(scanner)
|
|
self.addWidget(scanner, 0, 0, -1, 1)
|
|
|
|
start = ScientificSpinBox()
|
|
start.setStyleSheet("QDoubleSpinBox {color:blue}")
|
|
disable_scroll_wheel(start)
|
|
self.addWidget(start, 0, 1)
|
|
|
|
npoints = QtWidgets.QSpinBox()
|
|
npoints.setMinimum(1)
|
|
npoints.setMaximum((1 << 31) - 1)
|
|
disable_scroll_wheel(npoints)
|
|
self.addWidget(npoints, 1, 1)
|
|
|
|
stop = ScientificSpinBox()
|
|
stop.setStyleSheet("QDoubleSpinBox {color:red}")
|
|
disable_scroll_wheel(stop)
|
|
self.addWidget(stop, 2, 1)
|
|
|
|
randomize = QtWidgets.QCheckBox("Randomize")
|
|
self.addWidget(randomize, 3, 1)
|
|
|
|
apply_properties(start)
|
|
start.setPrecision()
|
|
start.setRelativeStep()
|
|
apply_properties(stop)
|
|
stop.setPrecision()
|
|
stop.setRelativeStep()
|
|
apply_properties(scanner)
|
|
|
|
def update_start(value):
|
|
state["start"] = value*scale
|
|
scanner.setStart(value)
|
|
if start.value() != value:
|
|
start.setValue(value)
|
|
|
|
def update_stop(value):
|
|
state["stop"] = value*scale
|
|
scanner.setStop(value)
|
|
if stop.value() != value:
|
|
stop.setValue(value)
|
|
|
|
def update_npoints(value):
|
|
state["npoints"] = value
|
|
scanner.setNum(value)
|
|
if npoints.value() != value:
|
|
npoints.setValue(value)
|
|
|
|
def update_randomize(value):
|
|
state["randomize"] = value
|
|
randomize.setChecked(value)
|
|
|
|
scanner.startChanged.connect(update_start)
|
|
scanner.numChanged.connect(update_npoints)
|
|
scanner.stopChanged.connect(update_stop)
|
|
start.valueChanged.connect(update_start)
|
|
npoints.valueChanged.connect(update_npoints)
|
|
stop.valueChanged.connect(update_stop)
|
|
randomize.stateChanged.connect(update_randomize)
|
|
scanner.setStart(state["start"]/scale)
|
|
scanner.setNum(state["npoints"])
|
|
scanner.setStop(state["stop"]/scale)
|
|
randomize.setChecked(state["randomize"])
|
|
|
|
|
|
class _CenterScan(LayoutWidget):
|
|
def __init__(self, procdesc, state):
|
|
LayoutWidget.__init__(self)
|
|
|
|
scale = procdesc["scale"]
|
|
|
|
def apply_properties(widget):
|
|
widget.setDecimals(procdesc["ndecimals"])
|
|
if procdesc["global_min"] is not None:
|
|
widget.setMinimum(procdesc["global_min"]/scale)
|
|
else:
|
|
widget.setMinimum(float("-inf"))
|
|
if procdesc["global_max"] is not None:
|
|
widget.setMaximum(procdesc["global_max"]/scale)
|
|
else:
|
|
widget.setMaximum(float("inf"))
|
|
if procdesc["global_step"] is not None:
|
|
widget.setSingleStep(procdesc["global_step"]/scale)
|
|
if procdesc["unit"]:
|
|
widget.setSuffix(" " + procdesc["unit"])
|
|
|
|
center = ScientificSpinBox()
|
|
disable_scroll_wheel(center)
|
|
apply_properties(center)
|
|
center.setPrecision()
|
|
center.setRelativeStep()
|
|
center.setValue(state["center"])
|
|
self.addWidget(center, 0, 1)
|
|
self.addWidget(QtWidgets.QLabel("Center:"), 0, 0)
|
|
|
|
span = ScientificSpinBox()
|
|
disable_scroll_wheel(span)
|
|
apply_properties(span)
|
|
span.setPrecision()
|
|
span.setRelativeStep()
|
|
span.setMinimum(0)
|
|
span.setValue(state["span"])
|
|
self.addWidget(span, 1, 1)
|
|
self.addWidget(QtWidgets.QLabel("Span:"), 1, 0)
|
|
|
|
step = ScientificSpinBox()
|
|
disable_scroll_wheel(step)
|
|
apply_properties(step)
|
|
step.setPrecision()
|
|
step.setRelativeStep()
|
|
step.setMinimum(0)
|
|
step.setValue(state["step"])
|
|
self.addWidget(step, 2, 1)
|
|
self.addWidget(QtWidgets.QLabel("Step:"), 2, 0)
|
|
|
|
randomize = QtWidgets.QCheckBox("Randomize")
|
|
self.addWidget(randomize, 3, 1)
|
|
randomize.setChecked(state["randomize"])
|
|
|
|
def update_center(value):
|
|
state["center"] = value*scale
|
|
|
|
def update_span(value):
|
|
state["span"] = value*scale
|
|
|
|
def update_step(value):
|
|
state["step"] = value*scale
|
|
|
|
def update_randomize(value):
|
|
state["randomize"] = value
|
|
|
|
center.valueChanged.connect(update_center)
|
|
span.valueChanged.connect(update_span)
|
|
step.valueChanged.connect(update_step)
|
|
randomize.stateChanged.connect(update_randomize)
|
|
|
|
|
|
class _ExplicitScan(LayoutWidget):
|
|
def __init__(self, state):
|
|
LayoutWidget.__init__(self)
|
|
|
|
self.value = QtWidgets.QLineEdit()
|
|
self.addWidget(QtWidgets.QLabel("Sequence:"), 0, 0)
|
|
self.addWidget(self.value, 0, 1)
|
|
|
|
float_regexp = r"(([+-]?\d+(\.\d*)?|\.\d+)([eE][+-]?\d+)?)"
|
|
regexp = "(float)?( +float)* *".replace("float", float_regexp)
|
|
self.value.setValidator(QtGui.QRegExpValidator(QtCore.QRegExp(regexp)))
|
|
|
|
self.value.setText(" ".join([str(x) for x in state["sequence"]]))
|
|
def update(text):
|
|
if self.value.hasAcceptableInput():
|
|
state["sequence"] = [float(x) for x in text.split()]
|
|
self.value.textEdited.connect(update)
|
|
|
|
|
|
class ScanEntry(LayoutWidget):
|
|
def __init__(self, argument):
|
|
LayoutWidget.__init__(self)
|
|
self.argument = argument
|
|
|
|
self.stack = QtWidgets.QStackedWidget()
|
|
self.addWidget(self.stack, 1, 0, colspan=4)
|
|
|
|
procdesc = argument["desc"]
|
|
state = argument["state"]
|
|
self.widgets = OrderedDict()
|
|
self.widgets["NoScan"] = _NoScan(procdesc, state["NoScan"])
|
|
self.widgets["RangeScan"] = _RangeScan(procdesc, state["RangeScan"])
|
|
self.widgets["CenterScan"] = _CenterScan(procdesc, state["CenterScan"])
|
|
self.widgets["ExplicitScan"] = _ExplicitScan(state["ExplicitScan"])
|
|
for widget in self.widgets.values():
|
|
self.stack.addWidget(widget)
|
|
|
|
self.radiobuttons = OrderedDict()
|
|
self.radiobuttons["NoScan"] = QtWidgets.QRadioButton("No scan")
|
|
self.radiobuttons["RangeScan"] = QtWidgets.QRadioButton("Range")
|
|
self.radiobuttons["CenterScan"] = QtWidgets.QRadioButton("Center")
|
|
self.radiobuttons["ExplicitScan"] = QtWidgets.QRadioButton("Explicit")
|
|
scan_type = QtWidgets.QButtonGroup()
|
|
for n, b in enumerate(self.radiobuttons.values()):
|
|
self.addWidget(b, 0, n)
|
|
scan_type.addButton(b)
|
|
b.toggled.connect(self._scan_type_toggled)
|
|
|
|
selected = argument["state"]["selected"]
|
|
self.radiobuttons[selected].setChecked(True)
|
|
|
|
def disable(self):
|
|
self.radiobuttons["NoScan"].setChecked(True)
|
|
self.widgets["NoScan"].repetitions.setValue(1)
|
|
|
|
@staticmethod
|
|
def state_to_value(state):
|
|
selected = state["selected"]
|
|
r = dict(state[selected])
|
|
r["ty"] = selected
|
|
return r
|
|
|
|
@staticmethod
|
|
def default_state(procdesc):
|
|
scale = procdesc["scale"]
|
|
state = {
|
|
"selected": "NoScan",
|
|
"NoScan": {"value": 0.0, "repetitions": 1},
|
|
"RangeScan": {"start": 0.0, "stop": 100.0*scale, "npoints": 10,
|
|
"randomize": False},
|
|
"CenterScan": {"center": 0.*scale, "span": 100.*scale,
|
|
"step": 10.*scale, "randomize": False},
|
|
"ExplicitScan": {"sequence": []}
|
|
}
|
|
if "default" in procdesc:
|
|
defaults = procdesc["default"]
|
|
if not isinstance(defaults, list):
|
|
defaults = [defaults]
|
|
state["selected"] = defaults[0]["ty"]
|
|
for default in reversed(defaults):
|
|
ty = default["ty"]
|
|
if ty == "NoScan":
|
|
state[ty]["value"] = default["value"]
|
|
state[ty]["repetitions"] = default["repetitions"]
|
|
elif ty == "RangeScan":
|
|
state[ty]["start"] = default["start"]
|
|
state[ty]["stop"] = default["stop"]
|
|
state[ty]["npoints"] = default["npoints"]
|
|
state[ty]["randomize"] = default["randomize"]
|
|
state[ty]["seed"] = default["seed"]
|
|
elif ty == "CenterScan":
|
|
for key in "center span step randomize seed".split():
|
|
state[ty][key] = default[key]
|
|
elif ty == "ExplicitScan":
|
|
state[ty]["sequence"] = default["sequence"]
|
|
else:
|
|
logger.warning("unknown default type: %s", ty)
|
|
return state
|
|
|
|
def _scan_type_toggled(self):
|
|
for ty, button in self.radiobuttons.items():
|
|
if button.isChecked():
|
|
self.stack.setCurrentWidget(self.widgets[ty])
|
|
self.argument["state"]["selected"] = ty
|
|
break
|
|
|
|
|
|
def procdesc_to_entry(procdesc):
|
|
ty = procdesc["ty"]
|
|
if ty == "NumberValue":
|
|
is_int = (procdesc["ndecimals"] == 0
|
|
and int(procdesc["step"]) == procdesc["step"]
|
|
and procdesc["scale"] == 1)
|
|
if is_int:
|
|
return NumberEntryInt
|
|
else:
|
|
return NumberEntryFloat
|
|
else:
|
|
return {
|
|
"PYONValue": StringEntry,
|
|
"BooleanValue": BooleanEntry,
|
|
"EnumerationValue": EnumerationEntry,
|
|
"StringValue": StringEntry,
|
|
"Scannable": ScanEntry
|
|
}[ty]
|