65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
|
from setuptools import setup
|
|||
|
|
from setuptools.extension import Extension
|
|||
|
|
from Cython.Distutils import build_ext
|
|||
|
|
from Cython.Build import cythonize
|
|||
|
|
import numpy
|
|||
|
|
from pathlib import Path
|
|||
|
|
import shutil
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
class MyBuildExt(build_ext):
|
|||
|
|
def run(self):
|
|||
|
|
build_ext.run(self)
|
|||
|
|
|
|||
|
|
build_dir = Path(self.build_lib)
|
|||
|
|
root_dir = Path(__file__).parent
|
|||
|
|
target_dir = build_dir if not self.inplace else root_dir
|
|||
|
|
|
|||
|
|
self.copy_file(Path('./WindSpeedTools') / '__init__.py', root_dir, target_dir)
|
|||
|
|
#self.copy_file(Path('./pkg2') / '__init__.py', root_dir, target_dir)
|
|||
|
|
self.copy_file(Path('.') / '__init__.py', root_dir, target_dir)
|
|||
|
|
def copy_file(self, path, source_dir, destination_dir):
|
|||
|
|
if not (source_dir / path).exists():
|
|||
|
|
return
|
|||
|
|
shutil.copyfile(str(source_dir / path), str(destination_dir / path))
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 判断平台和编译器，选择正确的OpenMP标志
|
|||
|
|
compile_args = []
|
|||
|
|
link_args = []
|
|||
|
|
|
|||
|
|
if sys.platform == "win32":
|
|||
|
|
# 在Windows上使用MSVC编译器
|
|||
|
|
compile_args = ['/openmp'] # 仅传递给编译器
|
|||
|
|
# 链接器不需要此选项，因此 link_args 为空
|
|||
|
|
else:
|
|||
|
|
# 在Linux/macOS上使用GCC/Clang编译器
|
|||
|
|
compile_args = ['-fopenmp']
|
|||
|
|
link_args = ['-fopenmp'] # 对于GCC，通常需要同时传递给编译器和链接器
|
|||
|
|
|
|||
|
|
setup(
|
|||
|
|
name="MyModule",
|
|||
|
|
ext_modules=cythonize(
|
|||
|
|
[
|
|||
|
|
#Extension("pkg1.*", ["root/pkg1/*.py"]),
|
|||
|
|
Extension("pkg2.*", ["WindSpeedModel.pyx"],
|
|||
|
|
# 添加OpenMP支持
|
|||
|
|
extra_compile_args=compile_args, # 选项只给编译器
|
|||
|
|
extra_link_args=link_args, # 链接器使用单独的选项列表
|
|||
|
|
),
|
|||
|
|
#Extension("1.*", ["root/*.py"])
|
|||
|
|
],
|
|||
|
|
build_dir="build",
|
|||
|
|
compiler_directives=dict(
|
|||
|
|
always_allow_keywords=True
|
|||
|
|
)),
|
|||
|
|
cmdclass=dict(
|
|||
|
|
build_ext=MyBuildExt
|
|||
|
|
),
|
|||
|
|
packages=[],
|
|||
|
|
include_dirs=[numpy.get_include()],
|
|||
|
|
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 指令： python setup.py build_ext --inplace
|