def get_eigenpair(self, i, r_vec=None, c_vec=None):
if isinstance(r_vec, BlockFunction):
r_vec_in = r_vec._cpp_object
else:
r_vec_in = r_vec
if isinstance(c_vec, BlockFunction):
c_vec_in = c_vec._cpp_object
else:
c_vec_in = c_vec
(lr, lc, r_vec_out,
c_vec_out) = BlockSLEPcEigenSolver.get_eigenpair(
self, i, r_vec_in, c_vec_in)
if isinstance(r_vec_out, cpp.function.BlockFunction):
r_vec_out = BlockFunction(r_vec.block_function_space(),
r_vec_out)
if isinstance(c_vec_out, cpp.function.BlockFunction):
c_vec_out = BlockFunction(c_vec.block_function_space(),
c_vec_out)
return (lr, lc, r_vec_out, c_vec_out)
def block_restrict(block_input, block_function_sub_space):
assert isinstance(
block_input,
(array, list, BlockDirichletBC, BlockForm1, BlockForm2, BlockFunction))
if isinstance(block_input, (array, list, BlockForm1, BlockForm2)):
if isinstance(block_input, (array, list)):
(block_form, _,
block_form_rank) = _block_form_preprocessing(block_input)
elif isinstance(block_input, BlockForm2):
block_form = block_input
block_form_rank = 2
elif isinstance(block_input, BlockForm1):
block_form = block_input
block_form_rank = 1
if block_form_rank == 2:
assert isinstance(block_function_sub_space, list)
assert len(block_function_sub_space) == 2
assert isinstance(block_function_sub_space[0], BlockFunctionSpace)
assert isinstance(block_function_sub_space[1], BlockFunctionSpace)
N_sub_space = block_function_sub_space[0].num_sub_spaces()
M_sub_space = block_function_sub_space[1].num_sub_spaces()
sub_block_form = zeros((N_sub_space, M_sub_space), dtype=object)
for I_sub_space in range(N_sub_space):
I_space = _sub_component_to_component(
block_function_sub_space[0], I_sub_space)
for J_sub_space in range(M_sub_space):
J_space = _sub_component_to_component(
block_function_sub_space[1], J_sub_space)
sub_block_form[I_sub_space,
J_sub_space] = block_form[I_space, J_space]
return BlockForm2(sub_block_form, block_function_sub_space)
elif block_form_rank == 1:
assert isinstance(block_function_sub_space,
(BlockFunctionSpace, list))
if isinstance(block_function_sub_space, BlockFunctionSpace):
block_function_sub_space = [block_function_sub_space]
assert len(block_function_sub_space) == 1
assert isinstance(block_function_sub_space[0], BlockFunctionSpace)
N_sub_space = block_function_sub_space[0].num_sub_spaces()
sub_block_form = zeros((N_sub_space, ), dtype=object)
for I_sub_space in range(N_sub_space):
I_space = _sub_component_to_component(
block_function_sub_space[0], I_sub_space)
sub_block_form[I_sub_space] = block_form[I_space]
return BlockForm1(sub_block_form, block_function_sub_space)
elif isinstance(block_input, BlockFunction):
assert isinstance(block_function_sub_space, (BlockFunctionSpace, list))
if isinstance(block_function_sub_space, list):
assert len(block_function_sub_space) == 1
block_function_sub_space = block_function_sub_space[0]
assert isinstance(block_function_sub_space, BlockFunctionSpace)
N_sub_space = block_function_sub_space.num_sub_spaces()
sub_functions = list()
for I_sub_space in range(N_sub_space):
I_space = _sub_component_to_component(block_function_sub_space,
I_sub_space)
sub_functions.append(block_input[I_space])
return BlockFunction(block_function_sub_space, sub_functions)
elif isinstance(block_input, BlockDirichletBC):
assert isinstance(block_function_sub_space, (BlockFunctionSpace, list))
if isinstance(block_function_sub_space, list):
assert len(block_function_sub_space) == 1
block_function_sub_space = block_function_sub_space[0]
assert isinstance(block_function_sub_space, BlockFunctionSpace)
N_sub_space = block_function_sub_space.num_sub_spaces()
sub_bcs = list()
for I_sub_space in range(N_sub_space):
I_space = _sub_component_to_component(block_function_sub_space,
I_sub_space)
sub_bcs.append(block_input[I_space])
return BlockDirichletBC(sub_bcs, block_function_sub_space)
else:
raise AssertionError("Invalid arguments to block_restrict")