151 lines
6.3 KiB
Python
151 lines
6.3 KiB
Python
from PyQt6.QtWidgets import QWidget, QRadioButton, QVBoxLayout, QHBoxLayout, QPushButton, QSpinBox, QFileDialog, QLabel
|
|
from PyQt6.QtWidgets import QFormLayout, QDoubleSpinBox, QCheckBox, QLineEdit, QGroupBox
|
|
|
|
from os import path
|
|
|
|
from ...utility.config import AppConfig
|
|
|
|
class ScriptSelection(QGroupBox):
|
|
def __init__(self, parent=None):
|
|
super().__init__(parent=parent, title="LED Script")
|
|
self.layout = QVBoxLayout()
|
|
|
|
# Radio buttons
|
|
self.radio_script_file = QRadioButton("Script file")
|
|
self.radio_constant_value = QRadioButton("Constant value")
|
|
self.radio_script_file.toggled.connect(self.on_radio_button_toggled)
|
|
self.radio_constant_value.toggled.connect(self.on_radio_button_toggled)
|
|
|
|
# Load from file button
|
|
self.btn_load_script = QPushButton("Load from file")
|
|
self.btn_load_script.clicked.connect(self.load_file)
|
|
self.w_script_file = QLineEdit()
|
|
self.w_script_file.setEnabled(False)
|
|
|
|
# QSpinBox for constant value
|
|
self.w_constant_value = QSpinBox()
|
|
self.w_constant_value.setRange(0, 100)
|
|
|
|
# Layouts
|
|
l_constant_value = QVBoxLayout()
|
|
l_constant_value.addWidget(self.radio_constant_value)
|
|
l_constant_value.addWidget(self.w_constant_value)
|
|
|
|
l_file = QVBoxLayout()
|
|
l_file.addWidget(self.radio_script_file)
|
|
l_file.addWidget(self.btn_load_script)
|
|
l_file.addWidget(self.w_script_file)
|
|
|
|
# Add layouts to main layout
|
|
self.layout.addLayout(l_constant_value)
|
|
self.layout.addLayout(l_file)
|
|
|
|
self.setLayout(self.layout)
|
|
|
|
# Initial state
|
|
self.radio_constant_value.setChecked(True)
|
|
self.on_radio_button_toggled()
|
|
|
|
self.layout.addStretch(1)
|
|
self.file_path = None
|
|
|
|
def on_radio_button_toggled(self):
|
|
if self.radio_script_file.isChecked():
|
|
self.btn_load_script.setEnabled(True)
|
|
self.w_constant_value.setEnabled(False)
|
|
else:
|
|
self.btn_load_script.setEnabled(False)
|
|
self.w_constant_value.setEnabled(True)
|
|
|
|
def load_file(self):
|
|
# options = QFileDialog.Options()
|
|
last_dir = AppConfig.MAIN_CFG.get_or("tmp_last_script_dir", "")
|
|
file_path, _ = QFileDialog.getOpenFileName(self, "Open Script File", last_dir, "All Files (*);;Text files (*.led)")
|
|
if file_path:
|
|
dir_name = path.dirname(file_path)
|
|
AppConfig.MAIN_CFG.set("tmp_last_script_dir", dir_name)
|
|
self.file_path = file_path
|
|
self.w_script_file.setText(self.file_path)
|
|
|
|
|
|
def get_script(self):
|
|
if self.radio_script_file.isChecked():
|
|
return self.file_path
|
|
else:
|
|
return int(self.w_constant_value.value())
|
|
|
|
|
|
class MeasurementSettings(QWidget):
|
|
"""
|
|
Widget allowing the user to set the measurement settings.
|
|
It loads the values from AppConfig.MEAS_CFG and automatically
|
|
updates the values in AppConfig.MEAS_CFG when the user changes them.
|
|
"""
|
|
def __init__(self, parent=None):
|
|
super().__init__(parent)
|
|
self.l_vbox = QVBoxLayout()
|
|
# self.label = QLabel("Measurement Settings")
|
|
# self.layout.addWidget(self.label)
|
|
self.setLayout(self.l_vbox)
|
|
self.l_form = QFormLayout()
|
|
|
|
# - script
|
|
self.w_led_script = ScriptSelection()
|
|
self.l_vbox.addWidget(self.w_led_script)
|
|
# key-value stuff in a form
|
|
self.l_vbox.addLayout(self.l_form)
|
|
self.ws_form = {}
|
|
self._add_form_field("interval", "Interval [s]", 1.0, QDoubleSpinBox(self), "Amount of seconds to wait between voltage measurements and LED device updates")
|
|
self._add_form_field("max_measurements", "Max Measurements", 0, QSpinBox(self), "Number of measurements to take. Set to 0 for infinite measurements")
|
|
self._add_form_field("stop_on_script_end", "Stop on Script End", False, QCheckBox(self), "Stop measurement when LED script ends")
|
|
self._add_form_field("use_buffer", "Use Buffer", False, QCheckBox(self), "If available, use device buffer for more accurate measurement timings.\nLeads to a lower accuracy of LED update timings, up to 1*interval")
|
|
self._add_form_field("flush_after", "Flush after", 0, QSpinBox(self), "Number of measurements to take before writing the data to an intermediate file")
|
|
self.l_vbox.addStretch(1)
|
|
|
|
def _add_form_field(self, key: str, label: str, default_value, widget: QWidget, tooltip: str = None):
|
|
if tooltip: widget.setToolTip(tooltip)
|
|
value = AppConfig.MEAS_CFG.get_or(key, default_value)
|
|
# set the value depending on the type of the widget
|
|
if isinstance(widget, QSpinBox) or isinstance(widget, QDoubleSpinBox):
|
|
widget.setValue(value)
|
|
widget.valueChanged.connect(lambda value: self.value_updated(key, value))
|
|
elif isinstance(widget, QCheckBox):
|
|
widget.setChecked(value)
|
|
widget.stateChanged.connect(lambda value: self.value_updated(key, value))
|
|
else:
|
|
raise ValueError(f"Unknown widget type: {type(widget)}")
|
|
self.l_form.addRow(QLabel(label), widget)
|
|
self.ws_form[key] = widget
|
|
|
|
|
|
def value_updated(self, key, value):
|
|
AppConfig.MEAS_CFG.set(key, value)
|
|
|
|
def set_value(self, key, value):
|
|
if key in self.ws_form:
|
|
# set depending on widget type
|
|
if isinstance(self.ws_form[key], QSpinBox) or isinstance(self.ws_form[key], QDoubleSpinBox):
|
|
self.ws_form[key].setValue(value)
|
|
elif isinstance(self.ws_form[key], QCheckBox):
|
|
self.ws_form[key].setChecked(value)
|
|
else:
|
|
raise ValueError(f"Unknown widget type: {type(self.ws_form[key])}")
|
|
else:
|
|
raise ValueError(f"Unknown key: {key}")
|
|
|
|
def get_value(self, key):
|
|
if key in self.ws_form:
|
|
# get depending on widget type
|
|
if isinstance(self.ws_form[key], QSpinBox) or isinstance(self.ws_form[key], QDoubleSpinBox):
|
|
return self.ws_form[key].value()
|
|
elif isinstance(self.ws_form[key], QCheckBox):
|
|
return self.ws_form[key].isChecked()
|
|
else:
|
|
raise ValueError(f"Unknown widget type: {type(self.ws_form[key])}")
|
|
elif key == "led_script":
|
|
return self.w_led_script.get_script()
|
|
else:
|
|
raise ValueError(f"Unknown key: {key}")
|
|
|
|
|