# -*- coding: utf-8 -*- import platform import shutil from pathlib import Path from subprocess import check_output from setuptools import Extension, setup try: from Cython.Build import cythonize from Cython.Distutils import build_ext except ImportError: have_cython = False else: have_cython = True LATEST_WAVPACK_VERSION = "5.8.0" SRC_FOLDER = "src/wavpack_numcodecs" def get_build_extensions(): include_dirs = [f"{SRC_FOLDER}/include"] runtime_library_dirs = [] extra_link_args = [] if platform.system() == "Linux": libraries = ["wavpack"] if shutil.which("wavpack") is not None: out = check_output(["wavpack", "--version"]) print(f"WavPack is installed!\n{out.decode()}") extra_link_args = ["-L/usr/local/lib/", "-L/usr/bin/"] runtime_library_dirs = ["/usr/local/lib/", "/usr/bin/"] else: print("Using shipped libraries") wavpack_libraries_folder = Path(f"{SRC_FOLDER}/libraries/{LATEST_WAVPACK_VERSION}") available_glibc_builds = [ p.name for p in wavpack_libraries_folder.iterdir() if p.is_dir() and "linux" in p.name ] available_glibc_versions = [ p[p.find("glibc") + len("glibc") :] for p in available_glibc_builds ] glibc_version = platform.libc_ver()[1] if glibc_version not in available_glibc_versions: raise RuntimeError( f"Could not find a matching the system's glibc version {glibc_version}. " f"Available builds: {available_glibc_versions}" ) distr_folder = f"linux-x86_64-glibc{glibc_version}" extra_link_args = [f"-L{wavpack_libraries_folder}/{distr_folder}"] runtime_library_dirs = [ f"$ORIGIN/libraries/{LATEST_WAVPACK_VERSION}/{distr_folder}", ] # hack shutil.copy( f"{wavpack_libraries_folder}/{distr_folder}/libwavpack.so", f"{wavpack_libraries_folder}/{distr_folder}/libwavpack.so.1", ) elif platform.system() == "Darwin": libraries = ["wavpack"] assert shutil.which("wavpack") is not None, ( f"WavPack needs to be installed externally for MacOS platforms.\n" f"You can use homebrew: \n\t >>> brew install wavpack\nor compile it from source:" f"\n\t >>> wget https://www.wavpack.com/wavpack-{LATEST_WAVPACK_VERSION}.tar.bz2" f"\n\t >>> tar -xf wavpack-{LATEST_WAVPACK_VERSION}.tar.bz2" f"\n\t >>> cd wavpack-{LATEST_WAVPACK_VERSION}\n\t >>> ./configure\n\t >>> sudo make install\n\t >>> cd .." ) print("wavpack is installed!") extra_link_args = ["-L~/include/", "-L/usr/local/include/", "-L/usr/include"] runtime_library_dirs = ["/opt/homebrew/lib/", "/opt/homebrew/Cellar"] for rt_dir in runtime_library_dirs: extra_link_args.append(f"-L{rt_dir}") else: # windows libraries = ["wavpackdll"] # add library folder to PATH and copy .dll in the src if "64" in platform.architecture()[0]: lib_path = f"{SRC_FOLDER}/libraries/{LATEST_WAVPACK_VERSION}/windows-x86_64" else: lib_path = f"{SRC_FOLDER}/libraries/{LATEST_WAVPACK_VERSION}/windows-x86_32" extra_link_args = [f"/LIBPATH:{lib_path}"] for libfile in Path(lib_path).iterdir(): shutil.copy(libfile, SRC_FOLDER) if have_cython: print("Building with Cython") sources_compat_ext = [f"{SRC_FOLDER}/compat_ext.pyx"] sources_wavpack_ext = [f"{SRC_FOLDER}/wavpack.pyx"] else: sources_compat_ext = [f"{SRC_FOLDER}/compat_ext.c"] sources_wavpack_ext = [f"{SRC_FOLDER}/wavpack.c"] extensions = [ Extension("wavpack_numcodecs.compat_ext", sources=sources_compat_ext, extra_compile_args=[]), Extension( "wavpack_numcodecs.wavpack", sources=sources_wavpack_ext, include_dirs=include_dirs, libraries=libraries, extra_link_args=extra_link_args, runtime_library_dirs=runtime_library_dirs, ), ] if have_cython: extensions = cythonize(extensions) return extensions extensions = get_build_extensions() cmdclass = {"build_ext": build_ext} if have_cython else {} setup( ext_modules=extensions, cmdclass=cmdclass, )