def __init__(self, function_space, count=None):
FormArgument.__init__(self)
counted_init(self, count, Coefficient)
if isinstance(function_space, FiniteElementBase):
# For legacy support for .ufl files using cells, we map
# the cell to The Default Mesh
element = function_space
domain = default_domain(element.cell())
function_space = FunctionSpace(domain, element)
elif not isinstance(function_space, AbstractFunctionSpace):
error("Expecting a FunctionSpace or FiniteElement.")
self._ufl_function_space = function_space
self._ufl_shape = function_space.ufl_element().value_shape()
self._repr = as_native_str("Coefficient(%s, %s)" % (
repr(self._ufl_function_space), repr(self._count)))
def __init__(self, function_space, count=None):
FormArgument.__init__(self)
counted_init(self, count, Coefficient)
if isinstance(function_space, FiniteElementBase):
# For legacy support for .ufl files using cells, we map
# the cell to The Default Mesh
element = function_space
domain = default_domain(element.cell())
function_space = FunctionSpace(domain, element)
elif not isinstance(function_space, AbstractFunctionSpace):
error("Expecting a FunctionSpace or FiniteElement.")
self._ufl_function_space = function_space
self._ufl_shape = function_space.ufl_element().value_shape()
self._repr = "Coefficient(%s, %s)" % (repr(
self._ufl_function_space), repr(self._count))
def test_construct_domains_from_cells():
for cell in all_cells:
D0 = Mesh(cell)
D1 = default_domain(cell)
D2 = as_domain(cell)
assert D0 is not D1
assert D0 is not D2
assert D1 is D2
if 0:
print()
for D in (D1, D2):
print(('id', id(D)))
print(('str', str(D)))
print(('repr', repr(D)))
print()
assert D0 != D1
assert D0 != D2
assert D1 == D2
def test_construct_domains_from_cells():
for cell in all_cells:
D0 = Mesh(cell)
D1 = default_domain(cell)
D2 = as_domain(cell)
assert D0 is not D1
assert D0 is not D2
assert D1 is D2
if 0:
print()
for D in (D1, D2):
print(('id', id(D)))
print(('str', str(D)))
print(('repr', repr(D)))
print()
assert D0 != D1
assert D0 != D2
assert D1 == D2
def __init__(self, function_space, number, part=None):
FormArgument.__init__(self)
if isinstance(function_space, FiniteElementBase):
# For legacy support for UFL files using cells, we map the cell to
# the default Mesh
element = function_space
domain = default_domain(element.cell())
function_space = FunctionSpace(domain, element)
elif not isinstance(function_space, AbstractFunctionSpace):
error("Expecting a FunctionSpace or FiniteElement.")
self._ufl_function_space = function_space
self._ufl_shape = function_space.ufl_element().value_shape()
if not isinstance(number, numbers.Integral):
error("Expecting an int for number, not %s" % (number, ))
if part is not None and not isinstance(part, numbers.Integral):
error("Expecting None or an int for part, not %s" % (part, ))
self._number = number
self._part = part
self._repr = "Argument(%s, %s, %s)" % (repr(
self._ufl_function_space), repr(self._number), repr(self._part))
def __init__(self, function_space, number, part=None):
FormArgument.__init__(self)
if isinstance(function_space, FiniteElementBase):
# For legacy support for .ufl files using cells, we map the cell to
# the default Mesh
element = function_space
domain = default_domain(element.cell())
function_space = FunctionSpace(domain, element)
elif not isinstance(function_space, AbstractFunctionSpace):
error("Expecting a FunctionSpace or FiniteElement.")
self._ufl_function_space = function_space
self._ufl_shape = function_space.ufl_element().value_shape()
if not isinstance(number, numbers.Integral):
error("Expecting an int for number, not %s" % (number,))
if part is not None and not isinstance(part, numbers.Integral):
error("Expecting None or an int for part, not %s" % (part,))
self._number = number
self._part = part
self._repr = as_native_str("Argument(%s, %s, %s)" % (
repr(self._ufl_function_space), repr(self._number), repr(self._part)))
def test_mixed_functionspace(self):
# Domains
domain_3d = default_domain(tetrahedron)
domain_2d = default_domain(triangle)
domain_1d = default_domain(interval)
# Finite elements
f_1d = FiniteElement("CG", interval, 1)
f_2d = FiniteElement("CG", triangle, 1)
f_3d = FiniteElement("CG", tetrahedron, 1)
# Function spaces
V_3d = FunctionSpace(domain_3d, f_3d)
V_2d = FunctionSpace(domain_2d, f_2d)
V_1d = FunctionSpace(domain_1d, f_1d)
# MixedFunctionSpace = V_3d x V_2d x V_1d
V = MixedFunctionSpace(V_3d, V_2d, V_1d)
# Check sub spaces
assert( V.num_sub_spaces() == 3 )
assert( V.ufl_sub_space(0) == V_3d )
assert( V.ufl_sub_space(1) == V_2d )
assert( V.ufl_sub_space(2) == V_1d )
# Arguments from MixedFunctionSpace
(u_3d, u_2d, u_1d) = TrialFunctions(V)
(v_3d, v_2d, v_1d) = TestFunctions(V)
# Measures
dx3 = Measure("dx", domain=V_3d)
dx2 = Measure("dx", domain=V_2d)
dx1 = Measure("dx", domain=V_1d)
# Mixed variational form
# LHS
a_11 = u_1d*v_1d*dx1
a_22 = u_2d*v_2d*dx2
a_33 = u_3d*v_3d*dx3
a_21 = u_2d*v_1d*dx1
a_12 = u_1d*v_2d*dx1
a_32 = u_3d*v_2d*dx2
a_23 = u_2d*v_3d*dx2
a_31 = u_3d*v_1d*dx1
a_13 = u_1d*v_3d*dx1
a = a_11 + a_22 + a_33 + a_21 + a_12 + a_32 + a_23 + a_31 + a_13
# RHS
f_1 = v_1d*dx1
f_2 = v_2d*dx2
f_3 = v_3d*dx3
f = f_1 + f_2 + f_3
# Check extract_block algorithm
# LHS
assert ( extract_blocks(a,0,0) == a_33 )
assert ( extract_blocks(a,0,1) == a_23 )
assert ( extract_blocks(a,0,2) == a_13 )
assert ( extract_blocks(a,1,0) == a_32 )
assert ( extract_blocks(a,1,1) == a_22 )
assert ( extract_blocks(a,1,2) == a_12 )
assert ( extract_blocks(a,2,0) == a_31 )
assert ( extract_blocks(a,2,1) == a_21 )
assert ( extract_blocks(a,2,2) == a_11 )
# RHS
assert ( extract_blocks(f,0) == f_3 )
assert ( extract_blocks(f,1) == f_2 )
assert ( extract_blocks(f,2) == f_1 )