netgen/setup.py

160 lines
4.6 KiB
Python
Raw Normal View History

2021-10-27 19:50:07 +05:00
import glob
2023-10-03 16:05:22 +05:00
import os.path
2021-10-27 19:50:07 +05:00
import sys
2023-05-12 14:26:38 +05:00
import pathlib
2023-10-03 16:05:22 +05:00
import sysconfig
2021-10-27 19:50:07 +05:00
from skbuild import setup
import skbuild.cmaker
from subprocess import check_output
setup_requires = ['pybind11-stubgen==2.5']
2021-10-27 19:50:07 +05:00
2024-05-13 16:43:53 +05:00
pyprefix = pathlib.Path(sys.prefix).as_posix()
2021-10-27 19:50:07 +05:00
def install_filter(cmake_manifest):
print(cmake_manifest)
return cmake_manifest
def _patched_parse_manifests(self):
paths = \
glob.glob(os.path.join(skbuild.cmaker.CMAKE_BUILD_DIR(), "netgen", "install_manifest*.txt"))
try:
return [self._parse_manifest(path) for path in paths][0]
except IndexError:
return []
# we are using the netgen superbuild (to download and build some dependencies)
# patch the parse_manifests function to point to the actual netgen cmake project within the superbuild
skbuild.cmaker.CMaker._parse_manifests = _patched_parse_manifests
git_version = check_output(['git', 'describe', '--tags']).decode('utf-8').strip()
version = git_version[1:].split('-')
if len(version)>2:
version = version[:2]
if len(version)>1:
version = '.post'.join(version) + '.dev'
else:
version = version[0]
2021-10-27 19:50:07 +05:00
2023-10-03 16:05:22 +05:00
py_install_dir = os.path.relpath(sysconfig.get_path('platlib'), sysconfig.get_path('data')).replace('\\','/')
2021-10-27 19:50:07 +05:00
name = "netgen-mesher"
arch = None
cmake_args = [
f'-DNETGEN_VERSION_GIT={git_version}',
f'-DNETGEN_VERSION_PYTHON={version}',
2021-10-27 19:50:07 +05:00
]
if 'NETGEN_ARCH' in os.environ and os.environ['NETGEN_ARCH'] == 'avx2':
# build for avx2 architecture
if 'darwin' in sys.platform:
flag = "'-Xarch_x86_64;-march=core-avx2'"
elif 'win' in sys.platform:
flag = '/arch:AVX2'
else:
flag = '-march=core-avx2'
cmake_args += [f'-DNG_COMPILE_FLAGS={flag}']
2021-10-27 19:50:07 +05:00
if 'NETGEN_CCACHE' in os.environ:
cmake_args += [f'-DUSE_CCACHE=ON']
2022-02-16 14:03:42 +05:00
packages = ['netgen', 'pyngcore']
2024-05-13 16:43:53 +05:00
have_mpi = False
2021-10-27 19:50:07 +05:00
if 'darwin' in sys.platform:
cmake_args += [
2022-02-23 14:49:27 +05:00
'-DNG_INSTALL_DIR_LIB=netgen',
'-DNG_INSTALL_DIR_PYTHON=.',
'-DNG_INSTALL_DIR_BIN=bin',
'-DNG_INSTALL_DIR_CMAKE=netgen/cmake',
'-DNG_INSTALL_DIR_INCLUDE=netgen/include',
'-DNG_INSTALL_DIR_RES=share',
2021-10-27 19:50:07 +05:00
]
2024-05-13 16:43:53 +05:00
if os.path.exists('/usr/local/include/mpi.h'):
have_mpi = True
cmake_args += [
'-DOPENMPI_INCLUDE_DIR=/usr/local/include',
]
2021-10-27 19:50:07 +05:00
elif 'win' in sys.platform:
cmake_args += [
'-A Win64',
2022-02-23 14:49:27 +05:00
'-DNG_INSTALL_DIR_BIN=netgen',
'-DNG_INSTALL_DIR_PYTHON=.',
'-DNG_INSTALL_DIR_LIB=netgen/lib',
'-DNG_INSTALL_DIR_CMAKE=netgen/cmake',
'-DNG_INSTALL_DIR_INCLUDE=netgen/include',
2021-10-27 19:50:07 +05:00
]
2024-05-13 16:43:53 +05:00
py_libdir = pathlib.Path(sys.prefix) / 'Library'
lib_file = py_libdir / 'lib' / 'impi.lib'
include_dir = py_libdir / 'include'
if lib_file.exists():
have_mpi = True
cmake_args += [
f'-DINTEL_MPI_INCLUDE_DIR={include_dir.as_posix()}',
f'-DINTEL_MPI_LIBRARY={lib_file.as_posix()}',
]
2021-10-27 19:50:07 +05:00
elif 'linux' in sys.platform:
name_dir = name.replace('-','_')
cmake_args += [
f'-DNG_INSTALL_DIR_LIB={py_install_dir}/{name_dir}.libs',
2022-02-23 14:49:27 +05:00
'-DNG_INSTALL_DIR_BIN=bin',
'-DNG_INSTALL_DIR_INCLUDE=include/netgen',
2022-09-15 13:46:19 +05:00
'-DTCL_INCLUDE_PATH=/usr/include',
'-DTK_INCLUDE_PATH=/usr/include',
2021-10-27 19:50:07 +05:00
]
2024-05-13 16:43:53 +05:00
mpich_include = '/opt/mpich/include'
openmpi_include = '/opt/openmpi/include'
if os.path.exists(mpich_include+'/mpi.h'):
have_mpi = True
cmake_args += [
f'-DMPICH_INCLUDE_DIR={mpich_include}',
]
if os.path.exists(openmpi_include+'/mpi.h'):
have_mpi = True
cmake_args += [
f'-DOPENMPI_INCLUDE_DIR={openmpi_include}',
]
2022-02-05 02:06:19 +05:00
packages = []
2021-10-27 19:50:07 +05:00
2024-05-13 16:43:53 +05:00
if have_mpi:
cmake_args += [
'-DUSE_MPI=ON',
2024-05-29 13:51:45 +05:00
'-DUSE_MPI_WRAPPER=ON',
2024-05-13 16:43:53 +05:00
]
2021-10-27 19:50:07 +05:00
cmake_args += [
'-DUSE_SUPERBUILD:BOOL=ON',
'-DUSE_CCACHE:BOOL=ON',
'-DUSE_GUI=ON',
'-DUSE_NATIVE_ARCH=OFF',
'-DBUILD_ZLIB=ON',
'-DBUILD_OCC=ON',
'-DUSE_OCC=ON',
'-DBUILD_FOR_CONDA=ON',
2021-10-28 16:59:38 +05:00
f'-DNETGEN_PYTHON_PACKAGE_NAME={name}',
'-DBUILD_STUB_FILES=ON',
2021-10-27 19:50:07 +05:00
]
2023-08-30 19:04:22 +05:00
cmake_args += [f'-DCMAKE_PREFIX_PATH={pyprefix}', f'-DPython3_ROOT_DIR={pyprefix}']
2021-10-27 19:50:07 +05:00
setup(
name=name,
version=version,
description="Netgen",
author='The Netgen team',
license="LGPL2.1",
2022-02-05 02:06:19 +05:00
packages=packages,
#package_dir={'netgen': 'python'},
2021-10-27 19:50:07 +05:00
tests_require=['pytest'],
2022-02-04 20:44:22 +05:00
#include_package_data=True,
2021-10-27 19:50:07 +05:00
cmake_process_manifest_hook=install_filter,
cmake_args = cmake_args,
setup_requires=setup_requires,
entry_points={
'console_scripts': [
'netgen = netgen.__main__:main',
],
},
)