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
38
39
40
41
42
43
44
45
46
|
from ufl import Measure, Mesh, triangle
class MockMesh:
def __init__(self, ufl_id):
self._ufl_id = ufl_id
def ufl_id(self):
return self._ufl_id
def ufl_domain(self):
return Mesh(triangle, ufl_id=self.ufl_id(), cargo=self)
def ufl_measure(
self, integral_type="dx", subdomain_id="everywhere", metadata=None, subdomain_data=None
):
return Measure(
integral_type,
subdomain_id=subdomain_id,
metadata=metadata,
domain=self,
subdomain_data=subdomain_data,
)
class MockMeshFunction:
"""Mock class for the pydolfin compatibility hack for domain data with [] syntax."""
def __init__(self, ufl_id, mesh):
self._mesh = mesh
self._ufl_id = ufl_id
def ufl_id(self):
return self._ufl_id
def mesh(self):
return self._mesh
def ufl_measure(self, integral_type=None, subdomain_id="everywhere", metadata=None):
return Measure(
integral_type,
subdomain_id=subdomain_id,
metadata=metadata,
domain=self.mesh(),
subdomain_data=self,
)
|
