-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
83 lines (69 loc) · 2.95 KB
/
setup.py
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
71
72
73
74
75
76
77
78
79
80
81
82
83
import os
import re
import sys
import platform
import subprocess
from setuptools import setup, find_packages, Extension
from setuptools.command.build_ext import build_ext
from distutils.version import LooseVersion
class CMakeExtension(Extension):
def __init__(self, name, sourcedir=''):
Extension.__init__(self, name, sources=[])
self.sourcedir = os.path.abspath(sourcedir)
class CMakeBuild(build_ext):
def run(self):
try:
out = subprocess.check_output(['cmake', '--version'])
except OSError:
raise RuntimeError("CMake must be installed to build the following extensions: " +
", ".join(e.name for e in self.extensions))
if platform.system() == "Windows":
raise NotImplementedError
for ext in self.extensions:
self.build_extension(ext)
def build_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 = 'Release'
build_args = ['--config', cfg]
if platform.system() == "Windows":
raise NotImplementedError
else:
cmake_args += ['-DCMAKE_BUILD_TYPE=' + cfg]
build_args += ['--', '-j']
self.build_temp = self.build_temp.replace("temp", "lib")
env = os.environ.copy()
if not os.path.exists(self.build_temp):
os.makedirs(self.build_temp)
vcpkg_dir = os.path.join(self.build_temp, 'vcpkg')
vcpkg_dir = os.path.abspath(vcpkg_dir)
if not os.path.exists(vcpkg_dir):
subprocess.check_call(['git', 'clone', 'https://github.com/microsoft/vcpkg.git', vcpkg_dir])
subprocess.check_call(['./bootstrap-vcpkg.sh'], cwd=vcpkg_dir)
vcpkg_triplet = 'x64-linux'
subprocess.check_call(['./vcpkg', 'install',
'eigen3',
'spectra',
'directx-headers',
'directxmath',
'directxtex',
'directxmesh',
'--triplet', vcpkg_triplet], cwd=vcpkg_dir)
subprocess.check_call(['./vcpkg', 'integrate', 'install'], cwd=vcpkg_dir)
cmake_args += [
'-DCMAKE_TOOLCHAIN_FILE={}/scripts/buildsystems/vcpkg.cmake'.format(vcpkg_dir),
'-DVCPKG_TARGET_TRIPLET={}'.format(vcpkg_triplet)]
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='pyatlas',
version='0.0.1',
author='Xiang Feng',
author_email='xfeng.cg@gmail.com',
description='Minimal wrapper of UVAtlas on linux',
long_description='',
ext_modules=[CMakeExtension('pyatlas', '.')],
cmdclass=dict(build_ext=CMakeBuild),
zip_safe=False,
)