def test_gradients(self):
from sfepy.fem.mappings import VolumeMapping
ok = True
order = 3
bads = []
bad_families = set()
for (geom, poly_space_base, qp_weights, mesh, ir, ic,
ap, ps, rrc, crc, vec,
edofs, fdofs) in _gen_common_data(order, self.gels, self.report):
conn = mesh.conns[0]
gel = self.gels[geom]
geo_ps = ap.interp.get_geom_poly_space('v')
rmapping = VolumeMapping(mesh.coors, conn[:1],
poly_space=geo_ps)
rori = ap.ori[:1] if ap.ori is not None else None
rvg = rmapping.get_mapping(rrc, qp_weights,
poly_space=ps, ori=rori)
rbfg = rvg.bfg
cmapping = VolumeMapping(mesh.coors, conn[1:],
poly_space=geo_ps)
cori = ap.ori[1:] if ap.ori is not None else None
cvg = cmapping.get_mapping(crc, qp_weights,
poly_space=ps, ori=cori)
cbfg = cvg.bfg
dofs = nm.r_[edofs, fdofs]
res = nm.zeros((2, dofs.shape[0]), dtype=nm.int32)
res[0, :] = dofs
for ii, ip in enumerate(dofs):
vec.fill(0.0)
vec[ip] = 1.0
evec = vec[ap.econn]
rvals = nm.dot(rbfg, evec[0])[0]
cvals = nm.dot(cbfg, evec[1])[0]
okx = nm.allclose(rvals[:, 0], cvals[:, 0],
atol=1e-14, rtol=0.0)
if gel.dim == 2:
oky = nm.allclose(rvals[:, 1], -cvals[:, 1],
atol=1e-14, rtol=0.0)
_ok = okx and oky
else:
oky = nm.allclose(rvals[:, 1], cvals[:, 1],
atol=1e-14, rtol=0.0)
okz = nm.allclose(rvals[:, 2], -cvals[:, 2],
atol=1e-14, rtol=0.0)
_ok = okx and oky and okz
res[1, ii] = _ok
if not _ok:
bads.append([geom, poly_space_base, ir, ic, ip])
bad_families.add((geom, poly_space_base))
ok = ok and _ok
self.report('results (dofs, status: 1 ok, 0 failure):\n%s' % res)
if not ok:
self.report('gradient continuity errors:\n', bads)
self.report('%d in total!' % len(bads))
self.report('gradient continuity errors occurred in these'
' spaces:\n', bad_families)
return ok
def describe_geometry(self,
field,
gtype,
region,
integral=None,
return_mapping=False):
"""
Compute jacobians, element volumes and base function derivatives
for Volume-type geometries (volume mappings), and jacobians,
normals and base function derivatives for Surface-type
geometries (surface mappings).
Notes
-----
- volume mappings can be defined on a part of an element group,
although the field has to be defined always on the whole group.
- surface mappings are defined on the surface region
- surface mappings require field order to be > 0
"""
domain = field.domain
group = domain.groups[self.ig]
coors = domain.get_mesh_coors(actual=True)
if gtype == 'volume':
if integral is None:
from sfepy.fem import Integral
integral = Integral('i_tmp', 'v', 1)
qp = self.get_qp('v', integral)
iels = region.cells[self.ig]
geo_ps = self.interp.get_geom_poly_space('v')
ps = self.interp.poly_spaces['v']
bf = self.get_base('v', 0, integral, iels=iels)
conn = nm.take(group.conn, iels, axis=0)
mapping = VolumeMapping(coors, conn, poly_space=geo_ps)
vg = mapping.get_mapping(qp.vals,
qp.weights,
poly_space=ps,
ori=self.ori)
out = vg
elif (gtype == 'surface') or (gtype == 'surface_extra'):
assert_(field.approx_order > 0)
if self.ori is not None:
msg = 'surface integrals do not work yet with the' \
' hierarchical basis!'
raise ValueError(msg)
sd = domain.surface_groups[self.ig][region.name]
esd = self.surface_data[region.name]
qp = self.get_qp(sd.face_type, integral)
geo_ps = self.interp.get_geom_poly_space(sd.face_type)
ps = self.interp.poly_spaces[esd.face_type]
bf = self.get_base(esd.face_type, 0, integral)
conn = sd.get_connectivity()
mapping = SurfaceMapping(coors, conn, poly_space=geo_ps)
sg = mapping.get_mapping(qp.vals, qp.weights, poly_space=ps)
if gtype == 'surface_extra':
sg.alloc_extra_data(self.get_v_data_shape()[2])
self.create_bqp(region.name, integral)
qp = self.qp_coors[(integral.name, esd.bkey)]
v_geo_ps = self.interp.get_geom_poly_space('v')
bf_bg = v_geo_ps.eval_base(qp.vals, diff=True)
ebf_bg = self.get_base(esd.bkey, 1, integral)
sg.evaluate_bfbgm(bf_bg, ebf_bg, coors, sd.fis, group.conn)
out = sg
elif gtype == 'point':
out = mapping = None
else:
raise ValueError('unknown geometry type: %s' % gtype)
if out is not None:
# Store the integral used.
out.integral = integral
out.qp = qp
out.ps = ps
# Update base.
out.bf[:] = bf
if return_mapping:
out = (out, mapping)
return out
def describe_geometry(self, field, gtype, region, integral=None,
return_mapping=False):
"""
Compute jacobians, element volumes and base function derivatives
for Volume-type geometries (volume mappings), and jacobians,
normals and base function derivatives for Surface-type
geometries (surface mappings).
Notes
-----
- volume mappings can be defined on a part of an element group,
although the field has to be defined always on the whole group.
- surface mappings are defined on the surface region
- surface mappings require field order to be > 0
"""
domain = field.domain
group = domain.groups[self.ig]
coors = domain.get_mesh_coors(actual=True)
if gtype == 'volume':
if integral is None:
from sfepy.fem import Integral
integral = Integral('i_tmp', 'v', 1)
qp = self.get_qp('v', integral)
iels = region.cells[self.ig]
geo_ps = self.interp.get_geom_poly_space('v')
ps = self.interp.poly_spaces['v']
bf = self.get_base('v', 0, integral, iels=iels)
conn = nm.take(group.conn, iels, axis=0)
mapping = VolumeMapping(coors, conn, poly_space=geo_ps)
vg = mapping.get_mapping(qp.vals, qp.weights, poly_space=ps,
ori=self.ori)
out = vg
elif (gtype == 'surface') or (gtype == 'surface_extra'):
assert_(field.approx_order > 0)
if self.ori is not None:
msg = 'surface integrals do not work yet with the' \
' hierarchical basis!'
raise ValueError(msg)
sd = domain.surface_groups[self.ig][region.name]
esd = self.surface_data[region.name]
qp = self.get_qp(sd.face_type, integral)
geo_ps = self.interp.get_geom_poly_space(sd.face_type)
ps = self.interp.poly_spaces[esd.face_type]
bf = self.get_base(esd.face_type, 0, integral)
conn = sd.get_connectivity()
mapping = SurfaceMapping(coors, conn, poly_space=geo_ps)
sg = mapping.get_mapping(qp.vals, qp.weights, poly_space=ps)
if gtype == 'surface_extra':
sg.alloc_extra_data(self.get_v_data_shape()[2])
self.create_bqp(region.name, integral)
qp = self.qp_coors[(integral.name, esd.bkey)]
v_geo_ps = self.interp.get_geom_poly_space('v')
bf_bg = v_geo_ps.eval_base(qp.vals, diff=True)
ebf_bg = self.get_base(esd.bkey, 1, integral)
sg.evaluate_bfbgm(bf_bg, ebf_bg, coors, sd.fis, group.conn)
out = sg
elif gtype == 'point':
out = mapping = None
else:
raise ValueError('unknown geometry type: %s' % gtype)
if out is not None:
# Store the integral used.
out.integral = integral
out.qp = qp
out.ps = ps
# Update base.
out.bf[:] = bf
if return_mapping:
out = (out, mapping)
return out
def test_gradients(self):
from sfepy.fem.mappings import VolumeMapping
ok = True
orders = {'2_3' : 3, '2_4' : 3, '3_4' : 4, '3_8' : 3}
bads = []
bad_families = set()
for (geom, poly_space_base, qp_weights, mesh, ir, ic,
ap, ps, rrc, rcell, crc, ccell, vec,
edofs, fdofs) in _gen_common_data(orders, self.gels, self.report):
conn = mesh.conns[0]
gel = self.gels[geom]
geo_ps = ap.interp.get_geom_poly_space('v')
rmapping = VolumeMapping(mesh.coors, conn[rcell:rcell+1],
poly_space=geo_ps)
rori = ap.ori[:1] if ap.ori is not None else None
rvg = rmapping.get_mapping(rrc, qp_weights,
poly_space=ps, ori=rori)
rbfg = rvg.bfg
cmapping = VolumeMapping(mesh.coors, conn[ccell:ccell+1],
poly_space=geo_ps)
cori = ap.ori[1:] if ap.ori is not None else None
cvg = cmapping.get_mapping(crc, qp_weights,
poly_space=ps, ori=cori)
cbfg = cvg.bfg
dofs = nm.r_[edofs, fdofs]
res = nm.zeros((2, dofs.shape[0]), dtype=nm.int32)
res[0, :] = dofs
for ii, ip in enumerate(dofs):
vec.fill(0.0)
vec[ip] = 1.0
evec = vec[ap.econn]
rvals = nm.dot(rbfg, evec[rcell])[0]
cvals = nm.dot(cbfg, evec[ccell])[0]
okx = nm.allclose(rvals[:, 0], cvals[:, 0],
atol=1e-12, rtol=0.0)
if gel.dim == 2:
oky = nm.allclose(rvals[:, 1], -cvals[:, 1],
atol=1e-12, rtol=0.0)
_ok = okx and oky
else:
oky = nm.allclose(rvals[:, 1], cvals[:, 1],
atol=1e-12, rtol=0.0)
okz = nm.allclose(rvals[:, 2], -cvals[:, 2],
atol=1e-12, rtol=0.0)
_ok = okx and oky and okz
res[1, ii] = _ok
if not _ok:
bads.append([geom, poly_space_base, ir, ic, ip])
bad_families.add((geom, poly_space_base))
ok = ok and _ok
self.report('results (dofs, status: 1 ok, 0 failure):\n%s' % res)
if not ok:
self.report('gradient continuity errors:\n', bads)
self.report('%d in total!' % len(bads))
self.report('gradient continuity errors occurred in these'
' spaces:\n', bad_families)
return ok
def describe_geometry(self, field, gtype, region, integral=None, coors=None):
"""Compute jacobians, element volumes and base function derivatives
for Volume-type geometries, and jacobians, normals and base function
derivatives for Surface-type geometries.
"""
if coors is None:
coors = field.aps.coors
if gtype == 'volume':
if integral is None:
from sfepy.fem import Integral
dim = field.domain.shape.dim
quad_name = 'gauss_o1_d%d' % dim
integral = Integral('i_tmp', 'v', quad_name)
qp = self.get_qp('v', integral)
mapping = VolumeMapping(coors, self.econn,
poly_space=self.interp.poly_spaces['v'])
try:
vg = mapping.get_mapping(qp.vals, qp.weights)
except ValueError: # Constant bubble.
domain = self.region.domain
group = domain.groups[self.ig]
coors = domain.get_mesh_coors()
ps = self.interp.gel.interp.poly_spaces['v']
mapping = VolumeMapping(coors, group.conn, poly_space=ps)
vg = mapping.get_mapping(qp.vals, qp.weights)
out = vg
elif (gtype == 'surface') or (gtype == 'surface_extra'):
sd = self.surface_data[region.name]
qp = self.get_qp(sd.face_type, integral)
if not self.is_surface:
ps = self.interp.poly_spaces[sd.face_type]
else:
ps = self.interp.poly_spaces['v']
mapping = SurfaceMapping(coors, sd.econn, poly_space=ps)
sg = mapping.get_mapping(qp.vals, qp.weights)
if gtype == 'surface_extra':
sg.alloc_extra_data( self.get_v_data_shape()[2] )
self.create_bqp( region.name, integral )
bf_bg = self.get_base(sd.bkey, 1, integral)
sg.evaluate_bfbgm( bf_bg, coors, sd.fis, self.econn )
out = sg
elif gtype == 'point':
out = None
else:
raise ValueError('unknown geometry type: %s' % gtype)
if out is not None:
# Store the integral used.
out.integral = integral
return out
