#!/usr/bin/env python # Two environmental variables influence this script. # # GDAL_CONFIG: the path to a gdal-config program that points to GDAL headers, # libraries, and data files. # # PACKAGE_DATA: if defined, GDAL and PROJ4 data files will be copied into the # source or binary distribution. This is essential when creating self-contained # binary wheels. import logging import os import pprint import shutil import subprocess import sys from setuptools import setup from setuptools.extension import Extension logging.basicConfig() log = logging.getLogger() # python -W all setup.py ... if 'all' in sys.warnoptions: log.level = logging.DEBUG def check_output(cmd): # since subprocess.check_output doesn't exist in 2.6 # we wrap it here. try: out = subprocess.check_output(cmd) return out.decode('utf') except AttributeError: # For some reasone check_output doesn't exist # So fall back on Popen p = subprocess.Popen(cmd, stdout=subprocess.PIPE) out, err = p.communicate() return out def copy_data_tree(datadir, destdir): try: shutil.rmtree(destdir) except OSError: pass shutil.copytree(datadir, destdir) # Parse the version from the rasterio module. with open('rasterio/__init__.py') as f: for line in f: if line.find("__version__") >= 0: version = line.split("=")[1].strip() version = version.strip('"') version = version.strip("'") continue with open('VERSION.txt', 'w') as f: f.write(version) # Use Cython if available. try: from Cython.Build import cythonize except ImportError: cythonize = None # By default we'll try to get options via gdal-config. On systems without, # options will need to be set in setup.cfg or on the setup command line. include_dirs = [] library_dirs = [] libraries = [] extra_link_args = [] gdal_output = [None]*3 try: import numpy include_dirs.append(numpy.get_include()) except ImportError: log.critical("Numpy and its headers are required to run setup(). Exiting.") sys.exit(1) try: gdal_config = os.environ.get('GDAL_CONFIG', 'gdal-config') for i, flag in enumerate(("--cflags", "--libs", "--datadir")): gdal_output[i] = check_output([gdal_config, flag]).strip() for item in gdal_output[0].split(): if item.startswith("-I"): include_dirs.extend(item[2:].split(":")) for item in gdal_output[1].split(): if item.startswith("-L"): library_dirs.extend(item[2:].split(":")) elif item.startswith("-l"): libraries.append(item[2:]) else: # e.g. -framework GDAL extra_link_args.append(item) except Exception as e: if os.name == "nt": log.info(("Building on Windows requires extra options to setup.py to locate needed GDAL files.\n" "More information is available in the README.")) else: log.warning("Failed to get options via gdal-config: %s", str(e)) # Conditionally copy the GDAL data. To be used in conjunction with # the bdist_wheel command to make self-contained binary wheels. if os.environ.get('PACKAGE_DATA'): destdir = 'rasterio/gdal_data' if gdal_output[2]: log.info("Copying gdal data from %s" % gdal_output[2]) copy_data_tree(gdal_output[2], destdir) else: # check to see if GDAL_DATA is defined gdal_data = os.environ.get('GDAL_DATA', None) if gdal_data: log.info("Copying gdal_data from %s" % gdal_data) copy_data_tree(gdal_data, destdir) # Conditionally copy PROJ.4 data. projdatadir = os.environ.get('PROJ_LIB', '/usr/local/share/proj') if os.path.exists(projdatadir): log.info("Copying proj_data from %s" % projdatadir) copy_data_tree(projdatadir, 'rasterio/proj_data') ext_options = dict( include_dirs=include_dirs, library_dirs=library_dirs, libraries=libraries, extra_link_args=extra_link_args) if not os.name == "nt": # These options fail on Windows if using Visual Studio ext_options['extra_compile_args'] = ['-Wno-unused-parameter', '-Wno-unused-function'] log.debug('ext_options:\n%s', pprint.pformat(ext_options)) # When building from a repo, Cython is required. if os.path.exists("MANIFEST.in") and "clean" not in sys.argv: log.info("MANIFEST.in found, presume a repo, cythonizing...") if not cythonize: log.critical( "Cython.Build.cythonize not found. " "Cython is required to build from a repo.") sys.exit(1) ext_modules = cythonize([ Extension( 'rasterio._base', ['rasterio/_base.pyx'], **ext_options), Extension( 'rasterio._io', ['rasterio/_io.pyx'], **ext_options), Extension( 'rasterio._copy', ['rasterio/_copy.pyx'], **ext_options), Extension( 'rasterio._features', ['rasterio/_features.pyx'], **ext_options), Extension( 'rasterio._drivers', ['rasterio/_drivers.pyx'], **ext_options), Extension( 'rasterio._warp', ['rasterio/_warp.pyx'], **ext_options), Extension( 'rasterio._fill', ['rasterio/_fill.pyx', 'rasterio/rasterfill.cpp'], **ext_options), Extension( 'rasterio._err', ['rasterio/_err.pyx'], **ext_options), Extension( 'rasterio._example', ['rasterio/_example.pyx'], **ext_options), ], quiet=True) # If there's no manifest template, as in an sdist, we just specify .c files. else: ext_modules = [ Extension( 'rasterio._base', ['rasterio/_base.c'], **ext_options), Extension( 'rasterio._io', ['rasterio/_io.c'], **ext_options), Extension( 'rasterio._copy', ['rasterio/_copy.c'], **ext_options), Extension( 'rasterio._features', ['rasterio/_features.c'], **ext_options), Extension( 'rasterio._drivers', ['rasterio/_drivers.c'], **ext_options), Extension( 'rasterio._warp', ['rasterio/_warp.cpp'], **ext_options), Extension( 'rasterio._fill', ['rasterio/_fill.cpp', 'rasterio/rasterfill.cpp'], **ext_options), Extension( 'rasterio._err', ['rasterio/_err.c'], **ext_options), Extension( 'rasterio._example', ['rasterio/_example.c'], **ext_options), ] with open('README.rst') as f: readme = f.read() # Runtime requirements. inst_reqs = [ 'affine>=1.0', 'cligj>=0.2.0', 'Numpy>=1.7', 'snuggs>=1.3.1'] if sys.version_info < (3, 4): inst_reqs.append('enum34') setup_args = dict( name='rasterio', version=version, description="Fast and direct raster I/O for use with Numpy and SciPy", long_description=readme, classifiers=[ 'Development Status :: 4 - Beta', 'Intended Audience :: Developers', 'Intended Audience :: Information Technology', 'Intended Audience :: Science/Research', 'License :: OSI Approved :: BSD License', 'Programming Language :: C', 'Programming Language :: Python :: 2.6', 'Programming Language :: Python :: 2.7', 'Programming Language :: Python :: 3.3', 'Programming Language :: Python :: 3.4', 'Topic :: Multimedia :: Graphics :: Graphics Conversion', 'Topic :: Scientific/Engineering :: GIS'], keywords='raster gdal', author='Sean Gillies', author_email='sean@mapbox.com', url='https://github.com/mapbox/rasterio', license='BSD', package_dir={'': '.'}, packages=['rasterio', 'rasterio.rio'], entry_points=''' [console_scripts] rio=rasterio.rio.main:main_group [rasterio.rio_commands] bounds=rasterio.rio.features:bounds calc=rasterio.rio.calc:calc edit-info=rasterio.rio.info:edit env=rasterio.rio.info:env info=rasterio.rio.info:info insp=rasterio.rio.info:insp mask=rasterio.rio.features:mask merge=rasterio.rio.merge:merge rasterize=rasterio.rio.features:rasterize sample=rasterio.rio.sample:sample shapes=rasterio.rio.features:shapes stack=rasterio.rio.bands:stack warp=rasterio.rio.warp:warp transform=rasterio.rio.info:transform ''', include_package_data=True, ext_modules=ext_modules, zip_safe=False, install_requires=inst_reqs, extras_require={ 'ipython': ['ipython>=2.0']}) if os.environ.get('PACKAGE_DATA'): setup_args['package_data'] = {'rasterio': ['gdal_data/*', 'proj_data/*']} setup(**setup_args)