From 35186e4048be38f644daa6f9cd13bf6245996922 Mon Sep 17 00:00:00 2001
From: huaibo <huaibovip@gmail.com>
Date: Wed, 27 May 2026 10:08:42 +0800
Subject: [PATCH] update setup.py

---
 setup.py                                      | 35 +++++++++----------
 src/{ => leapct}/__init__.py                  |  0
 src/{ => leapct}/leap_filter_sequence.py      |  0
 .../leap_preprocessing_algorithms.py          |  0
 src/{ => leapct}/leapctype.py                 |  0
 src/{ => leapct}/leaptorch.py                 |  0
 6 files changed, 16 insertions(+), 19 deletions(-)
 rename src/{ => leapct}/__init__.py (100%)
 rename src/{ => leapct}/leap_filter_sequence.py (100%)
 rename src/{ => leapct}/leap_preprocessing_algorithms.py (100%)
 rename src/{ => leapct}/leapctype.py (100%)
 rename src/{ => leapct}/leaptorch.py (100%)

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