def simple_button(self):
sleep_btn = QPushButton('Sleep Button', self)
# sleep_btn = QPushButton('&Sleep Button', self) # Alt + S
sleep_btn.setToolTip('This is a <b>QPushButton</b> widget')
sleep_btn.resize(sleep_btn.sizeHint())
sleep_btn.move(50, 50)
sleep_btn.clicked.connect(self.show_question_message_box)
def gui_components(self):
update_data = QPushButton("Update Data", self)
update_data.clicked.connect(self.update_data)
update_data.resize(update_data.sizeHint())
update_data.move(20, 25)
render_data_button = QPushButton("Render data analysis", self)
render_data_button.clicked.connect(self.render_data)
render_data_button.move(140, 25)
render_data_button.resize(render_data_button.sizeHint())
quit_button = QPushButton("Exit", self)
quit_button.clicked.connect(QApplication.instance().quit)
quit_button.clicked.connect(QCloseEvent)
quit_button.resize(quit_button.sizeHint())
quit_button.move(300, 25)
quit_button.setToolTip("Quit program")
def setup(self):
btn_quit = QPushButton('Quit', self)
btn_quit.clicked.connect(QApplication.instance().quit)
btn_quit.resize(btn_quit.sizeHint())
btn_quit.move(100, 100)
self.setGeometry(300, 300, 300, 300)
self.setWindowTitle('PySide6 Example')
self.show()
def menu_button(self):
sleep_btn = QPushButton('Sleep Button', self)
# sleep_btn = QPushButton('&Sleep Button', self) # Alt + S
sleep_btn.setToolTip('This is a menu')
sleep_btn.resize(sleep_btn.sizeHint())
sleep_btn.move(200, 50)
menu = QMenu("kek", self)
menu.addAction(self.new_action('1', shortcut="Ctrl+A"))
menu.addAction(self.new_action('2'))
sleep_btn.setMenu(menu)
def __init__(self, config: Config) -> None:
super().__init__()
hbox = QtWidgets.QHBoxLayout(self)
btn_ingest = QPushButton('Ingest Data', self)
btn_ingest.resize(btn_ingest.sizeHint())
btn_ingest.clicked.connect(self.open_ingest)
btn_train = QPushButton('Train Model', self)
btn_train.resize(btn_train.sizeHint())
btn_train.clicked.connect(self.open_training)
btn_predict = QPushButton('Predict', self)
btn_predict.resize(btn_predict.sizeHint())
btn_predict.clicked.connect(self.open_prediction)
btn_quit = QPushButton('Force Quit', self)
btn_quit.clicked.connect(QApplication.instance().quit)
btn_quit.resize(btn_quit.sizeHint())
hbox.addWidget(btn_ingest)
hbox.addWidget(btn_train)
hbox.addWidget(btn_predict)
hbox.addWidget(btn_quit)
self.setWindowTitle('Crypto Futures')
self.show()
self.setFixedSize(self.size())
self._ingest = IngestWindow(config)
self._ingest.setWindowTitle('Ingest Data')
self._training = TrainingWindow()
self._training.setWindowTitle('Training')
self._predict = PredictionWindow()
self._predict.setWindowTitle('Predictions')
def setup(self):
#self.setGeometry(200, 200, 400, 300)
self.setWindowTitle('Quizlet Scraper')
layout = QHBoxLayout()
layout.setSpacing(20)
self.setLayout(layout)
left = QWidget()
left_layout = QVBoxLayout()
left.setLayout(left_layout)
layout.addWidget(left)
right = QWidget()
right_layout = QVBoxLayout()
right.setLayout(right_layout)
layout.addWidget(right)
url_text = QLabel('URL:')
left_layout.addWidget(url_text)
self.urlbox = QLineEdit(self)
left_layout.addWidget(self.urlbox)
self.add_deck = QWidget()
add_deck_layout = QVBoxLayout()
self.add_deck.setLayout(add_deck_layout)
right_layout.addWidget(self.add_deck)
self.add_deck.hide()
add_deck_label = QLabel('Deck name:')
add_deck_layout.addWidget(add_deck_label)
self.add_deck_box = QLineEdit(self)
add_deck_layout.addWidget(self.add_deck_box)
self.combo = QComboBox()
self.combo.addItems([
'.csv (Comma separated)',
'.txt (Tab separated)',
'.xls (Excel)',
'.apkg (Anki package)'
])
self.combo.currentIndexChanged.connect(self.index_changed)
right_layout.addWidget(self.combo)
spreadsheet_button = QPushButton('Get Spreadsheet', self)
spreadsheet_button.clicked.connect(self.get_spreadsheet)
spreadsheet_button.resize(spreadsheet_button.sizeHint())
right_layout.addWidget(spreadsheet_button)
def setup_window(self):
self.setWindowTitle("API Data GUI")
display_list = QListWidget(self)
self.list_control = display_list
self.put_data_in_list(self.data)
display_list.resize(400, 350)
self.setGeometry(300, 100, 400, 500)
quit_button = QPushButton("Quit Now", self)
quit_button.clicked.connect(QApplication.instance().quit)
quit_button.resize(quit_button.sizeHint())
quit_button.move(300, 400)
api_db_demo_button = QPushButton("Push me for Data Visual", self)
api_db_demo_button.move(100, 400)
api_db_demo_button.clicked.connect(self.do_something_to_test)
self.show()
def setup_window(self):
self.setWindowTitle("GUI Demo for Capstone")
display_list = QListWidget(self)
self.list_control = display_list
self.put_data_in_list(self.data)
display_list.resize(400, 350)
self.setGeometry(300, 100, 400, 500)
quit_button = QPushButton("Quit Now", self)
quit_button.clicked.connect(QApplication.instance().quit)
quit_button.resize(quit_button.sizeHint())
quit_button.move(300, 400)
comp490_demo_button = QPushButton("Push me for Demo", self)
comp490_demo_button.move(100, 400)
comp490_demo_button.clicked.connect(self.do_something_to_demo)
# comp490_demo_button.resize(123, 35)
self.show()
class JobsWindow(QWidget):
def __init__(self):
super().__init__()
self.quit_button = QPushButton("Quit", self)
self.update_button = QPushButton("Update Data", self)
self.enter_data = QPushButton("Enter Data", self)
self.data_button = QPushButton("Run Data Visualization", self)
self.back_button = QPushButton("Back", self)
self.text_visualization_button = QPushButton("Text Visualization", self)
self.map_visualization = QPushButton("Map Visualization", self)
self.order_selector_text = QComboBox(self)
self.data_selector_map = QComboBox(self)
self.data_visualization_label = QLabel("Welcome to data visualization!", self)
self.welcome_label = QLabel("Welcome to Jobs data Visualization.", self)
self.list_control = None
self.update_label_01 = QLabel("", self)
self.update_label_02 = QLabel("", self)
self.update_label_03 = QLabel("", self)
self.update_label_04 = QLabel("", self)
self.update_label_05 = QLabel("", self)
self.update_label_06 = QLabel("", self)
self.update_label_07 = QLabel("", self)
self.update_label_08 = QLabel("", self)
self.excel_label = QLabel("Excel Spreadsheet:", self)
self.update_information = QLabel(
"If you would like to update a single entry please select the table and enter all of its information.\n"
"If you would like to import from a spreadsheet type it into the box on the right and select the table.\n"
"When you are ready with which ever function press Enter Data.\n"
"For any data you want to append make sure the id is None.\n"
"Also only work from one method at a time and make sure he other is empty.", self)
self.table_selection = QComboBox(self)
self.update_box_01 = QLineEdit(self)
self.update_box_02 = QLineEdit(self)
self.update_box_03 = QLineEdit(self)
self.update_box_04 = QLineEdit(self)
self.update_box_05 = QLineEdit(self)
self.update_box_06 = QLineEdit(self)
self.update_box_07 = QLineEdit(self)
self.update_box_08 = QLineEdit(self)
self.update_excel_selection = QLineEdit(self)
self.list_control = None
self.setup_window()
def setup_window(self):
self.setWindowTitle("Jobs Window")
display_list = QListWidget(self)
self.list_control = display_list
display_list.resize(500, 350)
self.setGeometry(50, 50, 500, 500)
self.quit_button.clicked.connect(QApplication.instance().quit)
self.quit_button.resize(self.quit_button.sizeHint())
self.quit_button.move(415, 450)
self.update_button.clicked.connect(self.update_data)
self.update_button.move(200, 200)
self.data_button.clicked.connect(self.run_data_visualization)
self.data_button.move(175, 250)
self.back_button.clicked.connect(self.go_back)
self.back_button.move(25, 450)
self.data_visualization_label.move(20, 400)
self.text_visualization_button.move(400 - self.text_visualization_button.width(), 600)
self.order_selector_text.move(self.text_visualization_button.x(),
self.text_visualization_button.y()+self.text_visualization_button.height()+10)
self.text_visualization_button.clicked.connect(self.text_visualization)
self.map_visualization.move(self.text_visualization_button.x() + self.text_visualization_button.width(),
self.text_visualization_button.y())
self.data_selector_map.move(self.map_visualization.x(),
self.map_visualization.y()+self.map_visualization.height()+10)
self.map_visualization.clicked.connect(self.run_map_visualization)
self.welcome_label.move(150, 150)
self.update_box_01.move(150, 30)
self.update_box_02.move(150, 60)
self.update_box_03.move(150, 90)
self.update_box_04.move(150, 120)
self.update_box_05.move(150, 150)
self.update_box_06.move(150, 180)
self.update_box_07.move(150, 210)
self.update_box_08.move(150, 240)
self.excel_label.move(340, 70)
self.update_excel_selection.move(340, 90)
self.update_information.move(25, 325)
self.update_label_01.setGeometry(5, 30, 145, 20)
self.update_label_02.setGeometry(5, 60, 145, 20)
self.update_label_03.setGeometry(5, 90, 145, 20)
self.update_label_04.setGeometry(5, 120, 145, 20)
self.update_label_05.setGeometry(5, 150, 145, 20)
self.update_label_06.setGeometry(5, 180, 145, 20)
self.update_label_07.setGeometry(5, 210, 145, 20)
self.update_label_08.setGeometry(5, 240, 145, 20)
self.table_selection.setGeometry(10, 10, 145, 20)
self.enter_data.move(10, 260)
self.enter_data.clicked.connect(self.import_data)
self.table_selection.addItem("---")
self.table_selection.addItem("Schools")
self.table_selection.addItem("Jobs")
self.order_selector_text.addItem("---")
self.order_selector_text.addItem("ASC")
self.order_selector_text.addItem("DESC")
self.data_selector_map.addItem("---")
self.data_selector_map.addItem("Employment to Graduates")
self.data_selector_map.addItem("Average Salary to Average Declining Balance Percent")
self.table_selection.currentIndexChanged.connect(self.update_selection)
self.hidden_at_start()
self.show()
def hidden_at_start(self):
self.hide_update_boxes()
self.list_control.hide()
self.update_information.hide()
self.map_visualization.hide()
self.text_visualization_button.hide()
self.data_visualization_label.hide()
self.back_button.hide()
self.table_selection.hide()
self.enter_data.hide()
self.update_excel_selection.hide()
self.excel_label.hide()
self.data_selector_map.hide()
self.order_selector_text.hide()
def hide_update_boxes(self):
self.update_box_01.hide()
self.update_box_02.hide()
self.update_box_03.hide()
self.update_box_04.hide()
self.update_box_05.hide()
self.update_box_06.hide()
self.update_box_07.hide()
self.update_box_08.hide()
self.update_label_01.setText("")
self.update_label_02.setText("")
self.update_label_03.setText("")
self.update_label_04.setText("")
self.update_label_05.setText("")
self.update_label_06.setText("")
self.update_label_07.setText("")
self.update_label_08.setText("")
def update_data(self):
self.update_button.hide()
self.data_button.hide()
self.back_button.show()
self.welcome_label.hide()
self.table_selection.show()
self.enter_data.show()
self.update_information.show()
self.update_excel_selection.show()
self.excel_label.show()
def update_selection(self):
self.hide_update_boxes()
if self.table_selection.currentText() == "Jobs":
self.update_box_01.show()
self.update_label_01.setText("jobs_id")
self.update_box_02.show()
self.update_label_02.setText("state_name")
self.update_box_03.show()
self.update_label_03.setText("occupation_code")
self.update_box_04.show()
self.update_label_04.setText("tittle")
self.update_box_05.show()
self.update_label_05.setText("employment")
self.update_box_06.show()
self.update_label_06.setText("salary_25th_percentile")
elif self.table_selection.currentText() == "Schools":
self.update_box_01.show()
self.update_label_01.setText("school_id")
self.update_box_02.show()
self.update_label_02.setText("name")
self.update_box_03.show()
self.update_label_03.setText("state_abrev")
self.update_box_04.show()
self.update_label_04.setText("size_2017")
self.update_box_05.show()
self.update_label_05.setText("size_2018")
self.update_box_06.show()
self.update_label_06.setText("earnings")
self.update_box_07.show()
self.update_label_07.setText("repayment_overall")
self.update_box_08.show()
self.update_label_08.setText("repayment_cohort")
def import_data(self):
if self.update_excel_selection.text() == "":
if self.table_selection.currentText() == "Jobs":
information_to_update = [self.update_box_01.text(), self.update_box_02.text(),
self.update_box_03.text(), self.update_box_04.text(),
self.update_box_05.text(), self.update_box_06.text()]
jobs.update_data_from_list(information_to_update, "Jobs", "jobs_db.sqlite")
elif self.table_selection.currentText() == "Schools":
information_to_update = [self.update_box_01.text(), self.update_box_02.text(),
self.update_box_03.text(), self.update_box_04.text(),
self.update_box_05.text(), self.update_box_06.text(),
self.update_box_07.text(), self.update_box_08.text()]
jobs.update_data_from_list(information_to_update, "Schools", "jobs_db.sqlite")
else:
if self.table_selection.currentText() == "Jobs":
jobs.update_data_from_excel(self.update_excel_selection.text(), "Jobs", "jobs_db.sqlite")
elif self.table_selection.currentText() == "Schools":
jobs.update_data_from_excel(self.update_excel_selection.text(), "Jobs", "jobs_db.sqlite")
def run_data_visualization(self):
self.update_button.hide()
self.data_button.hide()
self.data_visualization_label.show()
self.back_button.show()
self.setGeometry(50, 50, 800, 800)
self.back_button.move(25, 750)
self.quit_button.move(715, 750)
self.welcome_label.hide()
self.map_visualization.show()
self.text_visualization_button.show()
self.data_selector_map.show()
self.order_selector_text.show()
def go_back(self):
self.back_button.hide()
self.update_button.show()
self.data_button.show()
self.data_visualization_label.hide()
self.setGeometry(50, 50, 500, 500)
self.back_button.move(25, 450)
self.quit_button.move(415, 450)
self.welcome_label.show()
self.hidden_at_start()
def text_visualization(self):
self.list_control.clear()
conn, cursor = jobs.open_db("jobs_db.sqlite")
if self.order_selector_text.currentText() == "ASC":
data_visualization_per_state = jobs.query_run('''SELECT state_abrev, state_name, ''' + '''
total(jobs.employment) as employment,
total(school.size_2018/4),
round(avg(school.repayment_cohort),3) as repayment_cohort,
round(avg(jobs.salary_25th_percentile)) as averge_entry_salary
FROM school
JOIN states using(state_abrev)
JOIN jobs using(state_name)
GROUP BY state_name
ORDER BY employment ASC;''', cursor)
elif self.order_selector_text.currentText() == "DESC":
data_visualization_per_state = jobs.query_run('''SELECT state_abrev, state_name, ''' + '''
total(jobs.employment) as employment,
total(school.size_2018/4),
round(avg(school.repayment_cohort),3) as repayment_cohort,
round(avg(jobs.salary_25th_percentile)) as averge_entry_salary
FROM school
JOIN states using(state_abrev)
JOIN jobs using(state_name)
GROUP BY state_name
ORDER BY employment DESC;''', cursor)
else:
data_visualization_per_state = jobs.query_run('''SELECT state_abrev, state_name, ''' + '''
total(jobs.employment) as employment,
total(school.size_2018/4),
round(avg(school.repayment_cohort),3) as repayment_cohort,
round(avg(jobs.salary_25th_percentile)) as averge_entry_salary
FROM school
JOIN states using(state_abrev)
JOIN jobs using(state_name)
GROUP BY state_name
;''', cursor)
QListWidgetItem("State", listview=self.list_control)
for state in data_visualization_per_state:
state_display_data = f"{state[0]}, {state[1]}"
grad_employ_data = f"Employment/Graduates: {state[2]/state[3]}"
repayment_data = f"Average Entry Salary/Average Declining Balance Percent: {state[5]/state[4]}"
state_item = QListWidgetItem(state_display_data, listview=self.list_control)
grad_item = QListWidgetItem(grad_employ_data, listview=self.list_control)
repayment_item = QListWidgetItem(repayment_data, listview=self.list_control)
grad_item.setForeground(Qt.darkGreen)
repayment_item.setForeground(Qt.blue)
state_item.setForeground(Qt.white)
state_item.setBackground(Qt.black)
self.list_control.show()
jobs.close_db(conn)
def run_map_visualization(self):
conn, cursor = jobs.open_db("jobs_db.sqlite")
data_visualization_per_state = jobs.query_run('''SELECT state_abrev, state_name, ''' + '''
total(jobs.employment) as employment,
total(school.size_2018/4),
round(avg(school.repayment_cohort),3) as repayment_cohort,
round(avg(jobs.salary_25th_percentile)) as averge_entry_salary
FROM school
JOIN states using(state_abrev)
JOIN jobs using(state_name)
GROUP BY state_name
;''', cursor)
state_abrev = []
state_grads = []
state_repayment = []
for state in data_visualization_per_state:
state_abrev.append(state[0])
state_grads.append(state[2]/state[3])
state_repayment.append(state[5]/state[4])
if self.data_selector_map.currentText() == "Graduates to Employment":
us_map = px.Figure(data=px.Choropleth(locations=state_abrev, z=state_grads,
locationmode='USA-states', colorbar_title="Employment/Graduates"
))
us_map.update_layout(geo_scope='usa', title_text='Employment VS Graduates By State')
us_map.show()
elif self.data_selector_map.currentText() == "Average Declining Balance Percent":
us_map = px.Figure(data=px.Choropleth(locations=state_abrev, z=state_repayment,
locationmode='USA-states', colorbar_title="Salary/Average Percent"
))
us_map.update_layout(geo_scope='usa', title_text='Average Salary VS Average '
'Percent of People with Declining Loans')
us_map.show()
else:
us_map = px.Figure(data=px.Choropleth(locations=state_abrev, z=state_grads,
locationmode='USA-states', colorbar_title="Graduates"
))
us_map.update_layout(geo_scope='usa', title_text='Graduates By State')
us_map.show()
jobs.close_db(conn)
self.list_control.hide()
class UIOpenPatientWindow(object):
patient_info_initialized = QtCore.Signal(object)
def setup_ui(self, open_patient_window_instance):
if platform.system() == 'Darwin':
self.stylesheet_path = "res/stylesheet.qss"
else:
self.stylesheet_path = "res/stylesheet-win-linux.qss"
window_icon = QIcon()
window_icon.addPixmap(QPixmap(resource_path("res/images/icon.ico")),
QIcon.Normal, QIcon.Off)
open_patient_window_instance.setObjectName("OpenPatientWindowInstance")
open_patient_window_instance.setWindowIcon(window_icon)
open_patient_window_instance.resize(840, 530)
# Create a vertical box for containing the other elements and layouts
self.open_patient_window_instance_vertical_box = QVBoxLayout()
self.open_patient_window_instance_vertical_box.setObjectName(
"OpenPatientWindowInstanceVerticalBox")
# Create a label to prompt the user to enter the path to the directory that contains the DICOM files
self.open_patient_directory_prompt = QLabel()
self.open_patient_directory_prompt.setObjectName(
"OpenPatientDirectoryPrompt")
self.open_patient_directory_prompt.setAlignment(Qt.AlignLeft)
self.open_patient_window_instance_vertical_box.addWidget(
self.open_patient_directory_prompt)
# Create a horizontal box to hold the input box for the directory and the choose button
self.open_patient_directory_input_horizontal_box = QHBoxLayout()
self.open_patient_directory_input_horizontal_box.setObjectName(
"OpenPatientDirectoryInputHorizontalBox")
# Create a textbox to contain the path to the directory that contains the DICOM files
self.open_patient_directory_input_box = UIOpenPatientWindowDragAndDropEvent(
self)
self.open_patient_directory_input_box.setObjectName(
"OpenPatientDirectoryInputBox")
self.open_patient_directory_input_box.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.open_patient_directory_input_box.returnPressed.connect(
self.scan_directory_for_patient)
self.open_patient_directory_input_horizontal_box.addWidget(
self.open_patient_directory_input_box)
# Create a choose button to open the file dialog
self.open_patient_directory_choose_button = QPushButton()
self.open_patient_directory_choose_button.setObjectName(
"OpenPatientDirectoryChooseButton")
self.open_patient_directory_choose_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.open_patient_directory_choose_button.resize(
self.open_patient_directory_choose_button.sizeHint().width(),
self.open_patient_directory_input_box.height())
self.open_patient_directory_choose_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.open_patient_directory_input_horizontal_box.addWidget(
self.open_patient_directory_choose_button)
self.open_patient_directory_choose_button.clicked.connect(
self.choose_button_clicked)
# Create a widget to hold the input fields
self.open_patient_directory_input_widget = QWidget()
self.open_patient_directory_input_horizontal_box.setStretch(0, 4)
self.open_patient_directory_input_widget.setLayout(
self.open_patient_directory_input_horizontal_box)
self.open_patient_window_instance_vertical_box.addWidget(
self.open_patient_directory_input_widget)
# Create a horizontal box to hold the stop button and direction to the user on where to select the patient
self.open_patient_appear_prompt_and_stop_horizontal_box = QHBoxLayout()
self.open_patient_appear_prompt_and_stop_horizontal_box.setObjectName(
"OpenPatientAppearPromptAndStopHorizontalBox")
# Create a label to show direction on where the files will appear
self.open_patient_directory_appear_prompt = QLabel()
self.open_patient_directory_appear_prompt.setObjectName(
"OpenPatientDirectoryAppearPrompt")
self.open_patient_directory_appear_prompt.setAlignment(Qt.AlignLeft)
self.open_patient_appear_prompt_and_stop_horizontal_box.addWidget(
self.open_patient_directory_appear_prompt)
self.open_patient_appear_prompt_and_stop_horizontal_box.addStretch(1)
# Create a button to stop searching
self.open_patient_window_stop_button = QPushButton()
self.open_patient_window_stop_button.setObjectName(
"OpenPatientWindowStopButton")
self.open_patient_window_stop_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.open_patient_window_stop_button.resize(
self.open_patient_window_stop_button.sizeHint().width(),
self.open_patient_window_stop_button.sizeHint().height())
self.open_patient_window_stop_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.open_patient_window_stop_button.clicked.connect(
self.stop_button_clicked)
self.open_patient_window_stop_button.setProperty(
"QPushButtonClass", "fail-button")
self.open_patient_window_stop_button.setVisible(
False) # Button doesn't show until a search commences
self.open_patient_appear_prompt_and_stop_horizontal_box.addWidget(
self.open_patient_window_stop_button)
# Create a widget to hold the layout
self.open_patient_appear_prompt_and_stop_widget = QWidget()
self.open_patient_appear_prompt_and_stop_widget.setLayout(
self.open_patient_appear_prompt_and_stop_horizontal_box)
self.open_patient_window_instance_vertical_box.addWidget(
self.open_patient_appear_prompt_and_stop_widget)
# Create a tree view list to list out all patients in the directory selected above
self.open_patient_window_patients_tree = QTreeWidget()
self.open_patient_window_patients_tree.setObjectName(
"OpenPatientWindowPatientsTree")
self.open_patient_window_patients_tree.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.open_patient_window_patients_tree.resize(
self.open_patient_window_patients_tree.sizeHint().width(),
self.open_patient_window_patients_tree.sizeHint().height())
self.open_patient_window_patients_tree.setHeaderHidden(False)
self.open_patient_window_patients_tree.setHeaderLabels([""])
self.open_patient_window_patients_tree.itemChanged.connect(
self.tree_item_changed)
self.open_patient_window_instance_vertical_box.addWidget(
self.open_patient_window_patients_tree)
self.last_patient = None
# Create a label to show what would happen if they select the patient
self.open_patient_directory_result_label = QtWidgets.QLabel()
self.open_patient_directory_result_label.setObjectName(
"OpenPatientDirectoryResultLabel")
self.open_patient_directory_result_label.setAlignment(Qt.AlignLeft)
self.open_patient_window_instance_vertical_box.addWidget(
self.open_patient_directory_result_label)
# Create a horizontal box to hold the Cancel and Open button
self.open_patient_window_patient_open_actions_horizontal_box = QHBoxLayout(
)
self.open_patient_window_patient_open_actions_horizontal_box.setObjectName(
"OpenPatientWindowPatientOpenActionsHorizontalBox")
self.open_patient_window_patient_open_actions_horizontal_box.addStretch(
1)
# Add a button to go back/exit from the application
self.open_patient_window_exit_button = QPushButton()
self.open_patient_window_exit_button.setObjectName(
"OpenPatientWindowExitButton")
self.open_patient_window_exit_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.open_patient_window_exit_button.resize(
self.open_patient_window_stop_button.sizeHint().width(),
self.open_patient_window_stop_button.sizeHint().height())
self.open_patient_window_exit_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.open_patient_window_exit_button.clicked.connect(
self.exit_button_clicked)
self.open_patient_window_exit_button.setProperty(
"QPushButtonClass", "fail-button")
self.open_patient_window_patient_open_actions_horizontal_box.addWidget(
self.open_patient_window_exit_button)
# Add a button to confirm opening of the patient
self.open_patient_window_confirm_button = QPushButton()
self.open_patient_window_confirm_button.setObjectName(
"OpenPatientWindowConfirmButton")
self.open_patient_window_confirm_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.open_patient_window_confirm_button.resize(
self.open_patient_window_confirm_button.sizeHint().width(),
self.open_patient_window_confirm_button.sizeHint().height())
self.open_patient_window_confirm_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.open_patient_window_confirm_button.setDisabled(True)
self.open_patient_window_confirm_button.clicked.connect(
self.confirm_button_clicked)
self.open_patient_window_confirm_button.setProperty(
"QPushButtonClass", "success-button")
self.open_patient_window_patient_open_actions_horizontal_box.addWidget(
self.open_patient_window_confirm_button)
# Create a widget to house all of the actions button for open patient window
self.open_patient_window_patient_open_actions_widget = QWidget()
self.open_patient_window_patient_open_actions_widget.setLayout(
self.open_patient_window_patient_open_actions_horizontal_box)
self.open_patient_window_instance_vertical_box.addWidget(
self.open_patient_window_patient_open_actions_widget)
# Set the vertical box fourth element, the tree view, to stretch out as far as possible
self.open_patient_window_instance_vertical_box.setStretch(
3, 4) # Stretch the treeview out as far as possible
self.open_patient_window_instance_central_widget = QWidget()
self.open_patient_window_instance_central_widget.setObjectName(
"OpenPatientWindowInstanceCentralWidget")
self.open_patient_window_instance_central_widget.setLayout(
self.open_patient_window_instance_vertical_box)
# Create threadpool for multithreading
self.threadpool = QThreadPool()
print("Multithreading with maximum %d threads" %
self.threadpool.maxThreadCount())
# Create interrupt event for stopping the directory search
self.interrupt_flag = threading.Event()
# Bind all texts into the buttons and labels
self.retranslate_ui(open_patient_window_instance)
# Set the central widget, ready for display
open_patient_window_instance.setCentralWidget(
self.open_patient_window_instance_central_widget)
# Set the current stylesheet to the instance and connect it back to the caller through slot
_stylesheet = open(resource_path(self.stylesheet_path)).read()
open_patient_window_instance.setStyleSheet(_stylesheet)
QtCore.QMetaObject.connectSlotsByName(open_patient_window_instance)
def retranslate_ui(self, open_patient_window_instance):
_translate = QtCore.QCoreApplication.translate
open_patient_window_instance.setWindowTitle(
_translate("OpenPatientWindowInstance",
"OnkoDICOM - Select Patient"))
self.open_patient_directory_prompt.setText(
_translate(
"OpenPatientWindowInstance",
"Choose the path of the folder containing DICOM files to load Patient's details:"
))
self.open_patient_directory_input_box.setPlaceholderText(
_translate(
"OpenPatientWindowInstance",
"Enter DICOM Files Path (For example, C:\path\\to\your\DICOM\Files)"
))
self.open_patient_directory_choose_button.setText(
_translate("OpenPatientWindowInstance", "Choose"))
self.open_patient_directory_appear_prompt.setText(
_translate(
"OpenPatientWindowInstance",
"Patient File directory shown below once file path chosen. Please select the file(s) you want to open:"
))
self.open_patient_directory_result_label.setText(
"The selected directory(s) above will be opened in the OnkoDICOM program."
)
self.open_patient_window_stop_button.setText(
_translate("OpenPatientWindowInstance", "Stop Search"))
self.open_patient_window_exit_button.setText(
_translate("OpenPatientWindowInstance", "Exit"))
self.open_patient_window_confirm_button.setText(
_translate("OpenPatientWindowInstance", "Confirm"))
def exit_button_clicked(self):
QCoreApplication.exit(0)
def scan_directory_for_patient(self):
# Reset tree view header and last patient
self.open_patient_window_patients_tree.setHeaderLabels([""])
self.last_patient = None
self.filepath = self.open_patient_directory_input_box.text()
# Proceed if a folder was selected
if self.filepath != "":
# Update the QTreeWidget to reflect data being loaded
# First, clear the widget of any existing data
self.open_patient_window_patients_tree.clear()
# Next, update the tree widget
self.open_patient_window_patients_tree.addTopLevelItem(
QTreeWidgetItem(["Loading selected directory..."]))
# The choose button is disabled until the thread finishes executing
self.open_patient_directory_choose_button.setEnabled(False)
# Reveals the Stop Search button for the duration of the search
self.open_patient_window_stop_button.setVisible(True)
# The interrupt flag is then un-set if a previous search has been stopped.
self.interrupt_flag.clear()
# Then, create a new thread that will load the selected folder
worker = Worker(DICOMDirectorySearch.get_dicom_structure,
self.filepath,
self.interrupt_flag,
progress_callback=True)
worker.signals.result.connect(self.on_search_complete)
worker.signals.progress.connect(self.search_progress)
# Execute the thread
self.threadpool.start(worker)
def choose_button_clicked(self):
"""
Executes when the choose button is clicked.
Gets filepath from the user and loads all files and subdirectories.
"""
# Get folder path from pop up dialog box
self.filepath = QtWidgets.QFileDialog.getExistingDirectory(
None, 'Select patient folder...', '')
self.open_patient_directory_input_box.setText(self.filepath)
self.scan_directory_for_patient()
def stop_button_clicked(self):
self.interrupt_flag.set()
def search_progress(self, progress_update):
"""
Current progress of the file search.
"""
self.open_patient_window_patients_tree.clear()
self.open_patient_window_patients_tree.addTopLevelItem(
QTreeWidgetItem([
"Loading selected directory... (%s files searched)" %
progress_update
]))
def on_search_complete(self, dicom_structure):
"""
Executes once the directory search is complete.
:param dicom_structure: DICOMStructure object constructed by the directory search.
"""
self.open_patient_directory_choose_button.setEnabled(True)
self.open_patient_window_stop_button.setVisible(False)
self.open_patient_window_patients_tree.clear()
if dicom_structure is None: # dicom_structure will be None if function was interrupted.
return
for patient_item in dicom_structure.get_tree_items_list():
self.open_patient_window_patients_tree.addTopLevelItem(
patient_item)
if len(dicom_structure.patients) == 0:
QMessageBox.about(self, "No files found",
"Selected directory contains no DICOM files.")
def tree_item_changed(self, item, _):
"""
Executes when a tree item is checked or unchecked.
If a different patient is checked, uncheck the previous patient.
Inform user about missing DICOM files.
"""
selected_patient = item
# If the item is not top-level, bubble up to see which top-level item this item belongs to
if self.open_patient_window_patients_tree.invisibleRootItem(
).indexOfChild(item) == -1:
while self.open_patient_window_patients_tree.invisibleRootItem(
).indexOfChild(selected_patient) == -1:
selected_patient = selected_patient.parent()
# Uncheck previous patient if a different patient is selected
if item.checkState(
0
) == Qt.CheckState.Checked and self.last_patient != selected_patient:
if self.last_patient is not None:
self.last_patient.setCheckState(0, Qt.CheckState.Unchecked)
self.last_patient.setSelected(False)
self.last_patient = selected_patient
# Get the types of all selected leaves
self.selected_series_types = set()
for checked_item in self.get_checked_leaves():
series_type = checked_item.dicom_object.get_series_type()
if type(series_type) == str:
self.selected_series_types.add(series_type)
else:
self.selected_series_types.update(series_type)
# Check the existence of IMAGE, RTSTRUCT and RTDOSE files
if len(list({'CT', 'MR', 'PT'} & self.selected_series_types)) == 0:
header = "Cannot proceed without an image file."
self.open_patient_window_confirm_button.setDisabled(True)
elif 'RTSTRUCT' not in self.selected_series_types:
header = "DVH and Radiomics calculations are not available without a RTSTRUCT file."
elif 'RTDOSE' not in self.selected_series_types:
header = "DVH calculations are not available without a RTDOSE file."
else:
header = ""
self.open_patient_window_patients_tree.setHeaderLabel(header)
if len(list({'CT', 'MR', 'PT'} & self.selected_series_types)) != 0:
self.open_patient_window_confirm_button.setDisabled(False)
def confirm_button_clicked(self):
"""
Begins loading of the selected files.
"""
selected_files = []
for item in self.get_checked_leaves():
selected_files += item.dicom_object.get_files()
self.progress_window = ProgressWindow(
self, QtCore.Qt.WindowTitleHint | QtCore.Qt.WindowCloseButtonHint)
self.progress_window.signal_loaded.connect(self.on_loaded)
self.progress_window.signal_error.connect(self.on_loading_error)
self.progress_window.start_loading(selected_files)
self.progress_window.exec_()
def on_loaded(self, results):
"""
Executes when the progress bar finishes loaded the selected files.
"""
if results[0] is True: # Will be NoneType if loading was interrupted.
self.patient_info_initialized.emit(
results[1]) # Emits the progress window.
def on_loading_error(self, error_code):
"""
Error handling for progress window.
"""
if error_code == 0:
QMessageBox.about(
self.progress_window, "Unable to open selection",
"Selected files cannot be opened as they are not a DICOM-RT set."
)
self.progress_window.close()
elif error_code == 1:
QMessageBox.about(
self.progress_window, "Unable to open selection",
"Selected files cannot be opened as they contain unsupported DICOM classes."
)
self.progress_window.close()
def get_checked_leaves(self):
"""
:return: A list of all QTreeWidgetItems in the QTreeWidget that are both leaves and checked.
"""
checked_items = []
def recurse(parent_item: QTreeWidgetItem):
for i in range(parent_item.childCount()):
child = parent_item.child(i)
grand_children = child.childCount()
if grand_children > 0:
recurse(child)
else:
if child.checkState(0) == Qt.Checked:
checked_items.append(child)
recurse(self.open_patient_window_patients_tree.invisibleRootItem())
return checked_items
class UIDeleteROIWindow():
def setup_ui(self, delete_roi_window_instance, regions_of_interest,
dataset_rtss, deleting_rois_structure_tuple):
# Initialise the 2 lists for containing the ROI(s) that we are going to keep and delete respectively
self.regions_of_interest_to_keep = []
self.regions_of_interest_to_delete = []
# This is for holding the original list of ROI(s)
self.regions_of_interest_list = regions_of_interest
# This is for holding the DICOM dataset of that specific patient
self.dataset_rtss = dataset_rtss
# Assigning new tuple for holding the deleting ROI(s)
self.deleting_rois_structure_tuple = deleting_rois_structure_tuple
# Initialise a DeleteROIWindow
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)
delete_roi_window_instance.setObjectName("DeleteRoiWindowInstance")
delete_roi_window_instance.setWindowIcon(window_icon)
delete_roi_window_instance.resize(800, 606)
# Create a vertical box to hold all widgets
self.delete_roi_window_instance_vertical_box = QVBoxLayout()
self.delete_roi_window_instance_vertical_box.setObjectName(
"DeleteRoiWindowInstanceVerticalBox")
# Create a label for holding the window's title
self.delete_roi_window_title = QLabel()
self.delete_roi_window_title.setObjectName("DeleteRoiWindowTitle")
self.delete_roi_window_title.setProperty("QLabelClass", "window-title")
self.delete_roi_window_title.setAlignment(Qt.AlignLeft)
self.delete_roi_window_instance_vertical_box.addWidget(
self.delete_roi_window_title)
# Create a label for holding the instruction of how to delete the ROIs
self.delete_roi_window_instruction = QLabel()
self.delete_roi_window_instruction.setObjectName(
"DeleteRoiWindowInstruction")
self.delete_roi_window_instruction.setAlignment(Qt.AlignCenter)
self.delete_roi_window_instance_vertical_box.addWidget(
self.delete_roi_window_instruction)
# Create a horizontal box for holding the 2 lists and the move left, move right buttons
self.delete_roi_window_keep_and_delete_box = QHBoxLayout()
self.delete_roi_window_keep_and_delete_box.setObjectName(
"DeleteRoiWindowKeepAndDeleteBox")
# ================================= KEEP BOX =================================
# Create a vertical box for holding the label and the tree view for holding the ROIs that we are keeping
self.delete_roi_window_keep_vertical_box = QVBoxLayout()
self.delete_roi_window_keep_vertical_box.setObjectName(
"DeleteRoiWindowKeepVerticalBox")
# Create a label for the tree view with the list of ROIs to keep
self.delete_roi_window_keep_tree_view_label = QLabel()
self.delete_roi_window_keep_tree_view_label.setObjectName(
"DeleteRoiWindowKeepTreeViewLabel")
self.delete_roi_window_keep_tree_view_label.setProperty(
"QLabelClass", ["tree-view-label", "tree-view-label-keep-delete"])
self.delete_roi_window_keep_tree_view_label.setAlignment(
Qt.AlignCenter)
self.delete_roi_window_keep_tree_view_label.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.delete_roi_window_keep_tree_view_label.resize(
self.delete_roi_window_keep_tree_view_label.sizeHint().width(),
self.delete_roi_window_keep_tree_view_label.sizeHint().height())
self.delete_roi_window_keep_vertical_box.addWidget(
self.delete_roi_window_keep_tree_view_label)
# Create a tree view for containing the list of ROIs to keep
self.delete_roi_window_keep_tree_view = QTreeWidget()
self.delete_roi_window_keep_tree_view.setObjectName(
"DeleteRoiWindowKeepTreeView")
self.delete_roi_window_keep_tree_view.setHeaderHidden(True)
self.delete_roi_window_keep_tree_view.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.delete_roi_window_keep_tree_view.resize(
self.delete_roi_window_keep_tree_view.sizeHint().width(),
self.delete_roi_window_keep_tree_view.sizeHint().height())
self.delete_roi_window_keep_vertical_box.addWidget(
self.delete_roi_window_keep_tree_view)
self.delete_roi_window_keep_vertical_box.setStretch(1, 4)
# Create a widget to hold the keep vertical box
self.delete_roi_window_keep_widget = QWidget()
self.delete_roi_window_keep_widget.setObjectName(
"DeleteRoiWindowKeepWidget")
self.delete_roi_window_keep_widget.setLayout(
self.delete_roi_window_keep_vertical_box)
self.delete_roi_window_keep_and_delete_box.addStretch(1)
self.delete_roi_window_keep_and_delete_box.addWidget(
self.delete_roi_window_keep_widget)
# ================================= KEEP BOX =================================
# ================================= MOVE LEFT/RIGHT BOX =================================
# Create a vertical box for holding the 2 buttons for moving left and right
self.delete_roi_window_move_left_right_vertical_box = QVBoxLayout()
self.delete_roi_window_move_left_right_vertical_box.setObjectName(
"DeleteRoiWindowMoveLeftRightVerticalBox")
# Create Move Right Button / Delete Button
self.move_right_button = QPushButton()
self.move_right_button.setObjectName("MoveRightButton")
self.move_right_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.move_right_button.resize(
self.move_right_button.sizeHint().width(),
self.move_right_button.sizeHint().height())
self.move_right_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.move_right_button.clicked.connect(
self.move_right_button_onClicked)
self.move_right_button.setProperty("QPushButtonClass", "fail-button")
self.delete_roi_window_move_left_right_vertical_box.addStretch(1)
self.delete_roi_window_move_left_right_vertical_box.addWidget(
self.move_right_button)
# Create Move Left Button / Keep Button
self.move_left_button = QPushButton()
self.move_left_button.setObjectName("MoveLeftButton")
self.move_left_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.move_left_button.resize(self.move_left_button.sizeHint().width(),
self.move_left_button.sizeHint().height())
self.move_left_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.move_left_button.clicked.connect(self.move_left_button_onClicked)
self.move_left_button.setProperty("QPushButtonClass", "success-button")
self.delete_roi_window_move_left_right_vertical_box.addWidget(
self.move_left_button)
self.delete_roi_window_move_left_right_vertical_box.addStretch(1)
# Create a widget for holding the 2 buttons
self.delete_roi_window_move_left_right_widget = QWidget()
self.delete_roi_window_move_left_right_widget.setObjectName(
"DeleteRoiWindowMoveLeftRightWidget")
self.delete_roi_window_move_left_right_widget.setLayout(
self.delete_roi_window_move_left_right_vertical_box)
self.delete_roi_window_keep_and_delete_box.addWidget(
self.delete_roi_window_move_left_right_widget)
# ================================= MOVE LEFT/RIGHT BOX =================================
# ================================= DELETE BOX =================================
# Create a vertical box for holding the label and the tree view for holding the ROIs that we are deleting
self.delete_roi_window_delete_vertical_box = QVBoxLayout()
self.delete_roi_window_delete_vertical_box.setObjectName(
"DeleteRoiWindowDeleteVerticalBox")
# Create a label for the tree view with the list of ROIs to delete
self.delete_roi_window_delete_tree_view_label = QLabel()
self.delete_roi_window_delete_tree_view_label.setObjectName(
"DeleteRoiWindowDeleteTreeViewLabel")
self.delete_roi_window_delete_tree_view_label.setProperty(
"QLabelClass", ["tree-view-label", "tree-view-label-keep-delete"])
self.delete_roi_window_delete_tree_view_label.setAlignment(
Qt.AlignCenter)
self.delete_roi_window_delete_tree_view_label.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.delete_roi_window_delete_tree_view_label.resize(
self.delete_roi_window_delete_tree_view_label.sizeHint().width(),
self.delete_roi_window_delete_tree_view_label.sizeHint().height())
self.delete_roi_window_delete_vertical_box.addWidget(
self.delete_roi_window_delete_tree_view_label)
# Create a tree view for containing the list of ROIs to delete
self.delete_roi_window_delete_tree_view = QTreeWidget()
self.delete_roi_window_delete_tree_view.setObjectName(
"DeleteRoiWindowDeleteTreeView")
self.delete_roi_window_delete_tree_view.setHeaderHidden(True)
self.delete_roi_window_delete_tree_view.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.delete_roi_window_delete_tree_view.resize(
self.delete_roi_window_delete_tree_view.sizeHint().width(),
self.delete_roi_window_delete_tree_view.sizeHint().height())
self.delete_roi_window_delete_vertical_box.addWidget(
self.delete_roi_window_delete_tree_view)
self.delete_roi_window_delete_vertical_box.setStretch(1, 4)
# Create a widget to hold the delete vertical box
self.delete_roi_window_delete_widget = QWidget()
self.delete_roi_window_delete_widget.setObjectName(
"DeleteRoiWindowDeleteWidget")
self.delete_roi_window_delete_widget.setLayout(
self.delete_roi_window_delete_vertical_box)
self.delete_roi_window_keep_and_delete_box.addWidget(
self.delete_roi_window_delete_widget)
self.delete_roi_window_keep_and_delete_box.addStretch(1)
self.delete_roi_window_keep_and_delete_box.setStretch(1, 4)
self.delete_roi_window_keep_and_delete_box.setStretch(3, 4)
# ================================= DELETE BOX =================================
# Create a widget to hold the keep and delete box
self.delete_roi_window_keep_and_delete_widget = QWidget()
self.delete_roi_window_keep_and_delete_widget.setObjectName(
"DeleteRoiWindowKeepAndDeleteWidget")
self.delete_roi_window_keep_and_delete_widget.setLayout(
self.delete_roi_window_keep_and_delete_box)
self.delete_roi_window_instance_vertical_box.addWidget(
self.delete_roi_window_keep_and_delete_widget)
# Create a horizontal box to hold 2 action buttons for this window
self.delete_roi_window_action_buttons_box = QHBoxLayout()
self.delete_roi_window_action_buttons_box.setObjectName(
"DeleteRoiWindowActionButtonsBox")
# Create the cancel button
self.delete_roi_window_cancel_button = QPushButton()
self.delete_roi_window_cancel_button.setObjectName(
"DeleteRoiWindowCancelButton")
self.delete_roi_window_cancel_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.delete_roi_window_cancel_button.resize(
self.delete_roi_window_cancel_button.sizeHint().width(),
self.delete_roi_window_cancel_button.sizeHint().height())
self.delete_roi_window_cancel_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.delete_roi_window_cancel_button.clicked.connect(
self.on_cancel_button_clicked)
self.delete_roi_window_cancel_button.setProperty(
"QPushButtonClass", "fail-button")
self.delete_roi_window_action_buttons_box.addStretch(1)
self.delete_roi_window_action_buttons_box.addWidget(
self.delete_roi_window_cancel_button)
# Create the confirm button
self.delete_roi_window_confirm_button = QPushButton()
self.delete_roi_window_confirm_button.setObjectName(
"DeleteRoiWindowConfirmButton")
self.delete_roi_window_confirm_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.delete_roi_window_confirm_button.resize(
self.delete_roi_window_confirm_button.sizeHint().width(),
self.delete_roi_window_confirm_button.sizeHint().height())
self.delete_roi_window_confirm_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.delete_roi_window_confirm_button.clicked.connect(
self.confirm_button_onClicked)
self.delete_roi_window_confirm_button.setEnabled(False)
self.delete_roi_window_confirm_button.setProperty(
"QPushButtonClass", "success-button")
self.delete_roi_window_action_buttons_box.addWidget(
self.delete_roi_window_confirm_button)
# Create a widget to hold the action buttons
self.delete_roi_window_action_buttons_widget = QWidget()
self.delete_roi_window_action_buttons_widget.setObjectName(
"DeleteRoiWindowActionButtonsWidget")
self.delete_roi_window_action_buttons_widget.setLayout(
self.delete_roi_window_action_buttons_box)
self.delete_roi_window_instance_vertical_box.addWidget(
self.delete_roi_window_action_buttons_widget)
# Set text for all attributes
self.retranslate_ui(delete_roi_window_instance)
# Create a central widget to hold the vertical layout box
self.delete_roi_window_instance_central_widget = QWidget()
self.delete_roi_window_instance_central_widget.setObjectName(
"DeleteRoiWindowInstanceCentralWidget")
self.delete_roi_window_instance_central_widget.setLayout(
self.delete_roi_window_instance_vertical_box)
self.delete_roi_window_instance_vertical_box.setStretch(2, 4)
# Set the central widget for the main window and style the window
delete_roi_window_instance.setCentralWidget(
self.delete_roi_window_instance_central_widget)
delete_roi_window_instance.setStyleSheet(stylesheet)
# Load the ROIs in
self.display_rois_in_listViewKeep()
# Set the selection mode to multi so that we can select multiple ROIs to delete
self.delete_roi_window_keep_tree_view.setSelectionMode(
QAbstractItemView.MultiSelection)
self.delete_roi_window_delete_tree_view.setSelectionMode(
QAbstractItemView.MultiSelection)
QtCore.QMetaObject.connectSlotsByName(delete_roi_window_instance)
def retranslate_ui(self, delete_roi_window_instance):
_translate = QtCore.QCoreApplication.translate
delete_roi_window_instance.setWindowTitle(
_translate("DeleteRoiWindowInstance", "OnkoDICOM - Delete ROI(s)"))
self.delete_roi_window_title.setText(
_translate("DeleteRoiWindowTitle", "Delete ROI(s)"))
self.delete_roi_window_instruction.setText(
_translate(
"DeleteRoiWindowInstruction",
"Move the Regions of Interest to be deleted to the right-hand side or vice versa"
))
self.delete_roi_window_keep_tree_view_label.setText(
_translate("DeleteRoiWindowKeepTreeViewLabel", "To Keep"))
self.delete_roi_window_delete_tree_view_label.setText(
_translate("DeleteRoiWindowDeleteTreeViewLabel", "To Delete"))
self.move_right_button.setText(
_translate("MoveRightButton", "Move Right ->>>"))
self.move_left_button.setText(
_translate("MoveLeftButton", "<<<- Move Left"))
self.delete_roi_window_cancel_button.setText(
_translate("DeleteRoiWindowCancelButton", "Cancel"))
self.delete_roi_window_confirm_button.setText(
_translate("DeleteRoiWindowConfirmButton", "Confirm"))
def on_cancel_button_clicked(self):
self.close()
def display_rois_in_listViewKeep(self):
self.regions_of_interest_to_keep.clear()
for roi_id, roi_dict in self.regions_of_interest_list.items():
self.regions_of_interest_to_keep.append(roi_dict['name'])
self.delete_roi_window_keep_tree_view.clear()
self.delete_roi_window_keep_tree_view.setIndentation(0)
self.item = QTreeWidgetItem(["item"])
for index in self.regions_of_interest_to_keep:
item = QTreeWidgetItem([index])
self.delete_roi_window_keep_tree_view.addTopLevelItem(item)
def move_right_button_onClicked(self):
root_item = self.delete_roi_window_keep_tree_view.invisibleRootItem()
for index in range(root_item.childCount()):
item = root_item.child(index)
if item in self.delete_roi_window_keep_tree_view.selectedItems():
# This will get ROI name
self.regions_of_interest_to_delete.append(item.text(0))
# Move to the right column list
self.delete_roi_window_delete_tree_view.clear()
self.delete_roi_window_delete_tree_view.setIndentation(0)
for roi in self.regions_of_interest_to_delete:
item = QTreeWidgetItem([roi])
self.delete_roi_window_delete_tree_view.addTopLevelItem(item)
self.delete_roi_window_confirm_button.setEnabled(True)
# Delete moved items from the left column list
self.regions_of_interest_to_keep = [
x for x in self.regions_of_interest_to_keep
if x not in self.regions_of_interest_to_delete
]
self.delete_roi_window_keep_tree_view.clear()
for index in self.regions_of_interest_to_keep:
item = QTreeWidgetItem([index])
self.delete_roi_window_keep_tree_view.addTopLevelItem(item)
def move_left_button_onClicked(self):
root_item = self.delete_roi_window_delete_tree_view.invisibleRootItem()
for index in range(root_item.childCount()):
item = root_item.child(index)
if item in self.delete_roi_window_delete_tree_view.selectedItems():
# This will get ROI name
self.regions_of_interest_to_keep.append(item.text(0))
# Move to the left column list
self.delete_roi_window_keep_tree_view.clear()
self.delete_roi_window_keep_tree_view.setIndentation(0)
for roi in self.regions_of_interest_to_keep:
item = QTreeWidgetItem([roi])
self.delete_roi_window_keep_tree_view.addTopLevelItem(item)
# Delete moved items from the right column list
self.regions_of_interest_to_delete = [
x for x in self.regions_of_interest_to_delete
if x not in self.regions_of_interest_to_keep
]
self.delete_roi_window_delete_tree_view.clear()
for index in self.regions_of_interest_to_delete:
item = QTreeWidgetItem([index])
self.delete_roi_window_delete_tree_view.addTopLevelItem(item)
if len(self.regions_of_interest_to_delete) == 0:
self.delete_roi_window_confirm_button.setEnabled(False)
def confirm_button_onClicked(self):
confirmation_dialog = QMessageBox.information(
self, 'Delete ROI(s)?',
'Region(s) of Interest in the To Delete table will be deleted. '
'Would you like to continue?', QMessageBox.Yes | QMessageBox.No)
if confirmation_dialog == QMessageBox.Yes:
progress_window = DeleteROIProgressWindow(
self, QtCore.Qt.WindowTitleHint)
progress_window.signal_roi_deleted.connect(self.on_rois_deleted)
progress_window.start_deleting(self.dataset_rtss,
self.regions_of_interest_to_delete)
progress_window.show()
def on_rois_deleted(self, new_rtss):
self.deleting_rois_structure_tuple.emit((new_rtss, {
"delete":
self.regions_of_interest_to_delete
}))
QMessageBox.about(self, "Saved",
"Regions of interest successfully deleted!")
self.close()
class UIImageFusionWindow(object):
image_fusion_info_initialized = QtCore.Signal(object)
def setup_ui(self, open_image_fusion_select_instance):
"""Sets up a UI"""
if platform.system() == 'Darwin':
self.stylesheet_path = "res/stylesheet.qss"
else:
self.stylesheet_path = "res/stylesheet-win-linux.qss"
window_icon = QIcon()
window_icon.addPixmap(QPixmap(resource_path("res/images/icon.ico")),
QIcon.Normal, QIcon.Off)
open_image_fusion_select_instance.setObjectName(
"OpenPatientWindowInstance")
open_image_fusion_select_instance.setWindowIcon(window_icon)
open_image_fusion_select_instance.resize(840, 530)
# Create a vertical box for containing the other elements and layouts
self.open_patient_window_instance_vertical_box = QVBoxLayout()
self.open_patient_window_instance_vertical_box.setObjectName(
"OpenPatientWindowInstanceVerticalBox")
# Create a label to prompt the user to enter the path to the
# directory that contains the DICOM files
self.open_patient_directory_prompt = QLabel()
self.open_patient_directory_prompt.setObjectName(
"OpenPatientDirectoryPrompt")
self.open_patient_directory_prompt.setAlignment(Qt.AlignLeft)
self.open_patient_window_instance_vertical_box.addWidget(
self.open_patient_directory_prompt)
# Create a horizontal box to hold the input box for the directory
# and the choose button
self.open_patient_directory_input_horizontal_box = QHBoxLayout()
self.open_patient_directory_input_horizontal_box.setObjectName(
"OpenPatientDirectoryInputHorizontalBox")
# Create a textbox to contain the path to the directory that contains
# the DICOM files
self.open_patient_directory_input_box = \
UIImageFusionWindowDragAndDropEvent(self)
self.open_patient_directory_input_box.setObjectName(
"OpenPatientDirectoryInputBox")
self.open_patient_directory_input_box.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.open_patient_directory_input_box.returnPressed.connect(
self.scan_directory_for_patient)
self.open_patient_directory_input_horizontal_box.addWidget(
self.open_patient_directory_input_box)
# Create a choose button to open the file dialog
self.open_patient_directory_choose_button = QPushButton()
self.open_patient_directory_choose_button.setObjectName(
"OpenPatientDirectoryChooseButton")
self.open_patient_directory_choose_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.open_patient_directory_choose_button.resize(
self.open_patient_directory_choose_button.sizeHint().width(),
self.open_patient_directory_input_box.height())
self.open_patient_directory_choose_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.open_patient_directory_input_horizontal_box.addWidget(
self.open_patient_directory_choose_button)
self.open_patient_directory_choose_button.clicked.connect(
self.choose_button_clicked)
# Create a widget to hold the input fields
self.open_patient_directory_input_widget = QWidget()
self.open_patient_directory_input_horizontal_box.setStretch(0, 4)
self.open_patient_directory_input_widget.setLayout(
self.open_patient_directory_input_horizontal_box)
self.open_patient_window_instance_vertical_box.addWidget(
self.open_patient_directory_input_widget)
# Create a horizontal box to hold the stop button and direction to
# the user on where to select the patient
self.open_patient_appear_prompt_and_stop_horizontal_box = QHBoxLayout()
self.open_patient_appear_prompt_and_stop_horizontal_box.setObjectName(
"OpenPatientAppearPromptAndStopHorizontalBox")
# Create a label to show direction on where the files will appear
self.open_patient_directory_appear_prompt = QLabel()
self.open_patient_directory_appear_prompt.setObjectName(
"OpenPatientDirectoryAppearPrompt")
self.open_patient_directory_appear_prompt.setAlignment(Qt.AlignLeft)
self.open_patient_appear_prompt_and_stop_horizontal_box.addWidget(
self.open_patient_directory_appear_prompt)
self.open_patient_appear_prompt_and_stop_horizontal_box.addStretch(1)
# Create a button to stop searching
self.open_patient_window_stop_button = QPushButton()
self.open_patient_window_stop_button.setObjectName(
"OpenPatientWindowStopButton")
self.open_patient_window_stop_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.open_patient_window_stop_button.resize(
self.open_patient_window_stop_button.sizeHint().width(),
self.open_patient_window_stop_button.sizeHint().height())
self.open_patient_window_stop_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.open_patient_window_stop_button.clicked.connect(
self.stop_button_clicked)
self.open_patient_window_stop_button.setProperty(
"QPushButtonClass", "fail-button")
self.open_patient_window_stop_button.setVisible(False)
self.open_patient_appear_prompt_and_stop_horizontal_box.addWidget(
self.open_patient_window_stop_button)
# Create a widget to hold the layout
self.open_patient_appear_prompt_and_stop_widget = QWidget()
self.open_patient_appear_prompt_and_stop_widget.setLayout(
self.open_patient_appear_prompt_and_stop_horizontal_box)
self.open_patient_window_instance_vertical_box.addWidget(
self.open_patient_appear_prompt_and_stop_widget)
# Create a tree view list to list out all patients in the directory
# selected above
self.open_patient_window_patients_tree = QTreeWidget()
self.open_patient_window_patients_tree.setObjectName(
"OpenPatientWindowPatientsTree")
self.open_patient_window_patients_tree.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.open_patient_window_patients_tree.resize(
self.open_patient_window_patients_tree.sizeHint().width(),
self.open_patient_window_patients_tree.sizeHint().height())
self.open_patient_window_patients_tree.setHeaderHidden(False)
self.open_patient_window_patients_tree.setHeaderLabels([""])
self.open_patient_window_patients_tree.itemChanged.connect(
self.tree_item_clicked)
self.open_patient_window_instance_vertical_box.addWidget(
self.open_patient_window_patients_tree)
self.last_patient = None
# Create a label to show what would happen if they select the patient
self.open_patient_directory_result_label = QtWidgets.QLabel()
self.open_patient_directory_result_label.setObjectName(
"OpenPatientDirectoryResultLabel")
self.open_patient_directory_result_label.setAlignment(Qt.AlignLeft)
self.open_patient_window_instance_vertical_box.addWidget(
self.open_patient_directory_result_label)
# Create a horizontal box to hold the Cancel and Open button
self.open_patient_window_patient_open_actions_horizontal_box = \
QHBoxLayout()
self.open_patient_window_patient_open_actions_horizontal_box. \
setObjectName("OpenPatientWindowPatientOpenActionsHorizontalBox")
self.open_patient_window_patient_open_actions_horizontal_box. \
addStretch(1)
# Add a button to go back/close from the application
self.open_patient_window_close_button = QPushButton()
self.open_patient_window_close_button.setObjectName(
"OpenPatientWindowcloseButton")
self.open_patient_window_close_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.open_patient_window_close_button.resize(
self.open_patient_window_stop_button.sizeHint().width(),
self.open_patient_window_stop_button.sizeHint().height())
self.open_patient_window_close_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.open_patient_window_close_button.clicked.connect(
self.close_button_clicked)
self.open_patient_window_close_button.setProperty(
"QPushButtonClass", "fail-button")
self.open_patient_window_patient_open_actions_horizontal_box. \
addWidget(self.open_patient_window_close_button)
# Add a button to confirm opening of the patient
self.open_patient_window_confirm_button = QPushButton()
self.open_patient_window_confirm_button.setObjectName(
"OpenPatientWindowConfirmButton")
self.open_patient_window_confirm_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.open_patient_window_confirm_button.resize(
self.open_patient_window_confirm_button.sizeHint().width(),
self.open_patient_window_confirm_button.sizeHint().height())
self.open_patient_window_confirm_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.open_patient_window_confirm_button.setDisabled(True)
self.open_patient_window_confirm_button.clicked.connect(
self.confirm_button_clicked)
self.open_patient_window_confirm_button.setProperty(
"QPushButtonClass", "success-button")
self.open_patient_window_patient_open_actions_horizontal_box. \
addWidget(
self.open_patient_window_confirm_button)
# Create a widget to house all of the actions button for open patient
# window
self.open_patient_window_patient_open_actions_widget = QWidget()
self.open_patient_window_patient_open_actions_widget.setLayout(
self.open_patient_window_patient_open_actions_horizontal_box)
self.open_patient_window_instance_vertical_box.addWidget(
self.open_patient_window_patient_open_actions_widget)
# Set the vertical box fourth element, the tree view, to stretch
# out as far as possible
self.open_patient_window_instance_vertical_box.setStretch(3, 4)
self.open_patient_window_instance_central_widget = QWidget()
self.open_patient_window_instance_central_widget.setObjectName(
"OpenPatientWindowInstanceCentralWidget")
self.open_patient_window_instance_central_widget.setLayout(
self.open_patient_window_instance_vertical_box)
# Create threadpool for multithreading
self.threadpool = QThreadPool()
# print("Multithreading with maximum %d threads" % self.threadpool.
# maxThreadCount())
# Create interrupt event for stopping the directory search
self.interrupt_flag = threading.Event()
# Bind all texts into the buttons and labels
self.retranslate_ui(open_image_fusion_select_instance)
# Set the central widget, ready for display
open_image_fusion_select_instance.setCentralWidget(
self.open_patient_window_instance_central_widget)
# Set the current stylesheet to the instance and connect it back
# to the caller through slot
_stylesheet = open(resource_path(self.stylesheet_path)).read()
open_image_fusion_select_instance.setStyleSheet(_stylesheet)
QtCore.QMetaObject.connectSlotsByName(
open_image_fusion_select_instance)
def retranslate_ui(self, open_image_fusion_select_instance):
"""Translates UI"""
_translate = QtCore.QCoreApplication.translate
open_image_fusion_select_instance.setWindowTitle(
_translate("OpenPatientWindowInstance",
"OnkoDICOM - Select Patient"))
self.open_patient_directory_prompt.setText(_translate(
"OpenPatientWindowInstance",
"Choose an image to merge with:"))
self.open_patient_directory_input_box.setPlaceholderText(
_translate("OpenPatientWindowInstance",
"Enter DICOM Files Path (For example, "
"C:\path\\to\your\DICOM\Files)"))
self.open_patient_directory_choose_button.setText(_translate(
"OpenPatientWindowInstance",
"Choose"))
self.open_patient_directory_appear_prompt.setText(_translate(
"OpenPatientWindowInstance",
"Please select below the image set you wish to overlay:"))
self.open_patient_directory_result_label. \
setText("The selected imageset(s) above will be "
"co-registered with the current imageset.")
self.open_patient_window_stop_button.setText(_translate(
"OpenPatientWindowInstance", "Stop Search"))
self.open_patient_window_close_button.setText(_translate(
"OpenPatientWindowInstance", "Close"))
self.open_patient_window_confirm_button.setText(_translate(
"OpenPatientWindowInstance", "Confirm"))
def update_patient(self):
self.clear_checked_leaves()
self.patient_dict_container = PatientDictContainer()
self.patient = self.patient_dict_container.get("basic_info")
self.patient_id = self.patient['id']
dataset = self.patient_dict_container.dataset[0]
self.patient_current_image_series_uid = \
dataset.get("SeriesInstanceUID")
def clear_checked_leaves(self):
"""
Resets all leaves to their unchecked state
"""
def recurse(parent_item: QTreeWidgetItem):
for i in range(parent_item.childCount()):
child = parent_item.child(i)
grand_children = child.childCount()
if grand_children > 0:
recurse(child)
else:
if child.checkState(0) == Qt.Checked:
child.setCheckState(0, Qt.CheckState.Unchecked)
child.setSelected(False)
recurse(self.open_patient_window_patients_tree.invisibleRootItem())
self.open_patient_window_patients_tree.collapseAll()
def close_button_clicked(self):
"""Closes the window."""
self.close()
def scan_directory_for_patient(self):
# Reset tree view header and last patient
self.open_patient_window_confirm_button.setDisabled(True)
self.open_patient_window_patients_tree.setHeaderLabels([""])
self.last_patient = None
self.filepath = self.open_patient_directory_input_box.text()
# Proceed if a folder was selected
if self.filepath != "":
# Update the QTreeWidget to reflect data being loaded
# First, clear the widget of any existing data
self.open_patient_window_patients_tree.clear()
# Next, update the tree widget
self.open_patient_window_patients_tree.addTopLevelItem(
QTreeWidgetItem(["Loading selected directory..."]))
# The choose button is disabled until the thread finishes executing
self.open_patient_directory_choose_button.setEnabled(False)
# Reveals the Stop Search button for the duration of the search
self.open_patient_window_stop_button.setVisible(True)
# The interrupt flag is then un-set if a previous search has been
# stopped.
self.interrupt_flag.clear()
# Then, create a new thread that will load the selected folder
worker = Worker(DICOMDirectorySearch.get_dicom_structure,
self.filepath,
self.interrupt_flag, progress_callback=True)
worker.signals.result.connect(self.on_search_complete)
worker.signals.progress.connect(self.search_progress)
# Execute the thread
self.threadpool.start(worker)
def choose_button_clicked(self):
"""
Executes when the choose button is clicked.
Gets filepath from the user and loads all files and subdirectories.
"""
# Get folder path from pop up dialog box
self.filepath = QtWidgets.QFileDialog.getExistingDirectory(
None, 'Select patient folder...', '')
self.open_patient_directory_input_box.setText(self.filepath)
self.scan_directory_for_patient()
def stop_button_clicked(self):
self.interrupt_flag.set()
def search_progress(self, progress_update):
"""
Current progress of the file search.
"""
self.open_patient_window_patients_tree.clear()
self.open_patient_window_patients_tree.addTopLevelItem(
QTreeWidgetItem(["Loading selected directory... "
"(%s files searched)" % progress_update]))
def on_search_complete(self, dicom_structure):
"""
Executes once the directory search is complete.
:param dicom_structure: DICOMStructure object constructed by the
directory search.
"""
self.open_patient_directory_choose_button.setEnabled(True)
self.open_patient_window_stop_button.setVisible(False)
self.open_patient_window_patients_tree.clear()
# dicom_structure will be None if function was interrupted.
if dicom_structure is None:
return
for patient_item in dicom_structure.get_tree_items_list():
self.open_patient_window_patients_tree.addTopLevelItem(
patient_item)
patient_item.setExpanded(True) # Display all studies
# Display all image sets
for i in range(patient_item.childCount()):
study = patient_item.child(i)
study.setExpanded(True)
if len(dicom_structure.patients) == 0:
QMessageBox.about(self, "No files found",
"Selected directory contains no DICOM files.")
def tree_item_clicked(self, item, _):
"""
Executes when a tree item is checked or unchecked.
If a different patient is checked, uncheck the previous patient.
Inform user about missing DICOM files.
"""
# If patient is only selected, but not checked, set it to "focus" to
# coincide with stylesheet. And if the selected item is an image set,
# display its child branches.
if item.checkState(0) == Qt.CheckState.Unchecked:
self.open_patient_window_patients_tree.setCurrentItem(item)
else: # Otherwise don't "focus", then set patient as selected
self.open_patient_window_patients_tree.setCurrentItem(None)
item.setSelected(True)
# Expand or collapse the tree branch if item is an image series
# Only collapse if the selected image series is expanded but unchecked
# Otherwise, expand its tree branch to show RT files
is_expanded = False \
if (item.isExpanded() is True and
item.checkState(0) == Qt.CheckState.Unchecked) else True
self.display_a_tree_branch(item, is_expanded)
selected_patient = item
# If the item is not top-level, bubble up to see which top-level item
# this item belongs to
if self.open_patient_window_patients_tree.invisibleRootItem(). \
indexOfChild(item) == -1:
while self.open_patient_window_patients_tree.invisibleRootItem(). \
indexOfChild(selected_patient) == -1:
selected_patient = selected_patient.parent()
# Uncheck previous patient if a different patient is selected
if item.checkState(0) == Qt.CheckState.Checked and self.last_patient \
!= selected_patient:
if self.last_patient is not None:
last_patient_checked_items = self.get_checked_nodes(
self.last_patient)
for checked_item in last_patient_checked_items:
checked_item.setCheckState(0, Qt.Unchecked)
self.last_patient = selected_patient
# Check selected items and display warning messages
self.check_selected_items(selected_patient)
def display_a_tree_branch(self, node, is_expanded):
# TO DO:
# Could Team 23 please update the defintion of this docstring as
# well as same function presented in OpenPatientWindow.
"""
Displays a tree branch
Parameters:
node : root node the tree
is_expanded (boolean): flag for checking if a particular
node/leaf is expanded.
"""
node.setExpanded(is_expanded)
if node.childCount() > 0:
for i in range(node.childCount()):
self.display_a_tree_branch(node.child(i), is_expanded)
else:
return
def check_selected_items(self, selected_patient):
"""
Check and display warning messages based on the existence and quantity
of image series, RTSTRUCT, RTPLAN, RTDOSE and SR files
Parameters:
selected_patient (DICOMStructure): DICOM Object of patient
"""
# Get the types of all selected leaves & Get the names of all selected
# studies
checked_nodes = self.get_checked_nodes(
self.open_patient_window_patients_tree.invisibleRootItem())
selected_series_types = [checked_node.dicom_object.get_series_type()
for checked_node in checked_nodes]
selected_series_id = [checked_node.dicom_object.series_uid
for checked_node in checked_nodes]
# Total number of selected image series
total_selected_image_series = selected_series_types.count('CT') + \
selected_series_types.count('MR') + \
selected_series_types.count('PT')
# Check the existence of IMAGE, RTSTRUCT, RTPLAN and RTDOSE files
proceed = True
if total_selected_image_series < 1:
header = "Cannot proceed without an image."
proceed = False
elif total_selected_image_series > 1:
header = "Cannot proceed with more than 1 selected image."
proceed = False
elif selected_patient.dicom_object.patient_id.strip() != \
self.patient_id:
header = "Cannot proceed with different patient."
proceed = False
elif self.patient_current_image_series_uid in selected_series_id:
header = "Cannot fuse with the same series."
proceed = False
elif not self.check_selected_items_referencing(checked_nodes):
# Check that selected items properly reference each other
header = "Selected series do not reference each other."
proceed = False
elif 'RTSTRUCT' not in selected_series_types and \
self.check_existing_rtss(checked_nodes):
header = "The associated RTSTRUCT must be selected."
proceed = False
elif 'RTDOSE' in selected_series_types:
header = "Cannot fuse with a RTDOSE file."
proceed = False
else:
header = ""
self.open_patient_window_confirm_button.setDisabled(not proceed)
# Set the tree header
self.open_patient_window_patients_tree.setHeaderLabel(header)
def check_selected_items_referencing(self, items):
"""
Check if selected tree items properly reference each other.
:param items: List of selected DICOMWidgetItems.
:return: True if the selected items belong to the same tree branch.
"""
# Dictionary of series of different file types
series = {
"IMAGE": None,
"RTSTRUCT": None,
"RTPLAN": None,
"RTDOSE": None,
"SR": None
}
for item in items:
series_type = item.dicom_object.get_series_type()
if series_type in series:
series[series_type] = item
else:
series["IMAGE"] = item
# Check if the RTSTRUCT, RTPLAN, and RTDOSE are a child item of the
# image series
if series["IMAGE"]:
if series["RTSTRUCT"] and series["RTSTRUCT"].parent() != \
series["IMAGE"]:
return False
if series["RTPLAN"] and \
series["RTPLAN"].parent().parent() != series["IMAGE"]:
return False
if series["SR"] and series["SR"].parent() != series["IMAGE"]:
return False
return True
def check_existing_rtss(self, items):
"""
Check for existing rtss
:return: bool, whether there is a rtss associated with the selected
image series
"""
image_series = ['CT', 'MR', 'PT']
for item in items:
if item.dicom_object.get_series_type() in image_series:
for i in range(item.childCount()):
if item.child(i).dicom_object:
return True
return False
def get_checked_nodes(self, root):
"""
:param root: QTreeWidgetItem as a root.
:return: A list of all QTreeWidgetItems in the QTreeWidget that are
checked under the root.
"""
checked_items = []
def recurse(parent_item: QTreeWidgetItem):
for i in range(parent_item.childCount()):
child = parent_item.child(i)
if int(child.flags()) & int(Qt.ItemIsUserCheckable) and \
child.checkState(0) == Qt.Checked:
checked_items.append(child)
grand_children = child.childCount()
if grand_children > 0:
recurse(child)
recurse(root)
return checked_items
def confirm_button_clicked(self):
"""
Begins loading of the selected files.
"""
selected_files = []
for item in self.get_checked_nodes(
self.open_patient_window_patients_tree.invisibleRootItem()):
selected_files += item.dicom_object.get_files()
self.progress_window = ImageFusionProgressWindow(self)
self.progress_window.signal_loaded.connect(self.on_loaded)
self.progress_window.signal_error.connect(self.on_loading_error)
self.progress_window.start_loading(selected_files)
def on_loaded(self, results):
"""
Executes when the progress bar finishes loaded the selected files.
"""
if results[0] is True: # Will be NoneType if loading was interrupted.
self.image_fusion_info_initialized.emit(results[1])
def on_loading_error(self, exception):
"""
Error handling for progress window.
"""
if type(exception[1]) == ImageLoading.NotRTSetError:
QMessageBox.about(self.progress_window,
"Unable to open selection",
"Selected files cannot be opened as they are not"
" a DICOM-RT set.")
self.progress_window.close()
elif type(exception[1]) == ImageLoading.NotAllowedClassError:
QMessageBox.about(self.progress_window,
"Unable to open selection",
"Selected files cannot be opened as they contain"
" unsupported DICOM classes.")
self.progress_window.close()
class UIDrawROIWindow:
def setup_ui(self, draw_roi_window_instance, rois, dataset_rtss,
signal_roi_drawn):
"""
this function is responsible for setting up the UI
for DrawROIWindow
param draw_roi_window_instance: the current drawing
window instance.
:param rois: the rois to be drawn
:param dataset_rtss: the rtss to be written to
:param signal_roi_drawn: the signal to be triggered
when roi is drawn
"""
self.patient_dict_container = PatientDictContainer()
self.rois = rois
self.dataset_rtss = dataset_rtss
self.signal_roi_drawn = signal_roi_drawn
self.drawn_roi_list = {}
self.standard_organ_names = []
self.standard_volume_names = []
self.standard_names = [] # Combination of organ and volume
self.ROI_name = None # Selected ROI name
self.target_pixel_coords = [] # This will contain the new pixel
# coordinates specified by the min and max
self.drawingROI = None
self.slice_changed = False
self.drawing_tool_radius = INITIAL_DRAWING_TOOL_RADIUS
self.keep_empty_pixel = False
# pixel density
self.target_pixel_coords_single_array = [] # 1D array
self.draw_roi_window_instance = draw_roi_window_instance
self.colour = None
self.ds = None
self.zoom = 1.0
self.upper_limit = None
self.lower_limit = None
# is_four_view is set to True to stop the SUV2ROI button from appearing
self.dicom_view = DicomAxialView(is_four_view=True)
self.current_slice = self.dicom_view.slider.value()
self.dicom_view.slider.valueChanged.connect(self.slider_value_changed)
self.init_layout()
QtCore.QMetaObject.connectSlotsByName(draw_roi_window_instance)
def retranslate_ui(self, draw_roi_window_instance):
"""
this function retranslate the ui for draw roi window
:param draw_roi_window_instance: the current drawing
window instance.
"""
_translate = QtCore.QCoreApplication.translate
draw_roi_window_instance.setWindowTitle(
_translate("DrawRoiWindowInstance",
"OnkoDICOM - Draw Region Of Interest"))
self.roi_name_label.setText(
_translate("ROINameLabel", "Region of Interest: "))
self.roi_name_line_edit.setText(_translate("ROINameLineEdit", ""))
self.image_slice_number_label.setText(
_translate("ImageSliceNumberLabel", "Slice Number: "))
self.image_slice_number_line_edit.setText(
_translate("ImageSliceNumberLineEdit",
str(self.dicom_view.current_slice_number)))
self.image_slice_number_transect_button.setText(
_translate("ImageSliceNumberTransectButton", "Transect"))
self.image_slice_number_box_draw_button.setText(
_translate("ImageSliceNumberBoxDrawButton", "Set Bounds"))
self.image_slice_number_draw_button.setText(
_translate("ImageSliceNumberDrawButton", "Draw"))
self.image_slice_number_move_forward_button.setText(
_translate("ImageSliceNumberMoveForwardButton", ""))
self.image_slice_number_move_backward_button.setText(
_translate("ImageSliceNumberMoveBackwardButton", ""))
self.draw_roi_window_instance_save_button.setText(
_translate("DrawRoiWindowInstanceSaveButton", "Save"))
self.draw_roi_window_instance_cancel_button.setText(
_translate("DrawRoiWindowInstanceCancelButton", "Cancel"))
self.internal_hole_max_label.setText(
_translate("InternalHoleLabel",
"Maximum internal hole size (pixels): "))
self.internal_hole_max_line_edit.setText(
_translate("InternalHoleInput", "9"))
self.isthmus_width_max_label.setText(
_translate("IsthmusWidthLabel",
"Maximum isthmus width size (pixels): "))
self.isthmus_width_max_line_edit.setText(
_translate("IsthmusWidthInput", "5"))
self.min_pixel_density_label.setText(
_translate("MinPixelDensityLabel", "Minimum density (pixels): "))
self.min_pixel_density_line_edit.setText(
_translate("MinPixelDensityInput", ""))
self.max_pixel_density_label.setText(
_translate("MaxPixelDensityLabel", "Maximum density (pixels): "))
self.max_pixel_density_line_edit.setText(
_translate("MaxPixelDensityInput", ""))
self.toggle_keep_empty_pixel_label.setText(
_translate("ToggleKeepEmptyPixelLabel", "Keep empty pixel: "))
self.draw_roi_window_viewport_zoom_label.setText(
_translate("DrawRoiWindowViewportZoomLabel", "Zoom: "))
self.draw_roi_window_cursor_radius_change_label.setText(
_translate("DrawRoiWindowCursorRadiusChangeLabel",
"Cursor Radius: "))
self.draw_roi_window_instance_action_reset_button.setText(
_translate("DrawRoiWindowInstanceActionClearButton", "Reset"))
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 DrawROIWindow
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.draw_roi_window_instance.setObjectName("DrawRoiWindowInstance")
self.draw_roi_window_instance.setWindowIcon(window_icon)
# Creating a form box to hold all buttons and input fields
self.draw_roi_window_input_container_box = QFormLayout()
self.draw_roi_window_input_container_box. \
setObjectName("DrawRoiWindowInputContainerBox")
self.draw_roi_window_input_container_box. \
setLabelAlignment(Qt.AlignLeft)
# Create a label for denoting the ROI name
self.roi_name_label = QLabel()
self.roi_name_label.setObjectName("ROINameLabel")
self.roi_name_line_edit = QLineEdit()
# Create an input box for ROI name
self.roi_name_line_edit.setObjectName("ROINameLineEdit")
self.roi_name_line_edit.setSizePolicy(QSizePolicy.Minimum,
QSizePolicy.Minimum)
self.roi_name_line_edit.resize(
self.roi_name_line_edit.sizeHint().width(),
self.roi_name_line_edit.sizeHint().height())
self.roi_name_line_edit.setEnabled(False)
self.draw_roi_window_input_container_box. \
addRow(self.roi_name_label, self.roi_name_line_edit)
# Create horizontal box to store image slice number and backward,
# forward buttons
self.image_slice_number_box = QHBoxLayout()
self.image_slice_number_box.setObjectName("ImageSliceNumberBox")
# Create a label for denoting the Image Slice Number
self.image_slice_number_label = QLabel()
self.image_slice_number_label.setObjectName("ImageSliceNumberLabel")
self.image_slice_number_box.addWidget(self.image_slice_number_label)
# Create a line edit for containing the image slice number
self.image_slice_number_line_edit = QLineEdit()
self.image_slice_number_line_edit. \
setObjectName("ImageSliceNumberLineEdit")
self.image_slice_number_line_edit. \
setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Minimum)
self.image_slice_number_line_edit.resize(
self.image_slice_number_line_edit.sizeHint().width(),
self.image_slice_number_line_edit.sizeHint().height())
self.image_slice_number_line_edit.setCursorPosition(0)
self.image_slice_number_line_edit.setEnabled(False)
self.image_slice_number_box. \
addWidget(self.image_slice_number_line_edit)
# Create a button to move backward to the previous image
self.image_slice_number_move_backward_button = QPushButton()
self.image_slice_number_move_backward_button. \
setObjectName("ImageSliceNumberMoveBackwardButton")
self.image_slice_number_move_backward_button.setSizePolicy(
QSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed))
self.image_slice_number_move_backward_button.resize(QSize(24, 24))
self.image_slice_number_move_backward_button.clicked. \
connect(self.onBackwardClicked)
icon_move_backward = QtGui.QIcon()
icon_move_backward.addPixmap(
QtGui.QPixmap(
resource_path('res/images/btn-icons/backward_slide_icon.png')))
self.image_slice_number_move_backward_button.setIcon(
icon_move_backward)
self.image_slice_number_box. \
addWidget(self.image_slice_number_move_backward_button)
# Create a button to move forward to the next image
self.image_slice_number_move_forward_button = QPushButton()
self.image_slice_number_move_forward_button. \
setObjectName("ImageSliceNumberMoveForwardButton")
self.image_slice_number_move_forward_button.setSizePolicy(
QSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed))
self.image_slice_number_move_forward_button.resize(QSize(24, 24))
self.image_slice_number_move_forward_button.clicked. \
connect(self.onForwardClicked)
icon_move_forward = QtGui.QIcon()
icon_move_forward.addPixmap(
QtGui.QPixmap(
resource_path('res/images/btn-icons/forward_slide_icon.png')))
self.image_slice_number_move_forward_button.setIcon(icon_move_forward)
self.image_slice_number_box. \
addWidget(self.image_slice_number_move_forward_button)
self.draw_roi_window_input_container_box. \
addRow(self.image_slice_number_box)
# Create a horizontal box for containing the zoom function
self.draw_roi_window_viewport_zoom_box = QHBoxLayout()
self.draw_roi_window_viewport_zoom_box.setObjectName(
"DrawRoiWindowViewportZoomBox")
# Create a label for zooming
self.draw_roi_window_viewport_zoom_label = QLabel()
self.draw_roi_window_viewport_zoom_label. \
setObjectName("DrawRoiWindowViewportZoomLabel")
# Create an input box for zoom factor
self.draw_roi_window_viewport_zoom_input = QLineEdit()
self.draw_roi_window_viewport_zoom_input. \
setObjectName("DrawRoiWindowViewportZoomInput")
self.draw_roi_window_viewport_zoom_input. \
setText("{:.2f}".format(self.zoom * 100) + "%")
self.draw_roi_window_viewport_zoom_input.setCursorPosition(0)
self.draw_roi_window_viewport_zoom_input.setEnabled(False)
self.draw_roi_window_viewport_zoom_input. \
setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Minimum)
self.draw_roi_window_viewport_zoom_input.resize(
self.draw_roi_window_viewport_zoom_input.sizeHint().width(),
self.draw_roi_window_viewport_zoom_input.sizeHint().height())
# Create 2 buttons for zooming in and out
# Zoom In Button
self.draw_roi_window_viewport_zoom_in_button = QPushButton()
self.draw_roi_window_viewport_zoom_in_button. \
setObjectName("DrawRoiWindowViewportZoomInButton")
self.draw_roi_window_viewport_zoom_in_button.setSizePolicy(
QSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed))
self.draw_roi_window_viewport_zoom_in_button.resize(QSize(24, 24))
self.draw_roi_window_viewport_zoom_in_button. \
setProperty("QPushButtonClass", "zoom-button")
icon_zoom_in = QtGui.QIcon()
icon_zoom_in.addPixmap(
QtGui.QPixmap(
resource_path('res/images/btn-icons/zoom_in_icon.png')))
self.draw_roi_window_viewport_zoom_in_button.setIcon(icon_zoom_in)
self.draw_roi_window_viewport_zoom_in_button.clicked. \
connect(self.onZoomInClicked)
# Zoom Out Button
self.draw_roi_window_viewport_zoom_out_button = QPushButton()
self.draw_roi_window_viewport_zoom_out_button. \
setObjectName("DrawRoiWindowViewportZoomOutButton")
self.draw_roi_window_viewport_zoom_out_button.setSizePolicy(
QSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed))
self.draw_roi_window_viewport_zoom_out_button.resize(QSize(24, 24))
self.draw_roi_window_viewport_zoom_out_button. \
setProperty("QPushButtonClass", "zoom-button")
icon_zoom_out = QtGui.QIcon()
icon_zoom_out.addPixmap(
QtGui.QPixmap(
resource_path('res/images/btn-icons/zoom_out_icon.png')))
self.draw_roi_window_viewport_zoom_out_button.setIcon(icon_zoom_out)
self.draw_roi_window_viewport_zoom_out_button.clicked. \
connect(self.onZoomOutClicked)
self.draw_roi_window_viewport_zoom_box. \
addWidget(self.draw_roi_window_viewport_zoom_label)
self.draw_roi_window_viewport_zoom_box. \
addWidget(self.draw_roi_window_viewport_zoom_input)
self.draw_roi_window_viewport_zoom_box. \
addWidget(self.draw_roi_window_viewport_zoom_out_button)
self.draw_roi_window_viewport_zoom_box. \
addWidget(self.draw_roi_window_viewport_zoom_in_button)
self.draw_roi_window_input_container_box. \
addRow(self.draw_roi_window_viewport_zoom_box)
self.init_cursor_radius_change_box()
# Create field to toggle two options: Keep empty pixel or fill empty
# pixel when using draw cursor
self.toggle_keep_empty_pixel_box = QHBoxLayout()
self.toggle_keep_empty_pixel_label = QLabel()
self.toggle_keep_empty_pixel_label. \
setObjectName("ToggleKeepEmptyPixelLabel")
# Create input for min pixel size
self.toggle_keep_empty_pixel_combo_box = QComboBox()
self.toggle_keep_empty_pixel_combo_box.addItems(["Off", "On"])
self.toggle_keep_empty_pixel_combo_box.setCurrentIndex(0)
self.toggle_keep_empty_pixel_combo_box.setEnabled(False)
self.toggle_keep_empty_pixel_combo_box. \
setObjectName("ToggleKeepEmptyPixelComboBox")
self.toggle_keep_empty_pixel_combo_box. \
setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Minimum)
self.toggle_keep_empty_pixel_combo_box.resize(
self.toggle_keep_empty_pixel_combo_box.sizeHint().width(),
self.toggle_keep_empty_pixel_combo_box.sizeHint().height())
self.toggle_keep_empty_pixel_combo_box.currentIndexChanged.connect(
self.toggle_keep_empty_pixel_box_index_changed)
self.toggle_keep_empty_pixel_box. \
addWidget(self.toggle_keep_empty_pixel_label)
self.toggle_keep_empty_pixel_box. \
addWidget(self.toggle_keep_empty_pixel_combo_box)
self.draw_roi_window_input_container_box. \
addRow(self.toggle_keep_empty_pixel_box)
# Create a horizontal box for transect and draw button
self.draw_roi_window_transect_draw_box = QHBoxLayout()
self.draw_roi_window_transect_draw_box. \
setObjectName("DrawRoiWindowTransectDrawBox")
# Create a transect button
self.image_slice_number_transect_button = QPushButton()
self.image_slice_number_transect_button. \
setObjectName("ImageSliceNumberTransectButton")
self.image_slice_number_transect_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum))
self.image_slice_number_transect_button.resize(
self.image_slice_number_transect_button.sizeHint().width(),
self.image_slice_number_transect_button.sizeHint().height())
self.image_slice_number_transect_button.clicked. \
connect(self.transect_handler)
icon_transect = QtGui.QIcon()
icon_transect.addPixmap(
QtGui.QPixmap(
resource_path('res/images/btn-icons/transect_icon.png')))
self.image_slice_number_transect_button.setIcon(icon_transect)
self.draw_roi_window_transect_draw_box. \
addWidget(self.image_slice_number_transect_button)
# Create a bounding box button
self.image_slice_number_box_draw_button = QPushButton()
self.image_slice_number_box_draw_button. \
setObjectName("ImageSliceNumberBoxDrawButton")
self.image_slice_number_box_draw_button.setSizePolicy(
QSizePolicy.MinimumExpanding, QSizePolicy.Minimum)
self.image_slice_number_box_draw_button.resize(
self.image_slice_number_box_draw_button.sizeHint().width(),
self.image_slice_number_box_draw_button.sizeHint().height())
self.image_slice_number_box_draw_button.clicked. \
connect(self.onBoxDrawClicked)
icon_box_draw = QtGui.QIcon()
icon_box_draw.addPixmap(
QtGui.QPixmap(
resource_path('res/images/btn-icons/draw_bound_icon.png')))
self.image_slice_number_box_draw_button.setIcon(icon_box_draw)
self.draw_roi_window_transect_draw_box. \
addWidget(self.image_slice_number_box_draw_button)
# Create a draw button
self.image_slice_number_draw_button = QPushButton()
self.image_slice_number_draw_button. \
setObjectName("ImageSliceNumberDrawButton")
self.image_slice_number_draw_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum))
self.image_slice_number_draw_button.resize(
self.image_slice_number_draw_button.sizeHint().width(),
self.image_slice_number_draw_button.sizeHint().height())
self.image_slice_number_draw_button.clicked.connect(self.onDrawClicked)
icon_draw = QtGui.QIcon()
icon_draw.addPixmap(
QtGui.QPixmap(resource_path('res/images/btn-icons/draw_icon.png')))
self.image_slice_number_draw_button.setIcon(icon_draw)
self.draw_roi_window_transect_draw_box. \
addWidget(self.image_slice_number_draw_button)
self.draw_roi_window_input_container_box. \
addRow(self.draw_roi_window_transect_draw_box)
# Create a contour preview button
self.row_preview_layout = QtWidgets.QHBoxLayout()
self.button_contour_preview = QtWidgets.QPushButton("Preview contour")
self.button_contour_preview.clicked.connect(self.onPreviewClicked)
self.row_preview_layout.addWidget(self.button_contour_preview)
self.draw_roi_window_input_container_box. \
addRow(self.row_preview_layout)
icon_preview = QtGui.QIcon()
icon_preview.addPixmap(
QtGui.QPixmap(
resource_path('res/images/btn-icons/preview_icon.png')))
self.button_contour_preview.setIcon(icon_preview)
# Create input line edit for alpha value
self.label_alpha_value = QtWidgets.QLabel("Alpha value:")
self.input_alpha_value = QtWidgets.QLineEdit("0.2")
self.input_alpha_value. \
setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum)
self.input_alpha_value.resize(
self.input_alpha_value.sizeHint().width(),
self.input_alpha_value.sizeHint().height())
self.input_alpha_value.setValidator(
QRegularExpressionValidator(
QRegularExpression("^[0-9]*[.]?[0-9]*$")))
self.draw_roi_window_input_container_box. \
addRow(self.label_alpha_value, self.input_alpha_value)
# Create a label for denoting the max internal hole size
self.internal_hole_max_label = QLabel()
self.internal_hole_max_label.setObjectName("InternalHoleLabel")
# Create input for max internal hole size
self.internal_hole_max_line_edit = QLineEdit()
self.internal_hole_max_line_edit.setObjectName("InternalHoleInput")
self.internal_hole_max_line_edit. \
setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum)
self.internal_hole_max_line_edit.resize(
self.internal_hole_max_line_edit.sizeHint().width(),
self.internal_hole_max_line_edit.sizeHint().height())
self.internal_hole_max_line_edit.setValidator(
QRegularExpressionValidator(
QRegularExpression("^[0-9]*[.]?[0-9]*$")))
self.draw_roi_window_input_container_box.addRow(
self.internal_hole_max_label, self.internal_hole_max_line_edit)
# Create a label for denoting the isthmus width size
self.isthmus_width_max_label = QLabel()
self.isthmus_width_max_label.setObjectName("IsthmusWidthLabel")
# Create input for max isthmus width size
self.isthmus_width_max_line_edit = QLineEdit()
self.isthmus_width_max_line_edit.setObjectName("IsthmusWidthInput")
self.isthmus_width_max_line_edit.setSizePolicy(
QSizePolicy.MinimumExpanding, QSizePolicy.Minimum)
self.isthmus_width_max_line_edit.resize(
self.isthmus_width_max_line_edit.sizeHint().width(),
self.isthmus_width_max_line_edit.sizeHint().height())
self.isthmus_width_max_line_edit.setValidator(
QRegularExpressionValidator(
QRegularExpression("^[0-9]*[.]?[0-9]*$")))
self.draw_roi_window_input_container_box.addRow(
self.isthmus_width_max_label, self.isthmus_width_max_line_edit)
# Create a label for denoting the minimum pixel density
self.min_pixel_density_label = QLabel()
self.min_pixel_density_label.setObjectName("MinPixelDensityLabel")
# Create input for min pixel size
self.min_pixel_density_line_edit = QLineEdit()
self.min_pixel_density_line_edit.setObjectName("MinPixelDensityInput")
self.min_pixel_density_line_edit.setSizePolicy(
QSizePolicy.MinimumExpanding, QSizePolicy.Minimum)
self.min_pixel_density_line_edit.resize(
self.min_pixel_density_line_edit.sizeHint().width(),
self.min_pixel_density_line_edit.sizeHint().height())
self.min_pixel_density_line_edit.setValidator(
QRegularExpressionValidator(
QRegularExpression("^[0-9]*[.]?[0-9]*$")))
self.draw_roi_window_input_container_box.addRow(
self.min_pixel_density_label, self.min_pixel_density_line_edit)
# Create a label for denoting the minimum pixel density
self.max_pixel_density_label = QLabel()
self.max_pixel_density_label.setObjectName("MaxPixelDensityLabel")
# Create input for min pixel size
self.max_pixel_density_line_edit = QLineEdit()
self.max_pixel_density_line_edit.setObjectName("MaxPixelDensityInput")
self.max_pixel_density_line_edit. \
setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum)
self.max_pixel_density_line_edit.resize(
self.max_pixel_density_line_edit.sizeHint().width(),
self.max_pixel_density_line_edit.sizeHint().height())
self.max_pixel_density_line_edit.setValidator(
QRegularExpressionValidator(
QRegularExpression("^[0-9]*[.]?[0-9]*$")))
self.draw_roi_window_input_container_box.addRow(
self.max_pixel_density_label, self.max_pixel_density_line_edit)
# Create a button to clear the draw
self.draw_roi_window_instance_action_reset_button = QPushButton()
self.draw_roi_window_instance_action_reset_button. \
setObjectName("DrawRoiWindowInstanceActionClearButton")
self.draw_roi_window_instance_action_reset_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum))
reset_button = self.draw_roi_window_instance_action_reset_button
self.draw_roi_window_instance_action_reset_button.resize(
reset_button.sizeHint().width(),
reset_button.sizeHint().height())
self.draw_roi_window_instance_action_reset_button.clicked. \
connect(self.onResetClicked)
self.draw_roi_window_instance_action_reset_button. \
setProperty("QPushButtonClass", "fail-button")
icon_clear_roi_draw = QtGui.QIcon()
icon_clear_roi_draw.addPixmap(
QtGui.QPixmap(
resource_path('res/images/btn-icons/reset_roi_draw_icon.png')))
self.draw_roi_window_instance_action_reset_button. \
setIcon(icon_clear_roi_draw)
self.draw_roi_window_input_container_box. \
addRow(self.draw_roi_window_instance_action_reset_button)
# Create a horizontal box for saving and cancel the drawing
self.draw_roi_window_cancel_save_box = QHBoxLayout()
self.draw_roi_window_cancel_save_box. \
setObjectName("DrawRoiWindowCancelSaveBox")
# Create an exit button to cancel the drawing
# Add a button to go back/exit from the application
self.draw_roi_window_instance_cancel_button = QPushButton()
self.draw_roi_window_instance_cancel_button. \
setObjectName("DrawRoiWindowInstanceCancelButton")
self.draw_roi_window_instance_cancel_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum))
self.draw_roi_window_instance_cancel_button.resize(
self.draw_roi_window_instance_cancel_button.sizeHint().width(),
self.draw_roi_window_instance_cancel_button.sizeHint().height())
self.draw_roi_window_instance_cancel_button. \
setCursor(QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.draw_roi_window_instance_cancel_button.clicked. \
connect(self.onCancelButtonClicked)
self.draw_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.draw_roi_window_instance_cancel_button.setIcon(icon_cancel)
self.draw_roi_window_cancel_save_box. \
addWidget(self.draw_roi_window_instance_cancel_button)
# Create a save button to save all the changes
self.draw_roi_window_instance_save_button = QPushButton()
self.draw_roi_window_instance_save_button. \
setObjectName("DrawRoiWindowInstanceSaveButton")
self.draw_roi_window_instance_save_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum))
self.draw_roi_window_instance_save_button.resize(
self.draw_roi_window_instance_save_button.sizeHint().width(),
self.draw_roi_window_instance_save_button.sizeHint().height())
self.draw_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.draw_roi_window_instance_save_button.setIcon(icon_save)
self.draw_roi_window_instance_save_button.clicked. \
connect(self.onSaveClicked)
self.draw_roi_window_cancel_save_box. \
addWidget(self.draw_roi_window_instance_save_button)
self.draw_roi_window_input_container_box. \
addRow(self.draw_roi_window_cancel_save_box)
# Creating a horizontal box to hold the ROI view and slider
self.draw_roi_window_instance_view_box = QHBoxLayout()
self.draw_roi_window_instance_view_box. \
setObjectName("DrawRoiWindowInstanceViewBox")
# Add View and Slider into horizontal box
self.draw_roi_window_instance_view_box.addWidget(self.dicom_view)
# Create a widget to hold the image slice box
self.draw_roi_window_instance_view_widget = QWidget()
self.draw_roi_window_instance_view_widget.setObjectName(
"DrawRoiWindowInstanceActionWidget")
self.draw_roi_window_instance_view_widget.setLayout(
self.draw_roi_window_instance_view_box)
# Create a horizontal box for containing the input fields and the
# viewport
self.draw_roi_window_main_box = QHBoxLayout()
self.draw_roi_window_main_box.setObjectName("DrawRoiWindowMainBox")
self.draw_roi_window_main_box. \
addLayout(self.draw_roi_window_input_container_box, 1)
self.draw_roi_window_main_box. \
addWidget(self.draw_roi_window_instance_view_widget, 11)
# Create a new central widget to hold the vertical box layout
self.draw_roi_window_instance_central_widget = QWidget()
self.draw_roi_window_instance_central_widget. \
setObjectName("DrawRoiWindowInstanceCentralWidget")
self.draw_roi_window_instance_central_widget.setLayout(
self.draw_roi_window_main_box)
self.retranslate_ui(self.draw_roi_window_instance)
self.draw_roi_window_instance.setStyleSheet(stylesheet)
self.draw_roi_window_instance. \
setCentralWidget(self.draw_roi_window_instance_central_widget)
QtCore.QMetaObject.connectSlotsByName(self.draw_roi_window_instance)
def slider_value_changed(self):
"""
actions to be taken when slider value changes
"""
image_slice_number = self.current_slice
# save progress
self.save_drawing_progress(image_slice_number)
self.set_current_slice(self.dicom_view.slider.value())
def set_current_slice(self, slice_number):
"""
set the current slice
:param slice_number: the slice number to be set
"""
self.image_slice_number_line_edit.setText(str(slice_number + 1))
self.current_slice = slice_number
self.dicom_view.update_view()
# check if this slice has any drawings before
if self.drawn_roi_list.get(self.current_slice) is not None:
self.drawingROI = self.drawn_roi_list[
self.current_slice]['drawingROI']
self.ds = self.drawn_roi_list[self.current_slice]['ds']
self.dicom_view.view.setScene(self.drawingROI)
self.enable_cursor_radius_change_box()
self.drawingROI.clear_cursor(self.drawing_tool_radius)
else:
self.disable_cursor_radius_change_box()
self.ds = None
def onZoomInClicked(self):
"""
This function is used for zooming in button
"""
self.dicom_view.zoom *= 1.05
self.dicom_view.update_view(zoom_change=True)
if self.drawingROI \
and self.drawingROI.current_slice == self.current_slice:
self.dicom_view.view.setScene(self.drawingROI)
self.draw_roi_window_viewport_zoom_input.setText(
"{:.2f}".format(self.dicom_view.zoom * 100) + "%")
self.draw_roi_window_viewport_zoom_input.setCursorPosition(0)
def onZoomOutClicked(self):
"""
This function is used for zooming out button
"""
self.dicom_view.zoom /= 1.05
self.dicom_view.update_view(zoom_change=True)
if self.drawingROI \
and self.drawingROI.current_slice == self.current_slice:
self.dicom_view.view.setScene(self.drawingROI)
self.draw_roi_window_viewport_zoom_input. \
setText("{:.2f}".format(self.dicom_view.zoom * 100) + "%")
self.draw_roi_window_viewport_zoom_input.setCursorPosition(0)
def toggle_keep_empty_pixel_box_index_changed(self):
self.keep_empty_pixel = self.toggle_keep_empty_pixel_combo_box. \
currentText() == "On"
self.drawingROI.keep_empty_pixel = self.keep_empty_pixel
def onCancelButtonClicked(self):
"""
This function is used for canceling the drawing
"""
self.closeWindow()
def onBackwardClicked(self):
"""
This function is used when backward button is clicked
"""
image_slice_number = self.current_slice
# save progress
if self.save_drawing_progress(image_slice_number):
# Backward will only execute if current image slice is above 0.
if int(image_slice_number) > 0:
# decrements slice by 1 and update slider to move to correct
# position
self.dicom_view.slider.setValue(image_slice_number - 1)
def onForwardClicked(self):
"""
This function is used when forward button is clicked
"""
image_slice_number = self.current_slice
# save progress
if self.save_drawing_progress(image_slice_number):
pixmaps = self.patient_dict_container.get("pixmaps_axial")
total_slices = len(pixmaps)
# Forward will only execute if current image slice is below the
# total number of slices.
if int(image_slice_number) < total_slices:
# increments slice by 1 and update slider to move to correct
# position
self.dicom_view.slider.setValue(image_slice_number + 1)
def onResetClicked(self):
"""
This function is used when reset button is clicked
"""
self.dicom_view.image_display()
self.dicom_view.update_view()
self.isthmus_width_max_line_edit.setText("5")
self.internal_hole_max_line_edit.setText("9")
self.min_pixel_density_line_edit.setText("")
self.max_pixel_density_line_edit.setText("")
if hasattr(self, 'bounds_box_draw'):
delattr(self, 'bounds_box_draw')
if hasattr(self, 'drawingROI'):
delattr(self, 'drawingROI')
self.ds = None
def transect_handler(self):
"""
Function triggered when the Transect button is pressed from the menu.
"""
pixmaps = self.patient_dict_container.get("pixmaps_axial")
id = self.current_slice
dt = self.patient_dict_container.dataset[id]
rowS = dt.PixelSpacing[0]
colS = dt.PixelSpacing[1]
dt.convert_pixel_data()
MainPageCallClass().run_transect(
self.draw_roi_window_instance,
self.dicom_view.view,
pixmaps[id],
dt._pixel_array.transpose(),
rowS,
colS,
is_roi_draw=True,
)
def save_drawing_progress(self, image_slice_number):
"""
this function saves the drawing progress on current slice
:param image_slice_number: the slice number to be saved
"""
if self.slice_changed:
if hasattr(self, 'drawingROI') and self.drawingROI \
and self.ds is not None \
and len(self.drawingROI.target_pixel_coords) != 0:
alpha = float(self.input_alpha_value.text())
pixel_hull_list = calculate_concave_hull_of_points(
self.drawingROI.target_pixel_coords, alpha)
coord_list = []
for pixel_hull in pixel_hull_list:
coord_list.append(pixel_hull)
self.drawn_roi_list[image_slice_number] = {
'coords': coord_list,
'ds': self.ds,
'drawingROI': self.drawingROI
}
self.slice_changed = False
return True
else:
return True
return True
def on_transect_close(self):
"""
Function triggered when transect is closed
"""
if self.upper_limit and self.lower_limit:
self.min_pixel_density_line_edit.setText(str(self.lower_limit))
self.max_pixel_density_line_edit.setText(str(self.upper_limit))
self.dicom_view.update_view()
def onDrawClicked(self):
"""
Function triggered when the Draw button is pressed from the menu.
"""
pixmaps = self.patient_dict_container.get("pixmaps_axial")
if self.min_pixel_density_line_edit.text() == "" \
or self.max_pixel_density_line_edit.text() == "":
QMessageBox.about(self.draw_roi_window_instance, "Not Enough Data",
"Not all values are specified or correct.")
else:
# Getting most updated selected slice
id = self.current_slice
dt = self.patient_dict_container.dataset[id]
dt.convert_pixel_data()
# Path to the selected .dcm file
location = self.patient_dict_container.filepaths[id]
self.ds = pydicom.dcmread(location)
min_pixel = self.min_pixel_density_line_edit.text()
max_pixel = self.max_pixel_density_line_edit.text()
# If they are number inputs
if min_pixel.isdecimal() and max_pixel.isdecimal():
min_pixel = int(min_pixel)
max_pixel = int(max_pixel)
if min_pixel >= max_pixel:
QMessageBox.about(
self.draw_roi_window_instance, "Incorrect Input",
"Please ensure maximum density is "
"atleast higher than minimum density.")
self.drawingROI = Drawing(
pixmaps[id], dt._pixel_array.transpose(), min_pixel,
max_pixel, self.patient_dict_container.dataset[id],
self.draw_roi_window_instance, self.slice_changed,
self.current_slice, self.drawing_tool_radius,
self.keep_empty_pixel, set())
self.slice_changed = True
self.dicom_view.view.setScene(self.drawingROI)
self.enable_cursor_radius_change_box()
else:
QMessageBox.about(self.draw_roi_window_instance,
"Not Enough Data",
"Not all values are specified or correct.")
def onBoxDrawClicked(self):
"""
Function triggered when bounding box button is pressed
"""
id = self.current_slice
dt = self.patient_dict_container.dataset[id]
dt.convert_pixel_data()
pixmaps = self.patient_dict_container.get("pixmaps_axial")
self.bounds_box_draw = DrawBoundingBox(pixmaps[id], dt)
self.dicom_view.view.setScene(self.bounds_box_draw)
self.disable_cursor_radius_change_box()
def onSaveClicked(self):
"""
Function triggered when Save button is clicked
"""
# Make sure the user has clicked Draw first
if self.save_drawing_progress(image_slice_number=self.current_slice):
self.saveROIList()
def saveROIList(self):
"""
Function triggered when saving ROI list
"""
roi_list = ROI.convert_hull_list_to_contours_data(
self.drawn_roi_list, self.patient_dict_container)
if len(roi_list) == 0:
QMessageBox.about(self.draw_roi_window_instance, "No ROI Detected",
"Please ensure you have drawn your ROI first.")
return
# The list of points will need to be converted into a
# single-dimensional array, as RTSTRUCT contour data is stored in
# such a way. i.e. [x, y, z, x, y, z, x, y, z, ..., ...] Create a
# popup window that modifies the RTSTRUCT and tells the user that
# processing is happening.
connectSaveROIProgress(self, roi_list, self.dataset_rtss,
self.ROI_name, self.roi_saved)
def roi_saved(self, new_rtss):
"""
Function to call save ROI and display progress
"""
self.signal_roi_drawn.emit((new_rtss, {"draw": self.ROI_name}))
QMessageBox.about(self.draw_roi_window_instance, "Saved",
"New contour successfully created!")
self.closeWindow()
def onPreviewClicked(self):
"""
function triggered when Preview button is clicked
"""
if hasattr(self, 'drawingROI') and self.drawingROI and len(
self.drawingROI.target_pixel_coords) > 0:
alpha = float(self.input_alpha_value.text())
polygon_list = calculate_concave_hull_of_points(
self.drawingROI.target_pixel_coords, alpha)
self.drawingROI.draw_contour_preview(polygon_list)
else:
QMessageBox.about(self.draw_roi_window_instance, "Not Enough Data",
"Please ensure you have drawn your ROI first.")
def set_selected_roi_name(self, roi_name):
"""
function to set selected roi name
:param roi_name: roi name selected
"""
roi_exists = False
patient_dict_container = PatientDictContainer()
existing_rois = patient_dict_container.get("rois")
number_of_rois = len(existing_rois)
# Check to see if the ROI already exists
for key, value in existing_rois.items():
if roi_name in value['name']:
roi_exists = True
if roi_exists:
QMessageBox.about(self.draw_roi_window_instance,
"ROI already exists in RTSS",
"Would you like to continue?")
self.ROI_name = roi_name
self.roi_name_line_edit.setText(self.ROI_name)
def onRadiusReduceClicked(self):
"""
function triggered when user reduce cursor radius
"""
self.drawing_tool_radius = max(self.drawing_tool_radius - 1, 4)
self.draw_roi_window_cursor_radius_change_input.setText(
str(self.drawing_tool_radius))
self.draw_roi_window_cursor_radius_change_input.setCursorPosition(0)
self.draw_cursor_when_radius_changed()
def onRadiusIncreaseClicked(self):
"""
function triggered when user increase cursor radius
"""
self.drawing_tool_radius = min(self.drawing_tool_radius + 1, 25)
self.draw_roi_window_cursor_radius_change_input.setText(
str(self.drawing_tool_radius))
self.draw_cursor_when_radius_changed()
def draw_cursor_when_radius_changed(self):
"""
function to update drawing cursor when radius changed
"""
if self.drawingROI.cursor:
self.drawingROI.draw_cursor(
self.drawingROI.current_cursor_x + self.drawing_tool_radius,
self.drawingROI.current_cursor_y + self.drawing_tool_radius,
self.drawing_tool_radius)
else:
self.drawingROI.draw_cursor(
(self.drawingROI.min_x + self.drawingROI.max_x) / 2,
(self.drawingROI.min_y + self.drawingROI.max_y) / 2,
self.drawing_tool_radius, True)
def init_cursor_radius_change_box(self):
"""
function to init cursor radius change box
"""
# Create a horizontal box for containing the cursor radius changing
# function
self.draw_roi_window_cursor_radius_change_box = QHBoxLayout()
self.draw_roi_window_cursor_radius_change_box.setObjectName(
"DrawRoiWindowCursorRadiusChangeBox")
# Create a label for cursor radius change
self.draw_roi_window_cursor_radius_change_label = QLabel()
self.draw_roi_window_cursor_radius_change_label.setObjectName(
"DrawRoiWindowCursorRadiusChangeLabel")
# Create an input box for cursor radius
self.draw_roi_window_cursor_radius_change_input = QLineEdit()
self.draw_roi_window_cursor_radius_change_input.setObjectName(
"DrawRoiWindowCursorRadiusChangeInput")
self.draw_roi_window_cursor_radius_change_input.setText(str(19))
self.draw_roi_window_cursor_radius_change_input.setCursorPosition(0)
self.draw_roi_window_cursor_radius_change_input.setEnabled(False)
self.draw_roi_window_cursor_radius_change_input.setSizePolicy(
QSizePolicy.Minimum, QSizePolicy.Minimum)
self.draw_roi_window_cursor_radius_change_input.resize(
self.draw_roi_window_cursor_radius_change_input.sizeHint().width(),
self.draw_roi_window_cursor_radius_change_input.sizeHint().height(
))
# Create 2 buttons for increasing and reducing cursor radius
# Increase Button
self.draw_roi_window_cursor_radius_change_increase_button = \
QPushButton()
self.draw_roi_window_cursor_radius_change_increase_button. \
setObjectName("DrawRoiWindowCursorRadiusIncreaseButton")
self.draw_roi_window_cursor_radius_change_increase_button. \
setSizePolicy(QSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed))
self.draw_roi_window_cursor_radius_change_increase_button.resize(
QSize(24, 24))
self.draw_roi_window_cursor_radius_change_increase_button.setProperty(
"QPushButtonClass", "zoom-button")
icon_zoom_in = QtGui.QIcon()
icon_zoom_in.addPixmap(
QtGui.QPixmap(
resource_path('res/images/btn-icons/zoom_in_icon.png')))
self.draw_roi_window_cursor_radius_change_increase_button.setIcon(
icon_zoom_in)
self.draw_roi_window_cursor_radius_change_increase_button.clicked. \
connect(self.onRadiusIncreaseClicked)
# Reduce Button
self.draw_roi_window_cursor_radius_change_reduce_button = QPushButton()
self.draw_roi_window_cursor_radius_change_reduce_button.setObjectName(
"DrawRoiWindowCursorRadiusReduceButton")
self.draw_roi_window_cursor_radius_change_reduce_button.setSizePolicy(
QSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed))
self.draw_roi_window_cursor_radius_change_reduce_button.resize(
QSize(24, 24))
self.draw_roi_window_cursor_radius_change_reduce_button.setProperty(
"QPushButtonClass", "zoom-button")
icon_zoom_out = QtGui.QIcon()
icon_zoom_out.addPixmap(
QtGui.QPixmap(
resource_path('res/images/btn-icons/zoom_out_icon.png')))
self.draw_roi_window_cursor_radius_change_reduce_button.setIcon(
icon_zoom_out)
self.draw_roi_window_cursor_radius_change_reduce_button.clicked. \
connect(self.onRadiusReduceClicked)
self.draw_roi_window_cursor_radius_change_box.addWidget(
self.draw_roi_window_cursor_radius_change_label)
self.draw_roi_window_cursor_radius_change_box.addWidget(
self.draw_roi_window_cursor_radius_change_input)
self.draw_roi_window_cursor_radius_change_box.addWidget(
self.draw_roi_window_cursor_radius_change_reduce_button)
self.draw_roi_window_cursor_radius_change_box.addWidget(
self.draw_roi_window_cursor_radius_change_increase_button)
self.draw_roi_window_input_container_box.addRow(
self.draw_roi_window_cursor_radius_change_box)
self.draw_roi_window_cursor_radius_change_increase_button.setEnabled(
False)
self.draw_roi_window_cursor_radius_change_reduce_button.setEnabled(
False)
def disable_cursor_radius_change_box(self):
"""
function to disable cursor radius change box
"""
self.draw_roi_window_cursor_radius_change_reduce_button.setEnabled(
False)
self.draw_roi_window_cursor_radius_change_increase_button.setEnabled(
False)
self.toggle_keep_empty_pixel_combo_box.setEnabled(False)
def enable_cursor_radius_change_box(self):
"""
function to enable cursor radius change box
"""
self.draw_roi_window_cursor_radius_change_reduce_button.setEnabled(
True)
self.draw_roi_window_cursor_radius_change_increase_button.setEnabled(
True)
self.toggle_keep_empty_pixel_combo_box.setEnabled(True)
def closeWindow(self):
"""
function to close draw roi window
"""
self.drawn_roi_list = {}
if hasattr(self, 'bounds_box_draw'):
delattr(self, 'bounds_box_draw')
if hasattr(self, 'drawingROI'):
delattr(self, 'drawingROI')
self.ds = None
self.close()
class UIFirstTimeWelcomeWindow(object):
configured = QtCore.Signal(str)
# the ui constructor function
def setup_ui(self, first_time_welcome_window_instance):
self.filepath = ""
if platform.system() == 'Darwin':
self.stylesheet_path = "res/stylesheet.qss"
else:
self.stylesheet_path = "res/stylesheet-win-linux.qss"
window_icon = QtGui.QIcon()
window_icon.addPixmap(
QtGui.QPixmap(resource_path("res/images/icon.ico")),
QtGui.QIcon.Normal, QtGui.QIcon.Off) # adding icon
first_time_welcome_window_instance.setObjectName(
"FirstTimeWelcomeWindowInstance")
first_time_welcome_window_instance.setWindowIcon(window_icon)
first_time_welcome_window_instance.setFixedSize(840, 530)
# Create a vertical layout to manage layout in a vertical manner
self.window_vertical_layout_box = QVBoxLayout()
self.window_vertical_layout_box.setObjectName(
"FirstTimeWelcomeWindowInstanceVerticalLayoutBox")
# Create a horizontal box to hold the Onko logo & welcome message
self.first_time_welcome_logo_message_horizontal_box = QHBoxLayout()
self.first_time_welcome_logo_message_horizontal_box.setObjectName(
"FirstTimeWelcomeLogoMessageBox")
# Set up the Logo Holder
self.logo_holder = QtWidgets.QHBoxLayout()
self.first_time_welcome_logo = QtWidgets.QLabel()
self.first_time_welcome_logo.setPixmap(
QtGui.QPixmap(resource_path("res/images/image.png")))
self.first_time_welcome_logo.setScaledContents(True)
self.first_time_welcome_logo.setObjectName(
"FirstTimeWelcomeWindowLogo")
self.first_time_welcome_logo.setFixedSize(240, 130)
self.logo_holder.addWidget(self.first_time_welcome_logo)
self.first_time_welcome_logo_message_horizontal_box.addLayout(
self.logo_holder)
# Create a vertical box to hold the welcome message
self.first_time_welcome_message_vertical_box = QVBoxLayout()
self.first_time_welcome_message_vertical_box.setObjectName(
"FirstTimeWelcomeWindowWelcomeMessage")
# Set up the Label
self.first_time_welcome_message_label = QtWidgets.QLabel()
self.first_time_welcome_message_label.setObjectName(
"FirstTimeWelcomeWindowLabel")
self.first_time_welcome_message_label.setAlignment(Qt.AlignLeft)
self.first_time_welcome_message_vertical_box.addWidget(
self.first_time_welcome_message_label)
# Set up the Slogan
self.first_time_welcome_message_slogan = QtWidgets.QLabel()
self.first_time_welcome_message_slogan.setObjectName(
"FirstTimeWelcomeWindowSlogan")
self.first_time_welcome_message_slogan.setAlignment(Qt.AlignLeft)
self.first_time_welcome_message_slogan.setWordWrap(True)
self.first_time_welcome_message_vertical_box.addWidget(
self.first_time_welcome_message_slogan)
self.first_time_welcome_message_slogan_widget = QWidget()
self.first_time_welcome_message_slogan_widget.setLayout(
self.first_time_welcome_message_vertical_box)
self.first_time_welcome_logo_message_horizontal_box.addWidget(
self.first_time_welcome_message_slogan_widget)
self.first_time_welcome_logo_message_widget = QWidget()
self.first_time_welcome_logo_message_widget.setLayout(
self.first_time_welcome_logo_message_horizontal_box)
self.window_vertical_layout_box.addWidget(
self.first_time_welcome_logo_message_widget)
# Create a label to prompt the user to enter the path to the default directory
self.first_time_welcome_default_dir_prompt = QtWidgets.QLabel()
self.first_time_welcome_default_dir_prompt.setObjectName(
"FirstTimeWelcomeWindowPrompt")
self.first_time_welcome_default_dir_prompt.setAlignment(Qt.AlignLeft)
self.window_vertical_layout_box.addWidget(
self.first_time_welcome_default_dir_prompt)
# Create a horizontal box to hold the input box for the directory and the choose button
self.first_time_welcome_input_horizontal_box = QHBoxLayout()
self.first_time_welcome_input_horizontal_box.setObjectName(
"FirstTimeWelcomeWindowInputHorizontalBox")
# Create a textbox to contain the path to the default directory
self.first_time_welcome_input_box = UIFirstTimeUserDragAndDropEvent(
self)
self.first_time_welcome_input_box.setObjectName(
"FirstTimeWelcomeWindowInputBox")
self.first_time_welcome_input_box.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.first_time_welcome_input_horizontal_box.addWidget(
self.first_time_welcome_input_box)
# Create a choose button to open the file dialog
self.first_time_welcome_choose_button = QPushButton()
self.first_time_welcome_choose_button.setObjectName(
"FirstTimeWelcomeWindowChooseButton")
self.first_time_welcome_choose_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.first_time_welcome_choose_button.resize(
self.first_time_welcome_choose_button.sizeHint().width(),
self.first_time_welcome_input_box.height())
self.first_time_welcome_choose_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.first_time_welcome_input_horizontal_box.addWidget(
self.first_time_welcome_choose_button)
self.first_time_welcome_choose_button.clicked.connect(
self.choose_button_clicked)
# Create a widget to hold the input field
self.first_time_welcome_input_widget = QWidget()
self.first_time_welcome_input_horizontal_box.setStretch(0, 4)
self.first_time_welcome_input_widget.setLayout(
self.first_time_welcome_input_horizontal_box)
self.window_vertical_layout_box.addWidget(
self.first_time_welcome_input_widget)
self.window_vertical_layout_box.addStretch(1)
# Create widgets for the clinical data CSV file path
self.clinical_data_csv_dir_label = QtWidgets.QLabel()
self.clinical_data_csv_dir_label.setObjectName("ClinicalDataCSVPrompt")
self.clinical_data_csv_dir_label.setAlignment(Qt.AlignLeft)
self.window_vertical_layout_box.addWidget(
self.clinical_data_csv_dir_label)
# Create a horizontal box to hold the input box for the
# directory and the choose button
self.clinical_data_csv_horizontal_box = QHBoxLayout()
self.clinical_data_csv_horizontal_box.setObjectName(
"ClinicalDataCSVHorizontalBox")
# Create a textbox to contain the path to the default directory
self.clinical_data_csv_input_box = \
UIFirstTimeUserDragAndDropEvent(self)
self.clinical_data_csv_input_box.setObjectName(
"ClinicalDataCSVInputBox")
self.clinical_data_csv_input_box.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.clinical_data_csv_horizontal_box.addWidget(
self.clinical_data_csv_input_box)
# Create a choose button to open the file dialog
self.clinical_data_csv_choose_button = QPushButton()
self.clinical_data_csv_choose_button.setObjectName(
"ClinicalDataCSVChooseButton")
self.clinical_data_csv_choose_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.clinical_data_csv_choose_button.resize(
self.clinical_data_csv_choose_button.sizeHint().width(),
self.clinical_data_csv_input_box.height())
self.clinical_data_csv_choose_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.clinical_data_csv_horizontal_box.addWidget(
self.clinical_data_csv_choose_button)
self.clinical_data_csv_choose_button.clicked.connect(
self.change_clinical_data_csv_button_clicked)
# Create a widget to hold the input field
self.clinical_data_input_widget = QWidget()
self.clinical_data_csv_horizontal_box.setStretch(0, 4)
self.clinical_data_input_widget.setLayout(
self.clinical_data_csv_horizontal_box)
self.window_vertical_layout_box.addWidget(
self.clinical_data_input_widget)
self.window_vertical_layout_box.addStretch(1)
# Create a horizontal box to hold the Skip and Confirm button
self.first_time_window_configure_actions_horizontal_box = QHBoxLayout()
self.first_time_window_configure_actions_horizontal_box.setObjectName(
"FirstTimeWelcomeWindowConfigureActionsHorizontalBox")
self.first_time_window_configure_actions_horizontal_box.addStretch(1)
# Add a button to skip the configuration setting
self.skip_button = QPushButton()
self.skip_button.setObjectName("SkipButton")
self.skip_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.skip_button.resize(self.skip_button.sizeHint().width(),
self.skip_button.sizeHint().height())
self.skip_button.setCursor(QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.skip_button.setProperty("QPushButtonClass", "fail-button")
self.first_time_window_configure_actions_horizontal_box.addWidget(
self.skip_button)
# Add a button to save the default directory
self.save_dir_button = QPushButton()
self.save_dir_button.setObjectName("SaveDirButton")
self.save_dir_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding))
self.save_dir_button.resize(self.save_dir_button.sizeHint().width(),
self.save_dir_button.sizeHint().height())
self.save_dir_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.save_dir_button.clicked.connect(self.save_options)
self.save_dir_button.setProperty("QPushButtonClass", "success-button")
self.first_time_window_configure_actions_horizontal_box.addWidget(
self.save_dir_button)
# Create a widget to house all of the actions button for first-time window
self.first_time_window_configure_actions_widget = QWidget()
self.first_time_window_configure_actions_widget.setLayout(
self.first_time_window_configure_actions_horizontal_box)
self.window_vertical_layout_box.addWidget(
self.first_time_window_configure_actions_widget)
self.first_time_window_instance_central_widget = QWidget()
self.first_time_window_instance_central_widget.setLayout(
self.window_vertical_layout_box)
first_time_welcome_window_instance.setCentralWidget(
self.first_time_window_instance_central_widget)
# Set the current stylesheet to the instance and connect it back to the caller through slot
_stylesheet = open(resource_path(self.stylesheet_path)).read()
first_time_welcome_window_instance.setStyleSheet(_stylesheet)
self.retranslate_ui(first_time_welcome_window_instance)
QtCore.QMetaObject.connectSlotsByName(
first_time_welcome_window_instance)
# this function inserts all the text in the first time page
def retranslate_ui(self, first_time_welcome_window_instance):
_translate = QtCore.QCoreApplication.translate
first_time_welcome_window_instance.setWindowTitle(
_translate("FirstTimeWelcomeWindowInstance",
"OnkoDICOM - First-time User Setting"))
self.first_time_welcome_message_label.setText(
_translate("FirstTimeWelcomeWindowInstance",
"Welcome to OnkoDICOM!"))
self.first_time_welcome_message_slogan.setText(
_translate(
"FirstTimeWelcomeWindowInstance",
"OnkoDICOM - the solution for producing data for analysis from your oncology plans and scans."
))
self.first_time_welcome_default_dir_prompt.setText(
_translate(
"FirstTimeWelcomeWindowInstance",
"Choose the path of the default directory containing all DICOM files:"
))
self.first_time_welcome_input_box.setPlaceholderText(
_translate(
"FirstTimeWelcomeWindowInstance",
"Enter DICOM Files Path (For example, C:\path\\to\your\DICOM\Files)"
))
self.first_time_welcome_choose_button.setText(
_translate("FirstTimeWelcomeWindowInstance", "Choose"))
# Clinical data CSV widgets
self.clinical_data_csv_dir_label.setText(
_translate(
"FirstTimeWelcomeWindowInstance",
"Choose the CSV file containing patient clinical data:"))
self.clinical_data_csv_input_box.setPlaceholderText(
_translate("FirstTimeWelcomeWindowInstance",
"Enter CSV File Path (for example, C:\CSV\\file.csv)"))
self.clinical_data_csv_choose_button.setText(
_translate("FirstTimeWelcomeWindowInstance", "Choose"))
self.save_dir_button.setText(
_translate("FirstTimeWelcomeWindowInstance", "Confirm"))
self.skip_button.setText(
_translate("FirstTimeWelcomeWindowInstance", "Skip"))
def save_options(self):
"""
Saves options selected in the first time welcome window.
"""
self.filepath = self.first_time_welcome_input_box.text()
self.csv_path = self.clinical_data_csv_input_box.text()
if self.filepath == "" and self.csv_path == "":
QMessageBox.about(self, "Unable to proceed",
"No directories selected.")
elif not os.path.exists(self.filepath) or not \
os.path.exists(self.csv_path):
QMessageBox.about(self, "Unable to proceed",
"Directories do not exist")
else:
config = Configuration()
try:
config.update_default_directory(self.filepath)
# Update CSV path if it exists
if self.csv_path != "" and os.path.exists(self.csv_path):
config.update_clinical_data_csv_dir(self.csv_path)
except SqlError:
config.set_up_config_db()
QMessageBox.critical(
self, "Config file error",
"Failed to access configuration file.\nPlease try again.")
else:
self.configured.emit(self.filepath)
def choose_button_clicked(self):
"""
Executes when the choose button is clicked.
Gets filepath from the user and loads all files and subdirectories.
"""
# Get folder path from pop up dialog box
self.filepath = QtWidgets.QFileDialog.getExistingDirectory(
None, 'Select default directory...', '')
self.first_time_welcome_input_box.setText(self.filepath)
def change_clinical_data_csv_button_clicked(self):
"""
Executes when the choose button is clicked.
Gets filepath from the user.
"""
# Get folder path from pop up dialog box
self.csv_path = QtWidgets.QFileDialog.getOpenFileName(
None, "Open Clinical Data File", "", "CSV data files (*.csv)")[0]
if len(self.csv_path) > 0:
self.clinical_data_csv_input_box.setText(self.csv_path)
class UIManipulateROIWindow:
def setup_ui(self, manipulate_roi_window_instance, rois, dataset_rtss,
roi_color, signal_roi_manipulated):
self.patient_dict_container = PatientDictContainer()
self.rois = rois
self.dataset_rtss = dataset_rtss
self.signal_roi_manipulated = signal_roi_manipulated
self.roi_color = roi_color
self.roi_names = [] # Names of selected ROIs
self.all_roi_names = [] # Names of all existing ROIs
for roi_id, roi_dict in self.rois.items():
self.all_roi_names.append(roi_dict['name'])
# Operation names
self.single_roi_operation_names = [
"Expand", "Contract", "Inner Rind (annulus)",
"Outer Rind (annulus)"
]
self.multiple_roi_operation_names = [
"Union", "Intersection", "Difference"
]
self.operation_names = self.multiple_roi_operation_names + \
self.single_roi_operation_names
self.new_ROI_contours = None
self.manipulate_roi_window_instance = manipulate_roi_window_instance
self.dicom_view = DicomAxialView(metadata_formatted=True,
is_four_view=True)
self.dicom_preview = DicomAxialView(metadata_formatted=True,
is_four_view=True)
self.dicom_view.slider.valueChanged.connect(
self.dicom_view_slider_value_changed)
self.dicom_preview.slider.valueChanged.connect(
self.dicom_preview_slider_value_changed)
self.init_layout()
QtCore.QMetaObject.connectSlotsByName(manipulate_roi_window_instance)
def retranslate_ui(self, manipulate_roi_window_instance):
_translate = QtCore.QCoreApplication.translate
manipulate_roi_window_instance.setWindowTitle(
_translate("ManipulateRoiWindowInstance",
"OnkoDICOM - Draw Region Of Interest"))
self.first_roi_name_label.setText(
_translate("FirstROINameLabel", "ROI 1: "))
self.first_roi_name_dropdown_list.setPlaceholderText("ROI 1")
self.first_roi_name_dropdown_list.addItems(self.all_roi_names)
self.operation_name_label.setText(
_translate("OperationNameLabel", "Operation"))
self.operation_name_dropdown_list.setPlaceholderText("Operation")
self.operation_name_dropdown_list.addItems(self.operation_names)
self.second_roi_name_label.setText(
_translate("SecondROINameLabel", "ROI 2: "))
self.second_roi_name_dropdown_list.setPlaceholderText("ROI 2")
self.second_roi_name_dropdown_list.addItems(self.all_roi_names)
self.manipulate_roi_window_instance_draw_button.setText(
_translate("ManipulateRoiWindowInstanceDrawButton", "Draw"))
self.manipulate_roi_window_instance_save_button.setText(
_translate("ManipulateRoiWindowInstanceSaveButton", "Save"))
self.manipulate_roi_window_instance_cancel_button.setText(
_translate("ManipulateRoiWindowInstanceCancelButton", "Cancel"))
self.margin_label.setText(_translate("MarginLabel", "Margin (mm): "))
self.new_roi_name_label.setText(
_translate("NewROINameLabel", "New ROI Name"))
self.ROI_view_box_label.setText("ROI")
self.preview_box_label.setText("Preview")
def init_layout(self):
"""
Initialize the layout for the DICOM View tab.
Add the view widget and the slider in the layout.
Add the whole container 'tab2_view' as a tab in the main page.
"""
# Initialise a ManipulateROIWindow
if platform.system() == 'Darwin':
self.stylesheet_path = "res/stylesheet.qss"
else:
self.stylesheet_path = "res/stylesheet-win-linux.qss"
stylesheet = open(resource_path(self.stylesheet_path)).read()
window_icon = QIcon()
window_icon.addPixmap(QPixmap(resource_path("res/images/icon.ico")),
QIcon.Normal, QIcon.Off)
self.manipulate_roi_window_instance.setObjectName(
"ManipulateRoiWindowInstance")
self.manipulate_roi_window_instance.setWindowIcon(window_icon)
# Creating a form box to hold all buttons and input fields
self.manipulate_roi_window_input_container_box = QFormLayout()
self.manipulate_roi_window_input_container_box.setObjectName(
"ManipulateRoiWindowInputContainerBox")
self.manipulate_roi_window_input_container_box.setLabelAlignment(
Qt.AlignLeft)
# Create a label for denoting the first ROI name
self.first_roi_name_label = QLabel()
self.first_roi_name_label.setObjectName("FirstROINameLabel")
self.first_roi_name_dropdown_list = QComboBox()
# Create an dropdown list for ROI name
self.first_roi_name_dropdown_list.setObjectName(
"FirstROINameDropdownList")
self.first_roi_name_dropdown_list.setSizePolicy(
QSizePolicy.Minimum, QSizePolicy.Minimum)
self.first_roi_name_dropdown_list.resize(
self.first_roi_name_dropdown_list.sizeHint().width(),
self.first_roi_name_dropdown_list.sizeHint().height())
self.first_roi_name_dropdown_list.activated.connect(
self.update_selected_rois)
self.manipulate_roi_window_input_container_box.addRow(
self.first_roi_name_label, self.first_roi_name_dropdown_list)
# Create a label for denoting the operation
self.operation_name_label = QLabel()
self.operation_name_label.setObjectName("OperationNameLabel")
self.operation_name_dropdown_list = QComboBox()
# Create an dropdown list for operation name
self.operation_name_dropdown_list.setObjectName(
"OperationNameDropdownList")
self.operation_name_dropdown_list.setSizePolicy(
QSizePolicy.Minimum, QSizePolicy.Minimum)
self.operation_name_dropdown_list.resize(
self.operation_name_dropdown_list.sizeHint().width(),
self.operation_name_dropdown_list.sizeHint().height())
self.operation_name_dropdown_list.activated.connect(
self.operation_changed)
self.manipulate_roi_window_input_container_box.addRow(
self.operation_name_label, self.operation_name_dropdown_list)
# Create a label for denoting the second ROI name
self.second_roi_name_label = QLabel()
self.second_roi_name_label.setObjectName("SecondROINameLabel")
self.second_roi_name_label.setVisible(False)
self.second_roi_name_dropdown_list = QComboBox()
# Create an dropdown list for ROI name
self.second_roi_name_dropdown_list.setObjectName(
"SecondROINameDropdownList")
self.second_roi_name_dropdown_list.setSizePolicy(
QSizePolicy.Minimum, QSizePolicy.Minimum)
self.second_roi_name_dropdown_list.resize(
self.second_roi_name_dropdown_list.sizeHint().width(),
self.second_roi_name_dropdown_list.sizeHint().height())
self.second_roi_name_dropdown_list.setVisible(False)
self.second_roi_name_dropdown_list.activated.connect(
self.update_selected_rois)
self.manipulate_roi_window_input_container_box.addRow(
self.second_roi_name_label, self.second_roi_name_dropdown_list)
# Create a label for denoting the margin
self.margin_label = QLabel()
self.margin_label.setObjectName("MarginLabel")
self.margin_label.setVisible(False)
# Create input for the new ROI name
self.margin_line_edit = QLineEdit()
self.margin_line_edit.setObjectName("MarginInput")
self.margin_line_edit.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.Minimum)
self.margin_line_edit.resize(self.margin_line_edit.sizeHint().width(),
self.margin_line_edit.sizeHint().height())
self.margin_line_edit.setVisible(False)
self.margin_line_edit.setValidator(
QRegularExpressionValidator(
QRegularExpression("^[0-9]*[.]?[0-9]*$")))
self.manipulate_roi_window_input_container_box.addRow(
self.margin_label, self.margin_line_edit)
# Create a label for denoting the new ROI name
self.new_roi_name_label = QLabel()
self.new_roi_name_label.setObjectName("NewROINameLabel")
# Create input for the new ROI name
self.new_roi_name_line_edit = QLineEdit()
self.new_roi_name_line_edit.setObjectName("NewROINameInput")
self.new_roi_name_line_edit.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.Minimum)
self.new_roi_name_line_edit.resize(
self.new_roi_name_line_edit.sizeHint().width(),
self.new_roi_name_line_edit.sizeHint().height())
self.manipulate_roi_window_input_container_box.addRow(
self.new_roi_name_label, self.new_roi_name_line_edit)
# Create a spacer between inputs and buttons
spacer = QWidget()
spacer.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
spacer.setFocusPolicy(Qt.NoFocus)
self.manipulate_roi_window_input_container_box.addRow(spacer)
# Create a warning message when missing inputs
self.warning_message = QWidget()
self.warning_message.setContentsMargins(8, 5, 8, 5)
warning_message_layout = QHBoxLayout()
warning_message_layout.setAlignment(QtCore.Qt.AlignLeft
| QtCore.Qt.AlignLeft)
warning_message_icon = QLabel()
warning_message_icon.setPixmap(
QtGui.QPixmap(
resource_path("res/images/btn-icons/alert_icon.png")))
warning_message_layout.addWidget(warning_message_icon)
self.warning_message_text = QLabel()
self.warning_message_text.setStyleSheet("color: red")
warning_message_layout.addWidget(self.warning_message_text)
self.warning_message.setLayout(warning_message_layout)
self.warning_message.setVisible(False)
self.manipulate_roi_window_input_container_box.addRow(
self.warning_message)
# Create a draw button
self.manipulate_roi_window_instance_draw_button = QPushButton()
self.manipulate_roi_window_instance_draw_button.setObjectName(
"ManipulateRoiWindowInstanceDrawButton")
self.manipulate_roi_window_instance_draw_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum))
self.manipulate_roi_window_instance_draw_button.resize(
self.manipulate_roi_window_instance_draw_button.sizeHint().width(),
self.manipulate_roi_window_instance_draw_button.sizeHint().height(
))
self.manipulate_roi_window_instance_draw_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.manipulate_roi_window_instance_draw_button.clicked.connect(
self.onDrawButtonClicked)
self.manipulate_roi_window_input_container_box.addRow(
self.manipulate_roi_window_instance_draw_button)
# Create a horizontal box for saving and cancel the drawing
self.manipulate_roi_window_cancel_save_box = QHBoxLayout()
self.manipulate_roi_window_cancel_save_box.setObjectName(
"ManipulateRoiWindowCancelSaveBox")
# Create an exit button to cancel the drawing
# Add a button to go back/exit from the application
self.manipulate_roi_window_instance_cancel_button = QPushButton()
self.manipulate_roi_window_instance_cancel_button.setObjectName(
"ManipulateRoiWindowInstanceCancelButton")
self.manipulate_roi_window_instance_cancel_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum))
self.manipulate_roi_window_instance_cancel_button.resize(
self.manipulate_roi_window_instance_cancel_button.sizeHint().width(
),
self.manipulate_roi_window_instance_cancel_button.sizeHint().
height())
self.manipulate_roi_window_instance_cancel_button.setCursor(
QtGui.QCursor(QtCore.Qt.PointingHandCursor))
self.manipulate_roi_window_instance_cancel_button.clicked.connect(
self.onCancelButtonClicked)
self.manipulate_roi_window_instance_cancel_button.setProperty(
"QPushButtonClass", "fail-button")
icon_cancel = QtGui.QIcon()
icon_cancel.addPixmap(
QtGui.QPixmap(
resource_path('res/images/btn-icons/cancel_icon.png')))
self.manipulate_roi_window_instance_cancel_button.setIcon(icon_cancel)
self.manipulate_roi_window_cancel_save_box.addWidget(
self.manipulate_roi_window_instance_cancel_button)
# Create a save button to save all the changes
self.manipulate_roi_window_instance_save_button = QPushButton()
self.manipulate_roi_window_instance_save_button.setObjectName(
"ManipulateRoiWindowInstanceSaveButton")
self.manipulate_roi_window_instance_save_button.setSizePolicy(
QSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Minimum))
self.manipulate_roi_window_instance_save_button.resize(
self.manipulate_roi_window_instance_save_button.sizeHint().width(),
self.manipulate_roi_window_instance_save_button.sizeHint().height(
))
self.manipulate_roi_window_instance_save_button.setProperty(
"QPushButtonClass", "success-button")
icon_save = QtGui.QIcon()
icon_save.addPixmap(
QtGui.QPixmap(resource_path('res/images/btn-icons/save_icon.png')))
self.manipulate_roi_window_instance_save_button.setIcon(icon_save)
self.manipulate_roi_window_instance_save_button.clicked.connect(
self.onSaveClicked)
self.manipulate_roi_window_cancel_save_box.addWidget(
self.manipulate_roi_window_instance_save_button)
self.manipulate_roi_window_input_container_box.addRow(
self.manipulate_roi_window_cancel_save_box)
# Creating a horizontal box to hold the ROI view and the preview
self.manipulate_roi_window_instance_view_box = QHBoxLayout()
self.manipulate_roi_window_instance_view_box.setObjectName(
"ManipulateRoiWindowInstanceViewBoxes")
# Font for the ROI view and preview's labels
font = QFont()
font.setBold(True)
font.setPixelSize(20)
# Creating the ROI view
self.ROI_view_box_layout = QVBoxLayout()
self.ROI_view_box_label = QLabel()
self.ROI_view_box_label.setFont(font)
self.ROI_view_box_label.setAlignment(Qt.AlignHCenter)
self.ROI_view_box_layout.addWidget(self.ROI_view_box_label)
self.ROI_view_box_layout.addWidget(self.dicom_view)
self.ROI_view_box_widget = QWidget()
self.ROI_view_box_widget.setLayout(self.ROI_view_box_layout)
# Creating the preview
self.preview_box_layout = QVBoxLayout()
self.preview_box_label = QLabel()
self.preview_box_label.setFont(font)
self.preview_box_label.setAlignment(Qt.AlignHCenter)
self.preview_box_layout.addWidget(self.preview_box_label)
self.preview_box_layout.addWidget(self.dicom_preview)
self.preview_box_widget = QWidget()
self.preview_box_widget.setLayout(self.preview_box_layout)
# Add View and Slider into horizontal box
self.manipulate_roi_window_instance_view_box.addWidget(
self.ROI_view_box_widget)
self.manipulate_roi_window_instance_view_box.addWidget(
self.preview_box_widget)
# Create a widget to hold the image slice box
self.manipulate_roi_window_instance_view_widget = QWidget()
self.manipulate_roi_window_instance_view_widget.setObjectName(
"ManipulateRoiWindowInstanceActionWidget")
self.manipulate_roi_window_instance_view_widget.setLayout(
self.manipulate_roi_window_instance_view_box)
# Create a horizontal box for containing the input fields and the
# viewports
self.manipulate_roi_window_main_box = QHBoxLayout()
self.manipulate_roi_window_main_box.setObjectName(
"ManipulateRoiWindowMainBox")
self.manipulate_roi_window_main_box.addLayout(
self.manipulate_roi_window_input_container_box, 1)
self.manipulate_roi_window_main_box.addWidget(
self.manipulate_roi_window_instance_view_widget, 11)
# Create a new central widget to hold the horizontal box layout
self.manipulate_roi_window_instance_central_widget = QWidget()
self.manipulate_roi_window_instance_central_widget.setObjectName(
"ManipulateRoiWindowInstanceCentralWidget")
self.manipulate_roi_window_instance_central_widget.setLayout(
self.manipulate_roi_window_main_box)
self.retranslate_ui(self.manipulate_roi_window_instance)
self.manipulate_roi_window_instance.setStyleSheet(stylesheet)
self.manipulate_roi_window_instance.setCentralWidget(
self.manipulate_roi_window_instance_central_widget)
QtCore.QMetaObject.connectSlotsByName(
self.manipulate_roi_window_instance)
def dicom_view_slider_value_changed(self):
"""
Display selected ROIs in dropbox when moving to another image slice
"""
self.display_selected_roi()
if self.dicom_preview.slider.value() != self.dicom_view.slider.value():
self.dicom_preview.slider.setValue(self.dicom_view.slider.value())
def dicom_preview_slider_value_changed(self):
"""
Display generated ROI when moving to another image slice
"""
self.draw_roi()
if self.dicom_preview.slider.value() != self.dicom_view.slider.value():
self.dicom_view.slider.setValue(self.dicom_preview.slider.value())
def onCancelButtonClicked(self):
"""
This function is used for canceling the drawing
"""
self.close()
def onDrawButtonClicked(self):
"""
Function triggered when the Draw button is pressed from the menu.
"""
# Hide warning message
self.warning_message.setVisible(False)
# Check inputs
selected_operation = self.operation_name_dropdown_list.currentText()
roi_1 = self.first_roi_name_dropdown_list.currentText()
roi_2 = self.second_roi_name_dropdown_list.currentText()
new_roi_name = self.new_roi_name_line_edit.text()
# Check the selected inputs and execute the operations
if roi_1 != "" and new_roi_name != "" and \
self.margin_line_edit.text() != "" and \
selected_operation in self.single_roi_operation_names:
# Single ROI operations
dict_rois_contours = ROI.get_roi_contour_pixel(
self.patient_dict_container.get("raw_contour"), [roi_1],
self.patient_dict_container.get("pixluts"))
roi_geometry = ROI.roi_to_geometry(dict_rois_contours[roi_1])
margin = float(self.margin_line_edit.text())
if selected_operation == self.single_roi_operation_names[0]:
new_geometry = ROI.scale_roi(roi_geometry, margin)
elif selected_operation == self.single_roi_operation_names[1]:
new_geometry = ROI.scale_roi(roi_geometry, -margin)
elif selected_operation == self.single_roi_operation_names[2]:
new_geometry = ROI.rind_roi(roi_geometry, -margin)
else:
new_geometry = ROI.rind_roi(roi_geometry, margin)
self.new_ROI_contours = ROI.geometry_to_roi(new_geometry)
self.draw_roi()
return True
elif roi_1 != "" and roi_2 != "" and new_roi_name != "" and \
selected_operation in self.multiple_roi_operation_names:
# Multiple ROI operations
dict_rois_contours = ROI.get_roi_contour_pixel(
self.patient_dict_container.get("raw_contour"), [roi_1, roi_2],
self.patient_dict_container.get("pixluts"))
roi_1_geometry = ROI.roi_to_geometry(dict_rois_contours[roi_1])
roi_2_geometry = ROI.roi_to_geometry(dict_rois_contours[roi_2])
# Execute the selected operation
new_geometry = ROI.manipulate_rois(roi_1_geometry, roi_2_geometry,
selected_operation.upper())
self.new_ROI_contours = ROI.geometry_to_roi(new_geometry)
self.draw_roi()
return True
self.warning_message_text.setText("Not all values are specified.")
self.warning_message.setVisible(True)
return False
def onSaveClicked(self):
""" Save the new ROI """
# Get the name of the new ROI
new_roi_name = self.new_roi_name_line_edit.text()
# If the new ROI hasn't been drawn, draw the new ROI. Then if the new
# ROI is drawn successfully, proceed to save the new ROI.
if self.new_ROI_contours is None:
if not self.onDrawButtonClicked():
return
# Get a dict to convert SOPInstanceUID to slice id
slice_ids_dict = get_dict_slice_to_uid(PatientDictContainer())
# Transform new_ROI_contours to a list of roi information
rois_to_save = {}
for uid, contour_sequence in self.new_ROI_contours.items():
slider_id = slice_ids_dict[uid]
location = self.patient_dict_container.filepaths[slider_id]
ds = pydicom.dcmread(location)
slice_info = {'coords': contour_sequence, 'ds': ds}
rois_to_save[slider_id] = slice_info
roi_list = ROI.convert_hull_list_to_contours_data(
rois_to_save, self.patient_dict_container)
connectSaveROIProgress(self, roi_list, self.dataset_rtss, new_roi_name,
self.roi_saved)
def draw_roi(self):
""" Draw the new ROI """
# Get the new ROI's name
new_roi_name = self.new_roi_name_line_edit.text()
# Check if the new ROI contour is None
if self.new_ROI_contours is None:
return
# Get the info required to draw the new ROI
slider_id = self.dicom_preview.slider.value()
curr_slice = self.patient_dict_container.get("dict_uid")[slider_id]
# Calculate the new ROI's polygon
dict_ROI_contours = {}
dict_ROI_contours[new_roi_name] = self.new_ROI_contours
polygons = ROI.calc_roi_polygon(new_roi_name, curr_slice,
dict_ROI_contours)
# Set the new ROI color
color = QtGui.QColor()
color.setRgb(90, 250, 175, 200)
pen_color = QtGui.QColor(color.red(), color.green(), color.blue())
pen = QtGui.QPen(pen_color)
pen.setStyle(QtCore.Qt.PenStyle(1))
pen.setWidthF(2.0)
# Draw the new ROI
self.dicom_preview.update_view()
for i in range(len(polygons)):
self.dicom_preview.scene.addPolygon(polygons[i], pen,
QtGui.QBrush(color))
def update_selected_rois(self):
""" Get the names of selected ROIs """
# Hide warning message
self.warning_message.setVisible(False)
self.roi_names = []
if self.first_roi_name_dropdown_list.currentText() != "":
self.roi_names.append(
self.first_roi_name_dropdown_list.currentText())
if self.second_roi_name_dropdown_list.currentText() != "" and \
self.second_roi_name_dropdown_list.isVisible():
self.roi_names.append(
self.second_roi_name_dropdown_list.currentText())
self.dict_rois_contours_axial = ROI.get_roi_contour_pixel(
self.patient_dict_container.get("raw_contour"), self.roi_names,
self.patient_dict_container.get("pixluts"))
self.display_selected_roi()
def display_selected_roi(self):
""" Display selected ROIs """
# Get the info required to display the selected ROIs
slider_id = self.dicom_view.slider.value()
curr_slice = self.patient_dict_container.get("dict_uid")[slider_id]
self.rois = self.patient_dict_container.get("rois")
# Display the selected ROIs
self.dicom_view.update_view()
for roi_id, roi_dict in self.rois.items():
roi_name = roi_dict['name']
if roi_name in self.roi_names:
polygons = ROI.calc_roi_polygon(roi_name, curr_slice,
self.dict_rois_contours_axial)
self.dicom_view.draw_roi_polygons(roi_id, polygons,
self.roi_color)
def operation_changed(self):
""" Change the form when users select different operations """
# Hide warning message
self.warning_message.setVisible(False)
selected_operation = self.operation_name_dropdown_list.currentText()
if selected_operation in self.single_roi_operation_names:
self.second_roi_name_label.setVisible(False)
self.second_roi_name_dropdown_list.setVisible(False)
self.margin_label.setVisible(True)
self.margin_line_edit.setVisible(True)
self.update_selected_rois()
else:
self.second_roi_name_label.setVisible(True)
self.second_roi_name_dropdown_list.setVisible(True)
self.margin_label.setVisible(False)
self.margin_line_edit.setVisible(False)
self.update_selected_rois()
def roi_saved(self, new_rtss):
""" Create a new ROI in Structure Tab and notify user """
new_roi_name = self.new_roi_name_line_edit.text()
self.signal_roi_manipulated.emit((new_rtss, {"draw": new_roi_name}))
QMessageBox.about(self.manipulate_roi_window_instance, "Saved",
"New contour successfully created!")
self.close()
