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) # 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 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 #Initiliaze enviromental 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()