diff --git a/setup.py b/setup.py index ea2eb59..842564b 100644 --- a/setup.py +++ b/setup.py @@ -7,36 +7,33 @@ # setup.py for pytorch module ################################################################################ import os -import pathlib +import shutil +from sys import platform as _platform from setuptools import setup, find_packages -from setuptools.command.install import install -from sys import platform as _platform + +retVal = -1 if _platform == "linux" or _platform == "linux2": lib_fname = 'build/lib/libleapct.so' retVal = os.system(r'sh ./etc/build.sh') - if retVal != 0: - print('Failed to compile!') - quit() - + elif _platform == "win32": lib_fname = r'win_build\bin\Release\libleapct.dll' retVal = os.system(r'.\etc\win_build_agn.bat') - if retVal != 0: - print('Failed to compile!') - quit() - - import site - copy_text = 'copy ' + str(lib_fname) + ' ' + str(os.path.join(site.getsitepackages()[1], 'libleapct.dll')) - os.system(copy_text) elif _platform == "darwin": lib_fname = 'build/lib/libleapct.dylib' retVal = os.system(r'sh ./etc/build.sh') - if retVal != 0: - print('Failed to compile!') - quit() + +if retVal != 0: + print('Failed to compile!') + quit() + +shutil.copy( + lib_fname, + os.path.join('src', 'leapct', os.path.basename(lib_fname)), +) setup( name='leapct', @@ -49,6 +46,6 @@ packages=find_packages("src"), package_dir={'': 'src'}, install_requires=['numpy', 'torch'], - py_modules=['leaptorch', 'leapctype', 'leap_filter_sequence', 'leap_preprocessing_algorithms'], - package_data={'': [lib_fname]}, + package_data={'leapct': [os.path.basename(lib_fname)]}, + include_package_data=True, ) diff --git a/src/__init__.py b/src/leapct/__init__.py similarity index 100% rename from src/__init__.py rename to src/leapct/__init__.py diff --git a/src/leap_filter_sequence.py b/src/leapct/leap_filter_sequence.py similarity index 100% rename from src/leap_filter_sequence.py rename to src/leapct/leap_filter_sequence.py diff --git a/src/leap_preprocessing_algorithms.py b/src/leapct/leap_preprocessing_algorithms.py similarity index 100% rename from src/leap_preprocessing_algorithms.py rename to src/leapct/leap_preprocessing_algorithms.py diff --git a/src/leapctype.py b/src/leapct/leapctype.py similarity index 100% rename from src/leapctype.py rename to src/leapct/leapctype.py diff --git a/src/leaptorch.py b/src/leapct/leaptorch.py similarity index 100% rename from src/leaptorch.py rename to src/leapct/leaptorch.py