Changeset ed8ad21 in sasview for guiframe/local_perspectives/plotting/Plotter2D.py

Timestamp:

Jan 31, 2011 9:51:23 AM (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:

d0ecb36

Parents:

ec02ddd

Message:

added 2D save as a file in context menu

File:

• guiframe/local_perspectives/plotting/Plotter2D.py (modified) (3 diffs)

guiframe/local_perspectives/plotting/Plotter2D.py

@@ -13 +13 @@
 import wx
 import sys
+import os
 import math
 import pylab
@@ -240 +241 @@
         wx.EVT_MENU(self, id, self.onPrinterPreview)
         
+        # saving data
+        plot = self.data2D
+        id = wx.NewId()
+        name = plot.name
+        slicerpop.Append(id, "&Save as a file (DAT)" )
+        self.action_ids[str(id)] = plot
+        wx.EVT_MENU(self, id, self._onSave)
+
         slicerpop.AppendSeparator()
         if len(self.data2D.detector) == 1:        
@@ -554 +563 @@
             event = self._getEmptySlicerEvent()
             wx.PostEvent(self, event)
-   
+            
+    def _onSave(self, evt):
+        """
+        Save a data set to a dat(text) file
+        
+        :param evt: Menu event
+        
+        """
+        id = str(evt.GetId())
+        if id in self.action_ids:         
+            
+            path = None
+            wildcard = "IGOR/DAT 2D file in Q_map (*.dat)|*.DAT"
+            dlg = wx.FileDialog(self, "Choose a file",
+                                self._default_save_location,
+                                 "", wildcard , wx.SAVE)
+           
+            if dlg.ShowModal() == wx.ID_OK:
+                path = dlg.GetPath()
+                mypath = os.path.basename(path)
+                
+                #TODO: This is bad design. The DataLoader is designed 
+                #to recognize extensions.
+                # It should be a simple matter of calling the .
+                #save(file, data, '.xml') method
+                # of the DataLoader.loader.Loader class.
+                from DataLoader.loader import  Loader
+                #Instantiate a loader 
+                loader = Loader() 
+                data = self.data2D
+
+                format = ".dat"
+                if os.path.splitext(mypath)[1].lower() == format:
+                    loader.save(path, data, format)
+                try:
+                    self._default_save_location = os.path.dirname(path)
+                except:
+                    pass    
+            dlg.Destroy()