setup.py

# -*- coding: utf-8 -*-
#File: setup.py
#Created: Sat Dec 15 19:40:30 2012
#Last Change: Sat Dec 15 19:42:45 2012

import os
import glob
from distutils.core import setup
from distutils.command.build_ext import build_ext
from distutils.extension import Extension
import distutils.ccompiler
import distutils.command.config
import distutils.sysconfig
from pkg_resources import require

topDir = os.getcwd()
sourceDir = "PyWCSTools" + os.path.sep + "wcssubs-3.9.0" + os.path.sep

#oFiles=glob.glob(sourceDir+"*.o")
#print oFiles
oFiles = [
    'PyWCSTools/wcssubs-3.9.0/cel.o', 'PyWCSTools/wcssubs-3.9.0/wcs.o',
    'PyWCSTools/wcssubs-3.9.0/proj.o', 'PyWCSTools/wcssubs-3.9.0/distort.o',
    'PyWCSTools/wcssubs-3.9.0/wcsinit.o', 'PyWCSTools/wcssubs-3.9.0/wcslib.o',
    'PyWCSTools/wcssubs-3.9.0/poly.o', 'PyWCSTools/wcssubs-3.9.0/platepos.o',
    'PyWCSTools/wcssubs-3.9.0/zpxpos.o', 'PyWCSTools/wcssubs-3.9.0/iget.o',
    'PyWCSTools/wcssubs-3.9.0/imio.o', 'PyWCSTools/wcssubs-3.9.0/dsspos.o',
    'PyWCSTools/wcssubs-3.9.0/tnxpos.o', 'PyWCSTools/wcssubs-3.9.0/wcscon.o',
    'PyWCSTools/wcssubs-3.9.0/fitsfile.o',
    'PyWCSTools/wcssubs-3.9.0/dateutil.o',
    'PyWCSTools/wcssubs-3.9.0/imhfile.o', 'PyWCSTools/wcssubs-3.9.0/lin.o',
    'PyWCSTools/wcssubs-3.9.0/fileutil.o',
    'PyWCSTools/wcssubs-3.9.0/wcstrig.o', 'PyWCSTools/wcssubs-3.9.0/slasubs.o',
    'PyWCSTools/wcssubs-3.9.0/sph.o', 'PyWCSTools/wcssubs-3.9.0/worldpos.o',
    'PyWCSTools/wcssubs-3.9.0/hget.o', 'PyWCSTools/wcssubs-3.9.0/hput.o'
]

exampleScripts = glob.glob("scripts" + os.path.sep + "*.py")


class build_PyWCSTools_ext(build_ext):
    def build_extensions(self):

        os.chdir(sourceDir)

        # This line is tough to make match the style guide
        cc = distutils.ccompiler.new_compiler(
            distutils.ccompiler.get_default_compiler())
        distutils.command.config.customize_compiler(cc)

        # Suppress warnings from compiling WCSTools wcssubs-3.9.0
        if "-Wstrict-prototypes" in cc.compiler_so:
            cc.compiler_so.pop(cc.compiler_so.index("-Wstrict-prototypes"))
        if "-Wall" in cc.compiler_so:
            cc.compiler_so.pop(cc.compiler_so.index("-Wall"))

        WCSToolsCFiles = glob.glob("*.c")
        WCSToolsCFiles.pop(WCSToolsCFiles.index("wcs_wrap.c"))
        WCSToolsCFiles.pop(WCSToolsCFiles.index("wcscon_wrap.c"))
        cc.compile(WCSToolsCFiles)

        os.chdir(topDir)

        build_ext.build_extensions(self)


setup(
    name='astLib',
    version='99.0.0',
    url='http://astlib.sourceforge.net',
    download_url='http://sourceforge.net/project/platformdownload.php?group_id=202537',
    author='Matt Hilton',
    author_email='matt.hilton@mykolab.com',
    classifiers=[
        'Development Status :: 5 - Production/Stable',
        'Environment :: Console', 'Intended Audience :: Science/Research',
        'License :: OSI Approved :: GNU Library or Lesser General Public License (LGPL)',
        'Natural Language :: English', 'Operating System :: POSIX',
        'Programming Language :: Python',
        'Topic :: Scientific/Engineering :: Astronomy',
        'Topic :: Software Development :: Libraries'
    ],
    description=('A set of python modules for producing simple plots, '
    'statistics, common calculations, coordinate conversions, and manipulating '
    'FITS images with World Coordinate System (WCS) information.'),
    long_description="""astLib is a set of Python modules that provides some
    tools for research astronomers. It can be used for simple plots,
    statistics, common calculations, coordinate conversions, and manipulating
    FITS images with World Coordinate System (WCS) information through
    PyWCSTools - a simple wrapping of WCSTools by Jessica Mink.
    PyWCSTools is distributed (and developed) as part of astLib.""",
    packages=['astLib', 'PyWCSTools'],
    package_data={'astLib': ['data/*']},
    cmdclass={"build_ext": build_PyWCSTools_ext},
    scripts=exampleScripts,
    ext_modules=[
        Extension('PyWCSTools._wcscon', [sourceDir + "wcscon_wrap.c"],
                  extra_objects=oFiles), Extension('PyWCSTools._wcs',
                                                   [sourceDir + "wcs_wrap.c"],
                                                   extra_objects=oFiles)
    ])