Changeset 96d293da in sasview for src/sas/sasgui/perspectives/corfunc/corfunc_panel.py

Timestamp:

Aug 24, 2016 11:15:28 AM (8 years ago)

Author:

lewis

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, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

d45d590

Parents:

5a525e1

git-author:

Lewis O'Driscoll <lewis.o'driscoll@…> (08/24/16 10:59:41)

git-committer:

Lewis O'Driscoll <lewis.o'driscoll@…> (08/24/16 11:15:28)

Message:

Implement save analysis in corfunc

File:

• src/sas/sasgui/perspectives/corfunc/corfunc_panel.py (modified) (2 diffs)

src/sas/sasgui/perspectives/corfunc/corfunc_panel.py

@@ -1 +1 @@
 import wx
 import sys
+import os
 import numpy as np
 from wx.lib.scrolledpanel import ScrolledPanel
 from sas.sasgui.guiframe.events import PlotQrangeEvent
 from sas.sasgui.guiframe.events import StatusEvent
+from sas.sasgui.guiframe.events import PanelOnFocusEvent
 from sas.sasgui.guiframe.panel_base import PanelBase
 from sas.sasgui.guiframe.utils import check_float
@@ -299 +301 @@
         doc_viewer = DocumentationWindow(self, -1, tree_location, "",
                                           "Correlation Function Help")
+
+    def get_save_flag(self):
+        if self._data is not None:
+            return True
+        return False
+
+    def on_set_focus(self, event=None):
+        if self._manager.parent is not None:
+            wx.PostEvent(self._manager.parent, PanelOnFocusEvent(panel=self))
+
+    def on_save(self, event=None):
+        """
+        Save corfunc state into a file
+        """
+        # Ask the user the location of the file to write to.
+        path = None
+        default_save_location = os.getcwd()
+        if self._manager.parent != None:
+            default_save_location = self._manager.parent.get_save_location()
+
+        dlg = wx.FileDialog(self, "Choose a file",
+                            default_save_location, \
+                            self.window_caption, "*.cor", wx.SAVE)
+        if dlg.ShowModal() == wx.ID_OK:
+            path = dlg.GetPath()
+            default_save_location = os.path.dirname(path)
+            if self._manager.parent != None:
+                self._manager.parent._default_save_location = default_save_location
+        else:
+            return None
+
+        dlg.Destroy()
+        # MAC always needs the extension for saving
+        extens = ".cor"
+        # Make sure the ext included in the file name
+        f_name = os.path.splitext(path)[0] + extens
+        self._manager.state_reader.write(f_name, self._data, self.get_state())
 
     def save_project(self, doc=None):