source: sasview/src/sas/sasview/sasview.py @ 8faef63

magnetic_scattrelease-4.2.2ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249unittest-saveload
Last change on this file since 8faef63 was 8faef63, checked in by wojciech, 6 years ago

Moved dialog to SasView? class to avoid calling wx.App twice

  • Property mode set to 100644
File size: 11.9 KB
Line 
1# -*- coding: utf-8 -*-
2"""
3Base module for loading and running the main SasView application.
4"""
5################################################################################
6# This software was developed by the University of Tennessee as part of the
7# Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
8# project funded by the US National Science Foundation.
9#
10# See the license text in license.txt
11#
12# copyright 2009, University of Tennessee
13################################################################################
14import os
15import os.path
16import sys
17import traceback
18import logging
19
20try:
21    reload(sys)
22    sys.setdefaultencoding("iso-8859-1")
23except NameError:
24    # On python 3 sys.setdefaultencoding does nothing, so pass.
25    # We know we are in python 3 at this point since reload is no longer in
26    # builtins, but instead has been moved to importlib, hence the NameError.
27    pass
28
29import sas
30
31APP_NAME = 'SasView'
32PLUGIN_MODEL_DIR = 'plugin_models'
33
34class SasView(object):
35    """
36    Main class for running the SasView application
37    """
38    def __init__(self):
39        """
40        """
41        logger = logging.getLogger(__name__)
42
43        from sas.sasgui.guiframe.gui_manager import SasViewApp
44        self.gui = SasViewApp(0)
45        if sys.platform == "darwin":
46            self.check_sasmodels_compiler()
47        # Set the application manager for the GUI
48        self.gui.set_manager(self)
49        # Add perspectives to the basic application
50        # Additional perspectives can still be loaded
51        # dynamically
52        # Note: py2exe can't find dynamically loaded
53        # modules. We load the fitting module here
54        # to ensure a complete Windows executable build.
55
56        # Rebuild .sasview/categories.json.  This triggers a load of sasmodels
57        # and all the plugins.
58        try:
59            from sas.sascalc.fit.models import ModelManager
60            from sas.sasgui.guiframe.CategoryInstaller import CategoryInstaller
61            model_list = ModelManager().cat_model_list()
62            CategoryInstaller.check_install(model_list=model_list)
63        except Exception:
64            logger.error("%s: could not load SasView models")
65            logger.error(traceback.format_exc())
66
67        # Fitting perspective
68        try:
69            import sas.sasgui.perspectives.fitting as module
70            fitting_plug = module.Plugin()
71            self.gui.add_perspective(fitting_plug)
72        except Exception:
73            logger.error("%s: could not find Fitting plug-in module", APP_NAME)
74            logger.error(traceback.format_exc())
75
76        # P(r) perspective
77        try:
78            import sas.sasgui.perspectives.pr as module
79            pr_plug = module.Plugin()
80            self.gui.add_perspective(pr_plug)
81        except Exception:
82            logger.error("%s: could not find P(r) plug-in module", APP_NAME)
83            logger.error(traceback.format_exc())
84
85        # Invariant perspective
86        try:
87            import sas.sasgui.perspectives.invariant as module
88            invariant_plug = module.Plugin()
89            self.gui.add_perspective(invariant_plug)
90        except Exception:
91            logger.error("%s: could not find Invariant plug-in module",
92                         APP_NAME)
93            logger.error(traceback.format_exc())
94
95        # Corfunc perspective
96        try:
97            import sas.sasgui.perspectives.corfunc as module
98            corfunc_plug = module.Plugin()
99            self.gui.add_perspective(corfunc_plug)
100        except Exception:
101            logger.error("Unable to load corfunc module")
102
103        # Calculator perspective
104        try:
105            import sas.sasgui.perspectives.calculator as module
106            calculator_plug = module.Plugin()
107            self.gui.add_perspective(calculator_plug)
108        except Exception:
109            logger.error("%s: could not find Calculator plug-in module",
110                         APP_NAME)
111            logger.error(traceback.format_exc())
112
113        # File converter tool
114        try:
115            import sas.sasgui.perspectives.file_converter as module
116            converter_plug = module.Plugin()
117            self.gui.add_perspective(converter_plug)
118        except Exception:
119            logger.error("%s: could not find File Converter plug-in module",
120                         APP_NAME)
121            logger.error(traceback.format_exc())
122
123        # Add welcome page
124        from .welcome_panel import WelcomePanel
125        self.gui.set_welcome_panel(WelcomePanel)
126
127        # Build the GUI
128        self.gui.build_gui()
129        # delete unused model folder
130        self.gui.clean_plugin_models(PLUGIN_MODEL_DIR)
131        # Start the main loop
132        self.gui.MainLoop()
133
134    def check_sasmodels_compiler(self):
135        """
136        Checking c compiler for sasmodels and raises xcode command line
137        tools for installation
138        """
139        import wx
140        import subprocess
141        #Generic message box created becuase standard MessageBox is not moveable
142        class GenericMessageBox(wx.Dialog):
143            def __init__(self, parent, text, title = ''):
144
145                wx.Dialog.__init__(self, parent, -1, title = title,
146                               size = (360,180), pos=(20,60),
147                               style = wx.STAY_ON_TOP | wx.DEFAULT_DIALOG_STYLE)
148                panel = wx.Panel(self, -1)
149                top_row_sizer = wx.BoxSizer(wx.HORIZONTAL)
150
151                error_bitmap = wx.ArtProvider.GetBitmap(
152                    wx.ART_ERROR, wx.ART_MESSAGE_BOX
153                )
154                error_bitmap_ctrl = wx.StaticBitmap(panel, -1)
155                error_bitmap_ctrl.SetBitmap(error_bitmap)
156                label = wx.StaticText(panel, -1, text)
157                top_row_sizer.Add(error_bitmap_ctrl, flag=wx.ALL, border=10)
158                top_row_sizer.Add(label, flag=wx.ALIGN_CENTER_VERTICAL)
159
160                #Create the OK button in the bottom row.
161                #ok_button = wx.Button(panel, wx.ID_OK )
162                ok_button = wx.Button(panel, wx.ID_OK )
163                self.Bind(wx.EVT_BUTTON, self.on_ok, source=ok_button)
164                ok_button.SetFocus()
165                ok_button.SetDefault()
166
167                sizer = wx.BoxSizer(wx.VERTICAL)
168                sizer.Add(top_row_sizer)
169                sizer.Add(ok_button, flag=wx.ALIGN_CENTER | wx.ALL, border=5)
170                panel.SetSizer(sizer)
171
172            def on_ok(self, event):
173                self.Destroy()
174        logger = logging.getLogger(__name__)
175        try:
176            subprocess.check_output(["cc","--version"], stderr=subprocess.STDOUT)
177            dlg = GenericMessageBox(parent=None,
178            text='No compiler installed. Please follow instruction for\n'
179                'command line developers tools installation and restart SasView\n\n'
180                'Alternatively one can use OpenCL compiler,\n'
181                 'which can be setup from menu Fitting->OpenCL Options\n\n',
182            title = 'Compiler Info')
183            dlg.Show(1)
184
185        except subprocess.CalledProcessError as exc:
186
187            dlg = GenericMessageBox(parent=None,
188            text='No compiler installed. Please follow instruction for\n'
189                'command line developers tools installation and restart SasView\n\n'
190                'Alternatively one can use OpenCL compiler,\n'
191                 'which can be setup from menu Fitting->OpenCL Options\n\n',
192            title = 'Compiler Info')
193            dlg.ShowModal()
194
195            logger.error("No compiler installed. %s\n"%(exc))
196            logger.error(traceback.format_exc())
197
198        #raise RuntimeError("Terminating sasview")
199
200def setup_logging():
201    from sas.logger_config import SetupLogger
202
203    logger = SetupLogger(__name__).config_production()
204    # Log the start of the session
205    logger.info(" --- SasView session started ---")
206    # Log the python version
207    logger.info("Python: %s" % sys.version)
208    return logger
209
210
211def setup_wx():
212    # Allow the dynamic selection of wxPython via an environment variable, when devs
213    # who have multiple versions of the module installed want to pick between them.
214    # This variable does not have to be set of course, and through normal usage will
215    # probably not be, but this can make things a little easier when upgrading to a
216    # new version of wx.
217    logger = logging.getLogger(__name__)
218    WX_ENV_VAR = "SASVIEW_WX_VERSION"
219    if WX_ENV_VAR in os.environ:
220        logger.info("You have set the %s environment variable to %s.",
221                    WX_ENV_VAR, os.environ[WX_ENV_VAR])
222        import wxversion
223        if wxversion.checkInstalled(os.environ[WX_ENV_VAR]):
224            logger.info("Version %s of wxPython is installed, so using that version.",
225                        os.environ[WX_ENV_VAR])
226            wxversion.select(os.environ[WX_ENV_VAR])
227        else:
228            logger.error("Version %s of wxPython is not installed, so using default version.",
229                         os.environ[WX_ENV_VAR])
230    else:
231        logger.info("You have not set the %s environment variable, so using default version of wxPython.",
232                    WX_ENV_VAR)
233
234    import wx
235
236    try:
237        logger.info("Wx version: %s", wx.__version__)
238    except AttributeError:
239        logger.error("Wx version: error reading version")
240
241    from . import wxcruft
242    wxcruft.call_later_fix()
243    #wxcruft.trace_new_id()
244    #Always use private .matplotlib setup to avoid conflicts with other
245    #uses of matplotlib
246
247
248def setup_mpl(backend=None):
249    # Always use private .matplotlib setup to avoid conflicts with other
250    mplconfigdir = os.path.join(sas.get_user_dir(), '.matplotlib')
251    if not os.path.exists(mplconfigdir):
252        os.mkdir(mplconfigdir)
253    os.environ['MPLCONFIGDIR'] = mplconfigdir
254    # Set backend to WXAgg; this overrides matplotlibrc, but shouldn't override
255    # mpl.use().  Note: Don't import matplotlib here since the script that
256    # we are running may not actually need it; also, putting as little on the
257    # path as we can
258    if backend:
259        os.environ['MPLBACKEND'] = backend
260
261    # TODO: ... so much for not importing matplotlib unless we need it...
262    from matplotlib import backend_bases
263    backend_bases.FigureCanvasBase.filetypes.pop('pgf', None)
264
265def setup_sasmodels():
266    """
267    Prepare sasmodels for running within sasview.
268    """
269    # Set SAS_MODELPATH so sasmodels can find our custom models
270    plugin_dir = os.path.join(sas.get_user_dir(), PLUGIN_MODEL_DIR)
271    os.environ['SAS_MODELPATH'] = plugin_dir
272    #Initialize environment variable with custom setting but only if variable not set
273    SAS_OPENCL = sas.get_custom_config().SAS_OPENCL
274    if SAS_OPENCL and "SAS_OPENCL" not in os.environ:
275        os.environ["SAS_OPENCL"] = SAS_OPENCL
276
277def run_gui():
278    """
279    __main__ method for loading and running SasView
280    """
281    from multiprocessing import freeze_support
282    freeze_support()
283    setup_logging()
284    setup_mpl(backend='WXAgg')
285    setup_sasmodels()
286    setup_wx()
287    SasView()
288
289
290def run_cli():
291    from multiprocessing import freeze_support
292    freeze_support()
293    setup_logging()
294    # Use default matplotlib backend on mac/linux, but wx on windows.
295    # The problem on mac is that the wx backend requires pythonw.  On windows
296    # we are sure to wx since it is the shipped with the app.
297    setup_mpl(backend='WXAgg' if os.name == 'nt' else None)
298    setup_sasmodels()
299    if len(sys.argv) == 1 or sys.argv[1] == '-i':
300        # Run sasview as an interactive python interpreter
301        try:
302            from IPython import start_ipython
303            sys.argv = ["ipython", "--pylab"]
304            sys.exit(start_ipython())
305        except ImportError:
306            import code
307            code.interact(local={'exit': sys.exit})
308    elif sys.argv[1] == '-c':
309        exec(sys.argv[2])
310    else:
311        thing_to_run = sys.argv[1]
312        sys.argv = sys.argv[1:]
313        import runpy
314        if os.path.exists(thing_to_run):
315            runpy.run_path(thing_to_run, run_name="__main__")
316        else:
317            runpy.run_module(thing_to_run, run_name="__main__")
318
319
320if __name__ == "__main__":
321    run_gui()
Note: See TracBrowser for help on using the repository browser.