# Fix numpy installation using setuptools

__builtins__.__NUMPY_SETUP__ = False

import numpy

try:

return check_output([gdal_config, '--%s' % option]).decode('utf-8').strip()

except CalledProcessError as e:

ret = False

with open('gdal_python_cxx11_test.cpp', 'wt') as f:

f.write("""

#if __cplusplus < 201103L

#error "C++11 required"

#endif

int main () { return 0; }""")

f.close()

extra_postargs = None

if compiler_flag:

extra_postargs = [compiler_flag]

if os.name == 'posix':

# Redirect stderr to /dev/null to hide any error messages

# from the compiler.

devnull = open(os.devnull, 'w')

oldstderr = os.dup(sys.stderr.fileno())

os.dup2(devnull.fileno(), sys.stderr.fileno())

try:

compiler.compile([f.name], extra_postargs=extra_postargs)

ret = True

except CompileError:

pass

os.dup2(oldstderr, sys.stderr.fileno())

devnull.close()

else:

try:

compiler.compile([f.name], extra_postargs=extra_postargs)

ret = True

except CompileError:

pass

os.unlink('gdal_python_cxx11_test.cpp')

if os.path.exists('gdal_python_cxx11_test.o'):

os.unlink('gdal_python_cxx11_test.o')

GDAL_CONFIG = 'gdal-config'

def run(self):

inst_gdal_version = self.get_gdal_config('version')

if inst_gdal_version != GDAL_VERSION:

raise GDALConfigError('Version mismatch %s != %s' % (

inst_gdal_version, GDAL_VERSION))

build_ext.run(self)

def initialize_options(self):

build_ext.initialize_options(self)

self.gdaldir = None

self.gdal_config = self.GDAL_CONFIG

self.extra_cflags = []

def get_gdal_config(self, option):

return fetch_config(option, gdal_config=self.gdal_config)

def build_extensions(self):

# Add a -std=c++11 or similar flag if needed

ct = self.compiler.compiler_type

if ct == 'unix' and not supports_cxx11(self.compiler):

cxx11_flag = None

if supports_cxx11(self.compiler, '-std=c++11'):

cxx11_flag = '-std=c++11'

if cxx11_flag:

for ext in self.extensions:

# gdalconst builds as a .c file

if ext.name != 'osgeo._gdalconst':

ext.extra_compile_args += [cxx11_flag]

build_ext.build_extensions(self)

def finalize_options(self):

if self.libraries is None:

self.libraries = ['gdal', ]

build_ext.finalize_options(self)

if ENV_GDALHOME in os.environ:

print('GDAL prefix from environment variable %s' % ENV_GDALHOME)

self.gdaldir = os.environ[ENV_GDALHOME]

else:

self.gdaldir = self.get_gdal_config('prefix')

self.include_dirs.append(get_numpy_include())

self.library_dirs.append(os.path.join(self.gdaldir, 'lib'))

self.include_dirs.append(os.path.join(self.gdaldir, 'include', 'gdal'))

self.include_dirs.append(os.path.join(self.gdaldir, 'include'))

cflags = self.get_gdal_config('cflags')

if cflags:

self.extra_cflags = cflags.split()

def build_extension(self, ext):

ext.extra_compile_args.extend(self.extra_cflags)

'osgeo._gdal',

sources=['extensions/gdal_wrap.cpp'],

extra_compile_args=extra_compile_args,

'osgeo._gdalconst',

sources=['extensions/gdalconst_wrap.c'],

extra_compile_args=extra_compile_args,

'osgeo._osr',

sources=['extensions/osr_wrap.cpp'],

extra_compile_args=extra_compile_args,

'osgeo._ogr',

sources=['extensions/ogr_wrap.cpp'],

extra_compile_args=extra_compile_args,

'osgeo._gdal_array',

sources=['extensions/gdal_array_wrap.cpp'],

extra_compile_args=extra_compile_args,

gdal_module,

gdalconst_module,

osr_module,

ogr_module,

gdal_array_module,

'Development Status :: 5 - Production/Stable',

'Intended Audience :: Developers',

'Intended Audience :: Science/Research',

'License :: OSI Approved :: MIT License',

'Operating System :: OS Independent',

'Programming Language :: Python :: 2',

'Programming Language :: C',

'Programming Language :: C++',

'Topic :: Scientific/Engineering :: GIS',

'Topic :: Scientific/Engineering :: Information Analysis',