def build_extension(self, ext):
extdir = os.path.abspath(
os.path.dirname(self.get_ext_fullpath(ext.name)))
cmake_args = [
'-DCMAKE_LIBRARY_OUTPUT_DIRECTORY=' + extdir + '\\' + ext.name,
'-DPYTHON_EXECUTABLE=' + sys.executable
]
cfg = 'Debug' if self.debug else 'Release'
build_args = ['--config', cfg]
if platform.system() == "Windows":
cmake_args += [
'-DCMAKE_LIBRARY_OUTPUT_DIRECTORY_{}={}'.format(
cfg.upper(), extdir)
]
if sys.maxsize > 2**32:
cmake_args += ['-A', 'x64']
build_args += ['--', '/m']
else:
cmake_args += ['-DCMAKE_BUILD_TYPE=' + cfg]
build_args += ['--', '-j2']
import petsc4py
cmake_args += ['-DPETSC_DIR=' + petsc4py.get_config()['PETSC_DIR']]
cmake_args += ['-DPETSC_ARCH=' + petsc4py.get_config()['PETSC_ARCH']]
env = os.environ.copy()
env['CXXFLAGS'] = '{} -DVERSION_INFO=\\"{}\\"'.format(
env.get('CXXFLAGS', ''), self.distribution.get_version())
if not os.path.exists(self.build_temp):
os.makedirs(self.build_temp)
subprocess.check_call(['cmake', ext.sourcedir] + cmake_args,
cwd=self.build_temp,
env=env)
subprocess.check_call(['cmake', '--build', '.'] + build_args,
cwd=self.build_temp)
def get_petsc_variables():
"""Get dict of PETSc environment variables from the file:
$PETSC_DIR/$PETSC_ARCH/lib/petsc/conf/petscvariables
The result is memoized to avoid constantly reading the file.
"""
config = petsc4py.get_config()
path = [config["PETSC_DIR"], config["PETSC_ARCH"], "lib/petsc/conf/petscvariables"]
variables_path = os.path.join(*path)
with open(variables_path) as fh:
# Split lines on first '=' (assignment)
splitlines = (line.split("=", maxsplit=1) for line in fh.readlines())
return {k.strip(): v.strip() for k, v in splitlines}
def configure():
INCLUDE_DIRS = []
LIBRARY_DIRS = []
LIBRARIES = []
# PETSc
import os
try:
PETSC_DIR = os.environ['PETSC_DIR']
PETSC_ARCH = os.environ.get('PETSC_ARCH', '')
except KeyError as e:
p4pyconfig = petsc4py.get_config()
PETSC_DIR = p4pyconfig['PETSC_DIR']
PETSC_ARCH = p4pyconfig['PETSC_ARCH']
from os.path import join, isdir
if PETSC_ARCH and isdir(join(PETSC_DIR, PETSC_ARCH)):
INCLUDE_DIRS += [
join(PETSC_DIR, PETSC_ARCH, 'include'),
join(PETSC_DIR, 'include')
]
LIBRARY_DIRS += [join(PETSC_DIR, PETSC_ARCH, 'lib')]
else:
if PETSC_ARCH: pass # XXX should warn ...
INCLUDE_DIRS += [join(PETSC_DIR, 'include')]
LIBRARY_DIRS += [join(PETSC_DIR, 'lib')]
LIBRARIES += ['petsc']
# PETSc for Python
INCLUDE_DIRS = [petsc4py.get_include()] + INCLUDE_DIRS
# NumPy
INCLUDE_DIRS = [numpy.get_include()] + INCLUDE_DIRS
return dict(
include_dirs=INCLUDE_DIRS + [os.curdir],
libraries=LIBRARIES,
library_dirs=LIBRARY_DIRS,
runtime_library_dirs=LIBRARY_DIRS,
)
self.plex = PETSc.DMPlex().createBoxMesh([1, 1, 1], simplex=True)
class TestPlex_3D_Boundary(BaseTestPlex_3D, unittest.TestCase):
CELLS = None
COORDS = None
def setUp(self):
boundary = PETSc.DMPlex().create(self.COMM)
boundary.createCubeBoundary([0., 0., 0.], [1., 1., 1.], [1, 1, 1])
boundary.setDimension(self.DIM - 1)
self.plex = PETSc.DMPlex().generate(boundary)
# --------------------------------------------------------------------
PETSC_DIR = petsc4py.get_config()['PETSC_DIR']
def check_dtype(method):
def wrapper(self, *args, **kwargs):
if PETSc.ScalarType is PETSc.ComplexType:
return
else:
return method(self, *args, **kwargs)
return wrapper
def check_package(method):
def wrapper(self, *args, **kwargs):
if not PETSc.Sys.hasExternalPackage("hdf5"):