setup.py

import os
import re
import sys
import pathlib
import platform

from setuptools import Extension, setup


def mkconfig(srcpath: str, outpath: str, have_long_double: int = 0) -> None:
    data = pathlib.Path(srcpath).joinpath("CMakeLists.txt").read_text()
    mobj = re.search(
        r"""\
set\s*\(\s*PROJECT_VERSION_MAJOR\s+(?P<VERSION_MAJOR>\d+)\s*\)\s*
set\s*\(\s*PROJECT_VERSION_MINOR\s+(?P<VERSION_MINOR>\d+)\s*\)\s*
set\s*\(\s*PROJECT_VERSION_PATCH\s+(?P<VERSION_PATCH>\d+)\s*\)\s*
""",  # noqa: C092
        data,
        re.MULTILINE,
    )

    version_major = mobj.group("VERSION_MAJOR")
    version_minor = mobj.group("VERSION_MINOR")
    version_patch = mobj.group("VERSION_PATCH")

    if version_patch == "0":
        version_string = ".".join([version_major, version_minor])
    else:
        version_string = ".".join(
            [version_major, version_minor, version_patch],
        )

    bigendian = 0 if sys.byteorder == "little" else 1

    configdata = f"""\
#define GEOGRAPHICLIB_VERSION_STRING "{version_string}"
#define GEOGRAPHICLIB_VERSION_MAJOR {version_major}
#define GEOGRAPHICLIB_VERSION_MINOR {version_minor}
#define GEOGRAPHICLIB_VERSION_PATCH {version_patch}
#define GEOGRAPHICLIB_DATA "/usr/local/share/GeographicLib"

// These are macros which affect the building of the library
#define GEOGRAPHICLIB_HAVE_LONG_DOUBLE {have_long_double}
#define GEOGRAPHICLIB_WORDS_BIGENDIAN {bigendian}
#define GEOGRAPHICLIB_PRECISION 2

#if !defined(GEOGRAPHICLIB_SHARED_LIB)
#define GEOGRAPHICLIB_SHARED_LIB 0
#endif
"""  # noqa: C092

    outpath = pathlib.Path(outpath)
    if not outpath.exists():
        outpath.write_text(configdata)
        print(f"configuration file written to: {outpath}")
    else:
        print(f"configuration file already exists: {outpath}")


if os.name == "posix":
    if platform.system() == "FreeBSD":
        mandir = pathlib.Path("man")
    else:
        mandir = pathlib.Path("share/man")
    datafiles = [(
        str(mandir / "man1"),
        ["docs/man/geomodels-cli.1"],
    )]
else:
    datafiles = []


IGNORE_BUNDLED_LIBS_STR = os.environ.get("GEOMODELS_IGNORE_BUNDLED_LIBS")
IGNORE_BUNDLED_LIBS = IGNORE_BUNDLED_LIBS_STR in {"1", "ON", "TRUE", "YES"}
SRCPATH = pathlib.Path("extern/geographiclib")


def get_extra_compile_args():
    from setuptools._distutils.ccompiler import new_compiler

    compiler = new_compiler()
    if hasattr(compiler, "initialize"):
        compiler.initialize()
    if hasattr(compiler, "cc") and "cl" in compiler.cc:
        return ["/std:c++17"]

    return ["-std=c++17", "-Wall"]


if SRCPATH.exists() and not IGNORE_BUNDLED_LIBS:
    geographiclib_src = SRCPATH.glob("src/*.cpp")
    geographiclib_include = list(SRCPATH.glob("include"))[0]
    geomodels_ext = Extension(
        "geomodels._ext",
        sources=["geomodels/_ext.pyx"] + [str(p) for p in geographiclib_src],
        include_dirs=[str(geographiclib_include)],
        libraries=[],
        language="c++",
        extra_compile_args=get_extra_compile_args(),
    )
    outpath = geographiclib_include / "GeographicLib" / "Config.h"
    mkconfig(SRCPATH, outpath)
else:
    libname = os.environ.get("GEOMODELS_GEOGRAPHICLIB_NAME", "GeographicLib")
    geomodels_ext = Extension(
        "geomodels._ext",
        sources=["geomodels/_ext.pyx"],
        libraries=[libname],
        language="c++",
        extra_compile_args=get_extra_compile_args(),
    )

cython_directives = (
    geomodels_ext.cython_directives
    if hasattr(geomodels_ext, "cython_directives")
    else {}
)
cython_directives["freethreading_compatible"] = True
geomodels_ext.cython_directives = cython_directives

description = pathlib.Path("README.rst").read_text()
description = description.replace(".. doctest", "").replace(":doc:", "")


setup(
    long_description=description,
    long_description_content_type="text/x-rst",
    ext_modules=[geomodels_ext],
    data_files=datafiles,
)