setup.py

#!/usr/bin/env python
import sys
from os import path
from setuptools import setup, Extension
from setuptools.command.build_ext import build_ext

desc = """Local Field Components (or lighting fast calculator) at
muon sites for the muesr package."""
with open('README.md','r') as file:
    long_desc = file.read()

# ugly hack to keep tests under the python folder
sys.path.insert(0, path.join(path.dirname(__file__),'python'))

sources = ['simplesum.c', \
           'rotatesum.c', \
           'fastincommsum.c', \
           'vec3.c', \
           'mat3.c', \
           'pile.c', \
           'dipolartensor.c']

src_sources = []
for s in sources:
    src_sources.append(
    path.join('.','src',s)
    )


openmp_compile_args = []
openmp_link_args = []


## In case of missing numpy array headers
try:
    from numpy import get_include as numpy_get_include
    numpy_include_dir = [numpy_get_include()]
except:
    numpy_include_dir = []
    print("Warning: could not add numpy include dir.")

# Ugly hack to set compiler flags 
COMPILE_ARGS = {'msvc':[],'gcc':[],'unix':[]}
LINK_ARGS = {'msvc':[],'gcc':[],'unix':[]}

for compiler, args in [
        ('msvc', ['/EHsc', '/DHUNSPELL_STATIC']),
        ('gcc', ['-O3', '-g0']),
        ('unix', ['-O3', '-g0'])]:
    COMPILE_ARGS[compiler] += args
# add math lib if needed
for compiler, args in [
        ('msvc', []),
        ('unix', ['-lm']),
        ('gcc', ['-lm'])]:
    LINK_ARGS[compiler] += args

# Ugly hack to have openMP as option
if "--with-openmp" in sys.argv:
    for compiler, args in [
            ('msvc', ['/openmp']),
            ('unix', ['-fopenmp']),
            ('gcc', ['-fopenmp'])]:
        COMPILE_ARGS[compiler] += args
    for compiler, args in [
            ('msvc', []),
            ('unix', ['-lgomp']),
            ('gcc', ['-lgomp'])]:
        LINK_ARGS[compiler] += args

    sys.argv.remove("--with-openmp")

class build_ext_compiler_check(build_ext):
    def build_extensions(self):
        
        compiler = self.compiler.compiler_type
        cargs = COMPILE_ARGS[compiler]
        for ext in self.extensions:
            ext.extra_compile_args = cargs
            
        largs = LINK_ARGS[compiler]
        for ext in self.extensions:
            ext.extra_link_args = largs
        
        build_ext.build_extensions(self)

setup(name='mulfc',
      version='0.0.2',
      description=desc,
      long_description=long_desc,
      author='Pietro Bonfa',
      author_email='pietro.bonfa@fis.unipr.it',
      url='https://github.com/bonfus/muLFC',
      packages=['mulfc',],
      ext_modules=[Extension('lfclib', sources = ['python/lfclib.c',]+src_sources,
                                      libraries=[],
                                      include_dirs=numpy_include_dir + [path.join('.','src')])],
     package_dir={'mulfc': 'python' },
     install_requires=[
          'numpy >= 1.8',
     ],
     test_suite="tests",
     cmdclass={ 'build_ext': build_ext_compiler_check }
     )