simple-cpp-python/setup.py at master · fteicht/simple-cpp-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
importos
importre
importsys
importplatform
importsubprocess

fromsetuptoolsimportsetup, Extension
fromsetuptools.command.build_extimportbuild_ext
fromdistutils.versionimportLooseVersion


classCMakeExtension(Extension):
def__init__(self, name, sourcedir=''):
Extension.__init__(self, name, sources=[])
self.sourcedir=os.path.abspath(sourcedir)


classCMakeBuild(build_ext):
defrun(self):
try:
out=subprocess.check_output(['cmake', '--version'])
exceptOSError:
raiseRuntimeError("CMake must be installed to build the following extensions: "+
", ".join(e.nameforeinself.extensions))

ifplatform.system() =="Windows":
cmake_version=LooseVersion(re.search(r'version\s*([\d.]+)', out.decode()).group(1))
ifcmake_version<'3.1.0':
raiseRuntimeError("CMake >= 3.1.0 is required on Windows")

forextinself.extensions:
self.build_extension(ext)

defbuild_extension(self, ext):
extdir=os.path.abspath(os.path.dirname(self.get_ext_fullpath(ext.name)))
cmake_args= ['-DCMAKE_LIBRARY_OUTPUT_DIRECTORY='+extdir,
'-DPYTHON_EXECUTABLE='+sys.executable]

cfg='Debug'ifself.debugelse'Release'
build_args= ['--config', cfg]

ifplatform.system() =="Windows":
cmake_args+= ['-DCMAKE_LIBRARY_OUTPUT_DIRECTORY_{}={}'.format(cfg.upper(), extdir)]
ifsys.maxsize>2**32:
cmake_args+= ['-A', 'x64']
build_args+= ['--', '/m']
else:
cmake_args+= ['-DCMAKE_BUILD_TYPE='+cfg]
build_args+= ['--', '-j2']

env=os.environ.copy()
env['CXXFLAGS'] ='{} -DVERSION_INFO=\\"{}\\"'.format(env.get('CXXFLAGS', ''),
self.distribution.get_version())
ifnotos.path.exists(self.build_temp):
os.makedirs(self.build_temp)
subprocess.check_call(['cmake', ext.sourcedir] +cmake_args, cwd=self.build_temp, env=env)
subprocess.check_call(['cmake', '--build', '.'] +build_args, cwd=self.build_temp)

setup(
name='simple_cpp_python',
packages=['simple_cpp_python'],
version='0.0.1',
author='Florent Teichteil-Koenigsbuch',
author_email='florent.teichteil@gmail.com',
description='Simple demonstration of mixing parallel C++ and Python codes',
long_description='',
ext_modules=[CMakeExtension('simple_cpp_python/simple_cpp_python')],
cmdclass=dict(build_ext=CMakeBuild),
zip_safe=False,
)