old_setup.py

"""A setuptools based setup module for EQcorrscan package.

:copyright:
    EQcorrscan developers.

:license:
    GNU Lesser General Public License, Version 3
    (https://www.gnu.org/copyleft/lesser.html)
"""

# Always prefer setuptools over distutils
import setuptools
from setuptools import setup, find_packages
import sys
import os
import shutil
import eqcorrscan
import numpy as np     # @UnusedImport # NOQA
from numpy.distutils.ccompiler import get_default_compiler
# To use a consistent encoding
from codecs import open
from os import path
from distutils.extension import Extension
import glob
import platform


try:
    from pypandoc import convert
    read_md = lambda f: convert(f, 'rst')
except ImportError:
    msg = ' '.join([])
    msg = ' '.join(["warning: pypandoc module not found,",
                    " could not convert Markdown to RST"])
    print(msg)
    read_md = lambda f: open(f, 'r').read()

# helper function for collecting export symbols from .def files

def export_symbols(*path):
    lines = open(os.path.join(*path), 'r').readlines()[2:]
    return [s.strip() for s in lines if s.strip() != '']
# check for MSVC
if platform.system() == "Windows" and (
        'msvc' in sys.argv or
        '-c' not in sys.argv and
        get_default_compiler() == 'msvc'):
    IS_MSVC = True
else:
    IS_MSVC = False

if IS_MSVC:
    # extra_args = []
    # extra_links = []
    # libs = []
    # lib_dirs = []
    # inc_dirs = []
    extra_args = ['/openmp', '/DFFTW_DLL']
    extra_links = ["/DFFTW_DLL"]
    libs = ['libfftw3-3']
    lib_dirs = [os.path.join(os.getcwd(), 'fftw'),
                os.path.join(sys.prefix, 'bin')]
    inc_dirs = [os.path.join(os.getcwd(), 'fftw')]
else:
    extra_args = ['-fopenmp', '-ftree-vectorize', '-msse2']
    extra_links = ['-lm', '-lgomp']
    libs = ['fftw3']
    lib_dirs = []
    inc_dirs = []

static_fftw_path = os.environ.get('STATIC_FFTW_DIR', None)
link_static_fftw = static_fftw_path is not None
if link_static_fftw:
    from pkg_resources import get_build_platform
    if get_build_platform() in ('win32', 'win-amd64'):
        lib_pre = ''
        lib_ext = '.lib'
    else:
        lib_pre = 'lib'
        lib_ext = '.a'
    extra_link_args = []
    for lib in libs:
        extra_link_args.append(
            os.path.join(static_fftw_path, lib_pre + lib + lib_ext))

# Check if we are on RTD and don't build extensions if we are.
READ_THE_DOCS = os.environ.get('READTHEDOCS', None) == 'True'

if not READ_THE_DOCS:
    ext = [Extension(
        "eqcorrscan.lib.libutils",
        sources=["eqcorrscan/utils/src/multi_corr.c"],
        export_symbols=export_symbols("eqcorrscan/utils/src/libutils.def"),
        extra_compile_args=extra_args, extra_link_args=extra_links,
        libraries=libs, library_dirs=lib_dirs, include_dirs=inc_dirs)]
    cmd_class = {}
else:
    ext = []
    cmd_class = {}


here = path.abspath(path.dirname(__file__))

# Get the long description from the relevant file
long_description = "EQcorrscan: matched-filter earthquake detection and " +\
    "analysis in Python.  Open-source routines for: systematic template " +\
    "creation, multi-parallel matched-filter detection, clustering of " +\
    "events, integration with SEISAN, SAC, QuakeML and NonLinLoc, " +\
    "magnitude calculation by singular value decomposition, and more!"

# Get a list of all the scripts not to be installed
scriptfiles = glob.glob('eqcorrscan/tutorials/*.py')
scriptfiles += glob.glob('eqcorrscan/scripts/*.py')

