1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
"""numerix imports either Numeric or numarray based on various selectors.
0. If the value "--numarray" or "--Numeric" is specified on the
command line, then numerix imports the specified array package.
1. If the environment variable NUMERIX exists, it's value is used to
choose Numeric or numarray.
2. The value of numerix in ~/.matplotlibrc: either Numeric or numarray
<currently not implemented for scipy>
3. If none of the above is done, the default array package is Numeric.
Because the .matplotlibrc always provides *some* value for numerix (it
has it's own system of default values), this default is most likely
never used.
To summarize: the commandline is examined first, the rc file second,
and the default array package is Numeric.
"""
import sys, os
# from matplotlib import rcParams, verbose
which = None, None
# First, see if --numarray or --Numeric was specified on the command
# line:
if hasattr(sys, 'argv'): #Once again, Apache mod_python has no argv
for a in sys.argv:
if a in ["--Numeric", "--numeric", "--NUMERIC",
"--Numarray", "--numarray", "--NUMARRAY"]:
which = a[2:], "command line"
sys.argv.remove(a)
break
del a
if os.getenv("NUMERIX"):
which = os.getenv("NUMERIX"), "environment var"
# if which[0] is None:
# try: # In theory, rcParams always has *some* value for numerix.
# which = rcParams['numerix'], "rc"
# except KeyError:
# pass
# If all the above fail, default to Numeric.
if which[0] is None:
which = "numeric", "defaulted"
which = which[0].strip().lower(), which[1]
if which[0] not in ["numeric", "numarray"]:
verbose.report_error(__doc__)
raise ValueError("numerix selector must be either 'Numeric' or 'numarray' but the value obtained from the %s was '%s'." % (which[1], which[0]))
# Tweak the environment for f2py/scipy_distutils, e.g. setenv NUMERIC 1
os.environ[ which[0].upper() ] = "1"
if which[0] == "numarray":
from _na_imports import *
import numarray
elif which[0] == "numeric":
from _nc_imports import *
import Numeric
else:
raise RuntimeError("invalid numerix selector")
#print 'numerix %s'% NX_VERSION
# ---------------------------------------------------------------
# Common imports and fixes
# ---------------------------------------------------------------
# a bug fix for blas numeric suggested by Fernando Perez
matrixmultiply=dot
def any(x):
"""Return true if any elements of x are true: sometrue(ravel(x))
"""
return sometrue(ravel(x))
def all(x):
"""Return true if all elements of x are true: alltrue(ravel(x))
"""
return alltrue(ravel(x))
def _import_fail_message(module, version):
"""Prints a message when the array package specific version of an extension
fails to import correctly.
"""
_dict = { "which" : which[0],
"module" : module,
"specific" : version + module
}
print """\nThe import of the %(which)s version of the %(module)s module, %(specific)s, failed.\nThis is either because %(which)s was unavailable when scipy was compiled,\nor because a dependency of %(specific)s could not be satisfied.\nIf it appears that %(specific)s was not built, make sure you have a working copy of\n%(which)s and then re-install scipy. Otherwise, the following traceback gives more details:\n""" % _dict
|
