dune-localfunctions 3.0-git
meta/power/basis.hh
Go to the documentation of this file.
1// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
2// vi: set et ts=4 sw=2 sts=2:
3
4#ifndef DUNE_LOCALFUNCTIONS_META_POWER_BASIS_HH
5#define DUNE_LOCALFUNCTIONS_META_POWER_BASIS_HH
6
7#include <numeric>
8#include <cstddef>
9#include <vector>
10
11#include <dune/common/fmatrix.hh>
12#include <dune/common/fvector.hh>
13
14namespace Dune {
15
17
23 template<class Backend, std::size_t dimR>
24 class PowerBasis {
25 static_assert(Backend::Traits::dimRange == 1,
26 "PowerBasis works only with scalar backends");
27
28 // don't use a reference here so this class stays copyable
29 const Backend *backend;
30
31 public:
33 struct Traits : public Backend::Traits
34 {
36 static const std::size_t dimRange = dimR;
38 typedef FieldVector<typename Traits::RangeField, dimR> Range;
39
41
45 typedef FieldMatrix<typename Traits::RangeField, dimR,
46 Traits::dimDomainGlobal> Jacobian;
47
49 static const std::size_t diffOrder =
50 Backend::Traits::diffOrder > 0 ? 1 : 0;
51 };
52
54
60 PowerBasis(const Backend &backend_) : backend(&backend_) { }
61
63 std::size_t size () const { return backend->size()*dimR; }
65 std::size_t order () const { return backend->order(); }
66
68 void evaluateFunction(const typename Traits::DomainLocal& in,
69 std::vector<typename Traits::Range>& out) const
70 {
71 std::vector<typename Backend::Traits::Range> backendValues;
72 backend->evaluateFunction(in, backendValues);
73 out.assign(size(), typename Traits::Range(0));
74 for(std::size_t d = 0; d < dimR; ++d)
75 for(std::size_t i = 0; i < backend->size(); ++i)
76 out[d*backend->size()+i][d] = backendValues[i][0];
77 }
78
80 void evaluateJacobian(const typename Traits::DomainLocal& in,
81 std::vector<typename Traits::Jacobian>& out) const
82 {
83 std::vector<typename Backend::Traits::Jacobian> backendValues;
84 backend->evaluateJacobian(in, backendValues);
85 out.assign(size(), typename Traits::Jacobian(0));
86 for(std::size_t d = 0; d < dimR; ++d)
87 for(std::size_t i = 0; i < backend->size(); ++i)
88 out[d*backend->size()+i][d] = backendValues[i][0];
89 }
90
92 void partial (const std::array<unsigned int, Backend::Traits::dimDomainGlobal>& order,
93 const typename Traits::DomainLocal& in, // position
94 std::vector<typename Traits::Range>& out) const // return value
95 {
96 auto totalOrder = std::accumulate(order.begin(), order.end(), 0);
97 if (totalOrder == 0) {
98 evaluateFunction(in, out);
99 } else {
100 DUNE_THROW(NotImplemented, "Desired derivative order is not implemented");
101 }
102 }
103 };
104
105} // namespace Dune
106
107#endif // DUNE_LOCALFUNCTIONS_META_POWER_BASIS_HH
Definition brezzidouglasmarini1cube2d.hh:14
Meta-basis turning a scalar basis into vector-valued basis.
Definition meta/power/basis.hh:24
PowerBasis(const Backend &backend_)
Construct a PowerBasis.
Definition meta/power/basis.hh:60
std::size_t size() const
Number of shape functions.
Definition meta/power/basis.hh:63
void partial(const std::array< unsigned int, Backend::Traits::dimDomainGlobal > &order, const typename Traits::DomainLocal &in, std::vector< typename Traits::Range > &out) const
Evaluate partial derivatives of all shape functions.
Definition meta/power/basis.hh:92
void evaluateJacobian(const typename Traits::DomainLocal &in, std::vector< typename Traits::Jacobian > &out) const
Evaluate Jacobian of all shape functions at given position.
Definition meta/power/basis.hh:80
void evaluateFunction(const typename Traits::DomainLocal &in, std::vector< typename Traits::Range > &out) const
Evaluate all shape functions at given position.
Definition meta/power/basis.hh:68
std::size_t order() const
Polynomial order of the shape functions for quadrature.
Definition meta/power/basis.hh:65
types of domain and range
Definition meta/power/basis.hh:34
FieldMatrix< typename Traits::RangeField, dimR, Traits::dimDomainGlobal > Jacobian
Jacobian properties.
Definition meta/power/basis.hh:46
static const std::size_t diffOrder
maximum number of partial derivatives supported
Definition meta/power/basis.hh:49
FieldVector< typename Traits::RangeField, dimR > Range
Type used for range values.
Definition meta/power/basis.hh:38
static const std::size_t dimRange
Dimension of the range values.
Definition meta/power/basis.hh:36