Homec4science
Diffusion GooseFEM (gh-pages) Browse

rGOOSEFEM/setup.py
996c1fc89a77gh-pages

Last Change
Raw File

setup.py
View Options

from setuptools import setup, Extension
import re
import os
import pybind11
import pyxtensor
import subprocess
header = open('include/GooseFEM/config.h','r').read()
major = re.split(r'(.*)(\#define GOOSEFEM_VERSION_MAJOR\ )([0-9]+)(.*)', header)[3]
minor = re.split(r'(.*)(\#define GOOSEFEM_VERSION_MINOR\ )([0-9]+)(.*)', header)[3]
patch = re.split(r'(.*)(\#define GOOSEFEM_VERSION_PATCH\ )([0-9]+)(.*)', header)[3]
__version__ = '.'.join([major, minor, patch])
include_dirs = [
os.path.abspath('include/'),
pyxtensor.find_pyxtensor(),
pyxtensor.find_pybind11(),
pyxtensor.find_xtensor(),
pyxtensor.find_xtl(),
pyxtensor.find_eigen()]
build = pyxtensor.BuildExt
xsimd = pyxtensor.find_xsimd()
if xsimd:
if len(xsimd) > 0:
include_dirs += [xsimd]
build.c_opts['unix'] += ['-march=native', '-DXTENSOR_USE_XSIMD']
build.c_opts['msvc'] += ['/DXTENSOR_USE_XSIMD']
git_hash = subprocess.check_output(["git", "rev-parse", "HEAD"]).strip().decode('UTF-8')
git_branch = subprocess.check_output(["git", "rev-parse", "--abbrev-ref", "HEAD"]).strip().decode('UTF-8')
build.c_opts['unix'] += [
'-DGOOSEFEM_GIT_HASH="{0:s}"'.format(git_hash),
'-DGOOSEFEM_GIT_BRANCH="{0:s}"'.format(git_branch)]
build.c_opts['msvc'] += [
'/DGOOSEFEM_GIT_HASH="{0:s}"'.format(git_hash),
'/DGOOSEFEM_GIT_BRANCH="{0:s}"'.format(git_branch)]
ext_modules = [Extension(
'GooseFEM',
['python/main.cpp'],
include_dirs = include_dirs,
language = 'c++')]
setup(
name = 'GooseFEM',
description = 'Finite element meshes, quadrature, and assembly tools',
long_description = 'Finite element meshes, quadrature, and assembly tools',
version = __version__,
license = 'GPLv3',
author = 'Tom de Geus',
author_email = 'tom@geus.me',
url = 'https://github.com/tdegeus/GooseFEM',
ext_modules = ext_modules,
install_requires = ['pybind11', 'pyxtensor'],
cmdclass = {'build_ext': build},
zip_safe = False)
c4science · Help