Changeset 278cc25 in sasview for guiframe/local_perspectives/plotting

Timestamp:

May 27, 2008 2:32:45 PM (17 years ago)

Author:

Mathieu Doucet <doucetm@…>

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:

4a5de6f

Parents:

14b8154

Message:

Allow adding base perspectives to an app (as opposed of only dynamically finding them out).
Modified plot label update

File:

• guiframe/local_perspectives/plotting/plotting.py (modified) (4 diffs)

guiframe/local_perspectives/plotting/plotting.py

@@ -52 +52 @@
         ## Unique ID (from gui_manager)
         self.uid = None
+        
+        ## Action IDs for internal call-backs
+        self.action_ids = {}
         
         ## Graph        
@@ -83 +86 @@
         # Check axis labels
         #TODO: Should re-factor this
-        if event.plot._xunit != self.graph.prop["xunit"]:
-            self.graph.xaxis(event.plot._xaxis, event.plot._xunit)
-            
-        if event.plot._yunit != self.graph.prop["yunit"]:
-            self.graph.yaxis(event.plot._yaxis, event.plot._yunit)
+        #if event.plot._xunit != self.graph.prop["xunit"]:
+        self.graph.xaxis(event.plot._xaxis, event.plot._xunit)
+            
+        #if event.plot._yunit != self.graph.prop["yunit"]:
+        self.graph.yaxis(event.plot._yaxis, event.plot._yunit)
       
         self.graph.render(self)
@@ -100 +103 @@
         slicerpop = PanelMenu()
         slicerpop.set_plots(self.plots)
+                
+        # Option to save the data displayed
+        id = wx.NewId()
+        for plot in self.graph.plottables:
+            name = plot.name
+            slicerpop.Append(id, "&Save %s points" % name)
+            self.action_ids[str(id)] = plot
+            wx.EVT_MENU(self, id, self._onSave)
                 
         # Various plot options
@@ -127 +138 @@
         pos = self.ScreenToClient(pos)
         self.PopupMenu(slicerpop, pos)
+    
+    def _onSave(self, evt):
+        """
+            Save a data set to a text file
+            @param evt: Menu event
+        """
+        import os
+        id = str(evt.GetId())
+        if id in self.action_ids:         
+            
+            path = None
+            dlg = wx.FileDialog(self, "Choose a file", os.getcwd(), "", "*.txt", wx.SAVE)
+            if dlg.ShowModal() == wx.ID_OK:
+                path = dlg.GetPath()
+                mypath = os.path.basename(path)
+                print path
+            dlg.Destroy()
+            
+            if not path == None:
+                out = open(path, 'w')
+                has_errors = True
+                if self.action_ids[id].dy==None or self.action_ids[id].dy==[]:
+                    has_errors = False
+                    
+                # Sanity check
+                if has_errors:
+                    try:
+                        if len(self.action_ids[id].y) != len(self.action_ids[id].dy):
+                            print "Y and dY have different lengths"
+                            has_errors = False
+                    except:
+                        has_errors = False
+                
+                if has_errors:
+                    out.write("<X>   <Y>   <dY>\n")
+                else:
+                    out.write("<X>   <Y>\n")
+                    
+                for i in range(len(self.action_ids[id].x)):
+                    if has_errors:
+                        out.write("%g  %g  %g\n" % (self.action_ids[id].x[i], 
+                                                    self.action_ids[id].y[i],
+                                                    self.action_ids[id].dy[i]))
+                    else:
+                        out.write("%g  %g\n" % (self.action_ids[id].x[i], 
+                                                self.action_ids[id].y[i]))
+                        
+                out.close()
+    
     
     def _onToggleScale(self, event):