Changeset b35d3d1 in sasview for guiframe

Timestamp:

Aug 6, 2010 7:36:07 PM (14 years ago)

Author:

Jae Cho <jhjcho@…>

Branches:

master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, release_4.0.1, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

028a0e8

Parents:

a93f525

Message:

save state file: working, but more work to do

Location:

guiframe

Files:

• data_loader.py (modified) (3 diffs)
• gui_manager.py (modified) (2 diffs)

guiframe/data_loader.py

@@ -150 +150 @@
                                                 type="stop"))
     
-def plot_data(parent, path):
+def plot_data(parent, path, format=None):
     """
     Use the DataLoader loader to created data to plot.
     
     :param path: the path of the data to load
-    
+    : param format: file format (as file extension)
     """
     from sans.guicomm.events import NewPlotEvent, StatusEvent
@@ -165 +165 @@
     # Load data 
     try:
-        output = L.load(path)
+        output = L.load(path, format)
     except:
         load_error(sys.exc_value)
@@ -235 +235 @@
         i=1
         for item in output:
-            msg = "Loading 1D data: %s"%str(item.run[0])
-            wx.PostEvent(parent, StatusEvent(status=msg, info="info", type="stop"))
             try:
-                dx = item.dx
-                dxl = item.dxl
-                dxw = item.dxw
+                msg = "Loading 1D data: %s"%str(item.run[0])
+                wx.PostEvent(parent, StatusEvent(status=msg, info="info", type="stop"))
+                try:
+                    dx = item.dx
+                    dxl = item.dxl
+                    dxw = item.dxw
+                except:
+                    dx = None
+                    dxl = None
+                    dxw = None
+    
+                new_plot = Data1D(x=item.x,y=item.y,dx=dx,dy=item.dy)
+                new_plot.copy_from_datainfo(item)
+                item.clone_without_data(clone=new_plot)
+                new_plot.dxl = dxl
+                new_plot.dxw = dxw
+                
+                name = parse_name(name=str(item.run[0]), expression="_")
+                #if not name in parent.indice_load_data.keys():
+                #    parent.indice_load_data[name] = 0
+                #else:
+                    ## create a copy of the loaded data
+                    
+                    #TODO: this is a very annoying feature. We should make this
+                    # an option. Excel doesn't do this. Why should we?
+                    # What is the requirement for this feature, and are the
+                    # counter arguments stronger? Is this feature developed
+                    # to please at least 80% of the users or a special few?
+                    #parent.indice_load_data[name] += 1
+                    #name = name + "(copy %i)"%parent.indice_load_data[name]
+                    
+                new_plot.name = name
+                new_plot.interactive = True
+                new_plot.group_id = name
+                new_plot.id = name
+                new_plot.is_data = True
+            
+                if hasattr(item,"title"):
+                    title = item.title.lstrip().rstrip()
+                    if title == "":
+                        title = str(name)
+                else:
+                    title = name
+                if hasattr(parent, "panel_on_focus") and not(parent.panel_on_focus is None):
+                    existing_panel  = parent.panel_on_focus
+                    panel_name = existing_panel.window_caption
+                    data_name = new_plot.name
+                    if enable_add_data(existing_panel, new_plot):
+                        if open_dialog_append_data(panel_name, data_name):
+                            #add this plot the an existing panel
+                            new_plot.group_id = existing_panel.group_id
+                wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title=str(title)))
+                i+=1
             except:
-                dx = None
-                dxl = None
-                dxw = None
-
-            new_plot = Data1D(x=item.x,y=item.y,dx=dx,dy=item.dy)
-            new_plot.copy_from_datainfo(item)
-            item.clone_without_data(clone=new_plot)
-            new_plot.dxl = dxl
-            new_plot.dxw = dxw
-           
-            name = parse_name(name=str(item.run[0]), expression="_")
-            #if not name in parent.indice_load_data.keys():
-            #    parent.indice_load_data[name] = 0
-            #else:
-                ## create a copy of the loaded data
-                
-                #TODO: this is a very annoying feature. We should make this
-                # an option. Excel doesn't do this. Why should we?
-                # What is the requirement for this feature, and are the
-                # counter arguments stronger? Is this feature developed
-                # to please at least 80% of the users or a special few?
-                #parent.indice_load_data[name] += 1
-                #name = name + "(copy %i)"%parent.indice_load_data[name]
-                
-            new_plot.name = name
-            new_plot.interactive = True
-            new_plot.group_id = name
-            new_plot.id = name
-            new_plot.is_data = True
-        
-            if hasattr(item,"title"):
-                title = item.title.lstrip().rstrip()
-                if title == "":
-                    title = str(name)
-            else:
-                title = name
-            if hasattr(parent, "panel_on_focus") and not(parent.panel_on_focus is None):
-                existing_panel  = parent.panel_on_focus
-                panel_name = existing_panel.window_caption
-                data_name = new_plot.name
-                if enable_add_data(existing_panel, new_plot):
-                    if open_dialog_append_data(panel_name, data_name):
-                        #add this plot the an existing panel
-                        new_plot.group_id = existing_panel.group_id
-            wx.PostEvent(parent, NewPlotEvent(plot=new_plot, title=str(title)))
-            i+=1
-         
+                raise

