compiler, uv for windows

Author: alexlib

pyproject.toml

@@ -11,7 +11,7 @@ requires-python = ">=3.11"
 dependencies = ["numpy>=1.20"]
 
 [tool.cibuildwheel]
-build-frontend = "build[uv]"
+build-frontend = "build"
 skip = "*-musllinux*"
 test-skip = "*"
 environment = { EASYSBA_LAPACK_LIBS = "openblas" }

setup.py

@@ -11,6 +11,14 @@ def _parse_csv_env(name, default_value):
     return [item.strip() for item in raw.split(",") if item.strip()]
 
 
+def _parse_path_env(name):
+    raw = os.environ.get(name, "")
+    if not raw:
+        return []
+    normalized = raw.replace(",", os.pathsep)
+    return [item.strip() for item in normalized.split(os.pathsep) if item.strip()]
+
+
 sources = [
     "easysba/_easysba.cpp",
     "src/sba_chkjac.c",
@@ -22,6 +30,26 @@ def _parse_csv_env(name, default_value):
 
 libraries = _parse_csv_env("EASYSBA_LAPACK_LIBS", "lapack,blas")
 
+include_dirs = [pybind11.get_include(), "src"]
+library_dirs = []
+extra_link_args = []
+
+include_dirs.extend(_parse_path_env("EASYSBA_INCLUDE_DIRS"))
+library_dirs.extend(_parse_path_env("EASYSBA_LIBRARY_DIRS"))
+
+if sys.platform == "darwin" and not library_dirs:
+    for prefix in ("/opt/homebrew/opt/openblas", "/usr/local/opt/openblas"):
+        inc = os.path.join(prefix, "include")
+        lib = os.path.join(prefix, "lib")
+        if os.path.isdir(lib):
+            if os.path.isdir(inc):
+                include_dirs.append(inc)
+            library_dirs.append(lib)
+
+if sys.platform == "darwin":
+    for libdir in library_dirs:
+        extra_link_args.append(f"-Wl,-rpath,{libdir}")
+
 extra_compile_args = []
 if sys.platform == "win32":
     extra_compile_args.extend(["/O2"])
@@ -34,10 +62,12 @@ def _parse_csv_env(name, default_value):
     Extension(
         "easysba._easysba",
         sources=sources,
-        include_dirs=[pybind11.get_include(), "src"],
+        include_dirs=include_dirs,
         language="c++",
         libraries=libraries,
+        library_dirs=library_dirs,
         extra_compile_args=extra_compile_args,
+        extra_link_args=extra_link_args,
     )
 ]