class UIManipulateROIWindow:
def setup_ui(self, manipulate_roi_window_instance, rois, dataset_rtss,
roi_color, signal_roi_manipulated):
self.patient_dict_container = PatientDictContainer()
self.rois = rois
self.dataset_rtss = dataset_rtss
self.signal_roi_manipulated = signal_roi_manipulated
self.roi_color = roi_color
self.roi_names = [] # Names of selected ROIs
self.all_roi_names = [] # Names of all existing ROIs
for roi_id, roi_dict in self.rois.items():
self.all_roi_names.append(roi_dict['name'])
# Operation names
self.single_roi_operation_names = [
"Expand", "Contract", "Inner Rind (annulus)",
"Outer Rind (annulus)"
]
self.multiple_roi_operation_names = [
"Union", "Intersection", "Difference"
]
self.operation_names = self.multiple_roi_operation_names + \
self.single_roi_operation_names
self.new_ROI_contours = None
self.manipulate_roi_window_instance = manipulate_roi_window_instance
self.dicom_view = DicomAxialView(metadata_formatted=True,
is_four_view=True)
self.dicom_preview = DicomAxialView(metadata_formatted=True,
is_four_view=True)
self.dicom_view.slider.valueChanged.connect(
self.dicom_view_slider_value_changed)
self.dicom_preview.slider.valueChanged.connect(
self.dicom_preview_slider_value_changed)
self.init_layout()
QtCore.QMetaObject.connectSlotsByName(manipulate_roi_window_instance)
def retranslate_ui(self, manipulate_roi_window_instance):
_translate = QtCore.QCoreApplication.translate
manipulate_roi_window_instance.setWindowTitle(
_translate("ManipulateRoiWindowInstance",
"OnkoDICOM - Draw Region Of Interest"))
self.first_roi_name_label.setText(
_translate("FirstROINameLabel", "ROI 1: "))
self.first_roi_name_dropdown_list.setPlaceholderText("ROI 1")
self.first_roi_name_dropdown_list.addItems(self.all_roi_names)
self.operation_name_label.setText(
_translate("OperationNameLabel", "Operation"))
self.operation_name_dropdown_list.setPlaceholderText("Operation")
self.operation_name_dropdown_list.addItems(self.operation_names)
self.second_roi_name_label.setText(
_translate("SecondROINameLabel", "ROI 2: "))
self.second_roi_name_dropdown_list.setPlaceholderText("ROI 2")
self.second_roi_name_dropdown_list.addItems(self.all_roi_names)
self.manipulate_roi_window_instance_draw_button.setText(
_translate("ManipulateRoiWindowInstanceDrawButton", "Draw"))
self.manipulate_roi_window_instance_save_button.setText(
_translate("ManipulateRoiWindowInstanceSaveButton", "Save"))
self.manipulate_roi_window_instance_cancel_button.setText(
_translate("ManipulateRoiWindowInstanceCancelButton", "Cancel"))
self.margin_label.setText(_translate("MarginLabel", "Margin (mm): "))
self.new_roi_name_label.setText(
_translate("NewROINameLabel", "New ROI Name"))
self.ROI_view_box_label.setText("ROI")
self.preview_box_label.setText("Preview")
def init_layout(self):
"""
Initialize the layout for the DICOM View tab.
Add the view widget and the slider in the layout.
Add the whole container 'tab2_view' as a tab in the main page.
"""
# Initialise a ManipulateROIWindow
if platform.system() == 'Darwin':
self.stylesheet_path = "res/stylesheet.qss"
else:
self.stylesheet_path = "res/stylesheet-win-linux.qss"
stylesheet = open(resource_path(self.stylesheet_path)).read()
window_icon = QIcon()
window_icon.addPixmap(QPixmap(resource_path("res/images/icon.ico")),
QIcon.Normal, QIcon.Off)
self.manipulate_roi_window_instance.setObjectName(
"ManipulateRoiWindowInstance")
self.manipulate_roi_window_instance.setWindowIcon(window_icon)
# Creating a form box to hold all buttons and input fields
self.manipulate_roi_window_input_container_box = QFormLayout()
self.manipulate_roi_window_input_container_box.setObjectName(
"ManipulateRoiWindowInputContainerBox")
self.manipulate_roi_window_input_container_box.setLabelAlignment(
Qt.AlignLeft)
# Create a label for denoting the first ROI name
self.first_roi_name_label = QLabel()
self.first_roi_name_label.setObjectName("FirstROINameLabel")
self.first_roi_name_dropdown_list = QComboBox()
# Create an dropdown list for ROI name
self.first_roi_name_dropdown_list.setObjectName(
"FirstROINameDropdownList")
self.first_roi_name_dropdown_list.setSizePolicy(
QSizePolicy.Minimum, QSizePolicy.Minimum)
self.first_roi_name_dropdown_list.resize(
self.first_roi_name_dropdown_list.sizeHint().width(),
self.first_roi_name_dropdown_list.sizeHint().height())
self.first_roi_name_dropdown_list.activated.connect(
self.update_selected_rois)
self.manipulate_roi_window_input_container_box.addRow(
self.first_roi_name_label, self.first_roi_name_dropdown_list)
# Create a label for denoting the operation
self.operation_name_label = QLabel()
self.operation_name_label.setObjectName("OperationNameLabel")
self.operation_name_dropdown_list = QComboBox()
# Create an dropdown list for operation name
self.operation_name_dropdown_list.setObjectName(
"OperationNameDropdownList")
self.operation_name_dropdown_list.setSizePolicy(
QSizePolicy.Minimum, QSizePolicy.Minimum)
self.operation_name_dropdown_list.resize(
self.operation_name_dropdown_list.sizeHint().width(),
self.operation_name_dropdown_list.sizeHint().height())
self.operation_name_dropdown_list.activated.connect(
self.operation_changed)
self.manipulate_roi_window_input_container_box.addRow(
self.operation_name_label, self.operation_name_dropdown_list)
# Create a label for denoting the second ROI name
self.second_roi_name_label = QLabel()
self.second_roi_name_label.setObjectName("SecondROINameLabel")
self.second_roi_name_label.setVisible(False)
self.second_roi_name_dropdown_list = QComboBox()
# Create an dropdown list for ROI name
self.second_roi_name_dropdown_list.setObjectName(
"SecondROINameDropdownList")
self.second_roi_name_dropdown_list.setSizePolicy(
QSizePolicy.Minimum, QSizePolicy.Minimum)
self.second_roi_name_dropdown_list.resize(
self.second_roi_name_dropdown_list.sizeHint().width(),
self.second_roi_name_dropdown_list.sizeHint().height())
self.second_roi_name_dropdown_list.setVisible(False)
self.second_roi_name_dropdown_list.activated.connect(
self.update_selected_rois)
self.manipulate_roi_window_input_container_box.addRow(
self.second_roi_name_label, self.second_roi_name_dropdown_list)
# Create a label for denoting the margin
self.margin_label = QLabel()
self.margin_label.setObjectName("MarginLabel")
self.margin_label.setVisible(False)
# Create input for the new ROI name
self.margin_line_edit = QLineEdit()
self.margin_line_edit.setObjectName("MarginInput")
self.margin_line_edit.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.Minimum)
self.margin_line_edit.resize(self.margin_line_edit.sizeHint().width(),
self.margin_line_edit.sizeHint().height())
self.margin_line_edit.setVisible(False)
self.margin_line_edit.setValidator(
QRegularExpressionValidator(
QRegularExpression("^[0-9]*[.]?[0-9]*$")))
self.manipulate_roi_window_input_container_box.addRow(
self.margin_label, self.margin_line_edit)
# Create a label for denoting the new ROI name
self.new_roi_name_label = QLabel()
self.new_roi_name_label.setObjectName("NewROINameLabel")
# Create input for the new ROI name
self.new_roi_name_line_edit = QLineEdit()
self.new_roi_name_line_edit.setObjectName("NewROINameInput")
self.new_roi_name_line_edit.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.Minimum)
self.new_roi_name_line_edit.resize(
self.new_roi_name_line_edit.sizeHint().width(),
self.new_roi_name_line_edit.sizeHint().height())
self.manipulate_roi_window_input_container_box.addRow(
self.new_roi_name_label, self.new_roi_name_line_edit)
# Create a spacer between inputs and buttons
spacer = QWidget()
spacer.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
spacer.setFocusPolicy(Qt.NoFocus)
self.manipulate_roi_window_input_container_box.addRow(spacer)
# Create a warning message when missing inputs
self.warning_message = QWidget()
self.warning_message.setContentsMargins(8, 5, 8, 5)
warning_message_layout = QHBoxLayout()
warning_message_layout.setAlignment(QtCore.Qt.AlignLeft
| QtCore.Qt.AlignLeft)
warning_message_icon = QLabel()
warning_message_icon.setPixmap(
QtGui.QPixmap(
resource_path("res/images/btn-icons/alert_icon.png")))
warning_message_layout.addWidget(warning_message_icon)
self.warning_message_text = QLabel()
self.warning_message_text.setStyleSheet("color: red")
warning_message_layout.addWidget(self.warning_message_text)
self.warning_message.setLayout(warning_message_layout)
self.warning_message.setVisible(False)
self.manipulate_roi_window_input_container_box.addRow(
self.warning_message)
# Create a draw button
self.manipulate_roi_window_instance_draw_button = QPushButton()
self.manipulate_roi_window_instance_draw_button.setObjectName(
"ManipulateRoiWindowInstanceDrawButton")
self.manipulate_roi_window_instance_draw_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum))
self.manipulate_roi_window_instance_draw_button.resize(
self.manipulate_roi_window_instance_draw_button.sizeHint().width(),
self.manipulate_roi_window_instance_draw_button.sizeHint().height(
))
self.manipulate_roi_window_instance_draw_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.manipulate_roi_window_instance_draw_button.clicked.connect(
self.onDrawButtonClicked)
self.manipulate_roi_window_input_container_box.addRow(
self.manipulate_roi_window_instance_draw_button)
# Create a horizontal box for saving and cancel the drawing
self.manipulate_roi_window_cancel_save_box = QHBoxLayout()
self.manipulate_roi_window_cancel_save_box.setObjectName(
"ManipulateRoiWindowCancelSaveBox")
# Create an exit button to cancel the drawing
# Add a button to go back/exit from the application
self.manipulate_roi_window_instance_cancel_button = QPushButton()
self.manipulate_roi_window_instance_cancel_button.setObjectName(
"ManipulateRoiWindowInstanceCancelButton")
self.manipulate_roi_window_instance_cancel_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum))
self.manipulate_roi_window_instance_cancel_button.resize(
self.manipulate_roi_window_instance_cancel_button.sizeHint().width(
),
self.manipulate_roi_window_instance_cancel_button.sizeHint().
height())
self.manipulate_roi_window_instance_cancel_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.manipulate_roi_window_instance_cancel_button.clicked.connect(
self.onCancelButtonClicked)
self.manipulate_roi_window_instance_cancel_button.setProperty(
"QPushButtonClass", "fail-button")
icon_cancel = QtGui.QIcon()
icon_cancel.addPixmap(
QtGui.QPixmap(
resource_path('res/images/btn-icons/cancel_icon.png')))
self.manipulate_roi_window_instance_cancel_button.setIcon(icon_cancel)
self.manipulate_roi_window_cancel_save_box.addWidget(
self.manipulate_roi_window_instance_cancel_button)
# Create a save button to save all the changes
self.manipulate_roi_window_instance_save_button = QPushButton()
self.manipulate_roi_window_instance_save_button.setObjectName(
"ManipulateRoiWindowInstanceSaveButton")
self.manipulate_roi_window_instance_save_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum))
self.manipulate_roi_window_instance_save_button.resize(
self.manipulate_roi_window_instance_save_button.sizeHint().width(),
self.manipulate_roi_window_instance_save_button.sizeHint().height(
))
self.manipulate_roi_window_instance_save_button.setProperty(
"QPushButtonClass", "success-button")
icon_save = QtGui.QIcon()
icon_save.addPixmap(
QtGui.QPixmap(resource_path('res/images/btn-icons/save_icon.png')))
self.manipulate_roi_window_instance_save_button.setIcon(icon_save)
self.manipulate_roi_window_instance_save_button.clicked.connect(
self.onSaveClicked)
self.manipulate_roi_window_cancel_save_box.addWidget(
self.manipulate_roi_window_instance_save_button)
self.manipulate_roi_window_input_container_box.addRow(
self.manipulate_roi_window_cancel_save_box)
# Creating a horizontal box to hold the ROI view and the preview
self.manipulate_roi_window_instance_view_box = QHBoxLayout()
self.manipulate_roi_window_instance_view_box.setObjectName(
"ManipulateRoiWindowInstanceViewBoxes")
# Font for the ROI view and preview's labels
font = QFont()
font.setBold(True)
font.setPixelSize(20)
# Creating the ROI view
self.ROI_view_box_layout = QVBoxLayout()
self.ROI_view_box_label = QLabel()
self.ROI_view_box_label.setFont(font)
self.ROI_view_box_label.setAlignment(Qt.AlignHCenter)
self.ROI_view_box_layout.addWidget(self.ROI_view_box_label)
self.ROI_view_box_layout.addWidget(self.dicom_view)
self.ROI_view_box_widget = QWidget()
self.ROI_view_box_widget.setLayout(self.ROI_view_box_layout)
# Creating the preview
self.preview_box_layout = QVBoxLayout()
self.preview_box_label = QLabel()
self.preview_box_label.setFont(font)
self.preview_box_label.setAlignment(Qt.AlignHCenter)
self.preview_box_layout.addWidget(self.preview_box_label)
self.preview_box_layout.addWidget(self.dicom_preview)
self.preview_box_widget = QWidget()
self.preview_box_widget.setLayout(self.preview_box_layout)
# Add View and Slider into horizontal box
self.manipulate_roi_window_instance_view_box.addWidget(
self.ROI_view_box_widget)
self.manipulate_roi_window_instance_view_box.addWidget(
self.preview_box_widget)
# Create a widget to hold the image slice box
self.manipulate_roi_window_instance_view_widget = QWidget()
self.manipulate_roi_window_instance_view_widget.setObjectName(
"ManipulateRoiWindowInstanceActionWidget")
self.manipulate_roi_window_instance_view_widget.setLayout(
self.manipulate_roi_window_instance_view_box)
# Create a horizontal box for containing the input fields and the
# viewports
self.manipulate_roi_window_main_box = QHBoxLayout()
self.manipulate_roi_window_main_box.setObjectName(
"ManipulateRoiWindowMainBox")
self.manipulate_roi_window_main_box.addLayout(
self.manipulate_roi_window_input_container_box, 1)
self.manipulate_roi_window_main_box.addWidget(
self.manipulate_roi_window_instance_view_widget, 11)
# Create a new central widget to hold the horizontal box layout
self.manipulate_roi_window_instance_central_widget = QWidget()
self.manipulate_roi_window_instance_central_widget.setObjectName(
"ManipulateRoiWindowInstanceCentralWidget")
self.manipulate_roi_window_instance_central_widget.setLayout(
self.manipulate_roi_window_main_box)
self.retranslate_ui(self.manipulate_roi_window_instance)
self.manipulate_roi_window_instance.setStyleSheet(stylesheet)
self.manipulate_roi_window_instance.setCentralWidget(
self.manipulate_roi_window_instance_central_widget)
QtCore.QMetaObject.connectSlotsByName(
self.manipulate_roi_window_instance)
def dicom_view_slider_value_changed(self):
"""
Display selected ROIs in dropbox when moving to another image slice
"""
self.display_selected_roi()
if self.dicom_preview.slider.value() != self.dicom_view.slider.value():
self.dicom_preview.slider.setValue(self.dicom_view.slider.value())
def dicom_preview_slider_value_changed(self):
"""
Display generated ROI when moving to another image slice
"""
self.draw_roi()
if self.dicom_preview.slider.value() != self.dicom_view.slider.value():
self.dicom_view.slider.setValue(self.dicom_preview.slider.value())
def onCancelButtonClicked(self):
"""
This function is used for canceling the drawing
"""
self.close()
def onDrawButtonClicked(self):
"""
Function triggered when the Draw button is pressed from the menu.
"""
# Hide warning message
self.warning_message.setVisible(False)
# Check inputs
selected_operation = self.operation_name_dropdown_list.currentText()
roi_1 = self.first_roi_name_dropdown_list.currentText()
roi_2 = self.second_roi_name_dropdown_list.currentText()
new_roi_name = self.new_roi_name_line_edit.text()
# Check the selected inputs and execute the operations
if roi_1 != "" and new_roi_name != "" and \
self.margin_line_edit.text() != "" and \
selected_operation in self.single_roi_operation_names:
# Single ROI operations
dict_rois_contours = ROI.get_roi_contour_pixel(
self.patient_dict_container.get("raw_contour"), [roi_1],
self.patient_dict_container.get("pixluts"))
roi_geometry = ROI.roi_to_geometry(dict_rois_contours[roi_1])
margin = float(self.margin_line_edit.text())
if selected_operation == self.single_roi_operation_names[0]:
new_geometry = ROI.scale_roi(roi_geometry, margin)
elif selected_operation == self.single_roi_operation_names[1]:
new_geometry = ROI.scale_roi(roi_geometry, -margin)
elif selected_operation == self.single_roi_operation_names[2]:
new_geometry = ROI.rind_roi(roi_geometry, -margin)
else:
new_geometry = ROI.rind_roi(roi_geometry, margin)
self.new_ROI_contours = ROI.geometry_to_roi(new_geometry)
self.draw_roi()
return True
elif roi_1 != "" and roi_2 != "" and new_roi_name != "" and \
selected_operation in self.multiple_roi_operation_names:
# Multiple ROI operations
dict_rois_contours = ROI.get_roi_contour_pixel(
self.patient_dict_container.get("raw_contour"), [roi_1, roi_2],
self.patient_dict_container.get("pixluts"))
roi_1_geometry = ROI.roi_to_geometry(dict_rois_contours[roi_1])
roi_2_geometry = ROI.roi_to_geometry(dict_rois_contours[roi_2])
# Execute the selected operation
new_geometry = ROI.manipulate_rois(roi_1_geometry, roi_2_geometry,
selected_operation.upper())
self.new_ROI_contours = ROI.geometry_to_roi(new_geometry)
self.draw_roi()
return True
self.warning_message_text.setText("Not all values are specified.")
self.warning_message.setVisible(True)
return False
def onSaveClicked(self):
""" Save the new ROI """
# Get the name of the new ROI
new_roi_name = self.new_roi_name_line_edit.text()
# If the new ROI hasn't been drawn, draw the new ROI. Then if the new
# ROI is drawn successfully, proceed to save the new ROI.
if self.new_ROI_contours is None:
if not self.onDrawButtonClicked():
return
# Get a dict to convert SOPInstanceUID to slice id
slice_ids_dict = get_dict_slice_to_uid(PatientDictContainer())
# Transform new_ROI_contours to a list of roi information
rois_to_save = {}
for uid, contour_sequence in self.new_ROI_contours.items():
slider_id = slice_ids_dict[uid]
location = self.patient_dict_container.filepaths[slider_id]
ds = pydicom.dcmread(location)
slice_info = {'coords': contour_sequence, 'ds': ds}
rois_to_save[slider_id] = slice_info
roi_list = ROI.convert_hull_list_to_contours_data(
rois_to_save, self.patient_dict_container)
connectSaveROIProgress(self, roi_list, self.dataset_rtss, new_roi_name,
self.roi_saved)
def draw_roi(self):
""" Draw the new ROI """
# Get the new ROI's name
new_roi_name = self.new_roi_name_line_edit.text()
# Check if the new ROI contour is None
if self.new_ROI_contours is None:
return
# Get the info required to draw the new ROI
slider_id = self.dicom_preview.slider.value()
curr_slice = self.patient_dict_container.get("dict_uid")[slider_id]
# Calculate the new ROI's polygon
dict_ROI_contours = {}
dict_ROI_contours[new_roi_name] = self.new_ROI_contours
polygons = ROI.calc_roi_polygon(new_roi_name, curr_slice,
dict_ROI_contours)
# Set the new ROI color
color = QtGui.QColor()
color.setRgb(90, 250, 175, 200)
pen_color = QtGui.QColor(color.red(), color.green(), color.blue())
pen = QtGui.QPen(pen_color)
pen.setStyle(QtCore.Qt.PenStyle(1))
pen.setWidthF(2.0)
# Draw the new ROI
self.dicom_preview.update_view()
for i in range(len(polygons)):
self.dicom_preview.scene.addPolygon(polygons[i], pen,
QtGui.QBrush(color))
def update_selected_rois(self):
""" Get the names of selected ROIs """
# Hide warning message
self.warning_message.setVisible(False)
self.roi_names = []
if self.first_roi_name_dropdown_list.currentText() != "":
self.roi_names.append(
self.first_roi_name_dropdown_list.currentText())
if self.second_roi_name_dropdown_list.currentText() != "" and \
self.second_roi_name_dropdown_list.isVisible():
self.roi_names.append(
self.second_roi_name_dropdown_list.currentText())
self.dict_rois_contours_axial = ROI.get_roi_contour_pixel(
self.patient_dict_container.get("raw_contour"), self.roi_names,
self.patient_dict_container.get("pixluts"))
self.display_selected_roi()
def display_selected_roi(self):
""" Display selected ROIs """
# Get the info required to display the selected ROIs
slider_id = self.dicom_view.slider.value()
curr_slice = self.patient_dict_container.get("dict_uid")[slider_id]
self.rois = self.patient_dict_container.get("rois")
# Display the selected ROIs
self.dicom_view.update_view()
for roi_id, roi_dict in self.rois.items():
roi_name = roi_dict['name']
if roi_name in self.roi_names:
polygons = ROI.calc_roi_polygon(roi_name, curr_slice,
self.dict_rois_contours_axial)
self.dicom_view.draw_roi_polygons(roi_id, polygons,
self.roi_color)
def operation_changed(self):
""" Change the form when users select different operations """
# Hide warning message
self.warning_message.setVisible(False)
selected_operation = self.operation_name_dropdown_list.currentText()
if selected_operation in self.single_roi_operation_names:
self.second_roi_name_label.setVisible(False)
self.second_roi_name_dropdown_list.setVisible(False)
self.margin_label.setVisible(True)
self.margin_line_edit.setVisible(True)
self.update_selected_rois()
else:
self.second_roi_name_label.setVisible(True)
self.second_roi_name_dropdown_list.setVisible(True)
self.margin_label.setVisible(False)
self.margin_line_edit.setVisible(False)
self.update_selected_rois()
def roi_saved(self, new_rtss):
""" Create a new ROI in Structure Tab and notify user """
new_roi_name = self.new_roi_name_line_edit.text()
self.signal_roi_manipulated.emit((new_rtss, {"draw": new_roi_name}))
QMessageBox.about(self.manipulate_roi_window_instance, "Saved",
"New contour successfully created!")
self.close()
class ParameterEditorItem(QWidget):
onRemoveParameter = Signal(Parameter)
onMoveParameterUp = Signal(Parameter)
onMoveParameterDown = Signal(Parameter)
onChanged = Signal()
def __init__(self, parameter: Parameter):
super().__init__()
self.parameter = parameter
deleteButton = QToolButton()
deleteButton.setText("X")
deleteButton.clicked.connect(
lambda: self.onRemoveParameter.emit(self.parameter))
upButton = QToolButton()
upButton.setText("\u2191")
upButton.clicked.connect(
lambda: self.onMoveParameterUp.emit(self.parameter))
downButton = QToolButton()
downButton.setText("\u2193")
downButton.clicked.connect(
lambda: self.onMoveParameterDown.emit(self.parameter))
buttonsLayout = QVBoxLayout()
buttonsLayout.setAlignment(Qt.AlignTop)
buttonsLayout.addWidget(deleteButton)
buttonsLayout.addWidget(upButton)
buttonsLayout.addWidget(downButton)
self._nameLabel = QLabel("Name")
self._nameField = QLineEdit()
self._nameField.textChanged.connect(self.OnChanged)
self._dataTypeLabel = QLabel("Data Type")
self._dataTypeField = QComboBox()
for dataType in DataType:
self._dataTypeField.addItem(dataType.ToString(), userData=dataType)
self._dataTypeField.currentIndexChanged.connect(self.OnChanged)
self._defaultValueLabel = QLabel("Default Value")
self._defaultInteger = QSpinBox()
self._defaultInteger.valueChanged.connect(self.OnChanged)
self._defaultFloat = QDoubleSpinBox()
self._defaultFloat.valueChanged.connect(self.OnChanged)
self._defaultBoolean = QComboBox()
self._defaultBoolean.addItem("True", True)
self._defaultBoolean.addItem("False", False)
self._defaultBoolean.currentIndexChanged.connect(self.OnChanged)
self._defaultString = QLineEdit()
self._defaultString.textChanged.connect(self.OnChanged)
self._minimumLabel = QLabel("Minimum")
self._minimumFloat = QDoubleSpinBox()
self._minimumFloat.valueChanged.connect(self.OnChanged)
self._minimumInteger = QSpinBox()
self._minimumInteger.valueChanged.connect(self.OnChanged)
self._maximumLabel = QLabel("Maximum")
self._maximumFloat = QDoubleSpinBox()
self._maximumFloat.valueChanged.connect(self.OnChanged)
self._maximumInteger = QSpinBox()
self._maximumInteger.valueChanged.connect(self.OnChanged)
gridLayout = QGridLayout()
gridLayout.setAlignment(Qt.AlignTop)
gridLayout.addWidget(self._nameLabel, 0, 0)
gridLayout.addWidget(self._nameField, 0, 1)
gridLayout.addWidget(self._dataTypeLabel, 1, 0)
gridLayout.addWidget(self._dataTypeField, 1, 1)
gridLayout.addWidget(self._defaultValueLabel, 2, 0)
for defaultField in [
self._defaultInteger, self._defaultFloat, self._defaultBoolean,
self._defaultString
]:
gridLayout.addWidget(defaultField, 2, 1)
gridLayout.addWidget(self._minimumLabel, 3, 0)
gridLayout.addWidget(self._minimumInteger, 3, 1)
gridLayout.addWidget(self._minimumFloat, 3, 1)
gridLayout.addWidget(self._maximumLabel, 4, 0)
gridLayout.addWidget(self._maximumInteger, 4, 1)
gridLayout.addWidget(self._maximumFloat, 4, 1)
layout = QHBoxLayout()
layout.addLayout(buttonsLayout)
layout.addLayout(gridLayout)
self.setLayout(layout)
self.SetFieldsFromParameter()
def SetFieldsFromParameter(self):
self._nameField.setText(self.parameter.name)
self._dataTypeField.setCurrentText(self.parameter.dataType.ToString())
minFloat, maxFloat = self.parameter.minimumFloat, self.parameter.maximumFloat
minInt, maxInt = self.parameter.minimumInteger, self.parameter.maximumInteger
self._minimumFloat.setRange(-2**30, maxFloat)
self._maximumFloat.setRange(minFloat, 2**30)
self._minimumInteger.setRange(-2**30, maxInt)
self._maximumInteger.setRange(minInt, 2**30)
if self._minimumFloat.value() != minFloat:
self._minimumFloat.setValue(minFloat)
if self._maximumFloat.value() != maxFloat:
self._maximumFloat.setValue(maxFloat)
if self._minimumInteger.value() != minInt:
self._minimumInteger.setValue(minInt)
if self._maximumInteger.value() != maxInt:
self._maximumInteger.setValue(maxInt)
self._defaultInteger.setRange(minInt, maxInt)
self._defaultFloat.setRange(minFloat, maxFloat)
if self._defaultInteger.value() != self.parameter.defaultValueDict[
DataType.INTEGER]:
self._defaultInteger.setValue(
self.parameter.defaultValueDict[DataType.INTEGER])
if self._defaultFloat.value() != self.parameter.defaultValueDict[
DataType.FLOAT]:
self._defaultFloat.setValue(
self.parameter.defaultValueDict[DataType.FLOAT])
if self._defaultBoolean.currentData(
) != self.parameter.defaultValueDict[DataType.BOOLEAN]:
self._defaultBoolean.setCurrentText(
str(self.parameter.defaultValueDict[DataType.BOOLEAN]))
if self._defaultString.text() != self.parameter.defaultValueDict[
DataType.STRING]:
self._defaultString.setText(
self.parameter.defaultValueDict[DataType.STRING])
self.UpdateVisibility()
def UpdateVisibility(self):
self._defaultInteger.setVisible(
self._dataTypeField.currentData() is DataType.INTEGER)
self._defaultFloat.setVisible(
self._dataTypeField.currentData() is DataType.FLOAT)
self._defaultBoolean.setVisible(
self._dataTypeField.currentData() is DataType.BOOLEAN)
self._defaultString.setVisible(
self._dataTypeField.currentData() is DataType.STRING)
self._minimumLabel.setVisible(self._dataTypeField.currentData() in
[DataType.INTEGER, DataType.FLOAT])
self._maximumLabel.setVisible(self._dataTypeField.currentData() in
[DataType.INTEGER, DataType.FLOAT])
self._minimumInteger.setVisible(
self._dataTypeField.currentData() is DataType.INTEGER)
self._maximumInteger.setVisible(
self._dataTypeField.currentData() is DataType.INTEGER)
self._minimumFloat.setVisible(
self._dataTypeField.currentData() is DataType.FLOAT)
self._maximumFloat.setVisible(
self._dataTypeField.currentData() is DataType.FLOAT)
self._minimumLabel.setVisible(self._dataTypeField.currentData() in
[DataType.INTEGER, DataType.FLOAT])
self._maximumLabel.setVisible(self._dataTypeField.currentData() in
[DataType.INTEGER, DataType.FLOAT])
self._defaultValueLabel.setVisible(self._dataTypeField.currentData(
) not in [DataType.VALVE, DataType.PROGRAM, DataType.PROGRAM_PRESET])
def OnChanged(self):
self.UpdateVisibility()
self._minimumFloat.setMaximum(self._maximumFloat.value())
self._maximumFloat.setMinimum(self._minimumFloat.value())
self._minimumInteger.setMaximum(self._maximumInteger.value())
self._maximumInteger.setMinimum(self._minimumInteger.value())
self._defaultInteger.setRange(self._minimumInteger.value(),
self._maximumInteger.value())
self._defaultFloat.setRange(self._minimumFloat.value(),
self._maximumFloat.value())
self.onChanged.emit()
def UpdateParameter(self):
self.parameter.name = self._nameField.text()
self.parameter.dataType = self._dataTypeField.currentData()
self.parameter.defaultValueDict[
DataType.INTEGER] = self._defaultInteger.value()
self.parameter.defaultValueDict[
DataType.FLOAT] = self._defaultFloat.value()
self.parameter.defaultValueDict[
DataType.BOOLEAN] = self._defaultBoolean.currentData()
self.parameter.defaultValueDict[
DataType.STRING] = self._defaultString.text()
self.parameter.minimumInteger = self._minimumInteger.value()
self.parameter.maximumInteger = self._maximumInteger.value()
self.parameter.minimumFloat = self._minimumFloat.value()
self.parameter.maximumFloat = self._maximumFloat.value()
self.parameter.ValidateDefaultValues()
self.SetFieldsFromParameter()
