Changeset 25594ca in sasview for src/sas/sasgui/perspectives/fitting/gpu_options.py

Timestamp:

Jan 2, 2017 7:18:34 AM (7 years ago)

Author:

wojciech

Children:

2a1b92eb

Parents:

b39debba

Message:

Added message dialog for testing results

File:

• src/sas/sasgui/perspectives/fitting/gpu_options.py (modified) (7 diffs)

src/sas/sasgui/perspectives/fitting/gpu_options.py

@@ -14 +14 @@
 from sas.sasgui.guiframe.documentation_window import DocumentationWindow
 import imp
+
+
+class CustomMessageBox(wx.Dialog):
+    def __init__(self, parent, msg, title):
+        wx.Dialog.__init__(self, parent, title=title)
+        text = wx.TextCtrl(self, style=wx.TE_READONLY|wx.BORDER_NONE)
+        text.SetValue(msg)
+        text.SetBackgroundColour(self.GetBackgroundColour())
+        self.ShowModal()
+        self.Destroy()
+
 
 class GpuOptions(wx.Dialog):
@@ -137 +148 @@
         return clinfo
 
-    def write_to_config_file(self):
+
+    def _write_to_config_file(self):
         from sas.sasgui.guiframe.customdir import SetupCustom
-        from sas.sasgui.guiframe.gui_manager import get_app_dir, _find_local_config
-
-        PATH_APP = get_app_dir()
-        c_conf_dir = SetupCustom().setup_dir(PATH_APP)
+        from sas.sasgui.guiframe.gui_manager import _find_local_config
+
+        c_conf_dir = SetupCustom().find_dir()
         self.custom_config = _find_local_config('custom_config', c_conf_dir)
         if self.custom_config is None:
             self.custom_config = _find_local_config('custom_config', os.getcwd())
-            if self.custom_config is None:
-                msgConfig = "Custom_config file was not imported"
-                logging.info(msgConfig)
-            else:
-                logging.info("using custom_config in %s" % os.getcwd())
-        else:
-            logging.info("using custom_config from %s" % c_conf_dir)
 
         #How to store it in file
-        self.custom_config.SAS_OPENCL = self.sas_opencl
+        new_config_lines = []
+        config_file = open(self.custom_config.__file__[:-1])
+        if self.custom_config is not None:
+            config_lines = config_file.readlines()
+            for line in config_lines:
+                if "SAS_OPENCL" in line:
+                    if self.sas_opencl:
+                        new_config_lines.append("SAS_OPENCL = \""+self.sas_opencl+"\"")
+                    else:
+                        new_config_lines.append("SAS_OPENCL = None")
+                else:
+                    new_config_lines.append(line)
+        config_file.close()
+        new_config_file = open(self.custom_config.__file__[:-1],"w")
+        new_config_file.writelines(new_config_lines)
+        new_config_file.close()
+
+        #After file is touched module needs to be reloaded
+        self.custom_config = _find_local_config('custom_config', c_conf_dir)
+        if self.custom_config is None:
+            self.custom_config = _find_local_config('custom_config', os.getcwd())
 
     def on_check(self, event):
@@ -183 +207 @@
             if "SAS_OPENCL" in os.environ:
                 del(os.environ["SAS_OPENCL"])
-
         try:
             imp.find_module('sasmodels.kernelcl')
@@ -191 +214 @@
         if kernelcl_exist:
             sasmodels.kernelcl.ENV = None
-            #self.custom_config.SAS_OPENCL = self.sas_opencl
-
+
+        self._write_to_config_file()
         #Need to reload sasmodels.core module to account SAS_OPENCL = "None"
         reload(sasmodels.core)
@@ -201 +224 @@
         Close window on accpetance
         """
+        import sasmodels
         for btn in self.buttons:
             btn.SetValue(0)
+
         self.sas_opencl=None
+        del(os.environ["SAS_OPENCL"])
+
+        try:
+            imp.find_module('sasmodels.kernelcl')
+            kernelcl_exist = True
+        except:
+            kernelcl_exist = False
+        if kernelcl_exist:
+            sasmodels.kernelcl.ENV = None
+
+        self._write_to_config_file()
+        reload(sasmodels.core)
+        event.Skip()
 
     def on_test(self, event):
@@ -211 +249 @@
         import json
         import platform
-
         import sasmodels
         from sasmodels.model_test import model_tests
+
         try:
             from sasmodels.kernelcl import environment
@@ -239 +277 @@
             'failing tests': failures,
         }
-        print(json.dumps(info['failing tests']))
+
+        msg_info = "OpenCL tests"
+        msg = "OpenCL test completed!\n"
+        msg += "Sasmodels version: "
+        msg += info['version']+"\n"
+        msg += "Platform used: "
+        msg += json.dumps(info['platform'])+"\n"
+        msg += "OpenCL driver: "
+        msg += json.dumps(info['opencl'])+"\n"
+        if len(failures) > 0:
+            msg += 'Failing tests:\n'
+            msg += json.dumps(info['failing tests'])
+        else:
+            msg+="All tests passed\n"
+
+        CustomMessageBox(self.panel1, msg, msg_info)
 
     def on_help(self, event):