bec/bec_widgets
0
0
Fork 0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-14 11:41:49 +02:00
Code Activity

fix: scan_control.py args_size_max fixed

Browse Source
This commit is contained in:
wyzula-jan
2023-11-09 12:13:24 +01:00
parent 5c67026637
commit da9025e032
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
bec_widgets/widgets/scan_control/scan_control.py
View File

@ -29,6 +29,7 @@ class ScanArgType:
FLOAT = "float" FLOAT = "float"
INT = "int" INT = "int"
BOOL = "bool" BOOL = "bool"
STR = "str"
class ScanControl(QWidget): class ScanControl(QWidget):
@ -37,6 +38,7 @@ class ScanControl(QWidget):
ScanArgType.FLOAT: QDoubleSpinBox, ScanArgType.FLOAT: QDoubleSpinBox,
ScanArgType.INT: QSpinBox, ScanArgType.INT: QSpinBox,
ScanArgType.BOOL: QCheckBox, ScanArgType.BOOL: QCheckBox,
ScanArgType.STR: QLineEdit,
} }
def __init__(self, parent=None, client=None, allowed_scans=None): def __init__(self, parent=None, client=None, allowed_scans=None):
@ -257,10 +259,6 @@ class ScanControl(QWidget):
# Clear the previous input fields # Clear the previous input fields
table.setRowCount(0) # Wipe table table.setRowCount(0) # Wipe table
# Set max number of args rows for table
if self.arg_size_max is not None:
table.setRowCount(self.arg_size_max)
def add_widgets_row_to_layout( def add_widgets_row_to_layout(
self, grid_layout: QGridLayout, widgets: list, row_index: int = None self, grid_layout: QGridLayout, widgets: list, row_index: int = None
) -> None: ) -> None:
@ -294,6 +292,9 @@ class ScanControl(QWidget):
# If row_index is not specified, add to the end of the table # If row_index is not specified, add to the end of the table
if row_index is None or row_index > table_widget.rowCount(): if row_index is None or row_index > table_widget.rowCount():
row_index = table_widget.rowCount() row_index = table_widget.rowCount()
if self.arg_size_max is not None: # ensure the max args size is not exceeded
if row_index >= self.arg_size_max:
return
table_widget.insertRow(row_index) table_widget.insertRow(row_index)
for column_index, widget in enumerate(widgets): for column_index, widget in enumerate(widgets):
@ -430,7 +431,9 @@ if __name__ == "__main__":
client.start() client.start()
app = QApplication([]) app = QApplication([])
scan_control = ScanControl(client=client, allowed_scans=["line_scan", "grid_scan"]) scan_control = ScanControl(
client=client,
) # allowed_scans=["line_scan", "grid_scan"])
window = scan_control window = scan_control
window.show() window.show()