def linear_projection(pb, data_qp):
svar = pb.create_variables(['svar'])['svar']
aux = []
for ii in range(data_qp.shape[2]):
make_l2_projection_data(svar, data_qp[..., ii, :].copy())
aux.append(svar())
return nm.ascontiguousarray(nm.array(aux).T)
def test_projection_iga_fem(self):
from sfepy.discrete import FieldVariable
from sfepy.discrete.fem import FEDomain, Field
from sfepy.discrete.iga.domain import IGDomain
from sfepy.mesh.mesh_generators import gen_block_mesh
from sfepy.discrete.iga.domain_generators import gen_patch_block_domain
from sfepy.discrete.projections import (make_l2_projection,
make_l2_projection_data)
shape = [10, 12, 12]
dims = [5, 6, 6]
centre = [0, 0, 0]
degrees = [2, 2, 2]
nurbs, bmesh, regions = gen_patch_block_domain(dims, shape, centre,
degrees,
cp_mode='greville',
name='iga')
ig_domain = IGDomain('iga', nurbs, bmesh, regions=regions)
ig_omega = ig_domain.create_region('Omega', 'all')
ig_field = Field.from_args('iga', nm.float64, 1, ig_omega,
approx_order='iga', poly_space_base='iga')
ig_u = FieldVariable('ig_u', 'parameter', ig_field,
primary_var_name='(set-to-None)')
mesh = gen_block_mesh(dims, shape, centre, name='fem')
fe_domain = FEDomain('fem', mesh)
fe_omega = fe_domain.create_region('Omega', 'all')
fe_field = Field.from_args('fem', nm.float64, 1, fe_omega,
approx_order=2)
fe_u = FieldVariable('fe_u', 'parameter', fe_field,
primary_var_name='(set-to-None)')
def _eval_data(ts, coors, mode, **kwargs):
return nm.prod(coors**2, axis=1)[:, None, None]
make_l2_projection_data(ig_u, _eval_data)
make_l2_projection(fe_u, ig_u) # This calls ig_u.evaluate_at().
coors = 0.5 * nm.random.rand(20, 3) * dims
ig_vals = ig_u.evaluate_at(coors)
fe_vals = fe_u.evaluate_at(coors)
ok = nm.allclose(ig_vals, fe_vals, rtol=0.0, atol=1e-12)
if not ok:
self.report('iga-fem projection failed!')
self.report('coors:')
self.report(coors)
self.report('iga fem diff:')
self.report(nm.c_[ig_vals, fe_vals, nm.abs(ig_vals - fe_vals)])
return ok
def test_projection_iga_fem(self):
from sfepy.discrete import FieldVariable
from sfepy.discrete.fem import FEDomain, Field
from sfepy.discrete.iga.domain import IGDomain
from sfepy.mesh.mesh_generators import gen_block_mesh
from sfepy.discrete.iga.domain_generators import gen_patch_block_domain
from sfepy.discrete.projections import (make_l2_projection,
make_l2_projection_data)
shape = [10, 12, 12]
dims = [5, 6, 6]
centre = [0, 0, 0]
degrees = [2, 2, 2]
nurbs, bmesh, regions = gen_patch_block_domain(dims, shape, centre,
degrees,
cp_mode='greville',
name='iga')
ig_domain = IGDomain('iga', nurbs, bmesh, regions=regions)
ig_omega = ig_domain.create_region('Omega', 'all')
ig_field = Field.from_args('iga', nm.float64, 1, ig_omega,
approx_order='iga', poly_space_base='iga')
ig_u = FieldVariable('ig_u', 'parameter', ig_field,
primary_var_name='(set-to-None)')
mesh = gen_block_mesh(dims, shape, centre, name='fem')
fe_domain = FEDomain('fem', mesh)
fe_omega = fe_domain.create_region('Omega', 'all')
fe_field = Field.from_args('fem', nm.float64, 1, fe_omega,
approx_order=2)
fe_u = FieldVariable('fe_u', 'parameter', fe_field,
primary_var_name='(set-to-None)')
def _eval_data(ts, coors, mode, **kwargs):
return nm.prod(coors**2, axis=1)[:, None, None]
make_l2_projection_data(ig_u, _eval_data)
make_l2_projection(fe_u, ig_u) # This calls ig_u.evaluate_at().
coors = 0.5 * nm.random.rand(20, 3) * dims
ig_vals = ig_u.evaluate_at(coors)
fe_vals = fe_u.evaluate_at(coors)
ok = nm.allclose(ig_vals, fe_vals, rtol=0.0, atol=1e-12)
if not ok:
self.report('iga-fem projection failed!')
self.report('coors:')
self.report(coors)
self.report('iga fem diff:')
self.report(nm.c_[ig_vals, fe_vals, nm.abs(ig_vals - fe_vals)])
return ok
def project_by_component(tensor, tensor_qp, component, order):
"""
Wrapper around make_l2_projection_data() for non-scalar fields.
"""
aux = []
for ic in range(3):
make_l2_projection_data(component, tensor_qp[..., ic, :].copy(),
order=order)
aux.append(component())
tensor.set_data(nm.array(aux).T.ravel())
def test_continuity(self):
from sfepy.base.base import Struct
from sfepy.mesh.mesh_generators import gen_block_mesh
from sfepy.discrete import FieldVariable
from sfepy.discrete.fem import FEDomain, Field
from sfepy.discrete.projections import make_l2_projection_data
import sfepy.discrete.fem.refine_hanging as rh
dims = [1.5, 2.0, 1.3]
shape = [3, 3, 3]
centre = [0.0, 0.0, 0.0]
probe_gens = {'2_4': _gen_lines_2_4, '3_8': _gen_grid_3_8}
ok = True
for key in self.gel_names:
gel = self.gels[key]
probe_gen = probe_gens[key]
perms = gel.get_conn_permutations()
dim = gel.dim
for io, order in enumerate(range(1, 4)):
mesh00 = gen_block_mesh(dims[:dim],
shape[:dim],
centre[:dim],
name='block')
for ip, perm in enumerate(perms):
self.report(
'geometry: %s, order: %d, permutation: %d: %s' %
(key, order, ip, perm))
mesh0 = mesh00.copy()
conn = mesh0.cmesh.get_conn(dim, 0).indices
conn = conn.reshape((mesh0.n_el, -1))
conn[-1, :] = conn[-1, perm]
domain0 = FEDomain('d', mesh0)
refine = nm.zeros(mesh0.n_el, dtype=nm.uint8)
refine[:-1] = 1
subs = None
domain, subs = rh.refine(domain0, refine, subs=subs)
omega = domain.create_region('Omega', 'all')
field = Field.from_args('fu',
nm.float64,
1,
omega,
approx_order=order)
field.substitute_dofs(subs)
uvar = FieldVariable('u',
'parameter',
field,
primary_var_name='(set-to-None)')
field.restore_dofs(store=True)
field.substitute_dofs(subs=None, restore=True)
make_l2_projection_data(uvar, eval_fun)
field.restore_dofs()
bbox = domain.get_mesh_bounding_box()
eps = 1e-7
save = False
for ii, (probe0, probe1) in enumerate(probe_gen(bbox,
eps)):
probe0.set_options(close_limit=0.0)
probe1.set_options(close_limit=0.0)
pars0, vals0 = probe0(uvar)
pars1, vals1 = probe1(uvar)
assert_(nm.allclose(pars0, pars1, atol=1e-14,
rtol=0.0))
_ok = nm.allclose(vals0,
vals1,
atol=20.0 * eps,
rtol=0.0)
if not _ok:
save = True
self.report('probe %d failed! (max. error: %e)' %
(ii, nm.abs(vals0 - vals1).max()))
ok = ok and _ok
if (ip == 0) or save:
out = uvar.create_output()
filenames = _build_filenames(self.options.out_dir, key,
order, ip)
domain.mesh.write(filenames[0], out=out)
linearization = Struct(kind='adaptive',
min_level=0,
max_level=4,
eps=1e-2)
out = uvar.create_output(linearization=linearization)
val = out['u']
mesh = val.get('mesh', domain.mesh)
mesh.write(filenames[1], out=out)
return ok
def test_continuity(self):
from sfepy.base.base import Struct
from sfepy.mesh.mesh_generators import gen_block_mesh
from sfepy.discrete import FieldVariable
from sfepy.discrete.fem import FEDomain, Field
from sfepy.discrete.projections import make_l2_projection_data
import sfepy.discrete.fem.refine_hanging as rh
dims = [1.5, 2.0, 1.3]
shape = [3, 3, 3]
centre = [0.0, 0.0, 0.0]
probe_gens = {"2_4": _gen_lines_2_4, "3_8": _gen_grid_3_8}
ok = True
for key in self.gel_names:
gel = self.gels[key]
probe_gen = probe_gens[key]
perms = gel.get_conn_permutations()
dim = gel.dim
for io, order in enumerate(range(1, 4)):
mesh00 = gen_block_mesh(dims[:dim], shape[:dim], centre[:dim], name="block")
for ip, perm in enumerate(perms):
self.report("geometry: %s, order: %d, permutation: %d: %s" % (key, order, ip, perm))
mesh0 = mesh00.copy()
conn = mesh0.cmesh.get_conn(dim, 0).indices
conn = conn.reshape((mesh0.n_el, -1))
conn[-1, :] = conn[-1, perm]
domain0 = FEDomain("d", mesh0)
refine = nm.zeros(mesh0.n_el, dtype=nm.uint8)
refine[:-1] = 1
subs = None
domain, subs = rh.refine(domain0, refine, subs=subs)
omega = domain.create_region("Omega", "all")
field = Field.from_args("fu", nm.float64, 1, omega, approx_order=order)
field.substitute_dofs(subs)
uvar = FieldVariable("u", "parameter", field, primary_var_name="(set-to-None)")
field.restore_dofs(store=True)
field.substitute_dofs(subs=None, restore=True)
make_l2_projection_data(uvar, eval_fun)
field.restore_dofs()
bbox = domain.get_mesh_bounding_box()
eps = 1e-7
save = False
for ii, (probe0, probe1) in enumerate(probe_gen(bbox, eps)):
probe0.set_options(close_limit=0.0)
probe1.set_options(close_limit=0.0)
pars0, vals0 = probe0(uvar)
pars1, vals1 = probe1(uvar)
assert_(nm.allclose(pars0, pars1, atol=1e-14, rtol=0.0))
_ok = nm.allclose(vals0, vals1, atol=20.0 * eps, rtol=0.0)
if not _ok:
save = True
self.report("probe %d failed! (max. error: %e)" % (ii, nm.abs(vals0 - vals1).max()))
ok = ok and _ok
if (ip == 0) or save:
out = uvar.create_output()
filenames = _build_filenames(self.options.out_dir, key, order, ip)
domain.mesh.write(filenames[0], out=out)
linearization = Struct(kind="adaptive", min_level=0, max_level=4, eps=1e-2)
out = uvar.create_output(linearization=linearization)
val = out["u"]
mesh = val.get("mesh", domain.mesh)
mesh.write(filenames[1], out=out)
return ok