dune-localfunctions  2.4.1
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 <cstddef>
8 #include <vector>
9 
10 #include <dune/common/fmatrix.hh>
11 #include <dune/common/fvector.hh>
12 
13 namespace Dune {
14 
16 
22  template<class Backend, std::size_t dimR>
23  class PowerBasis {
24  static_assert(Backend::Traits::dimRange == 1,
25  "PowerBasis works only with scalar backends");
26 
27  // don't use a reference here so this class stays copyable
28  const Backend *backend;
29 
30  public:
32  struct Traits : public Backend::Traits
33  {
35  static const std::size_t dimRange = dimR;
37  typedef FieldVector<typename Traits::RangeField, dimR> Range;
38 
40 
44  typedef FieldMatrix<typename Traits::RangeField, dimR,
45  Traits::dimDomainGlobal> Jacobian;
46 
48  static const std::size_t diffOrder =
49  Backend::Traits::diffOrder > 0 ? 1 : 0;
50  };
51 
53 
59  PowerBasis(const Backend &backend_) : backend(&backend_) { }
60 
62  std::size_t size () const { return backend->size()*dimR; }
64  std::size_t order () const { return backend->order(); }
65 
67  void evaluateFunction(const typename Traits::DomainLocal& in,
68  std::vector<typename Traits::Range>& out) const
69  {
70  std::vector<typename Backend::Traits::Range> backendValues;
71  backend->evaluateFunction(in, backendValues);
72  out.assign(size(), typename Traits::Range(0));
73  for(std::size_t d = 0; d < dimR; ++d)
74  for(std::size_t i = 0; i < backend->size(); ++i)
75  out[d*backend->size()+i][d] = backendValues[i][0];
76  }
77 
79  void evaluateJacobian(const typename Traits::DomainLocal& in,
80  std::vector<typename Traits::Jacobian>& out) const
81  {
82  std::vector<typename Backend::Traits::Jacobian> backendValues;
83  backend->evaluateJacobian(in, backendValues);
84  out.assign(size(), typename Traits::Jacobian(0));
85  for(std::size_t d = 0; d < dimR; ++d)
86  for(std::size_t i = 0; i < backend->size(); ++i)
87  out[d*backend->size()+i][d] = backendValues[i][0];
88  }
89  };
90 
91 } // namespace Dune
92 
93 #endif // DUNE_LOCALFUNCTIONS_META_POWER_BASIS_HH
Definition: brezzidouglasmarini1cube2dlocalbasis.hh:14
Meta-basis turning a scalar basis into vector-valued basis.
Definition: meta/power/basis.hh:23