if sys.version_info.major == 2:
    if not READ_THE_DOCS:
        install_requires = ['numpy>=1.12.0',
                            'matplotlib>=1.3.0', 'scipy>=0.18', 'LatLon',
                            'cython']
    else:
        install_requires = ['numpy>=1.12.0',
                            'matplotlib>=1.3.0', 'LatLon']
else:
    if not READ_THE_DOCS:
        install_requires = ['numpy>=1.12.0',
                            'matplotlib>=1.3.0', 'scipy>=0.18', 'LatLon',
                            'cython']
    else:
        install_requires = ['numpy>=1.12.0',
                            'matplotlib>=1.3.0', 'LatLon']

if IS_MSVC:
    import distutils
    from distutils.msvc9compiler import MSVCCompiler

    if distutils.__version__.startswith('2.'):
        def _library_dir_option(self, dir):
            return '/LIBPATH:"%s"' % (dir)

        MSVCCompiler.library_dir_option = _library_dir_option

    # Sort linking issues with init exported symbols
    def _get_export_symbols(self, ext):
        return ext.export_symbols
    from distutils.command.build_ext import build_ext
    build_ext.get_export_symbols = _get_export_symbols

# add --with-system-libs command-line option if possible
def add_features():
    if 'setuptools' not in sys.modules:
        return {}

    class ExternalLibFeature(setuptools.Feature):
        def include_in(self, dist):
            global EXTERNAL_LIBS
            EXTERNAL_LIBS = True

        def exclude_from(self, dist):
            global EXTERNAL_LIBS
            EXTERNAL_LIBS = False

    return {
        'system-libs': ExternalLibFeature(
            'use of system C libraries',
            standard=False,
            EXTERNAL_LIBS=True
        )
    }


# install_requires.append('ConfigParser')
setup(
    name='EQcorrscan',

    # Versions should comply with PEP440.  For a discussion on single-sourcing
    # the version across setup.py and the project code, see
    # https://packaging.python.org/en/latest/single_source_version.html
    version=eqcorrscan.__version__,

    description='EQcorrscan - matched-filter earthquake detection and analysis',
    long_description=long_description,

    # The project's main homepage.
    url='https://github.com/calum-chamberlain/EQcorrscan',

    # Author details
    author='Calum Chamberlain',
    author_email='goride42@gmail.com',

    # Choose your license
    license='LGPL',

    # See https://pypi.python.org/pypi?%3Aaction=list_classifiers
    classifiers=[
        # How mature is this project? Common values are
        #   3 - Alpha
        #   4 - Beta
        #   5 - Production/Stable
        'Development Status :: 4 - Beta',

        # Indicate who your project is intended for
        'Intended Audience :: Science/Research',
        'Topic :: Scientific/Engineering',

        # Pick your license as you wish (should match "license" above)
        'License :: OSI Approved :: GNU Library or Lesser General Public ' +
        'License (LGPL)',

        # Specify the Python versions you support here. In particular, ensure
        # that you indicate whether you support Python 2, Python 3 or both.
        'Programming Language :: Python :: 2.7',
        'Programming Language :: Python :: 3.5',
        'Programming Language :: Python :: 3.4',
    ],

    # What does your project relate to?
    keywords='earthquake correlation detection match-filter',

    # You can just specify the packages manually here if your project is
    # simple. Or you can use find_packages().
    packages=find_packages(exclude=['docs', 'tests', 'test_data',
                                    'grid', 'detections', 'templates',
                                    'stack_templates', 'par', '.git']),

    scripts=scriptfiles,

    # List run-time dependencies here.  These will be installed by pip when
    # your project is installed. For an analysis of "install_requires" vs pip's
    # requirements files see:
    # https://packaging.python.org/en/latest/requirements.html
    install_requires=install_requires,

    # Test requirements for using pytest
    setup_requires=['pytest-runner'],
    tests_require=['pytest', 'pytest-cov', 'pytest-pep8', 'pytest-xdist'],

    # Build our extension
    cmdclass=cmd_class,
    ext_modules=ext,
    features=add_features()
)