Moving things to live under scipy

1 files changed, 95 insertions, 0 deletions

diff --git a/scipy/base/numerix.py b/scipy/base/numerix.py
new file mode 100644
index 000000000..d7db343f3
--- /dev/null
+++ b/scipy/base/numerix.py
@@ -0,0 +1,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