Detect if NumPy is installed before running setup

This is necessary because we use the setup function from
NumPy, so we need this installed before we can run setup
for GenSVM.

To avoid also needing a scikit-learn installation, the
get_blas_info() function has been inlined here.

1 files changed, 79 insertions, 8 deletions

diff --git a/setup.py b/setup.py
index 28d27ce..1d8a873 100644
--- a/setup.py
+++ b/setup.py
@@ -3,11 +3,6 @@
 
 import os
 import re
-import numpy
-
-from numpy.distutils.core import setup
-from numpy.distutils.misc_util import Configuration
-from sklearn._build_utils import get_blas_info
 
 # Set this to True to enable building extensions using Cython. Set it to False·
 # to build extensions from the C file (that was previously generated using·
@@ -30,11 +25,70 @@ if USE_CYTHON:
         else:
             raise
 
+def _skl_get_blas_info():
+    """Copyright notice for this function
+
+    Copyright (c) 2007–2017 The scikit-learn developers.
+    All rights reserved.
+
+
+    Redistribution and use in source and binary forms, with or without
+    modification, are permitted provided that the following conditions are met:
+
+      a. Redistributions of source code must retain the above copyright notice,
+         this list of conditions and the following disclaimer.
+      b. Redistributions in binary form must reproduce the above copyright
+         notice, this list of conditions and the following disclaimer in the
+         documentation and/or other materials provided with the distribution.
+      c. Neither the name of the Scikit-learn Developers  nor the names of
+         its contributors may be used to endorse or promote products
+         derived from this software without specific prior written
+         permission.
+
+
+    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+    AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+    IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+    ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR
+    ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+    DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+    SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+    CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+    LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+    OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
+    DAMAGE.
+
+    """
+    from numpy.distutils.system_info import get_info
+
+    def atlas_not_found(blas_info_):
+        def_macros = blas_info.get('define_macros', [])
+        for x in def_macros:
+            if x[0] == "NO_ATLAS_INFO":
+                # if x[1] != 1 we should have lapack
+                # how do we do that now?
+                return True
+            if x[0] == "ATLAS_INFO":
+                if "None" in x[1]:
+                    # this one turned up on FreeBSD
+                    return True
+        return False
+
+    blas_info = get_info('blas_opt', 0)
+    if (not blas_info) or atlas_not_found(blas_info):
+        cblas_libs = ['cblas']
+        blas_info.pop('libraries', None)
+    else:
+        cblas_libs = blas_info.pop('libraries', [])
+
+    return cblas_libs, blas_info
+
 
 def configuration():
+    from numpy.distutils.misc_util import Configuration
     config = Configuration('gensvm', '', None)
 
-    cblas_libs, blas_info = get_blas_info()
+    cblas_libs, blas_info = _skl_get_blas_info()
     if os.name == 'posix':
         cblas_libs.append('m')
 
@@ -60,13 +114,14 @@ def configuration():
             os.path.join('src', 'gensvm', 'gensvm_helper.c')
             ]
 
+    from numpy import get_include
     config.add_extension('wrapper',
             sources=gensvm_sources,
             libraries=cblas_libs,
             include_dirs=[
                 os.path.join('src', 'gensvm'),
                 os.path.join('src', 'gensvm', 'include'),
-                numpy.get_include(),
+                get_include(),
                 blas_info.pop('include_dirs', [])],
             extra_compile_args=blas_info.pop('extra_compile_args', []),
             depends=gensvm_depends,
@@ -83,13 +138,28 @@ def read(fname):
     return open(os.path.join(os.path.dirname(__file__), fname)).read()
 
 
+def check_requirements():
+    numpy_instructions = ("Numerical Python (NumPy) is not installed on your "
+                "system. This package is required for GenSVM. Please install "
+                "NumPy using the instructions available here: "
+                "https://docs.scipy.org/doc/numpy-1.13.0/user/install.html")
+
+    try:
+        import numpy
+        numpy_version = numpy.__version__
+    except ImportError:
+        raise ImportError(numpy_instructions)
+
+
 if __name__ == '__main__':
+    check_requirements()
 
     version = re.search("__version__ = '([^']+)'", 
             open('gensvm/__init__.py').read()).group(1)
 
     attr = configuration().todict()
 
+    attr['version'] = version
     attr['description'] = 'Python package for the GenSVM classifier'
     attr['long_description'] = read('README.rst')
     attr['packages'] = ['gensvm']
@@ -97,6 +167,7 @@ if __name__ == '__main__':
     attr['author'] = "G.J.J. van den Burg"
     attr['author_email'] = "gertjanvandenburg@gmail.com"
     attr['license'] = 'GPL v2'
-    attr['install_requires'] = ['scikit-learn', 'numpy']
+    attr['install_requires'] = ['scikit-learn', 'numpy', 'scipy']
 
+    from numpy.distutils.core import setup
     setup(**attr)