Tweak for Arm Mac

pyflct/setup_package.py

@@ -1,5 +1,6 @@
 import os
 import sys
+import platform
 import subprocess
 from glob import glob
 from collections import defaultdict
@@ -27,14 +28,17 @@ def get_extensions():
     else:
         cfg["libraries"].append("m")
         # WE assume we have brew installed on Mac OS to provide FFTW3
-        if sys.platform.lower() == "darwin":
+        if platform.system().lower() == "darwin":
             brew_path = (
                 subprocess.run(["brew", "--prefix"], stdout=subprocess.PIPE).stdout.decode("utf-8").replace("\n", "")
             )
             cfg["include_dirs"].append(f"{brew_path}/include")
-        if not (sys.platform.lower() == "darwin" and sys.processor().lower() == "arm"):
+        if platform.system().lower() == "darwin" and platform.processor().lower() == "arm":
+            os.environ["DYLD_LIBRARY_PATH"] = os.environ.get("DYLD_LIBRARY_PATH", []) + [f"{brew_path}/lib"]
+        else:
             # This does not get registered on ARM Mac via Brew
             cfg["libraries"].append("fftw3")
         cfg["include_dirs"].append("/usr/include")
+        cfg["include_dirs"].append("/usr/local/include")
         cfg["extra_compile_args"].extend(["-O3", "-w", "-fomit-frame-pointer", "-fPIC"])
     return [Extension("pyflct._flct", **cfg)]