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
"""
[docs] 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()