Add get_npymath_info function to reuse npymath outside numpy.

1 files changed, 16 insertions, 0 deletions

diff --git a/numpy/distutils/misc_util.py b/numpy/distutils/misc_util.py
index f3ce796b2..d7606d337 100644
--- a/numpy/distutils/misc_util.py
+++ b/numpy/distutils/misc_util.py
@@ -1539,6 +1539,22 @@ def get_numpy_include_dirs():
     # else running numpy/core/setup.py
     return include_dirs
 
+def get_npymath_info():
+    """Return a extra_info-compatible dict to link against the core npymath
+    library.
+    
+    Example
+    -------
+    >>> npymath_info = get_npymath_info()
+    >>> config.add_extension('foo', sources=['foo.c'], extra_info=npymath_info)
+    """
+    import numpy
+    p = numpy.__file__
+    info = {}
+    info["library_dirs"] = [os.path.join(os.path.dirname(p), "core", "lib")]
+    info["libraries"] = ["npymath"]
+    return info
+
 def is_bootstrapping():
     import __builtin__
     try: