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()
class SelectROIPopUp(QDialog):
signal_roi_name = QtCore.Signal(str)
def __init__(self):
QDialog.__init__(self)
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()
self.setStyleSheet(stylesheet)
self.standard_names = []
self.init_standard_names()
self.setWindowTitle("Select A Region of Interest To Draw")
self.setMinimumSize(350, 180)
self.icon = QtGui.QIcon()
self.icon.addPixmap(
QtGui.QPixmap(resource_path("res/images/icon.ico")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.setWindowIcon(self.icon)
self.explanation_text = QLabel("Search for ROI:")
self.input_field = QLineEdit()
self.input_field.textChanged.connect(self.on_text_edited)
self.button_area = QWidget()
self.cancel_button = QPushButton("Cancel")
self.cancel_button.clicked.connect(self.on_cancel_clicked)
self.begin_draw_button = QPushButton("Begin Draw Process")
self.begin_draw_button.clicked.connect(self.on_begin_clicked)
self.button_layout = QHBoxLayout()
self.button_layout.addWidget(self.cancel_button)
self.button_layout.addWidget(self.begin_draw_button)
self.button_area.setLayout(self.button_layout)
self.list_label = QLabel()
self.list_label.setText("Select a Standard Region of Interest")
self.list_of_ROIs = QListWidget()
for standard_name in self.standard_names:
self.list_of_ROIs.addItem(standard_name)
self.layout = QVBoxLayout()
self.layout.addWidget(self.explanation_text)
self.layout.addWidget(self.input_field)
self.layout.addWidget(self.list_label)
self.layout.addWidget(self.list_of_ROIs)
self.layout.addWidget(self.button_area)
self.setLayout(self.layout)
self.list_of_ROIs.clicked.connect(self.on_roi_clicked)
def init_standard_names(self):
"""
Create two lists containing standard organ
and standard volume names as set by the Add-On options.
"""
with open(data_path('organName.csv'), 'r') as f:
standard_organ_names = []
csv_input = csv.reader(f)
header = next(f) # Ignore the "header" of the column
for row in csv_input:
standard_organ_names.append(row[0])
with open(data_path('volumeName.csv'), 'r') as f:
standard_volume_names = []
csv_input = csv.reader(f)
next(f) # Ignore the "header" of the column
for row in csv_input:
standard_volume_names.append(row[1])
self.standard_names = standard_organ_names + standard_volume_names
def on_text_edited(self, text):
"""
function triggered when text in
the ROI popup is edited
-------
:param text: text in the pop up
"""
self.list_of_ROIs.clear()
text_upper_case = text.upper()
for item in self.standard_names:
if item.startswith(text) or item.startswith(text_upper_case):
self.list_of_ROIs.addItem(item)
def on_roi_clicked(self):
"""
function triggered when an ROI is clicked
"""
self.begin_draw_button.setEnabled(True)
self.begin_draw_button.setFocus()
def on_begin_clicked(self):
"""
function to start the draw ROI function
"""
# If there is a ROI Selected
if self.list_of_ROIs.currentItem() is not None:
roi = self.list_of_ROIs.currentItem()
self.roi_name = str(roi.text())
# Call function on UIDrawWindow so it has selected ROI
self.signal_roi_name.emit(self.roi_name)
self.close()
def on_cancel_clicked(self):
"""
function to cancel the operation
"""
self.close()
class ProgramList(QWidget):
onProgramEditRequest = Signal(Program)
def __init__(self, parent):
super().__init__(parent)
self._programsLabel = QLabel("Programs")
self._programsList = QListWidget()
self._programsList.itemClicked.connect(self.SelectProgram)
self._programsList.itemDoubleClicked.connect(self.EditProgram)
AppGlobals.Instance().onChipModified.connect(self.SyncInstances)
self._instances: Dict[Program, ProgramInstance] = {}
self._newButton = QPushButton("Create New Program")
self._newButton.clicked.connect(self.NewProgram)
layout = QVBoxLayout()
layout.addWidget(self._programsLabel)
layout.addWidget(self._programsList)
layout.addWidget(self._newButton)
self.setLayout(layout)
AppGlobals.Instance().onChipOpened.connect(self.SyncInstances)
def EditProgram(self, selectedProgram: 'ProgramListItem'):
self.onProgramEditRequest.emit(selectedProgram.program)
def SelectProgram(self, selectedProgram: 'ProgramListItem'):
contextDisplay = ProgramContextDisplay(self.topLevelWidget(),
selectedProgram.instance,
self._programsList)
contextDisplay.onDelete.connect(self.DeleteProgram)
contextDisplay.onEdit.connect(self.onProgramEditRequest)
def SyncInstances(self):
for program in AppGlobals.Chip().programs:
if program not in self._instances:
self._instances[program] = ProgramInstance(program)
for program in self._instances.copy():
if program not in AppGlobals.Chip().programs:
del self._instances[program]
else:
self._instances[program].SyncParameters()
self.RefreshList()
def NewProgram(self):
newProgram = Program()
AppGlobals.Chip().programs.append(newProgram)
AppGlobals.Instance().onChipModified.emit()
for item in [
self._programsList.item(row)
for row in range(self._programsList.count())
]:
if item.program is newProgram:
self._programsList.setCurrentItem(item)
self.onProgramEditRequest.emit(newProgram)
return
def ProgramItem(self, program: Program):
matches = [
item for item in [
self._programsList.item(row)
for row in range(self._programsList.count())
] if item.program is program
]
if matches:
return matches[0]
def RefreshList(self):
self._programsList.blockSignals(True)
self._programsList.clear()
for program in AppGlobals.Chip().programs:
self._programsList.addItem(
ProgramListItem(program, self._instances[program]))
self._programsList.blockSignals(False)
def DeleteProgram(self, program: Program):
if QMessageBox.question(
self, "Confirm Deletion", "Are you sure you want to delete " +
program.name + "?") is QMessageBox.Yes:
AppGlobals.Chip().programs.remove(program)
AppGlobals.Instance().onChipModified.emit()
def keyPressEvent(self, event) -> None:
if event.key() == Qt.Key.Key_Delete:
if self._programsList.currentItem():
self.DeleteProgram(self._programsList.currentItem().program)
class MainWidget(QWidget):
def __init__(self, base):
QWidget.__init__(self)
self.base = base
self.setWindowIcon(QIcon('icon.ico'))
self.setWindowTitle(TITLE)
self.set_background_color(QColor(255, 255, 255))
self.process_header_widget = QWidget()
self.process_header_layout = QHBoxLayout(self.process_header_widget)
self.process_header_layout.setContentsMargins(0, 0, 0, 0)
self.process_label = QLabel('Available processes:')
self.github_label = QLabel(
'<a href="https://github.com/darktohka/p3dephaser">GitHub</a>')
self.github_label.setOpenExternalLinks(True)
self.refresh_button = QPushButton('Refresh')
self.refresh_button.clicked.connect(self.refresh_processes)
self.refresh_button.setFixedSize(100, 23)
self.multifile_widget = QWidget()
self.multifile_layout = QHBoxLayout(self.multifile_widget)
self.multifile_layout.setContentsMargins(0, 0, 0, 0)
self.multifileLabel = QLabel('Requested multifile names:')
self.multifileBox = QLineEdit(self)
self.multifileBox.returnPressed.connect(self.begin_scan)
self.multifile_layout.addWidget(self.multifileLabel)
self.multifile_layout.addWidget(self.multifileBox)
self.scan_button = QPushButton('Scan')
self.scan_button.clicked.connect(self.begin_scan)
self.process_list_box = QListWidget()
self.process_header_layout.addWidget(self.process_label)
self.process_header_layout.addStretch(1)
self.process_header_layout.addWidget(self.github_label)
self.process_header_layout.addWidget(self.refresh_button)
self.result_table = QTableWidget()
self.result_table.setColumnCount(3)
self.result_table.horizontalHeader().setStretchLastSection(True)
self.result_table.horizontalHeader().setSectionResizeMode(
QHeaderView.ResizeMode.Stretch)
for i, header in enumerate(('Process', 'Multifile', 'Password')):
self.result_table.setHorizontalHeaderItem(i,
QTableWidgetItem(header))
self.base_layout = QVBoxLayout(self)
self.base_layout.setContentsMargins(15, 15, 15, 15)
self.base_layout.addWidget(self.process_header_widget)
self.base_layout.addWidget(self.process_list_box)
self.base_layout.addWidget(self.multifile_widget)
self.base_layout.addWidget(self.scan_button)
self.base_layout.addWidget(self.result_table)
self.refresh_processes()
self.thread_pool = QThreadPool()
self.worker = None
self.process_name = None
self.stop_event = threading.Event()
def set_background_color(self, color):
self.setAutoFillBackground(True)
palette = self.palette()
palette.setColor(self.backgroundRole(), color)
self.setPalette(palette)
def get_processes(self):
processes = []
for proc in psutil.process_iter():
processes.append(proc.as_dict(attrs=['pid', 'name']))
processes.sort(
key=lambda process: (process['name'].lower(), process['pid']))
return processes
def refresh_processes(self):
self.process_list_box.clear()
processes = self.get_processes()
for process in processes:
name = process['name']
pid = process['pid']
self.process_list_box.addItem(f'{name} (PID {pid})')
def begin_scan(self):
if self.worker:
self.stop_event.set()
self.scan_button.setEnabled(False)
return
items = self.process_list_box.selectedItems()
if not items:
QMessageBox.warning(self, TITLE,
'Please choose a process from the list!')
return
process = items[0].text()[:-1].split(' ')
self.process_name = ' '.join(process[:-2])
pid = int(process[-1])
multifiles = self.multifileBox.text().split()
if not multifiles:
QMessageBox.warning(self, TITLE,
'Please choose some multifiles to target!')
return
multifile_names = '\n'.join(
[f'- {multifile}' for multifile in multifiles])
question = f'Do you really want to scan {self.process_name} for the following multifiles?\n\n{multifile_names}'
if QMessageBox.question(
self, TITLE, question,
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No
) != QMessageBox.StandardButton.Yes:
return
self.count = 0
self.setWindowTitle(f'{TITLE} - Scanning...')
self.scan_button.setText('Stop')
self.worker = ScanWorker(self, pid, multifiles)
self.worker.signals.finished.connect(self.scan_over)
self.worker.signals.error.connect(self.error_occurred)
self.worker.signals.progress.connect(self.report_progress)
self.thread_pool.start(self.worker)
def scan_over(self):
self.worker = None
self.stop_event.clear()
self.scan_button.setText('Scan')
self.scan_button.setEnabled(True)
self.setWindowTitle(TITLE)
QMessageBox.information(
self, TITLE,
f'Scan complete!\n\n{self.count} password{"s have" if self.count != 1 else " has"} been found.'
)
def error_occurred(self, error):
exc, value, message = error
QMessageBox.critical(
self, TITLE,
f'An error has occurred while trying to scan this process!\n\n{exc} {value}\n\n{message}'
)
def report_progress(self, multifile, password):
self.count += 1
index = self.result_table.rowCount()
self.result_table.insertRow(index)
for i, value in enumerate((self.process_name, multifile, password)):
self.result_table.setItem(index, i, QTableWidgetItem(value))
class TableClothGenerator(QMainWindow):
def __init__(self, parent=None):
super().__init__(parent)
# Main UI settings
self.setWindowTitle('Tablecloth Generator')
self.setWindowIcon(QIcon('icon.ico'))
self.centralWidget = QWidget()
self.setCentralWidget(self.centralWidget)
self.resize(350, 350)
self.center()
self._createMenuBar()
self.MainUI()
def MainUI(self):
# Obtain the configs
fp_config = open(THISDIR + "\\config\\config.json", "r",
encoding="utf-8")
self.config = json.loads(fp_config.read())
fp_config.close()
# Obtain and List the teams
fp_teams = open(THISDIR + "\\config\\teams.json", "r",
encoding="utf-8")
conf_teams = json.loads(fp_teams.read())
fp_teams.close()
self.teams = conf_teams["teams"]
self.players = conf_teams["players"]
# Obtain all images needed to create the tablecloth
self.background = Image.open(THISDIR + "\\images\\mat.png")
self.table_border = Image.open(THISDIR + "\\images\\table_border.png")
self.tech_lines = Image.open(THISDIR + "\\images\\technical_lines.png")
# Check if there's no configuration set up
# and prompt to create/import one
if self.config["total_teams"] == 0:
self.no_config = QMessageBox.question(self, "No configuration",
"No configuration has been found. Do you wish to set up a new one?",
QMessageBox.Yes | QMessageBox.No)
if self.no_config == QMessageBox.Yes:
self.CreateTeamsWindow()
self.bg_image = self.config["image_route"]
self.players_combobox = QComboBox()
self.UpdatePlayersList()
self.players_combobox.setEditable(True)
self.players_combobox.completer()\
.setCompletionMode(QCompleter.PopupCompletion)
self.players_combobox.setInsertPolicy(QComboBox.NoInsert)
# Set up the GUI
self.statusBar().showMessage("Remember: Rig responsibly.")
# Bottom (EAST)
self.label_east = QLabel(self)
self.label_east.setText("<h1>East Seat</h1>")
self.label_east.setAlignment(QtCore.Qt.AlignCenter)
self.image_east = QLabel(self)
self.image_east.setPixmap(QPixmap("images/logos/team1.png")\
.scaled(100,100))
self.image_east.setAlignment(QtCore.Qt.AlignCenter)
self.search_east = QLineEdit()
self.search_east.setAlignment(QtCore.Qt.AlignCenter)
self.search_east.editingFinished.connect(
lambda: self.searchPlayer(self.search_east.text(),
self.cloth_east))
self.cloth_east = QComboBox()
self.cloth_east.setModel(self.players_combobox.model())
self.cloth_east.currentIndexChanged.connect(
lambda: self.SwitchImage(self.cloth_east, self.image_east))
# Right (SOUTH)
self.label_south = QLabel(self)
self.label_south.setText("<h1>South Seat</h1>")
self.label_south.setAlignment(QtCore.Qt.AlignCenter)
self.image_south = QLabel(self)
self.image_south.setPixmap(QPixmap("images/logos/team1.png")\
.scaled(100,100))
self.image_south.setAlignment(QtCore.Qt.AlignCenter)
self.image_south.show()
self.search_south = QLineEdit()
self.search_south.setAlignment(QtCore.Qt.AlignCenter)
self.search_south.editingFinished.connect(
lambda: self.searchPlayer(self.search_south.text(),
self.cloth_south))
self.cloth_south = QComboBox()
self.cloth_south.setModel(self.players_combobox.model())
self.cloth_south.currentIndexChanged.connect(
lambda: self.SwitchImage(self.cloth_south, self.image_south))
# Top (WEST)
self.label_west = QLabel(self)
self.label_west.setText("<h1>West Seat</h1>")
self.label_west.setAlignment(QtCore.Qt.AlignCenter)
self.image_west = QLabel(self)
self.image_west.setPixmap(QPixmap("images/logos/team1.png")\
.scaled(100,100))
self.image_west.setAlignment(QtCore.Qt.AlignCenter)
self.image_west.show()
self.cloth_west = QComboBox()
self.search_west = QLineEdit()
self.search_west.setAlignment(QtCore.Qt.AlignCenter)
self.search_west.editingFinished.connect(
lambda: self.searchPlayer(self.search_west.text(),
self.cloth_west))
self.cloth_west.setModel(self.players_combobox.model())
self.cloth_west.currentIndexChanged.connect(
lambda: self.SwitchImage(self.cloth_west, self.image_west))
# Left (NORTH)
self.label_north = QLabel(self)
self.label_north.setText("<h1>North Seat</h1>")
self.label_north.setAlignment(QtCore.Qt.AlignCenter)
self.image_north = QLabel(self)
self.image_north.setPixmap(QPixmap("images/logos/team1.png")\
.scaled(100,100))
self.image_north.setAlignment(QtCore.Qt.AlignCenter)
self.image_north.show()
self.cloth_north = QComboBox()
self.search_north = QLineEdit()
self.search_north.setAlignment(QtCore.Qt.AlignCenter)
self.search_north.editingFinished.connect(
lambda: self.searchPlayer(self.search_north.text(),
self.cloth_north))
self.cloth_north.setModel(self.players_combobox.model())
self.cloth_north.currentIndexChanged.connect(
lambda: self.SwitchImage(self.cloth_north, self.image_north))
# Technical lines
self.technical_lines = QCheckBox("Show Technical lines", self)
# Generate button
self.generate = QPushButton(self)
self.generate.setText("Generate Tablecloth")
self.generate.clicked.connect(self.GeneratePreview)
# Add custom mat
self.custom_mat = QPushButton(self)
self.custom_mat.setText("Add Mat")
self.custom_mat.clicked.connect(self.MatDialog)
# Create the layout
grid_layout = QGridLayout()
grid_layout.setAlignment(QtCore.Qt.AlignCenter)
grid_layout.setAlignment(QtCore.Qt.AlignTop)
# Labels East, West
grid_layout.addWidget(self.label_east, 1, 1)
grid_layout.addWidget(self.label_west, 1, 2)
# Image preview East, West
grid_layout.addWidget(self.image_east, 2, 1)
grid_layout.addWidget(self.image_west, 2, 2)
# Search player East, West
grid_layout.addWidget(self.search_east, 3, 1)
grid_layout.addWidget(self.search_west, 3, 2)
# Player combobox East, West
grid_layout.addWidget(self.cloth_east, 4, 1)
grid_layout.addWidget(self.cloth_west, 4, 2)
# Labes South, North
grid_layout.addWidget(self.label_south, 5, 1)
grid_layout.addWidget(self.label_north, 5, 2)
# Image preview South, North
grid_layout.addWidget(self.image_south, 6, 1)
grid_layout.addWidget(self.image_north, 6, 2)
# Search player South, North
grid_layout.addWidget(self.search_south, 7, 1)
grid_layout.addWidget(self.search_north, 7, 2)
# Player combobox South, North
grid_layout.addWidget(self.cloth_south, 8, 1)
grid_layout.addWidget(self.cloth_north, 8, 2)
# Technical lines
grid_layout.addWidget(self.technical_lines, 9, 1)
# Custom mat/bg
grid_layout.addWidget(self.custom_mat, 10, 1)
# Generate
grid_layout.addWidget(self.generate, 10, 2)
self.centralWidget.setLayout(grid_layout)
# Create the window
self.show()
def _createMenuBar(self):
# Settings and stuff for the toolbar
menubar = QMenuBar(self)
file_menu = QMenu("&File", self)
file_menu.addAction("Create Team(s)", self.CreateTeamsWindow)
file_menu.addAction("Edit Team(s)", self.EditTeamsWindow)
file_menu.addAction("Exit", self.close)
settings_menu = QMenu("&Settings", self)
settings_menu.addAction("Version", self.SeeVersion)
settings_menu.addAction("Help", self.GetHelp)
menubar.addMenu(file_menu)
menubar.addMenu(settings_menu)
self.setMenuBar(menubar)
def _createProgressBar(self):
self.progress_bar = QProgressBar()
self.progress_bar.minimum = 0
self.progress_bar.maximum = 100
self.progress_bar.setValue(0)
self.progress_bar.setTextVisible(False)
self.progress_bar.setGeometry(50, 50, 10, 10)
self.progress_bar.setAlignment(QtCore.Qt.AlignRight)
self.progress_bar.adjustSize()
self.statusBar().addPermanentWidget(self.progress_bar)
self.ChangeAppStatus(False)
def SwitchImage(self, cloth, image):
# It shows you the team logo. No way you can miss those, right?
team_id = self.SearchTeamID(cloth, True)
image.setPixmap(QPixmap(
"images/logos/team%d.png" % team_id).scaled(100,100))
def searchPlayer(self, text, combobox):
# It even searches the player for you. What more could you want?
search_index = combobox.findText(text, QtCore.Qt.MatchContains)
if search_index == -1:
QMessageBox.warning(self, "Error", "No player found")
else:
combobox.setCurrentIndex(search_index)
def CreateTeamsWindow(self):
self.teamcreation_wid = EditionWidget()
self.teamcreation_wid.resize(400, 200)
self.teamcreation_wid.setWindowTitle("Teams configuration")
self.new_config = {}
id_label = QLabel(self)
id_label.setText("Team ID: ")
self.num_id = QLabel(self)
current_id = str(self.config["total_teams"] + 1)
self.num_id.setText(current_id)
name_label = QLabel(self)
name_label.setText("Team Name:")
name_label.setFocus()
self.name_input = QLineEdit(self)
members_label = QLabel(self)
members_label.setText("Members (write and press enter):")
members_input = QLineEdit(self)
members_input.editingFinished.connect(
lambda: self.AddMember(members_input))
self.members_list = QListWidget(self)
import_image = QPushButton(self)
import_image.setText("Import Team Image")
import_image.clicked.connect(self.ImportTeamImage)
add_team = QPushButton(self)
add_team.setText("Add Team")
add_team.clicked.connect(
lambda: self.addTeamFunction(self.name_input.text(),
self.members_list))
import_config = QPushButton(self)
import_config.setText("Import configuration")
import_config.clicked.connect(self.importTeamFunction)
config_lay = QGridLayout()
config_lay.addWidget(id_label, 1, 0)
config_lay.addWidget(self.num_id, 1, 1)
config_lay.addWidget(name_label, 2, 0)
config_lay.addWidget(self.name_input, 2, 1)
config_lay.addWidget(members_label, 3, 0)
config_lay.addWidget(members_input, 3, 1)
config_lay.addWidget(self.members_list, 4, 0, 2, 2)
config_lay.addWidget(add_team, 6, 0)
config_lay.addWidget(import_image, 6, 1)
config_lay.addWidget(import_config, 7, 0, 1, 2)
self.teamcreation_wid.setLayout(config_lay)
self.teamcreation_wid.setWindowModality(QtCore.Qt.ApplicationModal)
self.teamcreation_wid.activateWindow()
self.teamcreation_wid.raise_()
self.teamcreation_wid.show()
def addTeamFunction(self, name, members):
fp_teams = open(THISDIR + "\\config\\teams.json", "r",
encoding="utf-8")
current_teams = json.loads(fp_teams.read())
fp_teams.close()
team = {}
current_teams["teams"].append(name)
current_teams["players"][name] = [str(self.members_list.item(i).text())\
for i in range(self.members_list.count())]
new_team = open(THISDIR + "\\config\\teams.json", "w+",
encoding="utf-8")
add_config = open(THISDIR + "\\config\\config.json", "w+",
encoding="utf-8")
self.teams = current_teams["teams"]
self.players = current_teams["players"]
self.config["total_teams"] += 1
new_id = self.config["total_teams"] + 1
self.num_id.setText(str(new_id))
add_config.write(json.dumps(self.config, indent=4))
new_team.write(json.dumps(current_teams, indent=4))
new_team.close()
self.name_input.clear()
self.members_list.clear()
self.UpdatePlayersList()
def ImportTeamImage(self):
image_dialog = QFileDialog(self)
image_dialog = QFileDialog.getOpenFileName(filter="Images (*.png)",
selectedFilter="Images (*.png)")
if image_dialog[0] != "":
new_team_logo = Image.open(image_dialog[0]).convert("RGBA")
if new_team_logo.size != (250, 250):
new_team_logo.resize((250, 250))
new_team_logo.save(THISDIR+"\\images\\logos\\team%s.png"\
% self.num_id.text())
QMessageBox.information(self, "Team Image", "Team image added.")
def importTeamFunction(self):
file_dialog = QFileDialog(self)
file_dialog = QFileDialog.getOpenFileName(
filter="Team Files (*.json *.zip)",
selectedFilter="Team Files (*.json *.zip)")
if file_dialog[0] != "":
if is_zipfile(file_dialog[0]):
with ZipFile(file_dialog[0]) as zip_import:
list_of_files = zip_import.namelist()
for fimp in list_of_files:
if fimp.startswith('logos'):
zip_import.extract(fimp, path=THISDIR+'\\images\\')
imported_teams = zip_import.read('teams.json')
imported_teams = imported_teams.decode('utf-8')
else:
imported_teams = open(file_dialog[0], "r",
encoding="utf-8").read()
json_teams = json.loads(imported_teams)
self.teams = json_teams["teams"]
self.players = json_teams["players"]
new_teams = open(THISDIR + "\\config\\teams.json", "w+",
encoding="utf-8")
new_teams.write(json.dumps(json_teams, indent=4))
new_teams.close()
old_config = open(THISDIR + "\\config\\config.json", "r",
encoding="utf-8").read()
old_config = json.loads(old_config)
old_config["total_teams"] = len(json_teams["teams"])
self.config = old_config
new_config = open(THISDIR + "\\config\\config.json", "w+",
encoding="utf-8")
new_config.write(json.dumps(self.config, indent=4))
new_config.close()
self.UpdatePlayersList()
self.image_east.setPixmap(QPixmap("images/logos/team1.png")\
.scaled(100,100))
self.cloth_east.setModel(self.players_combobox.model())
self.image_south.setPixmap(QPixmap("images/logos/team1.png")\
.scaled(100,100))
self.cloth_south.setModel(self.players_combobox.model())
self.image_west.setPixmap(QPixmap("images/logos/team1.png")\
.scaled(100,100))
self.cloth_west.setModel(self.players_combobox.model())
self.image_north.setPixmap(QPixmap("images/logos/team1.png")\
.scaled(100,100))
self.cloth_north.setModel(self.players_combobox.model())
self.statusBar().showMessage("Teams imported successfully.")
self.teamcreation_wid.close()
def AddMember(self, member):
self.members_list.addItem(member.text())
member.clear()
def EditTeamsWindow(self):
self.teamedit_wid = EditionWidget()
self.teamedit_wid.resize(400, 320)
self.teamedit_wid.setWindowTitle("Edit Teams")
self.teams_list = QComboBox(self)
self.teams_list.addItem("--- Select a team ---")
for team in self.teams:
self.teams_list.addItem(team)
self.teams_list.currentIndexChanged.connect(self.UpdateTeamInfo)
team_id_label = QLabel(self)
team_id_label.setText("Team ID: ")
self.config_team_id = QLabel(self)
team_name_label = QLabel(self)
team_name_label.setText("Team name: ")
self.config_team_name = QLabel(self)
team_members_label = QLabel(self)
team_members_label.setText("Team members: ")
self.config_team_members = QListWidget(self)
add_member_label = QLabel(self)
add_member_label.setText("Add new member: ")
add_member_input = QLineEdit(self)
add_member_input.editingFinished.connect(self.AddNewMember)
delete_member = QPushButton(self)
delete_member.setText("Delete member")
delete_member.clicked.connect(self.DeleteMember)
delete_team = QPushButton(self)
delete_team.setText("Delete Team")
delete_team.clicked.connect(self.DeleteTeam)
save_changes = QPushButton(self)
save_changes.setText("Save changes")
save_changes.clicked.connect(self.SaveEdits)
export_config = QPushButton(self)
export_config.setText("Export Configuration")
export_config.clicked.connect(self.ExportTeams)
config_lay = QGridLayout()
config_lay.addWidget(self.teams_list, 1, 0)
config_lay.addWidget(team_id_label, 2, 0)
config_lay.addWidget(self.config_team_id, 2, 1)
config_lay.addWidget(team_name_label, 3, 0)
config_lay.addWidget(self.config_team_name, 3, 1, 1, 2)
config_lay.addWidget(team_members_label, 4, 0)
config_lay.addWidget(self.config_team_members, 5, 0)
config_lay.addWidget(add_member_label, 6, 0)
config_lay.addWidget(add_member_input, 6, 1, 1, 2)
config_lay.addWidget(delete_member, 7, 0)
config_lay.addWidget(delete_team, 7, 1)
config_lay.addWidget(save_changes, 8, 0)
config_lay.addWidget(export_config, 8, 1)
self.teamedit_wid.setLayout(config_lay)
self.teamedit_wid.setWindowModality(QtCore.Qt.ApplicationModal)
self.teamedit_wid.activateWindow()
self.teamedit_wid.raise_()
self.teamedit_wid.show()
def UpdateTeamInfo(self):
sender = self.sender()
if sender.currentIndex() > 0:
team_id = sender.currentIndex()
self.config_team_id.setText(str(team_id))
self.config_team_name.setText(sender.currentText())
if self.config_team_members.count() > 0:
self.config_team_members.clear()
self.config_team_members.addItems(
self.players[sender.currentText()])
def AddNewMember(self):
sender = self.sender()
self.config_team_members.addItem(sender.text())
sender.clear()
def DeleteMember(self):
list_members = self.config_team_members.selectedItems()
if len(list_members) == 0:
QMessageBox.warning(self, "Error", "No player selected")
else:
for member in list_members:
self.config_team_members.takeItem(
self.config_team_members.row(member))
def DeleteTeam(self):
team_id = int(self.config_team_id.text())
is_last_item = self.teams[self.teams.index(
self.config_team_name.text())] == (self.teams[len(self.teams)-1])
self.teams.pop(self.teams.index(self.config_team_name.text()))
self.players.pop(self.config_team_name.text())
new_teamlist = {}
new_teamlist["teams"] = self.teams
new_teamlist["players"] = self.players
current_teams = open(THISDIR + "\\config\\teams.json", "w+",
encoding="utf-8")
current_teams.write(json.dumps(new_teamlist, indent=4))
current_teams.close()
if is_last_item == True:
self.teams_list.setCurrentIndex(1)
else:
self.teams_list.setCurrentIndex(team_id+1)
self.teams_list.removeItem(team_id)
self.UpdatePlayersList()
self.cloth_east.setModel(self.players_combobox.model())
self.cloth_south.setModel(self.players_combobox.model())
self.cloth_west.setModel(self.players_combobox.model())
self.cloth_north.setModel(self.players_combobox.model())
def ExportTeams(self):
export_dir = self.config["save_route"] if self.config["save_route"] \
is not None else THISDIR
exported_file = QFileDialog.getSaveFileName(self, "Save File",
export_dir, "Save files (*.zip)")
if exported_file[0] != "":
export_filename = exported_file[0]
if export_filename.endswith(".zip") is False:
export_filename += ".zip"
files_to_export = []
files_to_export.append("config\\teams.json")
for root, directories, files in os.walk(THISDIR+"\\images\\logos"):
for filename in files:
filepath = os.path.join(root, filename)
files_to_export.append(filepath)
with ZipFile(export_filename, "w") as export_zip:
for exp_file in files_to_export:
export_name = exp_file
if exp_file.endswith(".json"):
split_name = exp_file.split("\\")
export_name = split_name[-1]
if exp_file.endswith(".png"):
split_name = exp_file.split("\\")
export_name = "\\logos\\" + split_name[-1]
export_zip.write(exp_file, arcname=export_name)
export_zip.close()
if os.path.exists(export_filename):
QMessageBox.information(self, "Export",
"The export was successful")
def SaveEdits(self):
list_members = [str(self.config_team_members.item(i).text()) for i in \
range(self.config_team_members.count())]
self.players[self.config_team_name.text()] = list_members
new_teamlist = {}
new_teamlist["teams"] = self.teams
new_teamlist["players"] = self.players
current_teams = open(THISDIR + "\\config\\teams.json", "w+",
encoding="utf-8")
current_teams.write(json.dumps(new_teamlist, indent=4))
current_teams.close()
self.teamedit_wid.close()
self.statusBar().showMessage("Settings saved.")
def MatDialog(self):
mat_dialog = QFileDialog(self)
mat_dialog = QFileDialog.getOpenFileName(filter="Images (*.png *.jpg)",
selectedFilter="Images (*.png *.jpg)")
if mat_dialog[0] != "":
self.GenerateMat(mat_dialog[0])
def GenerateMat(self, image):
self.background = image
background = Image.open(self.background).resize((2048,2048))\
.convert("RGBA")
self.mat_thread = QThread()
east_id = self.SearchTeamID(self.cloth_east, True)
south_id = self.SearchTeamID(self.cloth_south, True)
west_id = self.SearchTeamID(self.cloth_west, True)
north_id = self.SearchTeamID(self.cloth_north, True)
if self.config["save_route"] is None:
save_to_route = THISDIR
else:
save_to_route = self.config["save_route"]
self._createProgressBar()
self.mat_worker = GenerateImageThread(background, self.table_border,
east_id, south_id, west_id, north_id,
self.technical_lines.isChecked(), save_to_route,
self.bg_image, True)
self.mat_worker.moveToThread(self.mat_thread)
self.mat_thread.started.connect(self.mat_worker.run)
self.mat_worker.update_progress.connect(self.UpdateStatus)
self.mat_worker.finished.connect(self.mat_thread.quit)
self.mat_worker.finished.connect(self.mat_worker.deleteLater)
self.mat_thread.finished.connect(self.mat_thread.deleteLater)
self.mat_thread.finished.connect(self.MatPreviewWindow)
self.mat_thread.start()
def MatPreviewWindow(self):
self.statusBar().showMessage('Mat preview generated.')
self.statusBar().removeWidget(self.progress_bar)
# Now you can go back to rigging
self.ChangeAppStatus(True)
self.mat_wid = QWidget()
self.mat_wid.resize(600, 600)
self.mat_wid.setWindowTitle("Background preview")
mat_preview_title = QLabel(self)
mat_preview_title.setText("Selected image (1/4 scale)")
mat_preview = QLabel(self)
mat_preview.setPixmap(QPixmap(tempfile.gettempdir()+"\\Table_Dif.jpg")\
.scaled(512,512))
confirm = QPushButton(self)
confirm.setText("Confirm")
confirm.clicked.connect(
lambda: self.ChangeMatImage(self.background))
vbox = QVBoxLayout()
vbox.setAlignment(QtCore.Qt.AlignCenter)
vbox.addWidget(mat_preview_title)
vbox.addWidget(mat_preview)
vbox.addWidget(confirm)
self.mat_wid.setLayout(vbox)
self.mat_wid.setWindowModality(QtCore.Qt.ApplicationModal)
self.mat_wid.activateWindow()
self.mat_wid.raise_()
self.mat_wid.show()
def ChangeMatImage(self, image):
new_bg = Image.open(image)
if new_bg.size != (2048, 2048):
new_bg = new_bg.resize((2048, 2048))
if new_bg.mode != "RGBA":
new_bg = new_bg.convert("RGBA")
if self.config["save_route"] is not None:
new_bg.save(self.config["save_route"]+"\\images\\mat.png")
self.bg_image = self.config["save_route"]+"\\images\\mat.png"
else:
new_bg.save(THISDIR+"\\images\\mat.png")
self.bg_image = THISDIR+"\\images\\mat.png"
self.background = new_bg
self.config["image_route"] = self.bg_image
new_file = open(THISDIR + "\\config\\config.json", "w+",
encoding="utf-8")
new_file.write(json.dumps(self.config, indent=4))
new_file.close()
self.statusBar().showMessage('New background added.')
self.statusBar().removeWidget(self.progress_bar)
self.ChangeAppStatus(True)
self.mat_wid.close()
def GeneratePreview(self):
self.preview_thread = QThread()
east_id = self.SearchTeamID(self.cloth_east, True)
south_id = self.SearchTeamID(self.cloth_south, True)
west_id = self.SearchTeamID(self.cloth_west, True)
north_id = self.SearchTeamID(self.cloth_north, True)
if self.config["save_route"] is None:
save_to_route = THISDIR
else:
save_to_route = self.config["save_route"]
self._createProgressBar()
self.preview_worker = GenerateImageThread(self.background,
self.table_border, east_id, south_id, west_id, north_id,
self.technical_lines.isChecked(), save_to_route,
self.bg_image, True)
self.preview_worker.moveToThread(self.preview_thread)
self.preview_thread.started.connect(self.preview_worker.run)
self.preview_worker.update_progress.connect(self.UpdateStatus)
self.preview_worker.finished.connect(self.preview_thread.quit)
self.preview_worker.finished.connect(self.preview_worker.deleteLater)
self.preview_thread.finished.connect(self.preview_thread.deleteLater)
self.preview_thread.finished.connect(self.PreviewWindow)
self.preview_thread.start()
def PreviewWindow(self):
self.statusBar().showMessage('Tablecloth preview generated.')
self.statusBar().removeWidget(self.progress_bar)
# Now you can go back to rigging
self.ChangeAppStatus(True)
self.preview_wid = QWidget()
self.preview_wid.resize(600, 600)
self.preview_wid.setWindowTitle("Tablecloth preview")
tablecloth = QPixmap(tempfile.gettempdir()+"\\Table_Dif.jpg")
tablecloth_preview_title = QLabel(self)
tablecloth_preview_title.setText("Tablecloth preview (1/4 scale)")
tablecloth_preview = QLabel(self)
tablecloth_preview.setPixmap(tablecloth.scaled(512,512))
confirm = QPushButton(self)
confirm.setText("Confirm")
confirm.clicked.connect(self.GenerateImage)
confirm.clicked.connect(self.preview_wid.close)
vbox = QVBoxLayout()
vbox.setAlignment(QtCore.Qt.AlignCenter)
vbox.addWidget(tablecloth_preview_title)
vbox.addWidget(tablecloth_preview)
vbox.addWidget(confirm)
self.preview_wid.setLayout(vbox)
self.preview_wid.setWindowModality(QtCore.Qt.ApplicationModal)
self.preview_wid.activateWindow()
self.preview_wid.raise_()
self.preview_wid.show()
def GeneratedDialog(self):
self.statusBar().showMessage('Tablecloth generated. Happy rigging!')
self.statusBar().removeWidget(self.progress_bar)
# Now you can go back to rigging
self.ChangeAppStatus(True)
mbox = QMessageBox()
mbox.setWindowTitle("Tablecloth Generator")
mbox.setText("Tablecloth Generated!")
mbox.setStandardButtons(QMessageBox.Ok)
mbox.exec()
def UpdateStatus(self, status):
self.progress_bar.setValue(status)
def GenerateImage(self):
self.statusBar().showMessage('Generating image...')
self._createProgressBar()
if self.config["save_route"] is None:
self.config["save_route"] = THISDIR
save_to_route = QFileDialog.getExistingDirectory(self,
"Where to save the image", self.config["save_route"],
QFileDialog.ShowDirsOnly | QFileDialog.DontResolveSymlinks)
if self.config["save_route"] != save_to_route:
temp_file = open(THISDIR + "\\config\\config.json", "r",
encoding="utf-8")
fp_teams = json.loads(temp_file.read())
fp_teams["save_route"] = save_to_route
fp_teams["image_route"] = self.bg_image
new_file = open(THISDIR + "\\config\\config.json", "w+",
encoding="utf-8")
new_file.write(json.dumps(fp_teams, indent=4))
new_file.close()
self.background = Image.open(THISDIR + "\\images\\mat.png")
self.table_border = Image.open(THISDIR + "\\images\\table_border.png")
self.tech_lines = Image.open(THISDIR + "\\images\\technical_lines.png")
self.thread = QThread()
east_id = self.SearchTeamID(self.cloth_east, True)
south_id = self.SearchTeamID(self.cloth_south, True)
west_id = self.SearchTeamID(self.cloth_west, True)
north_id = self.SearchTeamID(self.cloth_north, True)
self.worker = GenerateImageThread(self.background, self.table_border,
east_id, south_id, west_id, north_id,
self.technical_lines.isChecked(), save_to_route, self.bg_image)
self.worker.moveToThread(self.thread)
self.thread.started.connect(self.worker.run)
self.worker.update_progress.connect(self.UpdateStatus)
self.worker.finished.connect(self.thread.quit)
self.worker.finished.connect(self.worker.deleteLater)
self.thread.finished.connect(self.thread.deleteLater)
self.thread.finished.connect(self.GeneratedDialog)
self.thread.start()
def ChangeAppStatus(self, status):
# True for enable, False for disable.
self.cloth_east.setEnabled(status)
self.search_east.setEnabled(status)
self.cloth_south.setEnabled(status)
self.search_south.setEnabled(status)
self.cloth_west.setEnabled(status)
self.search_west.setEnabled(status)
self.cloth_north.setEnabled(status)
self.search_north.setEnabled(status)
self.generate.setEnabled(status)
def SearchTeamID(self, cloth, plus_one=False):
team_id = self.teams.index(cloth.itemData(cloth.currentIndex()))
if plus_one:
team_id += 1
return team_id
def UpdatePlayersList(self):
for team, members in self.players.items():
for member in members:
self.players_combobox.addItem(member, team)
def center(self):
qr = self.frameGeometry()
cp = QScreen().availableGeometry().center()
qr.moveCenter(cp)
def SeeVersion(self):
git_url = "https://raw.githubusercontent.com/vg-mjg/tablecloth-"
git_url += "generator/main/version.txt"
with urllib.request.urlopen(git_url) as response:
url_version = response.read().decode("utf-8")
version = "Your version is up to date!"
if url_version != VERSION:
version = "Your version is outdated."
version += "Please check the <a href='https://github.com/vg-mjg/"
version += "tablecloth-generator/releases'>Github page</a>"
version +=" for updates."
version_message = QMessageBox(self)
version_message.setWindowTitle("Checking version")
version_message.setText("""<h1>Tablecloth generator</h1>
<br>
<b>Current Version:</b> %s<br>
<b>Your Version:</b> %s<br>
<i>%s</i>
""" % (url_version, VERSION, version))
version_message.exec()
def GetHelp(self):
webbrowser.open("https://github.com/vg-mjg/tablecloth-generator/wiki")
