setup.py

Go to the documentation of this file.

import os
import numpy
from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext

DATAVERSION = os.getenv("DATAVERSION")
ITMLIBDIR = os.getenv("ITMLIBDIR")
SYS = os.getenv("SYS")
UAL = os.getenv("UAL")
SVNVERSION = os.getenv("SVNVERSION")
ITMCONSTANTS = "%s/%s/include" % (os.getenv("ITM_ITMCONSTANTS_DIR"), os.getenv("ITM_ITMCONSTANTS_VERSION"))

library_dirs=[UAL + "/lib", UAL + "/lowlevel", UAL + "/fortraninterface"]

libraries=["amns-"+SVNVERSION+"-"+DATAVERSION+"-"+SYS, "UALLowLevel"]

if SYS == "amd64_pgi_10":
    library_dirs.extend(["../lib/amd64_pgi", "/E/L/pgi-10.2-libso"])
    libraries.extend(["UALFORTRANInterface_pgi", "pgf90rtl", "pgc"])
elif SYS == "amd64_g95_0.93":
    library_dirs.extend(["../lib/amd64_g95_0.93", "/E/L/g95-0.92/gcc-lib/x86_64-unknown-linux-gnu/4.0.3"])
    libraries.extend(["UALFORTRANInterface_g95"])
elif SYS == "amd64_intel_12":
    library_dirs.extend(["../lib/amd64_intel_12", ])
    libraries.extend(["UALFORTRANInterface_ifort"])
elif SYS == "amd64_intel_17":
    library_dirs.extend(["../lib/amd64_intel_17", ])
    libraries.extend(["UALFORTRANInterface_ifort"])
elif SYS == "amd64_intel_18":
    library_dirs.extend(["../lib/amd64_intel_18", ])
    libraries.extend(["UALFORTRANInterface_ifort"])
elif SYS == "amd64_gfortran_4.7":
    library_dirs.extend(["../lib/amd64_gfortran_4.7", ])
    libraries.extend(["UALFORTRANInterface_ifort"])
elif SYS == "amd64_gfortran_6.1":
    library_dirs.extend(["../lib/amd64_gfortran_6.1", ])
    libraries.extend(["UALFORTRANInterface_ifort"])
else:
    raise ValueError("Unsupported Fortran compiler (SYS=" + SYS + ")")


setup(
    name = "amns",
    version = DATAVERSION,
    author = "David Coster",
    author_email = "David.Coster@ipp.mpg.de",
    description = ("Python interface to the ITM AMNS system"),
    license = "JRC",
    url = "http://www.efda-itm.eu/ITM/html/amns_public.html",
    cmdclass = {'build_ext': build_ext},
    ext_modules = [Extension("amns", ["amns.pyx"],
                             library_dirs=library_dirs,
                             libraries=libraries,
                             include_dirs = [numpy.get_include(), ITMCONSTANTS]),
                   Extension("camns_interface", ["camns_interface.pxd"],
                             library_dirs=library_dirs,
                             libraries=libraries,
                             include_dirs = [ITMCONSTANTS])]
)