DOLFIN-X
DOLFIN-X C++ interface
FunctionSpace.h
1 // Copyright (C) 2008-2019 Anders Logg and Garth N. Wells
2 //
3 // This file is part of DOLFINX (https://www.fenicsproject.org)
4 //
5 // SPDX-License-Identifier: LGPL-3.0-or-later
6 
7 #pragma once
8 
9 #include <Eigen/Dense>
10 #include <cstddef>
11 #include <functional>
12 #include <map>
13 #include <memory>
14 #include <vector>
15 
16 namespace dolfinx
17 {
18 
19 namespace mesh
20 {
21 class Mesh;
22 }
23 
24 namespace fem
25 {
26 class DofMap;
27 class FiniteElement;
28 
32 
34 {
35 public:
40  FunctionSpace(std::shared_ptr<const mesh::Mesh> mesh,
41  std::shared_ptr<const fem::FiniteElement> element,
42  std::shared_ptr<const fem::DofMap> dofmap);
43 
44  // Copy constructor (deleted)
45  FunctionSpace(const FunctionSpace& V) = delete;
46 
48  FunctionSpace(FunctionSpace&& V) = default;
49 
51  virtual ~FunctionSpace() = default;
52 
53  // Assignment operator (delete)
54  FunctionSpace& operator=(const FunctionSpace& V) = delete;
55 
58 
61  bool operator==(const FunctionSpace& V) const;
62 
65  bool operator!=(const FunctionSpace& V) const;
66 
70  std::shared_ptr<FunctionSpace> sub(const std::vector<int>& component) const;
71 
75  bool contains(const FunctionSpace& V) const;
76 
80  std::pair<std::shared_ptr<FunctionSpace>, std::vector<std::int32_t>>
81  collapse() const;
82 
86  std::vector<int> component() const;
87 
90  Eigen::Array<double, Eigen::Dynamic, 3, Eigen::RowMajor>
92 
98  Eigen::Array<double, Eigen::Dynamic, 3, Eigen::RowMajor>
100 
102  std::size_t id() const;
103 
105  std::shared_ptr<const mesh::Mesh> mesh() const;
106 
108  std::shared_ptr<const fem::FiniteElement> element() const;
109 
111  std::shared_ptr<const fem::DofMap> dofmap() const;
112 
113 private:
114  // The mesh
115  std::shared_ptr<const mesh::Mesh> _mesh;
116 
117  // The finite element
118  std::shared_ptr<const fem::FiniteElement> _element;
119 
120  // The dofmap
121  std::shared_ptr<const fem::DofMap> _dofmap;
122 
123  // The component w.r.t. to root space
124  std::vector<int> _component;
125 
126  // Unique identifier
127  std::size_t _id;
128 
129  // The identifier of root space
130  std::size_t _root_space_id;
131 
132  // Cache of subspaces
133  mutable std::map<std::vector<int>, std::weak_ptr<FunctionSpace>> _subspaces;
134 };
135 
145 std::array<std::vector<std::shared_ptr<const FunctionSpace>>, 2>
147  const std::vector<
148  std::vector<std::array<std::shared_ptr<const FunctionSpace>, 2>>>& V);
149 } // namespace fem
150 } // namespace dolfinx
This class represents a finite element function space defined by a mesh, a finite element,...
Definition: FunctionSpace.h:34
virtual ~FunctionSpace()=default
Destructor.
std::vector< int > component() const
Get the component with respect to the root superspace.
Definition: FunctionSpace.cpp:189
std::shared_ptr< FunctionSpace > sub(const std::vector< int > &component) const
Extract subspace for component.
Definition: FunctionSpace.cpp:134
Eigen::Array< double, Eigen::Dynamic, 3, Eigen::RowMajor > tabulate_dof_coordinates() const
Tabulate the physical coordinates of all dofs on this process.
Definition: FunctionSpace.cpp:192
bool contains(const FunctionSpace &V) const
Check whether V is subspace of this, or this itself.
Definition: FunctionSpace.cpp:229
std::pair< std::shared_ptr< FunctionSpace >, std::vector< std::int32_t > > collapse() const
Collapse a subspace and return a new function space and a map from new to old dofs.
Definition: FunctionSpace.cpp:171
FunctionSpace(std::shared_ptr< const mesh::Mesh > mesh, std::shared_ptr< const fem::FiniteElement > element, std::shared_ptr< const fem::DofMap > dofmap)
Create function space for given mesh, element and dofmap.
Definition: FunctionSpace.cpp:114
Eigen::Array< double, Eigen::Dynamic, 3, Eigen::RowMajor > tabulate_scalar_subspace_dof_coordinates() const
Tabulate the physical coordinates of all dofs of scalar subspace on this process. For a VectorFunctio...
Definition: FunctionSpace.cpp:204
std::shared_ptr< const fem::DofMap > dofmap() const
The dofmap.
Definition: FunctionSpace.cpp:224
std::shared_ptr< const fem::FiniteElement > element() const
The finite element.
Definition: FunctionSpace.cpp:219
bool operator!=(const FunctionSpace &V) const
Inequality operator.
Definition: FunctionSpace.cpp:128
FunctionSpace(FunctionSpace &&V)=default
Move constructor.
bool operator==(const FunctionSpace &V) const
Equality operator.
Definition: FunctionSpace.cpp:123
FunctionSpace & operator=(FunctionSpace &&V)=default
Move assignment operator.
std::shared_ptr< const mesh::Mesh > mesh() const
The mesh.
Definition: FunctionSpace.cpp:217
std::size_t id() const
Unique identifier.
Definition: FunctionSpace.cpp:215
std::array< std::vector< std::shared_ptr< const FunctionSpace > >, 2 > common_function_spaces(const std::vector< std::vector< std::array< std::shared_ptr< const FunctionSpace >, 2 >>> &V)
Extract FunctionSpaces for (0) rows blocks and (1) columns blocks from a rectangular array of (test,...
Definition: FunctionSpace.cpp:248