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}") start.setMinimumSize(110, 0) start.setSizePolicy(QtWidgets.QSizePolicy( QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed)) 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}") stop.setMinimumSize(110, 0) 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 _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["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["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}, "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 == "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]