#!/usr/bin/env python # -*- coding: utf-8 -*- # # Copyright (C) 2013 Ghislain Vaillant # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see . import os import sys import subprocess # Define global path variables setup_dir = dir = os.path.dirname(os.path.abspath(__file__)) package_name = 'pynfft' package_dir = os.path.join(setup_dir, package_name) # Define utility functions to build the extensions def get_common_extension_args(): import numpy common_extension_args = dict( libraries=['nfft3_threads', 'nfft3', 'fftw3_threads', 'fftw3', 'm'], library_dirs=[], include_dirs=[numpy.get_include()], extra_compile_args='-O3 -fomit-frame-pointer ' '-fstrict-aliasing -ffast-math'.split(), ) return common_extension_args def get_extensions(): from setuptools import Extension ext_modules = [] common_extension_args = get_common_extension_args() ext_modules.append(Extension( name=package_name+'.nfft', sources=[os.path.join(package_dir, 'nfft.c')], **common_extension_args ) ) ext_modules.append(Extension( name=package_name+'.solver', sources=[os.path.join(package_dir, 'solver.c')], **common_extension_args ) ) ext_modules.append(Extension( name=package_name+'.util', sources=[os.path.join(package_dir, 'util.c')], **common_extension_args ) ) return ext_modules def get_cython_extensions(): from setuptools import Extension from Cython.Build import cythonize ext_modules = [] common_extension_args = get_common_extension_args() ext_modules.append(Extension( name=package_name+'.nfft', sources=[os.path.join(package_dir, 'nfft.pyx')], **common_extension_args ) ) ext_modules.append(Extension( name=package_name+'.solver', sources=[os.path.join(package_dir, 'solver.pyx')], **common_extension_args ) ) ext_modules.append(Extension( name=package_name+'.util', sources=[os.path.join(package_dir, 'util.pyx')], **common_extension_args ) ) return cythonize(ext_modules, language_level=2) # BEFORE importing distutils, remove MANIFEST. distutils doesn't properly # update it when the contents of directories change. if os.path.exists('MANIFEST'): os.remove('MANIFEST') # Define custom clean command from setuptools import Command class CleanCommand(Command): """Custom distutils command to clean the .so and .pyc files.""" user_options = [("all", "a", "")] def initialize_options(self): self.all = True self._clean_me = [] self._clean_trees = [] self._clean_exclude = [] # clean Cython generated files and cache for root, dirs, files in os.walk(package_dir): for f in files: if f in self._clean_exclude: continue if os.path.splitext(f)[-1] in ('.pyc', '.so', '.o', '.pyo', '.pyd', '.c', '.orig'): self._clean_me.append(os.path.join(root, f)) for d in dirs: if d == '__pycache__': self._clean_trees.append(os.path.join(root, d)) # clean build and sdist directories in root for d in ('build', 'dist'): if os.path.exists(d): self._clean_trees.append(d) def finalize_options(self): pass def run(self): for clean_me in self._clean_me: try: os.unlink(clean_me) except Exception: pass for clean_tree in self._clean_trees: try: shutil.rmtree(clean_tree) except Exception: pass cmdclass = {'clean': CleanCommand} LONG_DESCRIPTION = '''"The NFFT is a C subroutine library for computing the nonequispaced discrete Fourier transform (NDFT) in one or more dimensions, of arbitrary input size, and of complex data." The NFFT library is licensed under GPLv2 and available at: http://www-user.tu-chemnitz.de/~potts/nfft/index.php This wrapper provides a somewhat Pythonic access to some of the core NFFT library functionalities and is largely inspired from the pyFFTW project developped by Henry Gomersall (http://hgomersall.github.io/pyFFTW/). This project is still work in progress and is still considered beta quality. In particular, the API is not yet frozen and is likely to change as the development continues. Please consult the documentation and changelog for more information.''' CLASSIFIERS = [ 'Programming Language :: Cython', 'Programming Language :: Python', 'Programming Language :: Python :: 2', 'Programming Language :: Python :: 2.6', 'Programming Language :: Python :: 2.7', 'Programming Language :: Python :: 3', 'Programming Language :: Python :: 3.2', 'Programming Language :: Python :: 3.3', 'Programming Language :: Python :: 3.4', 'Development Status :: 4 - Beta', 'License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)', 'Operating System :: POSIX :: Linux', 'Intended Audience :: Developers', 'Intended Audience :: Science/Research', 'Topic :: Scientific/Engineering', 'Topic :: Scientific/Engineering :: Mathematics', 'Topic :: Multimedia :: Sound/Audio :: Analysis', ] MAJOR = 1 MINOR = 4 MICRO = 2 ISRELEASED = True VERSION = '%d.%d.%d' % (MAJOR, MINOR, MICRO) # borrowed from scipy def git_version(): def _minimal_ext_cmd(cmd): # construct minimal environment env = {} for k in ['SYSTEMROOT', 'PATH']: v = os.environ.get(k) if v is not None: env[k] = v # LANGUAGE is used on win32 env['LANGUAGE'] = 'C' env['LANG'] = 'C' env['LC_ALL'] = 'C' out = subprocess.Popen(cmd, stdout = subprocess.PIPE, env=env).communicate()[0] return out try: out = _minimal_ext_cmd(['git', 'rev-parse', 'HEAD']) GIT_REVISION = out.strip().decode('ascii') except OSError: GIT_REVISION = "Unknown" return GIT_REVISION # borrowed from scipy def get_version_info(): FULLVERSION = VERSION if os.path.exists('.git'): GIT_REVISION = git_version() elif os.path.exists('pynfft/version.py'): # must be a source distribution, use existing version file # load it as a separate module in order not to load __init__.py import imp version = imp.load_source('pynfft.version', 'pynfft/version.py') GIT_REVISION = version.git_revision else: GIT_REVISION = "Unknown" if not ISRELEASED: FULLVERSION += '.dev-' + GIT_REVISION[:7] return FULLVERSION, GIT_REVISION # borrowed from scipy def write_version_py(filename='pynfft/version.py'): cnt = """ # THIS FILE IS GENERATED FROM SETUP.PY short_version = '%(version)s' version = '%(version)s' full_version = '%(full_version)s' git_revision = '%(git_revision)s' release = %(isrelease)s if not release: version = full_version """ FULLVERSION, GIT_REVISION = get_version_info() f = open(filename, 'w') try: f.write(cnt % {'version': VERSION, 'full_version' : FULLVERSION, 'git_revision' : GIT_REVISION, 'isrelease': str(ISRELEASED)}) finally: f.close() def setup_package(): from setuptools import setup # Get current version FULLVERSION, GIT_REVISION = get_version_info() # Refresh version file # write_version_py() # Figure out whether to add ``*_requires = ['numpy']``. build_requires = [] try: import numpy except: build_requires = ['numpy>=1.6',] # Common setup args setup_args = dict( name = 'pyNFFT', packages = ['pynfft'], version = FULLVERSION, author = 'Ghislain Vaillant', author_email = 'ghisvail@gmail.com', description = 'A pythonic wrapper around NFFT', long_description = LONG_DESCRIPTION, url = 'https://github.com/ghisvail/pyNFFT.git', cmdclass = cmdclass, classifiers = CLASSIFIERS, platforms=['Linux', 'Unix'], test_suite='nose.collector', setup_requires = build_requires, install_requires = build_requires, ) if len(sys.argv) >= 2 and ('--help' in sys.argv[1:] or sys.argv[1] in ('--help-commands', 'egg_info', '--version', 'clean')): # For these actions, NumPy is not required. pass else: try: from Cython.Distutils import build_ext have_cython = True except: have_cython = False if have_cython: extensions = get_cython_extensions() else: extensions = get_extensions() setup_args['packages'] = ['pynfft', 'pynfft.tests'] setup_args['ext_modules'] = extensions setup(**setup_args) if __name__ == '__main__': setup_package()