-
Notifications
You must be signed in to change notification settings - Fork 5
/
setup.py
114 lines (93 loc) · 4.43 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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
import os
import re
import sys
import platform
import subprocess
from setuptools import setup, Extension, find_packages
from setuptools.command.build_ext import build_ext
from distutils.version import LooseVersion
with open('README.rst') as f:
readme = f.read()
class CMakeExtension(Extension):
def __init__(self, name, modName, sourcedir=''):
Extension.__init__(self, name, sources=[], language='c++')
self.modName = modName
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":
cmake_version = LooseVersion(re.search(r'version\s*([\d.]+)', out.decode()).group(1))
if cmake_version < '3.1.0':
raise RuntimeError("CMake >= 3.1.0 is required on Windows")
for ext in self.extensions:
self.build_extension(ext)
def build_extension(self, ext):
print('ext_full_path', self.get_ext_fullpath(ext.name))
extdir = os.path.abspath(os.path.dirname(self.get_ext_fullpath(ext.name)))
# required for auto-detection of auxiliary "native" libs
if not extdir.endswith(os.path.sep):
extdir += os.path.sep
cmake_args = ['-DCMAKE_LIBRARY_OUTPUT_DIRECTORY=' + extdir,
'-DPYTHON_EXECUTABLE=' + sys.executable,
'-DBUILD_PYTHON=ON']
cfg = 'Debug' if self.debug else 'Release'
build_args = ['--config', cfg]
if platform.system() == "Windows":
cmake_args += ['-DCMAKE_LIBRARY_OUTPUT_DIRECTORY_{}={}'.format(cfg.upper(), extdir)]
if sys.maxsize > 2**32:
cmake_args += ['-A', 'x64']
build_args += ['--', '/m']
else:
cmake_args += ['-DCMAKE_BUILD_TYPE=' + cfg]
build_args += ['--', '-j4']
cmake_args += ['-DCMAKE_INSTALL_PREFIX=' + '.']
env = os.environ.copy()
env['CXXFLAGS'] = '{} -DVERSION_INFO=\\"{}\\"'.format(env.get('CXXFLAGS', ''),
self.distribution.get_version())
if not os.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', '.', '--target', ext.modName] + build_args, cwd=self.build_temp)
# subprocess.check_call(['cmake', '--build', '.', '--target', 'install'], cwd=self.build_temp)
setup(
name='libSLM',
version='0.2.7',
author='Luke Parry',
url='https://github.com/drlukeparry/libslm',
long_description=readme,
long_description_content_type = 'text/x-rst',
description='libSLM is a python library for reading and writing to common machine build files used by commerical Additive Manufacturing systems',
ext_modules=[CMakeExtension('libSLM.slm', 'slm'),
CMakeExtension('libSLM.translators.eos', 'eos'),
CMakeExtension('libSLM.translators.mtt', 'mtt'),
CMakeExtension('libSLM.translators.slmsol', 'slmsol'),
CMakeExtension('libSLM.translators.realizer', 'realizer')],
cmdclass=dict(build_ext=CMakeBuild),
packages = ['libSLM', 'libSLM.translators'],
package_dir={'': 'python'},
keywords=['3D Printing', 'AM', 'Additive Manufacturing', 'Geometry', 'SLM', 'Selective Laser Melting'],
# tell setuptools that all packages will be under the 'src' directory
# and nowhere else
zip_safe=False,
classifiers=[
'Programming Language :: Python',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
'Programming Language :: Python :: 3.9',
'Natural Language :: English',
'Topic :: Scientific/Engineering'],
license="",
project_urls = {
'Documentation': 'https://pyslm.readthedocs.io/en/latest/',
'Source': 'https://github.com/drylukeparry/libslm',
'Tracker': 'https://github.com/drlukeparry/libslm/issues'
}
)