source: sasview/run.py @ 075c460

Last change on this file since 075c460 was 18e7309, checked in by lewis, 8 years ago

Add C extension for reading 2d BSL files

  • Property mode set to 100755
File size: 4.8 KB
Line 
1#!/usr/bin/env python
2"""
3Run sasview in place.  This allows sasview to use the python
4files in the source tree without having to call setup.py install
5first.  A rebuild is still necessary when working on sas models
6or c modules.
7
8Usage:
9
10./run.py [(module|script) args...]
11
12Without arguments run.py runs sasview.  With arguments, run.py will run
13the given module or script.
14"""
15
16import os
17import sys
18import imp
19from contextlib import contextmanager
20from os.path import abspath, dirname, join as joinpath
21
22
23def addpath(path):
24    """
25    Add a directory to the python path environment, and to the PYTHONPATH
26    environment variable for subprocesses.
27    """
28    path = abspath(path)
29    if 'PYTHONPATH' in os.environ:
30        PYTHONPATH = path + os.pathsep + os.environ['PYTHONPATH']
31    else:
32        PYTHONPATH = path
33    os.environ['PYTHONPATH'] = PYTHONPATH
34    sys.path.insert(0, path)
35
36@contextmanager
37def cd(path):
38    """
39    Change directory for duration of "with" context.
40    """
41    old_dir = os.getcwd()
42    os.chdir(path)
43    yield
44    os.chdir(old_dir)
45
46def import_package(modname, path):
47    """Import a package into a particular point in the python namespace"""
48    mod = imp.load_source(modname, abspath(joinpath(path,'__init__.py')))
49    sys.modules[modname] = mod
50    mod.__path__ = [abspath(path)]
51    return mod
52
53def import_dll(modname, build_path):
54    """Import a DLL from the build directory"""
55    import sysconfig
56    ext = sysconfig.get_config_var('SO')
57    # build_path comes from context
58    path = joinpath(build_path, *modname.split('.'))+ext
59    #print "importing", modname, "from", path
60    return imp.load_dynamic(modname, path)
61
62def prepare():
63    # Don't create *.pyc files
64    sys.dont_write_bytecode = True
65
66    # Debug numpy warnings
67    #import numpy; numpy.seterr(all='raise')
68
69    # find the directories for the source and build
70    from distutils.util import get_platform
71    root = abspath(dirname(__file__))
72    platform = '%s-%s'%(get_platform(),sys.version[:3])
73    build_path = joinpath(root, 'build','lib.'+platform)
74
75    # Notify the help menu that the Sphinx documentation is in a different
76    # place than it otherwise would be.
77    os.environ['SASVIEW_DOC_PATH'] = joinpath(build_path, "doc")
78
79    # Make sure that we have a private version of mplconfig
80    #mplconfig = joinpath(abspath(dirname(__file__)), '.mplconfig')
81    #os.environ['MPLCONFIGDIR'] = mplconfig
82    #if not os.path.exists(mplconfig): os.mkdir(mplconfig)
83    #import matplotlib
84    #matplotlib.use('Agg')
85    #print matplotlib.__file__
86    #import pylab; pylab.hold(False)
87    # add periodictable to the path
88    try: import periodictable
89    except: addpath(joinpath(root, '..','periodictable'))
90
91    try: import bumps
92    except: addpath(joinpath(root, '..','bumps'))
93
94    # select wx version
95    #addpath(os.path.join(root, '..','wxPython-src-3.0.0.0','wxPython'))
96
97    # Build project if the build directory does not already exist.
98    if not os.path.exists(build_path):
99        import subprocess
100        with cd(root):
101            subprocess.call((sys.executable, "setup.py", "build"), shell=False)
102
103    # Put the source trees on the path
104    addpath(joinpath(root, 'src'))
105
106    # sasmodels on the path
107    addpath(joinpath(root, '../sasmodels/'))
108
109    # Import the sasview package from root/sasview as sas.sasview.  It would
110    # be better to just store the package in src/sas/sasview.
111    import sas
112    sas.sasview = import_package('sas.sasview', joinpath(root,'sasview'))
113
114    # The sas.models package Compiled Model files should be pulled in from the build directory even though
115    # the source is stored in src/sas/models.
116
117    # Compiled modules need to be pulled from the build directory.
118    # Some packages are not where they are needed, so load them explicitly.
119    import sas.sascalc.pr
120    sas.sascalc.pr.core = import_package('sas.sascalc.pr.core',
121                                  joinpath(build_path, 'sas', 'sascalc', 'pr', 'core'))
122
123    # Compiled modules need to be pulled from the build directory.
124    # Some packages are not where they are needed, so load them explicitly.
125    import sas.sascalc.file_converter
126    sas.sascalc.file_converter.core = import_package('sas.sascalc.file_converter.core',
127                                  joinpath(build_path, 'sas', 'sascalc', 'file_converter', 'core'))                   
128
129    # Compiled modules need to be pulled from the build directory.
130    # Some packages are not where they are needed, so load them explicitly.
131    import sas.sascalc.calculator
132    sas.sascalc.calculator.core = import_package('sas.sascalc.calculator.core',
133                                  joinpath(build_path, 'sas', 'sascalc', 'calculator', 'core'))
134
135    sys.path.append(build_path)
136
137    #print "\n".join(sys.path)
138
139if __name__ == "__main__":
140    prepare()
141    from sas.sasview.sasview import run
142    run()
Note: See TracBrowser for help on using the repository browser.