setup.py

from setuptools import setup, Extension
from Cython.Build import cythonize
import os
import numpy

# make the c compiler clang in llvm (use this, because it allows for openmp)
# the below one is if you're on macos and are using Johnny's old laptop!
#os.environ["CC"] = "/usr/local/opt/llvm/bin/clang"
# os.environ["CC"] = "gcc"
# this one is for the new homebrew llvm clang
os.environ["CC"] = "/opt/homebrew/opt/llvm/bin/clang"


ext_modules = [
        Extension(
            "musr_decoherence.cython_polarisation",
            ["musr_decoherence/cython_polarisation.pyx"],
            extra_compile_args = ['-fopenmp'],
            extra_link_args = ['-fopenmp']
            )
        ]

setup(
    name='musr_decoherence',
    version='0.1',
    packages=['musr_decoherence'],
    install_requires=['numpy', 'scipy', 'matplotlib', 'ase', 'lmfit'],
    extras_require = {
        'gpu' : ['cupy']
    },
    ext_modules = cythonize(ext_modules),
    include_dirs=[numpy.get_include()],
    url='',
    license='',
    author='John Wilkinson',
    author_email='john.wilkinson@physics.ox.ac.uk',
    description='musr_decoherence -- calculate and fit F--mu--F states by taking into account further nearest-neighbours.'
)