def set_mesh_entity(self, mesh_entity):
self._mesh_entity = mesh_entity
vertices = mesh_entity.vertices()
self._ref_element = LagrangeLine(len(vertices) - 1, self._space_dim)
self._mapping = FEMapping(self._ref_element)
self._mapping.set_mesh_entity(mesh_entity)
class IsoparametricContinuousLagrange1d(MappedElement):
def __init__(self, dim, space_dim=1):
MappedElement.__init__(self)
self._dim = dim
self._space_dim = space_dim
self._ref_element = None
self._mesh_entity = None
self._cache = {}
def number_of_global_dofs_per_vertex(self):
return self._dim
def number_of_global_non_vertex_dofs(self):
return 0
def number_of_global_dofs(self):
return self.number_of_global_dofs_per_vertex() * self._mesh_entity.number_of_vertices() + \
self.number_of_global_non_vertex_dofs()
def set_mesh_entity(self, mesh_entity):
self._mesh_entity = mesh_entity
vertices = mesh_entity.vertices()
self._ref_element = LagrangeLine(len(vertices) - 1, self._space_dim)
self._mapping = FEMapping(self._ref_element)
self._mapping.set_mesh_entity(mesh_entity)
def set_mapping(self, mapping):
raise Exception("Isoparametric element sets mapping internally!")
def topological_dimension(self):
return 1
def value_dimension(self):
return self._dim
def _clear_cache(self):
self._cache = {}
def _mapping_is_valid(self, ref_point):
raise NotImplementedError("This kind of check is not implemented yet.")
def interpolate_function(self, function):
return self._ref_element.interpolate_function(function, self._mapping)
def function_value(self, dof_values, ref_point):
return self._ref_element.function_value(dof_values, ref_point)
def function_gradient(self, dof_values, ref_point):
jac_inv = self._mapping.inverse_jacobian(ref_point)
return self._ref_element.function_gradient(dof_values, ref_point, jac_inv)
def shape_function_values(self, ref_point):
return self._ref_element.basis_function_values(ref_point)
def shape_function_gradients(self, ref_point):
jac_inv = self._mapping.inverse_jacobian(ref_point)
return self._ref_element.basis_function_gradients(ref_point, jac_inv)
def boundary_normal(self, local_boundary_index, boundary_ref_point):
if self._space_dim > 1:
raise Exception("boundary_normal() for this element only defined for space_dim == 1")
if local_boundary_index == 0:
return sp.array([[-1]])
elif local_boundary_index == 1:
return sp.array([[1]])
def boundary_orientation(self, local_boundary_index):
return self._mesh_entity.get_sub_entity(local_boundary_index).orientation
def physical_coords(self, ref_point):
return self._mapping.map_point(ref_point)
def director(self, ref_point):
jac = self._mapping.jacobian(ref_point)
return jac/sp.linalg.norm(jac)
def global_vertex_indices(self):
return self._mesh_entity.global_vertex_indices()
def index(self):
return self._mesh_entity.index
def domain_indicator(self):
return self._mesh_entity.domain_indicator
def boundary_indicator(self):
return self._mesh_entity.boundary_indicator
