57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
import os
|
|
import shutil
|
|
from setuptools import setup, Extension
|
|
from setuptools.command.build_ext import build_ext
|
|
|
|
|
|
class CMakeExtension(Extension):
|
|
def __init__(self, name):
|
|
super().__init__(name, sources=[])
|
|
|
|
|
|
class CMakeBuild(build_ext):
|
|
def run(self):
|
|
# build and install
|
|
cwd = os.getcwd()
|
|
build_dir = os.path.join(cwd, 'build')
|
|
os.makedirs(build_dir, exist_ok=True)
|
|
os.chdir(build_dir)
|
|
install_dir = os.path.join(build_dir, 'install')
|
|
if not self.dry_run:
|
|
self.spawn(
|
|
['cmake', '..', '-DCMAKE_INSTALL_PREFIX=%s' % install_dir])
|
|
if self.debug:
|
|
self.spawn(['cmake', '--build', '.', '--config',
|
|
'Debug', '--target', 'install'])
|
|
else:
|
|
self.spawn(['cmake', '--build', '.', '--config',
|
|
'Release', '--target', 'install'])
|
|
os.chdir(cwd)
|
|
# move
|
|
for ext in self.extensions:
|
|
dst_dir = os.path.dirname(self.get_ext_fullpath(ext.name))
|
|
lib_dir = os.path.join(dst_dir, 'NDIlib')
|
|
os.makedirs(lib_dir, exist_ok=True)
|
|
files = os.listdir(install_dir)
|
|
for filename in files:
|
|
filepath = os.path.join(install_dir, filename)
|
|
if os.path.isfile(filepath):
|
|
shutil.copy(filepath, lib_dir)
|
|
|
|
|
|
setup(
|
|
name='ndi-python',
|
|
version='5.1.1',
|
|
description='Wrapper package for NDI SDK python bindings.',
|
|
author='Naoto Kondo <cgigcp3yqt@gmail.com>',
|
|
url='https://github.com/buresu/ndi-python',
|
|
license="MIT",
|
|
python_requires='>=3.6',
|
|
install_requires=['numpy'],
|
|
ext_modules=[CMakeExtension('NDIlib')],
|
|
cmdclass={'build_ext': CMakeBuild},
|
|
packages=['NDIlib'],
|
|
package_data={'NDIlib': ['*.so*', '*.pyd', '*.dll', '*.dylib', '*.txt']},
|
|
zip_safe=False
|
|
)
|