def InitCanvas(self): global db, algorithmName print "before call" self.sessionController = PreventionSessionController(db, PreventionSession, None, None) self.experimentController = ExperimentController(db, Experiment, None, None) experiment = self.experimentController.getDefaultExperiment(algorithmName) self.experiment_id = experiment.id self.depth_canvas = PreventionDepthCameraCanvas (self.depth_video_panel, 40, "PUDepthCanvas", db, experiment.id) w = self.depth_canvas.GetWindow() #w.SetMinSize(wx.Size(640, 480)) depthPanelSizer = wx.BoxSizer(wx.VERTICAL) depthPanelSizer.Add(w, 0, wx.EXPAND, 0, self.depth_canvas) self.depth_video_panel.SetSizer(depthPanelSizer) depthPanelSizer.Layout() self.Layout() #wx.Sleep(2) #print "Canvas ready? %s" % self.depth_canvas.canvas_.is_ready() #print "after call" while w: w.Layout() w.Refresh() w = w.GetParent()
class MainFrame(wx.Frame): """ This class implements the main window (or frame) for the Prevention System GUI. It defines the window, the menubars, the tabbed notebook panes, and a status line. For the Prevention GUI the notebook panes consist of a collection pane and a clinical pane. The collection pane will contain a GLCanvas which will display the video from the depth camera. Methods: __init__(*args, **kwds) - creates the widgets in the frame and performs initialization __set_properties() - set various properties of the widgets __do_layout() - lays out the widgets doNewPatient(event) - Menu handler for adding new patients doSelectPatient(event) - Menu handler for selecting a patient to record """ def __init__(self, *args, **kwds): """ Creates the widgets in the frame and performs initialization. Also, creates and attaches an OpenGL canvas for the camera process to use. """ # begin wxGlade: MainFrame.__init__ kwds["style"] = wx.DEFAULT_FRAME_STYLE wx.Frame.__init__(self, *args, **kwds) # Menu Bar self.main_frame_menubar = wx.MenuBar() self.mainFileMenu = wx.Menu() self.exitMenuItem = wx.MenuItem(self.mainFileMenu, wx.ID_ANY, _("Exit"), "", wx.ITEM_NORMAL) self.mainFileMenu.AppendItem(self.exitMenuItem) self.main_frame_menubar.Append(self.mainFileMenu, _("File")) self.mainPatientMenu = wx.Menu() self.newPatientMenuItem = wx.MenuItem(self.mainPatientMenu, wx.ID_ANY, _("New/Modify..."), "", wx.ITEM_NORMAL) self.mainPatientMenu.AppendItem(self.newPatientMenuItem) self.selectPatientMenuItem = wx.MenuItem(self.mainPatientMenu, wx.ID_ANY, _("Select..."), "", wx.ITEM_NORMAL) self.mainPatientMenu.AppendItem(self.selectPatientMenuItem) self.main_frame_menubar.Append(self.mainPatientMenu, _("Patient")) self.mainAlgorithmMenu = wx.Menu() self.newAlgorithmItem = wx.MenuItem(self.mainAlgorithmMenu, wx.ID_ANY, _("New/Modify/Delete Algorithm..."), "", wx.ITEM_NORMAL) self.mainAlgorithmMenu.AppendItem(self.newAlgorithmItem) self.newExperimentItem = wx.MenuItem(self.mainAlgorithmMenu, wx.ID_ANY, _("New/Modify/Delete Experiment..."), "", wx.ITEM_NORMAL) self.mainAlgorithmMenu.AppendItem(self.newExperimentItem) self.main_frame_menubar.Append(self.mainAlgorithmMenu, _("Algorithm")) self.mainAdminMenu = wx.Menu() self.changePasswordItem = wx.MenuItem(self.mainAdminMenu, wx.ID_ANY, _("Change Password..."), "", wx.ITEM_NORMAL) self.mainAdminMenu.AppendItem(self.changePasswordItem) self.manageUsersItem = wx.MenuItem(self.mainAdminMenu, wx.ID_ANY, _("Manage Users..."), "", wx.ITEM_NORMAL) self.mainAdminMenu.AppendItem(self.manageUsersItem) self.accessControlItem = wx.MenuItem(self.mainAdminMenu, wx.ID_ANY, _("Access Control..."), "", wx.ITEM_NORMAL) self.mainAdminMenu.AppendItem(self.accessControlItem) self.newSystemConfiguration = wx.MenuItem(self.mainAdminMenu, wx.ID_ANY, _("System Configuration..."), "", wx.ITEM_NORMAL) self.mainAdminMenu.AppendItem(self.newSystemConfiguration) self.main_frame_menubar.Append(self.mainAdminMenu, _("Admin")) self.mainHelpMenu = wx.Menu() self.main_frame_menubar.Append(self.mainHelpMenu, _("Help")) self.SetMenuBar(self.main_frame_menubar) # Menu Bar end self.main_frame_statusbar = self.CreateStatusBar(1, 0) self.panel_1 = wx.Panel(self, wx.ID_ANY) self.prevention_notebook = wx.Notebook(self.panel_1, wx.ID_ANY, style=0) self.prevention_collection_pane = wx.Panel(self.prevention_notebook, wx.ID_ANY) self.depth_video_panel = wx.Panel(self.prevention_collection_pane, wx.ID_ANY) self.label_1 = wx.StaticText(self.prevention_collection_pane, wx.ID_ANY, _("Patient"), style=wx.ALIGN_CENTRE) self.patient_name = wx.TextCtrl(self.prevention_collection_pane, wx.ID_ANY, _("None"), style=wx.TE_READONLY) self.start_button = wx.Button(self.prevention_collection_pane, wx.ID_ANY, _("Start Recording")) self.stop_button = wx.Button(self.prevention_collection_pane, wx.ID_ANY, _("Stop Recording")) self.olv = FastObjectListView(self.prevention_collection_pane, wx.ID_ANY, style=wx.LC_REPORT|wx.SUNKEN_BORDER) self.prevention_clinical_pane = wx.Panel(self.prevention_notebook, wx.ID_ANY) self.clinical_label_1 = wx.StaticText(self.prevention_clinical_pane, wx.ID_ANY, _("Prevention Care Reporting"), style=wx.ALIGN_CENTRE) self.clinical_container_panel = wx.Panel(self.prevention_clinical_pane, wx.ID_ANY) self.__set_properties() self.__do_layout() self.Bind(wx.EVT_MENU, self.OnExit, self.exitMenuItem) self.Bind(wx.EVT_MENU, self.DoAlgorithmList, self.newAlgorithmItem) self.Bind(wx.EVT_MENU, self.DoExperimentList, self.newExperimentItem) self.Bind(wx.EVT_MENU, self.DoChangePassword, self.changePasswordItem) self.Bind(wx.EVT_MENU, self.DoConfigList, self.newSystemConfiguration) self.Bind(wx.EVT_BUTTON, self.doStartRecording, self.start_button) self.Bind(wx.EVT_BUTTON, self.DoStopRecording, self.stop_button) # end wxGlade self.currentPatient = None self.Bind(wx.EVT_CLOSE, self.OnExit) self.Bind(wx.EVT_MENU, self.doNewPatient, self.newPatientMenuItem) self.Bind(wx.EVT_MENU, self.doSelectPatient, self.selectPatientMenuItem) self.start_button.Disable() self.stop_button.Disable() # Wrap the ObjectListView in a batch updater # self.olv = BatchedUpdate(self.olv, 10) self.olv.SetEmptyListMsg("No Turning Events Found") self.olv.SetColumns(OlvPatientTurningCols().getColumnDefinitions()) # ---------- # ---------- def __set_properties(self): """ Sets various properties of the widgets """ # begin wxGlade: MainFrame.__set_properties self.SetTitle(_("Multi-Modality Portable System for Pressure Ulcer Prevention")) self.main_frame_statusbar.SetStatusWidths([-1]) # statusbar fields main_frame_statusbar_fields = [_("frame_1_statusbar")] for i in range(len(main_frame_statusbar_fields)): self.main_frame_statusbar.SetStatusText(main_frame_statusbar_fields[i], i) self.depth_video_panel.SetMinSize((640, 480)) self.start_button.SetMinSize((100, 23)) self.stop_button.SetMinSize((100, 23)) self.prevention_collection_pane.SetFocus() self.clinical_label_1.SetFont(wx.Font(14, wx.DEFAULT, wx.NORMAL, wx.BOLD, 0, "")) self.prevention_notebook.SetMinSize((844, 1001)) # end wxGlade def __do_layout(self): """ Lays out the widgets in the frame """ # begin wxGlade: MainFrame.__do_layout sizer_1 = wx.BoxSizer(wx.VERTICAL) sizer_2 = wx.BoxSizer(wx.VERTICAL) clinical_pane_sizer = wx.BoxSizer(wx.VERTICAL) collection_sizer_1 = wx.BoxSizer(wx.VERTICAL) sizer_6 = wx.BoxSizer(wx.HORIZONTAL) sizer_3 = wx.BoxSizer(wx.VERTICAL) sizer_6.Add(self.depth_video_panel, 0, wx.EXPAND, 0) sizer_3.Add((20, 20), 0, 0, 0) sizer_3.Add(self.label_1, 0, 0, 0) sizer_3.Add(self.patient_name, 0, 0, 0) sizer_3.Add((20, 40), 0, 0, 0) sizer_3.Add(self.start_button, 0, 0, 2) sizer_3.Add((20, 40), 0, 0, 0) sizer_3.Add(self.stop_button, 0, 0, 2) sizer_6.Add(sizer_3, 1, wx.LEFT | wx.EXPAND, 20) collection_sizer_1.Add(sizer_6, 0, wx.LEFT | wx.TOP | wx.EXPAND, 10) collection_sizer_1.Add((20, 20), 0, 0, 0) collection_sizer_1.Add(self.olv, 1, wx.EXPAND, 0) self.prevention_collection_pane.SetSizer(collection_sizer_1) clinical_pane_sizer.Add(self.clinical_label_1, 0, wx.ALIGN_CENTER_HORIZONTAL, 0) clinical_pane_sizer.Add((20, 20), 0, 0, 0) clinical_pane_sizer.Add(self.clinical_container_panel, 1, wx.EXPAND, 0) self.prevention_clinical_pane.SetSizer(clinical_pane_sizer) self.prevention_notebook.AddPage(self.prevention_collection_pane, _("Collection")) self.prevention_notebook.AddPage(self.prevention_clinical_pane, _("Clinical")) sizer_2.Add(self.prevention_notebook, 1, wx.EXPAND, 0) self.panel_1.SetSizer(sizer_2) sizer_1.Add(self.panel_1, 1000, wx.EXPAND, 0) self.SetSizer(sizer_1) sizer_1.Fit(self) self.Layout() # end wxGlade self.clinical_pane_sizer = clinical_pane_sizer def InitCanvas(self): global db, algorithmName print "before call" self.sessionController = PreventionSessionController(db, PreventionSession, None, None) self.experimentController = ExperimentController(db, Experiment, None, None) experiment = self.experimentController.getDefaultExperiment(algorithmName) self.experiment_id = experiment.id self.depth_canvas = PreventionDepthCameraCanvas (self.depth_video_panel, 40, "PUDepthCanvas", db, experiment.id) w = self.depth_canvas.GetWindow() #w.SetMinSize(wx.Size(640, 480)) depthPanelSizer = wx.BoxSizer(wx.VERTICAL) depthPanelSizer.Add(w, 0, wx.EXPAND, 0, self.depth_canvas) self.depth_video_panel.SetSizer(depthPanelSizer) depthPanelSizer.Layout() self.Layout() #wx.Sleep(2) #print "Canvas ready? %s" % self.depth_canvas.canvas_.is_ready() #print "after call" while w: w.Layout() w.Refresh() w = w.GetParent() def InitMisc(self): """ Method to initialize various misc. items """ global db # Set up file director self.director = PUFileDirector(self.system_config['BASE_FILE_DIRECTORY']) self.director.SetSystemType("prevention") # Set up database writer self.preventionDatabaseWriter = PreventionDatabaseWriter(db) def InitClinicalPane(self): global db self.prevention_clinical_panel = PreventionClinicalDataDialog(self.clinical_container_panel, db) cpsizer = wx.BoxSizer(wx.VERTICAL) cpsizer.Add(self.prevention_clinical_panel, 1, wx.EXPAND, 0) self.clinical_container_panel.SetSizer(cpsizer) cpsizer.Fit(self.clinical_container_panel) self.clinical_pane_sizer.RecalcSizes() def doNewPatient(self, event): """ Menu handler for adding new patients """ global db self.panel_1.Disable() olvDialog = OLVDialog(None, db, PatientIdentification, OlvPatientIdentification, OlvPatientIdentificationCols) rc = olvDialog.ShowModal() self.panel_1.Enable() def doSelectPatient(self, event): """ Menu handler for selecting a patient to record """ global db self.panel_1.Disable() olvDialog = OLVDialog(None, db, PatientIdentification, OlvPatientIdentification, OlvPatientIdentificationCols,"Select-Only") rc = olvDialog.ShowModal() if rc == 0: self.currentPatient = olvDialog.getSelectedObject() self.patient_id = self.currentPatient.patient_id self.patient_name.SetValue(self.currentPatient.patient_name) self.director.SetPatientId(self.patient_id) self.preventionDatabaseWriter.setPatientId(self.patient_id) self.start_button.Enable() self.prevention_clinical_panel.setPatient(self.currentPatient.patient_id, self.currentPatient.patient_name) self.prevention_clinical_panel.showAllRecords() self.panel_1.Enable() def doStartRecording(self, event): # wxGlade: MainFrame.<event_handler> """ Button handler for starting the recording of a patient TBD """ # print "doStartRecording event handler" # self.depth_canvas.post_redraw_event() global recording self.SendSizeEvent() self.session = PreventionSession() self.session.patient_id = self.patient_id self.session.start_time = datetime.today() self.session.end_time = self.session.start_time self.session.depth_video_file_directory = "" (rc,msg) = self.sessionController.add(self.session) if rc != 0: self.SetStatusText(msg) return self.stop_button.Enable() self.start_button.Disable() self.director.SetSessionId(self.session.id) self.preventionDatabaseWriter.setSessionId(self.session.id) self.preventionDatabaseWriter.setExperimentId(self.experiment_id) self.session.depth_video_file_directory = self.director.GetDepthFileDirectory() try: os.makedirs(self.session.depth_video_file_directory) except OSError as exc: # Python >2.5 if exc.errno == errno.EEXIST and os.path.isdir(self.session.depth_video_file_directory): pass else: # raise msg = "Unable to access directory: " + str(self.session.depth_video_file_directory) + ". Please insure that the drive is attached and unlocked." rc = util.showMessageDialog(msg, "Drive unavailable") self.start_button.Enable() self.stop_button.Disable() (rc,msg) = self.sessionController.deleteRecord(self.session.id) if rc != 0: self.SetStatusText(msg) return freespace = util.GetFreeSpaceGB(self.director.GetDepthFileDirectory()) if freespace < int(self.system_config['PREVENTION_FILE_SIZE_GB']): msg = "Not enough space to record patient. Need " + str(self.system_config['PREVENTION_FILE_SIZE_GB']) + " GB but found " + str(freespace) + " GB." rc = util.showMessageDialog(msg, "Out of space") (rc,msg) = self.sessionController.deleteRecord(self.session.id) if rc != 0: self.SetStatusText(msg) return self.depth_canvas.proc_.set_database_writer(self.preventionDatabaseWriter) self.depth_canvas.proc_.start_recording(str(self.session.depth_video_file_directory)) recording = True # patientTurningController = PatientTurningController(db, PatientTurning, OlvPatientTurning, OlvPatientTurningCols) # self.results = patientTurningController.getByPatientBySessionByExperiment(patientId, sessionId, experimentId) # self.olv.SetObjects(self.results) self.CreateUpdatingList() def CreateUpdatingList(self): self.olv = BatchedUpdate(self.olv, 10) self.timerThread = IntervalTimer(5, self.UpdateList) self.timerThread.start() def UpdateList(self): print "patientId=%d, sessionId=%d, experimentId=%d" % (self.patient_id, self.session.id, self.experiment_id) patientTurningController = PatientTurningController(db, PatientTurning, OlvPatientTurning, OlvPatientTurningCols) # self.results = patientTurningController.getByPatientBySessionByExperiment(self.patient_id, self.session.id, self.experiment_id) self.results = patientTurningController.getByPatientBySessionByExperiment(self.patient_id, 1, self.experiment_id) wx.CallAfter(self.olv.SetObjects,self.results) def DoStopRecording(self, event): # wxGlade: MainFrame.<event_handler> """ Button handler stopping recording """ global recording if recording: recording = False self.session.end_time = datetime.today() (rc,msg) = self.sessionController.update(self.session) if rc != 0: self.SetStatusText(msg) return self.depth_canvas.proc_.stop_recording() self.stop_button.Disable() self.start_button.Enable() if self.timerThread: print "stopping thread" self.timerThread.stop() def OnExit(self, event): # wxGlade: MainFrame.<event_handler> """ Menu handler for exiting """ evt = wx.PyCommandEvent(wx.EVT_BUTTON.typeId,self.stop_button.GetId()) wx.PostEvent(self, evt) sys.exit() def DoAlgorithmList(self, event): # wxGlade: MainFrame.<event_handler> """ Menu handler for adding, modifying, deleting algorithms and algorithm defaults """ global db self.panel_1.Disable() olvDialog = AlgorithmDialog(None, db) rc = olvDialog.ShowModal() olvDialog.Destroy() # self.RefreshTabs() self.panel_1.Enable() def DoExperimentList(self, event): # wxGlade: MainFrame.<event_handler> """ Menu handler for adding, modifying, deleting experiments and experiment configurations """ global db self.panel_1.Disable() olvDialog = ExperimentDialog(None, db) rc = olvDialog.ShowModal() olvDialog.Destroy() self.panel_1.Enable() def DoConfigList(self, event): # wxGlade: MainFrame.<event_handler> """ Menu handler for adding, editing, deleting system configuration parameters """ global db self.panel_1.Disable() olvDialog = OLVDialog(None, db, SystemConfiguration, OlvSystemConfiguration, OlvSystemConfigurationCols) rc = olvDialog.ShowModal() olvDialog.Destroy() self.panel_1.Enable() def DoChangePassword(self, event): # wxGlade: MainFrame.<event_handler> """ Menu handler for adding, editing, deleting system configuration parameters """ global db self.panel_1.Disable() changePasswordDialog = ChangePasswordDialog(None) changePasswordDialog.SetDb(db) rc = changePasswordDialog.ShowModal() # print "rc=%d" % rc if rc == 1: self.SetStatusText("Password successfully changed") changePasswordDialog.Destroy() self.panel_1.Enable()