setup.py 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. from __future__ import division, print_function, absolute_import
  2. from os.path import join, dirname
  3. import sys
  4. import os
  5. import glob
  6. def configuration(parent_package='',top_path=None):
  7. from numpy.distutils.misc_util import Configuration
  8. from scipy._build_utils.system_info import get_info
  9. from scipy._build_utils import numpy_nodepr_api
  10. config = Configuration('dsolve',parent_package,top_path)
  11. config.add_data_dir('tests')
  12. lapack_opt = get_info('lapack_opt',notfound_action=2)
  13. if sys.platform == 'win32':
  14. superlu_defs = [('NO_TIMER',1)]
  15. else:
  16. superlu_defs = []
  17. superlu_defs.append(('USE_VENDOR_BLAS',1))
  18. superlu_src = join(dirname(__file__), 'SuperLU', 'SRC')
  19. sources = sorted(glob.glob(join(superlu_src, '*.c')))
  20. headers = list(glob.glob(join(superlu_src, '*.h')))
  21. config.add_library('superlu_src',
  22. sources=sources,
  23. macros=superlu_defs,
  24. include_dirs=[superlu_src],
  25. )
  26. # Extension
  27. ext_sources = ['_superlumodule.c',
  28. '_superlu_utils.c',
  29. '_superluobject.c']
  30. config.add_extension('_superlu',
  31. sources=ext_sources,
  32. libraries=['superlu_src'],
  33. depends=(sources + headers),
  34. extra_info=lapack_opt,
  35. **numpy_nodepr_api
  36. )
  37. # Add license files
  38. config.add_data_files('SuperLU/License.txt')
  39. return config
  40. if __name__ == '__main__':
  41. from numpy.distutils.core import setup
  42. setup(**configuration(top_path='').todict())