1import os 2import re 3import sys 4import platform 5import subprocess 6 7from setuptools import setup, Extension 8from setuptools.command.build_ext import build_ext 9from distutils.version import LooseVersion 10from distutils.sysconfig import get_config_var 11from distutils.sysconfig import get_python_inc 12 13CMAKE_ADDITIONAL_OPT = [] 14if '--' in sys.argv: 15 i = sys.argv.index('--') 16 CMAKE_ADDITIONAL_OPT = sys.argv[i+1:] 17 sys.argv = sys.argv[:i] 18 19class CMakeExtension(Extension): 20 21 def __init__(self, name, sourcedir=''): 22 Extension.__init__(self, name, sources=[]) 23 self.sourcedir = os.path.abspath(sourcedir) 24 25 26class CMakeBuild(build_ext): 27 28 def run(self): 29 try: 30 out = subprocess.check_output(['cmake', '--version']) 31 except OSError: 32 raise RuntimeError("CMake must be installed to build the following extensions: " + 33 ", ".join(e.name for e in self.extensions)) 34 35 if platform.system() == "Windows": 36 cmake_version = LooseVersion( 37 re.search(r'version\s*([\d.]+)', out.decode()).group(1)) 38 if cmake_version < '3.1.0': 39 raise RuntimeError("CMake >= 3.1.0 is required on Windows") 40 41 for ext in self.extensions: 42 self.build_extension(ext) 43 44 def build_extension(self, ext): 45 extdir = os.path.abspath(os.path.dirname( 46 self.get_ext_fullpath(ext.name))) 47 48 python_library = str(get_config_var('LIBDIR')) 49 python_include_directory = str(get_python_inc()) 50 51 cmake_args = ['-DCMAKE_LIBRARY_OUTPUT_DIRECTORY=' + extdir, 52 '-DPYTHON_EXECUTABLE=' + sys.executable, 53 '-DPYTHON_INCLUDE_DIR=' + python_include_directory, ] 54 55 cfg = 'Debug' if self.debug else 'Release' 56 build_args = ['--config', cfg] 57 58 if platform.system() == "Windows": 59 cmake_args += [ 60 '-DCMAKE_LIBRARY_OUTPUT_DIRECTORY_{}={}'.format(cfg.upper(), extdir)] 61 if sys.maxsize > 2**32: 62 cmake_args += ['-A', 'x64'] 63 build_args += ['--', '/m'] 64 else: 65 cmake_args += ['-DCMAKE_BUILD_TYPE=' + cfg] 66 build_args += ['--', '-j2'] 67 cmake_args += CMAKE_ADDITIONAL_OPT 68 69 env = os.environ.copy() 70 env['CXXFLAGS'] = '{} -DVERSION_INFO=\\"{}\\"'.format(env.get('CXXFLAGS', ''), 71 self.distribution.get_version()) 72 if not os.path.exists(self.build_temp): 73 os.makedirs(self.build_temp) 74 75 subprocess.check_call(['cmake', ext.sourcedir] + 76 cmake_args, cwd=self.build_temp, env=env) 77 subprocess.check_call(['cmake', '--build', '.'] + 78 build_args, cwd=self.build_temp) 79 80setup( 81 name='pyigl', 82 version='0.0.1', 83 author='Geometric Computing Lab @ NYU', 84 author_email='info@geometriccomputing.org', 85 description='', 86 long_description='', 87 ext_modules=[CMakeExtension('pyigl')], 88 cmdclass=dict(build_ext=CMakeBuild), 89 zip_safe=False, 90) 91