123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148 |
- import os
- import platform
- import sys
- from os.path import join
- from numpy.distutils.system_info import platform_bits
- is_msvc = (platform.platform().startswith('Windows') and
- platform.python_compiler().startswith('MS'))
- def configuration(parent_package='', top_path=None):
- from numpy.distutils.misc_util import Configuration, get_mathlibs
- config = Configuration('random', parent_package, top_path)
- def generate_libraries(ext, build_dir):
- config_cmd = config.get_config_cmd()
- libs = get_mathlibs()
- if sys.platform == 'win32':
- libs.extend(['Advapi32', 'Kernel32'])
- ext.libraries.extend(libs)
- return None
- # enable unix large file support on 32 bit systems
- # (64 bit off_t, lseek -> lseek64 etc.)
- if sys.platform[:3] == 'aix':
- defs = [('_LARGE_FILES', None)]
- else:
- defs = [('_FILE_OFFSET_BITS', '64'),
- ('_LARGEFILE_SOURCE', '1'),
- ('_LARGEFILE64_SOURCE', '1')]
- defs.append(('NPY_NO_DEPRECATED_API', 0))
- config.add_subpackage('tests')
- config.add_data_dir('tests/data')
- config.add_data_dir('_examples')
- EXTRA_LINK_ARGS = []
- EXTRA_LIBRARIES = ['npyrandom']
- if os.name != 'nt':
- # Math lib
- EXTRA_LIBRARIES.append('m')
- # Some bit generators exclude GCC inlining
- EXTRA_COMPILE_ARGS = ['-U__GNUC_GNU_INLINE__']
- if is_msvc and platform_bits == 32:
- # 32-bit windows requires explicit sse2 option
- EXTRA_COMPILE_ARGS += ['/arch:SSE2']
- elif not is_msvc:
- # Some bit generators require c99
- EXTRA_COMPILE_ARGS += ['-std=c99']
- # Use legacy integer variable sizes
- LEGACY_DEFS = [('NP_RANDOM_LEGACY', '1')]
- PCG64_DEFS = []
- # One can force emulated 128-bit arithmetic if one wants.
- #PCG64_DEFS += [('PCG_FORCE_EMULATED_128BIT_MATH', '1')]
- depends = ['__init__.pxd', 'c_distributions.pxd', 'bit_generator.pxd']
- # npyrandom - a library like npymath
- npyrandom_sources = [
- 'src/distributions/logfactorial.c',
- 'src/distributions/distributions.c',
- 'src/distributions/random_mvhg_count.c',
- 'src/distributions/random_mvhg_marginals.c',
- 'src/distributions/random_hypergeometric.c',
- ]
- config.add_installed_library('npyrandom',
- sources=npyrandom_sources,
- install_dir='lib',
- build_info={
- 'include_dirs' : [], # empty list required for creating npyrandom.h
- 'extra_compiler_args' : (['/GL-'] if is_msvc else []),
- })
- for gen in ['mt19937']:
- # gen.pyx, src/gen/gen.c, src/gen/gen-jump.c
- config.add_extension(f'_{gen}',
- sources=[f'_{gen}.c',
- f'src/{gen}/{gen}.c',
- f'src/{gen}/{gen}-jump.c'],
- include_dirs=['.', 'src', join('src', gen)],
- libraries=EXTRA_LIBRARIES,
- extra_compile_args=EXTRA_COMPILE_ARGS,
- extra_link_args=EXTRA_LINK_ARGS,
- depends=depends + [f'_{gen}.pyx'],
- define_macros=defs,
- )
- for gen in ['philox', 'pcg64', 'sfc64']:
- # gen.pyx, src/gen/gen.c
- _defs = defs + PCG64_DEFS if gen == 'pcg64' else defs
- config.add_extension(f'_{gen}',
- sources=[f'_{gen}.c',
- f'src/{gen}/{gen}.c'],
- include_dirs=['.', 'src', join('src', gen)],
- libraries=EXTRA_LIBRARIES,
- extra_compile_args=EXTRA_COMPILE_ARGS,
- extra_link_args=EXTRA_LINK_ARGS,
- depends=depends + [f'_{gen}.pyx',
- 'bit_generator.pyx', 'bit_generator.pxd'],
- define_macros=_defs,
- )
- for gen in ['_common', 'bit_generator']:
- # gen.pyx
- config.add_extension(gen,
- sources=[f'{gen}.c'],
- libraries=EXTRA_LIBRARIES,
- extra_compile_args=EXTRA_COMPILE_ARGS,
- extra_link_args=EXTRA_LINK_ARGS,
- include_dirs=['.', 'src'],
- depends=depends + [f'{gen}.pyx', f'{gen}.pxd',],
- define_macros=defs,
- )
- config.add_data_files(f'{gen}.pxd')
- for gen in ['_generator', '_bounded_integers']:
- # gen.pyx, src/distributions/distributions.c
- config.add_extension(gen,
- sources=[f'{gen}.c'],
- libraries=EXTRA_LIBRARIES + ['npymath'],
- extra_compile_args=EXTRA_COMPILE_ARGS,
- include_dirs=['.', 'src'],
- extra_link_args=EXTRA_LINK_ARGS,
- depends=depends + [f'{gen}.pyx'],
- define_macros=defs,
- )
- config.add_data_files('_bounded_integers.pxd')
- mtrand_libs = ['m', 'npymath'] if os.name != 'nt' else ['npymath']
- config.add_extension('mtrand',
- sources=['mtrand.c',
- 'src/legacy/legacy-distributions.c',
- 'src/distributions/distributions.c',
- ],
- include_dirs=['.', 'src', 'src/legacy'],
- libraries=mtrand_libs,
- extra_compile_args=EXTRA_COMPILE_ARGS,
- extra_link_args=EXTRA_LINK_ARGS,
- depends=depends + ['mtrand.pyx'],
- define_macros=defs + LEGACY_DEFS,
- )
- config.add_data_files(*depends)
- config.add_data_files('*.pyi')
- return config
- if __name__ == '__main__':
- from numpy.distutils.core import setup
- setup(configuration=configuration)
|