guiframe/gui_manager.py

@@ -537 +537 @@
         
         id = wx.NewId()
-        self.filemenu.Append(id, '&Load Data', 'Load data file into the application')
+        self.filemenu.Append(id, '&Open', 'Load data file into the application')
         wx.EVT_MENU(self, id, self._on_open)
+        #self.filemenu.AppendSeparator()
+        
+        id = wx.NewId()
+        self.filemenu.Append(id, '&Save', 'Save project as a SanaView (svs) file')
+        wx.EVT_MENU(self, id, self._on_save)
         #self.filemenu.AppendSeparator()
         
@@ -711 +716 @@
         from data_loader import plot_data
         if path and os.path.isfile(path):
-            plot_data(self, path)
+             basename  = os.path.basename(path)
+             if  basename.endswith('.svs'):
+                 plot_data(self, path,'.inv')
+                 plot_data(self, path,'.prv')
+                 plot_data(self, path,'.fitv')
+             else: plot_data(self,path)
         if self.defaultPanel is not None and \
             self._mgr.GetPane(self.panels["default"].window_name).IsShown():
             self.on_close_welcome_panel()
+            
+    def _on_save(self, event):
+        """
+        Save state into a file
+        """
+        # Ask the user the location of the file to write to.
+        
+        ## Default file location for save
+        self._default_save_location = os.getcwd()
+        path = None
+        dlg = wx.FileDialog(self, "Choose a file", self._default_save_location, "", "*.svs", wx.SAVE)
+        if dlg.ShowModal() == wx.ID_OK:
+            path = dlg.GetPath()
+            self._default_save_location = os.path.dirname(path)
+        else:
+            return None
+        
+        dlg.Destroy()
+        if path is None:
+            return
+        # default cansas xml doc
+        doc = None
+        for item in self.panels:
+            try:
+                if self.panels[item].window_name == 'Invariant' or self.panels[item].window_name == 'pr_control' or self.panels[item].window_name == 'Fit panel':
+                    if self.panels[item].window_name == 'Invariant':
+                        data = self.panels[item]._data
+                        state = self.panels[item].state
+                        new_doc =self.panels[item]._manager.state_reader.write_toXML(data,state)
+                        if hasattr(doc, "firstChild"):
+                            doc.firstChild.appendChild (new_doc.firstChild.firstChild)  
+                        else:
+                            doc = new_doc 
+                    elif self.panels[item].window_name == 'pr_control':
+                        data = self.panels[item].manager.current_plottable
+                        state = self.panels[item].get_state()
+                        new_doc =self.panels[item].manager.state_reader.write_toXML(data,state)
+                        if hasattr(doc, "firstChild"):
+                            doc.firstChild.appendChild (new_doc.firstChild.firstChild)  
+                        else:
+                            doc = new_doc 
+                    elif self.panels[item].window_name == 'Fit panel':
+                        for index in range(self.panels[item].GetPageCount()):
+                            selected_page = self.panels[item].GetPage(index) 
+                            if hasattr(selected_page,"get_data"):
+                                data = selected_page.get_data()
+                                state = selected_page.state
+                                new_doc =selected_page.manager.state_reader.write_toXML(data,state)
+                                if doc != None and hasattr(doc, "firstChild"):
+                                    doc.firstChild.appendChild (new_doc.firstChild.firstChild)
+                                else:
+                                    doc = new_doc
+
+            except: 
+                pass
+
+        # Write the XML document
+        if doc != None:
+            fd = open(path, 'w')
+            fd.write(doc.toprettyxml())
+            fd.close()
+        else:
+            raise RuntimeError, "%s is not a SansView (.svs) file..." % path
+
        
     def _onClose(self, event):