import sys
from os.path import join
import setuptools
import versioneer
from setuptools.extension import Extension
from Cython.Build import cythonize
import numpy

COMMON_INCLUDE_DIRS = [
    './geoist/model/magmod',
    './geoist/model/magmod/include',
    join(sys.prefix, 'include'),
]
COMMON_INCLUDE_DIRS.append(numpy.get_include())

with open("README.md", "r") as fh:
    long_description = fh.read()

extensions = [
    Extension("geoist.forward.pfm._prism",
              ["geoist/forward/pfm/_prism.pyx"],
              include_dirs=[numpy.get_include()]
    ),
    Extension("geoist.inversion._ttime2d",
              ["geoist/inversion/_ttime2d.pyx"],
              include_dirs=[numpy.get_include()]
    ),
    Extension("geoist.inversion._wavefd",
              ["geoist/inversion/_wavefd.pyx"],
              include_dirs=[numpy.get_include()]
    ),
    Extension(
            'geoist.model.magmod._pymm',
            sources=[
                'geoist/model/magmod/pymm.c',
            ],
            libraries=[],
            library_dirs=[],
            include_dirs=COMMON_INCLUDE_DIRS,
        ),
    Extension(
            'geoist.model.magmod._pysunpos',
            sources=[
                'geoist/model/magmod/pysunpos.c',
            ],
            libraries=[],
            library_dirs=[],
            include_dirs=COMMON_INCLUDE_DIRS,
        ),
    Extension(
            'geoist.model.magmod._pytimeconv',
            sources=[
                'geoist/model/magmod/pytimeconv.c',
            ],
            libraries=[],
            library_dirs=[],
            include_dirs=COMMON_INCLUDE_DIRS,
        ),
]

exts = cythonize(extensions)


install_requires = ["numpy","matplotlib","scipy","h5py","numba","pandas",
                    "pytest","future","Cython",'statsmodels>=0.9.0',"PyWavelets",
                    "seaborn","patsy", "appdirs"]
setuptools.setup(
    name="geoist",
    version=versioneer.get_version(),
    cmdclass=versioneer.get_cmdclass(),
    author="IGP-GRAVITY",
    author_email="gravity_igp@sina.com",
    description="An Open-Source Geophysical Python Library for Geoscience Prototype Research",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/igp-gravity/geoist",
    packages=setuptools.find_packages(),
    ext_modules=exts,
    include_package_data=True,
    install_requires=install_requires,
    classifiers=(
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
    ),
)

# Build the f2py fortran extension
from numpy.distutils.core import setup, Extension

ext_cpwt = Extension(name='geoist.model.flex.cpwt',
                     sources=['geoist/model/flex/cpwt/cpwt.f90', 'geoist/model/flex/cpwt/cpwt_sub.f90'])
ext_flex = Extension(name='geoist.model.flex.flex',
                     sources=['geoist/model/flex/flex/flex.f90'])
ext_fa2boug= Extension(name='geoist.model.fa2boug',
                     sources=['geoist/model/fa2boug/FA2Bouguer.f90'])
setup(
    ext_modules=[ext_cpwt, ext_flex, ext_fa2boug],
    include_package_data=True
)