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 <cstddef>
10 #include <dolfinx/common/array2d.h>
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 
96  array2d<double> tabulate_dof_coordinates(bool transpose) const;
97 
99  std::size_t id() const;
100 
102  std::shared_ptr<const mesh::Mesh> mesh() const;
103 
105  std::shared_ptr<const fem::FiniteElement> element() const;
106 
108  std::shared_ptr<const fem::DofMap> dofmap() const;
109 
110 private:
111  // The mesh
112  std::shared_ptr<const mesh::Mesh> _mesh;
113 
114  // The finite element
115  std::shared_ptr<const fem::FiniteElement> _element;
116 
117  // The dofmap
118  std::shared_ptr<const fem::DofMap> _dofmap;
119 
120  // The component w.r.t. to root space
121  std::vector<int> _component;
122 
123  // Unique identifier
124  std::size_t _id;
125 
126  // The identifier of root space
127  std::size_t _root_space_id;
128 
129  // Cache of subspaces
130  mutable std::map<std::vector<int>, std::weak_ptr<FunctionSpace>> _subspaces;
131 };
132 
142 std::array<std::vector<std::shared_ptr<const FunctionSpace>>, 2>
144  const std::vector<
145  std::vector<std::array<std::shared_ptr<const FunctionSpace>, 2>>>& V);
146 } // namespace fem
147 } // namespace dolfinx
This class provides a dynamic 2-dimensional row-wise array data structure.
Definition: array2d.h:21
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:98
std::shared_ptr< FunctionSpace > sub(const std::vector< int > &component) const
Extract subspace for component.
Definition: FunctionSpace.cpp:43
bool contains(const FunctionSpace &V) const
Check whether V is subspace of this, or this itself.
Definition: FunctionSpace.cpp:207
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:80
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:23
array2d< double > tabulate_dof_coordinates(bool transpose) const
Definition: FunctionSpace.cpp:100
std::shared_ptr< const fem::DofMap > dofmap() const
The dofmap.
Definition: FunctionSpace.cpp:202
std::shared_ptr< const fem::FiniteElement > element() const
The finite element.
Definition: FunctionSpace.cpp:197
bool operator!=(const FunctionSpace &V) const
Inequality operator.
Definition: FunctionSpace.cpp:37
FunctionSpace(FunctionSpace &&V)=default
Move constructor.
bool operator==(const FunctionSpace &V) const
Equality operator.
Definition: FunctionSpace.cpp:32
FunctionSpace & operator=(FunctionSpace &&V)=default
Move assignment operator.
std::shared_ptr< const mesh::Mesh > mesh() const
The mesh.
Definition: FunctionSpace.cpp:195
std::size_t id() const
Unique identifier.
Definition: FunctionSpace.cpp:193
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:226