class RoiManager(wx.Dialog): """ Dialog accessible from the DICOM import GUI to allow user to easily update the ROI map with new plans """ def __init__(self, parent, roi_map, physician, physician_roi): """ :param parent: GUI parent :param roi_map: roi_map object :type roi_map: DatabaseROIs :param physician: initial physician value :type physician: str :param physician_roi: initial physician roi :type physician_roi: str """ wx.Dialog.__init__(self, parent) self.parent = parent self.roi_map = roi_map self.initial_physician = physician self.initial_physician_roi = physician_roi self.combo_box_physician = wx.ComboBox( self, wx.ID_ANY, choices=list(self.roi_map.physicians), style=wx.CB_DROPDOWN | wx.CB_READONLY) self.combo_box_physician_roi = wx.ComboBox(self, wx.ID_ANY, choices=[], style=wx.CB_DROPDOWN | wx.CB_READONLY) self.list_ctrl_variations = wx.ListCtrl( self, wx.ID_ANY, style=wx.LC_NO_HEADER | wx.LC_REPORT | wx.BORDER_SUNKEN) self.button_select_all = wx.Button(self, wx.ID_ANY, "Select All") self.button_deselect_all = wx.Button(self, wx.ID_ANY, "Deselect All") self.button_add = wx.Button(self, wx.ID_ANY, "Add") self.button_delete = wx.Button(self, wx.ID_ANY, "Delete") self.button_move = wx.Button(self, wx.ID_ANY, "Move") self.button_dismiss = wx.Button(self, wx.ID_CANCEL, "Dismiss") self.button_move.Disable() self.button_delete.Disable() self.button_deselect_all.Disable() self.button_add_physician = wx.Button(self, wx.ID_ANY, "Add") self.button_add_physician_roi = wx.Button(self, wx.ID_ANY, "Add") self.columns = ['Variations'] self.data_table = DataTable(self.list_ctrl_variations, columns=self.columns, widths=[400]) self.__set_properties() self.__do_layout() self.__do_bind() self.run() def __set_properties(self): self.SetTitle("ROI Manager") self.combo_box_physician.SetValue(self.initial_physician) self.update_physician_rois() self.combo_box_physician_roi.SetValue(self.initial_physician_roi) self.update_variations() def __do_bind(self): self.Bind(wx.EVT_COMBOBOX, self.physician_ticker, id=self.combo_box_physician.GetId()) self.Bind(wx.EVT_COMBOBOX, self.physician_roi_ticker, id=self.combo_box_physician_roi.GetId()) self.Bind(wx.EVT_BUTTON, self.add_physician, id=self.button_add_physician.GetId()) self.Bind(wx.EVT_BUTTON, self.add_physician_roi, id=self.button_add_physician_roi.GetId()) self.Bind(wx.EVT_BUTTON, self.select_all, id=self.button_select_all.GetId()) self.Bind(wx.EVT_BUTTON, self.deselect_all, id=self.button_deselect_all.GetId()) self.Bind(wx.EVT_BUTTON, self.add_variation, id=self.button_add.GetId()) self.Bind(wx.EVT_BUTTON, self.move_variations, id=self.button_move.GetId()) self.Bind(wx.EVT_BUTTON, self.delete_variations, id=self.button_delete.GetId()) self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.update_button_enable, id=self.list_ctrl_variations.GetId()) self.Bind(wx.EVT_LIST_ITEM_DESELECTED, self.update_button_enable, id=self.list_ctrl_variations.GetId()) def __do_layout(self): sizer_wrapper = wx.BoxSizer(wx.VERTICAL) sizer_buttons = wx.BoxSizer(wx.HORIZONTAL) sizer_variation_buttons = wx.BoxSizer(wx.VERTICAL) sizer_variation_table = wx.BoxSizer(wx.VERTICAL) sizer_select = wx.StaticBoxSizer(wx.StaticBox(self, wx.ID_ANY, ""), wx.VERTICAL) sizer_variations = wx.BoxSizer(wx.HORIZONTAL) sizer_physician_roi = wx.BoxSizer(wx.VERTICAL) sizer_physician_roi_row_2 = wx.BoxSizer(wx.HORIZONTAL) sizer_physician = wx.BoxSizer(wx.VERTICAL) label_physician = wx.StaticText(self, wx.ID_ANY, "Physician:") sizer_physician.Add(label_physician, 0, wx.LEFT, 5) sizer_physician_row = wx.BoxSizer(wx.HORIZONTAL) sizer_physician_row.Add(self.combo_box_physician, 1, wx.EXPAND | wx.LEFT | wx.RIGHT, 5) sizer_physician_row.Add(self.button_add_physician, 0, wx.EXPAND | wx.LEFT | wx.RIGHT, 10) sizer_physician.Add(sizer_physician_row, 1, wx.EXPAND, 0) sizer_select.Add(sizer_physician, 0, wx.ALL | wx.EXPAND, 5) label_physician_roi = wx.StaticText(self, wx.ID_ANY, "Physician ROI:") sizer_physician_roi.Add(label_physician_roi, 0, wx.LEFT, 5) sizer_physician_roi_row_2.Add(self.combo_box_physician_roi, 1, wx.EXPAND | wx.LEFT | wx.RIGHT, 5) sizer_physician_roi_row_2.Add(self.button_add_physician_roi, 0, wx.EXPAND | wx.LEFT | wx.RIGHT, 10) sizer_physician_roi.Add(sizer_physician_roi_row_2, 0, wx.EXPAND, 0) sizer_select.Add(sizer_physician_roi, 0, wx.ALL | wx.EXPAND, 5) label_variations = wx.StaticText(self, wx.ID_ANY, "Variations:") label_variations_buttons = wx.StaticText(self, wx.ID_ANY, " ") sizer_variation_table.Add(label_variations, 0, 0, 0) sizer_variation_table.Add(self.list_ctrl_variations, 1, wx.ALL | wx.EXPAND, 0) sizer_variations.Add(sizer_variation_table, 0, wx.ALL, 5) sizer_variation_buttons.Add(label_variations_buttons, 0, 0, 0) sizer_variation_buttons.Add(self.button_add, 0, wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM, 5) sizer_variation_buttons.Add(self.button_delete, 0, wx.EXPAND | wx.ALL, 5) sizer_variation_buttons.Add(self.button_move, 0, wx.EXPAND | wx.ALL, 5) sizer_variation_buttons.Add(self.button_select_all, 0, wx.EXPAND | wx.ALL, 5) sizer_variation_buttons.Add(self.button_deselect_all, 0, wx.EXPAND | wx.ALL, 5) sizer_variations.Add(sizer_variation_buttons, 0, wx.EXPAND | wx.ALL, 5) sizer_select.Add(sizer_variations, 0, wx.ALL | wx.EXPAND, 5) sizer_wrapper.Add(sizer_select, 0, wx.ALL, 5) sizer_buttons.Add(self.button_dismiss, 0, wx.ALIGN_CENTER | wx.ALL, 5) sizer_wrapper.Add(sizer_buttons, 0, wx.ALIGN_CENTER | wx.ALL, 0) self.SetSizer(sizer_wrapper) self.Layout() self.Fit() self.Center() def run(self): self.ShowModal() self.Destroy() def update_physicians(self, old_physicians=None): choices = list(self.roi_map.physicians) self.update_combo_box_choices(self.combo_box_physician, choices, old_physicians) def update_physician_rois(self, old_physician_rois=None): choices = self.roi_map.get_physician_rois(self.physician) self.update_combo_box_choices(self.combo_box_physician_roi, choices, old_physician_rois) @staticmethod def update_combo_box_choices(combo_box, choices, old_choices): if choices: value = choices[0] if old_choices: value = list(set(choices) - set(old_choices)) value = value[0] if len(value) else combo_box.GetValue() combo_box.Clear() combo_box.AppendItems(choices) combo_box.SetValue(value) def update_variations(self): self.data_table.set_data(self.variation_table_data, self.columns) self.update_button_enable() def physician_ticker(self, evt): self.update_physician_rois() self.update_variations() def physician_roi_ticker(self, evt): self.update_variations() @property def physician(self): return self.combo_box_physician.GetValue() @property def physician_roi(self): return self.combo_box_physician_roi.GetValue() @property def variations(self): variations = self.roi_map.get_variations( self.combo_box_physician.GetValue(), self.combo_box_physician_roi.GetValue()) variations = list( set(variations) - {self.combo_box_physician_roi.GetValue()}) # remove physician roi variations.sort() return variations @property def variation_table_data(self): return {'Variations': self.variations} @property def selected_indices(self): return get_selected_listctrl_items(self.list_ctrl_variations) @property def selected_values(self): return [ self.list_ctrl_variations.GetItem(i, 0).GetText() for i in self.selected_indices ] @property def variation_count(self): return len(self.variations) def select_all(self, evt): self.apply_global_selection() def deselect_all(self, evt): self.apply_global_selection(on=0) def apply_global_selection(self, on=1): for i in range(self.variation_count): self.list_ctrl_variations.Select(i, on=on) def delete_variations(self, evt): self.roi_map.delete_variations(self.physician, self.physician_roi, self.selected_values) self.update_variations() def add_variation(self, evt): AddVariation(self.parent, self.physician, self.roi_map, self.physician_roi) self.update_variations() def move_variations(self, evt): choices = [ roi for roi in self.roi_map.get_physician_rois(self.physician) if roi != self.physician_roi ] MoveVariationDialog(self, self.selected_values, self.physician, self.physician_roi, choices, self.roi_map) self.update_variations() def update_button_enable(self, *evt): if self.selected_indices: self.button_move.Enable() self.button_delete.Enable() self.button_deselect_all.Enable() else: self.button_move.Disable() self.button_delete.Disable() self.button_deselect_all.Disable() def add_physician_roi(self, evt): old_physician_rois = self.roi_map.get_physician_rois(self.physician) AddPhysicianROI(self.parent, self.physician, self.roi_map) self.update_physician_rois(old_physician_rois=old_physician_rois) def add_physician(self, evt): old_physicians = list(self.roi_map.physicians) AddPhysician(self.roi_map) self.update_physicians(old_physicians=old_physicians)
class DatabaseEditorFrame(wx.Frame): """ Various viewing and editing tools for the SQL database. This object is called on Database toolbar click. """ def __init__(self, roi_map, options): """ :param roi_map: roi_map object :type roi_map: DatabaseROIs """ wx.Frame.__init__(self, None, title='Database Administrator') set_frame_icon(self) set_msw_background_color( self) # If windows, change the background color self.roi_map = roi_map self.options = options self.db_tree = self.get_db_tree() self.SetSize(get_window_size(0.792, 0.781)) self.window_db_editor = wx.SplitterWindow(self, wx.ID_ANY, style=wx.SP_3D) self.window_pane_db_tree = wx.ScrolledWindow(self.window_db_editor, wx.ID_ANY, style=wx.BORDER_SUNKEN | wx.TAB_TRAVERSAL) self.tree_ctrl_db = wx.TreeCtrl(self.window_pane_db_tree, wx.ID_ANY, style=wx.TR_HAS_BUTTONS | wx.TR_MULTIPLE) self.window_pane_query = wx.Panel(self.window_db_editor, wx.ID_ANY, style=wx.BORDER_SUNKEN | wx.TAB_TRAVERSAL) self.text_ctrl_condition = wx.TextCtrl(self.window_pane_query, wx.ID_ANY, "") self.list_ctrl_query_results = wx.ListCtrl( self.window_pane_query, wx.ID_ANY, style=wx.LC_HRULES | wx.LC_REPORT | wx.LC_VRULES) self.data_query_results = DataTable( self.list_ctrl_query_results, columns=['mrn', 'study_instance_uid']) self.combo_box_query_table = wx.ComboBox(self.window_pane_query, wx.ID_ANY, choices=list(self.db_tree), style=wx.CB_DROPDOWN | wx.CB_READONLY) self.button = { 'delete_all_data': wx.Button(self, wx.ID_ANY, "Delete All Data"), 'rebuild_db': wx.Button(self, wx.ID_ANY, "Rebuild Database"), 'calculations': wx.Button(self, wx.ID_ANY, "Calculations"), 'edit_db': wx.Button(self, wx.ID_ANY, "Edit Database"), 'reimport': wx.Button(self, wx.ID_ANY, "Reimport from DICOM"), 'delete_study': wx.Button(self, wx.ID_ANY, "Delete Study"), 'change_mrn_uid': wx.Button(self, wx.ID_ANY, "Change MRN/UID"), 'query': wx.Button(self.window_pane_query, wx.ID_ANY, "Query"), 'clear': wx.Button(self.window_pane_query, wx.ID_ANY, "Clear"), 'export_csv': wx.Button(self.window_pane_query, wx.ID_ANY, "Export"), 'remap_roi_names': wx.Button(self, wx.ID_ANY, "Remap ROI Names"), 'update_from_csv': wx.Button(self, wx.ID_ANY, "Update from CSV"), # 'plan_complexity': wx.Button(self, wx.ID_ANY, "Recalc Plan Complexities"), 'auto_fit_columns': wx.Button(self.window_pane_query, wx.ID_ANY, "Auto-fit Columns") } self.checkbox_auto_backup = wx.CheckBox( self, wx.ID_ANY, "Auto Backup SQLite DB After Import") self.__set_properties() self.__do_layout() self.__do_bind() self.selected_columns = { table: {c: False for c in list(self.db_tree[table])} for table in list(self.db_tree) } self.selected_tables = {table: False for table in list(self.db_tree)} self.window_db_editor.SetSashPosition(250) self.tree_ctrl_db.Expand(self.db_tree_root) self.allow_tree_select_change = True self.Show() def __set_properties(self): self.SetTitle("Database Administrator") self.window_pane_db_tree.SetScrollRate(10, 10) self.window_db_editor.SetMinimumPaneSize(20) self.db_tree_root = self.tree_ctrl_db.AddRoot('DVH Analytics') self.table_nodes = { table: self.tree_ctrl_db.AppendItem(self.db_tree_root, table) for table in self.db_tree } self.column_nodes = {table: {} for table in self.db_tree} for table in self.db_tree: for column in self.db_tree[table]: self.column_nodes[table][ column] = self.tree_ctrl_db.AppendItem( self.table_nodes[table], column) self.combo_box_query_table.SetValue('Plans') self.checkbox_auto_backup.SetFont( wx.Font(11, wx.FONTFAMILY_DEFAULT, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_NORMAL, 0, "")) self.checkbox_auto_backup.SetValue(self.options.AUTO_SQL_DB_BACKUP) def __do_layout(self): sizer_wrapper = wx.BoxSizer(wx.VERTICAL) sizer_query = wx.BoxSizer(wx.VERTICAL) sizer_query_table = wx.StaticBoxSizer( wx.StaticBox(self.window_pane_query, wx.ID_ANY, "Query Results"), wx.VERTICAL) sizer_condition_buttons = wx.BoxSizer(wx.HORIZONTAL) sizer_download_button = wx.BoxSizer(wx.VERTICAL) sizer_clear_button = wx.BoxSizer(wx.VERTICAL) sizer_query_button = wx.BoxSizer(wx.VERTICAL) sizer_auto_fit_columns_button = wx.BoxSizer(wx.VERTICAL) sizer_condition = wx.BoxSizer(wx.VERTICAL) sizer_combo_box = wx.BoxSizer(wx.VERTICAL) sizer_db_tree = wx.BoxSizer(wx.HORIZONTAL) sizer_dialog_buttons = wx.BoxSizer(wx.VERTICAL) sizer_dialog_buttons_1 = wx.BoxSizer(wx.HORIZONTAL) sizer_dialog_buttons_1.Add(self.button['calculations'], 0, wx.ALL, 5) sizer_dialog_buttons_1.Add(self.button['edit_db'], 0, wx.ALL, 5) sizer_dialog_buttons_1.Add(self.button['reimport'], 0, wx.ALL, 5) sizer_dialog_buttons_1.Add(self.button['delete_study'], 0, wx.ALL, 5) sizer_dialog_buttons_1.Add(self.button['change_mrn_uid'], 0, wx.ALL, 5) sizer_dialog_buttons_1.Add(self.button['rebuild_db'], 0, wx.ALL, 5) sizer_dialog_buttons_1.Add(self.button['delete_all_data'], 0, wx.ALL, 5) sizer_dialog_buttons_1.Add(self.button['remap_roi_names'], 0, wx.ALL, 5) sizer_dialog_buttons_1.Add(self.button['update_from_csv'], 0, wx.ALL, 5) # sizer_dialog_buttons.Add(self.button['plan_complexity'], 0, wx.ALL, 5) sizer_dialog_buttons.Add(sizer_dialog_buttons_1, 0, 0, 0) sizer_dialog_buttons.Add(self.checkbox_auto_backup, 0, wx.LEFT, 5) sizer_wrapper.Add(sizer_dialog_buttons, 0, wx.ALL, 5) sizer_db_tree.Add(self.tree_ctrl_db, 1, wx.EXPAND, 0) self.window_pane_db_tree.SetSizer(sizer_db_tree) label_table = wx.StaticText(self.window_pane_query, wx.ID_ANY, "Table:") sizer_combo_box.Add(label_table, 0, wx.BOTTOM | wx.TOP, 5) sizer_combo_box.Add(self.combo_box_query_table, 0, 0, 0) sizer_condition_buttons.Add(sizer_combo_box, 0, wx.ALL | wx.EXPAND, 5) label_condition = wx.StaticText(self.window_pane_query, wx.ID_ANY, "Condition:") sizer_condition.Add(label_condition, 0, wx.BOTTOM | wx.TOP, 5) sizer_condition.Add(self.text_ctrl_condition, 0, wx.ALL | wx.EXPAND, 0) sizer_condition_buttons.Add(sizer_condition, 1, wx.ALL | wx.EXPAND, 5) label_spacer_1 = wx.StaticText(self.window_pane_query, wx.ID_ANY, "") sizer_query_button.Add(label_spacer_1, 0, wx.BOTTOM, 5) sizer_query_button.Add(self.button['query'], 0, wx.ALL, 5) sizer_condition_buttons.Add(sizer_query_button, 0, wx.ALL, 5) label_spacer_2 = wx.StaticText(self.window_pane_query, wx.ID_ANY, "") sizer_download_button.Add(label_spacer_2, 0, wx.BOTTOM, 5) sizer_download_button.Add(self.button['export_csv'], 0, wx.TOP | wx.BOTTOM, 5) sizer_condition_buttons.Add(sizer_download_button, 0, wx.ALL, 5) label_spacer_3 = wx.StaticText(self.window_pane_query, wx.ID_ANY, "") sizer_clear_button.Add(label_spacer_3, 0, wx.BOTTOM, 5) sizer_clear_button.Add(self.button['clear'], 0, wx.TOP | wx.BOTTOM, 5) sizer_condition_buttons.Add(sizer_clear_button, 0, wx.ALL, 5) label_spacer_4 = wx.StaticText(self.window_pane_query, wx.ID_ANY, "") sizer_auto_fit_columns_button.Add(label_spacer_4, 0, wx.BOTTOM, 5) sizer_auto_fit_columns_button.Add(self.button['auto_fit_columns'], 0, wx.ALL, 5) sizer_condition_buttons.Add(sizer_auto_fit_columns_button, 0, wx.ALL, 5) sizer_query.Add(sizer_condition_buttons, 0, wx.BOTTOM | wx.EXPAND | wx.LEFT | wx.RIGHT, 5) sizer_query_table.Add(self.list_ctrl_query_results, 1, wx.EXPAND, 0) sizer_query.Add(sizer_query_table, 1, wx.ALL | wx.EXPAND, 5) self.window_pane_query.SetSizer(sizer_query) self.window_db_editor.SplitVertically(self.window_pane_db_tree, self.window_pane_query) sizer_wrapper.Add(self.window_db_editor, 1, wx.EXPAND, 0) self.SetSizer(sizer_wrapper) self.Layout() self.Center() def __do_bind(self): # self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnTreeAdd, self.tree_ctrl_db, id=self.tree_ctrl_db.GetId()) # All buttons are bound to a function based on their key prepended with 'on_' # For example, query button calls on_query when clicked for key, button in self.button.items(): self.Bind(wx.EVT_BUTTON, getattr(self, 'on_' + key), id=button.GetId()) self.Bind(wx.EVT_CHECKBOX, self.on_auto_backup, id=self.checkbox_auto_backup.GetId()) self.Bind(wx.EVT_LIST_COL_CLICK, self.sort_query_results, self.list_ctrl_query_results) def on_auto_fit_columns(self, *evt): self.data_query_results.set_column_widths(auto=True) def on_tree_add(self, evt): self.update_selected_tree_items() # def UnselectOtherTables(self, selected_table): # self.allow_tree_select_change = False # for table, node in self.table_nodes.items(): # if table != selected_table: # self.tree_ctrl_db.UnselectItem(node) # self.allow_tree_select_change = True def on_query(self, evt): self.update_selected_tree_items() table = self.combo_box_query_table.GetValue() columns = [ c for c, sel in self.selected_columns[table].items() if sel and c not in {'mrn', 'study_instance_uid'} ] columns.sort() if not columns: columns = [ c for c in self.db_tree[table] if c not in {'mrn', 'study_instance_uid'} ] columns.insert(0, 'study_instance_uid') columns.insert(0, 'mrn') condition = self.text_ctrl_condition.GetValue() with DVH_SQL() as cnx: try: data = cnx.query(table, ','.join(columns), condition, bokeh_cds=True) self.data_query_results.set_data(data, columns) except SQLError as e: SQLErrorDialog(self, e) self.data_query_results.clear() def on_clear(self, evt): self.data_query_results.clear() @staticmethod def get_db_tree(): tree = get_database_tree() for table in list(tree): tree[table] = [ column for column in tree[table] if 'string' not in column ] return tree def update_selected_tree_items(self): for table, table_item in self.table_nodes.items(): self.selected_tables[table] = self.tree_ctrl_db.IsSelected( table_item) for column, column_item in self.column_nodes[table].items(): self.selected_columns[table][ column] = self.tree_ctrl_db.IsSelected(column_item) def on_change_mrn_uid(self, evt): self.change_or_delete_dlg(ChangePatientIdentifierDialog) def on_delete_study(self, evt): self.change_or_delete_dlg(DeletePatientDialog) def change_or_delete_dlg(self, class_type): selected_data = self.data_query_results.selected_row_data if selected_data: class_type(mrn=selected_data[0][0], study_instance_uid=selected_data[0][1]) else: class_type() def on_reimport(self, evt): selected_data = self.data_query_results.selected_row_data if selected_data: ReimportDialog(self.roi_map, self.options, mrn=selected_data[0][0], study_instance_uid=selected_data[0][1]) else: ReimportDialog(self.roi_map, self.options) @staticmethod def on_edit_db(evt): EditDatabaseDialog() @staticmethod def on_calculations(evt): CalculationsDialog() def on_rebuild_db(self, evt): RebuildDB(self, self.roi_map, self.options) def on_delete_all_data(self, evt): DeleteAllData(self, self.options) def on_export_csv(self, evt): save_data_to_file(self, "Export Data Table to CSV", self.data_query_results.get_csv()) def on_remap_roi_names(self, evt): RemapROIFrame(self.roi_map, remap_all=True) def sort_query_results(self, evt): self.data_query_results.sort_table(evt) def on_auto_backup(self, *evt): self.options.AUTO_SQL_DB_BACKUP = self.checkbox_auto_backup.GetValue() # @staticmethod # def on_plan_complexity(*evt): # with wx.BusyCursor() as busy: # recalculate_plan_complexities_from_beams() def on_update_from_csv(self, *evt): msg = "Load a CSV file and update your database. Please see manual for " \ "more information. Formatting is critical, this can not be undone!" caption = "USE WITH EXTREME CAUTION" ErrorDialog(self, msg, caption, flags=wx.ICON_WARNING | wx.OK | wx.OK_DEFAULT) with wx.FileDialog(self, "Load CSV into Database", style=wx.FD_FILE_MUST_EXIST | wx.FD_OPEN) as fileDialog: if fileDialog.ShowModal() == wx.ID_CANCEL: return pathname = fileDialog.GetPath() with open(pathname, 'r') as fp: lines = fp.readlines() header_row = "table,column,value,condition" if lines and header_row in lines[0].strip(): params = [csv_to_list(line.strip()) for line in lines[1:]] ProgressFrame(params, update_db_with_csv, title="Update Database from CSV file", action_msg="Processing row", star_map=True) else: msg = "The selected file was not recognized. " \ "Be sure you have a header row of: %s" % header_row caption = "CSV load failure" ErrorDialog(self, msg, caption, flags=wx.ICON_WARNING | wx.OK | wx.OK_DEFAULT)
class ExportCSVDialog(wx.Dialog): """ Allow user to select available data for export to CSV into one file. This class leverages the get_csv functions built-into each of the data types / tabs """ def __init__(self, app): """ :param app: easier to pass main frame pointer than several links to each data type :type app: DVHAMainFrame """ wx.Dialog.__init__(self, None) self.app = app # Each of these objects shoudl have a has_data property, if False, those UI elements in this class will # be disabled (i.e., don't allow user to export empty tables self.enabled = { 'DVHs': self.app.dvh.has_data, 'DVHs Summary': self.app.dvh.has_data, 'Endpoints': self.app.endpoint.has_data, 'Radbio': self.app.radbio.has_data, 'Charting Variables': self.app.time_series.has_data } checkbox_keys = [ 'DVHs', 'DVHs Summary', 'Endpoints', 'Radbio', 'Charting Variables' ] self.checkbox = { key: wx.CheckBox(self, wx.ID_ANY, key) for key in checkbox_keys } # set to a dictionary because previous versions had a tree with Regression self.list_ctrl = { 'Charting Variables': wx.ListCtrl(self, wx.ID_ANY, style=wx.LC_HRULES | wx.LC_REPORT | wx.LC_VRULES) } time_series_column = "Variables" time_series_variables = self.app.time_series.combo_box_y_axis.GetItems( ) time_series_data = {time_series_column: time_series_variables} self.data_table_time_series = DataTable( self.list_ctrl['Charting Variables'], columns=[time_series_column], widths=[400]) self.data_table_time_series.set_data(time_series_data, [time_series_column]) # set to a dictionary because previous versions had a table with Regression self.button_select_data = { 'Charting Variables': { 'Select': wx.Button(self, wx.ID_ANY, "Select All"), 'Deselect': wx.Button(self, wx.ID_ANY, "Deselect All") } } self.button_save = wx.Button(self, wx.ID_OK, "Save") self.button_cancel = wx.Button(self, wx.ID_CANCEL, "Cancel") self.button_select_all = wx.Button(self, wx.ID_ANY, 'Select All') self.button_deselect_all = wx.Button(self, wx.ID_ANY, 'Deselect All') self.__set_properties() self.__do_bind() self.__do_layout() self.run() def __set_properties(self): self.SetTitle("Export Data to CSV") self.button_select_all.SetToolTip( 'Only data objects with data will be enabled.') self.validate_ui_objects() def __do_bind(self): self.Bind(wx.EVT_BUTTON, self.on_select_all, id=self.button_select_all.GetId()) self.Bind(wx.EVT_BUTTON, self.on_deselect_all, id=self.button_deselect_all.GetId()) self.Bind( wx.EVT_BUTTON, self.on_time_series_select_all, id=self.button_select_data['Charting Variables']['Select'].GetId()) self.Bind(wx.EVT_BUTTON, self.on_time_series_deselect_all, id=self.button_select_data['Charting Variables'] ['Deselect'].GetId()) def __do_layout(self): sizer_wrapper = wx.BoxSizer(wx.VERTICAL) sizer_main = wx.BoxSizer(wx.VERTICAL) sizer_main_buttons = wx.BoxSizer(wx.HORIZONTAL) sizer_data = wx.StaticBoxSizer( wx.StaticBox(self, wx.ID_ANY, "Data Selection"), wx.VERTICAL) sizer_time_series = wx.BoxSizer(wx.VERTICAL) sizer_time_series_listctrl = wx.BoxSizer(wx.HORIZONTAL) sizer_time_series_checkboxes = wx.BoxSizer(wx.HORIZONTAL) sizer_time_series_buttons = wx.BoxSizer(wx.HORIZONTAL) sizer_radbio = wx.BoxSizer(wx.VERTICAL) sizer_endpoints = wx.BoxSizer(wx.VERTICAL) sizer_dvhs = wx.BoxSizer(wx.VERTICAL) sizer_dvhs_checkboxes = wx.BoxSizer(wx.HORIZONTAL) keys = ['DVHs', 'Endpoints', 'Radbio'] static_line = {key: wx.StaticLine(self, wx.ID_ANY) for key in keys} sizer_dvhs_checkboxes.Add(self.checkbox['DVHs'], 1, wx.ALL | wx.EXPAND, 5) sizer_dvhs_checkboxes.Add(self.checkbox['DVHs Summary'], 1, wx.ALL | wx.EXPAND, 5) sizer_dvhs.Add(sizer_dvhs_checkboxes, 1, wx.EXPAND, 0) sizer_dvhs.Add(static_line['DVHs'], 0, wx.EXPAND | wx.TOP, 5) sizer_data.Add(sizer_dvhs, 0, wx.ALL | wx.EXPAND, 5) sizer_endpoints.Add(self.checkbox['Endpoints'], 0, wx.ALL, 5) sizer_endpoints.Add(static_line['Endpoints'], 0, wx.EXPAND | wx.TOP, 5) sizer_data.Add(sizer_endpoints, 0, wx.ALL | wx.EXPAND, 5) sizer_radbio.Add(self.checkbox['Radbio'], 0, wx.ALL, 5) sizer_radbio.Add(static_line['Radbio'], 0, wx.EXPAND | wx.TOP, 5) sizer_data.Add(sizer_radbio, 0, wx.ALL | wx.EXPAND, 5) sizer_time_series_checkboxes.Add(self.checkbox['Charting Variables'], 1, wx.EXPAND, 0) sizer_time_series_buttons.Add( self.button_select_data['Charting Variables']['Select'], 0, wx.ALL | wx.EXPAND, 5) sizer_time_series_buttons.Add( self.button_select_data['Charting Variables']['Deselect'], 0, wx.ALL | wx.EXPAND, 5) sizer_time_series_checkboxes.Add(sizer_time_series_buttons, 1, wx.EXPAND, 0) sizer_time_series.Add(sizer_time_series_checkboxes, 0, wx.ALL | wx.EXPAND, 5) sizer_time_series_listctrl.Add((20, 20), 0, 0, 0) sizer_time_series_listctrl.Add(self.list_ctrl['Charting Variables'], 1, wx.ALL | wx.EXPAND, 5) sizer_time_series.Add(sizer_time_series_listctrl, 0, wx.ALL | wx.EXPAND, 5) sizer_data.Add(sizer_time_series, 0, wx.ALL | wx.EXPAND, 5) sizer_main.Add(sizer_data, 0, wx.ALL | wx.EXPAND, 5) sizer_main_buttons.Add(self.button_select_all, 0, wx.ALL, 5) sizer_main_buttons.Add(self.button_deselect_all, 0, wx.ALL, 5) sizer_main_buttons.Add(self.button_save, 0, wx.ALL, 5) sizer_main_buttons.Add(self.button_cancel, 0, wx.ALL, 5) sizer_main.Add(sizer_main_buttons, 0, wx.ALIGN_RIGHT | wx.ALL, 5) sizer_wrapper.Add(sizer_main, 1, wx.EXPAND, 0) self.SetSizer(sizer_wrapper) self.Layout() self.Fit() self.Center() def set_checkbox_values(self, value): for checkbox in self.checkbox.values(): checkbox.SetValue(value) def on_select_all(self, evt): self.set_checkbox_values(True) self.validate_ui_objects(allow_enable=False) def on_deselect_all(self, evt): self.set_checkbox_values(False) self.validate_ui_objects(allow_enable=False) def on_time_series_select_all(self, evt): self.data_table_time_series.apply_selection_to_all(1) def on_time_series_deselect_all(self, evt): self.data_table_time_series.apply_selection_to_all(0) def validate_ui_objects(self, allow_enable=True): tables = {'Charting Variables': self.data_table_time_series} for key, data_table in tables.items(): state = data_table.has_data if not state or (state and allow_enable): self.list_ctrl[key].Enable(state) self.button_select_data[key]['Select'].Enable(state) self.button_select_data[key]['Deselect'].Enable(state) for key, value in self.enabled.items(): if not value or (value and allow_enable): self.checkbox[key].SetValue(value) self.checkbox[key].Enable(value) def run(self): res = self.ShowModal() if res == wx.ID_OK: save_data_to_file(self, 'Export CSV Data', self.csv) self.Destroy() def is_checked(self, key): return self.checkbox[key].GetValue() def on_dvh_check(self, evt): if not self.is_checked('DVHs'): self.checkbox['DVHs Summary'].SetValue(False) @property def csv(self): csv_data = [] csv_key = ['DVHs', 'Endpoints', 'Radbio', 'Charting Variables'] csv_obj = [ None, self.app.endpoint, self.app.radbio, self.app.time_series, self.app.control_chart ] for i, key in enumerate(csv_key): if self.is_checked(key): csv_data.append('%s\n' % key) if key == 'DVHs': # DVHs has a summary and plot data for export csv_data.append( self.app.plot.get_csv( include_summary=self.is_checked('DVHs Summary'), include_dvhs=self.is_checked('DVHs'))) else: if key == 'Charting Variables': selection_indices = get_selected_listctrl_items( self.list_ctrl['Charting Variables']) y_choices = self.app.time_series.combo_box_y_axis.GetItems( ) selection = [ y for i, y in enumerate(y_choices) if i in selection_indices ] else: selection = None csv_data.append(csv_obj[i].get_csv(selection=selection)) csv_data.append('\n\n') return '\n'.join(csv_data)
class DatabaseEditorFrame(wx.Frame): """ Various viewing and editing tools for the SQL database. This object is called on Database toolbar click. """ def __init__(self, roi_map): """ :param roi_map: roi_map object :type roi_map: DatabaseROIs """ wx.Frame.__init__(self, None, title='Database Administrator') set_msw_background_color(self) # If windows, change the background color self.roi_map = roi_map self.db_tree = self.get_db_tree() self.SetSize(get_window_size(0.792, 0.781)) self.window_db_editor = wx.SplitterWindow(self, wx.ID_ANY, style=wx.SP_3D) self.window_pane_db_tree = wx.ScrolledWindow(self.window_db_editor, wx.ID_ANY, style=wx.BORDER_SUNKEN | wx.TAB_TRAVERSAL) self.tree_ctrl_db = wx.TreeCtrl(self.window_pane_db_tree, wx.ID_ANY, style=wx.TR_HAS_BUTTONS | wx.TR_MULTIPLE) self.window_pane_query = wx.Panel(self.window_db_editor, wx.ID_ANY, style=wx.BORDER_SUNKEN | wx.TAB_TRAVERSAL) self.text_ctrl_condition = wx.TextCtrl(self.window_pane_query, wx.ID_ANY, "") self.list_ctrl_query_results = wx.ListCtrl(self.window_pane_query, wx.ID_ANY, style=wx.LC_HRULES | wx.LC_REPORT | wx.LC_VRULES) self.data_query_results = DataTable(self.list_ctrl_query_results, columns=['mrn', 'study_instance_uid']) self.combo_box_query_table = wx.ComboBox(self.window_pane_query, wx.ID_ANY, choices=list(self.db_tree), style=wx.CB_DROPDOWN | wx.CB_READONLY) self.button = {'delete_all_data': wx.Button(self, wx.ID_ANY, "Delete All Data"), 'rebuild_db': wx.Button(self, wx.ID_ANY, "Rebuild Database"), # 'calculations': wx.Button(self, wx.ID_ANY, "Calculations"), 'edit_db': wx.Button(self, wx.ID_ANY, "Edit Database"), 'reimport': wx.Button(self, wx.ID_ANY, "Reimport from DICOM"), 'delete_study': wx.Button(self, wx.ID_ANY, "Delete Study"), 'change_mrn_uid': wx.Button(self, wx.ID_ANY, "Change MRN/UID"), 'query': wx.Button(self.window_pane_query, wx.ID_ANY, "Query"), 'clear': wx.Button(self.window_pane_query, wx.ID_ANY, "Clear"), 'export_csv': wx.Button(self.window_pane_query, wx.ID_ANY, "Export"), 'remap_roi_names': wx.Button(self, wx.ID_ANY, "Remap ROI Names")} self.__set_properties() self.__do_layout() self.__do_bind() self.selected_columns = {table: {c: False for c in list(self.db_tree[table])} for table in list(self.db_tree)} self.selected_tables = {table: False for table in list(self.db_tree)} self.window_db_editor.SetSashPosition(250) self.tree_ctrl_db.Expand(self.db_tree_root) self.allow_tree_select_change = True self.Show() def __set_properties(self): self.SetTitle("Database Administrator") self.window_pane_db_tree.SetScrollRate(10, 10) self.window_db_editor.SetMinimumPaneSize(20) self.db_tree_root = self.tree_ctrl_db.AddRoot('DVH Analytics') self.table_nodes = {table: self.tree_ctrl_db.AppendItem(self.db_tree_root, table) for table in self.db_tree} self.column_nodes = {table: {} for table in self.db_tree} for table in self.db_tree: for column in self.db_tree[table]: self.column_nodes[table][column] = self.tree_ctrl_db.AppendItem(self.table_nodes[table], column) self.combo_box_query_table.SetValue('Plans') def __do_layout(self): sizer_wrapper = wx.BoxSizer(wx.VERTICAL) sizer_query = wx.BoxSizer(wx.VERTICAL) sizer_query_table = wx.StaticBoxSizer(wx.StaticBox(self.window_pane_query, wx.ID_ANY, "Query Results"), wx.VERTICAL) sizer_condition_buttons = wx.BoxSizer(wx.HORIZONTAL) sizer_download_button = wx.BoxSizer(wx.VERTICAL) sizer_clear_button = wx.BoxSizer(wx.VERTICAL) sizer_query_button = wx.BoxSizer(wx.VERTICAL) sizer_condition = wx.BoxSizer(wx.VERTICAL) sizer_combo_box = wx.BoxSizer(wx.VERTICAL) sizer_db_tree = wx.BoxSizer(wx.HORIZONTAL) sizer_dialog_buttons = wx.BoxSizer(wx.HORIZONTAL) # sizer_dialog_buttons.Add(self.button['calculations'], 0, wx.ALL, 5) sizer_dialog_buttons.Add(self.button['edit_db'], 0, wx.ALL, 5) sizer_dialog_buttons.Add(self.button['reimport'], 0, wx.ALL, 5) sizer_dialog_buttons.Add(self.button['delete_study'], 0, wx.ALL, 5) sizer_dialog_buttons.Add(self.button['change_mrn_uid'], 0, wx.ALL, 5) sizer_dialog_buttons.Add(self.button['rebuild_db'], 0, wx.ALL, 5) sizer_dialog_buttons.Add(self.button['delete_all_data'], 0, wx.ALL, 5) sizer_dialog_buttons.Add(self.button['remap_roi_names'], 0, wx.ALL, 5) sizer_wrapper.Add(sizer_dialog_buttons, 0, wx.ALL, 5) sizer_db_tree.Add(self.tree_ctrl_db, 1, wx.EXPAND, 0) self.window_pane_db_tree.SetSizer(sizer_db_tree) label_table = wx.StaticText(self.window_pane_query, wx.ID_ANY, "Table:") sizer_combo_box.Add(label_table, 0, wx.BOTTOM | wx.TOP, 5) sizer_combo_box.Add(self.combo_box_query_table, 0, 0, 0) sizer_condition_buttons.Add(sizer_combo_box, 0, wx.ALL | wx.EXPAND, 5) label_condition = wx.StaticText(self.window_pane_query, wx.ID_ANY, "Condition:") sizer_condition.Add(label_condition, 0, wx.BOTTOM | wx.TOP, 5) sizer_condition.Add(self.text_ctrl_condition, 0, wx.ALL | wx.EXPAND, 0) sizer_condition_buttons.Add(sizer_condition, 1, wx.ALL | wx.EXPAND, 5) label_spacer_1 = wx.StaticText(self.window_pane_query, wx.ID_ANY, "") sizer_query_button.Add(label_spacer_1, 0, wx.BOTTOM, 5) sizer_query_button.Add(self.button['query'], 0, wx.ALL, 5) sizer_condition_buttons.Add(sizer_query_button, 0, wx.ALL, 5) label_spacer_2 = wx.StaticText(self.window_pane_query, wx.ID_ANY, "") sizer_download_button.Add(label_spacer_2, 0, wx.BOTTOM, 5) sizer_download_button.Add(self.button['export_csv'], 0, wx.TOP | wx.BOTTOM, 5) sizer_condition_buttons.Add(sizer_download_button, 0, wx.ALL, 5) label_spacer_3 = wx.StaticText(self.window_pane_query, wx.ID_ANY, "") sizer_clear_button.Add(label_spacer_3, 0, wx.BOTTOM, 5) sizer_clear_button.Add(self.button['clear'], 0, wx.ALL, 5) sizer_condition_buttons.Add(sizer_clear_button, 0, wx.ALL, 5) sizer_query.Add(sizer_condition_buttons, 0, wx.BOTTOM | wx.EXPAND | wx.LEFT | wx.RIGHT, 5) sizer_query_table.Add(self.list_ctrl_query_results, 1, wx.EXPAND, 0) sizer_query.Add(sizer_query_table, 1, wx.ALL | wx.EXPAND, 5) self.window_pane_query.SetSizer(sizer_query) self.window_db_editor.SplitVertically(self.window_pane_db_tree, self.window_pane_query) sizer_wrapper.Add(self.window_db_editor, 1, wx.EXPAND, 0) self.SetSizer(sizer_wrapper) self.Layout() self.Center() def __do_bind(self): # self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnTreeAdd, self.tree_ctrl_db, id=self.tree_ctrl_db.GetId()) # All buttons are bound to a function based on their key prepended with 'on_' # For example, query button calls on_query when clicked for key, button in self.button.items(): self.Bind(wx.EVT_BUTTON, getattr(self, 'on_' + key), id=button.GetId()) self.Bind(wx.EVT_LIST_COL_CLICK, self.sort_query_results, self.list_ctrl_query_results) def on_tree_add(self, evt): self.update_selected_tree_items() # def UnselectOtherTables(self, selected_table): # self.allow_tree_select_change = False # for table, node in self.table_nodes.items(): # if table != selected_table: # self.tree_ctrl_db.UnselectItem(node) # self.allow_tree_select_change = True def on_query(self, evt): self.update_selected_tree_items() table = self.combo_box_query_table.GetValue() columns = [c for c, sel in self.selected_columns[table].items() if sel and c not in {'mrn', 'study_instance_uid'}] columns.sort() if not columns: columns = [c for c in self.db_tree[table] if c not in {'mrn', 'study_instance_uid'}] columns.insert(0, 'study_instance_uid') columns.insert(0, 'mrn') condition = self.text_ctrl_condition.GetValue() wait = wx.BusyInfo("Querying data\nPlease wait...") with DVH_SQL() as cnx: try: data = cnx.query(table, ','.join(columns), condition, bokeh_cds=True) self.data_query_results.set_data(data, columns) except SQLError as e: SQLErrorDialog(self, e) self.data_query_results.clear() del wait def on_clear(self, evt): self.data_query_results.clear() @staticmethod def get_db_tree(): tree = get_database_tree() for table in list(tree): tree[table] = [column for column in tree[table] if 'string' not in column] return tree def update_selected_tree_items(self): for table, table_item in self.table_nodes.items(): self.selected_tables[table] = self.tree_ctrl_db.IsSelected(table_item) for column, column_item in self.column_nodes[table].items(): self.selected_columns[table][column] = self.tree_ctrl_db.IsSelected(column_item) def on_change_mrn_uid(self, evt): self.change_or_delete_dlg(ChangePatientIdentifierDialog) def on_delete_study(self, evt): # TODO: Needs an Are you sure? dialog self.change_or_delete_dlg(DeletePatientDialog) def change_or_delete_dlg(self, class_type): selected_data = self.data_query_results.selected_row_data if selected_data: class_type(mrn=selected_data[0][0], study_instance_uid=selected_data[0][1]) else: class_type() def on_reimport(self, evt): selected_data = self.data_query_results.selected_row_data if selected_data: ReimportDialog(mrn=selected_data[0][0], study_instance_uid=selected_data[0][1]) else: ReimportDialog() @staticmethod def on_edit_db(evt): EditDatabaseDialog() @staticmethod def on_calculations(evt): CalculationsDialog() def on_rebuild_db(self, evt): RebuildDB(self) def on_delete_all_data(self, evt): DeleteAllData(self) def on_export_csv(self, evt): save_data_to_file(self, "Export Data Table to CSV", self.data_query_results.get_csv()) def on_remap_roi_names(self, evt): RemapROIFrame(self.roi_map, remap_all=True) def sort_query_results(self, evt): self.data_query_results.sort_table(evt)
class QueriedDataFrame(wx.Frame): """ Generate a simple table to view data of the current query for a specified SQL table """ def __init__(self, data_obj, columns, data_key, menu, menu_item_id): """ :param data_obj: object containing data to be viewed for each group :type data_obj: dict :param columns: columns to be displayed in table :type columns: list :param data_key: either 'DVHs', 'Plans', 'Beams', 'Rxs', or 'StatsData' :type data_key: str :param menu: a link to the main app menu, used to toggle Show/Hide status :type menu: Menu :param menu_item_id: the ID of the menu item associated with the specified data_obj """ wx.Frame.__init__(self, None, title='%s Data' % data_key[0:-1]) self.data = data_obj self.columns = columns self.sql_table = data_key self.menu = menu self.menu_item_id = menu_item_id self.list_ctrl = wx.ListCtrl(self, wx.ID_ANY, style=wx.BORDER_SUNKEN | wx.LC_HRULES | wx.LC_REPORT | wx.LC_VRULES) self.button_export = wx.Button(self, wx.ID_ANY, "Export to CSV") self.radio_button_query_group = wx.RadioBox(self, wx.ID_ANY, 'Query Group', choices=['1', '2']) self.data_table = DataTable(self.list_ctrl) self.data_table.set_data(self.table_data, self.columns) if not self.data[2]: self.radio_button_query_group.Disable() self.__do_bind() self.__set_properties() self.__do_layout() self.run() def __set_properties(self): self.SetSize(get_window_size(0.714, 0.762)) set_msw_background_color(self) set_frame_icon(self) def __do_bind(self): self.Bind(wx.EVT_BUTTON, self.on_export, id=self.button_export.GetId()) self.Bind(wx.EVT_LIST_COL_CLICK, self.sort_table, self.list_ctrl) self.Bind(wx.EVT_CLOSE, self.on_close) self.Bind(wx.EVT_RADIOBOX, self.on_group_select, id=self.radio_button_query_group.GetId()) def __do_layout(self): sizer_wrapper = wx.BoxSizer(wx.VERTICAL) sizer_widgets = wx.BoxSizer(wx.HORIZONTAL) sizer_button = wx.BoxSizer(wx.HORIZONTAL) sizer_button.Add(self.button_export, 0, wx.TOP, 15) sizer_widgets.Add(sizer_button, 0, wx.ALL, 10) sizer_widgets.Add(self.radio_button_query_group, 0, wx.ALL, 10) sizer_wrapper.Add(sizer_widgets, 0, 0, 0) sizer_wrapper.Add(self.list_ctrl, 1, wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, 10) self.SetSizer(sizer_wrapper) self.Center() @property def table_data(self): return { column: getattr(self.selected_data, column) for column in self.columns } @property def selected_group(self): return self.radio_button_query_group.GetSelection() + 1 @property def selected_data(self): return self.data[self.selected_group] def run(self): self.toggle_data_menu_item() self.Show() def on_close(self, *args): self.toggle_data_menu_item() self.Destroy() def on_export(self, *args): save_data_to_file( self, "Export Group %s %s to CSV" % (self.selected_group, self.sql_table), self.data_table.get_csv()) def toggle_data_menu_item(self): short_cut = ['DVHs', 'Plans', 'Rxs', 'Beams'].index(self.sql_table) + 1 show_hide = ['Show', 'Hide']['Show' in self.menu.GetLabel(self.menu_item_id)] self.menu.SetLabel( self.menu_item_id, '%s %s\tCtrl+%s' % (show_hide, self.sql_table, short_cut)) def sort_table(self, evt): self.data_table.sort_table(evt) def on_group_select(self, *evt): self.data_table.clear() self.data_table.set_data(self.table_data, self.columns)
class RadBioFrame: """ Object to be passed into notebook panel for the Rad Bio tab """ def __init__(self, parent, dvh, time_series, regression, control_chart): """ :param parent: notebook panel in main view :type parent: Panel :param dvh: dvh data object :type dvh: DVH :param time_series: Time Series object in notebook :type time_series: TimeSeriesFrame :param regression: Regression frame object in notebook :type regression: RegressionFrame :param control_chart: Control Chart frame object in notebook :type control_chart: ControlChartFrame """ self.parent = parent self.dvh = dvh self.time_series = time_series self.regression = regression self.control_chart = control_chart self.table_published_values = wx.ListCtrl(self.parent, wx.ID_ANY, style=wx.BORDER_SUNKEN | wx.LC_HRULES | wx.LC_REPORT | wx.LC_VRULES) self.text_input_eud_a = wx.TextCtrl(self.parent, wx.ID_ANY, "") self.text_input_gamma_50 = wx.TextCtrl(self.parent, wx.ID_ANY, "") self.text_input_td_50 = wx.TextCtrl(self.parent, wx.ID_ANY, "") self.button_apply_parameters = wx.Button(self.parent, wx.ID_ANY, "Apply Parameters") self.button_export = wx.Button(self.parent, wx.ID_ANY, "Export") self.table_rad_bio = wx.ListCtrl(self.parent, wx.ID_ANY, style=wx.BORDER_SUNKEN | wx.LC_HRULES | wx.LC_REPORT | wx.LC_VRULES) self.columns = ['MRN', 'ROI Name', 'a', u'\u03b3_50', 'TD or TCD', 'EUD', 'NTCP or TCP', 'PTV Overlap', 'ROI Type', 'Rx Dose', 'Total Fxs', 'Fx Dose'] self.width = [100, 175, 50, 50, 80, 80, 80, 100, 100, 100, 100, 100] formats = [wx.LIST_FORMAT_RIGHT] * len(self.columns) formats[0] = wx.LIST_FORMAT_LEFT formats[1] = wx.LIST_FORMAT_LEFT self.data_table_rad_bio = DataTable(self.table_rad_bio, columns=self.columns, widths=self.width, formats=formats) self.__set_properties() self.__do_layout() parent.Bind(wx.EVT_LIST_ITEM_SELECTED, self.on_parameter_select, self.table_published_values) parent.Bind(wx.EVT_BUTTON, self.apply_parameters, id=self.button_apply_parameters.GetId()) parent.Bind(wx.EVT_BUTTON, self.on_export_csv, id=self.button_export.GetId()) self.disable_buttons() def __set_properties(self): self.table_published_values.AppendColumn("Structure", format=wx.LIST_FORMAT_LEFT, width=150) self.table_published_values.AppendColumn("Endpoint", format=wx.LIST_FORMAT_LEFT, width=300) self.table_published_values.AppendColumn("a", format=wx.LIST_FORMAT_LEFT, width=-1) self.table_published_values.AppendColumn(u"\u03b3_50", format=wx.LIST_FORMAT_LEFT, width=-1) self.table_published_values.AppendColumn("TD_50", format=wx.LIST_FORMAT_LEFT, width=-1) for i, col in enumerate(self.columns): self.table_rad_bio.AppendColumn(col, width=self.width[i]) self.published_data = [['Brain', 'Necrosis', 5, 3, 60], ['Brainstem', 'Necrosis', 7, 3, 65], ['Optic Chasm', 'Blindness', 25, 3, 65], ['Colon', 'Obstruction/Perforation', 6, 4, 55], ['Ear (mid/ext)', 'Acute serous otitus', 31, 3, 40], ['Ear (mid/ext)', 'Chronic serous otitus', 31, 4, 65], ['Esophagus', 'Perforation', 19, 4, 68], ['Heart', 'Pericarditus', 3, 3, 50], ['Kidney', 'Nephritis', 1, 3, 28], ['Lens', 'Cataract', 3, 1, 18], ['Liver', 'Liver Failure', 3, 3, 40], ['Lung', 'Pneumontis', 1, 2, 24.5], ['Optic Nerve', 'Blindness', 25, 3, 65], ['Retina', 'Blindness', 15, 2, 65]] for row in self.published_data: index = self.table_published_values.InsertItem(50000, str(row[0])) for i in [1, 2, 3, 4]: self.table_published_values.SetItem(index, i, str(row[i])) def __do_layout(self): sizer_main = wx.BoxSizer(wx.VERTICAL) sizer_parameters = wx.BoxSizer(wx.VERTICAL) sizer_parameters_input = wx.StaticBoxSizer(wx.StaticBox(self.parent, wx.ID_ANY, ""), wx.HORIZONTAL) sizer_button = wx.BoxSizer(wx.VERTICAL) sizer_button_2 = wx.BoxSizer(wx.VERTICAL) sizer_td_50 = wx.BoxSizer(wx.VERTICAL) sizer_gamma_50 = wx.BoxSizer(wx.VERTICAL) sizer_eud = wx.BoxSizer(wx.VERTICAL) sizer_published_values = wx.BoxSizer(wx.VERTICAL) label_published_values = wx.StaticText(self.parent, wx.ID_ANY, "Published EUD Parameters from Emami et. al. for 1.8-2.0Gy " "fractions (Click to apply)") label_published_values.SetFont(wx.Font(11, wx.FONTFAMILY_DEFAULT, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_BOLD, 0, "")) sizer_published_values.Add(label_published_values, 0, wx.ALL, 5) sizer_published_values.Add(self.table_published_values, 1, wx.ALL, 10) sizer_published_values.SetMinSize(get_window_size(0.298, 0.319)) sizer_main.Add(sizer_published_values, 1, wx.ALL | wx.EXPAND, 10) label_parameters = wx.StaticText(self.parent, wx.ID_ANY, "Parameters:") label_parameters.SetFont(wx.Font(11, wx.FONTFAMILY_DEFAULT, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_BOLD, 0, "")) sizer_parameters.Add(label_parameters, 0, 0, 0) label_eud = wx.StaticText(self.parent, wx.ID_ANY, "EUD a-value:") sizer_eud.Add(label_eud, 0, 0, 0) sizer_eud.Add(self.text_input_eud_a, 0, wx.ALL | wx.EXPAND, 5) sizer_parameters_input.Add(sizer_eud, 1, wx.EXPAND, 0) label_gamma_50 = wx.StaticText(self.parent, wx.ID_ANY, u"\u03b3_50:") sizer_gamma_50.Add(label_gamma_50, 0, 0, 0) sizer_gamma_50.Add(self.text_input_gamma_50, 0, wx.ALL | wx.EXPAND, 5) sizer_parameters_input.Add(sizer_gamma_50, 1, wx.EXPAND, 0) label_td_50 = wx.StaticText(self.parent, wx.ID_ANY, "TD_50 or TCD_50:") sizer_td_50.Add(label_td_50, 0, 0, 0) sizer_td_50.Add(self.text_input_td_50, 0, wx.ALL | wx.EXPAND, 5) sizer_parameters_input.Add(sizer_td_50, 1, wx.EXPAND, 0) sizer_button.Add(self.button_apply_parameters, 1, wx.ALL | wx.EXPAND, 15) sizer_button_2.Add(self.button_export, 1, wx.ALL | wx.EXPAND, 15) sizer_parameters_input.Add(sizer_button, 1, wx.EXPAND, 0) sizer_parameters_input.Add(sizer_button_2, 1, wx.EXPAND, 0) sizer_parameters.Add(sizer_parameters_input, 1, wx.ALL | wx.EXPAND, 5) sizer_main.Add(sizer_parameters, 0, wx.ALL | wx.EXPAND, 10) sizer_main.Add(self.table_rad_bio, 1, wx.ALL | wx.EXPAND, 10) self.layout = sizer_main def __set_tooltips(self): self.button_apply_parameters.SetToolTip("Shift or Ctrl click for targeted application.") def enable_buttons(self): self.button_apply_parameters.Enable() def disable_buttons(self): self.button_apply_parameters.Disable() def enable_initial_buttons(self): self.button_apply_parameters.Enable() def on_parameter_select(self, evt): index = self.table_published_values.GetFirstSelected() self.text_input_eud_a.SetValue(str(self.published_data[index][2])) self.text_input_gamma_50.SetValue(str(self.published_data[index][3])) self.text_input_td_50.SetValue(str(self.published_data[index][4])) def update_dvh_data(self, dvh): """ Import dvh data, store into self.dvh and set data in data_table :param dvh: dvh object from main frame :type dvh: DVH """ self.dvh = dvh data = {'MRN': self.dvh.mrn, 'ROI Name': self.dvh.roi_name, 'a': [''] * self.dvh.count, u'\u03b3_50': [''] * self.dvh.count, 'TD or TCD': [''] * self.dvh.count, 'EUD': [''] * self.dvh.count, 'NTCP or TCP': [''] * self.dvh.count, 'PTV Overlap': self.dvh.ptv_overlap, 'ROI Type': self.dvh.roi_type, 'Rx Dose': self.dvh.rx_dose, 'Total Fxs': self.dvh.total_fxs, 'Fx Dose': self.dvh.fx_dose} self.data_table_rad_bio.set_data(data, self.columns) def apply_parameters(self, evt): """ Calculate rad bio values based on parameters supplied by user, pass information on to other tabs in GUI """ # Get the indices of the selected rows, or assume all should be updated selected_indices = get_selected_listctrl_items(self.table_rad_bio) if not selected_indices: selected_indices = range(self.data_table_rad_bio.row_count) # Concert user supplied parameters from text to floats eud_a = float_or_none(self.text_input_eud_a.GetValue()) gamma_50 = float_or_none(self.text_input_gamma_50.GetValue()) td_50 = float_or_none(self.text_input_td_50.GetValue()) # set the data in the datatable for the selected indices for i in selected_indices: current_row = self.data_table_rad_bio.get_row(i) for j in [7, 9]: current_row[j] = convert_value_to_str(current_row[j]) new_row = deepcopy(current_row) new_row[2], new_row[3], new_row[4] = eud_a, gamma_50, td_50 try: new_row[5] = "%0.2f" % round(calc_eud(self.dvh.dvh[:, i], eud_a), 2) except: new_row[5] = 'None' try: new_row[6] = "%0.2f" % round(calc_tcp(gamma_50, td_50, float(new_row[5])), 3) except: new_row[6] = 'None' self.data_table_rad_bio.edit_row(new_row, i) # Update data in in dvh object self.dvh.eud = [] self.dvh.ntcp_or_tcp = [] for i, eud in enumerate(self.data_table_rad_bio.data['EUD']): self.dvh.eud.append(float_or_none(eud)) self.dvh.ntcp_or_tcp.append(float_or_none(self.data_table_rad_bio.data['NTCP or TCP'][i])) # update data in time series self.time_series.update_y_axis_options() if self.time_series.combo_box_y_axis.GetValue() in ['EUD', 'NTCP or TCP']: self.time_series.update_plot() # update data in regression self.regression.stats_data.update_endpoints_and_radbio() self.regression.update_combo_box_y_choices() # update data in control chart self.control_chart.update_combo_box_y_choices() if self.control_chart.combo_box_y_axis.GetValue() in ['EUD', 'NTCP or TCP']: self.control_chart.update_plot() def clear_data(self): self.data_table_rad_bio.delete_all_rows() def get_csv(self, selection=None): return self.data_table_rad_bio.get_csv() def on_export_csv(self, evt): save_data_to_file(self.parent, "Export RadBio table to CSV", self.get_csv()) def get_save_data(self): return self.data_table_rad_bio.get_save_data() def load_save_data(self, save_data): self.data_table_rad_bio.load_save_data(save_data) @property def has_data(self): return any(self.data_table_rad_bio.data['a'])
class ROIMapFrame(wx.Frame): """ Class to view and edit roi map """ def __init__(self, roi_map): """ :param roi_map: roi map object :type roi_map: DatabaseROIs """ wx.Frame.__init__(self, None, title='ROI Map') set_frame_icon(self) self.roi_map = roi_map self.window_size = get_window_size(0.893, 0.762) self.SetSize(self.window_size) self.window = wx.SplitterWindow(self, wx.ID_ANY) self.window_tree = wx.Panel(self.window, wx.ID_ANY, style=wx.BORDER_SUNKEN) self.combo_box_tree_plot_data = wx.ComboBox(self.window_tree, wx.ID_ANY, choices=['All', 'Linked', 'Unlinked', 'Branched'], style=wx.CB_DROPDOWN | wx.CB_READONLY) self.plot = PlotROIMap(self.window_tree, roi_map) self.window_editor = wx.Panel(self.window, wx.ID_ANY, style=wx.BORDER_SUNKEN) self.combo_box_physician = wx.ComboBox(self.window_editor, wx.ID_ANY, choices=self.roi_map.get_physicians(), style=wx.CB_DROPDOWN | wx.CB_READONLY) self.combo_box_physician_roi = wx.ComboBox(self.window_editor, wx.ID_ANY, choices=[], style=wx.CB_DROPDOWN | wx.CB_READONLY) self.list_ctrl_variations = wx.ListCtrl(self.window_editor, wx.ID_ANY, style=wx.LC_NO_HEADER | wx.LC_REPORT | wx.BORDER_SUNKEN) self.button_variation_select_all = wx.Button(self.window_editor, wx.ID_ANY, "Select All") self.button_variation_deselect_all = wx.Button(self.window_editor, wx.ID_ANY, "Deselect All") self.button_variation_add = wx.Button(self.window_editor, wx.ID_ANY, "Add") self.button_variation_delete = wx.Button(self.window_editor, wx.ID_ANY, "Delete") self.button_variation_move = wx.Button(self.window_editor, wx.ID_ANY, "Move") self.button_variation_move.Disable() self.button_variation_delete.Disable() self.button_variation_deselect_all.Disable() self.button_physician = {'add': wx.Button(self.window_editor, wx.ID_ANY, "+"), 'del': wx.Button(self.window_editor, wx.ID_ANY, "-"), 'edit': wx.Button(self.window_editor, wx.ID_ANY, "Δ")} self.button_physician_roi = {'add': wx.Button(self.window_editor, wx.ID_ANY, "+"), 'del': wx.Button(self.window_editor, wx.ID_ANY, "-"), 'edit': wx.Button(self.window_editor, wx.ID_ANY, "Δ")} self.button_link_physician_roi = wx.Button(self.window_editor, wx.ID_ANY, "Link") self.button_link_physician_roi.Disable() self.combo_box_uncategorized_ignored = wx.ComboBox(self.window_editor, wx.ID_ANY, choices=["Uncategorized", "Ignored"], style=wx.CB_DROPDOWN | wx.CB_READONLY) self.combo_box_uncategorized_ignored_roi = wx.ComboBox(self.window_editor, wx.ID_ANY, choices=[], style=wx.CB_DROPDOWN) self.button_uncategorized_ignored_delete = wx.Button(self.window_editor, wx.ID_ANY, "Delete DVH") self.button_uncategorized_ignored_ignore = wx.Button(self.window_editor, wx.ID_ANY, "Ignore DVH") self.combo_box_physician_roi_merge = {'a': wx.ComboBox(self.window_editor, wx.ID_ANY, style=wx.CB_DROPDOWN), 'b': wx.ComboBox(self.window_editor, wx.ID_ANY, style=wx.CB_DROPDOWN)} self.button_merge = wx.Button(self.window_editor, wx.ID_ANY, "Merge") self.button_save_and_update = wx.Button(self.window_editor, wx.ID_ANY, "Save and Update Database") self.button_cancel = wx.Button(self.window_editor, wx.ID_ANY, "Cancel Changes and Reload") self.uncategorized_variations = {} self.columns = ['Variations'] self.data_table = DataTable(self.list_ctrl_variations, columns=self.columns, widths=[490]) self.__set_properties() self.__do_bind() self.__do_layout() self.plot.update_roi_map_source_data(self.physician) self.run() def __set_properties(self): self.combo_box_uncategorized_ignored.SetSelection(0) self.button_uncategorized_ignored_ignore.SetMinSize((110, 20)) self.combo_box_physician.SetValue('DEFAULT') self.update_physician_rois() self.update_variations() self.combo_box_physician.SetValue('DEFAULT') self.combo_box_tree_plot_data.SetValue('ALL') self.update_uncategorized_ignored_choices() self.window_tree.SetBackgroundColour('white') for button in self.button_physician.values(): button.SetMaxSize((25, 25)) for button in self.button_physician_roi.values(): button.SetMaxSize((25, 25)) self.update_physician_enable() self.update_merge_physician_rois() def __do_bind(self): self.window_tree.Bind(wx.EVT_COMBOBOX, self.on_plot_data_type_change, id=self.combo_box_tree_plot_data.GetId()) self.window_editor.Bind(wx.EVT_COMBOBOX, self.update_uncategorized_ignored_choices, id=self.combo_box_uncategorized_ignored.GetId()) self.window_editor.Bind(wx.EVT_COMBOBOX, self.physician_ticker, id=self.combo_box_physician.GetId()) self.window_editor.Bind(wx.EVT_COMBOBOX, self.physician_roi_ticker, id=self.combo_box_physician_roi.GetId()) self.window_editor.Bind(wx.EVT_COMBOBOX, self.uncategorized_ticker, id=self.combo_box_uncategorized_ignored.GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.add_physician, id=self.button_physician['add'].GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.on_delete_physician, id=self.button_physician['del'].GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.on_edit_physician, id=self.button_physician['edit'].GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.on_link_physician_roi, id=self.button_link_physician_roi.GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.add_physician_roi, id=self.button_physician_roi['add'].GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.on_delete_physician_roi, id=self.button_physician_roi['del'].GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.on_edit_physician_roi, id=self.button_physician_roi['edit'].GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.select_all_variations, id=self.button_variation_select_all.GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.deselect_all_variations, id=self.button_variation_deselect_all.GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.add_variation, id=self.button_variation_add.GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.move_variations, id=self.button_variation_move.GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.delete_variations, id=self.button_variation_delete.GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.on_delete_dvh, id=self.button_uncategorized_ignored_delete.GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.on_ignore_dvh, id=self.button_uncategorized_ignored_ignore.GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.on_merge, id=self.button_merge.GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.save_and_update, id=self.button_save_and_update.GetId()) self.window_editor.Bind(wx.EVT_BUTTON, self.on_cancel, id=self.button_cancel.GetId()) self.Bind(wx.EVT_CLOSE, self.on_close) self.window_editor.Bind(wx.EVT_COMBOBOX, self.update_merge_enable, id=self.combo_box_physician_roi_merge['a'].GetId()) self.window_editor.Bind(wx.EVT_COMBOBOX, self.update_merge_enable, id=self.combo_box_physician_roi_merge['b'].GetId()) self.window_editor.Bind(wx.EVT_LIST_ITEM_SELECTED, self.update_button_variation_enable, id=self.list_ctrl_variations.GetId()) self.window_editor.Bind(wx.EVT_LIST_ITEM_DESELECTED, self.update_button_variation_enable, id=self.list_ctrl_variations.GetId()) def __do_layout(self): sizer_wrapper = wx.BoxSizer(wx.HORIZONTAL) sizer_editor = wx.BoxSizer(wx.VERTICAL) sizer_physician_roi_merger = wx.BoxSizer(wx.HORIZONTAL) sizer_physician_roi_merger_merge = wx.BoxSizer(wx.VERTICAL) sizer_physician_roi_b = wx.BoxSizer(wx.VERTICAL) sizer_physician_roi_a = wx.BoxSizer(wx.VERTICAL) sizer_uncategorized_ignored = wx.StaticBoxSizer( wx.StaticBox(self.window_editor, wx.ID_ANY, "Uncategorized / Ignored"), wx.HORIZONTAL) sizer_uncategorized_ignored_ignore = wx.BoxSizer(wx.VERTICAL) sizer_uncategorized_ignored_delete = wx.BoxSizer(wx.VERTICAL) sizer_uncategorized_ignored_roi = wx.BoxSizer(wx.VERTICAL) sizer_uncategorized_ignored_type = wx.BoxSizer(wx.VERTICAL) sizer_tree = wx.BoxSizer(wx.VERTICAL) sizer_tree_input = wx.BoxSizer(wx.HORIZONTAL) sizer_tree_plot_data = wx.BoxSizer(wx.VERTICAL) sizer_roi_manager = wx.BoxSizer(wx.VERTICAL) sizer_variation_buttons = wx.BoxSizer(wx.VERTICAL) sizer_variation_table = wx.BoxSizer(wx.VERTICAL) sizer_map_editor = wx.StaticBoxSizer(wx.StaticBox(self.window_editor, wx.ID_ANY, "ROI Map Editor"), wx.VERTICAL) sizer_variations = wx.BoxSizer(wx.HORIZONTAL) sizer_physician_roi = wx.BoxSizer(wx.VERTICAL) sizer_physician = wx.BoxSizer(wx.VERTICAL) sizer_physician_row = wx.BoxSizer(wx.HORIZONTAL) sizer_physician_roi_row = wx.BoxSizer(wx.HORIZONTAL) sizer_save_cancel_buttons = wx.BoxSizer(wx.HORIZONTAL) label_physician = wx.StaticText(self.window_editor, wx.ID_ANY, "Physician:") sizer_physician.Add(label_physician, 0, 0, 0) sizer_physician_row.Add(self.combo_box_physician, 1, wx.EXPAND | wx.RIGHT, 5) sizer_physician_row.Add(self.button_physician['add'], 0, wx.LEFT | wx.RIGHT, 5) sizer_physician_row.Add(self.button_physician['del'], 0, wx.RIGHT, 5) sizer_physician_row.Add(self.button_physician['edit'], 0, wx.RIGHT, 10) sizer_physician.Add(sizer_physician_row, 1, wx.EXPAND, 0) self.label_physician_roi = wx.StaticText(self.window_editor, wx.ID_ANY, "Institutional ROI:") sizer_physician_roi.Add(self.label_physician_roi, 0, 0, 0) sizer_physician_roi_row.Add(self.combo_box_physician_roi, 1, wx.EXPAND | wx.RIGHT, 5) sizer_physician_roi_row.Add(self.button_link_physician_roi, 0, wx.EXPAND | wx.LEFT | wx.RIGHT, 5) sizer_physician_roi_row.Add(self.button_physician_roi['add'], 0, wx.LEFT | wx.RIGHT, 5) sizer_physician_roi_row.Add(self.button_physician_roi['del'], 0, wx.RIGHT, 5) sizer_physician_roi_row.Add(self.button_physician_roi['edit'], 0, wx.RIGHT, 10) sizer_physician_roi.Add(sizer_physician_roi_row, 0, wx.EXPAND, 0) sizer_map_editor.Add(sizer_physician, 0, wx.ALL | wx.EXPAND, 5) sizer_map_editor.Add(sizer_physician_roi, 0, wx.ALL | wx.EXPAND, 5) label_variations = wx.StaticText(self.window_editor, wx.ID_ANY, "Variations:") label_variations_buttons = wx.StaticText(self.window_editor, wx.ID_ANY, " ") sizer_variation_table.Add(label_variations, 0, 0, 0) sizer_variation_table.Add(self.list_ctrl_variations, 1, wx.BOTTOM | wx.EXPAND, 15) sizer_variations.Add(sizer_variation_table, 1, wx.EXPAND | wx.ALL, 5) sizer_variation_buttons.Add(label_variations_buttons, 0, 0, 0) sizer_variation_buttons.Add(self.button_variation_add, 0, wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM, 5) sizer_variation_buttons.Add(self.button_variation_delete, 0, wx.EXPAND | wx.ALL, 5) sizer_variation_buttons.Add(self.button_variation_move, 0, wx.EXPAND | wx.ALL, 5) sizer_variation_buttons.Add(self.button_variation_select_all, 0, wx.EXPAND | wx.ALL, 5) sizer_variation_buttons.Add(self.button_variation_deselect_all, 0, wx.EXPAND | wx.ALL, 5) sizer_variations.Add(sizer_variation_buttons, 0, wx.EXPAND | wx.ALL, 5) sizer_map_editor.Add(sizer_variations, 0, wx.EXPAND, 0) label_physician_roi_a = wx.StaticText(self.window_editor, wx.ID_ANY, "Merge Physician ROI A:") sizer_physician_roi_a.Add(label_physician_roi_a, 0, wx.EXPAND | wx.LEFT | wx.RIGHT | wx.TOP, 5) sizer_physician_roi_a.Add(self.combo_box_physician_roi_merge['a'], 1, wx.EXPAND | wx.BOTTOM | wx.LEFT | wx.RIGHT, 5) sizer_physician_roi_merger.Add(sizer_physician_roi_a, 1, wx.EXPAND, 0) label_physician_roi_b = wx.StaticText(self.window_editor, wx.ID_ANY, "Into Physician ROI B:") sizer_physician_roi_b.Add(label_physician_roi_b, 0, wx.EXPAND | wx.LEFT | wx.RIGHT | wx.TOP, 5) sizer_physician_roi_b.Add(self.combo_box_physician_roi_merge['b'], 1, wx.EXPAND | wx.BOTTOM | wx.LEFT | wx.RIGHT, 5) sizer_physician_roi_merger.Add(sizer_physician_roi_b, 1, wx.EXPAND, 0) sizer_physician_roi_merger_merge.Add((20, 16), 0, 0, 0) sizer_physician_roi_merger_merge.Add(self.button_merge, 0, wx.ALL, 5) sizer_physician_roi_merger.Add(sizer_physician_roi_merger_merge, 0, wx.ALL | wx.EXPAND, 0) sizer_map_editor.Add(sizer_physician_roi_merger, 0, wx.EXPAND, 0) sizer_save_cancel_buttons.Add(self.button_save_and_update, 1, wx.EXPAND | wx.LEFT | wx.RIGHT, 40) sizer_save_cancel_buttons.Add(self.button_cancel, 1, wx.EXPAND | wx.LEFT| wx.RIGHT, 40) sizer_map_editor.Add((10, 10), 0, 0, 0) sizer_map_editor.Add(sizer_save_cancel_buttons, 0, wx.EXPAND | wx.ALL, 10) sizer_roi_manager.Add(sizer_map_editor, 1, wx.EXPAND, 0) sizer_editor.Add(sizer_roi_manager, 0, wx.EXPAND | wx.ALL, 5) label_tree_plot_data = wx.StaticText(self.window_tree, wx.ID_ANY, 'Institutional Data to Display:') sizer_tree_plot_data.Add(label_tree_plot_data, 0, wx.LEFT | wx.RIGHT | wx.TOP, 5) sizer_tree_plot_data.Add(self.combo_box_tree_plot_data, 0, wx.EXPAND | wx.LEFT | wx.BOTTOM | wx.RIGHT, 5) sizer_tree_input.Add(sizer_tree_plot_data, 0, wx.EXPAND | wx.LEFT | wx.TOP, 5) sizer_tree.Add(sizer_tree_input, 0, wx.EXPAND, 0) sizer_tree.Add(self.plot.layout, 1, wx.EXPAND, 0) self.window_tree.SetSizer(sizer_tree) label_uncategorized_ignored = wx.StaticText(self.window_editor, wx.ID_ANY, "Type:") label_uncategorized_ignored.SetMinSize((38, 16)) sizer_uncategorized_ignored_type.Add(label_uncategorized_ignored, 0, wx.EXPAND | wx.LEFT | wx.RIGHT | wx.TOP, 5) sizer_uncategorized_ignored_type.Add(self.combo_box_uncategorized_ignored, 1, wx.EXPAND | wx.BOTTOM | wx.LEFT | wx.RIGHT, 5) sizer_uncategorized_ignored.Add(sizer_uncategorized_ignored_type, 1, wx.EXPAND, 0) label_uncategorized_ignored_roi = wx.StaticText(self.window_editor, wx.ID_ANY, "ROI:") label_uncategorized_ignored_roi.SetMinSize((30, 16)) sizer_uncategorized_ignored_roi.Add(label_uncategorized_ignored_roi, 0, wx.EXPAND | wx.LEFT | wx.RIGHT | wx.TOP, 5) sizer_uncategorized_ignored_roi.Add(self.combo_box_uncategorized_ignored_roi, 1, wx.EXPAND | wx.BOTTOM | wx.LEFT | wx.RIGHT, 5) sizer_uncategorized_ignored.Add(sizer_uncategorized_ignored_roi, 1, wx.EXPAND, 0) sizer_uncategorized_ignored_delete.Add((20, 16), 0, 0, 0) sizer_uncategorized_ignored_delete.Add(self.button_uncategorized_ignored_delete, 0, wx.ALL, 5) sizer_uncategorized_ignored.Add(sizer_uncategorized_ignored_delete, 0, wx.EXPAND, 0) sizer_uncategorized_ignored_ignore.Add((20, 16), 0, 0, 0) sizer_uncategorized_ignored_ignore.Add(self.button_uncategorized_ignored_ignore, 0, wx.ALL, 5) sizer_uncategorized_ignored.Add(sizer_uncategorized_ignored_ignore, 0, wx.EXPAND, 0) sizer_editor.Add(sizer_uncategorized_ignored, 0, wx.ALL | wx.EXPAND, 5) self.window_editor.SetSizer(sizer_editor) self.window.SplitVertically(self.window_tree, self.window_editor) self.window.SetSashPosition(int(self.window_size[0] * 0.55)) sizer_wrapper.Add(self.window, 1, wx.EXPAND, 0) self.SetSizer(sizer_wrapper) self.Layout() self.Centre() def run(self): self.Show() @staticmethod def update_combo_box_choices(combo_box, choices, value): if not value: value = combo_box.GetValue() combo_box.Clear() combo_box.AppendItems(choices) combo_box.SetValue(value) def update_roi_map(self): self.plot.update_roi_map_source_data(self.physician, plot_type=self.plot_data_type) @property def physician(self): return self.combo_box_physician.GetValue() @property def physician_roi(self): return self.combo_box_physician_roi.GetValue() @property def plot_data_type(self): return self.combo_box_tree_plot_data.GetValue() def physician_ticker(self, evt): self.update_physician_roi_label() self.update_physician_enable() self.update_all(skip_physicians=True) def on_plot_data_type_change(self, evt): self.update_roi_map() def update_uncategorized_ignored_choices(self, *args): ignored_variations = self.combo_box_uncategorized_ignored.GetValue() == 'Ignored' self.uncategorized_variations = self.get_uncategorized_variations(self.physician, ignored_variations=ignored_variations) choices = list(self.uncategorized_variations) choices.sort() if not choices: choices = ['None'] self.button_uncategorized_ignored_delete.Disable() self.button_uncategorized_ignored_ignore.Disable() else: self.button_uncategorized_ignored_delete.Enable() self.button_uncategorized_ignored_ignore.Enable() self.combo_box_uncategorized_ignored_roi.Clear() self.combo_box_uncategorized_ignored_roi.Append(choices) self.combo_box_uncategorized_ignored_roi.SetValue(choices[0]) @staticmethod def get_uncategorized_variations(physician, ignored_variations=False): if echo_sql_db(): with DVH_SQL() as cnx: physician = clean_name(physician).upper() condition = "physician_roi = '%s'" % ['uncategorized', 'ignored'][ignored_variations] cursor_rtn = cnx.query('dvhs', 'roi_name, study_instance_uid', condition) new_variations = {} for row in cursor_rtn: variation = clean_name(str(row[0])) study_instance_uid = str(row[1]) physician_db = cnx.get_unique_values('Plans', 'physician', "study_instance_uid = '%s'" % study_instance_uid) if physician_db and physician_db[0] == physician: if variation not in list(new_variations): new_variations[variation] = {'roi_name': variation, 'study_instance_uid': [study_instance_uid]} else: new_variations[variation]['study_instance_uid'].append(study_instance_uid) return new_variations @property def selected_indices(self): return get_selected_listctrl_items(self.list_ctrl_variations) def update_button_variation_enable(self, *args): if self.selected_indices: self.button_variation_move.Enable() self.button_variation_delete.Enable() self.button_variation_deselect_all.Enable() else: self.button_variation_move.Disable() self.button_variation_delete.Disable() self.button_variation_deselect_all.Disable() self.button_variation_select_all.Enable(self.variation_count > 0) def update_variations(self): self.data_table.set_data(self.variation_table_data, self.columns) self.update_button_variation_enable() def physician_roi_ticker(self, evt): self.update_variations() def update_physicians(self, old_physicians=None): choices = self.roi_map.get_physicians() new = choices[0] if old_physicians: new = list(set(choices) - set(old_physicians)) if new: new = clean_name(new[0]).upper() self.update_combo_box_choices(self.combo_box_physician, choices, new) self.update_physician_roi_label() self.update_physician_enable() def update_physician_rois(self, old_physician_rois=None): choices = self.roi_map.get_physician_rois(self.physician) new = choices[0] if old_physician_rois: new = list(set(choices) - set(old_physician_rois)) if new: new = clean_name(new[0]) self.update_combo_box_choices(self.combo_box_physician_roi, choices, new) @property def variations(self): variations = self.roi_map.get_variations(self.physician, self.physician_roi) variations = list(set(variations) - {self.physician_roi}) # remove physician roi variations.sort() return variations @property def variation_table_data(self): return {'Variations': self.variations} def add_physician_roi(self, evt): old_physician_rois = self.roi_map.get_physician_rois(self.physician) dlg = AddPhysicianROI(self, self.physician, self.roi_map, institutional_mode=self.physician == 'DEFAULT') if dlg.res == wx.ID_OK: self.update_all(old_physician_rois=old_physician_rois) def add_physician(self, evt): old_physicians = self.roi_map.get_physicians() dlg = AddPhysician(self.roi_map) if dlg.res == wx.ID_OK: self.update_all(old_physicians=old_physicians) @property def variation_count(self): return len(self.variations) @property def selected_values(self): return [self.list_ctrl_variations.GetItem(i, 0).GetText() for i in self.selected_indices] def select_all_variations(self, evt): self.apply_global_selection() def deselect_all_variations(self, evt): self.apply_global_selection(on=0) def apply_global_selection(self, on=1): for i in range(self.variation_count): self.list_ctrl_variations.Select(i, on=on) def delete_variations(self, evt): self.roi_map.delete_variations(self.physician, self.physician_roi, self.selected_values) self.update_variations() def add_variation(self, evt): dlg = AddVariationDialog(self, self.physician, self.physician_roi, self.roi_map) res = dlg.ShowModal() if res == wx.ID_OK: try: self.roi_map.add_variation(self.physician, self.physician_roi, dlg.text_ctrl_variation.GetValue()) self.update_variations() except ROIVariationError as e: ROIVariationErrorDialog(self, e) dlg.Destroy() def move_variations(self, evt): choices = [roi for roi in self.roi_map.get_physician_rois(self.physician) if roi != self.physician_roi] MoveVariationDialog(self, self.selected_values, self.physician, self.physician_roi, choices, self.roi_map) self.update_variations() def on_delete_physician(self, evt): MessageDialog(self, 'Delete Physician %s?' % self.physician, action_yes_func=self.delete_physician) def delete_physician(self): self.roi_map.delete_physician(self.physician) self.update_all() def on_delete_physician_roi(self, evt): if self.physician == 'DEFAULT': MessageDialog(self, "Delete Institutional ROI %s?" % self.physician_roi, action_yes_func=self.delete_institutional_roi) else: MessageDialog(self, "Delete Physician ROI %s?" % self.physician_roi, action_yes_func=self.delete_physician_roi) def delete_physician_roi(self): self.roi_map.delete_physician_roi(self.physician, self.physician_roi) self.update_all(skip_physicians=True) def delete_institutional_roi(self): self.roi_map.delete_institutional_roi(self.physician_roi) self.update_all(skip_physicians=True) def on_delete_dvh(self, evt): MessageDialog(self, "Delete all DVHs named %s for %s?" % (self.dvh, self.physician), message="Are you sure? This cannot be undone!", action_yes_func=self.delete_dvh) def delete_dvh(self): with DVH_SQL() as cnx: for uid in self.dvh_uids: cnx.delete_dvh(self.dvh, uid) self.update_uncategorized_ignored_choices() def on_ignore_dvh(self, evt): msg_type = ['Unignore', 'Ignore'][self.button_uncategorized_ignored_ignore.GetLabelText() == 'Ignore DVH'] MessageDialog(self, "%s all DVHs named %s for %s?" % (msg_type, self.dvh, self.physician), action_yes_func=self.ignore_dvh) def ignore_dvh(self): unignore = self.button_uncategorized_ignored_ignore.GetLabelText() == 'Unignore DVH' with DVH_SQL() as cnx: for uid in self.dvh_uids: cnx.ignore_dvh(self.dvh, uid, unignore=unignore) self.update_uncategorized_ignored_choices() @property def dvh(self): return self.combo_box_uncategorized_ignored_roi.GetValue() @property def dvh_uids(self): return self.uncategorized_variations[self.dvh]['study_instance_uid'] def update_all(self, old_physicians=None, old_physician_rois=None, skip_physicians=False): if not skip_physicians: self.update_physicians(old_physicians=old_physicians) self.update_physician_rois(old_physician_rois=old_physician_rois) self.update_variations() self.update_uncategorized_ignored_choices() self.update_merge_physician_rois() self.update_roi_map() def on_edit_physician(self, evt): current_physicians = self.roi_map.get_physicians() dlg = RenamePhysicianDialog(self.physician, self.roi_map) if dlg.res == wx.ID_OK: self.update_all(old_physicians=current_physicians) def on_edit_physician_roi(self, evt): current_physician_rois = self.roi_map.get_physician_rois(self.physician) if self.physician == 'DEFAULT': dlg = RenameInstitutionalROIDialog(self.physician_roi, self.roi_map) else: dlg = RenamePhysicianROIDialog(self.physician, self.physician_roi, self.roi_map) if dlg.res == wx.ID_OK: self.update_all(old_physician_rois=current_physician_rois, skip_physicians=True) def update_physician_enable(self): self.button_physician['del'].Enable(self.physician != 'DEFAULT') self.button_physician['edit'].Enable(self.physician != 'DEFAULT') self.button_variation_add.Enable(self.physician != 'DEFAULT') def update_physician_roi_label(self): label_text = ['Physician ROI:', 'Institutional ROI:'][self.physician == 'DEFAULT'] self.label_physician_roi.SetLabelText(label_text) self.button_link_physician_roi.Enable(self.physician != 'DEFAULT') def uncategorized_ticker(self, evt): if self.combo_box_uncategorized_ignored.GetValue() == 'Uncategorized': self.button_uncategorized_ignored_ignore.SetLabelText('Ignore DVH') else: self.button_uncategorized_ignored_ignore.SetLabelText('Unignore DVH') self.update_uncategorized_ignored_choices() def update_merge_physician_rois(self): options = [] if self.physician != 'DEFAULT': options = self.roi_map.get_physician_rois(self.physician) if not options: options = [''] for combo_box in self.combo_box_physician_roi_merge.values(): combo_box.Clear() combo_box.Append(options) combo_box.SetValue(options[0]) self.update_merge_enable() @property def merge_a(self): return self.combo_box_physician_roi_merge['a'].GetValue() @property def merge_b(self): return self.combo_box_physician_roi_merge['b'].GetValue() def on_merge(self, evt): self.roi_map.merge_physician_rois(self.physician, [self.merge_a, self.merge_b], self.merge_b) self.update_all(skip_physicians=True) def update_merge_enable(self, *args): # *args to catch wx.EVT_BUTTON self.combo_box_physician_roi_merge['a'].Enable(bool(self.merge_a and self.merge_b)) self.combo_box_physician_roi_merge['b'].Enable(bool(self.merge_a and self.merge_b)) self.button_merge.Enable(bool(self.merge_a) and bool(self.merge_b) and self.merge_a != self.merge_b) def on_link_physician_roi(self, evt): dlg = LinkPhysicianROI(self, self.physician, self.physician_roi, self.roi_map) if dlg.res == wx.ID_OK: self.update_roi_map() def save_and_update(self, evt): RemapROIFrame(self.roi_map) self.Destroy() def on_cancel(self, *args): self.roi_map.import_from_file() self.update_roi_map() def on_close(self, *args): self.Destroy() self.roi_map.import_from_file()
class QueriedDataFrame(wx.Frame): """ Generate a simple table to view data of the current query for a specified SQL table """ def __init__(self, data_obj, sql_table, menu, menu_item_id): """ :param data_obj: object containing data to be viewed :param sql_table: either 'DVHs', 'Plans', 'Beams', or 'Rxs' :type sql_table: str :param menu: a link to the main app menu, used to toggle Show/Hide status :type menu: Menu :param menu_item_id: the ID of the menu item associated with the specified data_obj """ wx.Frame.__init__(self, None, title='%s Data' % sql_table[0:-1]) self.data = data_obj self.sql_table = sql_table self.menu = menu self.menu_item_id = menu_item_id self.list_ctrl = wx.ListCtrl(self, wx.ID_ANY, style=wx.BORDER_SUNKEN | wx.LC_HRULES | wx.LC_REPORT | wx.LC_VRULES) # self.data_table = DataTable(self.list_ctrl, data=self.table_data, columns=self.columns) self.data_table = DataTable(self.list_ctrl) self.data_table.set_data(self.table_data, self.columns) self.button_export = wx.Button(self, wx.ID_ANY, "Export to CSV") self.__do_bind() self.__set_properties() self.__do_layout() self.run() def __set_properties(self): self.SetSize(get_window_size(0.714, 0.762)) def __do_bind(self): self.Bind(wx.EVT_BUTTON, self.on_export, id=self.button_export.GetId()) self.Bind(wx.EVT_LIST_COL_CLICK, self.sort_table, self.list_ctrl) self.Bind(wx.EVT_CLOSE, self.on_close) def __do_layout(self): sizer_wrapper = wx.BoxSizer(wx.VERTICAL) sizer_wrapper.Add(self.button_export, 0, wx.ALL, 10) sizer_wrapper.Add(self.list_ctrl, 1, wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.EXPAND, 10) self.SetSizer(sizer_wrapper) self.Center() @property def columns(self): if self.sql_table == 'DVHs': columns = self.data.keys elif self.sql_table == 'Rxs': columns = [ 'plan_name', 'fx_dose', 'rx_percent', 'fxs', 'rx_dose', 'fx_grp_number', 'fx_grp_count', 'fx_grp_name', 'normalization_method', 'normalization_object' ] else: columns = [ obj['var_name'] for obj in sql_column_info.values() if obj['table'] == self.sql_table ] for starter_column in ['study_instance_uid', 'mrn']: if starter_column in columns: columns.pop(columns.index(starter_column)) columns.insert(0, starter_column) return columns @property def table_data(self): return {column: getattr(self.data, column) for column in self.columns} def run(self): self.toggle_data_menu_item() self.Show() def on_close(self, *args): self.toggle_data_menu_item() self.Destroy() def on_export(self, *args): save_data_to_file(self, "Export %s to CSV" % self.sql_table, self.data_table.get_csv()) def toggle_data_menu_item(self): short_cut = ['DVHs', 'Plans', 'Rxs', 'Beams'].index(self.sql_table) + 1 show_hide = ['Show', 'Hide']['Show' in self.menu.GetLabel(self.menu_item_id)] self.menu.SetLabel( self.menu_item_id, '%s %s\tCtrl+%s' % (show_hide, self.sql_table, short_cut)) def sort_table(self, evt): self.data_table.sort_table(evt)
class EndpointFrame: """ Object to be passed into notebook panel for the Endpoint tab """ def __init__(self, parent, dvh, time_series, regression, control_chart): """ :param parent: notebook panel in main view :type parent: Panel :param dvh: dvh data object :type dvh: DVH :param time_series: Time Series object in notebook :type time_series: TimeSeriesFrame :param regression: Regression frame object in notebook :type regression: RegressionFrame :param control_chart: Control Chart frame object in notebook :type control_chart: ControlChartFrame """ self.parent = parent self.dvh = dvh self.time_series = time_series self.regression = regression self.control_chart = control_chart self.initial_columns = ['MRN', 'Tx Site', 'ROI Name', 'Volume (cc)'] self.widths = [150, 150, 250, 100] self.button = {'add': wx.Button(self.parent, wx.ID_ANY, "Add Endpoint"), 'del': wx.Button(self.parent, wx.ID_ANY, "Delete Endpoint"), 'exp': wx.Button(self.parent, wx.ID_ANY, 'Export')} self.table = wx.ListCtrl(self.parent, wx.ID_ANY, style=wx.BORDER_SUNKEN | wx.LC_HRULES | wx.LC_REPORT | wx.LC_VRULES) self.table.SetMinSize(get_window_size(0.623, 0.762)) self.data_table = DataTable(self.table) self.endpoint_defs = DataTable(None, columns=['label', 'output_type', 'input_type', 'input_value', 'units_in', 'units_out']) if dvh: self.dvh.endpoints['data'] = self.data_table.data self.dvh.endpoints['defs'] = self.endpoint_defs.data self.__do_bind() self.__set_properties() self.__do_layout() self.disable_buttons() def __do_bind(self): self.parent.Bind(wx.EVT_BUTTON, self.add_ep_button_click, id=self.button['add'].GetId()) self.parent.Bind(wx.EVT_BUTTON, self.del_ep_button_click, id=self.button['del'].GetId()) self.parent.Bind(wx.EVT_BUTTON, self.on_export_csv, id=self.button['exp'].GetId()) def __set_properties(self): for i, column in enumerate(self.initial_columns): self.table.AppendColumn(column, format=wx.LIST_FORMAT_LEFT, width=self.widths[i]) def __do_layout(self): sizer_wrapper = wx.BoxSizer(wx.VERTICAL) vbox = wx.BoxSizer(wx.VERTICAL) hbox = wx.BoxSizer(wx.HORIZONTAL) for key in list(self.button): hbox.Add(self.button[key], 0, wx.ALL, 5) vbox.Add(hbox, 0, wx.ALL | wx.EXPAND, 5) vbox.Add(self.table, 1, wx.EXPAND, 0) sizer_wrapper.Add(vbox, 1, wx.ALL | wx.EXPAND, 20) self.layout = sizer_wrapper def calculate_endpoints(self): columns = [c for c in self.initial_columns] if self.data_table.data: current_labels = [key for key in list(self.data_table.data) if key not in columns] else: current_labels = [] ep = {'MRN': self.dvh.mrn, 'Tx Site': self.dvh.get_plan_values('tx_site'), 'ROI Name': self.dvh.roi_name, 'Volume (cc)': self.dvh.volume} ep_defs = self.endpoint_defs.data if ep_defs: for i, ep_name in enumerate(ep_defs['label']): if ep_name not in columns: columns.append(ep_name) if ep_name in current_labels: ep[ep_name] = deepcopy(self.data_table.data[ep_name]) else: endpoint_input = ep_defs['input_type'][i] endpoint_output = ep_defs['output_type'][i] x = float(ep_defs['input_value'][i]) if endpoint_input == 'relative': x /= 100. if 'V' in ep_name: ep[ep_name] = self.dvh.get_volume_of_dose(x, volume_scale=endpoint_output, dose_scale=endpoint_input) else: ep[ep_name] = self.dvh.get_dose_to_volume(x, dose_scale=endpoint_output, volume_scale=endpoint_input) self.data_table.set_data(ep, columns) self.data_table.set_column_width(0, 150) self.data_table.set_column_width(1, 150) self.data_table.set_column_width(2, 200) def add_ep_button_click(self, evt): dlg = AddEndpointDialog() res = dlg.ShowModal() if res == wx.ID_OK and dlg.is_endpoint_valid: self.endpoint_defs.append_row(dlg.endpoint_row) self.calculate_endpoints() self.enable_buttons() self.update_endpoints_in_dvh() self.time_series.update_y_axis_options() dlg.Destroy() self.regression.stats_data.update_endpoints_and_radbio() self.regression.update_combo_box_choices() self.control_chart.update_combo_box_y_choices() def del_ep_button_click(self, evt): dlg = DelEndpointDialog(self.data_table.columns) res = dlg.ShowModal() if res == wx.ID_OK: for value in dlg.selected_values: self.data_table.delete_column(value) endpoint_def_row = self.endpoint_defs.data['label'].index(value) self.update_endpoints_in_dvh() self.endpoint_defs.delete_row(endpoint_def_row) self.time_series.update_y_axis_options() dlg.Destroy() self.regression.stats_data.update_endpoints_and_radbio() self.regression.update_combo_box_choices() self.control_chart.update_combo_box_y_choices() if self.data_table.column_count == 3: self.button['del'].Disable() self.button['exp'].Disable() def update_dvh(self, dvh): self.dvh = dvh self.update_endpoints_in_dvh() def update_endpoints_in_dvh(self): if self.dvh: self.dvh.endpoints['data'] = self.data_table.data self.dvh.endpoints['defs'] = self.endpoint_defs.data def clear_data(self): self.data_table.delete_all_rows() self.endpoint_defs.delete_all_rows(force_delete_data=True) # no attached layout, force delete if self.data_table.data: for column in list(self.data_table.data): if column not in self.initial_columns: self.data_table.delete_column(column) def enable_buttons(self): for key in list(self.button): self.button[key].Enable() def disable_buttons(self): for key in list(self.button): self.button[key].Disable() def enable_initial_buttons(self): self.button['add'].Enable() def get_csv(self, selection=None): uid = {1: {'title': 'Study Instance UID', 'data': self.dvh.uid}} return self.data_table.get_csv(extra_column_data=uid) def on_export_csv(self, evt): save_data_to_file(self.parent, "Export Endpoints to CSV", self.get_csv()) def get_save_data(self): return deepcopy({'data_table': self.data_table.get_save_data(), 'endpoint_defs': self.endpoint_defs.get_save_data()}) def load_save_data(self, save_data): self.data_table.load_save_data(save_data['data_table'], ignore_layout=True) self.endpoint_defs.load_save_data(save_data['endpoint_defs'], ignore_layout=True) self.calculate_endpoints() @property def has_data(self): if self.endpoint_defs.data and self.endpoint_defs.data['label']: return True return False