def __init__(self, data):
global instance_id
QWidget.__init__(self)
self.actionHandler = UIActionHandler()
self.actionHandler.setupActionHandler(self)
offset_layout = QHBoxLayout()
offset_layout.addWidget(QLabel("Offset: "))
self.offset = QLabel(hex(0))
offset_layout.addWidget(self.offset)
offset_layout.setAlignment(QtCore.Qt.AlignCenter)
datatype_layout = QHBoxLayout()
datatype_layout.addWidget(QLabel("Data Type: "))
self.datatype = QLabel("")
datatype_layout.addWidget(self.datatype)
datatype_layout.setAlignment(QtCore.Qt.AlignCenter)
layout = QVBoxLayout()
title = QLabel("Hello Pane", self)
title.setAlignment(QtCore.Qt.AlignCenter)
instance = QLabel("Instance: " + str(instance_id), self)
instance.setAlignment(QtCore.Qt.AlignCenter)
layout.addStretch()
layout.addWidget(title)
layout.addWidget(instance)
layout.addLayout(datatype_layout)
layout.addLayout(offset_layout)
layout.addStretch()
self.setLayout(layout)
instance_id += 1
self.data = data
# Populate initial state
self.updateState()
# Set up view and address change notifications
self.notifications = HelloNotifications(self)
def __init__(self, name):
global instance_id
GlobalAreaWidget.__init__(self, name)
self.actionHandler = UIActionHandler()
self.actionHandler.setupActionHandler(self)
offset_layout = QHBoxLayout()
offset_layout.addWidget(QLabel("Offset: "))
self.offset = QLabel(hex(0))
offset_layout.addWidget(self.offset)
offset_layout.setAlignment(QtCore.Qt.AlignCenter)
datatype_layout = QHBoxLayout()
datatype_layout.addWidget(QLabel("Data Type: "))
self.datatype = QLabel("")
datatype_layout.addWidget(self.datatype)
datatype_layout.setAlignment(QtCore.Qt.AlignCenter)
layout = QVBoxLayout()
title = QLabel(name, self)
title.setAlignment(QtCore.Qt.AlignCenter)
instance = QLabel("Instance: " + str(instance_id), self)
instance.setAlignment(QtCore.Qt.AlignCenter)
layout.addStretch()
layout.addWidget(title)
layout.addWidget(instance)
layout.addLayout(datatype_layout)
layout.addLayout(offset_layout)
layout.addStretch()
self.setLayout(layout)
instance_id += 1
self.data = None
def set_item_data(self, i, DisplayName, name, sid):
check_box_layout = QHBoxLayout()
check_box_layout.setAlignment(Qt.AlignmentFlag.AlignCenter)
check_box = QCheckBox()
check_box.setObjectName('check_box_' + str(i))
check_box_layout.addWidget(check_box)
check_box_layout.setContentsMargins(0, 0, 0, 0)
check_box_widget = QWidget()
check_box_widget.setLayout(check_box_layout)
self.tableWidget.item(i, 2).setText(DisplayName)
self.tableWidget.item(i, 3).setText(name)
self.tableWidget.item(i, 4).setText(sid)
self.tableWidget.setCellWidget(i, 0, check_box_widget)
def set_item_data(i, fun_name, fun_content):
# 显示虚拟机名称、状态
self.tableWidget.item(i, 0).setText(fun_name)
if fun_name in self.fun_explain:
self.tableWidget.item(i, 1).setText(self.fun_explain[fun_name])
self.tableWidget.item(i, 2).setText(fun_content)
# 显示修改框
if fun_name in self.fun_modifiable:
lineedit = QLineEdit()
lineedit.setObjectName("lineEdit_1_{}".format(i))
lineedit.setPlaceholderText("请输入修改值")
lineedit.setStyleSheet("QLineEdit {border: none;}")
self.lineedit_list.append(lineedit) # 存储lineedit
lineedit_layout = QHBoxLayout()
lineedit_layout.setAlignment(Qt.AlignmentFlag.AlignCenter)
lineedit_layout.setContentsMargins(0, 0, 0, 0)
lineedit_layout.addWidget(lineedit)
lineedit_widget = QWidget()
lineedit_widget.setLayout(lineedit_layout)
self.tableWidget.setCellWidget(i, 3, lineedit_widget)
def __init__(self, parent, name, data):
QWidget.__init__(self, parent)
DockContextHandler.__init__(self, self, name)
self.actionHandler = UIActionHandler()
self.actionHandler.setupActionHandler(self)
status_layout = QHBoxLayout()
status_layout.addWidget(QLabel('Status: '))
self.status = QLabel('idle')
status_layout.addWidget(self.status)
status_layout.setAlignment(QtCore.Qt.AlignCenter)
client_dbg_layout = QHBoxLayout()
client_dbg_layout.addWidget(QLabel('Client debugger: '))
self.client_dbg = QLabel('n/a')
client_dbg_layout.addWidget(self.client_dbg)
client_dbg_layout.setAlignment(QtCore.Qt.AlignCenter)
client_pgm_layout = QHBoxLayout()
client_pgm_layout.addWidget(QLabel('Client program: '))
self.client_pgm = QLabel('n/a')
client_pgm_layout.addWidget(self.client_pgm)
client_pgm_layout.setAlignment(QtCore.Qt.AlignCenter)
layout = QVBoxLayout()
layout.addStretch()
layout.addLayout(status_layout)
layout.addLayout(client_dbg_layout)
layout.addLayout(client_pgm_layout)
layout.addStretch()
self.setLayout(layout)
def __init__(self, text, parent):
super().__init__(parent)
layout = QHBoxLayout()
layout.setSpacing(0)
layout.setAlignment(Qt.AlignCenter)
if isinstance(text, str):
widget = QLabel(text, self)
widget.setFont(default_font)
layout.addWidget(widget)
if isinstance(text, list):
for t in text:
if isinstance(t, str):
widget = QLabel(t, self)
elif isinstance(t, list):
widget = GlossBox(t[0], t[1], self)
widget.setFont(default_font)
layout.addWidget(widget)
self.setLayout(layout)
class UITransferROIWindow:
def setup_ui(self, transfer_roi_window_instance,
signal_roi_transferred_to_fixed_container,
signal_roi_transferred_to_moving_container):
self.patient_dict_container = PatientDictContainer()
self.moving_dict_container = MovingDictContainer()
self.fixed_image_initial_rois = self.patient_dict_container.get("rois")
self.moving_image_initial_rois = self.moving_dict_container.get("rois")
self.transfer_roi_window_instance = transfer_roi_window_instance
self.signal_roi_transferred_to_fixed_container = \
signal_roi_transferred_to_fixed_container
self.signal_roi_transferred_to_moving_container = \
signal_roi_transferred_to_moving_container
self.fixed_to_moving_rois = {}
self.moving_to_fixed_rois = {}
self.add_suffix = True
self.progress_window = ProgressWindow(
self, QtCore.Qt.WindowTitleHint | QtCore.Qt.WindowCloseButtonHint)
self.progress_window.setFixedSize(250, 100)
self.progress_window.signal_loaded \
.connect(self.onTransferRoiFinished)
self.progress_window.signal_error.connect(self.onTransferRoiError)
self.init_layout()
def retranslate_ui(self, transfer_roi_window_instance):
_translate = QtCore.QCoreApplication.translate
transfer_roi_window_instance.setWindowTitle(
_translate("TransferRoiWindowInstance",
"OnkoDICOM - Transfer Region of Interest"))
self.add_suffix_checkbox.setText(
_translate("AddSuffixCheckBox", "Add Suffix"))
self.patient_A_label.setText(
_translate("PatientAROILabel", "First Image Set ROIs"))
self.patient_B_label.setText(
_translate("PatientBROILabel", "Second Image Set ROIs"))
self.transfer_all_rois_to_patient_B_button.setText(
_translate("ROITransferToBButton", "All"))
self.transfer_all_rois_to_patient_A_button.setText(
_translate("ROITransferToAButton", "All"))
self.save_button.setText(_translate("SaveButton", "Save"))
self.reset_button.setText(_translate("ResetButton", "Reset"))
def init_layout(self):
"""
Initialize the layout for the Transfer ROI Window.
"""
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.transfer_roi_window_instance.setObjectName(
"TransferRoiWindowInstance")
self.transfer_roi_window_instance.setWindowIcon(window_icon)
# Creating a grid layout to hold all elements
self.transfer_roi_window_grid_layout = QGridLayout()
self.transfer_roi_window_grid_layout.setColumnStretch(0, 1)
self.transfer_roi_window_grid_layout.setColumnStretch(1, 1)
self.transfer_roi_window_grid_layout.setColumnStretch(2, 1)
self.init_patient_labels()
self.init_transfer_arrow_buttons()
self.init_patient_A_initial_roi_list()
self.init_patient_B_rois_to_A_layout()
self.init_patient_A_rois_to_B_layout()
self.init_patient_B_initial_roi_list()
self.init_add_suffix_checkbox()
self.init_save_and_reset_button_layout()
# Create a new central widget to hold the grid layout
self.transfer_roi_window_instance_central_widget = QWidget()
self.transfer_roi_window_instance_central_widget.setLayout(
self.transfer_roi_window_grid_layout)
self.retranslate_ui(self.transfer_roi_window_instance)
self.transfer_roi_window_instance.setStyleSheet(stylesheet)
self.transfer_roi_window_instance.setCentralWidget(
self.transfer_roi_window_instance_central_widget)
QtCore.QMetaObject.connectSlotsByName(
self.transfer_roi_window_instance)
def init_transfer_arrow_buttons(self):
"""
Initialize the layout for arrow buttons
"""
self.transfer_all_rois_to_patient_B_button = QPushButton()
self.transfer_all_rois_to_patient_B_button.setObjectName(
"ROITransferToBButton")
transfer_all_rois_to_patient_B_icon = QtGui.QIcon()
transfer_all_rois_to_patient_B_icon.addPixmap(
QtGui.QPixmap(
resource_path('res/images/btn-icons/forward_slide_icon.png')),
QtGui.QIcon.Normal, QtGui.QIcon.On)
self.transfer_all_rois_to_patient_B_button \
.setIcon(transfer_all_rois_to_patient_B_icon)
self.transfer_all_rois_to_patient_B_button.clicked.connect(
self.transfer_all_rois_to_patient_B_button_clicked)
self.transfer_roi_window_grid_layout.addWidget(
self.transfer_all_rois_to_patient_B_button, 1, 1)
self.transfer_all_rois_to_patient_A_button = QPushButton()
self.transfer_all_rois_to_patient_A_button.setObjectName(
"ROITransferToAButton")
self.transfer_all_rois_to_patient_A_button.setMaximumWidth(100)
transfer_all_rois_to_patient_A_icon = QtGui.QIcon()
transfer_all_rois_to_patient_A_icon.addPixmap(
QtGui.QPixmap(
resource_path('res/images/btn-icons/backward_slide_icon.png')),
QtGui.QIcon.Normal, QtGui.QIcon.On)
self.transfer_all_rois_to_patient_A_button \
.setIcon(transfer_all_rois_to_patient_A_icon)
self.transfer_all_rois_to_patient_A_button.clicked.connect(
self.transfer_all_rois_to_patient_A_button_clicked)
self.transfer_roi_window_grid_layout.addWidget(
self.transfer_all_rois_to_patient_A_button, 2, 1)
def transfer_all_rois_to_patient_B_button_clicked(self):
"""
This function is triggered when the right arrow button is clicked.
"""
self.fixed_to_moving_rois.clear()
self.patient_A_rois_to_B_list_widget.clear()
for i in range(0, len(self.fixed_image_initial_rois)):
self.patient_A_initial_roi_double_clicked(
self.patient_A_initial_rois_list_widget.item(i))
def transfer_all_rois_to_patient_A_button_clicked(self):
"""
This function is triggered when the left arrow button is clicked.
"""
self.moving_to_fixed_rois.clear()
self.patient_B_rois_to_A_list_widget.clear()
for i in range(0, len(self.moving_image_initial_rois)):
self.patient_B_initial_roi_double_clicked(
self.patient_B_initial_rois_list_widget.item(i))
def init_add_suffix_checkbox(self):
"""
Initialize the layout for add suffix checkbox
"""
self.add_suffix_checkbox = QCheckBox()
self.add_suffix_checkbox.setObjectName("AddSuffixCheckBox")
self.add_suffix_checkbox.setChecked(self.add_suffix)
self.add_suffix_checkbox.clicked.connect(
self.add_suffix_checkbox_clicked)
self.transfer_roi_window_grid_layout.addWidget(
self.add_suffix_checkbox, 3, 0)
def init_patient_labels(self):
"""
Initialize the layout for two patient labels
"""
self.patient_A_label = QLabel()
self.patient_A_label.setObjectName("PatientAROILabel")
self.patient_A_label.setMinimumHeight(50)
self.patient_A_label.setAlignment(Qt.AlignCenter)
self.patient_A_label.setStyleSheet(
"QLabel { background-color : green; color : white; "
"font-size: 15pt; font-weight: bold;}")
self.patient_B_label = QLabel()
self.patient_B_label.setObjectName("PatientBROILabel")
self.patient_B_label.setMinimumHeight(50)
self.patient_B_label.setAlignment(Qt.AlignCenter)
self.patient_B_label.setStyleSheet(
"QLabel { background-color : red; color : white; "
"font-size: 15pt; font-weight: bold;}")
self.transfer_roi_window_grid_layout.addWidget(self.patient_A_label, 0,
0)
self.transfer_roi_window_grid_layout.addWidget(self.patient_B_label, 0,
2)
def init_save_and_reset_button_layout(self):
"""
Initialize the layout for save and reset buttons
"""
self.reset_and_save_buttons_layout = QHBoxLayout()
self.reset_button = QPushButton()
self.reset_button.setObjectName("ResetButton")
self.reset_button.clicked.connect(self.reset_clicked)
self.save_button = QPushButton()
self.save_button.setObjectName("SaveButton")
self.save_button.setDisabled(True)
self.save_button.clicked.connect(self.transfer_roi_clicked)
self.reset_and_save_buttons_layout.setAlignment(Qt.AlignRight)
self.reset_and_save_buttons_layout.addWidget(self.reset_button)
self.reset_and_save_buttons_layout.addWidget(self.save_button)
# Create a widget to hold Reset and Save buttons
self.reset_and_save_button_central_widget = QWidget()
self.reset_and_save_button_central_widget.setLayout(
self.reset_and_save_buttons_layout)
self.transfer_roi_window_grid_layout.addWidget(
self.reset_and_save_button_central_widget, 3, 2)
def add_suffix_checkbox_clicked(self):
"""
This function is triggered when the add suffix checkbox is clicked
"""
self.add_suffix = self.add_suffix_checkbox.isChecked()
def init_patient_B_rois_to_A_layout(self):
"""
Initialize the layout for transfer rois from B to A container
"""
# Create scrolling area widget to contain the content.
self.patient_B_rois_to_A_list_widget = QListWidget(self)
self.transfer_roi_window_grid_layout \
.addWidget(self.patient_B_rois_to_A_list_widget, 2, 0)
self.patient_B_rois_to_A_list_widget.itemDoubleClicked.connect(
self.patient_B_to_A_rois_double_clicked)
def init_patient_A_rois_to_B_layout(self):
"""
Initialize the layout for transfer rois from A to B container
"""
self.patient_A_rois_to_B_list_widget = QListWidget(self)
self.transfer_roi_window_grid_layout \
.addWidget(self.patient_A_rois_to_B_list_widget, 1, 2)
self.patient_A_rois_to_B_list_widget.itemDoubleClicked.connect(
self.patient_A_to_B_rois_double_clicked)
def init_patient_A_initial_roi_list(self):
"""
Initialize the layout for patient A's roi list
"""
self.patient_A_initial_rois_list_widget = QListWidget(self)
self.patient_A_initial_rois_list_widget.itemDoubleClicked.connect(
self.patient_A_initial_roi_double_clicked)
for idx in self.fixed_image_initial_rois:
roi_label = QListWidgetItem(
self.fixed_image_initial_rois[idx]['name'])
roi_label.setForeground(Qt.darkGreen)
roi_label.setData(Qt.UserRole, self.fixed_image_initial_rois[idx])
self.patient_A_initial_rois_list_widget.addItem(roi_label)
self.transfer_roi_window_grid_layout.addWidget(
self.patient_A_initial_rois_list_widget, 1, 0)
def init_patient_B_initial_roi_list(self):
"""
Initialize the layout for patient B's roi list
"""
self.patient_B_initial_rois_list_widget = QListWidget(self)
self.patient_B_initial_rois_list_widget.itemDoubleClicked.connect(
self.patient_B_initial_roi_double_clicked)
for idx in self.moving_image_initial_rois:
roi_label = QListWidgetItem(
self.moving_image_initial_rois[idx]['name'])
roi_label.setForeground(Qt.red)
roi_label.setData(Qt.UserRole, self.moving_image_initial_rois[idx])
self.patient_B_initial_rois_list_widget.addItem(roi_label)
self.transfer_roi_window_grid_layout.addWidget(
self.patient_B_initial_rois_list_widget, 2, 2)
def patient_A_to_B_rois_double_clicked(self, item):
"""
This function is triggered when a roi in "A to B" list is
double-clicked.
"""
roi_to_remove = item.data(Qt.UserRole)
to_delete_value = roi_to_remove['name']
self.fixed_to_moving_rois.pop(to_delete_value)
self.patient_A_rois_to_B_list_widget.clear()
for key, value in self.fixed_to_moving_rois.items():
roi_label = QListWidgetItem(value)
roi_label.setForeground(Qt.red)
roi_label.setData(Qt.UserRole, {'name': key})
self.patient_A_rois_to_B_list_widget.addItem(roi_label)
if self.transfer_list_is_empty():
self.save_button.setDisabled(True)
def patient_B_to_A_rois_double_clicked(self, item):
"""
This function is triggered when a roi in "B to A" list is
double-clicked.
"""
roi_to_remove = item.data(Qt.UserRole)
to_delete_value = roi_to_remove['name']
self.moving_to_fixed_rois.pop(to_delete_value)
self.patient_B_rois_to_A_list_widget.clear()
for key, value in self.moving_to_fixed_rois.items():
roi_label = QListWidgetItem(value)
roi_label.setForeground(Qt.red)
roi_label.setData(Qt.UserRole, {'name': key})
self.patient_B_rois_to_A_list_widget.addItem(roi_label)
if self.transfer_list_is_empty():
self.save_button.setDisabled(True)
def patient_A_initial_roi_double_clicked(self, item):
"""
This function is triggered when a roi in patient A's roi list is
double-clicked.
"""
roi_to_add = item.data(Qt.UserRole)
transferred_roi_name = roi_to_add['name']
# If the clicked roi is already transferred, return
if transferred_roi_name in self.fixed_to_moving_rois.keys():
QMessageBox.about(self, "Transfer Failed",
"Chosen ROI has already been transferred!")
return
# Create a set to store all current roi names in target patient
# including both initial rois name and added roi names so far
patient_B_initial_roi_name_list = set()
for item in self.fixed_to_moving_rois.values():
patient_B_initial_roi_name_list.add(item)
for idx in self.moving_image_initial_rois:
patient_B_initial_roi_name_list.add(
self.moving_image_initial_rois[idx]['name'])
# Check if clicked roi name has duplicate
# in patient B's initial roi names list
if transferred_roi_name in patient_B_initial_roi_name_list:
if self.add_suffix:
transferred_roi_name = generate_non_duplicated_name(
transferred_roi_name, patient_B_initial_roi_name_list)
else:
QMessageBox.about(
self, "Transfer Failed", "Duplicated ROI name. "
"Please consider adding suffix.")
return
# Add clicked roi to transferred list
self.fixed_to_moving_rois[roi_to_add['name']] = transferred_roi_name
roi_label = QListWidgetItem(transferred_roi_name)
roi_label.setForeground(Qt.red)
roi_label.setData(Qt.UserRole, roi_to_add)
self.patient_A_rois_to_B_list_widget.addItem(roi_label)
self.save_button.setDisabled(False)
def patient_B_initial_roi_double_clicked(self, item):
"""
This function is triggered when a roi in patient B's roi list is
double-clicked.
"""
roi_to_add = item.data(Qt.UserRole)
transferred_roi_name = roi_to_add['name']
# If the clicked roi is already transferred, return
if transferred_roi_name in self.moving_to_fixed_rois.keys():
QMessageBox.about(self, "Transfer Failed",
"Chosen ROI has already been transferred!")
return
# Create a set to store all current roi names in target patient
# including both initial rois name and added roi names so far
patient_A_current_roi_name_list = set()
for item in self.moving_to_fixed_rois.values():
patient_A_current_roi_name_list.add(item)
for idx in self.fixed_image_initial_rois:
patient_A_current_roi_name_list.add(
self.fixed_image_initial_rois[idx]['name'])
# Check if clicked roi name has duplicate in
# target patient's roi names list
if transferred_roi_name in patient_A_current_roi_name_list:
# if add suffix is ticked, iteratively try adding suffix
# from _A to _Z, stop when no duplicate found
if self.add_suffix:
transferred_roi_name = generate_non_duplicated_name(
transferred_roi_name, patient_A_current_roi_name_list)
else:
QMessageBox.about(
self, "Transfer Failed", "Duplicated ROI name. "
"Please consider adding suffix.")
return
# Add clicked roi to transferred list
self.moving_to_fixed_rois[roi_to_add['name']] = transferred_roi_name
roi_label = QListWidgetItem(transferred_roi_name)
roi_label.setForeground(Qt.red)
roi_label.setData(Qt.UserRole, roi_to_add)
self.patient_B_rois_to_A_list_widget.addItem(roi_label)
self.save_button.setDisabled(False)
def reset_clicked(self):
"""
This function is triggered when reset button is clicked.
"""
self.fixed_to_moving_rois.clear()
self.moving_to_fixed_rois.clear()
self.patient_A_rois_to_B_list_widget.clear()
self.patient_B_rois_to_A_list_widget.clear()
self.save_button.setDisabled(True)
def transfer_list_is_empty(self):
"""
This function is to check if the transfer list is empty
"""
return len(self.fixed_to_moving_rois) == 0 \
and len(self.moving_to_fixed_rois) == 0
def save_clicked(self, interrupt_flag, progress_callback):
"""
This function is triggered when the save button is clicked. It contains
all steps in the ROI transferring process.
:param interrupt_flag: interrupt flag to stop process
:param progress_callback: signal that receives the current
progress of the loading.
"""
progress_callback.emit(("Converting images to sitk", 0))
# check if interrupt flag is set
if not check_interrupt_flag(interrupt_flag):
return False
rtss = self.patient_dict_container.get("dataset_rtss")
# get sitk for the fixed image
dicom_image = read_dicom_image_to_sitk(
self.patient_dict_container.filepaths)
if not check_interrupt_flag(interrupt_flag):
return False
# get array of roi indexes from sitk images
rois_images_fixed = transform_point_set_from_dicom_struct(
dicom_image,
rtss,
self.fixed_to_moving_rois.keys(),
spacing_override=None,
interrupt_flag=interrupt_flag)
moving_rtss = self.moving_dict_container.get("dataset_rtss")
if not check_interrupt_flag(interrupt_flag):
return False
# get sitk for the moving image
moving_dicom_image = read_dicom_image_to_sitk(
self.moving_dict_container.filepaths)
if not check_interrupt_flag(interrupt_flag):
return False
# get array of roi indexes from sitk images
progress_callback \
.emit(("Retrieving ROIs from \nboth image sets", 20))
# check if interrupt flag is set
if not check_interrupt_flag(interrupt_flag):
return False
if moving_rtss:
rois_images_moving = transform_point_set_from_dicom_struct(
moving_dicom_image,
moving_rtss,
self.moving_to_fixed_rois.keys(),
spacing_override=None,
interrupt_flag=interrupt_flag)
else:
rois_images_moving = ([], [])
if not check_interrupt_flag(interrupt_flag):
return False
tfm = self.moving_dict_container.get("tfm")
progress_callback.emit(
("Transfering ROIs from moving \nto fixed image set", 40))
# check if interrupt flag is set
if not check_interrupt_flag(interrupt_flag):
return False
# transform roi from moving_dict to fixed_dict
self.transfer_rois(self.moving_to_fixed_rois, tfm, dicom_image,
rois_images_moving, self.patient_dict_container)
progress_callback.emit(
("Transfering ROIs from fixed \nto moving image set", 60))
if not check_interrupt_flag(interrupt_flag):
return False
# transform roi from moving_dict to fixed_dict
self.transfer_rois(self.fixed_to_moving_rois, tfm.GetInverse(),
moving_dicom_image, rois_images_fixed,
self.moving_dict_container)
progress_callback.emit(("Saving ROIs to RTSS", 80))
# check if interrupt flag is set
if not check_interrupt_flag(interrupt_flag):
return False
progress_callback.emit(("Reloading window", 90))
return True
def transfer_roi_clicked(self):
"""
telling progress window to start ROI transfer
"""
self.progress_window.start(self.save_clicked)
def onTransferRoiError(self, exception):
"""
This function is triggered when there is an error in the
ROI transferring process.
:param exception: exception thrown
"""
QMessageBox.about(self.progress_window, "Unable to transfer ROIs",
"Please check your image set and ROI data.")
self.progress_window.close()
def onTransferRoiFinished(self, result):
"""
This function is triggered when ROI transferring process is finished.
"""
# emit changed dataset to structure_modified function and
# auto_save_roi function
if result[0] is True:
if len(self.fixed_to_moving_rois) > 0:
self.signal_roi_transferred_to_moving_container.emit(
(self.moving_dict_container.get("dataset_rtss"), {
"transfer": None
}))
if len(self.moving_to_fixed_rois) > 0:
self.signal_roi_transferred_to_fixed_container.emit(
(self.patient_dict_container.get("dataset_rtss"), {
"transfer": None
}))
self.progress_window.close()
QMessageBox.about(self.transfer_roi_window_instance, "Saved",
"ROIs are successfully transferred!")
else:
QMessageBox.about(self.transfer_roi_window_instance, "Cancelled",
"ROIs Transfer is cancelled.")
self.closeWindow()
def transfer_rois(self, transfer_dict, tfm, reference_image,
original_roi_list, patient_dict_container):
"""
Converting (transferring) ROIs from one image set to another and save
the transferred rois to rtss.
:param transfer_dict: dictionary of rois to be transfer.
key is original roi names, value is the name after transferred.
:param original_roi_list: tuple of sitk rois from the base image.
:param tfm: the tfm that contains information for transferring rois
:param reference_image: the reference (base) image
:param patient_dict_container: container of the transfer image set.
"""
for roi_name, new_roi_name in transfer_dict.items():
for index, name in enumerate(original_roi_list[1]):
if name == roi_name:
sitk_image = original_roi_list[0][index]
new_contour = apply_linear_transform(
input_image=sitk_image,
transform=tfm,
reference_image=reference_image,
is_structure=True)
contour = sitk.GetArrayViewFromImage(new_contour)
contours = np.transpose(contour.nonzero())
self.save_roi_to_patient_dict_container(
contours, new_roi_name, patient_dict_container)
def save_roi_to_patient_dict_container(self, contours, roi_name,
patient_dict_container):
"""
Save the transferred ROI to the corresponding rtss.
:param contours: np array of coordinates of the ROI to be saved.
:param roi_name: name of the ROI to be saved
:param patient_dict_container: container of the transfer image set.
"""
pixels_coords_dict = {}
slice_ids_dict = get_dict_slice_to_uid(patient_dict_container)
total_slices = len(slice_ids_dict)
for contour in contours:
curr_slice_id = total_slices - contour[0]
if curr_slice_id >= total_slices:
curr_slice_id = 0
if curr_slice_id not in pixels_coords_dict:
pixels_coords_dict[curr_slice_id] = [
tuple([contour[2], contour[1]])
]
else:
pixels_coords_dict[curr_slice_id].append(
tuple([contour[2], contour[1]]))
rois_to_save = {}
for key in pixels_coords_dict.keys():
coords = pixels_coords_dict[key]
polygon_list = ROI.calculate_concave_hull_of_points(coords)
if len(polygon_list) > 0:
rois_to_save[key] = {
'ds': patient_dict_container.dataset[key],
'coords': polygon_list
}
roi_list = ROI.convert_hull_list_to_contours_data(
rois_to_save, patient_dict_container)
if len(roi_list) > 0:
print("Saving ", roi_name)
if isinstance(patient_dict_container, MovingDictContainer):
new_rtss = ROI.create_roi(
patient_dict_container.get("dataset_rtss"),
roi_name,
roi_list,
rtss_owner="MOVING")
self.moving_dict_container.set("dataset_rtss", new_rtss)
self.moving_dict_container.set("rtss_modified", True)
else:
new_rtss = ROI.create_roi(
patient_dict_container.get("dataset_rtss"), roi_name,
roi_list)
self.patient_dict_container.set("dataset_rtss", new_rtss)
self.patient_dict_container.set("rtss_modified", True)
def closeWindow(self):
"""
function to close transfer roi window
"""
self.close()
def __init__(self, parent: QWidget, model: Model) -> None:
super().__init__(parent)
logger.add(self.log)
settings = QSettings()
self.mainlayout = QVBoxLayout()
self.mainlayout.setContentsMargins(5, 5, 5, 5)
self.setLayout(self.mainlayout)
# summary
summarylayout = FlowLayout()
summarylayout.setContentsMargins(0, 0, 0, 0)
self.summary = QWidget()
self.summary.setLayout(summarylayout)
self.mainlayout.addWidget(self.summary)
self.summary.setVisible(
settings.value('showSummary', 'True') == 'True')
detailslayout = QHBoxLayout()
detailslayout.setContentsMargins(1, 0, 0, 0)
detailslayout.setAlignment(Qt.AlignLeft | Qt.AlignVCenter)
detailslayout.setSpacing(15)
details = QWidget()
details.setLayout(detailslayout)
summarylayout.addWidget(details)
self.modstotal = QLabel()
detailslayout.addWidget(self.modstotal)
self.modsenabled = QLabel()
detailslayout.addWidget(self.modsenabled)
self.overridden = QLabel()
detailslayout.addWidget(self.overridden)
self.conflicts = QLabel()
detailslayout.addWidget(self.conflicts)
buttonslayout = QHBoxLayout()
buttonslayout.setContentsMargins(0, 0, 0, 0)
buttonslayout.setAlignment(Qt.AlignRight | Qt.AlignVCenter)
buttons = QWidget()
buttons.setLayout(buttonslayout)
summarylayout.addWidget(buttons)
self.startscriptmerger = QPushButton('Start Script Merger')
self.startscriptmerger.setContentsMargins(0, 0, 0, 0)
self.startscriptmerger.setMinimumWidth(140)
self.startscriptmerger.setIcon(
QIcon(str(getRuntimePath('resources/icons/script.ico'))))
self.startscriptmerger.clicked.connect(lambda: [
openExecutable(Path(str(settings.value('scriptMergerPath'))), True)
])
self.startscriptmerger.setEnabled(
verifyScriptMergerPath(
Path(str(settings.value('scriptMergerPath')))) is not None)
buttonslayout.addWidget(self.startscriptmerger)
self.startgame = QPushButton('Start Game')
self.startgame.setContentsMargins(0, 0, 0, 0)
self.startgame.setMinimumWidth(100)
self.startgame.setIcon(
QIcon(str(getRuntimePath('resources/icons/w3b.ico'))))
buttonslayout.addWidget(self.startgame)
# splitter
self.splitter = QSplitter(Qt.Vertical)
self.stack = QStackedWidget()
self.splitter.addWidget(self.stack)
# mod list widget
self.modlistwidget = QWidget()
self.modlistlayout = QVBoxLayout()
self.modlistlayout.setContentsMargins(0, 0, 0, 0)
self.modlistwidget.setLayout(self.modlistlayout)
self.stack.addWidget(self.modlistwidget)
# search bar
self.searchbar = QLineEdit()
self.searchbar.setPlaceholderText('Search...')
self.modlistlayout.addWidget(self.searchbar)
# mod list
self.modlist = ModList(self, model)
self.modlistlayout.addWidget(self.modlist)
self.searchbar.textChanged.connect(lambda e: self.modlist.setFilter(e))
# welcome message
welcomelayout = QVBoxLayout()
welcomelayout.setAlignment(Qt.AlignHCenter | Qt.AlignVCenter)
welcomewidget = QWidget()
welcomewidget.setLayout(welcomelayout)
welcomewidget.dragEnterEvent = self.modlist.dragEnterEvent # type: ignore
welcomewidget.dragMoveEvent = self.modlist.dragMoveEvent # type: ignore
welcomewidget.dragLeaveEvent = self.modlist.dragLeaveEvent # type: ignore
welcomewidget.dropEvent = self.modlist.dropEvent # type: ignore
welcomewidget.setAcceptDrops(True)
icon = QIcon(str(getRuntimePath('resources/icons/open-folder.ico')))
iconpixmap = icon.pixmap(32, 32)
icon = QLabel()
icon.setPixmap(iconpixmap)
icon.setAlignment(Qt.AlignHCenter | Qt.AlignVCenter)
icon.setContentsMargins(4, 4, 4, 4)
welcomelayout.addWidget(icon)
welcome = QLabel('''<p><font>
No mod installed yet.
Drag a mod into this area to get started!
</font></p>''')
welcome.setAttribute(Qt.WA_TransparentForMouseEvents)
welcome.setAlignment(Qt.AlignHCenter | Qt.AlignVCenter)
welcomelayout.addWidget(welcome)
self.stack.addWidget(welcomewidget)
# output log
self.output = QTextEdit(self)
self.output.setTextInteractionFlags(Qt.NoTextInteraction)
self.output.setReadOnly(True)
self.output.setContextMenuPolicy(Qt.NoContextMenu)
self.output.setPlaceholderText('Program output...')
self.splitter.addWidget(self.output)
# TODO: enhancement: show indicator if scripts have to be merged
self.splitter.setStretchFactor(0, 1)
self.splitter.setStretchFactor(1, 0)
self.mainlayout.addWidget(self.splitter)
# TODO: incomplete: make start game button functional
if len(model):
self.stack.setCurrentIndex(0)
self.splitter.setSizes([self.splitter.size().height(), 50])
else:
self.stack.setCurrentIndex(1)
self.splitter.setSizes([self.splitter.size().height(), 0])
model.updateCallbacks.append(self.modelUpdateEvent)
asyncio.create_task(model.loadInstalled())
class RequirementEditor:
_editor: Union[None, ResourceRequirementEditor, ArrayRequirementEditor,
TemplateRequirementEditor]
def __init__(self,
parent: QWidget,
parent_layout: QVBoxLayout,
resource_database: ResourceDatabase,
*,
on_remove=None):
self.parent = parent
self.parent_layout = parent_layout
self.resource_database = resource_database
self._editor = None
self._last_resource = None
self._last_items = ()
self._last_comment = None
self.line_layout = QHBoxLayout()
self.line_layout.setAlignment(Qt.AlignLeft)
self.parent_layout.addLayout(self.line_layout)
if on_remove is not None:
self.remove_button = QtWidgets.QToolButton(parent)
self.remove_button.setText("X")
self.remove_button.setMaximumWidth(20)
self.remove_button.clicked.connect(on_remove)
self.line_layout.addWidget(self.remove_button)
else:
self.remove_button = None
self.requirement_type_combo = QComboBox(parent)
self.requirement_type_combo.addItem("Resource", ResourceRequirement)
self.requirement_type_combo.addItem("Or", RequirementOr)
self.requirement_type_combo.addItem("And", RequirementAnd)
if resource_database.requirement_template:
self.requirement_type_combo.addItem("Template",
RequirementTemplate)
self.requirement_type_combo.setMaximumWidth(75)
self.requirement_type_combo.activated.connect(
self._on_change_requirement_type)
self.line_layout.addWidget(self.requirement_type_combo)
def create_specialized_editor(self, requirement: Requirement):
self.requirement_type_combo.setCurrentIndex(
self.requirement_type_combo.findData(type(requirement)))
if isinstance(requirement, ResourceRequirement):
self._editor = ResourceRequirementEditor(self.parent,
self.line_layout,
self.resource_database,
requirement)
elif isinstance(requirement, RequirementArrayBase):
self._editor = ArrayRequirementEditor(self.parent,
self.parent_layout,
self.line_layout,
self.resource_database,
requirement)
elif isinstance(requirement, RequirementTemplate):
self._editor = TemplateRequirementEditor(self.parent,
self.line_layout,
self.resource_database,
requirement)
else:
raise RuntimeError(
f"Unknown requirement type: {type(requirement)} - {requirement}"
)
def _on_change_requirement_type(self):
current_requirement = self.current_requirement
self._editor.deleteLater()
if isinstance(current_requirement, ResourceRequirement):
self._last_resource = current_requirement
elif isinstance(current_requirement, RequirementArrayBase):
self._last_items = current_requirement.items
self._last_comment = current_requirement.comment
elif isinstance(current_requirement, RequirementTemplate):
pass
else:
raise RuntimeError(
f"Unknown requirement type: {type(current_requirement)} - {current_requirement}"
)
new_class = self.requirement_type_combo.currentData()
if new_class == ResourceRequirement:
if self._last_resource is None:
new_requirement = _create_default_resource_requirement(
self.resource_database)
else:
new_requirement = self._last_resource
elif new_class == RequirementTemplate:
new_requirement = _create_default_template_requirement(
self.resource_database)
else:
new_requirement = new_class(self._last_items, self._last_comment)
self.create_specialized_editor(new_requirement)
def deleteLater(self):
if self.remove_button is not None:
self.remove_button.deleteLater()
self.requirement_type_combo.deleteLater()
if self._editor is not None:
self._editor.deleteLater()
@property
def current_requirement(self) -> Requirement:
return self._editor.current_requirement
def __init__(self,
parent: Optional[QWidget] = None,
firstStart: bool = False) -> None:
super().__init__(parent, )
if parent:
self.setWindowTitle('Settings')
else:
self.setWindowTitle(getTitleString('Settings'))
self.setAttribute(Qt.WA_DeleteOnClose)
settings = QSettings()
mainLayout = QVBoxLayout(self)
mainLayout.setContentsMargins(5, 5, 5, 5)
# First Start info
if firstStart:
firstStartInfo = QLabel(
'''
<p><strong>Hello! It looks like this is your first time using w3modmanager,
or the game installation path recently changed.</strong></p>
<p>
Please review the settings below.
</p>
''', self)
firstStartInfo.setWordWrap(True)
firstStartInfo.setContentsMargins(10, 10, 10, 10)
firstStartInfo.setSizePolicy(QSizePolicy.Minimum,
QSizePolicy.Minimum)
mainLayout.addWidget(firstStartInfo)
# Game
gbGame = QGroupBox('Game Path', self)
mainLayout.addWidget(gbGame)
gbGameLayout = QVBoxLayout(gbGame)
gamePathLayout = QHBoxLayout()
self.gamePath = QLineEdit(self)
self.gamePath.setPlaceholderText('Path to witcher3.exe...')
if settings.value('gamePath'):
self.gamePath.setText(str(settings.value('gamePath')))
self.gamePath.textChanged.connect(
lambda: self.validateGamePath(self.gamePath.text()))
gamePathLayout.addWidget(self.gamePath)
self.locateGame = QPushButton('Detect', self)
self.locateGame.clicked.connect(self.locateGameEvent)
self.locateGame.setToolTip(
'Automatically detect the game path if possible')
gamePathLayout.addWidget(self.locateGame)
selectGame = QPushButton('Browse', self)
selectGame.clicked.connect(self.selectGameEvent)
gamePathLayout.addWidget(selectGame)
gbGameLayout.addLayout(gamePathLayout)
gamePathInfoLayout = QHBoxLayout()
self.gamePathInfo = QLabel('', self)
self.gamePathInfo.setContentsMargins(4, 4, 4, 4)
self.gamePathInfo.setMinimumHeight(40)
self.gamePathInfo.setWordWrap(True)
gamePathInfoLayout.addWidget(self.gamePathInfo)
gbGameLayout.addLayout(gamePathInfoLayout)
# Config
gbConfig = QGroupBox('Game Config', self)
mainLayout.addWidget(gbConfig)
gbConfigLayout = QVBoxLayout(gbConfig)
configPathLayout = QHBoxLayout()
self.configPath = QLineEdit(self)
self.configPath.setPlaceholderText('Path to config folder...')
if settings.value('configPath'):
self.configPath.setText(str(settings.value('configPath')))
self.configPath.textChanged.connect(
lambda: self.validateConfigPath(self.configPath.text()))
configPathLayout.addWidget(self.configPath)
self.locateConfig = QPushButton('Detect', self)
self.locateConfig.clicked.connect(self.locateConfigEvent)
self.locateConfig.setToolTip(
'Automatically detect the config folder if possible')
configPathLayout.addWidget(self.locateConfig)
selectConfig = QPushButton('Browse', self)
selectConfig.clicked.connect(self.selectConfigEvent)
configPathLayout.addWidget(selectConfig)
gbConfigLayout.addLayout(configPathLayout)
configPathInfoLayout = QHBoxLayout()
self.configPathInfo = QLabel('', self)
self.configPathInfo.setContentsMargins(4, 4, 4, 4)
self.configPathInfo.setMinimumHeight(40)
self.configPathInfo.setWordWrap(True)
configPathInfoLayout.addWidget(self.configPathInfo)
gbConfigLayout.addLayout(configPathInfoLayout)
# Script Merger
gbScriptMerger = QGroupBox('Script Merger', self)
mainLayout.addWidget(gbScriptMerger)
gbScriptMergerLayout = QVBoxLayout(gbScriptMerger)
scriptMergerPathLayout = QHBoxLayout()
self.scriptMergerPath = QLineEdit(self)
self.scriptMergerPath.setPlaceholderText(
'Path to WitcherScriptMerger.exe...')
if settings.value('scriptMergerPath'):
self.scriptMergerPath.setText(
str(settings.value('scriptMergerPath')))
self.scriptMergerPath.textChanged.connect(
lambda: self.validateScriptMergerPath(self.scriptMergerPath.text()
))
scriptMergerPathLayout.addWidget(self.scriptMergerPath)
self.locateScriptMerger = QPushButton('Detect', self)
self.locateScriptMerger.clicked.connect(self.locateScriptMergerEvent)
self.locateScriptMerger.setToolTip(
'Automatically detect the script merger path if possible')
scriptMergerPathLayout.addWidget(self.locateScriptMerger)
selectScriptMerger = QPushButton('Browse', self)
selectScriptMerger.clicked.connect(self.selectScriptMergerEvent)
scriptMergerPathLayout.addWidget(selectScriptMerger)
gbScriptMergerLayout.addLayout(scriptMergerPathLayout)
scriptMergerPathInfoLayout = QHBoxLayout()
self.scriptMergerPathInfo = QLabel('', self)
self.scriptMergerPathInfo.setOpenExternalLinks(True)
self.scriptMergerPathInfo.setContentsMargins(4, 4, 4, 4)
self.scriptMergerPathInfo.setMinimumHeight(40)
self.scriptMergerPathInfo.setWordWrap(True)
scriptMergerPathInfoLayout.addWidget(self.scriptMergerPathInfo)
gbScriptMergerLayout.addLayout(scriptMergerPathInfoLayout)
# Nexus Mods API
gbNexusModsAPI = QGroupBox('Nexus Mods API', self)
mainLayout.addWidget(gbNexusModsAPI)
gbNexusModsAPILayout = QVBoxLayout(gbNexusModsAPI)
self.nexusAPIKey = QLineEdit(self)
self.nexusAPIKey.setPlaceholderText('Personal API Key...')
if settings.value('nexusAPIKey'):
self.nexusAPIKey.setText(str(settings.value('nexusAPIKey')))
self.nexusAPIKey.textChanged.connect(
lambda: self.validateApiKey(self.nexusAPIKey.text()))
gbNexusModsAPILayout.addWidget(self.nexusAPIKey)
self.nexusAPIKeyInfo = QLabel('🌐', self)
self.nexusAPIKeyInfo.setOpenExternalLinks(True)
self.nexusAPIKeyInfo.setWordWrap(True)
self.nexusAPIKeyInfo.setContentsMargins(4, 4, 4, 4)
self.nexusAPIKeyInfo.setMinimumHeight(48)
gbNexusModsAPILayout.addWidget(self.nexusAPIKeyInfo)
self.nexusGetInfo = QCheckBox('Get Mod details after adding a new mod',
self)
self.nexusGetInfo.setChecked(
settings.value('nexusGetInfo', 'True') == 'True')
self.nexusGetInfo.setDisabled(True)
gbNexusModsAPILayout.addWidget(self.nexusGetInfo)
self.nexusCheckUpdates = QCheckBox('Check for Mod updates on startup',
self)
self.nexusCheckUpdates.setChecked(
settings.value('nexusCheckUpdates', 'False') == 'True')
self.nexusCheckUpdates.setDisabled(True)
gbNexusModsAPILayout.addWidget(self.nexusCheckUpdates)
self.nexusCheckClipboard = QCheckBox(
'Monitor the Clipboard for Nexus Mods URLs', self)
self.nexusCheckClipboard.setChecked(
settings.value('nexusCheckClipboard', 'False') == 'True')
self.nexusCheckClipboard.setDisabled(True)
gbNexusModsAPILayout.addWidget(self.nexusCheckClipboard)
# Output
gbOutput = QGroupBox('Output Preferences', self)
mainLayout.addWidget(gbOutput)
gbOutputLayout = QVBoxLayout(gbOutput)
self.unhideOutput = QCheckBox('Auto-show output panel', self)
self.unhideOutput.setChecked(
settings.value('unhideOutput', 'True') == 'True')
gbOutputLayout.addWidget(self.unhideOutput)
self.debugOutput = QCheckBox('Show debug output', self)
self.debugOutput.setChecked(
settings.value('debugOutput', 'False') == 'True')
gbOutputLayout.addWidget(self.debugOutput)
# Actions
actionsLayout = QHBoxLayout()
actionsLayout.setAlignment(Qt.AlignRight)
self.save = QPushButton('Save', self)
self.save.clicked.connect(self.saveEvent)
self.save.setAutoDefault(True)
self.save.setDefault(True)
actionsLayout.addWidget(self.save)
cancel = QPushButton('Cancel', self)
cancel.clicked.connect(self.cancelEvent)
actionsLayout.addWidget(cancel)
mainLayout.addLayout(actionsLayout)
# Setup
if not settings.value('gamePath'):
self.locateGameEvent()
self.setMinimumSize(QSize(440, 440))
self.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding)
self.validGamePath = False
self.validConfigPath = False
self.validNexusAPIKey = False
self.validScriptMergerPath = False
self.validateGamePath(self.gamePath.text())
self.validateConfigPath(self.configPath.text())
self.validateApiKey(self.nexusAPIKey.text())
self.validateScriptMergerPath(self.scriptMergerPath.text())
self.updateSaveButton()
self.finished.connect(
lambda: self.validateApiKey.cancel()) # type: ignore
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 __init__(self,
parent: Optional[QWidget] = None,
url: str = '') -> None:
super().__init__(parent, )
if parent:
self.setWindowTitle('Download Mod')
else:
self.setWindowTitle(getTitleString('Download Mod'))
self.setAttribute(Qt.WA_DeleteOnClose)
mainLayout = QVBoxLayout(self)
mainLayout.setContentsMargins(5, 5, 5, 5)
self.signals = DownloadWindowEvents(self)
# URL input
gbUrl = QGroupBox('Mod URL')
gbUrlLayout = QVBoxLayout()
gbUrl.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Minimum)
self.url = QLineEdit()
self.url.setPlaceholderText(
'https://www.nexusmods.com/witcher3/mods/...')
self.url.setText(url)
self.url.textChanged.connect(lambda: self.validateUrl(self.url.text()))
gbUrlLayout.addWidget(self.url)
self.urlInfo = QLabel('🌐')
self.urlInfo.setContentsMargins(4, 4, 4, 4)
self.urlInfo.setMinimumHeight(36)
self.urlInfo.setWordWrap(True)
gbUrlLayout.addWidget(self.urlInfo)
gbUrl.setLayout(gbUrlLayout)
mainLayout.addWidget(gbUrl)
# File selection
gbFiles = QGroupBox('Mod Files')
gbFilesLayout = QVBoxLayout()
gbFiles.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding)
self.files = QTableWidget(0, 4)
self.files.setVerticalScrollMode(QAbstractItemView.ScrollPerPixel)
self.files.setHorizontalScrollMode(QAbstractItemView.ScrollPerPixel)
self.files.setContextMenuPolicy(Qt.CustomContextMenu)
self.files.setSelectionMode(QAbstractItemView.ExtendedSelection)
self.files.setSelectionBehavior(QAbstractItemView.SelectRows)
self.files.setWordWrap(False)
self.files.setSortingEnabled(True)
self.files.setFocusPolicy(Qt.StrongFocus)
self.files.verticalHeader().hide()
self.files.setSortingEnabled(True)
self.files.sortByColumn(2, Qt.DescendingOrder)
self.files.verticalHeader().setVisible(False)
self.files.verticalHeader().setDefaultSectionSize(25)
self.files.horizontalHeader().setHighlightSections(False)
self.files.horizontalHeader().setStretchLastSection(True)
self.files.setHorizontalHeaderLabels(
['File Name', 'Version', 'Upload Date', 'Description'])
self.files.setEditTriggers(QAbstractItemView.NoEditTriggers)
self.files.verticalScrollBar().valueChanged.connect(
lambda: self.files.clearFocus())
self.files.itemSelectionChanged.connect(lambda: self.validateFiles())
self.files.setDisabled(True)
self.files.setStyleSheet('''
QTableView {
gridline-color: rgba(255,255,255,1);
}
QTableView::item {
padding: 5px;
margin: 1px 0;
}
QTableView::item:!selected:hover {
background-color: rgb(217, 235, 249);
padding: 0;
}
''')
gbFilesLayout.addWidget(self.files)
_mouseMoveEvent = self.files.mouseMoveEvent
self.files.hoverIndexRow = -1
def mouseMoveEvent(event: QMouseEvent) -> None:
self.files.hoverIndexRow = self.files.indexAt(event.pos()).row()
_mouseMoveEvent(event)
self.files.mouseMoveEvent = mouseMoveEvent # type: ignore
self.files.setItemDelegate(ModListItemDelegate(self.files))
self.files.setMouseTracking(True)
gbFiles.setLayout(gbFilesLayout)
mainLayout.addWidget(gbFiles)
# Actions
actionsLayout = QHBoxLayout()
actionsLayout.setAlignment(Qt.AlignRight)
self.download = QPushButton('Download', self)
self.download.clicked.connect(lambda: self.downloadEvent())
self.download.setAutoDefault(True)
self.download.setDefault(True)
self.download.setDisabled(True)
actionsLayout.addWidget(self.download)
cancel = QPushButton('Cancel', self)
cancel.clicked.connect(self.cancelEvent)
actionsLayout.addWidget(cancel)
mainLayout.addLayout(actionsLayout)
# Setup
self.setMinimumSize(QSize(420, 420))
self.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding)
self.resize(QSize(720, 420))
self.finished.connect(
lambda: self.validateUrl.cancel()) # type: ignore
self.finished.connect(
lambda: self.downloadEvent.cancel()) # type: ignore
self.modId = 0
self.validateUrl(self.url.text())
def __init__(self, model: SettingsModel, parent=None):
super(SetQuotaDiskWidget, self).__init__(parent)
self._model = model
self.setAccessibleName("InfoBox")
self.title = QLabel()
self.title.setText("Spazio di archiviazione")
self.title.setAccessibleName("Title2")
self.sottotitolo = QLabel()
self.sottotitolo.setAccessibleName('Sottotitolo')
self.sottotitolo.setText(
"Cambia lo spazio di archiviazione destinato alla cartella sincronizzata"
)
# Barra riempimento disco
self.progress_label = QLabel()
self.progress_label.setText("Spazio occupato:")
self.disk_progress = QProgressBar()
self.disk_progress.setFormat("")
self.disk_quota = QLabel()
# Modifica spazio dedicato
self.spaceLabel = QLabel(" ")
self.dedicated_space = QLineEdit()
self.dedicated_space.setValidator(QDoubleValidator())
self.sizes_box = QComboBox()
self.sizes_box.wheelEvent = lambda event: None
_path_size = bitmath.parse_string(model.convert_size(model.get_size()))
_disk_free = bitmath.parse_string(
model.convert_size(model.get_free_disk()))
self.populate_size_box(_path_size, _disk_free)
self.change_quota_button = QPushButton("Cambia quota disco")
self.change_quota_button.setMaximumWidth(150)
self.change_quota_button.clicked.connect(
self.Sl_dedicated_space_changed)
self.dedicated_space.returnPressed.connect(
self.Sl_dedicated_space_changed)
self.buttonLayout = QHBoxLayout()
self.buttonLayout.addWidget(self.spaceLabel)
self.buttonLayout.addWidget(self.change_quota_button)
self.buttonLayout.addWidget(self.spaceLabel)
set_space_layout = QHBoxLayout()
set_space_layout.addWidget(self.dedicated_space)
set_space_layout.addWidget(self.sizes_box)
quota_layout = QHBoxLayout()
quota_layout.setAlignment(Qt.AlignLeft)
quota_layout.addWidget(self.progress_label)
quota_layout.addWidget(self.disk_quota)
# layout
disk_layout = QVBoxLayout()
disk_layout.setAlignment(Qt.AlignLeft)
disk_layout.addWidget(self.title)
disk_layout.addWidget(self.sottotitolo)
disk_layout.addWidget(self.spaceLabel)
disk_layout.addLayout(quota_layout)
disk_layout.addWidget(self.disk_progress)
disk_layout.addWidget(self.spaceLabel)
disk_layout.addLayout(set_space_layout)
disk_layout.addLayout(self.buttonLayout)
self.setLayout(disk_layout)
self.Sl_model_changed()
