1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
# SPDX-FileCopyrightText: Copyright © DUNE Project contributors, see file LICENSE.md in module root
# SPDX-License-Identifier: LicenseRef-GPL-2.0-only-with-DUNE-exception
from dune.generator.generator import SimpleGenerator
from dune.common.hashit import hashIt
def _generateModule(dim):
# if code is changed here also change code in dune/python/geometry/referenceelements.hh
typeName = "Dune::Geo::ReferenceElement<Dune::Geo::ReferenceElementImplementation<double," + str(dim) + "> >"
includes = ["dune/python/geometry/referenceelements.hh"]
typeHash = "referenceelements_" + hashIt(typeName)
generator = SimpleGenerator("ReferenceElements", "Dune::Python")
m = generator.load(includes, typeName, typeHash)
return m
def module(dim):
# try to load default module added in _geometry.cc
# this needs the DUNE_ENABLE_PYTHONMODULE_PRECOMPILE to be enabled
# otherwise all modules will be generated and compiled
try:
import importlib
return importlib.import_module( "dune.geometry._geometry._defaultreferenceelements_"+str(dim) )
except ImportError:
return _generateModule(dim)
_duneReferenceElements = {}
def referenceElement(geometryType):
try:
geometryType = geometryType.type
except:
pass
try:
ref = _duneReferenceElements[geometryType]
except KeyError:
ref = module(geometryType.dim).general(geometryType)
_duneReferenceElements[geometryType] = ref
return ref
|
