Source code for sas.sasview.sasview

# -*- coding: utf-8 -*-
"""
Base module for loading and running the main SasView application.
"""
################################################################################
# This software was developed by the University of Tennessee as part of the
# Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
# project funded by the US National Science Foundation.
#
# See the license text in license.txt
#
# copyright 2009, University of Tennessee
################################################################################
import os
import os.path
import sys
import traceback
import logging

try:
    reload(sys)
    sys.setdefaultencoding("iso-8859-1")
except NameError:
    # On python 3 sys.setdefaultencoding does nothing, so pass.
    # We know we are in python 3 at this point since reload is no longer in
    # builtins, but instead has been moved to importlib, hence the NameError.
    pass

import sas

APP_NAME = 'SasView'
PLUGIN_MODEL_DIR = 'plugin_models'

[docs]class SasView(object): """ Main class for running the SasView application """ def __init__(self): """ """ logger = logging.getLogger(__name__) from sas.sasgui.guiframe.gui_manager import SasViewApp self.gui = SasViewApp(0) if sys.platform == "darwin": self.check_sasmodels_compiler() # Set the application manager for the GUI self.gui.set_manager(self) # Add perspectives to the basic application # Additional perspectives can still be loaded # dynamically # Note: py2exe can't find dynamically loaded # modules. We load the fitting module here # to ensure a complete Windows executable build. # Rebuild .sasview/categories.json. This triggers a load of sasmodels # and all the plugins. try: from sas.sascalc.fit.models import ModelManager from sas.sasgui.guiframe.CategoryInstaller import CategoryInstaller model_list = ModelManager().cat_model_list() CategoryInstaller.check_install(model_list=model_list) except Exception: logger.error("%s: could not load SasView models") logger.error(traceback.format_exc()) # Fitting perspective try: import sas.sasgui.perspectives.fitting as module fitting_plug = module.Plugin() self.gui.add_perspective(fitting_plug) except Exception: logger.error("%s: could not find Fitting plug-in module", APP_NAME) logger.error(traceback.format_exc()) # P(r) perspective try: import sas.sasgui.perspectives.pr as module pr_plug = module.Plugin() self.gui.add_perspective(pr_plug) except Exception: logger.error("%s: could not find P(r) plug-in module", APP_NAME) logger.error(traceback.format_exc()) # Invariant perspective try: import sas.sasgui.perspectives.invariant as module invariant_plug = module.Plugin() self.gui.add_perspective(invariant_plug) except Exception: logger.error("%s: could not find Invariant plug-in module", APP_NAME) logger.error(traceback.format_exc()) # Corfunc perspective try: import sas.sasgui.perspectives.corfunc as module corfunc_plug = module.Plugin() self.gui.add_perspective(corfunc_plug) except Exception: logger.error("Unable to load corfunc module") # Calculator perspective try: import sas.sasgui.perspectives.calculator as module calculator_plug = module.Plugin() self.gui.add_perspective(calculator_plug) except Exception: logger.error("%s: could not find Calculator plug-in module", APP_NAME) logger.error(traceback.format_exc()) # File converter tool try: import sas.sasgui.perspectives.file_converter as module converter_plug = module.Plugin() self.gui.add_perspective(converter_plug) except Exception: logger.error("%s: could not find File Converter plug-in module", APP_NAME) logger.error(traceback.format_exc()) # Add welcome page from .welcome_panel import WelcomePanel self.gui.set_welcome_panel(WelcomePanel) # Build the GUI self.gui.build_gui() # delete unused model folder self.gui.clean_plugin_models(PLUGIN_MODEL_DIR) # Start the main loop self.gui.MainLoop()
[docs] def check_sasmodels_compiler(self): """ Checking c compiler for sasmodels and raises xcode command line tools for installation """ #wx should be importable at this stage import wx import subprocess #Generic message box created becuase standard MessageBox is not moveable class GenericMessageBox(wx.Dialog): def __init__(self, parent, text, title = ''): wx.Dialog.__init__(self, parent, -1, title = title, size = (360,200), pos=(20,60), style = wx.STAY_ON_TOP | wx.DEFAULT_DIALOG_STYLE) panel = wx.Panel(self, -1) top_row_sizer = wx.BoxSizer(wx.HORIZONTAL) error_bitmap = wx.ArtProvider.GetBitmap( wx.ART_ERROR, wx.ART_MESSAGE_BOX ) error_bitmap_ctrl = wx.StaticBitmap(panel, -1) error_bitmap_ctrl.SetBitmap(error_bitmap) label = wx.StaticText(panel, -1, text) top_row_sizer.Add(error_bitmap_ctrl, flag=wx.ALL, border=10) top_row_sizer.Add(label, flag=wx.ALIGN_CENTER_VERTICAL) #Create the OK button in the bottom row. ok_button = wx.Button(panel, wx.ID_OK ) self.Bind(wx.EVT_BUTTON, self.on_ok, source=ok_button) ok_button.SetFocus() ok_button.SetDefault() sizer = wx.BoxSizer(wx.VERTICAL) sizer.Add(top_row_sizer) sizer.Add(ok_button, flag=wx.ALIGN_CENTER | wx.ALL, border=5) panel.SetSizer(sizer) def on_ok(self, event): self.Destroy() logger = logging.getLogger(__name__) try: subprocess.check_output(["cc","--version"], stderr=subprocess.STDOUT) except subprocess.CalledProcessError as exc: dlg = GenericMessageBox(parent=None, text='No compiler installed. Please install command line\n' 'developers tools by clicking \"Install\" in another winodw\n\n' 'Alternatively click \"Not Now\" and use OpenCL\n' 'compiler, which can be set up from menu Fitting->OpenCL Options\n\n', title = 'Compiler Info') dlg.Show() logger.error("No compiler installed. %s\n"%(exc)) logger.error(traceback.format_exc())
[docs]def setup_logging(): from sas.logger_config import SetupLogger logger = SetupLogger(__name__).config_production() # Log the start of the session logger.info(" --- SasView session started ---") # Log the python version logger.info("Python: %s" % sys.version) return logger
[docs]def setup_wx(): # Allow the dynamic selection of wxPython via an environment variable, when devs # who have multiple versions of the module installed want to pick between them. # This variable does not have to be set of course, and through normal usage will # probably not be, but this can make things a little easier when upgrading to a # new version of wx. logger = logging.getLogger(__name__) WX_ENV_VAR = "SASVIEW_WX_VERSION" if WX_ENV_VAR in os.environ: logger.info("You have set the %s environment variable to %s.", WX_ENV_VAR, os.environ[WX_ENV_VAR]) import wxversion if wxversion.checkInstalled(os.environ[WX_ENV_VAR]): logger.info("Version %s of wxPython is installed, so using that version.", os.environ[WX_ENV_VAR]) wxversion.select(os.environ[WX_ENV_VAR]) else: logger.error("Version %s of wxPython is not installed, so using default version.", os.environ[WX_ENV_VAR]) else: logger.info("You have not set the %s environment variable, so using default version of wxPython.", WX_ENV_VAR) import wx try: logger.info("Wx version: %s", wx.__version__) except AttributeError: logger.error("Wx version: error reading version") from . import wxcruft wxcruft.call_later_fix()
#wxcruft.trace_new_id() #Always use private .matplotlib setup to avoid conflicts with other #uses of matplotlib
[docs]def setup_mpl(backend=None): # Always use private .matplotlib setup to avoid conflicts with other mplconfigdir = os.path.join(sas.get_user_dir(), '.matplotlib') if not os.path.exists(mplconfigdir): os.mkdir(mplconfigdir) os.environ['MPLCONFIGDIR'] = mplconfigdir # Set backend to WXAgg; this overrides matplotlibrc, but shouldn't override # mpl.use(). Note: Don't import matplotlib here since the script that # we are running may not actually need it; also, putting as little on the # path as we can if backend: os.environ['MPLBACKEND'] = backend # TODO: ... so much for not importing matplotlib unless we need it... from matplotlib import backend_bases backend_bases.FigureCanvasBase.filetypes.pop('pgf', None)
[docs]def setup_sasmodels(): """ Prepare sasmodels for running within sasview. """ # Set SAS_MODELPATH so sasmodels can find our custom models plugin_dir = os.path.join(sas.get_user_dir(), PLUGIN_MODEL_DIR) os.environ['SAS_MODELPATH'] = plugin_dir #Initialize environment variable with custom setting but only if variable not set SAS_OPENCL = sas.get_custom_config().SAS_OPENCL if SAS_OPENCL and "SAS_OPENCL" not in os.environ: os.environ["SAS_OPENCL"] = SAS_OPENCL
[docs]def run_gui(): """ __main__ method for loading and running SasView """ from multiprocessing import freeze_support freeze_support() setup_logging() setup_mpl(backend='WXAgg') setup_sasmodels() setup_wx() SasView()
[docs]def run_cli(): from multiprocessing import freeze_support freeze_support() setup_logging() # Use default matplotlib backend on mac/linux, but wx on windows. # The problem on mac is that the wx backend requires pythonw. On windows # we are sure to wx since it is the shipped with the app. setup_mpl(backend='WXAgg' if os.name == 'nt' else None) setup_sasmodels() if len(sys.argv) == 1 or sys.argv[1] == '-i': # Run sasview as an interactive python interpreter try: from IPython import start_ipython sys.argv = ["ipython", "--pylab"] sys.exit(start_ipython()) except ImportError: import code code.interact(local={'exit': sys.exit}) elif sys.argv[1] == '-c': exec(sys.argv[2]) else: thing_to_run = sys.argv[1] sys.argv = sys.argv[1:] import runpy if os.path.exists(thing_to_run): runpy.run_path(thing_to_run, run_name="__main__") else: runpy.run_module(thing_to_run, run_name="__main__")
if __name__ == "__main__": run_gui()