setup.py 6.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160
  1. from __future__ import division, print_function, absolute_import
  2. import os
  3. import sys
  4. from os.path import join, dirname
  5. from distutils.sysconfig import get_python_inc
  6. import subprocess
  7. import numpy
  8. from numpy.distutils.misc_util import get_numpy_include_dirs
  9. try:
  10. from numpy.distutils.misc_util import get_info
  11. except ImportError:
  12. raise ValueError("numpy >= 1.4 is required (detected %s from %s)" %
  13. (numpy.__version__, numpy.__file__))
  14. def configuration(parent_package='',top_path=None):
  15. from numpy.distutils.misc_util import Configuration
  16. from scipy._build_utils.system_info import get_info as get_system_info
  17. config = Configuration('special', parent_package, top_path)
  18. define_macros = []
  19. if sys.platform == 'win32':
  20. # define_macros.append(('NOINFINITIES',None))
  21. # define_macros.append(('NONANS',None))
  22. define_macros.append(('_USE_MATH_DEFINES',None))
  23. curdir = os.path.abspath(os.path.dirname(__file__))
  24. inc_dirs = [get_python_inc(), os.path.join(curdir, "c_misc")]
  25. if inc_dirs[0] != get_python_inc(plat_specific=1):
  26. inc_dirs.append(get_python_inc(plat_specific=1))
  27. inc_dirs.insert(0, get_numpy_include_dirs())
  28. inc_dirs.append(join(dirname(dirname(__file__)), '_lib'))
  29. # C libraries
  30. c_misc_src = [join('c_misc','*.c')]
  31. c_misc_hdr = [join('c_misc','*.h')]
  32. cephes_src = [join('cephes','*.c')]
  33. cephes_hdr = [join('cephes', '*.h')]
  34. config.add_library('sc_c_misc',sources=c_misc_src,
  35. include_dirs=[curdir] + inc_dirs,
  36. depends=(cephes_hdr + cephes_src
  37. + c_misc_hdr + cephes_hdr
  38. + ['*.h']),
  39. macros=define_macros)
  40. config.add_library('sc_cephes',sources=cephes_src,
  41. include_dirs=[curdir] + inc_dirs,
  42. depends=(cephes_hdr + ['*.h']),
  43. macros=define_macros)
  44. # Fortran/C++ libraries
  45. mach_src = [join('mach','*.f')]
  46. amos_src = [join('amos','*.f')]
  47. cdf_src = [join('cdflib','*.f')]
  48. specfun_src = [join('specfun','*.f')]
  49. config.add_library('sc_mach',sources=mach_src,
  50. config_fc={'noopt':(__file__,1)})
  51. config.add_library('sc_amos',sources=amos_src)
  52. config.add_library('sc_cdf',sources=cdf_src)
  53. config.add_library('sc_specfun',sources=specfun_src)
  54. # Extension specfun
  55. config.add_extension('specfun',
  56. sources=['specfun.pyf'],
  57. f2py_options=['--no-wrap-functions'],
  58. depends=specfun_src,
  59. define_macros=[],
  60. libraries=['sc_specfun'])
  61. # Extension _ufuncs
  62. headers = ['*.h', join('c_misc', '*.h'), join('cephes', '*.h')]
  63. ufuncs_src = ['_ufuncs.c', 'sf_error.c', '_logit.c.src',
  64. "amos_wrappers.c", "cdf_wrappers.c", "specfun_wrappers.c"]
  65. ufuncs_dep = (headers + ufuncs_src + amos_src + c_misc_src + cephes_src
  66. + mach_src + cdf_src + specfun_src)
  67. cfg = dict(get_system_info('lapack_opt'))
  68. cfg.setdefault('include_dirs', []).extend([curdir] + inc_dirs + [numpy.get_include()])
  69. cfg.setdefault('libraries', []).extend(['sc_amos','sc_c_misc','sc_cephes','sc_mach',
  70. 'sc_cdf', 'sc_specfun'])
  71. cfg.setdefault('define_macros', []).extend(define_macros)
  72. config.add_extension('_ufuncs',
  73. depends=ufuncs_dep,
  74. sources=ufuncs_src,
  75. extra_info=get_info("npymath"),
  76. **cfg)
  77. # Extension _ufuncs_cxx
  78. ufuncs_cxx_src = ['_ufuncs_cxx.cxx', 'sf_error.c',
  79. '_faddeeva.cxx', 'Faddeeva.cc',
  80. '_wright.cxx', 'wright.cc']
  81. ufuncs_cxx_dep = (headers + ufuncs_cxx_src + cephes_src
  82. + ['*.hh'])
  83. config.add_extension('_ufuncs_cxx',
  84. sources=ufuncs_cxx_src,
  85. depends=ufuncs_cxx_dep,
  86. include_dirs=[curdir] + inc_dirs,
  87. define_macros=define_macros,
  88. extra_info=get_info("npymath"))
  89. cfg = dict(get_system_info('lapack_opt'))
  90. config.add_extension('_ellip_harm_2',
  91. sources=['_ellip_harm_2.c', 'sf_error.c',],
  92. **cfg
  93. )
  94. # Cython API
  95. config.add_data_files('cython_special.pxd')
  96. cython_special_src = ['cython_special.c', 'sf_error.c', '_logit.c.src',
  97. "amos_wrappers.c", "cdf_wrappers.c", "specfun_wrappers.c"]
  98. cython_special_dep = (headers + ufuncs_src + ufuncs_cxx_src + amos_src
  99. + c_misc_src + cephes_src + mach_src + cdf_src
  100. + specfun_src)
  101. cfg = dict(get_system_info('lapack_opt'))
  102. cfg.setdefault('include_dirs', []).extend([curdir] + inc_dirs + [numpy.get_include()])
  103. cfg.setdefault('libraries', []).extend(['sc_amos','sc_c_misc','sc_cephes','sc_mach',
  104. 'sc_cdf', 'sc_specfun'])
  105. cfg.setdefault('define_macros', []).extend(define_macros)
  106. config.add_extension('cython_special',
  107. depends=cython_special_dep,
  108. sources=cython_special_src,
  109. extra_info=get_info("npymath"),
  110. **cfg)
  111. # combinatorics
  112. config.add_extension('_comb',
  113. sources=['_comb.c'])
  114. # testing for _round.h
  115. config.add_extension('_test_round',
  116. sources=['_test_round.c'],
  117. depends=['_round.h', 'cephes/dd_idefs.h'],
  118. include_dirs=[numpy.get_include()] + inc_dirs,
  119. extra_info=get_info('npymath'))
  120. config.add_data_files('tests/*.py')
  121. config.add_data_files('tests/data/README')
  122. # regenerate npz data files
  123. makenpz = os.path.join(os.path.dirname(__file__),
  124. 'utils', 'makenpz.py')
  125. data_dir = os.path.join(os.path.dirname(__file__),
  126. 'tests', 'data')
  127. for name in ['boost', 'gsl', 'local']:
  128. subprocess.check_call([sys.executable, makenpz,
  129. '--use-timestamp',
  130. os.path.join(data_dir, name)])
  131. config.add_data_files('tests/data/*.npz')
  132. config.add_subpackage('_precompute')
  133. return config
  134. if __name__ == '__main__':
  135. from numpy.distutils.core import setup
  136. setup(**configuration(top_path='').todict())