setup.py

"""
Build with:
     python setup.py build_ext --inplace

See this site for building on windows-64:
	https://github.com/cython/cython/wiki/CythonExtensionsOnWindows

See this example for packaging cython modules:
        https://github.com/thearn/simple-cython-example/blob/master/setup.py
"""

#from distutils.core import setup
#from distutils.extension import Extension
try:
    from setuptools import setup
    from setuptools import Extension
    from setuptools.dist import Distribution
except ImportError:
    from distutils.core import setup
    from distutils.extension import Extension

class BinaryDistribution(Distribution):
    def is_pure(self):
        return False


try:
    from Cython.Build import cythonize
    use_cython = True
    ext = 'pyx'
except:
    use_cython = False
    ext = 'c'

#from Cython.Distutils import build_ext
import os
import numpy

# Check for a C compiler
import distutils.ccompiler
if isinstance(distutils.ccompiler.new_compiler(), distutils.ccompiler.CCompiler):
    compile=True
else:
    compile=False

if compile:
    os.environ["CC"]='cc'

    extensions =[
        Extension("sfoda.ugrid.ugridutils",["sfoda/ugrid/ugridutils.{}".format(ext)],
            include_dirs=[numpy.get_include()],
            extra_compile_args=
                ['-shared', '-pthread', '-fPIC', '-fwrapv', '-O2', '-Wall',
                '-fno-strict-aliasing'],),
        Extension("sfoda.ugrid.searchutils",["sfoda/ugrid/searchutils.{}".format(ext)],
            include_dirs=[numpy.get_include()],
            extra_compile_args=['-O3','-ffast-math','-march=native','-fopenmp'],
            extra_link_args=['-fopenmp'],),
    ]

    if use_cython:
        ext_modules = cythonize(extensions, language_level=3)
        cmdclass = {}
    else:
        cmdclass= {} #{"build_ext": extensions}
        ext_modules = extensions
else:
    cmdclass = {}
    ext_modules = None


setup(
    name = "sfoda",
    packages=[
        'sfoda',
        'sfoda.utils',
        'sfoda.ugrid',
        'sfoda.suntans',
        'sfoda.roms',
        'sfoda.tides',    
        'sfoda.dbase',
        'sfoda.dataio.conversion',
        'sfoda.dataio.datadownload',
        ],
    #package_dir={'sfoda':''},
    ext_modules = ext_modules,
    cmdclass = cmdclass,
    version="0.1.3",
    description='Stuff For Ocean Data Analysis',
    author='Matt Rayson',
    author_email='matt.rayson@uwa.edu.au',
    #packages=find_packages(exclude=["*.tests", "*.tests.*", "tests.*", "tests"]),
    install_requires=[
      'numpy>=1.20.0',
      'scipy',
      'matplotlib',
      'netcdf4',
      'xarray',
      'pyyaml',
      #'numba',
      'pyproj',
      'dask[complete]',
      #'gdal',
      #'shapely',
      ],
    license='LICENSE',
    include_package_data=True,
    distclass=BinaryDistribution,

)