def VectorFunctionSpace(mesh: cpp.mesh.Mesh, element: ElementMetaData, dim=None, restriction=None): """Create vector finite element (composition of scalar elements) function space.""" e = ElementMetaData(*element) ufl_element = ufl.VectorElement(e.family, mesh.ufl_cell(), e.degree, form_degree=e.form_degree, dim=dim) return FunctionSpace(mesh, ufl_element)
def TensorFunctionSpace(mesh: cpp.mesh.Mesh, element: ElementMetaData, shape=None, symmetry: bool = None, restriction=None): """Create tensor finite element (composition of scalar elements) function space.""" e = ElementMetaData(*element) ufl_element = ufl.TensorElement(e.family, mesh.ufl_cell(), e.degree, shape, symmetry) return FunctionSpace(mesh, ufl_element)
def __init__(self, mesh: cpp.mesh.Mesh, element: typing.Union[ufl.FiniteElementBase, ElementMetaData], cppV: typing.Optional[cpp.function.FunctionSpace] = None): """Create a finite element function space.""" # Create function space from a UFL element and existing cpp # FunctionSpace if cppV is not None: assert mesh is None ufl_domain = cppV.mesh.ufl_domain() super().__init__(ufl_domain, element) self._cpp_object = cppV return # Initialise the ufl.FunctionSpace if isinstance(element, ufl.FiniteElementBase): super().__init__(mesh.ufl_domain(), element) else: e = ElementMetaData(*element) ufl_element = ufl.FiniteElement(e.family, mesh.ufl_cell(), e.degree, form_degree=e.form_degree) super().__init__(mesh.ufl_domain(), ufl_element) # Compile dofmap and element and create DOLFIN objects ufc_element, ufc_dofmap_ptr = jit.ffc_jit( self.ufl_element(), form_compiler_parameters=None, mpi_comm=mesh.mpi_comm()) ffi = cffi.FFI() ufc_element = dofmap.make_ufc_finite_element( ffi.cast("uintptr_t", ufc_element)) cpp_element = cpp.fem.FiniteElement(ufc_element) ufc_dofmap = dofmap.make_ufc_dofmap( ffi.cast("uintptr_t", ufc_dofmap_ptr)) cpp_dofmap = cpp.fem.create_dofmap(ufc_dofmap, mesh) # Initialize the cpp.FunctionSpace self._cpp_object = cpp.function.FunctionSpace(mesh, cpp_element, cpp_dofmap)
def __init__(self, mesh: cpp.mesh.Mesh): """Create function that evaluates to the facet area/length on each facet. *Example of usage* .. code-block:: python mesh = UnitSquare(4,4) fa = FacetArea(mesh) """ # Initialize C++ part self._cpp_object = cpp.function.FacetArea(mesh) # Initialize UFL part # NB! This is defined as a piecewise constant function for # each cell, not for each facet! ufl_element = ufl.FiniteElement("Discontinuous Lagrange", mesh.ufl_cell(), 0) super().__init__(domain=mesh.ufl_domain(), element=ufl_element, name="FacetArea")