329 lines
12 KiB
Python
329 lines
12 KiB
Python
import sys
|
|
import os
|
|
|
|
if sys.version.find('MSC') == -1:
|
|
windows = False
|
|
else:
|
|
windows = True
|
|
|
|
# <rant>
|
|
# Setuptools is broken on Linux. The command "python setup.py install" no
|
|
# longer works. Only use setuptools on Windows to keep Appveyor happy.
|
|
# </rant>
|
|
|
|
if windows:
|
|
try:
|
|
from setuptools import setup, Extension
|
|
except ImportError:
|
|
from distutils.core import setup, Extension
|
|
else:
|
|
from distutils.core import setup, Extension
|
|
|
|
from distutils.command.clean import clean
|
|
from distutils.command.build_ext import build_ext
|
|
from distutils.command.install_data import install_data
|
|
|
|
def writeln(s):
|
|
sys.stdout.write('%s\n' % s)
|
|
sys.stdout.flush()
|
|
|
|
|
|
# Fail gracefully for old versions of Python.
|
|
|
|
if sys.version[:3] < '2.6':
|
|
writeln("GMPY2 requires Python 2.6 or later.")
|
|
writeln("Please use GMPY 1.x for earlier versions of Python.")
|
|
sys.exit()
|
|
|
|
# Initialize some global values.
|
|
|
|
lib_path = 'lib'
|
|
|
|
# Several command line options can be used to modify compilation of GMPY2.
|
|
#
|
|
# --msys2 -> build on Windows using MSYS2, MinGW, and GMP/MPFR/MPC
|
|
# --mpir -> build on Windows using Visual Studio and MPIR/MPFR/MPC
|
|
# --vector -> build the unsupport/development vector functions
|
|
# --lib64 -> use /prefix/lib64 instead of /prefix/lib
|
|
# --lib32 -> use /prefix/lib32 instead of /prefix/lib
|
|
# --shared=<...> -> add the specified directory prefix to the beginning of
|
|
# the list of directories that are searched for GMP, MPFR,
|
|
# and MPC shared libraries
|
|
# --static=<...> -> create a statically linked library using libraries from
|
|
# specified path, or from the operating system's default
|
|
# library location if no path is specified
|
|
# --vector -> include the vector_XXX() functions; they are unstable
|
|
# and under active development
|
|
# --fast -> depend on MPFR and MPC internal implementations details
|
|
# (even more than the standard build)
|
|
#
|
|
# Ugly hack ahead. Sorry.
|
|
#
|
|
# I haven't found any examples on how to extend distutils with user-defined
|
|
# options. And the documentation is not helpful, either. So instead of fighting
|
|
# distutils, some of the options are converted into macro definitions. Macros
|
|
# then get parsed by distutils and then a custom class reads the macros and
|
|
# tweaks the setup.
|
|
#
|
|
# The custom command line arguments are processed as follows:
|
|
#
|
|
# --force is temporarily added to the list of defines and then removed by
|
|
# gmpy_build_ext. It is used to allow setup.py install --force to work as
|
|
# expected.
|
|
#
|
|
# --msys2 is converted to -DMSYS2. Since MSYS2 needs to be defined as a macro
|
|
# to control options in the GMPY2 source code, this make sense.
|
|
#
|
|
# --mpir is converted to -DMPIR. Since MPI needs to be defined as a macro
|
|
# to control options in the GMPY2 source code, this make sense.
|
|
#
|
|
# --lib64 and --lib32 are removed from sys.argv and the global variable
|
|
# lib_path is set to 'lib64' or 'lib32' as appropriate.
|
|
#
|
|
# --shared and --static are converted to -DSHARED and -DSTATIC.
|
|
|
|
# Notes regarding coverage testing.
|
|
#
|
|
# The --gcov command line option configures GCC to collect code coverage data
|
|
# for testing purposes. The raw collected data can be converted to .gcov files
|
|
# with the command "gcov build/temp.linux-x86_64-3.4/src/gmpy2.gcno". To create
|
|
# a tidy HTML summary, use "gcovr -s -r . --html -o coverage.html". When tests
|
|
# are completed, delete the .gcov and coverage.html files.
|
|
#
|
|
# The following commands show the use of lcov to analyze the coverage data.
|
|
#
|
|
# py34 setup.py clean
|
|
# py34 setup.py install --gcov --force
|
|
# py34 test/runtests.py
|
|
# lcov --capture --directory build/temp.linux-x86_64-3.4/src/gmpy2.gcno --output-file build/coverage.info
|
|
# genhtml build/coverage.info --output-directory build/out
|
|
# firefox build/out/index.html
|
|
#
|
|
# Remember to remove the *.gcov file and the out sub-directory.
|
|
|
|
defines = []
|
|
build_type_specified = False
|
|
|
|
for token in sys.argv[:]:
|
|
if token.lower() == '--force':
|
|
defines.append( ('FORCE', 1) )
|
|
sys.argv.remove(token)
|
|
|
|
if token.lower() == '--lib64':
|
|
lib_path = 'lib64'
|
|
sys.argv.remove(token)
|
|
|
|
if token.lower() == '--lib32':
|
|
lib_path = 'lib32'
|
|
sys.argv.remove(token)
|
|
|
|
if token.lower() == '--vector':
|
|
defines.append( ('VECTOR', 1) )
|
|
sys.argv.remove(token)
|
|
|
|
if token.lower() == '--fast':
|
|
defines.append( ('FAST', 1) )
|
|
sys.argv.remove(token)
|
|
|
|
if token.lower() == '--msys2':
|
|
defines.append( ('MSYS2', 1) )
|
|
sys.argv.remove(token)
|
|
|
|
if token.lower() == '--mpir':
|
|
defines.append( ('MPIR', 1) )
|
|
sys.argv.remove(token)
|
|
|
|
if token.lower().startswith('--shared'):
|
|
build_type_specified = True
|
|
try:
|
|
defines.append( ('SHARED', token.split('=')[1]) )
|
|
except IndexError:
|
|
defines.append( ('SHARED', 1) )
|
|
sys.argv.remove(token)
|
|
|
|
if token.lower().startswith('--static'):
|
|
build_type_specified = True
|
|
try:
|
|
defines.append( ('STATIC', token.split('=')[1]) )
|
|
except IndexError:
|
|
defines.append( ('STATIC', 1) )
|
|
sys.argv.remove(token)
|
|
|
|
if token.lower() == '--gcov':
|
|
defines.append( ('GCOV', 1) )
|
|
sys.argv.remove(token)
|
|
|
|
if not build_type_specified:
|
|
defines.append( ('SHARED', 1) )
|
|
|
|
|
|
# Improved clean command.
|
|
|
|
class gmpy_clean(clean):
|
|
|
|
def run(self):
|
|
self.all = True
|
|
clean.run(self)
|
|
|
|
# Define a custom build class that parses to the defined macros to alter the
|
|
# build setup.
|
|
|
|
class gmpy_build_ext(build_ext):
|
|
|
|
def initialize_options(self):
|
|
build_ext.initialize_options(self)
|
|
|
|
def doit(self):
|
|
# Find the directory specfied for non-standard library location.
|
|
search_dirs = []
|
|
static = False
|
|
msys2 = False
|
|
mpir = False
|
|
|
|
for d in self.extensions[0].define_macros[:]:
|
|
if d[0] == 'MSYS2':
|
|
self.compiler = 'mingw32'
|
|
msys2 = True
|
|
|
|
if d[0] == 'MPIR':
|
|
mpir = True
|
|
|
|
if d[0] == 'GCOV':
|
|
self.extensions[0].libraries.append('gcov')
|
|
self.extensions[0].extra_compile_args.extend(['-O0', '--coverage'])
|
|
self.extensions[0].extra_link_args.append('--coverage')
|
|
self.extensions[0].define_macros.remove(d)
|
|
|
|
if d[0] == 'FORCE':
|
|
self.force = 1
|
|
try:
|
|
self.extensions[0].define_macros.remove(d)
|
|
except ValueError:
|
|
pass
|
|
|
|
if d[0] in ('SHARED', 'STATIC'):
|
|
if d[0] == 'STATIC':
|
|
static = True
|
|
if d[0] == 'SHARED':
|
|
static = False
|
|
if d[1] and d[1] != 1:
|
|
search_dirs.extend(map(os.path.expanduser, d[1].split(":")))
|
|
|
|
if mpir:
|
|
self.extensions[0].libraries.extend(['mpir', 'mpfr', 'mpc'])
|
|
else:
|
|
self.extensions[0].libraries.extend(['gmp', 'mpfr', 'mpc'])
|
|
|
|
# If non-default directories have been specified, we need to find the
|
|
# exact location of the libraries to allow static or runtime linking.
|
|
|
|
mp_found = ''
|
|
mpfr_found = ''
|
|
mpc_found = ''
|
|
if search_dirs:
|
|
|
|
for adir in search_dirs:
|
|
lookin = os.path.join(adir, 'include')
|
|
if mpir:
|
|
if os.path.isfile(os.path.join(lookin, 'mpir.h')):
|
|
mp_found = adir
|
|
else:
|
|
if os.path.isfile(os.path.join(lookin, 'gmp.h')):
|
|
mp_found = adir
|
|
if os.path.isfile(os.path.join(lookin, 'mpfr.h')):
|
|
mpfr_found = adir
|
|
if os.path.isfile(os.path.join(lookin, 'mpc.h')):
|
|
mpc_found = adir
|
|
|
|
# Add the directory information for location where valid versions were
|
|
# found. This can cause confusion if there are multiple installations of
|
|
# the same version of Python on the system.
|
|
|
|
for adir in (mp_found, mpfr_found, mpc_found):
|
|
if not adir:
|
|
continue
|
|
if os.path.join(adir, 'include') in self.extensions[0].include_dirs:
|
|
continue
|
|
self.extensions[0].include_dirs += [os.path.join(adir, 'include')]
|
|
self.extensions[0].library_dirs += [os.path.join(adir, lib_path)]
|
|
|
|
# Add the runtime linking options.
|
|
if not static and not windows:
|
|
self.extensions[0].runtime_library_dirs += [os.path.join(adir, lib_path)]
|
|
|
|
# Add the static linking options.
|
|
|
|
# Add MSVC specific options.
|
|
if windows and not msys2:
|
|
self.extensions[0].extra_link_args.append('/MANIFEST')
|
|
self.extensions[0].define_macros.append(("MPIR", 1))
|
|
if not static:
|
|
self.extensions[0].define_macros.append(("MSC_USE_DLL", None))
|
|
else:
|
|
if mpir:
|
|
if static and mp_found:
|
|
self.extensions[0].extra_objects.append(os.path.join(mp_found, lib_path, 'libmpir.a'))
|
|
else:
|
|
if static and mp_found:
|
|
self.extensions[0].extra_objects.append(os.path.join(mp_found, lib_path, 'libgmp.a'))
|
|
if static and mpfr_found:
|
|
self.extensions[0].extra_objects.append(os.path.join(mpfr_found, lib_path, 'libmpfr.a'))
|
|
if static and mpc_found:
|
|
self.extensions[0].extra_objects.append(os.path.join(mpc_found, lib_path, 'libmpc.a'))
|
|
|
|
|
|
def finalize_options(self):
|
|
build_ext.finalize_options(self)
|
|
gmpy_build_ext.doit(self)
|
|
|
|
# custom install data in order that data_files
|
|
# get installed together with the .so
|
|
|
|
class gmpy_install_data(install_data):
|
|
def finalize_options(self):
|
|
install_data.finalize_options(self)
|
|
install = self.distribution.get_command_obj('install')
|
|
self.install_dir = install.install_purelib
|
|
|
|
# prepare the extension for building
|
|
|
|
my_commands = {'clean' : gmpy_clean, 'build_ext' : gmpy_build_ext, 'install_data' : gmpy_install_data}
|
|
|
|
gmpy2_ext = Extension('gmpy2',
|
|
sources=[os.path.join('src', 'gmpy2.c')],
|
|
include_dirs=['./src'],
|
|
define_macros = defines)
|
|
|
|
setup(name = "gmpy2",
|
|
version = "2.1.0a1",
|
|
author = "Case Van Horsen",
|
|
author_email = "casevh@gmail.com",
|
|
license = "LGPL-3.0+",
|
|
url = "https://github.com/aleaxit/gmpy",
|
|
description = "gmpy2 interface to GMP/MPIR, MPFR, and MPC for Python 2.6+ and 3.4+",
|
|
data_files = [('', ['src/gmpy2.pxd']), ('gmpy2', ['src/gmpy2.h'])],
|
|
classifiers = [
|
|
'Development Status :: 3 - Alpha',
|
|
'Intended Audience :: Developers',
|
|
'Intended Audience :: Science/Research',
|
|
'License :: OSI Approved :: GNU Lesser General Public License v3 or later (LGPLv3+)',
|
|
'Natural Language :: English',
|
|
'Operating System :: MacOS :: MacOS X',
|
|
'Operating System :: Microsoft :: Windows',
|
|
'Operating System :: POSIX',
|
|
'Programming Language :: C',
|
|
'Programming Language :: Python :: 2',
|
|
'Programming Language :: Python :: 2.6',
|
|
'Programming Language :: Python :: 2.7',
|
|
'Programming Language :: Python :: 3',
|
|
'Programming Language :: Python :: 3.4',
|
|
'Programming Language :: Python :: 3.5',
|
|
'Programming Language :: Python :: 3.6',
|
|
'Programming Language :: Python :: Implementation :: CPython',
|
|
'Topic :: Scientific/Engineering :: Mathematics',
|
|
'Topic :: Software Development :: Libraries :: Python Modules',
|
|
],
|
|
keywords = "gmp mpir mpfr mpc multiple-precision arbitrary-precision precision bignum",
|
|
cmdclass = my_commands,
|
|
ext_modules = [gmpy2_ext]
|
|
)
|