"""
Minimal setup.py for building gswc.
"""


import os
import shutil
import sys

import numpy
from setuptools import Extension, setup
from setuptools.command.build_ext import build_ext as _build_ext

rootpath = os.path.abspath(os.path.dirname(__file__))


DEFINE_MACROS = [
    ("NPY_NO_DEPRECATED_API", "NPY_1_7_API_VERSION"),
    # 0x030B0000 -> 3.11
    ("Py_LIMITED_API", "0x030B0000"),
    ("CYTHON_LIMITED_API", None),
]

def read(*parts):
    return open(os.path.join(rootpath, *parts)).read()


class build_ext(_build_ext):
    # Extension builder from pandas without the cython stuff
    def build_extensions(self):
        numpy_incl = numpy.get_include()

        for ext in self.extensions:
            if hasattr(ext, "include_dirs") and not numpy_incl in ext.include_dirs:
                ext.include_dirs.append(numpy_incl)
        _build_ext.build_extensions(self)


# MSVC can't handle C complex, and distutils doesn't seem to be able to
# let us force C++ compilation of .c files, so we use the following hack for
# Windows.
if sys.platform == "win32":
    cext = "cpp"
    shutil.copy(
        "src/c_gsw/gsw_oceanographic_toolbox.c",
        "src/c_gsw/gsw_oceanographic_toolbox.cpp",
    )
    shutil.copy("src/c_gsw/gsw_saar.c", "src/c_gsw/gsw_saar.cpp")
else:
    cext = "c"

ufunc_src_list = [
    "src/_ufuncs.c",
    "src/c_gsw/gsw_oceanographic_toolbox." + cext,
    "src/c_gsw/gsw_saar." + cext,
]

config = {
    "ext_modules": [
        Extension(
            "gsw._gsw_ufuncs",
            ufunc_src_list,
            define_macros=DEFINE_MACROS,
            py_limited_api=True)],
    "include_dirs": [os.path.join(rootpath, "src", "c_gsw")],
    "cmdclass": {"build_ext": build_ext},
    "options": {"bdist_wheel": {"py_limited_api": "cp311"}},
}

setup(**config)